diff --git a/.github/assets/screenshot-integrations.png b/.github/assets/screenshot-integrations.png
index 8d71bf538d6..abbc0f76ff0 100644
Binary files a/.github/assets/screenshot-integrations.png and b/.github/assets/screenshot-integrations.png differ
diff --git a/.github/copilot-instructions.md b/.github/copilot-instructions.md
new file mode 100644
index 00000000000..06499d62b9e
--- /dev/null
+++ b/.github/copilot-instructions.md
@@ -0,0 +1,100 @@
+# Instructions for GitHub Copilot
+
+This repository holds the core of Home Assistant, a Python 3 based home
+automation application.
+
+- Python code must be compatible with Python 3.13
+- Use the newest Python language features if possible:
+ - Pattern matching
+ - Type hints
+ - f-strings for string formatting over `%` or `.format()`
+ - Dataclasses
+ - Walrus operator
+- Code quality tools:
+ - Formatting: Ruff
+ - Linting: PyLint and Ruff
+ - Type checking: MyPy
+ - Testing: pytest with plain functions and fixtures
+- Inline code documentation:
+ - File headers should be short and concise:
+ ```python
+ """Integration for Peblar EV chargers."""
+ ```
+ - Every method and function needs a docstring:
+ ```python
+ async def async_setup_entry(hass: HomeAssistant, entry: PeblarConfigEntry) -> bool:
+ """Set up Peblar from a config entry."""
+ ...
+ ```
+- All code and comments and other text are written in American English
+- Follow existing code style patterns as much as possible
+- Core locations:
+ - Shared constants: `homeassistant/const.py`, use them instead of hardcoding
+ strings or creating duplicate integration constants.
+ - Integration files:
+ - Constants: `homeassistant/components/{domain}/const.py`
+ - Models: `homeassistant/components/{domain}/models.py`
+ - Coordinator: `homeassistant/components/{domain}/coordinator.py`
+ - Config flow: `homeassistant/components/{domain}/config_flow.py`
+ - Platform code: `homeassistant/components/{domain}/{platform}.py`
+- All external I/O operations must be async
+- Async patterns:
+ - Avoid sleeping in loops
+ - Avoid awaiting in loops, gather instead
+ - No blocking calls
+- Polling:
+ - Follow update coordinator pattern, when possible
+ - Polling interval may not be configurable by the user
+ - For local network polling, the minimum interval is 5 seconds
+ - For cloud polling, the minimum interval is 60 seconds
+- Error handling:
+ - Use specific exceptions from `homeassistant.exceptions`
+ - Setup failures:
+ - Temporary: Raise `ConfigEntryNotReady`
+ - Permanent: Use `ConfigEntryError`
+- Logging:
+ - Message format:
+ - No periods at end
+ - No integration names or domains (added automatically)
+ - No sensitive data (keys, tokens, passwords), even when those are incorrect.
+ - Be very restrictive on the use of logging info messages, use debug for
+ anything which is not targeting the user.
+ - Use lazy logging (no f-strings):
+ ```python
+ _LOGGER.debug("This is a log message with %s", variable)
+ ```
+- Entities:
+ - Ensure unique IDs for state persistence:
+ - Unique IDs should not contain values that are subject to user or network change.
+ - An ID needs to be unique per platform, not per integration.
+ - The ID does not have to contain the integration domain or platform.
+ - Acceptable examples:
+ - Serial number of a device
+ - MAC address of a device formatted using `homeassistant.helpers.device_registry.format_mac`
+ Do not obtain the MAC address through arp cache of local network access,
+ only use the MAC address provided by discovery or the device itself.
+ - Unique identifier that is physically printed on the device or burned into an EEPROM
+ - Not acceptable examples:
+ - IP Address
+ - Device name
+ - Hostname
+ - URL
+ - Email address
+ - Username
+ - For entities that are setup by a config entry, the config entry ID
+ can be used as a last resort if no other Unique ID is available.
+ For example: `f"{entry.entry_id}-battery"`
+ - If the state value is unknown, use `None`
+ - Do not use the `unavailable` string as a state value,
+ implement the `available()` property method instead
+ - Do not use the `unknown` string as a state value, use `None` instead
+- Extra entity state attributes:
+ - The keys of all state attributes should always be present
+ - If the value is unknown, use `None`
+ - Provide descriptive state attributes
+- Testing:
+ - Test location: `tests/components/{domain}/`
+ - Use pytest fixtures from `tests.common`
+ - Mock external dependencies
+ - Use snapshots for complex data
+ - Follow existing test patterns
diff --git a/.github/workflows/builder.yml b/.github/workflows/builder.yml
index cdffcbe4d5b..fcf707fef3d 100644
--- a/.github/workflows/builder.yml
+++ b/.github/workflows/builder.yml
@@ -69,7 +69,7 @@ jobs:
run: find ./homeassistant/components/*/translations -name "*.json" | tar zcvf translations.tar.gz -T -
- name: Upload translations
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: translations
path: translations.tar.gz
@@ -94,7 +94,7 @@ jobs:
- name: Download nightly wheels of frontend
if: needs.init.outputs.channel == 'dev'
- uses: dawidd6/action-download-artifact@v8
+ uses: dawidd6/action-download-artifact@v9
with:
github_token: ${{secrets.GITHUB_TOKEN}}
repo: home-assistant/frontend
@@ -105,7 +105,7 @@ jobs:
- name: Download nightly wheels of intents
if: needs.init.outputs.channel == 'dev'
- uses: dawidd6/action-download-artifact@v8
+ uses: dawidd6/action-download-artifact@v9
with:
github_token: ${{secrets.GITHUB_TOKEN}}
repo: home-assistant/intents-package
@@ -175,7 +175,7 @@ jobs:
sed -i "s|pykrakenapi|# pykrakenapi|g" requirements_all.txt
- name: Download translations
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: translations
@@ -190,14 +190,14 @@ jobs:
echo "${{ github.sha }};${{ github.ref }};${{ github.event_name }};${{ github.actor }}" > rootfs/OFFICIAL_IMAGE
- name: Login to GitHub Container Registry
- uses: docker/login-action@v3.3.0
+ uses: docker/login-action@v3.4.0
with:
registry: ghcr.io
username: ${{ github.repository_owner }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Build base image
- uses: home-assistant/builder@2024.08.2
+ uses: home-assistant/builder@2025.03.0
with:
args: |
$BUILD_ARGS \
@@ -256,14 +256,14 @@ jobs:
fi
- name: Login to GitHub Container Registry
- uses: docker/login-action@v3.3.0
+ uses: docker/login-action@v3.4.0
with:
registry: ghcr.io
username: ${{ github.repository_owner }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Build base image
- uses: home-assistant/builder@2024.08.2
+ uses: home-assistant/builder@2025.03.0
with:
args: |
$BUILD_ARGS \
@@ -324,20 +324,20 @@ jobs:
uses: actions/checkout@v4.2.2
- name: Install Cosign
- uses: sigstore/cosign-installer@v3.8.0
+ uses: sigstore/cosign-installer@v3.8.1
with:
cosign-release: "v2.2.3"
- name: Login to DockerHub
if: matrix.registry == 'docker.io/homeassistant'
- uses: docker/login-action@v3.3.0
+ uses: docker/login-action@v3.4.0
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_TOKEN }}
- name: Login to GitHub Container Registry
if: matrix.registry == 'ghcr.io/home-assistant'
- uses: docker/login-action@v3.3.0
+ uses: docker/login-action@v3.4.0
with:
registry: ghcr.io
username: ${{ github.repository_owner }}
@@ -448,6 +448,9 @@ jobs:
environment: ${{ needs.init.outputs.channel }}
needs: ["init", "build_base"]
runs-on: ubuntu-latest
+ permissions:
+ contents: read
+ id-token: write
if: github.repository_owner == 'home-assistant' && needs.init.outputs.publish == 'true'
steps:
- name: Checkout the repository
@@ -459,7 +462,7 @@ jobs:
python-version: ${{ env.DEFAULT_PYTHON }}
- name: Download translations
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: translations
@@ -473,16 +476,13 @@ jobs:
run: |
# Remove dist, build, and homeassistant.egg-info
# when build locally for testing!
- pip install twine build
+ pip install build
python -m build
- - name: Upload package
- shell: bash
- run: |
- export TWINE_USERNAME="__token__"
- export TWINE_PASSWORD="${{ secrets.TWINE_TOKEN }}"
-
- twine upload dist/* --skip-existing
+ - name: Upload package to PyPI
+ uses: pypa/gh-action-pypi-publish@v1.12.4
+ with:
+ skip-existing: true
hassfest-image:
name: Build and test hassfest image
@@ -502,14 +502,14 @@ jobs:
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
- name: Login to GitHub Container Registry
- uses: docker/login-action@9780b0c442fbb1117ed29e0efdff1e18412f7567 # v3.3.0
+ uses: docker/login-action@74a5d142397b4f367a81961eba4e8cd7edddf772 # v3.4.0
with:
registry: ghcr.io
username: ${{ github.repository_owner }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Build Docker image
- uses: docker/build-push-action@ca877d9245402d1537745e0e356eab47c3520991 # v6.13.0
+ uses: docker/build-push-action@471d1dc4e07e5cdedd4c2171150001c434f0b7a4 # v6.15.0
with:
context: . # So action will not pull the repository again
file: ./script/hassfest/docker/Dockerfile
@@ -522,7 +522,7 @@ jobs:
- name: Push Docker image
if: needs.init.outputs.channel != 'dev' && needs.init.outputs.publish == 'true'
id: push
- uses: docker/build-push-action@ca877d9245402d1537745e0e356eab47c3520991 # v6.13.0
+ uses: docker/build-push-action@471d1dc4e07e5cdedd4c2171150001c434f0b7a4 # v6.15.0
with:
context: . # So action will not pull the repository again
file: ./script/hassfest/docker/Dockerfile
@@ -531,7 +531,7 @@ jobs:
- name: Generate artifact attestation
if: needs.init.outputs.channel != 'dev' && needs.init.outputs.publish == 'true'
- uses: actions/attest-build-provenance@520d128f165991a6c774bcb264f323e3d70747f4 # v2.2.0
+ uses: actions/attest-build-provenance@c074443f1aee8d4aeeae555aebba3282517141b2 # v2.2.3
with:
subject-name: ${{ env.HASSFEST_IMAGE_NAME }}
subject-digest: ${{ steps.push.outputs.digest }}
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 2a9f1571830..2b1606568b5 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -37,10 +37,10 @@ on:
type: boolean
env:
- CACHE_VERSION: 11
+ CACHE_VERSION: 12
UV_CACHE_VERSION: 1
MYPY_CACHE_VERSION: 9
- HA_SHORT_VERSION: "2025.3"
+ HA_SHORT_VERSION: "2025.4"
DEFAULT_PYTHON: "3.13"
ALL_PYTHON_VERSIONS: "['3.13']"
# 10.3 is the oldest supported version
@@ -89,6 +89,7 @@ jobs:
test_groups: ${{ steps.info.outputs.test_groups }}
tests_glob: ${{ steps.info.outputs.tests_glob }}
tests: ${{ steps.info.outputs.tests }}
+ lint_only: ${{ steps.info.outputs.lint_only }}
skip_coverage: ${{ steps.info.outputs.skip_coverage }}
runs-on: ubuntu-24.04
steps:
@@ -142,6 +143,7 @@ jobs:
test_group_count=10
tests="[]"
tests_glob=""
+ lint_only=""
skip_coverage=""
if [[ "${{ steps.integrations.outputs.changes }}" != "[]" ]];
@@ -192,6 +194,17 @@ jobs:
test_full_suite="true"
fi
+ if [[ "${{ github.event.inputs.lint-only }}" == "true" ]] \
+ || [[ "${{ github.event.inputs.pylint-only }}" == "true" ]] \
+ || [[ "${{ github.event.inputs.mypy-only }}" == "true" ]] \
+ || [[ "${{ github.event.inputs.audit-licenses-only }}" == "true" ]] \
+ || [[ "${{ github.event_name }}" == "push" \
+ && "${{ github.event.repository.full_name }}" != "home-assistant/core" ]];
+ then
+ lint_only="true"
+ skip_coverage="true"
+ fi
+
if [[ "${{ github.event.inputs.skip-coverage }}" == "true" ]] \
|| [[ "${{ contains(github.event.pull_request.labels.*.name, 'ci-skip-coverage') }}" == "true" ]];
then
@@ -217,6 +230,8 @@ jobs:
echo "tests=${tests}" >> $GITHUB_OUTPUT
echo "tests_glob: ${tests_glob}"
echo "tests_glob=${tests_glob}" >> $GITHUB_OUTPUT
+ echo "lint_only": ${lint_only}
+ echo "lint_only=${lint_only}" >> $GITHUB_OUTPUT
echo "skip_coverage: ${skip_coverage}"
echo "skip_coverage=${skip_coverage}" >> $GITHUB_OUTPUT
@@ -240,7 +255,7 @@ jobs:
check-latest: true
- name: Restore base Python virtual environment
id: cache-venv
- uses: actions/cache@v4.2.0
+ uses: actions/cache@v4.2.3
with:
path: venv
key: >-
@@ -256,7 +271,7 @@ jobs:
uv pip install "$(cat requirements_test.txt | grep pre-commit)"
- name: Restore pre-commit environment from cache
id: cache-precommit
- uses: actions/cache@v4.2.0
+ uses: actions/cache@v4.2.3
with:
path: ${{ env.PRE_COMMIT_CACHE }}
lookup-only: true
@@ -286,7 +301,7 @@ jobs:
check-latest: true
- name: Restore base Python virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -295,7 +310,7 @@ jobs:
needs.info.outputs.pre-commit_cache_key }}
- name: Restore pre-commit environment from cache
id: cache-precommit
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: ${{ env.PRE_COMMIT_CACHE }}
fail-on-cache-miss: true
@@ -326,7 +341,7 @@ jobs:
check-latest: true
- name: Restore base Python virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -335,7 +350,7 @@ jobs:
needs.info.outputs.pre-commit_cache_key }}
- name: Restore pre-commit environment from cache
id: cache-precommit
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: ${{ env.PRE_COMMIT_CACHE }}
fail-on-cache-miss: true
@@ -366,7 +381,7 @@ jobs:
check-latest: true
- name: Restore base Python virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -375,7 +390,7 @@ jobs:
needs.info.outputs.pre-commit_cache_key }}
- name: Restore pre-commit environment from cache
id: cache-precommit
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: ${{ env.PRE_COMMIT_CACHE }}
fail-on-cache-miss: true
@@ -482,7 +497,7 @@ jobs:
env.HA_SHORT_VERSION }}-$(date -u '+%Y-%m-%dT%H:%M:%s')" >> $GITHUB_OUTPUT
- name: Restore base Python virtual environment
id: cache-venv
- uses: actions/cache@v4.2.0
+ uses: actions/cache@v4.2.3
with:
path: venv
key: >-
@@ -490,7 +505,7 @@ jobs:
needs.info.outputs.python_cache_key }}
- name: Restore uv wheel cache
if: steps.cache-venv.outputs.cache-hit != 'true'
- uses: actions/cache@v4.2.0
+ uses: actions/cache@v4.2.3
with:
path: ${{ env.UV_CACHE_DIR }}
key: >-
@@ -537,7 +552,7 @@ jobs:
python --version
uv pip freeze >> pip_freeze.txt
- name: Upload pip_freeze artifact
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: pip-freeze-${{ matrix.python-version }}
path: pip_freeze.txt
@@ -578,7 +593,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ env.DEFAULT_PYTHON }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -611,7 +626,7 @@ jobs:
check-latest: true
- name: Restore base Python virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -623,6 +638,25 @@ jobs:
. venv/bin/activate
python -m script.gen_requirements_all validate
+ dependency-review:
+ name: Dependency review
+ runs-on: ubuntu-24.04
+ needs:
+ - info
+ - base
+ if: |
+ github.event.inputs.pylint-only != 'true'
+ && github.event.inputs.mypy-only != 'true'
+ && needs.info.outputs.requirements == 'true'
+ && github.event_name == 'pull_request'
+ steps:
+ - name: Check out code from GitHub
+ uses: actions/checkout@v4.2.2
+ - name: Dependency review
+ uses: actions/dependency-review-action@v4.5.0
+ with:
+ license-check: false # We use our own license audit checks
+
audit-licenses:
name: Audit licenses
runs-on: ubuntu-24.04
@@ -649,7 +683,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ matrix.python-version }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -661,7 +695,7 @@ jobs:
. venv/bin/activate
python -m script.licenses extract --output-file=licenses-${{ matrix.python-version }}.json
- name: Upload licenses
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: licenses-${{ github.run_number }}-${{ matrix.python-version }}
path: licenses-${{ matrix.python-version }}.json
@@ -692,7 +726,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ env.DEFAULT_PYTHON }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -739,7 +773,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ env.DEFAULT_PYTHON }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -791,7 +825,7 @@ jobs:
env.HA_SHORT_VERSION }}-$(date -u '+%Y-%m-%dT%H:%M:%s')" >> $GITHUB_OUTPUT
- name: Restore full Python ${{ env.DEFAULT_PYTHON }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -799,7 +833,7 @@ jobs:
${{ runner.os }}-${{ steps.python.outputs.python-version }}-${{
needs.info.outputs.python_cache_key }}
- name: Restore mypy cache
- uses: actions/cache@v4.2.0
+ uses: actions/cache@v4.2.3
with:
path: .mypy_cache
key: >-
@@ -829,11 +863,7 @@ jobs:
prepare-pytest-full:
runs-on: ubuntu-24.04
if: |
- (github.event_name != 'push' || github.event.repository.full_name == 'home-assistant/core')
- && github.event.inputs.lint-only != 'true'
- && github.event.inputs.pylint-only != 'true'
- && github.event.inputs.mypy-only != 'true'
- && github.event.inputs.audit-licenses-only != 'true'
+ needs.info.outputs.lint_only != 'true'
&& needs.info.outputs.test_full_suite == 'true'
needs:
- info
@@ -865,7 +895,7 @@ jobs:
check-latest: true
- name: Restore base Python virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -877,7 +907,7 @@ jobs:
. venv/bin/activate
python -m script.split_tests ${{ needs.info.outputs.test_group_count }} tests
- name: Upload pytest_buckets
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: pytest_buckets
path: pytest_buckets.txt
@@ -886,11 +916,7 @@ jobs:
pytest-full:
runs-on: ubuntu-24.04
if: |
- (github.event_name != 'push' || github.event.repository.full_name == 'home-assistant/core')
- && github.event.inputs.lint-only != 'true'
- && github.event.inputs.pylint-only != 'true'
- && github.event.inputs.mypy-only != 'true'
- && github.event.inputs.audit-licenses-only != 'true'
+ needs.info.outputs.lint_only != 'true'
&& needs.info.outputs.test_full_suite == 'true'
needs:
- info
@@ -929,7 +955,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ matrix.python-version }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -942,7 +968,7 @@ jobs:
run: |
echo "::add-matcher::.github/workflows/matchers/pytest-slow.json"
- name: Download pytest_buckets
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: pytest_buckets
- name: Compile English translations
@@ -962,6 +988,7 @@ jobs:
if [[ "${{ needs.info.outputs.skip_coverage }}" != "true" ]]; then
cov_params+=(--cov="homeassistant")
cov_params+=(--cov-report=xml)
+ cov_params+=(--junitxml=junit.xml -o junit_family=legacy)
fi
echo "Test group ${{ matrix.group }}: $(sed -n "${{ matrix.group }},1p" pytest_buckets.txt)"
@@ -980,18 +1007,24 @@ jobs:
2>&1 | tee pytest-${{ matrix.python-version }}-${{ matrix.group }}.txt
- name: Upload pytest output
if: success() || failure() && steps.pytest-full.conclusion == 'failure'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: pytest-${{ github.run_number }}-${{ matrix.python-version }}-${{ matrix.group }}
path: pytest-*.txt
overwrite: true
- name: Upload coverage artifact
if: needs.info.outputs.skip_coverage != 'true'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: coverage-${{ matrix.python-version }}-${{ matrix.group }}
path: coverage.xml
overwrite: true
+ - name: Upload test results artifact
+ if: needs.info.outputs.skip_coverage != 'true' && !cancelled()
+ uses: actions/upload-artifact@v4.6.2
+ with:
+ name: test-results-full-${{ matrix.python-version }}-${{ matrix.group }}
+ path: junit.xml
- name: Remove pytest_buckets
run: rm pytest_buckets.txt
- name: Check dirty
@@ -1009,11 +1042,7 @@ jobs:
MYSQL_ROOT_PASSWORD: password
options: --health-cmd="mysqladmin ping -uroot -ppassword" --health-interval=5s --health-timeout=2s --health-retries=3
if: |
- (github.event_name != 'push' || github.event.repository.full_name == 'home-assistant/core')
- && github.event.inputs.lint-only != 'true'
- && github.event.inputs.pylint-only != 'true'
- && github.event.inputs.mypy-only != 'true'
- && github.event.inputs.audit-licenses-only != 'true'
+ needs.info.outputs.lint_only != 'true'
&& needs.info.outputs.mariadb_groups != '[]'
needs:
- info
@@ -1051,7 +1080,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ matrix.python-version }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -1088,6 +1117,7 @@ jobs:
cov_params+=(--cov="homeassistant.components.recorder")
cov_params+=(--cov-report=xml)
cov_params+=(--cov-report=term-missing)
+ cov_params+=(--junitxml=junit.xml -o junit_family=legacy)
fi
python3 -b -X dev -m pytest \
@@ -1108,7 +1138,7 @@ jobs:
2>&1 | tee pytest-${{ matrix.python-version }}-${mariadb}.txt
- name: Upload pytest output
if: success() || failure() && steps.pytest-partial.conclusion == 'failure'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: pytest-${{ github.run_number }}-${{ matrix.python-version }}-${{
steps.pytest-partial.outputs.mariadb }}
@@ -1116,12 +1146,19 @@ jobs:
overwrite: true
- name: Upload coverage artifact
if: needs.info.outputs.skip_coverage != 'true'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: coverage-${{ matrix.python-version }}-${{
steps.pytest-partial.outputs.mariadb }}
path: coverage.xml
overwrite: true
+ - name: Upload test results artifact
+ if: needs.info.outputs.skip_coverage != 'true' && !cancelled()
+ uses: actions/upload-artifact@v4.6.2
+ with:
+ name: test-results-mariadb-${{ matrix.python-version }}-${{
+ steps.pytest-partial.outputs.mariadb }}
+ path: junit.xml
- name: Check dirty
run: |
./script/check_dirty
@@ -1137,11 +1174,7 @@ jobs:
POSTGRES_PASSWORD: password
options: --health-cmd="pg_isready -hlocalhost -Upostgres" --health-interval=5s --health-timeout=2s --health-retries=3
if: |
- (github.event_name != 'push' || github.event.repository.full_name == 'home-assistant/core')
- && github.event.inputs.lint-only != 'true'
- && github.event.inputs.pylint-only != 'true'
- && github.event.inputs.mypy-only != 'true'
- && github.event.inputs.audit-licenses-only != 'true'
+ needs.info.outputs.lint_only != 'true'
&& needs.info.outputs.postgresql_groups != '[]'
needs:
- info
@@ -1181,7 +1214,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ matrix.python-version }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -1218,6 +1251,7 @@ jobs:
cov_params+=(--cov="homeassistant.components.recorder")
cov_params+=(--cov-report=xml)
cov_params+=(--cov-report=term-missing)
+ cov_params+=(--junitxml=junit.xml -o junit_family=legacy)
fi
python3 -b -X dev -m pytest \
@@ -1239,7 +1273,7 @@ jobs:
2>&1 | tee pytest-${{ matrix.python-version }}-${postgresql}.txt
- name: Upload pytest output
if: success() || failure() && steps.pytest-partial.conclusion == 'failure'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: pytest-${{ github.run_number }}-${{ matrix.python-version }}-${{
steps.pytest-partial.outputs.postgresql }}
@@ -1247,12 +1281,19 @@ jobs:
overwrite: true
- name: Upload coverage artifact
if: needs.info.outputs.skip_coverage != 'true'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: coverage-${{ matrix.python-version }}-${{
steps.pytest-partial.outputs.postgresql }}
path: coverage.xml
overwrite: true
+ - name: Upload test results artifact
+ if: needs.info.outputs.skip_coverage != 'true' && !cancelled()
+ uses: actions/upload-artifact@v4.6.2
+ with:
+ name: test-results-postgres-${{ matrix.python-version }}-${{
+ steps.pytest-partial.outputs.postgresql }}
+ path: junit.xml
- name: Check dirty
run: |
./script/check_dirty
@@ -1271,12 +1312,12 @@ jobs:
- name: Check out code from GitHub
uses: actions/checkout@v4.2.2
- name: Download all coverage artifacts
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
pattern: coverage-*
- name: Upload coverage to Codecov
if: needs.info.outputs.test_full_suite == 'true'
- uses: codecov/codecov-action@v5.3.1
+ uses: codecov/codecov-action@v5.4.0
with:
fail_ci_if_error: true
flags: full-suite
@@ -1285,11 +1326,7 @@ jobs:
pytest-partial:
runs-on: ubuntu-24.04
if: |
- (github.event_name != 'push' || github.event.repository.full_name == 'home-assistant/core')
- && github.event.inputs.lint-only != 'true'
- && github.event.inputs.pylint-only != 'true'
- && github.event.inputs.mypy-only != 'true'
- && github.event.inputs.audit-licenses-only != 'true'
+ needs.info.outputs.lint_only != 'true'
&& needs.info.outputs.tests_glob
&& needs.info.outputs.test_full_suite == 'false'
needs:
@@ -1328,7 +1365,7 @@ jobs:
check-latest: true
- name: Restore full Python ${{ matrix.python-version }} virtual environment
id: cache-venv
- uses: actions/cache/restore@v4.2.0
+ uses: actions/cache/restore@v4.2.3
with:
path: venv
fail-on-cache-miss: true
@@ -1365,6 +1402,7 @@ jobs:
cov_params+=(--cov="homeassistant.components.${{ matrix.group }}")
cov_params+=(--cov-report=xml)
cov_params+=(--cov-report=term-missing)
+ cov_params+=(--junitxml=junit.xml -o junit_family=legacy)
fi
python3 -b -X dev -m pytest \
@@ -1382,18 +1420,24 @@ jobs:
2>&1 | tee pytest-${{ matrix.python-version }}-${{ matrix.group }}.txt
- name: Upload pytest output
if: success() || failure() && steps.pytest-partial.conclusion == 'failure'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: pytest-${{ github.run_number }}-${{ matrix.python-version }}-${{ matrix.group }}
path: pytest-*.txt
overwrite: true
- name: Upload coverage artifact
if: needs.info.outputs.skip_coverage != 'true'
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: coverage-${{ matrix.python-version }}-${{ matrix.group }}
path: coverage.xml
overwrite: true
+ - name: Upload test results artifact
+ if: needs.info.outputs.skip_coverage != 'true' && !cancelled()
+ uses: actions/upload-artifact@v4.6.2
+ with:
+ name: test-results-partial-${{ matrix.python-version }}-${{ matrix.group }}
+ path: junit.xml
- name: Check dirty
run: |
./script/check_dirty
@@ -1410,12 +1454,37 @@ jobs:
- name: Check out code from GitHub
uses: actions/checkout@v4.2.2
- name: Download all coverage artifacts
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
pattern: coverage-*
- name: Upload coverage to Codecov
if: needs.info.outputs.test_full_suite == 'false'
- uses: codecov/codecov-action@v5.3.1
+ uses: codecov/codecov-action@v5.4.0
with:
fail_ci_if_error: true
token: ${{ secrets.CODECOV_TOKEN }}
+
+ upload-test-results:
+ name: Upload test results to Codecov
+ # codecov/test-results-action currently doesn't support tokenless uploads
+ # therefore we can't run it on forks
+ if: ${{ (github.event_name != 'pull_request' || !github.event.pull_request.head.repo.fork) && needs.info.outputs.skip_coverage != 'true' && !cancelled() }}
+ runs-on: ubuntu-24.04
+ needs:
+ - info
+ - pytest-partial
+ - pytest-full
+ - pytest-postgres
+ - pytest-mariadb
+ timeout-minutes: 10
+ steps:
+ - name: Download all coverage artifacts
+ uses: actions/download-artifact@v4.2.1
+ with:
+ pattern: test-results-*
+ - name: Upload test results to Codecov
+ uses: codecov/test-results-action@v1
+ with:
+ fail_ci_if_error: true
+ verbose: true
+ token: ${{ secrets.CODECOV_TOKEN }}
diff --git a/.github/workflows/codeql.yml b/.github/workflows/codeql.yml
index c1272759acc..f4d4144243c 100644
--- a/.github/workflows/codeql.yml
+++ b/.github/workflows/codeql.yml
@@ -24,11 +24,11 @@ jobs:
uses: actions/checkout@v4.2.2
- name: Initialize CodeQL
- uses: github/codeql-action/init@v3.28.8
+ uses: github/codeql-action/init@v3.28.12
with:
languages: python
- name: Perform CodeQL Analysis
- uses: github/codeql-action/analyze@v3.28.8
+ uses: github/codeql-action/analyze@v3.28.12
with:
category: "/language:python"
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
index 41e7b351184..cdf0c07cccf 100644
--- a/.github/workflows/wheels.yml
+++ b/.github/workflows/wheels.yml
@@ -91,7 +91,7 @@ jobs:
) > build_constraints.txt
- name: Upload env_file
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: env_file
path: ./.env_file
@@ -99,14 +99,14 @@ jobs:
overwrite: true
- name: Upload build_constraints
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: build_constraints
path: ./build_constraints.txt
overwrite: true
- name: Upload requirements_diff
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: requirements_diff
path: ./requirements_diff.txt
@@ -118,7 +118,7 @@ jobs:
python -m script.gen_requirements_all ci
- name: Upload requirements_all_wheels
- uses: actions/upload-artifact@v4.6.0
+ uses: actions/upload-artifact@v4.6.2
with:
name: requirements_all_wheels
path: ./requirements_all_wheels_*.txt
@@ -138,17 +138,17 @@ jobs:
uses: actions/checkout@v4.2.2
- name: Download env_file
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: env_file
- name: Download build_constraints
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: build_constraints
- name: Download requirements_diff
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: requirements_diff
@@ -159,7 +159,7 @@ jobs:
sed -i "/uv/d" requirements_diff.txt
- name: Build wheels
- uses: home-assistant/wheels@2024.11.0
+ uses: home-assistant/wheels@2025.02.0
with:
abi: ${{ matrix.abi }}
tag: musllinux_1_2
@@ -187,22 +187,22 @@ jobs:
uses: actions/checkout@v4.2.2
- name: Download env_file
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: env_file
- name: Download build_constraints
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: build_constraints
- name: Download requirements_diff
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: requirements_diff
- name: Download requirements_all_wheels
- uses: actions/download-artifact@v4.1.8
+ uses: actions/download-artifact@v4.2.1
with:
name: requirements_all_wheels
@@ -218,16 +218,8 @@ jobs:
sed -i "/uv/d" requirements.txt
sed -i "/uv/d" requirements_diff.txt
- - name: Split requirements all
- run: |
- # We split requirements all into multiple files.
- # This is to prevent the build from running out of memory when
- # resolving packages on 32-bits systems (like armhf, armv7).
-
- split -l $(expr $(expr $(cat requirements_all.txt | wc -l) + 1) / 3) requirements_all_wheels_${{ matrix.arch }}.txt requirements_all.txt
-
- - name: Build wheels (part 1)
- uses: home-assistant/wheels@2024.11.0
+ - name: Build wheels
+ uses: home-assistant/wheels@2025.02.0
with:
abi: ${{ matrix.abi }}
tag: musllinux_1_2
@@ -238,32 +230,4 @@ jobs:
skip-binary: aiohttp;charset-normalizer;grpcio;multidict;SQLAlchemy;propcache;protobuf;pymicro-vad;yarl
constraints: "homeassistant/package_constraints.txt"
requirements-diff: "requirements_diff.txt"
- requirements: "requirements_all.txtaa"
-
- - name: Build wheels (part 2)
- uses: home-assistant/wheels@2024.11.0
- with:
- abi: ${{ matrix.abi }}
- tag: musllinux_1_2
- arch: ${{ matrix.arch }}
- wheels-key: ${{ secrets.WHEELS_KEY }}
- env-file: true
- apk: "bluez-dev;libffi-dev;openssl-dev;glib-dev;eudev-dev;libxml2-dev;libxslt-dev;libpng-dev;libjpeg-turbo-dev;tiff-dev;cups-dev;gmp-dev;mpfr-dev;mpc1-dev;ffmpeg-dev;gammu-dev;yaml-dev;openblas-dev;fftw-dev;lapack-dev;gfortran;blas-dev;eigen-dev;freetype-dev;glew-dev;harfbuzz-dev;hdf5-dev;libdc1394-dev;libtbb-dev;mesa-dev;openexr-dev;openjpeg-dev;uchardet-dev;nasm;zlib-ng-dev"
- skip-binary: aiohttp;charset-normalizer;grpcio;multidict;SQLAlchemy;propcache;protobuf;pymicro-vad;yarl
- constraints: "homeassistant/package_constraints.txt"
- requirements-diff: "requirements_diff.txt"
- requirements: "requirements_all.txtab"
-
- - name: Build wheels (part 3)
- uses: home-assistant/wheels@2024.11.0
- with:
- abi: ${{ matrix.abi }}
- tag: musllinux_1_2
- arch: ${{ matrix.arch }}
- wheels-key: ${{ secrets.WHEELS_KEY }}
- env-file: true
- apk: "bluez-dev;libffi-dev;openssl-dev;glib-dev;eudev-dev;libxml2-dev;libxslt-dev;libpng-dev;libjpeg-turbo-dev;tiff-dev;cups-dev;gmp-dev;mpfr-dev;mpc1-dev;ffmpeg-dev;gammu-dev;yaml-dev;openblas-dev;fftw-dev;lapack-dev;gfortran;blas-dev;eigen-dev;freetype-dev;glew-dev;harfbuzz-dev;hdf5-dev;libdc1394-dev;libtbb-dev;mesa-dev;openexr-dev;openjpeg-dev;uchardet-dev;nasm;zlib-ng-dev"
- skip-binary: aiohttp;charset-normalizer;grpcio;multidict;SQLAlchemy;propcache;protobuf;pymicro-vad;yarl
- constraints: "homeassistant/package_constraints.txt"
- requirements-diff: "requirements_diff.txt"
- requirements: "requirements_all.txtac"
+ requirements: "requirements_all.txt"
diff --git a/.gitignore b/.gitignore
index 241255253c5..5aa51c9d762 100644
--- a/.gitignore
+++ b/.gitignore
@@ -69,6 +69,7 @@ test-reports/
test-results.xml
test-output.xml
pytest-*.txt
+junit.xml
# Translations
*.mo
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index a059710d3d7..42e05a869c3 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,6 +1,6 @@
repos:
- repo: https://github.com/astral-sh/ruff-pre-commit
- rev: v0.9.1
+ rev: v0.11.0
hooks:
- id: ruff
args:
diff --git a/.strict-typing b/.strict-typing
index 1e3187980cc..0e00c2e9e07 100644
--- a/.strict-typing
+++ b/.strict-typing
@@ -103,6 +103,7 @@ homeassistant.components.auth.*
homeassistant.components.automation.*
homeassistant.components.awair.*
homeassistant.components.axis.*
+homeassistant.components.azure_storage.*
homeassistant.components.backup.*
homeassistant.components.baf.*
homeassistant.components.bang_olufsen.*
@@ -135,6 +136,7 @@ homeassistant.components.clicksend.*
homeassistant.components.climate.*
homeassistant.components.cloud.*
homeassistant.components.co2signal.*
+homeassistant.components.comelit.*
homeassistant.components.command_line.*
homeassistant.components.config.*
homeassistant.components.configurator.*
@@ -234,6 +236,7 @@ homeassistant.components.here_travel_time.*
homeassistant.components.history.*
homeassistant.components.history_stats.*
homeassistant.components.holiday.*
+homeassistant.components.home_connect.*
homeassistant.components.homeassistant.*
homeassistant.components.homeassistant_alerts.*
homeassistant.components.homeassistant_green.*
@@ -394,6 +397,7 @@ homeassistant.components.pure_energie.*
homeassistant.components.purpleair.*
homeassistant.components.pushbullet.*
homeassistant.components.pvoutput.*
+homeassistant.components.pyload.*
homeassistant.components.python_script.*
homeassistant.components.qbus.*
homeassistant.components.qnap_qsw.*
@@ -406,7 +410,9 @@ homeassistant.components.raspberry_pi.*
homeassistant.components.rdw.*
homeassistant.components.recollect_waste.*
homeassistant.components.recorder.*
+homeassistant.components.remember_the_milk.*
homeassistant.components.remote.*
+homeassistant.components.remote_calendar.*
homeassistant.components.renault.*
homeassistant.components.reolink.*
homeassistant.components.repairs.*
@@ -437,6 +443,7 @@ homeassistant.components.select.*
homeassistant.components.sensibo.*
homeassistant.components.sensirion_ble.*
homeassistant.components.sensor.*
+homeassistant.components.sensorpush_cloud.*
homeassistant.components.sensoterra.*
homeassistant.components.senz.*
homeassistant.components.sfr_box.*
@@ -524,6 +531,7 @@ homeassistant.components.vallox.*
homeassistant.components.valve.*
homeassistant.components.velbus.*
homeassistant.components.vlc_telnet.*
+homeassistant.components.vodafone_station.*
homeassistant.components.wake_on_lan.*
homeassistant.components.wake_word.*
homeassistant.components.wallbox.*
diff --git a/.vscode/launch.json b/.vscode/launch.json
index 7b77a1c9bfd..459a9e6acc5 100644
--- a/.vscode/launch.json
+++ b/.vscode/launch.json
@@ -38,10 +38,17 @@
"module": "pytest",
"justMyCode": false,
"args": [
- "--timeout=10",
"--picked"
],
},
+ {
+ "name": "Home Assistant: Debug Current Test File",
+ "type": "debugpy",
+ "request": "launch",
+ "module": "pytest",
+ "console": "integratedTerminal",
+ "args": ["-vv", "${file}"]
+ },
{
// Debug by attaching to local Home Assistant server using Remote Python Debugger.
// See https://www.home-assistant.io/integrations/debugpy/
@@ -77,4 +84,4 @@
]
}
]
-}
\ No newline at end of file
+}
diff --git a/.vscode/tasks.json b/.vscode/tasks.json
index 7425e7a2533..09c1d374299 100644
--- a/.vscode/tasks.json
+++ b/.vscode/tasks.json
@@ -4,7 +4,7 @@
{
"label": "Run Home Assistant Core",
"type": "shell",
- "command": "hass -c ./config",
+ "command": "${command:python.interpreterPath} -m homeassistant -c ./config",
"group": "test",
"presentation": {
"reveal": "always",
@@ -148,7 +148,7 @@
{
"label": "Install all Test Requirements",
"type": "shell",
- "command": "uv pip install -r requirements_test_all.txt",
+ "command": "uv pip install -r requirements.txt -r requirements_test_all.txt",
"group": {
"kind": "build",
"isDefault": true
diff --git a/CODEOWNERS b/CODEOWNERS
index e510eec6dfa..1835e6d0be4 100644
--- a/CODEOWNERS
+++ b/CODEOWNERS
@@ -180,6 +180,8 @@ build.json @home-assistant/supervisor
/homeassistant/components/azure_event_hub/ @eavanvalkenburg
/tests/components/azure_event_hub/ @eavanvalkenburg
/homeassistant/components/azure_service_bus/ @hfurubotten
+/homeassistant/components/azure_storage/ @zweckj
+/tests/components/azure_storage/ @zweckj
/homeassistant/components/backup/ @home-assistant/core
/tests/components/backup/ @home-assistant/core
/homeassistant/components/baf/ @bdraco @jfroy
@@ -568,8 +570,8 @@ build.json @home-assistant/supervisor
/tests/components/google_cloud/ @lufton @tronikos
/homeassistant/components/google_drive/ @tronikos
/tests/components/google_drive/ @tronikos
-/homeassistant/components/google_generative_ai_conversation/ @tronikos
-/tests/components/google_generative_ai_conversation/ @tronikos
+/homeassistant/components/google_generative_ai_conversation/ @tronikos @ivanlh
+/tests/components/google_generative_ai_conversation/ @tronikos @ivanlh
/homeassistant/components/google_mail/ @tkdrob
/tests/components/google_mail/ @tkdrob
/homeassistant/components/google_photos/ @allenporter
@@ -967,8 +969,8 @@ build.json @home-assistant/supervisor
/tests/components/motionblinds_ble/ @LennP @jerrybboy
/homeassistant/components/motioneye/ @dermotduffy
/tests/components/motioneye/ @dermotduffy
-/homeassistant/components/motionmount/ @RJPoelstra
-/tests/components/motionmount/ @RJPoelstra
+/homeassistant/components/motionmount/ @laiho-vogels
+/tests/components/motionmount/ @laiho-vogels
/homeassistant/components/mqtt/ @emontnemery @jbouwh @bdraco
/tests/components/mqtt/ @emontnemery @jbouwh @bdraco
/homeassistant/components/msteams/ @peroyvind
@@ -1051,8 +1053,8 @@ build.json @home-assistant/supervisor
/tests/components/numato/ @clssn
/homeassistant/components/number/ @home-assistant/core @Shulyaka
/tests/components/number/ @home-assistant/core @Shulyaka
-/homeassistant/components/nut/ @bdraco @ollo69 @pestevez
-/tests/components/nut/ @bdraco @ollo69 @pestevez
+/homeassistant/components/nut/ @bdraco @ollo69 @pestevez @tdfountain
+/tests/components/nut/ @bdraco @ollo69 @pestevez @tdfountain
/homeassistant/components/nws/ @MatthewFlamm @kamiyo
/tests/components/nws/ @MatthewFlamm @kamiyo
/homeassistant/components/nyt_games/ @joostlek
@@ -1138,12 +1140,14 @@ build.json @home-assistant/supervisor
/tests/components/permobil/ @IsakNyberg
/homeassistant/components/persistent_notification/ @home-assistant/core
/tests/components/persistent_notification/ @home-assistant/core
+/homeassistant/components/pglab/ @pglab-electronics
+/tests/components/pglab/ @pglab-electronics
/homeassistant/components/philips_js/ @elupus
/tests/components/philips_js/ @elupus
/homeassistant/components/pi_hole/ @shenxn
/tests/components/pi_hole/ @shenxn
-/homeassistant/components/picnic/ @corneyl
-/tests/components/picnic/ @corneyl
+/homeassistant/components/picnic/ @corneyl @codesalatdev
+/tests/components/picnic/ @corneyl @codesalatdev
/homeassistant/components/ping/ @jpbede
/tests/components/ping/ @jpbede
/homeassistant/components/plaato/ @JohNan
@@ -1248,6 +1252,8 @@ build.json @home-assistant/supervisor
/tests/components/refoss/ @ashionky
/homeassistant/components/remote/ @home-assistant/core
/tests/components/remote/ @home-assistant/core
+/homeassistant/components/remote_calendar/ @Thomas55555
+/tests/components/remote_calendar/ @Thomas55555
/homeassistant/components/renault/ @epenet
/tests/components/renault/ @epenet
/homeassistant/components/renson/ @jimmyd-be
@@ -1340,6 +1346,8 @@ build.json @home-assistant/supervisor
/tests/components/sensorpro/ @bdraco
/homeassistant/components/sensorpush/ @bdraco
/tests/components/sensorpush/ @bdraco
+/homeassistant/components/sensorpush_cloud/ @sstallion
+/tests/components/sensorpush_cloud/ @sstallion
/homeassistant/components/sensoterra/ @markruys
/tests/components/sensoterra/ @markruys
/homeassistant/components/sentry/ @dcramer @frenck
@@ -1395,6 +1403,8 @@ build.json @home-assistant/supervisor
/tests/components/smappee/ @bsmappee
/homeassistant/components/smart_meter_texas/ @grahamwetzler
/tests/components/smart_meter_texas/ @grahamwetzler
+/homeassistant/components/smartthings/ @joostlek
+/tests/components/smartthings/ @joostlek
/homeassistant/components/smarttub/ @mdz
/tests/components/smarttub/ @mdz
/homeassistant/components/smarty/ @z0mbieprocess
@@ -1409,6 +1419,8 @@ build.json @home-assistant/supervisor
/tests/components/snapcast/ @luar123
/homeassistant/components/snmp/ @nmaggioni
/tests/components/snmp/ @nmaggioni
+/homeassistant/components/snoo/ @Lash-L
+/tests/components/snoo/ @Lash-L
/homeassistant/components/snooz/ @AustinBrunkhorst
/tests/components/snooz/ @AustinBrunkhorst
/homeassistant/components/solaredge/ @frenck @bdraco
@@ -1519,8 +1531,8 @@ build.json @home-assistant/supervisor
/tests/components/tedee/ @patrickhilker @zweckj
/homeassistant/components/tellduslive/ @fredrike
/tests/components/tellduslive/ @fredrike
-/homeassistant/components/template/ @PhracturedBlue @home-assistant/core
-/tests/components/template/ @PhracturedBlue @home-assistant/core
+/homeassistant/components/template/ @Petro31 @PhracturedBlue @home-assistant/core
+/tests/components/template/ @Petro31 @PhracturedBlue @home-assistant/core
/homeassistant/components/tesla_fleet/ @Bre77
/tests/components/tesla_fleet/ @Bre77
/homeassistant/components/tesla_wall_connector/ @einarhauks
@@ -1689,6 +1701,8 @@ build.json @home-assistant/supervisor
/tests/components/weatherflow_cloud/ @jeeftor
/homeassistant/components/weatherkit/ @tjhorner
/tests/components/weatherkit/ @tjhorner
+/homeassistant/components/webdav/ @jpbede
+/tests/components/webdav/ @jpbede
/homeassistant/components/webhook/ @home-assistant/core
/tests/components/webhook/ @home-assistant/core
/homeassistant/components/webmin/ @autinerd
diff --git a/Dockerfile b/Dockerfile
index 19b2c97b181..2efb9d59a44 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,8 +12,26 @@ ENV \
ARG QEMU_CPU
+# Home Assistant S6-Overlay
+COPY rootfs /
+
+# Needs to be redefined inside the FROM statement to be set for RUN commands
+ARG BUILD_ARCH
+# Get go2rtc binary
+RUN \
+ case "${BUILD_ARCH}" in \
+ "aarch64") go2rtc_suffix='arm64' ;; \
+ "armhf") go2rtc_suffix='armv6' ;; \
+ "armv7") go2rtc_suffix='arm' ;; \
+ *) go2rtc_suffix=${BUILD_ARCH} ;; \
+ esac \
+ && curl -L https://github.com/AlexxIT/go2rtc/releases/download/v1.9.9/go2rtc_linux_${go2rtc_suffix} --output /bin/go2rtc \
+ && chmod +x /bin/go2rtc \
+ # Verify go2rtc can be executed
+ && go2rtc --version
+
# Install uv
-RUN pip3 install uv==0.5.27
+RUN pip3 install uv==0.6.8
WORKDIR /usr/src
@@ -42,22 +60,4 @@ RUN \
&& python3 -m compileall \
homeassistant/homeassistant
-# Home Assistant S6-Overlay
-COPY rootfs /
-
-# Needs to be redefined inside the FROM statement to be set for RUN commands
-ARG BUILD_ARCH
-# Get go2rtc binary
-RUN \
- case "${BUILD_ARCH}" in \
- "aarch64") go2rtc_suffix='arm64' ;; \
- "armhf") go2rtc_suffix='armv6' ;; \
- "armv7") go2rtc_suffix='arm' ;; \
- *) go2rtc_suffix=${BUILD_ARCH} ;; \
- esac \
- && curl -L https://github.com/AlexxIT/go2rtc/releases/download/v1.9.8/go2rtc_linux_${go2rtc_suffix} --output /bin/go2rtc \
- && chmod +x /bin/go2rtc \
- # Verify go2rtc can be executed
- && go2rtc --version
-
WORKDIR /config
diff --git a/build.yaml b/build.yaml
index e6e149cf700..cd54e410493 100644
--- a/build.yaml
+++ b/build.yaml
@@ -1,10 +1,10 @@
image: ghcr.io/home-assistant/{arch}-homeassistant
build_from:
- aarch64: ghcr.io/home-assistant/aarch64-homeassistant-base:2024.12.0
- armhf: ghcr.io/home-assistant/armhf-homeassistant-base:2024.12.0
- armv7: ghcr.io/home-assistant/armv7-homeassistant-base:2024.12.0
- amd64: ghcr.io/home-assistant/amd64-homeassistant-base:2024.12.0
- i386: ghcr.io/home-assistant/i386-homeassistant-base:2024.12.0
+ aarch64: ghcr.io/home-assistant/aarch64-homeassistant-base:2025.02.1
+ armhf: ghcr.io/home-assistant/armhf-homeassistant-base:2025.02.1
+ armv7: ghcr.io/home-assistant/armv7-homeassistant-base:2025.02.1
+ amd64: ghcr.io/home-assistant/amd64-homeassistant-base:2025.02.1
+ i386: ghcr.io/home-assistant/i386-homeassistant-base:2025.02.1
codenotary:
signer: notary@home-assistant.io
base_image: notary@home-assistant.io
diff --git a/homeassistant/block_async_io.py b/homeassistant/block_async_io.py
index d224b0b151d..eb81268434b 100644
--- a/homeassistant/block_async_io.py
+++ b/homeassistant/block_async_io.py
@@ -178,6 +178,15 @@ _BLOCKING_CALLS: tuple[BlockingCall, ...] = (
strict_core=False,
skip_for_tests=True,
),
+ BlockingCall(
+ original_func=SSLContext.set_default_verify_paths,
+ object=SSLContext,
+ function="set_default_verify_paths",
+ check_allowed=None,
+ strict=False,
+ strict_core=False,
+ skip_for_tests=True,
+ ),
BlockingCall(
original_func=Path.open,
object=Path,
diff --git a/homeassistant/bootstrap.py b/homeassistant/bootstrap.py
index 490ce5559a9..02a3b8c8fcc 100644
--- a/homeassistant/bootstrap.py
+++ b/homeassistant/bootstrap.py
@@ -74,12 +74,14 @@ from .core_config import async_process_ha_core_config
from .exceptions import HomeAssistantError
from .helpers import (
area_registry,
+ backup,
category_registry,
config_validation as cv,
device_registry,
entity,
entity_registry,
floor_registry,
+ frame,
issue_registry,
label_registry,
recorder,
@@ -91,6 +93,7 @@ from .helpers.dispatcher import async_dispatcher_send_internal
from .helpers.storage import get_internal_store_manager
from .helpers.system_info import async_get_system_info
from .helpers.typing import ConfigType
+from .loader import Integration
from .setup import (
# _setup_started is marked as protected to make it clear
# that it is not part of the public API and should not be used
@@ -134,14 +137,12 @@ DATA_REGISTRIES_LOADED: HassKey[None] = HassKey("bootstrap_registries_loaded")
LOG_SLOW_STARTUP_INTERVAL = 60
SLOW_STARTUP_CHECK_INTERVAL = 1
+STAGE_0_SUBSTAGE_TIMEOUT = 60
STAGE_1_TIMEOUT = 120
STAGE_2_TIMEOUT = 300
WRAP_UP_TIMEOUT = 300
COOLDOWN_TIME = 60
-
-DEBUGGER_INTEGRATIONS = {"debugpy"}
-
# Core integrations are unconditionally loaded
CORE_INTEGRATIONS = {"homeassistant", "persistent_notification"}
@@ -152,6 +153,10 @@ LOGGING_AND_HTTP_DEPS_INTEGRATIONS = {
"isal",
# Set log levels
"logger",
+ # Ensure network config is available
+ # before hassio or any other integration is
+ # loaded that might create an aiohttp client session
+ "network",
# Error logging
"system_log",
"sentry",
@@ -161,23 +166,28 @@ FRONTEND_INTEGRATIONS = {
# integrations can be removed and database migration status is
# visible in frontend
"frontend",
- # Hassio is an after dependency of backup, after dependencies
- # are not promoted from stage 2 to earlier stages, so we need to
- # add it here. Hassio needs to be setup before backup, otherwise
- # the backup integration will think we are a container/core install
- # when using HAOS or Supervised install.
- "hassio",
- # Backup is an after dependency of frontend, after dependencies
- # are not promoted from stage 2 to earlier stages, so we need to
- # add it here.
- "backup",
}
-RECORDER_INTEGRATIONS = {
- # Setup after frontend
- # To record data
- "recorder",
-}
-DISCOVERY_INTEGRATIONS = ("bluetooth", "dhcp", "ssdp", "usb", "zeroconf")
+# Stage 0 is divided into substages. Each substage has a name, a set of integrations and a timeout.
+# The substage containing recorder should have no timeout, as it could cancel a database migration.
+# Recorder freezes "recorder" timeout during a migration, but it does not freeze other timeouts.
+# The substages preceding it should also have no timeout, until we ensure that the recorder
+# is not accidentally promoted as a dependency of any of the integrations in them.
+# If we add timeouts to the frontend substages, we should make sure they don't apply in recovery mode.
+STAGE_0_INTEGRATIONS = (
+ # Load logging and http deps as soon as possible
+ ("logging, http deps", LOGGING_AND_HTTP_DEPS_INTEGRATIONS, None),
+ # Setup frontend
+ ("frontend", FRONTEND_INTEGRATIONS, None),
+ # Setup recorder
+ ("recorder", {"recorder"}, None),
+ # Start up debuggers. Start these first in case they want to wait.
+ ("debugger", {"debugpy"}, STAGE_0_SUBSTAGE_TIMEOUT),
+ # Zeroconf is used for mdns resolution in aiohttp client helper.
+ ("zeroconf", {"zeroconf"}, STAGE_0_SUBSTAGE_TIMEOUT),
+)
+
+DISCOVERY_INTEGRATIONS = ("bluetooth", "dhcp", "ssdp", "usb")
+# Stage 1 integrations are not to be preimported in bootstrap.
STAGE_1_INTEGRATIONS = {
# We need to make sure discovery integrations
# update their deps before stage 2 integrations
@@ -192,6 +202,7 @@ STAGE_1_INTEGRATIONS = {
# Ensure supervisor is available
"hassio",
}
+
DEFAULT_INTEGRATIONS = {
# These integrations are set up unless recovery mode is activated.
#
@@ -232,22 +243,12 @@ DEFAULT_INTEGRATIONS_SUPERVISOR = {
# These integrations are set up if using the Supervisor
"hassio",
}
+
CRITICAL_INTEGRATIONS = {
# Recovery mode is activated if these integrations fail to set up
"frontend",
}
-SETUP_ORDER = (
- # Load logging and http deps as soon as possible
- ("logging, http deps", LOGGING_AND_HTTP_DEPS_INTEGRATIONS),
- # Setup frontend
- ("frontend", FRONTEND_INTEGRATIONS),
- # Setup recorder
- ("recorder", RECORDER_INTEGRATIONS),
- # Start up debuggers. Start these first in case they want to wait.
- ("debugger", DEBUGGER_INTEGRATIONS),
-)
-
#
# Storage keys we are likely to load during startup
# in order of when we expect to load them.
@@ -299,14 +300,6 @@ async def async_setup_hass(
return hass
- async def stop_hass(hass: core.HomeAssistant) -> None:
- """Stop hass."""
- # Ask integrations to shut down. It's messy but we can't
- # do a clean stop without knowing what is broken
- with contextlib.suppress(TimeoutError):
- async with hass.timeout.async_timeout(10):
- await hass.async_stop()
-
hass = await create_hass()
if runtime_config.skip_pip or runtime_config.skip_pip_packages:
@@ -322,10 +315,10 @@ async def async_setup_hass(
block_async_io.enable()
- config_dict = None
- basic_setup_success = False
-
if not (recovery_mode := runtime_config.recovery_mode):
+ config_dict = None
+ basic_setup_success = False
+
await hass.async_add_executor_job(conf_util.process_ha_config_upgrade, hass)
try:
@@ -343,39 +336,43 @@ async def async_setup_hass(
await async_from_config_dict(config_dict, hass) is not None
)
- if config_dict is None:
- recovery_mode = True
- await stop_hass(hass)
- hass = await create_hass()
+ if config_dict is None:
+ recovery_mode = True
+ await hass.async_stop(force=True)
+ hass = await create_hass()
- elif not basic_setup_success:
- _LOGGER.warning("Unable to set up core integrations. Activating recovery mode")
- recovery_mode = True
- await stop_hass(hass)
- hass = await create_hass()
+ elif not basic_setup_success:
+ _LOGGER.warning(
+ "Unable to set up core integrations. Activating recovery mode"
+ )
+ recovery_mode = True
+ await hass.async_stop(force=True)
+ hass = await create_hass()
- elif any(domain not in hass.config.components for domain in CRITICAL_INTEGRATIONS):
- _LOGGER.warning(
- "Detected that %s did not load. Activating recovery mode",
- ",".join(CRITICAL_INTEGRATIONS),
- )
+ elif any(
+ domain not in hass.config.components for domain in CRITICAL_INTEGRATIONS
+ ):
+ _LOGGER.warning(
+ "Detected that %s did not load. Activating recovery mode",
+ ",".join(CRITICAL_INTEGRATIONS),
+ )
- old_config = hass.config
- old_logging = hass.data.get(DATA_LOGGING)
+ old_config = hass.config
+ old_logging = hass.data.get(DATA_LOGGING)
- recovery_mode = True
- await stop_hass(hass)
- hass = await create_hass()
+ recovery_mode = True
+ await hass.async_stop(force=True)
+ hass = await create_hass()
- if old_logging:
- hass.data[DATA_LOGGING] = old_logging
- hass.config.debug = old_config.debug
- hass.config.skip_pip = old_config.skip_pip
- hass.config.skip_pip_packages = old_config.skip_pip_packages
- hass.config.internal_url = old_config.internal_url
- hass.config.external_url = old_config.external_url
- # Setup loader cache after the config dir has been set
- loader.async_setup(hass)
+ if old_logging:
+ hass.data[DATA_LOGGING] = old_logging
+ hass.config.debug = old_config.debug
+ hass.config.skip_pip = old_config.skip_pip
+ hass.config.skip_pip_packages = old_config.skip_pip_packages
+ hass.config.internal_url = old_config.internal_url
+ hass.config.external_url = old_config.external_url
+ # Setup loader cache after the config dir has been set
+ loader.async_setup(hass)
if recovery_mode:
_LOGGER.info("Starting in recovery mode")
@@ -438,9 +435,10 @@ async def async_load_base_functionality(hass: core.HomeAssistant) -> None:
if DATA_REGISTRIES_LOADED in hass.data:
return
hass.data[DATA_REGISTRIES_LOADED] = None
- translation.async_setup(hass)
entity.async_setup(hass)
+ frame.async_setup(hass)
template.async_setup(hass)
+ translation.async_setup(hass)
await asyncio.gather(
create_eager_task(get_internal_store_manager(hass).async_initialize()),
create_eager_task(area_registry.async_load(hass)),
@@ -661,11 +659,10 @@ def _create_log_file(
err_handler = _RotatingFileHandlerWithoutShouldRollOver(
err_log_path, backupCount=1
)
-
- try:
- err_handler.doRollover()
- except OSError as err:
- _LOGGER.error("Error rolling over log file: %s", err)
+ try:
+ err_handler.doRollover()
+ except OSError as err:
+ _LOGGER.error("Error rolling over log file: %s", err)
return err_handler
@@ -694,7 +691,6 @@ async def async_mount_local_lib_path(config_dir: str) -> str:
return deps_dir
-@core.callback
def _get_domains(hass: core.HomeAssistant, config: dict[str, Any]) -> set[str]:
"""Get domains of components to set up."""
# Filter out the repeating and common config section [homeassistant]
@@ -716,6 +712,258 @@ def _get_domains(hass: core.HomeAssistant, config: dict[str, Any]) -> set[str]:
return domains
+async def _async_resolve_domains_and_preload(
+ hass: core.HomeAssistant, config: dict[str, Any]
+) -> tuple[dict[str, Integration], dict[str, Integration]]:
+ """Resolve all dependencies and return integrations to set up.
+
+ The return value is a tuple of two dictionaries:
+ - The first dictionary contains integrations
+ specified by the configuration (including config entries).
+ - The second dictionary contains the same integrations as the first dictionary
+ together with all their dependencies.
+ """
+ domains_to_setup = _get_domains(hass, config)
+ platform_integrations = conf_util.extract_platform_integrations(
+ config, BASE_PLATFORMS
+ )
+ # Ensure base platforms that have platform integrations are added to `domains`,
+ # so they can be setup first instead of discovering them later when a config
+ # entry setup task notices that it's needed and there is already a long line
+ # to use the import executor.
+ #
+ # For example if we have
+ # sensor:
+ # - platform: template
+ #
+ # `template` has to be loaded to validate the config for sensor
+ # so we want to start loading `sensor` as soon as we know
+ # it will be needed. The more platforms under `sensor:`, the longer
+ # it will take to finish setup for `sensor` because each of these
+ # platforms has to be imported before we can validate the config.
+ #
+ # Thankfully we are migrating away from the platform pattern
+ # so this will be less of a problem in the future.
+ domains_to_setup.update(platform_integrations)
+
+ # Additionally process base platforms since we do not require the manifest
+ # to list them as dependencies.
+ # We want to later avoid lock contention when multiple integrations try to load
+ # their manifests at once.
+ # Also process integrations that are defined under base platforms
+ # to speed things up.
+ additional_domains_to_process = {
+ *BASE_PLATFORMS,
+ *chain.from_iterable(platform_integrations.values()),
+ }
+
+ # Resolve all dependencies so we know all integrations
+ # that will have to be loaded and start right-away
+ integrations_or_excs = await loader.async_get_integrations(
+ hass, {*domains_to_setup, *additional_domains_to_process}
+ )
+ # Eliminate those missing or with invalid manifest
+ integrations_to_process = {
+ domain: itg
+ for domain, itg in integrations_or_excs.items()
+ if isinstance(itg, Integration)
+ }
+ integrations_dependencies = await loader.resolve_integrations_dependencies(
+ hass, integrations_to_process.values()
+ )
+ # Eliminate those without valid dependencies
+ integrations_to_process = {
+ domain: integrations_to_process[domain] for domain in integrations_dependencies
+ }
+
+ integrations_to_setup = {
+ domain: itg
+ for domain, itg in integrations_to_process.items()
+ if domain in domains_to_setup
+ }
+ all_integrations_to_setup = integrations_to_setup.copy()
+ all_integrations_to_setup.update(
+ (dep, loader.async_get_loaded_integration(hass, dep))
+ for domain in integrations_to_setup
+ for dep in integrations_dependencies[domain].difference(
+ all_integrations_to_setup
+ )
+ )
+
+ # Gather requirements for all integrations,
+ # their dependencies and after dependencies.
+ # To gather all the requirements we must ignore exceptions here.
+ # The exceptions will be detected and handled later in the bootstrap process.
+ integrations_after_dependencies = (
+ await loader.resolve_integrations_after_dependencies(
+ hass, integrations_to_process.values(), ignore_exceptions=True
+ )
+ )
+ integrations_requirements = {
+ domain: itg.requirements for domain, itg in integrations_to_process.items()
+ }
+ integrations_requirements.update(
+ (dep, loader.async_get_loaded_integration(hass, dep).requirements)
+ for deps in integrations_after_dependencies.values()
+ for dep in deps.difference(integrations_requirements)
+ )
+ all_requirements = set(chain.from_iterable(integrations_requirements.values()))
+
+ # Optimistically check if requirements are already installed
+ # ahead of setting up the integrations so we can prime the cache
+ # We do not wait for this since it's an optimization only
+ hass.async_create_background_task(
+ requirements.async_load_installed_versions(hass, all_requirements),
+ "check installed requirements",
+ eager_start=True,
+ )
+
+ # Start loading translations for all integrations we are going to set up
+ # in the background so they are ready when we need them. This avoids a
+ # lot of waiting for the translation load lock and a thundering herd of
+ # tasks trying to load the same translations at the same time as each
+ # integration is loaded.
+ #
+ # We do not wait for this since as soon as the task runs it will
+ # hold the translation load lock and if anything is fast enough to
+ # wait for the translation load lock, loading will be done by the
+ # time it gets to it.
+ translations_to_load = {*all_integrations_to_setup, *additional_domains_to_process}
+ hass.async_create_background_task(
+ translation.async_load_integrations(hass, translations_to_load),
+ "load translations",
+ eager_start=True,
+ )
+
+ # Preload storage for all integrations we are going to set up
+ # so we do not have to wait for it to be loaded when we need it
+ # in the setup process.
+ hass.async_create_background_task(
+ get_internal_store_manager(hass).async_preload(
+ [*PRELOAD_STORAGE, *all_integrations_to_setup]
+ ),
+ "preload storage",
+ eager_start=True,
+ )
+
+ return integrations_to_setup, all_integrations_to_setup
+
+
+async def _async_set_up_integrations(
+ hass: core.HomeAssistant, config: dict[str, Any]
+) -> None:
+ """Set up all the integrations."""
+ watcher = _WatchPendingSetups(hass, _setup_started(hass))
+ watcher.async_start()
+
+ integrations, all_integrations = await _async_resolve_domains_and_preload(
+ hass, config
+ )
+ all_domains = set(all_integrations)
+ domains = set(integrations)
+
+ _LOGGER.info(
+ "Domains to be set up: %s | %s",
+ domains,
+ all_domains - domains,
+ )
+
+ # Initialize recorder
+ if "recorder" in all_domains:
+ recorder.async_initialize_recorder(hass)
+
+ # Initialize backup
+ if "backup" in all_domains:
+ backup.async_initialize_backup(hass)
+
+ stages: list[tuple[str, set[str], int | None]] = [
+ *(
+ (name, domain_group, timeout)
+ for name, domain_group, timeout in STAGE_0_INTEGRATIONS
+ ),
+ ("1", STAGE_1_INTEGRATIONS, STAGE_1_TIMEOUT),
+ ("2", domains, STAGE_2_TIMEOUT),
+ ]
+
+ _LOGGER.info("Setting up stage 0")
+ for name, domain_group, timeout in stages:
+ stage_domains_unfiltered = domain_group & all_domains
+ if not stage_domains_unfiltered:
+ _LOGGER.info("Nothing to set up in stage %s: %s", name, domain_group)
+ continue
+
+ stage_domains = stage_domains_unfiltered - hass.config.components
+ if not stage_domains:
+ _LOGGER.info("Already set up stage %s: %s", name, stage_domains_unfiltered)
+ continue
+
+ stage_dep_domains_unfiltered = {
+ dep
+ for domain in stage_domains
+ for dep in all_integrations[domain].all_dependencies
+ if dep not in stage_domains
+ }
+ stage_dep_domains = stage_dep_domains_unfiltered - hass.config.components
+
+ stage_all_domains = stage_domains | stage_dep_domains
+ stage_all_integrations = {
+ domain: all_integrations[domain] for domain in stage_all_domains
+ }
+ # Detect all cycles
+ stage_integrations_after_dependencies = (
+ await loader.resolve_integrations_after_dependencies(
+ hass, stage_all_integrations.values(), stage_all_domains
+ )
+ )
+ stage_all_domains = set(stage_integrations_after_dependencies)
+ stage_domains &= stage_all_domains
+ stage_dep_domains &= stage_all_domains
+
+ _LOGGER.info(
+ "Setting up stage %s: %s | %s\nDependencies: %s | %s",
+ name,
+ stage_domains,
+ stage_domains_unfiltered - stage_domains,
+ stage_dep_domains,
+ stage_dep_domains_unfiltered - stage_dep_domains,
+ )
+
+ async_set_domains_to_be_loaded(hass, stage_all_domains)
+
+ if timeout is None:
+ await _async_setup_multi_components(hass, stage_all_domains, config)
+ continue
+ try:
+ async with hass.timeout.async_timeout(timeout, cool_down=COOLDOWN_TIME):
+ await _async_setup_multi_components(hass, stage_all_domains, config)
+ except TimeoutError:
+ _LOGGER.warning(
+ "Setup timed out for stage %s waiting on %s - moving forward",
+ name,
+ hass._active_tasks, # noqa: SLF001
+ )
+
+ # Wrap up startup
+ _LOGGER.debug("Waiting for startup to wrap up")
+ try:
+ async with hass.timeout.async_timeout(WRAP_UP_TIMEOUT, cool_down=COOLDOWN_TIME):
+ await hass.async_block_till_done()
+ except TimeoutError:
+ _LOGGER.warning(
+ "Setup timed out for bootstrap waiting on %s - moving forward",
+ hass._active_tasks, # noqa: SLF001
+ )
+
+ watcher.async_stop()
+
+ if _LOGGER.isEnabledFor(logging.DEBUG):
+ setup_time = async_get_setup_timings(hass)
+ _LOGGER.debug(
+ "Integration setup times: %s",
+ dict(sorted(setup_time.items(), key=itemgetter(1), reverse=True)),
+ )
+
+
class _WatchPendingSetups:
"""Periodic log and dispatch of setups that are pending."""
@@ -787,14 +1035,12 @@ class _WatchPendingSetups:
self._handle = None
-async def async_setup_multi_components(
+async def _async_setup_multi_components(
hass: core.HomeAssistant,
domains: set[str],
config: dict[str, Any],
) -> None:
"""Set up multiple domains. Log on failure."""
- # Avoid creating tasks for domains that were setup in a previous stage
- domains_not_yet_setup = domains - hass.config.components
# Create setup tasks for base platforms first since everything will have
# to wait to be imported, and the sooner we can get the base platforms
# loaded the sooner we can start loading the rest of the integrations.
@@ -804,9 +1050,7 @@ async def async_setup_multi_components(
f"setup component {domain}",
eager_start=True,
)
- for domain in sorted(
- domains_not_yet_setup, key=SETUP_ORDER_SORT_KEY, reverse=True
- )
+ for domain in sorted(domains, key=SETUP_ORDER_SORT_KEY, reverse=True)
}
results = await asyncio.gather(*futures.values(), return_exceptions=True)
for idx, domain in enumerate(futures):
@@ -817,278 +1061,3 @@ async def async_setup_multi_components(
domain,
exc_info=(type(result), result, result.__traceback__),
)
-
-
-async def _async_resolve_domains_to_setup(
- hass: core.HomeAssistant, config: dict[str, Any]
-) -> tuple[set[str], dict[str, loader.Integration]]:
- """Resolve all dependencies and return list of domains to set up."""
- domains_to_setup = _get_domains(hass, config)
- needed_requirements: set[str] = set()
- platform_integrations = conf_util.extract_platform_integrations(
- config, BASE_PLATFORMS
- )
- # Ensure base platforms that have platform integrations are added to
- # to `domains_to_setup so they can be setup first instead of
- # discovering them when later when a config entry setup task
- # notices its needed and there is already a long line to use
- # the import executor.
- #
- # For example if we have
- # sensor:
- # - platform: template
- #
- # `template` has to be loaded to validate the config for sensor
- # so we want to start loading `sensor` as soon as we know
- # it will be needed. The more platforms under `sensor:`, the longer
- # it will take to finish setup for `sensor` because each of these
- # platforms has to be imported before we can validate the config.
- #
- # Thankfully we are migrating away from the platform pattern
- # so this will be less of a problem in the future.
- domains_to_setup.update(platform_integrations)
-
- # Load manifests for base platforms and platform based integrations
- # that are defined under base platforms right away since we do not require
- # the manifest to list them as dependencies and we want to avoid the lock
- # contention when multiple integrations try to load them at once
- additional_manifests_to_load = {
- *BASE_PLATFORMS,
- *chain.from_iterable(platform_integrations.values()),
- }
-
- translations_to_load = additional_manifests_to_load.copy()
-
- # Resolve all dependencies so we know all integrations
- # that will have to be loaded and start right-away
- integration_cache: dict[str, loader.Integration] = {}
- to_resolve: set[str] = domains_to_setup
- while to_resolve or additional_manifests_to_load:
- old_to_resolve: set[str] = to_resolve
- to_resolve = set()
-
- if additional_manifests_to_load:
- to_get = {*old_to_resolve, *additional_manifests_to_load}
- additional_manifests_to_load.clear()
- else:
- to_get = old_to_resolve
-
- manifest_deps: set[str] = set()
- resolve_dependencies_tasks: list[asyncio.Task[bool]] = []
- integrations_to_process: list[loader.Integration] = []
-
- for domain, itg in (await loader.async_get_integrations(hass, to_get)).items():
- if not isinstance(itg, loader.Integration):
- continue
- integration_cache[domain] = itg
- needed_requirements.update(itg.requirements)
-
- # Make sure manifests for dependencies are loaded in the next
- # loop to try to group as many as manifest loads in a single
- # call to avoid the creating one-off executor jobs later in
- # the setup process
- additional_manifests_to_load.update(
- dep
- for dep in chain(itg.dependencies, itg.after_dependencies)
- if dep not in integration_cache
- )
-
- if domain not in old_to_resolve:
- continue
-
- integrations_to_process.append(itg)
- manifest_deps.update(itg.dependencies)
- manifest_deps.update(itg.after_dependencies)
- if not itg.all_dependencies_resolved:
- resolve_dependencies_tasks.append(
- create_eager_task(
- itg.resolve_dependencies(),
- name=f"resolve dependencies {domain}",
- loop=hass.loop,
- )
- )
-
- if unseen_deps := manifest_deps - integration_cache.keys():
- # If there are dependencies, try to preload all
- # the integrations manifest at once and add them
- # to the list of requirements we need to install
- # so we can try to check if they are already installed
- # in a single call below which avoids each integration
- # having to wait for the lock to do it individually
- deps = await loader.async_get_integrations(hass, unseen_deps)
- for dependant_domain, dependant_itg in deps.items():
- if isinstance(dependant_itg, loader.Integration):
- integration_cache[dependant_domain] = dependant_itg
- needed_requirements.update(dependant_itg.requirements)
-
- if resolve_dependencies_tasks:
- await asyncio.gather(*resolve_dependencies_tasks)
-
- for itg in integrations_to_process:
- try:
- all_deps = itg.all_dependencies
- except RuntimeError:
- # Integration.all_dependencies raises RuntimeError if
- # dependencies could not be resolved
- continue
- for dep in all_deps:
- if dep in domains_to_setup:
- continue
- domains_to_setup.add(dep)
- to_resolve.add(dep)
-
- _LOGGER.info("Domains to be set up: %s", domains_to_setup)
-
- # Optimistically check if requirements are already installed
- # ahead of setting up the integrations so we can prime the cache
- # We do not wait for this since its an optimization only
- hass.async_create_background_task(
- requirements.async_load_installed_versions(hass, needed_requirements),
- "check installed requirements",
- eager_start=True,
- )
-
- #
- # Only add the domains_to_setup after we finish resolving
- # as new domains are likely to added in the process
- #
- translations_to_load.update(domains_to_setup)
- # Start loading translations for all integrations we are going to set up
- # in the background so they are ready when we need them. This avoids a
- # lot of waiting for the translation load lock and a thundering herd of
- # tasks trying to load the same translations at the same time as each
- # integration is loaded.
- #
- # We do not wait for this since as soon as the task runs it will
- # hold the translation load lock and if anything is fast enough to
- # wait for the translation load lock, loading will be done by the
- # time it gets to it.
- hass.async_create_background_task(
- translation.async_load_integrations(hass, translations_to_load),
- "load translations",
- eager_start=True,
- )
-
- # Preload storage for all integrations we are going to set up
- # so we do not have to wait for it to be loaded when we need it
- # in the setup process.
- hass.async_create_background_task(
- get_internal_store_manager(hass).async_preload(
- [*PRELOAD_STORAGE, *domains_to_setup]
- ),
- "preload storage",
- eager_start=True,
- )
-
- return domains_to_setup, integration_cache
-
-
-async def _async_set_up_integrations(
- hass: core.HomeAssistant, config: dict[str, Any]
-) -> None:
- """Set up all the integrations."""
- watcher = _WatchPendingSetups(hass, _setup_started(hass))
- watcher.async_start()
-
- domains_to_setup, integration_cache = await _async_resolve_domains_to_setup(
- hass, config
- )
-
- # Initialize recorder
- if "recorder" in domains_to_setup:
- recorder.async_initialize_recorder(hass)
-
- pre_stage_domains = [
- (name, domains_to_setup & domain_group) for name, domain_group in SETUP_ORDER
- ]
-
- # calculate what components to setup in what stage
- stage_1_domains: set[str] = set()
-
- # Find all dependencies of any dependency of any stage 1 integration that
- # we plan on loading and promote them to stage 1. This is done only to not
- # get misleading log messages
- deps_promotion: set[str] = STAGE_1_INTEGRATIONS
- while deps_promotion:
- old_deps_promotion = deps_promotion
- deps_promotion = set()
-
- for domain in old_deps_promotion:
- if domain not in domains_to_setup or domain in stage_1_domains:
- continue
-
- stage_1_domains.add(domain)
-
- if (dep_itg := integration_cache.get(domain)) is None:
- continue
-
- deps_promotion.update(dep_itg.all_dependencies)
-
- stage_2_domains = domains_to_setup - stage_1_domains
-
- for name, domain_group in pre_stage_domains:
- if domain_group:
- stage_2_domains -= domain_group
- _LOGGER.info("Setting up %s: %s", name, domain_group)
- to_be_loaded = domain_group.copy()
- to_be_loaded.update(
- dep
- for domain in domain_group
- if (integration := integration_cache.get(domain)) is not None
- for dep in integration.all_dependencies
- )
- async_set_domains_to_be_loaded(hass, to_be_loaded)
- await async_setup_multi_components(hass, domain_group, config)
-
- # Enables after dependencies when setting up stage 1 domains
- async_set_domains_to_be_loaded(hass, stage_1_domains)
-
- # Start setup
- if stage_1_domains:
- _LOGGER.info("Setting up stage 1: %s", stage_1_domains)
- try:
- async with hass.timeout.async_timeout(
- STAGE_1_TIMEOUT, cool_down=COOLDOWN_TIME
- ):
- await async_setup_multi_components(hass, stage_1_domains, config)
- except TimeoutError:
- _LOGGER.warning(
- "Setup timed out for stage 1 waiting on %s - moving forward",
- hass._active_tasks, # noqa: SLF001
- )
-
- # Add after dependencies when setting up stage 2 domains
- async_set_domains_to_be_loaded(hass, stage_2_domains)
-
- if stage_2_domains:
- _LOGGER.info("Setting up stage 2: %s", stage_2_domains)
- try:
- async with hass.timeout.async_timeout(
- STAGE_2_TIMEOUT, cool_down=COOLDOWN_TIME
- ):
- await async_setup_multi_components(hass, stage_2_domains, config)
- except TimeoutError:
- _LOGGER.warning(
- "Setup timed out for stage 2 waiting on %s - moving forward",
- hass._active_tasks, # noqa: SLF001
- )
-
- # Wrap up startup
- _LOGGER.debug("Waiting for startup to wrap up")
- try:
- async with hass.timeout.async_timeout(WRAP_UP_TIMEOUT, cool_down=COOLDOWN_TIME):
- await hass.async_block_till_done()
- except TimeoutError:
- _LOGGER.warning(
- "Setup timed out for bootstrap waiting on %s - moving forward",
- hass._active_tasks, # noqa: SLF001
- )
-
- watcher.async_stop()
-
- if _LOGGER.isEnabledFor(logging.DEBUG):
- setup_time = async_get_setup_timings(hass)
- _LOGGER.debug(
- "Integration setup times: %s",
- dict(sorted(setup_time.items(), key=itemgetter(1), reverse=True)),
- )
diff --git a/homeassistant/brands/microsoft.json b/homeassistant/brands/microsoft.json
index 0e00c4a7bc3..918f67f06dd 100644
--- a/homeassistant/brands/microsoft.json
+++ b/homeassistant/brands/microsoft.json
@@ -6,6 +6,7 @@
"azure_devops",
"azure_event_hub",
"azure_service_bus",
+ "azure_storage",
"microsoft_face_detect",
"microsoft_face_identify",
"microsoft_face",
diff --git a/homeassistant/brands/sensorpush.json b/homeassistant/brands/sensorpush.json
new file mode 100644
index 00000000000..b7e528948f8
--- /dev/null
+++ b/homeassistant/brands/sensorpush.json
@@ -0,0 +1,5 @@
+{
+ "domain": "sensorpush",
+ "name": "SensorPush",
+ "integrations": ["sensorpush", "sensorpush_cloud"]
+}
diff --git a/homeassistant/components/abode/alarm_control_panel.py b/homeassistant/components/abode/alarm_control_panel.py
index 4ec59ca4c39..554cf932fca 100644
--- a/homeassistant/components/abode/alarm_control_panel.py
+++ b/homeassistant/components/abode/alarm_control_panel.py
@@ -11,7 +11,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AbodeSystem
from .const import DOMAIN
@@ -19,7 +19,9 @@ from .entity import AbodeDevice
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode alarm control panel device."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/abode/binary_sensor.py b/homeassistant/components/abode/binary_sensor.py
index ca9679a5aaa..6f64fa46c0a 100644
--- a/homeassistant/components/abode/binary_sensor.py
+++ b/homeassistant/components/abode/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.enum import try_parse_enum
from . import AbodeSystem
@@ -21,7 +21,9 @@ from .entity import AbodeDevice
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode binary sensor devices."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/abode/camera.py b/homeassistant/components/abode/camera.py
index 58107f16462..3587c8c1799 100644
--- a/homeassistant/components/abode/camera.py
+++ b/homeassistant/components/abode/camera.py
@@ -15,7 +15,7 @@ from homeassistant.components.camera import Camera
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import Event, HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle
from . import AbodeSystem
@@ -26,7 +26,9 @@ MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=90)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode camera devices."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/abode/cover.py b/homeassistant/components/abode/cover.py
index b5b1e878b96..bc9df9d4a25 100644
--- a/homeassistant/components/abode/cover.py
+++ b/homeassistant/components/abode/cover.py
@@ -7,7 +7,7 @@ from jaraco.abode.devices.cover import Cover
from homeassistant.components.cover import CoverEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AbodeSystem
from .const import DOMAIN
@@ -15,7 +15,9 @@ from .entity import AbodeDevice
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode cover devices."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/abode/light.py b/homeassistant/components/abode/light.py
index e2d0a331f0a..9614e84ebad 100644
--- a/homeassistant/components/abode/light.py
+++ b/homeassistant/components/abode/light.py
@@ -18,7 +18,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AbodeSystem
from .const import DOMAIN
@@ -26,7 +26,9 @@ from .entity import AbodeDevice
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode light devices."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/abode/lock.py b/homeassistant/components/abode/lock.py
index ceff263e6b5..94832ed41f9 100644
--- a/homeassistant/components/abode/lock.py
+++ b/homeassistant/components/abode/lock.py
@@ -7,7 +7,7 @@ from jaraco.abode.devices.lock import Lock
from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AbodeSystem
from .const import DOMAIN
@@ -15,7 +15,9 @@ from .entity import AbodeDevice
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode lock devices."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/abode/sensor.py b/homeassistant/components/abode/sensor.py
index d6a5389029b..ee168c16509 100644
--- a/homeassistant/components/abode/sensor.py
+++ b/homeassistant/components/abode/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import LIGHT_LUX, PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AbodeSystem
from .const import DOMAIN
@@ -61,7 +61,9 @@ SENSOR_TYPES: tuple[AbodeSensorDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode sensor devices."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/abode/switch.py b/homeassistant/components/abode/switch.py
index 7dad750c8d5..f6018d99fc8 100644
--- a/homeassistant/components/abode/switch.py
+++ b/homeassistant/components/abode/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AbodeSystem
from .const import DOMAIN
@@ -20,7 +20,9 @@ DEVICE_TYPES = ["switch", "valve"]
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Abode switch devices."""
data: AbodeSystem = hass.data[DOMAIN]
diff --git a/homeassistant/components/acaia/binary_sensor.py b/homeassistant/components/acaia/binary_sensor.py
index ecb7ac06eb5..d720488faa0 100644
--- a/homeassistant/components/acaia/binary_sensor.py
+++ b/homeassistant/components/acaia/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AcaiaConfigEntry
from .entity import AcaiaEntity
@@ -40,7 +40,7 @@ BINARY_SENSORS: tuple[AcaiaBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AcaiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors."""
diff --git a/homeassistant/components/acaia/button.py b/homeassistant/components/acaia/button.py
index a41233bfc17..446f6134789 100644
--- a/homeassistant/components/acaia/button.py
+++ b/homeassistant/components/acaia/button.py
@@ -8,7 +8,7 @@ from aioacaia.acaiascale import AcaiaScale
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AcaiaConfigEntry
from .entity import AcaiaEntity
@@ -45,7 +45,7 @@ BUTTONS: tuple[AcaiaButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AcaiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button entities and services."""
diff --git a/homeassistant/components/acaia/sensor.py b/homeassistant/components/acaia/sensor.py
index 7ba44958eca..f62b93ddf1d 100644
--- a/homeassistant/components/acaia/sensor.py
+++ b/homeassistant/components/acaia/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfMass, UnitOfVolumeFlowRate
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AcaiaConfigEntry
from .entity import AcaiaEntity
@@ -77,7 +77,7 @@ RESTORE_SENSORS: tuple[AcaiaSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AcaiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
diff --git a/homeassistant/components/accuweather/const.py b/homeassistant/components/accuweather/const.py
index 1bbf5a36187..7216f5a0b9b 100644
--- a/homeassistant/components/accuweather/const.py
+++ b/homeassistant/components/accuweather/const.py
@@ -24,7 +24,7 @@ from homeassistant.components.weather import (
API_METRIC: Final = "Metric"
ATTRIBUTION: Final = "Data provided by AccuWeather"
-ATTR_CATEGORY: Final = "Category"
+ATTR_CATEGORY_VALUE = "CategoryValue"
ATTR_DIRECTION: Final = "Direction"
ATTR_ENGLISH: Final = "English"
ATTR_LEVEL: Final = "level"
@@ -55,5 +55,18 @@ CONDITION_MAP = {
for cond_ha, cond_codes in CONDITION_CLASSES.items()
for cond_code in cond_codes
}
+AIR_QUALITY_CATEGORY_MAP = {
+ 1: "good",
+ 2: "moderate",
+ 3: "unhealthy",
+ 4: "very_unhealthy",
+ 5: "hazardous",
+}
+POLLEN_CATEGORY_MAP = {
+ 1: "low",
+ 2: "moderate",
+ 3: "high",
+ 4: "very_high",
+}
UPDATE_INTERVAL_OBSERVATION = timedelta(minutes=40)
UPDATE_INTERVAL_DAILY_FORECAST = timedelta(hours=6)
diff --git a/homeassistant/components/accuweather/coordinator.py b/homeassistant/components/accuweather/coordinator.py
index 40ff3ad2c87..780c977f930 100644
--- a/homeassistant/components/accuweather/coordinator.py
+++ b/homeassistant/components/accuweather/coordinator.py
@@ -75,7 +75,11 @@ class AccuWeatherObservationDataUpdateCoordinator(
async with timeout(10):
result = await self.accuweather.async_get_current_conditions()
except EXCEPTIONS as error:
- raise UpdateFailed(error) from error
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="current_conditions_update_error",
+ translation_placeholders={"error": repr(error)},
+ ) from error
_LOGGER.debug("Requests remaining: %d", self.accuweather.requests_remaining)
@@ -117,9 +121,15 @@ class AccuWeatherDailyForecastDataUpdateCoordinator(
"""Update data via library."""
try:
async with timeout(10):
- result = await self.accuweather.async_get_daily_forecast()
+ result = await self.accuweather.async_get_daily_forecast(
+ language=self.hass.config.language
+ )
except EXCEPTIONS as error:
- raise UpdateFailed(error) from error
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="forecast_update_error",
+ translation_placeholders={"error": repr(error)},
+ ) from error
_LOGGER.debug("Requests remaining: %d", self.accuweather.requests_remaining)
diff --git a/homeassistant/components/accuweather/manifest.json b/homeassistant/components/accuweather/manifest.json
index 75f4a265b5f..810557519eb 100644
--- a/homeassistant/components/accuweather/manifest.json
+++ b/homeassistant/components/accuweather/manifest.json
@@ -7,6 +7,6 @@
"integration_type": "service",
"iot_class": "cloud_polling",
"loggers": ["accuweather"],
- "requirements": ["accuweather==4.0.0"],
+ "requirements": ["accuweather==4.2.0"],
"single_config_entry": true
}
diff --git a/homeassistant/components/accuweather/sensor.py b/homeassistant/components/accuweather/sensor.py
index 001edc5f197..415df402d55 100644
--- a/homeassistant/components/accuweather/sensor.py
+++ b/homeassistant/components/accuweather/sensor.py
@@ -25,12 +25,13 @@ from homeassistant.const import (
UnitOfVolumetricFlux,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
+ AIR_QUALITY_CATEGORY_MAP,
API_METRIC,
- ATTR_CATEGORY,
+ ATTR_CATEGORY_VALUE,
ATTR_DIRECTION,
ATTR_ENGLISH,
ATTR_LEVEL,
@@ -38,6 +39,7 @@ from .const import (
ATTR_VALUE,
ATTRIBUTION,
MAX_FORECAST_DAYS,
+ POLLEN_CATEGORY_MAP,
)
from .coordinator import (
AccuWeatherConfigEntry,
@@ -59,9 +61,9 @@ class AccuWeatherSensorDescription(SensorEntityDescription):
FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
AccuWeatherSensorDescription(
key="AirQuality",
- value_fn=lambda data: cast(str, data[ATTR_CATEGORY]),
+ value_fn=lambda data: AIR_QUALITY_CATEGORY_MAP[data[ATTR_CATEGORY_VALUE]],
device_class=SensorDeviceClass.ENUM,
- options=["good", "hazardous", "high", "low", "moderate", "unhealthy"],
+ options=list(AIR_QUALITY_CATEGORY_MAP.values()),
translation_key="air_quality",
),
AccuWeatherSensorDescription(
@@ -83,7 +85,9 @@ FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
entity_registry_enabled_default=False,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
value_fn=lambda data: cast(int, data[ATTR_VALUE]),
- attr_fn=lambda data: {ATTR_LEVEL: data[ATTR_CATEGORY]},
+ attr_fn=lambda data: {
+ ATTR_LEVEL: POLLEN_CATEGORY_MAP[data[ATTR_CATEGORY_VALUE]]
+ },
translation_key="grass_pollen",
),
AccuWeatherSensorDescription(
@@ -107,7 +111,9 @@ FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
entity_registry_enabled_default=False,
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
value_fn=lambda data: cast(int, data[ATTR_VALUE]),
- attr_fn=lambda data: {ATTR_LEVEL: data[ATTR_CATEGORY]},
+ attr_fn=lambda data: {
+ ATTR_LEVEL: POLLEN_CATEGORY_MAP[data[ATTR_CATEGORY_VALUE]]
+ },
translation_key="mold_pollen",
),
AccuWeatherSensorDescription(
@@ -115,7 +121,9 @@ FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
entity_registry_enabled_default=False,
value_fn=lambda data: cast(int, data[ATTR_VALUE]),
- attr_fn=lambda data: {ATTR_LEVEL: data[ATTR_CATEGORY]},
+ attr_fn=lambda data: {
+ ATTR_LEVEL: POLLEN_CATEGORY_MAP[data[ATTR_CATEGORY_VALUE]]
+ },
translation_key="ragweed_pollen",
),
AccuWeatherSensorDescription(
@@ -181,14 +189,18 @@ FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER,
entity_registry_enabled_default=False,
value_fn=lambda data: cast(int, data[ATTR_VALUE]),
- attr_fn=lambda data: {ATTR_LEVEL: data[ATTR_CATEGORY]},
+ attr_fn=lambda data: {
+ ATTR_LEVEL: POLLEN_CATEGORY_MAP[data[ATTR_CATEGORY_VALUE]]
+ },
translation_key="tree_pollen",
),
AccuWeatherSensorDescription(
key="UVIndex",
native_unit_of_measurement=UV_INDEX,
value_fn=lambda data: cast(int, data[ATTR_VALUE]),
- attr_fn=lambda data: {ATTR_LEVEL: data[ATTR_CATEGORY]},
+ attr_fn=lambda data: {
+ ATTR_LEVEL: POLLEN_CATEGORY_MAP[data[ATTR_CATEGORY_VALUE]]
+ },
translation_key="uv_index_forecast",
),
AccuWeatherSensorDescription(
@@ -375,7 +387,7 @@ SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AccuWeatherConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add AccuWeather entities from a config_entry."""
observation_coordinator: AccuWeatherObservationDataUpdateCoordinator = (
diff --git a/homeassistant/components/accuweather/strings.json b/homeassistant/components/accuweather/strings.json
index d0250a382e9..e1a71c5e1a5 100644
--- a/homeassistant/components/accuweather/strings.json
+++ b/homeassistant/components/accuweather/strings.json
@@ -26,10 +26,20 @@
"state": {
"good": "Good",
"hazardous": "Hazardous",
- "high": "High",
- "low": "Low",
"moderate": "Moderate",
- "unhealthy": "Unhealthy"
+ "unhealthy": "Unhealthy",
+ "very_unhealthy": "Very unhealthy"
+ },
+ "state_attributes": {
+ "options": {
+ "state": {
+ "good": "[%key:component::accuweather::entity::sensor::air_quality::state::good%]",
+ "hazardous": "[%key:component::accuweather::entity::sensor::air_quality::state::hazardous%]",
+ "moderate": "[%key:component::accuweather::entity::sensor::air_quality::state::moderate%]",
+ "unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::unhealthy%]",
+ "very_unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::very_unhealthy%]"
+ }
+ }
}
},
"apparent_temperature": {
@@ -62,12 +72,10 @@
"level": {
"name": "Level",
"state": {
- "good": "[%key:component::accuweather::entity::sensor::air_quality::state::good%]",
- "hazardous": "[%key:component::accuweather::entity::sensor::air_quality::state::hazardous%]",
- "high": "[%key:component::accuweather::entity::sensor::air_quality::state::high%]",
- "low": "[%key:component::accuweather::entity::sensor::air_quality::state::low%]",
- "moderate": "[%key:component::accuweather::entity::sensor::air_quality::state::moderate%]",
- "unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::unhealthy%]"
+ "high": "High",
+ "low": "Low",
+ "moderate": "Moderate",
+ "very_high": "Very high"
}
}
}
@@ -81,12 +89,10 @@
"level": {
"name": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::name%]",
"state": {
- "good": "[%key:component::accuweather::entity::sensor::air_quality::state::good%]",
- "hazardous": "[%key:component::accuweather::entity::sensor::air_quality::state::hazardous%]",
- "high": "[%key:component::accuweather::entity::sensor::air_quality::state::high%]",
- "low": "[%key:component::accuweather::entity::sensor::air_quality::state::low%]",
- "moderate": "[%key:component::accuweather::entity::sensor::air_quality::state::moderate%]",
- "unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::unhealthy%]"
+ "high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::high%]",
+ "low": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::low%]",
+ "moderate": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::moderate%]",
+ "very_high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::very_high%]"
}
}
}
@@ -100,6 +106,15 @@
"steady": "Steady",
"rising": "Rising",
"falling": "Falling"
+ },
+ "state_attributes": {
+ "options": {
+ "state": {
+ "falling": "[%key:component::accuweather::entity::sensor::pressure_tendency::state::falling%]",
+ "rising": "[%key:component::accuweather::entity::sensor::pressure_tendency::state::rising%]",
+ "steady": "[%key:component::accuweather::entity::sensor::pressure_tendency::state::steady%]"
+ }
+ }
}
},
"ragweed_pollen": {
@@ -108,12 +123,10 @@
"level": {
"name": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::name%]",
"state": {
- "good": "[%key:component::accuweather::entity::sensor::air_quality::state::good%]",
- "hazardous": "[%key:component::accuweather::entity::sensor::air_quality::state::hazardous%]",
- "high": "[%key:component::accuweather::entity::sensor::air_quality::state::high%]",
- "low": "[%key:component::accuweather::entity::sensor::air_quality::state::low%]",
- "moderate": "[%key:component::accuweather::entity::sensor::air_quality::state::moderate%]",
- "unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::unhealthy%]"
+ "high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::high%]",
+ "low": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::low%]",
+ "moderate": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::moderate%]",
+ "very_high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::very_high%]"
}
}
}
@@ -154,12 +167,10 @@
"level": {
"name": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::name%]",
"state": {
- "good": "[%key:component::accuweather::entity::sensor::air_quality::state::good%]",
- "hazardous": "[%key:component::accuweather::entity::sensor::air_quality::state::hazardous%]",
- "high": "[%key:component::accuweather::entity::sensor::air_quality::state::high%]",
- "low": "[%key:component::accuweather::entity::sensor::air_quality::state::low%]",
- "moderate": "[%key:component::accuweather::entity::sensor::air_quality::state::moderate%]",
- "unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::unhealthy%]"
+ "high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::high%]",
+ "low": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::low%]",
+ "moderate": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::moderate%]",
+ "very_high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::very_high%]"
}
}
}
@@ -170,12 +181,10 @@
"level": {
"name": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::name%]",
"state": {
- "good": "[%key:component::accuweather::entity::sensor::air_quality::state::good%]",
- "hazardous": "[%key:component::accuweather::entity::sensor::air_quality::state::hazardous%]",
- "high": "[%key:component::accuweather::entity::sensor::air_quality::state::high%]",
- "low": "[%key:component::accuweather::entity::sensor::air_quality::state::low%]",
- "moderate": "[%key:component::accuweather::entity::sensor::air_quality::state::moderate%]",
- "unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::unhealthy%]"
+ "high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::high%]",
+ "low": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::low%]",
+ "moderate": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::moderate%]",
+ "very_high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::very_high%]"
}
}
}
@@ -186,12 +195,10 @@
"level": {
"name": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::name%]",
"state": {
- "good": "[%key:component::accuweather::entity::sensor::air_quality::state::good%]",
- "hazardous": "[%key:component::accuweather::entity::sensor::air_quality::state::hazardous%]",
- "high": "[%key:component::accuweather::entity::sensor::air_quality::state::high%]",
- "low": "[%key:component::accuweather::entity::sensor::air_quality::state::low%]",
- "moderate": "[%key:component::accuweather::entity::sensor::air_quality::state::moderate%]",
- "unhealthy": "[%key:component::accuweather::entity::sensor::air_quality::state::unhealthy%]"
+ "high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::high%]",
+ "low": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::low%]",
+ "moderate": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::moderate%]",
+ "very_high": "[%key:component::accuweather::entity::sensor::grass_pollen::state_attributes::level::state::very_high%]"
}
}
}
@@ -222,6 +229,14 @@
}
}
},
+ "exceptions": {
+ "current_conditions_update_error": {
+ "message": "An error occurred while retrieving weather current conditions data from the AccuWeather API: {error}"
+ },
+ "forecast_update_error": {
+ "message": "An error occurred while retrieving weather forecast data from the AccuWeather API: {error}"
+ }
+ },
"system_health": {
"info": {
"can_reach_server": "Reach AccuWeather server",
diff --git a/homeassistant/components/accuweather/weather.py b/homeassistant/components/accuweather/weather.py
index 7d754278d91..770f2b64f20 100644
--- a/homeassistant/components/accuweather/weather.py
+++ b/homeassistant/components/accuweather/weather.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utc_from_timestamp
from .const import (
@@ -54,7 +54,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: AccuWeatherConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a AccuWeather weather entity from a config_entry."""
async_add_entities([AccuWeatherEntity(entry.runtime_data)])
diff --git a/homeassistant/components/acmeda/cover.py b/homeassistant/components/acmeda/cover.py
index 77099e86adc..d09ba4bac08 100644
--- a/homeassistant/components/acmeda/cover.py
+++ b/homeassistant/components/acmeda/cover.py
@@ -11,7 +11,7 @@ from homeassistant.components.cover import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AcmedaConfigEntry
from .const import ACMEDA_HUB_UPDATE
@@ -22,7 +22,7 @@ from .helpers import async_add_acmeda_entities
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AcmedaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Acmeda Rollers from a config entry."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/acmeda/helpers.py b/homeassistant/components/acmeda/helpers.py
index 52af7d586de..4c0f9b32cff 100644
--- a/homeassistant/components/acmeda/helpers.py
+++ b/homeassistant/components/acmeda/helpers.py
@@ -9,7 +9,7 @@ from aiopulse import Roller
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, LOGGER
@@ -23,7 +23,7 @@ def async_add_acmeda_entities(
entity_class: type,
config_entry: AcmedaConfigEntry,
current: set[int],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add any new entities."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/acmeda/sensor.py b/homeassistant/components/acmeda/sensor.py
index f5df1bf013d..515146f3d1a 100644
--- a/homeassistant/components/acmeda/sensor.py
+++ b/homeassistant/components/acmeda/sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AcmedaConfigEntry
from .const import ACMEDA_HUB_UPDATE
@@ -17,7 +17,7 @@ from .helpers import async_add_acmeda_entities
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AcmedaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Acmeda Rollers from a config entry."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/adax/climate.py b/homeassistant/components/adax/climate.py
index 15022ba3c9f..078640cd367 100644
--- a/homeassistant/components/adax/climate.py
+++ b/homeassistant/components/adax/climate.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ACCOUNT_ID, CONNECTION_TYPE, DOMAIN, LOCAL
@@ -33,7 +33,7 @@ from .const import ACCOUNT_ID, CONNECTION_TYPE, DOMAIN, LOCAL
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Adax thermostat with config flow."""
if entry.data.get(CONNECTION_TYPE) == LOCAL:
diff --git a/homeassistant/components/adax/strings.json b/homeassistant/components/adax/strings.json
index 6157b7dfc91..9ba497a9aca 100644
--- a/homeassistant/components/adax/strings.json
+++ b/homeassistant/components/adax/strings.json
@@ -5,14 +5,14 @@
"data": {
"connection_type": "Select connection type"
},
- "description": "Select connection type. Local requires heaters with bluetooth"
+ "description": "Select connection type. Local requires heaters with Bluetooth"
},
"local": {
"data": {
"wifi_ssid": "Wi-Fi SSID",
- "wifi_pswd": "Wi-Fi Password"
+ "wifi_pswd": "Wi-Fi password"
},
- "description": "Reset the heater by pressing + and OK until display shows 'Reset'. Then press and hold OK button on the heater until the blue led starts blinking before pressing Submit. Configuring heater might take some minutes."
+ "description": "Reset the heater by pressing + and OK until display shows 'Reset'. Then press and hold OK button on the heater until the blue LED starts blinking before pressing Submit. Configuring heater might take some minutes."
},
"cloud": {
"data": {
diff --git a/homeassistant/components/adguard/__init__.py b/homeassistant/components/adguard/__init__.py
index f8ddeba6767..bbc763d7ec3 100644
--- a/homeassistant/components/adguard/__init__.py
+++ b/homeassistant/components/adguard/__init__.py
@@ -7,7 +7,7 @@ from dataclasses import dataclass
from adguardhome import AdGuardHome, AdGuardHomeConnectionError
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_HOST,
CONF_NAME,
@@ -123,12 +123,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: AdGuardConfigEntry) -> b
async def async_unload_entry(hass: HomeAssistant, entry: AdGuardConfigEntry) -> bool:
"""Unload AdGuard Home config entry."""
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
# This is the last loaded instance of AdGuard, deregister any services
hass.services.async_remove(DOMAIN, SERVICE_ADD_URL)
hass.services.async_remove(DOMAIN, SERVICE_REMOVE_URL)
diff --git a/homeassistant/components/adguard/sensor.py b/homeassistant/components/adguard/sensor.py
index b2404a88278..f1af8ac32a4 100644
--- a/homeassistant/components/adguard/sensor.py
+++ b/homeassistant/components/adguard/sensor.py
@@ -12,7 +12,7 @@ from adguardhome import AdGuardHome
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.const import PERCENTAGE, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdGuardConfigEntry, AdGuardData
from .const import DOMAIN
@@ -85,7 +85,7 @@ SENSORS: tuple[AdGuardHomeEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AdGuardConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdGuard Home sensor based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/adguard/strings.json b/homeassistant/components/adguard/strings.json
index 5b6a5a546f7..44f4a388e6e 100644
--- a/homeassistant/components/adguard/strings.json
+++ b/homeassistant/components/adguard/strings.json
@@ -79,7 +79,7 @@
"services": {
"add_url": {
"name": "Add URL",
- "description": "Add a new filter subscription to AdGuard Home.",
+ "description": "Adds a new filter subscription to AdGuard Home.",
"fields": {
"name": {
"name": "[%key:common::config_flow::data::name%]",
@@ -123,11 +123,11 @@
},
"refresh": {
"name": "Refresh",
- "description": "Refresh all filter subscriptions in AdGuard Home.",
+ "description": "Refreshes all filter subscriptions in AdGuard Home.",
"fields": {
"force": {
"name": "Force",
- "description": "Force update (bypasses AdGuard Home throttling). \"true\" to force, or \"false\" to omit for a regular refresh."
+ "description": "Force update (bypasses AdGuard Home throttling), omit for a regular refresh."
}
}
}
diff --git a/homeassistant/components/adguard/switch.py b/homeassistant/components/adguard/switch.py
index 3ea4f9d1d93..5128102a955 100644
--- a/homeassistant/components/adguard/switch.py
+++ b/homeassistant/components/adguard/switch.py
@@ -11,7 +11,7 @@ from adguardhome import AdGuardHome, AdGuardHomeError
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdGuardConfigEntry, AdGuardData
from .const import DOMAIN, LOGGER
@@ -79,7 +79,7 @@ SWITCHES: tuple[AdGuardHomeSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AdGuardConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdGuard Home switch based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/advantage_air/binary_sensor.py b/homeassistant/components/advantage_air/binary_sensor.py
index 601b10aeb4a..dd306b82c8a 100644
--- a/homeassistant/components/advantage_air/binary_sensor.py
+++ b/homeassistant/components/advantage_air/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .entity import AdvantageAirAcEntity, AdvantageAirZoneEntity
@@ -20,7 +20,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir Binary Sensor platform."""
diff --git a/homeassistant/components/advantage_air/climate.py b/homeassistant/components/advantage_air/climate.py
index d07a3182ed7..1d593c5c3c8 100644
--- a/homeassistant/components/advantage_air/climate.py
+++ b/homeassistant/components/advantage_air/climate.py
@@ -2,6 +2,7 @@
from __future__ import annotations
+from decimal import Decimal
import logging
from typing import Any
@@ -14,12 +15,13 @@ from homeassistant.components.climate import (
FAN_MEDIUM,
ClimateEntity,
ClimateEntityFeature,
+ HVACAction,
HVACMode,
)
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .const import (
@@ -49,6 +51,14 @@ ADVANTAGE_AIR_MYTEMP_ENABLED = "climateControlModeEnabled"
ADVANTAGE_AIR_HEAT_TARGET = "myAutoHeatTargetTemp"
ADVANTAGE_AIR_COOL_TARGET = "myAutoCoolTargetTemp"
ADVANTAGE_AIR_MYFAN = "autoAA"
+ADVANTAGE_AIR_MYAUTO_MODE_SET = "myAutoModeCurrentSetMode"
+
+HVAC_ACTIONS = {
+ "cool": HVACAction.COOLING,
+ "heat": HVACAction.HEATING,
+ "vent": HVACAction.FAN,
+ "dry": HVACAction.DRYING,
+}
HVAC_MODES = [
HVACMode.OFF,
@@ -76,7 +86,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir climate platform."""
@@ -175,6 +185,17 @@ class AdvantageAirAC(AdvantageAirAcEntity, ClimateEntity):
return ADVANTAGE_AIR_HVAC_MODES.get(self._ac["mode"])
return HVACMode.OFF
+ @property
+ def hvac_action(self) -> HVACAction | None:
+ """Return the current running HVAC action."""
+ if self._ac["state"] == ADVANTAGE_AIR_STATE_OFF:
+ return HVACAction.OFF
+ if self._ac["mode"] == "myauto":
+ return HVAC_ACTIONS.get(
+ self._ac.get(ADVANTAGE_AIR_MYAUTO_MODE_SET, HVACAction.OFF)
+ )
+ return HVAC_ACTIONS.get(self._ac["mode"])
+
@property
def fan_mode(self) -> str | None:
"""Return the current fan modes."""
@@ -273,6 +294,22 @@ class AdvantageAirZone(AdvantageAirZoneEntity, ClimateEntity):
return HVACMode.HEAT_COOL
return HVACMode.OFF
+ @property
+ def hvac_action(self) -> HVACAction | None:
+ """Return the HVAC action, inheriting from master AC if zone is open but idle if air is <= 5%."""
+ if self._ac["state"] == ADVANTAGE_AIR_STATE_OFF:
+ return HVACAction.OFF
+ master_action = HVAC_ACTIONS.get(self._ac["mode"], HVACAction.OFF)
+ if self._ac["mode"] == "myauto":
+ master_action = HVAC_ACTIONS.get(
+ str(self._ac.get(ADVANTAGE_AIR_MYAUTO_MODE_SET)), HVACAction.OFF
+ )
+ if self._zone["state"] == ADVANTAGE_AIR_STATE_OPEN:
+ if self._zone["value"] <= Decimal(5):
+ return HVACAction.IDLE
+ return master_action
+ return HVACAction.OFF
+
@property
def current_temperature(self) -> float | None:
"""Return the current temperature."""
diff --git a/homeassistant/components/advantage_air/const.py b/homeassistant/components/advantage_air/const.py
index 6ae0a0e06d5..103ca57f6ef 100644
--- a/homeassistant/components/advantage_air/const.py
+++ b/homeassistant/components/advantage_air/const.py
@@ -7,3 +7,4 @@ ADVANTAGE_AIR_STATE_CLOSE = "close"
ADVANTAGE_AIR_STATE_ON = "on"
ADVANTAGE_AIR_STATE_OFF = "off"
ADVANTAGE_AIR_AUTOFAN_ENABLED = "aaAutoFanModeEnabled"
+ADVANTAGE_AIR_NIGHT_MODE_ENABLED = "quietNightModeEnabled"
diff --git a/homeassistant/components/advantage_air/cover.py b/homeassistant/components/advantage_air/cover.py
index b091f0077a1..e764d484128 100644
--- a/homeassistant/components/advantage_air/cover.py
+++ b/homeassistant/components/advantage_air/cover.py
@@ -9,7 +9,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .const import ADVANTAGE_AIR_STATE_CLOSE, ADVANTAGE_AIR_STATE_OPEN
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir cover platform."""
@@ -41,7 +41,7 @@ async def async_setup_entry(
entities.append(
AdvantageAirThingCover(instance, thing, CoverDeviceClass.BLIND)
)
- elif thing["channelDipState"] == 3: # 3 = "Garage door"
+ elif thing["channelDipState"] in [3, 10]: # 3 & 10 = "Garage door"
entities.append(
AdvantageAirThingCover(instance, thing, CoverDeviceClass.GARAGE)
)
diff --git a/homeassistant/components/advantage_air/light.py b/homeassistant/components/advantage_air/light.py
index 7dd0a0a183b..ffd502663b0 100644
--- a/homeassistant/components/advantage_air/light.py
+++ b/homeassistant/components/advantage_air/light.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .const import ADVANTAGE_AIR_STATE_ON, DOMAIN as ADVANTAGE_AIR_DOMAIN
@@ -16,7 +16,7 @@ from .models import AdvantageAirData
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir light platform."""
diff --git a/homeassistant/components/advantage_air/select.py b/homeassistant/components/advantage_air/select.py
index 84c37f38d7f..320bfd35aba 100644
--- a/homeassistant/components/advantage_air/select.py
+++ b/homeassistant/components/advantage_air/select.py
@@ -2,7 +2,7 @@
from homeassistant.components.select import SelectEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .entity import AdvantageAirAcEntity
@@ -14,7 +14,7 @@ ADVANTAGE_AIR_INACTIVE = "Inactive"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir select platform."""
diff --git a/homeassistant/components/advantage_air/sensor.py b/homeassistant/components/advantage_air/sensor.py
index ab1a1c4f9a0..2abcc3b5a68 100644
--- a/homeassistant/components/advantage_air/sensor.py
+++ b/homeassistant/components/advantage_air/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .const import ADVANTAGE_AIR_STATE_OPEN
@@ -32,7 +32,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir sensor platform."""
diff --git a/homeassistant/components/advantage_air/switch.py b/homeassistant/components/advantage_air/switch.py
index 876875a2510..8560c9a9138 100644
--- a/homeassistant/components/advantage_air/switch.py
+++ b/homeassistant/components/advantage_air/switch.py
@@ -4,11 +4,12 @@ from typing import Any
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .const import (
ADVANTAGE_AIR_AUTOFAN_ENABLED,
+ ADVANTAGE_AIR_NIGHT_MODE_ENABLED,
ADVANTAGE_AIR_STATE_OFF,
ADVANTAGE_AIR_STATE_ON,
)
@@ -19,7 +20,7 @@ from .models import AdvantageAirData
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir switch platform."""
@@ -32,6 +33,8 @@ async def async_setup_entry(
entities.append(AdvantageAirFreshAir(instance, ac_key))
if ADVANTAGE_AIR_AUTOFAN_ENABLED in ac_device["info"]:
entities.append(AdvantageAirMyFan(instance, ac_key))
+ if ADVANTAGE_AIR_NIGHT_MODE_ENABLED in ac_device["info"]:
+ entities.append(AdvantageAirNightMode(instance, ac_key))
if things := instance.coordinator.data.get("myThings"):
entities.extend(
AdvantageAirRelay(instance, thing)
@@ -93,6 +96,32 @@ class AdvantageAirMyFan(AdvantageAirAcEntity, SwitchEntity):
await self.async_update_ac({ADVANTAGE_AIR_AUTOFAN_ENABLED: False})
+class AdvantageAirNightMode(AdvantageAirAcEntity, SwitchEntity):
+ """Representation of Advantage 'MySleep$aver' Mode control."""
+
+ _attr_icon = "mdi:weather-night"
+ _attr_name = "MySleep$aver"
+ _attr_device_class = SwitchDeviceClass.SWITCH
+
+ def __init__(self, instance: AdvantageAirData, ac_key: str) -> None:
+ """Initialize an Advantage Air Night Mode control."""
+ super().__init__(instance, ac_key)
+ self._attr_unique_id += "-nightmode"
+
+ @property
+ def is_on(self) -> bool:
+ """Return the Night Mode status."""
+ return self._ac[ADVANTAGE_AIR_NIGHT_MODE_ENABLED]
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn Night Mode on."""
+ await self.async_update_ac({ADVANTAGE_AIR_NIGHT_MODE_ENABLED: True})
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn Night Mode off."""
+ await self.async_update_ac({ADVANTAGE_AIR_NIGHT_MODE_ENABLED: False})
+
+
class AdvantageAirRelay(AdvantageAirThingEntity, SwitchEntity):
"""Representation of Advantage Air Thing."""
diff --git a/homeassistant/components/advantage_air/update.py b/homeassistant/components/advantage_air/update.py
index b639e4df867..92a162303dd 100644
--- a/homeassistant/components/advantage_air/update.py
+++ b/homeassistant/components/advantage_air/update.py
@@ -3,7 +3,7 @@
from homeassistant.components.update import UpdateEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AdvantageAirDataConfigEntry
from .const import DOMAIN as ADVANTAGE_AIR_DOMAIN
@@ -14,7 +14,7 @@ from .models import AdvantageAirData
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AdvantageAirDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AdvantageAir update platform."""
diff --git a/homeassistant/components/aemet/image.py b/homeassistant/components/aemet/image.py
index ffc53022e4c..ba9986a5ccc 100644
--- a/homeassistant/components/aemet/image.py
+++ b/homeassistant/components/aemet/image.py
@@ -9,7 +9,7 @@ from aemet_opendata.helpers import dict_nested_value
from homeassistant.components.image import Image, ImageEntity, ImageEntityDescription
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AemetConfigEntry, WeatherUpdateCoordinator
from .entity import AemetEntity
@@ -25,7 +25,7 @@ AEMET_IMAGES: Final[tuple[ImageEntityDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AemetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AEMET OpenData image entities based on a config entry."""
domain_data = config_entry.runtime_data
diff --git a/homeassistant/components/aemet/sensor.py b/homeassistant/components/aemet/sensor.py
index 88eb34b6f84..9077b2bc44d 100644
--- a/homeassistant/components/aemet/sensor.py
+++ b/homeassistant/components/aemet/sensor.py
@@ -52,7 +52,7 @@ from homeassistant.const import (
UnitOfVolumetricFlux,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import (
@@ -358,7 +358,7 @@ WEATHER_SENSORS: Final[tuple[AemetSensorEntityDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AemetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AEMET OpenData sensor entities based on a config entry."""
domain_data = config_entry.runtime_data
diff --git a/homeassistant/components/aemet/weather.py b/homeassistant/components/aemet/weather.py
index a156652eadd..3a17430300d 100644
--- a/homeassistant/components/aemet/weather.py
+++ b/homeassistant/components/aemet/weather.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONDITIONS_MAP
from .coordinator import AemetConfigEntry, WeatherUpdateCoordinator
@@ -35,7 +35,7 @@ from .entity import AemetEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AemetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AEMET OpenData weather entity based on a config entry."""
domain_data = config_entry.runtime_data
diff --git a/homeassistant/components/aftership/sensor.py b/homeassistant/components/aftership/sensor.py
index 085be2499d4..7e0c6f524ab 100644
--- a/homeassistant/components/aftership/sensor.py
+++ b/homeassistant/components/aftership/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle
from . import AfterShipConfigEntry
@@ -42,7 +42,7 @@ PLATFORM_SCHEMA: Final = cv.removed(DOMAIN, raise_if_present=False)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AfterShipConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AfterShip sensor entities based on a config entry."""
aftership = config_entry.runtime_data
diff --git a/homeassistant/components/aftership/strings.json b/homeassistant/components/aftership/strings.json
index ace8eb6d2d3..c3817a0cd24 100644
--- a/homeassistant/components/aftership/strings.json
+++ b/homeassistant/components/aftership/strings.json
@@ -51,7 +51,7 @@
"issues": {
"deprecated_yaml_import_issue_cannot_connect": {
"title": "The {integration_title} YAML configuration import failed",
- "description": "Configuring {integration_title} using YAML is being removed but there was an connection error importing your YAML configuration.\n\nEnsure connection to {integration_title} works and restart Home Assistant to try again or remove the {integration_title} YAML configuration from your configuration.yaml file and continue to [set up the integration]({url}) manually."
+ "description": "Configuring {integration_title} using YAML is being removed but there was a connection error importing your YAML configuration.\n\nEnsure connection to {integration_title} works and restart Home Assistant to try again or remove the {integration_title} YAML configuration from your configuration.yaml file and continue to [set up the integration]({url}) manually."
}
}
}
diff --git a/homeassistant/components/agent_dvr/alarm_control_panel.py b/homeassistant/components/agent_dvr/alarm_control_panel.py
index 23328315e42..1ac808c87ad 100644
--- a/homeassistant/components/agent_dvr/alarm_control_panel.py
+++ b/homeassistant/components/agent_dvr/alarm_control_panel.py
@@ -9,7 +9,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AgentDVRConfigEntry
from .const import DOMAIN as AGENT_DOMAIN
@@ -24,7 +24,7 @@ CONST_ALARM_CONTROL_PANEL_NAME = "Alarm Panel"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AgentDVRConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Agent DVR Alarm Control Panels."""
async_add_entities([AgentBaseStation(config_entry.runtime_data)])
diff --git a/homeassistant/components/agent_dvr/camera.py b/homeassistant/components/agent_dvr/camera.py
index 933d0c6b40b..3de7f095b13 100644
--- a/homeassistant/components/agent_dvr/camera.py
+++ b/homeassistant/components/agent_dvr/camera.py
@@ -10,7 +10,7 @@ from homeassistant.components.mjpeg import MjpegCamera, filter_urllib3_logging
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
@@ -39,7 +39,7 @@ CAMERA_SERVICES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AgentDVRConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Agent cameras."""
filter_urllib3_logging()
diff --git a/homeassistant/components/airgradient/button.py b/homeassistant/components/airgradient/button.py
index ea7b12062e8..5e6f857f686 100644
--- a/homeassistant/components/airgradient/button.py
+++ b/homeassistant/components/airgradient/button.py
@@ -13,7 +13,7 @@ from homeassistant.components.button import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AirGradientConfigEntry
from .const import DOMAIN
@@ -47,7 +47,7 @@ LED_BAR_TEST = AirGradientButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: AirGradientConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirGradient button entities based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airgradient/manifest.json b/homeassistant/components/airgradient/manifest.json
index 13764142697..afaf2698ced 100644
--- a/homeassistant/components/airgradient/manifest.json
+++ b/homeassistant/components/airgradient/manifest.json
@@ -6,6 +6,6 @@
"documentation": "https://www.home-assistant.io/integrations/airgradient",
"integration_type": "device",
"iot_class": "local_polling",
- "requirements": ["airgradient==0.9.1"],
+ "requirements": ["airgradient==0.9.2"],
"zeroconf": ["_airgradient._tcp.local."]
}
diff --git a/homeassistant/components/airgradient/number.py b/homeassistant/components/airgradient/number.py
index 4265215fa25..eeb24867845 100644
--- a/homeassistant/components/airgradient/number.py
+++ b/homeassistant/components/airgradient/number.py
@@ -14,7 +14,7 @@ from homeassistant.components.number import (
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AirGradientConfigEntry
from .const import DOMAIN
@@ -60,7 +60,7 @@ LED_BAR_BRIGHTNESS = AirGradientNumberEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: AirGradientConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirGradient number entities based on a config entry."""
diff --git a/homeassistant/components/airgradient/select.py b/homeassistant/components/airgradient/select.py
index 8c15102ad3a..f288055ebf4 100644
--- a/homeassistant/components/airgradient/select.py
+++ b/homeassistant/components/airgradient/select.py
@@ -14,7 +14,7 @@ from homeassistant.components.select import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AirGradientConfigEntry
from .const import DOMAIN, PM_STANDARD, PM_STANDARD_REVERSE
@@ -142,7 +142,7 @@ CONTROL_ENTITIES: tuple[AirGradientSelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirGradientConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirGradient select entities based on a config entry."""
diff --git a/homeassistant/components/airgradient/sensor.py b/homeassistant/components/airgradient/sensor.py
index 3b20b31f923..a4944a6196e 100644
--- a/homeassistant/components/airgradient/sensor.py
+++ b/homeassistant/components/airgradient/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import AirGradientConfigEntry
@@ -225,7 +225,7 @@ CONFIG_DISPLAY_SENSOR_TYPES: tuple[AirGradientConfigSensorEntityDescription, ...
async def async_setup_entry(
hass: HomeAssistant,
entry: AirGradientConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirGradient sensor entities based on a config entry."""
diff --git a/homeassistant/components/airgradient/strings.json b/homeassistant/components/airgradient/strings.json
index 4cf3a6a34ea..2d9b6be529d 100644
--- a/homeassistant/components/airgradient/strings.json
+++ b/homeassistant/components/airgradient/strings.json
@@ -11,7 +11,7 @@
}
},
"discovery_confirm": {
- "description": "Do you want to setup {model}?"
+ "description": "Do you want to set up {model}?"
}
},
"abort": {
diff --git a/homeassistant/components/airgradient/switch.py b/homeassistant/components/airgradient/switch.py
index 55835fa30a6..0d404ed0f15 100644
--- a/homeassistant/components/airgradient/switch.py
+++ b/homeassistant/components/airgradient/switch.py
@@ -15,7 +15,7 @@ from homeassistant.components.switch import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AirGradientConfigEntry
from .const import DOMAIN
@@ -45,7 +45,7 @@ POST_DATA_TO_AIRGRADIENT = AirGradientSwitchEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: AirGradientConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirGradient switch entities based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airgradient/update.py b/homeassistant/components/airgradient/update.py
index 12cec65f791..97cb8576e79 100644
--- a/homeassistant/components/airgradient/update.py
+++ b/homeassistant/components/airgradient/update.py
@@ -6,7 +6,7 @@ from propcache.api import cached_property
from homeassistant.components.update import UpdateDeviceClass, UpdateEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AirGradientConfigEntry, AirGradientCoordinator
from .entity import AirGradientEntity
@@ -18,7 +18,7 @@ SCAN_INTERVAL = timedelta(hours=1)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AirGradientConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Airgradient update platform."""
diff --git a/homeassistant/components/airly/coordinator.py b/homeassistant/components/airly/coordinator.py
index b255c5f078f..668cabdae63 100644
--- a/homeassistant/components/airly/coordinator.py
+++ b/homeassistant/components/airly/coordinator.py
@@ -105,7 +105,14 @@ class AirlyDataUpdateCoordinator(DataUpdateCoordinator[dict[str, str | float | i
try:
await measurements.update()
except (AirlyError, ClientConnectorError) as error:
- raise UpdateFailed(error) from error
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="update_error",
+ translation_placeholders={
+ "entry": self.config_entry.title,
+ "error": repr(error),
+ },
+ ) from error
_LOGGER.debug(
"Requests remaining: %s/%s",
@@ -126,7 +133,11 @@ class AirlyDataUpdateCoordinator(DataUpdateCoordinator[dict[str, str | float | i
standards = measurements.current["standards"]
if index["description"] == NO_AIRLY_SENSORS:
- raise UpdateFailed("Can't retrieve data: no Airly sensors in this area")
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="no_station",
+ translation_placeholders={"entry": self.config_entry.title},
+ )
for value in values:
data[value["name"]] = value["value"]
for standard in standards:
diff --git a/homeassistant/components/airly/sensor.py b/homeassistant/components/airly/sensor.py
index fbf73ed753e..2aa99d9c792 100644
--- a/homeassistant/components/airly/sensor.py
+++ b/homeassistant/components/airly/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -175,7 +175,7 @@ SENSOR_TYPES: tuple[AirlySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirlyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Airly sensor entities based on a config entry."""
name = entry.data[CONF_NAME]
diff --git a/homeassistant/components/airly/strings.json b/homeassistant/components/airly/strings.json
index 33ee8bbe4c9..fe4ccbb4745 100644
--- a/homeassistant/components/airly/strings.json
+++ b/homeassistant/components/airly/strings.json
@@ -36,5 +36,13 @@
"name": "[%key:component::sensor::entity_component::carbon_monoxide::name%]"
}
}
+ },
+ "exceptions": {
+ "update_error": {
+ "message": "An error occurred while retrieving data from the Airly API for {entry}: {error}"
+ },
+ "no_station": {
+ "message": "An error occurred while retrieving data from the Airly API for {entry}: no measuring stations in this area"
+ }
}
}
diff --git a/homeassistant/components/airnow/sensor.py b/homeassistant/components/airnow/sensor.py
index c8b1c985c8b..db579de4976 100644
--- a/homeassistant/components/airnow/sensor.py
+++ b/homeassistant/components/airnow/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -130,7 +130,7 @@ SENSOR_TYPES: tuple[AirNowEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AirNowConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirNow sensor entities based on a config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/airnow/strings.json b/homeassistant/components/airnow/strings.json
index d5fb22106f9..a69f67948cb 100644
--- a/homeassistant/components/airnow/strings.json
+++ b/homeassistant/components/airnow/strings.json
@@ -7,7 +7,7 @@
"api_key": "[%key:common::config_flow::data::api_key%]",
"latitude": "[%key:common::config_flow::data::latitude%]",
"longitude": "[%key:common::config_flow::data::longitude%]",
- "radius": "Station Radius (miles; optional)"
+ "radius": "Station radius (miles; optional)"
}
}
},
@@ -25,7 +25,7 @@
"step": {
"init": {
"data": {
- "radius": "Station Radius (miles)"
+ "radius": "Station radius (miles)"
}
}
}
diff --git a/homeassistant/components/airq/config_flow.py b/homeassistant/components/airq/config_flow.py
index 0c57b399b1b..f87b73b5283 100644
--- a/homeassistant/components/airq/config_flow.py
+++ b/homeassistant/components/airq/config_flow.py
@@ -83,6 +83,7 @@ class AirQConfigFlow(ConfigFlow, domain=DOMAIN):
await self.async_set_unique_id(device_info["id"])
self._abort_if_unique_id_configured()
+ _LOGGER.debug("Creating an entry for %s", device_info["name"])
return self.async_create_entry(title=device_info["name"], data=user_input)
return self.async_show_form(
diff --git a/homeassistant/components/airq/coordinator.py b/homeassistant/components/airq/coordinator.py
index 362b65b5828..743d12d40e5 100644
--- a/homeassistant/components/airq/coordinator.py
+++ b/homeassistant/components/airq/coordinator.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from datetime import timedelta
import logging
-from aioairq import AirQ
+from aioairq.core import AirQ, identify_warming_up_sensors
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_IP_ADDRESS, CONF_PASSWORD
@@ -22,6 +22,8 @@ _LOGGER = logging.getLogger(__name__)
class AirQCoordinator(DataUpdateCoordinator):
"""Coordinator is responsible for querying the device at a specified route."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
@@ -33,6 +35,7 @@ class AirQCoordinator(DataUpdateCoordinator):
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=DOMAIN,
update_interval=timedelta(seconds=UPDATE_INTERVAL),
)
@@ -52,6 +55,9 @@ class AirQCoordinator(DataUpdateCoordinator):
async def _async_update_data(self) -> dict:
"""Fetch the data from the device."""
if "name" not in self.device_info:
+ _LOGGER.debug(
+ "'name' not found in AirQCoordinator.device_info, fetching from the device"
+ )
info = await self.airq.fetch_device_info()
self.device_info.update(
DeviceInfo(
@@ -61,7 +67,16 @@ class AirQCoordinator(DataUpdateCoordinator):
hw_version=info["hw_version"],
)
)
- return await self.airq.get_latest_data( # type: ignore[no-any-return]
+ _LOGGER.debug(
+ "Updated AirQCoordinator.device_info for 'name' %s",
+ self.device_info.get("name"),
+ )
+ data: dict = await self.airq.get_latest_data(
return_average=self.return_average,
clip_negative_values=self.clip_negative,
)
+ if warming_up_sensors := identify_warming_up_sensors(data):
+ _LOGGER.debug(
+ "Following sensors are still warming up: %s", warming_up_sensors
+ )
+ return data
diff --git a/homeassistant/components/airq/sensor.py b/homeassistant/components/airq/sensor.py
index c465d710406..08a344ae9f4 100644
--- a/homeassistant/components/airq/sensor.py
+++ b/homeassistant/components/airq/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import AirQConfigEntry, AirQCoordinator
@@ -399,7 +399,7 @@ SENSOR_TYPES: list[AirQEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: AirQConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor entities based on a config entry."""
diff --git a/homeassistant/components/airq/strings.json b/homeassistant/components/airq/strings.json
index 26b944467e6..9c16975a3ab 100644
--- a/homeassistant/components/airq/strings.json
+++ b/homeassistant/components/airq/strings.json
@@ -91,7 +91,7 @@
"name": "Hydrogen fluoride"
},
"health_index": {
- "name": "Health Index"
+ "name": "Health index"
},
"absolute_humidity": {
"name": "Absolute humidity"
@@ -112,10 +112,10 @@
"name": "Oxygen"
},
"performance_index": {
- "name": "Performance Index"
+ "name": "Performance index"
},
"hydrogen_phosphide": {
- "name": "Hydrogen Phosphide"
+ "name": "Hydrogen phosphide"
},
"relative_pressure": {
"name": "Relative pressure"
@@ -127,22 +127,22 @@
"name": "Refrigerant"
},
"silicon_hydride": {
- "name": "Silicon Hydride"
+ "name": "Silicon hydride"
},
"noise": {
"name": "Noise"
},
"maximum_noise": {
- "name": "Noise (Maximum)"
+ "name": "Noise (maximum)"
},
"radon": {
"name": "Radon"
},
"industrial_volatile_organic_compounds": {
- "name": "VOCs (Industrial)"
+ "name": "VOCs (industrial)"
},
"virus_index": {
- "name": "Virus Index"
+ "name": "Virus index"
}
}
}
diff --git a/homeassistant/components/airthings/sensor.py b/homeassistant/components/airthings/sensor.py
index 1b604d72032..a0d9c97c8c8 100644
--- a/homeassistant/components/airthings/sensor.py
+++ b/homeassistant/components/airthings/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -114,7 +114,7 @@ SENSORS: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: AirthingsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Airthings sensor."""
diff --git a/homeassistant/components/airthings_ble/sensor.py b/homeassistant/components/airthings_ble/sensor.py
index 248561706a3..9c1a2af7a9f 100644
--- a/homeassistant/components/airthings_ble/sensor.py
+++ b/homeassistant/components/airthings_ble/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.entity_registry import (
RegistryEntry,
async_entries_for_device,
@@ -153,7 +153,7 @@ def async_migrate(hass: HomeAssistant, address: str, sensor_name: str) -> None:
async def async_setup_entry(
hass: HomeAssistant,
entry: AirthingsBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Airthings BLE sensors."""
is_metric = hass.config.units is METRIC_SYSTEM
diff --git a/homeassistant/components/airtouch4/climate.py b/homeassistant/components/airtouch4/climate.py
index 0af920bd7a9..6d393ed0c99 100644
--- a/homeassistant/components/airtouch4/climate.py
+++ b/homeassistant/components/airtouch4/climate.py
@@ -19,7 +19,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import AirTouch4ConfigEntry
@@ -64,7 +64,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AirTouch4ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Airtouch 4."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/airtouch5/climate.py b/homeassistant/components/airtouch5/climate.py
index 16566f5d664..f3b914bf341 100644
--- a/homeassistant/components/airtouch5/climate.py
+++ b/homeassistant/components/airtouch5/climate.py
@@ -37,7 +37,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Airtouch5ConfigEntry
from .const import DOMAIN, FAN_INTELLIGENT_AUTO, FAN_TURBO
@@ -93,7 +93,7 @@ FAN_MODE_TO_SET_AC_FAN_SPEED = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: Airtouch5ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Airtouch 5 Climate entities."""
client = config_entry.runtime_data
diff --git a/homeassistant/components/airtouch5/cover.py b/homeassistant/components/airtouch5/cover.py
index 62cf7938fc2..b811ed5c451 100644
--- a/homeassistant/components/airtouch5/cover.py
+++ b/homeassistant/components/airtouch5/cover.py
@@ -20,7 +20,7 @@ from homeassistant.components.cover import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Airtouch5ConfigEntry
from .const import DOMAIN
@@ -32,7 +32,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: Airtouch5ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Airtouch 5 Cover entities."""
client = config_entry.runtime_data
diff --git a/homeassistant/components/airvisual/sensor.py b/homeassistant/components/airvisual/sensor.py
index 88a670edb82..1f406bd8f36 100644
--- a/homeassistant/components/airvisual/sensor.py
+++ b/homeassistant/components/airvisual/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
CONF_STATE,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from . import AirVisualConfigEntry
@@ -108,7 +108,7 @@ POLLUTANT_UNITS = {
async def async_setup_entry(
hass: HomeAssistant,
entry: AirVisualConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirVisual sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airvisual_pro/sensor.py b/homeassistant/components/airvisual_pro/sensor.py
index 58ad730bc31..215370736fe 100644
--- a/homeassistant/components/airvisual_pro/sensor.py
+++ b/homeassistant/components/airvisual_pro/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AirVisualProConfigEntry
from .entity import AirVisualProEntity
@@ -130,7 +130,7 @@ def async_get_aqi_locale(settings: dict[str, Any]) -> str:
async def async_setup_entry(
hass: HomeAssistant,
entry: AirVisualProConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AirVisual sensors based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/airzone/binary_sensor.py b/homeassistant/components/airzone/binary_sensor.py
index 48f6ce8fd94..7274df44261 100644
--- a/homeassistant/components/airzone/binary_sensor.py
+++ b/homeassistant/components/airzone/binary_sensor.py
@@ -23,7 +23,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AirzoneConfigEntry, AirzoneUpdateCoordinator
from .entity import AirzoneEntity, AirzoneSystemEntity, AirzoneZoneEntity
@@ -76,7 +76,7 @@ ZONE_BINARY_SENSOR_TYPES: Final[tuple[AirzoneBinarySensorEntityDescription, ...]
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone binary sensors from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone/climate.py b/homeassistant/components/airzone/climate.py
index 23355a070ab..39e70e58e6d 100644
--- a/homeassistant/components/airzone/climate.py
+++ b/homeassistant/components/airzone/climate.py
@@ -48,7 +48,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import API_TEMPERATURE_STEP, TEMP_UNIT_LIB_TO_HASS
from .coordinator import AirzoneConfigEntry, AirzoneUpdateCoordinator
@@ -100,7 +100,7 @@ HVAC_MODE_HASS_TO_LIB: Final[dict[HVACMode, OperationMode]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone climate from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone/select.py b/homeassistant/components/airzone/select.py
index 56a9b06ea21..c00e83f2c5b 100644
--- a/homeassistant/components/airzone/select.py
+++ b/homeassistant/components/airzone/select.py
@@ -25,7 +25,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AirzoneConfigEntry, AirzoneUpdateCoordinator
from .entity import AirzoneEntity, AirzoneZoneEntity
@@ -117,7 +117,7 @@ ZONE_SELECT_TYPES: Final[tuple[AirzoneSelectDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone select from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone/sensor.py b/homeassistant/components/airzone/sensor.py
index 0b5c5666c89..f76eb1466a3 100644
--- a/homeassistant/components/airzone/sensor.py
+++ b/homeassistant/components/airzone/sensor.py
@@ -28,7 +28,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import TEMP_UNIT_LIB_TO_HASS
from .coordinator import AirzoneConfigEntry, AirzoneUpdateCoordinator
@@ -79,7 +79,7 @@ ZONE_SENSOR_TYPES: Final[tuple[SensorEntityDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone sensors from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone/switch.py b/homeassistant/components/airzone/switch.py
index 69bf33666a5..07278970e03 100644
--- a/homeassistant/components/airzone/switch.py
+++ b/homeassistant/components/airzone/switch.py
@@ -14,7 +14,7 @@ from homeassistant.components.switch import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AirzoneConfigEntry, AirzoneUpdateCoordinator
from .entity import AirzoneEntity, AirzoneZoneEntity
@@ -39,7 +39,7 @@ ZONE_SWITCH_TYPES: Final[tuple[AirzoneSwitchDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone switch from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone/water_heater.py b/homeassistant/components/airzone/water_heater.py
index 2a1ca72db21..eb1537dc222 100644
--- a/homeassistant/components/airzone/water_heater.py
+++ b/homeassistant/components/airzone/water_heater.py
@@ -28,7 +28,7 @@ from homeassistant.components.water_heater import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, STATE_OFF
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import TEMP_UNIT_LIB_TO_HASS
from .coordinator import AirzoneConfigEntry, AirzoneUpdateCoordinator
@@ -58,7 +58,7 @@ OPERATION_MODE_TO_DHW_PARAMS: Final[dict[str, dict[str, Any]]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone Water Heater from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone_cloud/__init__.py b/homeassistant/components/airzone_cloud/__init__.py
index 5baa0bcea10..a5a29263140 100644
--- a/homeassistant/components/airzone_cloud/__init__.py
+++ b/homeassistant/components/airzone_cloud/__init__.py
@@ -5,12 +5,11 @@ from __future__ import annotations
from aioairzone_cloud.cloudapi import AirzoneCloudApi
from aioairzone_cloud.common import ConnectionOptions
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import aiohttp_client
-from .coordinator import AirzoneUpdateCoordinator
+from .coordinator import AirzoneCloudConfigEntry, AirzoneUpdateCoordinator
PLATFORMS: list[Platform] = [
Platform.BINARY_SENSOR,
@@ -21,8 +20,6 @@ PLATFORMS: list[Platform] = [
Platform.WATER_HEATER,
]
-type AirzoneCloudConfigEntry = ConfigEntry[AirzoneUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, entry: AirzoneCloudConfigEntry
@@ -42,7 +39,7 @@ async def async_setup_entry(
airzone.select_installation(inst)
await airzone.update_installation(inst)
- coordinator = AirzoneUpdateCoordinator(hass, airzone)
+ coordinator = AirzoneUpdateCoordinator(hass, entry, airzone)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/airzone_cloud/binary_sensor.py b/homeassistant/components/airzone_cloud/binary_sensor.py
index 3d6f6b42901..64fa8cb5151 100644
--- a/homeassistant/components/airzone_cloud/binary_sensor.py
+++ b/homeassistant/components/airzone_cloud/binary_sensor.py
@@ -26,10 +26,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AirzoneCloudConfigEntry
-from .coordinator import AirzoneUpdateCoordinator
+from .coordinator import AirzoneCloudConfigEntry, AirzoneUpdateCoordinator
from .entity import (
AirzoneAidooEntity,
AirzoneEntity,
@@ -112,7 +111,7 @@ ZONE_BINARY_SENSOR_TYPES: Final[tuple[AirzoneBinarySensorEntityDescription, ...]
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone Cloud binary sensors from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone_cloud/climate.py b/homeassistant/components/airzone_cloud/climate.py
index b98473072e4..115f6e32dbf 100644
--- a/homeassistant/components/airzone_cloud/climate.py
+++ b/homeassistant/components/airzone_cloud/climate.py
@@ -56,10 +56,9 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AirzoneCloudConfigEntry
-from .coordinator import AirzoneUpdateCoordinator
+from .coordinator import AirzoneCloudConfigEntry, AirzoneUpdateCoordinator
from .entity import (
AirzoneAidooEntity,
AirzoneEntity,
@@ -120,7 +119,7 @@ HVAC_MODE_HASS_TO_LIB: Final[dict[HVACMode, OperationMode]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone climate from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone_cloud/coordinator.py b/homeassistant/components/airzone_cloud/coordinator.py
index e510dcfb401..840bfec0d1b 100644
--- a/homeassistant/components/airzone_cloud/coordinator.py
+++ b/homeassistant/components/airzone_cloud/coordinator.py
@@ -10,6 +10,7 @@ from typing import Any
from aioairzone_cloud.cloudapi import AirzoneCloudApi
from aioairzone_cloud.exceptions import AirzoneCloudError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -19,11 +20,20 @@ SCAN_INTERVAL = timedelta(seconds=60)
_LOGGER = logging.getLogger(__name__)
+type AirzoneCloudConfigEntry = ConfigEntry[AirzoneUpdateCoordinator]
+
class AirzoneUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching data from the Airzone Cloud device."""
- def __init__(self, hass: HomeAssistant, airzone: AirzoneCloudApi) -> None:
+ config_entry: AirzoneCloudConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: AirzoneCloudConfigEntry,
+ airzone: AirzoneCloudApi,
+ ) -> None:
"""Initialize."""
self.airzone = airzone
self.airzone.set_update_callback(self.async_set_updated_data)
@@ -31,6 +41,7 @@ class AirzoneUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/airzone_cloud/diagnostics.py b/homeassistant/components/airzone_cloud/diagnostics.py
index b6744e36d8c..04aac7e2aa8 100644
--- a/homeassistant/components/airzone_cloud/diagnostics.py
+++ b/homeassistant/components/airzone_cloud/diagnostics.py
@@ -25,7 +25,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
-from . import AirzoneCloudConfigEntry
+from .coordinator import AirzoneCloudConfigEntry
TO_REDACT_API = [
API_CITY,
diff --git a/homeassistant/components/airzone_cloud/select.py b/homeassistant/components/airzone_cloud/select.py
index 895796a1073..816544efdf8 100644
--- a/homeassistant/components/airzone_cloud/select.py
+++ b/homeassistant/components/airzone_cloud/select.py
@@ -21,10 +21,9 @@ from aioairzone_cloud.const import (
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AirzoneCloudConfigEntry
-from .coordinator import AirzoneUpdateCoordinator
+from .coordinator import AirzoneCloudConfigEntry, AirzoneUpdateCoordinator
from .entity import AirzoneEntity, AirzoneZoneEntity
@@ -90,7 +89,7 @@ ZONE_SELECT_TYPES: Final[tuple[AirzoneSelectDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone Cloud select from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone_cloud/sensor.py b/homeassistant/components/airzone_cloud/sensor.py
index 70d2fd079d4..43526c3aa52 100644
--- a/homeassistant/components/airzone_cloud/sensor.py
+++ b/homeassistant/components/airzone_cloud/sensor.py
@@ -47,10 +47,9 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AirzoneCloudConfigEntry
-from .coordinator import AirzoneUpdateCoordinator
+from .coordinator import AirzoneCloudConfigEntry, AirzoneUpdateCoordinator
from .entity import (
AirzoneAidooEntity,
AirzoneEntity,
@@ -222,7 +221,7 @@ ZONE_SENSOR_TYPES: Final[tuple[SensorEntityDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone Cloud sensors from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone_cloud/switch.py b/homeassistant/components/airzone_cloud/switch.py
index 0eb907ff792..ab703cd537a 100644
--- a/homeassistant/components/airzone_cloud/switch.py
+++ b/homeassistant/components/airzone_cloud/switch.py
@@ -13,10 +13,9 @@ from homeassistant.components.switch import (
SwitchEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AirzoneCloudConfigEntry
-from .coordinator import AirzoneUpdateCoordinator
+from .coordinator import AirzoneCloudConfigEntry, AirzoneUpdateCoordinator
from .entity import AirzoneEntity, AirzoneZoneEntity
@@ -39,7 +38,7 @@ ZONE_SWITCH_TYPES: Final[tuple[AirzoneSwitchDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone Cloud switch from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/airzone_cloud/water_heater.py b/homeassistant/components/airzone_cloud/water_heater.py
index 51228ae6b90..41d43002569 100644
--- a/homeassistant/components/airzone_cloud/water_heater.py
+++ b/homeassistant/components/airzone_cloud/water_heater.py
@@ -29,10 +29,9 @@ from homeassistant.components.water_heater import (
)
from homeassistant.const import ATTR_TEMPERATURE, STATE_OFF, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AirzoneCloudConfigEntry
-from .coordinator import AirzoneUpdateCoordinator
+from .coordinator import AirzoneCloudConfigEntry, AirzoneUpdateCoordinator
from .entity import AirzoneHotWaterEntity
OPERATION_LIB_TO_HASS: Final[dict[HotWaterOperation, str]] = {
@@ -69,7 +68,7 @@ OPERATION_MODE_TO_DHW_PARAMS: Final[dict[str, dict[str, Any]]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: AirzoneCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Airzone Cloud Water Heater from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/aladdin_connect/__init__.py b/homeassistant/components/aladdin_connect/__init__.py
index 6d3f1d642b5..af50147a8ef 100644
--- a/homeassistant/components/aladdin_connect/__init__.py
+++ b/homeassistant/components/aladdin_connect/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
@@ -28,11 +28,13 @@ async def async_setup_entry(hass: HomeAssistant, _: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
- if all(
- config_entry.state is ConfigEntryState.NOT_LOADED
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.entry_id != entry.entry_id
- ):
- ir.async_delete_issue(hass, DOMAIN, DOMAIN)
-
return True
+
+
+async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Remove a config entry."""
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
+ ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ # Remove any remaining disabled or ignored entries
+ for _entry in hass.config_entries.async_entries(DOMAIN):
+ hass.async_create_task(hass.config_entries.async_remove(_entry.entry_id))
diff --git a/homeassistant/components/alarm_control_panel/strings.json b/homeassistant/components/alarm_control_panel/strings.json
index 5f718280566..ed02b2d0ee8 100644
--- a/homeassistant/components/alarm_control_panel/strings.json
+++ b/homeassistant/components/alarm_control_panel/strings.json
@@ -90,7 +90,7 @@
},
"alarm_arm_home": {
"name": "Arm home",
- "description": "Sets the alarm to: _armed, but someone is home_.",
+ "description": "Arms the alarm in the home mode.",
"fields": {
"code": {
"name": "[%key:component::alarm_control_panel::services::alarm_disarm::fields::code::name%]",
@@ -100,7 +100,7 @@
},
"alarm_arm_away": {
"name": "Arm away",
- "description": "Sets the alarm to: _armed, no one home_.",
+ "description": "Arms the alarm in the away mode.",
"fields": {
"code": {
"name": "[%key:component::alarm_control_panel::services::alarm_disarm::fields::code::name%]",
@@ -110,7 +110,7 @@
},
"alarm_arm_night": {
"name": "Arm night",
- "description": "Sets the alarm to: _armed for the night_.",
+ "description": "Arms the alarm in the night mode.",
"fields": {
"code": {
"name": "[%key:component::alarm_control_panel::services::alarm_disarm::fields::code::name%]",
@@ -120,7 +120,7 @@
},
"alarm_arm_vacation": {
"name": "Arm vacation",
- "description": "Sets the alarm to: _armed for vacation_.",
+ "description": "Arms the alarm in the vacation mode.",
"fields": {
"code": {
"name": "[%key:component::alarm_control_panel::services::alarm_disarm::fields::code::name%]",
@@ -130,7 +130,7 @@
},
"alarm_trigger": {
"name": "Trigger",
- "description": "Trigger the alarm manually.",
+ "description": "Triggers the alarm manually.",
"fields": {
"code": {
"name": "[%key:component::alarm_control_panel::services::alarm_disarm::fields::code::name%]",
diff --git a/homeassistant/components/alarmdecoder/alarm_control_panel.py b/homeassistant/components/alarmdecoder/alarm_control_panel.py
index d7092bbe1c4..52687f04bf9 100644
--- a/homeassistant/components/alarmdecoder/alarm_control_panel.py
+++ b/homeassistant/components/alarmdecoder/alarm_control_panel.py
@@ -14,7 +14,7 @@ from homeassistant.const import ATTR_CODE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AlarmDecoderConfigEntry
from .const import (
@@ -36,7 +36,7 @@ ATTR_KEYPRESS = "keypress"
async def async_setup_entry(
hass: HomeAssistant,
entry: AlarmDecoderConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up for AlarmDecoder alarm panels."""
options = entry.options
diff --git a/homeassistant/components/alarmdecoder/binary_sensor.py b/homeassistant/components/alarmdecoder/binary_sensor.py
index 1234c9f349b..b025da70d59 100644
--- a/homeassistant/components/alarmdecoder/binary_sensor.py
+++ b/homeassistant/components/alarmdecoder/binary_sensor.py
@@ -5,7 +5,7 @@ import logging
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AlarmDecoderConfigEntry
from .const import (
@@ -40,7 +40,7 @@ ATTR_RF_LOOP1 = "rf_loop1"
async def async_setup_entry(
hass: HomeAssistant,
entry: AlarmDecoderConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up for AlarmDecoder sensor."""
diff --git a/homeassistant/components/alarmdecoder/manifest.json b/homeassistant/components/alarmdecoder/manifest.json
index ae1a2f4684d..c2c12792801 100644
--- a/homeassistant/components/alarmdecoder/manifest.json
+++ b/homeassistant/components/alarmdecoder/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "device",
"iot_class": "local_push",
"loggers": ["adext", "alarmdecoder"],
- "requirements": ["adext==0.4.3"]
+ "requirements": ["adext==0.4.4"]
}
diff --git a/homeassistant/components/alarmdecoder/sensor.py b/homeassistant/components/alarmdecoder/sensor.py
index f5e744457fd..8fdc6d57c67 100644
--- a/homeassistant/components/alarmdecoder/sensor.py
+++ b/homeassistant/components/alarmdecoder/sensor.py
@@ -3,7 +3,7 @@
from homeassistant.components.sensor import SensorEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AlarmDecoderConfigEntry
from .const import SIGNAL_PANEL_MESSAGE
@@ -13,7 +13,7 @@ from .entity import AlarmDecoderEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: AlarmDecoderConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up for AlarmDecoder sensor."""
diff --git a/homeassistant/components/alert/entity.py b/homeassistant/components/alert/entity.py
index 629047b15ba..a11b281428f 100644
--- a/homeassistant/components/alert/entity.py
+++ b/homeassistant/components/alert/entity.py
@@ -14,7 +14,7 @@ from homeassistant.components.notify import (
)
from homeassistant.const import STATE_IDLE, STATE_OFF, STATE_ON
from homeassistant.core import Event, EventStateChangedData, HassJob, HomeAssistant
-from homeassistant.exceptions import ServiceNotFound
+from homeassistant.exceptions import ServiceNotFound, ServiceValidationError
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.event import (
async_track_point_in_time,
@@ -195,7 +195,8 @@ class AlertEntity(Entity):
async def async_turn_off(self, **kwargs: Any) -> None:
"""Async Acknowledge alert."""
- LOGGER.debug("Acknowledged Alert: %s", self._attr_name)
+ if not self._can_ack:
+ raise ServiceValidationError("This alert cannot be acknowledged")
self._ack = True
self.async_write_ha_state()
diff --git a/homeassistant/components/amberelectric/binary_sensor.py b/homeassistant/components/amberelectric/binary_sensor.py
index 66292ea5524..3ee27f19849 100644
--- a/homeassistant/components/amberelectric/binary_sensor.py
+++ b/homeassistant/components/amberelectric/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ATTRIBUTION
@@ -85,7 +85,7 @@ class AmberDemandWindowBinarySensor(AmberPriceGridSensor):
async def async_setup_entry(
hass: HomeAssistant,
entry: AmberConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/amberelectric/sensor.py b/homeassistant/components/amberelectric/sensor.py
index 49d6e5f4eac..7276ddb26a5 100644
--- a/homeassistant/components/amberelectric/sensor.py
+++ b/homeassistant/components/amberelectric/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import CURRENCY_DOLLAR, PERCENTAGE, UnitOfEnergy
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ATTRIBUTION
@@ -196,7 +196,7 @@ class AmberGridSensor(CoordinatorEntity[AmberUpdateCoordinator], SensorEntity):
async def async_setup_entry(
hass: HomeAssistant,
entry: AmberConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ambient_network/sensor.py b/homeassistant/components/ambient_network/sensor.py
index 9d262e5a987..9ec6db6ff45 100644
--- a/homeassistant/components/ambient_network/sensor.py
+++ b/homeassistant/components/ambient_network/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
UnitOfVolumetricFlux,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .coordinator import AmbientNetworkConfigEntry, AmbientNetworkDataUpdateCoordinator
@@ -239,6 +239,7 @@ SENSOR_DESCRIPTIONS = (
native_unit_of_measurement=DEGREE,
suggested_display_precision=0,
entity_registry_enabled_default=False,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key=TYPE_WINDGUSTMPH,
@@ -270,7 +271,7 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AmbientNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ambient Network sensor entities."""
diff --git a/homeassistant/components/ambient_station/binary_sensor.py b/homeassistant/components/ambient_station/binary_sensor.py
index a79788a4c38..9a7c89db95e 100644
--- a/homeassistant/components/ambient_station/binary_sensor.py
+++ b/homeassistant/components/ambient_station/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import ATTR_NAME, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AmbientStationConfigEntry
from .const import ATTR_LAST_DATA
@@ -381,7 +381,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AmbientStationConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ambient PWS binary sensors based on a config entry."""
ambient = entry.runtime_data
diff --git a/homeassistant/components/ambient_station/sensor.py b/homeassistant/components/ambient_station/sensor.py
index dfbd2d1b4a0..730b798bd15 100644
--- a/homeassistant/components/ambient_station/sensor.py
+++ b/homeassistant/components/ambient_station/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AmbientStation, AmbientStationConfigEntry
from .const import ATTR_LAST_DATA, TYPE_SOLARRADIATION, TYPE_SOLARRADIATION_LX
@@ -608,21 +608,25 @@ SENSOR_DESCRIPTIONS = (
key=TYPE_WINDDIR,
translation_key="wind_direction",
native_unit_of_measurement=DEGREE,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key=TYPE_WINDDIR_AVG10M,
translation_key="wind_direction_average_10m",
native_unit_of_measurement=DEGREE,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key=TYPE_WINDDIR_AVG2M,
translation_key="wind_direction_average_2m",
native_unit_of_measurement=DEGREE,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key=TYPE_WINDGUSTDIR,
translation_key="wind_gust_direction",
native_unit_of_measurement=DEGREE,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key=TYPE_WINDGUSTMPH,
@@ -662,7 +666,7 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AmbientStationConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ambient PWS sensors based on a config entry."""
ambient = entry.runtime_data
diff --git a/homeassistant/components/analytics_insights/__init__.py b/homeassistant/components/analytics_insights/__init__.py
index 69ad98db9df..ee7f6611c65 100644
--- a/homeassistant/components/analytics_insights/__init__.py
+++ b/homeassistant/components/analytics_insights/__init__.py
@@ -48,7 +48,7 @@ async def async_setup_entry(
continue
names[integration] = integrations[integration].title
- coordinator = HomeassistantAnalyticsDataUpdateCoordinator(hass, client)
+ coordinator = HomeassistantAnalyticsDataUpdateCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/analytics_insights/config_flow.py b/homeassistant/components/analytics_insights/config_flow.py
index da77a35f789..b2648f7c13c 100644
--- a/homeassistant/components/analytics_insights/config_flow.py
+++ b/homeassistant/components/analytics_insights/config_flow.py
@@ -8,7 +8,7 @@ from python_homeassistant_analytics import (
HomeassistantAnalyticsClient,
HomeassistantAnalyticsConnectionError,
)
-from python_homeassistant_analytics.models import IntegrationType
+from python_homeassistant_analytics.models import Environment, IntegrationType
import voluptuous as vol
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult, OptionsFlow
@@ -81,7 +81,7 @@ class HomeassistantAnalyticsConfigFlow(ConfigFlow, domain=DOMAIN):
)
try:
addons = await client.get_addons()
- integrations = await client.get_integrations()
+ integrations = await client.get_integrations(Environment.NEXT)
custom_integrations = await client.get_custom_integrations()
except HomeassistantAnalyticsConnectionError:
LOGGER.exception("Error connecting to Home Assistant analytics")
@@ -165,7 +165,7 @@ class HomeassistantAnalyticsOptionsFlowHandler(OptionsFlow):
)
try:
addons = await client.get_addons()
- integrations = await client.get_integrations()
+ integrations = await client.get_integrations(Environment.NEXT)
custom_integrations = await client.get_custom_integrations()
except HomeassistantAnalyticsConnectionError:
LOGGER.exception("Error connecting to Home Assistant analytics")
diff --git a/homeassistant/components/analytics_insights/coordinator.py b/homeassistant/components/analytics_insights/coordinator.py
index 701f1a8dbd4..fefd43ed8df 100644
--- a/homeassistant/components/analytics_insights/coordinator.py
+++ b/homeassistant/components/analytics_insights/coordinator.py
@@ -46,12 +46,16 @@ class HomeassistantAnalyticsDataUpdateCoordinator(DataUpdateCoordinator[Analytic
config_entry: AnalyticsInsightsConfigEntry
def __init__(
- self, hass: HomeAssistant, client: HomeassistantAnalyticsClient
+ self,
+ hass: HomeAssistant,
+ config_entry: AnalyticsInsightsConfigEntry,
+ client: HomeassistantAnalyticsClient,
) -> None:
"""Initialize the Homeassistant Analytics data coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(hours=12),
)
diff --git a/homeassistant/components/analytics_insights/sensor.py b/homeassistant/components/analytics_insights/sensor.py
index 324ca6991d2..830d914b311 100644
--- a/homeassistant/components/analytics_insights/sensor.py
+++ b/homeassistant/components/analytics_insights/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -94,7 +94,7 @@ GENERAL_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: AnalyticsInsightsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize the entries."""
diff --git a/homeassistant/components/android_ip_webcam/binary_sensor.py b/homeassistant/components/android_ip_webcam/binary_sensor.py
index 1846889bfda..67816664752 100644
--- a/homeassistant/components/android_ip_webcam/binary_sensor.py
+++ b/homeassistant/components/android_ip_webcam/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import MOTION_ACTIVE
from .coordinator import AndroidIPCamConfigEntry, AndroidIPCamDataUpdateCoordinator
@@ -24,7 +24,7 @@ BINARY_SENSOR_DESCRIPTION = BinarySensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AndroidIPCamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the IP Webcam sensors from config entry."""
diff --git a/homeassistant/components/android_ip_webcam/camera.py b/homeassistant/components/android_ip_webcam/camera.py
index 95d4fb9f67a..833b9a0d296 100644
--- a/homeassistant/components/android_ip_webcam/camera.py
+++ b/homeassistant/components/android_ip_webcam/camera.py
@@ -11,7 +11,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import AndroidIPCamConfigEntry, AndroidIPCamDataUpdateCoordinator
@@ -20,7 +20,7 @@ from .coordinator import AndroidIPCamConfigEntry, AndroidIPCamDataUpdateCoordina
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AndroidIPCamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the IP Webcam camera from config entry."""
filter_urllib3_logging()
diff --git a/homeassistant/components/android_ip_webcam/coordinator.py b/homeassistant/components/android_ip_webcam/coordinator.py
index fd6e1fcc4b9..c72d6ae1177 100644
--- a/homeassistant/components/android_ip_webcam/coordinator.py
+++ b/homeassistant/components/android_ip_webcam/coordinator.py
@@ -35,6 +35,7 @@ class AndroidIPCamDataUpdateCoordinator(DataUpdateCoordinator[None]):
super().__init__(
self.hass,
_LOGGER,
+ config_entry=config_entry,
name=f"{DOMAIN} {config_entry.data[CONF_HOST]}",
update_interval=timedelta(seconds=10),
)
diff --git a/homeassistant/components/android_ip_webcam/sensor.py b/homeassistant/components/android_ip_webcam/sensor.py
index 9b2454d6c09..e9d5f8514e8 100644
--- a/homeassistant/components/android_ip_webcam/sensor.py
+++ b/homeassistant/components/android_ip_webcam/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import AndroidIPCamConfigEntry, AndroidIPCamDataUpdateCoordinator
@@ -119,7 +119,7 @@ SENSOR_TYPES: tuple[AndroidIPWebcamSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AndroidIPCamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the IP Webcam sensors from config entry."""
diff --git a/homeassistant/components/android_ip_webcam/switch.py b/homeassistant/components/android_ip_webcam/switch.py
index f813415df0b..3ceaf6e59b9 100644
--- a/homeassistant/components/android_ip_webcam/switch.py
+++ b/homeassistant/components/android_ip_webcam/switch.py
@@ -11,7 +11,7 @@ from pydroid_ipcam import PyDroidIPCam
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AndroidIPCamConfigEntry, AndroidIPCamDataUpdateCoordinator
from .entity import AndroidIPCamBaseEntity
@@ -112,7 +112,7 @@ SWITCH_TYPES: tuple[AndroidIPWebcamSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AndroidIPCamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the IP Webcam switches from config entry."""
diff --git a/homeassistant/components/androidtv/media_player.py b/homeassistant/components/androidtv/media_player.py
index 728411ddf42..c9e62908cac 100644
--- a/homeassistant/components/androidtv/media_player.py
+++ b/homeassistant/components/androidtv/media_player.py
@@ -21,7 +21,7 @@ from homeassistant.const import ATTR_COMMAND
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from . import AndroidTVConfigEntry
@@ -65,7 +65,7 @@ ANDROIDTV_STATES = {
async def async_setup_entry(
hass: HomeAssistant,
entry: AndroidTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Android Debug Bridge entity."""
device_class = entry.runtime_data.aftv.DEVICE_CLASS
diff --git a/homeassistant/components/androidtv/remote.py b/homeassistant/components/androidtv/remote.py
index db48b0cf1b6..026d1485e07 100644
--- a/homeassistant/components/androidtv/remote.py
+++ b/homeassistant/components/androidtv/remote.py
@@ -12,7 +12,7 @@ from homeassistant.components.remote import ATTR_NUM_REPEATS, RemoteEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_TURN_OFF_COMMAND, CONF_TURN_ON_COMMAND, DOMAIN
from .entity import AndroidTVEntity, adb_decorator
@@ -21,7 +21,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the AndroidTV remote from a config entry."""
async_add_entities([AndroidTVRemote(entry)])
diff --git a/homeassistant/components/androidtv_remote/__init__.py b/homeassistant/components/androidtv_remote/__init__.py
index 6a55e9971ac..28a372da4ea 100644
--- a/homeassistant/components/androidtv_remote/__init__.py
+++ b/homeassistant/components/androidtv_remote/__init__.py
@@ -35,16 +35,12 @@ async def async_setup_entry(
@callback
def is_available_updated(is_available: bool) -> None:
- if is_available:
- _LOGGER.info(
- "Reconnected to %s at %s", entry.data[CONF_NAME], entry.data[CONF_HOST]
- )
- else:
- _LOGGER.warning(
- "Disconnected from %s at %s",
- entry.data[CONF_NAME],
- entry.data[CONF_HOST],
- )
+ _LOGGER.info(
+ "%s %s at %s",
+ "Reconnected to" if is_available else "Disconnected from",
+ entry.data[CONF_NAME],
+ entry.data[CONF_HOST],
+ )
api.add_is_available_updated_callback(is_available_updated)
diff --git a/homeassistant/components/androidtv_remote/manifest.json b/homeassistant/components/androidtv_remote/manifest.json
index d9c2dd05c44..1c45e825359 100644
--- a/homeassistant/components/androidtv_remote/manifest.json
+++ b/homeassistant/components/androidtv_remote/manifest.json
@@ -7,6 +7,6 @@
"integration_type": "device",
"iot_class": "local_push",
"loggers": ["androidtvremote2"],
- "requirements": ["androidtvremote2==0.1.2"],
+ "requirements": ["androidtvremote2==0.2.0"],
"zeroconf": ["_androidtvremote2._tcp.local."]
}
diff --git a/homeassistant/components/androidtv_remote/media_player.py b/homeassistant/components/androidtv_remote/media_player.py
index cdc307a0472..3d3a97092bc 100644
--- a/homeassistant/components/androidtv_remote/media_player.py
+++ b/homeassistant/components/androidtv_remote/media_player.py
@@ -18,7 +18,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AndroidTVRemoteConfigEntry
from .const import CONF_APP_ICON, CONF_APP_NAME
@@ -30,7 +30,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AndroidTVRemoteConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Android TV media player entity based on a config entry."""
api = config_entry.runtime_data
diff --git a/homeassistant/components/androidtv_remote/remote.py b/homeassistant/components/androidtv_remote/remote.py
index c9a261c8735..212b0491d2d 100644
--- a/homeassistant/components/androidtv_remote/remote.py
+++ b/homeassistant/components/androidtv_remote/remote.py
@@ -18,7 +18,7 @@ from homeassistant.components.remote import (
RemoteEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AndroidTVRemoteConfigEntry
from .const import CONF_APP_NAME
@@ -30,7 +30,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AndroidTVRemoteConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Android TV remote entity based on a config entry."""
api = config_entry.runtime_data
diff --git a/homeassistant/components/anova/coordinator.py b/homeassistant/components/anova/coordinator.py
index 811c32c97b5..61d118ed0a5 100644
--- a/homeassistant/components/anova/coordinator.py
+++ b/homeassistant/components/anova/coordinator.py
@@ -1,5 +1,7 @@
"""Support for Anova Coordinators."""
+from __future__ import annotations
+
from dataclasses import dataclass
import logging
@@ -20,7 +22,7 @@ class AnovaData:
"""Data for the Anova integration."""
api_jwt: str
- coordinators: list["AnovaCoordinator"]
+ coordinators: list[AnovaCoordinator]
api: AnovaApi
diff --git a/homeassistant/components/anova/sensor.py b/homeassistant/components/anova/sensor.py
index 7365e4597ba..c3a3d3861f2 100644
--- a/homeassistant/components/anova/sensor.py
+++ b/homeassistant/components/anova/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfTemperature, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import AnovaConfigEntry, AnovaCoordinator
@@ -97,7 +97,7 @@ SENSOR_DESCRIPTIONS: list[AnovaSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: AnovaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Anova device."""
anova_data = entry.runtime_data
@@ -108,7 +108,7 @@ async def async_setup_entry(
def setup_coordinator(
coordinator: AnovaCoordinator,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an individual Anova Coordinator."""
diff --git a/homeassistant/components/anthemav/media_player.py b/homeassistant/components/anthemav/media_player.py
index be5a6ad2258..317498e96b5 100644
--- a/homeassistant/components/anthemav/media_player.py
+++ b/homeassistant/components/anthemav/media_player.py
@@ -16,18 +16,19 @@ from homeassistant.const import CONF_MAC, CONF_MODEL
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AnthemavConfigEntry
from .const import ANTHEMAV_UPDATE_SIGNAL, DOMAIN, MANUFACTURER
_LOGGER = logging.getLogger(__name__)
+VOLUME_STEP = 0.01
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AnthemavConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
name = config_entry.title
@@ -60,6 +61,7 @@ class AnthemAVR(MediaPlayerEntity):
| MediaPlayerEntityFeature.TURN_OFF
| MediaPlayerEntityFeature.SELECT_SOURCE
)
+ _attr_volume_step = VOLUME_STEP
def __init__(
self,
diff --git a/homeassistant/components/anthemav/strings.json b/homeassistant/components/anthemav/strings.json
index 1f1dd0ec75b..15e365b3e63 100644
--- a/homeassistant/components/anthemav/strings.json
+++ b/homeassistant/components/anthemav/strings.json
@@ -10,7 +10,7 @@
},
"error": {
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
- "cannot_receive_deviceinfo": "Failed to retreive MAC Address. Make sure the device is turned on"
+ "cannot_receive_deviceinfo": "Failed to retrieve MAC Address. Make sure the device is turned on"
},
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
diff --git a/homeassistant/components/anthropic/__init__.py b/homeassistant/components/anthropic/__init__.py
index aa6cf509fa1..a9745d1a6a5 100644
--- a/homeassistant/components/anthropic/__init__.py
+++ b/homeassistant/components/anthropic/__init__.py
@@ -2,6 +2,8 @@
from __future__ import annotations
+from functools import partial
+
import anthropic
from homeassistant.config_entries import ConfigEntry
@@ -10,7 +12,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import config_validation as cv
-from .const import DOMAIN, LOGGER
+from .const import CONF_CHAT_MODEL, DOMAIN, LOGGER, RECOMMENDED_CHAT_MODEL
PLATFORMS = (Platform.CONVERSATION,)
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
@@ -20,14 +22,13 @@ type AnthropicConfigEntry = ConfigEntry[anthropic.AsyncClient]
async def async_setup_entry(hass: HomeAssistant, entry: AnthropicConfigEntry) -> bool:
"""Set up Anthropic from a config entry."""
- client = anthropic.AsyncAnthropic(api_key=entry.data[CONF_API_KEY])
+ client = await hass.async_add_executor_job(
+ partial(anthropic.AsyncAnthropic, api_key=entry.data[CONF_API_KEY])
+ )
try:
- await client.messages.create(
- model="claude-3-haiku-20240307",
- max_tokens=1,
- messages=[{"role": "user", "content": "Hi"}],
- timeout=10.0,
- )
+ model_id = entry.options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL)
+ model = await client.models.retrieve(model_id=model_id, timeout=10.0)
+ LOGGER.debug("Anthropic model: %s", model.display_name)
except anthropic.AuthenticationError as err:
LOGGER.error("Invalid API key: %s", err)
return False
diff --git a/homeassistant/components/anthropic/config_flow.py b/homeassistant/components/anthropic/config_flow.py
index fa43a3c4bcc..e53a479d7d4 100644
--- a/homeassistant/components/anthropic/config_flow.py
+++ b/homeassistant/components/anthropic/config_flow.py
@@ -2,6 +2,7 @@
from __future__ import annotations
+from functools import partial
import logging
from types import MappingProxyType
from typing import Any
@@ -33,10 +34,12 @@ from .const import (
CONF_PROMPT,
CONF_RECOMMENDED,
CONF_TEMPERATURE,
+ CONF_THINKING_BUDGET,
DOMAIN,
RECOMMENDED_CHAT_MODEL,
RECOMMENDED_MAX_TOKENS,
RECOMMENDED_TEMPERATURE,
+ RECOMMENDED_THINKING_BUDGET,
)
_LOGGER = logging.getLogger(__name__)
@@ -59,13 +62,10 @@ async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> None:
Data has the keys from STEP_USER_DATA_SCHEMA with values provided by the user.
"""
- client = anthropic.AsyncAnthropic(api_key=data[CONF_API_KEY])
- await client.messages.create(
- model="claude-3-haiku-20240307",
- max_tokens=1,
- messages=[{"role": "user", "content": "Hi"}],
- timeout=10.0,
+ client = await hass.async_add_executor_job(
+ partial(anthropic.AsyncAnthropic, api_key=data[CONF_API_KEY])
)
+ await client.models.list(timeout=10.0)
class AnthropicConfigFlow(ConfigFlow, domain=DOMAIN):
@@ -130,21 +130,29 @@ class AnthropicOptionsFlow(OptionsFlow):
) -> ConfigFlowResult:
"""Manage the options."""
options: dict[str, Any] | MappingProxyType[str, Any] = self.config_entry.options
+ errors: dict[str, str] = {}
if user_input is not None:
if user_input[CONF_RECOMMENDED] == self.last_rendered_recommended:
if user_input[CONF_LLM_HASS_API] == "none":
user_input.pop(CONF_LLM_HASS_API)
- return self.async_create_entry(title="", data=user_input)
- # Re-render the options again, now with the recommended options shown/hidden
- self.last_rendered_recommended = user_input[CONF_RECOMMENDED]
+ if user_input.get(
+ CONF_THINKING_BUDGET, RECOMMENDED_THINKING_BUDGET
+ ) >= user_input.get(CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS):
+ errors[CONF_THINKING_BUDGET] = "thinking_budget_too_large"
- options = {
- CONF_RECOMMENDED: user_input[CONF_RECOMMENDED],
- CONF_PROMPT: user_input[CONF_PROMPT],
- CONF_LLM_HASS_API: user_input[CONF_LLM_HASS_API],
- }
+ if not errors:
+ return self.async_create_entry(title="", data=user_input)
+ else:
+ # Re-render the options again, now with the recommended options shown/hidden
+ self.last_rendered_recommended = user_input[CONF_RECOMMENDED]
+
+ options = {
+ CONF_RECOMMENDED: user_input[CONF_RECOMMENDED],
+ CONF_PROMPT: user_input[CONF_PROMPT],
+ CONF_LLM_HASS_API: user_input[CONF_LLM_HASS_API],
+ }
suggested_values = options.copy()
if not suggested_values.get(CONF_PROMPT):
@@ -158,6 +166,7 @@ class AnthropicOptionsFlow(OptionsFlow):
return self.async_show_form(
step_id="init",
data_schema=schema,
+ errors=errors or None,
)
@@ -207,6 +216,10 @@ def anthropic_config_option_schema(
CONF_TEMPERATURE,
default=RECOMMENDED_TEMPERATURE,
): NumberSelector(NumberSelectorConfig(min=0, max=1, step=0.05)),
+ vol.Optional(
+ CONF_THINKING_BUDGET,
+ default=RECOMMENDED_THINKING_BUDGET,
+ ): int,
}
)
return schema
diff --git a/homeassistant/components/anthropic/const.py b/homeassistant/components/anthropic/const.py
index 0dbf9c51ac1..38e4270e6e1 100644
--- a/homeassistant/components/anthropic/const.py
+++ b/homeassistant/components/anthropic/const.py
@@ -13,3 +13,8 @@ CONF_MAX_TOKENS = "max_tokens"
RECOMMENDED_MAX_TOKENS = 1024
CONF_TEMPERATURE = "temperature"
RECOMMENDED_TEMPERATURE = 1.0
+CONF_THINKING_BUDGET = "thinking_budget"
+RECOMMENDED_THINKING_BUDGET = 0
+MIN_THINKING_BUDGET = 1024
+
+THINKING_MODELS = ["claude-3-7-sonnet-20250219", "claude-3-7-sonnet-latest"]
diff --git a/homeassistant/components/anthropic/conversation.py b/homeassistant/components/anthropic/conversation.py
index 259d1295809..5e5ad464eaa 100644
--- a/homeassistant/components/anthropic/conversation.py
+++ b/homeassistant/components/anthropic/conversation.py
@@ -1,33 +1,46 @@
"""Conversation support for Anthropic."""
-from collections.abc import Callable
+from collections.abc import AsyncGenerator, Callable, Iterable
import json
from typing import Any, Literal, cast
import anthropic
+from anthropic import AsyncStream
from anthropic._types import NOT_GIVEN
from anthropic.types import (
- Message,
+ InputJSONDelta,
MessageParam,
+ MessageStreamEvent,
+ RawContentBlockDeltaEvent,
+ RawContentBlockStartEvent,
+ RawContentBlockStopEvent,
+ RawMessageStartEvent,
+ RawMessageStopEvent,
+ RedactedThinkingBlock,
+ RedactedThinkingBlockParam,
+ SignatureDelta,
TextBlock,
TextBlockParam,
+ TextDelta,
+ ThinkingBlock,
+ ThinkingBlockParam,
+ ThinkingConfigDisabledParam,
+ ThinkingConfigEnabledParam,
+ ThinkingDelta,
ToolParam,
ToolResultBlockParam,
ToolUseBlock,
ToolUseBlockParam,
)
-import voluptuous as vol
from voluptuous_openapi import convert
from homeassistant.components import conversation
-from homeassistant.components.conversation import trace
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_LLM_HASS_API, MATCH_ALL
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import HomeAssistantError, TemplateError
-from homeassistant.helpers import device_registry as dr, intent, llm, template
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.util import ulid as ulid_util
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers import device_registry as dr, intent, llm
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AnthropicConfigEntry
from .const import (
@@ -35,11 +48,15 @@ from .const import (
CONF_MAX_TOKENS,
CONF_PROMPT,
CONF_TEMPERATURE,
+ CONF_THINKING_BUDGET,
DOMAIN,
LOGGER,
+ MIN_THINKING_BUDGET,
RECOMMENDED_CHAT_MODEL,
RECOMMENDED_MAX_TOKENS,
RECOMMENDED_TEMPERATURE,
+ RECOMMENDED_THINKING_BUDGET,
+ THINKING_MODELS,
)
# Max number of back and forth with the LLM to generate a response
@@ -49,7 +66,7 @@ MAX_TOOL_ITERATIONS = 10
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AnthropicConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up conversation entities."""
agent = AnthropicConversationEntity(config_entry)
@@ -67,26 +84,211 @@ def _format_tool(
)
-def _message_convert(
- message: Message,
-) -> MessageParam:
- """Convert from class to TypedDict."""
- param_content: list[TextBlockParam | ToolUseBlockParam] = []
+def _convert_content(
+ chat_content: Iterable[conversation.Content],
+) -> list[MessageParam]:
+ """Transform HA chat_log content into Anthropic API format."""
+ messages: list[MessageParam] = []
- for message_content in message.content:
- if isinstance(message_content, TextBlock):
- param_content.append(TextBlockParam(type="text", text=message_content.text))
- elif isinstance(message_content, ToolUseBlock):
- param_content.append(
- ToolUseBlockParam(
- type="tool_use",
- id=message_content.id,
- name=message_content.name,
- input=message_content.input,
- )
+ for content in chat_content:
+ if isinstance(content, conversation.ToolResultContent):
+ tool_result_block = ToolResultBlockParam(
+ type="tool_result",
+ tool_use_id=content.tool_call_id,
+ content=json.dumps(content.tool_result),
)
+ if not messages or messages[-1]["role"] != "user":
+ messages.append(
+ MessageParam(
+ role="user",
+ content=[tool_result_block],
+ )
+ )
+ elif isinstance(messages[-1]["content"], str):
+ messages[-1]["content"] = [
+ TextBlockParam(type="text", text=messages[-1]["content"]),
+ tool_result_block,
+ ]
+ else:
+ messages[-1]["content"].append(tool_result_block) # type: ignore[attr-defined]
+ elif isinstance(content, conversation.UserContent):
+ # Combine consequent user messages
+ if not messages or messages[-1]["role"] != "user":
+ messages.append(
+ MessageParam(
+ role="user",
+ content=content.content,
+ )
+ )
+ elif isinstance(messages[-1]["content"], str):
+ messages[-1]["content"] = [
+ TextBlockParam(type="text", text=messages[-1]["content"]),
+ TextBlockParam(type="text", text=content.content),
+ ]
+ else:
+ messages[-1]["content"].append( # type: ignore[attr-defined]
+ TextBlockParam(type="text", text=content.content)
+ )
+ elif isinstance(content, conversation.AssistantContent):
+ # Combine consequent assistant messages
+ if not messages or messages[-1]["role"] != "assistant":
+ messages.append(
+ MessageParam(
+ role="assistant",
+ content=[],
+ )
+ )
- return MessageParam(role=message.role, content=param_content)
+ if content.content:
+ messages[-1]["content"].append( # type: ignore[union-attr]
+ TextBlockParam(type="text", text=content.content)
+ )
+ if content.tool_calls:
+ messages[-1]["content"].extend( # type: ignore[union-attr]
+ [
+ ToolUseBlockParam(
+ type="tool_use",
+ id=tool_call.id,
+ name=tool_call.tool_name,
+ input=tool_call.tool_args,
+ )
+ for tool_call in content.tool_calls
+ ]
+ )
+ else:
+ # Note: We don't pass SystemContent here as its passed to the API as the prompt
+ raise TypeError(f"Unexpected content type: {type(content)}")
+
+ return messages
+
+
+async def _transform_stream(
+ result: AsyncStream[MessageStreamEvent],
+ messages: list[MessageParam],
+) -> AsyncGenerator[conversation.AssistantContentDeltaDict]:
+ """Transform the response stream into HA format.
+
+ A typical stream of responses might look something like the following:
+ - RawMessageStartEvent with no content
+ - RawContentBlockStartEvent with an empty ThinkingBlock (if extended thinking is enabled)
+ - RawContentBlockDeltaEvent with a ThinkingDelta
+ - RawContentBlockDeltaEvent with a ThinkingDelta
+ - RawContentBlockDeltaEvent with a ThinkingDelta
+ - ...
+ - RawContentBlockDeltaEvent with a SignatureDelta
+ - RawContentBlockStopEvent
+ - RawContentBlockStartEvent with a RedactedThinkingBlock (occasionally)
+ - RawContentBlockStopEvent (RedactedThinkingBlock does not have a delta)
+ - RawContentBlockStartEvent with an empty TextBlock
+ - RawContentBlockDeltaEvent with a TextDelta
+ - RawContentBlockDeltaEvent with a TextDelta
+ - RawContentBlockDeltaEvent with a TextDelta
+ - ...
+ - RawContentBlockStopEvent
+ - RawContentBlockStartEvent with ToolUseBlock specifying the function name
+ - RawContentBlockDeltaEvent with a InputJSONDelta
+ - RawContentBlockDeltaEvent with a InputJSONDelta
+ - ...
+ - RawContentBlockStopEvent
+ - RawMessageDeltaEvent with a stop_reason='tool_use'
+ - RawMessageStopEvent(type='message_stop')
+
+ Each message could contain multiple blocks of the same type.
+ """
+ if result is None:
+ raise TypeError("Expected a stream of messages")
+
+ current_message: MessageParam | None = None
+ current_block: (
+ TextBlockParam
+ | ToolUseBlockParam
+ | ThinkingBlockParam
+ | RedactedThinkingBlockParam
+ | None
+ ) = None
+ current_tool_args: str
+
+ async for response in result:
+ LOGGER.debug("Received response: %s", response)
+
+ if isinstance(response, RawMessageStartEvent):
+ if response.message.role != "assistant":
+ raise ValueError("Unexpected message role")
+ current_message = MessageParam(role=response.message.role, content=[])
+ elif isinstance(response, RawContentBlockStartEvent):
+ if isinstance(response.content_block, ToolUseBlock):
+ current_block = ToolUseBlockParam(
+ type="tool_use",
+ id=response.content_block.id,
+ name=response.content_block.name,
+ input="",
+ )
+ current_tool_args = ""
+ elif isinstance(response.content_block, TextBlock):
+ current_block = TextBlockParam(
+ type="text", text=response.content_block.text
+ )
+ yield {"role": "assistant"}
+ if response.content_block.text:
+ yield {"content": response.content_block.text}
+ elif isinstance(response.content_block, ThinkingBlock):
+ current_block = ThinkingBlockParam(
+ type="thinking",
+ thinking=response.content_block.thinking,
+ signature=response.content_block.signature,
+ )
+ elif isinstance(response.content_block, RedactedThinkingBlock):
+ current_block = RedactedThinkingBlockParam(
+ type="redacted_thinking", data=response.content_block.data
+ )
+ LOGGER.debug(
+ "Some of Claude’s internal reasoning has been automatically "
+ "encrypted for safety reasons. This doesn’t affect the quality of "
+ "responses"
+ )
+ elif isinstance(response, RawContentBlockDeltaEvent):
+ if current_block is None:
+ raise ValueError("Unexpected delta without a block")
+ if isinstance(response.delta, InputJSONDelta):
+ current_tool_args += response.delta.partial_json
+ elif isinstance(response.delta, TextDelta):
+ text_block = cast(TextBlockParam, current_block)
+ text_block["text"] += response.delta.text
+ yield {"content": response.delta.text}
+ elif isinstance(response.delta, ThinkingDelta):
+ thinking_block = cast(ThinkingBlockParam, current_block)
+ thinking_block["thinking"] += response.delta.thinking
+ elif isinstance(response.delta, SignatureDelta):
+ thinking_block = cast(ThinkingBlockParam, current_block)
+ thinking_block["signature"] += response.delta.signature
+ elif isinstance(response, RawContentBlockStopEvent):
+ if current_block is None:
+ raise ValueError("Unexpected stop event without a current block")
+ if current_block["type"] == "tool_use":
+ tool_block = cast(ToolUseBlockParam, current_block)
+ tool_args = json.loads(current_tool_args)
+ tool_block["input"] = tool_args
+ yield {
+ "tool_calls": [
+ llm.ToolInput(
+ id=tool_block["id"],
+ tool_name=tool_block["name"],
+ tool_args=tool_args,
+ )
+ ]
+ }
+ elif current_block["type"] == "thinking":
+ thinking_block = cast(ThinkingBlockParam, current_block)
+ LOGGER.debug("Thinking: %s", thinking_block["thinking"])
+
+ if current_message is None:
+ raise ValueError("Unexpected stop event without a current message")
+ current_message["content"].append(current_block) # type: ignore[union-attr]
+ current_block = None
+ elif isinstance(response, RawMessageStopEvent):
+ if current_message is not None:
+ messages.append(current_message)
+ current_message = None
class AnthropicConversationEntity(
@@ -100,7 +302,6 @@ class AnthropicConversationEntity(
def __init__(self, entry: AnthropicConfigEntry) -> None:
"""Initialize the agent."""
self.entry = entry
- self.history: dict[str, list[MessageParam]] = {}
self._attr_unique_id = entry.entry_id
self._attr_device_info = dr.DeviceInfo(
identifiers={(DOMAIN, entry.entry_id)},
@@ -125,188 +326,92 @@ class AnthropicConversationEntity(
self.entry.add_update_listener(self._async_entry_update_listener)
)
- async def async_process(
- self, user_input: conversation.ConversationInput
+ async def _async_handle_message(
+ self,
+ user_input: conversation.ConversationInput,
+ chat_log: conversation.ChatLog,
) -> conversation.ConversationResult:
- """Process a sentence."""
+ """Call the API."""
options = self.entry.options
- intent_response = intent.IntentResponse(language=user_input.language)
- llm_api: llm.APIInstance | None = None
- tools: list[ToolParam] | None = None
- user_name: str | None = None
- llm_context = llm.LLMContext(
- platform=DOMAIN,
- context=user_input.context,
- user_prompt=user_input.text,
- language=user_input.language,
- assistant=conversation.DOMAIN,
- device_id=user_input.device_id,
- )
-
- if options.get(CONF_LLM_HASS_API):
- try:
- llm_api = await llm.async_get_api(
- self.hass,
- options[CONF_LLM_HASS_API],
- llm_context,
- )
- except HomeAssistantError as err:
- LOGGER.error("Error getting LLM API: %s", err)
- intent_response.async_set_error(
- intent.IntentResponseErrorCode.UNKNOWN,
- f"Error preparing LLM API: {err}",
- )
- return conversation.ConversationResult(
- response=intent_response, conversation_id=user_input.conversation_id
- )
- tools = [
- _format_tool(tool, llm_api.custom_serializer) for tool in llm_api.tools
- ]
-
- if user_input.conversation_id is None:
- conversation_id = ulid_util.ulid_now()
- messages = []
-
- elif user_input.conversation_id in self.history:
- conversation_id = user_input.conversation_id
- messages = self.history[conversation_id]
-
- else:
- # Conversation IDs are ULIDs. We generate a new one if not provided.
- # If an old OLID is passed in, we will generate a new one to indicate
- # a new conversation was started. If the user picks their own, they
- # want to track a conversation and we respect it.
- try:
- ulid_util.ulid_to_bytes(user_input.conversation_id)
- conversation_id = ulid_util.ulid_now()
- except ValueError:
- conversation_id = user_input.conversation_id
-
- messages = []
-
- if (
- user_input.context
- and user_input.context.user_id
- and (
- user := await self.hass.auth.async_get_user(user_input.context.user_id)
- )
- ):
- user_name = user.name
try:
- prompt_parts = [
- template.Template(
- llm.BASE_PROMPT
- + options.get(CONF_PROMPT, llm.DEFAULT_INSTRUCTIONS_PROMPT),
- self.hass,
- ).async_render(
- {
- "ha_name": self.hass.config.location_name,
- "user_name": user_name,
- "llm_context": llm_context,
- },
- parse_result=False,
- )
+ await chat_log.async_update_llm_data(
+ DOMAIN,
+ user_input,
+ options.get(CONF_LLM_HASS_API),
+ options.get(CONF_PROMPT),
+ )
+ except conversation.ConverseError as err:
+ return err.as_conversation_result()
+
+ tools: list[ToolParam] | None = None
+ if chat_log.llm_api:
+ tools = [
+ _format_tool(tool, chat_log.llm_api.custom_serializer)
+ for tool in chat_log.llm_api.tools
]
- except TemplateError as err:
- LOGGER.error("Error rendering prompt: %s", err)
- intent_response.async_set_error(
- intent.IntentResponseErrorCode.UNKNOWN,
- f"Sorry, I had a problem with my template: {err}",
- )
- return conversation.ConversationResult(
- response=intent_response, conversation_id=conversation_id
- )
-
- if llm_api:
- prompt_parts.append(llm_api.api_prompt)
-
- prompt = "\n".join(prompt_parts)
-
- # Create a copy of the variable because we attach it to the trace
- messages = [*messages, MessageParam(role="user", content=user_input.text)]
-
- LOGGER.debug("Prompt: %s", messages)
- LOGGER.debug("Tools: %s", tools)
- trace.async_conversation_trace_append(
- trace.ConversationTraceEventType.AGENT_DETAIL,
- {"system": prompt, "messages": messages},
- )
+ system = chat_log.content[0]
+ if not isinstance(system, conversation.SystemContent):
+ raise TypeError("First message must be a system message")
+ messages = _convert_content(chat_log.content[1:])
client = self.entry.runtime_data
+ thinking_budget = options.get(CONF_THINKING_BUDGET, RECOMMENDED_THINKING_BUDGET)
+ model = options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL)
+
# To prevent infinite loops, we limit the number of iterations
for _iteration in range(MAX_TOOL_ITERATIONS):
+ model_args = {
+ "model": model,
+ "messages": messages,
+ "tools": tools or NOT_GIVEN,
+ "max_tokens": options.get(CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS),
+ "system": system.content,
+ "stream": True,
+ }
+ if model in THINKING_MODELS and thinking_budget >= MIN_THINKING_BUDGET:
+ model_args["thinking"] = ThinkingConfigEnabledParam(
+ type="enabled", budget_tokens=thinking_budget
+ )
+ else:
+ model_args["thinking"] = ThinkingConfigDisabledParam(type="disabled")
+ model_args["temperature"] = options.get(
+ CONF_TEMPERATURE, RECOMMENDED_TEMPERATURE
+ )
+
try:
- response = await client.messages.create(
- model=options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL),
- messages=messages,
- tools=tools or NOT_GIVEN,
- max_tokens=options.get(CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS),
- system=prompt,
- temperature=options.get(CONF_TEMPERATURE, RECOMMENDED_TEMPERATURE),
- )
+ stream = await client.messages.create(**model_args)
except anthropic.AnthropicError as err:
- intent_response.async_set_error(
- intent.IntentResponseErrorCode.UNKNOWN,
- f"Sorry, I had a problem talking to Anthropic: {err}",
- )
- return conversation.ConversationResult(
- response=intent_response, conversation_id=conversation_id
+ raise HomeAssistantError(
+ f"Sorry, I had a problem talking to Anthropic: {err}"
+ ) from err
+
+ messages.extend(
+ _convert_content(
+ [
+ content
+ async for content in chat_log.async_add_delta_content_stream(
+ user_input.agent_id, _transform_stream(stream, messages)
+ )
+ if not isinstance(content, conversation.AssistantContent)
+ ]
)
+ )
- LOGGER.debug("Response %s", response)
-
- messages.append(_message_convert(response))
-
- if response.stop_reason != "tool_use" or not llm_api:
- break
-
- tool_results: list[ToolResultBlockParam] = []
- for tool_call in response.content:
- if isinstance(tool_call, TextBlock):
- LOGGER.info(tool_call.text)
-
- if not isinstance(tool_call, ToolUseBlock):
- continue
-
- tool_input = llm.ToolInput(
- id=tool_call.id,
- tool_name=tool_call.name,
- tool_args=cast(dict[str, Any], tool_call.input),
- )
- LOGGER.debug(
- "Tool call: %s(%s)", tool_input.tool_name, tool_input.tool_args
- )
-
- try:
- tool_response = await llm_api.async_call_tool(tool_input)
- except (HomeAssistantError, vol.Invalid) as e:
- tool_response = {"error": type(e).__name__}
- if str(e):
- tool_response["error_text"] = str(e)
-
- LOGGER.debug("Tool response: %s", tool_response)
- tool_results.append(
- ToolResultBlockParam(
- type="tool_result",
- tool_use_id=tool_call.id,
- content=json.dumps(tool_response),
- )
- )
-
- messages.append(MessageParam(role="user", content=tool_results))
-
- self.history[conversation_id] = messages
-
- for content in response.content:
- if isinstance(content, TextBlock):
- intent_response.async_set_speech(content.text)
+ if not chat_log.unresponded_tool_results:
break
+ response_content = chat_log.content[-1]
+ if not isinstance(response_content, conversation.AssistantContent):
+ raise TypeError("Last message must be an assistant message")
+ intent_response = intent.IntentResponse(language=user_input.language)
+ intent_response.async_set_speech(response_content.content or "")
return conversation.ConversationResult(
- response=intent_response, conversation_id=conversation_id
+ response=intent_response,
+ conversation_id=chat_log.conversation_id,
+ continue_conversation=chat_log.continue_conversation,
)
async def _async_entry_update_listener(
diff --git a/homeassistant/components/anthropic/manifest.json b/homeassistant/components/anthropic/manifest.json
index b5cbb36c034..797a7299d16 100644
--- a/homeassistant/components/anthropic/manifest.json
+++ b/homeassistant/components/anthropic/manifest.json
@@ -8,5 +8,5 @@
"documentation": "https://www.home-assistant.io/integrations/anthropic",
"integration_type": "service",
"iot_class": "cloud_polling",
- "requirements": ["anthropic==0.44.0"]
+ "requirements": ["anthropic==0.47.2"]
}
diff --git a/homeassistant/components/anthropic/strings.json b/homeassistant/components/anthropic/strings.json
index 9550a1a6672..c2caf3a6666 100644
--- a/homeassistant/components/anthropic/strings.json
+++ b/homeassistant/components/anthropic/strings.json
@@ -23,12 +23,17 @@
"max_tokens": "Maximum tokens to return in response",
"temperature": "Temperature",
"llm_hass_api": "[%key:common::config_flow::data::llm_hass_api%]",
- "recommended": "Recommended model settings"
+ "recommended": "Recommended model settings",
+ "thinking_budget_tokens": "Thinking budget"
},
"data_description": {
- "prompt": "Instruct how the LLM should respond. This can be a template."
+ "prompt": "Instruct how the LLM should respond. This can be a template.",
+ "thinking_budget_tokens": "The number of tokens the model can use to think about the response out of the total maximum number of tokens. Set to 1024 or greater to enable extended thinking."
}
}
+ },
+ "error": {
+ "thinking_budget_too_large": "Maximum tokens must be greater than the thinking budget."
}
}
}
diff --git a/homeassistant/components/aosmith/__init__.py b/homeassistant/components/aosmith/__init__.py
index dd60f69c4b9..7593365c573 100644
--- a/homeassistant/components/aosmith/__init__.py
+++ b/homeassistant/components/aosmith/__init__.py
@@ -2,31 +2,22 @@
from __future__ import annotations
-from dataclasses import dataclass
-
from py_aosmith import AOSmithAPIClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import aiohttp_client, device_registry as dr
from .const import DOMAIN
-from .coordinator import AOSmithEnergyCoordinator, AOSmithStatusCoordinator
+from .coordinator import (
+ AOSmithConfigEntry,
+ AOSmithData,
+ AOSmithEnergyCoordinator,
+ AOSmithStatusCoordinator,
+)
PLATFORMS: list[Platform] = [Platform.SENSOR, Platform.WATER_HEATER]
-type AOSmithConfigEntry = ConfigEntry[AOSmithData]
-
-
-@dataclass
-class AOSmithData:
- """Data for the A. O. Smith integration."""
-
- client: AOSmithAPIClient
- status_coordinator: AOSmithStatusCoordinator
- energy_coordinator: AOSmithEnergyCoordinator
-
async def async_setup_entry(hass: HomeAssistant, entry: AOSmithConfigEntry) -> bool:
"""Set up A. O. Smith from a config entry."""
@@ -36,7 +27,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: AOSmithConfigEntry) -> b
session = aiohttp_client.async_get_clientsession(hass)
client = AOSmithAPIClient(email, password, session)
- status_coordinator = AOSmithStatusCoordinator(hass, client)
+ status_coordinator = AOSmithStatusCoordinator(hass, entry, client)
await status_coordinator.async_config_entry_first_refresh()
device_registry = dr.async_get(hass)
@@ -53,7 +44,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: AOSmithConfigEntry) -> b
)
energy_coordinator = AOSmithEnergyCoordinator(
- hass, client, list(status_coordinator.data)
+ hass, entry, client, list(status_coordinator.data)
)
await energy_coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/aosmith/coordinator.py b/homeassistant/components/aosmith/coordinator.py
index 3bf97e49cae..16cacfcbc10 100644
--- a/homeassistant/components/aosmith/coordinator.py
+++ b/homeassistant/components/aosmith/coordinator.py
@@ -1,5 +1,8 @@
"""The data update coordinator for the A. O. Smith integration."""
+from __future__ import annotations
+
+from dataclasses import dataclass
import logging
from py_aosmith import (
@@ -9,6 +12,7 @@ from py_aosmith import (
)
from py_aosmith.models import Device as AOSmithDevice
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -17,13 +21,37 @@ from .const import DOMAIN, ENERGY_USAGE_INTERVAL, FAST_INTERVAL, REGULAR_INTERVA
_LOGGER = logging.getLogger(__name__)
+type AOSmithConfigEntry = ConfigEntry[AOSmithData]
+
+
+@dataclass
+class AOSmithData:
+ """Data for the A. O. Smith integration."""
+
+ client: AOSmithAPIClient
+ status_coordinator: AOSmithStatusCoordinator
+ energy_coordinator: AOSmithEnergyCoordinator
+
class AOSmithStatusCoordinator(DataUpdateCoordinator[dict[str, AOSmithDevice]]):
"""Coordinator for device status, updating with a frequent interval."""
- def __init__(self, hass: HomeAssistant, client: AOSmithAPIClient) -> None:
+ config_entry: AOSmithConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: AOSmithConfigEntry,
+ client: AOSmithAPIClient,
+ ) -> None:
"""Initialize the coordinator."""
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=REGULAR_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=REGULAR_INTERVAL,
+ )
self.client = client
async def _async_update_data(self) -> dict[str, AOSmithDevice]:
@@ -51,15 +79,22 @@ class AOSmithStatusCoordinator(DataUpdateCoordinator[dict[str, AOSmithDevice]]):
class AOSmithEnergyCoordinator(DataUpdateCoordinator[dict[str, float]]):
"""Coordinator for energy usage data, updating with a slower interval."""
+ config_entry: AOSmithConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: AOSmithConfigEntry,
client: AOSmithAPIClient,
junction_ids: list[str],
) -> None:
"""Initialize the coordinator."""
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=ENERGY_USAGE_INTERVAL
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=ENERGY_USAGE_INTERVAL,
)
self.client = client
self.junction_ids = junction_ids
diff --git a/homeassistant/components/aosmith/diagnostics.py b/homeassistant/components/aosmith/diagnostics.py
index 94726731f75..4019bee4dc8 100644
--- a/homeassistant/components/aosmith/diagnostics.py
+++ b/homeassistant/components/aosmith/diagnostics.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant
-from . import AOSmithConfigEntry
+from .coordinator import AOSmithConfigEntry
TO_REDACT = {
"address",
diff --git a/homeassistant/components/aosmith/sensor.py b/homeassistant/components/aosmith/sensor.py
index b1c9852f647..071407e7b17 100644
--- a/homeassistant/components/aosmith/sensor.py
+++ b/homeassistant/components/aosmith/sensor.py
@@ -13,10 +13,13 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfEnergy
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AOSmithConfigEntry
-from .coordinator import AOSmithEnergyCoordinator, AOSmithStatusCoordinator
+from .coordinator import (
+ AOSmithConfigEntry,
+ AOSmithEnergyCoordinator,
+ AOSmithStatusCoordinator,
+)
from .entity import AOSmithEnergyEntity, AOSmithStatusEntity
@@ -40,7 +43,7 @@ STATUS_ENTITY_DESCRIPTIONS: tuple[AOSmithStatusSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AOSmithConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up A. O. Smith sensor platform."""
data = entry.runtime_data
diff --git a/homeassistant/components/aosmith/water_heater.py b/homeassistant/components/aosmith/water_heater.py
index f3dc8b3413f..d29b00955b6 100644
--- a/homeassistant/components/aosmith/water_heater.py
+++ b/homeassistant/components/aosmith/water_heater.py
@@ -15,10 +15,9 @@ from homeassistant.components.water_heater import (
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AOSmithConfigEntry
-from .coordinator import AOSmithStatusCoordinator
+from .coordinator import AOSmithConfigEntry, AOSmithStatusCoordinator
from .entity import AOSmithStatusEntity
MODE_HA_TO_AOSMITH = {
@@ -46,7 +45,7 @@ DEFAULT_OPERATION_MODE_PRIORITY = [
async def async_setup_entry(
hass: HomeAssistant,
entry: AOSmithConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up A. O. Smith water heater platform."""
data = entry.runtime_data
diff --git a/homeassistant/components/apcupsd/__init__.py b/homeassistant/components/apcupsd/__init__.py
index 44edc5c151f..e444f1cd735 100644
--- a/homeassistant/components/apcupsd/__init__.py
+++ b/homeassistant/components/apcupsd/__init__.py
@@ -4,13 +4,10 @@ from __future__ import annotations
from typing import Final
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT, Platform
from homeassistant.core import HomeAssistant
-from .coordinator import APCUPSdCoordinator
-
-type APCUPSdConfigEntry = ConfigEntry[APCUPSdCoordinator]
+from .coordinator import APCUPSdConfigEntry, APCUPSdCoordinator
PLATFORMS: Final = (Platform.BINARY_SENSOR, Platform.SENSOR)
@@ -20,7 +17,7 @@ async def async_setup_entry(
) -> bool:
"""Use config values to set up a function enabling status retrieval."""
host, port = config_entry.data[CONF_HOST], config_entry.data[CONF_PORT]
- coordinator = APCUPSdCoordinator(hass, host, port)
+ coordinator = APCUPSdCoordinator(hass, config_entry, host, port)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/apcupsd/binary_sensor.py b/homeassistant/components/apcupsd/binary_sensor.py
index cd9e60f7ae4..f3829b41f61 100644
--- a/homeassistant/components/apcupsd/binary_sensor.py
+++ b/homeassistant/components/apcupsd/binary_sensor.py
@@ -9,11 +9,10 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import APCUPSdConfigEntry
-from .coordinator import APCUPSdCoordinator
+from .coordinator import APCUPSdConfigEntry, APCUPSdCoordinator
PARALLEL_UPDATES = 0
@@ -28,7 +27,7 @@ _VALUE_ONLINE_MASK: Final = 0b1000
async def async_setup_entry(
hass: HomeAssistant,
config_entry: APCUPSdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an APCUPSd Online Status binary sensor."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/apcupsd/coordinator.py b/homeassistant/components/apcupsd/coordinator.py
index 1ae12d8c4b0..e2c1af50cee 100644
--- a/homeassistant/components/apcupsd/coordinator.py
+++ b/homeassistant/components/apcupsd/coordinator.py
@@ -25,6 +25,8 @@ _LOGGER = logging.getLogger(__name__)
UPDATE_INTERVAL: Final = timedelta(seconds=60)
REQUEST_REFRESH_COOLDOWN: Final = 5
+type APCUPSdConfigEntry = ConfigEntry[APCUPSdCoordinator]
+
class APCUPSdData(dict[str, str]):
"""Store data about an APCUPSd and provide a few helper methods for easier accesses."""
@@ -57,13 +59,20 @@ class APCUPSdCoordinator(DataUpdateCoordinator[APCUPSdData]):
updates from the server.
"""
- config_entry: ConfigEntry
+ config_entry: APCUPSdConfigEntry
- def __init__(self, hass: HomeAssistant, host: str, port: int) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: APCUPSdConfigEntry,
+ host: str,
+ port: int,
+ ) -> None:
"""Initialize the data object."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=UPDATE_INTERVAL,
request_refresh_debouncer=Debouncer(
diff --git a/homeassistant/components/apcupsd/diagnostics.py b/homeassistant/components/apcupsd/diagnostics.py
index fa0908f3144..a4bbf2191d2 100644
--- a/homeassistant/components/apcupsd/diagnostics.py
+++ b/homeassistant/components/apcupsd/diagnostics.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant
-from . import APCUPSdConfigEntry
+from .coordinator import APCUPSdConfigEntry
TO_REDACT = {"SERIALNO", "HOSTNAME"}
diff --git a/homeassistant/components/apcupsd/sensor.py b/homeassistant/components/apcupsd/sensor.py
index 9e0abcb1dd9..02016efa4ca 100644
--- a/homeassistant/components/apcupsd/sensor.py
+++ b/homeassistant/components/apcupsd/sensor.py
@@ -21,12 +21,11 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import APCUPSdConfigEntry
from .const import LAST_S_TEST
-from .coordinator import APCUPSdCoordinator
+from .coordinator import APCUPSdConfigEntry, APCUPSdCoordinator
PARALLEL_UPDATES = 0
@@ -407,7 +406,7 @@ INFERRED_UNITS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: APCUPSdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the APCUPSd sensors from config entries."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/apcupsd/strings.json b/homeassistant/components/apcupsd/strings.json
index 93102ac1393..fb5df9ec390 100644
--- a/homeassistant/components/apcupsd/strings.json
+++ b/homeassistant/components/apcupsd/strings.json
@@ -57,7 +57,7 @@
"name": "Status date"
},
"dip_switch_settings": {
- "name": "Dip switch settings"
+ "name": "DIP switch settings"
},
"low_battery_signal": {
"name": "Low battery signal"
diff --git a/homeassistant/components/apollo_automation/__init__.py b/homeassistant/components/apollo_automation/__init__.py
new file mode 100644
index 00000000000..7815b17818f
--- /dev/null
+++ b/homeassistant/components/apollo_automation/__init__.py
@@ -0,0 +1 @@
+"""Virtual integration: Apollo Automation."""
diff --git a/homeassistant/components/apollo_automation/manifest.json b/homeassistant/components/apollo_automation/manifest.json
new file mode 100644
index 00000000000..8e4c58f3f3d
--- /dev/null
+++ b/homeassistant/components/apollo_automation/manifest.json
@@ -0,0 +1,6 @@
+{
+ "domain": "apollo_automation",
+ "name": "Apollo Automation",
+ "integration_type": "virtual",
+ "supported_by": "esphome"
+}
diff --git a/homeassistant/components/apple_tv/__init__.py b/homeassistant/components/apple_tv/__init__.py
index f4417134b37..b911b3cec99 100644
--- a/homeassistant/components/apple_tv/__init__.py
+++ b/homeassistant/components/apple_tv/__init__.py
@@ -233,7 +233,6 @@ class AppleTVManager(DeviceListener):
pass
except Exception:
_LOGGER.exception("Failed to connect")
- await self.disconnect()
async def _connect_loop(self) -> None:
"""Connect loop background task function."""
diff --git a/homeassistant/components/apple_tv/media_player.py b/homeassistant/components/apple_tv/media_player.py
index 8a2336eea3b..b68d74e6115 100644
--- a/homeassistant/components/apple_tv/media_player.py
+++ b/homeassistant/components/apple_tv/media_player.py
@@ -39,7 +39,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import AppleTvConfigEntry, AppleTVManager
@@ -100,7 +100,7 @@ SUPPORT_FEATURE_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AppleTvConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Apple TV media player based on a config entry."""
name: str = config_entry.data[CONF_NAME]
diff --git a/homeassistant/components/apple_tv/remote.py b/homeassistant/components/apple_tv/remote.py
index 7f2c9f1b591..97e31bd4bb0 100644
--- a/homeassistant/components/apple_tv/remote.py
+++ b/homeassistant/components/apple_tv/remote.py
@@ -17,7 +17,7 @@ from homeassistant.components.remote import (
)
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AppleTvConfigEntry
from .entity import AppleTVEntity
@@ -38,7 +38,7 @@ COMMAND_TO_ATTRIBUTE = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AppleTvConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Apple TV remote based on a config entry."""
name: str = config_entry.data[CONF_NAME]
diff --git a/homeassistant/components/aprilaire/climate.py b/homeassistant/components/aprilaire/climate.py
index 194453046e6..5116e06b58f 100644
--- a/homeassistant/components/aprilaire/climate.py
+++ b/homeassistant/components/aprilaire/climate.py
@@ -18,7 +18,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import PRECISION_HALVES, PRECISION_WHOLE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
FAN_CIRCULATE,
@@ -63,7 +63,7 @@ FAN_MODE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AprilaireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add climates for passed config_entry in HA."""
diff --git a/homeassistant/components/aprilaire/humidifier.py b/homeassistant/components/aprilaire/humidifier.py
index 8a173e5e95e..fdb9233a0e3 100644
--- a/homeassistant/components/aprilaire/humidifier.py
+++ b/homeassistant/components/aprilaire/humidifier.py
@@ -15,7 +15,7 @@ from homeassistant.components.humidifier import (
HumidifierEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import AprilaireConfigEntry, AprilaireCoordinator
@@ -40,7 +40,7 @@ DEHUMIDIFIER_ACTION_MAP: dict[StateType, HumidifierAction] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AprilaireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Aprilaire humidifier devices."""
diff --git a/homeassistant/components/aprilaire/manifest.json b/homeassistant/components/aprilaire/manifest.json
index 577de8ae88d..b40460dd61b 100644
--- a/homeassistant/components/aprilaire/manifest.json
+++ b/homeassistant/components/aprilaire/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "device",
"iot_class": "local_push",
"loggers": ["pyaprilaire"],
- "requirements": ["pyaprilaire==0.7.7"]
+ "requirements": ["pyaprilaire==0.8.1"]
}
diff --git a/homeassistant/components/aprilaire/select.py b/homeassistant/components/aprilaire/select.py
index d8f6137f53d..c38c9e94501 100644
--- a/homeassistant/components/aprilaire/select.py
+++ b/homeassistant/components/aprilaire/select.py
@@ -10,7 +10,7 @@ from pyaprilaire.const import Attribute
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AprilaireConfigEntry, AprilaireCoordinator
from .entity import BaseAprilaireEntity
@@ -24,7 +24,7 @@ FRESH_AIR_MODE_MAP = {0: "off", 1: "automatic"}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AprilaireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Aprilaire select devices."""
diff --git a/homeassistant/components/aprilaire/sensor.py b/homeassistant/components/aprilaire/sensor.py
index e1909746364..bf3bd12f43d 100644
--- a/homeassistant/components/aprilaire/sensor.py
+++ b/homeassistant/components/aprilaire/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import AprilaireConfigEntry, AprilaireCoordinator
@@ -75,7 +75,7 @@ def get_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AprilaireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Aprilaire sensor devices."""
diff --git a/homeassistant/components/apsystems/__init__.py b/homeassistant/components/apsystems/__init__.py
index c437f5584db..cdc4563b92d 100644
--- a/homeassistant/components/apsystems/__init__.py
+++ b/homeassistant/components/apsystems/__init__.py
@@ -2,16 +2,13 @@
from __future__ import annotations
-from dataclasses import dataclass
-
from APsystemsEZ1 import APsystemsEZ1M
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_IP_ADDRESS, CONF_PORT, Platform
from homeassistant.core import HomeAssistant
from .const import DEFAULT_PORT
-from .coordinator import ApSystemsDataCoordinator
+from .coordinator import ApSystemsConfigEntry, ApSystemsData, ApSystemsDataCoordinator
PLATFORMS: list[Platform] = [
Platform.BINARY_SENSOR,
@@ -21,17 +18,6 @@ PLATFORMS: list[Platform] = [
]
-@dataclass
-class ApSystemsData:
- """Store runtime data."""
-
- coordinator: ApSystemsDataCoordinator
- device_id: str
-
-
-type ApSystemsConfigEntry = ConfigEntry[ApSystemsData]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: ApSystemsConfigEntry) -> bool:
"""Set up this integration using UI."""
api = APsystemsEZ1M(
@@ -40,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ApSystemsConfigEntry) ->
timeout=8,
enable_debounce=True,
)
- coordinator = ApSystemsDataCoordinator(hass, api)
+ coordinator = ApSystemsDataCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
assert entry.unique_id
entry.runtime_data = ApSystemsData(
@@ -51,6 +37,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: ApSystemsConfigEntry) ->
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: ApSystemsConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/apsystems/binary_sensor.py b/homeassistant/components/apsystems/binary_sensor.py
index 9e361ca883e..202d878014d 100644
--- a/homeassistant/components/apsystems/binary_sensor.py
+++ b/homeassistant/components/apsystems/binary_sensor.py
@@ -14,11 +14,10 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import ApSystemsConfigEntry, ApSystemsData
-from .coordinator import ApSystemsDataCoordinator
+from .coordinator import ApSystemsConfigEntry, ApSystemsData, ApSystemsDataCoordinator
from .entity import ApSystemsEntity
@@ -64,7 +63,7 @@ BINARY_SENSORS: tuple[ApsystemsLocalApiBinarySensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ApSystemsConfigEntry,
- add_entities: AddEntitiesCallback,
+ add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary sensor platform."""
config = config_entry.runtime_data
diff --git a/homeassistant/components/apsystems/coordinator.py b/homeassistant/components/apsystems/coordinator.py
index 2535c66c4ac..ca423055176 100644
--- a/homeassistant/components/apsystems/coordinator.py
+++ b/homeassistant/components/apsystems/coordinator.py
@@ -12,6 +12,7 @@ from APsystemsEZ1 import (
ReturnOutputData,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -26,16 +27,34 @@ class ApSystemsSensorData:
alarm_info: ReturnAlarmInfo
+@dataclass
+class ApSystemsData:
+ """Store runtime data."""
+
+ coordinator: ApSystemsDataCoordinator
+ device_id: str
+
+
+type ApSystemsConfigEntry = ConfigEntry[ApSystemsData]
+
+
class ApSystemsDataCoordinator(DataUpdateCoordinator[ApSystemsSensorData]):
"""Coordinator used for all sensors."""
+ config_entry: ApSystemsConfigEntry
device_version: str
- def __init__(self, hass: HomeAssistant, api: APsystemsEZ1M) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: ApSystemsConfigEntry,
+ api: APsystemsEZ1M,
+ ) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="APSystems Data",
update_interval=timedelta(seconds=12),
)
diff --git a/homeassistant/components/apsystems/entity.py b/homeassistant/components/apsystems/entity.py
index 7770b451680..2ce8becbf80 100644
--- a/homeassistant/components/apsystems/entity.py
+++ b/homeassistant/components/apsystems/entity.py
@@ -5,8 +5,8 @@ from __future__ import annotations
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from . import ApSystemsData
from .const import DOMAIN
+from .coordinator import ApSystemsData
class ApSystemsEntity(Entity):
@@ -19,10 +19,20 @@ class ApSystemsEntity(Entity):
data: ApSystemsData,
) -> None:
"""Initialize the APsystems entity."""
+
+ # Handle device version safely
+ sw_version = None
+ if data.coordinator.device_version:
+ version_parts = data.coordinator.device_version.split(" ")
+ if len(version_parts) > 1:
+ sw_version = version_parts[1]
+ else:
+ sw_version = version_parts[0]
+
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, data.device_id)},
manufacturer="APsystems",
model="EZ1-M",
serial_number=data.device_id,
- sw_version=data.coordinator.device_version.split(" ")[1],
+ sw_version=sw_version,
)
diff --git a/homeassistant/components/apsystems/number.py b/homeassistant/components/apsystems/number.py
index b5ed60a7754..b43b21f2b71 100644
--- a/homeassistant/components/apsystems/number.py
+++ b/homeassistant/components/apsystems/number.py
@@ -7,17 +7,17 @@ from aiohttp import ClientConnectorError
from homeassistant.components.number import NumberDeviceClass, NumberEntity, NumberMode
from homeassistant.const import UnitOfPower
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import DiscoveryInfoType
-from . import ApSystemsConfigEntry, ApSystemsData
+from .coordinator import ApSystemsConfigEntry, ApSystemsData
from .entity import ApSystemsEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ApSystemsConfigEntry,
- add_entities: AddEntitiesCallback,
+ add_entities: AddConfigEntryEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up the sensor platform."""
diff --git a/homeassistant/components/apsystems/sensor.py b/homeassistant/components/apsystems/sensor.py
index f87bc0f3f26..6e654cfbf61 100644
--- a/homeassistant/components/apsystems/sensor.py
+++ b/homeassistant/components/apsystems/sensor.py
@@ -15,12 +15,11 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import DiscoveryInfoType, StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import ApSystemsConfigEntry, ApSystemsData
-from .coordinator import ApSystemsDataCoordinator
+from .coordinator import ApSystemsConfigEntry, ApSystemsData, ApSystemsDataCoordinator
from .entity import ApSystemsEntity
@@ -110,7 +109,7 @@ SENSORS: tuple[ApsystemsLocalApiSensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ApSystemsConfigEntry,
- add_entities: AddEntitiesCallback,
+ add_entities: AddConfigEntryEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up the sensor platform."""
diff --git a/homeassistant/components/apsystems/switch.py b/homeassistant/components/apsystems/switch.py
index 73914845445..e1017f95448 100644
--- a/homeassistant/components/apsystems/switch.py
+++ b/homeassistant/components/apsystems/switch.py
@@ -9,16 +9,16 @@ from APsystemsEZ1 import InverterReturnedError
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import ApSystemsConfigEntry, ApSystemsData
+from .coordinator import ApSystemsConfigEntry, ApSystemsData
from .entity import ApSystemsEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ApSystemsConfigEntry,
- add_entities: AddEntitiesCallback,
+ add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switch platform."""
diff --git a/homeassistant/components/aquacell/sensor.py b/homeassistant/components/aquacell/sensor.py
index a76d26244ad..77cd3cdd60a 100644
--- a/homeassistant/components/aquacell/sensor.py
+++ b/homeassistant/components/aquacell/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import AquacellConfigEntry, AquacellCoordinator
@@ -83,7 +83,7 @@ SENSORS: tuple[SoftenerSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AquacellConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
softeners = config_entry.runtime_data.data
diff --git a/homeassistant/components/aranet/sensor.py b/homeassistant/components/aranet/sensor.py
index b5187cba1f4..ee2eb8c8a75 100644
--- a/homeassistant/components/aranet/sensor.py
+++ b/homeassistant/components/aranet/sensor.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from dataclasses import dataclass
from typing import Any
-from aranet4.client import Aranet4Advertisement
+from aranet4.client import Aranet4Advertisement, Color
from bleak.backends.device import BLEDevice
from homeassistant.components.bluetooth.passive_update_processor import (
@@ -35,7 +35,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AranetConfigEntry
from .const import ARANET_MANUFACTURER_NAME
@@ -74,6 +74,13 @@ SENSOR_DESCRIPTIONS = {
native_unit_of_measurement=UnitOfPressure.HPA,
state_class=SensorStateClass.MEASUREMENT,
),
+ "status": AranetSensorEntityDescription(
+ key="threshold",
+ translation_key="threshold",
+ name="Threshold",
+ device_class=SensorDeviceClass.ENUM,
+ options=[status.name.lower() for status in Color],
+ ),
"co2": AranetSensorEntityDescription(
key="co2",
name="Carbon Dioxide",
@@ -161,7 +168,10 @@ def sensor_update_to_bluetooth_data_update(
val = getattr(adv.readings, key)
if val == -1:
continue
- val *= desc.scale
+ if key == "status":
+ val = val.name.lower()
+ else:
+ val *= desc.scale
data[tag] = val
names[tag] = desc.name
descs[tag] = desc
@@ -176,7 +186,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: AranetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Aranet sensors."""
processor = PassiveBluetoothDataProcessor(sensor_update_to_bluetooth_data_update)
diff --git a/homeassistant/components/aranet/strings.json b/homeassistant/components/aranet/strings.json
index 1cc695637d4..f786f4b2d4d 100644
--- a/homeassistant/components/aranet/strings.json
+++ b/homeassistant/components/aranet/strings.json
@@ -21,5 +21,17 @@
"no_devices_found": "No unconfigured Aranet devices found.",
"outdated_version": "This device is using outdated firmware. Please update it to at least v1.2.0 and try again."
}
+ },
+ "entity": {
+ "sensor": {
+ "threshold": {
+ "state": {
+ "error": "Error",
+ "green": "Green",
+ "yellow": "Yellow",
+ "red": "Red"
+ }
+ }
+ }
}
}
diff --git a/homeassistant/components/arcam_fmj/manifest.json b/homeassistant/components/arcam_fmj/manifest.json
index 39d289f9cb1..41396eca5d6 100644
--- a/homeassistant/components/arcam_fmj/manifest.json
+++ b/homeassistant/components/arcam_fmj/manifest.json
@@ -6,7 +6,7 @@
"documentation": "https://www.home-assistant.io/integrations/arcam_fmj",
"iot_class": "local_polling",
"loggers": ["arcam"],
- "requirements": ["arcam-fmj==1.5.2"],
+ "requirements": ["arcam-fmj==1.8.1"],
"ssdp": [
{
"deviceType": "urn:schemas-upnp-org:device:MediaRenderer:1",
diff --git a/homeassistant/components/arcam_fmj/media_player.py b/homeassistant/components/arcam_fmj/media_player.py
index 7a133777a0a..cd4ed7bbb05 100644
--- a/homeassistant/components/arcam_fmj/media_player.py
+++ b/homeassistant/components/arcam_fmj/media_player.py
@@ -24,7 +24,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ArcamFmjConfigEntry
from .const import (
@@ -41,7 +41,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ArcamFmjConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the configuration entry."""
diff --git a/homeassistant/components/arve/__init__.py b/homeassistant/components/arve/__init__.py
index a1b4aa7042e..c5900967bde 100644
--- a/homeassistant/components/arve/__init__.py
+++ b/homeassistant/components/arve/__init__.py
@@ -13,7 +13,7 @@ PLATFORMS: list[Platform] = [Platform.SENSOR]
async def async_setup_entry(hass: HomeAssistant, entry: ArveConfigEntry) -> bool:
"""Set up Arve from a config entry."""
- coordinator = ArveCoordinator(hass)
+ coordinator = ArveCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/arve/coordinator.py b/homeassistant/components/arve/coordinator.py
index f02220e28e2..4b08efd639e 100644
--- a/homeassistant/components/arve/coordinator.py
+++ b/homeassistant/components/arve/coordinator.py
@@ -30,11 +30,12 @@ class ArveCoordinator(DataUpdateCoordinator[ArveSensProData]):
config_entry: ArveConfigEntry
devices: ArveDevices
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: ArveConfigEntry) -> None:
"""Initialize Arve coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/arve/sensor.py b/homeassistant/components/arve/sensor.py
index 64d9f6f8874..dea7110f611 100644
--- a/homeassistant/components/arve/sensor.py
+++ b/homeassistant/components/arve/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ArveConfigEntry
from .entity import ArveDeviceEntity
@@ -83,7 +83,9 @@ SENSORS: tuple[ArveDeviceEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ArveConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ArveConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Arve device based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/arwn/sensor.py b/homeassistant/components/arwn/sensor.py
index ada96c07340..a31156bbba6 100644
--- a/homeassistant/components/arwn/sensor.py
+++ b/homeassistant/components/arwn/sensor.py
@@ -92,7 +92,12 @@ def discover_sensors(topic: str, payload: dict[str, Any]) -> list[ArwnSensor] |
device_class=SensorDeviceClass.WIND_SPEED,
),
ArwnSensor(
- topic + "/dir", "Wind Direction", "direction", DEGREE, "mdi:compass"
+ topic + "/dir",
+ "Wind Direction",
+ "direction",
+ DEGREE,
+ "mdi:compass",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
]
return None
diff --git a/homeassistant/components/aseko_pool_live/__init__.py b/homeassistant/components/aseko_pool_live/__init__.py
index 52d74398818..012b5a19b0f 100644
--- a/homeassistant/components/aseko_pool_live/__init__.py
+++ b/homeassistant/components/aseko_pool_live/__init__.py
@@ -26,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: AsekoConfigEntry) -> boo
except AsekoNotLoggedIn as err:
raise ConfigEntryAuthFailed from err
- coordinator = AsekoDataUpdateCoordinator(hass, aseko)
+ coordinator = AsekoDataUpdateCoordinator(hass, entry, aseko)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/aseko_pool_live/binary_sensor.py b/homeassistant/components/aseko_pool_live/binary_sensor.py
index c8cc31dc795..15c72614ee1 100644
--- a/homeassistant/components/aseko_pool_live/binary_sensor.py
+++ b/homeassistant/components/aseko_pool_live/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AsekoConfigEntry
from .entity import AsekoEntity
@@ -37,7 +37,7 @@ BINARY_SENSORS: tuple[AsekoBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AsekoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Aseko Pool Live binary sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/aseko_pool_live/coordinator.py b/homeassistant/components/aseko_pool_live/coordinator.py
index 96893912361..d54aa756ddd 100644
--- a/homeassistant/components/aseko_pool_live/coordinator.py
+++ b/homeassistant/components/aseko_pool_live/coordinator.py
@@ -21,13 +21,18 @@ type AsekoConfigEntry = ConfigEntry[AsekoDataUpdateCoordinator]
class AsekoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Unit]]):
"""Class to manage fetching Aseko unit data from single endpoint."""
- def __init__(self, hass: HomeAssistant, aseko: Aseko) -> None:
+ config_entry: AsekoConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: AsekoConfigEntry, aseko: Aseko
+ ) -> None:
"""Initialize global Aseko unit data updater."""
self._aseko = aseko
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=2),
)
diff --git a/homeassistant/components/aseko_pool_live/sensor.py b/homeassistant/components/aseko_pool_live/sensor.py
index 3fe7cdd5272..f9a7287a9f1 100644
--- a/homeassistant/components/aseko_pool_live/sensor.py
+++ b/homeassistant/components/aseko_pool_live/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfElectricPotential, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import AsekoConfigEntry
@@ -86,7 +86,7 @@ SENSORS: list[AsekoSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AsekoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Aseko Pool Live sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/assist_pipeline/__init__.py b/homeassistant/components/assist_pipeline/__init__.py
index 9a32821e3a0..59bd987d90e 100644
--- a/homeassistant/components/assist_pipeline/__init__.py
+++ b/homeassistant/components/assist_pipeline/__init__.py
@@ -117,7 +117,7 @@ async def async_pipeline_from_audio_stream(
"""
with chat_session.async_get_chat_session(hass, conversation_id) as session:
pipeline_input = PipelineInput(
- conversation_id=session.conversation_id,
+ session=session,
device_id=device_id,
stt_metadata=stt_metadata,
stt_stream=stt_stream,
diff --git a/homeassistant/components/assist_pipeline/pipeline.py b/homeassistant/components/assist_pipeline/pipeline.py
index 94e2b04d7ae..42bb2d4ced8 100644
--- a/homeassistant/components/assist_pipeline/pipeline.py
+++ b/homeassistant/components/assist_pipeline/pipeline.py
@@ -13,24 +13,17 @@ from pathlib import Path
from queue import Empty, Queue
from threading import Thread
import time
-from typing import Any, Literal, cast
+from typing import TYPE_CHECKING, Any, Literal, cast
import wave
import hass_nabucasa
import voluptuous as vol
-from homeassistant.components import (
- conversation,
- media_source,
- stt,
- tts,
- wake_word,
- websocket_api,
-)
+from homeassistant.components import conversation, stt, tts, wake_word, websocket_api
from homeassistant.components.tts import (
generate_media_source_id as tts_generate_media_source_id,
)
-from homeassistant.const import MATCH_ALL
+from homeassistant.const import ATTR_SUPPORTED_FEATURES, MATCH_ALL
from homeassistant.core import Context, HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import chat_session, intent
@@ -81,6 +74,9 @@ from .error import (
)
from .vad import AudioBuffer, VoiceActivityTimeout, VoiceCommandSegmenter, chunk_samples
+if TYPE_CHECKING:
+ from hassil.recognize import RecognizeResult
+
_LOGGER = logging.getLogger(__name__)
STORAGE_KEY = f"{DOMAIN}.pipelines"
@@ -93,6 +89,9 @@ ENGINE_LANGUAGE_PAIRS = (
)
KEY_ASSIST_PIPELINE: HassKey[PipelineData] = HassKey(DOMAIN)
+KEY_PIPELINE_CONVERSATION_DATA: HassKey[dict[str, PipelineConversationData]] = HassKey(
+ "pipeline_conversation_data"
+)
def validate_language(data: dict[str, Any]) -> Any:
@@ -123,6 +122,12 @@ STORED_PIPELINE_RUNS = 10
SAVE_DELAY = 10
+@callback
+def _async_local_fallback_intent_filter(result: RecognizeResult) -> bool:
+ """Filter out intents that are not local fallback."""
+ return result.intent.name in (intent.INTENT_GET_STATE, intent.INTENT_NEVERMIND)
+
+
@callback
def _async_resolve_default_pipeline_settings(
hass: HomeAssistant,
@@ -374,6 +379,7 @@ class PipelineEventType(StrEnum):
STT_VAD_END = "stt-vad-end"
STT_END = "stt-end"
INTENT_START = "intent-start"
+ INTENT_PROGRESS = "intent-progress"
INTENT_END = "intent-end"
TTS_START = "tts-start"
TTS_END = "tts-end"
@@ -556,8 +562,7 @@ class PipelineRun:
id: str = field(default_factory=ulid_util.ulid_now)
stt_provider: stt.SpeechToTextEntity | stt.Provider = field(init=False, repr=False)
- tts_engine: str = field(init=False, repr=False)
- tts_options: dict | None = field(init=False, default=None)
+ tts_stream: tts.ResultStream | None = field(init=False, default=None)
wake_word_entity_id: str | None = field(init=False, default=None, repr=False)
wake_word_entity: wake_word.WakeWordDetectionEntity = field(init=False, repr=False)
@@ -580,6 +585,12 @@ class PipelineRun:
_device_id: str | None = None
"""Optional device id set during run start."""
+ _conversation_data: PipelineConversationData | None = None
+ """Data tied to the conversation ID."""
+
+ _intent_agent_only = False
+ """If request should only be handled by agent, ignoring sentence triggers and local processing."""
+
def __post_init__(self) -> None:
"""Set language for pipeline."""
self.language = self.pipeline.language or self.hass.config.language
@@ -629,13 +640,19 @@ class PipelineRun:
self._device_id = device_id
self._start_debug_recording_thread()
- data = {
+ data: dict[str, Any] = {
"pipeline": self.pipeline.id,
"language": self.language,
"conversation_id": conversation_id,
}
if self.runner_data is not None:
data["runner_data"] = self.runner_data
+ if self.tts_stream:
+ data["tts_output"] = {
+ "token": self.tts_stream.token,
+ "url": self.tts_stream.url,
+ "mime_type": self.tts_stream.content_type,
+ }
self.process_event(PipelineEvent(PipelineEventType.RUN_START, data))
@@ -997,19 +1014,36 @@ class PipelineRun:
yield chunk.audio
- async def prepare_recognize_intent(self) -> None:
+ async def prepare_recognize_intent(self, session: chat_session.ChatSession) -> None:
"""Prepare recognizing an intent."""
- agent_info = conversation.async_get_agent_info(
- self.hass,
- self.pipeline.conversation_engine or conversation.HOME_ASSISTANT_AGENT,
+ self._conversation_data = async_get_pipeline_conversation_data(
+ self.hass, session
)
- if agent_info is None:
- engine = self.pipeline.conversation_engine or "default"
- raise IntentRecognitionError(
- code="intent-not-supported",
- message=f"Intent recognition engine {engine} is not found",
+ if self._conversation_data.continue_conversation_agent is not None:
+ agent_info = conversation.async_get_agent_info(
+ self.hass, self._conversation_data.continue_conversation_agent
)
+ self._conversation_data.continue_conversation_agent = None
+ if agent_info is None:
+ raise IntentRecognitionError(
+ code="intent-agent-not-found",
+ message=f"Intent recognition engine {self._conversation_data.continue_conversation_agent} asked for follow-up but is no longer found",
+ )
+ self._intent_agent_only = True
+
+ else:
+ agent_info = conversation.async_get_agent_info(
+ self.hass,
+ self.pipeline.conversation_engine or conversation.HOME_ASSISTANT_AGENT,
+ )
+
+ if agent_info is None:
+ engine = self.pipeline.conversation_engine or "default"
+ raise IntentRecognitionError(
+ code="intent-not-supported",
+ message=f"Intent recognition engine {engine} is not found",
+ )
self.intent_agent = agent_info.id
@@ -1021,7 +1055,7 @@ class PipelineRun:
conversation_extra_system_prompt: str | None,
) -> str:
"""Run intent recognition portion of pipeline. Returns text to speak."""
- if self.intent_agent is None:
+ if self.intent_agent is None or self._conversation_data is None:
raise RuntimeError("Recognize intent was not prepared")
if self.pipeline.conversation_language == MATCH_ALL:
@@ -1068,7 +1102,7 @@ class PipelineRun:
agent_id = self.intent_agent
processed_locally = agent_id == conversation.HOME_ASSISTANT_AGENT
intent_response: intent.IntentResponse | None = None
- if not processed_locally:
+ if not processed_locally and not self._intent_agent_only:
# Sentence triggers override conversation agent
if (
trigger_response_text
@@ -1083,26 +1117,59 @@ class PipelineRun:
)
intent_response.async_set_speech(trigger_response_text)
- # Try local intents first, if preferred.
- elif self.pipeline.prefer_local_intents and (
- intent_response := await conversation.async_handle_intents(
- self.hass, user_input
+ intent_filter: Callable[[RecognizeResult], bool] | None = None
+ # If the LLM has API access, we filter out some sentences that are
+ # interfering with LLM operation.
+ if (
+ intent_agent_state := self.hass.states.get(self.intent_agent)
+ ) and intent_agent_state.attributes.get(
+ ATTR_SUPPORTED_FEATURES, 0
+ ) & conversation.ConversationEntityFeature.CONTROL:
+ intent_filter = _async_local_fallback_intent_filter
+
+ # Try local intents
+ if (
+ intent_response is None
+ and self.pipeline.prefer_local_intents
+ and (
+ intent_response := await conversation.async_handle_intents(
+ self.hass,
+ user_input,
+ intent_filter=intent_filter,
+ )
)
):
# Local intent matched
agent_id = conversation.HOME_ASSISTANT_AGENT
processed_locally = True
- # It was already handled, create response and add to chat history
- if intent_response is not None:
- with (
- chat_session.async_get_chat_session(
- self.hass, user_input.conversation_id
- ) as session,
- conversation.async_get_chat_log(
- self.hass, session, user_input
- ) as chat_log,
- ):
+ @callback
+ def chat_log_delta_listener(
+ chat_log: conversation.ChatLog, delta: dict
+ ) -> None:
+ """Handle chat log delta."""
+ self.process_event(
+ PipelineEvent(
+ PipelineEventType.INTENT_PROGRESS,
+ {
+ "chat_log_delta": delta,
+ },
+ )
+ )
+
+ with (
+ chat_session.async_get_chat_session(
+ self.hass, user_input.conversation_id
+ ) as session,
+ conversation.async_get_chat_log(
+ self.hass,
+ session,
+ user_input,
+ chat_log_delta_listener=chat_log_delta_listener,
+ ) as chat_log,
+ ):
+ # It was already handled, create response and add to chat history
+ if intent_response is not None:
speech: str = intent_response.speech.get("plain", {}).get(
"speech", ""
)
@@ -1117,21 +1184,21 @@ class PipelineRun:
conversation_id=session.conversation_id,
)
- else:
- # Fall back to pipeline conversation agent
- conversation_result = await conversation.async_converse(
- hass=self.hass,
- text=user_input.text,
- conversation_id=user_input.conversation_id,
- device_id=user_input.device_id,
- context=user_input.context,
- language=user_input.language,
- agent_id=user_input.agent_id,
- extra_system_prompt=user_input.extra_system_prompt,
- )
- speech = conversation_result.response.speech.get("plain", {}).get(
- "speech", ""
- )
+ else:
+ # Fall back to pipeline conversation agent
+ conversation_result = await conversation.async_converse(
+ hass=self.hass,
+ text=user_input.text,
+ conversation_id=user_input.conversation_id,
+ device_id=user_input.device_id,
+ context=user_input.context,
+ language=user_input.language,
+ agent_id=user_input.agent_id,
+ extra_system_prompt=user_input.extra_system_prompt,
+ )
+ speech = conversation_result.response.speech.get("plain", {}).get(
+ "speech", ""
+ )
except Exception as src_error:
_LOGGER.exception("Unexpected error during intent recognition")
@@ -1152,6 +1219,9 @@ class PipelineRun:
)
)
+ if conversation_result.continue_conversation:
+ self._conversation_data.continue_conversation_agent = agent_id
+
return speech
async def prepare_text_to_speech(self) -> None:
@@ -1174,36 +1244,31 @@ class PipelineRun:
tts_options[tts.ATTR_PREFERRED_SAMPLE_BYTES] = SAMPLE_WIDTH
try:
- options_supported = await tts.async_support_options(
- self.hass,
- engine,
- self.pipeline.tts_language,
- tts_options,
+ self.tts_stream = tts.async_create_stream(
+ hass=self.hass,
+ engine=engine,
+ language=self.pipeline.tts_language,
+ options=tts_options,
)
except HomeAssistantError as err:
- raise TextToSpeechError(
- code="tts-not-supported",
- message=f"Text-to-speech engine '{engine}' not found",
- ) from err
- if not options_supported:
raise TextToSpeechError(
code="tts-not-supported",
message=(
f"Text-to-speech engine {engine} "
- f"does not support language {self.pipeline.tts_language} or options {tts_options}"
+ f"does not support language {self.pipeline.tts_language} or options {tts_options}:"
+ f" {err}"
),
- )
-
- self.tts_engine = engine
- self.tts_options = tts_options
+ ) from err
async def text_to_speech(self, tts_input: str) -> None:
"""Run text-to-speech portion of pipeline."""
+ assert self.tts_stream is not None
+
self.process_event(
PipelineEvent(
PipelineEventType.TTS_START,
{
- "engine": self.tts_engine,
+ "engine": self.tts_stream.engine,
"language": self.pipeline.tts_language,
"voice": self.pipeline.tts_voice,
"tts_input": tts_input,
@@ -1216,14 +1281,9 @@ class PipelineRun:
tts_media_id = tts_generate_media_source_id(
self.hass,
tts_input,
- engine=self.tts_engine,
- language=self.pipeline.tts_language,
- options=self.tts_options,
- )
- tts_media = await media_source.async_resolve_media(
- self.hass,
- tts_media_id,
- None,
+ engine=self.tts_stream.engine,
+ language=self.tts_stream.language,
+ options=self.tts_stream.options,
)
except Exception as src_error:
_LOGGER.exception("Unexpected error during text-to-speech")
@@ -1232,10 +1292,13 @@ class PipelineRun:
message="Unexpected error during text-to-speech",
) from src_error
- _LOGGER.debug("TTS result %s", tts_media)
+ self.tts_stream.async_set_message(tts_input)
+
tts_output = {
"media_id": tts_media_id,
- **asdict(tts_media),
+ "token": self.tts_stream.token,
+ "url": self.tts_stream.url,
+ "mime_type": self.tts_stream.content_type,
}
self.process_event(
@@ -1415,8 +1478,8 @@ class PipelineInput:
run: PipelineRun
- conversation_id: str
- """Identifier for the conversation."""
+ session: chat_session.ChatSession
+ """Session for the conversation."""
stt_metadata: stt.SpeechMetadata | None = None
"""Metadata of stt input audio. Required when start_stage = stt."""
@@ -1441,7 +1504,9 @@ class PipelineInput:
async def execute(self) -> None:
"""Run pipeline."""
- self.run.start(conversation_id=self.conversation_id, device_id=self.device_id)
+ self.run.start(
+ conversation_id=self.session.conversation_id, device_id=self.device_id
+ )
current_stage: PipelineStage | None = self.run.start_stage
stt_audio_buffer: list[EnhancedAudioChunk] = []
stt_processed_stream: AsyncIterable[EnhancedAudioChunk] | None = None
@@ -1525,7 +1590,7 @@ class PipelineInput:
assert intent_input is not None
tts_input = await self.run.recognize_intent(
intent_input,
- self.conversation_id,
+ self.session.conversation_id,
self.device_id,
self.conversation_extra_system_prompt,
)
@@ -1609,7 +1674,7 @@ class PipelineInput:
<= PIPELINE_STAGE_ORDER.index(PipelineStage.INTENT)
<= end_stage_index
):
- prepare_tasks.append(self.run.prepare_recognize_intent())
+ prepare_tasks.append(self.run.prepare_recognize_intent(self.session))
if (
start_stage_index
@@ -1888,7 +1953,7 @@ class PipelineRunDebug:
class PipelineStore(Store[SerializedPipelineStorageCollection]):
- """Store entity registry data."""
+ """Store pipeline data."""
async def _async_migrate_func(
self,
@@ -1970,3 +2035,37 @@ async def async_run_migrations(hass: HomeAssistant) -> None:
for pipeline, attr_updates in updates:
await async_update_pipeline(hass, pipeline, **attr_updates)
+
+
+@dataclass
+class PipelineConversationData:
+ """Hold data for the duration of a conversation."""
+
+ continue_conversation_agent: str | None = None
+ """The agent that requested the conversation to be continued."""
+
+
+@callback
+def async_get_pipeline_conversation_data(
+ hass: HomeAssistant, session: chat_session.ChatSession
+) -> PipelineConversationData:
+ """Get the pipeline data for a specific conversation."""
+ all_conversation_data = hass.data.get(KEY_PIPELINE_CONVERSATION_DATA)
+ if all_conversation_data is None:
+ all_conversation_data = {}
+ hass.data[KEY_PIPELINE_CONVERSATION_DATA] = all_conversation_data
+
+ data = all_conversation_data.get(session.conversation_id)
+
+ if data is not None:
+ return data
+
+ @callback
+ def do_cleanup() -> None:
+ """Handle cleanup."""
+ all_conversation_data.pop(session.conversation_id)
+
+ session.async_on_cleanup(do_cleanup)
+
+ data = all_conversation_data[session.conversation_id] = PipelineConversationData()
+ return data
diff --git a/homeassistant/components/assist_pipeline/websocket_api.py b/homeassistant/components/assist_pipeline/websocket_api.py
index d2d54a1b7c3..937b3a0ea45 100644
--- a/homeassistant/components/assist_pipeline/websocket_api.py
+++ b/homeassistant/components/assist_pipeline/websocket_api.py
@@ -239,7 +239,7 @@ async def websocket_run(
with chat_session.async_get_chat_session(
hass, msg.get("conversation_id")
) as session:
- input_args["conversation_id"] = session.conversation_id
+ input_args["session"] = session
pipeline_input = PipelineInput(**input_args)
try:
diff --git a/homeassistant/components/assist_satellite/entity.py b/homeassistant/components/assist_satellite/entity.py
index e43abb4539c..3db38a23889 100644
--- a/homeassistant/components/assist_satellite/entity.py
+++ b/homeassistant/components/assist_satellite/entity.py
@@ -23,9 +23,6 @@ from homeassistant.components.assist_pipeline import (
vad,
)
from homeassistant.components.media_player import async_process_play_media_url
-from homeassistant.components.tts import (
- generate_media_source_id as tts_generate_media_source_id,
-)
from homeassistant.core import Context, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import chat_session, entity
@@ -98,6 +95,9 @@ class AssistSatelliteAnnouncement:
original_media_id: str
"""The raw media ID before processing."""
+ tts_token: str | None
+ """The TTS token of the media."""
+
media_id_source: Literal["url", "media_id", "tts"]
"""Source of the media ID."""
@@ -405,7 +405,10 @@ class AssistSatelliteEntity(entity.Entity):
def _internal_on_pipeline_event(self, event: PipelineEvent) -> None:
"""Set state based on pipeline stage."""
if event.type is PipelineEventType.WAKE_WORD_START:
- self._set_state(AssistSatelliteState.IDLE)
+ # Only return to idle if we're not currently responding.
+ # The state will return to idle in tts_response_finished.
+ if self.state != AssistSatelliteState.RESPONDING:
+ self._set_state(AssistSatelliteState.IDLE)
elif event.type is PipelineEventType.STT_START:
self._set_state(AssistSatelliteState.LISTENING)
elif event.type is PipelineEventType.INTENT_START:
@@ -471,6 +474,7 @@ class AssistSatelliteEntity(entity.Entity):
) -> AssistSatelliteAnnouncement:
"""Resolve the media ID."""
media_id_source: Literal["url", "media_id", "tts"] | None = None
+ tts_token: str | None = None
if media_id:
original_media_id = media_id
@@ -481,6 +485,10 @@ class AssistSatelliteEntity(entity.Entity):
pipeline_id = self._resolve_pipeline()
pipeline = async_get_pipeline(self.hass, pipeline_id)
+ engine = tts.async_resolve_engine(self.hass, pipeline.tts_engine)
+ if engine is None:
+ raise HomeAssistantError(f"TTS engine {pipeline.tts_engine} not found")
+
tts_options: dict[str, Any] = {}
if pipeline.tts_voice is not None:
tts_options[tts.ATTR_VOICE] = pipeline.tts_voice
@@ -488,14 +496,23 @@ class AssistSatelliteEntity(entity.Entity):
if self.tts_options is not None:
tts_options.update(self.tts_options)
- media_id = tts_generate_media_source_id(
+ stream = tts.async_create_stream(
self.hass,
- message,
- engine=pipeline.tts_engine,
+ engine=engine,
+ language=pipeline.tts_language,
+ options=tts_options,
+ )
+ stream.async_set_message(message)
+
+ tts_token = stream.token
+ media_id = stream.url
+ original_media_id = tts.generate_media_source_id(
+ self.hass,
+ message,
+ engine=engine,
language=pipeline.tts_language,
options=tts_options,
)
- original_media_id = media_id
if media_source.is_media_source_id(media_id):
if not media_id_source:
@@ -514,5 +531,9 @@ class AssistSatelliteEntity(entity.Entity):
media_id = async_process_play_media_url(self.hass, media_id)
return AssistSatelliteAnnouncement(
- message, media_id, original_media_id, media_id_source
+ message=message,
+ media_id=media_id,
+ original_media_id=original_media_id,
+ tts_token=tts_token,
+ media_id_source=media_id_source,
)
diff --git a/homeassistant/components/assist_satellite/websocket_api.py b/homeassistant/components/assist_satellite/websocket_api.py
index 6cd7af2bbdb..4fc1708b866 100644
--- a/homeassistant/components/assist_satellite/websocket_api.py
+++ b/homeassistant/components/assist_satellite/websocket_api.py
@@ -19,6 +19,7 @@ from .const import (
DOMAIN,
AssistSatelliteEntityFeature,
)
+from .entity import AssistSatelliteConfiguration
CONNECTION_TEST_TIMEOUT = 30
@@ -91,7 +92,16 @@ def websocket_get_configuration(
)
return
- config_dict = asdict(satellite.async_get_configuration())
+ try:
+ config_dict = asdict(satellite.async_get_configuration())
+ except NotImplementedError:
+ # Stub configuration
+ config_dict = asdict(
+ AssistSatelliteConfiguration(
+ available_wake_words=[], active_wake_words=[], max_active_wake_words=1
+ )
+ )
+
config_dict["pipeline_entity_id"] = satellite.pipeline_entity_id
config_dict["vad_entity_id"] = satellite.vad_sensitivity_entity_id
diff --git a/homeassistant/components/asuswrt/device_tracker.py b/homeassistant/components/asuswrt/device_tracker.py
index 95d2e4c8000..ee6c3f96fc4 100644
--- a/homeassistant/components/asuswrt/device_tracker.py
+++ b/homeassistant/components/asuswrt/device_tracker.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.device_tracker import ScannerEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AsusWrtConfigEntry
from .router import AsusWrtDevInfo, AsusWrtRouter
@@ -18,7 +18,7 @@ DEFAULT_DEVICE_NAME = "Unknown device"
async def async_setup_entry(
hass: HomeAssistant,
entry: AsusWrtConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for AsusWrt component."""
router = entry.runtime_data
@@ -38,7 +38,9 @@ async def async_setup_entry(
@callback
def add_entities(
- router: AsusWrtRouter, async_add_entities: AddEntitiesCallback, tracked: set[str]
+ router: AsusWrtRouter,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+ tracked: set[str],
) -> None:
"""Add new tracker entities from the router."""
new_tracked = []
diff --git a/homeassistant/components/asuswrt/sensor.py b/homeassistant/components/asuswrt/sensor.py
index fb43e574379..c4bd5e4bded 100644
--- a/homeassistant/components/asuswrt/sensor.py
+++ b/homeassistant/components/asuswrt/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -246,7 +246,7 @@ CONNECTION_SENSORS: tuple[AsusWrtSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AsusWrtConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
router = entry.runtime_data
diff --git a/homeassistant/components/asuswrt/strings.json b/homeassistant/components/asuswrt/strings.json
index 9d50f50c7e9..cac37c0cfd0 100644
--- a/homeassistant/components/asuswrt/strings.json
+++ b/homeassistant/components/asuswrt/strings.json
@@ -66,28 +66,28 @@
"name": "Upload"
},
"load_avg_1m": {
- "name": "Average load (1m)"
+ "name": "Average load (1 min)"
},
"load_avg_5m": {
- "name": "Average load (5m)"
+ "name": "Average load (5 min)"
},
"load_avg_15m": {
- "name": "Average load (15m)"
+ "name": "Average load (15 min)"
},
"24ghz_temperature": {
- "name": "2.4GHz Temperature"
+ "name": "2.4GHz temperature"
},
"5ghz_temperature": {
- "name": "5GHz Temperature"
+ "name": "5GHz temperature"
},
"cpu_temperature": {
- "name": "CPU Temperature"
+ "name": "CPU temperature"
},
"5ghz_2_temperature": {
- "name": "5GHz Temperature (Radio 2)"
+ "name": "5GHz temperature (Radio 2)"
},
"6ghz_temperature": {
- "name": "6GHz Temperature"
+ "name": "6GHz temperature"
},
"cpu_usage": {
"name": "CPU usage"
diff --git a/homeassistant/components/atag/climate.py b/homeassistant/components/atag/climate.py
index a362b71fbc8..8f1ded150f1 100644
--- a/homeassistant/components/atag/climate.py
+++ b/homeassistant/components/atag/climate.py
@@ -14,7 +14,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.enum import try_parse_enum
from .coordinator import AtagConfigEntry, AtagDataUpdateCoordinator
@@ -32,7 +32,9 @@ HVAC_MODES = [HVACMode.AUTO, HVACMode.HEAT]
async def async_setup_entry(
- hass: HomeAssistant, entry: AtagConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: AtagConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load a config entry."""
async_add_entities([AtagThermostat(entry.runtime_data, "climate")])
diff --git a/homeassistant/components/atag/coordinator.py b/homeassistant/components/atag/coordinator.py
index 6d542471384..f590bc1dc6a 100644
--- a/homeassistant/components/atag/coordinator.py
+++ b/homeassistant/components/atag/coordinator.py
@@ -19,17 +19,22 @@ type AtagConfigEntry = ConfigEntry[AtagDataUpdateCoordinator]
class AtagDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Atag data update coordinator."""
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
+ config_entry: AtagConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: AtagConfigEntry) -> None:
"""Initialize Atag coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Atag",
update_interval=timedelta(seconds=60),
)
self.atag = AtagOne(
- session=async_get_clientsession(hass), **entry.data, device=entry.unique_id
+ session=async_get_clientsession(hass),
+ **config_entry.data,
+ device=config_entry.unique_id,
)
async def _async_update_data(self) -> None:
diff --git a/homeassistant/components/atag/sensor.py b/homeassistant/components/atag/sensor.py
index bd39f0b3458..ca5bbd5e614 100644
--- a/homeassistant/components/atag/sensor.py
+++ b/homeassistant/components/atag/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AtagConfigEntry, AtagDataUpdateCoordinator
from .entity import AtagEntity
@@ -28,7 +28,7 @@ SENSORS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AtagConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize sensor platform from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/atag/water_heater.py b/homeassistant/components/atag/water_heater.py
index 6b013b36885..00761f47324 100644
--- a/homeassistant/components/atag/water_heater.py
+++ b/homeassistant/components/atag/water_heater.py
@@ -9,7 +9,7 @@ from homeassistant.components.water_heater import (
)
from homeassistant.const import ATTR_TEMPERATURE, STATE_OFF, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AtagConfigEntry
from .entity import AtagEntity
@@ -20,7 +20,7 @@ OPERATION_LIST = [STATE_OFF, STATE_ECO, STATE_PERFORMANCE]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AtagConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize DHW device from config entry."""
async_add_entities(
diff --git a/homeassistant/components/august/binary_sensor.py b/homeassistant/components/august/binary_sensor.py
index fb877252010..b4c440599c4 100644
--- a/homeassistant/components/august/binary_sensor.py
+++ b/homeassistant/components/august/binary_sensor.py
@@ -21,7 +21,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from . import AugustConfigEntry, AugustData
@@ -92,7 +92,7 @@ SENSOR_TYPES_DOORBELL: tuple[AugustDoorbellBinarySensorEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AugustConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the August binary sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/august/button.py b/homeassistant/components/august/button.py
index 79f2b67888a..4971d0cccf5 100644
--- a/homeassistant/components/august/button.py
+++ b/homeassistant/components/august/button.py
@@ -2,7 +2,7 @@
from homeassistant.components.button import ButtonEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AugustConfigEntry
from .entity import AugustEntity
@@ -11,7 +11,7 @@ from .entity import AugustEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AugustConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up August lock wake buttons."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/august/camera.py b/homeassistant/components/august/camera.py
index f4398455256..7b013022299 100644
--- a/homeassistant/components/august/camera.py
+++ b/homeassistant/components/august/camera.py
@@ -12,7 +12,7 @@ from yalexs.util import update_doorbell_image_from_activity
from homeassistant.components.camera import Camera
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AugustConfigEntry, AugustData
from .const import DEFAULT_NAME, DEFAULT_TIMEOUT
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AugustConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up August cameras."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/august/event.py b/homeassistant/components/august/event.py
index 49b14630337..0abc840bc69 100644
--- a/homeassistant/components/august/event.py
+++ b/homeassistant/components/august/event.py
@@ -16,7 +16,7 @@ from homeassistant.components.event import (
EventEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AugustConfigEntry, AugustData
from .entity import AugustDescriptionEntity
@@ -59,7 +59,7 @@ TYPES_DOORBELL: tuple[AugustEventEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AugustConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the august event platform."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/august/lock.py b/homeassistant/components/august/lock.py
index c681cc98808..4a37149772a 100644
--- a/homeassistant/components/august/lock.py
+++ b/homeassistant/components/august/lock.py
@@ -14,7 +14,7 @@ from yalexs.util import get_latest_activity, update_lock_detail_from_activity
from homeassistant.components.lock import ATTR_CHANGED_BY, LockEntity, LockEntityFeature
from homeassistant.const import ATTR_BATTERY_LEVEL
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.util import dt as dt_util
@@ -29,7 +29,7 @@ LOCK_JAMMED_ERR = 531
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AugustConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up August locks."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/august/sensor.py b/homeassistant/components/august/sensor.py
index b7c0d618492..94a5461149f 100644
--- a/homeassistant/components/august/sensor.py
+++ b/homeassistant/components/august/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
EntityCategory,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AugustConfigEntry
from .const import (
@@ -82,7 +82,7 @@ SENSOR_TYPE_KEYPAD_BATTERY = AugustSensorEntityDescription[KeypadDetail](
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AugustConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the August sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/aurora/__init__.py b/homeassistant/components/aurora/__init__.py
index b6c47cf36b2..a48d704141f 100644
--- a/homeassistant/components/aurora/__init__.py
+++ b/homeassistant/components/aurora/__init__.py
@@ -1,20 +1,17 @@
"""The aurora component."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from .const import CONF_THRESHOLD, DEFAULT_THRESHOLD
-from .coordinator import AuroraDataUpdateCoordinator
+from .coordinator import AuroraConfigEntry, AuroraDataUpdateCoordinator
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR]
-type AuroraConfigEntry = ConfigEntry[AuroraDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: AuroraConfigEntry) -> bool:
"""Set up Aurora from a config entry."""
- coordinator = AuroraDataUpdateCoordinator(hass=hass)
+ coordinator = AuroraDataUpdateCoordinator(hass=hass, config_entry=entry)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/aurora/binary_sensor.py b/homeassistant/components/aurora/binary_sensor.py
index b8fb5002ff5..73e732dc44a 100644
--- a/homeassistant/components/aurora/binary_sensor.py
+++ b/homeassistant/components/aurora/binary_sensor.py
@@ -4,16 +4,16 @@ from __future__ import annotations
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AuroraConfigEntry
+from .coordinator import AuroraConfigEntry
from .entity import AuroraEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: AuroraConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary_sensor platform."""
async_add_entities(
diff --git a/homeassistant/components/aurora/coordinator.py b/homeassistant/components/aurora/coordinator.py
index 9771cc53652..a7b87baec22 100644
--- a/homeassistant/components/aurora/coordinator.py
+++ b/homeassistant/components/aurora/coordinator.py
@@ -4,11 +4,11 @@ from __future__ import annotations
from datetime import timedelta
import logging
-from typing import TYPE_CHECKING
from aiohttp import ClientError
from auroranoaa import AuroraForecast
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -16,23 +16,23 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import CONF_THRESHOLD, DEFAULT_THRESHOLD
-if TYPE_CHECKING:
- from . import AuroraConfigEntry
-
_LOGGER = logging.getLogger(__name__)
+type AuroraConfigEntry = ConfigEntry[AuroraDataUpdateCoordinator]
+
class AuroraDataUpdateCoordinator(DataUpdateCoordinator[int]):
"""Class to manage fetching data from the NOAA Aurora API."""
config_entry: AuroraConfigEntry
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: AuroraConfigEntry) -> None:
"""Initialize the data updater."""
super().__init__(
hass=hass,
logger=_LOGGER,
+ config_entry=config_entry,
name="Aurora",
update_interval=timedelta(minutes=5),
)
diff --git a/homeassistant/components/aurora/sensor.py b/homeassistant/components/aurora/sensor.py
index 35d39289598..d424b7e98ab 100644
--- a/homeassistant/components/aurora/sensor.py
+++ b/homeassistant/components/aurora/sensor.py
@@ -5,16 +5,16 @@ from __future__ import annotations
from homeassistant.components.sensor import SensorEntity, SensorStateClass
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import AuroraConfigEntry
+from .coordinator import AuroraConfigEntry
from .entity import AuroraEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: AuroraConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
diff --git a/homeassistant/components/aurora_abb_powerone/__init__.py b/homeassistant/components/aurora_abb_powerone/__init__.py
index 749d40aeb5c..91166d5c8f5 100644
--- a/homeassistant/components/aurora_abb_powerone/__init__.py
+++ b/homeassistant/components/aurora_abb_powerone/__init__.py
@@ -23,7 +23,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: AuroraAbbConfigEntry) ->
comport = entry.data[CONF_PORT]
address = entry.data[CONF_ADDRESS]
- coordinator = AuroraAbbDataUpdateCoordinator(hass, comport, address)
+ coordinator = AuroraAbbDataUpdateCoordinator(hass, entry, comport, address)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/aurora_abb_powerone/coordinator.py b/homeassistant/components/aurora_abb_powerone/coordinator.py
index c3d05da95f3..d38f0716b44 100644
--- a/homeassistant/components/aurora_abb_powerone/coordinator.py
+++ b/homeassistant/components/aurora_abb_powerone/coordinator.py
@@ -21,12 +21,26 @@ type AuroraAbbConfigEntry = ConfigEntry[AuroraAbbDataUpdateCoordinator]
class AuroraAbbDataUpdateCoordinator(DataUpdateCoordinator[dict[str, float]]):
"""Class to manage fetching AuroraAbbPowerone data."""
- def __init__(self, hass: HomeAssistant, comport: str, address: int) -> None:
+ config_entry: AuroraAbbConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: AuroraAbbConfigEntry,
+ comport: str,
+ address: int,
+ ) -> None:
"""Initialize the data update coordinator."""
self.available_prev = False
self.available = False
self.client = AuroraSerialClient(address, comport, parity="N", timeout=1)
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
+ )
def _update_data(self) -> dict[str, float]:
"""Fetch new state data for the sensors.
diff --git a/homeassistant/components/aurora_abb_powerone/sensor.py b/homeassistant/components/aurora_abb_powerone/sensor.py
index 29d5cab2667..d35d8a2d8cb 100644
--- a/homeassistant/components/aurora_abb_powerone/sensor.py
+++ b/homeassistant/components/aurora_abb_powerone/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -130,7 +130,7 @@ SENSOR_TYPES = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AuroraAbbConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up aurora_abb_powerone sensor based on a config entry."""
diff --git a/homeassistant/components/aurora_abb_powerone/strings.json b/homeassistant/components/aurora_abb_powerone/strings.json
index 319bcb0adc4..6b28d9d8c1c 100644
--- a/homeassistant/components/aurora_abb_powerone/strings.json
+++ b/homeassistant/components/aurora_abb_powerone/strings.json
@@ -11,7 +11,7 @@
},
"error": {
"cannot_connect": "Unable to connect, please check serial port, address, electrical connection and that inverter is on (in daylight)",
- "invalid_serial_port": "Serial port is not a valid device or could not be openned",
+ "invalid_serial_port": "Serial port is not a valid device or could not be opened",
"cannot_open_serial_port": "Cannot open serial port, please check and try again"
},
"abort": {
diff --git a/homeassistant/components/aussie_broadband/sensor.py b/homeassistant/components/aussie_broadband/sensor.py
index 49da78da8de..41a2f164095 100644
--- a/homeassistant/components/aussie_broadband/sensor.py
+++ b/homeassistant/components/aussie_broadband/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfInformation, UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -123,7 +123,7 @@ SENSOR_DESCRIPTIONS: tuple[SensorValueEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AussieBroadbandConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Aussie Broadband sensor platform from a config entry."""
diff --git a/homeassistant/components/autarco/sensor.py b/homeassistant/components/autarco/sensor.py
index b7c4312815b..1635adefdb8 100644
--- a/homeassistant/components/autarco/sensor.py
+++ b/homeassistant/components/autarco/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import PERCENTAGE, UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -172,7 +172,7 @@ SENSORS_INVERTER: tuple[AutarcoInverterSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AutarcoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Autarco sensors based on a config entry."""
entities: list[SensorEntity] = []
diff --git a/homeassistant/components/awair/coordinator.py b/homeassistant/components/awair/coordinator.py
index 78f0d9d65f2..62725693522 100644
--- a/homeassistant/components/awair/coordinator.py
+++ b/homeassistant/components/awair/coordinator.py
@@ -40,17 +40,24 @@ class AwairResult:
class AwairDataUpdateCoordinator(DataUpdateCoordinator[dict[str, AwairResult]]):
"""Define a wrapper class to update Awair data."""
+ config_entry: AwairConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- config_entry: ConfigEntry,
+ config_entry: AwairConfigEntry,
update_interval: timedelta | None,
) -> None:
"""Set up the AwairDataUpdateCoordinator class."""
- self._config_entry = config_entry
self.title = config_entry.title
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=update_interval)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=update_interval,
+ )
async def _fetch_air_data(self, device: AwairBaseDevice) -> AwairResult:
"""Fetch latest air quality data."""
@@ -64,7 +71,10 @@ class AwairCloudDataUpdateCoordinator(AwairDataUpdateCoordinator):
"""Define a wrapper class to update Awair data from Cloud API."""
def __init__(
- self, hass: HomeAssistant, config_entry: ConfigEntry, session: ClientSession
+ self,
+ hass: HomeAssistant,
+ config_entry: AwairConfigEntry,
+ session: ClientSession,
) -> None:
"""Set up the AwairCloudDataUpdateCoordinator class."""
access_token = config_entry.data[CONF_ACCESS_TOKEN]
@@ -95,7 +105,10 @@ class AwairLocalDataUpdateCoordinator(AwairDataUpdateCoordinator):
_device: AwairLocalDevice | None = None
def __init__(
- self, hass: HomeAssistant, config_entry: ConfigEntry, session: ClientSession
+ self,
+ hass: HomeAssistant,
+ config_entry: AwairConfigEntry,
+ session: ClientSession,
) -> None:
"""Set up the AwairLocalDataUpdateCoordinator class."""
self._awair = AwairLocal(
diff --git a/homeassistant/components/awair/sensor.py b/homeassistant/components/awair/sensor.py
index c92009d9b1b..a0c4b5ba8fe 100644
--- a/homeassistant/components/awair/sensor.py
+++ b/homeassistant/components/awair/sensor.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -133,7 +133,7 @@ SENSOR_TYPES_DUST: tuple[AwairSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AwairConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Awair sensor entity based on a config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/axis/binary_sensor.py b/homeassistant/components/axis/binary_sensor.py
index d6f132874b6..6933380c094 100644
--- a/homeassistant/components/axis/binary_sensor.py
+++ b/homeassistant/components/axis/binary_sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from . import AxisConfigEntry
@@ -178,7 +178,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AxisConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Axis binary sensor."""
config_entry.runtime_data.entity_loader.register_platform(
diff --git a/homeassistant/components/axis/camera.py b/homeassistant/components/axis/camera.py
index a5a00bcd1ab..089a018ee5b 100644
--- a/homeassistant/components/axis/camera.py
+++ b/homeassistant/components/axis/camera.py
@@ -7,7 +7,7 @@ from homeassistant.components.mjpeg import MjpegCamera, filter_urllib3_logging
from homeassistant.const import HTTP_DIGEST_AUTHENTICATION
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AxisConfigEntry
from .const import DEFAULT_STREAM_PROFILE, DEFAULT_VIDEO_SOURCE
@@ -18,7 +18,7 @@ from .hub import AxisHub
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AxisConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Axis camera video stream."""
filter_urllib3_logging()
diff --git a/homeassistant/components/axis/light.py b/homeassistant/components/axis/light.py
index d0d144a28fa..0c6015efced 100644
--- a/homeassistant/components/axis/light.py
+++ b/homeassistant/components/axis/light.py
@@ -12,7 +12,7 @@ from homeassistant.components.light import (
LightEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AxisConfigEntry
from .entity import TOPIC_TO_EVENT_TYPE, AxisEventDescription, AxisEventEntity
@@ -46,7 +46,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AxisConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Axis light platform."""
config_entry.runtime_data.entity_loader.register_platform(
diff --git a/homeassistant/components/axis/switch.py b/homeassistant/components/axis/switch.py
index 17824302871..55250b5f489 100644
--- a/homeassistant/components/axis/switch.py
+++ b/homeassistant/components/axis/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AxisConfigEntry
from .entity import AxisEventDescription, AxisEventEntity
@@ -39,7 +39,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: AxisConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Axis switch platform."""
config_entry.runtime_data.entity_loader.register_platform(
diff --git a/homeassistant/components/azure_devops/sensor.py b/homeassistant/components/azure_devops/sensor.py
index 1d590032a85..55c821a119e 100644
--- a/homeassistant/components/azure_devops/sensor.py
+++ b/homeassistant/components/azure_devops/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -145,7 +145,7 @@ def parse_datetime(value: str | None) -> datetime | None:
async def async_setup_entry(
hass: HomeAssistant,
entry: AzureDevOpsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Azure DevOps sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/azure_devops/strings.json b/homeassistant/components/azure_devops/strings.json
index f5fe5cd06a7..611a8b9a758 100644
--- a/homeassistant/components/azure_devops/strings.json
+++ b/homeassistant/components/azure_devops/strings.json
@@ -14,7 +14,7 @@
"personal_access_token": "Personal Access Token (PAT)"
},
"description": "Set up an Azure DevOps instance to access your project. A Personal Access Token is only required for a private project.",
- "title": "Add Azure DevOps Project"
+ "title": "Add Azure DevOps project"
},
"reauth_confirm": {
"data": {
@@ -32,7 +32,7 @@
"entity": {
"sensor": {
"build_id": {
- "name": "{definition_name} latest build id"
+ "name": "{definition_name} latest build ID"
},
"finish_time": {
"name": "{definition_name} latest build finish time"
@@ -59,7 +59,7 @@
"name": "{definition_name} latest build start time"
},
"url": {
- "name": "{definition_name} latest build url"
+ "name": "{definition_name} latest build URL"
},
"work_item_count": {
"name": "{item_type} {item_state} work items"
@@ -68,7 +68,7 @@
},
"exceptions": {
"authentication_failed": {
- "message": "Could not authorize with Azure DevOps for {title}. You will need to update your personal access token."
+ "message": "Could not authorize with Azure DevOps for {title}. You will need to update your Personal Access Token."
}
}
}
diff --git a/homeassistant/components/azure_storage/__init__.py b/homeassistant/components/azure_storage/__init__.py
new file mode 100644
index 00000000000..f22e7b70c12
--- /dev/null
+++ b/homeassistant/components/azure_storage/__init__.py
@@ -0,0 +1,86 @@
+"""The Azure Storage integration."""
+
+from aiohttp import ClientTimeout
+from azure.core.exceptions import (
+ ClientAuthenticationError,
+ HttpResponseError,
+ ResourceNotFoundError,
+)
+from azure.core.pipeline.transport._aiohttp import (
+ AioHttpTransport,
+) # need to import from private file, as it is not properly imported in the init
+from azure.storage.blob.aio import ContainerClient
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import (
+ ConfigEntryAuthFailed,
+ ConfigEntryError,
+ ConfigEntryNotReady,
+)
+from homeassistant.helpers.aiohttp_client import async_create_clientsession
+
+from .const import (
+ CONF_ACCOUNT_NAME,
+ CONF_CONTAINER_NAME,
+ CONF_STORAGE_ACCOUNT_KEY,
+ DATA_BACKUP_AGENT_LISTENERS,
+ DOMAIN,
+)
+
+type AzureStorageConfigEntry = ConfigEntry[ContainerClient]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant, entry: AzureStorageConfigEntry
+) -> bool:
+ """Set up Azure Storage integration."""
+ # set increase aiohttp timeout for long running operations (up/download)
+ session = async_create_clientsession(
+ hass, timeout=ClientTimeout(connect=10, total=12 * 60 * 60)
+ )
+ container_client = ContainerClient(
+ account_url=f"https://{entry.data[CONF_ACCOUNT_NAME]}.blob.core.windows.net/",
+ container_name=entry.data[CONF_CONTAINER_NAME],
+ credential=entry.data[CONF_STORAGE_ACCOUNT_KEY],
+ transport=AioHttpTransport(session=session),
+ )
+
+ try:
+ if not await container_client.exists():
+ await container_client.create_container()
+ except ResourceNotFoundError as err:
+ raise ConfigEntryError(
+ translation_domain=DOMAIN,
+ translation_key="account_not_found",
+ translation_placeholders={CONF_ACCOUNT_NAME: entry.data[CONF_ACCOUNT_NAME]},
+ ) from err
+ except ClientAuthenticationError as err:
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN,
+ translation_key="invalid_auth",
+ translation_placeholders={CONF_ACCOUNT_NAME: entry.data[CONF_ACCOUNT_NAME]},
+ ) from err
+ except HttpResponseError as err:
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="cannot_connect",
+ translation_placeholders={CONF_ACCOUNT_NAME: entry.data[CONF_ACCOUNT_NAME]},
+ ) from err
+
+ entry.runtime_data = container_client
+
+ def _async_notify_backup_listeners() -> None:
+ for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
+ listener()
+
+ entry.async_on_unload(entry.async_on_state_change(_async_notify_backup_listeners))
+
+ return True
+
+
+async def async_unload_entry(
+ hass: HomeAssistant, entry: AzureStorageConfigEntry
+) -> bool:
+ """Unload an Azure Storage config entry."""
+ return True
diff --git a/homeassistant/components/azure_storage/backup.py b/homeassistant/components/azure_storage/backup.py
new file mode 100644
index 00000000000..4d897126d3d
--- /dev/null
+++ b/homeassistant/components/azure_storage/backup.py
@@ -0,0 +1,182 @@
+"""Support for Azure Storage backup."""
+
+from __future__ import annotations
+
+from collections.abc import AsyncIterator, Callable, Coroutine
+from functools import wraps
+import json
+import logging
+from typing import Any, Concatenate
+
+from azure.core.exceptions import HttpResponseError
+from azure.storage.blob import BlobProperties
+
+from homeassistant.components.backup import (
+ AgentBackup,
+ BackupAgent,
+ BackupAgentError,
+ BackupNotFound,
+ suggested_filename,
+)
+from homeassistant.core import HomeAssistant, callback
+
+from . import AzureStorageConfigEntry
+from .const import DATA_BACKUP_AGENT_LISTENERS, DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+METADATA_VERSION = "1"
+
+
+async def async_get_backup_agents(
+ hass: HomeAssistant,
+) -> list[BackupAgent]:
+ """Return a list of backup agents."""
+ entries: list[AzureStorageConfigEntry] = hass.config_entries.async_loaded_entries(
+ DOMAIN
+ )
+ return [AzureStorageBackupAgent(hass, entry) for entry in entries]
+
+
+@callback
+def async_register_backup_agents_listener(
+ hass: HomeAssistant,
+ *,
+ listener: Callable[[], None],
+ **kwargs: Any,
+) -> Callable[[], None]:
+ """Register a listener to be called when agents are added or removed."""
+ hass.data.setdefault(DATA_BACKUP_AGENT_LISTENERS, []).append(listener)
+
+ @callback
+ def remove_listener() -> None:
+ """Remove the listener."""
+ hass.data[DATA_BACKUP_AGENT_LISTENERS].remove(listener)
+ if not hass.data[DATA_BACKUP_AGENT_LISTENERS]:
+ hass.data.pop(DATA_BACKUP_AGENT_LISTENERS)
+
+ return remove_listener
+
+
+def handle_backup_errors[_R, **P](
+ func: Callable[Concatenate[AzureStorageBackupAgent, P], Coroutine[Any, Any, _R]],
+) -> Callable[Concatenate[AzureStorageBackupAgent, P], Coroutine[Any, Any, _R]]:
+ """Handle backup errors."""
+
+ @wraps(func)
+ async def wrapper(
+ self: AzureStorageBackupAgent, *args: P.args, **kwargs: P.kwargs
+ ) -> _R:
+ try:
+ return await func(self, *args, **kwargs)
+ except HttpResponseError as err:
+ _LOGGER.debug(
+ "Error during backup in %s: Status %s, message %s",
+ func.__name__,
+ err.status_code,
+ err.message,
+ exc_info=True,
+ )
+ raise BackupAgentError(
+ f"Error during backup operation in {func.__name__}:"
+ f" Status {err.status_code}, message: {err.message}"
+ ) from err
+
+ return wrapper
+
+
+class AzureStorageBackupAgent(BackupAgent):
+ """Azure storage backup agent."""
+
+ domain = DOMAIN
+
+ def __init__(self, hass: HomeAssistant, entry: AzureStorageConfigEntry) -> None:
+ """Initialize the Azure storage backup agent."""
+ super().__init__()
+ self._client = entry.runtime_data
+ self.name = entry.title
+ self.unique_id = entry.entry_id
+
+ @handle_backup_errors
+ async def async_download_backup(
+ self,
+ backup_id: str,
+ **kwargs: Any,
+ ) -> AsyncIterator[bytes]:
+ """Download a backup file."""
+ blob = await self._find_blob_by_backup_id(backup_id)
+ if blob is None:
+ raise BackupNotFound(f"Backup {backup_id} not found")
+ download_stream = await self._client.download_blob(blob.name)
+ return download_stream.chunks()
+
+ @handle_backup_errors
+ async def async_upload_backup(
+ self,
+ *,
+ open_stream: Callable[[], Coroutine[Any, Any, AsyncIterator[bytes]]],
+ backup: AgentBackup,
+ **kwargs: Any,
+ ) -> None:
+ """Upload a backup."""
+
+ metadata = {
+ "metadata_version": METADATA_VERSION,
+ "backup_id": backup.backup_id,
+ "backup_metadata": json.dumps(backup.as_dict()),
+ }
+
+ await self._client.upload_blob(
+ name=suggested_filename(backup),
+ metadata=metadata,
+ data=await open_stream(),
+ length=backup.size,
+ )
+
+ @handle_backup_errors
+ async def async_delete_backup(
+ self,
+ backup_id: str,
+ **kwargs: Any,
+ ) -> None:
+ """Delete a backup file."""
+ blob = await self._find_blob_by_backup_id(backup_id)
+ if blob is None:
+ raise BackupNotFound(f"Backup {backup_id} not found")
+ await self._client.delete_blob(blob.name)
+
+ @handle_backup_errors
+ async def async_list_backups(self, **kwargs: Any) -> list[AgentBackup]:
+ """List backups."""
+ backups: list[AgentBackup] = []
+ async for blob in self._client.list_blobs(include="metadata"):
+ metadata = blob.metadata
+
+ if metadata.get("metadata_version") == METADATA_VERSION:
+ backups.append(
+ AgentBackup.from_dict(json.loads(metadata["backup_metadata"]))
+ )
+
+ return backups
+
+ @handle_backup_errors
+ async def async_get_backup(
+ self,
+ backup_id: str,
+ **kwargs: Any,
+ ) -> AgentBackup:
+ """Return a backup."""
+ blob = await self._find_blob_by_backup_id(backup_id)
+ if blob is None:
+ raise BackupNotFound(f"Backup {backup_id} not found")
+
+ return AgentBackup.from_dict(json.loads(blob.metadata["backup_metadata"]))
+
+ async def _find_blob_by_backup_id(self, backup_id: str) -> BlobProperties | None:
+ """Find a blob by backup id."""
+ async for blob in self._client.list_blobs(include="metadata"):
+ if (
+ backup_id == blob.metadata.get("backup_id", "")
+ and blob.metadata.get("metadata_version") == METADATA_VERSION
+ ):
+ return blob
+ return None
diff --git a/homeassistant/components/azure_storage/config_flow.py b/homeassistant/components/azure_storage/config_flow.py
new file mode 100644
index 00000000000..2862d290f95
--- /dev/null
+++ b/homeassistant/components/azure_storage/config_flow.py
@@ -0,0 +1,160 @@
+"""Config flow for Azure Storage integration."""
+
+from collections.abc import Mapping
+import logging
+from typing import Any
+
+from azure.core.exceptions import ClientAuthenticationError, ResourceNotFoundError
+from azure.core.pipeline.transport._aiohttp import (
+ AioHttpTransport,
+) # need to import from private file, as it is not properly imported in the init
+from azure.storage.blob.aio import ContainerClient
+import voluptuous as vol
+
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+from .const import (
+ CONF_ACCOUNT_NAME,
+ CONF_CONTAINER_NAME,
+ CONF_STORAGE_ACCOUNT_KEY,
+ DOMAIN,
+)
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class AzureStorageConfigFlow(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow for azure storage."""
+
+ def get_account_url(self, account_name: str) -> str:
+ """Get the account URL."""
+ return f"https://{account_name}.blob.core.windows.net/"
+
+ async def validate_config(
+ self, container_client: ContainerClient
+ ) -> dict[str, str]:
+ """Validate the configuration."""
+ errors: dict[str, str] = {}
+ try:
+ await container_client.exists()
+ except ResourceNotFoundError:
+ errors["base"] = "cannot_connect"
+ except ClientAuthenticationError:
+ errors[CONF_STORAGE_ACCOUNT_KEY] = "invalid_auth"
+ except Exception:
+ _LOGGER.exception("Unknown exception occurred")
+ errors["base"] = "unknown"
+ return errors
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """User step for Azure Storage."""
+
+ errors: dict[str, str] = {}
+
+ if user_input is not None:
+ self._async_abort_entries_match(
+ {CONF_ACCOUNT_NAME: user_input[CONF_ACCOUNT_NAME]}
+ )
+ container_client = ContainerClient(
+ account_url=self.get_account_url(user_input[CONF_ACCOUNT_NAME]),
+ container_name=user_input[CONF_CONTAINER_NAME],
+ credential=user_input[CONF_STORAGE_ACCOUNT_KEY],
+ transport=AioHttpTransport(session=async_get_clientsession(self.hass)),
+ )
+ errors = await self.validate_config(container_client)
+
+ if not errors:
+ return self.async_create_entry(
+ title=f"{user_input[CONF_ACCOUNT_NAME]}/{user_input[CONF_CONTAINER_NAME]}",
+ data=user_input,
+ )
+
+ return self.async_show_form(
+ data_schema=vol.Schema(
+ {
+ vol.Required(CONF_ACCOUNT_NAME): str,
+ vol.Required(
+ CONF_CONTAINER_NAME, default="home-assistant-backups"
+ ): str,
+ vol.Required(CONF_STORAGE_ACCOUNT_KEY): str,
+ }
+ ),
+ errors=errors,
+ )
+
+ async def async_step_reauth(
+ self, entry_data: Mapping[str, Any]
+ ) -> ConfigFlowResult:
+ """Perform reauth upon an API authentication error."""
+ return await self.async_step_reauth_confirm()
+
+ async def async_step_reauth_confirm(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Confirm reauth dialog."""
+ errors: dict[str, str] = {}
+ reauth_entry = self._get_reauth_entry()
+
+ if user_input is not None:
+ container_client = ContainerClient(
+ account_url=self.get_account_url(reauth_entry.data[CONF_ACCOUNT_NAME]),
+ container_name=reauth_entry.data[CONF_CONTAINER_NAME],
+ credential=user_input[CONF_STORAGE_ACCOUNT_KEY],
+ transport=AioHttpTransport(session=async_get_clientsession(self.hass)),
+ )
+ errors = await self.validate_config(container_client)
+ if not errors:
+ return self.async_update_reload_and_abort(
+ reauth_entry,
+ data={**reauth_entry.data, **user_input},
+ )
+ return self.async_show_form(
+ step_id="reauth_confirm",
+ data_schema=vol.Schema(
+ {
+ vol.Required(CONF_STORAGE_ACCOUNT_KEY): str,
+ }
+ ),
+ errors=errors,
+ )
+
+ async def async_step_reconfigure(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Reconfigure the entry."""
+ errors: dict[str, str] = {}
+ reconfigure_entry = self._get_reconfigure_entry()
+
+ if user_input is not None:
+ container_client = ContainerClient(
+ account_url=self.get_account_url(
+ reconfigure_entry.data[CONF_ACCOUNT_NAME]
+ ),
+ container_name=user_input[CONF_CONTAINER_NAME],
+ credential=user_input[CONF_STORAGE_ACCOUNT_KEY],
+ transport=AioHttpTransport(session=async_get_clientsession(self.hass)),
+ )
+ errors = await self.validate_config(container_client)
+ if not errors:
+ return self.async_update_reload_and_abort(
+ reconfigure_entry,
+ data={**reconfigure_entry.data, **user_input},
+ )
+ return self.async_show_form(
+ data_schema=vol.Schema(
+ {
+ vol.Required(
+ CONF_CONTAINER_NAME,
+ default=reconfigure_entry.data[CONF_CONTAINER_NAME],
+ ): str,
+ vol.Required(
+ CONF_STORAGE_ACCOUNT_KEY,
+ default=reconfigure_entry.data[CONF_STORAGE_ACCOUNT_KEY],
+ ): str,
+ }
+ ),
+ errors=errors,
+ )
diff --git a/homeassistant/components/azure_storage/const.py b/homeassistant/components/azure_storage/const.py
new file mode 100644
index 00000000000..efcb338a096
--- /dev/null
+++ b/homeassistant/components/azure_storage/const.py
@@ -0,0 +1,16 @@
+"""Constants for the Azure Storage integration."""
+
+from collections.abc import Callable
+from typing import Final
+
+from homeassistant.util.hass_dict import HassKey
+
+DOMAIN: Final = "azure_storage"
+
+CONF_STORAGE_ACCOUNT_KEY: Final = "storage_account_key"
+CONF_ACCOUNT_NAME: Final = "account_name"
+CONF_CONTAINER_NAME: Final = "container_name"
+
+DATA_BACKUP_AGENT_LISTENERS: HassKey[list[Callable[[], None]]] = HassKey(
+ f"{DOMAIN}.backup_agent_listeners"
+)
diff --git a/homeassistant/components/azure_storage/manifest.json b/homeassistant/components/azure_storage/manifest.json
new file mode 100644
index 00000000000..729334f851d
--- /dev/null
+++ b/homeassistant/components/azure_storage/manifest.json
@@ -0,0 +1,12 @@
+{
+ "domain": "azure_storage",
+ "name": "Azure Storage",
+ "codeowners": ["@zweckj"],
+ "config_flow": true,
+ "documentation": "https://www.home-assistant.io/integrations/azure_storage",
+ "integration_type": "service",
+ "iot_class": "cloud_polling",
+ "loggers": ["azure-storage-blob"],
+ "quality_scale": "platinum",
+ "requirements": ["azure-storage-blob==12.24.0"]
+}
diff --git a/homeassistant/components/azure_storage/quality_scale.yaml b/homeassistant/components/azure_storage/quality_scale.yaml
new file mode 100644
index 00000000000..6199ba514a3
--- /dev/null
+++ b/homeassistant/components/azure_storage/quality_scale.yaml
@@ -0,0 +1,133 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: Integration does not register custom actions.
+ appropriate-polling:
+ status: exempt
+ comment: |
+ This integration does not poll.
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: |
+ This integration does not have any custom actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: exempt
+ comment: |
+ Entities of this integration does not explicitly subscribe to events.
+ entity-unique-id:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ has-entity-name:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions: done
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: |
+ This integration does not have any configuration parameters.
+ docs-installation-parameters: done
+ entity-unavailable:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ integration-owner: done
+ log-when-unavailable:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ parallel-updates:
+ status: exempt
+ comment: |
+ This integration does not have platforms.
+ reauthentication-flow: done
+ test-coverage: done
+
+ # Gold
+ devices:
+ status: exempt
+ comment: |
+ This integration connects to a single service.
+ diagnostics:
+ status: exempt
+ comment: |
+ There is no data to diagnose.
+ discovery-update-info:
+ status: exempt
+ comment: |
+ This integration is a cloud service and does not support discovery.
+ discovery:
+ status: exempt
+ comment: |
+ This integration is a cloud service and does not support discovery.
+ docs-data-update:
+ status: exempt
+ comment: |
+ This integration does not poll or push.
+ docs-examples:
+ status: exempt
+ comment: |
+ This integration only serves backup.
+ docs-known-limitations: done
+ docs-supported-devices:
+ status: exempt
+ comment: |
+ This integration is a cloud service.
+ docs-supported-functions:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ docs-troubleshooting: done
+ docs-use-cases: done
+ dynamic-devices:
+ status: exempt
+ comment: |
+ This integration connects to a single service.
+ entity-category:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ entity-device-class:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ entity-disabled-by-default:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ entity-translations:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ exception-translations: done
+ icon-translations:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ reconfiguration-flow: done
+ repair-issues: done
+ stale-devices:
+ status: exempt
+ comment: |
+ This integration connects to a single service.
+ # Platinum
+ async-dependency: done
+ inject-websession: done
+ strict-typing: done
diff --git a/homeassistant/components/azure_storage/strings.json b/homeassistant/components/azure_storage/strings.json
new file mode 100644
index 00000000000..e9053f113cc
--- /dev/null
+++ b/homeassistant/components/azure_storage/strings.json
@@ -0,0 +1,72 @@
+{
+ "config": {
+ "error": {
+ "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
+ "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
+ "unknown": "[%key:common::config_flow::error::unknown%]"
+ },
+ "step": {
+ "user": {
+ "data": {
+ "storage_account_key": "Storage account key",
+ "account_name": "Account name",
+ "container_name": "Container name"
+ },
+ "data_description": {
+ "storage_account_key": "Storage account access key used for authorization",
+ "account_name": "Name of the storage account",
+ "container_name": "Name of the storage container to be used (will be created if it does not exist)"
+ },
+ "description": "Set up an Azure (Blob) storage account to be used for backups.",
+ "title": "Add Azure storage account"
+ },
+ "reauth_confirm": {
+ "data": {
+ "storage_account_key": "[%key:component::azure_storage::config::step::user::data::storage_account_key%]"
+ },
+ "data_description": {
+ "storage_account_key": "[%key:component::azure_storage::config::step::user::data_description::storage_account_key%]"
+ },
+ "description": "Provide a new storage account key.",
+ "title": "Reauthenticate Azure storage account"
+ },
+ "reconfigure": {
+ "data": {
+ "container_name": "[%key:component::azure_storage::config::step::user::data::container_name%]",
+ "storage_account_key": "[%key:component::azure_storage::config::step::user::data::storage_account_key%]"
+ },
+ "data_description": {
+ "container_name": "[%key:component::azure_storage::config::step::user::data_description::container_name%]",
+ "storage_account_key": "[%key:component::azure_storage::config::step::user::data_description::storage_account_key%]"
+ },
+ "description": "Change the settings of the Azure storage integration.",
+ "title": "Reconfigure Azure storage account"
+ }
+ },
+ "abort": {
+ "already_configured": "[%key:common::config_flow::abort::already_configured_account%]",
+ "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",
+ "reconfigure_successful": "[%key:common::config_flow::abort::reconfigure_successful%]"
+ }
+ },
+ "issues": {
+ "container_not_found": {
+ "title": "Storage container not found",
+ "description": "The storage container {container_name} has not been found in the storage account. Please re-create it manually, then fix this issue."
+ }
+ },
+ "exceptions": {
+ "account_not_found": {
+ "message": "Storage account {account_name} not found"
+ },
+ "cannot_connect": {
+ "message": "Can not connect to storage account {account_name}"
+ },
+ "invalid_auth": {
+ "message": "Authentication failed for storage account {account_name}"
+ },
+ "container_not_found": {
+ "message": "Storage container {container_name} not found"
+ }
+ }
+}
diff --git a/homeassistant/components/backup/__init__.py b/homeassistant/components/backup/__init__.py
index 71a4f5ea41a..d9d1c3cc2fe 100644
--- a/homeassistant/components/backup/__init__.py
+++ b/homeassistant/components/backup/__init__.py
@@ -1,8 +1,8 @@
"""The Backup integration."""
-from homeassistant.core import HomeAssistant, ServiceCall, callback
-from homeassistant.exceptions import HomeAssistantError
+from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.backup import DATA_BACKUP
from homeassistant.helpers.hassio import is_hassio
from homeassistant.helpers.typing import ConfigType
@@ -16,6 +16,7 @@ from .agent import (
BackupAgentPlatformProtocol,
LocalBackupAgent,
)
+from .config import BackupConfig, CreateBackupParametersDict
from .const import DATA_MANAGER, DOMAIN
from .http import async_register_http_views
from .manager import (
@@ -31,6 +32,7 @@ from .manager import (
IdleEvent,
IncorrectPasswordError,
ManagerBackup,
+ ManagerStateEvent,
NewBackup,
RestoreBackupEvent,
RestoreBackupStage,
@@ -47,12 +49,14 @@ __all__ = [
"BackupAgent",
"BackupAgentError",
"BackupAgentPlatformProtocol",
+ "BackupConfig",
"BackupManagerError",
"BackupNotFound",
"BackupPlatformProtocol",
"BackupReaderWriter",
"BackupReaderWriterError",
"CreateBackupEvent",
+ "CreateBackupParametersDict",
"CreateBackupStage",
"CreateBackupState",
"Folder",
@@ -60,12 +64,12 @@ __all__ = [
"IncorrectPasswordError",
"LocalBackupAgent",
"ManagerBackup",
+ "ManagerStateEvent",
"NewBackup",
"RestoreBackupEvent",
"RestoreBackupStage",
"RestoreBackupState",
"WrittenBackup",
- "async_get_manager",
"suggested_filename",
"suggested_filename_from_name_date",
]
@@ -88,7 +92,13 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
backup_manager = BackupManager(hass, reader_writer)
hass.data[DATA_MANAGER] = backup_manager
- await backup_manager.async_setup()
+ try:
+ await backup_manager.async_setup()
+ except Exception as err:
+ hass.data[DATA_BACKUP].manager_ready.set_exception(err)
+ raise
+ else:
+ hass.data[DATA_BACKUP].manager_ready.set_result(None)
async_register_websocket_handlers(hass, with_hassio)
@@ -119,15 +129,3 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
async_register_http_views(hass)
return True
-
-
-@callback
-def async_get_manager(hass: HomeAssistant) -> BackupManager:
- """Get the backup manager instance.
-
- Raises HomeAssistantError if the backup integration is not available.
- """
- if DATA_MANAGER not in hass.data:
- raise HomeAssistantError("Backup integration is not available")
-
- return hass.data[DATA_MANAGER]
diff --git a/homeassistant/components/backup/agent.py b/homeassistant/components/backup/agent.py
index 9530f386c7b..8093ac88338 100644
--- a/homeassistant/components/backup/agent.py
+++ b/homeassistant/components/backup/agent.py
@@ -41,6 +41,8 @@ class BackupAgent(abc.ABC):
) -> AsyncIterator[bytes]:
"""Download a backup file.
+ Raises BackupNotFound if the backup does not exist.
+
:param backup_id: The ID of the backup that was returned in async_list_backups.
:return: An async iterator that yields bytes.
"""
@@ -67,6 +69,8 @@ class BackupAgent(abc.ABC):
) -> None:
"""Delete a backup file.
+ Raises BackupNotFound if the backup does not exist.
+
:param backup_id: The ID of the backup that was returned in async_list_backups.
"""
@@ -79,8 +83,11 @@ class BackupAgent(abc.ABC):
self,
backup_id: str,
**kwargs: Any,
- ) -> AgentBackup | None:
- """Return a backup."""
+ ) -> AgentBackup:
+ """Return a backup.
+
+ Raises BackupNotFound if the backup does not exist.
+ """
class LocalBackupAgent(BackupAgent):
diff --git a/homeassistant/components/backup/backup.py b/homeassistant/components/backup/backup.py
index c3a46a6ab1f..de2cfecb1a5 100644
--- a/homeassistant/components/backup/backup.py
+++ b/homeassistant/components/backup/backup.py
@@ -88,13 +88,13 @@ class CoreLocalBackupAgent(LocalBackupAgent):
self,
backup_id: str,
**kwargs: Any,
- ) -> AgentBackup | None:
+ ) -> AgentBackup:
"""Return a backup."""
if not self._loaded_backups:
await self._load_backups()
if backup_id not in self._backups:
- return None
+ raise BackupNotFound(f"Backup {backup_id} not found")
backup, backup_path = self._backups[backup_id]
if not await self._hass.async_add_executor_job(backup_path.exists):
@@ -107,7 +107,7 @@ class CoreLocalBackupAgent(LocalBackupAgent):
backup_path,
)
self._backups.pop(backup_id)
- return None
+ raise BackupNotFound(f"Backup {backup_id} not found")
return backup
@@ -130,10 +130,7 @@ class CoreLocalBackupAgent(LocalBackupAgent):
if not self._loaded_backups:
await self._load_backups()
- try:
- backup_path = self.get_backup_path(backup_id)
- except BackupNotFound:
- return
+ backup_path = self.get_backup_path(backup_id)
await self._hass.async_add_executor_job(backup_path.unlink, True)
LOGGER.debug("Deleted backup located at %s", backup_path)
self._backups.pop(backup_id)
diff --git a/homeassistant/components/backup/basic_websocket.py b/homeassistant/components/backup/basic_websocket.py
new file mode 100644
index 00000000000..614dc23a927
--- /dev/null
+++ b/homeassistant/components/backup/basic_websocket.py
@@ -0,0 +1,38 @@
+"""Websocket commands for the Backup integration."""
+
+from typing import Any
+
+import voluptuous as vol
+
+from homeassistant.components import websocket_api
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.backup import async_subscribe_events
+
+from .const import DATA_MANAGER
+from .manager import ManagerStateEvent
+
+
+@callback
+def async_register_websocket_handlers(hass: HomeAssistant) -> None:
+ """Register websocket commands."""
+ websocket_api.async_register_command(hass, handle_subscribe_events)
+
+
+@websocket_api.require_admin
+@websocket_api.websocket_command({vol.Required("type"): "backup/subscribe_events"})
+@websocket_api.async_response
+async def handle_subscribe_events(
+ hass: HomeAssistant,
+ connection: websocket_api.ActiveConnection,
+ msg: dict[str, Any],
+) -> None:
+ """Subscribe to backup events."""
+
+ def on_event(event: ManagerStateEvent) -> None:
+ connection.send_message(websocket_api.event_message(msg["id"], event))
+
+ if DATA_MANAGER in hass.data:
+ manager = hass.data[DATA_MANAGER]
+ on_event(manager.last_event)
+ connection.subscriptions[msg["id"]] = async_subscribe_events(hass, on_event)
+ connection.send_result(msg["id"])
diff --git a/homeassistant/components/backup/config.py b/homeassistant/components/backup/config.py
index 4d0cd82bc44..f4fa2e8bac6 100644
--- a/homeassistant/components/backup/config.py
+++ b/homeassistant/components/backup/config.py
@@ -12,16 +12,19 @@ from typing import TYPE_CHECKING, Self, TypedDict
from cronsim import CronSim
from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers import issue_registry as ir
from homeassistant.helpers.event import async_call_later, async_track_point_in_time
from homeassistant.helpers.typing import UNDEFINED, UndefinedType
from homeassistant.util import dt as dt_util
-from .const import LOGGER
+from .const import DOMAIN, LOGGER
from .models import BackupManagerError, Folder
if TYPE_CHECKING:
from .manager import BackupManager, ManagerBackup
+AUTOMATIC_BACKUP_AGENTS_UNAVAILABLE_ISSUE_ID = "automatic_backup_agents_unavailable"
+
CRON_PATTERN_DAILY = "{m} {h} * * *"
CRON_PATTERN_WEEKLY = "{m} {h} * * {d}"
@@ -39,6 +42,7 @@ class StoredBackupConfig(TypedDict):
"""Represent the stored backup config."""
agents: dict[str, StoredAgentConfig]
+ automatic_backups_configured: bool
create_backup: StoredCreateBackupConfig
last_attempted_automatic_backup: str | None
last_completed_automatic_backup: str | None
@@ -51,6 +55,7 @@ class BackupConfigData:
"""Represent loaded backup config data."""
agents: dict[str, AgentConfig]
+ automatic_backups_configured: bool # only used by frontend
create_backup: CreateBackupConfig
last_attempted_automatic_backup: datetime | None = None
last_completed_automatic_backup: datetime | None = None
@@ -88,6 +93,7 @@ class BackupConfigData:
agent_id: AgentConfig(protected=agent_data["protected"])
for agent_id, agent_data in data["agents"].items()
},
+ automatic_backups_configured=data["automatic_backups_configured"],
create_backup=CreateBackupConfig(
agent_ids=data["create_backup"]["agent_ids"],
include_addons=data["create_backup"]["include_addons"],
@@ -127,6 +133,7 @@ class BackupConfigData:
agents={
agent_id: agent.to_dict() for agent_id, agent in self.agents.items()
},
+ automatic_backups_configured=self.automatic_backups_configured,
create_backup=self.create_backup.to_dict(),
last_attempted_automatic_backup=last_attempted,
last_completed_automatic_backup=last_completed,
@@ -142,10 +149,12 @@ class BackupConfig:
"""Initialize backup config."""
self.data = BackupConfigData(
agents={},
+ automatic_backups_configured=False,
create_backup=CreateBackupConfig(),
retention=RetentionConfig(),
schedule=BackupSchedule(),
)
+ self._hass = hass
self._manager = manager
def load(self, stored_config: StoredBackupConfig) -> None:
@@ -154,10 +163,12 @@ class BackupConfig:
self.data.retention.apply(self._manager)
self.data.schedule.apply(self._manager)
- async def update(
+ @callback
+ def update(
self,
*,
agents: dict[str, AgentParametersDict] | UndefinedType = UNDEFINED,
+ automatic_backups_configured: bool | UndefinedType = UNDEFINED,
create_backup: CreateBackupParametersDict | UndefinedType = UNDEFINED,
retention: RetentionParametersDict | UndefinedType = UNDEFINED,
schedule: ScheduleParametersDict | UndefinedType = UNDEFINED,
@@ -171,8 +182,12 @@ class BackupConfig:
self.data.agents[agent_id] = replace(
self.data.agents[agent_id], **agent_config
)
+ if automatic_backups_configured is not UNDEFINED:
+ self.data.automatic_backups_configured = automatic_backups_configured
if create_backup is not UNDEFINED:
self.data.create_backup = replace(self.data.create_backup, **create_backup)
+ if "agent_ids" in create_backup:
+ check_unavailable_agents(self._hass, self._manager)
if retention is not UNDEFINED:
new_retention = RetentionConfig(**retention)
if new_retention != self.data.retention:
@@ -553,3 +568,46 @@ async def delete_backups_exceeding_configured_count(manager: BackupManager) -> N
await manager.async_delete_filtered_backups(
include_filter=_automatic_backups_filter, delete_filter=_delete_filter
)
+
+
+@callback
+def check_unavailable_agents(hass: HomeAssistant, manager: BackupManager) -> None:
+ """Check for unavailable agents."""
+ if missing_agent_ids := set(manager.config.data.create_backup.agent_ids) - set(
+ manager.backup_agents
+ ):
+ LOGGER.debug(
+ "Agents %s are configured for automatic backup but are unavailable",
+ missing_agent_ids,
+ )
+
+ # Remove issues for unavailable agents that are not unavailable anymore.
+ issue_registry = ir.async_get(hass)
+ existing_missing_agent_issue_ids = {
+ issue_id
+ for domain, issue_id in issue_registry.issues
+ if domain == DOMAIN
+ and issue_id.startswith(AUTOMATIC_BACKUP_AGENTS_UNAVAILABLE_ISSUE_ID)
+ }
+ current_missing_agent_issue_ids = {
+ f"{AUTOMATIC_BACKUP_AGENTS_UNAVAILABLE_ISSUE_ID}_{agent_id}": agent_id
+ for agent_id in missing_agent_ids
+ }
+ for issue_id in existing_missing_agent_issue_ids - set(
+ current_missing_agent_issue_ids
+ ):
+ ir.async_delete_issue(hass, DOMAIN, issue_id)
+ for issue_id, agent_id in current_missing_agent_issue_ids.items():
+ ir.async_create_issue(
+ hass,
+ DOMAIN,
+ issue_id,
+ is_fixable=False,
+ learn_more_url="homeassistant://config/backup",
+ severity=ir.IssueSeverity.WARNING,
+ translation_key="automatic_backup_agents_unavailable",
+ translation_placeholders={
+ "agent_id": agent_id,
+ "backup_settings": "/config/backup/settings",
+ },
+ )
diff --git a/homeassistant/components/backup/http.py b/homeassistant/components/backup/http.py
index 58f44d4a449..8f241e6363d 100644
--- a/homeassistant/components/backup/http.py
+++ b/homeassistant/components/backup/http.py
@@ -15,6 +15,7 @@ from multidict import istr
from homeassistant.components.http import KEY_HASS, HomeAssistantView, require_admin
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers import frame
from homeassistant.util import slugify
from . import util
@@ -59,11 +60,19 @@ class DownloadBackupView(HomeAssistantView):
if agent_id not in manager.backup_agents:
return Response(status=HTTPStatus.BAD_REQUEST)
agent = manager.backup_agents[agent_id]
- backup = await agent.async_get_backup(backup_id)
+ try:
+ backup = await agent.async_get_backup(backup_id)
+ except BackupNotFound:
+ return Response(status=HTTPStatus.NOT_FOUND)
- # We don't need to check if the path exists, aiohttp.FileResponse will handle
- # that
- if backup is None:
+ # Check for None to be backwards compatible with the old BackupAgent API,
+ # this can be removed in HA Core 2025.10
+ if not backup:
+ frame.report_usage(
+ "returns None from BackupAgent.async_get_backup",
+ breaks_in_ha_version="2025.10",
+ integration_domain=agent_id.partition(".")[0],
+ )
return Response(status=HTTPStatus.NOT_FOUND)
headers = {
@@ -92,6 +101,8 @@ class DownloadBackupView(HomeAssistantView):
) -> StreamResponse | FileResponse | Response:
if agent_id in manager.local_backup_agents:
local_agent = manager.local_backup_agents[agent_id]
+ # We don't need to check if the path exists, aiohttp.FileResponse will
+ # handle that
path = local_agent.get_backup_path(backup_id)
return FileResponse(path=path.as_posix(), headers=headers)
diff --git a/homeassistant/components/backup/manager.py b/homeassistant/components/backup/manager.py
index 25393a872cc..6dbe863185c 100644
--- a/homeassistant/components/backup/manager.py
+++ b/homeassistant/components/backup/manager.py
@@ -4,6 +4,7 @@ from __future__ import annotations
import abc
import asyncio
+from collections import defaultdict
from collections.abc import AsyncIterator, Callable, Coroutine
from dataclasses import dataclass, replace
from enum import StrEnum
@@ -13,6 +14,7 @@ from itertools import chain
import json
from pathlib import Path, PurePath
import shutil
+import sys
import tarfile
import time
from typing import IO, TYPE_CHECKING, Any, Protocol, TypedDict, cast
@@ -28,10 +30,13 @@ from homeassistant.backup_restore import (
from homeassistant.const import __version__ as HAVERSION
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import (
+ frame,
instance_id,
integration_platform,
issue_registry as ir,
+ start,
)
+from homeassistant.helpers.backup import DATA_BACKUP
from homeassistant.helpers.json import json_bytes
from homeassistant.util import dt as dt_util, json as json_util
@@ -42,7 +47,12 @@ from .agent import (
BackupAgentPlatformProtocol,
LocalBackupAgent,
)
-from .config import BackupConfig, delete_backups_exceeding_configured_count
+from .config import (
+ BackupConfig,
+ CreateBackupParametersDict,
+ check_unavailable_agents,
+ delete_backups_exceeding_configured_count,
+)
from .const import (
BUF_SIZE,
DATA_MANAGER,
@@ -55,6 +65,7 @@ from .models import (
AgentBackup,
BackupError,
BackupManagerError,
+ BackupNotFound,
BackupReaderWriterError,
BaseBackup,
Folder,
@@ -109,6 +120,7 @@ class BackupManagerState(StrEnum):
IDLE = "idle"
CREATE_BACKUP = "create_backup"
+ BLOCKED = "blocked"
RECEIVE_BACKUP = "receive_backup"
RESTORE_BACKUP = "restore_backup"
@@ -217,6 +229,13 @@ class RestoreBackupEvent(ManagerStateEvent):
state: RestoreBackupState
+@dataclass(frozen=True, kw_only=True, slots=True)
+class BlockedEvent(ManagerStateEvent):
+ """Backup manager blocked, Home Assistant is starting."""
+
+ manager_state: BackupManagerState = BackupManagerState.BLOCKED
+
+
class BackupPlatformProtocol(Protocol):
"""Define the format that backup platforms can have."""
@@ -281,6 +300,10 @@ class BackupReaderWriter(abc.ABC):
) -> None:
"""Get restore events after core restart."""
+ @abc.abstractmethod
+ async def async_validate_config(self, *, config: BackupConfig) -> None:
+ """Validate backup config."""
+
class IncorrectPasswordError(BackupReaderWriterError):
"""Raised when the password is incorrect."""
@@ -296,6 +319,12 @@ class DecryptOnDowloadNotSupported(BackupManagerError):
_message = "On-the-fly decryption is not supported for this backup."
+class BackupManagerExceptionGroup(BackupManagerError, ExceptionGroup):
+ """Raised when multiple exceptions occur."""
+
+ error_code = "multiple_errors"
+
+
class BackupManager:
"""Define the format that backup managers can have."""
@@ -321,9 +350,11 @@ class BackupManager:
self.remove_next_delete_event: Callable[[], None] | None = None
# Latest backup event and backup event subscribers
- self.last_event: ManagerStateEvent = IdleEvent()
- self.last_non_idle_event: ManagerStateEvent | None = None
- self._backup_event_subscriptions: list[Callable[[ManagerStateEvent], None]] = []
+ self.last_event: ManagerStateEvent = BlockedEvent()
+ self.last_action_event: ManagerStateEvent | None = None
+ self._backup_event_subscriptions = hass.data[
+ DATA_BACKUP
+ ].backup_event_subscriptions
async def async_setup(self) -> None:
"""Set up the backup manager."""
@@ -332,10 +363,20 @@ class BackupManager:
self.config.load(stored["config"])
self.known_backups.load(stored["backups"])
+ await self._reader_writer.async_validate_config(config=self.config)
+
await self._reader_writer.async_resume_restore_progress_after_restart(
on_progress=self.async_on_backup_event
)
+ async def set_manager_idle_after_start(hass: HomeAssistant) -> None:
+ """Set manager to idle after start."""
+ self.async_on_backup_event(IdleEvent())
+
+ if self.state == BackupManagerState.BLOCKED:
+ # If we're not finishing a restore job, set the manager to idle after start
+ start.async_at_started(self.hass, set_manager_idle_after_start)
+
await self.load_platforms()
@property
@@ -404,6 +445,13 @@ class BackupManager:
}
)
+ @callback
+ def check_unavailable_agents_after_start(hass: HomeAssistant) -> None:
+ """Check unavailable agents after start."""
+ check_unavailable_agents(hass, self)
+
+ start.async_at_started(self.hass, check_unavailable_agents_after_start)
+
async def _add_platform(
self,
hass: HomeAssistant,
@@ -560,8 +608,15 @@ class BackupManager:
return_exceptions=True,
)
for idx, result in enumerate(list_backups_results):
+ agent_id = agent_ids[idx]
if isinstance(result, BackupAgentError):
- agent_errors[agent_ids[idx]] = result
+ agent_errors[agent_id] = result
+ continue
+ if isinstance(result, Exception):
+ agent_errors[agent_id] = result
+ LOGGER.error(
+ "Unexpected error for %s: %s", agent_id, result, exc_info=result
+ )
continue
if isinstance(result, BaseException):
raise result # unexpected error
@@ -588,7 +643,7 @@ class BackupManager:
name=agent_backup.name,
with_automatic_settings=with_automatic_settings,
)
- backups[backup_id].agents[agent_ids[idx]] = AgentBackupStatus(
+ backups[backup_id].agents[agent_id] = AgentBackupStatus(
protected=agent_backup.protected,
size=agent_backup.size,
)
@@ -611,12 +666,28 @@ class BackupManager:
return_exceptions=True,
)
for idx, result in enumerate(get_backup_results):
+ agent_id = agent_ids[idx]
+ if isinstance(result, BackupNotFound):
+ continue
if isinstance(result, BackupAgentError):
- agent_errors[agent_ids[idx]] = result
+ agent_errors[agent_id] = result
+ continue
+ if isinstance(result, Exception):
+ agent_errors[agent_id] = result
+ LOGGER.error(
+ "Unexpected error for %s: %s", agent_id, result, exc_info=result
+ )
continue
if isinstance(result, BaseException):
raise result # unexpected error
+ # Check for None to be backwards compatible with the old BackupAgent API,
+ # this can be removed in HA Core 2025.10
if not result:
+ frame.report_usage(
+ "returns None from BackupAgent.async_get_backup",
+ breaks_in_ha_version="2025.10",
+ integration_domain=agent_id.partition(".")[0],
+ )
continue
if backup is None:
if known_backup := self.known_backups.get(backup_id):
@@ -640,7 +711,7 @@ class BackupManager:
name=result.name,
with_automatic_settings=with_automatic_settings,
)
- backup.agents[agent_ids[idx]] = AgentBackupStatus(
+ backup.agents[agent_id] = AgentBackupStatus(
protected=result.protected,
size=result.size,
)
@@ -663,21 +734,33 @@ class BackupManager:
return None
return with_automatic_settings
- async def async_delete_backup(self, backup_id: str) -> dict[str, Exception]:
+ async def async_delete_backup(
+ self, backup_id: str, *, agent_ids: list[str] | None = None
+ ) -> dict[str, Exception]:
"""Delete a backup."""
agent_errors: dict[str, Exception] = {}
- agent_ids = list(self.backup_agents)
+ if agent_ids is None:
+ agent_ids = list(self.backup_agents)
delete_backup_results = await asyncio.gather(
*(
- agent.async_delete_backup(backup_id)
- for agent in self.backup_agents.values()
+ self.backup_agents[agent_id].async_delete_backup(backup_id)
+ for agent_id in agent_ids
),
return_exceptions=True,
)
for idx, result in enumerate(delete_backup_results):
+ agent_id = agent_ids[idx]
+ if isinstance(result, BackupNotFound):
+ continue
if isinstance(result, BackupAgentError):
- agent_errors[agent_ids[idx]] = result
+ agent_errors[agent_id] = result
+ continue
+ if isinstance(result, Exception):
+ agent_errors[agent_id] = result
+ LOGGER.error(
+ "Unexpected error for %s: %s", agent_id, result, exc_info=result
+ )
continue
if isinstance(result, BaseException):
raise result # unexpected error
@@ -710,40 +793,76 @@ class BackupManager:
# Run the include filter first to ensure we only consider backups that
# should be included in the deletion process.
backups = include_filter(backups)
+ backups_by_agent: dict[str, dict[str, ManagerBackup]] = defaultdict(dict)
+ for backup_id, backup in backups.items():
+ for agent_id in backup.agents:
+ backups_by_agent[agent_id][backup_id] = backup
- LOGGER.debug("Total automatic backups: %s", backups)
+ LOGGER.debug("Backups returned by include filter: %s", backups)
+ LOGGER.debug(
+ "Backups returned by include filter by agent: %s",
+ {agent_id: list(backups) for agent_id, backups in backups_by_agent.items()},
+ )
backups_to_delete = delete_filter(backups)
+ LOGGER.debug("Backups returned by delete filter: %s", backups_to_delete)
+
if not backups_to_delete:
return
# always delete oldest backup first
- backups_to_delete = dict(
- sorted(
- backups_to_delete.items(),
- key=lambda backup_item: backup_item[1].date,
- )
+ backups_to_delete_by_agent: dict[str, dict[str, ManagerBackup]] = defaultdict(
+ dict
+ )
+ for backup_id, backup in sorted(
+ backups_to_delete.items(),
+ key=lambda backup_item: backup_item[1].date,
+ ):
+ for agent_id in backup.agents:
+ backups_to_delete_by_agent[agent_id][backup_id] = backup
+ LOGGER.debug(
+ "Backups returned by delete filter by agent: %s",
+ {
+ agent_id: list(backups)
+ for agent_id, backups in backups_to_delete_by_agent.items()
+ },
+ )
+ for agent_id, to_delete_from_agent in backups_to_delete_by_agent.items():
+ if len(to_delete_from_agent) >= len(backups_by_agent[agent_id]):
+ # Never delete the last backup.
+ last_backup = to_delete_from_agent.popitem()
+ LOGGER.debug(
+ "Keeping the last backup %s for agent %s", last_backup, agent_id
+ )
+
+ LOGGER.debug(
+ "Backups to delete by agent: %s",
+ {
+ agent_id: list(backups)
+ for agent_id, backups in backups_to_delete_by_agent.items()
+ },
)
- if len(backups_to_delete) >= len(backups):
- # Never delete the last backup.
- last_backup = backups_to_delete.popitem()
- LOGGER.debug("Keeping the last backup: %s", last_backup)
+ backup_ids_to_delete: dict[str, set[str]] = defaultdict(set)
+ for agent_id, to_delete in backups_to_delete_by_agent.items():
+ for backup_id in to_delete:
+ backup_ids_to_delete[backup_id].add(agent_id)
- LOGGER.debug("Backups to delete: %s", backups_to_delete)
-
- if not backups_to_delete:
+ if not backup_ids_to_delete:
return
- backup_ids = list(backups_to_delete)
+ backup_ids = list(backup_ids_to_delete)
delete_results = await asyncio.gather(
- *(self.async_delete_backup(backup_id) for backup_id in backups_to_delete)
+ *(
+ self.async_delete_backup(backup_id, agent_ids=list(agent_ids))
+ for backup_id, agent_ids in backup_ids_to_delete.items()
+ )
)
agent_errors = {
backup_id: error
for backup_id, error in zip(backup_ids, delete_results, strict=True)
- if error
+ if error and not isinstance(error, BackupNotFound)
}
if agent_errors:
LOGGER.error(
@@ -1175,7 +1294,20 @@ class BackupManager:
) -> None:
"""Initiate restoring a backup."""
agent = self.backup_agents[agent_id]
- if not await agent.async_get_backup(backup_id):
+ try:
+ backup = await agent.async_get_backup(backup_id)
+ except BackupNotFound as err:
+ raise BackupManagerError(
+ f"Backup {backup_id} not found in agent {agent_id}"
+ ) from err
+ # Check for None to be backwards compatible with the old BackupAgent API,
+ # this can be removed in HA Core 2025.10
+ if not backup:
+ frame.report_usage(
+ "returns None from BackupAgent.async_get_backup",
+ breaks_in_ha_version="2025.10",
+ integration_domain=agent_id.partition(".")[0],
+ )
raise BackupManagerError(
f"Backup {backup_id} not found in agent {agent_id}"
)
@@ -1204,24 +1336,11 @@ class BackupManager:
if (current_state := self.state) != (new_state := event.manager_state):
LOGGER.debug("Backup state: %s -> %s", current_state, new_state)
self.last_event = event
- if not isinstance(event, IdleEvent):
- self.last_non_idle_event = event
+ if not isinstance(event, (BlockedEvent, IdleEvent)):
+ self.last_action_event = event
for subscription in self._backup_event_subscriptions:
subscription(event)
- @callback
- def async_subscribe_events(
- self,
- on_event: Callable[[ManagerStateEvent], None],
- ) -> Callable[[], None]:
- """Subscribe events."""
-
- def remove_subscription() -> None:
- self._backup_event_subscriptions.remove(on_event)
-
- self._backup_event_subscriptions.append(on_event)
- return remove_subscription
-
def _update_issue_backup_failed(self) -> None:
"""Update issue registry when a backup fails."""
ir.async_create_issue(
@@ -1276,7 +1395,20 @@ class BackupManager:
agent = self.backup_agents[agent_id]
except KeyError as err:
raise BackupManagerError(f"Invalid agent selected: {agent_id}") from err
- if not await agent.async_get_backup(backup_id):
+ try:
+ backup = await agent.async_get_backup(backup_id)
+ except BackupNotFound as err:
+ raise BackupManagerError(
+ f"Backup {backup_id} not found in agent {agent_id}"
+ ) from err
+ # Check for None to be backwards compatible with the old BackupAgent API,
+ # this can be removed in HA Core 2025.10
+ if not backup:
+ frame.report_usage(
+ "returns None from BackupAgent.async_get_backup",
+ breaks_in_ha_version="2025.10",
+ integration_domain=agent_id.partition(".")[0],
+ )
raise BackupManagerError(
f"Backup {backup_id} not found in agent {agent_id}"
)
@@ -1536,10 +1668,24 @@ class CoreBackupReaderWriter(BackupReaderWriter):
)
finally:
# Inform integrations the backup is done
+ # If there's an unhandled exception, we keep it so we can rethrow it in case
+ # the post backup actions also fail.
+ unhandled_exc = sys.exception()
try:
- await manager.async_post_backup_actions()
- except BackupManagerError as err:
- raise BackupReaderWriterError(str(err)) from err
+ try:
+ await manager.async_post_backup_actions()
+ except BackupManagerError as err:
+ raise BackupReaderWriterError(str(err)) from err
+ except Exception as err:
+ if not unhandled_exc:
+ raise
+ # If there's an unhandled exception, we wrap both that and the exception
+ # from the post backup actions in an ExceptionGroup so the caller is
+ # aware of both exceptions.
+ raise BackupManagerExceptionGroup(
+ f"Multiple errors when creating backup: {unhandled_exc}, {err}",
+ [unhandled_exc, err],
+ ) from None
def _mkdir_and_generate_backup_contents(
self,
@@ -1551,7 +1697,13 @@ class CoreBackupReaderWriter(BackupReaderWriter):
"""Generate backup contents and return the size."""
if not tar_file_path:
tar_file_path = self.temp_backup_dir / f"{backup_data['slug']}.tar"
- make_backup_dir(tar_file_path.parent)
+ try:
+ make_backup_dir(tar_file_path.parent)
+ except OSError as err:
+ raise BackupReaderWriterError(
+ f"Failed to create dir {tar_file_path.parent}: "
+ f"{err} ({err.__class__.__name__})"
+ ) from err
excludes = EXCLUDE_FROM_BACKUP
if not database_included:
@@ -1589,7 +1741,14 @@ class CoreBackupReaderWriter(BackupReaderWriter):
file_filter=is_excluded_by_filter,
arcname="data",
)
- return (tar_file_path, tar_file_path.stat().st_size)
+ try:
+ stat_result = tar_file_path.stat()
+ except OSError as err:
+ raise BackupReaderWriterError(
+ f"Error getting size of {tar_file_path}: "
+ f"{err} ({err.__class__.__name__})"
+ ) from err
+ return (tar_file_path, stat_result.st_size)
async def async_receive_backup(
self,
@@ -1771,6 +1930,44 @@ class CoreBackupReaderWriter(BackupReaderWriter):
)
on_progress(IdleEvent())
+ async def async_validate_config(self, *, config: BackupConfig) -> None:
+ """Validate backup config.
+
+ Update automatic backup settings to not include addons or folders and remove
+ hassio agents in case a backup created by supervisor was restored.
+ """
+ create_backup = config.data.create_backup
+ if (
+ not create_backup.include_addons
+ and not create_backup.include_all_addons
+ and not create_backup.include_folders
+ and not any(a_id.startswith("hassio.") for a_id in create_backup.agent_ids)
+ ):
+ LOGGER.debug("Backup settings don't need to be adjusted")
+ return
+
+ LOGGER.info(
+ "Adjusting backup settings to not include addons, folders or supervisor locations"
+ )
+ automatic_agents = [
+ agent_id
+ for agent_id in create_backup.agent_ids
+ if not agent_id.startswith("hassio.")
+ ]
+ if (
+ self._local_agent_id not in automatic_agents
+ and "hassio.local" in create_backup.agent_ids
+ ):
+ automatic_agents = [self._local_agent_id, *automatic_agents]
+ config.update(
+ create_backup=CreateBackupParametersDict(
+ agent_ids=automatic_agents,
+ include_addons=None,
+ include_all_addons=False,
+ include_folders=None,
+ )
+ )
+
def _generate_backup_id(date: str, name: str) -> str:
"""Generate a backup ID."""
diff --git a/homeassistant/components/backup/manifest.json b/homeassistant/components/backup/manifest.json
index 6cbfb834c7f..db0719983b1 100644
--- a/homeassistant/components/backup/manifest.json
+++ b/homeassistant/components/backup/manifest.json
@@ -8,5 +8,5 @@
"integration_type": "system",
"iot_class": "calculated",
"quality_scale": "internal",
- "requirements": ["cronsim==2.6", "securetar==2025.1.4"]
+ "requirements": ["cronsim==2.6", "securetar==2025.2.1"]
}
diff --git a/homeassistant/components/backup/store.py b/homeassistant/components/backup/store.py
index 9b4af823c77..883447853e6 100644
--- a/homeassistant/components/backup/store.py
+++ b/homeassistant/components/backup/store.py
@@ -16,7 +16,7 @@ if TYPE_CHECKING:
STORE_DELAY_SAVE = 30
STORAGE_KEY = DOMAIN
STORAGE_VERSION = 1
-STORAGE_VERSION_MINOR = 3
+STORAGE_VERSION_MINOR = 5
class StoredBackupData(TypedDict):
@@ -60,6 +60,18 @@ class _BackupStore(Store[StoredBackupData]):
else:
data["config"]["schedule"]["days"] = [state]
data["config"]["schedule"]["recurrence"] = "custom_days"
+ if old_minor_version < 4:
+ # Workaround for a bug in frontend which incorrectly set days to 0
+ # instead of to None for unlimited retention.
+ if data["config"]["retention"]["copies"] == 0:
+ data["config"]["retention"]["copies"] = None
+ if data["config"]["retention"]["days"] == 0:
+ data["config"]["retention"]["days"] = None
+ if old_minor_version < 5:
+ # Version 1.5 adds automatic_backups_configured
+ data["config"]["automatic_backups_configured"] = (
+ data["config"]["create_backup"]["password"] is not None
+ )
# Note: We allow reading data with major version 2.
# Reject if major version is higher than 2.
diff --git a/homeassistant/components/backup/strings.json b/homeassistant/components/backup/strings.json
index 32d76ded049..c3047d3a4ac 100644
--- a/homeassistant/components/backup/strings.json
+++ b/homeassistant/components/backup/strings.json
@@ -1,5 +1,9 @@
{
"issues": {
+ "automatic_backup_agents_unavailable": {
+ "title": "The backup location {agent_id} is unavailable",
+ "description": "The backup location `{agent_id}` is unavailable but is still configured for automatic backups.\n\nPlease visit the [automatic backup configuration page]({backup_settings}) to review and update your backup locations. Backups will not be uploaded to selected locations that are unavailable."
+ },
"automatic_backup_failed_create": {
"title": "Automatic backup could not be created",
"description": "The automatic backup could not be created. Please check the logs for more information. Another attempt will be made at the next scheduled time if a backup schedule is configured."
diff --git a/homeassistant/components/backup/util.py b/homeassistant/components/backup/util.py
index 9d8f6e815dc..bd77880738e 100644
--- a/homeassistant/components/backup/util.py
+++ b/homeassistant/components/backup/util.py
@@ -104,12 +104,15 @@ def read_backup(backup_path: Path) -> AgentBackup:
bool, homeassistant.get("exclude_database", False)
)
+ extra_metadata = cast(dict[str, bool | str], data.get("extra", {}))
+ date = extra_metadata.get("supervisor.backup_request_date", data["date"])
+
return AgentBackup(
addons=addons,
backup_id=cast(str, data["slug"]),
database_included=database_included,
- date=cast(str, data["date"]),
- extra_metadata=cast(dict[str, bool | str], data.get("extra", {})),
+ date=cast(str, date),
+ extra_metadata=extra_metadata,
folders=folders,
homeassistant_included=homeassistant_included,
homeassistant_version=homeassistant_version,
diff --git a/homeassistant/components/backup/websocket.py b/homeassistant/components/backup/websocket.py
index b6d092e1913..4c370a4224d 100644
--- a/homeassistant/components/backup/websocket.py
+++ b/homeassistant/components/backup/websocket.py
@@ -10,11 +10,7 @@ from homeassistant.helpers import config_validation as cv
from .config import Day, ScheduleRecurrence
from .const import DATA_MANAGER, LOGGER
-from .manager import (
- DecryptOnDowloadNotSupported,
- IncorrectPasswordError,
- ManagerStateEvent,
-)
+from .manager import DecryptOnDowloadNotSupported, IncorrectPasswordError
from .models import BackupNotFound, Folder
@@ -34,7 +30,6 @@ def async_register_websocket_handlers(hass: HomeAssistant, with_hassio: bool) ->
websocket_api.async_register_command(hass, handle_create_with_automatic_settings)
websocket_api.async_register_command(hass, handle_delete)
websocket_api.async_register_command(hass, handle_restore)
- websocket_api.async_register_command(hass, handle_subscribe_events)
websocket_api.async_register_command(hass, handle_config_info)
websocket_api.async_register_command(hass, handle_config_update)
@@ -60,7 +55,7 @@ async def handle_info(
"backups": list(backups.values()),
"last_attempted_automatic_backup": manager.config.data.last_attempted_automatic_backup,
"last_completed_automatic_backup": manager.config.data.last_completed_automatic_backup,
- "last_non_idle_event": manager.last_non_idle_event,
+ "last_action_event": manager.last_action_event,
"next_automatic_backup": manager.config.data.schedule.next_automatic_backup,
"next_automatic_backup_additional": manager.config.data.schedule.next_automatic_backup_additional,
"state": manager.state,
@@ -346,11 +341,13 @@ async def handle_config_info(
)
+@callback
@websocket_api.require_admin
@websocket_api.websocket_command(
{
vol.Required("type"): "backup/config/update",
vol.Optional("agents"): vol.Schema({str: {"protected": bool}}),
+ vol.Optional("automatic_backups_configured"): bool,
vol.Optional("create_backup"): vol.Schema(
{
vol.Optional("agent_ids"): vol.All([str], vol.Unique()),
@@ -368,8 +365,10 @@ async def handle_config_info(
),
vol.Optional("retention"): vol.Schema(
{
- vol.Optional("copies"): vol.Any(int, None),
- vol.Optional("days"): vol.Any(int, None),
+ # Note: We can't use cv.positive_int because it allows 0 even
+ # though 0 is not positive.
+ vol.Optional("copies"): vol.Any(vol.All(int, vol.Range(min=1)), None),
+ vol.Optional("days"): vol.Any(vol.All(int, vol.Range(min=1)), None),
},
),
vol.Optional("schedule"): vol.Schema(
@@ -385,8 +384,7 @@ async def handle_config_info(
),
}
)
-@websocket_api.async_response
-async def handle_config_update(
+def handle_config_update(
hass: HomeAssistant,
connection: websocket_api.ActiveConnection,
msg: dict[str, Any],
@@ -396,24 +394,5 @@ async def handle_config_update(
changes = dict(msg)
changes.pop("id")
changes.pop("type")
- await manager.config.update(**changes)
- connection.send_result(msg["id"])
-
-
-@websocket_api.require_admin
-@websocket_api.websocket_command({vol.Required("type"): "backup/subscribe_events"})
-@websocket_api.async_response
-async def handle_subscribe_events(
- hass: HomeAssistant,
- connection: websocket_api.ActiveConnection,
- msg: dict[str, Any],
-) -> None:
- """Subscribe to backup events."""
-
- def on_event(event: ManagerStateEvent) -> None:
- connection.send_message(websocket_api.event_message(msg["id"], event))
-
- manager = hass.data[DATA_MANAGER]
- on_event(manager.last_event)
- connection.subscriptions[msg["id"]] = manager.async_subscribe_events(on_event)
+ manager.config.update(**changes)
connection.send_result(msg["id"])
diff --git a/homeassistant/components/baf/binary_sensor.py b/homeassistant/components/baf/binary_sensor.py
index 7c855711712..e12bfd8b90c 100644
--- a/homeassistant/components/baf/binary_sensor.py
+++ b/homeassistant/components/baf/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BAFConfigEntry
from .entity import BAFDescriptionEntity
@@ -41,7 +41,7 @@ OCCUPANCY_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: BAFConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BAF binary sensors."""
device = entry.runtime_data
diff --git a/homeassistant/components/baf/climate.py b/homeassistant/components/baf/climate.py
index c30d49e8c9d..abcc2afe254 100644
--- a/homeassistant/components/baf/climate.py
+++ b/homeassistant/components/baf/climate.py
@@ -12,7 +12,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BAFConfigEntry
from .entity import BAFEntity
@@ -21,7 +21,7 @@ from .entity import BAFEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BAFConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BAF fan auto comfort."""
device = entry.runtime_data
diff --git a/homeassistant/components/baf/fan.py b/homeassistant/components/baf/fan.py
index 8f7aab40b79..c990a248588 100644
--- a/homeassistant/components/baf/fan.py
+++ b/homeassistant/components/baf/fan.py
@@ -14,7 +14,7 @@ from homeassistant.components.fan import (
FanEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -28,7 +28,7 @@ from .entity import BAFEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BAFConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SenseME fans."""
device = entry.runtime_data
diff --git a/homeassistant/components/baf/light.py b/homeassistant/components/baf/light.py
index 4c0b1e353fe..e8298a8e4d4 100644
--- a/homeassistant/components/baf/light.py
+++ b/homeassistant/components/baf/light.py
@@ -13,7 +13,7 @@ from homeassistant.components.light import (
LightEntity,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BAFConfigEntry
from .entity import BAFEntity
@@ -22,7 +22,7 @@ from .entity import BAFEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BAFConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BAF lights."""
device = entry.runtime_data
diff --git a/homeassistant/components/baf/number.py b/homeassistant/components/baf/number.py
index a2e5e704e4d..87b5cdc095b 100644
--- a/homeassistant/components/baf/number.py
+++ b/homeassistant/components/baf/number.py
@@ -15,7 +15,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BAFConfigEntry
from .const import HALF_DAY_SECS, ONE_DAY_SECS, ONE_MIN_SECS, SPEED_RANGE
@@ -116,7 +116,7 @@ LIGHT_NUMBER_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: BAFConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BAF numbers."""
device = entry.runtime_data
diff --git a/homeassistant/components/baf/sensor.py b/homeassistant/components/baf/sensor.py
index 7e664254a38..e9b8965b7c4 100644
--- a/homeassistant/components/baf/sensor.py
+++ b/homeassistant/components/baf/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BAFConfigEntry
from .entity import BAFDescriptionEntity
@@ -93,7 +93,7 @@ FAN_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: BAFConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BAF fan sensors."""
device = entry.runtime_data
diff --git a/homeassistant/components/baf/strings.json b/homeassistant/components/baf/strings.json
index e2f02a6095e..64956984bb8 100644
--- a/homeassistant/components/baf/strings.json
+++ b/homeassistant/components/baf/strings.json
@@ -23,7 +23,7 @@
"entity": {
"climate": {
"auto_comfort": {
- "name": "Auto comfort"
+ "name": "Auto Comfort"
}
},
"fan": {
@@ -39,25 +39,25 @@
},
"number": {
"comfort_min_speed": {
- "name": "Auto Comfort Minimum Speed"
+ "name": "Auto Comfort minimum speed"
},
"comfort_max_speed": {
- "name": "Auto Comfort Maximum Speed"
+ "name": "Auto Comfort maximum speed"
},
"comfort_heat_assist_speed": {
- "name": "Auto Comfort Heat Assist Speed"
+ "name": "Auto Comfort Heat Assist speed"
},
"return_to_auto_timeout": {
- "name": "Return to Auto Timeout"
+ "name": "Return to Auto timeout"
},
"motion_sense_timeout": {
- "name": "Motion Sense Timeout"
+ "name": "Motion sense timeout"
},
"light_return_to_auto_timeout": {
- "name": "Light Return to Auto Timeout"
+ "name": "Light return to Auto timeout"
},
"light_auto_motion_timeout": {
- "name": "Light Motion Sense Timeout"
+ "name": "Light motion sense timeout"
}
},
"sensor": {
@@ -76,10 +76,10 @@
},
"switch": {
"legacy_ir_remote_enable": {
- "name": "Legacy IR Remote"
+ "name": "Legacy IR remote"
},
"led_indicators_enable": {
- "name": "Led Indicators"
+ "name": "LED indicators"
},
"comfort_heat_assist_enable": {
"name": "Auto Comfort Heat Assist"
@@ -88,10 +88,10 @@
"name": "Beep"
},
"eco_enable": {
- "name": "Eco Mode"
+ "name": "Eco mode"
},
"motion_sense_enable": {
- "name": "Motion Sense"
+ "name": "Motion sense"
},
"return_to_auto_enable": {
"name": "Return to Auto"
@@ -103,7 +103,7 @@
"name": "Dim to Warm"
},
"light_return_to_auto_enable": {
- "name": "Light Return to Auto"
+ "name": "Light return to Auto"
}
}
}
diff --git a/homeassistant/components/baf/switch.py b/homeassistant/components/baf/switch.py
index e18e26ddcaa..50bd90a6107 100644
--- a/homeassistant/components/baf/switch.py
+++ b/homeassistant/components/baf/switch.py
@@ -11,7 +11,7 @@ from aiobafi6 import Device
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BAFConfigEntry
from .entity import BAFDescriptionEntity
@@ -103,7 +103,7 @@ LIGHT_SWITCHES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: BAFConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BAF fan switches."""
device = entry.runtime_data
diff --git a/homeassistant/components/balboa/__init__.py b/homeassistant/components/balboa/__init__.py
index c982d59d513..54ae569bb78 100644
--- a/homeassistant/components/balboa/__init__.py
+++ b/homeassistant/components/balboa/__init__.py
@@ -21,12 +21,14 @@ _LOGGER = logging.getLogger(__name__)
PLATFORMS = [
Platform.BINARY_SENSOR,
Platform.CLIMATE,
+ Platform.EVENT,
Platform.FAN,
Platform.LIGHT,
Platform.SELECT,
+ Platform.SWITCH,
+ Platform.TIME,
]
-
KEEP_ALIVE_INTERVAL = timedelta(minutes=1)
SYNC_TIME_INTERVAL = timedelta(hours=1)
diff --git a/homeassistant/components/balboa/binary_sensor.py b/homeassistant/components/balboa/binary_sensor.py
index b8c62ce8abf..437a01866b8 100644
--- a/homeassistant/components/balboa/binary_sensor.py
+++ b/homeassistant/components/balboa/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BalboaConfigEntry
from .entity import BalboaEntity
@@ -22,7 +22,7 @@ from .entity import BalboaEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BalboaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the spa's binary sensors."""
spa = entry.runtime_data
diff --git a/homeassistant/components/balboa/climate.py b/homeassistant/components/balboa/climate.py
index 76b02f0e165..3fb2457d610 100644
--- a/homeassistant/components/balboa/climate.py
+++ b/homeassistant/components/balboa/climate.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BalboaConfigEntry
from .const import DOMAIN
@@ -47,7 +47,7 @@ TEMPERATURE_UNIT_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
entry: BalboaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the spa climate entity."""
async_add_entities([BalboaClimateEntity(entry.runtime_data)])
diff --git a/homeassistant/components/balboa/event.py b/homeassistant/components/balboa/event.py
new file mode 100644
index 00000000000..57263c34783
--- /dev/null
+++ b/homeassistant/components/balboa/event.py
@@ -0,0 +1,91 @@
+"""Support for Balboa events."""
+
+from __future__ import annotations
+
+from datetime import datetime, timedelta
+
+from pybalboa import EVENT_UPDATE, SpaClient
+
+from homeassistant.components.event import EventEntity
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.event import async_track_time_interval
+
+from . import BalboaConfigEntry
+from .entity import BalboaEntity
+
+FAULT = "fault"
+FAULT_DATE = "fault_date"
+REQUEST_FAULT_LOG_INTERVAL = timedelta(minutes=5)
+
+FAULT_MESSAGE_CODE_MAP: dict[int, str] = {
+ 15: "sensor_out_of_sync",
+ 16: "low_flow",
+ 17: "flow_failed",
+ 18: "settings_reset",
+ 19: "priming_mode",
+ 20: "clock_failed",
+ 21: "settings_reset",
+ 22: "memory_failure",
+ 26: "service_sensor_sync",
+ 27: "heater_dry",
+ 28: "heater_may_be_dry",
+ 29: "water_too_hot",
+ 30: "heater_too_hot",
+ 31: "sensor_a_fault",
+ 32: "sensor_b_fault",
+ 34: "pump_stuck",
+ 35: "hot_fault",
+ 36: "gfci_test_failed",
+ 37: "standby_mode",
+}
+FAULT_EVENT_TYPES = sorted(set(FAULT_MESSAGE_CODE_MAP.values()))
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: BalboaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the spa's events."""
+ async_add_entities([BalboaEventEntity(entry.runtime_data)])
+
+
+class BalboaEventEntity(BalboaEntity, EventEntity):
+ """Representation of a Balboa event entity."""
+
+ _attr_event_types = FAULT_EVENT_TYPES
+ _attr_translation_key = FAULT
+
+ def __init__(self, spa: SpaClient) -> None:
+ """Initialize a Balboa event entity."""
+ super().__init__(spa, FAULT)
+
+ @callback
+ def _async_handle_event(self) -> None:
+ """Handle the fault event."""
+ if not (fault := self._client.fault):
+ return
+ fault_date = fault.fault_datetime.isoformat()
+ if self.state_attributes.get(FAULT_DATE) != fault_date:
+ self._trigger_event(
+ FAULT_MESSAGE_CODE_MAP.get(fault.message_code, fault.message),
+ {FAULT_DATE: fault_date, "code": fault.message_code},
+ )
+ self.async_write_ha_state()
+
+ async def async_added_to_hass(self) -> None:
+ """Run when entity about to be added to hass."""
+ await super().async_added_to_hass()
+ self.async_on_remove(self._client.on(EVENT_UPDATE, self._async_handle_event))
+
+ async def request_fault_log(now: datetime | None = None) -> None:
+ """Request the most recent fault log."""
+ await self._client.request_fault_log()
+
+ await request_fault_log()
+ self.async_on_remove(
+ async_track_time_interval(
+ self.hass, request_fault_log, REQUEST_FAULT_LOG_INTERVAL
+ )
+ )
diff --git a/homeassistant/components/balboa/fan.py b/homeassistant/components/balboa/fan.py
index 3ecfec53a1e..b0d4379594b 100644
--- a/homeassistant/components/balboa/fan.py
+++ b/homeassistant/components/balboa/fan.py
@@ -10,7 +10,7 @@ from pybalboa.enums import OffOnState, UnknownState
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -23,7 +23,7 @@ from .entity import BalboaEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BalboaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the spa's pumps."""
spa = entry.runtime_data
diff --git a/homeassistant/components/balboa/light.py b/homeassistant/components/balboa/light.py
index 21e4dfc5e08..2f48747c084 100644
--- a/homeassistant/components/balboa/light.py
+++ b/homeassistant/components/balboa/light.py
@@ -9,7 +9,7 @@ from pybalboa.enums import OffOnState, UnknownState
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BalboaConfigEntry
from .entity import BalboaEntity
@@ -18,7 +18,7 @@ from .entity import BalboaEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BalboaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the spa's lights."""
spa = entry.runtime_data
diff --git a/homeassistant/components/balboa/manifest.json b/homeassistant/components/balboa/manifest.json
index 867e277358c..38d32adc4af 100644
--- a/homeassistant/components/balboa/manifest.json
+++ b/homeassistant/components/balboa/manifest.json
@@ -14,5 +14,5 @@
"documentation": "https://www.home-assistant.io/integrations/balboa",
"iot_class": "local_push",
"loggers": ["pybalboa"],
- "requirements": ["pybalboa==1.0.2"]
+ "requirements": ["pybalboa==1.1.3"]
}
diff --git a/homeassistant/components/balboa/select.py b/homeassistant/components/balboa/select.py
index e88e40ab063..ea82760744c 100644
--- a/homeassistant/components/balboa/select.py
+++ b/homeassistant/components/balboa/select.py
@@ -5,7 +5,7 @@ from pybalboa.enums import LowHighRange
from homeassistant.components.select import SelectEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BalboaConfigEntry
from .entity import BalboaEntity
@@ -14,7 +14,7 @@ from .entity import BalboaEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BalboaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the spa select entity."""
spa = entry.runtime_data
diff --git a/homeassistant/components/balboa/strings.json b/homeassistant/components/balboa/strings.json
index c00567a6052..784ce8533a8 100644
--- a/homeassistant/components/balboa/strings.json
+++ b/homeassistant/components/balboa/strings.json
@@ -57,6 +57,35 @@
}
}
},
+ "event": {
+ "fault": {
+ "name": "Fault",
+ "state_attributes": {
+ "event_type": {
+ "state": {
+ "sensor_out_of_sync": "Sensors are out of sync",
+ "low_flow": "The water flow is low",
+ "flow_failed": "The water flow has failed",
+ "settings_reset": "The settings have been reset",
+ "priming_mode": "Priming mode",
+ "clock_failed": "The clock has failed",
+ "memory_failure": "Program memory failure",
+ "service_sensor_sync": "Sensors are out of sync -- call for service",
+ "heater_dry": "The heater is dry",
+ "heater_may_be_dry": "The heater may be dry",
+ "water_too_hot": "The water is too hot",
+ "heater_too_hot": "The heater is too hot",
+ "sensor_a_fault": "Sensor A fault",
+ "sensor_b_fault": "Sensor B fault",
+ "pump_stuck": "A pump may be stuck on",
+ "hot_fault": "Hot fault",
+ "gfci_test_failed": "The GFCI test failed",
+ "standby_mode": "Standby mode (hold mode)"
+ }
+ }
+ }
+ }
+ },
"fan": {
"pump": {
"name": "Pump {index}"
@@ -78,6 +107,19 @@
"high": "High"
}
}
+ },
+ "switch": {
+ "filter_cycle_2_enabled": {
+ "name": "Filter cycle 2 enabled"
+ }
+ },
+ "time": {
+ "filter_cycle_start": {
+ "name": "Filter cycle {index} start"
+ },
+ "filter_cycle_end": {
+ "name": "Filter cycle {index} end"
+ }
}
}
}
diff --git a/homeassistant/components/balboa/switch.py b/homeassistant/components/balboa/switch.py
new file mode 100644
index 00000000000..c8c947f499d
--- /dev/null
+++ b/homeassistant/components/balboa/switch.py
@@ -0,0 +1,48 @@
+"""Support for Balboa switches."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from pybalboa import SpaClient
+
+from homeassistant.components.switch import SwitchEntity
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import BalboaConfigEntry
+from .entity import BalboaEntity
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: BalboaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the spa's switches."""
+ spa = entry.runtime_data
+ async_add_entities([BalboaSwitchEntity(spa)])
+
+
+class BalboaSwitchEntity(BalboaEntity, SwitchEntity):
+ """Representation of a Balboa switch entity."""
+
+ def __init__(self, spa: SpaClient) -> None:
+ """Initialize a Balboa switch entity."""
+ super().__init__(spa, "filter_cycle_2_enabled")
+ self._attr_entity_category = EntityCategory.CONFIG
+ self._attr_translation_key = "filter_cycle_2_enabled"
+
+ @property
+ def is_on(self) -> bool:
+ """Return True if entity is on."""
+ return self._client.filter_cycle_2_enabled
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the entity on."""
+ await self._client.configure_filter_cycle(2, enabled=True)
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the entity off."""
+ await self._client.configure_filter_cycle(2, enabled=False)
diff --git a/homeassistant/components/balboa/time.py b/homeassistant/components/balboa/time.py
new file mode 100644
index 00000000000..83467de8777
--- /dev/null
+++ b/homeassistant/components/balboa/time.py
@@ -0,0 +1,56 @@
+"""Support for Balboa times."""
+
+from __future__ import annotations
+
+from datetime import time
+import itertools
+from typing import Any
+
+from pybalboa import SpaClient
+
+from homeassistant.components.time import TimeEntity
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import BalboaConfigEntry
+from .entity import BalboaEntity
+
+FILTER_CYCLE = "filter_cycle_"
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: BalboaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the spa's times."""
+ spa = entry.runtime_data
+ async_add_entities(
+ BalboaTimeEntity(spa, index, period)
+ for index, period in itertools.product((1, 2), ("start", "end"))
+ )
+
+
+class BalboaTimeEntity(BalboaEntity, TimeEntity):
+ """Representation of a Balboa time entity."""
+
+ entity_category = EntityCategory.CONFIG
+
+ def __init__(self, spa: SpaClient, index: int, period: str) -> None:
+ """Initialize a Balboa time entity."""
+ super().__init__(spa, f"{FILTER_CYCLE}{index}_{period}")
+ self.index = index
+ self.period = period
+ self._attr_translation_key = f"{FILTER_CYCLE}{period}"
+ self._attr_translation_placeholders = {"index": str(index)}
+
+ @property
+ def native_value(self) -> time | None:
+ """Return the value reported by the time."""
+ return getattr(self._client, f"{FILTER_CYCLE}{self.index}_{self.period}")
+
+ async def async_set_value(self, value: time) -> None:
+ """Change the time."""
+ args: dict[str, Any] = {self.period: value}
+ await self._client.configure_filter_cycle(self.index, **args)
diff --git a/homeassistant/components/bang_olufsen/diagnostics.py b/homeassistant/components/bang_olufsen/diagnostics.py
index bf7b06e694a..3835de7c551 100644
--- a/homeassistant/components/bang_olufsen/diagnostics.py
+++ b/homeassistant/components/bang_olufsen/diagnostics.py
@@ -4,12 +4,13 @@ from __future__ import annotations
from typing import TYPE_CHECKING, Any
+from homeassistant.components.event import DOMAIN as EVENT_DOMAIN
from homeassistant.components.media_player import DOMAIN as MEDIA_PLAYER_DOMAIN
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from . import BangOlufsenConfigEntry
-from .const import DOMAIN
+from .const import DEVICE_BUTTONS, DOMAIN
async def async_get_config_entry_diagnostics(
@@ -25,8 +26,9 @@ async def async_get_config_entry_diagnostics(
if TYPE_CHECKING:
assert config_entry.unique_id
- # Add media_player entity's state
entity_registry = er.async_get(hass)
+
+ # Add media_player entity's state
if entity_id := entity_registry.async_get_entity_id(
MEDIA_PLAYER_DOMAIN, DOMAIN, config_entry.unique_id
):
@@ -37,4 +39,16 @@ async def async_get_config_entry_diagnostics(
state_dict.pop("context")
data["media_player"] = state_dict
+ # Add button Event entity states (if enabled)
+ for device_button in DEVICE_BUTTONS:
+ if entity_id := entity_registry.async_get_entity_id(
+ EVENT_DOMAIN, DOMAIN, f"{config_entry.unique_id}_{device_button}"
+ ):
+ if state := hass.states.get(entity_id):
+ state_dict = dict(state.as_dict())
+
+ # Remove context as it is not relevant
+ state_dict.pop("context")
+ data[f"{device_button}_event"] = state_dict
+
return data
diff --git a/homeassistant/components/bang_olufsen/event.py b/homeassistant/components/bang_olufsen/event.py
index 99e5c8bb6fd..91e04b92330 100644
--- a/homeassistant/components/bang_olufsen/event.py
+++ b/homeassistant/components/bang_olufsen/event.py
@@ -6,7 +6,7 @@ from homeassistant.components.event import EventDeviceClass, EventEntity
from homeassistant.const import CONF_MODEL
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BangOlufsenConfigEntry
from .const import (
@@ -25,7 +25,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BangOlufsenConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensor entities from config entry."""
diff --git a/homeassistant/components/bang_olufsen/media_player.py b/homeassistant/components/bang_olufsen/media_player.py
index 282ecdd2ae5..efb6843356b 100644
--- a/homeassistant/components/bang_olufsen/media_player.py
+++ b/homeassistant/components/bang_olufsen/media_player.py
@@ -64,7 +64,7 @@ from homeassistant.helpers import (
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.util.dt import utcnow
@@ -118,7 +118,7 @@ BANG_OLUFSEN_FEATURES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BangOlufsenConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Media Player entity from config entry."""
# Add MediaPlayer entity
diff --git a/homeassistant/components/bang_olufsen/strings.json b/homeassistant/components/bang_olufsen/strings.json
index 57ab828f9fb..422dc4be567 100644
--- a/homeassistant/components/bang_olufsen/strings.json
+++ b/homeassistant/components/bang_olufsen/strings.json
@@ -29,7 +29,7 @@
"description": "Manually configure your Bang & Olufsen device."
},
"zeroconf_confirm": {
- "title": "Setup Bang & Olufsen device",
+ "title": "Set up Bang & Olufsen device",
"description": "Confirm the configuration of the {model}-{serial_number} @ {host}."
}
}
@@ -197,11 +197,11 @@
"services": {
"beolink_allstandby": {
"name": "Beolink all standby",
- "description": "Set all Connected Beolink devices to standby."
+ "description": "Sets all connected Beolink devices to standby."
},
"beolink_expand": {
"name": "Beolink expand",
- "description": "Expand current Beolink experience.",
+ "description": "Adds devices to the current Beolink experience.",
"fields": {
"all_discovered": {
"name": "All discovered",
@@ -221,7 +221,7 @@
},
"beolink_join": {
"name": "Beolink join",
- "description": "Join a Beolink experience.",
+ "description": "Joins a Beolink experience.",
"fields": {
"beolink_jid": {
"name": "Beolink JID",
@@ -241,11 +241,11 @@
},
"beolink_leave": {
"name": "Beolink leave",
- "description": "Leave a Beolink experience."
+ "description": "Leaves a Beolink experience."
},
"beolink_unexpand": {
"name": "Beolink unexpand",
- "description": "Unexpand from current Beolink experience.",
+ "description": "Removes devices from the current Beolink experience.",
"fields": {
"beolink_jids": {
"name": "Beolink JIDs",
@@ -274,7 +274,7 @@
"message": "An error occurred while attempting to play {media_type}: {error_message}."
},
"invalid_grouping_entity": {
- "message": "Entity with id: {entity_id} can't be added to the Beolink session. Is the entity a Bang & Olufsen media_player?"
+ "message": "Entity with ID {entity_id} can't be added to the Beolink session. Is the entity a Bang & Olufsen media_player?"
},
"invalid_sound_mode": {
"message": "{invalid_sound_mode} is an invalid sound mode. Valid values are: {valid_sound_modes}."
diff --git a/homeassistant/components/bayesian/strings.json b/homeassistant/components/bayesian/strings.json
index 9ebccedc88d..00de79a2229 100644
--- a/homeassistant/components/bayesian/strings.json
+++ b/homeassistant/components/bayesian/strings.json
@@ -5,14 +5,14 @@
"title": "Manual YAML fix required for Bayesian"
},
"no_prob_given_false": {
- "description": "In the Bayesian integration `prob_given_false` is now a required configuration variable as there was no mathematical rationale for the previous default value. Please add this to your `configuration.yml` for `bayesian/{entity}`. These observations will be ignored until you do.",
+ "description": "In the Bayesian integration `prob_given_false` is now a required configuration variable as there was no mathematical rationale for the previous default value. Please add this to your `configuration.yaml` for `bayesian/{entity}`. These observations will be ignored until you do.",
"title": "Manual YAML addition required for Bayesian"
}
},
"services": {
"reload": {
"name": "[%key:common::action::reload%]",
- "description": "Reloads bayesian sensors from the YAML-configuration."
+ "description": "Reloads Bayesian sensors from the YAML-configuration."
}
}
}
diff --git a/homeassistant/components/blebox/binary_sensor.py b/homeassistant/components/blebox/binary_sensor.py
index 2aa86059ee2..b9032e6e705 100644
--- a/homeassistant/components/blebox/binary_sensor.py
+++ b/homeassistant/components/blebox/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BleBoxConfigEntry
from .entity import BleBoxEntity
@@ -24,7 +24,7 @@ BINARY_SENSOR_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BleBoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a BleBox entry."""
entities = [
diff --git a/homeassistant/components/blebox/button.py b/homeassistant/components/blebox/button.py
index 90356c8ae14..15867f84029 100644
--- a/homeassistant/components/blebox/button.py
+++ b/homeassistant/components/blebox/button.py
@@ -6,7 +6,7 @@ import blebox_uniapi.button
from homeassistant.components.button import ButtonEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BleBoxConfigEntry
from .entity import BleBoxEntity
@@ -15,7 +15,7 @@ from .entity import BleBoxEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BleBoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a BleBox button entry."""
entities = [
diff --git a/homeassistant/components/blebox/climate.py b/homeassistant/components/blebox/climate.py
index 2c528d50e3e..dbf4a326990 100644
--- a/homeassistant/components/blebox/climate.py
+++ b/homeassistant/components/blebox/climate.py
@@ -13,7 +13,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BleBoxConfigEntry
from .entity import BleBoxEntity
@@ -38,7 +38,7 @@ BLEBOX_TO_HVACACTION = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BleBoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a BleBox climate entity."""
entities = [
diff --git a/homeassistant/components/blebox/cover.py b/homeassistant/components/blebox/cover.py
index 4f2a7eeef11..c52c551bbac 100644
--- a/homeassistant/components/blebox/cover.py
+++ b/homeassistant/components/blebox/cover.py
@@ -16,7 +16,7 @@ from homeassistant.components.cover import (
CoverState,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BleBoxConfigEntry
from .entity import BleBoxEntity
@@ -45,7 +45,7 @@ BLEBOX_TO_HASS_COVER_STATES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BleBoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a BleBox entry."""
entities = [
diff --git a/homeassistant/components/blebox/light.py b/homeassistant/components/blebox/light.py
index c3c9de8be51..86ec8993779 100644
--- a/homeassistant/components/blebox/light.py
+++ b/homeassistant/components/blebox/light.py
@@ -21,7 +21,7 @@ from homeassistant.components.light import (
LightEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from . import BleBoxConfigEntry
@@ -35,7 +35,7 @@ SCAN_INTERVAL = timedelta(seconds=5)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BleBoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a BleBox entry."""
entities = [
diff --git a/homeassistant/components/blebox/sensor.py b/homeassistant/components/blebox/sensor.py
index c0abff31257..5120a7a3c98 100644
--- a/homeassistant/components/blebox/sensor.py
+++ b/homeassistant/components/blebox/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BleBoxConfigEntry
from .entity import BleBoxEntity
@@ -116,7 +116,7 @@ SENSOR_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BleBoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a BleBox entry."""
entities = [
diff --git a/homeassistant/components/blebox/switch.py b/homeassistant/components/blebox/switch.py
index c6f439e27c5..1598d4db6fa 100644
--- a/homeassistant/components/blebox/switch.py
+++ b/homeassistant/components/blebox/switch.py
@@ -7,7 +7,7 @@ import blebox_uniapi.switch
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BleBoxConfigEntry
from .entity import BleBoxEntity
@@ -18,7 +18,7 @@ SCAN_INTERVAL = timedelta(seconds=5)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BleBoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a BleBox switch entity."""
entities = [
diff --git a/homeassistant/components/blink/alarm_control_panel.py b/homeassistant/components/blink/alarm_control_panel.py
index bfb8aa9a3a0..17fd003742f 100644
--- a/homeassistant/components/blink/alarm_control_panel.py
+++ b/homeassistant/components/blink/alarm_control_panel.py
@@ -15,7 +15,7 @@ from homeassistant.const import ATTR_ATTRIBUTION
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DEFAULT_ATTRIBUTION, DEFAULT_BRAND, DOMAIN
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BlinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Blink Alarm Control Panels."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/blink/binary_sensor.py b/homeassistant/components/blink/binary_sensor.py
index c11d4cfea23..3d5430c8d1f 100644
--- a/homeassistant/components/blink/binary_sensor.py
+++ b/homeassistant/components/blink/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -48,7 +48,7 @@ BINARY_SENSORS_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BlinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the blink binary sensors."""
diff --git a/homeassistant/components/blink/camera.py b/homeassistant/components/blink/camera.py
index e35dd20eea7..04bd125d249 100644
--- a/homeassistant/components/blink/camera.py
+++ b/homeassistant/components/blink/camera.py
@@ -15,7 +15,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -38,7 +38,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BlinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Blink Camera."""
diff --git a/homeassistant/components/blink/sensor.py b/homeassistant/components/blink/sensor.py
index e0b5989cc80..1df708c3a10 100644
--- a/homeassistant/components/blink/sensor.py
+++ b/homeassistant/components/blink/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DEFAULT_BRAND, DOMAIN, TYPE_TEMPERATURE, TYPE_WIFI_STRENGTH
@@ -47,7 +47,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BlinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize a Blink sensor."""
diff --git a/homeassistant/components/blink/switch.py b/homeassistant/components/blink/switch.py
index 8eabd5c0e59..4f490e28310 100644
--- a/homeassistant/components/blink/switch.py
+++ b/homeassistant/components/blink/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DEFAULT_BRAND, DOMAIN, TYPE_CAMERA_ARMED
@@ -30,7 +30,7 @@ SWITCH_TYPES: tuple[SwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BlinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Blink switches."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/blue_current/sensor.py b/homeassistant/components/blue_current/sensor.py
index be39e9571ec..9ea444f9ec2 100644
--- a/homeassistant/components/blue_current/sensor.py
+++ b/homeassistant/components/blue_current/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BlueCurrentConfigEntry, Connector
from .const import DOMAIN
@@ -212,7 +212,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: BlueCurrentConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Blue Current sensors."""
connector = entry.runtime_data
diff --git a/homeassistant/components/blue_current/strings.json b/homeassistant/components/blue_current/strings.json
index 0154c794c33..2e48d768a74 100644
--- a/homeassistant/components/blue_current/strings.json
+++ b/homeassistant/components/blue_current/strings.json
@@ -28,7 +28,7 @@
"name": "Activity",
"state": {
"available": "Available",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"unavailable": "Unavailable",
"error": "Error",
"offline": "Offline"
diff --git a/homeassistant/components/bluemaestro/sensor.py b/homeassistant/components/bluemaestro/sensor.py
index 57702d4ff31..1163f8a1ff6 100644
--- a/homeassistant/components/bluemaestro/sensor.py
+++ b/homeassistant/components/bluemaestro/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from . import BlueMaestroConfigEntry
@@ -116,7 +116,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: BlueMaestroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the BlueMaestro BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/blueprint/importer.py b/homeassistant/components/blueprint/importer.py
index 8582761bafb..83afa511b68 100644
--- a/homeassistant/components/blueprint/importer.py
+++ b/homeassistant/components/blueprint/importer.py
@@ -6,6 +6,7 @@ from contextlib import suppress
from dataclasses import dataclass
import html
import re
+from typing import TYPE_CHECKING
import voluptuous as vol
import yarl
@@ -195,8 +196,8 @@ async def fetch_blueprint_from_github_gist_url(
)
gist = await resp.json()
- blueprint = None
- filename = None
+ blueprint: Blueprint | None = None
+ filename: str | None = None
content: str
for filename, info in gist["files"].items():
@@ -218,6 +219,8 @@ async def fetch_blueprint_from_github_gist_url(
"No valid blueprint found in the gist. The blueprint file needs to end with"
" '.yaml'"
)
+ if TYPE_CHECKING:
+ assert isinstance(filename, str)
return ImportedBlueprint(
f"{gist['owner']['login']}/{filename[:-5]}", content, blueprint
diff --git a/homeassistant/components/bluesound/config_flow.py b/homeassistant/components/bluesound/config_flow.py
index 2f002b70e1d..cfb6646d829 100644
--- a/homeassistant/components/bluesound/config_flow.py
+++ b/homeassistant/components/bluesound/config_flow.py
@@ -75,6 +75,9 @@ class BluesoundConfigFlow(ConfigFlow, domain=DOMAIN):
self, discovery_info: ZeroconfServiceInfo
) -> ConfigFlowResult:
"""Handle a flow initialized by zeroconf discovery."""
+ # the player can have an ipv6 address, but the api is only available on ipv4
+ if discovery_info.ip_address.version != 4:
+ return self.async_abort(reason="no_ipv4_address")
if discovery_info.port is not None:
self._port = discovery_info.port
diff --git a/homeassistant/components/bluesound/media_player.py b/homeassistant/components/bluesound/media_player.py
index 6bb3c101cd1..135d1b5d27e 100644
--- a/homeassistant/components/bluesound/media_player.py
+++ b/homeassistant/components/bluesound/media_player.py
@@ -32,7 +32,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -61,7 +61,7 @@ POLL_TIMEOUT = 120
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BluesoundConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Bluesound entry."""
bluesound_player = BluesoundPlayer(
diff --git a/homeassistant/components/bluesound/strings.json b/homeassistant/components/bluesound/strings.json
index b50c01a11bf..1170e0b92e0 100644
--- a/homeassistant/components/bluesound/strings.json
+++ b/homeassistant/components/bluesound/strings.json
@@ -19,7 +19,8 @@
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_service%]",
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
- "already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]"
+ "already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]",
+ "no_ipv4_address": "No IPv4 address found."
},
"error": {
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]"
diff --git a/homeassistant/components/bluetooth/__init__.py b/homeassistant/components/bluetooth/__init__.py
index c46ef22803e..7abc929fde5 100644
--- a/homeassistant/components/bluetooth/__init__.py
+++ b/homeassistant/components/bluetooth/__init__.py
@@ -311,11 +311,24 @@ async def async_update_device(
update the device with the new location so they can
figure out where the adapter is.
"""
+ address = details[ADAPTER_ADDRESS]
+ connections = {(dr.CONNECTION_BLUETOOTH, address)}
device_registry = dr.async_get(hass)
+ # We only have one device for the config entry
+ # so if the address has been corrected, make
+ # sure the device entry reflects the correct
+ # address
+ for device in dr.async_entries_for_config_entry(device_registry, entry.entry_id):
+ for conn_type, conn_value in device.connections:
+ if conn_type == dr.CONNECTION_BLUETOOTH and conn_value != address:
+ device_registry.async_update_device(
+ device.id, new_connections=connections
+ )
+ break
device_entry = device_registry.async_get_or_create(
config_entry_id=entry.entry_id,
- name=adapter_human_name(adapter, details[ADAPTER_ADDRESS]),
- connections={(dr.CONNECTION_BLUETOOTH, details[ADAPTER_ADDRESS])},
+ name=adapter_human_name(adapter, address),
+ connections=connections,
manufacturer=details[ADAPTER_MANUFACTURER],
model=adapter_model(details),
sw_version=details.get(ADAPTER_SW_VERSION),
@@ -342,9 +355,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
)
)
)
+ return True
address = entry.unique_id
assert address is not None
- assert source_entry is not None
source_domain = entry.data[CONF_SOURCE_DOMAIN]
if mac_manufacturer := await get_manufacturer_from_mac(address):
manufacturer = f"{mac_manufacturer} ({source_domain})"
diff --git a/homeassistant/components/bluetooth/config_flow.py b/homeassistant/components/bluetooth/config_flow.py
index e76277306f5..328707bd722 100644
--- a/homeassistant/components/bluetooth/config_flow.py
+++ b/homeassistant/components/bluetooth/config_flow.py
@@ -186,16 +186,28 @@ class BluetoothConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a flow initialized by an external scanner."""
source = user_input[CONF_SOURCE]
await self.async_set_unique_id(source)
+ source_config_entry_id = user_input[CONF_SOURCE_CONFIG_ENTRY_ID]
data = {
CONF_SOURCE: source,
CONF_SOURCE_MODEL: user_input[CONF_SOURCE_MODEL],
CONF_SOURCE_DOMAIN: user_input[CONF_SOURCE_DOMAIN],
- CONF_SOURCE_CONFIG_ENTRY_ID: user_input[CONF_SOURCE_CONFIG_ENTRY_ID],
+ CONF_SOURCE_CONFIG_ENTRY_ID: source_config_entry_id,
CONF_SOURCE_DEVICE_ID: user_input[CONF_SOURCE_DEVICE_ID],
}
self._abort_if_unique_id_configured(updates=data)
- manager = get_manager()
- scanner = manager.async_scanner_by_source(source)
+ for entry in self._async_current_entries(include_ignore=False):
+ # If the mac address needs to be corrected, migrate
+ # the config entry to the new mac address
+ if (
+ entry.data.get(CONF_SOURCE_CONFIG_ENTRY_ID) == source_config_entry_id
+ and entry.unique_id != source
+ ):
+ self.hass.config_entries.async_update_entry(
+ entry, unique_id=source, data={**entry.data, **data}
+ )
+ self.hass.config_entries.async_schedule_reload(entry.entry_id)
+ return self.async_abort(reason="already_configured")
+ scanner = get_manager().async_scanner_by_source(source)
assert scanner is not None
return self.async_create_entry(title=scanner.name, data=data)
diff --git a/homeassistant/components/bluetooth/manifest.json b/homeassistant/components/bluetooth/manifest.json
index 5d2b8ab6285..27fed6ad647 100644
--- a/homeassistant/components/bluetooth/manifest.json
+++ b/homeassistant/components/bluetooth/manifest.json
@@ -16,11 +16,11 @@
"quality_scale": "internal",
"requirements": [
"bleak==0.22.3",
- "bleak-retry-connector==3.8.1",
+ "bleak-retry-connector==3.9.0",
"bluetooth-adapters==0.21.4",
- "bluetooth-auto-recovery==1.4.2",
- "bluetooth-data-tools==1.23.4",
- "dbus-fast==2.33.0",
- "habluetooth==3.21.1"
+ "bluetooth-auto-recovery==1.4.5",
+ "bluetooth-data-tools==1.26.1",
+ "dbus-fast==2.41.1",
+ "habluetooth==3.37.0"
]
}
diff --git a/homeassistant/components/bluetooth/match.py b/homeassistant/components/bluetooth/match.py
index 6307d3ca93b..c37fa4615f6 100644
--- a/homeassistant/components/bluetooth/match.py
+++ b/homeassistant/components/bluetooth/match.py
@@ -411,7 +411,7 @@ def ble_device_matches(
) and service_data_uuid not in service_info.service_data:
return False
- if manufacturer_id := matcher.get(MANUFACTURER_ID):
+ if (manufacturer_id := matcher.get(MANUFACTURER_ID)) is not None:
if manufacturer_id not in service_info.manufacturer_data:
return False
diff --git a/homeassistant/components/bmw_connected_drive/binary_sensor.py b/homeassistant/components/bmw_connected_drive/binary_sensor.py
index 5a58c707d6a..01cdbbdc94d 100644
--- a/homeassistant/components/bmw_connected_drive/binary_sensor.py
+++ b/homeassistant/components/bmw_connected_drive/binary_sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_system import UnitSystem
from . import BMWConfigEntry
@@ -200,7 +200,7 @@ SENSOR_TYPES: tuple[BMWBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the BMW binary sensors from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bmw_connected_drive/button.py b/homeassistant/components/bmw_connected_drive/button.py
index a7c31d0ef79..f8980201f3f 100644
--- a/homeassistant/components/bmw_connected_drive/button.py
+++ b/homeassistant/components/bmw_connected_drive/button.py
@@ -14,7 +14,7 @@ from bimmer_connected.vehicle.remote_services import RemoteServiceStatus
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as BMW_DOMAIN, BMWConfigEntry
from .entity import BMWBaseEntity
@@ -69,7 +69,7 @@ BUTTON_TYPES: tuple[BMWButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the BMW buttons from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bmw_connected_drive/device_tracker.py b/homeassistant/components/bmw_connected_drive/device_tracker.py
index 74df8693f7a..23273cc8ba9 100644
--- a/homeassistant/components/bmw_connected_drive/device_tracker.py
+++ b/homeassistant/components/bmw_connected_drive/device_tracker.py
@@ -9,7 +9,7 @@ from bimmer_connected.vehicle import MyBMWVehicle
from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BMWConfigEntry
from .const import ATTR_DIRECTION
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the MyBMW tracker from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bmw_connected_drive/lock.py b/homeassistant/components/bmw_connected_drive/lock.py
index 4bec12e796b..9d8965d6ebf 100644
--- a/homeassistant/components/bmw_connected_drive/lock.py
+++ b/homeassistant/components/bmw_connected_drive/lock.py
@@ -12,7 +12,7 @@ from bimmer_connected.vehicle.doors_windows import LockState
from homeassistant.components.lock import LockEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as BMW_DOMAIN, BMWConfigEntry
from .coordinator import BMWDataUpdateCoordinator
@@ -28,7 +28,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the MyBMW lock from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bmw_connected_drive/number.py b/homeassistant/components/bmw_connected_drive/number.py
index c6a328ecc20..8361306ba9d 100644
--- a/homeassistant/components/bmw_connected_drive/number.py
+++ b/homeassistant/components/bmw_connected_drive/number.py
@@ -16,7 +16,7 @@ from homeassistant.components.number import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as BMW_DOMAIN, BMWConfigEntry
from .coordinator import BMWDataUpdateCoordinator
@@ -58,7 +58,7 @@ NUMBER_TYPES: list[BMWNumberEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the MyBMW number from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bmw_connected_drive/select.py b/homeassistant/components/bmw_connected_drive/select.py
index 385b45fd9fa..f144d3a71df 100644
--- a/homeassistant/components/bmw_connected_drive/select.py
+++ b/homeassistant/components/bmw_connected_drive/select.py
@@ -13,7 +13,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.const import UnitOfElectricCurrent
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as BMW_DOMAIN, BMWConfigEntry
from .coordinator import BMWDataUpdateCoordinator
@@ -65,7 +65,7 @@ SELECT_TYPES: tuple[BMWSelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the MyBMW lock from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bmw_connected_drive/sensor.py b/homeassistant/components/bmw_connected_drive/sensor.py
index b7be367d57d..114412ef9f2 100644
--- a/homeassistant/components/bmw_connected_drive/sensor.py
+++ b/homeassistant/components/bmw_connected_drive/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import BMWConfigEntry
@@ -190,7 +190,7 @@ SENSOR_TYPES: list[BMWSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the MyBMW sensors from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bmw_connected_drive/strings.json b/homeassistant/components/bmw_connected_drive/strings.json
index edb0d5cfb12..4b16b719d8d 100644
--- a/homeassistant/components/bmw_connected_drive/strings.json
+++ b/homeassistant/components/bmw_connected_drive/strings.json
@@ -138,7 +138,7 @@
"name": "Charging status",
"state": {
"default": "Default",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"error": "Error",
"complete": "Complete",
"fully_charged": "Fully charged",
diff --git a/homeassistant/components/bmw_connected_drive/switch.py b/homeassistant/components/bmw_connected_drive/switch.py
index 600ad41165a..f46969f3e9b 100644
--- a/homeassistant/components/bmw_connected_drive/switch.py
+++ b/homeassistant/components/bmw_connected_drive/switch.py
@@ -12,7 +12,7 @@ from bimmer_connected.vehicle.fuel_and_battery import ChargingState
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as BMW_DOMAIN, BMWConfigEntry
from .coordinator import BMWDataUpdateCoordinator
@@ -66,7 +66,7 @@ NUMBER_TYPES: list[BMWSwitchEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BMWConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the MyBMW switch from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bond/button.py b/homeassistant/components/bond/button.py
index 42915c7dc0b..9cea0251b41 100644
--- a/homeassistant/components/bond/button.py
+++ b/homeassistant/components/bond/button.py
@@ -8,7 +8,7 @@ from bond_async import Action
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BondConfigEntry
from .entity import BondEntity
@@ -91,6 +91,13 @@ BUTTONS: tuple[BondButtonEntityDescription, ...] = (
mutually_exclusive=Action.SET_BRIGHTNESS,
argument=None,
),
+ BondButtonEntityDescription(
+ key=Action.TOGGLE_LIGHT_TEMP,
+ name="Toggle Light Temperature",
+ translation_key="toggle_light_temp",
+ mutually_exclusive=None, # No mutually exclusive action
+ argument=None,
+ ),
BondButtonEntityDescription(
key=Action.START_UP_LIGHT_DIMMER,
name="Start Up Light Dimmer",
@@ -257,7 +264,7 @@ BUTTONS: tuple[BondButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: BondConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Bond button devices."""
data = entry.runtime_data
diff --git a/homeassistant/components/bond/cover.py b/homeassistant/components/bond/cover.py
index 66344a1913d..d2a78819fae 100644
--- a/homeassistant/components/bond/cover.py
+++ b/homeassistant/components/bond/cover.py
@@ -13,7 +13,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BondConfigEntry
from .entity import BondEntity
@@ -34,7 +34,7 @@ def _hass_to_bond_position(hass_position: int) -> int:
async def async_setup_entry(
hass: HomeAssistant,
entry: BondConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Bond cover devices."""
data = entry.runtime_data
diff --git a/homeassistant/components/bond/fan.py b/homeassistant/components/bond/fan.py
index 76a0daa46f9..c228c7355dd 100644
--- a/homeassistant/components/bond/fan.py
+++ b/homeassistant/components/bond/fan.py
@@ -19,7 +19,7 @@ from homeassistant.components.fan import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -40,7 +40,7 @@ PRESET_MODE_BREEZE = "Breeze"
async def async_setup_entry(
hass: HomeAssistant,
entry: BondConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Bond fan devices."""
data = entry.runtime_data
diff --git a/homeassistant/components/bond/light.py b/homeassistant/components/bond/light.py
index c3cf23e4fad..9c51165ebdb 100644
--- a/homeassistant/components/bond/light.py
+++ b/homeassistant/components/bond/light.py
@@ -14,7 +14,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BondConfigEntry
from .const import (
@@ -42,7 +42,7 @@ ENTITY_SERVICES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: BondConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Bond light devices."""
data = entry.runtime_data
diff --git a/homeassistant/components/bond/strings.json b/homeassistant/components/bond/strings.json
index 8986905c6ee..d65966d7701 100644
--- a/homeassistant/components/bond/strings.json
+++ b/homeassistant/components/bond/strings.json
@@ -31,7 +31,7 @@
"services": {
"set_fan_speed_tracked_state": {
"name": "Set fan speed tracked state",
- "description": "Sets the tracked fan speed for a bond fan.",
+ "description": "Sets the tracked fan speed for a Bond fan.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -45,7 +45,7 @@
},
"set_switch_power_tracked_state": {
"name": "Set switch power tracked state",
- "description": "Sets the tracked power state of a bond switch.",
+ "description": "Sets the tracked power state of a Bond switch.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -59,7 +59,7 @@
},
"set_light_power_tracked_state": {
"name": "Set light power tracked state",
- "description": "Sets the tracked power state of a bond light.",
+ "description": "Sets the tracked power state of a Bond light.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -73,7 +73,7 @@
},
"set_light_brightness_tracked_state": {
"name": "Set light brightness tracked state",
- "description": "Sets the tracked brightness state of a bond light.",
+ "description": "Sets the tracked brightness state of a Bond light.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -87,15 +87,15 @@
},
"start_increasing_brightness": {
"name": "Start increasing brightness",
- "description": "Start increasing the brightness of the light. (deprecated)."
+ "description": "Starts increasing the brightness of the light (deprecated)."
},
"start_decreasing_brightness": {
"name": "Start decreasing brightness",
- "description": "Start decreasing the brightness of the light. (deprecated)."
+ "description": "Starts decreasing the brightness of the light (deprecated)."
},
"stop": {
"name": "[%key:common::action::stop%]",
- "description": "Stop any in-progress action and empty the queue. (deprecated)."
+ "description": "Stops any in-progress action and empty the queue (deprecated)."
}
}
}
diff --git a/homeassistant/components/bond/switch.py b/homeassistant/components/bond/switch.py
index ace6d307e6d..fa2ccd2ca93 100644
--- a/homeassistant/components/bond/switch.py
+++ b/homeassistant/components/bond/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BondConfigEntry
from .const import ATTR_POWER_STATE, SERVICE_SET_POWER_TRACKED_STATE
@@ -22,7 +22,7 @@ from .entity import BondEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: BondConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Bond generic devices."""
data = entry.runtime_data
diff --git a/homeassistant/components/bosch_shc/binary_sensor.py b/homeassistant/components/bosch_shc/binary_sensor.py
index dd0f31ea6f9..30d823fd608 100644
--- a/homeassistant/components/bosch_shc/binary_sensor.py
+++ b/homeassistant/components/bosch_shc/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BoschConfigEntry
from .entity import SHCEntity
@@ -19,7 +19,7 @@ from .entity import SHCEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BoschConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SHC binary sensor platform."""
session = config_entry.runtime_data
diff --git a/homeassistant/components/bosch_shc/cover.py b/homeassistant/components/bosch_shc/cover.py
index 55d6bfc35de..766dcf37ce9 100644
--- a/homeassistant/components/bosch_shc/cover.py
+++ b/homeassistant/components/bosch_shc/cover.py
@@ -11,7 +11,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BoschConfigEntry
from .entity import SHCEntity
@@ -20,7 +20,7 @@ from .entity import SHCEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BoschConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SHC cover platform."""
session = config_entry.runtime_data
diff --git a/homeassistant/components/bosch_shc/sensor.py b/homeassistant/components/bosch_shc/sensor.py
index 6408e21654e..885908804c0 100644
--- a/homeassistant/components/bosch_shc/sensor.py
+++ b/homeassistant/components/bosch_shc/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import BoschConfigEntry
@@ -126,7 +126,7 @@ SENSOR_DESCRIPTIONS: dict[str, SHCSensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BoschConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SHC sensor platform."""
session = config_entry.runtime_data
diff --git a/homeassistant/components/bosch_shc/switch.py b/homeassistant/components/bosch_shc/switch.py
index 76b1da3e534..bf1d5d39ee5 100644
--- a/homeassistant/components/bosch_shc/switch.py
+++ b/homeassistant/components/bosch_shc/switch.py
@@ -21,7 +21,7 @@ from homeassistant.components.switch import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import BoschConfigEntry
@@ -79,7 +79,7 @@ SWITCH_TYPES: dict[str, SHCSwitchEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BoschConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SHC switch platform."""
session = config_entry.runtime_data
diff --git a/homeassistant/components/braviatv/button.py b/homeassistant/components/braviatv/button.py
index 626e5a225b7..20250949bcb 100644
--- a/homeassistant/components/braviatv/button.py
+++ b/homeassistant/components/braviatv/button.py
@@ -12,7 +12,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import BraviaTVConfigEntry, BraviaTVCoordinator
from .entity import BraviaTVEntity
@@ -44,7 +44,7 @@ BUTTONS: tuple[BraviaTVButtonDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BraviaTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Bravia TV Button entities."""
diff --git a/homeassistant/components/braviatv/coordinator.py b/homeassistant/components/braviatv/coordinator.py
index 1cc306bd5cf..039726de94d 100644
--- a/homeassistant/components/braviatv/coordinator.py
+++ b/homeassistant/components/braviatv/coordinator.py
@@ -39,7 +39,7 @@ _LOGGER = logging.getLogger(__name__)
SCAN_INTERVAL: Final = timedelta(seconds=10)
-type BraviaTVConfigEntry = ConfigEntry["BraviaTVCoordinator"]
+type BraviaTVConfigEntry = ConfigEntry[BraviaTVCoordinator]
def catch_braviatv_errors[_BraviaTVCoordinatorT: BraviaTVCoordinator, **_P](
diff --git a/homeassistant/components/braviatv/media_player.py b/homeassistant/components/braviatv/media_player.py
index ca48c6ee639..fe9c386b060 100644
--- a/homeassistant/components/braviatv/media_player.py
+++ b/homeassistant/components/braviatv/media_player.py
@@ -16,7 +16,7 @@ from homeassistant.components.media_player import (
MediaType,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SourceType
from .coordinator import BraviaTVConfigEntry
@@ -26,7 +26,7 @@ from .entity import BraviaTVEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BraviaTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Bravia TV Media Player from a config_entry."""
diff --git a/homeassistant/components/braviatv/remote.py b/homeassistant/components/braviatv/remote.py
index 9f4a573827b..0611e367445 100644
--- a/homeassistant/components/braviatv/remote.py
+++ b/homeassistant/components/braviatv/remote.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.remote import ATTR_NUM_REPEATS, RemoteEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import BraviaTVConfigEntry
from .entity import BraviaTVEntity
@@ -16,7 +16,7 @@ from .entity import BraviaTVEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BraviaTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Bravia TV Remote from a config entry."""
diff --git a/homeassistant/components/bring/diagnostics.py b/homeassistant/components/bring/diagnostics.py
index 6c2f779ef05..e5cafd30ab5 100644
--- a/homeassistant/components/bring/diagnostics.py
+++ b/homeassistant/components/bring/diagnostics.py
@@ -4,10 +4,14 @@ from __future__ import annotations
from typing import Any
+from homeassistant.components.diagnostics import async_redact_data
+from homeassistant.const import CONF_EMAIL, CONF_NAME
from homeassistant.core import HomeAssistant
from .coordinator import BringConfigEntry
+TO_REDACT = {CONF_NAME, CONF_EMAIL}
+
async def async_get_config_entry_diagnostics(
hass: HomeAssistant, config_entry: BringConfigEntry
@@ -15,7 +19,10 @@ async def async_get_config_entry_diagnostics(
"""Return diagnostics for a config entry."""
return {
- "data": {k: v.to_dict() for k, v in config_entry.runtime_data.data.items()},
+ "data": {
+ k: async_redact_data(v.to_dict(), TO_REDACT)
+ for k, v in config_entry.runtime_data.data.items()
+ },
"lists": [lst.to_dict() for lst in config_entry.runtime_data.lists],
"user_settings": config_entry.runtime_data.user_settings.to_dict(),
}
diff --git a/homeassistant/components/bring/event.py b/homeassistant/components/bring/event.py
index 699dba9015a..403856405ce 100644
--- a/homeassistant/components/bring/event.py
+++ b/homeassistant/components/bring/event.py
@@ -9,7 +9,7 @@ from bring_api import ActivityType, BringList
from homeassistant.components.event import EventEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BringConfigEntry
from .coordinator import BringDataUpdateCoordinator
@@ -21,7 +21,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BringConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the event platform."""
coordinator = config_entry.runtime_data
@@ -77,9 +77,12 @@ class BringEventEntity(BringBaseEntity, EventEntity):
attributes = asdict(activity.content)
attributes["last_activity_by"] = next(
- x.name
- for x in bring_list.users.users
- if x.publicUuid == activity.content.publicUserUuid
+ (
+ x.name
+ for x in bring_list.users.users
+ if x.publicUuid == activity.content.publicUserUuid
+ ),
+ None,
)
self._trigger_event(
diff --git a/homeassistant/components/bring/manifest.json b/homeassistant/components/bring/manifest.json
index 16767b7b0d6..b2d42835cce 100644
--- a/homeassistant/components/bring/manifest.json
+++ b/homeassistant/components/bring/manifest.json
@@ -7,5 +7,6 @@
"integration_type": "service",
"iot_class": "cloud_polling",
"loggers": ["bring_api"],
- "requirements": ["bring-api==1.0.1"]
+ "quality_scale": "platinum",
+ "requirements": ["bring-api==1.1.0"]
}
diff --git a/homeassistant/components/bring/quality_scale.yaml b/homeassistant/components/bring/quality_scale.yaml
index 58e67ab0e11..2d7d67be12e 100644
--- a/homeassistant/components/bring/quality_scale.yaml
+++ b/homeassistant/components/bring/quality_scale.yaml
@@ -10,9 +10,9 @@ rules:
config-flow: done
dependency-transparency: done
docs-actions: done
- docs-high-level-description: todo
- docs-installation-instructions: todo
- docs-removal-instructions: todo
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
entity-event-setup:
status: exempt
comment: The integration registers no events
@@ -26,8 +26,10 @@ rules:
# Silver
action-exceptions: done
config-entry-unloading: done
- docs-configuration-parameters: todo
- docs-installation-parameters: todo
+ docs-configuration-parameters:
+ status: exempt
+ comment: Integration has no configuration parameters
+ docs-installation-parameters: done
entity-unavailable: done
integration-owner: done
log-when-unavailable:
@@ -46,13 +48,15 @@ rules:
discovery:
status: exempt
comment: Integration is a service and has no devices.
- docs-data-update: todo
- docs-examples: todo
- docs-known-limitations: todo
- docs-supported-devices: todo
- docs-supported-functions: todo
- docs-troubleshooting: todo
- docs-use-cases: todo
+ docs-data-update: done
+ docs-examples: done
+ docs-known-limitations: done
+ docs-supported-devices:
+ status: exempt
+ comment: Integration is a service and has no devices.
+ docs-supported-functions: done
+ docs-troubleshooting: done
+ docs-use-cases: done
dynamic-devices: done
entity-category: done
entity-device-class: done
diff --git a/homeassistant/components/bring/sensor.py b/homeassistant/components/bring/sensor.py
index bfe93619dbb..2a09d574607 100644
--- a/homeassistant/components/bring/sensor.py
+++ b/homeassistant/components/bring/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import BringConfigEntry, BringData, BringDataUpdateCoordinator
@@ -85,7 +85,7 @@ SENSOR_DESCRIPTIONS: tuple[BringSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BringConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/bring/todo.py b/homeassistant/components/bring/todo.py
index 4de306273f3..d1eb9e78341 100644
--- a/homeassistant/components/bring/todo.py
+++ b/homeassistant/components/bring/todo.py
@@ -24,7 +24,7 @@ from homeassistant.components.todo import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_ITEM_NAME,
@@ -41,7 +41,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BringConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor from a config entry created in the integrations UI."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/broadlink/climate.py b/homeassistant/components/broadlink/climate.py
index 25a6bbd60a5..5be04c24f0d 100644
--- a/homeassistant/components/broadlink/climate.py
+++ b/homeassistant/components/broadlink/climate.py
@@ -13,7 +13,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PRECISION_HALVES, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, DOMAINS_AND_TYPES
from .device import BroadlinkDevice
@@ -31,7 +31,7 @@ class SensorMode(IntEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Broadlink climate entities."""
device = hass.data[DOMAIN].devices[config_entry.entry_id]
diff --git a/homeassistant/components/broadlink/entity.py b/homeassistant/components/broadlink/entity.py
index 6c956d8c80a..a97374680f9 100644
--- a/homeassistant/components/broadlink/entity.py
+++ b/homeassistant/components/broadlink/entity.py
@@ -17,13 +17,13 @@ class BroadlinkEntity(Entity):
self._device = device
self._coordinator = device.update_manager.coordinator
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Call when the entity is added to hass."""
self.async_on_remove(self._coordinator.async_add_listener(self._recv_data))
if self._coordinator.data:
self._update_state(self._coordinator.data)
- async def async_update(self):
+ async def async_update(self) -> None:
"""Update the state of the entity."""
await self._coordinator.async_request_refresh()
@@ -49,7 +49,7 @@ class BroadlinkEntity(Entity):
"""
@property
- def available(self):
+ def available(self) -> bool:
"""Return True if the entity is available."""
return self._device.available
diff --git a/homeassistant/components/broadlink/light.py b/homeassistant/components/broadlink/light.py
index 39d6caaa49f..64698e57249 100644
--- a/homeassistant/components/broadlink/light.py
+++ b/homeassistant/components/broadlink/light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import BroadlinkEntity
@@ -29,7 +29,7 @@ BROADLINK_COLOR_MODE_SCENES = 2
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Broadlink light."""
device = hass.data[DOMAIN].devices[config_entry.entry_id]
diff --git a/homeassistant/components/broadlink/remote.py b/homeassistant/components/broadlink/remote.py
index 18a3a82017c..c1196b03310 100644
--- a/homeassistant/components/broadlink/remote.py
+++ b/homeassistant/components/broadlink/remote.py
@@ -37,7 +37,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_COMMAND, STATE_OFF
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.storage import Store
from homeassistant.util import dt as dt_util
@@ -92,7 +92,7 @@ SERVICE_DELETE_SCHEMA = COMMAND_SCHEMA.extend(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Broadlink remote."""
device = hass.data[DOMAIN].devices[config_entry.entry_id]
diff --git a/homeassistant/components/broadlink/select.py b/homeassistant/components/broadlink/select.py
index 6253adc308a..661fc62600d 100644
--- a/homeassistant/components/broadlink/select.py
+++ b/homeassistant/components/broadlink/select.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.select import SelectEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BroadlinkDevice
from .const import DOMAIN
@@ -28,7 +28,7 @@ DAY_NAME_TO_ID = {v: k for k, v in DAY_ID_TO_NAME.items()}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Broadlink select."""
device = hass.data[DOMAIN].devices[config_entry.entry_id]
diff --git a/homeassistant/components/broadlink/sensor.py b/homeassistant/components/broadlink/sensor.py
index b7ae71ff803..e7d420f0c0e 100644
--- a/homeassistant/components/broadlink/sensor.py
+++ b/homeassistant/components/broadlink/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import BroadlinkEntity
@@ -86,7 +86,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Broadlink sensor."""
device = hass.data[DOMAIN].devices[config_entry.entry_id]
diff --git a/homeassistant/components/broadlink/switch.py b/homeassistant/components/broadlink/switch.py
index 9098440a5c4..d6869ac4c9c 100644
--- a/homeassistant/components/broadlink/switch.py
+++ b/homeassistant/components/broadlink/switch.py
@@ -30,7 +30,10 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -85,7 +88,7 @@ async def async_setup_platform(
if switches := config.get(CONF_SWITCHES):
platform_data = hass.data[DOMAIN].platforms.get(Platform.SWITCH, {})
- async_add_entities_config_entry: AddEntitiesCallback
+ async_add_entities_config_entry: AddConfigEntryEntitiesCallback
device: BroadlinkDevice
async_add_entities_config_entry, device = platform_data.get(
mac_addr, (None, None)
@@ -111,7 +114,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Broadlink switch."""
device = hass.data[DOMAIN].devices[config_entry.entry_id]
diff --git a/homeassistant/components/broadlink/time.py b/homeassistant/components/broadlink/time.py
index 3dcb045fead..4687df6b8b6 100644
--- a/homeassistant/components/broadlink/time.py
+++ b/homeassistant/components/broadlink/time.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.time import TimeEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import BroadlinkDevice
@@ -19,7 +19,7 @@ from .entity import BroadlinkEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Broadlink time."""
device = hass.data[DOMAIN].devices[config_entry.entry_id]
diff --git a/homeassistant/components/brother/__init__.py b/homeassistant/components/brother/__init__.py
index 464e6629224..1c1768b58fd 100644
--- a/homeassistant/components/brother/__init__.py
+++ b/homeassistant/components/brother/__init__.py
@@ -9,6 +9,7 @@ from homeassistant.const import CONF_HOST, CONF_TYPE, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
+from .const import DOMAIN
from .coordinator import BrotherConfigEntry, BrotherDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
@@ -25,7 +26,14 @@ async def async_setup_entry(hass: HomeAssistant, entry: BrotherConfigEntry) -> b
host, printer_type=printer_type, snmp_engine=snmp_engine
)
except (ConnectionError, SnmpError, TimeoutError) as error:
- raise ConfigEntryNotReady from error
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="cannot_connect",
+ translation_placeholders={
+ "device": entry.title,
+ "error": repr(error),
+ },
+ ) from error
coordinator = BrotherDataUpdateCoordinator(hass, entry, brother)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/brother/coordinator.py b/homeassistant/components/brother/coordinator.py
index 4f518ba8a25..a3c337f27f7 100644
--- a/homeassistant/components/brother/coordinator.py
+++ b/homeassistant/components/brother/coordinator.py
@@ -26,6 +26,7 @@ class BrotherDataUpdateCoordinator(DataUpdateCoordinator[BrotherSensors]):
) -> None:
"""Initialize."""
self.brother = brother
+ self.device_name = config_entry.title
super().__init__(
hass,
@@ -41,5 +42,12 @@ class BrotherDataUpdateCoordinator(DataUpdateCoordinator[BrotherSensors]):
async with timeout(20):
data = await self.brother.async_update()
except (ConnectionError, SnmpError, UnsupportedModelError) as error:
- raise UpdateFailed(error) from error
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="update_error",
+ translation_placeholders={
+ "device": self.device_name,
+ "error": repr(error),
+ },
+ ) from error
return data
diff --git a/homeassistant/components/brother/sensor.py b/homeassistant/components/brother/sensor.py
index 087a971f928..a09fe8ebc60 100644
--- a/homeassistant/components/brother/sensor.py
+++ b/homeassistant/components/brother/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -303,7 +303,7 @@ SENSOR_TYPES: tuple[BrotherSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: BrotherConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Brother entities from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/brother/strings.json b/homeassistant/components/brother/strings.json
index b502ed7e3b9..d0714a199c4 100644
--- a/homeassistant/components/brother/strings.json
+++ b/homeassistant/components/brother/strings.json
@@ -159,5 +159,13 @@
"name": "Last restart"
}
}
+ },
+ "exceptions": {
+ "cannot_connect": {
+ "message": "An error occurred while connecting to the {device} printer: {error}"
+ },
+ "update_error": {
+ "message": "An error occurred while retrieving data from the {device} printer: {error}"
+ }
}
}
diff --git a/homeassistant/components/brottsplatskartan/sensor.py b/homeassistant/components/brottsplatskartan/sensor.py
index 6725a32bb40..60f9a8163de 100644
--- a/homeassistant/components/brottsplatskartan/sensor.py
+++ b/homeassistant/components/brottsplatskartan/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_APP_ID, CONF_AREA, DOMAIN, LOGGER
@@ -21,7 +21,9 @@ SCAN_INTERVAL = timedelta(minutes=30)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Brottsplatskartan sensor entry."""
diff --git a/homeassistant/components/brunt/cover.py b/homeassistant/components/brunt/cover.py
index bb97f42bd36..95931d3449e 100644
--- a/homeassistant/components/brunt/cover.py
+++ b/homeassistant/components/brunt/cover.py
@@ -16,7 +16,7 @@ from homeassistant.components.cover import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -34,7 +34,7 @@ from .coordinator import BruntConfigEntry, BruntCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: BruntConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the brunt platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/bryant_evolution/climate.py b/homeassistant/components/bryant_evolution/climate.py
index 2d54ced8217..bd053229a1a 100644
--- a/homeassistant/components/bryant_evolution/climate.py
+++ b/homeassistant/components/bryant_evolution/climate.py
@@ -17,7 +17,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BryantEvolutionConfigEntry, names
from .const import CONF_SYSTEM_ZONE, DOMAIN
@@ -31,7 +31,7 @@ SCAN_INTERVAL = timedelta(seconds=60)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: BryantEvolutionConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
diff --git a/homeassistant/components/bsblan/climate.py b/homeassistant/components/bsblan/climate.py
index 2833d6549b4..bef0388a57d 100644
--- a/homeassistant/components/bsblan/climate.py
+++ b/homeassistant/components/bsblan/climate.py
@@ -19,7 +19,7 @@ from homeassistant.const import ATTR_TEMPERATURE
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers.device_registry import format_mac
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.enum import try_parse_enum
from . import BSBLanConfigEntry, BSBLanData
@@ -43,7 +43,7 @@ PRESET_MODES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: BSBLanConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BSBLAN device based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/bsblan/sensor.py b/homeassistant/components/bsblan/sensor.py
index c13b4ad7650..6a6784a4542 100644
--- a/homeassistant/components/bsblan/sensor.py
+++ b/homeassistant/components/bsblan/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import BSBLanConfigEntry, BSBLanData
@@ -51,7 +51,7 @@ SENSOR_TYPES: tuple[BSBLanSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: BSBLanConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BSB-Lan sensor based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/bsblan/strings.json b/homeassistant/components/bsblan/strings.json
index a73a89ca1cc..93562763999 100644
--- a/homeassistant/components/bsblan/strings.json
+++ b/homeassistant/components/bsblan/strings.json
@@ -30,7 +30,7 @@
"message": "Can't set preset mode to {preset_mode} when HVAC mode is not set to auto"
},
"set_data_error": {
- "message": "An error occurred while sending the data to the BSBLAN device"
+ "message": "An error occurred while sending the data to the BSB-Lan device"
},
"set_temperature_error": {
"message": "An error occurred while setting the temperature"
diff --git a/homeassistant/components/bsblan/water_heater.py b/homeassistant/components/bsblan/water_heater.py
index 318408a9124..a3aee4cdc15 100644
--- a/homeassistant/components/bsblan/water_heater.py
+++ b/homeassistant/components/bsblan/water_heater.py
@@ -16,7 +16,7 @@ from homeassistant.const import ATTR_TEMPERATURE, STATE_ON
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import format_mac
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BSBLanConfigEntry, BSBLanData
from .const import DOMAIN
@@ -37,7 +37,7 @@ OPERATION_MODES_REVERSE = {v: k for k, v in OPERATION_MODES.items()}
async def async_setup_entry(
hass: HomeAssistant,
entry: BSBLanConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BSBLAN water heater based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/bthome/binary_sensor.py b/homeassistant/components/bthome/binary_sensor.py
index bcc420df4a8..97ed85c1204 100644
--- a/homeassistant/components/bthome/binary_sensor.py
+++ b/homeassistant/components/bthome/binary_sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.bluetooth.passive_update_processor import (
PassiveBluetoothProcessorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .coordinator import BTHomePassiveBluetoothDataProcessor
@@ -169,7 +169,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: BTHomeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the BTHome BLE binary sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/bthome/event.py b/homeassistant/components/bthome/event.py
index a6ee79f4e05..99799819e43 100644
--- a/homeassistant/components/bthome/event.py
+++ b/homeassistant/components/bthome/event.py
@@ -12,7 +12,7 @@ from homeassistant.components.event import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import format_discovered_event_class, format_event_dispatcher_name
from .const import (
@@ -104,7 +104,7 @@ class BTHomeEventEntity(EventEntity):
async def async_setup_entry(
hass: HomeAssistant,
entry: BTHomeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up BTHome event."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/bthome/sensor.py b/homeassistant/components/bthome/sensor.py
index 23a058b0b0c..7025929abd8 100644
--- a/homeassistant/components/bthome/sensor.py
+++ b/homeassistant/components/bthome/sensor.py
@@ -42,7 +42,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .coordinator import BTHomePassiveBluetoothDataProcessor
@@ -423,7 +423,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: BTHomeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the BTHome BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/buienradar/camera.py b/homeassistant/components/buienradar/camera.py
index 45ff2d6de52..15d08281911 100644
--- a/homeassistant/components/buienradar/camera.py
+++ b/homeassistant/components/buienradar/camera.py
@@ -13,7 +13,7 @@ from homeassistant.components.camera import Camera
from homeassistant.const import CONF_COUNTRY_CODE, CONF_LATITUDE, CONF_LONGITUDE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import BuienRadarConfigEntry
@@ -31,7 +31,7 @@ SUPPORTED_COUNTRY_CODES = ["NL", "BE"]
async def async_setup_entry(
hass: HomeAssistant,
entry: BuienRadarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up buienradar radar-loop camera component."""
config = entry.data
diff --git a/homeassistant/components/buienradar/sensor.py b/homeassistant/components/buienradar/sensor.py
index 712f765237e..a4d39ea07cc 100644
--- a/homeassistant/components/buienradar/sensor.py
+++ b/homeassistant/components/buienradar/sensor.py
@@ -45,7 +45,7 @@ from homeassistant.const import (
UnitOfVolumetricFlux,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import BuienRadarConfigEntry
@@ -169,6 +169,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
translation_key="windazimuth",
native_unit_of_measurement=DEGREE,
icon="mdi:compass-outline",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key="pressure",
@@ -530,30 +531,35 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
translation_key="windazimuth_1d",
native_unit_of_measurement=DEGREE,
icon="mdi:compass-outline",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key="windazimuth_2d",
translation_key="windazimuth_2d",
native_unit_of_measurement=DEGREE,
icon="mdi:compass-outline",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key="windazimuth_3d",
translation_key="windazimuth_3d",
native_unit_of_measurement=DEGREE,
icon="mdi:compass-outline",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key="windazimuth_4d",
translation_key="windazimuth_4d",
native_unit_of_measurement=DEGREE,
icon="mdi:compass-outline",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key="windazimuth_5d",
translation_key="windazimuth_5d",
native_unit_of_measurement=DEGREE,
icon="mdi:compass-outline",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key="condition_1d",
@@ -691,7 +697,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: BuienRadarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the buienradar sensor."""
config = entry.data
diff --git a/homeassistant/components/buienradar/weather.py b/homeassistant/components/buienradar/weather.py
index 8b71032bace..4b71024c241 100644
--- a/homeassistant/components/buienradar/weather.py
+++ b/homeassistant/components/buienradar/weather.py
@@ -51,7 +51,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import BuienRadarConfigEntry
from .const import DEFAULT_TIMEFRAME
@@ -94,7 +94,7 @@ CONDITION_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
entry: BuienRadarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the buienradar platform."""
config = entry.data
diff --git a/homeassistant/components/burbank_water_and_power/__init__.py b/homeassistant/components/burbank_water_and_power/__init__.py
new file mode 100644
index 00000000000..2b82c8bd56b
--- /dev/null
+++ b/homeassistant/components/burbank_water_and_power/__init__.py
@@ -0,0 +1 @@
+"""Virtual integration: Burbank Water and Power (BWP)."""
diff --git a/homeassistant/components/burbank_water_and_power/manifest.json b/homeassistant/components/burbank_water_and_power/manifest.json
new file mode 100644
index 00000000000..7b938d3b98b
--- /dev/null
+++ b/homeassistant/components/burbank_water_and_power/manifest.json
@@ -0,0 +1,6 @@
+{
+ "domain": "burbank_water_and_power",
+ "name": "Burbank Water and Power (BWP)",
+ "integration_type": "virtual",
+ "supported_by": "opower"
+}
diff --git a/homeassistant/components/caldav/calendar.py b/homeassistant/components/caldav/calendar.py
index 7a426112d04..be909a02ea5 100644
--- a/homeassistant/components/caldav/calendar.py
+++ b/homeassistant/components/caldav/calendar.py
@@ -25,7 +25,10 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity import async_generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -143,7 +146,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: CalDavConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the CalDav calendar platform for a config entry."""
calendars = await async_get_calendars(hass, entry.runtime_data, SUPPORTED_COMPONENT)
diff --git a/homeassistant/components/caldav/todo.py b/homeassistant/components/caldav/todo.py
index cbd7963b595..73f172dabec 100644
--- a/homeassistant/components/caldav/todo.py
+++ b/homeassistant/components/caldav/todo.py
@@ -20,7 +20,7 @@ from homeassistant.components.todo import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import CalDavConfigEntry
@@ -46,7 +46,7 @@ TODO_STATUS_MAP_INV: dict[TodoItemStatus, str] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: CalDavConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the CalDav todo platform for a config entry."""
calendars = await async_get_calendars(hass, entry.runtime_data, SUPPORTED_COMPONENT)
@@ -138,6 +138,8 @@ class WebDavTodoListEntity(TodoListEntity):
await self.hass.async_add_executor_job(
partial(self._calendar.save_todo, **item_data),
)
+ # refreshing async otherwise it would take too much time
+ self.hass.async_create_task(self.async_update_ha_state(force_refresh=True))
except (requests.ConnectionError, DAVError) as err:
raise HomeAssistantError(f"CalDAV save error: {err}") from err
@@ -172,6 +174,8 @@ class WebDavTodoListEntity(TodoListEntity):
obj_type="todo",
),
)
+ # refreshing async otherwise it would take too much time
+ self.hass.async_create_task(self.async_update_ha_state(force_refresh=True))
except (requests.ConnectionError, DAVError) as err:
raise HomeAssistantError(f"CalDAV save error: {err}") from err
@@ -195,3 +199,5 @@ class WebDavTodoListEntity(TodoListEntity):
await self.hass.async_add_executor_job(item.delete)
except (requests.ConnectionError, DAVError) as err:
raise HomeAssistantError(f"CalDAV delete error: {err}") from err
+ # refreshing async otherwise it would take too much time
+ self.hass.async_create_task(self.async_update_ha_state(force_refresh=True))
diff --git a/homeassistant/components/calendar/__init__.py b/homeassistant/components/calendar/__init__.py
index 40d6952fa64..96bf717c3ac 100644
--- a/homeassistant/components/calendar/__init__.py
+++ b/homeassistant/components/calendar/__init__.py
@@ -153,6 +153,27 @@ def _has_min_duration(
return validate
+def _has_positive_interval(
+ start_key: str, end_key: str, duration_key: str
+) -> Callable[[dict[str, Any]], dict[str, Any]]:
+ """Verify that the time span between start and end is greater than zero."""
+
+ def validate(obj: dict[str, Any]) -> dict[str, Any]:
+ if (duration := obj.get(duration_key)) is not None:
+ if duration <= datetime.timedelta(seconds=0):
+ raise vol.Invalid(f"Expected positive duration ({duration})")
+ return obj
+
+ if (start := obj.get(start_key)) and (end := obj.get(end_key)):
+ if start >= end:
+ raise vol.Invalid(
+ f"Expected end time to be after start time ({start}, {end})"
+ )
+ return obj
+
+ return validate
+
+
def _has_same_type(*keys: Any) -> Callable[[dict[str, Any]], dict[str, Any]]:
"""Verify that all values are of the same type."""
@@ -281,6 +302,7 @@ SERVICE_GET_EVENTS_SCHEMA: Final = vol.All(
),
}
),
+ _has_positive_interval(EVENT_START_DATETIME, EVENT_END_DATETIME, EVENT_DURATION),
)
@@ -870,6 +892,7 @@ async def async_get_events_service(
end = start + service_call.data[EVENT_DURATION]
else:
end = service_call.data[EVENT_END_DATETIME]
+
calendar_event_list = await calendar.async_get_events(
calendar.hass, dt_util.as_local(start), dt_util.as_local(end)
)
diff --git a/homeassistant/components/cambridge_audio/manifest.json b/homeassistant/components/cambridge_audio/manifest.json
index 14a389587d2..88d28e256aa 100644
--- a/homeassistant/components/cambridge_audio/manifest.json
+++ b/homeassistant/components/cambridge_audio/manifest.json
@@ -8,6 +8,6 @@
"iot_class": "local_push",
"loggers": ["aiostreammagic"],
"quality_scale": "platinum",
- "requirements": ["aiostreammagic==2.10.0"],
+ "requirements": ["aiostreammagic==2.11.0"],
"zeroconf": ["_stream-magic._tcp.local.", "_smoip._tcp.local."]
}
diff --git a/homeassistant/components/cambridge_audio/media_player.py b/homeassistant/components/cambridge_audio/media_player.py
index 042178d5781..d18898fa916 100644
--- a/homeassistant/components/cambridge_audio/media_player.py
+++ b/homeassistant/components/cambridge_audio/media_player.py
@@ -23,7 +23,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import CambridgeAudioConfigEntry, media_browser
from .const import (
@@ -65,7 +65,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: CambridgeAudioConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Cambridge Audio device based on a config entry."""
client: StreamMagicClient = entry.runtime_data
diff --git a/homeassistant/components/cambridge_audio/select.py b/homeassistant/components/cambridge_audio/select.py
index 6bfe83c2539..e7d9136711f 100644
--- a/homeassistant/components/cambridge_audio/select.py
+++ b/homeassistant/components/cambridge_audio/select.py
@@ -9,7 +9,7 @@ from aiostreammagic.models import DisplayBrightness
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import CambridgeAudioConfigEntry
from .entity import CambridgeAudioEntity, command
@@ -82,7 +82,7 @@ CONTROL_ENTITIES: tuple[CambridgeAudioSelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: CambridgeAudioConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Cambridge Audio select entities based on a config entry."""
diff --git a/homeassistant/components/cambridge_audio/switch.py b/homeassistant/components/cambridge_audio/switch.py
index 065a1da4f94..0cebe8266c4 100644
--- a/homeassistant/components/cambridge_audio/switch.py
+++ b/homeassistant/components/cambridge_audio/switch.py
@@ -9,7 +9,7 @@ from aiostreammagic import StreamMagicClient
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import CambridgeAudioConfigEntry
from .entity import CambridgeAudioEntity, command
@@ -46,7 +46,7 @@ CONTROL_ENTITIES: tuple[CambridgeAudioSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: CambridgeAudioConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Cambridge Audio switch entities based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/canary/alarm_control_panel.py b/homeassistant/components/canary/alarm_control_panel.py
index 443944da8c3..9fe2dfb598d 100644
--- a/homeassistant/components/canary/alarm_control_panel.py
+++ b/homeassistant/components/canary/alarm_control_panel.py
@@ -13,7 +13,7 @@ from homeassistant.components.alarm_control_panel import (
AlarmControlPanelState,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import CanaryConfigEntry, CanaryDataUpdateCoordinator
@@ -22,7 +22,7 @@ from .coordinator import CanaryConfigEntry, CanaryDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: CanaryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Canary alarm control panels based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/canary/camera.py b/homeassistant/components/canary/camera.py
index 8f4a01c9968..07645f2f403 100644
--- a/homeassistant/components/canary/camera.py
+++ b/homeassistant/components/canary/camera.py
@@ -22,7 +22,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -48,7 +48,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: CanaryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Canary sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/canary/sensor.py b/homeassistant/components/canary/sensor.py
index 22f3eada2cb..d92166926e9 100644
--- a/homeassistant/components/canary/sensor.py
+++ b/homeassistant/components/canary/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER
@@ -64,7 +64,7 @@ STATE_AIR_QUALITY_VERY_ABNORMAL: Final = "very_abnormal"
async def async_setup_entry(
hass: HomeAssistant,
entry: CanaryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Canary sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/canary/strings.json b/homeassistant/components/canary/strings.json
index 699e8b25e11..8be11a48b5e 100644
--- a/homeassistant/components/canary/strings.json
+++ b/homeassistant/components/canary/strings.json
@@ -21,8 +21,8 @@
"step": {
"init": {
"data": {
- "ffmpeg_arguments": "Arguments passed to ffmpeg for cameras",
- "timeout": "Request Timeout (seconds)"
+ "ffmpeg_arguments": "Arguments passed to FFmpeg for cameras",
+ "timeout": "Request timeout (seconds)"
}
}
}
diff --git a/homeassistant/components/cast/config_flow.py b/homeassistant/components/cast/config_flow.py
index 034cf856023..6c33eac230f 100644
--- a/homeassistant/components/cast/config_flow.py
+++ b/homeassistant/components/cast/config_flow.py
@@ -16,12 +16,21 @@ from homeassistant.config_entries import (
from homeassistant.const import CONF_UUID
from homeassistant.core import callback
from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.selector import SelectSelector, SelectSelectorConfig
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from .const import CONF_IGNORE_CEC, CONF_KNOWN_HOSTS, DOMAIN
IGNORE_CEC_SCHEMA = vol.Schema(vol.All(cv.ensure_list, [cv.string]))
-KNOWN_HOSTS_SCHEMA = vol.Schema(vol.All(cv.ensure_list, [cv.string]))
+KNOWN_HOSTS_SCHEMA = vol.Schema(
+ {
+ vol.Optional(
+ CONF_KNOWN_HOSTS,
+ ): SelectSelector(
+ SelectSelectorConfig(custom_value=True, options=[], multiple=True),
+ )
+ }
+)
WANTED_UUID_SCHEMA = vol.Schema(vol.All(cv.ensure_list, [cv.string]))
@@ -30,12 +39,6 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
VERSION = 1
- def __init__(self) -> None:
- """Initialize flow."""
- self._ignore_cec = set[str]()
- self._known_hosts = set[str]()
- self._wanted_uuid = set[str]()
-
@staticmethod
@callback
def async_get_options_flow(
@@ -62,48 +65,31 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Confirm the setup."""
- errors = {}
- data = {CONF_KNOWN_HOSTS: self._known_hosts}
-
if user_input is not None:
- bad_hosts = False
- known_hosts = user_input[CONF_KNOWN_HOSTS]
- known_hosts = [x.strip() for x in known_hosts.split(",") if x.strip()]
- try:
- known_hosts = KNOWN_HOSTS_SCHEMA(known_hosts)
- except vol.Invalid:
- errors["base"] = "invalid_known_hosts"
- bad_hosts = True
- else:
- self._known_hosts = known_hosts
- data = self._get_data()
- if not bad_hosts:
- return self.async_create_entry(title="Google Cast", data=data)
+ known_hosts = _trim_items(user_input.get(CONF_KNOWN_HOSTS, []))
+ return self.async_create_entry(
+ title="Google Cast",
+ data=self._get_data(known_hosts=known_hosts),
+ )
- fields = {}
- fields[vol.Optional(CONF_KNOWN_HOSTS, default="")] = str
-
- return self.async_show_form(
- step_id="config", data_schema=vol.Schema(fields), errors=errors
- )
+ return self.async_show_form(step_id="config", data_schema=KNOWN_HOSTS_SCHEMA)
async def async_step_confirm(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Confirm the setup."""
-
- data = self._get_data()
-
if user_input is not None or not onboarding.async_is_onboarded(self.hass):
- return self.async_create_entry(title="Google Cast", data=data)
+ return self.async_create_entry(title="Google Cast", data=self._get_data())
return self.async_show_form(step_id="confirm")
- def _get_data(self):
+ def _get_data(
+ self, *, known_hosts: list[str] | None = None
+ ) -> dict[str, list[str]]:
return {
- CONF_IGNORE_CEC: list(self._ignore_cec),
- CONF_KNOWN_HOSTS: list(self._known_hosts),
- CONF_UUID: list(self._wanted_uuid),
+ CONF_IGNORE_CEC: [],
+ CONF_KNOWN_HOSTS: known_hosts or [],
+ CONF_UUID: [],
}
@@ -123,31 +109,24 @@ class CastOptionsFlowHandler(OptionsFlow):
) -> ConfigFlowResult:
"""Manage the Google Cast options."""
errors: dict[str, str] = {}
- current_config = self.config_entry.data
if user_input is not None:
- bad_hosts, known_hosts = _string_to_list(
- user_input.get(CONF_KNOWN_HOSTS, ""), KNOWN_HOSTS_SCHEMA
+ known_hosts = _trim_items(user_input.get(CONF_KNOWN_HOSTS, []))
+ self.updated_config = dict(self.config_entry.data)
+ self.updated_config[CONF_KNOWN_HOSTS] = known_hosts
+
+ if self.show_advanced_options:
+ return await self.async_step_advanced_options()
+
+ self.hass.config_entries.async_update_entry(
+ self.config_entry, data=self.updated_config
)
-
- if not bad_hosts:
- self.updated_config = dict(current_config)
- self.updated_config[CONF_KNOWN_HOSTS] = known_hosts
-
- if self.show_advanced_options:
- return await self.async_step_advanced_options()
-
- self.hass.config_entries.async_update_entry(
- self.config_entry, data=self.updated_config
- )
- return self.async_create_entry(title="", data={})
-
- fields: dict[vol.Marker, type[str]] = {}
- suggested_value = _list_to_string(current_config.get(CONF_KNOWN_HOSTS))
- _add_with_suggestion(fields, CONF_KNOWN_HOSTS, suggested_value)
+ return self.async_create_entry(title="", data={})
return self.async_show_form(
step_id="basic_options",
- data_schema=vol.Schema(fields),
+ data_schema=self.add_suggested_values_to_schema(
+ KNOWN_HOSTS_SCHEMA, self.config_entry.data
+ ),
errors=errors,
last_step=not self.show_advanced_options,
)
@@ -206,6 +185,10 @@ def _string_to_list(string, schema):
return invalid, items
+def _trim_items(items: list[str]) -> list[str]:
+ return [x.strip() for x in items if x.strip()]
+
+
def _add_with_suggestion(
fields: dict[vol.Marker, type[str]], key: str, suggested_value: str
) -> None:
diff --git a/homeassistant/components/cast/const.py b/homeassistant/components/cast/const.py
index 056ee054d1d..0a85a0007b3 100644
--- a/homeassistant/components/cast/const.py
+++ b/homeassistant/components/cast/const.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from typing import TYPE_CHECKING, TypedDict
+from typing import TYPE_CHECKING, NotRequired, TypedDict
from homeassistant.util.signal_type import SignalType
@@ -46,3 +46,4 @@ class HomeAssistantControllerData(TypedDict):
hass_uuid: str
client_id: str | None
refresh_token: str
+ app_id: NotRequired[str]
diff --git a/homeassistant/components/cast/helpers.py b/homeassistant/components/cast/helpers.py
index 8f4af197b8e..7f46100afca 100644
--- a/homeassistant/components/cast/helpers.py
+++ b/homeassistant/components/cast/helpers.py
@@ -7,6 +7,7 @@ from dataclasses import dataclass
import logging
from typing import TYPE_CHECKING, ClassVar
from urllib.parse import urlparse
+from uuid import UUID
import aiohttp
import attr
@@ -40,7 +41,7 @@ class ChromecastInfo:
is_dynamic_group = attr.ib(type=bool | None, default=None)
@property
- def friendly_name(self) -> str:
+ def friendly_name(self) -> str | None:
"""Return the Friendly Name."""
return self.cast_info.friendly_name
@@ -50,7 +51,7 @@ class ChromecastInfo:
return self.cast_info.cast_type == CAST_TYPE_GROUP
@property
- def uuid(self) -> bool:
+ def uuid(self) -> UUID:
"""Return the UUID."""
return self.cast_info.uuid
@@ -111,7 +112,10 @@ class ChromecastInfo:
is_dynamic_group = False
http_group_status = None
http_group_status = dial.get_multizone_status(
- None,
+ # We pass services which will be used for the HTTP request, and we
+ # don't care about the host in http_group_status.dynamic_groups so
+ # we pass an empty string to simplify the code.
+ "",
services=self.cast_info.services,
zconf=ChromeCastZeroconf.get_zeroconf(),
)
diff --git a/homeassistant/components/cast/manifest.json b/homeassistant/components/cast/manifest.json
index 0650f267544..feb613f4765 100644
--- a/homeassistant/components/cast/manifest.json
+++ b/homeassistant/components/cast/manifest.json
@@ -14,7 +14,7 @@
"documentation": "https://www.home-assistant.io/integrations/cast",
"iot_class": "local_polling",
"loggers": ["casttube", "pychromecast"],
- "requirements": ["PyChromecast==14.0.5"],
+ "requirements": ["PyChromecast==14.0.6"],
"single_config_entry": true,
"zeroconf": ["_googlecast._tcp.local."]
}
diff --git a/homeassistant/components/cast/media_player.py b/homeassistant/components/cast/media_player.py
index 3cc17fae43b..8ff078dfafd 100644
--- a/homeassistant/components/cast/media_player.py
+++ b/homeassistant/components/cast/media_player.py
@@ -51,7 +51,7 @@ from homeassistant.core import CALLBACK_TYPE, Event, HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.network import NoURLAvailableError, get_url, is_hass_url
from homeassistant.util import dt as dt_util
from homeassistant.util.logging import async_create_catching_coro
@@ -140,7 +140,7 @@ def _async_create_cast_device(hass: HomeAssistant, info: ChromecastInfo):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Cast from a config entry."""
hass.data.setdefault(ADDED_CAST_DEVICES_KEY, set())
diff --git a/homeassistant/components/cast/services.yaml b/homeassistant/components/cast/services.yaml
index e2e23ad40a2..45b36f6d983 100644
--- a/homeassistant/components/cast/services.yaml
+++ b/homeassistant/components/cast/services.yaml
@@ -7,11 +7,11 @@ show_lovelace_view:
integration: cast
domain: media_player
dashboard_path:
- required: true
example: lovelace-cast
selector:
text:
view_path:
+ required: true
example: downstairs
selector:
text:
diff --git a/homeassistant/components/cast/strings.json b/homeassistant/components/cast/strings.json
index 9c49813bd83..8c7c7c0cff0 100644
--- a/homeassistant/components/cast/strings.json
+++ b/homeassistant/components/cast/strings.json
@@ -6,9 +6,11 @@
},
"config": {
"title": "Google Cast configuration",
- "description": "Known Hosts - A comma-separated list of hostnames or IP-addresses of cast devices, use if mDNS discovery is not working.",
"data": {
- "known_hosts": "Known hosts"
+ "known_hosts": "Add known host"
+ },
+ "data_description": {
+ "known_hosts": "Hostnames or IP-addresses of cast devices, use if mDNS discovery is not working"
}
}
},
@@ -20,9 +22,11 @@
"step": {
"basic_options": {
"title": "[%key:component::cast::config::step::config::title%]",
- "description": "[%key:component::cast::config::step::config::description%]",
"data": {
"known_hosts": "[%key:component::cast::config::step::config::data::known_hosts%]"
+ },
+ "data_description": {
+ "known_hosts": "[%key:component::cast::config::step::config::data_description::known_hosts%]"
}
},
"advanced_options": {
@@ -49,7 +53,7 @@
},
"dashboard_path": {
"name": "Dashboard path",
- "description": "The URL path of the dashboard to show."
+ "description": "The URL path of the dashboard to show, defaults to lovelace if not specified."
},
"view_path": {
"name": "View path",
diff --git a/homeassistant/components/ccm15/climate.py b/homeassistant/components/ccm15/climate.py
index 099b91ec02c..df321395b9e 100644
--- a/homeassistant/components/ccm15/climate.py
+++ b/homeassistant/components/ccm15/climate.py
@@ -20,7 +20,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONST_CMD_FAN_MAP, CONST_CMD_STATE_MAP, DOMAIN
@@ -32,7 +32,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CCM15ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all climate."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/cert_expiry/__init__.py b/homeassistant/components/cert_expiry/__init__.py
index bc6ae29ee8e..adf1e0e981c 100644
--- a/homeassistant/components/cert_expiry/__init__.py
+++ b/homeassistant/components/cert_expiry/__init__.py
@@ -2,24 +2,21 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.start import async_at_started
-from .coordinator import CertExpiryDataUpdateCoordinator
+from .coordinator import CertExpiryConfigEntry, CertExpiryDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
-type CertExpiryConfigEntry = ConfigEntry[CertExpiryDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: CertExpiryConfigEntry) -> bool:
"""Load the saved entities."""
host: str = entry.data[CONF_HOST]
port: int = entry.data[CONF_PORT]
- coordinator = CertExpiryDataUpdateCoordinator(hass, host, port)
+ coordinator = CertExpiryDataUpdateCoordinator(hass, entry, host, port)
entry.runtime_data = coordinator
@@ -34,6 +31,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: CertExpiryConfigEntry) -
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: CertExpiryConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/cert_expiry/coordinator.py b/homeassistant/components/cert_expiry/coordinator.py
index 80c91f1d890..644e3ee3d00 100644
--- a/homeassistant/components/cert_expiry/coordinator.py
+++ b/homeassistant/components/cert_expiry/coordinator.py
@@ -5,6 +5,7 @@ from __future__ import annotations
from datetime import datetime, timedelta
import logging
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -14,11 +15,21 @@ from .helper import get_cert_expiry_timestamp
_LOGGER = logging.getLogger(__name__)
+type CertExpiryConfigEntry = ConfigEntry[CertExpiryDataUpdateCoordinator]
+
class CertExpiryDataUpdateCoordinator(DataUpdateCoordinator[datetime | None]):
"""Class to manage fetching Cert Expiry data from single endpoint."""
- def __init__(self, hass: HomeAssistant, host: str, port: int) -> None:
+ config_entry: CertExpiryConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: CertExpiryConfigEntry,
+ host: str,
+ port: int,
+ ) -> None:
"""Initialize global Cert Expiry data updater."""
self.host = host
self.port = port
@@ -31,6 +42,7 @@ class CertExpiryDataUpdateCoordinator(DataUpdateCoordinator[datetime | None]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=name,
update_interval=timedelta(hours=12),
always_update=False,
diff --git a/homeassistant/components/cert_expiry/sensor.py b/homeassistant/components/cert_expiry/sensor.py
index 4fd0846f0f3..3854dfc109e 100644
--- a/homeassistant/components/cert_expiry/sensor.py
+++ b/homeassistant/components/cert_expiry/sensor.py
@@ -7,18 +7,17 @@ from datetime import datetime
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import CertExpiryConfigEntry
from .const import DOMAIN
-from .coordinator import CertExpiryDataUpdateCoordinator
+from .coordinator import CertExpiryConfigEntry, CertExpiryDataUpdateCoordinator
from .entity import CertExpiryEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: CertExpiryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add cert-expiry entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/chacon_dio/cover.py b/homeassistant/components/chacon_dio/cover.py
index 3a4955adf5c..ea80116be8a 100644
--- a/homeassistant/components/chacon_dio/cover.py
+++ b/homeassistant/components/chacon_dio/cover.py
@@ -12,7 +12,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ChaconDioConfigEntry
from .entity import ChaconDioEntity
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ChaconDioConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Chacon Dio cover devices."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/chacon_dio/switch.py b/homeassistant/components/chacon_dio/switch.py
index be178c3c3b5..05b55552615 100644
--- a/homeassistant/components/chacon_dio/switch.py
+++ b/homeassistant/components/chacon_dio/switch.py
@@ -7,7 +7,7 @@ from dio_chacon_wifi_api.const import DeviceTypeEnum
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ChaconDioConfigEntry
from .entity import ChaconDioEntity
@@ -18,7 +18,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ChaconDioConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Chacon Dio switch devices."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/cisco_ios/device_tracker.py b/homeassistant/components/cisco_ios/device_tracker.py
index 0477ebb111c..6cc403817cf 100644
--- a/homeassistant/components/cisco_ios/device_tracker.py
+++ b/homeassistant/components/cisco_ios/device_tracker.py
@@ -104,7 +104,7 @@ class CiscoDeviceScanner(DeviceScanner):
"""Open connection to the router and get arp entries."""
try:
- cisco_ssh: pxssh.pxssh[str] = pxssh.pxssh(encoding="uft-8")
+ cisco_ssh: pxssh.pxssh[str] = pxssh.pxssh(encoding="utf-8")
cisco_ssh.login(
self.host,
self.username,
diff --git a/homeassistant/components/climate/__init__.py b/homeassistant/components/climate/__init__.py
index 3ea0f887e76..287a2397121 100644
--- a/homeassistant/components/climate/__init__.py
+++ b/homeassistant/components/climate/__init__.py
@@ -68,7 +68,6 @@ from .const import ( # noqa: F401
FAN_ON,
FAN_TOP,
HVAC_MODES,
- INTENT_GET_TEMPERATURE,
INTENT_SET_TEMPERATURE,
PRESET_ACTIVITY,
PRESET_AWAY,
diff --git a/homeassistant/components/climate/const.py b/homeassistant/components/climate/const.py
index d347ccbbb29..ecc0066cd93 100644
--- a/homeassistant/components/climate/const.py
+++ b/homeassistant/components/climate/const.py
@@ -126,7 +126,6 @@ DEFAULT_MAX_HUMIDITY = 99
DOMAIN = "climate"
-INTENT_GET_TEMPERATURE = "HassClimateGetTemperature"
INTENT_SET_TEMPERATURE = "HassClimateSetTemperature"
SERVICE_SET_AUX_HEAT = "set_aux_heat"
diff --git a/homeassistant/components/climate/intent.py b/homeassistant/components/climate/intent.py
index 9837a326188..7691a2db0f1 100644
--- a/homeassistant/components/climate/intent.py
+++ b/homeassistant/components/climate/intent.py
@@ -1,4 +1,4 @@
-"""Intents for the client integration."""
+"""Intents for the climate integration."""
from __future__ import annotations
@@ -11,7 +11,6 @@ from homeassistant.helpers import config_validation as cv, intent
from . import (
ATTR_TEMPERATURE,
DOMAIN,
- INTENT_GET_TEMPERATURE,
INTENT_SET_TEMPERATURE,
SERVICE_SET_TEMPERATURE,
ClimateEntityFeature,
@@ -20,49 +19,9 @@ from . import (
async def async_setup_intents(hass: HomeAssistant) -> None:
"""Set up the climate intents."""
- intent.async_register(hass, GetTemperatureIntent())
intent.async_register(hass, SetTemperatureIntent())
-class GetTemperatureIntent(intent.IntentHandler):
- """Handle GetTemperature intents."""
-
- intent_type = INTENT_GET_TEMPERATURE
- description = "Gets the current temperature of a climate device or entity"
- slot_schema = {
- vol.Optional("area"): intent.non_empty_string,
- vol.Optional("name"): intent.non_empty_string,
- }
- platforms = {DOMAIN}
-
- async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse:
- """Handle the intent."""
- hass = intent_obj.hass
- slots = self.async_validate_slots(intent_obj.slots)
-
- name: str | None = None
- if "name" in slots:
- name = slots["name"]["value"]
-
- area: str | None = None
- if "area" in slots:
- area = slots["area"]["value"]
-
- match_constraints = intent.MatchTargetsConstraints(
- name=name, area_name=area, domains=[DOMAIN], assistant=intent_obj.assistant
- )
- match_result = intent.async_match_targets(hass, match_constraints)
- if not match_result.is_match:
- raise intent.MatchFailedError(
- result=match_result, constraints=match_constraints
- )
-
- response = intent_obj.create_response()
- response.response_type = intent.IntentResponseType.QUERY_ANSWER
- response.async_set_states(matched_states=match_result.states)
- return response
-
-
class SetTemperatureIntent(intent.IntentHandler):
"""Handle SetTemperature intents."""
diff --git a/homeassistant/components/cloud/__init__.py b/homeassistant/components/cloud/__init__.py
index 55ffedd2781..97210b4197c 100644
--- a/homeassistant/components/cloud/__init__.py
+++ b/homeassistant/components/cloud/__init__.py
@@ -6,6 +6,7 @@ import asyncio
from collections.abc import Awaitable, Callable
from datetime import datetime, timedelta
from enum import Enum
+import logging
from typing import cast
from hass_nabucasa import Cloud
@@ -19,6 +20,7 @@ from homeassistant.const import (
CONF_NAME,
CONF_REGION,
EVENT_HOMEASSISTANT_STOP,
+ FORMAT_DATETIME,
Platform,
)
from homeassistant.core import Event, HassJob, HomeAssistant, ServiceCall, callback
@@ -33,7 +35,7 @@ from homeassistant.helpers.dispatcher import (
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.service import async_register_admin_service
from homeassistant.helpers.typing import ConfigType
-from homeassistant.loader import bind_hass
+from homeassistant.loader import async_get_integration, bind_hass
from homeassistant.util.signal_type import SignalType
# Pre-import backup to avoid it being imported
@@ -62,11 +64,13 @@ from .const import (
CONF_THINGTALK_SERVER,
CONF_USER_POOL_ID,
DATA_CLOUD,
+ DATA_CLOUD_LOG_HANDLER,
DATA_PLATFORMS_SETUP,
DOMAIN,
MODE_DEV,
MODE_PROD,
)
+from .helpers import FixedSizeQueueLogHandler
from .prefs import CloudPreferences
from .repairs import async_manage_legacy_subscription_issue
from .subscription import async_subscription_info
@@ -245,6 +249,8 @@ def async_remote_ui_url(hass: HomeAssistant) -> str:
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Initialize the Home Assistant cloud."""
+ log_handler = hass.data[DATA_CLOUD_LOG_HANDLER] = await _setup_log_handler(hass)
+
# Process configs
if DOMAIN in config:
kwargs = dict(config[DOMAIN])
@@ -267,6 +273,8 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
async def _shutdown(event: Event) -> None:
"""Shutdown event."""
await cloud.stop()
+ logging.root.removeHandler(log_handler)
+ del hass.data[DATA_CLOUD_LOG_HANDLER]
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _shutdown)
@@ -405,3 +413,19 @@ def _setup_services(hass: HomeAssistant, prefs: CloudPreferences) -> None:
async_register_admin_service(
hass, DOMAIN, SERVICE_REMOTE_DISCONNECT, _service_handler
)
+
+
+async def _setup_log_handler(hass: HomeAssistant) -> FixedSizeQueueLogHandler:
+ fmt = (
+ "%(asctime)s.%(msecs)03d %(levelname)s (%(threadName)s) [%(name)s] %(message)s"
+ )
+ handler = FixedSizeQueueLogHandler()
+ handler.setFormatter(logging.Formatter(fmt, datefmt=FORMAT_DATETIME))
+
+ integration = await async_get_integration(hass, DOMAIN)
+ loggers: set[str] = {integration.pkg_path, *(integration.loggers or [])}
+
+ for logger_name in loggers:
+ logging.getLogger(logger_name).addHandler(handler)
+
+ return handler
diff --git a/homeassistant/components/cloud/backup.py b/homeassistant/components/cloud/backup.py
index 9531604ccc7..b83c4725663 100644
--- a/homeassistant/components/cloud/backup.py
+++ b/homeassistant/components/cloud/backup.py
@@ -3,19 +3,27 @@
from __future__ import annotations
import asyncio
-import base64
from collections.abc import AsyncIterator, Callable, Coroutine, Mapping
-import hashlib
import logging
import random
-from typing import Any, Literal
+from typing import Any
from aiohttp import ClientError
from hass_nabucasa import Cloud, CloudError
from hass_nabucasa.api import CloudApiNonRetryableError
-from hass_nabucasa.cloud_api import async_files_delete_file, async_files_list
+from hass_nabucasa.cloud_api import (
+ FilesHandlerListEntry,
+ async_files_delete_file,
+ async_files_list,
+)
+from hass_nabucasa.files import FilesError, StorageType, calculate_b64md5
-from homeassistant.components.backup import AgentBackup, BackupAgent, BackupAgentError
+from homeassistant.components.backup import (
+ AgentBackup,
+ BackupAgent,
+ BackupAgentError,
+ BackupNotFound,
+)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.aiohttp_client import ChunkAsyncStreamIterator
from homeassistant.helpers.dispatcher import async_dispatcher_connect
@@ -24,20 +32,11 @@ from .client import CloudClient
from .const import DATA_CLOUD, DOMAIN, EVENT_CLOUD_EVENT
_LOGGER = logging.getLogger(__name__)
-_STORAGE_BACKUP: Literal["backup"] = "backup"
_RETRY_LIMIT = 5
_RETRY_SECONDS_MIN = 60
_RETRY_SECONDS_MAX = 600
-async def _b64md5(stream: AsyncIterator[bytes]) -> str:
- """Calculate the MD5 hash of a file."""
- file_hash = hashlib.md5()
- async for chunk in stream:
- file_hash.update(chunk)
- return base64.b64encode(file_hash.digest()).decode()
-
-
async def async_get_backup_agents(
hass: HomeAssistant,
**kwargs: Any,
@@ -86,11 +85,6 @@ class CloudBackupAgent(BackupAgent):
self._cloud = cloud
self._hass = hass
- @callback
- def _get_backup_filename(self) -> str:
- """Return the backup filename."""
- return f"{self._cloud.client.prefs.instance_id}.tar"
-
async def async_download_backup(
self,
backup_id: str,
@@ -101,13 +95,11 @@ class CloudBackupAgent(BackupAgent):
:param backup_id: The ID of the backup that was returned in async_list_backups.
:return: An async iterator that yields bytes.
"""
- if not await self.async_get_backup(backup_id):
- raise BackupAgentError("Backup not found")
-
+ backup = await self._async_get_backup(backup_id)
try:
content = await self._cloud.files.download(
- storage_type=_STORAGE_BACKUP,
- filename=self._get_backup_filename(),
+ storage_type=StorageType.BACKUP,
+ filename=backup["Key"],
)
except CloudError as err:
raise BackupAgentError(f"Failed to download backup: {err}") from err
@@ -129,16 +121,19 @@ class CloudBackupAgent(BackupAgent):
if not backup.protected:
raise BackupAgentError("Cloud backups must be protected")
- base64md5hash = await _b64md5(await open_stream())
- filename = self._get_backup_filename()
- metadata = backup.as_dict()
size = backup.size
+ try:
+ base64md5hash = await calculate_b64md5(open_stream, size)
+ except FilesError as err:
+ raise BackupAgentError(err) from err
+ filename = f"{self._cloud.client.prefs.instance_id}.tar"
+ metadata = backup.as_dict()
tries = 1
while tries <= _RETRY_LIMIT:
try:
await self._cloud.files.upload(
- storage_type=_STORAGE_BACKUP,
+ storage_type=StorageType.BACKUP,
open_stream=open_stream,
filename=filename,
base64md5hash=base64md5hash,
@@ -179,38 +174,48 @@ class CloudBackupAgent(BackupAgent):
:param backup_id: The ID of the backup that was returned in async_list_backups.
"""
- if not await self.async_get_backup(backup_id):
- return
-
+ backup = await self._async_get_backup(backup_id)
try:
await async_files_delete_file(
self._cloud,
- storage_type=_STORAGE_BACKUP,
- filename=self._get_backup_filename(),
+ storage_type=StorageType.BACKUP,
+ filename=backup["Key"],
)
except (ClientError, CloudError) as err:
raise BackupAgentError("Failed to delete backup") from err
async def async_list_backups(self, **kwargs: Any) -> list[AgentBackup]:
+ """List backups."""
+ backups = await self._async_list_backups()
+ return [AgentBackup.from_dict(backup["Metadata"]) for backup in backups]
+
+ async def _async_list_backups(self) -> list[FilesHandlerListEntry]:
"""List backups."""
try:
- backups = await async_files_list(self._cloud, storage_type=_STORAGE_BACKUP)
- _LOGGER.debug("Cloud backups: %s", backups)
+ backups = await async_files_list(
+ self._cloud, storage_type=StorageType.BACKUP
+ )
except (ClientError, CloudError) as err:
raise BackupAgentError("Failed to list backups") from err
- return [AgentBackup.from_dict(backup["Metadata"]) for backup in backups]
+ _LOGGER.debug("Cloud backups: %s", backups)
+ return backups
async def async_get_backup(
self,
backup_id: str,
**kwargs: Any,
- ) -> AgentBackup | None:
+ ) -> AgentBackup:
"""Return a backup."""
- backups = await self.async_list_backups()
+ backup = await self._async_get_backup(backup_id)
+ return AgentBackup.from_dict(backup["Metadata"])
+
+ async def _async_get_backup(self, backup_id: str) -> FilesHandlerListEntry:
+ """Return a backup."""
+ backups = await self._async_list_backups()
for backup in backups:
- if backup.backup_id == backup_id:
+ if backup["Metadata"]["backup_id"] == backup_id:
return backup
- return None
+ raise BackupNotFound(f"Backup {backup_id} not found")
diff --git a/homeassistant/components/cloud/binary_sensor.py b/homeassistant/components/cloud/binary_sensor.py
index 75cbd3c9f3d..0df13fe4c7b 100644
--- a/homeassistant/components/cloud/binary_sensor.py
+++ b/homeassistant/components/cloud/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .client import CloudClient
from .const import DATA_CLOUD, DISPATCHER_REMOTE_UPDATE
@@ -26,7 +26,7 @@ WAIT_UNTIL_CHANGE = 3
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Assistant Cloud binary sensors."""
cloud = hass.data[DATA_CLOUD]
diff --git a/homeassistant/components/cloud/const.py b/homeassistant/components/cloud/const.py
index 3883f19d1b7..e0c15c74cab 100644
--- a/homeassistant/components/cloud/const.py
+++ b/homeassistant/components/cloud/const.py
@@ -12,12 +12,14 @@ if TYPE_CHECKING:
from hass_nabucasa import Cloud
from .client import CloudClient
+ from .helpers import FixedSizeQueueLogHandler
DOMAIN = "cloud"
DATA_CLOUD: HassKey[Cloud[CloudClient]] = HassKey(DOMAIN)
DATA_PLATFORMS_SETUP: HassKey[dict[str, asyncio.Event]] = HassKey(
"cloud_platforms_setup"
)
+DATA_CLOUD_LOG_HANDLER: HassKey[FixedSizeQueueLogHandler] = HassKey("cloud_log_handler")
EVENT_CLOUD_EVENT = "cloud_event"
REQUEST_TIMEOUT = 10
diff --git a/homeassistant/components/cloud/helpers.py b/homeassistant/components/cloud/helpers.py
new file mode 100644
index 00000000000..7795a314fb7
--- /dev/null
+++ b/homeassistant/components/cloud/helpers.py
@@ -0,0 +1,31 @@
+"""Helpers for the cloud component."""
+
+from collections import deque
+import logging
+
+from homeassistant.core import HomeAssistant
+
+
+class FixedSizeQueueLogHandler(logging.Handler):
+ """Log handler to store messages, with auto rotation."""
+
+ MAX_RECORDS = 500
+
+ def __init__(self) -> None:
+ """Initialize a new LogHandler."""
+ super().__init__()
+ self._records: deque[logging.LogRecord] = deque(maxlen=self.MAX_RECORDS)
+
+ def emit(self, record: logging.LogRecord) -> None:
+ """Store log message."""
+ self._records.append(record)
+
+ async def get_logs(self, hass: HomeAssistant) -> list[str]:
+ """Get stored logs."""
+
+ def _get_logs() -> list[str]:
+ # copy the queue since it can mutate while iterating
+ records = self._records.copy()
+ return [self.format(record) for record in records]
+
+ return await hass.async_add_executor_job(_get_logs)
diff --git a/homeassistant/components/cloud/http_api.py b/homeassistant/components/cloud/http_api.py
index b1a845ef8b0..73952d80f6c 100644
--- a/homeassistant/components/cloud/http_api.py
+++ b/homeassistant/components/cloud/http_api.py
@@ -8,14 +8,15 @@ from contextlib import suppress
import dataclasses
from functools import wraps
from http import HTTPStatus
+import json
import logging
import time
-from typing import Any, Concatenate
+from typing import Any, Concatenate, cast
import aiohttp
from aiohttp import web
import attr
-from hass_nabucasa import Cloud, auth, thingtalk
+from hass_nabucasa import AlreadyConnectedError, Cloud, auth, thingtalk
from hass_nabucasa.const import STATE_DISCONNECTED
from hass_nabucasa.voice import TTS_VOICES
import voluptuous as vol
@@ -43,6 +44,7 @@ from .assist_pipeline import async_create_cloud_pipeline
from .client import CloudClient
from .const import (
DATA_CLOUD,
+ DATA_CLOUD_LOG_HANDLER,
EVENT_CLOUD_EVENT,
LOGIN_MFA_TIMEOUT,
PREF_ALEXA_REPORT_STATE,
@@ -63,7 +65,9 @@ from .subscription import async_subscription_info
_LOGGER = logging.getLogger(__name__)
-_CLOUD_ERRORS: dict[type[Exception], tuple[HTTPStatus, str]] = {
+_CLOUD_ERRORS: dict[
+ type[Exception], tuple[HTTPStatus, Callable[[Exception], str] | str]
+] = {
TimeoutError: (
HTTPStatus.BAD_GATEWAY,
"Unable to reach the Home Assistant cloud.",
@@ -132,6 +136,10 @@ def async_setup(hass: HomeAssistant) -> None:
HTTPStatus.BAD_REQUEST,
"Multi-factor authentication expired, or not started. Please try again.",
),
+ AlreadyConnectedError: (
+ HTTPStatus.CONFLICT,
+ lambda x: json.dumps(cast(AlreadyConnectedError, x).details),
+ ),
}
)
@@ -196,7 +204,11 @@ def _process_cloud_exception(exc: Exception, where: str) -> tuple[HTTPStatus, st
for err, value_info in _CLOUD_ERRORS.items():
if isinstance(exc, err):
- err_info = value_info
+ status, content = value_info
+ err_info = (
+ status,
+ content if isinstance(content, str) else content(exc),
+ )
break
if err_info is None:
@@ -239,6 +251,7 @@ class CloudLoginView(HomeAssistantView):
vol.All(
{
vol.Required("email"): str,
+ vol.Optional("check_connection", default=False): bool,
vol.Exclusive("password", "login"): str,
vol.Exclusive("code", "login"): str,
},
@@ -257,7 +270,11 @@ class CloudLoginView(HomeAssistantView):
code = data.get("code")
if email and password:
- await cloud.login(email, password)
+ await cloud.login(
+ email,
+ password,
+ check_connection=data["check_connection"],
+ )
else:
if (
@@ -269,7 +286,12 @@ class CloudLoginView(HomeAssistantView):
# Voluptuous should ensure that code is not None because password is
assert code is not None
- await cloud.login_verify_totp(email, code, self._mfa_tokens)
+ await cloud.login_verify_totp(
+ email,
+ code,
+ self._mfa_tokens,
+ check_connection=data["check_connection"],
+ )
self._mfa_tokens = {}
self._mfa_tokens_set_time = 0
@@ -397,8 +419,11 @@ class DownloadSupportPackageView(HomeAssistantView):
url = "/api/cloud/support_package"
name = "api:cloud:support_package"
- def _generate_markdown(
- self, hass_info: dict[str, Any], domains_info: dict[str, dict[str, str]]
+ async def _generate_markdown(
+ self,
+ hass: HomeAssistant,
+ hass_info: dict[str, Any],
+ domains_info: dict[str, dict[str, str]],
) -> str:
def get_domain_table_markdown(domain_info: dict[str, Any]) -> str:
if len(domain_info) == 0:
@@ -424,6 +449,17 @@ class DownloadSupportPackageView(HomeAssistantView):
"\n\n"
)
+ log_handler = hass.data[DATA_CLOUD_LOG_HANDLER]
+ logs = "\n".join(await log_handler.get_logs(hass))
+ markdown += (
+ "## Full logs\n\n"
+ "Logs
\n\n"
+ "```logs\n"
+ f"{logs}\n"
+ "```\n\n"
+ " \n"
+ )
+
return markdown
async def get(self, request: web.Request) -> web.Response:
@@ -433,7 +469,7 @@ class DownloadSupportPackageView(HomeAssistantView):
domain_health = await get_system_health_info(hass)
hass_info = domain_health.pop("homeassistant", {})
- markdown = self._generate_markdown(hass_info, domain_health)
+ markdown = await self._generate_markdown(hass, hass_info, domain_health)
return web.Response(
body=markdown,
diff --git a/homeassistant/components/cloud/manifest.json b/homeassistant/components/cloud/manifest.json
index 8e8ff4335db..7f448f2f614 100644
--- a/homeassistant/components/cloud/manifest.json
+++ b/homeassistant/components/cloud/manifest.json
@@ -12,7 +12,7 @@
"documentation": "https://www.home-assistant.io/integrations/cloud",
"integration_type": "system",
"iot_class": "cloud_push",
- "loggers": ["hass_nabucasa"],
- "requirements": ["hass-nabucasa==0.89.0"],
+ "loggers": ["acme", "hass_nabucasa", "snitun"],
+ "requirements": ["hass-nabucasa==0.94.0"],
"single_config_entry": true
}
diff --git a/homeassistant/components/cloud/stt.py b/homeassistant/components/cloud/stt.py
index b2154448d3a..df377c9a410 100644
--- a/homeassistant/components/cloud/stt.py
+++ b/homeassistant/components/cloud/stt.py
@@ -22,7 +22,7 @@ from homeassistant.components.stt import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.setup import async_when_setup
from .assist_pipeline import async_migrate_cloud_pipeline_engine
@@ -35,7 +35,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Home Assistant Cloud speech platform via config entry."""
stt_platform_loaded = hass.data[DATA_PLATFORMS_SETUP][Platform.STT]
diff --git a/homeassistant/components/cloud/tts.py b/homeassistant/components/cloud/tts.py
index 63f36554c65..f901adfa99e 100644
--- a/homeassistant/components/cloud/tts.py
+++ b/homeassistant/components/cloud/tts.py
@@ -23,7 +23,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PLATFORM, Platform
from homeassistant.core import HomeAssistant, async_get_hass, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.setup import async_when_setup
@@ -256,7 +256,7 @@ async def async_get_engine(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Home Assistant Cloud text-to-speech platform."""
tts_platform_loaded = hass.data[DATA_PLATFORMS_SETUP][Platform.TTS]
@@ -286,7 +286,7 @@ class CloudTTSEntity(TextToSpeechEntity):
return self._language
@property
- def default_options(self) -> dict[str, Any]:
+ def default_options(self) -> dict[str, str]:
"""Return a dict include default options."""
return {
ATTR_AUDIO_OUTPUT: AudioOutput.MP3,
@@ -363,7 +363,7 @@ class CloudTTSEntity(TextToSpeechEntity):
_LOGGER.error("Voice error: %s", err)
return (None, None)
- return (str(options[ATTR_AUDIO_OUTPUT].value), data)
+ return (options[ATTR_AUDIO_OUTPUT], data)
class CloudProvider(Provider):
@@ -404,7 +404,7 @@ class CloudProvider(Provider):
return [Voice(voice, voice) for voice in voices]
@property
- def default_options(self) -> dict[str, Any]:
+ def default_options(self) -> dict[str, str]:
"""Return a dict include default options."""
return {
ATTR_AUDIO_OUTPUT: AudioOutput.MP3,
@@ -444,7 +444,7 @@ class CloudProvider(Provider):
_LOGGER.error("Voice error: %s", err)
return (None, None)
- return (str(options[ATTR_AUDIO_OUTPUT].value), data)
+ return options[ATTR_AUDIO_OUTPUT], data
@callback
diff --git a/homeassistant/components/co2signal/__init__.py b/homeassistant/components/co2signal/__init__.py
index e84ba387194..612610eff43 100644
--- a/homeassistant/components/co2signal/__init__.py
+++ b/homeassistant/components/co2signal/__init__.py
@@ -4,23 +4,20 @@ from __future__ import annotations
from aioelectricitymaps import ElectricityMaps
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .coordinator import CO2SignalCoordinator
+from .coordinator import CO2SignalConfigEntry, CO2SignalCoordinator
PLATFORMS = [Platform.SENSOR]
-type CO2SignalConfigEntry = ConfigEntry[CO2SignalCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: CO2SignalConfigEntry) -> bool:
"""Set up CO2 Signal from a config entry."""
session = async_get_clientsession(hass)
coordinator = CO2SignalCoordinator(
- hass, ElectricityMaps(token=entry.data[CONF_API_KEY], session=session)
+ hass, entry, ElectricityMaps(token=entry.data[CONF_API_KEY], session=session)
)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/co2signal/config_flow.py b/homeassistant/components/co2signal/config_flow.py
index 530496811d9..00acd2829a6 100644
--- a/homeassistant/components/co2signal/config_flow.py
+++ b/homeassistant/components/co2signal/config_flow.py
@@ -3,11 +3,11 @@
from __future__ import annotations
from collections.abc import Mapping
+import logging
from typing import Any
from aioelectricitymaps import (
ElectricityMaps,
- ElectricityMapsError,
ElectricityMapsInvalidTokenError,
ElectricityMapsNoDataError,
)
@@ -36,6 +36,8 @@ TYPE_USE_HOME = "use_home_location"
TYPE_SPECIFY_COORDINATES = "specify_coordinates"
TYPE_SPECIFY_COUNTRY = "specify_country_code"
+_LOGGER = logging.getLogger(__name__)
+
class ElectricityMapsConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a config flow for Co2signal."""
@@ -158,7 +160,8 @@ class ElectricityMapsConfigFlow(ConfigFlow, domain=DOMAIN):
errors["base"] = "invalid_auth"
except ElectricityMapsNoDataError:
errors["base"] = "no_data"
- except ElectricityMapsError:
+ except Exception:
+ _LOGGER.exception("Unexpected error occurred while checking API key")
errors["base"] = "unknown"
else:
if self.source == SOURCE_REAUTH:
diff --git a/homeassistant/components/co2signal/coordinator.py b/homeassistant/components/co2signal/coordinator.py
index 475ebd1225d..be2036292e3 100644
--- a/homeassistant/components/co2signal/coordinator.py
+++ b/homeassistant/components/co2signal/coordinator.py
@@ -22,16 +22,27 @@ from .helpers import fetch_latest_carbon_intensity
_LOGGER = logging.getLogger(__name__)
+type CO2SignalConfigEntry = ConfigEntry[CO2SignalCoordinator]
+
class CO2SignalCoordinator(DataUpdateCoordinator[CarbonIntensityResponse]):
"""Data update coordinator."""
- config_entry: ConfigEntry
+ config_entry: CO2SignalConfigEntry
- def __init__(self, hass: HomeAssistant, client: ElectricityMaps) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: CO2SignalConfigEntry,
+ client: ElectricityMaps,
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=timedelta(minutes=15)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=timedelta(minutes=15),
)
self.client = client
diff --git a/homeassistant/components/co2signal/diagnostics.py b/homeassistant/components/co2signal/diagnostics.py
index a071950440f..840ba759a7b 100644
--- a/homeassistant/components/co2signal/diagnostics.py
+++ b/homeassistant/components/co2signal/diagnostics.py
@@ -9,7 +9,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant
-from . import CO2SignalConfigEntry
+from .coordinator import CO2SignalConfigEntry
TO_REDACT = {CONF_API_KEY}
diff --git a/homeassistant/components/co2signal/sensor.py b/homeassistant/components/co2signal/sensor.py
index 1b964edf591..a8e962532b8 100644
--- a/homeassistant/components/co2signal/sensor.py
+++ b/homeassistant/components/co2signal/sensor.py
@@ -15,12 +15,11 @@ from homeassistant.components.sensor import (
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import CO2SignalConfigEntry
from .const import ATTRIBUTION, DOMAIN
-from .coordinator import CO2SignalCoordinator
+from .coordinator import CO2SignalConfigEntry, CO2SignalCoordinator
@dataclass(frozen=True, kw_only=True)
@@ -55,7 +54,7 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: CO2SignalConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the CO2signal sensor."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/coinbase/__init__.py b/homeassistant/components/coinbase/__init__.py
index a29154d9c1b..317759f820d 100644
--- a/homeassistant/components/coinbase/__init__.py
+++ b/homeassistant/components/coinbase/__init__.py
@@ -140,8 +140,10 @@ def get_accounts(client, version):
API_ACCOUNT_ID: account[API_V3_ACCOUNT_ID],
API_ACCOUNT_NAME: account[API_ACCOUNT_NAME],
API_ACCOUNT_CURRENCY: account[API_ACCOUNT_CURRENCY],
- API_ACCOUNT_AMOUNT: account[API_ACCOUNT_AVALIABLE][API_ACCOUNT_VALUE]
- + account[API_ACCOUNT_HOLD][API_ACCOUNT_VALUE],
+ API_ACCOUNT_AMOUNT: (
+ float(account[API_ACCOUNT_AVALIABLE][API_ACCOUNT_VALUE])
+ + float(account[API_ACCOUNT_HOLD][API_ACCOUNT_VALUE])
+ ),
ACCOUNT_IS_VAULT: account[API_RESOURCE_TYPE] == API_V3_TYPE_VAULT,
}
for account in accounts
diff --git a/homeassistant/components/coinbase/sensor.py b/homeassistant/components/coinbase/sensor.py
index 37509160247..578877e7d90 100644
--- a/homeassistant/components/coinbase/sensor.py
+++ b/homeassistant/components/coinbase/sensor.py
@@ -7,7 +7,7 @@ import logging
from homeassistant.components.sensor import SensorEntity, SensorStateClass
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import CoinbaseConfigEntry, CoinbaseData
from .const import (
@@ -45,7 +45,7 @@ ATTRIBUTION = "Data provided by coinbase.com"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CoinbaseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Coinbase sensor platform."""
instance = config_entry.runtime_data
diff --git a/homeassistant/components/comelit/alarm_control_panel.py b/homeassistant/components/comelit/alarm_control_panel.py
index f694c2b392b..5ecc9a63599 100644
--- a/homeassistant/components/comelit/alarm_control_panel.py
+++ b/homeassistant/components/comelit/alarm_control_panel.py
@@ -6,7 +6,7 @@ import logging
from typing import cast
from aiocomelit.api import ComelitVedoAreaObject
-from aiocomelit.const import ALARM_AREAS, AlarmAreaState
+from aiocomelit.const import AlarmAreaState
from homeassistant.components.alarm_control_panel import (
AlarmControlPanelEntity,
@@ -15,11 +15,14 @@ from homeassistant.components.alarm_control_panel import (
CodeFormat,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import ComelitConfigEntry, ComelitVedoSystem
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
_LOGGER = logging.getLogger(__name__)
AWAY = "away"
@@ -48,7 +51,7 @@ ALARM_AREA_ARMED_STATUS: dict[str, int] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Comelit VEDO system alarm control panel devices."""
@@ -56,7 +59,7 @@ async def async_setup_entry(
async_add_entities(
ComelitAlarmEntity(coordinator, device, config_entry.entry_id)
- for device in coordinator.data[ALARM_AREAS].values()
+ for device in coordinator.data["alarm_areas"].values()
)
@@ -92,7 +95,7 @@ class ComelitAlarmEntity(CoordinatorEntity[ComelitVedoSystem], AlarmControlPanel
@property
def _area(self) -> ComelitVedoAreaObject:
"""Return area object."""
- return self.coordinator.data[ALARM_AREAS][self._area_index]
+ return self.coordinator.data["alarm_areas"][self._area_index]
@property
def available(self) -> bool:
diff --git a/homeassistant/components/comelit/binary_sensor.py b/homeassistant/components/comelit/binary_sensor.py
index fa51e0b1fda..dfa6d3e97f3 100644
--- a/homeassistant/components/comelit/binary_sensor.py
+++ b/homeassistant/components/comelit/binary_sensor.py
@@ -5,23 +5,25 @@ from __future__ import annotations
from typing import cast
from aiocomelit import ComelitVedoZoneObject
-from aiocomelit.const import ALARM_ZONES
from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
BinarySensorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import ComelitConfigEntry, ComelitVedoSystem
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit VEDO presence sensors."""
@@ -29,7 +31,7 @@ async def async_setup_entry(
async_add_entities(
ComelitVedoBinarySensorEntity(coordinator, device, config_entry.entry_id)
- for device in coordinator.data[ALARM_ZONES].values()
+ for device in coordinator.data["alarm_zones"].values()
)
@@ -49,7 +51,7 @@ class ComelitVedoBinarySensorEntity(
) -> None:
"""Init sensor entity."""
self._api = coordinator.api
- self._zone = zone
+ self._zone_index = zone.index
super().__init__(coordinator)
# Use config_entry.entry_id as base for unique_id
# because no serial number or mac is available
@@ -59,4 +61,6 @@ class ComelitVedoBinarySensorEntity(
@property
def is_on(self) -> bool:
"""Presence detected."""
- return self.coordinator.data[ALARM_ZONES][self._zone.index].status_api == "0001"
+ return (
+ self.coordinator.data["alarm_zones"][self._zone_index].status_api == "0001"
+ )
diff --git a/homeassistant/components/comelit/climate.py b/homeassistant/components/comelit/climate.py
index 1baa777bf99..3ec79001d55 100644
--- a/homeassistant/components/comelit/climate.py
+++ b/homeassistant/components/comelit/climate.py
@@ -3,7 +3,7 @@
from __future__ import annotations
from enum import StrEnum
-from typing import Any, cast
+from typing import Any, TypedDict, cast
from aiocomelit import ComelitSerialBridgeObject
from aiocomelit.const import CLIMATE
@@ -16,12 +16,17 @@ from homeassistant.components.climate import (
UnitOfTemperature,
)
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_TENTHS
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
+from .const import DOMAIN
from .coordinator import ComelitConfigEntry, ComelitSerialBridge
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
class ClimaComelitMode(StrEnum):
"""Serial Bridge clima modes."""
@@ -42,22 +47,23 @@ class ClimaComelitCommand(StrEnum):
AUTO = "auto"
-API_STATUS: dict[str, dict[str, Any]] = {
- ClimaComelitMode.OFF: {
- "action": "off",
- "hvac_mode": HVACMode.OFF,
- "hvac_action": HVACAction.OFF,
- },
- ClimaComelitMode.LOWER: {
- "action": "lower",
- "hvac_mode": HVACMode.COOL,
- "hvac_action": HVACAction.COOLING,
- },
- ClimaComelitMode.UPPER: {
- "action": "upper",
- "hvac_mode": HVACMode.HEAT,
- "hvac_action": HVACAction.HEATING,
- },
+class ClimaComelitApiStatus(TypedDict):
+ """Comelit Clima API status."""
+
+ hvac_mode: HVACMode
+ hvac_action: HVACAction
+
+
+API_STATUS: dict[str, ClimaComelitApiStatus] = {
+ ClimaComelitMode.OFF: ClimaComelitApiStatus(
+ hvac_mode=HVACMode.OFF, hvac_action=HVACAction.OFF
+ ),
+ ClimaComelitMode.LOWER: ClimaComelitApiStatus(
+ hvac_mode=HVACMode.COOL, hvac_action=HVACAction.COOLING
+ ),
+ ClimaComelitMode.UPPER: ClimaComelitApiStatus(
+ hvac_mode=HVACMode.HEAT, hvac_action=HVACAction.HEATING
+ ),
}
MODE_TO_ACTION: dict[HVACMode, ClimaComelitCommand] = {
@@ -71,7 +77,7 @@ MODE_TO_ACTION: dict[HVACMode, ClimaComelitCommand] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit climates."""
@@ -113,70 +119,50 @@ class ComelitClimateEntity(CoordinatorEntity[ComelitSerialBridge], ClimateEntity
# because no serial number or mac is available
self._attr_unique_id = f"{config_entry_entry_id}-{device.index}"
self._attr_device_info = coordinator.platform_device_info(device, device.type)
+ self._update_attributes()
+
+ def _update_attributes(self) -> None:
+ """Update class attributes."""
+ device = self.coordinator.data[CLIMATE][self._device.index]
+ if not isinstance(device.val, list):
+ raise HomeAssistantError(
+ translation_domain=DOMAIN, translation_key="invalid_clima_data"
+ )
- @property
- def _clima(self) -> list[Any]:
- """Return clima device data."""
# CLIMATE has a 2 item tuple:
# - first for Clima
# - second for Humidifier
- return self.coordinator.data[CLIMATE][self._device.index].val[0]
+ values = device.val[0]
- @property
- def _api_mode(self) -> str:
- """Return device mode."""
- # Values from API: "O", "L", "U"
- return self._clima[2]
+ _active = values[1]
+ _mode = values[2] # Values from API: "O", "L", "U"
+ _automatic = values[3] == ClimaComelitMode.AUTO
- @property
- def _api_active(self) -> bool:
- "Return device active/idle."
- return self._clima[1]
+ self._attr_current_temperature = values[0] / 10
- @property
- def _api_automatic(self) -> bool:
- """Return device in automatic/manual mode."""
- return self._clima[3] == ClimaComelitMode.AUTO
+ self._attr_hvac_action = None
+ if _mode == ClimaComelitMode.OFF:
+ self._attr_hvac_action = HVACAction.OFF
+ if not _active:
+ self._attr_hvac_action = HVACAction.IDLE
+ if _mode in API_STATUS:
+ self._attr_hvac_action = API_STATUS[_mode]["hvac_action"]
- @property
- def target_temperature(self) -> float:
- """Set target temperature."""
- return self._clima[4] / 10
+ self._attr_hvac_mode = None
+ if _mode == ClimaComelitMode.OFF:
+ self._attr_hvac_mode = HVACMode.OFF
+ if _automatic:
+ self._attr_hvac_mode = HVACMode.AUTO
+ if _mode in API_STATUS:
+ self._attr_hvac_mode = API_STATUS[_mode]["hvac_mode"]
- @property
- def current_temperature(self) -> float:
- """Return current temperature."""
- return self._clima[0] / 10
+ self._attr_target_temperature = values[4] / 10
- @property
- def hvac_mode(self) -> HVACMode | None:
- """HVAC current mode."""
-
- if self._api_mode == ClimaComelitMode.OFF:
- return HVACMode.OFF
-
- if self._api_automatic:
- return HVACMode.AUTO
-
- if self._api_mode in API_STATUS:
- return API_STATUS[self._api_mode]["hvac_mode"]
-
- return None
-
- @property
- def hvac_action(self) -> HVACAction | None:
- """HVAC current action."""
-
- if self._api_mode == ClimaComelitMode.OFF:
- return HVACAction.OFF
-
- if not self._api_active:
- return HVACAction.IDLE
-
- if self._api_mode in API_STATUS:
- return API_STATUS[self._api_mode]["hvac_action"]
-
- return None
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
+ self._update_attributes()
+ super()._handle_coordinator_update()
async def async_set_temperature(self, **kwargs: Any) -> None:
"""Set new target temperature."""
@@ -191,6 +177,8 @@ class ComelitClimateEntity(CoordinatorEntity[ComelitSerialBridge], ClimateEntity
await self.coordinator.api.set_clima_status(
self._device.index, ClimaComelitCommand.SET, target_temp
)
+ self._attr_target_temperature = target_temp
+ self.async_write_ha_state()
async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
"""Set hvac mode."""
@@ -202,3 +190,5 @@ class ComelitClimateEntity(CoordinatorEntity[ComelitSerialBridge], ClimateEntity
await self.coordinator.api.set_clima_status(
self._device.index, MODE_TO_ACTION[hvac_mode]
)
+ self._attr_hvac_mode = hvac_mode
+ self.async_write_ha_state()
diff --git a/homeassistant/components/comelit/coordinator.py b/homeassistant/components/comelit/coordinator.py
index fcb149b21d6..b3be3a47825 100644
--- a/homeassistant/components/comelit/coordinator.py
+++ b/homeassistant/components/comelit/coordinator.py
@@ -2,18 +2,19 @@
from abc import abstractmethod
from datetime import timedelta
-from typing import Any
+from typing import TypeVar
-from aiocomelit import (
+from aiocomelit.api import (
+ AlarmDataObject,
+ ComelitCommonApi,
ComeliteSerialBridgeApi,
ComelitSerialBridgeObject,
ComelitVedoApi,
ComelitVedoAreaObject,
ComelitVedoZoneObject,
- exceptions,
)
-from aiocomelit.api import ComelitCommonApi
from aiocomelit.const import BRIDGE, VEDO
+from aiocomelit.exceptions import CannotAuthenticate, CannotConnect, CannotRetrieveData
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
@@ -26,7 +27,13 @@ from .const import _LOGGER, DOMAIN
type ComelitConfigEntry = ConfigEntry[ComelitBaseCoordinator]
-class ComelitBaseCoordinator(DataUpdateCoordinator[dict[str, Any]]):
+T = TypeVar(
+ "T",
+ bound=dict[str, dict[int, ComelitSerialBridgeObject]] | AlarmDataObject,
+)
+
+
+class ComelitBaseCoordinator(DataUpdateCoordinator[T]):
"""Base coordinator for Comelit Devices."""
_hw_version: str
@@ -81,23 +88,25 @@ class ComelitBaseCoordinator(DataUpdateCoordinator[dict[str, Any]]):
hw_version=self._hw_version,
)
- async def _async_update_data(self) -> dict[str, Any]:
+ async def _async_update_data(self) -> T:
"""Update device data."""
_LOGGER.debug("Polling Comelit %s host: %s", self._device, self._host)
try:
await self.api.login()
return await self._async_update_system_data()
- except (exceptions.CannotConnect, exceptions.CannotRetrieveData) as err:
+ except (CannotConnect, CannotRetrieveData) as err:
raise UpdateFailed(repr(err)) from err
- except exceptions.CannotAuthenticate as err:
+ except CannotAuthenticate as err:
raise ConfigEntryAuthFailed from err
@abstractmethod
- async def _async_update_system_data(self) -> dict[str, Any]:
+ async def _async_update_system_data(self) -> T:
"""Class method for updating data."""
-class ComelitSerialBridge(ComelitBaseCoordinator):
+class ComelitSerialBridge(
+ ComelitBaseCoordinator[dict[str, dict[int, ComelitSerialBridgeObject]]]
+):
"""Queries Comelit Serial Bridge."""
_hw_version = "20003101"
@@ -115,12 +124,14 @@ class ComelitSerialBridge(ComelitBaseCoordinator):
self.api = ComeliteSerialBridgeApi(host, port, pin)
super().__init__(hass, entry, BRIDGE, host)
- async def _async_update_system_data(self) -> dict[str, Any]:
+ async def _async_update_system_data(
+ self,
+ ) -> dict[str, dict[int, ComelitSerialBridgeObject]]:
"""Specific method for updating data."""
return await self.api.get_all_devices()
-class ComelitVedoSystem(ComelitBaseCoordinator):
+class ComelitVedoSystem(ComelitBaseCoordinator[AlarmDataObject]):
"""Queries Comelit VEDO system."""
_hw_version = "VEDO IP"
@@ -138,6 +149,8 @@ class ComelitVedoSystem(ComelitBaseCoordinator):
self.api = ComelitVedoApi(host, port, pin)
super().__init__(hass, entry, VEDO, host)
- async def _async_update_system_data(self) -> dict[str, Any]:
+ async def _async_update_system_data(
+ self,
+ ) -> AlarmDataObject:
"""Specific method for updating data."""
return await self.api.get_all_areas_and_zones()
diff --git a/homeassistant/components/comelit/cover.py b/homeassistant/components/comelit/cover.py
index abb84824621..9bcf52ac111 100644
--- a/homeassistant/components/comelit/cover.py
+++ b/homeassistant/components/comelit/cover.py
@@ -9,17 +9,20 @@ from aiocomelit.const import COVER, STATE_COVER, STATE_OFF, STATE_ON
from homeassistant.components.cover import CoverDeviceClass, CoverEntity, CoverState
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import ComelitConfigEntry, ComelitSerialBridge
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit covers."""
diff --git a/homeassistant/components/comelit/humidifier.py b/homeassistant/components/comelit/humidifier.py
index d8058074c16..ad8f49ed5e2 100644
--- a/homeassistant/components/comelit/humidifier.py
+++ b/homeassistant/components/comelit/humidifier.py
@@ -16,14 +16,17 @@ from homeassistant.components.humidifier import (
HumidifierEntity,
HumidifierEntityFeature,
)
-from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
from .coordinator import ComelitConfigEntry, ComelitSerialBridge
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
class HumidifierComelitMode(StrEnum):
"""Serial Bridge humidifier modes."""
@@ -55,7 +58,7 @@ MODE_TO_ACTION: dict[str, HumidifierComelitCommand] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit humidifiers."""
@@ -121,62 +124,41 @@ class ComelitHumidifierEntity(CoordinatorEntity[ComelitSerialBridge], Humidifier
self._active_mode = active_mode
self._active_action = active_action
self._set_command = set_command
+ self._update_attributes()
+
+ def _update_attributes(self) -> None:
+ """Update class attributes."""
+ device = self.coordinator.data[CLIMATE][self._device.index]
+ if not isinstance(device.val, list):
+ raise HomeAssistantError(
+ translation_domain=DOMAIN, translation_key="invalid_clima_data"
+ )
- @property
- def _humidifier(self) -> list[Any]:
- """Return humidifier device data."""
# CLIMATE has a 2 item tuple:
# - first for Clima
# - second for Humidifier
- return self.coordinator.data[CLIMATE][self._device.index].val[1]
+ values = device.val[1]
- @property
- def _api_mode(self) -> str:
- """Return device mode."""
- # Values from API: "O", "L", "U"
- return self._humidifier[2]
+ _active = values[1]
+ _mode = values[2] # Values from API: "O", "L", "U"
+ _automatic = values[3] == HumidifierComelitMode.AUTO
- @property
- def _api_active(self) -> bool:
- "Return device active/idle."
- return self._humidifier[1]
+ self._attr_action = HumidifierAction.IDLE
+ if _mode == HumidifierComelitMode.OFF:
+ self._attr_action = HumidifierAction.OFF
+ if _active and _mode == self._active_mode:
+ self._attr_action = self._active_action
- @property
- def _api_automatic(self) -> bool:
- """Return device in automatic/manual mode."""
- return self._humidifier[3] == HumidifierComelitMode.AUTO
+ self._attr_current_humidity = values[0] / 10
+ self._attr_is_on = _mode == self._active_mode
+ self._attr_mode = MODE_AUTO if _automatic else MODE_NORMAL
+ self._attr_target_humidity = values[4] / 10
- @property
- def target_humidity(self) -> float:
- """Set target humidity."""
- return self._humidifier[4] / 10
-
- @property
- def current_humidity(self) -> float:
- """Return current humidity."""
- return self._humidifier[0] / 10
-
- @property
- def is_on(self) -> bool | None:
- """Return true is humidifier is on."""
- return self._api_mode == self._active_mode
-
- @property
- def mode(self) -> str | None:
- """Return current mode."""
- return MODE_AUTO if self._api_automatic else MODE_NORMAL
-
- @property
- def action(self) -> HumidifierAction | None:
- """Return current action."""
-
- if self._api_mode == HumidifierComelitMode.OFF:
- return HumidifierAction.OFF
-
- if self._api_active and self._api_mode == self._active_mode:
- return self._active_action
-
- return HumidifierAction.IDLE
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
+ self._update_attributes()
+ super()._handle_coordinator_update()
async def async_set_humidity(self, humidity: int) -> None:
"""Set new target humidity."""
@@ -192,12 +174,16 @@ class ComelitHumidifierEntity(CoordinatorEntity[ComelitSerialBridge], Humidifier
await self.coordinator.api.set_humidity_status(
self._device.index, HumidifierComelitCommand.SET, humidity
)
+ self._attr_target_humidity = humidity
+ self.async_write_ha_state()
async def async_set_mode(self, mode: str) -> None:
"""Set humidifier mode."""
await self.coordinator.api.set_humidity_status(
self._device.index, MODE_TO_ACTION[mode]
)
+ self._attr_mode = mode
+ self.async_write_ha_state()
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn on."""
diff --git a/homeassistant/components/comelit/light.py b/homeassistant/components/comelit/light.py
index 9736c9ac2a0..09180d628a6 100644
--- a/homeassistant/components/comelit/light.py
+++ b/homeassistant/components/comelit/light.py
@@ -9,16 +9,19 @@ from aiocomelit.const import LIGHT, STATE_OFF, STATE_ON
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import ComelitConfigEntry, ComelitSerialBridge
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit lights."""
diff --git a/homeassistant/components/comelit/manifest.json b/homeassistant/components/comelit/manifest.json
index 238dede8546..8836af4e8dd 100644
--- a/homeassistant/components/comelit/manifest.json
+++ b/homeassistant/components/comelit/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "hub",
"iot_class": "local_polling",
"loggers": ["aiocomelit"],
- "requirements": ["aiocomelit==0.10.1"]
+ "requirements": ["aiocomelit==0.11.2"]
}
diff --git a/homeassistant/components/comelit/sensor.py b/homeassistant/components/comelit/sensor.py
index efb2418244e..c93ccd30eb6 100644
--- a/homeassistant/components/comelit/sensor.py
+++ b/homeassistant/components/comelit/sensor.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from typing import Final, cast
from aiocomelit import ComelitSerialBridgeObject, ComelitVedoZoneObject
-from aiocomelit.const import ALARM_ZONES, BRIDGE, OTHER, AlarmZoneState
+from aiocomelit.const import BRIDGE, OTHER, AlarmZoneState
from homeassistant.components.sensor import (
SensorDeviceClass,
@@ -14,12 +14,15 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import CONF_TYPE, UnitOfPower
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import ComelitConfigEntry, ComelitSerialBridge, ComelitVedoSystem
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
SENSOR_BRIDGE_TYPES: Final = (
SensorEntityDescription(
key="power",
@@ -42,7 +45,7 @@ SENSOR_VEDO_TYPES: Final = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit sensors."""
@@ -55,7 +58,7 @@ async def async_setup_entry(
async def async_setup_bridge_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit Bridge sensors."""
@@ -75,14 +78,14 @@ async def async_setup_bridge_entry(
async def async_setup_vedo_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit VEDO sensors."""
coordinator = cast(ComelitVedoSystem, config_entry.runtime_data)
entities: list[ComelitVedoSensorEntity] = []
- for device in coordinator.data[ALARM_ZONES].values():
+ for device in coordinator.data["alarm_zones"].values():
entities.extend(
ComelitVedoSensorEntity(
coordinator, device, config_entry.entry_id, sensor_desc
@@ -119,9 +122,12 @@ class ComelitBridgeSensorEntity(CoordinatorEntity[ComelitSerialBridge], SensorEn
@property
def native_value(self) -> StateType:
"""Sensor value."""
- return getattr(
- self.coordinator.data[OTHER][self._device.index],
- self.entity_description.key,
+ return cast(
+ StateType,
+ getattr(
+ self.coordinator.data[OTHER][self._device.index],
+ self.entity_description.key,
+ ),
)
@@ -139,7 +145,7 @@ class ComelitVedoSensorEntity(CoordinatorEntity[ComelitVedoSystem], SensorEntity
) -> None:
"""Init sensor entity."""
self._api = coordinator.api
- self._zone = zone
+ self._zone_index = zone.index
super().__init__(coordinator)
# Use config_entry.entry_id as base for unique_id
# because no serial number or mac is available
@@ -151,7 +157,7 @@ class ComelitVedoSensorEntity(CoordinatorEntity[ComelitVedoSystem], SensorEntity
@property
def _zone_object(self) -> ComelitVedoZoneObject:
"""Zone object."""
- return self.coordinator.data[ALARM_ZONES][self._zone.index]
+ return self.coordinator.data["alarm_zones"][self._zone_index]
@property
def available(self) -> bool:
@@ -164,4 +170,4 @@ class ComelitVedoSensorEntity(CoordinatorEntity[ComelitVedoSystem], SensorEntity
if (status := self._zone_object.human_status) == AlarmZoneState.UNKNOWN:
return None
- return status.value
+ return cast(str, status.value)
diff --git a/homeassistant/components/comelit/strings.json b/homeassistant/components/comelit/strings.json
index 14d947c7323..496d62655a9 100644
--- a/homeassistant/components/comelit/strings.json
+++ b/homeassistant/components/comelit/strings.json
@@ -3,19 +3,25 @@
"flow_title": "{host}",
"step": {
"reauth_confirm": {
- "description": "Please enter the correct PIN for {host}",
"data": {
"pin": "[%key:common::config_flow::data::pin%]"
+ },
+ "data_description": {
+ "pin": "The PIN of your Comelit device."
}
},
"user": {
"data": {
"host": "[%key:common::config_flow::data::host%]",
"port": "[%key:common::config_flow::data::port%]",
- "pin": "[%key:common::config_flow::data::pin%]"
+ "pin": "[%key:common::config_flow::data::pin%]",
+ "type": "Device type"
},
"data_description": {
- "host": "The hostname or IP address of your Comelit device."
+ "host": "The hostname or IP address of your Comelit device.",
+ "port": "The port of your Comelit device.",
+ "pin": "[%key:component::comelit::config::step::reauth_confirm::data_description::pin%]",
+ "type": "The type of your Comelit device."
}
}
},
@@ -58,6 +64,9 @@
"exceptions": {
"humidity_while_off": {
"message": "Cannot change humidity while off"
+ },
+ "invalid_clima_data": {
+ "message": "Invalid 'clima' data"
}
}
}
diff --git a/homeassistant/components/comelit/switch.py b/homeassistant/components/comelit/switch.py
index 26d3b81ebde..db89bd082f6 100644
--- a/homeassistant/components/comelit/switch.py
+++ b/homeassistant/components/comelit/switch.py
@@ -9,16 +9,19 @@ from aiocomelit.const import IRRIGATION, OTHER, STATE_OFF, STATE_ON
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import ComelitConfigEntry, ComelitSerialBridge
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ComelitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Comelit switches."""
@@ -77,7 +80,4 @@ class ComelitSwitchEntity(CoordinatorEntity[ComelitSerialBridge], SwitchEntity):
@property
def is_on(self) -> bool:
"""Return True if switch is on."""
- return (
- self.coordinator.data[self._device.type][self._device.index].status
- == STATE_ON
- )
+ return self.coordinator.data[OTHER][self._device.index].status == STATE_ON
diff --git a/homeassistant/components/config/config_entries.py b/homeassistant/components/config/config_entries.py
index 52e3346002e..74c9b5a9d0c 100644
--- a/homeassistant/components/config/config_entries.py
+++ b/homeassistant/components/config/config_entries.py
@@ -46,6 +46,13 @@ def async_setup(hass: HomeAssistant) -> bool:
hass.http.register_view(OptionManagerFlowIndexView(hass.config_entries.options))
hass.http.register_view(OptionManagerFlowResourceView(hass.config_entries.options))
+ hass.http.register_view(
+ SubentryManagerFlowIndexView(hass.config_entries.subentries)
+ )
+ hass.http.register_view(
+ SubentryManagerFlowResourceView(hass.config_entries.subentries)
+ )
+
websocket_api.async_register_command(hass, config_entries_get)
websocket_api.async_register_command(hass, config_entry_disable)
websocket_api.async_register_command(hass, config_entry_get_single)
@@ -54,6 +61,9 @@ def async_setup(hass: HomeAssistant) -> bool:
websocket_api.async_register_command(hass, config_entries_progress)
websocket_api.async_register_command(hass, ignore_config_flow)
+ websocket_api.async_register_command(hass, config_subentry_delete)
+ websocket_api.async_register_command(hass, config_subentry_list)
+
return True
@@ -285,6 +295,66 @@ class OptionManagerFlowResourceView(
return await super().post(request, flow_id)
+class SubentryManagerFlowIndexView(
+ FlowManagerIndexView[config_entries.ConfigSubentryFlowManager]
+):
+ """View to create subentry flows."""
+
+ url = "/api/config/config_entries/subentries/flow"
+ name = "api:config:config_entries:subentries:flow"
+
+ @require_admin(
+ error=Unauthorized(perm_category=CAT_CONFIG_ENTRIES, permission=POLICY_EDIT)
+ )
+ @RequestDataValidator(
+ vol.Schema(
+ {
+ vol.Required("handler"): vol.All(vol.Coerce(tuple), (str, str)),
+ vol.Optional("show_advanced_options", default=False): cv.boolean,
+ },
+ extra=vol.ALLOW_EXTRA,
+ )
+ )
+ async def post(self, request: web.Request, data: dict[str, Any]) -> web.Response:
+ """Handle a POST request.
+
+ handler in request is [entry_id, subentry_type].
+ """
+ return await super()._post_impl(request, data)
+
+ def get_context(self, data: dict[str, Any]) -> dict[str, Any]:
+ """Return context."""
+ context = super().get_context(data)
+ context["source"] = config_entries.SOURCE_USER
+ if subentry_id := data.get("subentry_id"):
+ context["source"] = config_entries.SOURCE_RECONFIGURE
+ context["subentry_id"] = subentry_id
+ return context
+
+
+class SubentryManagerFlowResourceView(
+ FlowManagerResourceView[config_entries.ConfigSubentryFlowManager]
+):
+ """View to interact with the subentry flow manager."""
+
+ url = "/api/config/config_entries/subentries/flow/{flow_id}"
+ name = "api:config:config_entries:subentries:flow:resource"
+
+ @require_admin(
+ error=Unauthorized(perm_category=CAT_CONFIG_ENTRIES, permission=POLICY_EDIT)
+ )
+ async def get(self, request: web.Request, /, flow_id: str) -> web.Response:
+ """Get the current state of a data_entry_flow."""
+ return await super().get(request, flow_id)
+
+ @require_admin(
+ error=Unauthorized(perm_category=CAT_CONFIG_ENTRIES, permission=POLICY_EDIT)
+ )
+ async def post(self, request: web.Request, flow_id: str) -> web.Response:
+ """Handle a POST request."""
+ return await super().post(request, flow_id)
+
+
@websocket_api.require_admin
@websocket_api.websocket_command({"type": "config_entries/flow/progress"})
def config_entries_progress(
@@ -589,3 +659,63 @@ async def _async_matching_config_entries_json_fragments(
)
or (filter_is_not_helper and entry.domain not in integrations)
]
+
+
+@websocket_api.require_admin
+@websocket_api.websocket_command(
+ {
+ "type": "config_entries/subentries/list",
+ "entry_id": str,
+ }
+)
+@websocket_api.async_response
+async def config_subentry_list(
+ hass: HomeAssistant,
+ connection: websocket_api.ActiveConnection,
+ msg: dict[str, Any],
+) -> None:
+ """List subentries of a config entry."""
+ entry = get_entry(hass, connection, msg["entry_id"], msg["id"])
+ if entry is None:
+ return
+
+ result = [
+ {
+ "subentry_id": subentry.subentry_id,
+ "subentry_type": subentry.subentry_type,
+ "title": subentry.title,
+ "unique_id": subentry.unique_id,
+ }
+ for subentry in entry.subentries.values()
+ ]
+ connection.send_result(msg["id"], result)
+
+
+@websocket_api.require_admin
+@websocket_api.websocket_command(
+ {
+ "type": "config_entries/subentries/delete",
+ "entry_id": str,
+ "subentry_id": str,
+ }
+)
+@websocket_api.async_response
+async def config_subentry_delete(
+ hass: HomeAssistant,
+ connection: websocket_api.ActiveConnection,
+ msg: dict[str, Any],
+) -> None:
+ """Delete a subentry of a config entry."""
+ entry = get_entry(hass, connection, msg["entry_id"], msg["id"])
+ if entry is None:
+ return
+
+ try:
+ hass.config_entries.async_remove_subentry(entry, msg["subentry_id"])
+ except config_entries.UnknownSubEntry:
+ connection.send_error(
+ msg["id"], websocket_api.const.ERR_NOT_FOUND, "Config subentry not found"
+ )
+ return
+
+ connection.send_result(msg["id"])
diff --git a/homeassistant/components/control4/light.py b/homeassistant/components/control4/light.py
index cedfbeb49c3..d2d0f85f476 100644
--- a/homeassistant/components/control4/light.py
+++ b/homeassistant/components/control4/light.py
@@ -18,7 +18,7 @@ from homeassistant.components.light import (
LightEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from . import Control4ConfigEntry, Control4RuntimeData, get_items_of_category
@@ -36,7 +36,7 @@ CONTROL4_DIMMER_VARS = ["LIGHT_LEVEL", "Brightness Percent"]
async def async_setup_entry(
hass: HomeAssistant,
entry: Control4ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Control4 lights from a config entry."""
runtime_data = entry.runtime_data
diff --git a/homeassistant/components/control4/media_player.py b/homeassistant/components/control4/media_player.py
index bd8e3fb38fe..824ce431aea 100644
--- a/homeassistant/components/control4/media_player.py
+++ b/homeassistant/components/control4/media_player.py
@@ -19,7 +19,7 @@ from homeassistant.components.media_player import (
MediaType,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from . import Control4ConfigEntry, Control4RuntimeData
@@ -77,7 +77,7 @@ async def get_rooms(hass: HomeAssistant, entry: Control4ConfigEntry):
async def async_setup_entry(
hass: HomeAssistant,
entry: Control4ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Control4 rooms from a config entry."""
runtime_data = entry.runtime_data
diff --git a/homeassistant/components/conversation/__init__.py b/homeassistant/components/conversation/__init__.py
index 69e738205c5..14c5244c18b 100644
--- a/homeassistant/components/conversation/__init__.py
+++ b/homeassistant/components/conversation/__init__.py
@@ -2,10 +2,12 @@
from __future__ import annotations
+from collections.abc import Callable
import logging
import re
from typing import Literal
+from hassil.recognize import RecognizeResult
import voluptuous as vol
from homeassistant.config_entries import ConfigEntry
@@ -32,6 +34,7 @@ from .agent_manager import (
)
from .chat_log import (
AssistantContent,
+ AssistantContentDeltaDict,
ChatLog,
Content,
ConverseError,
@@ -65,6 +68,7 @@ __all__ = [
"HOME_ASSISTANT_AGENT",
"OLD_HOME_ASSISTANT_AGENT",
"AssistantContent",
+ "AssistantContentDeltaDict",
"ChatLog",
"Content",
"ConversationEntity",
@@ -239,7 +243,10 @@ async def async_handle_sentence_triggers(
async def async_handle_intents(
- hass: HomeAssistant, user_input: ConversationInput
+ hass: HomeAssistant,
+ user_input: ConversationInput,
+ *,
+ intent_filter: Callable[[RecognizeResult], bool] | None = None,
) -> intent.IntentResponse | None:
"""Try to match input against registered intents and return response.
@@ -248,7 +255,9 @@ async def async_handle_intents(
default_agent = async_get_agent(hass)
assert isinstance(default_agent, DefaultAgent)
- return await default_agent.async_handle_intents(user_input)
+ return await default_agent.async_handle_intents(
+ user_input, intent_filter=intent_filter
+ )
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
diff --git a/homeassistant/components/conversation/agent_manager.py b/homeassistant/components/conversation/agent_manager.py
index ce3a0cf028d..5ff47977d88 100644
--- a/homeassistant/components/conversation/agent_manager.py
+++ b/homeassistant/components/conversation/agent_manager.py
@@ -79,6 +79,9 @@ async def async_converse(
extra_system_prompt: str | None = None,
) -> ConversationResult:
"""Process text and get intent."""
+ if agent_id is None:
+ agent_id = HOME_ASSISTANT_AGENT
+
agent = async_get_agent(hass, agent_id)
if agent is None:
diff --git a/homeassistant/components/conversation/chat_log.py b/homeassistant/components/conversation/chat_log.py
index ad7a9d0ce9e..cb7b8dd22f7 100644
--- a/homeassistant/components/conversation/chat_log.py
+++ b/homeassistant/components/conversation/chat_log.py
@@ -1,11 +1,14 @@
-"""Conversation history."""
+"""Conversation chat log."""
from __future__ import annotations
-from collections.abc import AsyncGenerator, Generator
+import asyncio
+from collections.abc import AsyncGenerator, AsyncIterable, Callable, Generator
from contextlib import contextmanager
-from dataclasses import dataclass, field, replace
+from contextvars import ContextVar
+from dataclasses import asdict, dataclass, field, replace
import logging
+from typing import Any, Literal, TypedDict
import voluptuous as vol
@@ -19,53 +22,83 @@ from . import trace
from .const import DOMAIN
from .models import ConversationInput, ConversationResult
-DATA_CHAT_HISTORY: HassKey[dict[str, ChatLog]] = HassKey("conversation_chat_log")
+DATA_CHAT_LOGS: HassKey[dict[str, ChatLog]] = HassKey("conversation_chat_logs")
LOGGER = logging.getLogger(__name__)
+current_chat_log: ContextVar[ChatLog | None] = ContextVar(
+ "current_chat_log", default=None
+)
+
@contextmanager
def async_get_chat_log(
hass: HomeAssistant,
session: chat_session.ChatSession,
user_input: ConversationInput | None = None,
+ *,
+ chat_log_delta_listener: Callable[[ChatLog, dict], None] | None = None,
) -> Generator[ChatLog]:
"""Return chat log for a specific chat session."""
- all_history = hass.data.get(DATA_CHAT_HISTORY)
- if all_history is None:
- all_history = {}
- hass.data[DATA_CHAT_HISTORY] = all_history
+ # If a chat log is already active and it's the requested conversation ID,
+ # return that. We won't update the last updated time in this case.
+ if (
+ chat_log := current_chat_log.get()
+ ) and chat_log.conversation_id == session.conversation_id:
+ if chat_log_delta_listener is not None:
+ raise RuntimeError(
+ "Cannot attach chat log delta listener unless initial caller"
+ )
+ if user_input is not None and (
+ (content := chat_log.content[-1]).role != "user"
+ or content.content != user_input.text
+ ):
+ chat_log.async_add_user_content(UserContent(content=user_input.text))
- history = all_history.get(session.conversation_id)
+ yield chat_log
+ return
- if history:
- history = replace(history, content=history.content.copy())
+ all_chat_logs = hass.data.get(DATA_CHAT_LOGS)
+ if all_chat_logs is None:
+ all_chat_logs = {}
+ hass.data[DATA_CHAT_LOGS] = all_chat_logs
+
+ if chat_log := all_chat_logs.get(session.conversation_id):
+ chat_log = replace(chat_log, content=chat_log.content.copy())
else:
- history = ChatLog(hass, session.conversation_id)
+ chat_log = ChatLog(hass, session.conversation_id)
+
+ if chat_log_delta_listener:
+ chat_log.delta_listener = chat_log_delta_listener
if user_input is not None:
- history.async_add_user_content(UserContent(content=user_input.text))
+ chat_log.async_add_user_content(UserContent(content=user_input.text))
- last_message = history.content[-1]
+ last_message = chat_log.content[-1]
- yield history
+ token = current_chat_log.set(chat_log)
+ yield chat_log
+ current_chat_log.reset(token)
- if history.content[-1] is last_message:
+ if chat_log.content[-1] is last_message:
LOGGER.debug(
- "History opened but no assistant message was added, ignoring update"
+ "Chat Log opened but no assistant message was added, ignoring update"
)
return
- if session.conversation_id not in all_history:
+ if session.conversation_id not in all_chat_logs:
@callback
def do_cleanup() -> None:
"""Handle cleanup."""
- all_history.pop(session.conversation_id)
+ all_chat_logs.pop(session.conversation_id)
session.async_on_cleanup(do_cleanup)
- all_history[session.conversation_id] = history
+ if chat_log_delta_listener:
+ chat_log.delta_listener = None
+
+ all_chat_logs[session.conversation_id] = chat_log
class ConverseError(HomeAssistantError):
@@ -94,7 +127,7 @@ class ConverseError(HomeAssistantError):
class SystemContent:
"""Base class for chat messages."""
- role: str = field(init=False, default="system")
+ role: Literal["system"] = field(init=False, default="system")
content: str
@@ -102,7 +135,7 @@ class SystemContent:
class UserContent:
"""Assistant content."""
- role: str = field(init=False, default="user")
+ role: Literal["user"] = field(init=False, default="user")
content: str
@@ -110,9 +143,9 @@ class UserContent:
class AssistantContent:
"""Assistant content."""
- role: str = field(init=False, default="assistant")
+ role: Literal["assistant"] = field(init=False, default="assistant")
agent_id: str
- content: str
+ content: str | None = None
tool_calls: list[llm.ToolInput] | None = None
@@ -120,14 +153,22 @@ class AssistantContent:
class ToolResultContent:
"""Tool result content."""
- role: str = field(init=False, default="tool_result")
+ role: Literal["tool_result"] = field(init=False, default="tool_result")
agent_id: str
tool_call_id: str
tool_name: str
tool_result: JsonObjectType
-Content = SystemContent | UserContent | AssistantContent | ToolResultContent
+type Content = SystemContent | UserContent | AssistantContent | ToolResultContent
+
+
+class AssistantContentDeltaDict(TypedDict, total=False):
+ """Partial content to define an AssistantContent."""
+
+ role: Literal["assistant"]
+ content: str | None
+ tool_calls: list[llm.ToolInput] | None
@dataclass
@@ -139,10 +180,36 @@ class ChatLog:
content: list[Content] = field(default_factory=lambda: [SystemContent(content="")])
extra_system_prompt: str | None = None
llm_api: llm.APIInstance | None = None
+ delta_listener: Callable[[ChatLog, dict], None] | None = None
+
+ @property
+ def continue_conversation(self) -> bool:
+ """Return whether the conversation should continue."""
+ if not self.content:
+ return False
+
+ last_msg = self.content[-1]
+
+ return (
+ last_msg.role == "assistant"
+ and last_msg.content is not None
+ and last_msg.content.strip().endswith(
+ (
+ "?",
+ ";", # Greek question mark
+ )
+ )
+ )
+
+ @property
+ def unresponded_tool_results(self) -> bool:
+ """Return if there are unresponded tool results."""
+ return self.content[-1].role == "tool_result"
@callback
def async_add_user_content(self, content: UserContent) -> None:
"""Add user content to the log."""
+ LOGGER.debug("Adding user content: %s", content)
self.content.append(content)
@callback
@@ -150,14 +217,24 @@ class ChatLog:
self, content: AssistantContent
) -> None:
"""Add assistant content to the log."""
+ LOGGER.debug("Adding assistant content: %s", content)
if content.tool_calls is not None:
raise ValueError("Tool calls not allowed")
self.content.append(content)
async def async_add_assistant_content(
- self, content: AssistantContent
+ self,
+ content: AssistantContent,
+ /,
+ tool_call_tasks: dict[str, asyncio.Task] | None = None,
) -> AsyncGenerator[ToolResultContent]:
- """Add assistant content."""
+ """Add assistant content and execute tool calls.
+
+ tool_call_tasks can contains tasks for tool calls that are already in progress.
+
+ This method is an async generator and will yield the tool results as they come in.
+ """
+ LOGGER.debug("Adding assistant content: %s", content)
self.content.append(content)
if content.tool_calls is None:
@@ -166,13 +243,22 @@ class ChatLog:
if self.llm_api is None:
raise ValueError("No LLM API configured")
+ if tool_call_tasks is None:
+ tool_call_tasks = {}
+ for tool_input in content.tool_calls:
+ if tool_input.id not in tool_call_tasks:
+ tool_call_tasks[tool_input.id] = self.hass.async_create_task(
+ self.llm_api.async_call_tool(tool_input),
+ name=f"llm_tool_{tool_input.id}",
+ )
+
for tool_input in content.tool_calls:
LOGGER.debug(
"Tool call: %s(%s)", tool_input.tool_name, tool_input.tool_args
)
try:
- tool_result = await self.llm_api.async_call_tool(tool_input)
+ tool_result = await tool_call_tasks[tool_input.id]
except (HomeAssistantError, vol.Invalid) as e:
tool_result = {"error": type(e).__name__}
if str(e):
@@ -188,6 +274,86 @@ class ChatLog:
self.content.append(response_content)
yield response_content
+ async def async_add_delta_content_stream(
+ self, agent_id: str, stream: AsyncIterable[AssistantContentDeltaDict]
+ ) -> AsyncGenerator[AssistantContent | ToolResultContent]:
+ """Stream content into the chat log.
+
+ Returns a generator with all content that was added to the chat log.
+
+ stream iterates over dictionaries with optional keys role, content and tool_calls.
+
+ When a delta contains a role key, the current message is considered complete and
+ a new message is started.
+
+ The keys content and tool_calls will be concatenated if they appear multiple times.
+ """
+ current_content = ""
+ current_tool_calls: list[llm.ToolInput] = []
+ tool_call_tasks: dict[str, asyncio.Task] = {}
+
+ async for delta in stream:
+ LOGGER.debug("Received delta: %s", delta)
+
+ # Indicates update to current message
+ if "role" not in delta:
+ if delta_content := delta.get("content"):
+ current_content += delta_content
+ if delta_tool_calls := delta.get("tool_calls"):
+ if self.llm_api is None:
+ raise ValueError("No LLM API configured")
+ current_tool_calls += delta_tool_calls
+
+ # Start processing the tool calls as soon as we know about them
+ for tool_call in delta_tool_calls:
+ tool_call_tasks[tool_call.id] = self.hass.async_create_task(
+ self.llm_api.async_call_tool(tool_call),
+ name=f"llm_tool_{tool_call.id}",
+ )
+ if self.delta_listener:
+ self.delta_listener(self, delta) # type: ignore[arg-type]
+ continue
+
+ # Starting a new message
+
+ if delta["role"] != "assistant":
+ raise ValueError(f"Only assistant role expected. Got {delta['role']}")
+
+ # Yield the previous message if it has content
+ if current_content or current_tool_calls:
+ content = AssistantContent(
+ agent_id=agent_id,
+ content=current_content or None,
+ tool_calls=current_tool_calls or None,
+ )
+ yield content
+ async for tool_result in self.async_add_assistant_content(
+ content, tool_call_tasks=tool_call_tasks
+ ):
+ yield tool_result
+ if self.delta_listener:
+ self.delta_listener(self, asdict(tool_result))
+
+ current_content = delta.get("content") or ""
+ current_tool_calls = delta.get("tool_calls") or []
+
+ if self.delta_listener:
+ self.delta_listener(self, delta) # type: ignore[arg-type]
+
+ if current_content or current_tool_calls:
+ content = AssistantContent(
+ agent_id=agent_id,
+ content=current_content or None,
+ tool_calls=current_tool_calls or None,
+ )
+ yield content
+ async for tool_result in self.async_add_assistant_content(
+ content, tool_call_tasks=tool_call_tasks
+ ):
+ yield tool_result
+ if self.delta_listener:
+ self.delta_listener(self, asdict(tool_result))
+
async def async_update_llm_data(
self,
conversing_domain: str,
@@ -275,12 +441,10 @@ class ChatLog:
if llm_api:
prompt_parts.append(llm_api.api_prompt)
- extra_system_prompt = (
+ if extra_system_prompt := (
# Take new system prompt if one was given
user_input.extra_system_prompt or self.extra_system_prompt
- )
-
- if extra_system_prompt:
+ ):
prompt_parts.append(extra_system_prompt)
prompt = "\n".join(prompt_parts)
@@ -292,10 +456,16 @@ class ChatLog:
LOGGER.debug("Prompt: %s", self.content)
LOGGER.debug("Tools: %s", self.llm_api.tools if self.llm_api else None)
- trace.async_conversation_trace_append(
- trace.ConversationTraceEventType.AGENT_DETAIL,
+ self.async_trace(
{
"messages": self.content,
"tools": self.llm_api.tools if self.llm_api else None,
- },
+ }
+ )
+
+ def async_trace(self, agent_details: dict[str, Any]) -> None:
+ """Append agent specific details to the conversation trace."""
+ trace.async_conversation_trace_append(
+ trace.ConversationTraceEventType.AGENT_DETAIL,
+ agent_details,
)
diff --git a/homeassistant/components/conversation/default_agent.py b/homeassistant/components/conversation/default_agent.py
index bd7450e5a0f..c30e8bb4a92 100644
--- a/homeassistant/components/conversation/default_agent.py
+++ b/homeassistant/components/conversation/default_agent.py
@@ -42,7 +42,6 @@ from homeassistant.components.homeassistant.exposed_entities import (
from homeassistant.const import EVENT_STATE_CHANGED, MATCH_ALL
from homeassistant.helpers import (
area_registry as ar,
- chat_session,
device_registry as dr,
entity_registry as er,
floor_registry as fr,
@@ -53,9 +52,10 @@ from homeassistant.helpers import (
)
from homeassistant.helpers.entity_component import EntityComponent
from homeassistant.helpers.event import async_track_state_added_domain
+from homeassistant.util import language as language_util
from homeassistant.util.json import JsonObjectType, json_loads_object
-from .chat_log import AssistantContent, async_get_chat_log
+from .chat_log import AssistantContent, ChatLog
from .const import (
DATA_DEFAULT_ENTITY,
DEFAULT_EXPOSED_ATTRIBUTES,
@@ -184,21 +184,6 @@ class IntentCache:
self.cache.clear()
-def _get_language_variations(language: str) -> Iterable[str]:
- """Generate language codes with and without region."""
- yield language
-
- parts = re.split(r"([-_])", language)
- if len(parts) == 3:
- lang, sep, region = parts
- if sep == "_":
- # en_US -> en-US
- yield f"{lang}-{region}"
-
- # en-US -> en
- yield lang
-
-
async def async_setup_default_agent(
hass: core.HomeAssistant,
entity_component: EntityComponent[ConversationEntity],
@@ -346,49 +331,46 @@ class DefaultAgent(ConversationEntity):
return result
- async def async_process(self, user_input: ConversationInput) -> ConversationResult:
- """Process a sentence."""
+ async def _async_handle_message(
+ self,
+ user_input: ConversationInput,
+ chat_log: ChatLog,
+ ) -> ConversationResult:
+ """Handle a message."""
response: intent.IntentResponse | None = None
- with (
- chat_session.async_get_chat_session(
- self.hass, user_input.conversation_id
- ) as session,
- async_get_chat_log(self.hass, session, user_input) as chat_log,
- ):
- # Check if a trigger matched
- if trigger_result := await self.async_recognize_sentence_trigger(
- user_input
- ):
- # Process callbacks and get response
- response_text = await self._handle_trigger_result(
- trigger_result, user_input
- )
- # Convert to conversation result
- response = intent.IntentResponse(
- language=user_input.language or self.hass.config.language
- )
- response.response_type = intent.IntentResponseType.ACTION_DONE
- response.async_set_speech(response_text)
-
- if response is None:
- # Match intents
- intent_result = await self.async_recognize_intent(user_input)
- response = await self._async_process_intent_result(
- intent_result, user_input
- )
-
- speech: str = response.speech.get("plain", {}).get("speech", "")
- chat_log.async_add_assistant_content_without_tools(
- AssistantContent(
- agent_id=user_input.agent_id, # type: ignore[arg-type]
- content=speech,
- )
+ # Check if a trigger matched
+ if trigger_result := await self.async_recognize_sentence_trigger(user_input):
+ # Process callbacks and get response
+ response_text = await self._handle_trigger_result(
+ trigger_result, user_input
)
- return ConversationResult(
- response=response, conversation_id=session.conversation_id
+ # Convert to conversation result
+ response = intent.IntentResponse(
+ language=user_input.language or self.hass.config.language
)
+ response.response_type = intent.IntentResponseType.ACTION_DONE
+ response.async_set_speech(response_text)
+
+ if response is None:
+ # Match intents
+ intent_result = await self.async_recognize_intent(user_input)
+ response = await self._async_process_intent_result(
+ intent_result, user_input
+ )
+
+ speech: str = response.speech.get("plain", {}).get("speech", "")
+ chat_log.async_add_assistant_content_without_tools(
+ AssistantContent(
+ agent_id=user_input.agent_id,
+ content=speech,
+ )
+ )
+
+ return ConversationResult(
+ response=response, conversation_id=chat_log.conversation_id
+ )
async def _async_process_intent_result(
self,
@@ -914,26 +896,20 @@ class DefaultAgent(ConversationEntity):
def _load_intents(self, language: str) -> LanguageIntents | None:
"""Load all intents for language (run inside executor)."""
intents_dict: dict[str, Any] = {}
- language_variant: str | None = None
supported_langs = set(get_languages())
# Choose a language variant upfront and commit to it for custom
# sentences, etc.
- all_language_variants = {lang.lower(): lang for lang in supported_langs}
+ lang_matches = language_util.matches(language, supported_langs)
- # en-US, en_US, en, ...
- for maybe_variant in _get_language_variations(language):
- matching_variant = all_language_variants.get(maybe_variant.lower())
- if matching_variant:
- language_variant = matching_variant
- break
-
- if not language_variant:
+ if not lang_matches:
_LOGGER.warning(
"Unable to find supported language variant for %s", language
)
return None
+ language_variant = lang_matches[0]
+
# Load intents for this language variant
lang_variant_intents = get_intents(language_variant, json_load=json_load)
@@ -1329,6 +1305,8 @@ class DefaultAgent(ConversationEntity):
async def async_handle_intents(
self,
user_input: ConversationInput,
+ *,
+ intent_filter: Callable[[RecognizeResult], bool] | None = None,
) -> intent.IntentResponse | None:
"""Try to match sentence against registered intents and return response.
@@ -1336,7 +1314,9 @@ class DefaultAgent(ConversationEntity):
Returns None if no match or a matching error occurred.
"""
result = await self.async_recognize_intent(user_input, strict_intents_only=True)
- if not isinstance(result, RecognizeResult):
+ if not isinstance(result, RecognizeResult) or (
+ intent_filter is not None and intent_filter(result)
+ ):
# No error message on failed match
return None
diff --git a/homeassistant/components/conversation/entity.py b/homeassistant/components/conversation/entity.py
index d9598dee7eb..ca4d18ab9f5 100644
--- a/homeassistant/components/conversation/entity.py
+++ b/homeassistant/components/conversation/entity.py
@@ -4,9 +4,11 @@ from abc import abstractmethod
from typing import Literal, final
from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN
+from homeassistant.helpers.chat_session import async_get_chat_session
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.util import dt as dt_util
+from .chat_log import ChatLog, async_get_chat_log
from .const import ConversationEntityFeature
from .models import ConversationInput, ConversationResult
@@ -51,9 +53,21 @@ class ConversationEntity(RestoreEntity):
def supported_languages(self) -> list[str] | Literal["*"]:
"""Return a list of supported languages."""
- @abstractmethod
async def async_process(self, user_input: ConversationInput) -> ConversationResult:
"""Process a sentence."""
+ with (
+ async_get_chat_session(self.hass, user_input.conversation_id) as session,
+ async_get_chat_log(self.hass, session, user_input) as chat_log,
+ ):
+ return await self._async_handle_message(user_input, chat_log)
+
+ async def _async_handle_message(
+ self,
+ user_input: ConversationInput,
+ chat_log: ChatLog,
+ ) -> ConversationResult:
+ """Call the API."""
+ raise NotImplementedError
async def async_prepare(self, language: str | None = None) -> None:
"""Load intents for a language."""
diff --git a/homeassistant/components/conversation/http.py b/homeassistant/components/conversation/http.py
index 8134ecb0eee..4d8526a4fd4 100644
--- a/homeassistant/components/conversation/http.py
+++ b/homeassistant/components/conversation/http.py
@@ -195,7 +195,7 @@ async def websocket_hass_agent_debug(
conversation_id=None,
device_id=msg.get("device_id"),
language=msg.get("language", hass.config.language),
- agent_id=None,
+ agent_id=agent.entity_id,
)
result_dict: dict[str, Any] | None = None
diff --git a/homeassistant/components/conversation/manifest.json b/homeassistant/components/conversation/manifest.json
index 2d4a8053d75..ea950ace323 100644
--- a/homeassistant/components/conversation/manifest.json
+++ b/homeassistant/components/conversation/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/conversation",
"integration_type": "system",
"quality_scale": "internal",
- "requirements": ["hassil==2.2.3", "home-assistant-intents==2025.2.5"]
+ "requirements": ["hassil==2.2.3", "home-assistant-intents==2025.3.5"]
}
diff --git a/homeassistant/components/conversation/models.py b/homeassistant/components/conversation/models.py
index 9462c597f23..7bdd13afc01 100644
--- a/homeassistant/components/conversation/models.py
+++ b/homeassistant/components/conversation/models.py
@@ -37,7 +37,7 @@ class ConversationInput:
language: str
"""Language of the request."""
- agent_id: str | None = None
+ agent_id: str
"""Agent to use for processing."""
extra_system_prompt: str | None = None
@@ -62,12 +62,14 @@ class ConversationResult:
response: intent.IntentResponse
conversation_id: str | None = None
+ continue_conversation: bool = False
def as_dict(self) -> dict[str, Any]:
"""Return result as a dict."""
return {
"response": self.response.as_dict(),
"conversation_id": self.conversation_id,
+ "continue_conversation": self.continue_conversation,
}
diff --git a/homeassistant/components/cookidoo/button.py b/homeassistant/components/cookidoo/button.py
index b292a7309ba..97136deb031 100644
--- a/homeassistant/components/cookidoo/button.py
+++ b/homeassistant/components/cookidoo/button.py
@@ -8,7 +8,7 @@ from cookidoo_api import Cookidoo, CookidooException
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import CookidooConfigEntry, CookidooDataUpdateCoordinator
@@ -35,7 +35,7 @@ TODO_CLEAR = CookidooButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: CookidooConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Cookidoo button entities based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/cookidoo/sensor.py b/homeassistant/components/cookidoo/sensor.py
index 7fbacea18bc..6df41383a75 100644
--- a/homeassistant/components/cookidoo/sensor.py
+++ b/homeassistant/components/cookidoo/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -73,7 +73,7 @@ SENSOR_DESCRIPTIONS: tuple[CookidooSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CookidooConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/cookidoo/todo.py b/homeassistant/components/cookidoo/todo.py
index 3d5264f4e01..c577b845657 100644
--- a/homeassistant/components/cookidoo/todo.py
+++ b/homeassistant/components/cookidoo/todo.py
@@ -18,7 +18,7 @@ from homeassistant.components.todo import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import CookidooConfigEntry, CookidooDataUpdateCoordinator
@@ -30,7 +30,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CookidooConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the todo list from a config entry created in the integrations UI."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/coolmaster/binary_sensor.py b/homeassistant/components/coolmaster/binary_sensor.py
index ab2718b9352..5c1f19fd14c 100644
--- a/homeassistant/components/coolmaster/binary_sensor.py
+++ b/homeassistant/components/coolmaster/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import CoolmasterConfigEntry
from .entity import CoolmasterEntity
@@ -18,7 +18,7 @@ from .entity import CoolmasterEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CoolmasterConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the CoolMasterNet binary_sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/coolmaster/button.py b/homeassistant/components/coolmaster/button.py
index 5463566d1ef..7cc8fc56c80 100644
--- a/homeassistant/components/coolmaster/button.py
+++ b/homeassistant/components/coolmaster/button.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import CoolmasterConfigEntry
from .entity import CoolmasterEntity
@@ -14,7 +14,7 @@ from .entity import CoolmasterEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CoolmasterConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the CoolMasterNet button platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/coolmaster/climate.py b/homeassistant/components/coolmaster/climate.py
index cd1659e1666..52fdfaaca3f 100644
--- a/homeassistant/components/coolmaster/climate.py
+++ b/homeassistant/components/coolmaster/climate.py
@@ -15,7 +15,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_SUPPORTED_MODES
from .coordinator import CoolmasterConfigEntry, CoolmasterDataUpdateCoordinator
@@ -39,7 +39,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CoolmasterConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the CoolMasterNet climate platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/coolmaster/sensor.py b/homeassistant/components/coolmaster/sensor.py
index 2b835565bae..32dceb83c5f 100644
--- a/homeassistant/components/coolmaster/sensor.py
+++ b/homeassistant/components/coolmaster/sensor.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import CoolmasterConfigEntry
from .entity import CoolmasterEntity
@@ -14,7 +14,7 @@ from .entity import CoolmasterEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CoolmasterConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the CoolMasterNet sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/cover/reproduce_state.py b/homeassistant/components/cover/reproduce_state.py
index 307fe5f11bd..de3e0cebfb7 100644
--- a/homeassistant/components/cover/reproduce_state.py
+++ b/homeassistant/components/cover/reproduce_state.py
@@ -3,12 +3,14 @@
from __future__ import annotations
import asyncio
-from collections.abc import Iterable
+from collections.abc import Coroutine, Iterable
+from functools import partial
import logging
-from typing import Any
+from typing import Any, Final
from homeassistant.const import (
ATTR_ENTITY_ID,
+ ATTR_SUPPORTED_FEATURES,
SERVICE_CLOSE_COVER,
SERVICE_CLOSE_COVER_TILT,
SERVICE_OPEN_COVER,
@@ -16,7 +18,8 @@ from homeassistant.const import (
SERVICE_SET_COVER_POSITION,
SERVICE_SET_COVER_TILT_POSITION,
)
-from homeassistant.core import Context, HomeAssistant, State
+from homeassistant.core import Context, HomeAssistant, ServiceResponse, State
+from homeassistant.util.enum import try_parse_enum
from . import (
ATTR_CURRENT_POSITION,
@@ -24,17 +27,140 @@ from . import (
ATTR_POSITION,
ATTR_TILT_POSITION,
DOMAIN,
+ CoverEntityFeature,
CoverState,
)
_LOGGER = logging.getLogger(__name__)
-VALID_STATES = {
- CoverState.CLOSED,
- CoverState.CLOSING,
- CoverState.OPEN,
- CoverState.OPENING,
-}
+
+OPENING_STATES = {CoverState.OPENING, CoverState.OPEN}
+CLOSING_STATES = {CoverState.CLOSING, CoverState.CLOSED}
+VALID_STATES: set[CoverState] = OPENING_STATES | CLOSING_STATES
+
+FULL_OPEN: Final = 100
+FULL_CLOSE: Final = 0
+
+
+def _determine_features(current_attrs: dict[str, Any]) -> CoverEntityFeature:
+ """Determine supported features based on current attributes."""
+ features = CoverEntityFeature(0)
+ if ATTR_CURRENT_POSITION in current_attrs:
+ features |= (
+ CoverEntityFeature.SET_POSITION
+ | CoverEntityFeature.OPEN
+ | CoverEntityFeature.CLOSE
+ )
+ if ATTR_CURRENT_TILT_POSITION in current_attrs:
+ features |= (
+ CoverEntityFeature.SET_TILT_POSITION
+ | CoverEntityFeature.OPEN_TILT
+ | CoverEntityFeature.CLOSE_TILT
+ )
+ if features == CoverEntityFeature(0):
+ features = CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE
+ return features
+
+
+async def _async_set_position(
+ service_call: partial[Coroutine[Any, Any, ServiceResponse]],
+ service_data: dict[str, Any],
+ features: CoverEntityFeature,
+ target_position: int,
+) -> bool:
+ """Set the position of the cover.
+
+ Returns True if the position was set, False if there is no
+ supported method for setting the position.
+ """
+ if target_position == FULL_CLOSE and CoverEntityFeature.CLOSE in features:
+ await service_call(SERVICE_CLOSE_COVER, service_data)
+ elif target_position == FULL_OPEN and CoverEntityFeature.OPEN in features:
+ await service_call(SERVICE_OPEN_COVER, service_data)
+ elif CoverEntityFeature.SET_POSITION in features:
+ await service_call(
+ SERVICE_SET_COVER_POSITION, service_data | {ATTR_POSITION: target_position}
+ )
+ else:
+ # Requested a position but the cover doesn't support it
+ return False
+ return True
+
+
+async def _async_set_tilt_position(
+ service_call: partial[Coroutine[Any, Any, ServiceResponse]],
+ service_data: dict[str, Any],
+ features: CoverEntityFeature,
+ target_tilt_position: int,
+) -> bool:
+ """Set the tilt position of the cover.
+
+ Returns True if the tilt position was set, False if there is no
+ supported method for setting the tilt position.
+ """
+ if target_tilt_position == FULL_CLOSE and CoverEntityFeature.CLOSE_TILT in features:
+ await service_call(SERVICE_CLOSE_COVER_TILT, service_data)
+ elif target_tilt_position == FULL_OPEN and CoverEntityFeature.OPEN_TILT in features:
+ await service_call(SERVICE_OPEN_COVER_TILT, service_data)
+ elif CoverEntityFeature.SET_TILT_POSITION in features:
+ await service_call(
+ SERVICE_SET_COVER_TILT_POSITION,
+ service_data | {ATTR_TILT_POSITION: target_tilt_position},
+ )
+ else:
+ # Requested a tilt position but the cover doesn't support it
+ return False
+ return True
+
+
+async def _async_close_cover(
+ service_call: partial[Coroutine[Any, Any, ServiceResponse]],
+ service_data: dict[str, Any],
+ features: CoverEntityFeature,
+ set_position: bool,
+ set_tilt: bool,
+) -> None:
+ """Close the cover if it was not closed by setting the position."""
+ if not set_position:
+ if CoverEntityFeature.CLOSE in features:
+ await service_call(SERVICE_CLOSE_COVER, service_data)
+ elif CoverEntityFeature.SET_POSITION in features:
+ await service_call(
+ SERVICE_SET_COVER_POSITION, service_data | {ATTR_POSITION: FULL_CLOSE}
+ )
+ if not set_tilt:
+ if CoverEntityFeature.CLOSE_TILT in features:
+ await service_call(SERVICE_CLOSE_COVER_TILT, service_data)
+ elif CoverEntityFeature.SET_TILT_POSITION in features:
+ await service_call(
+ SERVICE_SET_COVER_TILT_POSITION,
+ service_data | {ATTR_TILT_POSITION: FULL_CLOSE},
+ )
+
+
+async def _async_open_cover(
+ service_call: partial[Coroutine[Any, Any, ServiceResponse]],
+ service_data: dict[str, Any],
+ features: CoverEntityFeature,
+ set_position: bool,
+ set_tilt: bool,
+) -> None:
+ """Open the cover if it was not opened by setting the position."""
+ if not set_position:
+ if CoverEntityFeature.OPEN in features:
+ await service_call(SERVICE_OPEN_COVER, service_data)
+ elif CoverEntityFeature.SET_POSITION in features:
+ await service_call(
+ SERVICE_SET_COVER_POSITION, service_data | {ATTR_POSITION: FULL_OPEN}
+ )
+ if not set_tilt:
+ if CoverEntityFeature.OPEN_TILT in features:
+ await service_call(SERVICE_OPEN_COVER_TILT, service_data)
+ elif CoverEntityFeature.SET_TILT_POSITION in features:
+ await service_call(
+ SERVICE_SET_COVER_TILT_POSITION,
+ service_data | {ATTR_TILT_POSITION: FULL_OPEN},
+ )
async def _async_reproduce_state(
@@ -45,74 +171,72 @@ async def _async_reproduce_state(
reproduce_options: dict[str, Any] | None = None,
) -> None:
"""Reproduce a single state."""
- if (cur_state := hass.states.get(state.entity_id)) is None:
- _LOGGER.warning("Unable to find entity %s", state.entity_id)
+ entity_id = state.entity_id
+ if (cur_state := hass.states.get(entity_id)) is None:
+ _LOGGER.warning("Unable to find entity %s", entity_id)
return
- if state.state not in VALID_STATES:
- _LOGGER.warning(
- "Invalid state specified for %s: %s", state.entity_id, state.state
- )
+ if (target_state := state.state) not in VALID_STATES:
+ _LOGGER.warning("Invalid state specified for %s: %s", entity_id, target_state)
return
+ current_attrs = cur_state.attributes
+ target_attrs = state.attributes
+
+ current_position = current_attrs.get(ATTR_CURRENT_POSITION)
+ target_position = target_attrs.get(ATTR_CURRENT_POSITION)
+ position_matches = current_position == target_position
+
+ current_tilt_position = current_attrs.get(ATTR_CURRENT_TILT_POSITION)
+ target_tilt_position = target_attrs.get(ATTR_CURRENT_TILT_POSITION)
+ tilt_position_matches = current_tilt_position == target_tilt_position
+
+ state_matches = cur_state.state == target_state
# Return if we are already at the right state.
- if (
- cur_state.state == state.state
- and cur_state.attributes.get(ATTR_CURRENT_POSITION)
- == state.attributes.get(ATTR_CURRENT_POSITION)
- and cur_state.attributes.get(ATTR_CURRENT_TILT_POSITION)
- == state.attributes.get(ATTR_CURRENT_TILT_POSITION)
- ):
+ if state_matches and position_matches and tilt_position_matches:
return
- service_data = {ATTR_ENTITY_ID: state.entity_id}
- service_data_tilting = {ATTR_ENTITY_ID: state.entity_id}
+ features = try_parse_enum(
+ CoverEntityFeature, current_attrs.get(ATTR_SUPPORTED_FEATURES)
+ )
+ if features is None:
+ # Backwards compatibility for integrations that
+ # don't set supported features since it previously
+ # worked without it.
+ _LOGGER.warning("Supported features is not set for %s", entity_id)
+ features = _determine_features(current_attrs)
- if not (
- cur_state.state == state.state
- and cur_state.attributes.get(ATTR_CURRENT_POSITION)
- == state.attributes.get(ATTR_CURRENT_POSITION)
- ):
- # Open/Close
- if state.state in [CoverState.CLOSED, CoverState.CLOSING]:
- service = SERVICE_CLOSE_COVER
- elif state.state in [CoverState.OPEN, CoverState.OPENING]:
- if (
- ATTR_CURRENT_POSITION in cur_state.attributes
- and ATTR_CURRENT_POSITION in state.attributes
- ):
- service = SERVICE_SET_COVER_POSITION
- service_data[ATTR_POSITION] = state.attributes[ATTR_CURRENT_POSITION]
- else:
- service = SERVICE_OPEN_COVER
+ service_call = partial(
+ hass.services.async_call,
+ DOMAIN,
+ context=context,
+ blocking=True,
+ )
+ service_data = {ATTR_ENTITY_ID: entity_id}
- await hass.services.async_call(
- DOMAIN, service, service_data, context=context, blocking=True
+ set_position = (
+ not position_matches
+ and target_position is not None
+ and await _async_set_position(
+ service_call, service_data, features, target_position
+ )
+ )
+ set_tilt = (
+ not tilt_position_matches
+ and target_tilt_position is not None
+ and await _async_set_tilt_position(
+ service_call, service_data, features, target_tilt_position
+ )
+ )
+
+ if target_state in CLOSING_STATES:
+ await _async_close_cover(
+ service_call, service_data, features, set_position, set_tilt
)
- if (
- ATTR_CURRENT_TILT_POSITION in state.attributes
- and ATTR_CURRENT_TILT_POSITION in cur_state.attributes
- and cur_state.attributes.get(ATTR_CURRENT_TILT_POSITION)
- != state.attributes.get(ATTR_CURRENT_TILT_POSITION)
- ):
- # Tilt position
- if state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 100:
- service_tilting = SERVICE_OPEN_COVER_TILT
- elif state.attributes.get(ATTR_CURRENT_TILT_POSITION) == 0:
- service_tilting = SERVICE_CLOSE_COVER_TILT
- else:
- service_tilting = SERVICE_SET_COVER_TILT_POSITION
- service_data_tilting[ATTR_TILT_POSITION] = state.attributes[
- ATTR_CURRENT_TILT_POSITION
- ]
-
- await hass.services.async_call(
- DOMAIN,
- service_tilting,
- service_data_tilting,
- context=context,
- blocking=True,
+ elif target_state in OPENING_STATES:
+ await _async_open_cover(
+ service_call, service_data, features, set_position, set_tilt
)
diff --git a/homeassistant/components/cpuspeed/sensor.py b/homeassistant/components/cpuspeed/sensor.py
index 6a14f7ad13f..11f683b1434 100644
--- a/homeassistant/components/cpuspeed/sensor.py
+++ b/homeassistant/components/cpuspeed/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfFrequency
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -24,7 +24,7 @@ HZ_ADVERTISED = "hz_advertised"
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from config_entry."""
async_add_entities([CPUSpeedSensor(entry)], True)
diff --git a/homeassistant/components/crownstone/light.py b/homeassistant/components/crownstone/light.py
index 70b7631fe6b..dc29ad93072 100644
--- a/homeassistant/components/crownstone/light.py
+++ b/homeassistant/components/crownstone/light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEnti
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CROWNSTONE_INCLUDE_TYPES,
@@ -30,7 +30,7 @@ from .helpers import map_from_to
async def async_setup_entry(
hass: HomeAssistant,
config_entry: CrownstoneConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up crownstones from a config entry."""
manager = config_entry.runtime_data
diff --git a/homeassistant/components/daikin/climate.py b/homeassistant/components/daikin/climate.py
index 06ee0a03860..648a65c0d30 100644
--- a/homeassistant/components/daikin/climate.py
+++ b/homeassistant/components/daikin/climate.py
@@ -21,7 +21,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_INSIDE_TEMPERATURE,
@@ -83,7 +83,7 @@ DAIKIN_ATTR_ADVANCED = "adv"
async def async_setup_entry(
hass: HomeAssistant,
entry: DaikinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Daikin climate based on config_entry."""
daikin_api = entry.runtime_data
diff --git a/homeassistant/components/daikin/manifest.json b/homeassistant/components/daikin/manifest.json
index f794d97a9ba..86fc804ec92 100644
--- a/homeassistant/components/daikin/manifest.json
+++ b/homeassistant/components/daikin/manifest.json
@@ -6,6 +6,6 @@
"documentation": "https://www.home-assistant.io/integrations/daikin",
"iot_class": "local_polling",
"loggers": ["pydaikin"],
- "requirements": ["pydaikin==2.13.8"],
+ "requirements": ["pydaikin==2.14.1"],
"zeroconf": ["_dkapi._tcp.local."]
}
diff --git a/homeassistant/components/daikin/sensor.py b/homeassistant/components/daikin/sensor.py
index 982aac1f3f2..c1aa28fbe67 100644
--- a/homeassistant/components/daikin/sensor.py
+++ b/homeassistant/components/daikin/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_COMPRESSOR_FREQUENCY,
@@ -134,7 +134,7 @@ SENSOR_TYPES: tuple[DaikinSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DaikinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Daikin climate based on config_entry."""
daikin_api = entry.runtime_data
diff --git a/homeassistant/components/daikin/switch.py b/homeassistant/components/daikin/switch.py
index 8a3a15d367f..20a56ac321c 100644
--- a/homeassistant/components/daikin/switch.py
+++ b/homeassistant/components/daikin/switch.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import DaikinConfigEntry, DaikinCoordinator
from .entity import DaikinEntity
@@ -19,7 +19,7 @@ DAIKIN_ATTR_MODE = "mode"
async def async_setup_entry(
hass: HomeAssistant,
entry: DaikinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Daikin climate based on config_entry."""
daikin_api = entry.runtime_data
diff --git a/homeassistant/components/deako/light.py b/homeassistant/components/deako/light.py
index 75b01935c9a..12f42c36f29 100644
--- a/homeassistant/components/deako/light.py
+++ b/homeassistant/components/deako/light.py
@@ -7,7 +7,7 @@ from pydeako import Deako
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeakoConfigEntry
from .const import DOMAIN
@@ -20,7 +20,7 @@ MODEL_DIMMER = "dimmer"
async def async_setup_entry(
hass: HomeAssistant,
config: DeakoConfigEntry,
- add_entities: AddEntitiesCallback,
+ add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure the platform."""
client = config.runtime_data
diff --git a/homeassistant/components/debugpy/manifest.json b/homeassistant/components/debugpy/manifest.json
index 078af8c67a5..21211d334df 100644
--- a/homeassistant/components/debugpy/manifest.json
+++ b/homeassistant/components/debugpy/manifest.json
@@ -6,5 +6,5 @@
"integration_type": "service",
"iot_class": "local_push",
"quality_scale": "internal",
- "requirements": ["debugpy==1.8.11"]
+ "requirements": ["debugpy==1.8.13"]
}
diff --git a/homeassistant/components/deconz/alarm_control_panel.py b/homeassistant/components/deconz/alarm_control_panel.py
index 94f4cd1ddd6..85ca32d76e6 100644
--- a/homeassistant/components/deconz/alarm_control_panel.py
+++ b/homeassistant/components/deconz/alarm_control_panel.py
@@ -17,7 +17,7 @@ from homeassistant.components.alarm_control_panel import (
CodeFormat,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzDevice
@@ -48,7 +48,7 @@ def get_alarm_system_id_for_unique_id(hub: DeconzHub, unique_id: str) -> str | N
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ alarm control panel devices."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/binary_sensor.py b/homeassistant/components/deconz/binary_sensor.py
index e3b0fc2f2c0..fcbb61a4e4f 100644
--- a/homeassistant/components/deconz/binary_sensor.py
+++ b/homeassistant/components/deconz/binary_sensor.py
@@ -25,7 +25,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import ATTR_TEMPERATURE, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .const import ATTR_DARK, ATTR_ON
@@ -161,7 +161,7 @@ ENTITY_DESCRIPTIONS: tuple[DeconzBinarySensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ binary sensor."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/button.py b/homeassistant/components/deconz/button.py
index 9fea1d02ab8..1d96f9867a7 100644
--- a/homeassistant/components/deconz/button.py
+++ b/homeassistant/components/deconz/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzDevice, DeconzSceneMixin
@@ -47,7 +47,7 @@ ENTITY_DESCRIPTIONS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ button entity."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/climate.py b/homeassistant/components/deconz/climate.py
index aa274e6c0c1..26597c195e7 100644
--- a/homeassistant/components/deconz/climate.py
+++ b/homeassistant/components/deconz/climate.py
@@ -30,7 +30,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .const import ATTR_LOCKED, ATTR_OFFSET, ATTR_VALVE
@@ -77,7 +77,7 @@ DECONZ_TO_PRESET_MODE = {value: key for key, value in PRESET_MODE_TO_DECONZ.item
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ climate devices."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/cover.py b/homeassistant/components/deconz/cover.py
index 6dee00248ff..d68e0fec09c 100644
--- a/homeassistant/components/deconz/cover.py
+++ b/homeassistant/components/deconz/cover.py
@@ -18,7 +18,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzDevice
@@ -34,7 +34,7 @@ DECONZ_TYPE_TO_DEVICE_CLASS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up covers for deCONZ component."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/fan.py b/homeassistant/components/deconz/fan.py
index aec078f771f..324ada807e0 100644
--- a/homeassistant/components/deconz/fan.py
+++ b/homeassistant/components/deconz/fan.py
@@ -13,7 +13,7 @@ from homeassistant.components.fan import (
FanEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -34,7 +34,7 @@ ORDERED_NAMED_FAN_SPEEDS: list[LightFanSpeed] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up fans for deCONZ component."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/light.py b/homeassistant/components/deconz/light.py
index 72ba7035c8e..b61a1d39333 100644
--- a/homeassistant/components/deconz/light.py
+++ b/homeassistant/components/deconz/light.py
@@ -30,7 +30,7 @@ from homeassistant.components.light import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.color import (
color_hs_to_xy,
color_temperature_kelvin_to_mired,
@@ -142,7 +142,7 @@ def update_color_state(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ lights and groups from a config entry."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/lock.py b/homeassistant/components/deconz/lock.py
index e5e2faf1d57..77b9ea435c7 100644
--- a/homeassistant/components/deconz/lock.py
+++ b/homeassistant/components/deconz/lock.py
@@ -10,7 +10,7 @@ from pydeconz.models.sensor.door_lock import DoorLock
from homeassistant.components.lock import DOMAIN as LOCK_DOMAIN, LockEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzDevice
@@ -19,7 +19,7 @@ from .entity import DeconzDevice
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up locks for deCONZ component."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/number.py b/homeassistant/components/deconz/number.py
index 9de86c1c79b..d5ba8cc28d5 100644
--- a/homeassistant/components/deconz/number.py
+++ b/homeassistant/components/deconz/number.py
@@ -19,7 +19,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzDevice
@@ -70,7 +70,7 @@ ENTITY_DESCRIPTIONS: tuple[DeconzNumberDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ number entity."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/scene.py b/homeassistant/components/deconz/scene.py
index 3f29b12b05f..0aff2b3ca8c 100644
--- a/homeassistant/components/deconz/scene.py
+++ b/homeassistant/components/deconz/scene.py
@@ -8,7 +8,7 @@ from pydeconz.models.event import EventType
from homeassistant.components.scene import DOMAIN as SCENE_DOMAIN, Scene
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzSceneMixin
@@ -17,7 +17,7 @@ from .entity import DeconzSceneMixin
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up scenes for deCONZ integration."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/select.py b/homeassistant/components/deconz/select.py
index a3109a278fc..4d92b465cdc 100644
--- a/homeassistant/components/deconz/select.py
+++ b/homeassistant/components/deconz/select.py
@@ -14,7 +14,7 @@ from pydeconz.models.sensor.presence import (
from homeassistant.components.select import DOMAIN as SELECT_DOMAIN, SelectEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzDevice
@@ -30,7 +30,7 @@ DECONZ_TO_SENSITIVITY = {value: key for key, value in SENSITIVITY_TO_DECONZ.item
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ button entity."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/sensor.py b/homeassistant/components/deconz/sensor.py
index 3003fb1008d..d318db6e2bf 100644
--- a/homeassistant/components/deconz/sensor.py
+++ b/homeassistant/components/deconz/sensor.py
@@ -50,7 +50,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -332,7 +332,7 @@ ENTITY_DESCRIPTIONS: tuple[DeconzSensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the deCONZ sensors."""
hub = config_entry.runtime_data
@@ -468,7 +468,7 @@ class DeconzBatteryTracker:
sensor_id: str,
hub: DeconzHub,
description: DeconzSensorDescription,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tracker."""
self.sensor = hub.api.sensors[sensor_id]
diff --git a/homeassistant/components/deconz/siren.py b/homeassistant/components/deconz/siren.py
index 28b606e30ba..4c15cf8ccfe 100644
--- a/homeassistant/components/deconz/siren.py
+++ b/homeassistant/components/deconz/siren.py
@@ -14,7 +14,7 @@ from homeassistant.components.siren import (
SirenEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .entity import DeconzDevice
@@ -23,7 +23,7 @@ from .entity import DeconzDevice
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sirens for deCONZ component."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/deconz/switch.py b/homeassistant/components/deconz/switch.py
index cd28871e35b..49904642804 100644
--- a/homeassistant/components/deconz/switch.py
+++ b/homeassistant/components/deconz/switch.py
@@ -9,7 +9,7 @@ from pydeconz.models.light.light import Light
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN, SwitchEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeconzConfigEntry
from .const import POWER_PLUGS
@@ -19,7 +19,7 @@ from .entity import DeconzDevice
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DeconzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches for deCONZ component.
diff --git a/homeassistant/components/deluge/__init__.py b/homeassistant/components/deluge/__init__.py
index 9b07ae9c875..f9972570df3 100644
--- a/homeassistant/components/deluge/__init__.py
+++ b/homeassistant/components/deluge/__init__.py
@@ -7,7 +7,6 @@ from ssl import SSLError
from deluge_client.client import DelugeRPCClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_HOST,
CONF_PASSWORD,
@@ -19,12 +18,11 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from .const import CONF_WEB_PORT
-from .coordinator import DelugeDataUpdateCoordinator
+from .coordinator import DelugeConfigEntry, DelugeDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR, Platform.SWITCH]
_LOGGER = logging.getLogger(__name__)
-type DelugeConfigEntry = ConfigEntry[DelugeDataUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: DelugeConfigEntry) -> bool:
diff --git a/homeassistant/components/deluge/coordinator.py b/homeassistant/components/deluge/coordinator.py
index 7f4bf9e884e..c5836243b9d 100644
--- a/homeassistant/components/deluge/coordinator.py
+++ b/homeassistant/components/deluge/coordinator.py
@@ -4,10 +4,11 @@ from __future__ import annotations
from datetime import timedelta
from ssl import SSLError
-from typing import TYPE_CHECKING, Any
+from typing import Any
from deluge_client.client import DelugeRPCClient, FailedToReconnectException
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
@@ -15,8 +16,7 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import LOGGER, DelugeGetSessionStatusKeys
-if TYPE_CHECKING:
- from . import DelugeConfigEntry
+type DelugeConfigEntry = ConfigEntry[DelugeDataUpdateCoordinator]
class DelugeDataUpdateCoordinator(
@@ -33,11 +33,11 @@ class DelugeDataUpdateCoordinator(
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=entry,
name=entry.title,
update_interval=timedelta(seconds=30),
)
self.api = api
- self.config_entry = entry
async def _async_update_data(self) -> dict[Platform, dict[str, Any]]:
"""Get the latest data from Deluge and updates the state."""
diff --git a/homeassistant/components/deluge/sensor.py b/homeassistant/components/deluge/sensor.py
index 5ebf3d01eeb..d6809967703 100644
--- a/homeassistant/components/deluge/sensor.py
+++ b/homeassistant/components/deluge/sensor.py
@@ -14,12 +14,11 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import STATE_IDLE, Platform, UnitOfDataRate
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import DelugeConfigEntry
from .const import DelugeGetSessionStatusKeys, DelugeSensorType
-from .coordinator import DelugeDataUpdateCoordinator
+from .coordinator import DelugeConfigEntry, DelugeDataUpdateCoordinator
from .entity import DelugeEntity
@@ -117,7 +116,7 @@ SENSOR_TYPES: tuple[DelugeSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DelugeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Deluge sensor."""
async_add_entities(
diff --git a/homeassistant/components/deluge/switch.py b/homeassistant/components/deluge/switch.py
index d81f02eee29..342442ee727 100644
--- a/homeassistant/components/deluge/switch.py
+++ b/homeassistant/components/deluge/switch.py
@@ -7,17 +7,16 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import DelugeConfigEntry
-from .coordinator import DelugeDataUpdateCoordinator
+from .coordinator import DelugeConfigEntry, DelugeDataUpdateCoordinator
from .entity import DelugeEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DelugeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Deluge switch."""
async_add_entities([DelugeSwitch(entry.runtime_data)])
diff --git a/homeassistant/components/demo/__init__.py b/homeassistant/components/demo/__init__.py
index 9314fc211de..dbc65119bfa 100644
--- a/homeassistant/components/demo/__init__.py
+++ b/homeassistant/components/demo/__init__.py
@@ -48,6 +48,7 @@ COMPONENTS_WITH_CONFIG_ENTRY_DEMO_PLATFORM = [
Platform.TIME,
Platform.UPDATE,
Platform.VACUUM,
+ Platform.VALVE,
Platform.WATER_HEATER,
Platform.WEATHER,
]
diff --git a/homeassistant/components/demo/air_quality.py b/homeassistant/components/demo/air_quality.py
index 551f2c8e88a..4e247812efe 100644
--- a/homeassistant/components/demo/air_quality.py
+++ b/homeassistant/components/demo/air_quality.py
@@ -5,13 +5,13 @@ from __future__ import annotations
from homeassistant.components.air_quality import AirQualityEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/alarm_control_panel.py b/homeassistant/components/demo/alarm_control_panel.py
index d34830042d7..64474b4beb6 100644
--- a/homeassistant/components/demo/alarm_control_panel.py
+++ b/homeassistant/components/demo/alarm_control_panel.py
@@ -9,13 +9,13 @@ from homeassistant.components.manual.alarm_control_panel import ManualAlarm
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ARMING_TIME, CONF_DELAY_TIME, CONF_TRIGGER_TIME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/binary_sensor.py b/homeassistant/components/demo/binary_sensor.py
index bc1d7b9daf2..b210e726205 100644
--- a/homeassistant/components/demo/binary_sensor.py
+++ b/homeassistant/components/demo/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -17,7 +17,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo binary sensor platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/button.py b/homeassistant/components/demo/button.py
index a3b8dd9ff0c..25212f38989 100644
--- a/homeassistant/components/demo/button.py
+++ b/homeassistant/components/demo/button.py
@@ -7,7 +7,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -15,7 +15,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo button platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/calendar.py b/homeassistant/components/demo/calendar.py
index 4e2fa7b3460..b0e82acfa61 100644
--- a/homeassistant/components/demo/calendar.py
+++ b/homeassistant/components/demo/calendar.py
@@ -7,14 +7,14 @@ import datetime
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo Calendar config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/camera.py b/homeassistant/components/demo/camera.py
index 9fae6468207..69ba7efda01 100644
--- a/homeassistant/components/demo/camera.py
+++ b/homeassistant/components/demo/camera.py
@@ -7,13 +7,13 @@ from pathlib import Path
from homeassistant.components.camera import Camera, CameraEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/climate.py b/homeassistant/components/demo/climate.py
index d5b763caa5a..f68714695f3 100644
--- a/homeassistant/components/demo/climate.py
+++ b/homeassistant/components/demo/climate.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -27,7 +27,7 @@ SUPPORT_FLAGS = ClimateEntityFeature(0)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo climate platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/cover.py b/homeassistant/components/demo/cover.py
index adddb6a3a7d..ed13f24cfd7 100644
--- a/homeassistant/components/demo/cover.py
+++ b/homeassistant/components/demo/cover.py
@@ -15,7 +15,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_utc_time_change
from . import DOMAIN
@@ -24,7 +24,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo cover platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/date.py b/homeassistant/components/demo/date.py
index b67c4248123..875075a381d 100644
--- a/homeassistant/components/demo/date.py
+++ b/homeassistant/components/demo/date.py
@@ -8,7 +8,7 @@ from homeassistant.components.date import DateEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -16,7 +16,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo date platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/datetime.py b/homeassistant/components/demo/datetime.py
index 920bc14cdc5..353ed8311bb 100644
--- a/homeassistant/components/demo/datetime.py
+++ b/homeassistant/components/demo/datetime.py
@@ -8,7 +8,7 @@ from homeassistant.components.datetime import DateTimeEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -16,7 +16,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo datetime platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/event.py b/homeassistant/components/demo/event.py
index c58b5f5fc2e..f593a833123 100644
--- a/homeassistant/components/demo/event.py
+++ b/homeassistant/components/demo/event.py
@@ -6,7 +6,7 @@ from homeassistant.components.event import EventDeviceClass, EventEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -14,7 +14,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo event platform."""
async_add_entities([DemoEvent()])
diff --git a/homeassistant/components/demo/fan.py b/homeassistant/components/demo/fan.py
index 42e7f9e2434..9f48628688e 100644
--- a/homeassistant/components/demo/fan.py
+++ b/homeassistant/components/demo/fan.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
PRESET_MODE_AUTO = "auto"
PRESET_MODE_SMART = "smart"
@@ -29,7 +29,7 @@ LIMITED_SUPPORT = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/humidifier.py b/homeassistant/components/demo/humidifier.py
index 7245d96eaf0..2bdbd22eef8 100644
--- a/homeassistant/components/demo/humidifier.py
+++ b/homeassistant/components/demo/humidifier.py
@@ -12,7 +12,7 @@ from homeassistant.components.humidifier import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
SUPPORT_FLAGS = HumidifierEntityFeature(0)
@@ -20,7 +20,7 @@ SUPPORT_FLAGS = HumidifierEntityFeature(0)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo humidifier devices config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/light.py b/homeassistant/components/demo/light.py
index ec98a056b3e..c00f2b42828 100644
--- a/homeassistant/components/demo/light.py
+++ b/homeassistant/components/demo/light.py
@@ -22,7 +22,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -39,7 +39,7 @@ SUPPORT_DEMO_HS_WHITE = {ColorMode.HS, ColorMode.WHITE}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo light platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/lock.py b/homeassistant/components/demo/lock.py
index 1f25445af7f..081e1cf1d53 100644
--- a/homeassistant/components/demo/lock.py
+++ b/homeassistant/components/demo/lock.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.lock import LockEntity, LockEntityFeature, LockState
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
LOCK_UNLOCK_DELAY = 2 # Used to give a realistic lock/unlock experience in frontend
@@ -16,7 +16,7 @@ LOCK_UNLOCK_DELAY = 2 # Used to give a realistic lock/unlock experience in fron
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/media_player.py b/homeassistant/components/demo/media_player.py
index fa3c3e3b2fc..de2a2cb3937 100644
--- a/homeassistant/components/demo/media_player.py
+++ b/homeassistant/components/demo/media_player.py
@@ -15,14 +15,14 @@ from homeassistant.components.media_player import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/notify.py b/homeassistant/components/demo/notify.py
index 7524517e6e8..d26e13cc541 100644
--- a/homeassistant/components/demo/notify.py
+++ b/homeassistant/components/demo/notify.py
@@ -10,7 +10,7 @@ from homeassistant.components.notify import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
EVENT_NOTIFY = "notify"
@@ -18,7 +18,7 @@ EVENT_NOTIFY = "notify"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo entity platform."""
async_add_entities([DemoNotifyEntity(unique_id="notify", device_name="Notifier")])
diff --git a/homeassistant/components/demo/number.py b/homeassistant/components/demo/number.py
index 8c3f5ec3477..c7b62bdc3e0 100644
--- a/homeassistant/components/demo/number.py
+++ b/homeassistant/components/demo/number.py
@@ -7,7 +7,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -15,7 +15,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo number platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/remote.py b/homeassistant/components/demo/remote.py
index 774f375dd27..ffd6fd6e609 100644
--- a/homeassistant/components/demo/remote.py
+++ b/homeassistant/components/demo/remote.py
@@ -9,13 +9,13 @@ from homeassistant.components.remote import RemoteEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import DEVICE_DEFAULT_NAME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/select.py b/homeassistant/components/demo/select.py
index ff664a31d2f..fce90bc9b4f 100644
--- a/homeassistant/components/demo/select.py
+++ b/homeassistant/components/demo/select.py
@@ -6,7 +6,7 @@ from homeassistant.components.select import SelectEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -14,7 +14,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo select platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/sensor.py b/homeassistant/components/demo/sensor.py
index 0c61faae00e..ae9ff26eca9 100644
--- a/homeassistant/components/demo/sensor.py
+++ b/homeassistant/components/demo/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_interval
from . import DOMAIN
@@ -33,7 +33,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo sensor platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/siren.py b/homeassistant/components/demo/siren.py
index 235d98f5875..ddaa5101e0f 100644
--- a/homeassistant/components/demo/siren.py
+++ b/homeassistant/components/demo/siren.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.siren import SirenEntity, SirenEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
SUPPORT_FLAGS = SirenEntityFeature.TURN_OFF | SirenEntityFeature.TURN_ON
@@ -15,7 +15,7 @@ SUPPORT_FLAGS = SirenEntityFeature.TURN_OFF | SirenEntityFeature.TURN_ON
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo siren devices config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/stt.py b/homeassistant/components/demo/stt.py
index 95eebe44588..1757e4a8b88 100644
--- a/homeassistant/components/demo/stt.py
+++ b/homeassistant/components/demo/stt.py
@@ -17,7 +17,7 @@ from homeassistant.components.stt import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
SUPPORT_LANGUAGES = ["en", "de"]
@@ -25,7 +25,7 @@ SUPPORT_LANGUAGES = ["en", "de"]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Demo speech platform via config entry."""
async_add_entities([DemoProviderEntity()])
diff --git a/homeassistant/components/demo/switch.py b/homeassistant/components/demo/switch.py
index 5dc05398bf1..dd288f285af 100644
--- a/homeassistant/components/demo/switch.py
+++ b/homeassistant/components/demo/switch.py
@@ -8,7 +8,7 @@ from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -16,7 +16,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo switch platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/text.py b/homeassistant/components/demo/text.py
index 1730f414fdf..3219821ef98 100644
--- a/homeassistant/components/demo/text.py
+++ b/homeassistant/components/demo/text.py
@@ -6,7 +6,7 @@ from homeassistant.components.text import TextEntity, TextMode
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -14,7 +14,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo text platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/time.py b/homeassistant/components/demo/time.py
index f5f0322f9be..296155e9bec 100644
--- a/homeassistant/components/demo/time.py
+++ b/homeassistant/components/demo/time.py
@@ -8,7 +8,7 @@ from homeassistant.components.time import TimeEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -16,7 +16,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo time platform."""
async_add_entities([DemoTime("time", "Time", time(12, 0, 0), False)])
diff --git a/homeassistant/components/demo/update.py b/homeassistant/components/demo/update.py
index 3fa037f6b02..916646416e9 100644
--- a/homeassistant/components/demo/update.py
+++ b/homeassistant/components/demo/update.py
@@ -13,7 +13,7 @@ from homeassistant.components.update import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -23,7 +23,7 @@ FAKE_INSTALL_SLEEP_TIME = 0.5
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up demo update platform."""
async_add_entities(
diff --git a/homeassistant/components/demo/vacuum.py b/homeassistant/components/demo/vacuum.py
index 3dd945ab82e..38019cff3c1 100644
--- a/homeassistant/components/demo/vacuum.py
+++ b/homeassistant/components/demo/vacuum.py
@@ -14,7 +14,7 @@ from homeassistant.components.vacuum import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import event
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
SUPPORT_MINIMAL_SERVICES = VacuumEntityFeature.TURN_ON | VacuumEntityFeature.TURN_OFF
@@ -63,7 +63,7 @@ DEMO_VACUUM_NONE = "4_Fourth_floor"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/demo/valve.py b/homeassistant/components/demo/valve.py
new file mode 100644
index 00000000000..03f0123dd96
--- /dev/null
+++ b/homeassistant/components/demo/valve.py
@@ -0,0 +1,89 @@
+"""Demo valve platform that implements valves."""
+
+from __future__ import annotations
+
+import asyncio
+from typing import Any
+
+from homeassistant.components.valve import ValveEntity, ValveEntityFeature, ValveState
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+OPEN_CLOSE_DELAY = 2 # Used to give a realistic open/close experience in frontend
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the Demo config entry."""
+ async_add_entities(
+ [
+ DemoValve("Front Garden", ValveState.OPEN),
+ DemoValve("Orchard", ValveState.CLOSED),
+ ]
+ )
+
+
+class DemoValve(ValveEntity):
+ """Representation of a Demo valve."""
+
+ _attr_should_poll = False
+
+ def __init__(
+ self,
+ name: str,
+ state: str,
+ moveable: bool = True,
+ ) -> None:
+ """Initialize the valve."""
+ self._attr_name = name
+ if moveable:
+ self._attr_supported_features = (
+ ValveEntityFeature.OPEN | ValveEntityFeature.CLOSE
+ )
+ self._state = state
+ self._moveable = moveable
+
+ @property
+ def is_open(self) -> bool:
+ """Return true if valve is open."""
+ return self._state == ValveState.OPEN
+
+ @property
+ def is_opening(self) -> bool:
+ """Return true if valve is opening."""
+ return self._state == ValveState.OPENING
+
+ @property
+ def is_closing(self) -> bool:
+ """Return true if valve is closing."""
+ return self._state == ValveState.CLOSING
+
+ @property
+ def is_closed(self) -> bool:
+ """Return true if valve is closed."""
+ return self._state == ValveState.CLOSED
+
+ @property
+ def reports_position(self) -> bool:
+ """Return True if entity reports position, False otherwise."""
+ return False
+
+ async def async_open_valve(self, **kwargs: Any) -> None:
+ """Open the valve."""
+ self._state = ValveState.OPENING
+ self.async_write_ha_state()
+ await asyncio.sleep(OPEN_CLOSE_DELAY)
+ self._state = ValveState.OPEN
+ self.async_write_ha_state()
+
+ async def async_close_valve(self, **kwargs: Any) -> None:
+ """Close the valve."""
+ self._state = ValveState.CLOSING
+ self.async_write_ha_state()
+ await asyncio.sleep(OPEN_CLOSE_DELAY)
+ self._state = ValveState.CLOSED
+ self.async_write_ha_state()
diff --git a/homeassistant/components/demo/water_heater.py b/homeassistant/components/demo/water_heater.py
index f295780b190..9e12bb9e1d5 100644
--- a/homeassistant/components/demo/water_heater.py
+++ b/homeassistant/components/demo/water_heater.py
@@ -11,7 +11,7 @@ from homeassistant.components.water_heater import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
SUPPORT_FLAGS_HEATER = (
WaterHeaterEntityFeature.TARGET_TEMPERATURE
@@ -24,16 +24,21 @@ SUPPORT_FLAGS_HEATER = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
[
DemoWaterHeater(
- "Demo Water Heater", 119, UnitOfTemperature.FAHRENHEIT, False, "eco"
+ "Demo Water Heater", 119, UnitOfTemperature.FAHRENHEIT, False, "eco", 1
),
DemoWaterHeater(
- "Demo Water Heater Celsius", 45, UnitOfTemperature.CELSIUS, True, "eco"
+ "Demo Water Heater Celsius",
+ 45,
+ UnitOfTemperature.CELSIUS,
+ True,
+ "eco",
+ 1,
),
]
)
@@ -52,6 +57,7 @@ class DemoWaterHeater(WaterHeaterEntity):
unit_of_measurement: str,
away: bool,
current_operation: str,
+ target_temperature_step: float,
) -> None:
"""Initialize the water_heater device."""
self._attr_name = name
@@ -74,6 +80,7 @@ class DemoWaterHeater(WaterHeaterEntity):
"gas",
"off",
]
+ self._attr_target_temperature_step = target_temperature_step
def set_temperature(self, **kwargs: Any) -> None:
"""Set new target temperatures."""
diff --git a/homeassistant/components/demo/weather.py b/homeassistant/components/demo/weather.py
index 2468c54dde3..d1f829fee1b 100644
--- a/homeassistant/components/demo/weather.py
+++ b/homeassistant/components/demo/weather.py
@@ -26,7 +26,7 @@ from homeassistant.components.weather import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfPressure, UnitOfSpeed, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_interval
from homeassistant.util import dt as dt_util
@@ -58,7 +58,7 @@ WEATHER_UPDATE_INTERVAL = timedelta(minutes=30)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/denonavr/media_player.py b/homeassistant/components/denonavr/media_player.py
index 818d530ddab..a67c76f6525 100644
--- a/homeassistant/components/denonavr/media_player.py
+++ b/homeassistant/components/denonavr/media_player.py
@@ -39,7 +39,7 @@ from homeassistant.const import ATTR_COMMAND, CONF_HOST, CONF_MODEL, CONF_TYPE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DenonavrConfigEntry
from .const import (
@@ -109,7 +109,7 @@ DENON_STATE_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DenonavrConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DenonAVR receiver from a config entry."""
entities = []
diff --git a/homeassistant/components/denonavr/strings.json b/homeassistant/components/denonavr/strings.json
index 6c055c5932a..192ab3bd71f 100644
--- a/homeassistant/components/denonavr/strings.json
+++ b/homeassistant/components/denonavr/strings.json
@@ -23,14 +23,14 @@
}
},
"error": {
- "discovery_error": "Failed to discover a Denon AVR Network Receiver"
+ "discovery_error": "Failed to discover a Denon AVR network receiver"
},
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]",
"already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]",
- "cannot_connect": "Failed to connect, please try again, disconnecting mains power and ethernet cables and reconnecting them may help",
- "not_denonavr_manufacturer": "Not a Denon AVR Network Receiver, discovered manufacturer did not match",
- "not_denonavr_missing": "Not a Denon AVR Network Receiver, discovery information not complete"
+ "cannot_connect": "Failed to connect, please try again, disconnecting mains power and Ethernet cables and reconnecting them may help",
+ "not_denonavr_manufacturer": "Not a Denon AVR network receiver, discovered manufacturer did not match",
+ "not_denonavr_missing": "Not a Denon AVR network receiver, discovery information not complete"
}
},
"options": {
@@ -64,7 +64,7 @@
"fields": {
"dynamic_eq": {
"name": "Dynamic equalizer",
- "description": "True/false for enable/disable."
+ "description": "Whether DynamicEQ should be enabled or disabled."
}
}
},
diff --git a/homeassistant/components/derivative/sensor.py b/homeassistant/components/derivative/sensor.py
index 988da5e938b..f6c2b45ef9c 100644
--- a/homeassistant/components/derivative/sensor.py
+++ b/homeassistant/components/derivative/sensor.py
@@ -24,12 +24,25 @@ from homeassistant.const import (
STATE_UNKNOWN,
UnitOfTime,
)
-from homeassistant.core import Event, EventStateChangedData, HomeAssistant, callback
+from homeassistant.core import (
+ Event,
+ EventStateChangedData,
+ EventStateReportedData,
+ HomeAssistant,
+ State,
+ callback,
+)
from homeassistant.helpers import config_validation as cv, entity_registry as er
from homeassistant.helpers.device import async_device_info_to_link_from_entity
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.event import async_track_state_change_event
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
+from homeassistant.helpers.event import (
+ async_track_state_change_event,
+ async_track_state_report_event,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import (
@@ -83,7 +96,7 @@ PLATFORM_SCHEMA = SENSOR_PLATFORM_SCHEMA.extend(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Derivative config entry."""
registry = er.async_get(hass)
@@ -197,13 +210,33 @@ class DerivativeSensor(RestoreSensor, SensorEntity):
_LOGGER.warning("Could not restore last state: %s", err)
@callback
- def calc_derivative(event: Event[EventStateChangedData]) -> None:
+ def on_state_reported(event: Event[EventStateReportedData]) -> None:
+ """Handle constant sensor state."""
+ if self._attr_native_value == Decimal(0):
+ # If the derivative is zero, and the source sensor hasn't
+ # changed state, then we know it will still be zero.
+ return
+ new_state = event.data["new_state"]
+ if new_state is not None:
+ calc_derivative(
+ new_state, new_state.state, event.data["old_last_reported"]
+ )
+
+ @callback
+ def on_state_changed(event: Event[EventStateChangedData]) -> None:
+ """Handle changed sensor state."""
+ new_state = event.data["new_state"]
+ old_state = event.data["old_state"]
+ if new_state is not None and old_state is not None:
+ calc_derivative(new_state, old_state.state, old_state.last_reported)
+
+ def calc_derivative(
+ new_state: State, old_value: str, old_last_reported: datetime
+ ) -> None:
"""Handle the sensor state changes."""
- if (
- (old_state := event.data["old_state"]) is None
- or old_state.state in (STATE_UNKNOWN, STATE_UNAVAILABLE)
- or (new_state := event.data["new_state"]) is None
- or new_state.state in (STATE_UNKNOWN, STATE_UNAVAILABLE)
+ if old_value in (STATE_UNKNOWN, STATE_UNAVAILABLE) or new_state.state in (
+ STATE_UNKNOWN,
+ STATE_UNAVAILABLE,
):
return
@@ -217,15 +250,15 @@ class DerivativeSensor(RestoreSensor, SensorEntity):
self._state_list = [
(time_start, time_end, state)
for time_start, time_end, state in self._state_list
- if (new_state.last_updated - time_end).total_seconds()
+ if (new_state.last_reported - time_end).total_seconds()
< self._time_window
]
try:
elapsed_time = (
- new_state.last_updated - old_state.last_updated
+ new_state.last_reported - old_last_reported
).total_seconds()
- delta_value = Decimal(new_state.state) - Decimal(old_state.state)
+ delta_value = Decimal(new_state.state) - Decimal(old_value)
new_derivative = (
delta_value
/ Decimal(elapsed_time)
@@ -237,7 +270,7 @@ class DerivativeSensor(RestoreSensor, SensorEntity):
_LOGGER.warning("While calculating derivative: %s", err)
except DecimalException as err:
_LOGGER.warning(
- "Invalid state (%s > %s): %s", old_state.state, new_state.state, err
+ "Invalid state (%s > %s): %s", old_value, new_state.state, err
)
except AssertionError as err:
_LOGGER.error("Could not calculate derivative: %s", err)
@@ -254,7 +287,7 @@ class DerivativeSensor(RestoreSensor, SensorEntity):
# add latest derivative to the window list
self._state_list.append(
- (old_state.last_updated, new_state.last_updated, new_derivative)
+ (old_last_reported, new_state.last_reported, new_derivative)
)
def calculate_weight(
@@ -274,13 +307,19 @@ class DerivativeSensor(RestoreSensor, SensorEntity):
else:
derivative = Decimal("0.00")
for start, end, value in self._state_list:
- weight = calculate_weight(start, end, new_state.last_updated)
+ weight = calculate_weight(start, end, new_state.last_reported)
derivative = derivative + (value * Decimal(weight))
self._attr_native_value = round(derivative, self._round_digits)
self.async_write_ha_state()
self.async_on_remove(
async_track_state_change_event(
- self.hass, self._sensor_source_id, calc_derivative
+ self.hass, self._sensor_source_id, on_state_changed
+ )
+ )
+
+ self.async_on_remove(
+ async_track_state_report_event(
+ self.hass, self._sensor_source_id, on_state_reported
)
)
diff --git a/homeassistant/components/devialet/media_player.py b/homeassistant/components/devialet/media_player.py
index 04ec58723cf..6fa6d40e17d 100644
--- a/homeassistant/components/devialet/media_player.py
+++ b/homeassistant/components/devialet/media_player.py
@@ -12,7 +12,7 @@ from homeassistant.components.media_player import (
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER, SOUND_MODES
@@ -38,7 +38,7 @@ DEVIALET_TO_HA_FEATURE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevialetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Devialet entry."""
async_add_entities([DevialetMediaPlayerEntity(entry.runtime_data)])
diff --git a/homeassistant/components/devolo_home_control/binary_sensor.py b/homeassistant/components/devolo_home_control/binary_sensor.py
index d24033a80b9..7a88b12c48a 100644
--- a/homeassistant/components/devolo_home_control/binary_sensor.py
+++ b/homeassistant/components/devolo_home_control/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeControlConfigEntry
from .entity import DevoloDeviceEntity
@@ -29,7 +29,7 @@ DEVICE_CLASS_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all binary sensor and multi level sensor devices and setup them via config entry."""
entities: list[BinarySensorEntity] = []
diff --git a/homeassistant/components/devolo_home_control/climate.py b/homeassistant/components/devolo_home_control/climate.py
index 1f407eb6804..3fdfa60870a 100644
--- a/homeassistant/components/devolo_home_control/climate.py
+++ b/homeassistant/components/devolo_home_control/climate.py
@@ -15,7 +15,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import PRECISION_HALVES, PRECISION_TENTHS, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeControlConfigEntry
from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
@@ -24,7 +24,7 @@ from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all cover devices and setup them via config entry."""
diff --git a/homeassistant/components/devolo_home_control/cover.py b/homeassistant/components/devolo_home_control/cover.py
index f49a9d0f0be..f23244f1b50 100644
--- a/homeassistant/components/devolo_home_control/cover.py
+++ b/homeassistant/components/devolo_home_control/cover.py
@@ -10,7 +10,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeControlConfigEntry
from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
@@ -19,7 +19,7 @@ from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all cover devices and setup them via config entry."""
diff --git a/homeassistant/components/devolo_home_control/light.py b/homeassistant/components/devolo_home_control/light.py
index c855574b83a..8a88081ed05 100644
--- a/homeassistant/components/devolo_home_control/light.py
+++ b/homeassistant/components/devolo_home_control/light.py
@@ -9,7 +9,7 @@ from devolo_home_control_api.homecontrol import HomeControl
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeControlConfigEntry
from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
@@ -18,7 +18,7 @@ from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all light devices and setup them via config entry."""
diff --git a/homeassistant/components/devolo_home_control/sensor.py b/homeassistant/components/devolo_home_control/sensor.py
index 8d0a7f0313c..22581267eea 100644
--- a/homeassistant/components/devolo_home_control/sensor.py
+++ b/homeassistant/components/devolo_home_control/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeControlConfigEntry
from .entity import DevoloDeviceEntity
@@ -40,7 +40,7 @@ STATE_CLASS_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all sensor devices and setup them via config entry."""
entities: list[SensorEntity] = []
diff --git a/homeassistant/components/devolo_home_control/siren.py b/homeassistant/components/devolo_home_control/siren.py
index e896f4d3ed8..5e4df944b3c 100644
--- a/homeassistant/components/devolo_home_control/siren.py
+++ b/homeassistant/components/devolo_home_control/siren.py
@@ -7,7 +7,7 @@ from devolo_home_control_api.homecontrol import HomeControl
from homeassistant.components.siren import ATTR_TONE, SirenEntity, SirenEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeControlConfigEntry
from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
@@ -16,7 +16,7 @@ from .devolo_multi_level_switch import DevoloMultiLevelSwitchDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all binary sensor and multi level sensor devices and setup them via config entry."""
diff --git a/homeassistant/components/devolo_home_control/switch.py b/homeassistant/components/devolo_home_control/switch.py
index a6f16229046..378e23a5f5f 100644
--- a/homeassistant/components/devolo_home_control/switch.py
+++ b/homeassistant/components/devolo_home_control/switch.py
@@ -9,7 +9,7 @@ from devolo_home_control_api.homecontrol import HomeControl
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeControlConfigEntry
from .entity import DevoloDeviceEntity
@@ -18,7 +18,7 @@ from .entity import DevoloDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and setup the switch devices via config entry."""
diff --git a/homeassistant/components/devolo_home_network/binary_sensor.py b/homeassistant/components/devolo_home_network/binary_sensor.py
index 5752956ffb5..2c258d758da 100644
--- a/homeassistant/components/devolo_home_network/binary_sensor.py
+++ b/homeassistant/components/devolo_home_network/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeNetworkConfigEntry
from .const import CONNECTED_PLC_DEVICES, CONNECTED_TO_ROUTER
@@ -54,7 +54,7 @@ SENSOR_TYPES: dict[str, DevoloBinarySensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and sensors and setup them via config entry."""
coordinators = entry.runtime_data.coordinators
diff --git a/homeassistant/components/devolo_home_network/button.py b/homeassistant/components/devolo_home_network/button.py
index 06822ff199e..fe6b1786363 100644
--- a/homeassistant/components/devolo_home_network/button.py
+++ b/homeassistant/components/devolo_home_network/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeNetworkConfigEntry
from .const import DOMAIN, IDENTIFY, PAIRING, RESTART, START_WPS
@@ -59,7 +59,7 @@ BUTTON_TYPES: dict[str, DevoloButtonEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and buttons and setup them via config entry."""
device = entry.runtime_data.device
diff --git a/homeassistant/components/devolo_home_network/device_tracker.py b/homeassistant/components/devolo_home_network/device_tracker.py
index 583f022df84..c5862738bd1 100644
--- a/homeassistant/components/devolo_home_network/device_tracker.py
+++ b/homeassistant/components/devolo_home_network/device_tracker.py
@@ -12,7 +12,7 @@ from homeassistant.components.device_tracker import (
from homeassistant.const import STATE_UNKNOWN, UnitOfFrequency
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import DevoloHomeNetworkConfigEntry
@@ -25,7 +25,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and sensors and setup them via config entry."""
device = entry.runtime_data.device
diff --git a/homeassistant/components/devolo_home_network/entity.py b/homeassistant/components/devolo_home_network/entity.py
index 93ec1b9a3a2..64d8ff131e8 100644
--- a/homeassistant/components/devolo_home_network/entity.py
+++ b/homeassistant/components/devolo_home_network/entity.py
@@ -8,6 +8,7 @@ from devolo_plc_api.device_api import (
WifiGuestAccessGet,
)
from devolo_plc_api.plcnet_api import DataRate, LogicalNetwork
+from yarl import URL
from homeassistant.const import ATTR_CONNECTIONS
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
@@ -43,7 +44,7 @@ class DevoloEntity(Entity):
self.entry = entry
self._attr_device_info = DeviceInfo(
- configuration_url=f"http://{self.device.ip}",
+ configuration_url=URL.build(scheme="http", host=self.device.ip),
identifiers={(DOMAIN, str(self.device.serial_number))},
manufacturer="devolo",
model=self.device.product,
diff --git a/homeassistant/components/devolo_home_network/image.py b/homeassistant/components/devolo_home_network/image.py
index 91e8dd83b7d..46a3eb3426a 100644
--- a/homeassistant/components/devolo_home_network/image.py
+++ b/homeassistant/components/devolo_home_network/image.py
@@ -12,7 +12,7 @@ from devolo_plc_api.device_api import WifiGuestAccessGet
from homeassistant.components.image import ImageEntity, ImageEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import DevoloHomeNetworkConfigEntry
@@ -42,7 +42,7 @@ IMAGE_TYPES: dict[str, DevoloImageEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and sensors and setup them via config entry."""
coordinators = entry.runtime_data.coordinators
diff --git a/homeassistant/components/devolo_home_network/sensor.py b/homeassistant/components/devolo_home_network/sensor.py
index 220ab66312a..d9a6f3f1110 100644
--- a/homeassistant/components/devolo_home_network/sensor.py
+++ b/homeassistant/components/devolo_home_network/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory, UnitOfDataRate
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from . import DevoloHomeNetworkConfigEntry
@@ -123,7 +123,7 @@ SENSOR_TYPES: dict[str, DevoloSensorEntityDescription[Any, Any]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and sensors and setup them via config entry."""
device = entry.runtime_data.device
diff --git a/homeassistant/components/devolo_home_network/switch.py b/homeassistant/components/devolo_home_network/switch.py
index 8ff35dcc4b6..0271270fa09 100644
--- a/homeassistant/components/devolo_home_network/switch.py
+++ b/homeassistant/components/devolo_home_network/switch.py
@@ -14,7 +14,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeNetworkConfigEntry
from .const import DOMAIN, SWITCH_GUEST_WIFI, SWITCH_LEDS
@@ -55,7 +55,7 @@ SWITCH_TYPES: dict[str, DevoloSwitchEntityDescription[Any]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and sensors and setup them via config entry."""
device = entry.runtime_data.device
diff --git a/homeassistant/components/devolo_home_network/update.py b/homeassistant/components/devolo_home_network/update.py
index 5091ce8e1e7..aaaf72af359 100644
--- a/homeassistant/components/devolo_home_network/update.py
+++ b/homeassistant/components/devolo_home_network/update.py
@@ -19,7 +19,7 @@ from homeassistant.components.update import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DevoloHomeNetworkConfigEntry
from .const import DOMAIN, REGULAR_FIRMWARE
@@ -51,7 +51,7 @@ UPDATE_TYPES: dict[str, DevoloUpdateEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: DevoloHomeNetworkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Get all devices and sensors and setup them via config entry."""
coordinators = entry.runtime_data.coordinators
diff --git a/homeassistant/components/dexcom/sensor.py b/homeassistant/components/dexcom/sensor.py
index cdb1894b675..eac0134f010 100644
--- a/homeassistant/components/dexcom/sensor.py
+++ b/homeassistant/components/dexcom/sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import CONF_USERNAME, UnitOfBloodGlucoseConcentration
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -26,7 +26,7 @@ TRENDS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DexcomConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Dexcom sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/dhcp/manifest.json b/homeassistant/components/dhcp/manifest.json
index 45af4f1b5dd..64fd2ff38c6 100644
--- a/homeassistant/components/dhcp/manifest.json
+++ b/homeassistant/components/dhcp/manifest.json
@@ -14,8 +14,8 @@
],
"quality_scale": "internal",
"requirements": [
- "aiodhcpwatcher==1.1.0",
- "aiodiscover==2.2.2",
- "cached-ipaddress==0.8.0"
+ "aiodhcpwatcher==1.1.1",
+ "aiodiscover==2.6.1",
+ "cached-ipaddress==0.10.0"
]
}
diff --git a/homeassistant/components/directv/media_player.py b/homeassistant/components/directv/media_player.py
index 8998e050a75..91934a2da3a 100644
--- a/homeassistant/components/directv/media_player.py
+++ b/homeassistant/components/directv/media_player.py
@@ -15,7 +15,7 @@ from homeassistant.components.media_player import (
MediaType,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import DirecTVConfigEntry
@@ -55,7 +55,7 @@ SUPPORT_DTV_CLIENT = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DirecTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DirecTV config entry."""
dtv = entry.runtime_data
diff --git a/homeassistant/components/directv/remote.py b/homeassistant/components/directv/remote.py
index dbaab5fa4e6..c9aacaae4d3 100644
--- a/homeassistant/components/directv/remote.py
+++ b/homeassistant/components/directv/remote.py
@@ -11,7 +11,7 @@ from directv import DIRECTV, DIRECTVError
from homeassistant.components.remote import ATTR_NUM_REPEATS, RemoteEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DirecTVConfigEntry
from .entity import DIRECTVEntity
@@ -24,7 +24,7 @@ SCAN_INTERVAL = timedelta(minutes=2)
async def async_setup_entry(
hass: HomeAssistant,
entry: DirecTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load DirecTV remote based on a config entry."""
dtv = entry.runtime_data
diff --git a/homeassistant/components/discovergy/coordinator.py b/homeassistant/components/discovergy/coordinator.py
index d4ef87049b8..e3f26ad49f8 100644
--- a/homeassistant/components/discovergy/coordinator.py
+++ b/homeassistant/components/discovergy/coordinator.py
@@ -51,9 +51,7 @@ class DiscovergyUpdateCoordinator(DataUpdateCoordinator[Reading]):
)
except InvalidLogin as err:
raise ConfigEntryAuthFailed(
- f"Auth expired while fetching last reading for meter {self.meter.meter_id}"
+ "Auth expired while fetching last reading"
) from err
except (HTTPError, DiscovergyClientError) as err:
- raise UpdateFailed(
- f"Error while fetching last reading for meter {self.meter.meter_id}"
- ) from err
+ raise UpdateFailed(f"Error while fetching last reading: {err}") from err
diff --git a/homeassistant/components/discovergy/sensor.py b/homeassistant/components/discovergy/sensor.py
index 65b1722e0d8..7d4bb6cb052 100644
--- a/homeassistant/components/discovergy/sensor.py
+++ b/homeassistant/components/discovergy/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER
@@ -166,7 +166,7 @@ ADDITIONAL_SENSORS: tuple[DiscovergySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DiscovergyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Discovergy sensors."""
entities: list[DiscovergySensor] = []
diff --git a/homeassistant/components/dlink/switch.py b/homeassistant/components/dlink/switch.py
index 54322cc6875..ef1348f613d 100644
--- a/homeassistant/components/dlink/switch.py
+++ b/homeassistant/components/dlink/switch.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DLinkConfigEntry
from .const import ATTR_TOTAL_CONSUMPTION
@@ -24,7 +24,7 @@ SWITCH_TYPE = SwitchEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: DLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the D-Link Power Plug switch."""
async_add_entities([SmartPlugSwitch(entry, SWITCH_TYPE)], True)
diff --git a/homeassistant/components/dlna_dmr/media_player.py b/homeassistant/components/dlna_dmr/media_player.py
index 563ed209b7d..d93d55e62be 100644
--- a/homeassistant/components/dlna_dmr/media_player.py
+++ b/homeassistant/components/dlna_dmr/media_player.py
@@ -32,7 +32,7 @@ from homeassistant.components.media_player import (
from homeassistant.const import CONF_DEVICE_ID, CONF_MAC, CONF_TYPE, CONF_URL
from homeassistant.core import CoreState, HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.ssdp import SsdpServiceInfo
from .const import (
@@ -92,7 +92,7 @@ def catch_request_errors[_DlnaDmrEntityT: DlnaDmrEntity, **_P, _R](
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DlnaDmrEntity from a config entry."""
_LOGGER.debug("media_player.async_setup_entry %s (%s)", entry.entry_id, entry.title)
diff --git a/homeassistant/components/dnsip/sensor.py b/homeassistant/components/dnsip/sensor.py
index 34730e934a0..6708baefe8c 100644
--- a/homeassistant/components/dnsip/sensor.py
+++ b/homeassistant/components/dnsip/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, CONF_PORT
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_HOSTNAME,
@@ -45,7 +45,9 @@ def sort_ips(ips: list, querytype: str) -> list:
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the dnsip sensor entry."""
diff --git a/homeassistant/components/doorbird/button.py b/homeassistant/components/doorbird/button.py
index 62631e51abc..173c2e923e4 100644
--- a/homeassistant/components/doorbird/button.py
+++ b/homeassistant/components/doorbird/button.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .device import ConfiguredDoorBird, async_reset_device_favorites
from .entity import DoorBirdEntity
@@ -45,7 +45,7 @@ BUTTON_DESCRIPTIONS: tuple[DoorbirdButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DoorBirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DoorBird button platform."""
door_bird_data = config_entry.runtime_data
diff --git a/homeassistant/components/doorbird/camera.py b/homeassistant/components/doorbird/camera.py
index 45f37527ac1..a41e7c41b28 100644
--- a/homeassistant/components/doorbird/camera.py
+++ b/homeassistant/components/doorbird/camera.py
@@ -9,7 +9,7 @@ import aiohttp
from homeassistant.components.camera import Camera, CameraEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .entity import DoorBirdEntity
@@ -25,7 +25,7 @@ _TIMEOUT = 15 # seconds
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DoorBirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DoorBird camera platform."""
door_bird_data = config_entry.runtime_data
diff --git a/homeassistant/components/doorbird/event.py b/homeassistant/components/doorbird/event.py
index 4c20098fc80..688f8b2fbeb 100644
--- a/homeassistant/components/doorbird/event.py
+++ b/homeassistant/components/doorbird/event.py
@@ -9,7 +9,7 @@ from homeassistant.components.event import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .device import DoorbirdEvent
@@ -35,7 +35,7 @@ EVENT_DESCRIPTIONS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DoorBirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DoorBird event platform."""
door_bird_data = config_entry.runtime_data
diff --git a/homeassistant/components/doorbird/strings.json b/homeassistant/components/doorbird/strings.json
index 090ba4f161f..ad43e8c1c1c 100644
--- a/homeassistant/components/doorbird/strings.json
+++ b/homeassistant/components/doorbird/strings.json
@@ -6,7 +6,7 @@
"events": "Comma separated list of events."
},
"data_description": {
- "events": "Add an comma separated event name for each event you wish to track. After entering them here, use the DoorBird app to assign them to a specific event.\n\nExample: somebody_pressed_the_button, motion"
+ "events": "Add a comma separated event name for each event you wish to track. After entering them here, use the DoorBird app to assign them to a specific event.\n\nExample: somebody_pressed_the_button, motion"
}
}
}
diff --git a/homeassistant/components/dormakaba_dkey/binary_sensor.py b/homeassistant/components/dormakaba_dkey/binary_sensor.py
index 56b991bf908..a8870ed224b 100644
--- a/homeassistant/components/dormakaba_dkey/binary_sensor.py
+++ b/homeassistant/components/dormakaba_dkey/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import DormakabaDkeyConfigEntry, DormakabaDkeyCoordinator
from .entity import DormakabaDkeyEntity
@@ -45,7 +45,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DormakabaDkeyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary sensor platform for Dormakaba dKey."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/dormakaba_dkey/config_flow.py b/homeassistant/components/dormakaba_dkey/config_flow.py
index 0d23b822231..369accb83d8 100644
--- a/homeassistant/components/dormakaba_dkey/config_flow.py
+++ b/homeassistant/components/dormakaba_dkey/config_flow.py
@@ -57,7 +57,7 @@ class DormkabaConfigFlow(ConfigFlow, domain=DOMAIN):
self._discovery_info = self._discovered_devices[address]
return await self.async_step_associate()
- current_addresses = self._async_current_ids()
+ current_addresses = self._async_current_ids(include_ignore=False)
for discovery in async_discovered_service_info(self.hass):
if (
discovery.address in current_addresses
diff --git a/homeassistant/components/dormakaba_dkey/lock.py b/homeassistant/components/dormakaba_dkey/lock.py
index 352e7cbe0ac..12a553adba3 100644
--- a/homeassistant/components/dormakaba_dkey/lock.py
+++ b/homeassistant/components/dormakaba_dkey/lock.py
@@ -8,7 +8,7 @@ from py_dormakaba_dkey.commands import UnlockStatus
from homeassistant.components.lock import LockEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import DormakabaDkeyConfigEntry, DormakabaDkeyCoordinator
from .entity import DormakabaDkeyEntity
@@ -17,7 +17,7 @@ from .entity import DormakabaDkeyEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DormakabaDkeyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the lock platform for Dormakaba dKey."""
async_add_entities([DormakabaDkeyLock(entry.runtime_data)])
diff --git a/homeassistant/components/dormakaba_dkey/sensor.py b/homeassistant/components/dormakaba_dkey/sensor.py
index b1e941bc7e1..413ea1c56b1 100644
--- a/homeassistant/components/dormakaba_dkey/sensor.py
+++ b/homeassistant/components/dormakaba_dkey/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import DormakabaDkeyConfigEntry, DormakabaDkeyCoordinator
from .entity import DormakabaDkeyEntity
@@ -28,7 +28,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DormakabaDkeyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the lock platform for Dormakaba dKey."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/dremel_3d_printer/__init__.py b/homeassistant/components/dremel_3d_printer/__init__.py
index 632c42d9b54..33a8ad0e67f 100644
--- a/homeassistant/components/dremel_3d_printer/__init__.py
+++ b/homeassistant/components/dremel_3d_printer/__init__.py
@@ -29,7 +29,7 @@ async def async_setup_entry(
f"Unable to connect to Dremel 3D Printer: {ex}"
) from ex
- coordinator = Dremel3DPrinterDataUpdateCoordinator(hass, api)
+ coordinator = Dremel3DPrinterDataUpdateCoordinator(hass, config_entry, api)
await coordinator.async_config_entry_first_refresh()
config_entry.runtime_data = coordinator
platforms = list(PLATFORMS)
diff --git a/homeassistant/components/dremel_3d_printer/binary_sensor.py b/homeassistant/components/dremel_3d_printer/binary_sensor.py
index 972945a84bb..923bcdad09c 100644
--- a/homeassistant/components/dremel_3d_printer/binary_sensor.py
+++ b/homeassistant/components/dremel_3d_printer/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import DremelConfigEntry
from .entity import Dremel3DPrinterEntity
@@ -43,7 +43,7 @@ BINARY_SENSOR_TYPES: tuple[Dremel3DPrinterBinarySensorEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DremelConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the available Dremel binary sensors."""
async_add_entities(
diff --git a/homeassistant/components/dremel_3d_printer/button.py b/homeassistant/components/dremel_3d_printer/button.py
index f91c1b0ea51..880b179650f 100644
--- a/homeassistant/components/dremel_3d_printer/button.py
+++ b/homeassistant/components/dremel_3d_printer/button.py
@@ -10,7 +10,7 @@ from dremel3dpy import Dremel3DPrinter
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import DremelConfigEntry
from .entity import Dremel3DPrinterEntity
@@ -45,7 +45,7 @@ BUTTON_TYPES: tuple[Dremel3DPrinterButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DremelConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Dremel 3D Printer control buttons."""
async_add_entities(
diff --git a/homeassistant/components/dremel_3d_printer/camera.py b/homeassistant/components/dremel_3d_printer/camera.py
index f4293915a25..ccb7eeaa658 100644
--- a/homeassistant/components/dremel_3d_printer/camera.py
+++ b/homeassistant/components/dremel_3d_printer/camera.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.camera import CameraEntityDescription
from homeassistant.components.mjpeg import MjpegCamera
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import Dremel3DPrinterDataUpdateCoordinator, DremelConfigEntry
from .entity import Dremel3DPrinterEntity
@@ -19,7 +19,7 @@ CAMERA_TYPE = CameraEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DremelConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a MJPEG IP Camera for the 3D45 Model. The 3D20 and 3D40 models don't have built in cameras."""
async_add_entities([Dremel3D45Camera(config_entry.runtime_data, CAMERA_TYPE)])
diff --git a/homeassistant/components/dremel_3d_printer/coordinator.py b/homeassistant/components/dremel_3d_printer/coordinator.py
index 3323569c05f..f2c1876fe0a 100644
--- a/homeassistant/components/dremel_3d_printer/coordinator.py
+++ b/homeassistant/components/dremel_3d_printer/coordinator.py
@@ -18,11 +18,14 @@ class Dremel3DPrinterDataUpdateCoordinator(DataUpdateCoordinator[None]):
config_entry: DremelConfigEntry
- def __init__(self, hass: HomeAssistant, api: Dremel3DPrinter) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: DremelConfigEntry, api: Dremel3DPrinter
+ ) -> None:
"""Initialize Dremel 3D Printer data update coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=10),
)
diff --git a/homeassistant/components/dremel_3d_printer/sensor.py b/homeassistant/components/dremel_3d_printer/sensor.py
index 002a5fc4adb..1f02b1fe239 100644
--- a/homeassistant/components/dremel_3d_printer/sensor.py
+++ b/homeassistant/components/dremel_3d_printer/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
from homeassistant.util.variance import ignore_variance
@@ -235,7 +235,7 @@ SENSOR_TYPES: tuple[Dremel3DPrinterSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DremelConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the available Dremel 3D Printer sensors."""
async_add_entities(
diff --git a/homeassistant/components/drop_connect/binary_sensor.py b/homeassistant/components/drop_connect/binary_sensor.py
index bc8cf900610..f133be431f0 100644
--- a/homeassistant/components/drop_connect/binary_sensor.py
+++ b/homeassistant/components/drop_connect/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_DEVICE_TYPE,
@@ -105,7 +105,7 @@ DEVICE_BINARY_SENSORS: dict[str, list[str]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DROPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DROP binary sensors from config entry."""
_LOGGER.debug(
diff --git a/homeassistant/components/drop_connect/select.py b/homeassistant/components/drop_connect/select.py
index 9e4c74b67e6..e198033d0f7 100644
--- a/homeassistant/components/drop_connect/select.py
+++ b/homeassistant/components/drop_connect/select.py
@@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_DEVICE_TYPE, DEV_HUB
from .coordinator import DROPConfigEntry, DROPDeviceDataUpdateCoordinator
@@ -50,7 +50,7 @@ DEVICE_SELECTS: dict[str, list[str]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DROPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DROP selects from config entry."""
_LOGGER.debug(
diff --git a/homeassistant/components/drop_connect/sensor.py b/homeassistant/components/drop_connect/sensor.py
index 5ec47ed9eb1..c69e2e12ea0 100644
--- a/homeassistant/components/drop_connect/sensor.py
+++ b/homeassistant/components/drop_connect/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_DEVICE_TYPE,
@@ -242,7 +242,7 @@ DEVICE_SENSORS: dict[str, list[str]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DROPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DROP sensors from config entry."""
_LOGGER.debug(
diff --git a/homeassistant/components/drop_connect/switch.py b/homeassistant/components/drop_connect/switch.py
index 404059d3196..d52d17c5ea0 100644
--- a/homeassistant/components/drop_connect/switch.py
+++ b/homeassistant/components/drop_connect/switch.py
@@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_DEVICE_TYPE,
@@ -65,7 +65,7 @@ DEVICE_SWITCHES: dict[str, list[str]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DROPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DROP switches from config entry."""
_LOGGER.debug(
diff --git a/homeassistant/components/dsmr/config_flow.py b/homeassistant/components/dsmr/config_flow.py
index 7d6a641b006..6f15f99517b 100644
--- a/homeassistant/components/dsmr/config_flow.py
+++ b/homeassistant/components/dsmr/config_flow.py
@@ -59,7 +59,7 @@ class DSMRConnection:
self._equipment_identifier = obis_ref.EQUIPMENT_IDENTIFIER
if dsmr_version == "5B":
self._equipment_identifier = obis_ref.BELGIUM_EQUIPMENT_IDENTIFIER
- if dsmr_version == "5L":
+ if dsmr_version in ("5L", "5EONHU"):
self._equipment_identifier = obis_ref.LUXEMBOURG_EQUIPMENT_IDENTIFIER
if dsmr_version == "Q3D":
self._equipment_identifier = obis_ref.Q3D_EQUIPMENT_IDENTIFIER
diff --git a/homeassistant/components/dsmr/const.py b/homeassistant/components/dsmr/const.py
index 4c6cb31ca4d..2682b4df1cc 100644
--- a/homeassistant/components/dsmr/const.py
+++ b/homeassistant/components/dsmr/const.py
@@ -28,7 +28,7 @@ DEVICE_NAME_GAS = "Gas Meter"
DEVICE_NAME_WATER = "Water Meter"
DEVICE_NAME_HEAT = "Heat Meter"
-DSMR_VERSIONS = {"2.2", "4", "5", "5B", "5L", "5S", "Q3D"}
+DSMR_VERSIONS = {"2.2", "4", "5", "5B", "5L", "5S", "Q3D", "5EONHU"}
DSMR_PROTOCOL = "dsmr_protocol"
RFXTRX_DSMR_PROTOCOL = "rfxtrx_dsmr_protocol"
diff --git a/homeassistant/components/dsmr/sensor.py b/homeassistant/components/dsmr/sensor.py
index e05785b8b26..ba528271824 100644
--- a/homeassistant/components/dsmr/sensor.py
+++ b/homeassistant/components/dsmr/sensor.py
@@ -43,7 +43,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import Throttle
@@ -115,7 +115,7 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
key="electricity_active_tariff",
translation_key="electricity_active_tariff",
obis_reference="ELECTRICITY_ACTIVE_TARIFF",
- dsmr_versions={"2.2", "4", "5", "5B", "5L"},
+ dsmr_versions={"2.2", "4", "5", "5B", "5L", "5EONHU"},
device_class=SensorDeviceClass.ENUM,
options=["low", "normal"],
),
@@ -123,7 +123,7 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
key="electricity_used_tariff_1",
translation_key="electricity_used_tariff_1",
obis_reference="ELECTRICITY_USED_TARIFF_1",
- dsmr_versions={"2.2", "4", "5", "5B", "5L"},
+ dsmr_versions={"2.2", "4", "5", "5B", "5L", "5EONHU"},
device_class=SensorDeviceClass.ENERGY,
state_class=SensorStateClass.TOTAL_INCREASING,
),
@@ -131,7 +131,25 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
key="electricity_used_tariff_2",
translation_key="electricity_used_tariff_2",
obis_reference="ELECTRICITY_USED_TARIFF_2",
- dsmr_versions={"2.2", "4", "5", "5B", "5L"},
+ dsmr_versions={"2.2", "4", "5", "5B", "5L", "5EONHU"},
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ ),
+ DSMRSensorEntityDescription(
+ key="electricity_used_tariff_3",
+ translation_key="electricity_used_tariff_3",
+ obis_reference="ELECTRICITY_USED_TARIFF_3",
+ dsmr_versions={"5EONHU"},
+ force_update=True,
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ ),
+ DSMRSensorEntityDescription(
+ key="electricity_used_tariff_4",
+ translation_key="electricity_used_tariff_4",
+ obis_reference="ELECTRICITY_USED_TARIFF_4",
+ dsmr_versions={"5EONHU"},
+ force_update=True,
device_class=SensorDeviceClass.ENERGY,
state_class=SensorStateClass.TOTAL_INCREASING,
),
@@ -139,7 +157,7 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
key="electricity_delivered_tariff_1",
translation_key="electricity_delivered_tariff_1",
obis_reference="ELECTRICITY_DELIVERED_TARIFF_1",
- dsmr_versions={"2.2", "4", "5", "5B", "5L"},
+ dsmr_versions={"2.2", "4", "5", "5B", "5L", "5EONHU"},
device_class=SensorDeviceClass.ENERGY,
state_class=SensorStateClass.TOTAL_INCREASING,
),
@@ -147,7 +165,25 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
key="electricity_delivered_tariff_2",
translation_key="electricity_delivered_tariff_2",
obis_reference="ELECTRICITY_DELIVERED_TARIFF_2",
- dsmr_versions={"2.2", "4", "5", "5B", "5L"},
+ dsmr_versions={"2.2", "4", "5", "5B", "5L", "5EONHU"},
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ ),
+ DSMRSensorEntityDescription(
+ key="electricity_delivered_tariff_3",
+ translation_key="electricity_delivered_tariff_3",
+ obis_reference="ELECTRICITY_DELIVERED_TARIFF_3",
+ dsmr_versions={"5EONHU"},
+ force_update=True,
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ ),
+ DSMRSensorEntityDescription(
+ key="electricity_delivered_tariff_4",
+ translation_key="electricity_delivered_tariff_4",
+ obis_reference="ELECTRICITY_DELIVERED_TARIFF_4",
+ dsmr_versions={"5EONHU"},
+ force_update=True,
device_class=SensorDeviceClass.ENERGY,
state_class=SensorStateClass.TOTAL_INCREASING,
),
@@ -341,7 +377,7 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
key="electricity_imported_total",
translation_key="electricity_imported_total",
obis_reference="ELECTRICITY_IMPORTED_TOTAL",
- dsmr_versions={"5L", "5S", "Q3D"},
+ dsmr_versions={"5L", "5S", "Q3D", "5EONHU"},
device_class=SensorDeviceClass.ENERGY,
state_class=SensorStateClass.TOTAL_INCREASING,
),
@@ -349,7 +385,7 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
key="electricity_exported_total",
translation_key="electricity_exported_total",
obis_reference="ELECTRICITY_EXPORTED_TOTAL",
- dsmr_versions={"5L", "5S", "Q3D"},
+ dsmr_versions={"5L", "5S", "Q3D", "5EONHU"},
device_class=SensorDeviceClass.ENERGY,
state_class=SensorStateClass.TOTAL_INCREASING,
),
@@ -387,6 +423,113 @@ SENSORS: tuple[DSMRSensorEntityDescription, ...] = (
device_class=SensorDeviceClass.GAS,
state_class=SensorStateClass.TOTAL_INCREASING,
),
+ DSMRSensorEntityDescription(
+ key="actual_threshold_electricity",
+ translation_key="actual_threshold_electricity",
+ obis_reference="ACTUAL_TRESHOLD_ELECTRICITY", # Misspelled in external tool
+ dsmr_versions={"5EONHU"},
+ device_class=SensorDeviceClass.POWER,
+ entity_registry_enabled_default=False,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="eon_hu_electricity_combined",
+ translation_key="electricity_combined",
+ obis_reference="EON_HU_ELECTRICITY_COMBINED",
+ dsmr_versions={"5EONHU"},
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL,
+ ),
+ DSMRSensorEntityDescription(
+ key="eon_hu_instantaneous_power_factor_total",
+ translation_key="instantaneous_power_factor_total",
+ obis_reference="EON_HU_INSTANTANEOUS_POWER_FACTOR_TOTAL",
+ dsmr_versions={"5EONHU"},
+ entity_registry_enabled_default=False,
+ device_class=SensorDeviceClass.POWER_FACTOR,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="eon_hu_instantaneous_power_factor_l1",
+ translation_key="instantaneous_power_factor_l1",
+ obis_reference="EON_HU_INSTANTANEOUS_POWER_FACTOR_L1",
+ dsmr_versions={"5EONHU"},
+ entity_registry_enabled_default=False,
+ device_class=SensorDeviceClass.POWER_FACTOR,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="eon_hu_instantaneous_power_factor_l2",
+ translation_key="instantaneous_power_factor_l2",
+ obis_reference="EON_HU_INSTANTANEOUS_POWER_FACTOR_L2",
+ dsmr_versions={"5EONHU"},
+ entity_registry_enabled_default=False,
+ device_class=SensorDeviceClass.POWER_FACTOR,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="eon_hu_instantaneous_power_factor_l3",
+ translation_key="instantaneous_power_factor_l3",
+ obis_reference="EON_HU_INSTANTANEOUS_POWER_FACTOR_L3",
+ dsmr_versions={"5EONHU"},
+ entity_registry_enabled_default=False,
+ device_class=SensorDeviceClass.POWER_FACTOR,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="eon_hu_frequency",
+ translation_key="frequency",
+ obis_reference="EON_HU_FREQUENCY",
+ dsmr_versions={"5EONHU"},
+ entity_registry_enabled_default=False,
+ device_class=SensorDeviceClass.FREQUENCY,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="fuse_threshold_l1",
+ translation_key="fuse_threshold_l1",
+ obis_reference="FUSE_THRESHOLD_L1",
+ dsmr_versions={"5EONHU"},
+ device_class=SensorDeviceClass.CURRENT,
+ entity_registry_enabled_default=False,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="fuse_threshold_l2",
+ translation_key="fuse_threshold_l2",
+ obis_reference="FUSE_THRESHOLD_L2",
+ dsmr_versions={"5EONHU"},
+ device_class=SensorDeviceClass.CURRENT,
+ entity_registry_enabled_default=False,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="fuse_threshold_l3",
+ translation_key="fuse_threshold_l3",
+ obis_reference="FUSE_THRESHOLD_L3",
+ dsmr_versions={"5EONHU"},
+ device_class=SensorDeviceClass.CURRENT,
+ entity_registry_enabled_default=False,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ DSMRSensorEntityDescription(
+ key="text_message",
+ translation_key="text_message",
+ obis_reference="TEXT_MESSAGE",
+ dsmr_versions={"5EONHU"},
+ entity_registry_enabled_default=False,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
)
SENSORS_MBUS_DEVICE_TYPE: dict[int, tuple[DSMRSensorEntityDescription, ...]] = {
@@ -549,7 +692,9 @@ def get_dsmr_object(
async def async_setup_entry(
- hass: HomeAssistant, entry: DsmrConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: DsmrConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the DSMR sensor."""
dsmr_version = entry.data[CONF_DSMR_VERSION]
@@ -842,8 +987,9 @@ class DSMREntity(SensorEntity):
def translate_tariff(value: str, dsmr_version: str) -> str | None:
"""Convert 2/1 to normal/low depending on DSMR version."""
# DSMR V5B: Note: In Belgium values are swapped:
+ # DSMR V5EONHU: Note: In EON HUngary values are swapped:
# Rate code 2 is used for low rate and rate code 1 is used for normal rate.
- if dsmr_version == "5B":
+ if dsmr_version in ("5B", "5EONHU"):
if value == "0001":
value = "0002"
elif value == "0002":
diff --git a/homeassistant/components/dsmr/strings.json b/homeassistant/components/dsmr/strings.json
index 055c0c41264..871dd382f2b 100644
--- a/homeassistant/components/dsmr/strings.json
+++ b/homeassistant/components/dsmr/strings.json
@@ -61,6 +61,12 @@
"electricity_delivered_tariff_2": {
"name": "Energy production (tarif 2)"
},
+ "electricity_delivered_tariff_3": {
+ "name": "Energy production (tarif 3)"
+ },
+ "electricity_delivered_tariff_4": {
+ "name": "Energy production (tarif 4)"
+ },
"electricity_exported_total": {
"name": "Energy production (total)"
},
@@ -73,6 +79,12 @@
"electricity_used_tariff_2": {
"name": "Energy consumption (tarif 2)"
},
+ "electricity_used_tariff_3": {
+ "name": "Energy consumption (tarif 3)"
+ },
+ "electricity_used_tariff_4": {
+ "name": "Energy consumption (tarif 4)"
+ },
"gas_meter_reading": {
"name": "Gas consumption"
},
@@ -150,6 +162,57 @@
},
"water_meter_reading": {
"name": "Water consumption"
+ },
+ "actual_threshold_electricity": {
+ "name": "Actual threshold electricity"
+ },
+ "electricity_reactive_imported_total": {
+ "name": "Imported reactive electricity (total)"
+ },
+ "electricity_reactive_exported_total": {
+ "name": "Exported reactive electricity (total)"
+ },
+ "electricity_reactive_total_q1": {
+ "name": "Reactive electricity (Q1)"
+ },
+ "electricity_reactive_total_q2": {
+ "name": "Reactive electricity (Q2)"
+ },
+ "electricity_reactive_total_q3": {
+ "name": "Reactive electricity (Q3)"
+ },
+ "electricity_reactive_total_q4": {
+ "name": "Reactive electricity (Q4)"
+ },
+ "electricity_combined": {
+ "name": "Energy combined"
+ },
+ "instantaneous_power_factor_total": {
+ "name": "Instantaneous power factor (total)"
+ },
+ "instantaneous_power_factor_l1": {
+ "name": "Instantaneous power factor L1"
+ },
+ "instantaneous_power_factor_l2": {
+ "name": "Instantaneous power factor L2"
+ },
+ "instantaneous_power_factor_l3": {
+ "name": "Instantaneous power factor L3"
+ },
+ "frequency": {
+ "name": "Frequency"
+ },
+ "fuse_threshold_l1": {
+ "name": "Fuse threshold on L1"
+ },
+ "fuse_threshold_l2": {
+ "name": "Fuse threshold on L2"
+ },
+ "fuse_threshold_l3": {
+ "name": "Fuse threshold on L3"
+ },
+ "text_message": {
+ "name": "Text message"
}
}
},
diff --git a/homeassistant/components/dsmr_reader/sensor.py b/homeassistant/components/dsmr_reader/sensor.py
index 784a4cdec51..c9bd9c9fff2 100644
--- a/homeassistant/components/dsmr_reader/sensor.py
+++ b/homeassistant/components/dsmr_reader/sensor.py
@@ -7,7 +7,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.util import slugify
@@ -18,7 +18,7 @@ from .definitions import SENSORS, DSMRReaderSensorEntityDescription
async def async_setup_entry(
_: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up DSMR Reader sensors from config entry."""
async_add_entities(DSMRSensor(description, config_entry) for description in SENSORS)
diff --git a/homeassistant/components/duke_energy/__init__.py b/homeassistant/components/duke_energy/__init__.py
index 6eacc15880f..bfa89d81c69 100644
--- a/homeassistant/components/duke_energy/__init__.py
+++ b/homeassistant/components/duke_energy/__init__.py
@@ -10,7 +10,7 @@ from .coordinator import DukeEnergyConfigEntry, DukeEnergyCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: DukeEnergyConfigEntry) -> bool:
"""Set up Duke Energy from a config entry."""
- coordinator = DukeEnergyCoordinator(hass, entry.data)
+ coordinator = DukeEnergyCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/duke_energy/coordinator.py b/homeassistant/components/duke_energy/coordinator.py
index 2b0ae46b405..12a2f5fd6ae 100644
--- a/homeassistant/components/duke_energy/coordinator.py
+++ b/homeassistant/components/duke_energy/coordinator.py
@@ -2,7 +2,6 @@
from datetime import datetime, timedelta
import logging
-from types import MappingProxyType
from typing import Any, cast
from aiodukeenergy import DukeEnergy
@@ -37,22 +36,21 @@ class DukeEnergyCoordinator(DataUpdateCoordinator[None]):
config_entry: DukeEnergyConfigEntry
def __init__(
- self,
- hass: HomeAssistant,
- entry_data: MappingProxyType[str, Any],
+ self, hass: HomeAssistant, config_entry: DukeEnergyConfigEntry
) -> None:
"""Initialize the data handler."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Duke Energy",
# Data is updated daily on Duke Energy.
# Refresh every 12h to be at most 12h behind.
update_interval=timedelta(hours=12),
)
self.api = DukeEnergy(
- entry_data[CONF_USERNAME],
- entry_data[CONF_PASSWORD],
+ config_entry.data[CONF_USERNAME],
+ config_entry.data[CONF_PASSWORD],
async_get_clientsession(hass),
)
self._statistic_ids: set = set()
diff --git a/homeassistant/components/dunehd/media_player.py b/homeassistant/components/dunehd/media_player.py
index db903cac2bf..b3093221385 100644
--- a/homeassistant/components/dunehd/media_player.py
+++ b/homeassistant/components/dunehd/media_player.py
@@ -17,7 +17,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DuneHDConfigEntry
from .const import ATTR_MANUFACTURER, DEFAULT_NAME, DOMAIN
@@ -39,7 +39,7 @@ DUNEHD_PLAYER_SUPPORT: Final[MediaPlayerEntityFeature] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DuneHDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Dune HD entities from a config_entry."""
async_add_entities(
diff --git a/homeassistant/components/duotecno/binary_sensor.py b/homeassistant/components/duotecno/binary_sensor.py
index aadef47b998..e2431b5eade 100644
--- a/homeassistant/components/duotecno/binary_sensor.py
+++ b/homeassistant/components/duotecno/binary_sensor.py
@@ -6,7 +6,7 @@ from duotecno.unit import ControlUnit, VirtualUnit
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DuotecnoConfigEntry
from .entity import DuotecnoEntity
@@ -15,7 +15,7 @@ from .entity import DuotecnoEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: DuotecnoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Duotecno binary sensor on config_entry."""
async_add_entities(
diff --git a/homeassistant/components/duotecno/climate.py b/homeassistant/components/duotecno/climate.py
index 83a211d97f5..0ae6735feb5 100644
--- a/homeassistant/components/duotecno/climate.py
+++ b/homeassistant/components/duotecno/climate.py
@@ -13,7 +13,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DuotecnoConfigEntry
from .entity import DuotecnoEntity, api_call
@@ -32,7 +32,7 @@ PRESETMODES_REVERSE: Final = {value: key for key, value in PRESETMODES.items()}
async def async_setup_entry(
hass: HomeAssistant,
entry: DuotecnoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Duotecno climate based on config_entry."""
async_add_entities(
diff --git a/homeassistant/components/duotecno/cover.py b/homeassistant/components/duotecno/cover.py
index 7d879741555..e184cf7ffb3 100644
--- a/homeassistant/components/duotecno/cover.py
+++ b/homeassistant/components/duotecno/cover.py
@@ -8,7 +8,7 @@ from duotecno.unit import DuoswitchUnit
from homeassistant.components.cover import CoverEntity, CoverEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DuotecnoConfigEntry
from .entity import DuotecnoEntity, api_call
@@ -17,7 +17,7 @@ from .entity import DuotecnoEntity, api_call
async def async_setup_entry(
hass: HomeAssistant,
entry: DuotecnoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the duoswitch endities."""
async_add_entities(
diff --git a/homeassistant/components/duotecno/light.py b/homeassistant/components/duotecno/light.py
index 7b41cbaef22..38617c92748 100644
--- a/homeassistant/components/duotecno/light.py
+++ b/homeassistant/components/duotecno/light.py
@@ -6,7 +6,7 @@ from duotecno.unit import DimUnit
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DuotecnoConfigEntry
from .entity import DuotecnoEntity, api_call
@@ -15,7 +15,7 @@ from .entity import DuotecnoEntity, api_call
async def async_setup_entry(
hass: HomeAssistant,
entry: DuotecnoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Duotecno light based on config_entry."""
async_add_entities(
diff --git a/homeassistant/components/duotecno/switch.py b/homeassistant/components/duotecno/switch.py
index 0c01a6ca4de..cef7715e946 100644
--- a/homeassistant/components/duotecno/switch.py
+++ b/homeassistant/components/duotecno/switch.py
@@ -6,7 +6,7 @@ from duotecno.unit import SwitchUnit
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DuotecnoConfigEntry
from .entity import DuotecnoEntity, api_call
@@ -15,7 +15,7 @@ from .entity import DuotecnoEntity, api_call
async def async_setup_entry(
hass: HomeAssistant,
entry: DuotecnoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
async_add_entities(
diff --git a/homeassistant/components/dwd_weather_warnings/__init__.py b/homeassistant/components/dwd_weather_warnings/__init__.py
index 7a56299a35b..727fcf95339 100644
--- a/homeassistant/components/dwd_weather_warnings/__init__.py
+++ b/homeassistant/components/dwd_weather_warnings/__init__.py
@@ -16,7 +16,7 @@ async def async_setup_entry(
device_registry = dr.async_get(hass)
if device_registry.async_get_device(identifiers={(DOMAIN, entry.entry_id)}):
device_registry.async_clear_config_entry(entry.entry_id)
- coordinator = DwdWeatherWarningsCoordinator(hass)
+ coordinator = DwdWeatherWarningsCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/dwd_weather_warnings/coordinator.py b/homeassistant/components/dwd_weather_warnings/coordinator.py
index be61304bc06..61656a82de6 100644
--- a/homeassistant/components/dwd_weather_warnings/coordinator.py
+++ b/homeassistant/components/dwd_weather_warnings/coordinator.py
@@ -28,10 +28,16 @@ class DwdWeatherWarningsCoordinator(DataUpdateCoordinator[None]):
config_entry: DwdWeatherWarningsConfigEntry
api: DwdWeatherWarningsAPI
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: DwdWeatherWarningsConfigEntry
+ ) -> None:
"""Initialize the dwd_weather_warnings coordinator."""
super().__init__(
- hass, LOGGER, name=DOMAIN, update_interval=DEFAULT_SCAN_INTERVAL
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=DEFAULT_SCAN_INTERVAL,
)
self._device_tracker = None
diff --git a/homeassistant/components/dwd_weather_warnings/sensor.py b/homeassistant/components/dwd_weather_warnings/sensor.py
index 0aaf1f2a801..1c2817350a5 100644
--- a/homeassistant/components/dwd_weather_warnings/sensor.py
+++ b/homeassistant/components/dwd_weather_warnings/sensor.py
@@ -16,7 +16,7 @@ from typing import Any
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -55,7 +55,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: DwdWeatherWarningsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities from config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/dynalite/cover.py b/homeassistant/components/dynalite/cover.py
index 17adf1947ec..2cd473a2977 100644
--- a/homeassistant/components/dynalite/cover.py
+++ b/homeassistant/components/dynalite/cover.py
@@ -8,7 +8,7 @@ from homeassistant.components.cover import (
CoverEntity,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.enum import try_parse_enum
from .bridge import DynaliteBridge, DynaliteConfigEntry
@@ -18,7 +18,7 @@ from .entity import DynaliteBase, async_setup_entry_base
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DynaliteConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Record the async_add_entities function to add them later when received from Dynalite."""
diff --git a/homeassistant/components/dynalite/light.py b/homeassistant/components/dynalite/light.py
index ea2bc2bc96f..e9816c828db 100644
--- a/homeassistant/components/dynalite/light.py
+++ b/homeassistant/components/dynalite/light.py
@@ -4,7 +4,7 @@ from typing import Any
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import DynaliteConfigEntry
from .entity import DynaliteBase, async_setup_entry_base
@@ -13,7 +13,7 @@ from .entity import DynaliteBase, async_setup_entry_base
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DynaliteConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Record the async_add_entities function to add them later when received from Dynalite."""
async_setup_entry_base(
diff --git a/homeassistant/components/dynalite/strings.json b/homeassistant/components/dynalite/strings.json
index 468cdebf0b1..4f73f91113b 100644
--- a/homeassistant/components/dynalite/strings.json
+++ b/homeassistant/components/dynalite/strings.json
@@ -36,7 +36,7 @@
},
"request_channel_level": {
"name": "Request channel level",
- "description": "Requests Dynalite to report the level of a specific channel.",
+ "description": "Requests Dynalite to report the brightness level of a specific channel.",
"fields": {
"host": {
"name": "[%key:common::config_flow::data::host%]",
@@ -48,7 +48,7 @@
},
"channel": {
"name": "Channel",
- "description": "Channel to request the level for."
+ "description": "Channel to request the brightness level for."
}
}
}
diff --git a/homeassistant/components/dynalite/switch.py b/homeassistant/components/dynalite/switch.py
index dd6aad8670c..29f78ecbc20 100644
--- a/homeassistant/components/dynalite/switch.py
+++ b/homeassistant/components/dynalite/switch.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.const import STATE_ON
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import DynaliteConfigEntry
from .entity import DynaliteBase, async_setup_entry_base
@@ -14,7 +14,7 @@ from .entity import DynaliteBase, async_setup_entry_base
async def async_setup_entry(
hass: HomeAssistant,
config_entry: DynaliteConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Record the async_add_entities function to add them later when received from Dynalite."""
async_setup_entry_base(
diff --git a/homeassistant/components/eafm/sensor.py b/homeassistant/components/eafm/sensor.py
index d9b18cbc663..5d0af596521 100644
--- a/homeassistant/components/eafm/sensor.py
+++ b/homeassistant/components/eafm/sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.sensor import SensorEntity, SensorStateClass
from homeassistant.const import UnitOfLength
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -20,7 +20,7 @@ UNIT_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EafmConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up UK Flood Monitoring Sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/easyenergy/sensor.py b/homeassistant/components/easyenergy/sensor.py
index 6976a38da49..35fab870af3 100644
--- a/homeassistant/components/easyenergy/sensor.py
+++ b/homeassistant/components/easyenergy/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, SERVICE_TYPE_DEVICE_NAMES
@@ -213,7 +213,7 @@ def get_gas_price(data: EasyEnergyData, hours: int) -> float | None:
async def async_setup_entry(
hass: HomeAssistant,
entry: EasyEnergyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up easyEnergy sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/easyenergy/strings.json b/homeassistant/components/easyenergy/strings.json
index 96afffdf78f..502db7920a3 100644
--- a/homeassistant/components/easyenergy/strings.json
+++ b/homeassistant/components/easyenergy/strings.json
@@ -60,12 +60,12 @@
"description": "Requests gas prices from easyEnergy.",
"fields": {
"config_entry": {
- "name": "Config Entry",
+ "name": "Config entry",
"description": "The configuration entry to use for this action."
},
"incl_vat": {
- "name": "VAT Included",
- "description": "Include or exclude VAT in the prices, default is true."
+ "name": "VAT included",
+ "description": "Whether the prices should include VAT."
},
"start": {
"name": "Start",
diff --git a/homeassistant/components/ecobee/binary_sensor.py b/homeassistant/components/ecobee/binary_sensor.py
index 9c9f2192f43..76b3399ec6e 100644
--- a/homeassistant/components/ecobee/binary_sensor.py
+++ b/homeassistant/components/ecobee/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcobeeConfigEntry
from .const import DOMAIN, ECOBEE_MODEL_TO_NAME, MANUFACTURER
@@ -17,7 +17,7 @@ from .const import DOMAIN, ECOBEE_MODEL_TO_NAME, MANUFACTURER
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ecobee binary (occupancy) sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/ecobee/climate.py b/homeassistant/components/ecobee/climate.py
index 743e2e1ba4b..26fb362ba03 100644
--- a/homeassistant/components/ecobee/climate.py
+++ b/homeassistant/components/ecobee/climate.py
@@ -38,7 +38,7 @@ from homeassistant.helpers import (
entity_platform,
)
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import TemperatureConverter
from . import EcobeeConfigEntry, EcobeeData
@@ -204,7 +204,7 @@ SUPPORT_FLAGS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ecobee thermostat."""
diff --git a/homeassistant/components/ecobee/humidifier.py b/homeassistant/components/ecobee/humidifier.py
index 982cbdd07f2..a6f3c16f84a 100644
--- a/homeassistant/components/ecobee/humidifier.py
+++ b/homeassistant/components/ecobee/humidifier.py
@@ -3,18 +3,20 @@
from __future__ import annotations
from datetime import timedelta
+from typing import Any
from homeassistant.components.humidifier import (
DEFAULT_MAX_HUMIDITY,
DEFAULT_MIN_HUMIDITY,
MODE_AUTO,
+ HumidifierAction,
HumidifierDeviceClass,
HumidifierEntity,
HumidifierEntityFeature,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcobeeConfigEntry
from .const import DOMAIN, ECOBEE_MODEL_TO_NAME, MANUFACTURER
@@ -28,7 +30,7 @@ MODE_OFF = "off"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ecobee thermostat humidifier entity."""
data = config_entry.runtime_data
@@ -41,6 +43,12 @@ async def async_setup_entry(
async_add_entities(entities, True)
+ECOBEE_HUMIDIFIER_ACTION_TO_HASS = {
+ "humidifier": HumidifierAction.HUMIDIFYING,
+ "dehumidifier": HumidifierAction.DRYING,
+}
+
+
class EcobeeHumidifier(HumidifierEntity):
"""A humidifier class for an ecobee thermostat with humidifier attached."""
@@ -52,7 +60,7 @@ class EcobeeHumidifier(HumidifierEntity):
_attr_has_entity_name = True
_attr_name = None
- def __init__(self, data, thermostat_index):
+ def __init__(self, data, thermostat_index) -> None:
"""Initialize ecobee humidifier platform."""
self.data = data
self.thermostat_index = thermostat_index
@@ -80,11 +88,11 @@ class EcobeeHumidifier(HumidifierEntity):
)
@property
- def available(self):
+ def available(self) -> bool:
"""Return if device is available."""
return self.thermostat["runtime"]["connected"]
- async def async_update(self):
+ async def async_update(self) -> None:
"""Get the latest state from the thermostat."""
if self.update_without_throttle:
await self.data.update(no_throttle=True)
@@ -96,12 +104,20 @@ class EcobeeHumidifier(HumidifierEntity):
self._last_humidifier_on_mode = self.mode
@property
- def is_on(self):
+ def action(self) -> HumidifierAction:
+ """Return the current action."""
+ for status in self.thermostat["equipmentStatus"].split(","):
+ if status in ECOBEE_HUMIDIFIER_ACTION_TO_HASS:
+ return ECOBEE_HUMIDIFIER_ACTION_TO_HASS[status]
+ return HumidifierAction.IDLE if self.is_on else HumidifierAction.OFF
+
+ @property
+ def is_on(self) -> bool:
"""Return True if the humidifier is on."""
return self.mode != MODE_OFF
@property
- def mode(self):
+ def mode(self) -> str:
"""Return the current mode, e.g., off, auto, manual."""
return self.thermostat["settings"]["humidifierMode"]
@@ -118,9 +134,11 @@ class EcobeeHumidifier(HumidifierEntity):
except KeyError:
return None
- def set_mode(self, mode):
+ def set_mode(self, mode: str) -> None:
"""Set humidifier mode (auto, off, manual)."""
- if mode.lower() not in (self.available_modes):
+ if self.available_modes is None:
+ raise NotImplementedError("Humidifier does not support modes.")
+ if mode.lower() not in self.available_modes:
raise ValueError(
f"Invalid mode value: {mode} Valid values are"
f" {', '.join(self.available_modes)}."
@@ -134,10 +152,10 @@ class EcobeeHumidifier(HumidifierEntity):
self.data.ecobee.set_humidity(self.thermostat_index, humidity)
self.update_without_throttle = True
- def turn_off(self, **kwargs):
+ def turn_off(self, **kwargs: Any) -> None:
"""Set humidifier to off mode."""
self.set_mode(MODE_OFF)
- def turn_on(self, **kwargs):
+ def turn_on(self, **kwargs: Any) -> None:
"""Set humidifier to on mode."""
self.set_mode(self._last_humidifier_on_mode)
diff --git a/homeassistant/components/ecobee/notify.py b/homeassistant/components/ecobee/notify.py
index 7c70d7ae4ac..2cf6a30acd7 100644
--- a/homeassistant/components/ecobee/notify.py
+++ b/homeassistant/components/ecobee/notify.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.components.notify import NotifyEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcobeeConfigEntry, EcobeeData
from .entity import EcobeeBaseEntity
@@ -13,7 +13,7 @@ from .entity import EcobeeBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ecobee thermostat."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/ecobee/number.py b/homeassistant/components/ecobee/number.py
index f047ea8f896..50e9170394d 100644
--- a/homeassistant/components/ecobee/number.py
+++ b/homeassistant/components/ecobee/number.py
@@ -14,7 +14,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import UnitOfTemperature, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcobeeConfigEntry, EcobeeData
from .entity import EcobeeBaseEntity
@@ -53,7 +53,7 @@ VENTILATOR_NUMBERS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ecobee thermostat number entity."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/ecobee/sensor.py b/homeassistant/components/ecobee/sensor.py
index 1b50fc21edf..759f167ec1c 100644
--- a/homeassistant/components/ecobee/sensor.py
+++ b/homeassistant/components/ecobee/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcobeeConfigEntry
from .const import DOMAIN, ECOBEE_MODEL_TO_NAME, MANUFACTURER
@@ -74,7 +74,7 @@ SENSOR_TYPES: tuple[EcobeeSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ecobee sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/ecobee/strings.json b/homeassistant/components/ecobee/strings.json
index 2b44c45edef..078643ee789 100644
--- a/homeassistant/components/ecobee/strings.json
+++ b/homeassistant/components/ecobee/strings.json
@@ -25,7 +25,7 @@
"state_attributes": {
"preset_mode": {
"state": {
- "away_indefinitely": "Away Indefinitely"
+ "away_indefinitely": "Away indefinitely"
}
}
}
@@ -91,7 +91,7 @@
},
"fan_min_on_time": {
"name": "Fan minimum on time",
- "description": "Minimum number of minutes to run the fan each hour (0 to 60) during the vacation."
+ "description": "Minimum number of minutes to run the fan each hour during the vacation."
}
}
},
@@ -125,7 +125,7 @@
},
"set_fan_min_on_time": {
"name": "Set fan minimum on time",
- "description": "Sets the minimum fan on time.",
+ "description": "Sets the minimum amount of time that the fan will run per hour.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -133,7 +133,7 @@
},
"fan_min_on_time": {
"name": "[%key:component::ecobee::services::create_vacation::fields::fan_min_on_time::name%]",
- "description": "New value of fan min on time."
+ "description": "Minimum number of minutes to run the fan each hour."
}
}
},
diff --git a/homeassistant/components/ecobee/switch.py b/homeassistant/components/ecobee/switch.py
index c92082b7b58..e0848913b39 100644
--- a/homeassistant/components/ecobee/switch.py
+++ b/homeassistant/components/ecobee/switch.py
@@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.climate import HVACMode
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import EcobeeConfigEntry, EcobeeData
@@ -25,7 +25,7 @@ DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ecobee thermostat switch entity."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/ecobee/weather.py b/homeassistant/components/ecobee/weather.py
index 39b2d30ddd8..126f8dd5061 100644
--- a/homeassistant/components/ecobee/weather.py
+++ b/homeassistant/components/ecobee/weather.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import EcobeeConfigEntry
@@ -40,7 +40,7 @@ from .const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcobeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ecobee weather platform."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/ecoforest/number.py b/homeassistant/components/ecoforest/number.py
index 878c150343e..c1d5f5f3055 100644
--- a/homeassistant/components/ecoforest/number.py
+++ b/homeassistant/components/ecoforest/number.py
@@ -9,7 +9,7 @@ from pyecoforest.models.device import Device
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EcoforestConfigEntry
from .entity import EcoforestEntity
@@ -37,7 +37,7 @@ NUMBER_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcoforestConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ecoforest number platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/ecoforest/sensor.py b/homeassistant/components/ecoforest/sensor.py
index 0babb476ab6..c1d4aca6f0c 100644
--- a/homeassistant/components/ecoforest/sensor.py
+++ b/homeassistant/components/ecoforest/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import EcoforestConfigEntry
@@ -143,7 +143,7 @@ SENSOR_TYPES: tuple[EcoforestSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EcoforestConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ecoforest sensor platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ecoforest/switch.py b/homeassistant/components/ecoforest/switch.py
index de52248e751..bd83bfc9ee5 100644
--- a/homeassistant/components/ecoforest/switch.py
+++ b/homeassistant/components/ecoforest/switch.py
@@ -11,7 +11,7 @@ from pyecoforest.models.device import Device
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EcoforestConfigEntry
from .entity import EcoforestEntity
@@ -38,7 +38,7 @@ SWITCH_TYPES: tuple[EcoforestSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcoforestConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ecoforest switch platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/econet/binary_sensor.py b/homeassistant/components/econet/binary_sensor.py
index 13ef8c4713b..0d041dfca5a 100644
--- a/homeassistant/components/econet/binary_sensor.py
+++ b/homeassistant/components/econet/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EconetConfigEntry
from .entity import EcoNetEntity
@@ -42,7 +42,7 @@ BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EconetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EcoNet binary sensor based on a config entry."""
equipment = entry.runtime_data
diff --git a/homeassistant/components/econet/climate.py b/homeassistant/components/econet/climate.py
index d46dbd8750a..e7ccec33310 100644
--- a/homeassistant/components/econet/climate.py
+++ b/homeassistant/components/econet/climate.py
@@ -22,8 +22,8 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.issue_registry import IssueSeverity, create_issue
from . import EconetConfigEntry
from .const import DOMAIN
@@ -35,8 +35,13 @@ ECONET_STATE_TO_HA = {
ThermostatOperationMode.OFF: HVACMode.OFF,
ThermostatOperationMode.AUTO: HVACMode.HEAT_COOL,
ThermostatOperationMode.FAN_ONLY: HVACMode.FAN_ONLY,
+ ThermostatOperationMode.EMERGENCY_HEAT: HVACMode.HEAT,
+}
+HA_STATE_TO_ECONET = {
+ value: key
+ for key, value in ECONET_STATE_TO_HA.items()
+ if key != ThermostatOperationMode.EMERGENCY_HEAT
}
-HA_STATE_TO_ECONET = {value: key for key, value in ECONET_STATE_TO_HA.items()}
ECONET_FAN_STATE_TO_HA = {
ThermostatFanMode.AUTO: FAN_AUTO,
@@ -57,7 +62,7 @@ SUPPORT_FLAGS_THERMOSTAT = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EconetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EcoNet thermostat based on a config entry."""
equipment = entry.runtime_data
@@ -209,7 +214,7 @@ class EcoNetThermostat(EcoNetEntity[Thermostat], ClimateEntity):
def turn_aux_heat_on(self) -> None:
"""Turn auxiliary heater on."""
- async_create_issue(
+ create_issue(
self.hass,
DOMAIN,
"migrate_aux_heat",
@@ -223,7 +228,7 @@ class EcoNetThermostat(EcoNetEntity[Thermostat], ClimateEntity):
def turn_aux_heat_off(self) -> None:
"""Turn auxiliary heater off."""
- async_create_issue(
+ create_issue(
self.hass,
DOMAIN,
"migrate_aux_heat",
diff --git a/homeassistant/components/econet/manifest.json b/homeassistant/components/econet/manifest.json
index bda52ee3d07..bc7505740d7 100644
--- a/homeassistant/components/econet/manifest.json
+++ b/homeassistant/components/econet/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/econet",
"iot_class": "cloud_push",
"loggers": ["paho_mqtt", "pyeconet"],
- "requirements": ["pyeconet==0.1.26"]
+ "requirements": ["pyeconet==0.1.28"]
}
diff --git a/homeassistant/components/econet/sensor.py b/homeassistant/components/econet/sensor.py
index 510906d699c..1cc806ca8d5 100644
--- a/homeassistant/components/econet/sensor.py
+++ b/homeassistant/components/econet/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EconetConfigEntry
from .entity import EcoNetEntity
@@ -83,7 +83,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EconetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EcoNet sensor based on a config entry."""
diff --git a/homeassistant/components/econet/switch.py b/homeassistant/components/econet/switch.py
index 9fcd38c860e..ff7f017b49f 100644
--- a/homeassistant/components/econet/switch.py
+++ b/homeassistant/components/econet/switch.py
@@ -10,7 +10,7 @@ from pyeconet.equipment.thermostat import Thermostat, ThermostatOperationMode
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EconetConfigEntry
from .entity import EcoNetEntity
@@ -21,7 +21,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: EconetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ecobee thermostat switch entity."""
equipment = entry.runtime_data
diff --git a/homeassistant/components/econet/water_heater.py b/homeassistant/components/econet/water_heater.py
index cfbff70b580..fb74ae8b4a5 100644
--- a/homeassistant/components/econet/water_heater.py
+++ b/homeassistant/components/econet/water_heater.py
@@ -19,7 +19,7 @@ from homeassistant.components.water_heater import (
)
from homeassistant.const import ATTR_TEMPERATURE, STATE_OFF, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EconetConfigEntry
from .entity import EcoNetEntity
@@ -48,7 +48,7 @@ SUPPORT_FLAGS_HEATER = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EconetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EcoNet water heater based on a config entry."""
equipment = entry.runtime_data
diff --git a/homeassistant/components/ecovacs/binary_sensor.py b/homeassistant/components/ecovacs/binary_sensor.py
index d755d01a4ae..552a8152cc5 100644
--- a/homeassistant/components/ecovacs/binary_sensor.py
+++ b/homeassistant/components/ecovacs/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .entity import EcovacsCapabilityEntityDescription, EcovacsDescriptionEntity, EventT
@@ -45,7 +45,7 @@ ENTITY_DESCRIPTIONS: tuple[EcovacsBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
async_add_entities(
diff --git a/homeassistant/components/ecovacs/button.py b/homeassistant/components/ecovacs/button.py
index 2759ca972df..04eb0af02e6 100644
--- a/homeassistant/components/ecovacs/button.py
+++ b/homeassistant/components/ecovacs/button.py
@@ -13,7 +13,7 @@ from deebot_client.events import LifeSpan
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .const import SUPPORTED_LIFESPANS, SUPPORTED_STATION_ACTIONS
@@ -81,7 +81,7 @@ LIFESPAN_ENTITY_DESCRIPTIONS = tuple(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/event.py b/homeassistant/components/ecovacs/event.py
index 3249b466c77..488767b472d 100644
--- a/homeassistant/components/ecovacs/event.py
+++ b/homeassistant/components/ecovacs/event.py
@@ -7,7 +7,7 @@ from deebot_client.events import CleanJobStatus, ReportStatsEvent
from homeassistant.components.event import EventEntity, EventEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .entity import EcovacsEntity
@@ -17,7 +17,7 @@ from .util import get_name_key
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/image.py b/homeassistant/components/ecovacs/image.py
index d8b69084cec..f8a89b0cfa0 100644
--- a/homeassistant/components/ecovacs/image.py
+++ b/homeassistant/components/ecovacs/image.py
@@ -7,7 +7,7 @@ from deebot_client.events.map import CachedMapInfoEvent, MapChangedEvent
from homeassistant.components.image import ImageEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .entity import EcovacsEntity
@@ -16,7 +16,7 @@ from .entity import EcovacsEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/lawn_mower.py b/homeassistant/components/ecovacs/lawn_mower.py
index bf773207dc5..b9af67fafcd 100644
--- a/homeassistant/components/ecovacs/lawn_mower.py
+++ b/homeassistant/components/ecovacs/lawn_mower.py
@@ -16,7 +16,7 @@ from homeassistant.components.lawn_mower import (
LawnMowerEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .entity import EcovacsEntity
@@ -37,7 +37,7 @@ _STATE_TO_MOWER_STATE = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ecovacs mowers."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/manifest.json b/homeassistant/components/ecovacs/manifest.json
index 33a251c22dc..6d3dc5c9be6 100644
--- a/homeassistant/components/ecovacs/manifest.json
+++ b/homeassistant/components/ecovacs/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/ecovacs",
"iot_class": "cloud_push",
"loggers": ["sleekxmppfs", "sucks", "deebot_client"],
- "requirements": ["py-sucks==0.9.10", "deebot-client==12.0.0"]
+ "requirements": ["py-sucks==0.9.10", "deebot-client==12.3.1"]
}
diff --git a/homeassistant/components/ecovacs/number.py b/homeassistant/components/ecovacs/number.py
index adf282560a9..7a74b02ceca 100644
--- a/homeassistant/components/ecovacs/number.py
+++ b/homeassistant/components/ecovacs/number.py
@@ -16,7 +16,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import DEGREE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .entity import (
@@ -83,7 +83,7 @@ ENTITY_DESCRIPTIONS: tuple[EcovacsNumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/select.py b/homeassistant/components/ecovacs/select.py
index 3c3852f05ec..a7b9baf1c4a 100644
--- a/homeassistant/components/ecovacs/select.py
+++ b/homeassistant/components/ecovacs/select.py
@@ -11,7 +11,7 @@ from deebot_client.events import WaterInfoEvent, WorkModeEvent
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .entity import EcovacsCapabilityEntityDescription, EcovacsDescriptionEntity, EventT
@@ -54,7 +54,7 @@ ENTITY_DESCRIPTIONS: tuple[EcovacsSelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/sensor.py b/homeassistant/components/ecovacs/sensor.py
index 0e906c6cb16..6c8ae080fc3 100644
--- a/homeassistant/components/ecovacs/sensor.py
+++ b/homeassistant/components/ecovacs/sensor.py
@@ -35,7 +35,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import EcovacsConfigEntry
@@ -192,7 +192,7 @@ LEGACY_LIFESPAN_SENSORS = tuple(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/strings.json b/homeassistant/components/ecovacs/strings.json
index 723bdef17f8..44c51c7ae43 100644
--- a/homeassistant/components/ecovacs/strings.json
+++ b/homeassistant/components/ecovacs/strings.json
@@ -250,7 +250,7 @@
"message": "Params are required for the command: {command}"
},
"vacuum_raw_get_positions_not_supported": {
- "message": "Getting the positions of the chargers and the device itself is not supported"
+ "message": "Retrieving the positions of the chargers and the device itself is not supported"
}
},
"selector": {
@@ -264,7 +264,7 @@
"services": {
"raw_get_positions": {
"name": "Get raw positions",
- "description": "Get the raw response for the positions of the chargers and the device itself."
+ "description": "Retrieves a raw response containing the positions of the chargers and the device itself."
}
}
}
diff --git a/homeassistant/components/ecovacs/switch.py b/homeassistant/components/ecovacs/switch.py
index 288d092d391..dd379dbb199 100644
--- a/homeassistant/components/ecovacs/switch.py
+++ b/homeassistant/components/ecovacs/switch.py
@@ -9,7 +9,7 @@ from deebot_client.events import EnableEvent
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcovacsConfigEntry
from .entity import (
@@ -105,7 +105,7 @@ ENTITY_DESCRIPTIONS: tuple[EcovacsSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for passed config_entry in HA."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/ecovacs/vacuum.py b/homeassistant/components/ecovacs/vacuum.py
index bc78981d1db..6570b80e920 100644
--- a/homeassistant/components/ecovacs/vacuum.py
+++ b/homeassistant/components/ecovacs/vacuum.py
@@ -21,7 +21,7 @@ from homeassistant.components.vacuum import (
from homeassistant.core import HomeAssistant, SupportsResponse
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level
from homeassistant.util import slugify
@@ -41,7 +41,7 @@ SERVICE_RAW_GET_POSITIONS = "raw_get_positions"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EcovacsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ecovacs vacuums."""
diff --git a/homeassistant/components/ecowitt/binary_sensor.py b/homeassistant/components/ecowitt/binary_sensor.py
index 5bc782e3589..1d36f5232db 100644
--- a/homeassistant/components/ecowitt/binary_sensor.py
+++ b/homeassistant/components/ecowitt/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EcowittConfigEntry
from .entity import EcowittEntity
@@ -26,13 +26,16 @@ ECOWITT_BINARYSENSORS_MAPPING: Final = {
device_class=BinarySensorDeviceClass.BATTERY,
entity_category=EntityCategory.DIAGNOSTIC,
),
+ EcoWittSensorTypes.RAIN_STATE: BinarySensorEntityDescription(
+ key="RAIN_STATE", device_class=BinarySensorDeviceClass.MOISTURE
+ ),
}
async def async_setup_entry(
hass: HomeAssistant,
entry: EcowittConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add sensors if new."""
ecowitt = entry.runtime_data
diff --git a/homeassistant/components/ecowitt/manifest.json b/homeassistant/components/ecowitt/manifest.json
index 175960ab57d..3ce66f48f95 100644
--- a/homeassistant/components/ecowitt/manifest.json
+++ b/homeassistant/components/ecowitt/manifest.json
@@ -6,5 +6,5 @@
"dependencies": ["webhook"],
"documentation": "https://www.home-assistant.io/integrations/ecowitt",
"iot_class": "local_push",
- "requirements": ["aioecowitt==2024.2.1"]
+ "requirements": ["aioecowitt==2025.3.1"]
}
diff --git a/homeassistant/components/ecowitt/sensor.py b/homeassistant/components/ecowitt/sensor.py
index 23af2f2a3af..6968acdfa4f 100644
--- a/homeassistant/components/ecowitt/sensor.py
+++ b/homeassistant/components/ecowitt/sensor.py
@@ -32,7 +32,7 @@ from homeassistant.const import (
UnitOfVolumetricFlux,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.unit_system import METRIC_SYSTEM, US_CUSTOMARY_SYSTEM
@@ -65,7 +65,9 @@ ECOWITT_SENSORS_MAPPING: Final = {
state_class=SensorStateClass.MEASUREMENT,
),
EcoWittSensorTypes.DEGREE: SensorEntityDescription(
- key="DEGREE", native_unit_of_measurement=DEGREE
+ key="DEGREE",
+ native_unit_of_measurement=DEGREE,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
EcoWittSensorTypes.WATT_METERS_SQUARED: SensorEntityDescription(
key="WATT_METERS_SQUARED",
@@ -218,7 +220,7 @@ ECOWITT_SENSORS_MAPPING: Final = {
async def async_setup_entry(
hass: HomeAssistant,
entry: EcowittConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add sensors if new."""
ecowitt = entry.runtime_data
diff --git a/homeassistant/components/edl21/sensor.py b/homeassistant/components/edl21/sensor.py
index 62d06a8a535..3194781d71c 100644
--- a/homeassistant/components/edl21/sensor.py
+++ b/homeassistant/components/edl21/sensor.py
@@ -30,7 +30,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from .const import (
@@ -289,7 +289,7 @@ SENSOR_UNIT_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the EDL21 sensor."""
api = EDL21(hass, config_entry.data, async_add_entities)
@@ -317,7 +317,7 @@ class EDL21:
self,
hass: HomeAssistant,
config: Mapping[str, Any],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize an EDL21 object."""
self._registered_obis: set[tuple[str, str]] = set()
diff --git a/homeassistant/components/efergy/sensor.py b/homeassistant/components/efergy/sensor.py
index 419c4da591d..6b54e4779a0 100644
--- a/homeassistant/components/efergy/sensor.py
+++ b/homeassistant/components/efergy/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import EfergyConfigEntry
@@ -108,7 +108,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EfergyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Efergy sensors."""
api = entry.runtime_data
diff --git a/homeassistant/components/eheimdigital/__init__.py b/homeassistant/components/eheimdigital/__init__.py
index a555a87cfbc..26e6bea4d4a 100644
--- a/homeassistant/components/eheimdigital/__init__.py
+++ b/homeassistant/components/eheimdigital/__init__.py
@@ -2,25 +2,22 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntry
from .const import DOMAIN
-from .coordinator import EheimDigitalUpdateCoordinator
+from .coordinator import EheimDigitalConfigEntry, EheimDigitalUpdateCoordinator
PLATFORMS = [Platform.CLIMATE, Platform.LIGHT]
-type EheimDigitalConfigEntry = ConfigEntry[EheimDigitalUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, entry: EheimDigitalConfigEntry
) -> bool:
"""Set up EHEIM Digital from a config entry."""
- coordinator = EheimDigitalUpdateCoordinator(hass)
+ coordinator = EheimDigitalUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/eheimdigital/climate.py b/homeassistant/components/eheimdigital/climate.py
index 7ad06659089..3cde9e758cd 100644
--- a/homeassistant/components/eheimdigital/climate.py
+++ b/homeassistant/components/eheimdigital/climate.py
@@ -21,11 +21,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import EheimDigitalConfigEntry
from .const import HEATER_BIO_MODE, HEATER_PRESET_TO_HEATER_MODE, HEATER_SMART_MODE
-from .coordinator import EheimDigitalUpdateCoordinator
+from .coordinator import EheimDigitalConfigEntry, EheimDigitalUpdateCoordinator
from .entity import EheimDigitalEntity
# Coordinator is used to centralize the data updates
@@ -35,18 +34,16 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: EheimDigitalConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the callbacks for the coordinator so climate entities can be added as devices are found."""
coordinator = entry.runtime_data
def async_setup_device_entities(
- device_address: str | dict[str, EheimDigitalDevice],
+ device_address: dict[str, EheimDigitalDevice],
) -> None:
"""Set up the climate entities for one or multiple devices."""
entities: list[EheimDigitalHeaterClimate] = []
- if isinstance(device_address, str):
- device_address = {device_address: coordinator.hub.devices[device_address]}
for device in device_address.values():
if isinstance(device, EheimDigitalHeater):
entities.append(EheimDigitalHeaterClimate(coordinator, device))
diff --git a/homeassistant/components/eheimdigital/coordinator.py b/homeassistant/components/eheimdigital/coordinator.py
index ee4f09426b7..df5475b6567 100644
--- a/homeassistant/components/eheimdigital/coordinator.py
+++ b/homeassistant/components/eheimdigital/coordinator.py
@@ -2,25 +2,27 @@
from __future__ import annotations
+import asyncio
from collections.abc import Callable
from aiohttp import ClientError
from eheimdigital.device import EheimDigitalDevice
from eheimdigital.hub import EheimDigitalHub
-from eheimdigital.types import EheimDeviceType
+from eheimdigital.types import EheimDeviceType, EheimDigitalClientError
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.entity_component import DEFAULT_SCAN_INTERVAL
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN, LOGGER
-type AsyncSetupDeviceEntitiesCallback = Callable[
- [str | dict[str, EheimDigitalDevice]], None
-]
+type AsyncSetupDeviceEntitiesCallback = Callable[[dict[str, EheimDigitalDevice]], None]
+
+type EheimDigitalConfigEntry = ConfigEntry[EheimDigitalUpdateCoordinator]
class EheimDigitalUpdateCoordinator(
@@ -28,19 +30,27 @@ class EheimDigitalUpdateCoordinator(
):
"""The EHEIM Digital data update coordinator."""
- config_entry: ConfigEntry
+ config_entry: EheimDigitalConfigEntry
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: EheimDigitalConfigEntry
+ ) -> None:
"""Initialize the EHEIM Digital data update coordinator."""
super().__init__(
- hass, LOGGER, name=DOMAIN, update_interval=DEFAULT_SCAN_INTERVAL
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=DEFAULT_SCAN_INTERVAL,
)
+ self.main_device_added_event = asyncio.Event()
self.hub = EheimDigitalHub(
host=self.config_entry.data[CONF_HOST],
session=async_get_clientsession(hass),
loop=hass.loop,
receive_callback=self._async_receive_callback,
device_found_callback=self._async_device_found,
+ main_device_added_event=self.main_device_added_event,
)
self.known_devices: set[str] = set()
self.platform_callbacks: set[AsyncSetupDeviceEntitiesCallback] = set()
@@ -62,14 +72,23 @@ class EheimDigitalUpdateCoordinator(
if device_address not in self.known_devices:
for platform_callback in self.platform_callbacks:
- platform_callback(device_address)
+ platform_callback({device_address: self.hub.devices[device_address]})
async def _async_receive_callback(self) -> None:
self.async_set_updated_data(self.hub.devices)
async def _async_setup(self) -> None:
- await self.hub.connect()
- await self.hub.update()
+ try:
+ await self.hub.connect()
+ async with asyncio.timeout(2):
+ # This event gets triggered when the first message is received from
+ # the device, it contains the data necessary to create the main device.
+ # This removes the race condition where the main device is accessed
+ # before the response from the device is parsed.
+ await self.main_device_added_event.wait()
+ await self.hub.update()
+ except (TimeoutError, EheimDigitalClientError) as err:
+ raise ConfigEntryNotReady from err
async def _async_update_data(self) -> dict[str, EheimDigitalDevice]:
try:
diff --git a/homeassistant/components/eheimdigital/light.py b/homeassistant/components/eheimdigital/light.py
index 5ae0a6e866a..2725315befd 100644
--- a/homeassistant/components/eheimdigital/light.py
+++ b/homeassistant/components/eheimdigital/light.py
@@ -16,12 +16,11 @@ from homeassistant.components.light import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.color import brightness_to_value, value_to_brightness
-from . import EheimDigitalConfigEntry
from .const import EFFECT_DAYCL_MODE, EFFECT_TO_LIGHT_MODE
-from .coordinator import EheimDigitalUpdateCoordinator
+from .coordinator import EheimDigitalConfigEntry, EheimDigitalUpdateCoordinator
from .entity import EheimDigitalEntity
BRIGHTNESS_SCALE = (1, 100)
@@ -33,18 +32,16 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: EheimDigitalConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the callbacks for the coordinator so lights can be added as devices are found."""
coordinator = entry.runtime_data
def async_setup_device_entities(
- device_address: str | dict[str, EheimDigitalDevice],
+ device_address: dict[str, EheimDigitalDevice],
) -> None:
"""Set up the light entities for one or multiple devices."""
entities: list[EheimDigitalClassicLEDControlLight] = []
- if isinstance(device_address, str):
- device_address = {device_address: coordinator.hub.devices[device_address]}
for device in device_address.values():
if isinstance(device, EheimDigitalClassicLEDControl):
for channel in range(2):
diff --git a/homeassistant/components/eight_sleep/__init__.py b/homeassistant/components/eight_sleep/__init__.py
index 9df39bbe314..cfb2cfba845 100644
--- a/homeassistant/components/eight_sleep/__init__.py
+++ b/homeassistant/components/eight_sleep/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
@@ -28,11 +28,13 @@ async def async_setup_entry(hass: HomeAssistant, _: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
- if all(
- config_entry.state is ConfigEntryState.NOT_LOADED
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.entry_id != entry.entry_id
- ):
- ir.async_delete_issue(hass, DOMAIN, DOMAIN)
-
return True
+
+
+async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Remove a config entry."""
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
+ ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ # Remove any remaining disabled or ignored entries
+ for _entry in hass.config_entries.async_entries(DOMAIN):
+ hass.async_create_task(hass.config_entries.async_remove(_entry.entry_id))
diff --git a/homeassistant/components/electrasmart/climate.py b/homeassistant/components/electrasmart/climate.py
index 84def436dfb..bdf94f606db 100644
--- a/homeassistant/components/electrasmart/climate.py
+++ b/homeassistant/components/electrasmart/climate.py
@@ -28,7 +28,7 @@ from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ElectraSmartConfigEntry
from .const import (
@@ -91,7 +91,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: ElectraSmartConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Electra AC devices."""
api = entry.runtime_data
diff --git a/homeassistant/components/electric_kiwi/icons.json b/homeassistant/components/electric_kiwi/icons.json
index 1932ce19432..e5dbbb5ac48 100644
--- a/homeassistant/components/electric_kiwi/icons.json
+++ b/homeassistant/components/electric_kiwi/icons.json
@@ -13,6 +13,11 @@
"hop_power_savings": {
"default": "mdi:percent"
}
+ },
+ "select": {
+ "hop_selector": {
+ "default": "mdi:lightning-bolt"
+ }
}
}
}
diff --git a/homeassistant/components/electric_kiwi/manifest.json b/homeassistant/components/electric_kiwi/manifest.json
index 45bb09ca475..b2f19000825 100644
--- a/homeassistant/components/electric_kiwi/manifest.json
+++ b/homeassistant/components/electric_kiwi/manifest.json
@@ -7,5 +7,6 @@
"documentation": "https://www.home-assistant.io/integrations/electric_kiwi",
"integration_type": "hub",
"iot_class": "cloud_polling",
+ "quality_scale": "bronze",
"requirements": ["electrickiwi-api==0.9.14"]
}
diff --git a/homeassistant/components/electric_kiwi/quality_scale.yaml b/homeassistant/components/electric_kiwi/quality_scale.yaml
new file mode 100644
index 00000000000..a7db8d203b6
--- /dev/null
+++ b/homeassistant/components/electric_kiwi/quality_scale.yaml
@@ -0,0 +1,105 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: |
+ No custom actions.
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: |
+ No custom actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: exempt
+ comment: |
+ Does not subscribe to event explicitly.
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions:
+ status: exempt
+ comment: |
+ No actions
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: |
+ Has no options flow
+ docs-installation-parameters:
+ status: exempt
+ comment: |
+ Handled by OAuth flow (HA is only one with credentials, users cannot get them)
+ entity-unavailable: done
+ integration-owner: done
+ log-when-unavailable: done
+ parallel-updates: done
+ reauthentication-flow: done
+ test-coverage: todo
+
+ # Gold
+ devices:
+ status: exempt
+ comment: |
+ Web services only
+ diagnostics: todo
+ discovery-update-info:
+ status: exempt
+ comment: |
+ Web services only
+ discovery:
+ status: exempt
+ comment: |
+ Web services only
+ docs-data-update: todo
+ docs-examples: done
+ docs-known-limitations: done
+ docs-supported-devices:
+ status: exempt
+ comment: |
+ No devices
+ docs-supported-functions: todo
+ docs-troubleshooting: todo
+ docs-use-cases: done
+ dynamic-devices:
+ status: exempt
+ comment: |
+ No devices
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default:
+ status: exempt
+ comment: |
+ No unnecessary or noisy entities
+ entity-translations: done
+ exception-translations: todo
+ icon-translations: done
+ reconfiguration-flow:
+ status: exempt
+ comment: |
+ Handled by OAuth
+ repair-issues:
+ status: exempt
+ comment: |
+ Does not have any repairs
+ stale-devices:
+ status: exempt
+ comment: |
+ Does not have devices
+
+ # Platinum
+ async-dependency: todo
+ inject-websession: todo
+ strict-typing: todo
diff --git a/homeassistant/components/electric_kiwi/select.py b/homeassistant/components/electric_kiwi/select.py
index 30e02b5c5b9..2ba2a089557 100644
--- a/homeassistant/components/electric_kiwi/select.py
+++ b/homeassistant/components/electric_kiwi/select.py
@@ -7,12 +7,14 @@ import logging
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ATTRIBUTION
from .coordinator import ElectricKiwiConfigEntry, ElectricKiwiHOPDataCoordinator
+PARALLEL_UPDATES = 1
+
_LOGGER = logging.getLogger(__name__)
ATTR_EK_HOP_SELECT = "hop_select"
@@ -26,7 +28,7 @@ HOP_SELECT = SelectEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: ElectricKiwiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Electric Kiwi select setup."""
hop_coordinator = entry.runtime_data.hop
diff --git a/homeassistant/components/electric_kiwi/sensor.py b/homeassistant/components/electric_kiwi/sensor.py
index 410d70808c3..27f13a82e09 100644
--- a/homeassistant/components/electric_kiwi/sensor.py
+++ b/homeassistant/components/electric_kiwi/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import CURRENCY_DOLLAR, PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -27,6 +27,8 @@ from .coordinator import (
ElectricKiwiHOPDataCoordinator,
)
+PARALLEL_UPDATES = 0
+
ATTR_EK_HOP_START = "hop_power_start"
ATTR_EK_HOP_END = "hop_power_end"
ATTR_TOTAL_RUNNING_BALANCE = "total_running_balance"
@@ -130,7 +132,7 @@ HOP_SENSOR_TYPES: tuple[ElectricKiwiHOPSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ElectricKiwiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Electric Kiwi Sensors Setup."""
account_coordinator = entry.runtime_data.account
diff --git a/homeassistant/components/elevenlabs/strings.json b/homeassistant/components/elevenlabs/strings.json
index b346f94a963..8b0205a9e9a 100644
--- a/homeassistant/components/elevenlabs/strings.json
+++ b/homeassistant/components/elevenlabs/strings.json
@@ -6,7 +6,7 @@
"api_key": "[%key:common::config_flow::data::api_key%]"
},
"data_description": {
- "api_key": "Your Elevenlabs API key."
+ "api_key": "Your ElevenLabs API key."
}
}
},
diff --git a/homeassistant/components/elevenlabs/tts.py b/homeassistant/components/elevenlabs/tts.py
index 008cd106615..efcadb3f440 100644
--- a/homeassistant/components/elevenlabs/tts.py
+++ b/homeassistant/components/elevenlabs/tts.py
@@ -20,7 +20,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ElevenLabsConfigEntry
from .const import (
@@ -58,7 +58,7 @@ def to_voice_settings(options: MappingProxyType[str, Any]) -> VoiceSettings:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElevenLabsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ElevenLabs tts platform via config entry."""
client = config_entry.runtime_data.client
diff --git a/homeassistant/components/elgato/button.py b/homeassistant/components/elgato/button.py
index 505eff36b44..23ed65ded33 100644
--- a/homeassistant/components/elgato/button.py
+++ b/homeassistant/components/elgato/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ElgatoConfigEntry, ElgatoDataUpdateCoordinator
from .entity import ElgatoEntity
@@ -50,7 +50,7 @@ BUTTONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ElgatoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Elgato button based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/elgato/light.py b/homeassistant/components/elgato/light.py
index 990a0606fce..7d2010f7ba4 100644
--- a/homeassistant/components/elgato/light.py
+++ b/homeassistant/components/elgato/light.py
@@ -16,7 +16,7 @@ from homeassistant.components.light import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.util import color as color_util
@@ -31,7 +31,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: ElgatoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Elgato Light based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/elgato/sensor.py b/homeassistant/components/elgato/sensor.py
index 529d2f7c76e..02dbc2aeef6 100644
--- a/homeassistant/components/elgato/sensor.py
+++ b/homeassistant/components/elgato/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ElgatoConfigEntry, ElgatoData, ElgatoDataUpdateCoordinator
from .entity import ElgatoEntity
@@ -104,7 +104,7 @@ SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ElgatoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Elgato sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/elgato/switch.py b/homeassistant/components/elgato/switch.py
index 3b2420b0ace..1b24f621807 100644
--- a/homeassistant/components/elgato/switch.py
+++ b/homeassistant/components/elgato/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ElgatoConfigEntry, ElgatoData, ElgatoDataUpdateCoordinator
from .entity import ElgatoEntity
@@ -54,7 +54,7 @@ SWITCHES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ElgatoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Elgato switches based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/elkm1/__init__.py b/homeassistant/components/elkm1/__init__.py
index 5286b7ad66f..4bf51b99de1 100644
--- a/homeassistant/components/elkm1/__init__.py
+++ b/homeassistant/components/elkm1/__init__.py
@@ -101,9 +101,11 @@ def hostname_from_url(url: str) -> str:
def _host_validator(config: dict[str, str]) -> dict[str, str]:
"""Validate that a host is properly configured."""
- if config[CONF_HOST].startswith("elks://"):
+ if config[CONF_HOST].startswith(("elks://", "elksv1_2://")):
if CONF_USERNAME not in config or CONF_PASSWORD not in config:
- raise vol.Invalid("Specify username and password for elks://")
+ raise vol.Invalid(
+ "Specify username and password for elks:// or elksv1_2://"
+ )
elif not config[CONF_HOST].startswith("elk://") and not config[
CONF_HOST
].startswith("serial://"):
diff --git a/homeassistant/components/elkm1/alarm_control_panel.py b/homeassistant/components/elkm1/alarm_control_panel.py
index ab51b6fe281..393845f65ff 100644
--- a/homeassistant/components/elkm1/alarm_control_panel.py
+++ b/homeassistant/components/elkm1/alarm_control_panel.py
@@ -20,7 +20,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import VolDictType
@@ -55,7 +55,7 @@ SERVICE_ALARM_CLEAR_BYPASS = "alarm_clear_bypass"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElkM1ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ElkM1 alarm platform."""
elk_data = config_entry.runtime_data
@@ -105,6 +105,7 @@ class ElkArea(ElkAttachedEntity, AlarmControlPanelEntity, RestoreEntity):
AlarmControlPanelEntityFeature.ARM_HOME
| AlarmControlPanelEntityFeature.ARM_AWAY
| AlarmControlPanelEntityFeature.ARM_NIGHT
+ | AlarmControlPanelEntityFeature.ARM_VACATION
)
_element: Area
@@ -204,7 +205,7 @@ class ElkArea(ElkAttachedEntity, AlarmControlPanelEntity, RestoreEntity):
ArmedStatus.ARMED_STAY_INSTANT: AlarmControlPanelState.ARMED_HOME,
ArmedStatus.ARMED_TO_NIGHT: AlarmControlPanelState.ARMED_NIGHT,
ArmedStatus.ARMED_TO_NIGHT_INSTANT: AlarmControlPanelState.ARMED_NIGHT,
- ArmedStatus.ARMED_TO_VACATION: AlarmControlPanelState.ARMED_AWAY,
+ ArmedStatus.ARMED_TO_VACATION: AlarmControlPanelState.ARMED_VACATION,
}
if self._element.alarm_state is None:
diff --git a/homeassistant/components/elkm1/binary_sensor.py b/homeassistant/components/elkm1/binary_sensor.py
index 73f6b925e8c..ba6a375c29b 100644
--- a/homeassistant/components/elkm1/binary_sensor.py
+++ b/homeassistant/components/elkm1/binary_sensor.py
@@ -10,7 +10,7 @@ from elkm1_lib.zones import Zone
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ElkM1ConfigEntry
from .entity import ElkAttachedEntity, ElkEntity
@@ -19,7 +19,7 @@ from .entity import ElkAttachedEntity, ElkEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElkM1ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the Elk-M1 sensor platform."""
elk_data = config_entry.runtime_data
diff --git a/homeassistant/components/elkm1/climate.py b/homeassistant/components/elkm1/climate.py
index 1448acc6079..55af0cfa29c 100644
--- a/homeassistant/components/elkm1/climate.py
+++ b/homeassistant/components/elkm1/climate.py
@@ -19,7 +19,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import PRECISION_WHOLE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from . import ElkM1ConfigEntry
@@ -60,7 +60,7 @@ ELK_TO_HASS_FAN_MODES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElkM1ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the Elk-M1 thermostat platform."""
elk_data = config_entry.runtime_data
diff --git a/homeassistant/components/elkm1/light.py b/homeassistant/components/elkm1/light.py
index c041c9c9d65..b5e2f0acacf 100644
--- a/homeassistant/components/elkm1/light.py
+++ b/homeassistant/components/elkm1/light.py
@@ -10,7 +10,7 @@ from elkm1_lib.lights import Light
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ElkM1ConfigEntry
from .entity import ElkEntity, create_elk_entities
@@ -20,7 +20,7 @@ from .models import ELKM1Data
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElkM1ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Elk light platform."""
elk_data = config_entry.runtime_data
diff --git a/homeassistant/components/elkm1/scene.py b/homeassistant/components/elkm1/scene.py
index d8a1d83f326..5da240aee2d 100644
--- a/homeassistant/components/elkm1/scene.py
+++ b/homeassistant/components/elkm1/scene.py
@@ -8,7 +8,7 @@ from elkm1_lib.tasks import Task
from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ElkM1ConfigEntry
from .entity import ElkAttachedEntity, ElkEntity, create_elk_entities
@@ -17,7 +17,7 @@ from .entity import ElkAttachedEntity, ElkEntity, create_elk_entities
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElkM1ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the Elk-M1 scene platform."""
elk_data = config_entry.runtime_data
diff --git a/homeassistant/components/elkm1/sensor.py b/homeassistant/components/elkm1/sensor.py
index 2ca932ec134..328672edbed 100644
--- a/homeassistant/components/elkm1/sensor.py
+++ b/homeassistant/components/elkm1/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import EntityCategory, UnitOfElectricPotential
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from . import ElkM1ConfigEntry
@@ -40,7 +40,7 @@ ELK_SET_COUNTER_SERVICE_SCHEMA: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElkM1ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the Elk-M1 sensor platform."""
elk_data = config_entry.runtime_data
diff --git a/homeassistant/components/elkm1/strings.json b/homeassistant/components/elkm1/strings.json
index f184483646d..b50c1817838 100644
--- a/homeassistant/components/elkm1/strings.json
+++ b/homeassistant/components/elkm1/strings.json
@@ -53,7 +53,7 @@
"fields": {
"code": {
"name": "Code",
- "description": "An code to authorize the bypass of the alarm control panel."
+ "description": "Alarm code to authorize the bypass of the alarm control panel."
}
}
},
@@ -63,7 +63,7 @@
"fields": {
"code": {
"name": "Code",
- "description": "An code to authorize the bypass clear of the alarm control panel."
+ "description": "Alarm code to authorize the bypass clear of the alarm control panel."
}
}
},
@@ -73,7 +73,7 @@
"fields": {
"code": {
"name": "Code",
- "description": "An code to arm the alarm control panel."
+ "description": "Alarm code to arm the alarm control panel."
}
}
},
@@ -181,7 +181,7 @@
"fields": {
"code": {
"name": "Code",
- "description": "An code to authorize the bypass of the zone."
+ "description": "Alarm code to authorize the bypass of the zone."
}
}
},
diff --git a/homeassistant/components/elkm1/switch.py b/homeassistant/components/elkm1/switch.py
index 3e0f4849518..d91d65512a2 100644
--- a/homeassistant/components/elkm1/switch.py
+++ b/homeassistant/components/elkm1/switch.py
@@ -12,7 +12,7 @@ from elkm1_lib.thermostats import Thermostat
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ElkM1ConfigEntry
from .entity import ElkAttachedEntity, ElkEntity, create_elk_entities
@@ -22,7 +22,7 @@ from .models import ELKM1Data
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElkM1ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the Elk-M1 switch platform."""
elk_data = config_entry.runtime_data
diff --git a/homeassistant/components/elmax/alarm_control_panel.py b/homeassistant/components/elmax/alarm_control_panel.py
index 139c9080c15..a90c8f2652c 100644
--- a/homeassistant/components/elmax/alarm_control_panel.py
+++ b/homeassistant/components/elmax/alarm_control_panel.py
@@ -15,7 +15,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError, InvalidStateError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ElmaxConfigEntry
@@ -25,7 +25,7 @@ from .entity import ElmaxEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElmaxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Elmax area platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/elmax/binary_sensor.py b/homeassistant/components/elmax/binary_sensor.py
index 351c386a084..d9ec3e75901 100644
--- a/homeassistant/components/elmax/binary_sensor.py
+++ b/homeassistant/components/elmax/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ElmaxConfigEntry
from .entity import ElmaxEntity
@@ -18,7 +18,7 @@ from .entity import ElmaxEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElmaxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Elmax sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/elmax/config_flow.py b/homeassistant/components/elmax/config_flow.py
index b8697552626..98e49cc8056 100644
--- a/homeassistant/components/elmax/config_flow.py
+++ b/homeassistant/components/elmax/config_flow.py
@@ -498,7 +498,11 @@ class ElmaxConfigFlow(ConfigFlow, domain=DOMAIN):
self, discovery_info: ZeroconfServiceInfo
) -> ConfigFlowResult:
"""Handle device found via zeroconf."""
- host = discovery_info.host
+ host = (
+ f"[{discovery_info.ip_address}]"
+ if discovery_info.ip_address.version == 6
+ else str(discovery_info.ip_address)
+ )
https_port = (
int(discovery_info.port)
if discovery_info.port is not None
diff --git a/homeassistant/components/elmax/cover.py b/homeassistant/components/elmax/cover.py
index e98477fe496..6993d5e44be 100644
--- a/homeassistant/components/elmax/cover.py
+++ b/homeassistant/components/elmax/cover.py
@@ -10,7 +10,7 @@ from elmax_api.model.cover_status import CoverStatus
from homeassistant.components.cover import CoverEntity, CoverEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ElmaxConfigEntry
from .entity import ElmaxEntity
@@ -27,7 +27,7 @@ _COMMAND_BY_MOTION_STATUS = { # Maps the stop command to use for every cover mo
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElmaxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Elmax cover platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/elmax/switch.py b/homeassistant/components/elmax/switch.py
index 70faa44cf01..28a97fefd91 100644
--- a/homeassistant/components/elmax/switch.py
+++ b/homeassistant/components/elmax/switch.py
@@ -9,7 +9,7 @@ from elmax_api.model.panel import PanelStatus
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ElmaxConfigEntry
from .entity import ElmaxEntity
@@ -20,7 +20,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ElmaxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Elmax switch platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/elvia/strings.json b/homeassistant/components/elvia/strings.json
index 888a5ab8e76..a2c3cb81f54 100644
--- a/homeassistant/components/elvia/strings.json
+++ b/homeassistant/components/elvia/strings.json
@@ -19,7 +19,7 @@
},
"abort": {
"metering_point_id_already_configured": "Metering point with ID `{metering_point_id}` is already configured.",
- "no_metering_points": "The provived API token has no metering points."
+ "no_metering_points": "The provided API token has no metering points."
}
}
}
diff --git a/homeassistant/components/emoncms/__init__.py b/homeassistant/components/emoncms/__init__.py
index 581948bbc6f..012abcc8c9a 100644
--- a/homeassistant/components/emoncms/__init__.py
+++ b/homeassistant/components/emoncms/__init__.py
@@ -2,7 +2,6 @@
from pyemoncms import EmoncmsClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY, CONF_URL, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
@@ -10,12 +9,10 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from .const import DOMAIN, EMONCMS_UUID_DOC_URL, LOGGER
-from .coordinator import EmoncmsCoordinator
+from .coordinator import EmonCMSConfigEntry, EmoncmsCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type EmonCMSConfigEntry = ConfigEntry[EmoncmsCoordinator]
-
def _migrate_unique_id(
hass: HomeAssistant, entry: EmonCMSConfigEntry, emoncms_unique_id: str
@@ -68,7 +65,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: EmonCMSConfigEntry) -> b
session=async_get_clientsession(hass),
)
await _check_unique_id_migration(hass, entry, emoncms_client)
- coordinator = EmoncmsCoordinator(hass, emoncms_client)
+ coordinator = EmoncmsCoordinator(hass, entry, emoncms_client)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
@@ -77,11 +74,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: EmonCMSConfigEntry) -> b
return True
-async def update_listener(hass: HomeAssistant, entry: ConfigEntry):
+async def update_listener(hass: HomeAssistant, entry: EmonCMSConfigEntry):
"""Handle options update."""
await hass.config_entries.async_reload(entry.entry_id)
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: EmonCMSConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/emoncms/coordinator.py b/homeassistant/components/emoncms/coordinator.py
index c6fda5ed7c8..ec439b400d5 100644
--- a/homeassistant/components/emoncms/coordinator.py
+++ b/homeassistant/components/emoncms/coordinator.py
@@ -5,24 +5,31 @@ from typing import Any
from pyemoncms import EmoncmsClient
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import CONF_MESSAGE, CONF_SUCCESS, LOGGER
+type EmonCMSConfigEntry = ConfigEntry[EmoncmsCoordinator]
+
class EmoncmsCoordinator(DataUpdateCoordinator[list[dict[str, Any]] | None]):
"""Emoncms Data Update Coordinator."""
+ config_entry: EmonCMSConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: EmonCMSConfigEntry,
emoncms_client: EmoncmsClient,
) -> None:
"""Initialize the emoncms data coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="emoncms_coordinator",
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/emoncms/sensor.py b/homeassistant/components/emoncms/sensor.py
index 1920e06a8e8..6321ccfafcd 100644
--- a/homeassistant/components/emoncms/sensor.py
+++ b/homeassistant/components/emoncms/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
+from homeassistant.config_entries import SOURCE_IMPORT
from homeassistant.const import (
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
CONCENTRATION_PARTS_PER_MILLION,
@@ -39,7 +39,10 @@ from homeassistant.const import (
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant, callback
from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers import config_validation as cv, template
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -53,7 +56,7 @@ from .const import (
FEED_NAME,
FEED_TAG,
)
-from .coordinator import EmoncmsCoordinator
+from .coordinator import EmonCMSConfigEntry, EmoncmsCoordinator
SENSORS: dict[str | None, SensorEntityDescription] = {
"kWh": SensorEntityDescription(
@@ -288,8 +291,8 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
- entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: EmonCMSConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the emoncms sensors."""
name = sensor_name(entry.data[CONF_URL])
diff --git a/homeassistant/components/emoncms/strings.json b/homeassistant/components/emoncms/strings.json
index 77216a3fb2f..451a3fb88e5 100644
--- a/homeassistant/components/emoncms/strings.json
+++ b/homeassistant/components/emoncms/strings.json
@@ -1,7 +1,7 @@
{
"config": {
"error": {
- "api_error": "An error occured in the pyemoncms API : {details}"
+ "api_error": "An error occurred in the pyemoncms API : {details}"
},
"step": {
"user": {
diff --git a/homeassistant/components/emonitor/sensor.py b/homeassistant/components/emonitor/sensor.py
index 39ed61741ae..be9e2ecb4cc 100644
--- a/homeassistant/components/emonitor/sensor.py
+++ b/homeassistant/components/emonitor/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.const import UnitOfPower
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -38,7 +38,7 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EmonitorConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/emulated_hue/hue_api.py b/homeassistant/components/emulated_hue/hue_api.py
index 464d2bcb7e7..9ccb8a64367 100644
--- a/homeassistant/components/emulated_hue/hue_api.py
+++ b/homeassistant/components/emulated_hue/hue_api.py
@@ -322,8 +322,10 @@ class HueOneLightStateView(HomeAssistantView):
if hass_entity_id is None:
_LOGGER.error(
- "Unknown entity number: %s not found in emulated_hue_ids.json",
+ "Unknown entity number: %s not found in emulated_hue_ids.json, "
+ "state request from %s",
entity_id,
+ request.remote,
)
return self.json_message("Entity not found", HTTPStatus.NOT_FOUND)
diff --git a/homeassistant/components/emulated_kasa/manifest.json b/homeassistant/components/emulated_kasa/manifest.json
index da3912a9d25..fc54fb50064 100644
--- a/homeassistant/components/emulated_kasa/manifest.json
+++ b/homeassistant/components/emulated_kasa/manifest.json
@@ -6,5 +6,5 @@
"iot_class": "local_push",
"loggers": ["sense_energy"],
"quality_scale": "internal",
- "requirements": ["sense-energy==0.13.4"]
+ "requirements": ["sense-energy==0.13.7"]
}
diff --git a/homeassistant/components/energenie_power_sockets/strings.json b/homeassistant/components/energenie_power_sockets/strings.json
index 4e4e49c68fb..bd536568d2c 100644
--- a/homeassistant/components/energenie_power_sockets/strings.json
+++ b/homeassistant/components/energenie_power_sockets/strings.json
@@ -1,5 +1,5 @@
{
- "title": "Energenie Power Sockets Integration.",
+ "title": "Energenie Power Sockets",
"config": {
"step": {
"user": {
diff --git a/homeassistant/components/energenie_power_sockets/switch.py b/homeassistant/components/energenie_power_sockets/switch.py
index e4fb7653e5e..825566887bc 100644
--- a/homeassistant/components/energenie_power_sockets/switch.py
+++ b/homeassistant/components/energenie_power_sockets/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EnergenieConfigEntry
from .const import DOMAIN
@@ -19,7 +19,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EnergenieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add EGPS sockets for passed config_entry in HA."""
powerstrip = config_entry.runtime_data
diff --git a/homeassistant/components/energyzero/sensor.py b/homeassistant/components/energyzero/sensor.py
index 141ac793fba..38349b89ff7 100644
--- a/homeassistant/components/energyzero/sensor.py
+++ b/homeassistant/components/energyzero/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, SERVICE_TYPE_DEVICE_NAMES
@@ -147,7 +147,7 @@ def get_gas_price(data: EnergyZeroData, hours: int) -> float | None:
async def async_setup_entry(
hass: HomeAssistant,
entry: EnergyZeroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EnergyZero Sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/energyzero/strings.json b/homeassistant/components/energyzero/strings.json
index 7788f4d4d8e..48682ab31ee 100644
--- a/homeassistant/components/energyzero/strings.json
+++ b/homeassistant/components/energyzero/strings.json
@@ -54,10 +54,10 @@
"services": {
"get_gas_prices": {
"name": "Get gas prices",
- "description": "Request gas prices from EnergyZero.",
+ "description": "Requests gas prices from EnergyZero.",
"fields": {
"config_entry": {
- "name": "Config Entry",
+ "name": "Config entry",
"description": "The config entry to use for this action."
},
"incl_vat": {
@@ -76,7 +76,7 @@
},
"get_energy_prices": {
"name": "Get energy prices",
- "description": "Request energy prices from EnergyZero.",
+ "description": "Requests energy prices from EnergyZero.",
"fields": {
"config_entry": {
"name": "[%key:component::energyzero::services::get_gas_prices::fields::config_entry::name%]",
diff --git a/homeassistant/components/enigma2/__init__.py b/homeassistant/components/enigma2/__init__.py
index da78f3dac5c..16295c7f228 100644
--- a/homeassistant/components/enigma2/__init__.py
+++ b/homeassistant/components/enigma2/__init__.py
@@ -1,12 +1,9 @@
"""Support for Enigma2 devices."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .coordinator import Enigma2UpdateCoordinator
-
-type Enigma2ConfigEntry = ConfigEntry[Enigma2UpdateCoordinator]
+from .coordinator import Enigma2ConfigEntry, Enigma2UpdateCoordinator
PLATFORMS = [Platform.MEDIA_PLAYER]
@@ -22,6 +19,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: Enigma2ConfigEntry) -> b
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: Enigma2ConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/enigma2/coordinator.py b/homeassistant/components/enigma2/coordinator.py
index d5bbf2c0ce5..9710d7f547f 100644
--- a/homeassistant/components/enigma2/coordinator.py
+++ b/homeassistant/components/enigma2/coordinator.py
@@ -30,18 +30,25 @@ from .const import CONF_SOURCE_BOUQUET, DOMAIN
LOGGER = logging.getLogger(__package__)
+type Enigma2ConfigEntry = ConfigEntry[Enigma2UpdateCoordinator]
+
class Enigma2UpdateCoordinator(DataUpdateCoordinator[OpenWebIfStatus]):
"""The Enigma2 data update coordinator."""
+ config_entry: Enigma2ConfigEntry
device: OpenWebIfDevice
unique_id: str | None
- def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: Enigma2ConfigEntry) -> None:
"""Initialize the Enigma2 data update coordinator."""
super().__init__(
- hass, logger=LOGGER, name=DOMAIN, update_interval=DEFAULT_SCAN_INTERVAL
+ hass,
+ logger=LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=DEFAULT_SCAN_INTERVAL,
)
base_url = URL.build(
diff --git a/homeassistant/components/enigma2/media_player.py b/homeassistant/components/enigma2/media_player.py
index 1012997ff7f..a3cdd1858ed 100644
--- a/homeassistant/components/enigma2/media_player.py
+++ b/homeassistant/components/enigma2/media_player.py
@@ -15,11 +15,10 @@ from homeassistant.components.media_player import (
MediaType,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import Enigma2ConfigEntry
-from .coordinator import Enigma2UpdateCoordinator
+from .coordinator import Enigma2ConfigEntry, Enigma2UpdateCoordinator
ATTR_MEDIA_CURRENTLY_RECORDING = "media_currently_recording"
ATTR_MEDIA_DESCRIPTION = "media_description"
@@ -32,7 +31,7 @@ _LOGGER = getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: Enigma2ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Enigma2 media player platform."""
async_add_entities([Enigma2Device(entry.runtime_data)])
diff --git a/homeassistant/components/enocean/entity.py b/homeassistant/components/enocean/entity.py
index 5c12fc12a68..b2d73e65443 100644
--- a/homeassistant/components/enocean/entity.py
+++ b/homeassistant/components/enocean/entity.py
@@ -16,7 +16,7 @@ class EnOceanEntity(Entity):
"""Initialize the device."""
self.dev_id = dev_id
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register callbacks."""
self.async_on_remove(
async_dispatcher_connect(
diff --git a/homeassistant/components/enphase_envoy/binary_sensor.py b/homeassistant/components/enphase_envoy/binary_sensor.py
index 0258281661a..dcffef8271b 100644
--- a/homeassistant/components/enphase_envoy/binary_sensor.py
+++ b/homeassistant/components/enphase_envoy/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import EnphaseConfigEntry, EnphaseUpdateCoordinator
@@ -75,7 +75,7 @@ ENPOWER_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EnphaseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up envoy binary sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/enphase_envoy/coordinator.py b/homeassistant/components/enphase_envoy/coordinator.py
index 8eb2b32ac7b..b8cda03a451 100644
--- a/homeassistant/components/enphase_envoy/coordinator.py
+++ b/homeassistant/components/enphase_envoy/coordinator.py
@@ -55,6 +55,7 @@ class EnphaseUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=entry_data[CONF_NAME],
update_interval=SCAN_INTERVAL,
always_update=False,
diff --git a/homeassistant/components/enphase_envoy/manifest.json b/homeassistant/components/enphase_envoy/manifest.json
index 0b1fd8b04b9..e51a7427504 100644
--- a/homeassistant/components/enphase_envoy/manifest.json
+++ b/homeassistant/components/enphase_envoy/manifest.json
@@ -6,7 +6,7 @@
"documentation": "https://www.home-assistant.io/integrations/enphase_envoy",
"iot_class": "local_polling",
"loggers": ["pyenphase"],
- "requirements": ["pyenphase==1.23.1"],
+ "requirements": ["pyenphase==1.25.1"],
"zeroconf": [
{
"type": "_enphase-envoy._tcp.local."
diff --git a/homeassistant/components/enphase_envoy/number.py b/homeassistant/components/enphase_envoy/number.py
index a62913a4c0b..91e93d9c59b 100644
--- a/homeassistant/components/enphase_envoy/number.py
+++ b/homeassistant/components/enphase_envoy/number.py
@@ -19,11 +19,11 @@ from homeassistant.components.number import (
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import EnphaseConfigEntry, EnphaseUpdateCoordinator
-from .entity import EnvoyBaseEntity
+from .entity import EnvoyBaseEntity, exception_handler
PARALLEL_UPDATES = 1
@@ -73,7 +73,7 @@ STORAGE_RESERVE_SOC_ENTITY = EnvoyStorageSettingsNumberEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EnphaseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Enphase Envoy number platform."""
coordinator = config_entry.runtime_data
@@ -132,6 +132,7 @@ class EnvoyRelayNumberEntity(EnvoyBaseEntity, NumberEntity):
self.data.dry_contact_settings[self._relay_id]
)
+ @exception_handler
async def async_set_native_value(self, value: float) -> None:
"""Update the relay."""
await self.envoy.update_dry_contact(
@@ -185,6 +186,7 @@ class EnvoyStorageSettingsNumberEntity(EnvoyBaseEntity, NumberEntity):
assert self.data.tariff.storage_settings is not None
return self.entity_description.value_fn(self.data.tariff.storage_settings)
+ @exception_handler
async def async_set_native_value(self, value: float) -> None:
"""Update the storage setting."""
await self.entity_description.update_fn(self.envoy, value)
diff --git a/homeassistant/components/enphase_envoy/select.py b/homeassistant/components/enphase_envoy/select.py
index 7dc275aab37..42b47e5d793 100644
--- a/homeassistant/components/enphase_envoy/select.py
+++ b/homeassistant/components/enphase_envoy/select.py
@@ -14,11 +14,11 @@ from pyenphase.models.tariff import EnvoyStorageMode, EnvoyStorageSettings
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import EnphaseConfigEntry, EnphaseUpdateCoordinator
-from .entity import EnvoyBaseEntity
+from .entity import EnvoyBaseEntity, exception_handler
PARALLEL_UPDATES = 1
@@ -130,7 +130,7 @@ STORAGE_MODE_ENTITY = EnvoyStorageSettingsSelectEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EnphaseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Enphase Envoy select platform."""
coordinator = config_entry.runtime_data
@@ -192,6 +192,7 @@ class EnvoyRelaySelectEntity(EnvoyBaseEntity, SelectEntity):
"""Return the state of the Enpower switch."""
return self.entity_description.value_fn(self.relay)
+ @exception_handler
async def async_select_option(self, option: str) -> None:
"""Update the relay."""
await self.entity_description.update_fn(self.envoy, self.relay, option)
@@ -243,6 +244,7 @@ class EnvoyStorageSettingsSelectEntity(EnvoyBaseEntity, SelectEntity):
assert self.data.tariff.storage_settings is not None
return self.entity_description.value_fn(self.data.tariff.storage_settings)
+ @exception_handler
async def async_select_option(self, option: str) -> None:
"""Update the relay."""
await self.entity_description.update_fn(self.envoy, option)
diff --git a/homeassistant/components/enphase_envoy/sensor.py b/homeassistant/components/enphase_envoy/sensor.py
index dcf062a5417..594f5f34088 100644
--- a/homeassistant/components/enphase_envoy/sensor.py
+++ b/homeassistant/components/enphase_envoy/sensor.py
@@ -49,7 +49,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN
@@ -806,7 +806,7 @@ AGGREGATE_BATTERY_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EnphaseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up envoy sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/enphase_envoy/strings.json b/homeassistant/components/enphase_envoy/strings.json
index e99c45c5c7a..b498c59e0d3 100644
--- a/homeassistant/components/enphase_envoy/strings.json
+++ b/homeassistant/components/enphase_envoy/strings.json
@@ -57,7 +57,7 @@
"init": {
"title": "Envoy {serial} {host} options",
"data": {
- "diagnostics_include_fixtures": "Include test fixture data in diagnostic report. Use when requested to provide test data for troubleshooting or development activies. With this option enabled the diagnostic report may take more time to download. When report is created best disable this option again.",
+ "diagnostics_include_fixtures": "Include test fixture data in diagnostic report. Use when requested to provide test data for troubleshooting or development activities. With this option enabled the diagnostic report may take more time to download. When report is created best disable this option again.",
"disable_keep_alive": "Always use a new connection when requesting data from the Envoy. May resolve communication issues with some Envoy firmwares."
},
"data_description": {
@@ -360,9 +360,9 @@
"acb_battery_state": {
"name": "Battery state",
"state": {
- "discharging": "Discharging",
+ "discharging": "[%key:common::state::discharging%]",
"idle": "[%key:common::state::idle%]",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"full": "Full"
}
},
diff --git a/homeassistant/components/enphase_envoy/switch.py b/homeassistant/components/enphase_envoy/switch.py
index 8a3ca493562..bb4ed874b1d 100644
--- a/homeassistant/components/enphase_envoy/switch.py
+++ b/homeassistant/components/enphase_envoy/switch.py
@@ -14,7 +14,7 @@ from pyenphase.models.tariff import EnvoyStorageSettings
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import EnphaseConfigEntry, EnphaseUpdateCoordinator
@@ -78,7 +78,7 @@ CHARGE_FROM_GRID_SWITCH = EnvoyStorageSettingsSwitchEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EnphaseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Enphase Envoy switch platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/environment_canada/camera.py b/homeassistant/components/environment_canada/camera.py
index 3ba059e2206..b051c572816 100644
--- a/homeassistant/components/environment_canada/camera.py
+++ b/homeassistant/components/environment_canada/camera.py
@@ -8,7 +8,7 @@ import voluptuous as vol
from homeassistant.components.camera import Camera
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.typing import VolDictType
@@ -26,7 +26,7 @@ SET_RADAR_TYPE_SCHEMA: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ECConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
coordinator = config_entry.runtime_data.radar_coordinator
diff --git a/homeassistant/components/environment_canada/const.py b/homeassistant/components/environment_canada/const.py
index f1f6db2e0df..c2b58d8dcce 100644
--- a/homeassistant/components/environment_canada/const.py
+++ b/homeassistant/components/environment_canada/const.py
@@ -5,3 +5,4 @@ ATTR_STATION = "station"
CONF_STATION = "station"
CONF_TITLE = "title"
DOMAIN = "environment_canada"
+SERVICE_ENVIRONMENT_CANADA_FORECASTS = "get_forecasts"
diff --git a/homeassistant/components/environment_canada/icons.json b/homeassistant/components/environment_canada/icons.json
index c3562ce1840..ca55254cc12 100644
--- a/homeassistant/components/environment_canada/icons.json
+++ b/homeassistant/components/environment_canada/icons.json
@@ -21,6 +21,9 @@
"services": {
"set_radar_type": {
"service": "mdi:radar"
+ },
+ "get_forecasts": {
+ "service": "mdi:weather-cloudy-clock"
}
}
}
diff --git a/homeassistant/components/environment_canada/manifest.json b/homeassistant/components/environment_canada/manifest.json
index 76534662ff7..fc05e093b33 100644
--- a/homeassistant/components/environment_canada/manifest.json
+++ b/homeassistant/components/environment_canada/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/environment_canada",
"iot_class": "cloud_polling",
"loggers": ["env_canada"],
- "requirements": ["env-canada==0.7.2"]
+ "requirements": ["env-canada==0.8.0"]
}
diff --git a/homeassistant/components/environment_canada/sensor.py b/homeassistant/components/environment_canada/sensor.py
index 989667fb1ac..3a789289c74 100644
--- a/homeassistant/components/environment_canada/sensor.py
+++ b/homeassistant/components/environment_canada/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ATTR_STATION
@@ -167,6 +167,7 @@ SENSOR_TYPES: tuple[ECSensorEntityDescription, ...] = (
translation_key="wind_bearing",
native_unit_of_measurement=DEGREE,
value_fn=lambda data: data.conditions.get("wind_bearing", {}).get("value"),
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
ECSensorEntityDescription(
key="wind_chill",
@@ -253,7 +254,7 @@ ALERT_TYPES: tuple[ECSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ECConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
weather_coordinator = config_entry.runtime_data.weather_coordinator
diff --git a/homeassistant/components/environment_canada/services.yaml b/homeassistant/components/environment_canada/services.yaml
index 4293b313f5c..0e33aeec933 100644
--- a/homeassistant/components/environment_canada/services.yaml
+++ b/homeassistant/components/environment_canada/services.yaml
@@ -1,3 +1,9 @@
+get_forecasts:
+ target:
+ entity:
+ integration: environment_canada
+ domain: weather
+
set_radar_type:
target:
entity:
diff --git a/homeassistant/components/environment_canada/strings.json b/homeassistant/components/environment_canada/strings.json
index 28ca55c6195..1ccff145bb3 100644
--- a/homeassistant/components/environment_canada/strings.json
+++ b/homeassistant/components/environment_canada/strings.json
@@ -113,6 +113,10 @@
}
},
"services": {
+ "get_forecasts": {
+ "name": "Get forecasts",
+ "description": "Retrieves the forecast from selected weather services."
+ },
"set_radar_type": {
"name": "Set radar type",
"description": "Sets the type of radar image to retrieve.",
diff --git a/homeassistant/components/environment_canada/weather.py b/homeassistant/components/environment_canada/weather.py
index 156b9f4152b..dd7632032ec 100644
--- a/homeassistant/components/environment_canada/weather.py
+++ b/homeassistant/components/environment_canada/weather.py
@@ -35,11 +35,16 @@ from homeassistant.const import (
UnitOfSpeed,
UnitOfTemperature,
)
-from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.core import (
+ HomeAssistant,
+ ServiceResponse,
+ SupportsResponse,
+ callback,
+)
+from homeassistant.helpers import entity_platform, entity_registry as er
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
+from .const import DOMAIN, SERVICE_ENVIRONMENT_CANADA_FORECASTS
from .coordinator import ECConfigEntry, ECDataUpdateCoordinator
# Icon codes from http://dd.weatheroffice.ec.gc.ca/citypage_weather/
@@ -63,7 +68,7 @@ ICON_CONDITION_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ECConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
entity_registry = er.async_get(hass)
@@ -78,6 +83,14 @@ async def async_setup_entry(
async_add_entities([ECWeatherEntity(config_entry.runtime_data.weather_coordinator)])
+ platform = entity_platform.async_get_current_platform()
+ platform.async_register_entity_service(
+ SERVICE_ENVIRONMENT_CANADA_FORECASTS,
+ None,
+ "_async_environment_canada_forecasts",
+ supports_response=SupportsResponse.ONLY,
+ )
+
def _calculate_unique_id(config_entry_unique_id: str | None, hourly: bool) -> str:
"""Calculate unique ID."""
@@ -185,6 +198,23 @@ class ECWeatherEntity(
"""Return the hourly forecast in native units."""
return get_forecast(self.ec_data, True)
+ def _async_environment_canada_forecasts(self) -> ServiceResponse:
+ """Return the native Environment Canada forecast."""
+ daily = []
+ for f in self.ec_data.daily_forecasts:
+ day = f.copy()
+ day["timestamp"] = day["timestamp"].isoformat()
+ daily.append(day)
+
+ hourly = []
+ for f in self.ec_data.hourly_forecasts:
+ hour = f.copy()
+ hour["timestamp"] = hour["period"].isoformat()
+ del hour["period"]
+ hourly.append(hour)
+
+ return {"daily_forecast": daily, "hourly_forecast": hourly}
+
def get_forecast(ec_data, hourly) -> list[Forecast] | None:
"""Build the forecast array."""
diff --git a/homeassistant/components/epic_games_store/calendar.py b/homeassistant/components/epic_games_store/calendar.py
index 5df1d6b756d..41edb5e31a7 100644
--- a/homeassistant/components/epic_games_store/calendar.py
+++ b/homeassistant/components/epic_games_store/calendar.py
@@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, CalendarType
@@ -21,7 +21,7 @@ DateRange = namedtuple("DateRange", ["start", "end"]) # noqa: PYI024
async def async_setup_entry(
hass: HomeAssistant,
entry: EGSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the local calendar platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/epion/sensor.py b/homeassistant/components/epion/sensor.py
index 78027813ffa..360c1f1d8a7 100644
--- a/homeassistant/components/epion/sensor.py
+++ b/homeassistant/components/epion/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -59,7 +59,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EpionConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add an Epion entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/epson/config_flow.py b/homeassistant/components/epson/config_flow.py
index c54bff2eea9..077b9cc31f7 100644
--- a/homeassistant/components/epson/config_flow.py
+++ b/homeassistant/components/epson/config_flow.py
@@ -72,5 +72,7 @@ class EpsonConfigFlow(ConfigFlow, domain=DOMAIN):
if projector:
projector.close()
return self.async_show_form(
- step_id="user", data_schema=DATA_SCHEMA, errors=errors
+ step_id="user",
+ data_schema=self.add_suggested_values_to_schema(DATA_SCHEMA, user_input),
+ errors=errors,
)
diff --git a/homeassistant/components/epson/media_player.py b/homeassistant/components/epson/media_player.py
index e0eac4a1cfb..c1582d6f0e5 100644
--- a/homeassistant/components/epson/media_player.py
+++ b/homeassistant/components/epson/media_player.py
@@ -43,7 +43,7 @@ from homeassistant.helpers import (
entity_registry as er,
)
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import EpsonConfigEntry
from .const import ATTR_CMODE, DOMAIN, SERVICE_SELECT_CMODE
@@ -54,7 +54,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EpsonConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Epson projector from a config entry."""
projector_entity = EpsonProjectorMediaPlayer(
diff --git a/homeassistant/components/eq3btsmart/binary_sensor.py b/homeassistant/components/eq3btsmart/binary_sensor.py
index 27525d47972..55b1f4d6ced 100644
--- a/homeassistant/components/eq3btsmart/binary_sensor.py
+++ b/homeassistant/components/eq3btsmart/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Eq3ConfigEntry
from .const import ENTITY_KEY_BATTERY, ENTITY_KEY_DST, ENTITY_KEY_WINDOW
@@ -51,7 +51,7 @@ BINARY_SENSOR_ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: Eq3ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the entry."""
diff --git a/homeassistant/components/eq3btsmart/climate.py b/homeassistant/components/eq3btsmart/climate.py
index ae01d0fc9a7..738efa99187 100644
--- a/homeassistant/components/eq3btsmart/climate.py
+++ b/homeassistant/components/eq3btsmart/climate.py
@@ -19,7 +19,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Eq3ConfigEntry
from .const import (
@@ -37,7 +37,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: Eq3ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Handle config entry setup."""
diff --git a/homeassistant/components/eq3btsmart/manifest.json b/homeassistant/components/eq3btsmart/manifest.json
index 0bc3ae55236..ab62c962982 100644
--- a/homeassistant/components/eq3btsmart/manifest.json
+++ b/homeassistant/components/eq3btsmart/manifest.json
@@ -22,5 +22,5 @@
"integration_type": "device",
"iot_class": "local_polling",
"loggers": ["eq3btsmart"],
- "requirements": ["eq3btsmart==1.4.1", "bleak-esphome==2.7.1"]
+ "requirements": ["eq3btsmart==1.4.1", "bleak-esphome==2.12.0"]
}
diff --git a/homeassistant/components/eq3btsmart/number.py b/homeassistant/components/eq3btsmart/number.py
index 2e069180fa3..c3cbd8eae31 100644
--- a/homeassistant/components/eq3btsmart/number.py
+++ b/homeassistant/components/eq3btsmart/number.py
@@ -21,7 +21,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory, UnitOfTemperature, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Eq3ConfigEntry
from .const import (
@@ -109,7 +109,7 @@ NUMBER_ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: Eq3ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the entry."""
diff --git a/homeassistant/components/eq3btsmart/sensor.py b/homeassistant/components/eq3btsmart/sensor.py
index bd2605042f4..aab3cbf1925 100644
--- a/homeassistant/components/eq3btsmart/sensor.py
+++ b/homeassistant/components/eq3btsmart/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.components.sensor.const import SensorStateClass
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Eq3ConfigEntry
from .const import ENTITY_KEY_AWAY_UNTIL, ENTITY_KEY_VALVE
@@ -51,7 +51,7 @@ SENSOR_ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: Eq3ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the entry."""
diff --git a/homeassistant/components/eq3btsmart/switch.py b/homeassistant/components/eq3btsmart/switch.py
index 7525d8ca494..61da133cb71 100644
--- a/homeassistant/components/eq3btsmart/switch.py
+++ b/homeassistant/components/eq3btsmart/switch.py
@@ -9,7 +9,7 @@ from eq3btsmart.models import Status
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Eq3ConfigEntry
from .const import ENTITY_KEY_AWAY, ENTITY_KEY_BOOST, ENTITY_KEY_LOCK
@@ -49,7 +49,7 @@ SWITCH_ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: Eq3ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the entry."""
diff --git a/homeassistant/components/escea/climate.py b/homeassistant/components/escea/climate.py
index c3fb0015e68..a1ac83844a2 100644
--- a/homeassistant/components/escea/climate.py
+++ b/homeassistant/components/escea/climate.py
@@ -21,7 +21,7 @@ from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTempera
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DATA_DISCOVERY_SERVICE,
@@ -47,7 +47,7 @@ _HA_FAN_TO_ESCEA = {v: k for k, v in _ESCEA_FAN_TO_HA.items()}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize an Escea Controller."""
discovery_service = hass.data[DATA_DISCOVERY_SERVICE]
diff --git a/homeassistant/components/esphome/__init__.py b/homeassistant/components/esphome/__init__.py
index fee2531fa20..1e1a2763b59 100644
--- a/homeassistant/components/esphome/__init__.py
+++ b/homeassistant/components/esphome/__init__.py
@@ -16,7 +16,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.typing import ConfigType
-from .const import CONF_NOISE_PSK, DATA_FFMPEG_PROXY, DOMAIN
+from .const import CONF_BLUETOOTH_MAC_ADDRESS, CONF_NOISE_PSK, DATA_FFMPEG_PROXY, DOMAIN
from .dashboard import async_setup as async_setup_dashboard
from .domain_data import DomainData
@@ -87,6 +87,6 @@ async def async_unload_entry(hass: HomeAssistant, entry: ESPHomeConfigEntry) ->
async def async_remove_entry(hass: HomeAssistant, entry: ESPHomeConfigEntry) -> None:
"""Remove an esphome config entry."""
- if mac_address := entry.unique_id:
- async_remove_scanner(hass, mac_address.upper())
+ if bluetooth_mac_address := entry.data.get(CONF_BLUETOOTH_MAC_ADDRESS):
+ async_remove_scanner(hass, bluetooth_mac_address.upper())
await DomainData.get(hass).get_or_create_store(hass, entry).async_remove()
diff --git a/homeassistant/components/esphome/assist_satellite.py b/homeassistant/components/esphome/assist_satellite.py
index f60668b0a06..fdd16d20d77 100644
--- a/homeassistant/components/esphome/assist_satellite.py
+++ b/homeassistant/components/esphome/assist_satellite.py
@@ -39,7 +39,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import EsphomeAssistEntity
@@ -87,7 +87,7 @@ _CONFIG_TIMEOUT_SEC = 5
async def async_setup_entry(
hass: HomeAssistant,
entry: ESPHomeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Assist satellite entity."""
entry_data = entry.runtime_data
@@ -284,7 +284,10 @@ class EsphomeAssistSatellite(
elif event_type == VoiceAssistantEventType.VOICE_ASSISTANT_INTENT_END:
assert event.data is not None
data_to_send = {
- "conversation_id": event.data["intent_output"]["conversation_id"] or "",
+ "conversation_id": event.data["intent_output"]["conversation_id"],
+ "continue_conversation": str(
+ int(event.data["intent_output"]["continue_conversation"])
+ ),
}
elif event_type == VoiceAssistantEventType.VOICE_ASSISTANT_TTS_START:
assert event.data is not None
diff --git a/homeassistant/components/esphome/binary_sensor.py b/homeassistant/components/esphome/binary_sensor.py
index ac759aa7b17..02b13748fb6 100644
--- a/homeassistant/components/esphome/binary_sensor.py
+++ b/homeassistant/components/esphome/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import issue_registry as ir
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.enum import try_parse_enum
from .const import DOMAIN
@@ -24,7 +24,7 @@ from .entry_data import ESPHomeConfigEntry
async def async_setup_entry(
hass: HomeAssistant,
entry: ESPHomeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ESPHome binary sensors based on a config entry."""
await platform_async_setup_entry(
diff --git a/homeassistant/components/esphome/climate.py b/homeassistant/components/esphome/climate.py
index 478ce9bae2c..b651f16dfd7 100644
--- a/homeassistant/components/esphome/climate.py
+++ b/homeassistant/components/esphome/climate.py
@@ -3,6 +3,7 @@
from __future__ import annotations
from functools import partial
+from math import isfinite
from typing import Any, cast
from aioesphomeapi import (
@@ -238,9 +239,13 @@ class EsphomeClimateEntity(EsphomeEntity[ClimateInfo, ClimateState], ClimateEnti
@esphome_state_property
def current_humidity(self) -> int | None:
"""Return the current humidity."""
- if not self._static_info.supports_current_humidity:
+ if (
+ not self._static_info.supports_current_humidity
+ or (val := self._state.current_humidity) is None
+ or not isfinite(val)
+ ):
return None
- return round(self._state.current_humidity)
+ return round(val)
@property
@esphome_float_state_property
diff --git a/homeassistant/components/esphome/config_flow.py b/homeassistant/components/esphome/config_flow.py
index 695131b19f7..955a93cd2b7 100644
--- a/homeassistant/components/esphome/config_flow.py
+++ b/homeassistant/components/esphome/config_flow.py
@@ -41,6 +41,7 @@ from .const import (
CONF_ALLOW_SERVICE_CALLS,
CONF_DEVICE_NAME,
CONF_NOISE_PSK,
+ CONF_SUBSCRIBE_LOGS,
DEFAULT_ALLOW_SERVICE_CALLS,
DEFAULT_NEW_CONFIG_ALLOW_ALLOW_SERVICE_CALLS,
DOMAIN,
@@ -508,6 +509,10 @@ class OptionsFlowHandler(OptionsFlow):
CONF_ALLOW_SERVICE_CALLS, DEFAULT_ALLOW_SERVICE_CALLS
),
): bool,
+ vol.Required(
+ CONF_SUBSCRIBE_LOGS,
+ default=self.config_entry.options.get(CONF_SUBSCRIBE_LOGS, False),
+ ): bool,
}
)
return self.async_show_form(step_id="init", data_schema=data_schema)
diff --git a/homeassistant/components/esphome/const.py b/homeassistant/components/esphome/const.py
index 143aaa6342a..c7cd7fdcdf0 100644
--- a/homeassistant/components/esphome/const.py
+++ b/homeassistant/components/esphome/const.py
@@ -5,18 +5,22 @@ from awesomeversion import AwesomeVersion
DOMAIN = "esphome"
CONF_ALLOW_SERVICE_CALLS = "allow_service_calls"
+CONF_SUBSCRIBE_LOGS = "subscribe_logs"
CONF_DEVICE_NAME = "device_name"
CONF_NOISE_PSK = "noise_psk"
+CONF_BLUETOOTH_MAC_ADDRESS = "bluetooth_mac_address"
DEFAULT_ALLOW_SERVICE_CALLS = True
DEFAULT_NEW_CONFIG_ALLOW_ALLOW_SERVICE_CALLS = False
-STABLE_BLE_VERSION_STR = "2023.8.0"
+STABLE_BLE_VERSION_STR = "2025.2.2"
STABLE_BLE_VERSION = AwesomeVersion(STABLE_BLE_VERSION_STR)
PROJECT_URLS = {
"esphome.bluetooth-proxy": "https://esphome.github.io/bluetooth-proxies/",
}
-DEFAULT_URL = f"https://esphome.io/changelog/{STABLE_BLE_VERSION_STR}.html"
+# ESPHome always uses .0 for the changelog URL
+STABLE_BLE_URL_VERSION = f"{STABLE_BLE_VERSION.major}.{STABLE_BLE_VERSION.minor}.0"
+DEFAULT_URL = f"https://esphome.io/changelog/{STABLE_BLE_URL_VERSION}.html"
DATA_FFMPEG_PROXY = f"{DOMAIN}.ffmpeg_proxy"
diff --git a/homeassistant/components/esphome/dashboard.py b/homeassistant/components/esphome/dashboard.py
index 334c16e5730..290feec1e2a 100644
--- a/homeassistant/components/esphome/dashboard.py
+++ b/homeassistant/components/esphome/dashboard.py
@@ -6,7 +6,7 @@ import asyncio
import logging
from typing import Any
-from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntryState
+from homeassistant.config_entries import SOURCE_REAUTH
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import CALLBACK_TYPE, Event, HomeAssistant, callback
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -108,8 +108,7 @@ class ESPHomeDashboardManager:
reloads = [
hass.config_entries.async_reload(entry.entry_id)
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state is ConfigEntryState.LOADED
+ for entry in hass.config_entries.async_loaded_entries(DOMAIN)
]
# Re-auth flows will check the dashboard for encryption key when the form is requested
# but we only trigger reauth if the dashboard is available.
diff --git a/homeassistant/components/esphome/diagnostics.py b/homeassistant/components/esphome/diagnostics.py
index 58c9a8fe666..0903e874a15 100644
--- a/homeassistant/components/esphome/diagnostics.py
+++ b/homeassistant/components/esphome/diagnostics.py
@@ -13,9 +13,7 @@ from . import CONF_NOISE_PSK
from .dashboard import async_get_dashboard
from .entry_data import ESPHomeConfigEntry
-CONF_MAC_ADDRESS = "mac_address"
-
-REDACT_KEYS = {CONF_NOISE_PSK, CONF_PASSWORD, CONF_MAC_ADDRESS}
+REDACT_KEYS = {CONF_NOISE_PSK, CONF_PASSWORD, "mac_address", "bluetooth_mac_address"}
async def async_get_config_entry_diagnostics(
@@ -27,13 +25,17 @@ async def async_get_config_entry_diagnostics(
diag["config"] = config_entry.as_dict()
entry_data = config_entry.runtime_data
+ device_info = entry_data.device_info
if (storage_data := await entry_data.store.async_load()) is not None:
diag["storage_data"] = storage_data
if (
- config_entry.unique_id
- and (scanner := async_scanner_by_source(hass, config_entry.unique_id.upper()))
+ device_info
+ and (
+ scanner_mac := device_info.bluetooth_mac_address or device_info.mac_address
+ )
+ and (scanner := async_scanner_by_source(hass, scanner_mac.upper()))
and (bluetooth_device := entry_data.bluetooth_device)
):
diag["bluetooth"] = {
diff --git a/homeassistant/components/esphome/manager.py b/homeassistant/components/esphome/manager.py
index 5f5ee1241f7..0a47fb66815 100644
--- a/homeassistant/components/esphome/manager.py
+++ b/homeassistant/components/esphome/manager.py
@@ -5,6 +5,7 @@ from __future__ import annotations
import asyncio
from functools import partial
import logging
+import re
from typing import TYPE_CHECKING, Any, NamedTuple
from aioesphomeapi import (
@@ -16,6 +17,7 @@ from aioesphomeapi import (
HomeassistantServiceCall,
InvalidAuthAPIError,
InvalidEncryptionKeyAPIError,
+ LogLevel,
ReconnectLogic,
RequiresEncryptionAPIError,
UserService,
@@ -33,6 +35,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import (
+ CALLBACK_TYPE,
Event,
EventStateChangedData,
HomeAssistant,
@@ -60,7 +63,9 @@ from homeassistant.util.async_ import create_eager_task
from .bluetooth import async_connect_scanner
from .const import (
CONF_ALLOW_SERVICE_CALLS,
+ CONF_BLUETOOTH_MAC_ADDRESS,
CONF_DEVICE_NAME,
+ CONF_SUBSCRIBE_LOGS,
DEFAULT_ALLOW_SERVICE_CALLS,
DEFAULT_URL,
DOMAIN,
@@ -74,8 +79,38 @@ from .domain_data import DomainData
# Import config flow so that it's added to the registry
from .entry_data import ESPHomeConfigEntry, RuntimeEntryData
+if TYPE_CHECKING:
+ from aioesphomeapi.api_pb2 import ( # type: ignore[attr-defined]
+ SubscribeLogsResponse,
+ )
+
+
_LOGGER = logging.getLogger(__name__)
+LOG_LEVEL_TO_LOGGER = {
+ LogLevel.LOG_LEVEL_NONE: logging.DEBUG,
+ LogLevel.LOG_LEVEL_ERROR: logging.ERROR,
+ LogLevel.LOG_LEVEL_WARN: logging.WARNING,
+ LogLevel.LOG_LEVEL_INFO: logging.INFO,
+ LogLevel.LOG_LEVEL_CONFIG: logging.INFO,
+ LogLevel.LOG_LEVEL_DEBUG: logging.DEBUG,
+ LogLevel.LOG_LEVEL_VERBOSE: logging.DEBUG,
+ LogLevel.LOG_LEVEL_VERY_VERBOSE: logging.DEBUG,
+}
+LOGGER_TO_LOG_LEVEL = {
+ logging.NOTSET: LogLevel.LOG_LEVEL_VERY_VERBOSE,
+ logging.DEBUG: LogLevel.LOG_LEVEL_VERY_VERBOSE,
+ logging.INFO: LogLevel.LOG_LEVEL_CONFIG,
+ logging.WARNING: LogLevel.LOG_LEVEL_WARN,
+ logging.ERROR: LogLevel.LOG_LEVEL_ERROR,
+ logging.CRITICAL: LogLevel.LOG_LEVEL_ERROR,
+}
+# 7-bit and 8-bit C1 ANSI sequences
+# https://stackoverflow.com/questions/14693701/how-can-i-remove-the-ansi-escape-sequences-from-a-string-in-python
+ANSI_ESCAPE_78BIT = re.compile(
+ rb"(?:\x1B[@-Z\\-_]|[\x80-\x9A\x9C-\x9F]|(?:\x1B\[|\x9B)[0-?]*[ -/]*[@-~])"
+)
+
@callback
def _async_check_firmware_version(
@@ -136,6 +171,8 @@ class ESPHomeManager:
"""Class to manage an ESPHome connection."""
__slots__ = (
+ "_cancel_subscribe_logs",
+ "_log_level",
"cli",
"device_id",
"domain_data",
@@ -169,6 +206,8 @@ class ESPHomeManager:
self.reconnect_logic: ReconnectLogic | None = None
self.zeroconf_instance = zeroconf_instance
self.entry_data = entry.runtime_data
+ self._cancel_subscribe_logs: CALLBACK_TYPE | None = None
+ self._log_level = LogLevel.LOG_LEVEL_NONE
async def on_stop(self, event: Event) -> None:
"""Cleanup the socket client on HA close."""
@@ -341,6 +380,34 @@ class ESPHomeManager:
# Re-connection logic will trigger after this
await self.cli.disconnect()
+ def _async_on_log(self, msg: SubscribeLogsResponse) -> None:
+ """Handle a log message from the API."""
+ log: bytes = msg.message
+ _LOGGER.log(
+ LOG_LEVEL_TO_LOGGER.get(msg.level, logging.DEBUG),
+ "%s: %s",
+ self.entry.title,
+ ANSI_ESCAPE_78BIT.sub(b"", log).decode("utf-8", "backslashreplace"),
+ )
+
+ @callback
+ def _async_get_equivalent_log_level(self) -> LogLevel:
+ """Get the equivalent ESPHome log level for the current logger."""
+ return LOGGER_TO_LOG_LEVEL.get(
+ _LOGGER.getEffectiveLevel(), LogLevel.LOG_LEVEL_VERY_VERBOSE
+ )
+
+ @callback
+ def _async_subscribe_logs(self, log_level: LogLevel) -> None:
+ """Subscribe to logs."""
+ if self._cancel_subscribe_logs is not None:
+ self._cancel_subscribe_logs()
+ self._cancel_subscribe_logs = None
+ self._log_level = log_level
+ self._cancel_subscribe_logs = self.cli.subscribe_logs(
+ self._async_on_log, self._log_level
+ )
+
async def _on_connnect(self) -> None:
"""Subscribe to states and list entities on successful API login."""
entry = self.entry
@@ -352,6 +419,8 @@ class ESPHomeManager:
cli = self.cli
stored_device_name = entry.data.get(CONF_DEVICE_NAME)
unique_id_is_mac_address = unique_id and ":" in unique_id
+ if entry.options.get(CONF_SUBSCRIBE_LOGS):
+ self._async_subscribe_logs(self._async_get_equivalent_log_level())
results = await asyncio.gather(
create_eager_task(cli.device_info()),
create_eager_task(cli.list_entities_services()),
@@ -363,6 +432,13 @@ class ESPHomeManager:
device_mac = format_mac(device_info.mac_address)
mac_address_matches = unique_id == device_mac
+ if (
+ bluetooth_mac_address := device_info.bluetooth_mac_address
+ ) and entry.data.get(CONF_BLUETOOTH_MAC_ADDRESS) != bluetooth_mac_address:
+ hass.config_entries.async_update_entry(
+ entry,
+ data={**entry.data, CONF_BLUETOOTH_MAC_ADDRESS: bluetooth_mac_address},
+ )
#
# Migrate config entry to new unique ID if the current
# unique id is not a mac address.
@@ -430,7 +506,9 @@ class ESPHomeManager:
)
)
else:
- bluetooth.async_remove_scanner(hass, device_info.mac_address)
+ bluetooth.async_remove_scanner(
+ hass, device_info.bluetooth_mac_address or device_info.mac_address
+ )
if device_info.voice_assistant_feature_flags_compat(api_version) and (
Platform.ASSIST_SATELLITE not in entry_data.loaded_platforms
@@ -503,6 +581,10 @@ class ESPHomeManager:
def _async_handle_logging_changed(self, _event: Event) -> None:
"""Handle when the logging level changes."""
self.cli.set_debug(_LOGGER.isEnabledFor(logging.DEBUG))
+ if self.entry.options.get(CONF_SUBSCRIBE_LOGS) and self._log_level != (
+ new_log_level := self._async_get_equivalent_log_level()
+ ):
+ self._async_subscribe_logs(new_log_level)
async def async_start(self) -> None:
"""Start the esphome connection manager."""
@@ -545,11 +627,22 @@ class ESPHomeManager:
)
_setup_services(hass, entry_data, services)
- if entry_data.device_info is not None and entry_data.device_info.name:
- reconnect_logic.name = entry_data.device_info.name
+ if (device_info := entry_data.device_info) is not None:
+ if device_info.name:
+ reconnect_logic.name = device_info.name
+ if (
+ bluetooth_mac_address := device_info.bluetooth_mac_address
+ ) and entry.data.get(CONF_BLUETOOTH_MAC_ADDRESS) != bluetooth_mac_address:
+ hass.config_entries.async_update_entry(
+ entry,
+ data={
+ **entry.data,
+ CONF_BLUETOOTH_MAC_ADDRESS: bluetooth_mac_address,
+ },
+ )
if entry.unique_id is None:
hass.config_entries.async_update_entry(
- entry, unique_id=format_mac(entry_data.device_info.mac_address)
+ entry, unique_id=format_mac(device_info.mac_address)
)
await reconnect_logic.start()
diff --git a/homeassistant/components/esphome/manifest.json b/homeassistant/components/esphome/manifest.json
index 185f9ea5cf0..075185dffbb 100644
--- a/homeassistant/components/esphome/manifest.json
+++ b/homeassistant/components/esphome/manifest.json
@@ -16,9 +16,9 @@
"loggers": ["aioesphomeapi", "noiseprotocol", "bleak_esphome"],
"mqtt": ["esphome/discover/#"],
"requirements": [
- "aioesphomeapi==29.0.0",
+ "aioesphomeapi==29.7.0",
"esphome-dashboard-api==1.2.3",
- "bleak-esphome==2.7.1"
+ "bleak-esphome==2.12.0"
],
"zeroconf": ["_esphomelib._tcp.local."]
}
diff --git a/homeassistant/components/esphome/select.py b/homeassistant/components/esphome/select.py
index 71a21186d3d..67bcbbbd221 100644
--- a/homeassistant/components/esphome/select.py
+++ b/homeassistant/components/esphome/select.py
@@ -13,7 +13,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import restore_state
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import (
@@ -29,7 +29,7 @@ from .entry_data import ESPHomeConfigEntry, RuntimeEntryData
async def async_setup_entry(
hass: HomeAssistant,
entry: ESPHomeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up esphome selects based on a config entry."""
await platform_async_setup_entry(
diff --git a/homeassistant/components/esphome/sensor.py b/homeassistant/components/esphome/sensor.py
index 670c92d291e..26f33f4fb47 100644
--- a/homeassistant/components/esphome/sensor.py
+++ b/homeassistant/components/esphome/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from homeassistant.util.enum import try_parse_enum
@@ -31,7 +31,9 @@ from .enum_mapper import EsphomeEnumMapper
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up esphome sensors based on a config entry."""
await platform_async_setup_entry(
diff --git a/homeassistant/components/esphome/strings.json b/homeassistant/components/esphome/strings.json
index 81b58de8df2..c6916a3636d 100644
--- a/homeassistant/components/esphome/strings.json
+++ b/homeassistant/components/esphome/strings.json
@@ -4,7 +4,7 @@
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]",
"already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]",
"reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",
- "mdns_missing_mac": "Missing MAC address in MDNS properties.",
+ "mdns_missing_mac": "Missing MAC address in mDNS properties.",
"service_received": "Action received",
"mqtt_missing_mac": "Missing MAC address in MQTT properties.",
"mqtt_missing_api": "Missing API port in MQTT properties.",
@@ -54,7 +54,8 @@
"step": {
"init": {
"data": {
- "allow_service_calls": "Allow the device to perform Home Assistant actions."
+ "allow_service_calls": "Allow the device to perform Home Assistant actions.",
+ "subscribe_logs": "Subscribe to logs from the device. When enabled, the device will send logs to Home Assistant and you can view them in the logs panel."
}
}
}
diff --git a/homeassistant/components/esphome/update.py b/homeassistant/components/esphome/update.py
index 2b593051742..60d4989063b 100644
--- a/homeassistant/components/esphome/update.py
+++ b/homeassistant/components/esphome/update.py
@@ -23,7 +23,7 @@ from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.enum import try_parse_enum
@@ -46,7 +46,7 @@ NO_FEATURES = UpdateEntityFeature(0)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ESPHome update based on a config entry."""
await platform_async_setup_entry(
diff --git a/homeassistant/components/eufylife_ble/sensor.py b/homeassistant/components/eufylife_ble/sensor.py
index d9cef45ce4d..0c65be89879 100644
--- a/homeassistant/components/eufylife_ble/sensor.py
+++ b/homeassistant/components/eufylife_ble/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN, UnitOfMass
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
from .models import EufyLifeConfigEntry, EufyLifeData
@@ -27,7 +27,7 @@ IGNORED_STATES = {STATE_UNAVAILABLE, STATE_UNKNOWN}
async def async_setup_entry(
hass: HomeAssistant,
entry: EufyLifeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the EufyLife sensors."""
data = entry.runtime_data
diff --git a/homeassistant/components/evil_genius_labs/light.py b/homeassistant/components/evil_genius_labs/light.py
index a6d1d9531b5..3dd9b763ae1 100644
--- a/homeassistant/components/evil_genius_labs/light.py
+++ b/homeassistant/components/evil_genius_labs/light.py
@@ -8,7 +8,7 @@ from typing import Any, cast
from homeassistant.components import light
from homeassistant.components.light import ColorMode, LightEntity, LightEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EvilGeniusConfigEntry, EvilGeniusUpdateCoordinator
from .entity import EvilGeniusEntity
@@ -21,7 +21,7 @@ FIB_NO_EFFECT = "Solid Color"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: EvilGeniusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Evil Genius light platform."""
async_add_entities([EvilGeniusLight(config_entry.runtime_data)])
diff --git a/homeassistant/components/evohome/__init__.py b/homeassistant/components/evohome/__init__.py
index 97f7c2db54d..9dce352df30 100644
--- a/homeassistant/components/evohome/__init__.py
+++ b/homeassistant/components/evohome/__init__.py
@@ -2,27 +2,30 @@
Such systems provide heating/cooling and DHW and include Evohome, Round Thermostat, and
others.
+
+Note that the API used by this integration's client does not support cooling.
"""
from __future__ import annotations
+from dataclasses import dataclass
from datetime import timedelta
import logging
-from typing import Any, Final
+from typing import Final
-import evohomeasync as ev1
-from evohomeasync.schema import SZ_SESSION_ID
-import evohomeasync2 as evo
-from evohomeasync2.schema.const import (
- SZ_AUTO_WITH_RESET,
- SZ_CAN_BE_TEMPORARY,
- SZ_SYSTEM_MODE,
- SZ_TIMING_MODE,
+import evohomeasync as ec1
+import evohomeasync2 as ec2
+from evohomeasync2.const import SZ_CAN_BE_TEMPORARY, SZ_SYSTEM_MODE, SZ_TIMING_MODE
+from evohomeasync2.schemas.const import (
+ S2_DURATION as SZ_DURATION,
+ S2_PERIOD as SZ_PERIOD,
+ SystemMode as EvoSystemMode,
)
import voluptuous as vol
from homeassistant.const import (
ATTR_ENTITY_ID,
+ ATTR_MODE,
CONF_PASSWORD,
CONF_SCAN_INTERVAL,
CONF_USERNAME,
@@ -34,31 +37,22 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.discovery import async_load_platform
from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.service import verify_domain_control
-from homeassistant.helpers.storage import Store
from homeassistant.helpers.typing import ConfigType
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
-from homeassistant.util import dt as dt_util
+from homeassistant.util.hass_dict import HassKey
from .const import (
- ACCESS_TOKEN,
- ACCESS_TOKEN_EXPIRES,
- ATTR_DURATION_DAYS,
- ATTR_DURATION_HOURS,
+ ATTR_DURATION,
ATTR_DURATION_UNTIL,
- ATTR_SYSTEM_MODE,
- ATTR_ZONE_TEMP,
+ ATTR_PERIOD,
+ ATTR_SETPOINT,
CONF_LOCATION_IDX,
DOMAIN,
- REFRESH_TOKEN,
SCAN_INTERVAL_DEFAULT,
SCAN_INTERVAL_MINIMUM,
- STORAGE_KEY,
- STORAGE_VER,
- USER_DATA,
EvoService,
)
-from .coordinator import EvoBroker
-from .helpers import dt_aware_to_naive, dt_local_to_aware, handle_evo_exception
+from .coordinator import EvoDataUpdateCoordinator
+from .storage import TokenManager
_LOGGER = logging.getLogger(__name__)
@@ -87,7 +81,7 @@ RESET_ZONE_OVERRIDE_SCHEMA: Final = vol.Schema(
SET_ZONE_OVERRIDE_SCHEMA: Final = vol.Schema(
{
vol.Required(ATTR_ENTITY_ID): cv.entity_id,
- vol.Required(ATTR_ZONE_TEMP): vol.All(
+ vol.Required(ATTR_SETPOINT): vol.All(
vol.Coerce(float), vol.Range(min=4.0, max=35.0)
),
vol.Optional(ATTR_DURATION_UNTIL): vol.All(
@@ -96,177 +90,69 @@ SET_ZONE_OVERRIDE_SCHEMA: Final = vol.Schema(
}
)
+EVOHOME_KEY: HassKey[EvoData] = HassKey(DOMAIN)
-class EvoSession:
- """Class for evohome client instantiation & authentication."""
- def __init__(self, hass: HomeAssistant) -> None:
- """Initialize the evohome broker and its data structure."""
+@dataclass
+class EvoData:
+ """Dataclass for storing evohome data."""
- self.hass = hass
-
- self._session = async_get_clientsession(hass)
- self._store = Store[dict[str, Any]](hass, STORAGE_VER, STORAGE_KEY)
-
- # the main client, which uses the newer API
- self.client_v2: evo.EvohomeClient | None = None
- self._tokens: dict[str, Any] = {}
-
- # the older client can be used to obtain high-precision temps (only)
- self.client_v1: ev1.EvohomeClient | None = None
- self.session_id: str | None = None
-
- async def authenticate(self, username: str, password: str) -> None:
- """Check the user credentials against the web API.
-
- Will raise evo.AuthenticationFailed if the credentials are invalid.
- """
-
- if (
- self.client_v2 is None
- or username != self.client_v2.username
- or password != self.client_v2.password
- ):
- await self._load_auth_tokens(username)
-
- client_v2 = evo.EvohomeClient(
- username,
- password,
- **self._tokens,
- session=self._session,
- )
-
- else: # force a re-authentication
- client_v2 = self.client_v2
- client_v2._user_account = None # noqa: SLF001
-
- await client_v2.login()
- self.client_v2 = client_v2 # only set attr if authentication succeeded
-
- await self.save_auth_tokens()
-
- self.client_v1 = ev1.EvohomeClient(
- username,
- password,
- session_id=self.session_id,
- session=self._session,
- )
-
- async def _load_auth_tokens(self, username: str) -> None:
- """Load access tokens and session_id from the store and validate them.
-
- Sets self._tokens and self._session_id to the latest values.
- """
-
- app_storage: dict[str, Any] = dict(await self._store.async_load() or {})
-
- if app_storage.pop(CONF_USERNAME, None) != username:
- # any tokens won't be valid, and store might be corrupt
- await self._store.async_save({})
-
- self.session_id = None
- self._tokens = {}
-
- return
-
- # evohomeasync2 requires naive/local datetimes as strings
- if app_storage.get(ACCESS_TOKEN_EXPIRES) is not None and (
- expires := dt_util.parse_datetime(app_storage[ACCESS_TOKEN_EXPIRES])
- ):
- app_storage[ACCESS_TOKEN_EXPIRES] = dt_aware_to_naive(expires)
-
- user_data: dict[str, str] = app_storage.pop(USER_DATA, {}) or {}
-
- self.session_id = user_data.get(SZ_SESSION_ID)
- self._tokens = app_storage
-
- async def save_auth_tokens(self) -> None:
- """Save access tokens and session_id to the store.
-
- Sets self._tokens and self._session_id to the latest values.
- """
-
- if self.client_v2 is None:
- await self._store.async_save({})
- return
-
- # evohomeasync2 uses naive/local datetimes
- access_token_expires = dt_local_to_aware(
- self.client_v2.access_token_expires # type: ignore[arg-type]
- )
-
- self._tokens = {
- CONF_USERNAME: self.client_v2.username,
- REFRESH_TOKEN: self.client_v2.refresh_token,
- ACCESS_TOKEN: self.client_v2.access_token,
- ACCESS_TOKEN_EXPIRES: access_token_expires.isoformat(),
- }
-
- self.session_id = self.client_v1.broker.session_id if self.client_v1 else None
-
- app_storage = self._tokens
- if self.client_v1:
- app_storage[USER_DATA] = {SZ_SESSION_ID: self.session_id}
-
- await self._store.async_save(app_storage)
+ coordinator: EvoDataUpdateCoordinator
+ loc_idx: int
+ tcs: ec2.ControlSystem
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the Evohome integration."""
- sess = EvoSession(hass)
-
- try:
- await sess.authenticate(
- config[DOMAIN][CONF_USERNAME],
- config[DOMAIN][CONF_PASSWORD],
- )
-
- except (evo.AuthenticationFailed, evo.RequestFailed) as err:
- handle_evo_exception(err)
- return False
-
- finally:
- config[DOMAIN][CONF_PASSWORD] = "REDACTED"
-
- broker = EvoBroker(sess)
-
- if not broker.validate_location(
- config[DOMAIN][CONF_LOCATION_IDX],
- ):
- return False
-
- coordinator = DataUpdateCoordinator(
+ token_manager = TokenManager(
+ hass,
+ config[DOMAIN][CONF_USERNAME],
+ config[DOMAIN][CONF_PASSWORD],
+ async_get_clientsession(hass),
+ )
+ coordinator = EvoDataUpdateCoordinator(
hass,
_LOGGER,
- config_entry=None,
+ ec2.EvohomeClient(token_manager),
name=f"{DOMAIN}_coordinator",
update_interval=config[DOMAIN][CONF_SCAN_INTERVAL],
- update_method=broker.async_update,
+ location_idx=config[DOMAIN][CONF_LOCATION_IDX],
+ client_v1=ec1.EvohomeClient(token_manager),
)
+
await coordinator.async_register_shutdown()
+ await coordinator.async_first_refresh()
- hass.data[DOMAIN] = {"broker": broker, "coordinator": coordinator}
+ if not coordinator.last_update_success:
+ _LOGGER.error(f"Failed to fetch initial data: {coordinator.last_exception}") # noqa: G004
+ return False
- # without a listener, _schedule_refresh() won't be invoked by _async_refresh()
- coordinator.async_add_listener(lambda: None)
- await coordinator.async_refresh() # get initial state
+ assert coordinator.tcs is not None # mypy
+
+ hass.data[EVOHOME_KEY] = EvoData(
+ coordinator=coordinator,
+ loc_idx=coordinator.loc_idx,
+ tcs=coordinator.tcs,
+ )
hass.async_create_task(
async_load_platform(hass, Platform.CLIMATE, DOMAIN, {}, config)
)
- if broker.tcs.hotwater:
+ if coordinator.tcs.hotwater:
hass.async_create_task(
async_load_platform(hass, Platform.WATER_HEATER, DOMAIN, {}, config)
)
- setup_service_functions(hass, broker)
+ setup_service_functions(hass, coordinator)
return True
@callback
-def setup_service_functions(hass: HomeAssistant, broker: EvoBroker) -> None:
+def setup_service_functions(
+ hass: HomeAssistant, coordinator: EvoDataUpdateCoordinator
+) -> None:
"""Set up the service handlers for the system/zone operating modes.
Not all Honeywell TCC-compatible systems support all operating modes. In addition,
@@ -279,13 +165,15 @@ def setup_service_functions(hass: HomeAssistant, broker: EvoBroker) -> None:
@verify_domain_control(hass, DOMAIN)
async def force_refresh(call: ServiceCall) -> None:
"""Obtain the latest state data via the vendor's RESTful API."""
- await broker.async_update()
+ await coordinator.async_refresh()
@verify_domain_control(hass, DOMAIN)
async def set_system_mode(call: ServiceCall) -> None:
"""Set the system mode."""
+ assert coordinator.tcs is not None # mypy
+
payload = {
- "unique_id": broker.tcs.systemId,
+ "unique_id": coordinator.tcs.id,
"service": call.service,
"data": call.data,
}
@@ -313,33 +201,39 @@ def setup_service_functions(hass: HomeAssistant, broker: EvoBroker) -> None:
async_dispatcher_send(hass, DOMAIN, payload)
+ assert coordinator.tcs is not None # mypy
+
hass.services.async_register(DOMAIN, EvoService.REFRESH_SYSTEM, force_refresh)
# Enumerate which operating modes are supported by this system
- modes = broker.tcs.allowedSystemModes
+ modes = list(coordinator.tcs.allowed_system_modes)
# Not all systems support "AutoWithReset": register this handler only if required
- if [m[SZ_SYSTEM_MODE] for m in modes if m[SZ_SYSTEM_MODE] == SZ_AUTO_WITH_RESET]:
+ if any(
+ m[SZ_SYSTEM_MODE]
+ for m in modes
+ if m[SZ_SYSTEM_MODE] == EvoSystemMode.AUTO_WITH_RESET
+ ):
hass.services.async_register(DOMAIN, EvoService.RESET_SYSTEM, set_system_mode)
system_mode_schemas = []
- modes = [m for m in modes if m[SZ_SYSTEM_MODE] != SZ_AUTO_WITH_RESET]
+ modes = [m for m in modes if m[SZ_SYSTEM_MODE] != EvoSystemMode.AUTO_WITH_RESET]
# Permanent-only modes will use this schema
perm_modes = [m[SZ_SYSTEM_MODE] for m in modes if not m[SZ_CAN_BE_TEMPORARY]]
if perm_modes: # any of: "Auto", "HeatingOff": permanent only
- schema = vol.Schema({vol.Required(ATTR_SYSTEM_MODE): vol.In(perm_modes)})
+ schema = vol.Schema({vol.Required(ATTR_MODE): vol.In(perm_modes)})
system_mode_schemas.append(schema)
modes = [m for m in modes if m[SZ_CAN_BE_TEMPORARY]]
# These modes are set for a number of hours (or indefinitely): use this schema
- temp_modes = [m[SZ_SYSTEM_MODE] for m in modes if m[SZ_TIMING_MODE] == "Duration"]
+ temp_modes = [m[SZ_SYSTEM_MODE] for m in modes if m[SZ_TIMING_MODE] == SZ_DURATION]
if temp_modes: # any of: "AutoWithEco", permanent or for 0-24 hours
schema = vol.Schema(
{
- vol.Required(ATTR_SYSTEM_MODE): vol.In(temp_modes),
- vol.Optional(ATTR_DURATION_HOURS): vol.All(
+ vol.Required(ATTR_MODE): vol.In(temp_modes),
+ vol.Optional(ATTR_DURATION): vol.All(
cv.time_period,
vol.Range(min=timedelta(hours=0), max=timedelta(hours=24)),
),
@@ -348,12 +242,12 @@ def setup_service_functions(hass: HomeAssistant, broker: EvoBroker) -> None:
system_mode_schemas.append(schema)
# These modes are set for a number of days (or indefinitely): use this schema
- temp_modes = [m[SZ_SYSTEM_MODE] for m in modes if m[SZ_TIMING_MODE] == "Period"]
+ temp_modes = [m[SZ_SYSTEM_MODE] for m in modes if m[SZ_TIMING_MODE] == SZ_PERIOD]
if temp_modes: # any of: "Away", "Custom", "DayOff", permanent or for 1-99 days
schema = vol.Schema(
{
- vol.Required(ATTR_SYSTEM_MODE): vol.In(temp_modes),
- vol.Optional(ATTR_DURATION_DAYS): vol.All(
+ vol.Required(ATTR_MODE): vol.In(temp_modes),
+ vol.Optional(ATTR_PERIOD): vol.All(
cv.time_period,
vol.Range(min=timedelta(days=1), max=timedelta(days=99)),
),
diff --git a/homeassistant/components/evohome/climate.py b/homeassistant/components/evohome/climate.py
index 64e7367bc32..b44dc9791b0 100644
--- a/homeassistant/components/evohome/climate.py
+++ b/homeassistant/components/evohome/climate.py
@@ -4,20 +4,20 @@ from __future__ import annotations
from datetime import datetime, timedelta
import logging
-from typing import TYPE_CHECKING, Any
+from typing import Any
import evohomeasync2 as evo
-from evohomeasync2.schema.const import (
- SZ_ACTIVE_FAULTS,
+from evohomeasync2.const import (
SZ_SETPOINT_STATUS,
- SZ_SYSTEM_ID,
SZ_SYSTEM_MODE,
SZ_SYSTEM_MODE_STATUS,
SZ_TEMPERATURE_STATUS,
- SZ_UNTIL,
- SZ_ZONE_ID,
- ZoneModelType,
- ZoneType,
+)
+from evohomeasync2.schemas.const import (
+ SystemMode as EvoSystemMode,
+ ZoneMode as EvoZoneMode,
+ ZoneModelType as EvoZoneModelType,
+ ZoneType as EvoZoneType,
)
from homeassistant.components.climate import (
@@ -29,68 +29,46 @@ from homeassistant.components.climate import (
ClimateEntityFeature,
HVACMode,
)
-from homeassistant.const import PRECISION_TENTHS, UnitOfTemperature
-from homeassistant.core import HomeAssistant
+from homeassistant.const import ATTR_MODE, PRECISION_TENTHS, UnitOfTemperature
+from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
+from . import EVOHOME_KEY
from .const import (
- ATTR_DURATION_DAYS,
- ATTR_DURATION_HOURS,
+ ATTR_DURATION,
ATTR_DURATION_UNTIL,
- ATTR_SYSTEM_MODE,
- ATTR_ZONE_TEMP,
- DOMAIN,
- EVO_AUTO,
- EVO_AUTOECO,
- EVO_AWAY,
- EVO_CUSTOM,
- EVO_DAYOFF,
- EVO_FOLLOW,
- EVO_HEATOFF,
- EVO_PERMOVER,
- EVO_RESET,
- EVO_TEMPOVER,
+ ATTR_PERIOD,
+ ATTR_SETPOINT,
EvoService,
)
-from .entity import EvoChild, EvoDevice
-
-if TYPE_CHECKING:
- from . import EvoBroker
-
+from .coordinator import EvoDataUpdateCoordinator
+from .entity import EvoChild, EvoEntity
_LOGGER = logging.getLogger(__name__)
-PRESET_RESET = "Reset" # reset all child zones to EVO_FOLLOW
+PRESET_RESET = "Reset" # reset all child zones to EvoZoneMode.FOLLOW_SCHEDULE
PRESET_CUSTOM = "Custom"
TCS_PRESET_TO_HA = {
- EVO_AWAY: PRESET_AWAY,
- EVO_CUSTOM: PRESET_CUSTOM,
- EVO_AUTOECO: PRESET_ECO,
- EVO_DAYOFF: PRESET_HOME,
- EVO_RESET: PRESET_RESET,
-} # EVO_AUTO: None,
+ EvoSystemMode.AWAY: PRESET_AWAY,
+ EvoSystemMode.CUSTOM: PRESET_CUSTOM,
+ EvoSystemMode.AUTO_WITH_ECO: PRESET_ECO,
+ EvoSystemMode.DAY_OFF: PRESET_HOME,
+ EvoSystemMode.AUTO_WITH_RESET: PRESET_RESET,
+} # EvoSystemMode.AUTO: None,
HA_PRESET_TO_TCS = {v: k for k, v in TCS_PRESET_TO_HA.items()}
EVO_PRESET_TO_HA = {
- EVO_FOLLOW: PRESET_NONE,
- EVO_TEMPOVER: "temporary",
- EVO_PERMOVER: "permanent",
+ EvoZoneMode.FOLLOW_SCHEDULE: PRESET_NONE,
+ EvoZoneMode.TEMPORARY_OVERRIDE: "temporary",
+ EvoZoneMode.PERMANENT_OVERRIDE: "permanent",
}
HA_PRESET_TO_EVO = {v: k for k, v in EVO_PRESET_TO_HA.items()}
-STATE_ATTRS_TCS = [SZ_SYSTEM_ID, SZ_ACTIVE_FAULTS, SZ_SYSTEM_MODE_STATUS]
-STATE_ATTRS_ZONES = [
- SZ_ZONE_ID,
- SZ_ACTIVE_FAULTS,
- SZ_SETPOINT_STATUS,
- SZ_TEMPERATURE_STATUS,
-]
-
async def async_setup_platform(
hass: HomeAssistant,
@@ -102,32 +80,34 @@ async def async_setup_platform(
if discovery_info is None:
return
- broker: EvoBroker = hass.data[DOMAIN]["broker"]
+ coordinator = hass.data[EVOHOME_KEY].coordinator
+ loc_idx = hass.data[EVOHOME_KEY].loc_idx
+ tcs = hass.data[EVOHOME_KEY].tcs
_LOGGER.debug(
"Found the Location/Controller (%s), id=%s, name=%s (location_idx=%s)",
- broker.tcs.modelType,
- broker.tcs.systemId,
- broker.loc.name,
- broker.loc_idx,
+ tcs.model,
+ tcs.id,
+ tcs.location.name,
+ loc_idx,
)
- entities: list[EvoClimateEntity] = [EvoController(broker, broker.tcs)]
+ entities: list[EvoController | EvoZone] = [EvoController(coordinator, tcs)]
- for zone in broker.tcs.zones.values():
+ for zone in tcs.zones:
if (
- zone.modelType == ZoneModelType.HEATING_ZONE
- or zone.zoneType == ZoneType.THERMOSTAT
+ zone.model == EvoZoneModelType.HEATING_ZONE
+ or zone.type == EvoZoneType.THERMOSTAT
):
_LOGGER.debug(
"Adding: %s (%s), id=%s, name=%s",
- zone.zoneType,
- zone.modelType,
- zone.zoneId,
+ zone.type,
+ zone.model,
+ zone.id,
zone.name,
)
- new_entity = EvoZone(broker, zone)
+ new_entity = EvoZone(coordinator, zone)
entities.append(new_entity)
else:
@@ -136,16 +116,19 @@ async def async_setup_platform(
"Ignoring: %s (%s), id=%s, name=%s: unknown/invalid zone type, "
"report as an issue if you feel this zone type should be supported"
),
- zone.zoneType,
- zone.modelType,
- zone.zoneId,
+ zone.type,
+ zone.model,
+ zone.id,
zone.name,
)
- async_add_entities(entities, update_before_add=True)
+ async_add_entities(entities)
+
+ for entity in entities:
+ await entity.update_attrs()
-class EvoClimateEntity(EvoDevice, ClimateEntity):
+class EvoClimateEntity(EvoEntity, ClimateEntity):
"""Base for any evohome-compatible climate entity (controller, zone)."""
_attr_hvac_modes = [HVACMode.OFF, HVACMode.HEAT]
@@ -157,25 +140,29 @@ class EvoZone(EvoChild, EvoClimateEntity):
_attr_preset_modes = list(HA_PRESET_TO_EVO)
- _evo_device: evo.Zone # mypy hint
+ _evo_device: evo.Zone
+ _evo_id_attr = "zone_id"
+ _evo_state_attr_names = (SZ_SETPOINT_STATUS, SZ_TEMPERATURE_STATUS)
- def __init__(self, evo_broker: EvoBroker, evo_device: evo.Zone) -> None:
+ def __init__(
+ self, coordinator: EvoDataUpdateCoordinator, evo_device: evo.Zone
+ ) -> None:
"""Initialize an evohome-compatible heating zone."""
- super().__init__(evo_broker, evo_device)
- self._evo_id = evo_device.zoneId
+ super().__init__(coordinator, evo_device)
+ self._evo_id = evo_device.id
- if evo_device.modelType.startswith("VisionProWifi"):
+ if evo_device.model.startswith("VisionProWifi"):
# this system does not have a distinct ID for the zone
- self._attr_unique_id = f"{evo_device.zoneId}z"
+ self._attr_unique_id = f"{evo_device.id}z"
else:
- self._attr_unique_id = evo_device.zoneId
+ self._attr_unique_id = evo_device.id
- if evo_broker.client_v1:
+ if coordinator.client_v1:
self._attr_precision = PRECISION_TENTHS
else:
- self._attr_precision = self._evo_device.setpointCapabilities[
- "valueResolution"
+ self._attr_precision = self._evo_device.setpoint_capabilities[
+ "value_resolution"
]
self._attr_supported_features = (
@@ -188,24 +175,24 @@ class EvoZone(EvoChild, EvoClimateEntity):
async def async_zone_svc_request(self, service: str, data: dict[str, Any]) -> None:
"""Process a service request (setpoint override) for a zone."""
if service == EvoService.RESET_ZONE_OVERRIDE:
- await self._evo_broker.call_client_api(self._evo_device.reset_mode())
+ await self.coordinator.call_client_api(self._evo_device.reset())
return
# otherwise it is EvoService.SET_ZONE_OVERRIDE
- temperature = max(min(data[ATTR_ZONE_TEMP], self.max_temp), self.min_temp)
+ temperature = max(min(data[ATTR_SETPOINT], self.max_temp), self.min_temp)
if ATTR_DURATION_UNTIL in data:
duration: timedelta = data[ATTR_DURATION_UNTIL]
if duration.total_seconds() == 0:
await self._update_schedule()
- until = dt_util.parse_datetime(self.setpoints.get("next_sp_from", ""))
+ until = self.setpoints.get("next_sp_from")
else:
until = dt_util.now() + data[ATTR_DURATION_UNTIL]
else:
until = None # indefinitely
until = dt_util.as_utc(until) if until else None
- await self._evo_broker.call_client_api(
+ await self.coordinator.call_client_api(
self._evo_device.set_temperature(temperature, until=until)
)
@@ -217,7 +204,7 @@ class EvoZone(EvoChild, EvoClimateEntity):
@property
def hvac_mode(self) -> HVACMode | None:
"""Return the current operating mode of a Zone."""
- if self._evo_tcs.system_mode in (EVO_AWAY, EVO_HEATOFF):
+ if self._evo_tcs.mode in (EvoSystemMode.AWAY, EvoSystemMode.HEATING_OFF):
return HVACMode.AUTO
if self.target_temperature is None:
return None
@@ -233,10 +220,8 @@ class EvoZone(EvoChild, EvoClimateEntity):
@property
def preset_mode(self) -> str | None:
"""Return the current preset mode, e.g., home, away, temp."""
- if self._evo_tcs.system_mode in (EVO_AWAY, EVO_HEATOFF):
- return TCS_PRESET_TO_HA.get(self._evo_tcs.system_mode)
- if self._evo_device.mode is None:
- return None
+ if self._evo_tcs.mode in (EvoSystemMode.AWAY, EvoSystemMode.HEATING_OFF):
+ return TCS_PRESET_TO_HA.get(self._evo_tcs.mode)
return EVO_PRESET_TO_HA.get(self._evo_device.mode)
@property
@@ -245,8 +230,6 @@ class EvoZone(EvoChild, EvoClimateEntity):
The default is 5, but is user-configurable within 5-21 (in Celsius).
"""
- if self._evo_device.min_heat_setpoint is None:
- return 5
return self._evo_device.min_heat_setpoint
@property
@@ -255,33 +238,27 @@ class EvoZone(EvoChild, EvoClimateEntity):
The default is 35, but is user-configurable within 21-35 (in Celsius).
"""
- if self._evo_device.max_heat_setpoint is None:
- return 35
return self._evo_device.max_heat_setpoint
async def async_set_temperature(self, **kwargs: Any) -> None:
"""Set a new target temperature."""
- assert self._evo_device.setpointStatus is not None # mypy check
-
temperature = kwargs["temperature"]
if (until := kwargs.get("until")) is None:
- if self._evo_device.mode == EVO_FOLLOW:
+ if self._evo_device.mode == EvoZoneMode.TEMPORARY_OVERRIDE:
+ until = self._evo_device.until
+ if self._evo_device.mode == EvoZoneMode.FOLLOW_SCHEDULE:
await self._update_schedule()
- until = dt_util.parse_datetime(self.setpoints.get("next_sp_from", ""))
- elif self._evo_device.mode == EVO_TEMPOVER:
- until = dt_util.parse_datetime(
- self._evo_device.setpointStatus[SZ_UNTIL]
- )
+ until = self.setpoints.get("next_sp_from")
until = dt_util.as_utc(until) if until else None
- await self._evo_broker.call_client_api(
+ await self.coordinator.call_client_api(
self._evo_device.set_temperature(temperature, until=until)
)
async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
- """Set a Zone to one of its native EVO_* operating modes.
+ """Set a Zone to one of its native operating modes.
Zones inherit their _effective_ operating mode from their Controller.
@@ -298,41 +275,34 @@ class EvoZone(EvoChild, EvoClimateEntity):
and 'Away', Zones to (by default) 12C.
"""
if hvac_mode == HVACMode.OFF:
- await self._evo_broker.call_client_api(
+ await self.coordinator.call_client_api(
self._evo_device.set_temperature(self.min_temp, until=None)
)
else: # HVACMode.HEAT
- await self._evo_broker.call_client_api(self._evo_device.reset_mode())
+ await self.coordinator.call_client_api(self._evo_device.reset())
async def async_set_preset_mode(self, preset_mode: str) -> None:
"""Set the preset mode; if None, then revert to following the schedule."""
- evo_preset_mode = HA_PRESET_TO_EVO.get(preset_mode, EVO_FOLLOW)
+ evo_preset_mode = HA_PRESET_TO_EVO.get(preset_mode, EvoZoneMode.FOLLOW_SCHEDULE)
- if evo_preset_mode == EVO_FOLLOW:
- await self._evo_broker.call_client_api(self._evo_device.reset_mode())
+ if evo_preset_mode == EvoZoneMode.FOLLOW_SCHEDULE:
+ await self.coordinator.call_client_api(self._evo_device.reset())
return
- if evo_preset_mode == EVO_TEMPOVER:
+ if evo_preset_mode == EvoZoneMode.TEMPORARY_OVERRIDE:
await self._update_schedule()
- until = dt_util.parse_datetime(self.setpoints.get("next_sp_from", ""))
- else: # EVO_PERMOVER
+ until = self.setpoints.get("next_sp_from")
+ else: # EvoZoneMode.PERMANENT_OVERRIDE
until = None
temperature = self._evo_device.target_heat_temperature
assert temperature is not None # mypy check
until = dt_util.as_utc(until) if until else None
- await self._evo_broker.call_client_api(
+ await self.coordinator.call_client_api(
self._evo_device.set_temperature(temperature, until=until)
)
- async def async_update(self) -> None:
- """Get the latest state data for a Zone."""
- await super().async_update()
-
- for attr in STATE_ATTRS_ZONES:
- self._device_state_attrs[attr] = getattr(self._evo_device, attr)
-
class EvoController(EvoClimateEntity):
"""Base for any evohome-compatible controller.
@@ -347,18 +317,22 @@ class EvoController(EvoClimateEntity):
_attr_icon = "mdi:thermostat"
_attr_precision = PRECISION_TENTHS
- _evo_device: evo.ControlSystem # mypy hint
+ _evo_device: evo.ControlSystem
+ _evo_id_attr = "system_id"
+ _evo_state_attr_names = (SZ_SYSTEM_MODE_STATUS,)
- def __init__(self, evo_broker: EvoBroker, evo_device: evo.ControlSystem) -> None:
+ def __init__(
+ self, coordinator: EvoDataUpdateCoordinator, evo_device: evo.ControlSystem
+ ) -> None:
"""Initialize an evohome-compatible controller."""
- super().__init__(evo_broker, evo_device)
- self._evo_id = evo_device.systemId
+ super().__init__(coordinator, evo_device)
+ self._evo_id = evo_device.id
- self._attr_unique_id = evo_device.systemId
+ self._attr_unique_id = evo_device.id
self._attr_name = evo_device.location.name
- self._evo_modes = [m[SZ_SYSTEM_MODE] for m in evo_device.allowedSystemModes]
+ self._evo_modes = [m[SZ_SYSTEM_MODE] for m in evo_device.allowed_system_modes]
self._attr_preset_modes = [
TCS_PRESET_TO_HA[m] for m in self._evo_modes if m in list(TCS_PRESET_TO_HA)
]
@@ -374,34 +348,40 @@ class EvoController(EvoClimateEntity):
Data validation is not required, it will have been done upstream.
"""
if service == EvoService.SET_SYSTEM_MODE:
- mode = data[ATTR_SYSTEM_MODE]
+ mode = data[ATTR_MODE]
else: # otherwise it is EvoService.RESET_SYSTEM
- mode = EVO_RESET
+ mode = EvoSystemMode.AUTO_WITH_RESET
- if ATTR_DURATION_DAYS in data:
+ if ATTR_PERIOD in data:
until = dt_util.start_of_local_day()
- until += data[ATTR_DURATION_DAYS]
+ until += data[ATTR_PERIOD]
- elif ATTR_DURATION_HOURS in data:
- until = dt_util.now() + data[ATTR_DURATION_HOURS]
+ elif ATTR_DURATION in data:
+ until = dt_util.now() + data[ATTR_DURATION]
else:
until = None
await self._set_tcs_mode(mode, until=until)
- async def _set_tcs_mode(self, mode: str, until: datetime | None = None) -> None:
- """Set a Controller to any of its native EVO_* operating modes."""
+ async def _set_tcs_mode(
+ self, mode: EvoSystemMode, until: datetime | None = None
+ ) -> None:
+ """Set a Controller to any of its native operating modes."""
until = dt_util.as_utc(until) if until else None
- await self._evo_broker.call_client_api(
- self._evo_device.set_mode(mode, until=until) # type: ignore[arg-type]
+ await self.coordinator.call_client_api(
+ self._evo_device.set_mode(mode, until=until)
)
@property
def hvac_mode(self) -> HVACMode:
"""Return the current operating mode of a Controller."""
- evo_mode = self._evo_device.system_mode
- return HVACMode.OFF if evo_mode in (EVO_HEATOFF, "Off") else HVACMode.HEAT
+ evo_mode = self._evo_device.mode
+ return (
+ HVACMode.OFF
+ if evo_mode in (EvoSystemMode.HEATING_OFF, EvoSystemMode.OFF)
+ else HVACMode.HEAT
+ )
@property
def current_temperature(self) -> float | None:
@@ -410,18 +390,14 @@ class EvoController(EvoClimateEntity):
Controllers do not have a current temp, but one is expected by HA.
"""
temps = [
- z.temperature
- for z in self._evo_device.zones.values()
- if z.temperature is not None
+ z.temperature for z in self._evo_device.zones if z.temperature is not None
]
return round(sum(temps) / len(temps), 1) if temps else None
@property
def preset_mode(self) -> str | None:
"""Return the current preset mode, e.g., home, away, temp."""
- if not self._evo_device.system_mode:
- return None
- return TCS_PRESET_TO_HA.get(self._evo_device.system_mode)
+ return TCS_PRESET_TO_HA.get(self._evo_device.mode)
async def async_set_temperature(self, **kwargs: Any) -> None:
"""Raise exception as Controllers don't have a target temperature."""
@@ -429,25 +405,40 @@ class EvoController(EvoClimateEntity):
async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
"""Set an operating mode for a Controller."""
+
+ evo_mode: EvoSystemMode
+
if hvac_mode == HVACMode.HEAT:
- evo_mode = EVO_AUTO if EVO_AUTO in self._evo_modes else "Heat"
+ evo_mode = (
+ EvoSystemMode.AUTO
+ if EvoSystemMode.AUTO in self._evo_modes
+ else EvoSystemMode.HEAT
+ )
elif hvac_mode == HVACMode.OFF:
- evo_mode = EVO_HEATOFF if EVO_HEATOFF in self._evo_modes else "Off"
+ evo_mode = (
+ EvoSystemMode.HEATING_OFF
+ if EvoSystemMode.HEATING_OFF in self._evo_modes
+ else EvoSystemMode.OFF
+ )
else:
raise HomeAssistantError(f"Invalid hvac_mode: {hvac_mode}")
await self._set_tcs_mode(evo_mode)
async def async_set_preset_mode(self, preset_mode: str) -> None:
"""Set the preset mode; if None, then revert to 'Auto' mode."""
- await self._set_tcs_mode(HA_PRESET_TO_TCS.get(preset_mode, EVO_AUTO))
+ await self._set_tcs_mode(HA_PRESET_TO_TCS.get(preset_mode, EvoSystemMode.AUTO))
- async def async_update(self) -> None:
- """Get the latest state data for a Controller."""
- self._device_state_attrs = {}
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
- attrs = self._device_state_attrs
- for attr in STATE_ATTRS_TCS:
- if attr == SZ_ACTIVE_FAULTS:
- attrs["activeSystemFaults"] = getattr(self._evo_device, attr)
- else:
- attrs[attr] = getattr(self._evo_device, attr)
+ self._device_state_attrs = {
+ "activeSystemFaults": self._evo_device.active_faults
+ + self._evo_device.gateway.active_faults
+ }
+
+ super()._handle_coordinator_update()
+
+ async def update_attrs(self) -> None:
+ """Update the entity's extra state attrs."""
+ self._handle_coordinator_update()
diff --git a/homeassistant/components/evohome/const.py b/homeassistant/components/evohome/const.py
index 3ebe6954fea..9da5969df1e 100644
--- a/homeassistant/components/evohome/const.py
+++ b/homeassistant/components/evohome/const.py
@@ -11,41 +11,17 @@ DOMAIN: Final = "evohome"
STORAGE_VER: Final = 1
STORAGE_KEY: Final = DOMAIN
-# The Parent's (i.e. TCS, Controller) operating mode is one of:
-EVO_RESET: Final = "AutoWithReset"
-EVO_AUTO: Final = "Auto"
-EVO_AUTOECO: Final = "AutoWithEco"
-EVO_AWAY: Final = "Away"
-EVO_DAYOFF: Final = "DayOff"
-EVO_CUSTOM: Final = "Custom"
-EVO_HEATOFF: Final = "HeatingOff"
-
-# The Children's (i.e. Dhw, Zone) operating mode is one of:
-EVO_FOLLOW: Final = "FollowSchedule" # the operating mode is 'inherited' from the TCS
-EVO_TEMPOVER: Final = "TemporaryOverride"
-EVO_PERMOVER: Final = "PermanentOverride"
-
-# These two are used only to help prevent E501 (line too long) violations
-GWS: Final = "gateways"
-TCS: Final = "temperatureControlSystems"
-
-UTC_OFFSET: Final = "currentOffsetMinutes"
-
CONF_LOCATION_IDX: Final = "location_idx"
-ACCESS_TOKEN: Final = "access_token"
-ACCESS_TOKEN_EXPIRES: Final = "access_token_expires"
-REFRESH_TOKEN: Final = "refresh_token"
USER_DATA: Final = "user_data"
SCAN_INTERVAL_DEFAULT: Final = timedelta(seconds=300)
SCAN_INTERVAL_MINIMUM: Final = timedelta(seconds=60)
-ATTR_SYSTEM_MODE: Final = "mode"
-ATTR_DURATION_DAYS: Final = "period"
-ATTR_DURATION_HOURS: Final = "duration"
+ATTR_PERIOD: Final = "period" # number of days
+ATTR_DURATION: Final = "duration" # number of minutes, <24h
-ATTR_ZONE_TEMP: Final = "setpoint"
+ATTR_SETPOINT: Final = "setpoint"
ATTR_DURATION_UNTIL: Final = "duration"
diff --git a/homeassistant/components/evohome/coordinator.py b/homeassistant/components/evohome/coordinator.py
index 943bd6605b4..33af90089a4 100644
--- a/homeassistant/components/evohome/coordinator.py
+++ b/homeassistant/components/evohome/coordinator.py
@@ -4,109 +4,147 @@ from __future__ import annotations
from collections.abc import Awaitable
from datetime import timedelta
+from http import HTTPStatus
import logging
-from typing import TYPE_CHECKING, Any
+from typing import Any
-import evohomeasync as ev1
-from evohomeasync.schema import SZ_ID, SZ_TEMP
-import evohomeasync2 as evo
-from evohomeasync2.schema.const import (
+import evohomeasync as ec1
+import evohomeasync2 as ec2
+from evohomeasync2.const import (
+ SZ_DHW,
SZ_GATEWAY_ID,
SZ_GATEWAY_INFO,
+ SZ_GATEWAYS,
SZ_LOCATION_ID,
SZ_LOCATION_INFO,
+ SZ_TEMPERATURE_CONTROL_SYSTEMS,
SZ_TIME_ZONE,
+ SZ_USE_DAYLIGHT_SAVE_SWITCHING,
+ SZ_ZONES,
)
+from evohomeasync2.schemas.typedefs import EvoLocStatusResponseT, EvoTcsConfigResponseT
-from homeassistant.helpers.dispatcher import async_dispatcher_send
-
-from .const import CONF_LOCATION_IDX, DOMAIN, GWS, TCS, UTC_OFFSET
-from .helpers import handle_evo_exception
-
-if TYPE_CHECKING:
- from . import EvoSession
-
-_LOGGER = logging.getLogger(__name__.rpartition(".")[0])
+from homeassistant.const import CONF_SCAN_INTERVAL
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-class EvoBroker:
- """Broker for evohome client broker."""
+class EvoDataUpdateCoordinator(DataUpdateCoordinator):
+ """Coordinator for evohome integration/client."""
- loc_idx: int
- loc: evo.Location
- loc_utc_offset: timedelta
- tcs: evo.ControlSystem
+ # These will not be None after _async_setup())
+ loc: ec2.Location
+ tcs: ec2.ControlSystem
- def __init__(self, sess: EvoSession) -> None:
- """Initialize the evohome broker and its data structure."""
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ logger: logging.Logger,
+ client_v2: ec2.EvohomeClient,
+ *,
+ name: str,
+ update_interval: timedelta,
+ location_idx: int,
+ client_v1: ec1.EvohomeClient | None = None,
+ ) -> None:
+ """Class to manage fetching data."""
- self._sess = sess
- self.hass = sess.hass
+ super().__init__(
+ hass,
+ logger,
+ config_entry=None,
+ name=name,
+ update_interval=update_interval,
+ )
- assert sess.client_v2 is not None # mypy
+ self.client = client_v2
+ self.client_v1 = client_v1
- self.client = sess.client_v2
- self.client_v1 = sess.client_v1
+ self.loc_idx = location_idx
+ self.data: EvoLocStatusResponseT = None # type: ignore[assignment]
self.temps: dict[str, float | None] = {}
- def validate_location(self, loc_idx: int) -> bool:
- """Get the default TCS of the specified location."""
+ self._first_refresh_done = False # get schedules only after first refresh
- self.loc_idx = loc_idx
+ # our version of async_config_entry_first_refresh()...
+ async def async_first_refresh(self) -> None:
+ """Refresh data for the first time when integration is setup.
- assert self.client.installation_info is not None # mypy
+ This integration does not have config flow, so it is inappropriate to
+ invoke `async_config_entry_first_refresh()`.
+ """
+
+ # can't replicate `if not await self.__wrap_async_setup():` (is mangled), so...
+ if not await self._DataUpdateCoordinator__wrap_async_setup(): # type: ignore[attr-defined]
+ return
+
+ await self._async_refresh(
+ log_failures=False, raise_on_auth_failed=True, raise_on_entry_error=True
+ )
+
+ async def _async_setup(self) -> None:
+ """Set up the coordinator.
+
+ Fetch the user information, and the configuration of their locations.
+ """
try:
- loc_config = self.client.installation_info[loc_idx]
- except IndexError:
- _LOGGER.error(
- (
- "Config error: '%s' = %s, but the valid range is 0-%s. "
- "Unable to continue. Fix any configuration errors and restart HA"
- ),
- CONF_LOCATION_IDX,
- loc_idx,
- len(self.client.installation_info) - 1,
- )
- return False
+ await self.client.update(dont_update_status=True) # only config for now
+ except ec2.EvohomeError as err:
+ raise UpdateFailed(err) from err
- self.loc = self.client.locations[loc_idx]
- self.loc_utc_offset = timedelta(minutes=self.loc.timeZone[UTC_OFFSET])
- self.tcs = self.loc._gateways[0]._control_systems[0] # noqa: SLF001
+ try:
+ self.loc = self.client.locations[self.loc_idx]
+ except IndexError as err:
+ raise UpdateFailed(
+ f"""
+ Config error: 'location_idx' = {self.loc_idx},
+ but the valid range is 0-{len(self.client.locations) - 1}.
+ Unable to continue. Fix any configuration errors and restart HA
+ """
+ ) from err
- if _LOGGER.isEnabledFor(logging.DEBUG):
+ self.tcs = self.loc.gateways[0].systems[0]
+
+ if self.logger.isEnabledFor(logging.DEBUG):
loc_info = {
- SZ_LOCATION_ID: loc_config[SZ_LOCATION_INFO][SZ_LOCATION_ID],
- SZ_TIME_ZONE: loc_config[SZ_LOCATION_INFO][SZ_TIME_ZONE],
+ SZ_LOCATION_ID: self.loc.id,
+ SZ_TIME_ZONE: self.loc.config[SZ_TIME_ZONE],
+ SZ_USE_DAYLIGHT_SAVE_SWITCHING: self.loc.config[
+ SZ_USE_DAYLIGHT_SAVE_SWITCHING
+ ],
}
+ tcs_info: EvoTcsConfigResponseT = self.tcs.config # type: ignore[assignment]
+ tcs_info[SZ_ZONES] = [zone.config for zone in self.tcs.zones]
+ if self.tcs.hotwater:
+ tcs_info[SZ_DHW] = self.tcs.hotwater.config
gwy_info = {
- SZ_GATEWAY_ID: loc_config[GWS][0][SZ_GATEWAY_INFO][SZ_GATEWAY_ID],
- TCS: loc_config[GWS][0][TCS],
+ SZ_GATEWAY_ID: self.loc.gateways[0].id,
+ SZ_TEMPERATURE_CONTROL_SYSTEMS: [tcs_info],
}
config = {
SZ_LOCATION_INFO: loc_info,
- GWS: [{SZ_GATEWAY_INFO: gwy_info}],
+ SZ_GATEWAYS: [{SZ_GATEWAY_INFO: gwy_info}],
}
- _LOGGER.debug("Config = %s", config)
-
- return True
+ self.logger.debug("Config = %s", [config])
async def call_client_api(
self,
client_api: Awaitable[dict[str, Any] | None],
- update_state: bool = True,
+ request_refresh: bool = True,
) -> dict[str, Any] | None:
- """Call a client API and update the broker state if required."""
+ """Call a client API and update the Coordinator state if required."""
try:
result = await client_api
- except evo.RequestFailed as err:
- handle_evo_exception(err)
+
+ except ec2.ApiRequestFailedError as err:
+ self.logger.error(err)
return None
- if update_state: # wait a moment for system to quiesce before updating state
- await self.hass.data[DOMAIN]["coordinator"].async_request_refresh()
+ if request_refresh: # wait a moment for system to quiesce before updating state
+ await self.async_request_refresh() # hass.async_create_task() won't help
return result
@@ -115,80 +153,90 @@ class EvoBroker:
assert self.client_v1 is not None # mypy check
- old_session_id = self._sess.session_id
-
try:
- temps = await self.client_v1.get_temperatures()
+ await self.client_v1.update()
- except ev1.InvalidSchema as err:
- _LOGGER.warning(
+ except ec1.BadUserCredentialsError as err:
+ self.logger.warning(
(
"Unable to obtain high-precision temperatures. "
- "It appears the JSON schema is not as expected, "
- "so the high-precision feature will be disabled until next restart."
- "Message is: %s"
+ "The feature will be disabled until next restart: %r"
),
err,
)
self.client_v1 = None
- except ev1.RequestFailed as err:
- _LOGGER.warning(
+ except ec1.EvohomeError as err:
+ self.logger.warning(
(
"Unable to obtain the latest high-precision temperatures. "
- "Check your network and the vendor's service status page. "
- "Proceeding without high-precision temperatures for now. "
- "Message is: %s"
+ "They will be ignored this refresh cycle: %r"
),
err,
)
self.temps = {} # high-precision temps now considered stale
- except Exception:
- self.temps = {} # high-precision temps now considered stale
- raise
-
else:
- if str(self.client_v1.location_id) != self.loc.locationId:
- _LOGGER.warning(
- "The v2 API's configured location doesn't match "
- "the v1 API's default location (there is more than one location), "
- "so the high-precision feature will be disabled until next restart"
- )
- self.client_v1 = None
- else:
- self.temps = {str(i[SZ_ID]): i[SZ_TEMP] for i in temps}
+ self.temps = await self.client_v1.location_by_id[
+ self.loc.id
+ ].get_temperatures(dont_update_status=True)
- finally:
- if self.client_v1 and self.client_v1.broker.session_id != old_session_id:
- await self._sess.save_auth_tokens()
-
- _LOGGER.debug("Temperatures = %s", self.temps)
+ self.logger.debug("Status (high-res temps) = %s", self.temps)
async def _update_v2_api_state(self, *args: Any) -> None:
"""Get the latest modes, temperatures, setpoints of a Location."""
- access_token = self.client.access_token # maybe receive a new token?
-
try:
- status = await self.loc.refresh_status()
- except evo.RequestFailed as err:
- handle_evo_exception(err)
- else:
- async_dispatcher_send(self.hass, DOMAIN)
- _LOGGER.debug("Status = %s", status)
- finally:
- if access_token != self.client.access_token:
- await self._sess.save_auth_tokens()
+ status = await self.loc.update()
- async def async_update(self, *args: Any) -> None:
- """Get the latest state data of an entire Honeywell TCC Location.
+ except ec2.ApiRequestFailedError as err:
+ if err.status != HTTPStatus.TOO_MANY_REQUESTS:
+ raise UpdateFailed(err) from err
+
+ raise UpdateFailed(
+ f"""
+ The vendor's API rate limit has been exceeded.
+ Consider increasing the {CONF_SCAN_INTERVAL}
+ """
+ ) from err
+
+ except ec2.EvohomeError as err:
+ raise UpdateFailed(err) from err
+
+ self.logger.debug("Status = %s", status)
+
+ async def _update_v2_schedules(self) -> None:
+ for zone in self.tcs.zones:
+ try:
+ await zone.get_schedule()
+ except ec2.InvalidScheduleError as err:
+ self.logger.warning(
+ "Zone '%s' has an invalid/missing schedule: %r", zone.name, err
+ )
+
+ if dhw := self.tcs.hotwater:
+ try:
+ await dhw.get_schedule()
+ except ec2.InvalidScheduleError as err:
+ self.logger.warning("DHW has an invalid/missing schedule: %r", err)
+
+ async def _async_update_data(self) -> EvoLocStatusResponseT: # type: ignore[override]
+ """Fetch the latest state of an entire TCC Location.
This includes state data for a Controller and all its child devices, such as the
operating mode of the Controller and the current temp of its children (e.g.
Zones, DHW controller).
"""
- await self._update_v2_api_state()
+ await self._update_v2_api_state() # may raise UpdateFailed
if self.client_v1:
- await self._update_v1_api_temps()
+ await self._update_v1_api_temps() # will never raise UpdateFailed
+
+ # to speed up HA startup, don't update entity schedules during initial
+ # async_first_refresh(), only during subsequent async_refresh()...
+ if self._first_refresh_done:
+ await self._update_v2_schedules()
+ else:
+ self._first_refresh_done = True
+
+ return self.loc.status
diff --git a/homeassistant/components/evohome/entity.py b/homeassistant/components/evohome/entity.py
index a42d8ef7582..2f93f0fb143 100644
--- a/homeassistant/components/evohome/entity.py
+++ b/homeassistant/components/evohome/entity.py
@@ -1,55 +1,50 @@
"""Base for evohome entity."""
-from datetime import datetime, timedelta, timezone
+from collections.abc import Mapping
+from datetime import UTC, datetime
import logging
from typing import Any
import evohomeasync2 as evo
-from evohomeasync2.schema.const import (
- SZ_HEAT_SETPOINT,
- SZ_SETPOINT_STATUS,
- SZ_STATE_STATUS,
- SZ_SYSTEM_MODE_STATUS,
- SZ_TIME_UNTIL,
- SZ_UNTIL,
-)
+from evohomeasync2.schemas.typedefs import DayOfWeekDhwT
+from homeassistant.core import callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity import Entity
-from homeassistant.util import dt as dt_util
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import EvoBroker, EvoService
-from .const import DOMAIN
-from .helpers import convert_dict, convert_until
+from .const import DOMAIN, EvoService
+from .coordinator import EvoDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
-class EvoDevice(Entity):
+class EvoEntity(CoordinatorEntity[EvoDataUpdateCoordinator]):
"""Base for any evohome-compatible entity (controller, DHW, zone).
This includes the controller, (1 to 12) heating zones and (optionally) a
DHW controller.
"""
- _attr_should_poll = False
+ _evo_device: evo.ControlSystem | evo.HotWater | evo.Zone
+ _evo_id_attr: str
+ _evo_state_attr_names: tuple[str, ...]
def __init__(
self,
- evo_broker: EvoBroker,
+ coordinator: EvoDataUpdateCoordinator,
evo_device: evo.ControlSystem | evo.HotWater | evo.Zone,
) -> None:
"""Initialize an evohome-compatible entity (TCS, DHW, zone)."""
+ super().__init__(coordinator, context=evo_device.id)
self._evo_device = evo_device
- self._evo_broker = evo_broker
self._device_state_attrs: dict[str, Any] = {}
- async def async_refresh(self, payload: dict | None = None) -> None:
+ async def process_signal(self, payload: dict | None = None) -> None:
"""Process any signals."""
+
if payload is None:
- self.async_schedule_update_ha_state(force_refresh=True)
- return
+ raise NotImplementedError
if payload["unique_id"] != self._attr_unique_id:
return
if payload["service"] in (
@@ -69,40 +64,46 @@ class EvoDevice(Entity):
raise NotImplementedError
@property
- def extra_state_attributes(self) -> dict[str, Any]:
+ def extra_state_attributes(self) -> Mapping[str, Any]:
"""Return the evohome-specific state attributes."""
- status = self._device_state_attrs
- if SZ_SYSTEM_MODE_STATUS in status:
- convert_until(status[SZ_SYSTEM_MODE_STATUS], SZ_TIME_UNTIL)
- if SZ_SETPOINT_STATUS in status:
- convert_until(status[SZ_SETPOINT_STATUS], SZ_UNTIL)
- if SZ_STATE_STATUS in status:
- convert_until(status[SZ_STATE_STATUS], SZ_UNTIL)
-
- return {"status": convert_dict(status)}
+ return {"status": self._device_state_attrs}
async def async_added_to_hass(self) -> None:
"""Run when entity about to be added to hass."""
- async_dispatcher_connect(self.hass, DOMAIN, self.async_refresh)
+ await super().async_added_to_hass()
+
+ async_dispatcher_connect(self.hass, DOMAIN, self.process_signal)
+
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
+
+ self._device_state_attrs[self._evo_id_attr] = self._evo_device.id
+
+ for attr in self._evo_state_attr_names:
+ self._device_state_attrs[attr] = getattr(self._evo_device, attr)
+
+ super()._handle_coordinator_update()
-class EvoChild(EvoDevice):
+class EvoChild(EvoEntity):
"""Base for any evohome-compatible child entity (DHW, zone).
This includes (1 to 12) heating zones and (optionally) a DHW controller.
"""
- _evo_id: str # mypy hint
+ _evo_device: evo.HotWater | evo.Zone
+ _evo_id: str
def __init__(
- self, evo_broker: EvoBroker, evo_device: evo.HotWater | evo.Zone
+ self, coordinator: EvoDataUpdateCoordinator, evo_device: evo.HotWater | evo.Zone
) -> None:
"""Initialize an evohome-compatible child entity (DHW, zone)."""
- super().__init__(evo_broker, evo_device)
+ super().__init__(coordinator, evo_device)
self._evo_tcs = evo_device.tcs
- self._schedule: dict[str, Any] = {}
+ self._schedule: list[DayOfWeekDhwT] | None = None
self._setpoints: dict[str, Any] = {}
@property
@@ -111,101 +112,81 @@ class EvoChild(EvoDevice):
assert isinstance(self._evo_device, evo.HotWater | evo.Zone) # mypy check
- if (temp := self._evo_broker.temps.get(self._evo_id)) is not None:
+ if (temp := self.coordinator.temps.get(self._evo_id)) is not None:
# use high-precision temps if available
return temp
return self._evo_device.temperature
@property
- def setpoints(self) -> dict[str, Any]:
+ def setpoints(self) -> Mapping[str, Any]:
"""Return the current/next setpoints from the schedule.
Only Zones & DHW controllers (but not the TCS) can have schedules.
"""
- def _dt_evo_to_aware(dt_naive: datetime, utc_offset: timedelta) -> datetime:
- dt_aware = dt_naive.replace(tzinfo=dt_util.UTC) - utc_offset
- return dt_util.as_local(dt_aware)
+ if not self._schedule:
+ return self._setpoints
- if not (schedule := self._schedule.get("DailySchedules")):
- return {} # no scheduled setpoints when {'DailySchedules': []}
+ this_sp_dtm, this_sp_val = self._evo_device.this_switchpoint
+ next_sp_dtm, next_sp_val = self._evo_device.next_switchpoint
- # get dt in the same TZ as the TCS location, so we can compare schedule times
- day_time = dt_util.now().astimezone(timezone(self._evo_broker.loc_utc_offset))
- day_of_week = day_time.weekday() # for evohome, 0 is Monday
- time_of_day = day_time.strftime("%H:%M:%S")
+ key = "temp" if isinstance(self._evo_device, evo.Zone) else "state"
- try:
- # Iterate today's switchpoints until past the current time of day...
- day = schedule[day_of_week]
- sp_idx = -1 # last switchpoint of the day before
- for i, tmp in enumerate(day["Switchpoints"]):
- if time_of_day > tmp["TimeOfDay"]:
- sp_idx = i # current setpoint
- else:
- break
-
- # Did this setpoint start yesterday? Does the next setpoint start tomorrow?
- this_sp_day = -1 if sp_idx == -1 else 0
- next_sp_day = 1 if sp_idx + 1 == len(day["Switchpoints"]) else 0
-
- for key, offset, idx in (
- ("this", this_sp_day, sp_idx),
- ("next", next_sp_day, (sp_idx + 1) * (1 - next_sp_day)),
- ):
- sp_date = (day_time + timedelta(days=offset)).strftime("%Y-%m-%d")
- day = schedule[(day_of_week + offset) % 7]
- switchpoint = day["Switchpoints"][idx]
-
- switchpoint_time_of_day = dt_util.parse_datetime(
- f"{sp_date}T{switchpoint['TimeOfDay']}"
- )
- assert switchpoint_time_of_day is not None # mypy check
- dt_aware = _dt_evo_to_aware(
- switchpoint_time_of_day, self._evo_broker.loc_utc_offset
- )
-
- self._setpoints[f"{key}_sp_from"] = dt_aware.isoformat()
- try:
- self._setpoints[f"{key}_sp_temp"] = switchpoint[SZ_HEAT_SETPOINT]
- except KeyError:
- self._setpoints[f"{key}_sp_state"] = switchpoint["DhwState"]
-
- except IndexError:
- self._setpoints = {}
- _LOGGER.warning(
- "Failed to get setpoints, report as an issue if this error persists",
- exc_info=True,
- )
+ self._setpoints = {
+ "this_sp_from": this_sp_dtm,
+ f"this_sp_{key}": this_sp_val,
+ "next_sp_from": next_sp_dtm,
+ f"next_sp_{key}": next_sp_val,
+ }
return self._setpoints
- async def _update_schedule(self) -> None:
+ async def _update_schedule(self, force_refresh: bool = False) -> None:
"""Get the latest schedule, if any."""
- assert isinstance(self._evo_device, evo.HotWater | evo.Zone) # mypy check
+ async def get_schedule() -> None:
+ try:
+ schedule = await self.coordinator.call_client_api(
+ self._evo_device.get_schedule(), # type: ignore[arg-type]
+ request_refresh=False,
+ )
+ except evo.InvalidScheduleError as err:
+ _LOGGER.warning(
+ "%s: Unable to retrieve a valid schedule: %s",
+ self._evo_device,
+ err,
+ )
+ self._schedule = []
+ return
+ else:
+ self._schedule = schedule # type: ignore[assignment]
- try:
- schedule = await self._evo_broker.call_client_api(
- self._evo_device.get_schedule(), update_state=False
+ _LOGGER.debug("Schedule['%s'] = %s", self.name, schedule)
+
+ if (
+ force_refresh
+ or self._schedule is None
+ or (
+ (until := self._setpoints.get("next_sp_from")) is not None
+ and until < datetime.now(UTC)
)
- except evo.InvalidSchedule as err:
- _LOGGER.warning(
- "%s: Unable to retrieve a valid schedule: %s",
- self._evo_device,
- err,
- )
- self._schedule = {}
- else:
- self._schedule = schedule or {}
+ ): # must use self._setpoints, not self.setpoints
+ await get_schedule()
- _LOGGER.debug("Schedule['%s'] = %s", self.name, self._schedule)
+ _ = self.setpoints # update the setpoints attr
- async def async_update(self) -> None:
- """Get the latest state data."""
- next_sp_from = self._setpoints.get("next_sp_from", "2000-01-01T00:00:00+00:00")
- next_sp_from_dt = dt_util.parse_datetime(next_sp_from)
- if next_sp_from_dt is None or dt_util.now() >= next_sp_from_dt:
- await self._update_schedule() # no schedule, or it's out-of-date
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
- self._device_state_attrs = {"setpoints": self.setpoints}
+ self._device_state_attrs = {
+ "activeFaults": self._evo_device.active_faults,
+ "setpoints": self._setpoints,
+ }
+
+ super()._handle_coordinator_update()
+
+ async def update_attrs(self) -> None:
+ """Update the entity's extra state attrs."""
+ await self._update_schedule()
+ self._handle_coordinator_update()
diff --git a/homeassistant/components/evohome/helpers.py b/homeassistant/components/evohome/helpers.py
deleted file mode 100644
index 0e2de36eb47..00000000000
--- a/homeassistant/components/evohome/helpers.py
+++ /dev/null
@@ -1,110 +0,0 @@
-"""Support for (EMEA/EU-based) Honeywell TCC systems."""
-
-from __future__ import annotations
-
-from datetime import datetime, timedelta
-from http import HTTPStatus
-import logging
-import re
-from typing import Any
-
-import evohomeasync2 as evo
-
-from homeassistant.const import CONF_SCAN_INTERVAL
-from homeassistant.util import dt as dt_util
-
-_LOGGER = logging.getLogger(__name__)
-
-
-def dt_local_to_aware(dt_naive: datetime) -> datetime:
- """Convert a local/naive datetime to TZ-aware."""
- dt_aware = dt_util.now() + (dt_naive - datetime.now())
- if dt_aware.microsecond >= 500000:
- dt_aware += timedelta(seconds=1)
- return dt_aware.replace(microsecond=0)
-
-
-def dt_aware_to_naive(dt_aware: datetime) -> datetime:
- """Convert a TZ-aware datetime to naive/local."""
- dt_naive = datetime.now() + (dt_aware - dt_util.now())
- if dt_naive.microsecond >= 500000:
- dt_naive += timedelta(seconds=1)
- return dt_naive.replace(microsecond=0)
-
-
-def convert_until(status_dict: dict, until_key: str) -> None:
- """Reformat a dt str from "%Y-%m-%dT%H:%M:%SZ" as local/aware/isoformat."""
- if until_key in status_dict and ( # only present for certain modes
- dt_utc_naive := dt_util.parse_datetime(status_dict[until_key])
- ):
- status_dict[until_key] = dt_util.as_local(dt_utc_naive).isoformat()
-
-
-def convert_dict(dictionary: dict[str, Any]) -> dict[str, Any]:
- """Recursively convert a dict's keys to snake_case."""
-
- def convert_key(key: str) -> str:
- """Convert a string to snake_case."""
- string = re.sub(r"[\-\.\s]", "_", str(key))
- return (
- (string[0]).lower()
- + re.sub(
- r"[A-Z]",
- lambda matched: f"_{matched.group(0).lower()}", # type:ignore[str-bytes-safe]
- string[1:],
- )
- )
-
- return {
- (convert_key(k) if isinstance(k, str) else k): (
- convert_dict(v) if isinstance(v, dict) else v
- )
- for k, v in dictionary.items()
- }
-
-
-def handle_evo_exception(err: evo.RequestFailed) -> None:
- """Return False if the exception can't be ignored."""
-
- try:
- raise err
-
- except evo.AuthenticationFailed:
- _LOGGER.error(
- (
- "Failed to authenticate with the vendor's server. Check your username"
- " and password. NB: Some special password characters that work"
- " correctly via the website will not work via the web API. Message"
- " is: %s"
- ),
- err,
- )
-
- except evo.RequestFailed:
- if err.status is None:
- _LOGGER.warning(
- (
- "Unable to connect with the vendor's server. "
- "Check your network and the vendor's service status page. "
- "Message is: %s"
- ),
- err,
- )
-
- elif err.status == HTTPStatus.SERVICE_UNAVAILABLE:
- _LOGGER.warning(
- "The vendor says their server is currently unavailable. "
- "Check the vendor's service status page"
- )
-
- elif err.status == HTTPStatus.TOO_MANY_REQUESTS:
- _LOGGER.warning(
- (
- "The vendor's API rate limit has been exceeded. "
- "If this message persists, consider increasing the %s"
- ),
- CONF_SCAN_INTERVAL,
- )
-
- else:
- raise # we don't expect/handle any other Exceptions
diff --git a/homeassistant/components/evohome/manifest.json b/homeassistant/components/evohome/manifest.json
index 22edadad7f4..44e4cdb1128 100644
--- a/homeassistant/components/evohome/manifest.json
+++ b/homeassistant/components/evohome/manifest.json
@@ -4,7 +4,7 @@
"codeowners": ["@zxdavb"],
"documentation": "https://www.home-assistant.io/integrations/evohome",
"iot_class": "cloud_polling",
- "loggers": ["evohomeasync", "evohomeasync2"],
+ "loggers": ["evohome", "evohomeasync", "evohomeasync2"],
"quality_scale": "legacy",
- "requirements": ["evohome-async==0.4.21"]
+ "requirements": ["evohome-async==1.0.4"]
}
diff --git a/homeassistant/components/evohome/storage.py b/homeassistant/components/evohome/storage.py
new file mode 100644
index 00000000000..b078c33b305
--- /dev/null
+++ b/homeassistant/components/evohome/storage.py
@@ -0,0 +1,118 @@
+"""Support for (EMEA/EU-based) Honeywell TCC systems."""
+
+from __future__ import annotations
+
+from datetime import UTC, datetime, timedelta
+from typing import Any, NotRequired, TypedDict
+
+from evohomeasync.auth import (
+ SZ_SESSION_ID,
+ SZ_SESSION_ID_EXPIRES,
+ AbstractSessionManager,
+)
+from evohomeasync2.auth import AbstractTokenManager
+
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.storage import Store
+
+from .const import STORAGE_KEY, STORAGE_VER
+
+
+class _SessionIdEntryT(TypedDict):
+ session_id: str
+ session_id_expires: NotRequired[str] # dt.isoformat() # TZ-aware
+
+
+class _TokenStoreT(TypedDict):
+ username: str
+ refresh_token: str
+ access_token: str
+ access_token_expires: str # dt.isoformat() # TZ-aware
+ session_id: NotRequired[str]
+ session_id_expires: NotRequired[str] # dt.isoformat() # TZ-aware
+
+
+class TokenManager(AbstractTokenManager, AbstractSessionManager):
+ """A token manager that uses a cache file to store the tokens."""
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ *args: Any,
+ **kwargs: Any,
+ ) -> None:
+ """Initialise the token manager."""
+ super().__init__(*args, **kwargs)
+
+ self._store = Store(hass, STORAGE_VER, STORAGE_KEY) # type: ignore[var-annotated]
+ self._store_initialized = False # True once cache loaded first time
+
+ async def get_access_token(self) -> str:
+ """Return a valid access token.
+
+ If the cached entry is not valid, will fetch a new access token.
+ """
+
+ if not self._store_initialized:
+ await self._load_cache_from_store()
+
+ return await super().get_access_token()
+
+ async def get_session_id(self) -> str:
+ """Return a valid session id.
+
+ If the cached entry is not valid, will fetch a new session id.
+ """
+
+ if not self._store_initialized:
+ await self._load_cache_from_store()
+
+ return await super().get_session_id()
+
+ async def _load_cache_from_store(self) -> None:
+ """Load the user entry from the cache.
+
+ Assumes single reader/writer. Reads only once, at initialization.
+ """
+
+ cache: _TokenStoreT = await self._store.async_load() or {} # type: ignore[assignment]
+ self._store_initialized = True
+
+ if not cache or cache["username"] != self._client_id:
+ return
+
+ if SZ_SESSION_ID in cache:
+ self._import_session_id(cache) # type: ignore[arg-type]
+ self._import_access_token(cache)
+
+ def _import_session_id(self, session: _SessionIdEntryT) -> None: # type: ignore[override]
+ """Extract the session id from a (serialized) dictionary."""
+ # base class method overridden because session_id_expired is NotRequired here
+
+ self._session_id = session[SZ_SESSION_ID]
+
+ session_id_expires = session.get(SZ_SESSION_ID_EXPIRES)
+ if session_id_expires is None:
+ self._session_id_expires = datetime.now(tz=UTC) + timedelta(minutes=15)
+ else:
+ self._session_id_expires = datetime.fromisoformat(session_id_expires)
+
+ async def save_access_token(self) -> None: # an abstractmethod
+ """Save the access token (and expiry dtm, refresh token) to the cache."""
+ await self.save_cache_to_store()
+
+ async def save_session_id(self) -> None: # an abstractmethod
+ """Save the session id (and expiry dtm) to the cache."""
+ await self.save_cache_to_store()
+
+ async def save_cache_to_store(self) -> None:
+ """Save the access token (and session id, if any) to the cache.
+
+ Assumes a single reader/writer. Writes whenever new data has been fetched.
+ """
+
+ cache = {"username": self._client_id} | self._export_access_token()
+ if self._session_id:
+ cache |= self._export_session_id()
+
+ await self._store.async_save(cache)
diff --git a/homeassistant/components/evohome/strings.json b/homeassistant/components/evohome/strings.json
index ca032643c9d..4fc51c30b97 100644
--- a/homeassistant/components/evohome/strings.json
+++ b/homeassistant/components/evohome/strings.json
@@ -10,17 +10,17 @@
},
"period": {
"name": "Period",
- "description": "A period of time in days; used only with Away, DayOff, or Custom. The system will revert to Auto at midnight (up to 99 days, today is day 1)."
+ "description": "A period of time in days; used only with Away, DayOff, or Custom mode. The system will revert to Auto mode at midnight (up to 99 days, today is day 1)."
},
"duration": {
"name": "Duration",
- "description": "The duration in hours; used only with AutoWithEco (up to 24 hours)."
+ "description": "The duration in hours; used only with AutoWithEco mode (up to 24 hours)."
}
}
},
"reset_system": {
"name": "Reset system",
- "description": "Sets the system to Auto mode and reset all the zones to follow their schedules. Not all Evohome systems support this feature (i.e. AutoWithReset mode)."
+ "description": "Sets the system to Auto mode and resets all the zones to follow their schedules. Not all Evohome systems support this feature (i.e. AutoWithReset mode)."
},
"refresh_system": {
"name": "Refresh system",
diff --git a/homeassistant/components/evohome/water_heater.py b/homeassistant/components/evohome/water_heater.py
index 2c3cf9de12d..7ea0fb3a2d9 100644
--- a/homeassistant/components/evohome/water_heater.py
+++ b/homeassistant/components/evohome/water_heater.py
@@ -3,17 +3,11 @@
from __future__ import annotations
import logging
-from typing import TYPE_CHECKING, Any
+from typing import Any
import evohomeasync2 as evo
-from evohomeasync2.schema.const import (
- SZ_ACTIVE_FAULTS,
- SZ_DHW_ID,
- SZ_OFF,
- SZ_ON,
- SZ_STATE_STATUS,
- SZ_TEMPERATURE_STATUS,
-)
+from evohomeasync2.const import SZ_STATE_STATUS, SZ_TEMPERATURE_STATUS
+from evohomeasync2.schemas.const import DhwState as EvoDhwState, ZoneMode as EvoZoneMode
from homeassistant.components.water_heater import (
WaterHeaterEntity,
@@ -31,22 +25,17 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
-from .const import DOMAIN, EVO_FOLLOW, EVO_PERMOVER
+from . import EVOHOME_KEY
+from .coordinator import EvoDataUpdateCoordinator
from .entity import EvoChild
-if TYPE_CHECKING:
- from . import EvoBroker
-
-
_LOGGER = logging.getLogger(__name__)
STATE_AUTO = "auto"
-HA_STATE_TO_EVO = {STATE_AUTO: "", STATE_ON: SZ_ON, STATE_OFF: SZ_OFF}
+HA_STATE_TO_EVO = {STATE_AUTO: "", STATE_ON: EvoDhwState.ON, STATE_OFF: EvoDhwState.OFF}
EVO_STATE_TO_HA = {v: k for k, v in HA_STATE_TO_EVO.items() if k != ""}
-STATE_ATTRS_DHW = [SZ_DHW_ID, SZ_ACTIVE_FAULTS, SZ_STATE_STATUS, SZ_TEMPERATURE_STATUS]
-
async def async_setup_platform(
hass: HomeAssistant,
@@ -58,19 +47,22 @@ async def async_setup_platform(
if discovery_info is None:
return
- broker: EvoBroker = hass.data[DOMAIN]["broker"]
+ coordinator = hass.data[EVOHOME_KEY].coordinator
+ tcs = hass.data[EVOHOME_KEY].tcs
- assert broker.tcs.hotwater is not None # mypy check
+ assert tcs.hotwater is not None # mypy check
_LOGGER.debug(
"Adding: DhwController (%s), id=%s",
- broker.tcs.hotwater.TYPE,
- broker.tcs.hotwater.dhwId,
+ tcs.hotwater.type,
+ tcs.hotwater.id,
)
- new_entity = EvoDHW(broker, broker.tcs.hotwater)
+ entity = EvoDHW(coordinator, tcs.hotwater)
- async_add_entities([new_entity], update_before_add=True)
+ async_add_entities([entity])
+
+ await entity.update_attrs()
class EvoDHW(EvoChild, WaterHeaterEntity):
@@ -81,19 +73,23 @@ class EvoDHW(EvoChild, WaterHeaterEntity):
_attr_operation_list = list(HA_STATE_TO_EVO)
_attr_temperature_unit = UnitOfTemperature.CELSIUS
- _evo_device: evo.HotWater # mypy hint
+ _evo_device: evo.HotWater
+ _evo_id_attr = "dhw_id"
+ _evo_state_attr_names = (SZ_STATE_STATUS, SZ_TEMPERATURE_STATUS)
- def __init__(self, evo_broker: EvoBroker, evo_device: evo.HotWater) -> None:
+ def __init__(
+ self, coordinator: EvoDataUpdateCoordinator, evo_device: evo.HotWater
+ ) -> None:
"""Initialize an evohome-compatible DHW controller."""
- super().__init__(evo_broker, evo_device)
- self._evo_id = evo_device.dhwId
+ super().__init__(coordinator, evo_device)
+ self._evo_id = evo_device.id
- self._attr_unique_id = evo_device.dhwId
+ self._attr_unique_id = evo_device.id
self._attr_name = evo_device.name # is static
self._attr_precision = (
- PRECISION_TENTHS if evo_broker.client_v1 else PRECISION_WHOLE
+ PRECISION_TENTHS if coordinator.client_v1 else PRECISION_WHOLE
)
self._attr_supported_features = (
WaterHeaterEntityFeature.AWAY_MODE | WaterHeaterEntityFeature.OPERATION_MODE
@@ -102,19 +98,15 @@ class EvoDHW(EvoChild, WaterHeaterEntity):
@property
def current_operation(self) -> str | None:
"""Return the current operating mode (Auto, On, or Off)."""
- if self._evo_device.mode == EVO_FOLLOW:
+ if self._evo_device.mode == EvoZoneMode.FOLLOW_SCHEDULE:
return STATE_AUTO
- if (device_state := self._evo_device.state) is None:
- return None
- return EVO_STATE_TO_HA[device_state]
+ return EVO_STATE_TO_HA[self._evo_device.state]
@property
def is_away_mode_on(self) -> bool | None:
"""Return True if away mode is on."""
- if self._evo_device.state is None:
- return None
is_off = EVO_STATE_TO_HA[self._evo_device.state] == STATE_OFF
- is_permanent = self._evo_device.mode == EVO_PERMOVER
+ is_permanent = self._evo_device.mode == EvoZoneMode.PERMANENT_OVERRIDE
return is_off and is_permanent
async def async_set_operation_mode(self, operation_mode: str) -> None:
@@ -123,40 +115,31 @@ class EvoDHW(EvoChild, WaterHeaterEntity):
Except for Auto, the mode is only until the next SetPoint.
"""
if operation_mode == STATE_AUTO:
- await self._evo_broker.call_client_api(self._evo_device.reset_mode())
+ await self.coordinator.call_client_api(self._evo_device.reset())
else:
await self._update_schedule()
- until = dt_util.parse_datetime(self.setpoints.get("next_sp_from", ""))
+ until = self.setpoints.get("next_sp_from")
until = dt_util.as_utc(until) if until else None
if operation_mode == STATE_ON:
- await self._evo_broker.call_client_api(
- self._evo_device.set_on(until=until)
- )
+ await self.coordinator.call_client_api(self._evo_device.on(until=until))
else: # STATE_OFF
- await self._evo_broker.call_client_api(
- self._evo_device.set_off(until=until)
+ await self.coordinator.call_client_api(
+ self._evo_device.off(until=until)
)
async def async_turn_away_mode_on(self) -> None:
"""Turn away mode on."""
- await self._evo_broker.call_client_api(self._evo_device.set_off())
+ await self.coordinator.call_client_api(self._evo_device.off())
async def async_turn_away_mode_off(self) -> None:
"""Turn away mode off."""
- await self._evo_broker.call_client_api(self._evo_device.reset_mode())
+ await self.coordinator.call_client_api(self._evo_device.reset())
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn on."""
- await self._evo_broker.call_client_api(self._evo_device.set_on())
+ await self.coordinator.call_client_api(self._evo_device.on())
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn off."""
- await self._evo_broker.call_client_api(self._evo_device.set_off())
-
- async def async_update(self) -> None:
- """Get the latest state data for a DHW controller."""
- await super().async_update()
-
- for attr in STATE_ATTRS_DHW:
- self._device_state_attrs[attr] = getattr(self._evo_device, attr)
+ await self.coordinator.call_client_api(self._evo_device.off())
diff --git a/homeassistant/components/ezviz/alarm_control_panel.py b/homeassistant/components/ezviz/alarm_control_panel.py
index 66a76df2cdc..08fa0a68ee8 100644
--- a/homeassistant/components/ezviz/alarm_control_panel.py
+++ b/homeassistant/components/ezviz/alarm_control_panel.py
@@ -18,7 +18,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, MANUFACTURER
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
@@ -50,7 +50,7 @@ ALARM_TYPE = EzvizAlarmControlPanelEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ezviz alarm control panel."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/binary_sensor.py b/homeassistant/components/ezviz/binary_sensor.py
index 6f0d87c8218..5e069e0277a 100644
--- a/homeassistant/components/ezviz/binary_sensor.py
+++ b/homeassistant/components/ezviz/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
from .entity import EzvizEntity
@@ -34,7 +34,7 @@ BINARY_SENSOR_TYPES: dict[str, BinarySensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/button.py b/homeassistant/components/ezviz/button.py
index b99674b0693..6dbb419c903 100644
--- a/homeassistant/components/ezviz/button.py
+++ b/homeassistant/components/ezviz/button.py
@@ -13,7 +13,7 @@ from pyezviz.exceptions import HTTPError, PyEzvizError
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
from .entity import EzvizEntity
@@ -68,7 +68,7 @@ BUTTON_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ button based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/camera.py b/homeassistant/components/ezviz/camera.py
index d96fc949c86..e3d01bef83e 100644
--- a/homeassistant/components/ezviz/camera.py
+++ b/homeassistant/components/ezviz/camera.py
@@ -15,7 +15,7 @@ from homeassistant.const import CONF_IP_ADDRESS, CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import discovery_flow
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
@@ -36,7 +36,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ cameras based on a config entry."""
@@ -141,11 +141,6 @@ class EzvizCamera(EzvizEntity, Camera):
if camera_password:
self._attr_supported_features = CameraEntityFeature.STREAM
- @property
- def available(self) -> bool:
- """Return True if entity is available."""
- return self.data["status"] != 2
-
@property
def is_on(self) -> bool:
"""Return true if on."""
diff --git a/homeassistant/components/ezviz/entity.py b/homeassistant/components/ezviz/entity.py
index 44de4a0c9c7..54614e4899a 100644
--- a/homeassistant/components/ezviz/entity.py
+++ b/homeassistant/components/ezviz/entity.py
@@ -42,6 +42,11 @@ class EzvizEntity(CoordinatorEntity[EzvizDataUpdateCoordinator], Entity):
"""Return coordinator data for this entity."""
return self.coordinator.data[self._serial]
+ @property
+ def available(self) -> bool:
+ """Return True if entity is available."""
+ return self.data["status"] != 2
+
class EzvizBaseEntity(Entity):
"""Generic entity for EZVIZ individual poll entities."""
@@ -72,3 +77,8 @@ class EzvizBaseEntity(Entity):
def data(self) -> dict[str, Any]:
"""Return coordinator data for this entity."""
return self.coordinator.data[self._serial]
+
+ @property
+ def available(self) -> bool:
+ """Return True if entity is available."""
+ return self.data["status"] != 2
diff --git a/homeassistant/components/ezviz/image.py b/homeassistant/components/ezviz/image.py
index d4c7a267b1e..28ebc7279e6 100644
--- a/homeassistant/components/ezviz/image.py
+++ b/homeassistant/components/ezviz/image.py
@@ -4,6 +4,7 @@ from __future__ import annotations
import logging
+from propcache.api import cached_property
from pyezviz.exceptions import PyEzvizError
from pyezviz.utils import decrypt_image
@@ -11,7 +12,7 @@ from homeassistant.components.image import Image, ImageEntity, ImageEntityDescri
from homeassistant.config_entries import SOURCE_IGNORE
from homeassistant.const import CONF_PASSWORD
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN
@@ -29,7 +30,7 @@ IMAGE_TYPE = ImageEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ image entities based on a config entry."""
@@ -62,6 +63,11 @@ class EzvizLastMotion(EzvizEntity, ImageEntity):
else None
)
+ @cached_property
+ def available(self) -> bool:
+ """Entity gets data from ezviz API so always available."""
+ return True
+
async def _async_load_image_from_url(self, url: str) -> Image | None:
"""Load an image by url."""
if response := await self._fetch_url(url):
diff --git a/homeassistant/components/ezviz/light.py b/homeassistant/components/ezviz/light.py
index 145c8b1ca20..ba398dd3ed4 100644
--- a/homeassistant/components/ezviz/light.py
+++ b/homeassistant/components/ezviz/light.py
@@ -10,7 +10,7 @@ from pyezviz.exceptions import HTTPError, PyEzvizError
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -26,7 +26,7 @@ BRIGHTNESS_RANGE = (1, 255)
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ lights based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/number.py b/homeassistant/components/ezviz/number.py
index 9e8a20f36dd..9bdd1feb81d 100644
--- a/homeassistant/components/ezviz/number.py
+++ b/homeassistant/components/ezviz/number.py
@@ -19,7 +19,7 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
from .entity import EzvizBaseEntity
@@ -51,7 +51,7 @@ NUMBER_TYPE = EzvizNumberEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/select.py b/homeassistant/components/ezviz/select.py
index 8e037fe6c33..486564bff6e 100644
--- a/homeassistant/components/ezviz/select.py
+++ b/homeassistant/components/ezviz/select.py
@@ -11,7 +11,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
from .entity import EzvizEntity
@@ -38,7 +38,7 @@ SELECT_TYPE = EzvizSelectEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ select entities based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/sensor.py b/homeassistant/components/ezviz/sensor.py
index f3d50836bc7..c441b34b42d 100644
--- a/homeassistant/components/ezviz/sensor.py
+++ b/homeassistant/components/ezviz/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
from .entity import EzvizEntity
@@ -72,7 +72,7 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/siren.py b/homeassistant/components/ezviz/siren.py
index 5a612aa0772..a2c88f58972 100644
--- a/homeassistant/components/ezviz/siren.py
+++ b/homeassistant/components/ezviz/siren.py
@@ -17,7 +17,7 @@ from homeassistant.const import STATE_ON
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import event as evt
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
@@ -35,7 +35,7 @@ SIREN_ENTITY_TYPE = SirenEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/strings.json b/homeassistant/components/ezviz/strings.json
index f1653661cdd..cd8bbc9d199 100644
--- a/homeassistant/components/ezviz/strings.json
+++ b/homeassistant/components/ezviz/strings.json
@@ -54,7 +54,7 @@
"init": {
"data": {
"timeout": "Request timeout (seconds)",
- "ffmpeg_arguments": "Arguments passed to ffmpeg for cameras"
+ "ffmpeg_arguments": "Arguments passed to FFmpeg for cameras"
}
}
}
diff --git a/homeassistant/components/ezviz/switch.py b/homeassistant/components/ezviz/switch.py
index 1a347c931a6..01f7cac1a55 100644
--- a/homeassistant/components/ezviz/switch.py
+++ b/homeassistant/components/ezviz/switch.py
@@ -15,7 +15,7 @@ from homeassistant.components.switch import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
from .entity import EzvizEntity
@@ -107,7 +107,7 @@ SWITCH_TYPES: dict[int, EzvizSwitchEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ switch based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ezviz/update.py b/homeassistant/components/ezviz/update.py
index 3027e048688..c9f8038b336 100644
--- a/homeassistant/components/ezviz/update.py
+++ b/homeassistant/components/ezviz/update.py
@@ -14,7 +14,7 @@ from homeassistant.components.update import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import EzvizConfigEntry, EzvizDataUpdateCoordinator
from .entity import EzvizEntity
@@ -30,7 +30,7 @@ UPDATE_ENTITY_TYPES = UpdateEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: EzvizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up EZVIZ sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/faa_delays/binary_sensor.py b/homeassistant/components/faa_delays/binary_sensor.py
index 0fbc028f111..6822e2620fd 100644
--- a/homeassistant/components/faa_delays/binary_sensor.py
+++ b/homeassistant/components/faa_delays/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import FAAConfigEntry, FAADataUpdateCoordinator
@@ -83,7 +83,9 @@ FAA_BINARY_SENSORS: tuple[FaaDelaysBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: FAAConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: FAAConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a FAA sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fastdotcom/sensor.py b/homeassistant/components/fastdotcom/sensor.py
index b633cb25628..5b6429b0754 100644
--- a/homeassistant/components/fastdotcom/sensor.py
+++ b/homeassistant/components/fastdotcom/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfDataRate
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -20,7 +20,7 @@ from .coordinator import FastdotcomConfigEntry, FastdotcomDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: FastdotcomConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fast.com sensor."""
async_add_entities([SpeedtestSensor(entry.entry_id, entry.runtime_data)])
diff --git a/homeassistant/components/feedreader/event.py b/homeassistant/components/feedreader/event.py
index ad6aed0fc76..578b5b1e175 100644
--- a/homeassistant/components/feedreader/event.py
+++ b/homeassistant/components/feedreader/event.py
@@ -10,7 +10,7 @@ from feedparser import FeedParserDict
from homeassistant.components.event import EventEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import FeedReaderConfigEntry
@@ -28,7 +28,7 @@ ATTR_TITLE = "title"
async def async_setup_entry(
hass: HomeAssistant,
entry: FeedReaderConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up event entities for feedreader."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/feedreader/strings.json b/homeassistant/components/feedreader/strings.json
index 0f0492eb6c9..35022e82bb1 100644
--- a/homeassistant/components/feedreader/strings.json
+++ b/homeassistant/components/feedreader/strings.json
@@ -36,7 +36,7 @@
"issues": {
"import_yaml_error_url_error": {
"title": "The Feedreader YAML configuration import failed",
- "description": "Configuring the Feedreader using YAML is being removed but there was a connection error when trying to import the YAML configuration for `{url}`.\n\nPlease verify that url is reachable and accessable for Home Assistant and restart Home Assistant to try again or remove the Feedreader YAML configuration from your configuration.yaml file and continue to set up the integration manually."
+ "description": "Configuring the Feedreader using YAML is being removed but there was a connection error when trying to import the YAML configuration for `{url}`.\n\nPlease verify that the URL is reachable and accessible for Home Assistant and restart Home Assistant to try again or remove the Feedreader YAML configuration from your configuration.yaml file and continue to set up the integration manually."
}
}
}
diff --git a/homeassistant/components/ffmpeg/strings.json b/homeassistant/components/ffmpeg/strings.json
index 66c1f19de5b..cac7fcfc48c 100644
--- a/homeassistant/components/ffmpeg/strings.json
+++ b/homeassistant/components/ffmpeg/strings.json
@@ -2,7 +2,7 @@
"services": {
"restart": {
"name": "[%key:common::action::restart%]",
- "description": "Sends a restart command to a ffmpeg based sensor.",
+ "description": "Sends a restart command to an FFmpeg-based sensor.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -12,7 +12,7 @@
},
"start": {
"name": "[%key:common::action::start%]",
- "description": "Sends a start command to a ffmpeg based sensor.",
+ "description": "Sends a start command to an FFmpeg-based sensor.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -22,7 +22,7 @@
},
"stop": {
"name": "[%key:common::action::stop%]",
- "description": "Sends a stop command to a ffmpeg based sensor.",
+ "description": "Sends a stop command to an FFmpeg-based sensor.",
"fields": {
"entity_id": {
"name": "Entity",
diff --git a/homeassistant/components/fibaro/__init__.py b/homeassistant/components/fibaro/__init__.py
index 8ede0169482..33b2598a636 100644
--- a/homeassistant/components/fibaro/__init__.py
+++ b/homeassistant/components/fibaro/__init__.py
@@ -7,21 +7,21 @@ from collections.abc import Callable, Mapping
import logging
from typing import Any
-from pyfibaro.fibaro_client import FibaroClient
+from pyfibaro.fibaro_client import (
+ FibaroAuthenticationFailed,
+ FibaroClient,
+ FibaroConnectFailed,
+)
+from pyfibaro.fibaro_data_helper import read_rooms
from pyfibaro.fibaro_device import DeviceModel
-from pyfibaro.fibaro_room import RoomModel
+from pyfibaro.fibaro_info import InfoModel
from pyfibaro.fibaro_scene import SceneModel
from pyfibaro.fibaro_state_resolver import FibaroEvent, FibaroStateResolver
-from requests.exceptions import HTTPError
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_URL, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import (
- ConfigEntryAuthFailed,
- ConfigEntryNotReady,
- HomeAssistantError,
-)
+from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceEntry, DeviceInfo
from homeassistant.util import slugify
@@ -74,63 +74,31 @@ FIBARO_TYPEMAP = {
class FibaroController:
"""Initiate Fibaro Controller Class."""
- def __init__(self, config: Mapping[str, Any]) -> None:
+ def __init__(
+ self, fibaro_client: FibaroClient, info: InfoModel, import_plugins: bool
+ ) -> None:
"""Initialize the Fibaro controller."""
-
- # The FibaroClient uses the correct API version automatically
- self._client = FibaroClient(config[CONF_URL])
- self._client.set_authentication(config[CONF_USERNAME], config[CONF_PASSWORD])
+ self._client = fibaro_client
+ self._fibaro_info = info
# Whether to import devices from plugins
- self._import_plugins = config[CONF_IMPORT_PLUGINS]
- self._room_map: dict[int, RoomModel] # Mapping roomId to room object
+ self._import_plugins = import_plugins
+ # Mapping roomId to room object
+ self._room_map = read_rooms(fibaro_client)
self._device_map: dict[int, DeviceModel] # Mapping deviceId to device object
self.fibaro_devices: dict[Platform, list[DeviceModel]] = defaultdict(
list
) # List of devices by entity platform
# All scenes
- self._scenes: list[SceneModel] = []
+ self._scenes = self._client.read_scenes()
self._callbacks: dict[int, list[Any]] = {} # Update value callbacks by deviceId
# Event callbacks by device id
self._event_callbacks: dict[int, list[Callable[[FibaroEvent], None]]] = {}
- self.hub_serial: str # Unique serial number of the hub
- self.hub_name: str # The friendly name of the hub
- self.hub_model: str
- self.hub_software_version: str
- self.hub_api_url: str = config[CONF_URL]
+ # Unique serial number of the hub
+ self.hub_serial = info.serial_number
# Device infos by fibaro device id
self._device_infos: dict[int, DeviceInfo] = {}
-
- def connect(self) -> None:
- """Start the communication with the Fibaro controller."""
-
- # Return value doesn't need to be checked,
- # it is only relevant when connecting without credentials
- self._client.connect()
- info = self._client.read_info()
- self.hub_serial = info.serial_number
- self.hub_name = info.hc_name
- self.hub_model = info.platform
- self.hub_software_version = info.current_version
-
- self._room_map = {room.fibaro_id: room for room in self._client.read_rooms()}
self._read_devices()
- self._scenes = self._client.read_scenes()
-
- def connect_with_error_handling(self) -> None:
- """Translate connect errors to easily differentiate auth and connect failures.
-
- When there is a better error handling in the used library this can be improved.
- """
- try:
- self.connect()
- except HTTPError as http_ex:
- if http_ex.response.status_code == 403:
- raise FibaroAuthFailed from http_ex
-
- raise FibaroConnectFailed from http_ex
- except Exception as ex:
- raise FibaroConnectFailed from ex
def enable_state_handler(self) -> None:
"""Start StateHandler thread for monitoring updates."""
@@ -302,14 +270,20 @@ class FibaroController:
def get_room_name(self, room_id: int) -> str | None:
"""Get the room name by room id."""
- assert self._room_map
- room = self._room_map.get(room_id)
- return room.name if room else None
+ return self._room_map.get(room_id)
def read_scenes(self) -> list[SceneModel]:
"""Return list of scenes."""
return self._scenes
+ def read_fibaro_info(self) -> InfoModel:
+ """Return the general info about the hub."""
+ return self._fibaro_info
+
+ def get_frontend_url(self) -> str:
+ """Return the url to the Fibaro hub web UI."""
+ return self._client.frontend_url()
+
def _read_devices(self) -> None:
"""Read and process the device list."""
devices = self._client.read_devices()
@@ -319,20 +293,17 @@ class FibaroController:
for device in devices:
try:
device.fibaro_controller = self
- if device.room_id == 0:
+ room_name = self.get_room_name(device.room_id)
+ if not room_name:
room_name = "Unknown"
- else:
- room_name = self._room_map[device.room_id].name
device.room_name = room_name
device.friendly_name = f"{room_name} {device.name}"
device.ha_id = (
f"{slugify(room_name)}_{slugify(device.name)}_{device.fibaro_id}"
)
if device.enabled and (not device.is_plugin or self._import_plugins):
- device.mapped_platform = self._map_device_to_platform(device)
- else:
- device.mapped_platform = None
- if (platform := device.mapped_platform) is None:
+ platform = self._map_device_to_platform(device)
+ if platform is None:
continue
device.unique_id_str = f"{slugify(self.hub_serial)}.{device.fibaro_id}"
self._create_device_info(device, devices)
@@ -375,11 +346,17 @@ class FibaroController:
pass
+def connect_fibaro_client(data: Mapping[str, Any]) -> tuple[InfoModel, FibaroClient]:
+ """Connect to the fibaro hub and read some basic data."""
+ client = FibaroClient(data[CONF_URL])
+ info = client.connect_with_credentials(data[CONF_USERNAME], data[CONF_PASSWORD])
+ return (info, client)
+
+
def init_controller(data: Mapping[str, Any]) -> FibaroController:
- """Validate the user input allows us to connect to fibaro."""
- controller = FibaroController(data)
- controller.connect_with_error_handling()
- return controller
+ """Connect to the fibaro hub and init the controller."""
+ info, client = connect_fibaro_client(data)
+ return FibaroController(client, info, data[CONF_IMPORT_PLUGINS])
async def async_setup_entry(hass: HomeAssistant, entry: FibaroConfigEntry) -> bool:
@@ -393,22 +370,24 @@ async def async_setup_entry(hass: HomeAssistant, entry: FibaroConfigEntry) -> bo
raise ConfigEntryNotReady(
f"Could not connect to controller at {entry.data[CONF_URL]}"
) from connect_ex
- except FibaroAuthFailed as auth_ex:
+ except FibaroAuthenticationFailed as auth_ex:
raise ConfigEntryAuthFailed from auth_ex
entry.runtime_data = controller
# register the hub device info separately as the hub has sometimes no entities
+ fibaro_info = controller.read_fibaro_info()
device_registry = dr.async_get(hass)
device_registry.async_get_or_create(
config_entry_id=entry.entry_id,
identifiers={(DOMAIN, controller.hub_serial)},
serial_number=controller.hub_serial,
- manufacturer="Fibaro",
- name=controller.hub_name,
- model=controller.hub_model,
- sw_version=controller.hub_software_version,
- configuration_url=controller.hub_api_url.removesuffix("/api/"),
+ manufacturer=fibaro_info.manufacturer_name,
+ name=fibaro_info.hc_name,
+ model=fibaro_info.model_name,
+ sw_version=fibaro_info.current_version,
+ configuration_url=controller.get_frontend_url(),
+ connections={(dr.CONNECTION_NETWORK_MAC, fibaro_info.mac_address)},
)
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
@@ -443,11 +422,3 @@ async def async_remove_config_entry_device(
return False
return True
-
-
-class FibaroConnectFailed(HomeAssistantError):
- """Error to indicate we cannot connect to fibaro home center."""
-
-
-class FibaroAuthFailed(HomeAssistantError):
- """Error to indicate that authentication failed on fibaro home center."""
diff --git a/homeassistant/components/fibaro/binary_sensor.py b/homeassistant/components/fibaro/binary_sensor.py
index 16e79c0c1d0..14c8f03f3ec 100644
--- a/homeassistant/components/fibaro/binary_sensor.py
+++ b/homeassistant/components/fibaro/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FibaroConfigEntry
from .entity import FibaroEntity
@@ -42,7 +42,7 @@ SENSOR_TYPES = {
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Fibaro controller devices."""
controller = entry.runtime_data
diff --git a/homeassistant/components/fibaro/climate.py b/homeassistant/components/fibaro/climate.py
index 45f700026a0..7a8cc3fd2a9 100644
--- a/homeassistant/components/fibaro/climate.py
+++ b/homeassistant/components/fibaro/climate.py
@@ -19,7 +19,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FibaroConfigEntry
from .entity import FibaroEntity
@@ -110,7 +110,7 @@ OP_MODE_ACTIONS = ("setMode", "setOperatingMode", "setThermostatMode")
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Fibaro controller devices."""
controller = entry.runtime_data
@@ -129,13 +129,13 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
def __init__(self, fibaro_device: DeviceModel) -> None:
"""Initialize the Fibaro device."""
super().__init__(fibaro_device)
- self._temp_sensor_device: FibaroEntity | None = None
- self._target_temp_device: FibaroEntity | None = None
- self._op_mode_device: FibaroEntity | None = None
- self._fan_mode_device: FibaroEntity | None = None
+ self._temp_sensor_device: DeviceModel | None = None
+ self._target_temp_device: DeviceModel | None = None
+ self._op_mode_device: DeviceModel | None = None
+ self._fan_mode_device: DeviceModel | None = None
self.entity_id = ENTITY_ID_FORMAT.format(self.ha_id)
- siblings = fibaro_device.fibaro_controller.get_siblings(fibaro_device)
+ siblings = self.controller.get_siblings(fibaro_device)
_LOGGER.debug("%s siblings: %s", fibaro_device.ha_id, siblings)
tempunit = "C"
for device in siblings:
@@ -147,23 +147,23 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
and (device.value.has_value or device.has_heating_thermostat_setpoint)
and device.unit in ("C", "F")
):
- self._temp_sensor_device = FibaroEntity(device)
+ self._temp_sensor_device = device
tempunit = device.unit
if any(
action for action in TARGET_TEMP_ACTIONS if action in device.actions
):
- self._target_temp_device = FibaroEntity(device)
+ self._target_temp_device = device
self._attr_supported_features |= ClimateEntityFeature.TARGET_TEMPERATURE
if device.has_unit:
tempunit = device.unit
if any(action for action in OP_MODE_ACTIONS if action in device.actions):
- self._op_mode_device = FibaroEntity(device)
+ self._op_mode_device = device
self._attr_supported_features |= ClimateEntityFeature.PRESET_MODE
if "setFanMode" in device.actions:
- self._fan_mode_device = FibaroEntity(device)
+ self._fan_mode_device = device
self._attr_supported_features |= ClimateEntityFeature.FAN_MODE
if tempunit == "F":
@@ -172,7 +172,7 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
self._attr_temperature_unit = UnitOfTemperature.CELSIUS
if self._fan_mode_device:
- fan_modes = self._fan_mode_device.fibaro_device.supported_modes
+ fan_modes = self._fan_mode_device.supported_modes
self._attr_fan_modes = []
for mode in fan_modes:
if mode not in FANMODES:
@@ -184,7 +184,7 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
if self._op_mode_device:
self._attr_preset_modes = []
self._attr_hvac_modes: list[HVACMode] = []
- device = self._op_mode_device.fibaro_device
+ device = self._op_mode_device
if device.has_supported_thermostat_modes:
for mode in device.supported_thermostat_modes:
try:
@@ -222,15 +222,15 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
"- _fan_mode_device %s"
),
self.ha_id,
- self._temp_sensor_device.ha_id if self._temp_sensor_device else "None",
- self._target_temp_device.ha_id if self._target_temp_device else "None",
- self._op_mode_device.ha_id if self._op_mode_device else "None",
- self._fan_mode_device.ha_id if self._fan_mode_device else "None",
+ self._temp_sensor_device.fibaro_id if self._temp_sensor_device else "None",
+ self._target_temp_device.fibaro_id if self._target_temp_device else "None",
+ self._op_mode_device.fibaro_id if self._op_mode_device else "None",
+ self._fan_mode_device.fibaro_id if self._fan_mode_device else "None",
)
await super().async_added_to_hass()
# Register update callback for child devices
- siblings = self.fibaro_device.fibaro_controller.get_siblings(self.fibaro_device)
+ siblings = self.controller.get_siblings(self.fibaro_device)
for device in siblings:
if device != self.fibaro_device:
self.controller.register(device.fibaro_id, self._update_callback)
@@ -240,14 +240,14 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
"""Return the fan setting."""
if not self._fan_mode_device:
return None
- mode = self._fan_mode_device.fibaro_device.mode
+ mode = self._fan_mode_device.mode
return FANMODES[mode]
def set_fan_mode(self, fan_mode: str) -> None:
"""Set new target fan mode."""
if not self._fan_mode_device:
return
- self._fan_mode_device.action("setFanMode", HA_FANMODES[fan_mode])
+ self._fan_mode_device.execute_action("setFanMode", [HA_FANMODES[fan_mode]])
@property
def fibaro_op_mode(self) -> str | int:
@@ -255,7 +255,7 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
if not self._op_mode_device:
return HA_OPMODES_HVAC[HVACMode.AUTO]
- device = self._op_mode_device.fibaro_device
+ device = self._op_mode_device
if device.has_operating_mode:
return device.operating_mode
@@ -281,17 +281,17 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
if not self._op_mode_device:
return
- if "setOperatingMode" in self._op_mode_device.fibaro_device.actions:
- self._op_mode_device.action("setOperatingMode", HA_OPMODES_HVAC[hvac_mode])
- elif "setThermostatMode" in self._op_mode_device.fibaro_device.actions:
- device = self._op_mode_device.fibaro_device
+ device = self._op_mode_device
+ if "setOperatingMode" in device.actions:
+ device.execute_action("setOperatingMode", [HA_OPMODES_HVAC[hvac_mode]])
+ elif "setThermostatMode" in device.actions:
if device.has_supported_thermostat_modes:
for mode in device.supported_thermostat_modes:
if mode.lower() == hvac_mode:
- self._op_mode_device.action("setThermostatMode", mode)
+ device.execute_action("setThermostatMode", [mode])
break
- elif "setMode" in self._op_mode_device.fibaro_device.actions:
- self._op_mode_device.action("setMode", HA_OPMODES_HVAC[hvac_mode])
+ elif "setMode" in device.actions:
+ device.execute_action("setMode", [HA_OPMODES_HVAC[hvac_mode]])
@property
def hvac_action(self) -> HVACAction | None:
@@ -299,7 +299,7 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
if not self._op_mode_device:
return None
- device = self._op_mode_device.fibaro_device
+ device = self._op_mode_device
if device.has_thermostat_operating_state:
with suppress(ValueError):
return HVACAction(device.thermostat_operating_state.lower())
@@ -315,15 +315,15 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
if not self._op_mode_device:
return None
- if self._op_mode_device.fibaro_device.has_thermostat_mode:
- mode = self._op_mode_device.fibaro_device.thermostat_mode
+ if self._op_mode_device.has_thermostat_mode:
+ mode = self._op_mode_device.thermostat_mode
if self.preset_modes is not None and mode in self.preset_modes:
return mode
return None
- if self._op_mode_device.fibaro_device.has_operating_mode:
- mode = self._op_mode_device.fibaro_device.operating_mode
+ if self._op_mode_device.has_operating_mode:
+ mode = self._op_mode_device.operating_mode
else:
- mode = self._op_mode_device.fibaro_device.mode
+ mode = self._op_mode_device.mode
if mode not in OPMODES_PRESET:
return None
@@ -334,20 +334,22 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
if self._op_mode_device is None:
return
- if "setThermostatMode" in self._op_mode_device.fibaro_device.actions:
- self._op_mode_device.action("setThermostatMode", preset_mode)
- elif "setOperatingMode" in self._op_mode_device.fibaro_device.actions:
- self._op_mode_device.action(
- "setOperatingMode", HA_OPMODES_PRESET[preset_mode]
+ if "setThermostatMode" in self._op_mode_device.actions:
+ self._op_mode_device.execute_action("setThermostatMode", [preset_mode])
+ elif "setOperatingMode" in self._op_mode_device.actions:
+ self._op_mode_device.execute_action(
+ "setOperatingMode", [HA_OPMODES_PRESET[preset_mode]]
+ )
+ elif "setMode" in self._op_mode_device.actions:
+ self._op_mode_device.execute_action(
+ "setMode", [HA_OPMODES_PRESET[preset_mode]]
)
- elif "setMode" in self._op_mode_device.fibaro_device.actions:
- self._op_mode_device.action("setMode", HA_OPMODES_PRESET[preset_mode])
@property
def current_temperature(self) -> float | None:
"""Return the current temperature."""
if self._temp_sensor_device:
- device = self._temp_sensor_device.fibaro_device
+ device = self._temp_sensor_device
if device.has_heating_thermostat_setpoint:
return device.heating_thermostat_setpoint
return device.value.float_value()
@@ -357,7 +359,7 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
def target_temperature(self) -> float | None:
"""Return the temperature we try to reach."""
if self._target_temp_device:
- device = self._target_temp_device.fibaro_device
+ device = self._target_temp_device
if device.has_heating_thermostat_setpoint_future:
return device.heating_thermostat_setpoint_future
return device.target_level
@@ -368,9 +370,11 @@ class FibaroThermostat(FibaroEntity, ClimateEntity):
temperature = kwargs.get(ATTR_TEMPERATURE)
target = self._target_temp_device
if target is not None and temperature is not None:
- if "setThermostatSetpoint" in target.fibaro_device.actions:
- target.action("setThermostatSetpoint", self.fibaro_op_mode, temperature)
- elif "setHeatingThermostatSetpoint" in target.fibaro_device.actions:
- target.action("setHeatingThermostatSetpoint", temperature)
+ if "setThermostatSetpoint" in target.actions:
+ target.execute_action(
+ "setThermostatSetpoint", [self.fibaro_op_mode, temperature]
+ )
+ elif "setHeatingThermostatSetpoint" in target.actions:
+ target.execute_action("setHeatingThermostatSetpoint", [temperature])
else:
- target.action("setTargetLevel", temperature)
+ target.execute_action("setTargetLevel", [temperature])
diff --git a/homeassistant/components/fibaro/config_flow.py b/homeassistant/components/fibaro/config_flow.py
index 0ffd9aaa48f..d941ceab37f 100644
--- a/homeassistant/components/fibaro/config_flow.py
+++ b/homeassistant/components/fibaro/config_flow.py
@@ -6,6 +6,7 @@ from collections.abc import Mapping
import logging
from typing import Any
+from pyfibaro.fibaro_client import FibaroAuthenticationFailed, FibaroConnectFailed
from slugify import slugify
import voluptuous as vol
@@ -13,7 +14,7 @@ from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_NAME, CONF_PASSWORD, CONF_URL, CONF_USERNAME
from homeassistant.core import HomeAssistant
-from . import FibaroAuthFailed, FibaroConnectFailed, init_controller
+from . import connect_fibaro_client
from .const import CONF_IMPORT_PLUGINS, DOMAIN
_LOGGER = logging.getLogger(__name__)
@@ -33,16 +34,16 @@ async def _validate_input(hass: HomeAssistant, data: dict[str, Any]) -> dict[str
Data has the keys from STEP_USER_DATA_SCHEMA with values provided by the user.
"""
- controller = await hass.async_add_executor_job(init_controller, data)
+ info, _ = await hass.async_add_executor_job(connect_fibaro_client, data)
_LOGGER.debug(
"Successfully connected to fibaro home center %s with name %s",
- controller.hub_serial,
- controller.hub_name,
+ info.serial_number,
+ info.hc_name,
)
return {
- "serial_number": slugify(controller.hub_serial),
- "name": controller.hub_name,
+ "serial_number": slugify(info.serial_number),
+ "name": info.hc_name,
}
@@ -75,7 +76,7 @@ class FibaroConfigFlow(ConfigFlow, domain=DOMAIN):
info = await _validate_input(self.hass, user_input)
except FibaroConnectFailed:
errors["base"] = "cannot_connect"
- except FibaroAuthFailed:
+ except FibaroAuthenticationFailed:
errors["base"] = "invalid_auth"
else:
await self.async_set_unique_id(info["serial_number"])
@@ -106,7 +107,7 @@ class FibaroConfigFlow(ConfigFlow, domain=DOMAIN):
await _validate_input(self.hass, new_data)
except FibaroConnectFailed:
errors["base"] = "cannot_connect"
- except FibaroAuthFailed:
+ except FibaroAuthenticationFailed:
errors["base"] = "invalid_auth"
else:
return self.async_update_reload_and_abort(
diff --git a/homeassistant/components/fibaro/cover.py b/homeassistant/components/fibaro/cover.py
index bfebbf87bd2..0008b56345e 100644
--- a/homeassistant/components/fibaro/cover.py
+++ b/homeassistant/components/fibaro/cover.py
@@ -15,7 +15,7 @@ from homeassistant.components.cover import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FibaroConfigEntry
from .entity import FibaroEntity
@@ -24,7 +24,7 @@ from .entity import FibaroEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fibaro covers."""
controller = entry.runtime_data
diff --git a/homeassistant/components/fibaro/entity.py b/homeassistant/components/fibaro/entity.py
index 6a8e12136c8..5375b058315 100644
--- a/homeassistant/components/fibaro/entity.py
+++ b/homeassistant/components/fibaro/entity.py
@@ -11,6 +11,8 @@ from pyfibaro.fibaro_device import DeviceModel
from homeassistant.const import ATTR_ARMED, ATTR_BATTERY_LEVEL
from homeassistant.helpers.entity import Entity
+from . import FibaroController
+
_LOGGER = logging.getLogger(__name__)
@@ -22,7 +24,7 @@ class FibaroEntity(Entity):
def __init__(self, fibaro_device: DeviceModel) -> None:
"""Initialize the device."""
self.fibaro_device = fibaro_device
- self.controller = fibaro_device.fibaro_controller
+ self.controller: FibaroController = fibaro_device.fibaro_controller
self.ha_id = fibaro_device.ha_id
self._attr_name = fibaro_device.friendly_name
self._attr_unique_id = fibaro_device.unique_id_str
@@ -54,15 +56,6 @@ class FibaroEntity(Entity):
return self.fibaro_device.value_2.int_value()
return None
- def dont_know_message(self, cmd: str) -> None:
- """Make a warning in case we don't know how to perform an action."""
- _LOGGER.warning(
- "Not sure how to %s: %s (available actions: %s)",
- cmd,
- str(self.ha_id),
- str(self.fibaro_device.actions),
- )
-
def set_level(self, level: int) -> None:
"""Set the level of Fibaro device."""
self.action("setValue", level)
@@ -97,11 +90,7 @@ class FibaroEntity(Entity):
def action(self, cmd: str, *args: Any) -> None:
"""Perform an action on the Fibaro HC."""
- if cmd in self.fibaro_device.actions:
- self.fibaro_device.execute_action(cmd, args)
- _LOGGER.debug("-> %s.%s%s called", str(self.ha_id), str(cmd), str(args))
- else:
- self.dont_know_message(cmd)
+ self.fibaro_device.execute_action(cmd, args)
@property
def current_binary_state(self) -> bool:
diff --git a/homeassistant/components/fibaro/event.py b/homeassistant/components/fibaro/event.py
index a2d5da7f877..0beea2e336e 100644
--- a/homeassistant/components/fibaro/event.py
+++ b/homeassistant/components/fibaro/event.py
@@ -12,7 +12,7 @@ from homeassistant.components.event import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FibaroConfigEntry
from .entity import FibaroEntity
@@ -21,7 +21,7 @@ from .entity import FibaroEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fibaro event entities."""
controller = entry.runtime_data
diff --git a/homeassistant/components/fibaro/light.py b/homeassistant/components/fibaro/light.py
index d40e26244f3..446b9b9f7ff 100644
--- a/homeassistant/components/fibaro/light.py
+++ b/homeassistant/components/fibaro/light.py
@@ -19,7 +19,7 @@ from homeassistant.components.light import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FibaroConfigEntry
from .entity import FibaroEntity
@@ -51,7 +51,7 @@ def scaleto99(value: int | None) -> int:
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Fibaro controller devices."""
controller = entry.runtime_data
diff --git a/homeassistant/components/fibaro/lock.py b/homeassistant/components/fibaro/lock.py
index 62a9dfa43b1..a1e76109e2d 100644
--- a/homeassistant/components/fibaro/lock.py
+++ b/homeassistant/components/fibaro/lock.py
@@ -9,7 +9,7 @@ from pyfibaro.fibaro_device import DeviceModel
from homeassistant.components.lock import ENTITY_ID_FORMAT, LockEntity
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FibaroConfigEntry
from .entity import FibaroEntity
@@ -18,7 +18,7 @@ from .entity import FibaroEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fibaro locks."""
controller = entry.runtime_data
diff --git a/homeassistant/components/fibaro/manifest.json b/homeassistant/components/fibaro/manifest.json
index d2a1186b05b..cd4d1de838c 100644
--- a/homeassistant/components/fibaro/manifest.json
+++ b/homeassistant/components/fibaro/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "hub",
"iot_class": "local_push",
"loggers": ["pyfibaro"],
- "requirements": ["pyfibaro==0.8.0"]
+ "requirements": ["pyfibaro==0.8.2"]
}
diff --git a/homeassistant/components/fibaro/scene.py b/homeassistant/components/fibaro/scene.py
index a4c0f1bd7f1..8a594506f27 100644
--- a/homeassistant/components/fibaro/scene.py
+++ b/homeassistant/components/fibaro/scene.py
@@ -9,7 +9,7 @@ from pyfibaro.fibaro_scene import SceneModel
from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import slugify
from . import FibaroConfigEntry, FibaroController
@@ -19,7 +19,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Fibaro scenes."""
controller = entry.runtime_data
diff --git a/homeassistant/components/fibaro/sensor.py b/homeassistant/components/fibaro/sensor.py
index 245a0d087d8..9034bd7d05e 100644
--- a/homeassistant/components/fibaro/sensor.py
+++ b/homeassistant/components/fibaro/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import convert
from . import FibaroConfigEntry
@@ -102,7 +102,7 @@ FIBARO_TO_HASS_UNIT: dict[str, str] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fibaro controller devices."""
diff --git a/homeassistant/components/fibaro/switch.py b/homeassistant/components/fibaro/switch.py
index f67683dff6a..8d77685c1e7 100644
--- a/homeassistant/components/fibaro/switch.py
+++ b/homeassistant/components/fibaro/switch.py
@@ -9,7 +9,7 @@ from pyfibaro.fibaro_device import DeviceModel
from homeassistant.components.switch import ENTITY_ID_FORMAT, SwitchEntity
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FibaroConfigEntry
from .entity import FibaroEntity
@@ -18,7 +18,7 @@ from .entity import FibaroEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FibaroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fibaro switches."""
controller = entry.runtime_data
diff --git a/homeassistant/components/file/notify.py b/homeassistant/components/file/notify.py
index 3d61dbb04e0..90af1677bce 100644
--- a/homeassistant/components/file/notify.py
+++ b/homeassistant/components/file/notify.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_FILE_PATH, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import CONF_TIMESTAMP, DEFAULT_NAME, DOMAIN, FILE_ICON
@@ -23,7 +23,7 @@ from .const import CONF_TIMESTAMP, DEFAULT_NAME, DOMAIN, FILE_ICON
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up notify entity."""
unique_id = entry.entry_id
diff --git a/homeassistant/components/file/sensor.py b/homeassistant/components/file/sensor.py
index 879c06e29f3..b8d174afe2c 100644
--- a/homeassistant/components/file/sensor.py
+++ b/homeassistant/components/file/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
CONF_VALUE_TEMPLATE,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.template import Template
from .const import DEFAULT_NAME, FILE_ICON
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the file sensor."""
config = dict(entry.data)
diff --git a/homeassistant/components/filesize/__init__.py b/homeassistant/components/filesize/__init__.py
index 602eac1f24d..b10125de67c 100644
--- a/homeassistant/components/filesize/__init__.py
+++ b/homeassistant/components/filesize/__init__.py
@@ -2,19 +2,15 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_FILE_PATH
from homeassistant.core import HomeAssistant
from .const import PLATFORMS
-from .coordinator import FileSizeCoordinator
-
-type FileSizeConfigEntry = ConfigEntry[FileSizeCoordinator]
+from .coordinator import FileSizeConfigEntry, FileSizeCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: FileSizeConfigEntry) -> bool:
"""Set up from a config entry."""
- coordinator = FileSizeCoordinator(hass, entry.data[CONF_FILE_PATH])
+ coordinator = FileSizeCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
@@ -22,6 +18,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: FileSizeConfigEntry) ->
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: FileSizeConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/filesize/coordinator.py b/homeassistant/components/filesize/coordinator.py
index 0c2a0277434..87f59f1a53e 100644
--- a/homeassistant/components/filesize/coordinator.py
+++ b/homeassistant/components/filesize/coordinator.py
@@ -7,6 +7,8 @@ import logging
import os
import pathlib
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_FILE_PATH
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt as dt_util
@@ -15,22 +17,26 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
+type FileSizeConfigEntry = ConfigEntry[FileSizeCoordinator]
+
class FileSizeCoordinator(DataUpdateCoordinator[dict[str, int | float | datetime]]):
"""Filesize coordinator."""
+ config_entry: FileSizeConfigEntry
path: pathlib.Path
- def __init__(self, hass: HomeAssistant, unresolved_path: str) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: FileSizeConfigEntry) -> None:
"""Initialize filesize coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=60),
always_update=False,
)
- self._unresolved_path = unresolved_path
+ self._unresolved_path = self.config_entry.data[CONF_FILE_PATH]
def _get_full_path(self) -> pathlib.Path:
"""Check if path is valid, allowed and return full path."""
diff --git a/homeassistant/components/filesize/sensor.py b/homeassistant/components/filesize/sensor.py
index 2eb170af99d..966e253660d 100644
--- a/homeassistant/components/filesize/sensor.py
+++ b/homeassistant/components/filesize/sensor.py
@@ -14,12 +14,11 @@ from homeassistant.components.sensor import (
from homeassistant.const import EntityCategory, UnitOfInformation
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import FileSizeConfigEntry
from .const import DOMAIN
-from .coordinator import FileSizeCoordinator
+from .coordinator import FileSizeConfigEntry, FileSizeCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -60,7 +59,7 @@ SENSOR_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FileSizeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from config entry."""
async_add_entities(
diff --git a/homeassistant/components/filter/sensor.py b/homeassistant/components/filter/sensor.py
index 330e61f499e..eb1337002e4 100644
--- a/homeassistant/components/filter/sensor.py
+++ b/homeassistant/components/filter/sensor.py
@@ -44,7 +44,10 @@ from homeassistant.core import (
callback,
)
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.reload import async_setup_reload_service
from homeassistant.helpers.start import async_at_started
@@ -201,7 +204,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Filter sensor entry."""
name: str = entry.options[CONF_NAME]
diff --git a/homeassistant/components/fireservicerota/__init__.py b/homeassistant/components/fireservicerota/__init__.py
index bf5385b6f2a..0f30a29cfba 100644
--- a/homeassistant/components/fireservicerota/__init__.py
+++ b/homeassistant/components/fireservicerota/__init__.py
@@ -4,50 +4,43 @@ from __future__ import annotations
from datetime import timedelta
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .const import DATA_CLIENT, DATA_COORDINATOR, DOMAIN
-from .coordinator import FireServiceRotaClient, FireServiceUpdateCoordinator
+from .coordinator import (
+ FireServiceConfigEntry,
+ FireServiceRotaClient,
+ FireServiceUpdateCoordinator,
+)
MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=60)
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR, Platform.SWITCH]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: FireServiceConfigEntry) -> bool:
"""Set up FireServiceRota from a config entry."""
- hass.data.setdefault(DOMAIN, {})
-
client = FireServiceRotaClient(hass, entry)
await client.setup()
if client.token_refresh_failure:
return False
+ entry.async_on_unload(client.async_stop_listener)
coordinator = FireServiceUpdateCoordinator(hass, client, entry)
await coordinator.async_config_entry_first_refresh()
- hass.data[DOMAIN][entry.entry_id] = {
- DATA_CLIENT: client,
- DATA_COORDINATOR: coordinator,
- }
+ entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, entry: FireServiceConfigEntry
+) -> bool:
"""Unload FireServiceRota config entry."""
-
- await hass.async_add_executor_job(
- hass.data[DOMAIN][entry.entry_id][DATA_CLIENT].websocket.stop_listener
- )
- unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- if unload_ok:
- del hass.data[DOMAIN][entry.entry_id]
- return unload_ok
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/fireservicerota/binary_sensor.py b/homeassistant/components/fireservicerota/binary_sensor.py
index b6d3aa67a0a..be7add191c0 100644
--- a/homeassistant/components/fireservicerota/binary_sensor.py
+++ b/homeassistant/components/fireservicerota/binary_sensor.py
@@ -7,25 +7,25 @@ from typing import Any
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .const import DATA_CLIENT, DATA_COORDINATOR, DOMAIN as FIRESERVICEROTA_DOMAIN
-from .coordinator import FireServiceRotaClient, FireServiceUpdateCoordinator
+from .coordinator import (
+ FireServiceConfigEntry,
+ FireServiceRotaClient,
+ FireServiceUpdateCoordinator,
+)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: FireServiceConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up FireServiceRota binary sensor based on a config entry."""
- client: FireServiceRotaClient = hass.data[FIRESERVICEROTA_DOMAIN][entry.entry_id][
- DATA_CLIENT
- ]
-
- coordinator: FireServiceUpdateCoordinator = hass.data[FIRESERVICEROTA_DOMAIN][
- entry.entry_id
- ][DATA_COORDINATOR]
+ coordinator = entry.runtime_data
+ client = coordinator.client
async_add_entities([ResponseBinarySensor(coordinator, client, entry)])
diff --git a/homeassistant/components/fireservicerota/coordinator.py b/homeassistant/components/fireservicerota/coordinator.py
index 14a8c40e469..6815bf39104 100644
--- a/homeassistant/components/fireservicerota/coordinator.py
+++ b/homeassistant/components/fireservicerota/coordinator.py
@@ -28,12 +28,19 @@ _LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR, Platform.SWITCH]
+type FireServiceConfigEntry = ConfigEntry[FireServiceUpdateCoordinator]
+
class FireServiceUpdateCoordinator(DataUpdateCoordinator[dict | None]):
"""Data update coordinator for FireServiceRota."""
+ config_entry: FireServiceConfigEntry
+
def __init__(
- self, hass: HomeAssistant, client: FireServiceRotaClient, entry: ConfigEntry
+ self,
+ hass: HomeAssistant,
+ client: FireServiceRotaClient,
+ entry: FireServiceConfigEntry,
) -> None:
"""Initialize the FireServiceRota DataUpdateCoordinator."""
super().__init__(
@@ -213,3 +220,7 @@ class FireServiceRotaClient:
)
await self.update_call(self.fsr.set_incident_response, self.incident_id, value)
+
+ async def async_stop_listener(self) -> None:
+ """Stop listener."""
+ await self._hass.async_add_executor_job(self.websocket.stop_listener)
diff --git a/homeassistant/components/fireservicerota/sensor.py b/homeassistant/components/fireservicerota/sensor.py
index b09d1295025..5ed65609dc8 100644
--- a/homeassistant/components/fireservicerota/sensor.py
+++ b/homeassistant/components/fireservicerota/sensor.py
@@ -4,25 +4,24 @@ import logging
from typing import Any
from homeassistant.components.sensor import SensorEntity
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
-from .const import DATA_CLIENT, DOMAIN as FIRESERVICEROTA_DOMAIN
-from .coordinator import FireServiceRotaClient
+from .const import DOMAIN as FIRESERVICEROTA_DOMAIN
+from .coordinator import FireServiceConfigEntry, FireServiceRotaClient
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: FireServiceConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up FireServiceRota sensor based on a config entry."""
- client = hass.data[FIRESERVICEROTA_DOMAIN][entry.entry_id][DATA_CLIENT]
-
- async_add_entities([IncidentsSensor(client)])
+ async_add_entities([IncidentsSensor(entry.runtime_data.client)])
# pylint: disable-next=hass-invalid-inheritance # needs fixing
diff --git a/homeassistant/components/fireservicerota/switch.py b/homeassistant/components/fireservicerota/switch.py
index affd46c91bd..d9fe382e4b1 100644
--- a/homeassistant/components/fireservicerota/switch.py
+++ b/homeassistant/components/fireservicerota/switch.py
@@ -7,21 +7,26 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DATA_CLIENT, DATA_COORDINATOR, DOMAIN as FIRESERVICEROTA_DOMAIN
-from .coordinator import FireServiceRotaClient, FireServiceUpdateCoordinator
+from .const import DOMAIN as FIRESERVICEROTA_DOMAIN
+from .coordinator import (
+ FireServiceConfigEntry,
+ FireServiceRotaClient,
+ FireServiceUpdateCoordinator,
+)
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: FireServiceConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up FireServiceRota switch based on a config entry."""
- client = hass.data[FIRESERVICEROTA_DOMAIN][entry.entry_id][DATA_CLIENT]
-
- coordinator = hass.data[FIRESERVICEROTA_DOMAIN][entry.entry_id][DATA_COORDINATOR]
+ coordinator = entry.runtime_data
+ client = coordinator.client
async_add_entities([ResponseSwitch(coordinator, client, entry)])
diff --git a/homeassistant/components/firmata/binary_sensor.py b/homeassistant/components/firmata/binary_sensor.py
index 4973afa6960..de79f676ae6 100644
--- a/homeassistant/components/firmata/binary_sensor.py
+++ b/homeassistant/components/firmata/binary_sensor.py
@@ -5,7 +5,7 @@ import logging
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.const import CONF_NAME, CONF_PIN
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FirmataConfigEntry
from .const import CONF_NEGATE_STATE, CONF_PIN_MODE
@@ -18,7 +18,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FirmataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Firmata binary sensors."""
new_entities = []
diff --git a/homeassistant/components/firmata/light.py b/homeassistant/components/firmata/light.py
index 4f27143b774..f866ce9dbe5 100644
--- a/homeassistant/components/firmata/light.py
+++ b/homeassistant/components/firmata/light.py
@@ -9,7 +9,7 @@ from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEnti
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_MAXIMUM, CONF_MINIMUM, CONF_NAME, CONF_PIN
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FirmataConfigEntry
from .board import FirmataPinType
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FirmataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Firmata lights."""
new_entities = []
diff --git a/homeassistant/components/firmata/sensor.py b/homeassistant/components/firmata/sensor.py
index 569d97fe1ec..7b49950e948 100644
--- a/homeassistant/components/firmata/sensor.py
+++ b/homeassistant/components/firmata/sensor.py
@@ -5,7 +5,7 @@ import logging
from homeassistant.components.sensor import SensorEntity
from homeassistant.const import CONF_NAME, CONF_PIN
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FirmataConfigEntry
from .const import CONF_DIFFERENTIAL, CONF_PIN_MODE
@@ -18,7 +18,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FirmataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Firmata sensors."""
new_entities = []
diff --git a/homeassistant/components/firmata/switch.py b/homeassistant/components/firmata/switch.py
index 33953b78974..e84b18e9c74 100644
--- a/homeassistant/components/firmata/switch.py
+++ b/homeassistant/components/firmata/switch.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.const import CONF_NAME, CONF_PIN
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FirmataConfigEntry
from .const import CONF_INITIAL_STATE, CONF_NEGATE_STATE, CONF_PIN_MODE
@@ -19,7 +19,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FirmataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Firmata switches."""
new_entities = []
diff --git a/homeassistant/components/fitbit/__init__.py b/homeassistant/components/fitbit/__init__.py
index 0c4a37198d6..f2378797d8d 100644
--- a/homeassistant/components/fitbit/__init__.py
+++ b/homeassistant/components/fitbit/__init__.py
@@ -1,6 +1,5 @@
"""The fitbit component."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -8,16 +7,13 @@ from homeassistant.helpers import config_entry_oauth2_flow
from . import api
from .const import FitbitScope
-from .coordinator import FitbitData, FitbitDeviceCoordinator
+from .coordinator import FitbitConfigEntry, FitbitData, FitbitDeviceCoordinator
from .exceptions import FitbitApiException, FitbitAuthException
from .model import config_from_entry_data
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type FitbitConfigEntry = ConfigEntry[FitbitData]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: FitbitConfigEntry) -> bool:
"""Set up fitbit from a config entry."""
implementation = (
@@ -39,7 +35,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: FitbitConfigEntry) -> bo
fitbit_config = config_from_entry_data(entry.data)
coordinator: FitbitDeviceCoordinator | None = None
if fitbit_config.is_allowed_resource(FitbitScope.DEVICE, "devices/battery"):
- coordinator = FitbitDeviceCoordinator(hass, fitbit_api)
+ coordinator = FitbitDeviceCoordinator(hass, entry, fitbit_api)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = FitbitData(api=fitbit_api, device_coordinator=coordinator)
diff --git a/homeassistant/components/fitbit/coordinator.py b/homeassistant/components/fitbit/coordinator.py
index 2126129d261..867723419dd 100644
--- a/homeassistant/components/fitbit/coordinator.py
+++ b/homeassistant/components/fitbit/coordinator.py
@@ -6,6 +6,7 @@ import datetime
import logging
from typing import Final
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -19,13 +20,25 @@ _LOGGER = logging.getLogger(__name__)
UPDATE_INTERVAL: Final = datetime.timedelta(minutes=30)
TIMEOUT = 10
+type FitbitConfigEntry = ConfigEntry[FitbitData]
+
class FitbitDeviceCoordinator(DataUpdateCoordinator[dict[str, FitbitDevice]]):
"""Coordinator for fetching fitbit devices from the API."""
- def __init__(self, hass: HomeAssistant, api: FitbitApi) -> None:
+ config_entry: FitbitConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: FitbitConfigEntry, api: FitbitApi
+ ) -> None:
"""Initialize FitbitDeviceCoordinator."""
- super().__init__(hass, _LOGGER, name="Fitbit", update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name="Fitbit",
+ update_interval=UPDATE_INTERVAL,
+ )
self._api = api
async def _async_update_data(self) -> dict[str, FitbitDevice]:
diff --git a/homeassistant/components/fitbit/quality_scale.yaml b/homeassistant/components/fitbit/quality_scale.yaml
index abf127cdb98..04ed2817a07 100644
--- a/homeassistant/components/fitbit/quality_scale.yaml
+++ b/homeassistant/components/fitbit/quality_scale.yaml
@@ -20,11 +20,7 @@ rules:
comment: Fitbit is a polling integration that does use async events.
entity-unique-id: done
has-entity-name: done
- runtime-data:
- status: todo
- comment: |
- The integration uses `hass.data` for data associated with a configuration
- entry and needs to be updated to use `runtime_data`.
+ runtime-data: done
test-before-configure: done
test-before-setup: done
unique-config-entry: done
diff --git a/homeassistant/components/fitbit/sensor.py b/homeassistant/components/fitbit/sensor.py
index 4ccbea97a66..f5c8a81ca26 100644
--- a/homeassistant/components/fitbit/sensor.py
+++ b/homeassistant/components/fitbit/sensor.py
@@ -24,14 +24,13 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import FitbitConfigEntry
from .api import FitbitApi
from .const import ATTRIBUTION, BATTERY_LEVELS, DOMAIN, FitbitScope, FitbitUnitSystem
-from .coordinator import FitbitDeviceCoordinator
+from .coordinator import FitbitConfigEntry, FitbitDeviceCoordinator
from .exceptions import FitbitApiException, FitbitAuthException
from .model import FitbitDevice, config_from_entry_data
@@ -525,7 +524,7 @@ FITBIT_RESOURCE_BATTERY_LEVEL = FitbitSensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: FitbitConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fitbit sensor platform."""
diff --git a/homeassistant/components/fivem/binary_sensor.py b/homeassistant/components/fivem/binary_sensor.py
index 42119939d4a..92aab58349b 100644
--- a/homeassistant/components/fivem/binary_sensor.py
+++ b/homeassistant/components/fivem/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import NAME_STATUS
from .coordinator import FiveMConfigEntry
@@ -34,7 +34,7 @@ BINARY_SENSORS: tuple[FiveMBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FiveMConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FiveM binary sensor platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fivem/sensor.py b/homeassistant/components/fivem/sensor.py
index 88290171756..c4f5856c636 100644
--- a/homeassistant/components/fivem/sensor.py
+++ b/homeassistant/components/fivem/sensor.py
@@ -4,7 +4,7 @@ from dataclasses import dataclass
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -50,7 +50,7 @@ SENSORS: tuple[FiveMSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FiveMConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FiveM sensor platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fjaraskupan/__init__.py b/homeassistant/components/fjaraskupan/__init__.py
index d95cb1d1006..961be04fd8d 100644
--- a/homeassistant/components/fjaraskupan/__init__.py
+++ b/homeassistant/components/fjaraskupan/__init__.py
@@ -3,7 +3,6 @@
from __future__ import annotations
from collections.abc import Callable
-from dataclasses import dataclass
import logging
from fjaraskupan import Device
@@ -16,7 +15,6 @@ from homeassistant.components.bluetooth import (
async_rediscover_address,
async_register_callback,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
@@ -29,7 +27,7 @@ from homeassistant.helpers.entity import Entity
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DISPATCH_DETECTION, DOMAIN
-from .coordinator import FjaraskupanCoordinator
+from .coordinator import FjaraskupanConfigEntry, FjaraskupanCoordinator
PLATFORMS = [
Platform.BINARY_SENSOR,
@@ -42,26 +40,17 @@ PLATFORMS = [
_LOGGER = logging.getLogger(__name__)
-@dataclass
-class EntryState:
- """Store state of config entry."""
-
- coordinators: dict[str, FjaraskupanCoordinator]
-
-
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: FjaraskupanConfigEntry) -> bool:
"""Set up Fjäråskupan from a config entry."""
- state = EntryState({})
- hass.data.setdefault(DOMAIN, {})
- hass.data[DOMAIN][entry.entry_id] = state
+ entry.runtime_data = {}
def detection_callback(
service_info: BluetoothServiceInfoBleak, change: BluetoothChange
) -> None:
if change != BluetoothChange.ADVERTISEMENT:
return
- if data := state.coordinators.get(service_info.address):
+ if data := entry.runtime_data.get(service_info.address):
_LOGGER.debug("Update: %s", service_info)
data.detection_callback(service_info)
else:
@@ -76,11 +65,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
)
coordinator: FjaraskupanCoordinator = FjaraskupanCoordinator(
- hass, device, device_info
+ hass, entry, device, device_info
)
coordinator.detection_callback(service_info)
- state.coordinators[service_info.address] = coordinator
+ entry.runtime_data[service_info.address] = coordinator
async_dispatcher_send(
hass, f"{DISPATCH_DETECTION}.{entry.entry_id}", coordinator
)
@@ -105,16 +94,15 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
@callback
def async_setup_entry_platform(
hass: HomeAssistant,
- entry: ConfigEntry,
+ entry: FjaraskupanConfigEntry,
async_add_entities: AddEntitiesCallback,
constructor: Callable[[FjaraskupanCoordinator], list[Entity]],
) -> None:
"""Set up a platform with added entities."""
- entry_state: EntryState = hass.data[DOMAIN][entry.entry_id]
async_add_entities(
entity
- for coordinator in entry_state.coordinators.values()
+ for coordinator in entry.runtime_data.values()
for entity in constructor(coordinator)
)
@@ -129,12 +117,12 @@ def async_setup_entry_platform(
)
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, entry: FjaraskupanConfigEntry
+) -> bool:
"""Unload a config entry."""
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
if unload_ok:
- hass.data[DOMAIN].pop(entry.entry_id)
-
for device_entry in dr.async_entries_for_config_entry(
dr.async_get(hass), entry.entry_id
):
diff --git a/homeassistant/components/fjaraskupan/binary_sensor.py b/homeassistant/components/fjaraskupan/binary_sensor.py
index 93886a2ac6a..7364fa85b2e 100644
--- a/homeassistant/components/fjaraskupan/binary_sensor.py
+++ b/homeassistant/components/fjaraskupan/binary_sensor.py
@@ -12,15 +12,14 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import async_setup_entry_platform
-from .coordinator import FjaraskupanCoordinator
+from .coordinator import FjaraskupanConfigEntry, FjaraskupanCoordinator
@dataclass(frozen=True)
@@ -48,8 +47,8 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FjaraskupanConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors dynamically through discovery."""
diff --git a/homeassistant/components/fjaraskupan/coordinator.py b/homeassistant/components/fjaraskupan/coordinator.py
index 90b2c617239..7fc4585a722 100644
--- a/homeassistant/components/fjaraskupan/coordinator.py
+++ b/homeassistant/components/fjaraskupan/coordinator.py
@@ -21,6 +21,7 @@ from homeassistant.components.bluetooth import (
async_address_present,
async_ble_device_from_address,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
@@ -28,6 +29,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN
+type FjaraskupanConfigEntry = ConfigEntry[dict[str, FjaraskupanCoordinator]]
+
_LOGGER = logging.getLogger(__name__)
@@ -64,8 +67,14 @@ class UnableToConnect(HomeAssistantError):
class FjaraskupanCoordinator(DataUpdateCoordinator[State]):
"""Update coordinator for each device."""
+ config_entry: FjaraskupanConfigEntry
+
def __init__(
- self, hass: HomeAssistant, device: Device, device_info: DeviceInfo
+ self,
+ hass: HomeAssistant,
+ config_entry: FjaraskupanConfigEntry,
+ device: Device,
+ device_info: DeviceInfo,
) -> None:
"""Initialize the coordinator."""
self.device = device
@@ -73,7 +82,11 @@ class FjaraskupanCoordinator(DataUpdateCoordinator[State]):
self._refresh_was_scheduled = False
super().__init__(
- hass, _LOGGER, name="Fjäråskupan", update_interval=timedelta(seconds=120)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name="Fjäråskupan",
+ update_interval=timedelta(seconds=120),
)
async def _async_refresh(
diff --git a/homeassistant/components/fjaraskupan/fan.py b/homeassistant/components/fjaraskupan/fan.py
index 540a7dd410d..b35bb728131 100644
--- a/homeassistant/components/fjaraskupan/fan.py
+++ b/homeassistant/components/fjaraskupan/fan.py
@@ -13,11 +13,10 @@ from fjaraskupan import (
)
from homeassistant.components.fan import FanEntity, FanEntityFeature
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
@@ -25,7 +24,7 @@ from homeassistant.util.percentage import (
)
from . import async_setup_entry_platform
-from .coordinator import FjaraskupanCoordinator
+from .coordinator import FjaraskupanConfigEntry, FjaraskupanCoordinator
ORDERED_NAMED_FAN_SPEEDS = ["1", "2", "3", "4", "5", "6", "7", "8"]
@@ -51,8 +50,8 @@ class UnsupportedPreset(HomeAssistantError):
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FjaraskupanConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors dynamically through discovery."""
diff --git a/homeassistant/components/fjaraskupan/light.py b/homeassistant/components/fjaraskupan/light.py
index f0083591d4d..c39e3ca4736 100644
--- a/homeassistant/components/fjaraskupan/light.py
+++ b/homeassistant/components/fjaraskupan/light.py
@@ -5,21 +5,20 @@ from __future__ import annotations
from typing import Any
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import async_setup_entry_platform
-from .coordinator import FjaraskupanCoordinator
+from .coordinator import FjaraskupanConfigEntry, FjaraskupanCoordinator
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FjaraskupanConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tuya sensors dynamically through tuya discovery."""
diff --git a/homeassistant/components/fjaraskupan/number.py b/homeassistant/components/fjaraskupan/number.py
index 1828c4cdea5..93fd31273e9 100644
--- a/homeassistant/components/fjaraskupan/number.py
+++ b/homeassistant/components/fjaraskupan/number.py
@@ -3,22 +3,21 @@
from __future__ import annotations
from homeassistant.components.number import NumberEntity
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import async_setup_entry_platform
-from .coordinator import FjaraskupanCoordinator
+from .coordinator import FjaraskupanConfigEntry, FjaraskupanCoordinator
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FjaraskupanConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number entities dynamically through discovery."""
diff --git a/homeassistant/components/fjaraskupan/sensor.py b/homeassistant/components/fjaraskupan/sensor.py
index 36db4d7ed9f..039feb5913c 100644
--- a/homeassistant/components/fjaraskupan/sensor.py
+++ b/homeassistant/components/fjaraskupan/sensor.py
@@ -9,23 +9,22 @@ from homeassistant.components.sensor import (
SensorEntity,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import SIGNAL_STRENGTH_DECIBELS_MILLIWATT, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import async_setup_entry_platform
-from .coordinator import FjaraskupanCoordinator
+from .coordinator import FjaraskupanConfigEntry, FjaraskupanCoordinator
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FjaraskupanConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors dynamically through discovery."""
diff --git a/homeassistant/components/flexit_bacnet/__init__.py b/homeassistant/components/flexit_bacnet/__init__.py
index 6b42310d181..01e0051f53f 100644
--- a/homeassistant/components/flexit_bacnet/__init__.py
+++ b/homeassistant/components/flexit_bacnet/__init__.py
@@ -2,12 +2,10 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_DEVICE_ID, Platform
+from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
-from .coordinator import FlexitCoordinator
+from .coordinator import FlexitConfigEntry, FlexitCoordinator
PLATFORMS: list[Platform] = [
Platform.BINARY_SENSOR,
@@ -18,23 +16,18 @@ PLATFORMS: list[Platform] = [
]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: FlexitConfigEntry) -> bool:
"""Set up Flexit Nordic (BACnet) from a config entry."""
- device_id = entry.data[CONF_DEVICE_ID]
-
- coordinator = FlexitCoordinator(hass, device_id)
+ coordinator = FlexitCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
- hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
+ entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: FlexitConfigEntry) -> bool:
"""Unload the Flexit Nordic (BACnet) config entry."""
- if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
- hass.data[DOMAIN].pop(entry.entry_id)
-
- return unload_ok
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/flexit_bacnet/binary_sensor.py b/homeassistant/components/flexit_bacnet/binary_sensor.py
index 901cc52de47..50c49f45e3e 100644
--- a/homeassistant/components/flexit_bacnet/binary_sensor.py
+++ b/homeassistant/components/flexit_bacnet/binary_sensor.py
@@ -10,12 +10,10 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FlexitCoordinator
-from .const import DOMAIN
+from .coordinator import FlexitConfigEntry, FlexitCoordinator
from .entity import FlexitEntity
@@ -38,17 +36,21 @@ SENSOR_TYPES: tuple[FlexitBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FlexitConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Flexit (bacnet) binary sensor from a config entry."""
- coordinator: FlexitCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+ coordinator = config_entry.runtime_data
async_add_entities(
FlexitBinarySensor(coordinator, description) for description in SENSOR_TYPES
)
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
+
class FlexitBinarySensor(FlexitEntity, BinarySensorEntity):
"""Representation of a Flexit binary Sensor."""
diff --git a/homeassistant/components/flexit_bacnet/climate.py b/homeassistant/components/flexit_bacnet/climate.py
index a2291dea9d6..878b63f938f 100644
--- a/homeassistant/components/flexit_bacnet/climate.py
+++ b/homeassistant/components/flexit_bacnet/climate.py
@@ -19,11 +19,10 @@ from homeassistant.components.climate import (
HVACAction,
HVACMode,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_HALVES, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
@@ -32,19 +31,20 @@ from .const import (
PRESET_TO_VENTILATION_MODE_MAP,
VENTILATION_TO_PRESET_MODE_MAP,
)
-from .coordinator import FlexitCoordinator
+from .coordinator import FlexitConfigEntry, FlexitCoordinator
from .entity import FlexitEntity
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FlexitConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flexit Nordic unit."""
- coordinator: FlexitCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+ async_add_entities([FlexitClimateEntity(config_entry.runtime_data)])
- async_add_entities([FlexitClimateEntity(coordinator)])
+
+PARALLEL_UPDATES = 1
class FlexitClimateEntity(FlexitEntity, ClimateEntity):
@@ -80,10 +80,6 @@ class FlexitClimateEntity(FlexitEntity, ClimateEntity):
super().__init__(coordinator)
self._attr_unique_id = coordinator.device.serial_number
- async def async_update(self) -> None:
- """Refresh unit state."""
- await self.device.update()
-
@property
def hvac_action(self) -> HVACAction | None:
"""Return current HVAC action."""
@@ -115,7 +111,13 @@ class FlexitClimateEntity(FlexitEntity, ClimateEntity):
else:
await self.device.set_air_temp_setpoint_home(temperature)
except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
- raise HomeAssistantError from exc
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_temperature",
+ translation_placeholders={
+ "temperature": str(temperature),
+ },
+ ) from exc
finally:
await self.coordinator.async_refresh()
@@ -134,7 +136,13 @@ class FlexitClimateEntity(FlexitEntity, ClimateEntity):
try:
await self.device.set_ventilation_mode(ventilation_mode)
except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
- raise HomeAssistantError from exc
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_preset_mode",
+ translation_placeholders={
+ "preset": str(ventilation_mode),
+ },
+ ) from exc
finally:
await self.coordinator.async_refresh()
@@ -154,6 +162,12 @@ class FlexitClimateEntity(FlexitEntity, ClimateEntity):
else:
await self.device.set_ventilation_mode(VENTILATION_MODE_HOME)
except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
- raise HomeAssistantError from exc
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_hvac_mode",
+ translation_placeholders={
+ "mode": str(hvac_mode),
+ },
+ ) from exc
finally:
await self.coordinator.async_refresh()
diff --git a/homeassistant/components/flexit_bacnet/coordinator.py b/homeassistant/components/flexit_bacnet/coordinator.py
index 79f3b6a05ad..9148ec87883 100644
--- a/homeassistant/components/flexit_bacnet/coordinator.py
+++ b/homeassistant/components/flexit_bacnet/coordinator.py
@@ -1,5 +1,7 @@
"""DataUpdateCoordinator for Flexit Nordic (BACnet) integration.."""
+from __future__ import annotations
+
import asyncio.exceptions
from datetime import timedelta
import logging
@@ -17,18 +19,21 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
+type FlexitConfigEntry = ConfigEntry[FlexitCoordinator]
+
class FlexitCoordinator(DataUpdateCoordinator[FlexitBACnet]):
"""Class to manage fetching data from a Flexit Nordic (BACnet) device."""
- config_entry: ConfigEntry
+ config_entry: FlexitConfigEntry
- def __init__(self, hass: HomeAssistant, device_id: str) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: FlexitConfigEntry) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
- name=f"{DOMAIN}_{device_id}",
+ config_entry=config_entry,
+ name=f"{DOMAIN}_{config_entry.data[CONF_DEVICE_ID]}",
update_interval=timedelta(seconds=60),
)
@@ -44,7 +49,11 @@ class FlexitCoordinator(DataUpdateCoordinator[FlexitBACnet]):
await self.device.update()
except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
raise ConfigEntryNotReady(
- f"Timeout while connecting to {self.config_entry.data[CONF_IP_ADDRESS]}"
+ translation_domain=DOMAIN,
+ translation_key="not_ready",
+ translation_placeholders={
+ "ip": str(self.config_entry.data[CONF_IP_ADDRESS]),
+ },
) from exc
return self.device
diff --git a/homeassistant/components/flexit_bacnet/icons.json b/homeassistant/components/flexit_bacnet/icons.json
index 7ce8b116a27..d03cffab9ad 100644
--- a/homeassistant/components/flexit_bacnet/icons.json
+++ b/homeassistant/components/flexit_bacnet/icons.json
@@ -30,6 +30,9 @@
},
"home_supply_fan_setpoint": {
"default": "mdi:fan-plus"
+ },
+ "fireplace_mode_runtime": {
+ "default": "mdi:fireplace"
}
},
"switch": {
@@ -38,6 +41,18 @@
"state": {
"off": "mdi:radiator-off"
}
+ },
+ "fireplace_mode": {
+ "default": "mdi:fireplace",
+ "state": {
+ "off": "mdi:fireplace-off"
+ }
+ },
+ "cooker_hood_mode": {
+ "default": "mdi:kettle-steam",
+ "state": {
+ "off": "mdi:kettle"
+ }
}
}
}
diff --git a/homeassistant/components/flexit_bacnet/manifest.json b/homeassistant/components/flexit_bacnet/manifest.json
index 40390162ce6..2e94dd2f4c7 100644
--- a/homeassistant/components/flexit_bacnet/manifest.json
+++ b/homeassistant/components/flexit_bacnet/manifest.json
@@ -6,5 +6,6 @@
"documentation": "https://www.home-assistant.io/integrations/flexit_bacnet",
"integration_type": "device",
"iot_class": "local_polling",
- "requirements": ["flexit_bacnet==2.2.1"]
+ "quality_scale": "silver",
+ "requirements": ["flexit_bacnet==2.2.3"]
}
diff --git a/homeassistant/components/flexit_bacnet/number.py b/homeassistant/components/flexit_bacnet/number.py
index 6e405e8e8ac..b8c329bd1d4 100644
--- a/homeassistant/components/flexit_bacnet/number.py
+++ b/homeassistant/components/flexit_bacnet/number.py
@@ -13,19 +13,21 @@ from homeassistant.components.number import (
NumberEntityDescription,
NumberMode,
)
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import PERCENTAGE
+from homeassistant.const import PERCENTAGE, UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FlexitCoordinator
from .const import DOMAIN
+from .coordinator import FlexitConfigEntry, FlexitCoordinator
from .entity import FlexitEntity
_MAX_FAN_SETPOINT = 100
_MIN_FAN_SETPOINT = 30
+_MAX_RUNTIME_DURATION = 360
+_MIN_RUNTIME_DURATION = 1
+
@dataclass(kw_only=True, frozen=True)
class FlexitNumberEntityDescription(NumberEntityDescription):
@@ -176,22 +178,37 @@ NUMBERS: tuple[FlexitNumberEntityDescription, ...] = (
native_max_value_fn=lambda _: _MAX_FAN_SETPOINT,
native_min_value_fn=lambda device: int(device.fan_setpoint_supply_air_away),
),
+ FlexitNumberEntityDescription(
+ key="fireplace_mode_runtime",
+ translation_key="fireplace_mode_runtime",
+ device_class=NumberDeviceClass.DURATION,
+ native_step=1,
+ mode=NumberMode.SLIDER,
+ native_value_fn=lambda device: device.fireplace_mode_runtime,
+ set_native_value_fn=lambda device: device.set_fireplace_mode_runtime,
+ native_unit_of_measurement=UnitOfTime.MINUTES,
+ native_max_value_fn=lambda _: _MAX_RUNTIME_DURATION,
+ native_min_value_fn=lambda _: _MIN_RUNTIME_DURATION,
+ ),
)
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FlexitConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Flexit (bacnet) number from a config entry."""
- coordinator: FlexitCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+ coordinator = config_entry.runtime_data
async_add_entities(
FlexitNumber(coordinator, description) for description in NUMBERS
)
+PARALLEL_UPDATES = 1
+
+
class FlexitNumber(FlexitEntity, NumberEntity):
"""Representation of a Flexit Number."""
@@ -233,6 +250,12 @@ class FlexitNumber(FlexitEntity, NumberEntity):
try:
await set_native_value_fn(int(value))
except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
- raise HomeAssistantError from exc
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_value_error",
+ translation_placeholders={
+ "value": str(value),
+ },
+ ) from exc
finally:
await self.coordinator.async_refresh()
diff --git a/homeassistant/components/flexit_bacnet/quality_scale.yaml b/homeassistant/components/flexit_bacnet/quality_scale.yaml
new file mode 100644
index 00000000000..7a98eda4eb3
--- /dev/null
+++ b/homeassistant/components/flexit_bacnet/quality_scale.yaml
@@ -0,0 +1,91 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: |
+ Integration does not define custom actions.
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: |
+ This integration does not use any actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: exempt
+ comment: |
+ Entities don't subscribe to events explicitly
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup:
+ status: done
+ comment: |
+ Done implicitly with `await coordinator.async_config_entry_first_refresh()`.
+ unique-config-entry: done
+ # Silver
+ action-exceptions: done
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: |
+ Integration does not use options flow.
+ docs-installation-parameters: done
+ entity-unavailable:
+ status: done
+ comment: |
+ Done implicitly with coordinator.
+ integration-owner: done
+ log-when-unavailable:
+ status: done
+ comment: |
+ Done implicitly with coordinator.
+ parallel-updates: done
+ reauthentication-flow:
+ status: exempt
+ comment: |
+ Integration doesn't require any form of authentication.
+ test-coverage: done
+ # Gold
+ entity-translations: done
+ entity-device-class: done
+ devices: done
+ entity-category: todo
+ entity-disabled-by-default: todo
+ discovery: todo
+ stale-devices:
+ status: exempt
+ comment: |
+ Device type integration.
+ diagnostics: todo
+ exception-translations: done
+ icon-translations: done
+ reconfiguration-flow: todo
+ dynamic-devices:
+ status: exempt
+ comment: |
+ Device type integration.
+ discovery-update-info: todo
+ repair-issues:
+ status: exempt
+ comment: |
+ This is not applicable for this integration.
+ docs-use-cases: todo
+ docs-supported-devices: todo
+ docs-supported-functions: todo
+ docs-data-update: done
+ docs-known-limitations: todo
+ docs-troubleshooting: todo
+ docs-examples: todo
+
+ # Platinum
+ async-dependency: todo
+ inject-websession: todo
+ strict-typing: done
diff --git a/homeassistant/components/flexit_bacnet/sensor.py b/homeassistant/components/flexit_bacnet/sensor.py
index be5f12e480e..0506b13892b 100644
--- a/homeassistant/components/flexit_bacnet/sensor.py
+++ b/homeassistant/components/flexit_bacnet/sensor.py
@@ -11,7 +11,6 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
PERCENTAGE,
REVOLUTIONS_PER_MINUTE,
@@ -20,11 +19,10 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import FlexitCoordinator
-from .const import DOMAIN
+from .coordinator import FlexitConfigEntry, FlexitCoordinator
from .entity import FlexitEntity
@@ -152,17 +150,21 @@ SENSOR_TYPES: tuple[FlexitSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FlexitConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Flexit (bacnet) sensor from a config entry."""
- coordinator: FlexitCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+ coordinator = config_entry.runtime_data
async_add_entities(
FlexitSensor(coordinator, description) for description in SENSOR_TYPES
)
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
+
class FlexitSensor(FlexitEntity, SensorEntity):
"""Representation of a Flexit (bacnet) Sensor."""
diff --git a/homeassistant/components/flexit_bacnet/strings.json b/homeassistant/components/flexit_bacnet/strings.json
index 7f763674d00..6364d59e4e8 100644
--- a/homeassistant/components/flexit_bacnet/strings.json
+++ b/homeassistant/components/flexit_bacnet/strings.json
@@ -5,6 +5,10 @@
"data": {
"ip_address": "[%key:common::config_flow::data::ip%]",
"device_id": "[%key:common::config_flow::data::device%]"
+ },
+ "data_description": {
+ "ip_address": "The IP address of the Flexit Nordic device",
+ "device_id": "The device ID of the Flexit Nordic device"
}
}
},
@@ -52,6 +56,9 @@
},
"home_supply_fan_setpoint": {
"name": "Home supply fan setpoint"
+ },
+ "fireplace_mode_runtime": {
+ "name": "Fireplace mode runtime"
}
},
"sensor": {
@@ -104,7 +111,33 @@
"switch": {
"electric_heater": {
"name": "Electric heater"
+ },
+ "fireplace_mode": {
+ "name": "Fireplace mode"
+ },
+ "cooker_hood_mode": {
+ "name": "Cooker hood mode"
}
}
+ },
+ "exceptions": {
+ "set_value_error": {
+ "message": "Failed setting the value {value}."
+ },
+ "switch_turn": {
+ "message": "Failed to turn the switch {state}."
+ },
+ "set_preset_mode": {
+ "message": "Failed to set preset mode {preset}."
+ },
+ "set_temperature": {
+ "message": "Failed to set temperature {temperature}."
+ },
+ "set_hvac_mode": {
+ "message": "Failed to set HVAC mode {mode}."
+ },
+ "not_ready": {
+ "message": "Timeout while connecting to {ip}."
+ }
}
}
diff --git a/homeassistant/components/flexit_bacnet/switch.py b/homeassistant/components/flexit_bacnet/switch.py
index c58e35cda75..bdeff006181 100644
--- a/homeassistant/components/flexit_bacnet/switch.py
+++ b/homeassistant/components/flexit_bacnet/switch.py
@@ -13,13 +13,12 @@ from homeassistant.components.switch import (
SwitchEntity,
SwitchEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FlexitCoordinator
from .const import DOMAIN
+from .coordinator import FlexitConfigEntry, FlexitCoordinator
from .entity import FlexitEntity
@@ -40,22 +39,39 @@ SWITCHES: tuple[FlexitSwitchEntityDescription, ...] = (
turn_on_fn=lambda data: data.enable_electric_heater(),
turn_off_fn=lambda data: data.disable_electric_heater(),
),
+ FlexitSwitchEntityDescription(
+ key="fireplace_mode",
+ translation_key="fireplace_mode",
+ is_on_fn=lambda data: data.fireplace_ventilation_status,
+ turn_on_fn=lambda data: data.trigger_fireplace_mode(),
+ turn_off_fn=lambda data: data.trigger_fireplace_mode(),
+ ),
+ FlexitSwitchEntityDescription(
+ key="cooker_hood_mode",
+ translation_key="cooker_hood_mode",
+ is_on_fn=lambda data: data.cooker_hood_status,
+ turn_on_fn=lambda data: data.activate_cooker_hood(),
+ turn_off_fn=lambda data: data.deactivate_cooker_hood(),
+ ),
)
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FlexitConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Flexit (bacnet) switch from a config entry."""
- coordinator: FlexitCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+ coordinator = config_entry.runtime_data
async_add_entities(
FlexitSwitch(coordinator, description) for description in SWITCHES
)
+PARALLEL_UPDATES = 1
+
+
class FlexitSwitch(FlexitEntity, SwitchEntity):
"""Representation of a Flexit Switch."""
@@ -82,19 +98,31 @@ class FlexitSwitch(FlexitEntity, SwitchEntity):
return self.entity_description.is_on_fn(self.coordinator.data)
async def async_turn_on(self, **kwargs: Any) -> None:
- """Turn electric heater on."""
+ """Turn switch on."""
try:
await self.entity_description.turn_on_fn(self.coordinator.data)
except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
- raise HomeAssistantError from exc
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="switch_turn",
+ translation_placeholders={
+ "state": "on",
+ },
+ ) from exc
finally:
await self.coordinator.async_refresh()
async def async_turn_off(self, **kwargs: Any) -> None:
- """Turn electric heater off."""
+ """Turn switch off."""
try:
await self.entity_description.turn_off_fn(self.coordinator.data)
except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
- raise HomeAssistantError from exc
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="switch_turn",
+ translation_placeholders={
+ "state": "off",
+ },
+ ) from exc
finally:
await self.coordinator.async_refresh()
diff --git a/homeassistant/components/flick_electric/__init__.py b/homeassistant/components/flick_electric/__init__.py
index 3ffddee1c7d..ad772c06b2e 100644
--- a/homeassistant/components/flick_electric/__init__.py
+++ b/homeassistant/components/flick_electric/__init__.py
@@ -9,7 +9,6 @@ from pyflick import FlickAPI
from pyflick.authentication import SimpleFlickAuth
from pyflick.const import DEFAULT_CLIENT_ID, DEFAULT_CLIENT_SECRET
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_ACCESS_TOKEN,
CONF_CLIENT_ID,
@@ -35,9 +34,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: FlickConfigEntry) -> boo
"""Set up Flick Electric from a config entry."""
auth = HassFlickAuth(hass, entry)
- coordinator = FlickElectricDataCoordinator(
- hass, FlickAPI(auth), entry.data[CONF_SUPPLY_NODE_REF]
- )
+ coordinator = FlickElectricDataCoordinator(hass, entry, FlickAPI(auth))
await coordinator.async_config_entry_first_refresh()
@@ -53,7 +50,9 @@ async def async_unload_entry(hass: HomeAssistant, entry: FlickConfigEntry) -> bo
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
-async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, config_entry: FlickConfigEntry
+) -> bool:
"""Migrate old entry."""
_LOGGER.debug(
"Migrating configuration from version %s.%s",
diff --git a/homeassistant/components/flick_electric/coordinator.py b/homeassistant/components/flick_electric/coordinator.py
index 474efc5297d..114b364635f 100644
--- a/homeassistant/components/flick_electric/coordinator.py
+++ b/homeassistant/components/flick_electric/coordinator.py
@@ -13,6 +13,8 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+from .const import CONF_SUPPLY_NODE_REF
+
_LOGGER = logging.getLogger(__name__)
SCAN_INTERVAL = timedelta(minutes=5)
@@ -23,17 +25,23 @@ type FlickConfigEntry = ConfigEntry[FlickElectricDataCoordinator]
class FlickElectricDataCoordinator(DataUpdateCoordinator[FlickPrice]):
"""Coordinator for flick power price."""
+ config_entry: FlickConfigEntry
+
def __init__(
- self, hass: HomeAssistant, api: FlickAPI, supply_node_ref: str
+ self,
+ hass: HomeAssistant,
+ config_entry: FlickConfigEntry,
+ api: FlickAPI,
) -> None:
"""Initialize FlickElectricDataCoordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Flick Electric",
update_interval=SCAN_INTERVAL,
)
- self.supply_node_ref = supply_node_ref
+ self.supply_node_ref = config_entry.data[CONF_SUPPLY_NODE_REF]
self._api = api
async def _async_update_data(self) -> FlickPrice:
diff --git a/homeassistant/components/flick_electric/sensor.py b/homeassistant/components/flick_electric/sensor.py
index 73b6f8793fb..636d12525ad 100644
--- a/homeassistant/components/flick_electric/sensor.py
+++ b/homeassistant/components/flick_electric/sensor.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.sensor import SensorEntity
from homeassistant.const import CURRENCY_CENT, UnitOfEnergy
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ATTR_COMPONENTS, ATTR_END_AT, ATTR_START_AT
@@ -21,7 +21,7 @@ SCAN_INTERVAL = timedelta(minutes=5)
async def async_setup_entry(
hass: HomeAssistant,
entry: FlickConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Flick Sensor Setup."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/flipr/__init__.py b/homeassistant/components/flipr/__init__.py
index 99bddb5a0d0..81e61f2554a 100644
--- a/homeassistant/components/flipr/__init__.py
+++ b/homeassistant/components/flipr/__init__.py
@@ -1,7 +1,6 @@
"""The Flipr integration."""
from collections import Counter
-from dataclasses import dataclass
import logging
from flipr_api import FliprAPIRestClient
@@ -13,24 +12,18 @@ from homeassistant.exceptions import ConfigEntryError
from homeassistant.helpers import issue_registry as ir
from .const import DOMAIN
-from .coordinator import FliprDataUpdateCoordinator, FliprHubDataUpdateCoordinator
+from .coordinator import (
+ FliprConfigEntry,
+ FliprData,
+ FliprDataUpdateCoordinator,
+ FliprHubDataUpdateCoordinator,
+)
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SELECT, Platform.SENSOR, Platform.SWITCH]
_LOGGER = logging.getLogger(__name__)
-@dataclass
-class FliprData:
- """The Flipr data class."""
-
- flipr_coordinators: list[FliprDataUpdateCoordinator]
- hub_coordinators: list[FliprHubDataUpdateCoordinator]
-
-
-type FliprConfigEntry = ConfigEntry[FliprData]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: FliprConfigEntry) -> bool:
"""Set up flipr from a config entry."""
@@ -50,13 +43,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: FliprConfigEntry) -> boo
flipr_coordinators = []
for flipr_id in ids["flipr"]:
- flipr_coordinator = FliprDataUpdateCoordinator(hass, client, flipr_id)
+ flipr_coordinator = FliprDataUpdateCoordinator(hass, entry, client, flipr_id)
await flipr_coordinator.async_config_entry_first_refresh()
flipr_coordinators.append(flipr_coordinator)
hub_coordinators = []
for hub_id in ids["hub"]:
- hub_coordinator = FliprHubDataUpdateCoordinator(hass, client, hub_id)
+ hub_coordinator = FliprHubDataUpdateCoordinator(hass, entry, client, hub_id)
await hub_coordinator.async_config_entry_first_refresh()
hub_coordinators.append(hub_coordinator)
diff --git a/homeassistant/components/flipr/binary_sensor.py b/homeassistant/components/flipr/binary_sensor.py
index cc6a9d36abc..899d045ad86 100644
--- a/homeassistant/components/flipr/binary_sensor.py
+++ b/homeassistant/components/flipr/binary_sensor.py
@@ -8,9 +8,9 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FliprConfigEntry
+from .coordinator import FliprConfigEntry
from .entity import FliprEntity
BINARY_SENSORS_TYPES: tuple[BinarySensorEntityDescription, ...] = (
@@ -30,7 +30,7 @@ BINARY_SENSORS_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FliprConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup of flipr binary sensors."""
diff --git a/homeassistant/components/flipr/coordinator.py b/homeassistant/components/flipr/coordinator.py
index 12fd174fe7d..82de5ae34d5 100644
--- a/homeassistant/components/flipr/coordinator.py
+++ b/homeassistant/components/flipr/coordinator.py
@@ -1,5 +1,8 @@
"""DataUpdateCoordinator for flipr integration."""
+from __future__ import annotations
+
+from dataclasses import dataclass
from datetime import timedelta
import logging
from typing import Any
@@ -14,13 +17,28 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
_LOGGER = logging.getLogger(__name__)
+@dataclass
+class FliprData:
+ """The Flipr data class."""
+
+ flipr_coordinators: list[FliprDataUpdateCoordinator]
+ hub_coordinators: list[FliprHubDataUpdateCoordinator]
+
+
+type FliprConfigEntry = ConfigEntry[FliprData]
+
+
class BaseDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
"""Parent class to hold Flipr and Hub data retrieval."""
- config_entry: ConfigEntry
+ config_entry: FliprConfigEntry
def __init__(
- self, hass: HomeAssistant, client: FliprAPIRestClient, flipr_or_hub_id: str
+ self,
+ hass: HomeAssistant,
+ config_entry: FliprConfigEntry,
+ client: FliprAPIRestClient,
+ flipr_or_hub_id: str,
) -> None:
"""Initialize."""
self.device_id = flipr_or_hub_id
@@ -29,6 +47,7 @@ class BaseDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"Flipr or Hub data measure for {self.device_id}",
update_interval=timedelta(minutes=15),
)
diff --git a/homeassistant/components/flipr/select.py b/homeassistant/components/flipr/select.py
index b8a8f0db60a..c10e9c6e91b 100644
--- a/homeassistant/components/flipr/select.py
+++ b/homeassistant/components/flipr/select.py
@@ -4,9 +4,9 @@ import logging
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FliprConfigEntry
+from .coordinator import FliprConfigEntry
from .entity import FliprEntity
_LOGGER = logging.getLogger(__name__)
@@ -23,7 +23,7 @@ SELECT_TYPES: tuple[SelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FliprConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select for Flipr hub mode."""
coordinators = config_entry.runtime_data.hub_coordinators
diff --git a/homeassistant/components/flipr/sensor.py b/homeassistant/components/flipr/sensor.py
index ba863718182..296bcaac68d 100644
--- a/homeassistant/components/flipr/sensor.py
+++ b/homeassistant/components/flipr/sensor.py
@@ -10,9 +10,9 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfElectricPotential, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FliprConfigEntry
+from .coordinator import FliprConfigEntry
from .entity import FliprEntity
SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
@@ -57,7 +57,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FliprConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinators = config_entry.runtime_data.flipr_coordinators
diff --git a/homeassistant/components/flipr/switch.py b/homeassistant/components/flipr/switch.py
index 65e729ec280..4db8b54af8a 100644
--- a/homeassistant/components/flipr/switch.py
+++ b/homeassistant/components/flipr/switch.py
@@ -5,9 +5,9 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FliprConfigEntry
+from .coordinator import FliprConfigEntry
from .entity import FliprEntity
_LOGGER = logging.getLogger(__name__)
@@ -23,7 +23,7 @@ SWITCH_TYPES: tuple[SwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FliprConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch for Flipr hub."""
coordinators = config_entry.runtime_data.hub_coordinators
diff --git a/homeassistant/components/flo/__init__.py b/homeassistant/components/flo/__init__.py
index b619df91d59..88824b041e7 100644
--- a/homeassistant/components/flo/__init__.py
+++ b/homeassistant/components/flo/__init__.py
@@ -6,27 +6,23 @@ import logging
from aioflo import async_get_api
from aioflo.errors import RequestError
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .const import CLIENT, DOMAIN
-from .coordinator import FloDeviceDataUpdateCoordinator
+from .coordinator import FloConfigEntry, FloDeviceDataUpdateCoordinator, FloRuntimeData
_LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR, Platform.SWITCH]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: FloConfigEntry) -> bool:
"""Set up flo from a config entry."""
session = async_get_clientsession(hass)
- hass.data.setdefault(DOMAIN, {})
- hass.data[DOMAIN][entry.entry_id] = {}
try:
- hass.data[DOMAIN][entry.entry_id][CLIENT] = client = await async_get_api(
+ client = await async_get_api(
entry.data[CONF_USERNAME], entry.data[CONF_PASSWORD], session=session
)
except RequestError as err:
@@ -36,8 +32,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
_LOGGER.debug("Flo user information with locations: %s", user_info)
- hass.data[DOMAIN][entry.entry_id]["devices"] = devices = [
- FloDeviceDataUpdateCoordinator(hass, client, location["id"], device["id"])
+ devices = [
+ FloDeviceDataUpdateCoordinator(
+ hass, entry, client, location["id"], device["id"]
+ )
for location in user_info["locations"]
for device in location["devices"]
]
@@ -45,14 +43,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
tasks = [device.async_refresh() for device in devices]
await asyncio.gather(*tasks)
+ entry.runtime_data = FloRuntimeData(client=client, devices=devices)
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: FloConfigEntry) -> bool:
"""Unload a config entry."""
- unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- if unload_ok:
- hass.data[DOMAIN].pop(entry.entry_id)
- return unload_ok
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/flo/binary_sensor.py b/homeassistant/components/flo/binary_sensor.py
index 20f5d7822d2..89f317fd3c6 100644
--- a/homeassistant/components/flo/binary_sensor.py
+++ b/homeassistant/components/flo/binary_sensor.py
@@ -6,24 +6,20 @@ from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
BinarySensorEntity,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN as FLO_DOMAIN
-from .coordinator import FloDeviceDataUpdateCoordinator
+from .coordinator import FloConfigEntry
from .entity import FloEntity
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FloConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flo sensors from config entry."""
- devices: list[FloDeviceDataUpdateCoordinator] = hass.data[FLO_DOMAIN][
- config_entry.entry_id
- ]["devices"]
+ devices = config_entry.runtime_data.devices
entities: list[BinarySensorEntity] = []
for device in devices:
if device.device_type == "puck_oem":
diff --git a/homeassistant/components/flo/const.py b/homeassistant/components/flo/const.py
index 9eb00ebfa62..5b1d926d9f4 100644
--- a/homeassistant/components/flo/const.py
+++ b/homeassistant/components/flo/const.py
@@ -4,7 +4,6 @@ import logging
LOGGER = logging.getLogger(__package__)
-CLIENT = "client"
DOMAIN = "flo"
FLO_HOME = "home"
FLO_AWAY = "away"
diff --git a/homeassistant/components/flo/coordinator.py b/homeassistant/components/flo/coordinator.py
index d0dd38bd490..9f540b230f4 100644
--- a/homeassistant/components/flo/coordinator.py
+++ b/homeassistant/components/flo/coordinator.py
@@ -3,6 +3,7 @@
from __future__ import annotations
import asyncio
+from dataclasses import dataclass
from datetime import datetime, timedelta
from typing import Any
@@ -10,20 +11,37 @@ from aioflo.api import API
from aioflo.errors import RequestError
from orjson import JSONDecodeError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt as dt_util
from .const import DOMAIN as FLO_DOMAIN, LOGGER
+type FloConfigEntry = ConfigEntry[FloRuntimeData]
+
+
+@dataclass
+class FloRuntimeData:
+ """Flo runtime data."""
+
+ client: API
+ devices: list[FloDeviceDataUpdateCoordinator]
+
class FloDeviceDataUpdateCoordinator(DataUpdateCoordinator):
"""Flo device object."""
+ config_entry: ConfigEntry
_failure_count: int = 0
def __init__(
- self, hass: HomeAssistant, api_client: API, location_id: str, device_id: str
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ api_client: API,
+ location_id: str,
+ device_id: str,
) -> None:
"""Initialize the device."""
self.hass: HomeAssistant = hass
@@ -36,6 +54,7 @@ class FloDeviceDataUpdateCoordinator(DataUpdateCoordinator):
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=f"{FLO_DOMAIN}-{device_id}",
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/flo/entity.py b/homeassistant/components/flo/entity.py
index b0cf8d04313..072afbae4f2 100644
--- a/homeassistant/components/flo/entity.py
+++ b/homeassistant/components/flo/entity.py
@@ -45,10 +45,10 @@ class FloEntity(Entity):
"""Return True if device is available."""
return self._device.available
- async def async_update(self):
+ async def async_update(self) -> None:
"""Update Flo entity."""
await self._device.async_request_refresh()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""When entity is added to hass."""
self.async_on_remove(self._device.async_add_listener(self.async_write_ha_state))
diff --git a/homeassistant/components/flo/sensor.py b/homeassistant/components/flo/sensor.py
index 7419b0a1c3b..ca763839b87 100644
--- a/homeassistant/components/flo/sensor.py
+++ b/homeassistant/components/flo/sensor.py
@@ -7,7 +7,6 @@ from homeassistant.components.sensor import (
SensorEntity,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
PERCENTAGE,
UnitOfPressure,
@@ -16,22 +15,19 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN as FLO_DOMAIN
-from .coordinator import FloDeviceDataUpdateCoordinator
+from .coordinator import FloConfigEntry
from .entity import FloEntity
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FloConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flo sensors from config entry."""
- devices: list[FloDeviceDataUpdateCoordinator] = hass.data[FLO_DOMAIN][
- config_entry.entry_id
- ]["devices"]
+ devices = config_entry.runtime_data.devices
entities = []
for device in devices:
if device.device_type == "puck_oem":
diff --git a/homeassistant/components/flo/strings.json b/homeassistant/components/flo/strings.json
index 3444911fbd4..64e22bedec3 100644
--- a/homeassistant/components/flo/strings.json
+++ b/homeassistant/components/flo/strings.json
@@ -60,11 +60,11 @@
"fields": {
"sleep_minutes": {
"name": "Sleep minutes",
- "description": "The time to sleep in minutes."
+ "description": "The duration to sleep in minutes."
},
"revert_to_mode": {
"name": "Revert to mode",
- "description": "The mode to revert to after sleep_minutes has elapsed."
+ "description": "The mode to revert to after the 'Sleep minutes' duration has elapsed."
}
}
},
@@ -78,7 +78,7 @@
},
"run_health_test": {
"name": "Run health test",
- "description": "Have the Flo device run a health test."
+ "description": "Requests the Flo device to run a health test."
}
}
}
diff --git a/homeassistant/components/flo/switch.py b/homeassistant/components/flo/switch.py
index f0460839837..12e242db5c8 100644
--- a/homeassistant/components/flo/switch.py
+++ b/homeassistant/components/flo/switch.py
@@ -8,13 +8,11 @@ from aioflo.location import SLEEP_MINUTE_OPTIONS, SYSTEM_MODE_HOME, SYSTEM_REVER
import voluptuous as vol
from homeassistant.components.switch import SwitchEntity
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN as FLO_DOMAIN
-from .coordinator import FloDeviceDataUpdateCoordinator
+from .coordinator import FloConfigEntry, FloDeviceDataUpdateCoordinator
from .entity import FloEntity
ATTR_REVERT_TO_MODE = "revert_to_mode"
@@ -27,13 +25,11 @@ SERVICE_RUN_HEALTH_TEST = "run_health_test"
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: FloConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flo switches from config entry."""
- devices: list[FloDeviceDataUpdateCoordinator] = hass.data[FLO_DOMAIN][
- config_entry.entry_id
- ]["devices"]
+ devices = config_entry.runtime_data.devices
async_add_entities(
[FloSwitch(device) for device in devices if device.device_type != "puck_oem"]
diff --git a/homeassistant/components/flume/__init__.py b/homeassistant/components/flume/__init__.py
index 24ed41b21c1..d229665ca62 100644
--- a/homeassistant/components/flume/__init__.py
+++ b/homeassistant/components/flume/__init__.py
@@ -7,7 +7,7 @@ from requests import Session
from requests.exceptions import RequestException
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import (
CONF_CLIENT_ID,
CONF_CLIENT_SECRET,
@@ -41,7 +41,7 @@ LIST_NOTIFICATIONS_SERVICE_SCHEMA = vol.All(
def _setup_entry(
- hass: HomeAssistant, entry: ConfigEntry
+ hass: HomeAssistant, entry: FlumeConfigEntry
) -> tuple[FlumeAuth, FlumeDeviceList, Session]:
"""Config entry set up in executor."""
config = entry.data
@@ -79,7 +79,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: FlumeConfigEntry) -> boo
_setup_entry, hass, entry
)
notification_coordinator = FlumeNotificationDataUpdateCoordinator(
- hass=hass, auth=flume_auth
+ hass=hass, config_entry=entry, auth=flume_auth
)
entry.runtime_data = FlumeRuntimeData(
diff --git a/homeassistant/components/flume/binary_sensor.py b/homeassistant/components/flume/binary_sensor.py
index 67cb71c5767..2c2dc285036 100644
--- a/homeassistant/components/flume/binary_sensor.py
+++ b/homeassistant/components/flume/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
FLUME_TYPE_BRIDGE,
@@ -69,7 +69,7 @@ FLUME_BINARY_NOTIFICATION_SENSORS: tuple[FlumeBinarySensorEntityDescription, ...
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FlumeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Flume binary sensor.."""
flume_domain_data = config_entry.runtime_data
@@ -80,7 +80,7 @@ async def async_setup_entry(
] = []
connection_coordinator = FlumeDeviceConnectionUpdateCoordinator(
- hass=hass, flume_devices=flume_devices
+ hass=hass, config_entry=config_entry, flume_devices=flume_devices
)
notification_coordinator = flume_domain_data.notifications_coordinator
flume_devices = get_valid_flume_devices(flume_devices)
diff --git a/homeassistant/components/flume/coordinator.py b/homeassistant/components/flume/coordinator.py
index fc76600cad4..1dabf5726b2 100644
--- a/homeassistant/components/flume/coordinator.py
+++ b/homeassistant/components/flume/coordinator.py
@@ -38,10 +38,18 @@ type FlumeConfigEntry = ConfigEntry[FlumeRuntimeData]
class FlumeDeviceDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Data update coordinator for an individual flume device."""
- def __init__(self, hass: HomeAssistant, flume_device: FlumeData) -> None:
+ config_entry: FlumeConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: FlumeConfigEntry,
+ flume_device: FlumeData,
+ ) -> None:
"""Initialize the Coordinator."""
super().__init__(
hass,
+ config_entry=config_entry,
name=DOMAIN,
logger=_LOGGER,
update_interval=DEVICE_SCAN_INTERVAL,
@@ -65,10 +73,18 @@ class FlumeDeviceDataUpdateCoordinator(DataUpdateCoordinator[None]):
class FlumeDeviceConnectionUpdateCoordinator(DataUpdateCoordinator[None]):
"""Date update coordinator to read connected status from Devices endpoint."""
- def __init__(self, hass: HomeAssistant, flume_devices: FlumeDeviceList) -> None:
+ config_entry: FlumeConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: FlumeConfigEntry,
+ flume_devices: FlumeDeviceList,
+ ) -> None:
"""Initialize the Coordinator."""
super().__init__(
hass,
+ config_entry=config_entry,
name=DOMAIN,
logger=_LOGGER,
update_interval=DEVICE_CONNECTION_SCAN_INTERVAL,
@@ -96,10 +112,15 @@ class FlumeDeviceConnectionUpdateCoordinator(DataUpdateCoordinator[None]):
class FlumeNotificationDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Data update coordinator for flume notifications."""
- def __init__(self, hass: HomeAssistant, auth: FlumeAuth) -> None:
+ config_entry: FlumeConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: FlumeConfigEntry, auth: FlumeAuth
+ ) -> None:
"""Initialize the Coordinator."""
super().__init__(
hass,
+ config_entry=config_entry,
name=DOMAIN,
logger=_LOGGER,
update_interval=NOTIFICATION_SCAN_INTERVAL,
diff --git a/homeassistant/components/flume/sensor.py b/homeassistant/components/flume/sensor.py
index 6c7cc0ab37d..0f0213ec984 100644
--- a/homeassistant/components/flume/sensor.py
+++ b/homeassistant/components/flume/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfVolume
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -108,7 +108,7 @@ def make_flume_datas(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FlumeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flume sensor."""
@@ -132,7 +132,7 @@ async def async_setup_entry(
flume_device = flume_datas[device_id]
coordinator = FlumeDeviceDataUpdateCoordinator(
- hass=hass, flume_device=flume_device
+ hass=hass, config_entry=config_entry, flume_device=flume_device
)
flume_entity_list.extend(
diff --git a/homeassistant/components/flux_led/__init__.py b/homeassistant/components/flux_led/__init__.py
index 7597a7c9c9a..7515b6b8dfc 100644
--- a/homeassistant/components/flux_led/__init__.py
+++ b/homeassistant/components/flux_led/__init__.py
@@ -11,7 +11,6 @@ from flux_led.aio import AIOWifiLedBulb
from flux_led.const import ATTR_ID, WhiteChannelType
from flux_led.scanner import FluxLEDDiscovery
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryNotReady
@@ -39,7 +38,7 @@ from .const import (
FLUX_LED_EXCEPTIONS,
SIGNAL_STATE_UPDATED,
)
-from .coordinator import FluxLedUpdateCoordinator
+from .coordinator import FluxLedConfigEntry, FluxLedUpdateCoordinator
from .discovery import (
async_build_cached_discovery,
async_clear_discovery_cache,
@@ -113,7 +112,9 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
return True
-async def _async_migrate_unique_ids(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def _async_migrate_unique_ids(
+ hass: HomeAssistant, entry: FluxLedConfigEntry
+) -> None:
"""Migrate entities when the mac address gets discovered."""
@callback
@@ -146,14 +147,16 @@ async def _async_migrate_unique_ids(hass: HomeAssistant, entry: ConfigEntry) ->
await er.async_migrate_entries(hass, entry.entry_id, _async_migrator)
-async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def _async_update_listener(
+ hass: HomeAssistant, entry: FluxLedConfigEntry
+) -> None:
"""Handle options update."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
if entry.title != coordinator.title:
await hass.config_entries.async_reload(entry.entry_id)
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: FluxLedConfigEntry) -> bool:
"""Set up Flux LED/MagicLight from a config entry."""
host = entry.data[CONF_HOST]
discovery_cached = True
@@ -206,7 +209,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await _async_migrate_unique_ids(hass, entry)
coordinator = FluxLedUpdateCoordinator(hass, device, entry)
- hass.data[DOMAIN][entry.entry_id] = coordinator
+ entry.runtime_data = coordinator
platforms = PLATFORMS_BY_TYPE[device.device_type]
await hass.config_entries.async_forward_entry_setups(entry, platforms)
@@ -239,13 +242,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: FluxLedConfigEntry) -> bool:
"""Unload a config entry."""
- device: AIOWifiLedBulb = hass.data[DOMAIN][entry.entry_id].device
+ device = entry.runtime_data.device
platforms = PLATFORMS_BY_TYPE[device.device_type]
if unload_ok := await hass.config_entries.async_unload_platforms(entry, platforms):
# Make sure we probe the device again in case something has changed externally
async_clear_discovery_cache(hass, entry.data[CONF_HOST])
- del hass.data[DOMAIN][entry.entry_id]
await device.async_stop()
return unload_ok
diff --git a/homeassistant/components/flux_led/button.py b/homeassistant/components/flux_led/button.py
index 90918a55bb2..c4a7ff6569c 100644
--- a/homeassistant/components/flux_led/button.py
+++ b/homeassistant/components/flux_led/button.py
@@ -5,7 +5,6 @@ from __future__ import annotations
from flux_led.aio import AIOWifiLedBulb
from flux_led.protocol import RemoteConfig
-from homeassistant import config_entries
from homeassistant.components.button import (
ButtonDeviceClass,
ButtonEntity,
@@ -13,10 +12,9 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import FluxLedUpdateCoordinator
+from .coordinator import FluxLedConfigEntry
from .entity import FluxBaseEntity
_RESTART_KEY = "restart"
@@ -34,11 +32,11 @@ UNPAIR_REMOTES_DESCRIPTION = ButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: FluxLedConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Magic Home button based on a config entry."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
device = coordinator.device
entities: list[FluxButton] = [
FluxButton(coordinator.device, entry, RESTART_BUTTON_DESCRIPTION)
@@ -59,7 +57,7 @@ class FluxButton(FluxBaseEntity, ButtonEntity):
def __init__(
self,
device: AIOWifiLedBulb,
- entry: config_entries.ConfigEntry,
+ entry: FluxLedConfigEntry,
description: ButtonEntityDescription,
) -> None:
"""Initialize the button."""
diff --git a/homeassistant/components/flux_led/config_flow.py b/homeassistant/components/flux_led/config_flow.py
index 035be5b115c..754ed0525b9 100644
--- a/homeassistant/components/flux_led/config_flow.py
+++ b/homeassistant/components/flux_led/config_flow.py
@@ -18,7 +18,6 @@ import voluptuous as vol
from homeassistant.config_entries import (
SOURCE_IGNORE,
- ConfigEntry,
ConfigEntryState,
ConfigFlow,
ConfigFlowResult,
@@ -46,6 +45,7 @@ from .const import (
TRANSITION_JUMP,
TRANSITION_STROBE,
)
+from .coordinator import FluxLedConfigEntry
from .discovery import (
async_discover_device,
async_discover_devices,
@@ -72,7 +72,7 @@ class FluxLedConfigFlow(ConfigFlow, domain=DOMAIN):
@staticmethod
@callback
def async_get_options_flow(
- config_entry: ConfigEntry,
+ config_entry: FluxLedConfigEntry,
) -> FluxLedOptionsFlow:
"""Get the options flow for the Flux LED component."""
return FluxLedOptionsFlow()
diff --git a/homeassistant/components/flux_led/coordinator.py b/homeassistant/components/flux_led/coordinator.py
index a473387a513..78d8bb947fd 100644
--- a/homeassistant/components/flux_led/coordinator.py
+++ b/homeassistant/components/flux_led/coordinator.py
@@ -20,21 +20,25 @@ _LOGGER = logging.getLogger(__name__)
REQUEST_REFRESH_DELAY: Final = 2.0
+type FluxLedConfigEntry = ConfigEntry[FluxLedUpdateCoordinator]
+
class FluxLedUpdateCoordinator(DataUpdateCoordinator[None]):
"""DataUpdateCoordinator to gather data for a specific flux_led device."""
+ config_entry: FluxLedConfigEntry
+
def __init__(
- self, hass: HomeAssistant, device: AIOWifiLedBulb, entry: ConfigEntry
+ self, hass: HomeAssistant, device: AIOWifiLedBulb, entry: FluxLedConfigEntry
) -> None:
"""Initialize DataUpdateCoordinator to gather data for specific device."""
self.device = device
self.title = entry.title
- self.entry = entry
self.force_next_update = False
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=self.device.ipaddr,
update_interval=timedelta(seconds=10),
# We don't want an immediate refresh since the device
diff --git a/homeassistant/components/flux_led/diagnostics.py b/homeassistant/components/flux_led/diagnostics.py
index e24c1aff9a4..683aa362377 100644
--- a/homeassistant/components/flux_led/diagnostics.py
+++ b/homeassistant/components/flux_led/diagnostics.py
@@ -4,22 +4,19 @@ from __future__ import annotations
from typing import Any
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
-from .coordinator import FluxLedUpdateCoordinator
+from .coordinator import FluxLedConfigEntry
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, entry: ConfigEntry
+ hass: HomeAssistant, entry: FluxLedConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
return {
"entry": {
"title": entry.title,
"data": dict(entry.data),
},
- "data": coordinator.device.diagnostics,
+ "data": entry.runtime_data.device.diagnostics,
}
diff --git a/homeassistant/components/flux_led/discovery.py b/homeassistant/components/flux_led/discovery.py
index d55f560193f..c3a3c5df3a7 100644
--- a/homeassistant/components/flux_led/discovery.py
+++ b/homeassistant/components/flux_led/discovery.py
@@ -23,9 +23,8 @@ from flux_led.const import (
from flux_led.models_db import get_model_description
from flux_led.scanner import FluxLEDDiscovery
-from homeassistant import config_entries
from homeassistant.components import network
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import SOURCE_INTEGRATION_DISCOVERY, ConfigEntryState
from homeassistant.const import CONF_HOST, CONF_MODEL, CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, discovery_flow
@@ -44,6 +43,7 @@ from .const import (
DOMAIN,
FLUX_LED_DISCOVERY,
)
+from .coordinator import FluxLedConfigEntry
from .util import format_as_flux_mac, mac_matches_by_one
_LOGGER = logging.getLogger(__name__)
@@ -63,7 +63,7 @@ CONF_TO_DISCOVERY: Final = {
@callback
-def async_build_cached_discovery(entry: ConfigEntry) -> FluxLEDDiscovery:
+def async_build_cached_discovery(entry: FluxLedConfigEntry) -> FluxLEDDiscovery:
"""When discovery is unavailable, load it from the config entry."""
data = entry.data
return FluxLEDDiscovery(
@@ -116,7 +116,7 @@ def async_populate_data_from_discovery(
@callback
def async_update_entry_from_discovery(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
+ entry: FluxLedConfigEntry,
device: FluxLEDDiscovery,
model_num: int | None,
allow_update_mac: bool,
@@ -230,6 +230,6 @@ def async_trigger_discovery(
discovery_flow.async_create_flow(
hass,
DOMAIN,
- context={"source": config_entries.SOURCE_INTEGRATION_DISCOVERY},
+ context={"source": SOURCE_INTEGRATION_DISCOVERY},
data={**device},
)
diff --git a/homeassistant/components/flux_led/entity.py b/homeassistant/components/flux_led/entity.py
index bcf7bfff9ed..f9b87dbb8c1 100644
--- a/homeassistant/components/flux_led/entity.py
+++ b/homeassistant/components/flux_led/entity.py
@@ -89,7 +89,9 @@ class FluxEntity(CoordinatorEntity[FluxLedUpdateCoordinator]):
self._attr_unique_id = f"{base_unique_id}_{key}"
else:
self._attr_unique_id = base_unique_id
- self._attr_device_info = _async_device_info(self._device, coordinator.entry)
+ self._attr_device_info = _async_device_info(
+ self._device, coordinator.config_entry
+ )
async def _async_ensure_device_on(self) -> None:
"""Turn the device on if it needs to be turned on before a command."""
diff --git a/homeassistant/components/flux_led/light.py b/homeassistant/components/flux_led/light.py
index 2a0b5795970..79dae33a2a5 100644
--- a/homeassistant/components/flux_led/light.py
+++ b/homeassistant/components/flux_led/light.py
@@ -11,7 +11,6 @@ from flux_led.protocol import MusicMode
from flux_led.utils import rgbcw_brightness, rgbcw_to_rgbwc, rgbw_brightness
import voluptuous as vol
-from homeassistant import config_entries
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_COLOR_TEMP_KELVIN,
@@ -26,7 +25,7 @@ from homeassistant.components.light import (
from homeassistant.const import CONF_EFFECT
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -38,7 +37,6 @@ from .const import (
CONF_SPEED_PCT,
CONF_TRANSITION,
DEFAULT_EFFECT_SPEED,
- DOMAIN,
MIN_CCT_BRIGHTNESS,
MIN_RGB_BRIGHTNESS,
MULTI_BRIGHTNESS_COLOR_MODES,
@@ -46,7 +44,7 @@ from .const import (
TRANSITION_JUMP,
TRANSITION_STROBE,
)
-from .coordinator import FluxLedUpdateCoordinator
+from .coordinator import FluxLedConfigEntry, FluxLedUpdateCoordinator
from .entity import FluxOnOffEntity
from .util import (
_effect_brightness,
@@ -134,11 +132,11 @@ SET_ZONES_DICT: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: FluxLedConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flux lights."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
platform = entity_platform.async_get_current_platform()
platform.async_register_entity_service(
diff --git a/homeassistant/components/flux_led/number.py b/homeassistant/components/flux_led/number.py
index 93687c0c579..edf6b8c9654 100644
--- a/homeassistant/components/flux_led/number.py
+++ b/homeassistant/components/flux_led/number.py
@@ -16,18 +16,16 @@ from flux_led.protocol import (
SEGMENTS_MAX,
)
-from homeassistant import config_entries
from homeassistant.components.light import EFFECT_RANDOM
from homeassistant.components.number import NumberEntity, NumberMode
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.debounce import Debouncer
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .const import DOMAIN
-from .coordinator import FluxLedUpdateCoordinator
+from .coordinator import FluxLedConfigEntry, FluxLedUpdateCoordinator
from .entity import FluxEntity
from .util import _effect_brightness
@@ -38,11 +36,11 @@ DEBOUNCE_TIME = 1
async def async_setup_entry(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: FluxLedConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flux lights."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
device = coordinator.device
entities: list[
FluxSpeedNumber
diff --git a/homeassistant/components/flux_led/select.py b/homeassistant/components/flux_led/select.py
index 3809e73147a..bcb44c995b8 100644
--- a/homeassistant/components/flux_led/select.py
+++ b/homeassistant/components/flux_led/select.py
@@ -13,14 +13,13 @@ from flux_led.const import (
)
from flux_led.protocol import PowerRestoreState, RemoteConfig
-from homeassistant import config_entries
from homeassistant.components.select import SelectEntity
from homeassistant.const import CONF_NAME, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import CONF_WHITE_CHANNEL_TYPE, DOMAIN, FLUX_COLOR_MODE_RGBW
-from .coordinator import FluxLedUpdateCoordinator
+from .const import CONF_WHITE_CHANNEL_TYPE, FLUX_COLOR_MODE_RGBW
+from .coordinator import FluxLedConfigEntry, FluxLedUpdateCoordinator
from .entity import FluxBaseEntity, FluxEntity
from .util import _human_readable_option
@@ -29,9 +28,7 @@ NAME_TO_POWER_RESTORE_STATE = {
}
-async def _async_delayed_reload(
- hass: HomeAssistant, entry: config_entries.ConfigEntry
-) -> None:
+async def _async_delayed_reload(hass: HomeAssistant, entry: FluxLedConfigEntry) -> None:
"""Reload after making a change that will effect the operation of the device."""
await asyncio.sleep(STATE_CHANGE_LATENCY)
hass.async_create_task(hass.config_entries.async_reload(entry.entry_id))
@@ -39,11 +36,11 @@ async def _async_delayed_reload(
async def async_setup_entry(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: FluxLedConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flux selects."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
device = coordinator.device
entities: list[
FluxPowerStateSelect
@@ -97,7 +94,7 @@ class FluxPowerStateSelect(FluxConfigAtStartSelect, SelectEntity):
def __init__(
self,
device: AIOWifiLedBulb,
- entry: config_entries.ConfigEntry,
+ entry: FluxLedConfigEntry,
) -> None:
"""Initialize the power state select."""
super().__init__(device, entry)
@@ -141,7 +138,7 @@ class FluxICTypeSelect(FluxConfigSelect):
async def async_select_option(self, option: str) -> None:
"""Change the ic type."""
await self._device.async_set_device_config(ic_type=option)
- await _async_delayed_reload(self.hass, self.coordinator.entry)
+ await _async_delayed_reload(self.hass, self.coordinator.config_entry)
class FluxWiringsSelect(FluxConfigSelect):
@@ -184,7 +181,7 @@ class FluxOperatingModesSelect(FluxConfigSelect):
async def async_select_option(self, option: str) -> None:
"""Change the ic type."""
await self._device.async_set_device_config(operating_mode=option)
- await _async_delayed_reload(self.hass, self.coordinator.entry)
+ await _async_delayed_reload(self.hass, self.coordinator.config_entry)
class FluxRemoteConfigSelect(FluxConfigSelect):
@@ -228,7 +225,7 @@ class FluxWhiteChannelSelect(FluxConfigAtStartSelect):
def __init__(
self,
device: AIOWifiLedBulb,
- entry: config_entries.ConfigEntry,
+ entry: FluxLedConfigEntry,
) -> None:
"""Initialize the white channel select."""
super().__init__(device, entry)
diff --git a/homeassistant/components/flux_led/sensor.py b/homeassistant/components/flux_led/sensor.py
index 5a6633669ae..ad4b9bacbbe 100644
--- a/homeassistant/components/flux_led/sensor.py
+++ b/homeassistant/components/flux_led/sensor.py
@@ -2,24 +2,22 @@
from __future__ import annotations
-from homeassistant import config_entries
from homeassistant.components.sensor import SensorEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import FluxLedUpdateCoordinator
+from .coordinator import FluxLedConfigEntry
from .entity import FluxEntity
async def async_setup_entry(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: FluxLedConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Magic Home sensors."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
if coordinator.device.paired_remotes is not None:
async_add_entities(
[
diff --git a/homeassistant/components/flux_led/switch.py b/homeassistant/components/flux_led/switch.py
index 3adcd9a9da9..5dea5408c84 100644
--- a/homeassistant/components/flux_led/switch.py
+++ b/homeassistant/components/flux_led/switch.py
@@ -8,31 +8,29 @@ from flux_led import DeviceType
from flux_led.aio import AIOWifiLedBulb
from flux_led.const import MODE_MUSIC
-from homeassistant import config_entries
from homeassistant.components.switch import SwitchEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
CONF_REMOTE_ACCESS_ENABLED,
CONF_REMOTE_ACCESS_HOST,
CONF_REMOTE_ACCESS_PORT,
- DOMAIN,
)
-from .coordinator import FluxLedUpdateCoordinator
+from .coordinator import FluxLedConfigEntry, FluxLedUpdateCoordinator
from .discovery import async_clear_discovery_cache
from .entity import FluxBaseEntity, FluxEntity, FluxOnOffEntity
async def async_setup_entry(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: FluxLedConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Flux lights."""
- coordinator: FluxLedUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
entities: list[FluxSwitch | FluxRemoteAccessSwitch | FluxMusicSwitch] = []
base_unique_id = entry.unique_id or entry.entry_id
@@ -70,7 +68,7 @@ class FluxRemoteAccessSwitch(FluxBaseEntity, SwitchEntity):
def __init__(
self,
device: AIOWifiLedBulb,
- entry: config_entries.ConfigEntry,
+ entry: FluxLedConfigEntry,
) -> None:
"""Initialize the light."""
super().__init__(device, entry)
diff --git a/homeassistant/components/folder_watcher/event.py b/homeassistant/components/folder_watcher/event.py
index 7158930e116..472599c4ead 100644
--- a/homeassistant/components/folder_watcher/event.py
+++ b/homeassistant/components/folder_watcher/event.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -25,7 +25,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Folder Watcher event."""
diff --git a/homeassistant/components/folder_watcher/strings.json b/homeassistant/components/folder_watcher/strings.json
index da1e3c1962a..5b1f72bf254 100644
--- a/homeassistant/components/folder_watcher/strings.json
+++ b/homeassistant/components/folder_watcher/strings.json
@@ -36,11 +36,11 @@
"issues": {
"import_failed_not_allowed_path": {
"title": "The Folder Watcher YAML configuration could not be imported",
- "description": "Configuring Folder Watcher using YAML is being removed but your configuration could not be imported as the folder {path} is not in the configured allowlist.\n\nPlease add it to `{config_variable}` in config.yaml and restart Home Assistant to import it and fix this issue."
+ "description": "Configuring Folder Watcher using YAML is being removed but your configuration could not be imported as the folder {path} is not in the configured allowlist.\n\nPlease add it to `{config_variable}` in configuration.yaml and restart Home Assistant to import it and fix this issue."
},
"setup_not_allowed_path": {
"title": "The Folder Watcher configuration for {path} could not start",
- "description": "The path {path} is not accessible or not allowed to be accessed.\n\nPlease check the path is accessible and add it to `{config_variable}` in config.yaml and restart Home Assistant to fix this issue."
+ "description": "The path {path} is not accessible or not allowed to be accessed.\n\nPlease check the path is accessible and add it to `{config_variable}` in configuration.yaml and restart Home Assistant to fix this issue."
}
},
"entity": {
diff --git a/homeassistant/components/forecast_solar/__init__.py b/homeassistant/components/forecast_solar/__init__.py
index 00be13f1235..171341f7226 100644
--- a/homeassistant/components/forecast_solar/__init__.py
+++ b/homeassistant/components/forecast_solar/__init__.py
@@ -2,7 +2,6 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
@@ -12,14 +11,14 @@ from .const import (
CONF_DAMPING_MORNING,
CONF_MODULES_POWER,
)
-from .coordinator import ForecastSolarDataUpdateCoordinator
+from .coordinator import ForecastSolarConfigEntry, ForecastSolarDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
-type ForecastSolarConfigEntry = ConfigEntry[ForecastSolarDataUpdateCoordinator]
-
-async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, entry: ForecastSolarConfigEntry
+) -> bool:
"""Migrate old config entry."""
if entry.version == 1:
@@ -53,11 +52,15 @@ async def async_setup_entry(
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, entry: ForecastSolarConfigEntry
+) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
-async def async_update_options(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def async_update_options(
+ hass: HomeAssistant, entry: ForecastSolarConfigEntry
+) -> None:
"""Update options."""
await hass.config_entries.async_reload(entry.entry_id)
diff --git a/homeassistant/components/forecast_solar/coordinator.py b/homeassistant/components/forecast_solar/coordinator.py
index c9c062a0c88..efed954e490 100644
--- a/homeassistant/components/forecast_solar/coordinator.py
+++ b/homeassistant/components/forecast_solar/coordinator.py
@@ -23,15 +23,16 @@ from .const import (
LOGGER,
)
+type ForecastSolarConfigEntry = ConfigEntry[ForecastSolarDataUpdateCoordinator]
+
class ForecastSolarDataUpdateCoordinator(DataUpdateCoordinator[Estimate]):
"""The Forecast.Solar Data Update Coordinator."""
- config_entry: ConfigEntry
+ config_entry: ForecastSolarConfigEntry
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
+ def __init__(self, hass: HomeAssistant, entry: ForecastSolarConfigEntry) -> None:
"""Initialize the Forecast.Solar coordinator."""
- self.config_entry = entry
# Our option flow may cause it to be an empty string,
# this if statement is here to catch that.
@@ -61,7 +62,13 @@ class ForecastSolarDataUpdateCoordinator(DataUpdateCoordinator[Estimate]):
if api_key is not None:
update_interval = timedelta(minutes=30)
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=update_interval)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=entry,
+ name=DOMAIN,
+ update_interval=update_interval,
+ )
async def _async_update_data(self) -> Estimate:
"""Fetch Forecast.Solar estimates."""
diff --git a/homeassistant/components/forecast_solar/sensor.py b/homeassistant/components/forecast_solar/sensor.py
index c1fa971a89d..13a4d5c2d23 100644
--- a/homeassistant/components/forecast_solar/sensor.py
+++ b/homeassistant/components/forecast_solar/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -135,7 +135,7 @@ SENSORS: tuple[ForecastSolarSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ForecastSolarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/forked_daapd/__init__.py b/homeassistant/components/forked_daapd/__init__.py
index 2172e60ba38..844a6a3eff9 100644
--- a/homeassistant/components/forked_daapd/__init__.py
+++ b/homeassistant/components/forked_daapd/__init__.py
@@ -1,29 +1,36 @@
"""The forked_daapd component."""
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import Platform
-from homeassistant.core import HomeAssistant
+from pyforked_daapd import ForkedDaapdAPI
-from .const import DOMAIN, HASS_DATA_UPDATER_KEY
+from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT, Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+from .coordinator import ForkedDaapdConfigEntry, ForkedDaapdUpdater
PLATFORMS = [Platform.MEDIA_PLAYER]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: ForkedDaapdConfigEntry) -> bool:
"""Set up forked-daapd from a config entry by forwarding to platform."""
+ host: str = entry.data[CONF_HOST]
+ port: int = entry.data[CONF_PORT]
+ password: str = entry.data[CONF_PASSWORD]
+ forked_daapd_api = ForkedDaapdAPI(
+ async_get_clientsession(hass), host, port, password
+ )
+ forked_daapd_updater = ForkedDaapdUpdater(hass, forked_daapd_api, entry.entry_id)
+ entry.runtime_data = forked_daapd_updater
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, entry: ForkedDaapdConfigEntry
+) -> bool:
"""Remove forked-daapd component."""
status = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- if status and hass.data.get(DOMAIN) and hass.data[DOMAIN].get(entry.entry_id):
- if websocket_handler := hass.data[DOMAIN][entry.entry_id][
- HASS_DATA_UPDATER_KEY
- ].websocket_handler:
+ if status:
+ if websocket_handler := entry.runtime_data.websocket_handler:
websocket_handler.cancel()
- del hass.data[DOMAIN][entry.entry_id]
- if not hass.data[DOMAIN]:
- del hass.data[DOMAIN]
return status
diff --git a/homeassistant/components/forked_daapd/config_flow.py b/homeassistant/components/forked_daapd/config_flow.py
index b2b2d498f60..890976c7503 100644
--- a/homeassistant/components/forked_daapd/config_flow.py
+++ b/homeassistant/components/forked_daapd/config_flow.py
@@ -7,12 +7,7 @@ from typing import Any
from pyforked_daapd import ForkedDaapdAPI
import voluptuous as vol
-from homeassistant.config_entries import (
- ConfigEntry,
- ConfigFlow,
- ConfigFlowResult,
- OptionsFlow,
-)
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult, OptionsFlow
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, CONF_PORT
from homeassistant.core import callback
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -28,6 +23,7 @@ from .const import (
DEFAULT_TTS_VOLUME,
DOMAIN,
)
+from .coordinator import ForkedDaapdConfigEntry
_LOGGER = logging.getLogger(__name__)
@@ -115,7 +111,7 @@ class ForkedDaapdFlowHandler(ConfigFlow, domain=DOMAIN):
@staticmethod
@callback
def async_get_options_flow(
- config_entry: ConfigEntry,
+ config_entry: ForkedDaapdConfigEntry,
) -> ForkedDaapdOptionsFlowHandler:
"""Return options flow handler."""
return ForkedDaapdOptionsFlowHandler()
diff --git a/homeassistant/components/forked_daapd/const.py b/homeassistant/components/forked_daapd/const.py
index dd7ed1bdf16..effd4c9454c 100644
--- a/homeassistant/components/forked_daapd/const.py
+++ b/homeassistant/components/forked_daapd/const.py
@@ -30,9 +30,8 @@ DEFAULT_SERVER_NAME = "My Server"
DEFAULT_TTS_PAUSE_TIME = 1.2
DEFAULT_TTS_VOLUME = 0.8
DEFAULT_UNMUTE_VOLUME = 0.6
-DOMAIN = "forked_daapd" # key for hass.data
+DOMAIN = "forked_daapd"
FD_NAME = "OwnTone"
-HASS_DATA_UPDATER_KEY = "UPDATER"
KNOWN_PIPES = {"librespot-java"}
PIPE_FUNCTION_MAP = {
"librespot-java": {
diff --git a/homeassistant/components/forked_daapd/coordinator.py b/homeassistant/components/forked_daapd/coordinator.py
new file mode 100644
index 00000000000..0ba339be505
--- /dev/null
+++ b/homeassistant/components/forked_daapd/coordinator.py
@@ -0,0 +1,155 @@
+"""Support forked_daapd media player."""
+
+from __future__ import annotations
+
+import asyncio
+from collections.abc import Sequence
+import logging
+from typing import Any
+
+from pyforked_daapd import ForkedDaapdAPI
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import PlatformNotReady
+from homeassistant.helpers.dispatcher import async_dispatcher_send
+
+from .const import (
+ SIGNAL_ADD_ZONES,
+ SIGNAL_UPDATE_DATABASE,
+ SIGNAL_UPDATE_MASTER,
+ SIGNAL_UPDATE_OUTPUTS,
+ SIGNAL_UPDATE_PLAYER,
+ SIGNAL_UPDATE_QUEUE,
+)
+
+type ForkedDaapdConfigEntry = ConfigEntry[ForkedDaapdUpdater]
+
+_LOGGER = logging.getLogger(__name__)
+
+WS_NOTIFY_EVENT_TYPES = ["player", "outputs", "volume", "options", "queue", "database"]
+WEBSOCKET_RECONNECT_TIME = 30 # seconds
+
+
+class ForkedDaapdUpdater:
+ """Manage updates for the forked-daapd device."""
+
+ def __init__(self, hass: HomeAssistant, api: ForkedDaapdAPI, entry_id: str) -> None:
+ """Initialize."""
+ self.hass = hass
+ self._api = api
+ self.websocket_handler: asyncio.Task[None] | None = None
+ self._all_output_ids: set[str] = set()
+ self._entry_id = entry_id
+
+ @property
+ def api(self) -> ForkedDaapdAPI:
+ """Return the API object."""
+ return self._api
+
+ async def async_init(self) -> None:
+ """Perform async portion of class initialization."""
+ if not (server_config := await self._api.get_request("config")):
+ raise PlatformNotReady
+ if websocket_port := server_config.get("websocket_port"):
+ self.websocket_handler = asyncio.create_task(
+ self._api.start_websocket_handler(
+ websocket_port,
+ WS_NOTIFY_EVENT_TYPES,
+ self._update,
+ WEBSOCKET_RECONNECT_TIME,
+ self._disconnected_callback,
+ )
+ )
+ else:
+ _LOGGER.error("Invalid websocket port")
+
+ async def _disconnected_callback(self) -> None:
+ """Send update signals when the websocket gets disconnected."""
+ async_dispatcher_send(
+ self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), False
+ )
+ async_dispatcher_send(
+ self.hass, SIGNAL_UPDATE_OUTPUTS.format(self._entry_id), []
+ )
+
+ async def _update(self, update_types_sequence: Sequence[str]) -> None:
+ """Private update method."""
+ update_types = set(update_types_sequence)
+ update_events = {}
+ _LOGGER.debug("Updating %s", update_types)
+ if (
+ "queue" in update_types
+ ): # update queue, queue before player for async_play_media
+ if queue := await self._api.get_request("queue"):
+ update_events["queue"] = asyncio.Event()
+ async_dispatcher_send(
+ self.hass,
+ SIGNAL_UPDATE_QUEUE.format(self._entry_id),
+ queue,
+ update_events["queue"],
+ )
+ # order of below don't matter
+ if not {"outputs", "volume"}.isdisjoint(update_types): # update outputs
+ if outputs := await self._api.get_request("outputs"):
+ outputs = outputs["outputs"]
+ update_events["outputs"] = (
+ asyncio.Event()
+ ) # only for master, zones should ignore
+ async_dispatcher_send(
+ self.hass,
+ SIGNAL_UPDATE_OUTPUTS.format(self._entry_id),
+ outputs,
+ update_events["outputs"],
+ )
+ self._add_zones(outputs)
+ if not {"database"}.isdisjoint(update_types):
+ pipes, playlists = await asyncio.gather(
+ self._api.get_pipes(), self._api.get_playlists()
+ )
+ update_events["database"] = asyncio.Event()
+ async_dispatcher_send(
+ self.hass,
+ SIGNAL_UPDATE_DATABASE.format(self._entry_id),
+ pipes,
+ playlists,
+ update_events["database"],
+ )
+ if not {"update", "config"}.isdisjoint(update_types): # not supported
+ _LOGGER.debug("update/config notifications neither requested nor supported")
+ if not {"player", "options", "volume"}.isdisjoint(
+ update_types
+ ): # update player
+ if player := await self._api.get_request("player"):
+ update_events["player"] = asyncio.Event()
+ if update_events.get("queue"):
+ await update_events[
+ "queue"
+ ].wait() # make sure queue done before player for async_play_media
+ async_dispatcher_send(
+ self.hass,
+ SIGNAL_UPDATE_PLAYER.format(self._entry_id),
+ player,
+ update_events["player"],
+ )
+ if update_events:
+ await asyncio.wait(
+ [asyncio.create_task(event.wait()) for event in update_events.values()]
+ ) # make sure callbacks done before update
+ async_dispatcher_send(
+ self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), True
+ )
+
+ def _add_zones(self, outputs: list[dict[str, Any]]) -> None:
+ outputs_to_add: list[dict[str, Any]] = []
+ for output in outputs:
+ if output["id"] not in self._all_output_ids:
+ self._all_output_ids.add(output["id"])
+ outputs_to_add.append(output)
+ if outputs_to_add:
+ async_dispatcher_send(
+ self.hass,
+ SIGNAL_ADD_ZONES.format(self._entry_id),
+ self._api,
+ outputs_to_add,
+ )
diff --git a/homeassistant/components/forked_daapd/media_player.py b/homeassistant/components/forked_daapd/media_player.py
index 0116cc57e7b..fd5390195a6 100644
--- a/homeassistant/components/forked_daapd/media_player.py
+++ b/homeassistant/components/forked_daapd/media_player.py
@@ -7,7 +7,6 @@ from collections import defaultdict
import logging
from typing import Any
-from pyforked_daapd import ForkedDaapdAPI
from pylibrespot_java import LibrespotJavaAPI
from homeassistant.components import media_source
@@ -28,16 +27,14 @@ from homeassistant.components.spotify import (
resolve_spotify_media_type,
spotify_uri_from_media_browser_url,
)
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT
+from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant, callback
-from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from .browse_media import (
@@ -56,9 +53,7 @@ from .const import (
DEFAULT_TTS_PAUSE_TIME,
DEFAULT_TTS_VOLUME,
DEFAULT_UNMUTE_VOLUME,
- DOMAIN,
FD_NAME,
- HASS_DATA_UPDATER_KEY,
KNOWN_PIPES,
PIPE_FUNCTION_MAP,
SIGNAL_ADD_ZONES,
@@ -75,31 +70,25 @@ from .const import (
SUPPORTED_FEATURES_ZONE,
TTS_TIMEOUT,
)
+from .coordinator import ForkedDaapdConfigEntry
_LOGGER = logging.getLogger(__name__)
-WS_NOTIFY_EVENT_TYPES = ["player", "outputs", "volume", "options", "queue", "database"]
-WEBSOCKET_RECONNECT_TIME = 30 # seconds
-
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: ForkedDaapdConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up forked-daapd from a config entry."""
- host = config_entry.data[CONF_HOST]
- port = config_entry.data[CONF_PORT]
- password = config_entry.data[CONF_PASSWORD]
- forked_daapd_api = ForkedDaapdAPI(
- async_get_clientsession(hass), host, port, password
- )
+ forked_daapd_updater = config_entry.runtime_data
+
+ host: str = config_entry.data[CONF_HOST]
+ forked_daapd_api = forked_daapd_updater.api
forked_daapd_master = ForkedDaapdMaster(
clientsession=async_get_clientsession(hass),
api=forked_daapd_api,
ip_address=host,
- api_port=port,
- api_password=password,
config_entry=config_entry,
)
@@ -116,20 +105,12 @@ async def async_setup_entry(
)
config_entry.async_on_unload(config_entry.add_update_listener(update_listener))
- if not hass.data.get(DOMAIN):
- hass.data[DOMAIN] = {config_entry.entry_id: {}}
-
async_add_entities([forked_daapd_master], False)
- forked_daapd_updater = ForkedDaapdUpdater(
- hass, forked_daapd_api, config_entry.entry_id
- )
- hass.data[DOMAIN][config_entry.entry_id][HASS_DATA_UPDATER_KEY] = (
- forked_daapd_updater
- )
+
await forked_daapd_updater.async_init()
-async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def update_listener(hass: HomeAssistant, entry: ForkedDaapdConfigEntry) -> None:
"""Handle options update."""
async_dispatcher_send(
hass, SIGNAL_CONFIG_OPTIONS_UPDATE.format(entry.entry_id), entry.options
@@ -243,9 +224,7 @@ class ForkedDaapdMaster(MediaPlayerEntity):
_attr_should_poll = False
- def __init__(
- self, clientsession, api, ip_address, api_port, api_password, config_entry
- ):
+ def __init__(self, clientsession, api, ip_address, config_entry):
"""Initialize the ForkedDaapd Master Device."""
# Leave the api public so the browse media helpers can use it
self.api = api
@@ -272,7 +251,7 @@ class ForkedDaapdMaster(MediaPlayerEntity):
self._on_remove = None
self._available = False
self._clientsession = clientsession
- self._config_entry = config_entry
+ self._entry_id = config_entry.entry_id
self.update_options(config_entry.options)
self._paused_event = asyncio.Event()
self._pause_requested = False
@@ -285,42 +264,42 @@ class ForkedDaapdMaster(MediaPlayerEntity):
self.async_on_remove(
async_dispatcher_connect(
self.hass,
- SIGNAL_UPDATE_PLAYER.format(self._config_entry.entry_id),
+ SIGNAL_UPDATE_PLAYER.format(self._entry_id),
self._update_player,
)
)
self.async_on_remove(
async_dispatcher_connect(
self.hass,
- SIGNAL_UPDATE_QUEUE.format(self._config_entry.entry_id),
+ SIGNAL_UPDATE_QUEUE.format(self._entry_id),
self._update_queue,
)
)
self.async_on_remove(
async_dispatcher_connect(
self.hass,
- SIGNAL_UPDATE_OUTPUTS.format(self._config_entry.entry_id),
+ SIGNAL_UPDATE_OUTPUTS.format(self._entry_id),
self._update_outputs,
)
)
self.async_on_remove(
async_dispatcher_connect(
self.hass,
- SIGNAL_UPDATE_MASTER.format(self._config_entry.entry_id),
+ SIGNAL_UPDATE_MASTER.format(self._entry_id),
self._update_callback,
)
)
self.async_on_remove(
async_dispatcher_connect(
self.hass,
- SIGNAL_CONFIG_OPTIONS_UPDATE.format(self._config_entry.entry_id),
+ SIGNAL_CONFIG_OPTIONS_UPDATE.format(self._entry_id),
self.update_options,
)
)
self.async_on_remove(
async_dispatcher_connect(
self.hass,
- SIGNAL_UPDATE_DATABASE.format(self._config_entry.entry_id),
+ SIGNAL_UPDATE_DATABASE.format(self._entry_id),
self._update_database,
)
)
@@ -414,9 +393,9 @@ class ForkedDaapdMaster(MediaPlayerEntity):
self._track_info = defaultdict(str)
@property
- def unique_id(self):
+ def unique_id(self) -> str:
"""Return unique ID."""
- return self._config_entry.entry_id
+ return self._entry_id
@property
def available(self) -> bool:
@@ -897,122 +876,3 @@ class ForkedDaapdMaster(MediaPlayerEntity):
if url := result.get("artwork_url"):
return await self._async_fetch_image(self.api.full_url(url))
return None, None
-
-
-class ForkedDaapdUpdater:
- """Manage updates for the forked-daapd device."""
-
- def __init__(self, hass, api, entry_id):
- """Initialize."""
- self.hass = hass
- self._api = api
- self.websocket_handler = None
- self._all_output_ids = set()
- self._entry_id = entry_id
-
- async def async_init(self):
- """Perform async portion of class initialization."""
- if not (server_config := await self._api.get_request("config")):
- raise PlatformNotReady
- if websocket_port := server_config.get("websocket_port"):
- self.websocket_handler = asyncio.create_task(
- self._api.start_websocket_handler(
- websocket_port,
- WS_NOTIFY_EVENT_TYPES,
- self._update,
- WEBSOCKET_RECONNECT_TIME,
- self._disconnected_callback,
- )
- )
- else:
- _LOGGER.error("Invalid websocket port")
-
- async def _disconnected_callback(self):
- """Send update signals when the websocket gets disconnected."""
- async_dispatcher_send(
- self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), False
- )
- async_dispatcher_send(
- self.hass, SIGNAL_UPDATE_OUTPUTS.format(self._entry_id), []
- )
-
- async def _update(self, update_types):
- """Private update method."""
- update_types = set(update_types)
- update_events = {}
- _LOGGER.debug("Updating %s", update_types)
- if (
- "queue" in update_types
- ): # update queue, queue before player for async_play_media
- if queue := await self._api.get_request("queue"):
- update_events["queue"] = asyncio.Event()
- async_dispatcher_send(
- self.hass,
- SIGNAL_UPDATE_QUEUE.format(self._entry_id),
- queue,
- update_events["queue"],
- )
- # order of below don't matter
- if not {"outputs", "volume"}.isdisjoint(update_types): # update outputs
- if outputs := await self._api.get_request("outputs"):
- outputs = outputs["outputs"]
- update_events["outputs"] = (
- asyncio.Event()
- ) # only for master, zones should ignore
- async_dispatcher_send(
- self.hass,
- SIGNAL_UPDATE_OUTPUTS.format(self._entry_id),
- outputs,
- update_events["outputs"],
- )
- self._add_zones(outputs)
- if not {"database"}.isdisjoint(update_types):
- pipes, playlists = await asyncio.gather(
- self._api.get_pipes(), self._api.get_playlists()
- )
- update_events["database"] = asyncio.Event()
- async_dispatcher_send(
- self.hass,
- SIGNAL_UPDATE_DATABASE.format(self._entry_id),
- pipes,
- playlists,
- update_events["database"],
- )
- if not {"update", "config"}.isdisjoint(update_types): # not supported
- _LOGGER.debug("update/config notifications neither requested nor supported")
- if not {"player", "options", "volume"}.isdisjoint(
- update_types
- ): # update player
- if player := await self._api.get_request("player"):
- update_events["player"] = asyncio.Event()
- if update_events.get("queue"):
- await update_events[
- "queue"
- ].wait() # make sure queue done before player for async_play_media
- async_dispatcher_send(
- self.hass,
- SIGNAL_UPDATE_PLAYER.format(self._entry_id),
- player,
- update_events["player"],
- )
- if update_events:
- await asyncio.wait(
- [asyncio.create_task(event.wait()) for event in update_events.values()]
- ) # make sure callbacks done before update
- async_dispatcher_send(
- self.hass, SIGNAL_UPDATE_MASTER.format(self._entry_id), True
- )
-
- def _add_zones(self, outputs):
- outputs_to_add = []
- for output in outputs:
- if output["id"] not in self._all_output_ids:
- self._all_output_ids.add(output["id"])
- outputs_to_add.append(output)
- if outputs_to_add:
- async_dispatcher_send(
- self.hass,
- SIGNAL_ADD_ZONES.format(self._entry_id),
- self._api,
- outputs_to_add,
- )
diff --git a/homeassistant/components/foscam/camera.py b/homeassistant/components/foscam/camera.py
index ed5ba1d4c21..353c7397d81 100644
--- a/homeassistant/components/foscam/camera.py
+++ b/homeassistant/components/foscam/camera.py
@@ -10,7 +10,7 @@ from homeassistant.components.camera import Camera, CameraEntityFeature
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_RTSP_PORT, CONF_STREAM, LOGGER, SERVICE_PTZ, SERVICE_PTZ_PRESET
from .coordinator import FoscamConfigEntry, FoscamCoordinator
@@ -49,7 +49,7 @@ PTZ_GOTO_PRESET_COMMAND = "ptz_goto_preset"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FoscamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a Foscam IP camera from a config entry."""
platform = entity_platform.async_get_current_platform()
diff --git a/homeassistant/components/foscam/strings.json b/homeassistant/components/foscam/strings.json
index 2784e541809..03351e3238f 100644
--- a/homeassistant/components/foscam/strings.json
+++ b/homeassistant/components/foscam/strings.json
@@ -35,7 +35,7 @@
"services": {
"ptz": {
"name": "PTZ",
- "description": "Pan/Tilt action for Foscam camera.",
+ "description": "Moves a Foscam camera to a specified direction.",
"fields": {
"movement": {
"name": "Movement",
@@ -49,7 +49,7 @@
},
"ptz_preset": {
"name": "PTZ preset",
- "description": "PTZ Preset action for Foscam camera.",
+ "description": "Moves a Foscam camera to a predefined position.",
"fields": {
"preset_name": {
"name": "Preset name",
diff --git a/homeassistant/components/foscam/switch.py b/homeassistant/components/foscam/switch.py
index 189271d2746..24b05b5aeaa 100644
--- a/homeassistant/components/foscam/switch.py
+++ b/homeassistant/components/foscam/switch.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import LOGGER
from .coordinator import FoscamConfigEntry, FoscamCoordinator
@@ -17,7 +17,7 @@ from .entity import FoscamEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FoscamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up foscam switch from a config entry."""
diff --git a/homeassistant/components/frankever/__init__.py b/homeassistant/components/frankever/__init__.py
new file mode 100644
index 00000000000..66eeecb1e59
--- /dev/null
+++ b/homeassistant/components/frankever/__init__.py
@@ -0,0 +1 @@
+"""FrankEver virtual integration."""
diff --git a/homeassistant/components/frankever/manifest.json b/homeassistant/components/frankever/manifest.json
new file mode 100644
index 00000000000..37d7be765ef
--- /dev/null
+++ b/homeassistant/components/frankever/manifest.json
@@ -0,0 +1,6 @@
+{
+ "domain": "frankever",
+ "name": "FrankEver",
+ "integration_type": "virtual",
+ "supported_by": "shelly"
+}
diff --git a/homeassistant/components/freebox/alarm_control_panel.py b/homeassistant/components/freebox/alarm_control_panel.py
index 9d8e85a14ca..89462b33a2f 100644
--- a/homeassistant/components/freebox/alarm_control_panel.py
+++ b/homeassistant/components/freebox/alarm_control_panel.py
@@ -9,7 +9,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, FreeboxHomeCategory
from .entity import FreeboxHomeEntity
@@ -28,7 +28,9 @@ FREEBOX_TO_STATUS = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up alarm panel."""
router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
diff --git a/homeassistant/components/freebox/binary_sensor.py b/homeassistant/components/freebox/binary_sensor.py
index 20c124efea6..9fc9929b869 100644
--- a/homeassistant/components/freebox/binary_sensor.py
+++ b/homeassistant/components/freebox/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, FreeboxHomeCategory
from .entity import FreeboxHomeEntity
@@ -34,7 +34,9 @@ RAID_SENSORS: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors."""
router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
diff --git a/homeassistant/components/freebox/button.py b/homeassistant/components/freebox/button.py
index 79e3c98b8b7..4f676fd46a1 100644
--- a/homeassistant/components/freebox/button.py
+++ b/homeassistant/components/freebox/button.py
@@ -13,7 +13,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .router import FreeboxRouter
@@ -44,7 +44,9 @@ BUTTON_DESCRIPTIONS: tuple[FreeboxButtonEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the buttons."""
router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
diff --git a/homeassistant/components/freebox/camera.py b/homeassistant/components/freebox/camera.py
index 33919df74f6..45bb5a34063 100644
--- a/homeassistant/components/freebox/camera.py
+++ b/homeassistant/components/freebox/camera.py
@@ -17,7 +17,7 @@ from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_DETECTION, DOMAIN, FreeboxHomeCategory
from .entity import FreeboxHomeEntity
@@ -27,7 +27,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up cameras."""
router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
@@ -49,7 +51,7 @@ async def async_setup_entry(
def add_entities(
hass: HomeAssistant,
router: FreeboxRouter,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
tracked: set[str],
) -> None:
"""Add new cameras from the router."""
diff --git a/homeassistant/components/freebox/device_tracker.py b/homeassistant/components/freebox/device_tracker.py
index 1fa37ebc270..dcb6eb104b2 100644
--- a/homeassistant/components/freebox/device_tracker.py
+++ b/homeassistant/components/freebox/device_tracker.py
@@ -9,14 +9,16 @@ from homeassistant.components.device_tracker import ScannerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DEFAULT_DEVICE_NAME, DEVICE_ICONS, DOMAIN
from .router import FreeboxRouter
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Freebox component."""
router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
@@ -36,7 +38,9 @@ async def async_setup_entry(
@callback
def add_entities(
- router: FreeboxRouter, async_add_entities: AddEntitiesCallback, tracked: set[str]
+ router: FreeboxRouter,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+ tracked: set[str],
) -> None:
"""Add new tracker entities from the router."""
new_tracked = []
diff --git a/homeassistant/components/freebox/sensor.py b/homeassistant/components/freebox/sensor.py
index 588992a7f21..cc62de9ae0d 100644
--- a/homeassistant/components/freebox/sensor.py
+++ b/homeassistant/components/freebox/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.const import PERCENTAGE, UnitOfDataRate, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN
@@ -60,7 +60,9 @@ DISK_PARTITION_SENSORS: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
diff --git a/homeassistant/components/freebox/switch.py b/homeassistant/components/freebox/switch.py
index 96c3bcc2496..c4618b014bf 100644
--- a/homeassistant/components/freebox/switch.py
+++ b/homeassistant/components/freebox/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .router import FreeboxRouter
@@ -29,7 +29,9 @@ SWITCH_DESCRIPTIONS = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switch."""
router: FreeboxRouter = hass.data[DOMAIN][entry.unique_id]
diff --git a/homeassistant/components/freedompro/binary_sensor.py b/homeassistant/components/freedompro/binary_sensor.py
index 840150e807d..9ff62446176 100644
--- a/homeassistant/components/freedompro/binary_sensor.py
+++ b/homeassistant/components/freedompro/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -34,7 +34,7 @@ SUPPORTED_SENSORS = {"smokeSensor", "occupancySensor", "motionSensor", "contactS
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro binary_sensor."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/freedompro/climate.py b/homeassistant/components/freedompro/climate.py
index a0146dc70b3..0145dea27bb 100644
--- a/homeassistant/components/freedompro/climate.py
+++ b/homeassistant/components/freedompro/climate.py
@@ -19,7 +19,7 @@ from homeassistant.const import ATTR_TEMPERATURE, CONF_API_KEY, UnitOfTemperatur
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -45,7 +45,7 @@ SUPPORTED_HVAC_MODES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro climate."""
api_key: str = entry.data[CONF_API_KEY]
diff --git a/homeassistant/components/freedompro/cover.py b/homeassistant/components/freedompro/cover.py
index ee61612428c..01e1b39d08f 100644
--- a/homeassistant/components/freedompro/cover.py
+++ b/homeassistant/components/freedompro/cover.py
@@ -15,7 +15,7 @@ from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -35,7 +35,7 @@ SUPPORTED_SENSORS = {"windowCovering", "gate", "garageDoor", "door", "window"}
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro cover."""
api_key: str = entry.data[CONF_API_KEY]
diff --git a/homeassistant/components/freedompro/fan.py b/homeassistant/components/freedompro/fan.py
index ad520ac8eb8..c65afb3a0e2 100644
--- a/homeassistant/components/freedompro/fan.py
+++ b/homeassistant/components/freedompro/fan.py
@@ -12,7 +12,7 @@ from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -22,7 +22,7 @@ from .coordinator import FreedomproConfigEntry, FreedomproDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro fan."""
api_key: str = entry.data[CONF_API_KEY]
diff --git a/homeassistant/components/freedompro/light.py b/homeassistant/components/freedompro/light.py
index c1b2e0ea17b..f9d90420c5d 100644
--- a/homeassistant/components/freedompro/light.py
+++ b/homeassistant/components/freedompro/light.py
@@ -17,7 +17,7 @@ from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -27,7 +27,7 @@ from .coordinator import FreedomproConfigEntry, FreedomproDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro light."""
api_key: str = entry.data[CONF_API_KEY]
diff --git a/homeassistant/components/freedompro/lock.py b/homeassistant/components/freedompro/lock.py
index 70423bb9514..4aee252abbe 100644
--- a/homeassistant/components/freedompro/lock.py
+++ b/homeassistant/components/freedompro/lock.py
@@ -10,7 +10,7 @@ from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -20,7 +20,7 @@ from .coordinator import FreedomproConfigEntry, FreedomproDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro lock."""
api_key: str = entry.data[CONF_API_KEY]
diff --git a/homeassistant/components/freedompro/sensor.py b/homeassistant/components/freedompro/sensor.py
index eaa96ac9fed..dbe1449d6e5 100644
--- a/homeassistant/components/freedompro/sensor.py
+++ b/homeassistant/components/freedompro/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import LIGHT_LUX, PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -42,7 +42,7 @@ SUPPORTED_SENSORS = {"temperatureSensor", "humiditySensor", "lightSensor"}
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro sensor."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/freedompro/switch.py b/homeassistant/components/freedompro/switch.py
index 12346825474..bda13b147b1 100644
--- a/homeassistant/components/freedompro/switch.py
+++ b/homeassistant/components/freedompro/switch.py
@@ -10,7 +10,7 @@ from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -20,7 +20,7 @@ from .coordinator import FreedomproConfigEntry, FreedomproDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: FreedomproConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Freedompro switch."""
api_key: str = entry.data[CONF_API_KEY]
diff --git a/homeassistant/components/fritz/binary_sensor.py b/homeassistant/components/fritz/binary_sensor.py
index 7553328a64c..6bc8bb571d4 100644
--- a/homeassistant/components/fritz/binary_sensor.py
+++ b/homeassistant/components/fritz/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ConnectionInfo, FritzConfigEntry
from .entity import FritzBoxBaseCoordinatorEntity, FritzEntityDescription
@@ -51,7 +51,7 @@ SENSOR_TYPES: tuple[FritzBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
_LOGGER.debug("Setting up FRITZ!Box binary sensors")
diff --git a/homeassistant/components/fritz/button.py b/homeassistant/components/fritz/button.py
index f3ffbe42099..74e8ab5e43e 100644
--- a/homeassistant/components/fritz/button.py
+++ b/homeassistant/components/fritz/button.py
@@ -16,7 +16,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import BUTTON_TYPE_WOL, CONNECTION_TYPE_LAN, DOMAIN, MeshRoles
from .coordinator import (
@@ -72,7 +72,7 @@ BUTTONS: Final = [
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set buttons for device."""
_LOGGER.debug("Setting up buttons")
diff --git a/homeassistant/components/fritz/coordinator.py b/homeassistant/components/fritz/coordinator.py
index 38d76c92871..d60232ec8ad 100644
--- a/homeassistant/components/fritz/coordinator.py
+++ b/homeassistant/components/fritz/coordinator.py
@@ -196,6 +196,7 @@ class FritzBoxTools(DataUpdateCoordinator[UpdateCoordinatorDataType]):
self.hass = hass
self.host = host
self.mesh_role = MeshRoles.NONE
+ self.mesh_wifi_uplink = False
self.device_conn_type: str | None = None
self.device_is_router: bool = False
self.password = password
@@ -610,6 +611,12 @@ class FritzBoxTools(DataUpdateCoordinator[UpdateCoordinatorDataType]):
ssid=interf.get("ssid", ""),
type=interf["type"],
)
+
+ if interf["type"].lower() == "wlan" and interf[
+ "name"
+ ].lower().startswith("uplink"):
+ self.mesh_wifi_uplink = True
+
if dr.format_mac(int_mac) == self.mac:
self.mesh_role = MeshRoles(node["mesh_role"])
diff --git a/homeassistant/components/fritz/device_tracker.py b/homeassistant/components/fritz/device_tracker.py
index ba3c9a5aab6..e066219342e 100644
--- a/homeassistant/components/fritz/device_tracker.py
+++ b/homeassistant/components/fritz/device_tracker.py
@@ -8,7 +8,7 @@ import logging
from homeassistant.components.device_tracker import ScannerEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import (
FRITZ_DATA_KEY,
@@ -26,7 +26,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for FRITZ!Box component."""
_LOGGER.debug("Starting FRITZ!Box device tracker")
@@ -48,7 +48,7 @@ async def async_setup_entry(
@callback
def _async_add_entities(
avm_wrapper: AvmWrapper,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
data_fritz: FritzData,
) -> None:
"""Add new tracker entities from the AVM device."""
diff --git a/homeassistant/components/fritz/image.py b/homeassistant/components/fritz/image.py
index d305551b097..d329ec318c5 100644
--- a/homeassistant/components/fritz/image.py
+++ b/homeassistant/components/fritz/image.py
@@ -10,7 +10,7 @@ from requests.exceptions import RequestException
from homeassistant.components.image import ImageEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util, slugify
from .coordinator import AvmWrapper, FritzConfigEntry
@@ -22,7 +22,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up guest WiFi QR code for device."""
avm_wrapper = entry.runtime_data
diff --git a/homeassistant/components/fritz/sensor.py b/homeassistant/components/fritz/sensor.py
index 81b50bd21ac..bcee590460f 100644
--- a/homeassistant/components/fritz/sensor.py
+++ b/homeassistant/components/fritz/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfInformation,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
@@ -268,7 +268,7 @@ SENSOR_TYPES: tuple[FritzSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
_LOGGER.debug("Setting up FRITZ!Box sensors")
diff --git a/homeassistant/components/fritz/switch.py b/homeassistant/components/fritz/switch.py
index 9c12fe0cecc..8b4816f7451 100644
--- a/homeassistant/components/fritz/switch.py
+++ b/homeassistant/components/fritz/switch.py
@@ -12,7 +12,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import slugify
@@ -207,8 +207,9 @@ async def async_all_entities_list(
local_ip: str,
) -> list[Entity]:
"""Get a list of all entities."""
-
if avm_wrapper.mesh_role == MeshRoles.SLAVE:
+ if not avm_wrapper.mesh_wifi_uplink:
+ return [*await _async_wifi_entities_list(avm_wrapper, device_friendly_name)]
return []
return [
@@ -222,7 +223,7 @@ async def async_all_entities_list(
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
_LOGGER.debug("Setting up switches")
@@ -565,6 +566,9 @@ class FritzBoxWifiSwitch(FritzBoxBaseSwitch):
self._attributes = {}
self._attr_entity_category = EntityCategory.CONFIG
+ self._attr_entity_registry_enabled_default = (
+ avm_wrapper.mesh_role is not MeshRoles.SLAVE
+ )
self._network_num = network_num
switch_info = SwitchInfo(
diff --git a/homeassistant/components/fritz/update.py b/homeassistant/components/fritz/update.py
index ad23a076ca6..5d064dc3035 100644
--- a/homeassistant/components/fritz/update.py
+++ b/homeassistant/components/fritz/update.py
@@ -13,7 +13,7 @@ from homeassistant.components.update import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import AvmWrapper, FritzConfigEntry
from .entity import FritzBoxBaseCoordinatorEntity, FritzEntityDescription
@@ -29,7 +29,7 @@ class FritzUpdateEntityDescription(UpdateEntityDescription, FritzEntityDescripti
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AVM FRITZ!Box update entities."""
_LOGGER.debug("Setting up AVM FRITZ!Box update entities")
diff --git a/homeassistant/components/fritzbox/binary_sensor.py b/homeassistant/components/fritzbox/binary_sensor.py
index 3c9cb6ada5c..75683017cb7 100644
--- a/homeassistant/components/fritzbox/binary_sensor.py
+++ b/homeassistant/components/fritzbox/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import FritzboxConfigEntry
from .entity import FritzBoxDeviceEntity
@@ -66,7 +66,7 @@ BINARY_SENSOR_TYPES: Final[tuple[FritzBinarySensorEntityDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FRITZ!SmartHome binary sensor from ConfigEntry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fritzbox/button.py b/homeassistant/components/fritzbox/button.py
index 44a6697e1c0..54baa97b11a 100644
--- a/homeassistant/components/fritzbox/button.py
+++ b/homeassistant/components/fritzbox/button.py
@@ -5,7 +5,7 @@ from pyfritzhome.devicetypes import FritzhomeTemplate
from homeassistant.components.button import ButtonEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import FritzboxConfigEntry
@@ -15,7 +15,7 @@ from .entity import FritzBoxEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FRITZ!SmartHome template from ConfigEntry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fritzbox/climate.py b/homeassistant/components/fritzbox/climate.py
index 87a87ac691f..118e03c391f 100644
--- a/homeassistant/components/fritzbox/climate.py
+++ b/homeassistant/components/fritzbox/climate.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_STATE_BATTERY_LOW,
@@ -59,7 +59,7 @@ OFF_REPORT_SET_TEMPERATURE = 0.0
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FRITZ!SmartHome thermostat from ConfigEntry."""
coordinator = entry.runtime_data
@@ -85,6 +85,8 @@ async def async_setup_entry(
class FritzboxThermostat(FritzBoxDeviceEntity, ClimateEntity):
"""The thermostat class for FRITZ!SmartHome thermostats."""
+ _attr_max_temp = MAX_TEMPERATURE
+ _attr_min_temp = MIN_TEMPERATURE
_attr_precision = PRECISION_HALVES
_attr_temperature_unit = UnitOfTemperature.CELSIUS
_attr_translation_key = "thermostat"
@@ -135,11 +137,13 @@ class FritzboxThermostat(FritzBoxDeviceEntity, ClimateEntity):
async def async_set_temperature(self, **kwargs: Any) -> None:
"""Set new target temperature."""
- target_temp = kwargs.get(ATTR_TEMPERATURE)
- hvac_mode = kwargs.get(ATTR_HVAC_MODE)
- if hvac_mode == HVACMode.OFF:
+ if (hvac_mode := kwargs.get(ATTR_HVAC_MODE)) is HVACMode.OFF:
await self.async_set_hvac_mode(hvac_mode)
- elif target_temp is not None:
+ elif (target_temp := kwargs.get(ATTR_TEMPERATURE)) is not None:
+ if target_temp == OFF_API_TEMPERATURE:
+ target_temp = OFF_REPORT_SET_TEMPERATURE
+ elif target_temp == ON_API_TEMPERATURE:
+ target_temp = ON_REPORT_SET_TEMPERATURE
await self.hass.async_add_executor_job(
self.data.set_target_temperature, target_temp, True
)
@@ -169,12 +173,12 @@ class FritzboxThermostat(FritzBoxDeviceEntity, ClimateEntity):
translation_domain=DOMAIN,
translation_key="change_hvac_while_active_mode",
)
- if self.hvac_mode == hvac_mode:
+ if self.hvac_mode is hvac_mode:
LOGGER.debug(
"%s is already in requested hvac mode %s", self.name, hvac_mode
)
return
- if hvac_mode == HVACMode.OFF:
+ if hvac_mode is HVACMode.OFF:
await self.async_set_temperature(temperature=OFF_REPORT_SET_TEMPERATURE)
else:
if value_scheduled_preset(self.data) == PRESET_ECO:
@@ -208,16 +212,6 @@ class FritzboxThermostat(FritzBoxDeviceEntity, ClimateEntity):
elif preset_mode == PRESET_ECO:
await self.async_set_temperature(temperature=self.data.eco_temperature)
- @property
- def min_temp(self) -> int:
- """Return the minimum temperature."""
- return MIN_TEMPERATURE
-
- @property
- def max_temp(self) -> int:
- """Return the maximum temperature."""
- return MAX_TEMPERATURE
-
@property
def extra_state_attributes(self) -> ClimateExtraAttributes:
"""Return the device specific state attributes."""
diff --git a/homeassistant/components/fritzbox/cover.py b/homeassistant/components/fritzbox/cover.py
index 070bb868298..c7ecfef6a32 100644
--- a/homeassistant/components/fritzbox/cover.py
+++ b/homeassistant/components/fritzbox/cover.py
@@ -11,7 +11,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import FritzboxConfigEntry
from .entity import FritzBoxDeviceEntity
@@ -20,7 +20,7 @@ from .entity import FritzBoxDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FRITZ!SmartHome cover from ConfigEntry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fritzbox/light.py b/homeassistant/components/fritzbox/light.py
index 94d7d320704..8603840630c 100644
--- a/homeassistant/components/fritzbox/light.py
+++ b/homeassistant/components/fritzbox/light.py
@@ -12,7 +12,7 @@ from homeassistant.components.light import (
LightEntity,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import COLOR_MODE, LOGGER
from .coordinator import FritzboxConfigEntry, FritzboxDataUpdateCoordinator
@@ -22,7 +22,7 @@ from .entity import FritzBoxDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FRITZ!SmartHome light from ConfigEntry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fritzbox/manifest.json b/homeassistant/components/fritzbox/manifest.json
index 2fbb75443b2..f6155024cbf 100644
--- a/homeassistant/components/fritzbox/manifest.json
+++ b/homeassistant/components/fritzbox/manifest.json
@@ -7,7 +7,7 @@
"integration_type": "hub",
"iot_class": "local_polling",
"loggers": ["pyfritzhome"],
- "requirements": ["pyfritzhome==0.6.14"],
+ "requirements": ["pyfritzhome==0.6.17"],
"ssdp": [
{
"st": "urn:schemas-upnp-org:device:fritzbox:1"
diff --git a/homeassistant/components/fritzbox/sensor.py b/homeassistant/components/fritzbox/sensor.py
index e610fd80f3e..bed7004bd6a 100644
--- a/homeassistant/components/fritzbox/sensor.py
+++ b/homeassistant/components/fritzbox/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utc_from_timestamp
@@ -229,7 +229,7 @@ SENSOR_TYPES: Final[tuple[FritzSensorEntityDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FRITZ!SmartHome sensor from ConfigEntry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fritzbox/strings.json b/homeassistant/components/fritzbox/strings.json
index c7c2439b566..e0df30875bc 100644
--- a/homeassistant/components/fritzbox/strings.json
+++ b/homeassistant/components/fritzbox/strings.json
@@ -89,7 +89,7 @@
"message": "Can't change preset while holiday or summer mode is active on the device."
},
"change_hvac_while_active_mode": {
- "message": "Can't change hvac mode while holiday or summer mode is active on the device."
+ "message": "Can't change HVAC mode while holiday or summer mode is active on the device."
}
}
}
diff --git a/homeassistant/components/fritzbox/switch.py b/homeassistant/components/fritzbox/switch.py
index d83793c77dc..c2679ef5243 100644
--- a/homeassistant/components/fritzbox/switch.py
+++ b/homeassistant/components/fritzbox/switch.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import FritzboxConfigEntry
@@ -17,7 +17,7 @@ from .entity import FritzBoxDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FritzboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FRITZ!SmartHome switch from ConfigEntry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fritzbox_callmonitor/sensor.py b/homeassistant/components/fritzbox_callmonitor/sensor.py
index df18ae5702a..574ae9ef7f2 100644
--- a/homeassistant/components/fritzbox_callmonitor/sensor.py
+++ b/homeassistant/components/fritzbox_callmonitor/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import CONF_HOST, CONF_PORT, EVENT_HOMEASSISTANT_STOP
from homeassistant.core import Event, HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FritzBoxCallMonitorConfigEntry
from .base import Contact, FritzBoxPhonebook
@@ -48,7 +48,7 @@ class CallState(StrEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FritzBoxCallMonitorConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the fritzbox_callmonitor sensor from config_entry."""
fritzbox_phonebook = config_entry.runtime_data
diff --git a/homeassistant/components/fronius/manifest.json b/homeassistant/components/fronius/manifest.json
index 94d0f90b0bd..661d808ad23 100644
--- a/homeassistant/components/fronius/manifest.json
+++ b/homeassistant/components/fronius/manifest.json
@@ -11,6 +11,6 @@
"documentation": "https://www.home-assistant.io/integrations/fronius",
"iot_class": "local_polling",
"loggers": ["pyfronius"],
- "quality_scale": "gold",
- "requirements": ["PyFronius==0.7.3"]
+ "quality_scale": "platinum",
+ "requirements": ["PyFronius==0.7.7"]
}
diff --git a/homeassistant/components/fronius/quality_scale.yaml b/homeassistant/components/fronius/quality_scale.yaml
index 2c4b892475b..522b8ab571f 100644
--- a/homeassistant/components/fronius/quality_scale.yaml
+++ b/homeassistant/components/fronius/quality_scale.yaml
@@ -83,7 +83,4 @@ rules:
# Platinum
async-dependency: done
inject-websession: done
- strict-typing:
- status: todo
- comment: |
- The pyfronius library isn't strictly typed and doesn't export type information.
+ strict-typing: done
diff --git a/homeassistant/components/fronius/sensor.py b/homeassistant/components/fronius/sensor.py
index c6c3ff4b602..c65f6072ba6 100644
--- a/homeassistant/components/fronius/sensor.py
+++ b/homeassistant/components/fronius/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -64,7 +64,7 @@ ENERGY_VOLT_AMPERE_REACTIVE_HOUR: Final = "varh"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FroniusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Fronius sensor entities based on a config entry."""
solar_net = config_entry.runtime_data
diff --git a/homeassistant/components/frontend/__init__.py b/homeassistant/components/frontend/__init__.py
index 6184d888004..9a0627f9f42 100644
--- a/homeassistant/components/frontend/__init__.py
+++ b/homeassistant/components/frontend/__init__.py
@@ -52,10 +52,9 @@ CONF_JS_VERSION = "javascript_version"
DEFAULT_THEME_COLOR = "#03A9F4"
-DATA_PANELS = "frontend_panels"
-DATA_JS_VERSION = "frontend_js_version"
-DATA_EXTRA_MODULE_URL = "frontend_extra_module_url"
-DATA_EXTRA_JS_URL_ES5 = "frontend_extra_js_url_es5"
+DATA_PANELS: HassKey[dict[str, Panel]] = HassKey("frontend_panels")
+DATA_EXTRA_MODULE_URL: HassKey[UrlManager] = HassKey("frontend_extra_module_url")
+DATA_EXTRA_JS_URL_ES5: HassKey[UrlManager] = HassKey("frontend_extra_js_url_es5")
DATA_WS_SUBSCRIBERS: HassKey[set[tuple[websocket_api.ActiveConnection, int]]] = HassKey(
"frontend_ws_subscribers"
@@ -64,8 +63,8 @@ DATA_WS_SUBSCRIBERS: HassKey[set[tuple[websocket_api.ActiveConnection, int]]] =
THEMES_STORAGE_KEY = f"{DOMAIN}_theme"
THEMES_STORAGE_VERSION = 1
THEMES_SAVE_DELAY = 60
-DATA_THEMES_STORE = "frontend_themes_store"
-DATA_THEMES = "frontend_themes"
+DATA_THEMES_STORE: HassKey[Store] = HassKey("frontend_themes_store")
+DATA_THEMES: HassKey[dict[str, Any]] = HassKey("frontend_themes")
DATA_DEFAULT_THEME = "frontend_default_theme"
DATA_DEFAULT_DARK_THEME = "frontend_default_dark_theme"
DEFAULT_THEME = "default"
@@ -242,7 +241,7 @@ class Panel:
sidebar_title: str | None = None
# Url to show the panel in the frontend
- frontend_url_path: str | None = None
+ frontend_url_path: str
# Config to pass to the webcomponent
config: dict[str, Any] | None = None
@@ -273,7 +272,7 @@ class Panel:
self.config_panel_domain = config_panel_domain
@callback
- def to_response(self) -> PanelRespons:
+ def to_response(self) -> PanelResponse:
"""Panel as dictionary."""
return {
"component_name": self.component_name,
@@ -631,7 +630,8 @@ class IndexView(web_urldispatcher.AbstractResource):
def get_info(self) -> dict[str, list[str]]: # type: ignore[override]
"""Return a dict with additional info useful for introspection."""
- return {"panels": list(self.hass.data[DATA_PANELS])}
+ panels = self.hass.data[DATA_PANELS]
+ return {"panels": list(panels)}
def raw_match(self, path: str) -> bool:
"""Perform a raw match against path."""
@@ -841,13 +841,13 @@ def websocket_subscribe_extra_js(
connection.send_message(websocket_api.result_message(msg["id"]))
-class PanelRespons(TypedDict):
+class PanelResponse(TypedDict):
"""Represent the panel response type."""
component_name: str
icon: str | None
title: str | None
config: dict[str, Any] | None
- url_path: str | None
+ url_path: str
require_admin: bool
config_panel_domain: str | None
diff --git a/homeassistant/components/frontend/manifest.json b/homeassistant/components/frontend/manifest.json
index d27785dcea5..b210fdb6661 100644
--- a/homeassistant/components/frontend/manifest.json
+++ b/homeassistant/components/frontend/manifest.json
@@ -1,7 +1,6 @@
{
"domain": "frontend",
"name": "Home Assistant Frontend",
- "after_dependencies": ["backup"],
"codeowners": ["@home-assistant/frontend"],
"dependencies": [
"api",
@@ -21,5 +20,5 @@
"documentation": "https://www.home-assistant.io/integrations/frontend",
"integration_type": "system",
"quality_scale": "internal",
- "requirements": ["home-assistant-frontend==20250205.0"]
+ "requirements": ["home-assistant-frontend==20250306.0"]
}
diff --git a/homeassistant/components/frontend/storage.py b/homeassistant/components/frontend/storage.py
index cbcc3024aa7..a33a9de7ac5 100644
--- a/homeassistant/components/frontend/storage.py
+++ b/homeassistant/components/frontend/storage.py
@@ -12,8 +12,11 @@ from homeassistant.components import websocket_api
from homeassistant.components.websocket_api import ActiveConnection
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.storage import Store
+from homeassistant.util.hass_dict import HassKey
-DATA_STORAGE = "frontend_storage"
+DATA_STORAGE: HassKey[tuple[dict[str, Store], dict[str, dict]]] = HassKey(
+ "frontend_storage"
+)
STORAGE_VERSION_USER_DATA = 1
diff --git a/homeassistant/components/frontier_silicon/media_player.py b/homeassistant/components/frontier_silicon/media_player.py
index 6b0f987baa2..4f5e55d1536 100644
--- a/homeassistant/components/frontier_silicon/media_player.py
+++ b/homeassistant/components/frontier_silicon/media_player.py
@@ -22,7 +22,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FrontierSiliconConfigEntry
from .browse_media import browse_node, browse_top_level
@@ -34,7 +34,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FrontierSiliconConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Frontier Silicon entity."""
diff --git a/homeassistant/components/fujitsu_fglair/__init__.py b/homeassistant/components/fujitsu_fglair/__init__.py
index 547545e4feb..699356a2e75 100644
--- a/homeassistant/components/fujitsu_fglair/__init__.py
+++ b/homeassistant/components/fujitsu_fglair/__init__.py
@@ -7,18 +7,15 @@ from contextlib import suppress
from ayla_iot_unofficial import new_ayla_api
from ayla_iot_unofficial.fujitsu_consts import FGLAIR_APP_CREDENTIALS
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import aiohttp_client
from .const import API_TIMEOUT, CONF_EUROPE, CONF_REGION, REGION_DEFAULT, REGION_EU
-from .coordinator import FGLairCoordinator
+from .coordinator import FGLairConfigEntry, FGLairCoordinator
PLATFORMS: list[Platform] = [Platform.CLIMATE, Platform.SENSOR]
-type FGLairConfigEntry = ConfigEntry[FGLairCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: FGLairConfigEntry) -> bool:
"""Set up Fujitsu HVAC (based on Ayla IOT) from a config entry."""
@@ -33,7 +30,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: FGLairConfigEntry) -> bo
timeout=API_TIMEOUT,
)
- coordinator = FGLairCoordinator(hass, api)
+ coordinator = FGLairCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/fujitsu_fglair/climate.py b/homeassistant/components/fujitsu_fglair/climate.py
index c0f5ab7dce4..bf1df07823c 100644
--- a/homeassistant/components/fujitsu_fglair/climate.py
+++ b/homeassistant/components/fujitsu_fglair/climate.py
@@ -25,10 +25,9 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_HALVES, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FGLairConfigEntry
-from .coordinator import FGLairCoordinator
+from .coordinator import FGLairConfigEntry, FGLairCoordinator
from .entity import FGLairEntity
HA_TO_FUJI_FAN = {
@@ -61,7 +60,7 @@ FUJI_TO_HA_SWING = {value: key for key, value in HA_TO_FUJI_SWING.items()}
async def async_setup_entry(
hass: HomeAssistant,
entry: FGLairConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up one Fujitsu HVAC device."""
async_add_entities(
diff --git a/homeassistant/components/fujitsu_fglair/coordinator.py b/homeassistant/components/fujitsu_fglair/coordinator.py
index d98464e4751..8c66548973b 100644
--- a/homeassistant/components/fujitsu_fglair/coordinator.py
+++ b/homeassistant/components/fujitsu_fglair/coordinator.py
@@ -5,6 +5,7 @@ import logging
from ayla_iot_unofficial import AylaApi, AylaAuthError
from ayla_iot_unofficial.fujitsu_hvac import FujitsuHVAC
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -13,15 +14,22 @@ from .const import API_REFRESH
_LOGGER = logging.getLogger(__name__)
+type FGLairConfigEntry = ConfigEntry[FGLairCoordinator]
+
class FGLairCoordinator(DataUpdateCoordinator[dict[str, FujitsuHVAC]]):
"""Coordinator for Fujitsu HVAC integration."""
- def __init__(self, hass: HomeAssistant, api: AylaApi) -> None:
+ config_entry: FGLairConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: FGLairConfigEntry, api: AylaApi
+ ) -> None:
"""Initialize coordinator for Fujitsu HVAC integration."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Fujitsu HVAC data",
update_interval=API_REFRESH,
)
diff --git a/homeassistant/components/fujitsu_fglair/manifest.json b/homeassistant/components/fujitsu_fglair/manifest.json
index 330685f89fc..c8fed9b45c9 100644
--- a/homeassistant/components/fujitsu_fglair/manifest.json
+++ b/homeassistant/components/fujitsu_fglair/manifest.json
@@ -5,5 +5,5 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/fujitsu_fglair",
"iot_class": "cloud_polling",
- "requirements": ["ayla-iot-unofficial==1.4.5"]
+ "requirements": ["ayla-iot-unofficial==1.4.7"]
}
diff --git a/homeassistant/components/fujitsu_fglair/sensor.py b/homeassistant/components/fujitsu_fglair/sensor.py
index 1426e2349ea..3bb693e1068 100644
--- a/homeassistant/components/fujitsu_fglair/sensor.py
+++ b/homeassistant/components/fujitsu_fglair/sensor.py
@@ -9,22 +9,22 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .climate import FGLairConfigEntry
-from .coordinator import FGLairCoordinator
+from .coordinator import FGLairConfigEntry, FGLairCoordinator
from .entity import FGLairEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FGLairConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up one Fujitsu HVAC device."""
async_add_entities(
FGLairOutsideTemperature(entry.runtime_data, device)
for device in entry.runtime_data.data.values()
+ if device.outdoor_temperature is not None
)
diff --git a/homeassistant/components/fully_kiosk/__init__.py b/homeassistant/components/fully_kiosk/__init__.py
index 074ec3feaa0..772e7f79242 100644
--- a/homeassistant/components/fully_kiosk/__init__.py
+++ b/homeassistant/components/fully_kiosk/__init__.py
@@ -1,17 +1,14 @@
"""The Fully Kiosk Browser integration."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.typing import ConfigType
from .const import DOMAIN
-from .coordinator import FullyKioskDataUpdateCoordinator
+from .coordinator import FullyKioskConfigEntry, FullyKioskDataUpdateCoordinator
from .services import async_setup_services
-type FullyKioskConfigEntry = ConfigEntry[FullyKioskDataUpdateCoordinator]
-
PLATFORMS = [
Platform.BINARY_SENSOR,
Platform.BUTTON,
diff --git a/homeassistant/components/fully_kiosk/binary_sensor.py b/homeassistant/components/fully_kiosk/binary_sensor.py
index c039baa0397..8a25376f635 100644
--- a/homeassistant/components/fully_kiosk/binary_sensor.py
+++ b/homeassistant/components/fully_kiosk/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FullyKioskConfigEntry
from .coordinator import FullyKioskDataUpdateCoordinator
@@ -38,7 +38,7 @@ SENSORS: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser sensor."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/button.py b/homeassistant/components/fully_kiosk/button.py
index 4b172d45ae2..112ead983b9 100644
--- a/homeassistant/components/fully_kiosk/button.py
+++ b/homeassistant/components/fully_kiosk/button.py
@@ -15,7 +15,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FullyKioskConfigEntry
from .coordinator import FullyKioskDataUpdateCoordinator
@@ -68,7 +68,7 @@ BUTTONS: tuple[FullyButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser button entities."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/camera.py b/homeassistant/components/fully_kiosk/camera.py
index 7dfbe9e9257..6357660f8e8 100644
--- a/homeassistant/components/fully_kiosk/camera.py
+++ b/homeassistant/components/fully_kiosk/camera.py
@@ -7,7 +7,7 @@ from fullykiosk import FullyKioskError
from homeassistant.components.camera import Camera, CameraEntityFeature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FullyKioskConfigEntry
from .coordinator import FullyKioskDataUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import FullyKioskEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the cameras."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/coordinator.py b/homeassistant/components/fully_kiosk/coordinator.py
index 405f0746437..dc3640020be 100644
--- a/homeassistant/components/fully_kiosk/coordinator.py
+++ b/homeassistant/components/fully_kiosk/coordinator.py
@@ -14,11 +14,15 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DEFAULT_PORT, LOGGER, UPDATE_INTERVAL
+type FullyKioskConfigEntry = ConfigEntry[FullyKioskDataUpdateCoordinator]
+
class FullyKioskDataUpdateCoordinator(DataUpdateCoordinator):
"""Define an object to hold Fully Kiosk Browser data."""
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
+ config_entry: FullyKioskConfigEntry
+
+ def __init__(self, hass: HomeAssistant, entry: FullyKioskConfigEntry) -> None:
"""Initialize."""
self.use_ssl = entry.data.get(CONF_SSL, False)
self.fully = FullyKiosk(
@@ -32,6 +36,7 @@ class FullyKioskDataUpdateCoordinator(DataUpdateCoordinator):
super().__init__(
hass,
LOGGER,
+ config_entry=entry,
name=entry.data[CONF_HOST],
update_interval=UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/fully_kiosk/image.py b/homeassistant/components/fully_kiosk/image.py
index e1a4240c9e9..158eae8671c 100644
--- a/homeassistant/components/fully_kiosk/image.py
+++ b/homeassistant/components/fully_kiosk/image.py
@@ -11,7 +11,7 @@ from fullykiosk import FullyKiosk, FullyKioskError
from homeassistant.components.image import ImageEntity, ImageEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import FullyKioskConfigEntry
@@ -38,7 +38,7 @@ IMAGES: tuple[FullyImageEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser image entities."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/media_player.py b/homeassistant/components/fully_kiosk/media_player.py
index 24f002a7544..f6333a2941d 100644
--- a/homeassistant/components/fully_kiosk/media_player.py
+++ b/homeassistant/components/fully_kiosk/media_player.py
@@ -14,7 +14,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FullyKioskConfigEntry
from .const import AUDIOMANAGER_STREAM_MUSIC, MEDIA_SUPPORT_FULLYKIOSK
@@ -25,7 +25,7 @@ from .entity import FullyKioskEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser media player entity."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/notify.py b/homeassistant/components/fully_kiosk/notify.py
index bddc07439b3..0a0c24c60e2 100644
--- a/homeassistant/components/fully_kiosk/notify.py
+++ b/homeassistant/components/fully_kiosk/notify.py
@@ -9,7 +9,7 @@ from fullykiosk import FullyKioskError
from homeassistant.components.notify import NotifyEntity, NotifyEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FullyKioskConfigEntry
from .coordinator import FullyKioskDataUpdateCoordinator
@@ -40,7 +40,7 @@ NOTIFIERS: tuple[FullyNotifyEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser notify entities."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/number.py b/homeassistant/components/fully_kiosk/number.py
index ef25a69f1ee..8c386e85418 100644
--- a/homeassistant/components/fully_kiosk/number.py
+++ b/homeassistant/components/fully_kiosk/number.py
@@ -7,7 +7,7 @@ from contextlib import suppress
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FullyKioskConfigEntry
from .coordinator import FullyKioskDataUpdateCoordinator
@@ -54,7 +54,7 @@ ENTITY_TYPES: tuple[NumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser number entities."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/sensor.py b/homeassistant/components/fully_kiosk/sensor.py
index d92c5c17341..6094a6c4c23 100644
--- a/homeassistant/components/fully_kiosk/sensor.py
+++ b/homeassistant/components/fully_kiosk/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfInformation
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import FullyKioskConfigEntry
@@ -114,7 +114,7 @@ SENSORS: tuple[FullySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser sensor."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/fully_kiosk/strings.json b/homeassistant/components/fully_kiosk/strings.json
index a4b466926f0..5841456c034 100644
--- a/homeassistant/components/fully_kiosk/strings.json
+++ b/homeassistant/components/fully_kiosk/strings.json
@@ -1,6 +1,6 @@
{
"common": {
- "data_description_password": "The Remote Admin Password from the Fully Kiosk Browser app settings.",
+ "data_description_password": "The Remote Admin password from the Fully Kiosk Browser app settings.",
"data_description_ssl": "Is the Fully Kiosk app configured to require SSL for the connection?",
"data_description_verify_ssl": "Should SSL certificartes be verified? This should be off for self-signed certificates."
},
@@ -151,7 +151,7 @@
}
},
"set_config": {
- "name": "Set Configuration",
+ "name": "Set configuration",
"description": "Sets a configuration parameter on Fully Kiosk Browser.",
"fields": {
"key": {
@@ -165,7 +165,7 @@
}
},
"start_application": {
- "name": "Start Application",
+ "name": "Start application",
"description": "Starts an application on the device running Fully Kiosk Browser.",
"fields": {
"application": {
diff --git a/homeassistant/components/fully_kiosk/switch.py b/homeassistant/components/fully_kiosk/switch.py
index 4adf8e8c924..804233dcc9e 100644
--- a/homeassistant/components/fully_kiosk/switch.py
+++ b/homeassistant/components/fully_kiosk/switch.py
@@ -11,7 +11,7 @@ from fullykiosk import FullyKiosk
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import FullyKioskConfigEntry
from .coordinator import FullyKioskDataUpdateCoordinator
@@ -84,7 +84,7 @@ SWITCHES: tuple[FullySwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: FullyKioskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fully Kiosk Browser switch."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/fyta/__init__.py b/homeassistant/components/fyta/__init__.py
index ab4a74c627a..1b00afc9c80 100644
--- a/homeassistant/components/fyta/__init__.py
+++ b/homeassistant/components/fyta/__init__.py
@@ -7,7 +7,6 @@ import logging
from fyta_cli.fyta_connector import FytaConnector
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_ACCESS_TOKEN,
CONF_PASSWORD,
@@ -19,7 +18,7 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.util.dt import async_get_time_zone
from .const import CONF_EXPIRATION
-from .coordinator import FytaCoordinator
+from .coordinator import FytaConfigEntry, FytaCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -28,7 +27,6 @@ PLATFORMS = [
Platform.IMAGE,
Platform.SENSOR,
]
-type FytaConfigEntry = ConfigEntry[FytaCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: FytaConfigEntry) -> bool:
@@ -46,7 +44,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: FytaConfigEntry) -> bool
username, password, access_token, expiration, tz, async_get_clientsession(hass)
)
- coordinator = FytaCoordinator(hass, fyta)
+ coordinator = FytaCoordinator(hass, entry, fyta)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/fyta/binary_sensor.py b/homeassistant/components/fyta/binary_sensor.py
index bcef609d01a..ac092f1d9cb 100644
--- a/homeassistant/components/fyta/binary_sensor.py
+++ b/homeassistant/components/fyta/binary_sensor.py
@@ -15,9 +15,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FytaConfigEntry
+from .coordinator import FytaConfigEntry
from .entity import FytaPlantEntity
@@ -83,7 +83,9 @@ BINARY_SENSORS: Final[list[FytaBinarySensorEntityDescription]] = [
async def async_setup_entry(
- hass: HomeAssistant, entry: FytaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: FytaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FYTA binary sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fyta/coordinator.py b/homeassistant/components/fyta/coordinator.py
index a0c42d449d5..012ed3b2af0 100644
--- a/homeassistant/components/fyta/coordinator.py
+++ b/homeassistant/components/fyta/coordinator.py
@@ -5,7 +5,6 @@ from __future__ import annotations
from collections.abc import Callable
from datetime import datetime, timedelta
import logging
-from typing import TYPE_CHECKING
from fyta_cli.fyta_connector import FytaConnector
from fyta_cli.fyta_exceptions import (
@@ -16,6 +15,7 @@ from fyta_cli.fyta_exceptions import (
)
from fyta_cli.fyta_models import Plant
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -24,22 +24,24 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import CONF_EXPIRATION, DOMAIN
-if TYPE_CHECKING:
- from . import FytaConfigEntry
-
_LOGGER = logging.getLogger(__name__)
+type FytaConfigEntry = ConfigEntry[FytaCoordinator]
+
class FytaCoordinator(DataUpdateCoordinator[dict[int, Plant]]):
"""Fyta custom coordinator."""
config_entry: FytaConfigEntry
- def __init__(self, hass: HomeAssistant, fyta: FytaConnector) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: FytaConfigEntry, fyta: FytaConnector
+ ) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="FYTA Coordinator",
update_interval=timedelta(minutes=4),
)
diff --git a/homeassistant/components/fyta/diagnostics.py b/homeassistant/components/fyta/diagnostics.py
index d02f8cacfa3..d6bda70d754 100644
--- a/homeassistant/components/fyta/diagnostics.py
+++ b/homeassistant/components/fyta/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
-from . import FytaConfigEntry
+from .coordinator import FytaConfigEntry
TO_REDACT = [
CONF_PASSWORD,
diff --git a/homeassistant/components/fyta/entity.py b/homeassistant/components/fyta/entity.py
index 0d0ec533c44..02cd73c54f9 100644
--- a/homeassistant/components/fyta/entity.py
+++ b/homeassistant/components/fyta/entity.py
@@ -6,9 +6,8 @@ from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import FytaConfigEntry
from .const import DOMAIN
-from .coordinator import FytaCoordinator
+from .coordinator import FytaConfigEntry, FytaCoordinator
class FytaPlantEntity(CoordinatorEntity[FytaCoordinator]):
diff --git a/homeassistant/components/fyta/image.py b/homeassistant/components/fyta/image.py
index f03df969dcc..326f2ddf570 100644
--- a/homeassistant/components/fyta/image.py
+++ b/homeassistant/components/fyta/image.py
@@ -7,15 +7,16 @@ from datetime import datetime
from homeassistant.components.image import ImageEntity, ImageEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import FytaConfigEntry
-from .coordinator import FytaCoordinator
+from .coordinator import FytaConfigEntry, FytaCoordinator
from .entity import FytaPlantEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: FytaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: FytaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FYTA plant images."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/fyta/manifest.json b/homeassistant/components/fyta/manifest.json
index ea628f55c6c..615197203a8 100644
--- a/homeassistant/components/fyta/manifest.json
+++ b/homeassistant/components/fyta/manifest.json
@@ -9,5 +9,5 @@
"iot_class": "cloud_polling",
"loggers": ["fyta_cli"],
"quality_scale": "platinum",
- "requirements": ["fyta_cli==0.7.0"]
+ "requirements": ["fyta_cli==0.7.2"]
}
diff --git a/homeassistant/components/fyta/sensor.py b/homeassistant/components/fyta/sensor.py
index 254e4522819..622945ae102 100644
--- a/homeassistant/components/fyta/sensor.py
+++ b/homeassistant/components/fyta/sensor.py
@@ -22,11 +22,10 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import FytaConfigEntry
-from .coordinator import FytaCoordinator
+from .coordinator import FytaConfigEntry, FytaCoordinator
from .entity import FytaPlantEntity
@@ -155,7 +154,9 @@ SENSORS: Final[list[FytaSensorEntityDescription]] = [
async def async_setup_entry(
- hass: HomeAssistant, entry: FytaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: FytaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the FYTA sensors."""
coordinator: FytaCoordinator = entry.runtime_data
diff --git a/homeassistant/components/garages_amsterdam/__init__.py b/homeassistant/components/garages_amsterdam/__init__.py
index 99d751cfcc8..854e41f2d89 100644
--- a/homeassistant/components/garages_amsterdam/__init__.py
+++ b/homeassistant/components/garages_amsterdam/__init__.py
@@ -4,24 +4,24 @@ from __future__ import annotations
from odp_amsterdam import ODPAmsterdam
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .coordinator import GaragesAmsterdamDataUpdateCoordinator
+from .coordinator import (
+ GaragesAmsterdamConfigEntry,
+ GaragesAmsterdamDataUpdateCoordinator,
+)
PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SENSOR]
-type GaragesAmsterdamConfigEntry = ConfigEntry[GaragesAmsterdamDataUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, entry: GaragesAmsterdamConfigEntry
) -> bool:
"""Set up Garages Amsterdam from a config entry."""
client = ODPAmsterdam(session=async_get_clientsession(hass))
- coordinator = GaragesAmsterdamDataUpdateCoordinator(hass, client)
+ coordinator = GaragesAmsterdamDataUpdateCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/garages_amsterdam/binary_sensor.py b/homeassistant/components/garages_amsterdam/binary_sensor.py
index b93b43e1173..6cfd68c8a00 100644
--- a/homeassistant/components/garages_amsterdam/binary_sensor.py
+++ b/homeassistant/components/garages_amsterdam/binary_sensor.py
@@ -13,10 +13,12 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import GaragesAmsterdamConfigEntry
-from .coordinator import GaragesAmsterdamDataUpdateCoordinator
+from .coordinator import (
+ GaragesAmsterdamConfigEntry,
+ GaragesAmsterdamDataUpdateCoordinator,
+)
from .entity import GaragesAmsterdamEntity
@@ -40,7 +42,7 @@ BINARY_SENSORS: tuple[GaragesAmsterdamBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GaragesAmsterdamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/garages_amsterdam/coordinator.py b/homeassistant/components/garages_amsterdam/coordinator.py
index 3d06aba79e2..74f2361980d 100644
--- a/homeassistant/components/garages_amsterdam/coordinator.py
+++ b/homeassistant/components/garages_amsterdam/coordinator.py
@@ -4,24 +4,31 @@ from __future__ import annotations
from odp_amsterdam import Garage, ODPAmsterdam, VehicleType
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN, LOGGER, SCAN_INTERVAL
+type GaragesAmsterdamConfigEntry = ConfigEntry[GaragesAmsterdamDataUpdateCoordinator]
+
class GaragesAmsterdamDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Garage]]):
"""Class to manage fetching Garages Amsterdam data from single endpoint."""
+ config_entry: GaragesAmsterdamConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: GaragesAmsterdamConfigEntry,
client: ODPAmsterdam,
) -> None:
"""Initialize global Garages Amsterdam data updater."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/garages_amsterdam/sensor.py b/homeassistant/components/garages_amsterdam/sensor.py
index b562fff841a..5467ae73b1e 100644
--- a/homeassistant/components/garages_amsterdam/sensor.py
+++ b/homeassistant/components/garages_amsterdam/sensor.py
@@ -13,11 +13,13 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import GaragesAmsterdamConfigEntry
-from .coordinator import GaragesAmsterdamDataUpdateCoordinator
+from .coordinator import (
+ GaragesAmsterdamConfigEntry,
+ GaragesAmsterdamDataUpdateCoordinator,
+)
from .entity import GaragesAmsterdamEntity
@@ -57,7 +59,7 @@ SENSORS: tuple[GaragesAmsterdamSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GaragesAmsterdamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/gardena_bluetooth/__init__.py b/homeassistant/components/gardena_bluetooth/__init__.py
index 47034e61fb9..34f72bf0a5a 100644
--- a/homeassistant/components/gardena_bluetooth/__init__.py
+++ b/homeassistant/components/gardena_bluetooth/__init__.py
@@ -10,7 +10,6 @@ from gardena_bluetooth.const import DeviceConfiguration, DeviceInformation
from gardena_bluetooth.exceptions import CommunicationFailure
from homeassistant.components import bluetooth
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ADDRESS, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@@ -18,7 +17,11 @@ from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.util import dt as dt_util
from .const import DOMAIN
-from .coordinator import DeviceUnavailable, GardenaBluetoothCoordinator
+from .coordinator import (
+ DeviceUnavailable,
+ GardenaBluetoothConfigEntry,
+ GardenaBluetoothCoordinator,
+)
PLATFORMS: list[Platform] = [
Platform.BINARY_SENSOR,
@@ -32,8 +35,6 @@ LOGGER = logging.getLogger(__name__)
TIMEOUT = 20.0
DISCONNECT_DELAY = 5
-type GardenaBluetoothConfigEntry = ConfigEntry[GardenaBluetoothCoordinator]
-
def get_connection(hass: HomeAssistant, address: str) -> CachedConnection:
"""Set up a cached client that keeps connection after last use."""
@@ -80,7 +81,7 @@ async def async_setup_entry(
)
coordinator = GardenaBluetoothCoordinator(
- hass, LOGGER, client, uuids, device, address
+ hass, entry, LOGGER, client, uuids, device, address
)
entry.runtime_data = coordinator
diff --git a/homeassistant/components/gardena_bluetooth/binary_sensor.py b/homeassistant/components/gardena_bluetooth/binary_sensor.py
index d3ae096e291..b41988afd8c 100644
--- a/homeassistant/components/gardena_bluetooth/binary_sensor.py
+++ b/homeassistant/components/gardena_bluetooth/binary_sensor.py
@@ -14,9 +14,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import GardenaBluetoothConfigEntry
+from .coordinator import GardenaBluetoothConfigEntry
from .entity import GardenaBluetoothDescriptorEntity
@@ -53,7 +53,7 @@ DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GardenaBluetoothConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/gardena_bluetooth/button.py b/homeassistant/components/gardena_bluetooth/button.py
index 9d87cba2446..6a4f0395fe0 100644
--- a/homeassistant/components/gardena_bluetooth/button.py
+++ b/homeassistant/components/gardena_bluetooth/button.py
@@ -10,9 +10,9 @@ from gardena_bluetooth.parse import CharacteristicBool
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import GardenaBluetoothConfigEntry
+from .coordinator import GardenaBluetoothConfigEntry
from .entity import GardenaBluetoothDescriptorEntity
@@ -42,7 +42,7 @@ DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GardenaBluetoothConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/gardena_bluetooth/config_flow.py b/homeassistant/components/gardena_bluetooth/config_flow.py
index c7631b62f47..613d0cf21db 100644
--- a/homeassistant/components/gardena_bluetooth/config_flow.py
+++ b/homeassistant/components/gardena_bluetooth/config_flow.py
@@ -41,6 +41,8 @@ def _is_supported(discovery_info: BluetoothServiceInfo):
ProductType.PUMP,
ProductType.VALVE,
ProductType.WATER_COMPUTER,
+ ProductType.AUTOMATS,
+ ProductType.PRESSURE_TANKS,
):
_LOGGER.debug("Unsupported device: %s", manufacturer_data)
return False
diff --git a/homeassistant/components/gardena_bluetooth/coordinator.py b/homeassistant/components/gardena_bluetooth/coordinator.py
index 5caafe0e794..f85fb839657 100644
--- a/homeassistant/components/gardena_bluetooth/coordinator.py
+++ b/homeassistant/components/gardena_bluetooth/coordinator.py
@@ -12,6 +12,7 @@ from gardena_bluetooth.exceptions import (
)
from gardena_bluetooth.parse import Characteristic, CharacteristicType
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
@@ -20,6 +21,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
SCAN_INTERVAL = timedelta(seconds=60)
LOGGER = logging.getLogger(__name__)
+type GardenaBluetoothConfigEntry = ConfigEntry[GardenaBluetoothCoordinator]
+
class DeviceUnavailable(HomeAssistantError):
"""Raised if device can't be found."""
@@ -28,9 +31,12 @@ class DeviceUnavailable(HomeAssistantError):
class GardenaBluetoothCoordinator(DataUpdateCoordinator[dict[str, bytes]]):
"""Class to manage fetching data."""
+ config_entry: GardenaBluetoothConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: GardenaBluetoothConfigEntry,
logger: logging.Logger,
client: Client,
characteristics: set[str],
@@ -41,6 +47,7 @@ class GardenaBluetoothCoordinator(DataUpdateCoordinator[dict[str, bytes]]):
super().__init__(
hass=hass,
logger=logger,
+ config_entry=config_entry,
name="Gardena Bluetooth Data Update Coordinator",
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/gardena_bluetooth/manifest.json b/homeassistant/components/gardena_bluetooth/manifest.json
index 28bba1015f5..8c9cda7d998 100644
--- a/homeassistant/components/gardena_bluetooth/manifest.json
+++ b/homeassistant/components/gardena_bluetooth/manifest.json
@@ -14,5 +14,5 @@
"documentation": "https://www.home-assistant.io/integrations/gardena_bluetooth",
"iot_class": "local_polling",
"loggers": ["bleak", "bleak_esphome", "gardena_bluetooth"],
- "requirements": ["gardena-bluetooth==1.5.0"]
+ "requirements": ["gardena-bluetooth==1.6.0"]
}
diff --git a/homeassistant/components/gardena_bluetooth/number.py b/homeassistant/components/gardena_bluetooth/number.py
index b55630fa797..41b4f1e79ba 100644
--- a/homeassistant/components/gardena_bluetooth/number.py
+++ b/homeassistant/components/gardena_bluetooth/number.py
@@ -19,10 +19,9 @@ from homeassistant.components.number import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import GardenaBluetoothConfigEntry
-from .coordinator import GardenaBluetoothCoordinator
+from .coordinator import GardenaBluetoothConfigEntry, GardenaBluetoothCoordinator
from .entity import GardenaBluetoothDescriptorEntity, GardenaBluetoothEntity
@@ -106,7 +105,7 @@ DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GardenaBluetoothConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entity based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/gardena_bluetooth/sensor.py b/homeassistant/components/gardena_bluetooth/sensor.py
index c07d2ba6866..602f5bdfd6e 100644
--- a/homeassistant/components/gardena_bluetooth/sensor.py
+++ b/homeassistant/components/gardena_bluetooth/sensor.py
@@ -16,11 +16,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
-from . import GardenaBluetoothConfigEntry
-from .coordinator import GardenaBluetoothCoordinator
+from .coordinator import GardenaBluetoothConfigEntry, GardenaBluetoothCoordinator
from .entity import GardenaBluetoothDescriptorEntity, GardenaBluetoothEntity
@@ -96,7 +95,7 @@ DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GardenaBluetoothConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Gardena Bluetooth sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/gardena_bluetooth/switch.py b/homeassistant/components/gardena_bluetooth/switch.py
index f82c39025a5..de1fbe22470 100644
--- a/homeassistant/components/gardena_bluetooth/switch.py
+++ b/homeassistant/components/gardena_bluetooth/switch.py
@@ -9,17 +9,16 @@ from gardena_bluetooth.const import Valve
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import GardenaBluetoothConfigEntry
-from .coordinator import GardenaBluetoothCoordinator
+from .coordinator import GardenaBluetoothConfigEntry, GardenaBluetoothCoordinator
from .entity import GardenaBluetoothEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: GardenaBluetoothConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/gardena_bluetooth/valve.py b/homeassistant/components/gardena_bluetooth/valve.py
index ae6bf56a7ff..4138c7c4472 100644
--- a/homeassistant/components/gardena_bluetooth/valve.py
+++ b/homeassistant/components/gardena_bluetooth/valve.py
@@ -8,10 +8,9 @@ from gardena_bluetooth.const import Valve
from homeassistant.components.valve import ValveEntity, ValveEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import GardenaBluetoothConfigEntry
-from .coordinator import GardenaBluetoothCoordinator
+from .coordinator import GardenaBluetoothConfigEntry, GardenaBluetoothCoordinator
from .entity import GardenaBluetoothEntity
FALLBACK_WATERING_TIME_IN_SECONDS = 60 * 60
@@ -20,7 +19,7 @@ FALLBACK_WATERING_TIME_IN_SECONDS = 60 * 60
async def async_setup_entry(
hass: HomeAssistant,
entry: GardenaBluetoothConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/gdacs/geo_location.py b/homeassistant/components/gdacs/geo_location.py
index 3f693241b24..d277ee54f6b 100644
--- a/homeassistant/components/gdacs/geo_location.py
+++ b/homeassistant/components/gdacs/geo_location.py
@@ -15,7 +15,7 @@ from homeassistant.const import UnitOfLength
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import DistanceConverter
from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
@@ -52,7 +52,9 @@ SOURCE = "gdacs"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GDACS Feed platform."""
manager: GdacsFeedEntityManager = hass.data[DOMAIN][FEED][entry.entry_id]
diff --git a/homeassistant/components/gdacs/sensor.py b/homeassistant/components/gdacs/sensor.py
index c8205730da4..a204addd414 100644
--- a/homeassistant/components/gdacs/sensor.py
+++ b/homeassistant/components/gdacs/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import GdacsFeedEntityManager
@@ -37,7 +37,9 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GDACS Feed platform."""
manager: GdacsFeedEntityManager = hass.data[DOMAIN][FEED][entry.entry_id]
diff --git a/homeassistant/components/generic/camera.py b/homeassistant/components/generic/camera.py
index edefbc55ca6..6821300fadf 100644
--- a/homeassistant/components/generic/camera.py
+++ b/homeassistant/components/generic/camera.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import TemplateError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.httpx_client import get_async_client
from homeassistant.helpers.template import Template
@@ -47,7 +47,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a generic IP Camera."""
diff --git a/homeassistant/components/generic_hygrostat/humidifier.py b/homeassistant/components/generic_hygrostat/humidifier.py
index 69c4fb3cdf4..6e699745279 100644
--- a/homeassistant/components/generic_hygrostat/humidifier.py
+++ b/homeassistant/components/generic_hygrostat/humidifier.py
@@ -43,7 +43,10 @@ from homeassistant.core import (
)
from homeassistant.helpers import condition, config_validation as cv
from homeassistant.helpers.device import async_device_info_to_link_from_entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import (
async_track_state_change_event,
async_track_state_report_event,
@@ -94,7 +97,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
@@ -116,7 +119,7 @@ async def _async_setup_config(
hass: HomeAssistant,
config: Mapping[str, Any],
unique_id: str | None,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddEntitiesCallback | AddConfigEntryEntitiesCallback,
) -> None:
name: str = config[CONF_NAME]
switch_entity_id: str = config[CONF_HUMIDIFIER]
diff --git a/homeassistant/components/generic_thermostat/climate.py b/homeassistant/components/generic_thermostat/climate.py
index fe6f0253f48..190caa58b3f 100644
--- a/homeassistant/components/generic_thermostat/climate.py
+++ b/homeassistant/components/generic_thermostat/climate.py
@@ -49,7 +49,10 @@ from homeassistant.core import (
from homeassistant.exceptions import ConditionError
from homeassistant.helpers import condition, config_validation as cv
from homeassistant.helpers.device import async_device_info_to_link_from_entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import (
async_track_state_change_event,
async_track_time_interval,
@@ -123,7 +126,7 @@ PLATFORM_SCHEMA = CLIMATE_PLATFORM_SCHEMA.extend(PLATFORM_SCHEMA_COMMON.schema)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
await _async_setup_config(
@@ -152,7 +155,7 @@ async def _async_setup_config(
hass: HomeAssistant,
config: Mapping[str, Any],
unique_id: str | None,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddEntitiesCallback | AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the generic thermostat platform."""
diff --git a/homeassistant/components/geniushub/binary_sensor.py b/homeassistant/components/geniushub/binary_sensor.py
index 01ccc950fd6..c2f25532453 100644
--- a/homeassistant/components/geniushub/binary_sensor.py
+++ b/homeassistant/components/geniushub/binary_sensor.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GeniusHubConfigEntry
from .entity import GeniusDevice
@@ -16,7 +16,7 @@ GH_TYPE = "Receiver"
async def async_setup_entry(
hass: HomeAssistant,
entry: GeniusHubConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Genius Hub binary sensor entities."""
diff --git a/homeassistant/components/geniushub/climate.py b/homeassistant/components/geniushub/climate.py
index e20d649541e..3c5cc4d4ad9 100644
--- a/homeassistant/components/geniushub/climate.py
+++ b/homeassistant/components/geniushub/climate.py
@@ -11,7 +11,7 @@ from homeassistant.components.climate import (
HVACMode,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GeniusHubConfigEntry
from .entity import GeniusHeatingZone
@@ -29,7 +29,7 @@ GH_ZONES = ["radiator", "wet underfloor"]
async def async_setup_entry(
hass: HomeAssistant,
entry: GeniusHubConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Genius Hub climate entities."""
diff --git a/homeassistant/components/geniushub/sensor.py b/homeassistant/components/geniushub/sensor.py
index a558ad18672..de7c047e934 100644
--- a/homeassistant/components/geniushub/sensor.py
+++ b/homeassistant/components/geniushub/sensor.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import GeniusHubConfigEntry
@@ -26,7 +26,7 @@ GH_LEVEL_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
entry: GeniusHubConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Genius Hub sensor entities."""
diff --git a/homeassistant/components/geniushub/strings.json b/homeassistant/components/geniushub/strings.json
index 42d53c7fa00..79eee2c9a1b 100644
--- a/homeassistant/components/geniushub/strings.json
+++ b/homeassistant/components/geniushub/strings.json
@@ -45,7 +45,7 @@
},
"mode": {
"name": "[%key:common::config_flow::data::mode%]",
- "description": "One of: off, timer or footprint."
+ "description": "The zone's operating mode."
}
}
},
diff --git a/homeassistant/components/geniushub/switch.py b/homeassistant/components/geniushub/switch.py
index 3af82eb4e92..890ca1578be 100644
--- a/homeassistant/components/geniushub/switch.py
+++ b/homeassistant/components/geniushub/switch.py
@@ -10,7 +10,7 @@ import voluptuous as vol
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from . import ATTR_DURATION, GeniusHubConfigEntry
@@ -31,7 +31,7 @@ SET_SWITCH_OVERRIDE_SCHEMA: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
entry: GeniusHubConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Genius Hub switch entities."""
diff --git a/homeassistant/components/geniushub/water_heater.py b/homeassistant/components/geniushub/water_heater.py
index 2807bd60611..60acf8f2cca 100644
--- a/homeassistant/components/geniushub/water_heater.py
+++ b/homeassistant/components/geniushub/water_heater.py
@@ -8,7 +8,7 @@ from homeassistant.components.water_heater import (
)
from homeassistant.const import STATE_OFF
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GeniusHubConfigEntry
from .entity import GeniusHeatingZone
@@ -36,7 +36,7 @@ GH_HEATERS = ["hot water temperature"]
async def async_setup_entry(
hass: HomeAssistant,
entry: GeniusHubConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Genius Hub water heater entities."""
diff --git a/homeassistant/components/geo_json_events/__init__.py b/homeassistant/components/geo_json_events/__init__.py
index d55fe6e3ee6..e38c17008a5 100644
--- a/homeassistant/components/geo_json_events/__init__.py
+++ b/homeassistant/components/geo_json_events/__init__.py
@@ -4,25 +4,27 @@ from __future__ import annotations
import logging
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from .const import DOMAIN, PLATFORMS
-from .manager import GeoJsonFeedEntityManager
+from .const import PLATFORMS
+from .manager import GeoJsonConfigEntry, GeoJsonFeedEntityManager
_LOGGER = logging.getLogger(__name__)
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_setup_entry(
+ hass: HomeAssistant, config_entry: GeoJsonConfigEntry
+) -> bool:
"""Set up the GeoJSON events component as config entry."""
- feeds = hass.data.setdefault(DOMAIN, {})
# Create feed entity manager for all platforms.
manager = GeoJsonFeedEntityManager(hass, config_entry)
- feeds[config_entry.entry_id] = manager
_LOGGER.debug("Feed entity manager added for %s", config_entry.entry_id)
await remove_orphaned_entities(hass, config_entry.entry_id)
+
+ config_entry.runtime_data = manager
+ config_entry.async_on_unload(manager.async_stop)
await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
await manager.async_init()
return True
@@ -46,10 +48,6 @@ async def remove_orphaned_entities(hass: HomeAssistant, entry_id: str) -> None:
entity_registry.async_remove(entry.entity_id)
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: GeoJsonConfigEntry) -> bool:
"""Unload the GeoJSON events config entry."""
- unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- if unload_ok:
- manager: GeoJsonFeedEntityManager = hass.data[DOMAIN].pop(entry.entry_id)
- await manager.async_stop()
- return unload_ok
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/geo_json_events/geo_location.py b/homeassistant/components/geo_json_events/geo_location.py
index e0067bcfdc9..a119571a0ca 100644
--- a/homeassistant/components/geo_json_events/geo_location.py
+++ b/homeassistant/components/geo_json_events/geo_location.py
@@ -9,29 +9,24 @@ from typing import Any
from aio_geojson_generic_client.feed_entry import GenericFeedEntry
from homeassistant.components.geo_location import GeolocationEvent
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfLength
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import GeoJsonFeedEntityManager
-from .const import (
- ATTR_EXTERNAL_ID,
- DOMAIN,
- SIGNAL_DELETE_ENTITY,
- SIGNAL_UPDATE_ENTITY,
- SOURCE,
-)
+from .const import ATTR_EXTERNAL_ID, SIGNAL_DELETE_ENTITY, SIGNAL_UPDATE_ENTITY, SOURCE
+from .manager import GeoJsonConfigEntry, GeoJsonFeedEntityManager
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: GeoJsonConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GeoJSON Events platform."""
- manager: GeoJsonFeedEntityManager = hass.data[DOMAIN][entry.entry_id]
+ manager = entry.runtime_data
@callback
def async_add_geolocation(
diff --git a/homeassistant/components/geo_json_events/manager.py b/homeassistant/components/geo_json_events/manager.py
index deff15436a6..223d3bf571f 100644
--- a/homeassistant/components/geo_json_events/manager.py
+++ b/homeassistant/components/geo_json_events/manager.py
@@ -25,6 +25,8 @@ from .const import (
_LOGGER = logging.getLogger(__name__)
+type GeoJsonConfigEntry = ConfigEntry[GeoJsonFeedEntityManager]
+
class GeoJsonFeedEntityManager:
"""Feed Entity Manager for GeoJSON feeds."""
diff --git a/homeassistant/components/geocaching/coordinator.py b/homeassistant/components/geocaching/coordinator.py
index 8f56cd9d846..41b59d049af 100644
--- a/homeassistant/components/geocaching/coordinator.py
+++ b/homeassistant/components/geocaching/coordinator.py
@@ -18,12 +18,13 @@ from .const import DOMAIN, ENVIRONMENT, LOGGER, UPDATE_INTERVAL
class GeocachingDataUpdateCoordinator(DataUpdateCoordinator[GeocachingStatus]):
"""Class to manage fetching Geocaching data from single endpoint."""
+ config_entry: ConfigEntry
+
def __init__(
self, hass: HomeAssistant, *, entry: ConfigEntry, session: OAuth2Session
) -> None:
"""Initialize global Geocaching data updater."""
self.session = session
- self.entry = entry
async def async_token_refresh() -> str:
await session.async_ensure_token_valid()
@@ -39,7 +40,13 @@ class GeocachingDataUpdateCoordinator(DataUpdateCoordinator[GeocachingStatus]):
token_refresh_method=async_token_refresh,
)
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=entry,
+ name=DOMAIN,
+ update_interval=UPDATE_INTERVAL,
+ )
async def _async_update_data(self) -> GeocachingStatus:
try:
diff --git a/homeassistant/components/geocaching/manifest.json b/homeassistant/components/geocaching/manifest.json
index 127519ca5d0..4617bd1c57b 100644
--- a/homeassistant/components/geocaching/manifest.json
+++ b/homeassistant/components/geocaching/manifest.json
@@ -6,5 +6,5 @@
"dependencies": ["application_credentials"],
"documentation": "https://www.home-assistant.io/integrations/geocaching",
"iot_class": "cloud_polling",
- "requirements": ["geocachingapi==0.2.1"]
+ "requirements": ["geocachingapi==0.3.0"]
}
diff --git a/homeassistant/components/geocaching/sensor.py b/homeassistant/components/geocaching/sensor.py
index c082e5308a1..c7894afc5ac 100644
--- a/homeassistant/components/geocaching/sensor.py
+++ b/homeassistant/components/geocaching/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import SensorEntity, SensorEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -64,7 +64,9 @@ SENSORS: tuple[GeocachingSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Geocaching sensor entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/geofency/device_tracker.py b/homeassistant/components/geofency/device_tracker.py
index 2ad3c1772de..c74dad1cebb 100644
--- a/homeassistant/components/geofency/device_tracker.py
+++ b/homeassistant/components/geofency/device_tracker.py
@@ -7,7 +7,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import DOMAIN as GF_DOMAIN, TRACKER_UPDATE
@@ -16,7 +16,7 @@ from . import DOMAIN as GF_DOMAIN, TRACKER_UPDATE
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Geofency config entry."""
diff --git a/homeassistant/components/geonetnz_quakes/geo_location.py b/homeassistant/components/geonetnz_quakes/geo_location.py
index 78313e102e0..96a1c3c09b2 100644
--- a/homeassistant/components/geonetnz_quakes/geo_location.py
+++ b/homeassistant/components/geonetnz_quakes/geo_location.py
@@ -14,7 +14,7 @@ from homeassistant.const import ATTR_TIME, UnitOfLength
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import DistanceConverter
from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
@@ -38,7 +38,9 @@ SOURCE = "geonetnz_quakes"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GeoNet NZ Quakes Feed platform."""
manager: GeonetnzQuakesFeedEntityManager = hass.data[DOMAIN][FEED][entry.entry_id]
diff --git a/homeassistant/components/geonetnz_quakes/sensor.py b/homeassistant/components/geonetnz_quakes/sensor.py
index 2fce3e93d12..b8a1e2dd4db 100644
--- a/homeassistant/components/geonetnz_quakes/sensor.py
+++ b/homeassistant/components/geonetnz_quakes/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN, FEED
@@ -31,7 +31,9 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GeoNet NZ Quakes Feed platform."""
manager = hass.data[DOMAIN][FEED][entry.entry_id]
diff --git a/homeassistant/components/geonetnz_volcano/sensor.py b/homeassistant/components/geonetnz_volcano/sensor.py
index 980679cc64f..bde04acb895 100644
--- a/homeassistant/components/geonetnz_volcano/sensor.py
+++ b/homeassistant/components/geonetnz_volcano/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE, UnitOfLength
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from homeassistant.util.unit_conversion import DistanceConverter
@@ -31,7 +31,9 @@ ATTR_LAST_UPDATE_SUCCESSFUL = "feed_last_update_successful"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GeoNet NZ Volcano Feed platform."""
manager = hass.data[DOMAIN][FEED][entry.entry_id]
diff --git a/homeassistant/components/gios/__init__.py b/homeassistant/components/gios/__init__.py
index b5a0e9d5371..31f704fcacc 100644
--- a/homeassistant/components/gios/__init__.py
+++ b/homeassistant/components/gios/__init__.py
@@ -2,32 +2,26 @@
from __future__ import annotations
-from dataclasses import dataclass
import logging
+from aiohttp.client_exceptions import ClientConnectorError
+from gios import Gios
+from gios.exceptions import GiosError
+
from homeassistant.components.air_quality import DOMAIN as AIR_QUALITY_PLATFORM
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import CONF_STATION_ID, DOMAIN
-from .coordinator import GiosDataUpdateCoordinator
+from .coordinator import GiosConfigEntry, GiosData, GiosDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.SENSOR]
-type GiosConfigEntry = ConfigEntry[GiosData]
-
-
-@dataclass
-class GiosData:
- """Data for GIOS integration."""
-
- coordinator: GiosDataUpdateCoordinator
-
async def async_setup_entry(hass: HomeAssistant, entry: GiosConfigEntry) -> bool:
"""Set up GIOS as config entry."""
@@ -47,8 +41,19 @@ async def async_setup_entry(hass: HomeAssistant, entry: GiosConfigEntry) -> bool
device_registry.async_update_device(device_entry.id, new_identifiers={new_ids})
websession = async_get_clientsession(hass)
+ try:
+ gios = await Gios.create(websession, station_id)
+ except (GiosError, ConnectionError, ClientConnectorError) as err:
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="cannot_connect",
+ translation_placeholders={
+ "entry": entry.title,
+ "error": repr(err),
+ },
+ ) from err
- coordinator = GiosDataUpdateCoordinator(hass, websession, station_id)
+ coordinator = GiosDataUpdateCoordinator(hass, entry, gios)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = GiosData(coordinator)
diff --git a/homeassistant/components/gios/config_flow.py b/homeassistant/components/gios/config_flow.py
index a089aeab820..9b242a8cc99 100644
--- a/homeassistant/components/gios/config_flow.py
+++ b/homeassistant/components/gios/config_flow.py
@@ -3,7 +3,7 @@
from __future__ import annotations
import asyncio
-from typing import Any
+from typing import TYPE_CHECKING, Any
from aiohttp.client_exceptions import ClientConnectorError
from gios import ApiError, Gios, InvalidSensorsDataError, NoStationError
@@ -12,6 +12,12 @@ import voluptuous as vol
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_NAME
from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.selector import (
+ SelectOptionDict,
+ SelectSelector,
+ SelectSelectorConfig,
+ SelectSelectorMode,
+)
from .const import API_TIMEOUT, CONF_STATION_ID, DOMAIN
@@ -27,40 +33,59 @@ class GiosFlowHandler(ConfigFlow, domain=DOMAIN):
"""Handle a flow initialized by the user."""
errors = {}
+ websession = async_get_clientsession(self.hass)
+
if user_input is not None:
+ station_id = user_input[CONF_STATION_ID]
+
try:
- await self.async_set_unique_id(
- str(user_input[CONF_STATION_ID]), raise_on_progress=False
- )
+ await self.async_set_unique_id(station_id, raise_on_progress=False)
self._abort_if_unique_id_configured()
- websession = async_get_clientsession(self.hass)
-
async with asyncio.timeout(API_TIMEOUT):
- gios = Gios(user_input[CONF_STATION_ID], websession)
+ gios = await Gios.create(websession, int(station_id))
await gios.async_update()
- assert gios.station_name is not None
+ # GIOS treats station ID as int
+ user_input[CONF_STATION_ID] = int(station_id)
+
+ if TYPE_CHECKING:
+ assert gios.station_name is not None
+
return self.async_create_entry(
title=gios.station_name,
data=user_input,
)
except (ApiError, ClientConnectorError, TimeoutError):
errors["base"] = "cannot_connect"
- except NoStationError:
- errors[CONF_STATION_ID] = "wrong_station_id"
except InvalidSensorsDataError:
errors[CONF_STATION_ID] = "invalid_sensors_data"
+ try:
+ gios = await Gios.create(websession)
+ except (ApiError, ClientConnectorError, NoStationError):
+ return self.async_abort(reason="cannot_connect")
+
+ options: list[SelectOptionDict] = [
+ SelectOptionDict(value=str(station.id), label=station.name)
+ for station in gios.measurement_stations.values()
+ ]
+
+ schema: vol.Schema = vol.Schema(
+ {
+ vol.Required(CONF_STATION_ID): SelectSelector(
+ SelectSelectorConfig(
+ options=options,
+ sort=True,
+ mode=SelectSelectorMode.DROPDOWN,
+ ),
+ ),
+ vol.Optional(CONF_NAME, default=self.hass.config.location_name): str,
+ }
+ )
+
return self.async_show_form(
step_id="user",
- data_schema=vol.Schema(
- {
- vol.Required(CONF_STATION_ID): int,
- vol.Optional(
- CONF_NAME, default=self.hass.config.location_name
- ): str,
- }
- ),
+ data_schema=schema,
errors=errors,
)
diff --git a/homeassistant/components/gios/const.py b/homeassistant/components/gios/const.py
index a8490511ab8..2294e89c961 100644
--- a/homeassistant/components/gios/const.py
+++ b/homeassistant/components/gios/const.py
@@ -13,7 +13,7 @@ SCAN_INTERVAL: Final = timedelta(minutes=30)
DOMAIN: Final = "gios"
MANUFACTURER: Final = "Główny Inspektorat Ochrony Środowiska"
-URL = "http://powietrze.gios.gov.pl/pjp/current/station_details/info/{station_id}"
+URL = "https://powietrze.gios.gov.pl/pjp/current/station_details/info/{station_id}"
API_TIMEOUT: Final = 30
diff --git a/homeassistant/components/gios/coordinator.py b/homeassistant/components/gios/coordinator.py
index 17b4b89174f..eb0dd82eb67 100644
--- a/homeassistant/components/gios/coordinator.py
+++ b/homeassistant/components/gios/coordinator.py
@@ -3,14 +3,15 @@
from __future__ import annotations
import asyncio
+from dataclasses import dataclass
import logging
-from aiohttp import ClientSession
from aiohttp.client_exceptions import ClientConnectorError
from gios import Gios
from gios.exceptions import GiosError
from gios.model import GiosSensors
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -18,17 +19,37 @@ from .const import API_TIMEOUT, DOMAIN, SCAN_INTERVAL
_LOGGER = logging.getLogger(__name__)
+type GiosConfigEntry = ConfigEntry[GiosData]
+
+
+@dataclass
+class GiosData:
+ """Data for GIOS integration."""
+
+ coordinator: GiosDataUpdateCoordinator
+
class GiosDataUpdateCoordinator(DataUpdateCoordinator[GiosSensors]):
"""Define an object to hold GIOS data."""
+ config_entry: GiosConfigEntry
+
def __init__(
- self, hass: HomeAssistant, session: ClientSession, station_id: int
+ self,
+ hass: HomeAssistant,
+ config_entry: GiosConfigEntry,
+ gios: Gios,
) -> None:
"""Class to manage fetching GIOS data API."""
- self.gios = Gios(station_id, session)
+ self.gios = gios
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
+ )
async def _async_update_data(self) -> GiosSensors:
"""Update data via library."""
@@ -36,4 +57,11 @@ class GiosDataUpdateCoordinator(DataUpdateCoordinator[GiosSensors]):
async with asyncio.timeout(API_TIMEOUT):
return await self.gios.async_update()
except (GiosError, ClientConnectorError) as error:
- raise UpdateFailed(error) from error
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="update_error",
+ translation_placeholders={
+ "entry": self.config_entry.title,
+ "error": repr(error),
+ },
+ ) from error
diff --git a/homeassistant/components/gios/diagnostics.py b/homeassistant/components/gios/diagnostics.py
index a94a95254de..7e938d5ac6b 100644
--- a/homeassistant/components/gios/diagnostics.py
+++ b/homeassistant/components/gios/diagnostics.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.core import HomeAssistant
-from . import GiosConfigEntry
+from .coordinator import GiosConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/gios/manifest.json b/homeassistant/components/gios/manifest.json
index 3d2e719fab6..8deb2eee414 100644
--- a/homeassistant/components/gios/manifest.json
+++ b/homeassistant/components/gios/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "service",
"iot_class": "cloud_polling",
"loggers": ["dacite", "gios"],
- "requirements": ["gios==5.0.0"]
+ "requirements": ["gios==6.0.0"]
}
diff --git a/homeassistant/components/gios/sensor.py b/homeassistant/components/gios/sensor.py
index 69e198d34df..67997a01dc6 100644
--- a/homeassistant/components/gios/sensor.py
+++ b/homeassistant/components/gios/sensor.py
@@ -19,11 +19,10 @@ from homeassistant.const import CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONF_N
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import GiosConfigEntry
from .const import (
ATTR_AQI,
ATTR_C6H6,
@@ -38,7 +37,7 @@ from .const import (
MANUFACTURER,
URL,
)
-from .coordinator import GiosDataUpdateCoordinator
+from .coordinator import GiosConfigEntry, GiosDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -159,7 +158,9 @@ SENSOR_TYPES: tuple[GiosSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: GiosConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: GiosConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a GIOS entities from a config_entry."""
name = entry.data[CONF_NAME]
diff --git a/homeassistant/components/gios/strings.json b/homeassistant/components/gios/strings.json
index fc82f1c843d..eca23159a13 100644
--- a/homeassistant/components/gios/strings.json
+++ b/homeassistant/components/gios/strings.json
@@ -5,17 +5,17 @@
"title": "GIO\u015a (Polish Chief Inspectorate Of Environmental Protection)",
"data": {
"name": "[%key:common::config_flow::data::name%]",
- "station_id": "ID of the measuring station"
+ "station_id": "Measuring station"
}
}
},
"error": {
- "wrong_station_id": "ID of the measuring station is not correct.",
"invalid_sensors_data": "Invalid sensors' data for this measuring station.",
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]"
},
"abort": {
- "already_configured": "[%key:common::config_flow::abort::already_configured_location%]"
+ "already_configured": "[%key:common::config_flow::abort::already_configured_location%]",
+ "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]"
}
},
"system_health": {
@@ -170,5 +170,13 @@
}
}
}
+ },
+ "exceptions": {
+ "cannot_connect": {
+ "message": "An error occurred while connecting to the GIOS API for {entry}: {error}"
+ },
+ "update_error": {
+ "message": "An error occurred while retrieving data from the GIOS API for {entry}: {error}"
+ }
}
}
diff --git a/homeassistant/components/github/__init__.py b/homeassistant/components/github/__init__.py
index 74575e38e09..dea2acf4f1b 100644
--- a/homeassistant/components/github/__init__.py
+++ b/homeassistant/components/github/__init__.py
@@ -4,7 +4,6 @@ from __future__ import annotations
from aiogithubapi import GitHubAPI
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
@@ -14,14 +13,11 @@ from homeassistant.helpers.aiohttp_client import (
)
from .const import CONF_REPOSITORIES, DOMAIN, LOGGER
-from .coordinator import GitHubDataUpdateCoordinator
+from .coordinator import GithubConfigEntry, GitHubDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type GithubConfigEntry = ConfigEntry[dict[str, GitHubDataUpdateCoordinator]]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: GithubConfigEntry) -> bool:
"""Set up GitHub from a config entry."""
client = GitHubAPI(
@@ -36,6 +32,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: GithubConfigEntry) -> bo
for repository in repositories:
coordinator = GitHubDataUpdateCoordinator(
hass=hass,
+ config_entry=entry,
client=client,
repository=repository,
)
@@ -57,7 +54,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: GithubConfigEntry) -> bo
@callback
def async_cleanup_device_registry(
hass: HomeAssistant,
- entry: ConfigEntry,
+ entry: GithubConfigEntry,
) -> None:
"""Remove entries form device registry if we no longer track the repository."""
device_registry = dr.async_get(hass)
@@ -92,6 +89,6 @@ async def async_unload_entry(hass: HomeAssistant, entry: GithubConfigEntry) -> b
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
-async def async_reload_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def async_reload_entry(hass: HomeAssistant, entry: GithubConfigEntry) -> None:
"""Handle an options update."""
await hass.config_entries.async_reload(entry.entry_id)
diff --git a/homeassistant/components/github/coordinator.py b/homeassistant/components/github/coordinator.py
index e73e02932e9..adeda1fd88a 100644
--- a/homeassistant/components/github/coordinator.py
+++ b/homeassistant/components/github/coordinator.py
@@ -13,6 +13,7 @@ from aiogithubapi import (
GitHubResponseModel,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -98,13 +99,18 @@ query ($owner: String!, $repository: String!) {
}
"""
+type GithubConfigEntry = ConfigEntry[dict[str, GitHubDataUpdateCoordinator]]
+
class GitHubDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Data update coordinator for the GitHub integration."""
+ config_entry: GithubConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: GithubConfigEntry,
client: GitHubAPI,
repository: str,
) -> None:
@@ -118,6 +124,7 @@ class GitHubDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=repository,
update_interval=FALLBACK_UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/github/diagnostics.py b/homeassistant/components/github/diagnostics.py
index 8d2d496a813..41fef9406a4 100644
--- a/homeassistant/components/github/diagnostics.py
+++ b/homeassistant/components/github/diagnostics.py
@@ -6,7 +6,6 @@ from typing import Any
from aiogithubapi import GitHubAPI, GitHubException
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import (
@@ -14,10 +13,12 @@ from homeassistant.helpers.aiohttp_client import (
async_get_clientsession,
)
+from .coordinator import GithubConfigEntry
+
async def async_get_config_entry_diagnostics(
hass: HomeAssistant,
- config_entry: ConfigEntry,
+ config_entry: GithubConfigEntry,
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
data = {"options": {**config_entry.options}}
diff --git a/homeassistant/components/github/sensor.py b/homeassistant/components/github/sensor.py
index 614ebe254c4..35985ed50d5 100644
--- a/homeassistant/components/github/sensor.py
+++ b/homeassistant/components/github/sensor.py
@@ -14,13 +14,12 @@ from homeassistant.components.sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import GithubConfigEntry
from .const import DOMAIN
-from .coordinator import GitHubDataUpdateCoordinator
+from .coordinator import GithubConfigEntry, GitHubDataUpdateCoordinator
@dataclass(frozen=True, kw_only=True)
@@ -140,7 +139,7 @@ SENSOR_DESCRIPTIONS: tuple[GitHubSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GithubConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up GitHub sensor based on a config entry."""
repositories = entry.runtime_data
diff --git a/homeassistant/components/glances/__init__.py b/homeassistant/components/glances/__init__.py
index 9d09e63606e..d7b645d9e11 100644
--- a/homeassistant/components/glances/__init__.py
+++ b/homeassistant/components/glances/__init__.py
@@ -10,7 +10,6 @@ from glances_api.exceptions import (
GlancesApiNoDataAvailable,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_HOST,
CONF_PASSWORD,
@@ -29,15 +28,13 @@ from homeassistant.exceptions import (
)
from homeassistant.helpers.httpx_client import get_async_client
-from .coordinator import GlancesDataUpdateCoordinator
+from .coordinator import GlancesConfigEntry, GlancesDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
_LOGGER = logging.getLogger(__name__)
-type GlancesConfigEntry = ConfigEntry[GlancesDataUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, config_entry: GlancesConfigEntry
diff --git a/homeassistant/components/glances/coordinator.py b/homeassistant/components/glances/coordinator.py
index 8882b097ba9..28cf40aae6e 100644
--- a/homeassistant/components/glances/coordinator.py
+++ b/homeassistant/components/glances/coordinator.py
@@ -17,21 +17,25 @@ from .const import DEFAULT_SCAN_INTERVAL, DOMAIN
_LOGGER = logging.getLogger(__name__)
+type GlancesConfigEntry = ConfigEntry[GlancesDataUpdateCoordinator]
+
class GlancesDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Get the latest data from Glances api."""
- config_entry: ConfigEntry
+ config_entry: GlancesConfigEntry
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry, api: Glances) -> None:
+ def __init__(
+ self, hass: HomeAssistant, entry: GlancesConfigEntry, api: Glances
+ ) -> None:
"""Initialize the Glances data."""
self.hass = hass
- self.config_entry = entry
self.host: str = entry.data[CONF_HOST]
self.api = api
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=f"{DOMAIN} - {self.host}",
update_interval=DEFAULT_SCAN_INTERVAL,
)
diff --git a/homeassistant/components/glances/sensor.py b/homeassistant/components/glances/sensor.py
index 0741926296e..67f57ee0fbf 100644
--- a/homeassistant/components/glances/sensor.py
+++ b/homeassistant/components/glances/sensor.py
@@ -19,11 +19,11 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import GlancesConfigEntry, GlancesDataUpdateCoordinator
from .const import CPU_ICON, DOMAIN
+from .coordinator import GlancesConfigEntry, GlancesDataUpdateCoordinator
@dataclass(frozen=True, kw_only=True)
@@ -288,7 +288,7 @@ SENSOR_TYPES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: GlancesConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Glances sensors."""
diff --git a/homeassistant/components/go2rtc/const.py b/homeassistant/components/go2rtc/const.py
index 234411936cb..491b2269043 100644
--- a/homeassistant/components/go2rtc/const.py
+++ b/homeassistant/components/go2rtc/const.py
@@ -6,4 +6,4 @@ CONF_DEBUG_UI = "debug_ui"
DEBUG_UI_URL_MESSAGE = "Url and debug_ui cannot be set at the same time."
HA_MANAGED_API_PORT = 11984
HA_MANAGED_URL = f"http://localhost:{HA_MANAGED_API_PORT}/"
-RECOMMENDED_VERSION = "1.9.8"
+RECOMMENDED_VERSION = "1.9.9"
diff --git a/homeassistant/components/goalzero/__init__.py b/homeassistant/components/goalzero/__init__.py
index 6698d1efc99..4a34927a585 100644
--- a/homeassistant/components/goalzero/__init__.py
+++ b/homeassistant/components/goalzero/__init__.py
@@ -36,7 +36,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: GoalZeroConfigEntry) ->
except exceptions.ConnectError as ex:
raise ConfigEntryNotReady(f"Failed to connect to device: {ex}") from ex
- entry.runtime_data = GoalZeroDataUpdateCoordinator(hass, api)
+ entry.runtime_data = GoalZeroDataUpdateCoordinator(hass, entry, api)
await entry.runtime_data.async_config_entry_first_refresh()
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/goalzero/binary_sensor.py b/homeassistant/components/goalzero/binary_sensor.py
index 6bd061879eb..86287dc35eb 100644
--- a/homeassistant/components/goalzero/binary_sensor.py
+++ b/homeassistant/components/goalzero/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import GoalZeroConfigEntry
from .entity import GoalZeroEntity
@@ -44,7 +44,7 @@ BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GoalZeroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Goal Zero Yeti sensor."""
async_add_entities(
diff --git a/homeassistant/components/goalzero/coordinator.py b/homeassistant/components/goalzero/coordinator.py
index 3c7cd967482..97a7f537759 100644
--- a/homeassistant/components/goalzero/coordinator.py
+++ b/homeassistant/components/goalzero/coordinator.py
@@ -18,11 +18,14 @@ class GoalZeroDataUpdateCoordinator(DataUpdateCoordinator[None]):
config_entry: GoalZeroConfigEntry
- def __init__(self, hass: HomeAssistant, api: Yeti) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: GoalZeroConfigEntry, api: Yeti
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/goalzero/sensor.py b/homeassistant/components/goalzero/sensor.py
index f565c216745..7b5f8955947 100644
--- a/homeassistant/components/goalzero/sensor.py
+++ b/homeassistant/components/goalzero/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import GoalZeroConfigEntry
@@ -131,7 +131,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GoalZeroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Goal Zero Yeti sensor."""
async_add_entities(
diff --git a/homeassistant/components/goalzero/switch.py b/homeassistant/components/goalzero/switch.py
index daff4ee5fec..00a1ad936d8 100644
--- a/homeassistant/components/goalzero/switch.py
+++ b/homeassistant/components/goalzero/switch.py
@@ -6,7 +6,7 @@ from typing import Any, cast
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import GoalZeroConfigEntry
from .entity import GoalZeroEntity
@@ -30,7 +30,7 @@ SWITCH_TYPES: tuple[SwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: GoalZeroConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Goal Zero Yeti switch."""
async_add_entities(
diff --git a/homeassistant/components/gogogate2/common.py b/homeassistant/components/gogogate2/common.py
index 52b1788c23e..8506414ca33 100644
--- a/homeassistant/components/gogogate2/common.py
+++ b/homeassistant/components/gogogate2/common.py
@@ -62,6 +62,7 @@ def get_data_update_coordinator(
config_entry_data[DATA_UPDATE_COORDINATOR] = DeviceDataUpdateCoordinator(
hass,
+ config_entry,
_LOGGER,
api,
# Name of the data. For logging purposes.
diff --git a/homeassistant/components/gogogate2/coordinator.py b/homeassistant/components/gogogate2/coordinator.py
index 7c15e8b1c32..c2e7cc47b46 100644
--- a/homeassistant/components/gogogate2/coordinator.py
+++ b/homeassistant/components/gogogate2/coordinator.py
@@ -8,6 +8,7 @@ import logging
from ismartgate import AbstractGateApi, GogoGate2InfoResponse, ISmartGateInfoResponse
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -18,9 +19,12 @@ class DeviceDataUpdateCoordinator(
):
"""Manages polling for state changes from the device."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
logger: logging.Logger,
api: AbstractGateApi,
*,
@@ -33,10 +37,10 @@ class DeviceDataUpdateCoordinator(
request_refresh_debouncer: Debouncer | None = None,
) -> None:
"""Initialize the data update coordinator."""
- DataUpdateCoordinator.__init__(
- self,
+ super().__init__(
hass,
logger,
+ config_entry=config_entry,
name=name,
update_interval=update_interval,
update_method=update_method,
diff --git a/homeassistant/components/gogogate2/cover.py b/homeassistant/components/gogogate2/cover.py
index 6bd38a0bc01..9492108d4b2 100644
--- a/homeassistant/components/gogogate2/cover.py
+++ b/homeassistant/components/gogogate2/cover.py
@@ -18,7 +18,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import cover_unique_id, get_data_update_coordinator
from .coordinator import DeviceDataUpdateCoordinator
@@ -28,7 +28,7 @@ from .entity import GoGoGate2Entity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the config entry."""
data_update_coordinator = get_data_update_coordinator(hass, config_entry)
diff --git a/homeassistant/components/gogogate2/sensor.py b/homeassistant/components/gogogate2/sensor.py
index c7740e24825..ce86ca9ac43 100644
--- a/homeassistant/components/gogogate2/sensor.py
+++ b/homeassistant/components/gogogate2/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import get_data_update_coordinator, sensor_unique_id
from .coordinator import DeviceDataUpdateCoordinator
@@ -26,7 +26,7 @@ SENSOR_ID_WIRED = "WIRE"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the config entry."""
data_update_coordinator = get_data_update_coordinator(hass, config_entry)
diff --git a/homeassistant/components/goodwe/button.py b/homeassistant/components/goodwe/button.py
index d3d96a19a76..e93b23570db 100644
--- a/homeassistant/components/goodwe/button.py
+++ b/homeassistant/components/goodwe/button.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, KEY_DEVICE_INFO, KEY_INVERTER
@@ -37,7 +37,7 @@ SYNCHRONIZE_CLOCK = GoodweButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the inverter button entities from a config entry."""
inverter = hass.data[DOMAIN][config_entry.entry_id][KEY_INVERTER]
diff --git a/homeassistant/components/goodwe/coordinator.py b/homeassistant/components/goodwe/coordinator.py
index a8ee7df6337..914ba3155b4 100644
--- a/homeassistant/components/goodwe/coordinator.py
+++ b/homeassistant/components/goodwe/coordinator.py
@@ -19,6 +19,8 @@ _LOGGER = logging.getLogger(__name__)
class GoodweUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Gather data for the energy device."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
@@ -29,6 +31,7 @@ class GoodweUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=entry.title,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/goodwe/number.py b/homeassistant/components/goodwe/number.py
index ce36bb35bf9..0a61ac19d64 100644
--- a/homeassistant/components/goodwe/number.py
+++ b/homeassistant/components/goodwe/number.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, KEY_DEVICE_INFO, KEY_INVERTER
@@ -87,7 +87,7 @@ NUMBERS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the inverter select entities from a config entry."""
inverter = hass.data[DOMAIN][config_entry.entry_id][KEY_INVERTER]
diff --git a/homeassistant/components/goodwe/select.py b/homeassistant/components/goodwe/select.py
index 4fa84c8401f..340e10bfa0f 100644
--- a/homeassistant/components/goodwe/select.py
+++ b/homeassistant/components/goodwe/select.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, KEY_DEVICE_INFO, KEY_INVERTER
@@ -40,7 +40,7 @@ OPERATION_MODE = SelectEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the inverter select entities from a config entry."""
inverter = hass.data[DOMAIN][config_entry.entry_id][KEY_INVERTER]
diff --git a/homeassistant/components/goodwe/sensor.py b/homeassistant/components/goodwe/sensor.py
index 5a88ac612da..d2dce2770e4 100644
--- a/homeassistant/components/goodwe/sensor.py
+++ b/homeassistant/components/goodwe/sensor.py
@@ -33,7 +33,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_point_in_time
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -166,7 +166,7 @@ TEXT_SENSOR = GoodweSensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GoodWe inverter from a config entry."""
entities: list[InverterSensor] = []
diff --git a/homeassistant/components/google/calendar.py b/homeassistant/components/google/calendar.py
index 5ac5dae616c..4ae8c8cce03 100644
--- a/homeassistant/components/google/calendar.py
+++ b/homeassistant/components/google/calendar.py
@@ -43,7 +43,7 @@ from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.exceptions import HomeAssistantError, PlatformNotReady
from homeassistant.helpers import entity_platform, entity_registry as er
from homeassistant.helpers.entity import generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -192,7 +192,7 @@ def _get_entity_descriptions(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the google calendar platform."""
calendar_service = hass.data[DOMAIN][config_entry.entry_id][DATA_SERVICE]
@@ -266,6 +266,7 @@ async def async_setup_entry(
if not entity_description.local_sync:
coordinator = CalendarQueryUpdateCoordinator(
hass,
+ config_entry,
calendar_service,
entity_description.name or entity_description.key,
calendar_id,
@@ -285,6 +286,7 @@ async def async_setup_entry(
)
coordinator = CalendarSyncUpdateCoordinator(
hass,
+ config_entry,
sync,
entity_description.name or entity_description.key,
)
@@ -381,9 +383,9 @@ class GoogleCalendarEntity(
for attendee in event.attendees
):
return False
-
- if event.event_type == EventTypeEnum.WORKING_LOCATION:
- return self.entity_description.working_location
+ is_working_location_event = event.event_type == EventTypeEnum.WORKING_LOCATION
+ if self.entity_description.working_location != is_working_location_event:
+ return False
if self._ignore_availability:
return True
return event.transparency == OPAQUE
diff --git a/homeassistant/components/google/coordinator.py b/homeassistant/components/google/coordinator.py
index 19198041c05..4a8a3d9f167 100644
--- a/homeassistant/components/google/coordinator.py
+++ b/homeassistant/components/google/coordinator.py
@@ -52,6 +52,7 @@ class CalendarSyncUpdateCoordinator(DataUpdateCoordinator[Timeline]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
sync: CalendarEventSyncManager,
name: str,
) -> None:
@@ -59,6 +60,7 @@ class CalendarSyncUpdateCoordinator(DataUpdateCoordinator[Timeline]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=name,
update_interval=MIN_TIME_BETWEEN_UPDATES,
)
@@ -111,6 +113,7 @@ class CalendarQueryUpdateCoordinator(DataUpdateCoordinator[list[Event]]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
calendar_service: GoogleCalendarService,
name: str,
calendar_id: str,
@@ -120,6 +123,7 @@ class CalendarQueryUpdateCoordinator(DataUpdateCoordinator[list[Event]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=name,
update_interval=MIN_TIME_BETWEEN_UPDATES,
)
diff --git a/homeassistant/components/google/manifest.json b/homeassistant/components/google/manifest.json
index bd04597e513..81fd2b07de4 100644
--- a/homeassistant/components/google/manifest.json
+++ b/homeassistant/components/google/manifest.json
@@ -7,5 +7,5 @@
"documentation": "https://www.home-assistant.io/integrations/google",
"iot_class": "cloud_polling",
"loggers": ["googleapiclient"],
- "requirements": ["gcal-sync==7.0.0", "oauth2client==4.1.3", "ical==8.3.0"]
+ "requirements": ["gcal-sync==7.0.0", "oauth2client==4.1.3", "ical==9.0.1"]
}
diff --git a/homeassistant/components/google_assistant/button.py b/homeassistant/components/google_assistant/button.py
index cf3a42e251b..58560d7b8d1 100644
--- a/homeassistant/components/google_assistant/button.py
+++ b/homeassistant/components/google_assistant/button.py
@@ -8,7 +8,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .const import CONF_PROJECT_ID, CONF_SERVICE_ACCOUNT, DATA_CONFIG, DOMAIN
@@ -18,7 +18,7 @@ from .http import GoogleConfig
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform."""
yaml_config: ConfigType = hass.data[DOMAIN][DATA_CONFIG]
diff --git a/homeassistant/components/google_assistant/const.py b/homeassistant/components/google_assistant/const.py
index 8132ecaae2c..71738c9d13e 100644
--- a/homeassistant/components/google_assistant/const.py
+++ b/homeassistant/components/google_assistant/const.py
@@ -14,6 +14,7 @@ from homeassistant.components import (
input_boolean,
input_button,
input_select,
+ lawn_mower,
light,
lock,
media_player,
@@ -58,6 +59,7 @@ DEFAULT_EXPOSED_DOMAINS = [
"humidifier",
"input_boolean",
"input_select",
+ "lawn_mower",
"light",
"lock",
"media_player",
@@ -88,6 +90,7 @@ TYPE_GATE = f"{PREFIX_TYPES}GATE"
TYPE_HUMIDIFIER = f"{PREFIX_TYPES}HUMIDIFIER"
TYPE_LIGHT = f"{PREFIX_TYPES}LIGHT"
TYPE_LOCK = f"{PREFIX_TYPES}LOCK"
+TYPE_MOWER = f"{PREFIX_TYPES}MOWER"
TYPE_OUTLET = f"{PREFIX_TYPES}OUTLET"
TYPE_RECEIVER = f"{PREFIX_TYPES}AUDIO_VIDEO_RECEIVER"
TYPE_SCENE = f"{PREFIX_TYPES}SCENE"
@@ -149,6 +152,7 @@ DOMAIN_TO_GOOGLE_TYPES = {
input_boolean.DOMAIN: TYPE_SWITCH,
input_button.DOMAIN: TYPE_SCENE,
input_select.DOMAIN: TYPE_SENSOR,
+ lawn_mower.DOMAIN: TYPE_MOWER,
light.DOMAIN: TYPE_LIGHT,
lock.DOMAIN: TYPE_LOCK,
media_player.DOMAIN: TYPE_SETTOP,
diff --git a/homeassistant/components/google_assistant/trait.py b/homeassistant/components/google_assistant/trait.py
index 44251a3be04..9edd340d7d9 100644
--- a/homeassistant/components/google_assistant/trait.py
+++ b/homeassistant/components/google_assistant/trait.py
@@ -21,6 +21,7 @@ from homeassistant.components import (
input_boolean,
input_button,
input_select,
+ lawn_mower,
light,
lock,
media_player,
@@ -42,6 +43,7 @@ from homeassistant.components.climate import ClimateEntityFeature
from homeassistant.components.cover import CoverEntityFeature
from homeassistant.components.fan import FanEntityFeature
from homeassistant.components.humidifier import HumidifierEntityFeature
+from homeassistant.components.lawn_mower import LawnMowerEntityFeature
from homeassistant.components.light import LightEntityFeature
from homeassistant.components.lock import LockState
from homeassistant.components.media_player import MediaPlayerEntityFeature, MediaType
@@ -714,7 +716,7 @@ class DockTrait(_Trait):
@staticmethod
def supported(domain, features, device_class, _):
"""Test if state is supported."""
- return domain == vacuum.DOMAIN
+ return domain in (vacuum.DOMAIN, lawn_mower.DOMAIN)
def sync_attributes(self) -> dict[str, Any]:
"""Return dock attributes for a sync request."""
@@ -722,17 +724,32 @@ class DockTrait(_Trait):
def query_attributes(self) -> dict[str, Any]:
"""Return dock query attributes."""
- return {"isDocked": self.state.state == vacuum.VacuumActivity.DOCKED}
+ domain = self.state.domain
+ state = self.state.state
+ if domain == vacuum.DOMAIN:
+ return {"isDocked": state == vacuum.VacuumActivity.DOCKED}
+ if domain == lawn_mower.DOMAIN:
+ return {"isDocked": state == lawn_mower.LawnMowerActivity.DOCKED}
+ raise NotImplementedError(f"Unsupported domain {domain}")
async def execute(self, command, data, params, challenge):
"""Execute a dock command."""
- await self.hass.services.async_call(
- self.state.domain,
- vacuum.SERVICE_RETURN_TO_BASE,
- {ATTR_ENTITY_ID: self.state.entity_id},
- blocking=not self.config.should_report_state,
- context=data.context,
- )
+ domain = self.state.domain
+ service: str | None = None
+
+ if domain == vacuum.DOMAIN:
+ service = vacuum.SERVICE_RETURN_TO_BASE
+ elif domain == lawn_mower.DOMAIN:
+ service = lawn_mower.SERVICE_DOCK
+
+ if service:
+ await self.hass.services.async_call(
+ self.state.domain,
+ service,
+ {ATTR_ENTITY_ID: self.state.entity_id},
+ blocking=not self.config.should_report_state,
+ context=data.context,
+ )
@register_trait
@@ -843,7 +860,7 @@ class StartStopTrait(_Trait):
@staticmethod
def supported(domain, features, device_class, _):
"""Test if state is supported."""
- if domain == vacuum.DOMAIN:
+ if domain in (vacuum.DOMAIN, lawn_mower.DOMAIN):
return True
if (
@@ -863,6 +880,12 @@ class StartStopTrait(_Trait):
& VacuumEntityFeature.PAUSE
!= 0
}
+ if domain == lawn_mower.DOMAIN:
+ return {
+ "pausable": self.state.attributes.get(ATTR_SUPPORTED_FEATURES, 0)
+ & LawnMowerEntityFeature.PAUSE
+ != 0
+ }
if domain in COVER_VALVE_DOMAINS:
return {}
@@ -878,6 +901,11 @@ class StartStopTrait(_Trait):
"isRunning": state == vacuum.VacuumActivity.CLEANING,
"isPaused": state == vacuum.VacuumActivity.PAUSED,
}
+ if domain == lawn_mower.DOMAIN:
+ return {
+ "isRunning": state == lawn_mower.LawnMowerActivity.MOWING,
+ "isPaused": state == lawn_mower.LawnMowerActivity.PAUSED,
+ }
if domain in COVER_VALVE_DOMAINS:
return {
@@ -896,46 +924,52 @@ class StartStopTrait(_Trait):
if domain == vacuum.DOMAIN:
await self._execute_vacuum(command, data, params, challenge)
return
+ if domain == lawn_mower.DOMAIN:
+ await self._execute_lawn_mower(command, data, params, challenge)
+ return
if domain in COVER_VALVE_DOMAINS:
await self._execute_cover_or_valve(command, data, params, challenge)
return
async def _execute_vacuum(self, command, data, params, challenge):
"""Execute a StartStop command."""
+ service: str | None = None
if command == COMMAND_START_STOP:
- if params["start"]:
- await self.hass.services.async_call(
- self.state.domain,
- vacuum.SERVICE_START,
- {ATTR_ENTITY_ID: self.state.entity_id},
- blocking=not self.config.should_report_state,
- context=data.context,
- )
- else:
- await self.hass.services.async_call(
- self.state.domain,
- vacuum.SERVICE_STOP,
- {ATTR_ENTITY_ID: self.state.entity_id},
- blocking=not self.config.should_report_state,
- context=data.context,
- )
+ service = vacuum.SERVICE_START if params["start"] else vacuum.SERVICE_STOP
elif command == COMMAND_PAUSE_UNPAUSE:
- if params["pause"]:
- await self.hass.services.async_call(
- self.state.domain,
- vacuum.SERVICE_PAUSE,
- {ATTR_ENTITY_ID: self.state.entity_id},
- blocking=not self.config.should_report_state,
- context=data.context,
- )
- else:
- await self.hass.services.async_call(
- self.state.domain,
- vacuum.SERVICE_START,
- {ATTR_ENTITY_ID: self.state.entity_id},
- blocking=not self.config.should_report_state,
- context=data.context,
- )
+ service = vacuum.SERVICE_PAUSE if params["pause"] else vacuum.SERVICE_START
+ if service:
+ await self.hass.services.async_call(
+ self.state.domain,
+ service,
+ {ATTR_ENTITY_ID: self.state.entity_id},
+ blocking=not self.config.should_report_state,
+ context=data.context,
+ )
+
+ async def _execute_lawn_mower(self, command, data, params, challenge):
+ """Execute a StartStop command."""
+ service: str | None = None
+ if command == COMMAND_START_STOP:
+ service = (
+ lawn_mower.SERVICE_START_MOWING
+ if params["start"]
+ else lawn_mower.SERVICE_DOCK
+ )
+ elif command == COMMAND_PAUSE_UNPAUSE:
+ service = (
+ lawn_mower.SERVICE_PAUSE
+ if params["pause"]
+ else lawn_mower.SERVICE_START_MOWING
+ )
+ if service:
+ await self.hass.services.async_call(
+ self.state.domain,
+ service,
+ {ATTR_ENTITY_ID: self.state.entity_id},
+ blocking=not self.config.should_report_state,
+ context=data.context,
+ )
async def _execute_cover_or_valve(self, command, data, params, challenge):
"""Execute a StartStop command."""
diff --git a/homeassistant/components/google_assistant_sdk/__init__.py b/homeassistant/components/google_assistant_sdk/__init__.py
index 4ea496f2824..a08d7554516 100644
--- a/homeassistant/components/google_assistant_sdk/__init__.py
+++ b/homeassistant/components/google_assistant_sdk/__init__.py
@@ -10,7 +10,7 @@ from google.oauth2.credentials import Credentials
import voluptuous as vol
from homeassistant.components import conversation
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_NAME, Platform
from homeassistant.core import (
HomeAssistant,
@@ -99,12 +99,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
hass.data[DOMAIN].pop(entry.entry_id)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
for service_name in hass.services.async_services_for_domain(DOMAIN):
hass.services.async_remove(DOMAIN, service_name)
diff --git a/homeassistant/components/google_cloud/stt.py b/homeassistant/components/google_cloud/stt.py
index ebca586d1a3..41c5a6710b7 100644
--- a/homeassistant/components/google_cloud/stt.py
+++ b/homeassistant/components/google_cloud/stt.py
@@ -22,7 +22,7 @@ from homeassistant.components.stt import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_SERVICE_ACCOUNT_INFO,
@@ -38,7 +38,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Google Cloud speech platform via config entry."""
service_account_info = config_entry.data[CONF_SERVICE_ACCOUNT_INFO]
diff --git a/homeassistant/components/google_cloud/tts.py b/homeassistant/components/google_cloud/tts.py
index 7f22dda4faf..1f5f838b593 100644
--- a/homeassistant/components/google_cloud/tts.py
+++ b/homeassistant/components/google_cloud/tts.py
@@ -21,7 +21,7 @@ from homeassistant.components.tts import (
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import (
@@ -88,7 +88,7 @@ async def async_get_engine(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Google Cloud text-to-speech."""
service_account_info = config_entry.data[CONF_SERVICE_ACCOUNT_INFO]
diff --git a/homeassistant/components/google_drive/__init__.py b/homeassistant/components/google_drive/__init__.py
index af93956931a..d5252bd01ea 100644
--- a/homeassistant/components/google_drive/__init__.py
+++ b/homeassistant/components/google_drive/__init__.py
@@ -49,6 +49,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: GoogleDriveConfigEntry)
except GoogleDriveApiError as err:
raise ConfigEntryNotReady from err
+ def async_notify_backup_listeners() -> None:
+ for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
+ listener()
+
+ entry.async_on_unload(entry.async_on_state_change(async_notify_backup_listeners))
+
return True
@@ -56,10 +62,4 @@ async def async_unload_entry(
hass: HomeAssistant, entry: GoogleDriveConfigEntry
) -> bool:
"""Unload a config entry."""
- hass.loop.call_soon(_notify_backup_listeners, hass)
return True
-
-
-def _notify_backup_listeners(hass: HomeAssistant) -> None:
- for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
- listener()
diff --git a/homeassistant/components/google_drive/api.py b/homeassistant/components/google_drive/api.py
index 475eddb6231..c21d42e0f3a 100644
--- a/homeassistant/components/google_drive/api.py
+++ b/homeassistant/components/google_drive/api.py
@@ -146,9 +146,10 @@ class DriveClient:
backup.backup_id,
backup_metadata,
)
- await self._api.upload_file(
+ await self._api.resumable_upload_file(
backup_metadata,
open_stream,
+ backup.size,
timeout=ClientTimeout(total=_UPLOAD_AND_DOWNLOAD_TIMEOUT),
)
_LOGGER.debug(
diff --git a/homeassistant/components/google_drive/application_credentials.py b/homeassistant/components/google_drive/application_credentials.py
index 1c4421623d4..8bcab2b039c 100644
--- a/homeassistant/components/google_drive/application_credentials.py
+++ b/homeassistant/components/google_drive/application_credentials.py
@@ -2,7 +2,10 @@
from homeassistant.components.application_credentials import AuthorizationServer
from homeassistant.core import HomeAssistant
-from homeassistant.helpers import config_entry_oauth2_flow
+from homeassistant.helpers.config_entry_oauth2_flow import (
+ AUTH_CALLBACK_PATH,
+ MY_AUTH_CALLBACK_PATH,
+)
async def async_get_authorization_server(hass: HomeAssistant) -> AuthorizationServer:
@@ -15,9 +18,14 @@ async def async_get_authorization_server(hass: HomeAssistant) -> AuthorizationSe
async def async_get_description_placeholders(hass: HomeAssistant) -> dict[str, str]:
"""Return description placeholders for the credentials dialog."""
+ if "my" in hass.config.components:
+ redirect_url = MY_AUTH_CALLBACK_PATH
+ else:
+ ha_host = hass.config.external_url or "https://YOUR_DOMAIN:PORT"
+ redirect_url = f"{ha_host}{AUTH_CALLBACK_PATH}"
return {
"oauth_consent_url": "https://console.cloud.google.com/apis/credentials/consent",
"more_info_url": "https://www.home-assistant.io/integrations/google_drive/",
"oauth_creds_url": "https://console.cloud.google.com/apis/credentials",
- "redirect_url": config_entry_oauth2_flow.async_get_redirect_uri(hass),
+ "redirect_url": redirect_url,
}
diff --git a/homeassistant/components/google_drive/backup.py b/homeassistant/components/google_drive/backup.py
index 73e5902f8f5..a4b7fc956ce 100644
--- a/homeassistant/components/google_drive/backup.py
+++ b/homeassistant/components/google_drive/backup.py
@@ -8,7 +8,12 @@ from typing import Any
from google_drive_api.exceptions import GoogleDriveApiError
-from homeassistant.components.backup import AgentBackup, BackupAgent, BackupAgentError
+from homeassistant.components.backup import (
+ AgentBackup,
+ BackupAgent,
+ BackupAgentError,
+ BackupNotFound,
+)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.aiohttp_client import ChunkAsyncStreamIterator
@@ -93,13 +98,13 @@ class GoogleDriveBackupAgent(BackupAgent):
self,
backup_id: str,
**kwargs: Any,
- ) -> AgentBackup | None:
+ ) -> AgentBackup:
"""Return a backup."""
backups = await self.async_list_backups()
for backup in backups:
if backup.backup_id == backup_id:
return backup
- return None
+ raise BackupNotFound(f"Backup {backup_id} not found")
async def async_download_backup(
self,
@@ -120,7 +125,7 @@ class GoogleDriveBackupAgent(BackupAgent):
return ChunkAsyncStreamIterator(stream)
except (GoogleDriveApiError, HomeAssistantError, TimeoutError) as err:
raise BackupAgentError(f"Failed to download backup: {err}") from err
- raise BackupAgentError("Backup not found")
+ raise BackupNotFound(f"Backup {backup_id} not found")
async def async_delete_backup(
self,
@@ -138,5 +143,7 @@ class GoogleDriveBackupAgent(BackupAgent):
_LOGGER.debug("Deleting file_id: %s", file_id)
await self._client.async_delete(file_id)
_LOGGER.debug("Deleted backup_id: %s", backup_id)
+ return
except (GoogleDriveApiError, HomeAssistantError, TimeoutError) as err:
raise BackupAgentError(f"Failed to delete backup: {err}") from err
+ raise BackupNotFound(f"Backup {backup_id} not found")
diff --git a/homeassistant/components/google_drive/manifest.json b/homeassistant/components/google_drive/manifest.json
index a1abb9b260a..6b199a5d3eb 100644
--- a/homeassistant/components/google_drive/manifest.json
+++ b/homeassistant/components/google_drive/manifest.json
@@ -10,5 +10,5 @@
"iot_class": "cloud_polling",
"loggers": ["google_drive_api"],
"quality_scale": "platinum",
- "requirements": ["python-google-drive-api==0.0.2"]
+ "requirements": ["python-google-drive-api==0.1.0"]
}
diff --git a/homeassistant/components/google_generative_ai_conversation/__init__.py b/homeassistant/components/google_generative_ai_conversation/__init__.py
index a5c55c2099d..c32d7b5ddea 100644
--- a/homeassistant/components/google_generative_ai_conversation/__init__.py
+++ b/homeassistant/components/google_generative_ai_conversation/__init__.py
@@ -5,11 +5,9 @@ from __future__ import annotations
import mimetypes
from pathlib import Path
-from google.ai import generativelanguage_v1beta
-from google.api_core.client_options import ClientOptions
-from google.api_core.exceptions import ClientError, DeadlineExceeded, GoogleAPIError
-import google.generativeai as genai
-import google.generativeai.types as genai_types
+from google import genai # type: ignore[attr-defined]
+from google.genai.errors import APIError, ClientError
+from requests.exceptions import Timeout
import voluptuous as vol
from homeassistant.config_entries import ConfigEntry
@@ -27,59 +25,91 @@ from homeassistant.exceptions import (
HomeAssistantError,
)
from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType
-from .const import CONF_CHAT_MODEL, CONF_PROMPT, DOMAIN, RECOMMENDED_CHAT_MODEL
+from .const import (
+ CONF_CHAT_MODEL,
+ CONF_PROMPT,
+ DOMAIN,
+ RECOMMENDED_CHAT_MODEL,
+ TIMEOUT_MILLIS,
+)
SERVICE_GENERATE_CONTENT = "generate_content"
CONF_IMAGE_FILENAME = "image_filename"
+CONF_FILENAMES = "filenames"
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
PLATFORMS = (Platform.CONVERSATION,)
+type GoogleGenerativeAIConfigEntry = ConfigEntry[genai.Client]
+
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up Google Generative AI Conversation."""
async def generate_content(call: ServiceCall) -> ServiceResponse:
"""Generate content from text and optionally images."""
- prompt_parts = [call.data[CONF_PROMPT]]
- image_filenames = call.data[CONF_IMAGE_FILENAME]
- for image_filename in image_filenames:
- if not hass.config.is_allowed_path(image_filename):
- raise HomeAssistantError(
- f"Cannot read `{image_filename}`, no access to path; "
- "`allowlist_external_dirs` may need to be adjusted in "
- "`configuration.yaml`"
- )
- if not Path(image_filename).exists():
- raise HomeAssistantError(f"`{image_filename}` does not exist")
- mime_type, _ = mimetypes.guess_type(image_filename)
- if mime_type is None or not mime_type.startswith("image"):
- raise HomeAssistantError(f"`{image_filename}` is not an image")
- prompt_parts.append(
- {
- "mime_type": mime_type,
- "data": await hass.async_add_executor_job(
- Path(image_filename).read_bytes
- ),
- }
+
+ if call.data[CONF_IMAGE_FILENAME]:
+ # Deprecated in 2025.3, to remove in 2025.9
+ async_create_issue(
+ hass,
+ DOMAIN,
+ "deprecated_image_filename_parameter",
+ breaks_in_ha_version="2025.9.0",
+ is_fixable=False,
+ severity=IssueSeverity.WARNING,
+ translation_key="deprecated_image_filename_parameter",
)
- model = genai.GenerativeModel(model_name=RECOMMENDED_CHAT_MODEL)
+ prompt_parts = [call.data[CONF_PROMPT]]
+
+ config_entry: GoogleGenerativeAIConfigEntry = (
+ hass.config_entries.async_loaded_entries(DOMAIN)[0]
+ )
+
+ client = config_entry.runtime_data
+
+ def append_files_to_prompt():
+ image_filenames = call.data[CONF_IMAGE_FILENAME]
+ filenames = call.data[CONF_FILENAMES]
+ for filename in set(image_filenames + filenames):
+ if not hass.config.is_allowed_path(filename):
+ raise HomeAssistantError(
+ f"Cannot read `{filename}`, no access to path; "
+ "`allowlist_external_dirs` may need to be adjusted in "
+ "`configuration.yaml`"
+ )
+ if not Path(filename).exists():
+ raise HomeAssistantError(f"`{filename}` does not exist")
+ mimetype = mimetypes.guess_type(filename)[0]
+ with open(filename, "rb") as file:
+ uploaded_file = client.files.upload(
+ file=file, config={"mime_type": mimetype}
+ )
+ prompt_parts.append(uploaded_file)
+
+ await hass.async_add_executor_job(append_files_to_prompt)
try:
- response = await model.generate_content_async(prompt_parts)
+ response = await client.aio.models.generate_content(
+ model=RECOMMENDED_CHAT_MODEL, contents=prompt_parts
+ )
except (
- GoogleAPIError,
+ APIError,
ValueError,
- genai_types.BlockedPromptException,
- genai_types.StopCandidateException,
) as err:
raise HomeAssistantError(f"Error generating content: {err}") from err
- if not response.parts:
- raise HomeAssistantError("Error generating content")
+ if response.prompt_feedback:
+ raise HomeAssistantError(
+ f"Error generating content due to content violations, reason: {response.prompt_feedback.block_reason_message}"
+ )
+
+ if not response.candidates[0].content.parts:
+ raise HomeAssistantError("Unknown error generating content")
return {"text": response.text}
@@ -93,6 +123,9 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
vol.Optional(CONF_IMAGE_FILENAME, default=[]): vol.All(
cv.ensure_list, [cv.string]
),
+ vol.Optional(CONF_FILENAMES, default=[]): vol.All(
+ cv.ensure_list, [cv.string]
+ ),
}
),
supports_response=SupportsResponse.ONLY,
@@ -100,30 +133,34 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
return True
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(
+ hass: HomeAssistant, entry: GoogleGenerativeAIConfigEntry
+) -> bool:
"""Set up Google Generative AI Conversation from a config entry."""
- genai.configure(api_key=entry.data[CONF_API_KEY])
try:
- client = generativelanguage_v1beta.ModelServiceAsyncClient(
- client_options=ClientOptions(api_key=entry.data[CONF_API_KEY])
+ client = genai.Client(api_key=entry.data[CONF_API_KEY])
+ await client.aio.models.get(
+ model=entry.options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL),
+ config={"http_options": {"timeout": TIMEOUT_MILLIS}},
)
- await client.get_model(
- name=entry.options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL), timeout=5.0
- )
- except (GoogleAPIError, ValueError) as err:
- if isinstance(err, ClientError) and err.reason == "API_KEY_INVALID":
- raise ConfigEntryAuthFailed(err) from err
- if isinstance(err, DeadlineExceeded):
+ except (APIError, Timeout) as err:
+ if isinstance(err, ClientError) and "API_KEY_INVALID" in str(err):
+ raise ConfigEntryAuthFailed(err.message) from err
+ if isinstance(err, Timeout):
raise ConfigEntryNotReady(err) from err
raise ConfigEntryError(err) from err
+ else:
+ entry.runtime_data = client
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, entry: GoogleGenerativeAIConfigEntry
+) -> bool:
"""Unload GoogleGenerativeAI."""
if not await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
return False
diff --git a/homeassistant/components/google_generative_ai_conversation/config_flow.py b/homeassistant/components/google_generative_ai_conversation/config_flow.py
index 83eec25ed15..00a016143f4 100644
--- a/homeassistant/components/google_generative_ai_conversation/config_flow.py
+++ b/homeassistant/components/google_generative_ai_conversation/config_flow.py
@@ -3,15 +3,13 @@
from __future__ import annotations
from collections.abc import Mapping
-from functools import partial
import logging
from types import MappingProxyType
from typing import Any
-from google.ai import generativelanguage_v1beta
-from google.api_core.client_options import ClientOptions
-from google.api_core.exceptions import ClientError, GoogleAPIError
-import google.generativeai as genai
+from google import genai # type: ignore[attr-defined]
+from google.genai.errors import APIError, ClientError
+from requests.exceptions import Timeout
import voluptuous as vol
from homeassistant.config_entries import (
@@ -53,6 +51,7 @@ from .const import (
RECOMMENDED_TEMPERATURE,
RECOMMENDED_TOP_K,
RECOMMENDED_TOP_P,
+ TIMEOUT_MILLIS,
)
_LOGGER = logging.getLogger(__name__)
@@ -70,15 +69,20 @@ RECOMMENDED_OPTIONS = {
}
-async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> None:
+async def validate_input(data: dict[str, Any]) -> None:
"""Validate the user input allows us to connect.
Data has the keys from STEP_USER_DATA_SCHEMA with values provided by the user.
"""
- client = generativelanguage_v1beta.ModelServiceAsyncClient(
- client_options=ClientOptions(api_key=data[CONF_API_KEY])
+ client = genai.Client(api_key=data[CONF_API_KEY])
+ await client.aio.models.list(
+ config={
+ "http_options": {
+ "timeout": TIMEOUT_MILLIS,
+ },
+ "query_base": True,
+ }
)
- await client.list_models(timeout=5.0)
class GoogleGenerativeAIConfigFlow(ConfigFlow, domain=DOMAIN):
@@ -93,9 +97,9 @@ class GoogleGenerativeAIConfigFlow(ConfigFlow, domain=DOMAIN):
errors: dict[str, str] = {}
if user_input is not None:
try:
- await validate_input(self.hass, user_input)
- except GoogleAPIError as err:
- if isinstance(err, ClientError) and err.reason == "API_KEY_INVALID":
+ await validate_input(user_input)
+ except (APIError, Timeout) as err:
+ if isinstance(err, ClientError) and "API_KEY_INVALID" in str(err):
errors["base"] = "invalid_auth"
else:
errors["base"] = "cannot_connect"
@@ -166,6 +170,7 @@ class GoogleGenerativeAIOptionsFlow(OptionsFlow):
self.last_rendered_recommended = config_entry.options.get(
CONF_RECOMMENDED, False
)
+ self._genai_client = config_entry.runtime_data
async def async_step_init(
self, user_input: dict[str, Any] | None = None
@@ -188,7 +193,9 @@ class GoogleGenerativeAIOptionsFlow(OptionsFlow):
CONF_LLM_HASS_API: user_input[CONF_LLM_HASS_API],
}
- schema = await google_generative_ai_config_option_schema(self.hass, options)
+ schema = await google_generative_ai_config_option_schema(
+ self.hass, options, self._genai_client
+ )
return self.async_show_form(
step_id="init",
data_schema=vol.Schema(schema),
@@ -198,6 +205,7 @@ class GoogleGenerativeAIOptionsFlow(OptionsFlow):
async def google_generative_ai_config_option_schema(
hass: HomeAssistant,
options: dict[str, Any] | MappingProxyType[str, Any],
+ genai_client: genai.Client,
) -> dict:
"""Return a schema for Google Generative AI completion options."""
hass_apis: list[SelectOptionDict] = [
@@ -236,18 +244,21 @@ async def google_generative_ai_config_option_schema(
if options.get(CONF_RECOMMENDED):
return schema
- api_models = await hass.async_add_executor_job(partial(genai.list_models))
-
+ api_models_pager = await genai_client.aio.models.list(config={"query_base": True})
+ api_models = [api_model async for api_model in api_models_pager]
models = [
SelectOptionDict(
label=api_model.display_name,
value=api_model.name,
)
- for api_model in sorted(api_models, key=lambda x: x.display_name)
+ for api_model in sorted(api_models, key=lambda x: x.display_name or "")
if (
api_model.name != "models/gemini-1.0-pro" # duplicate of gemini-pro
+ and api_model.display_name
+ and api_model.name
+ and api_model.supported_actions
and "vision" not in api_model.name
- and "generateContent" in api_model.supported_generation_methods
+ and "generateContent" in api_model.supported_actions
)
]
diff --git a/homeassistant/components/google_generative_ai_conversation/const.py b/homeassistant/components/google_generative_ai_conversation/const.py
index 4d83b935528..35834f6e7f9 100644
--- a/homeassistant/components/google_generative_ai_conversation/const.py
+++ b/homeassistant/components/google_generative_ai_conversation/const.py
@@ -22,3 +22,5 @@ CONF_HATE_BLOCK_THRESHOLD = "hate_block_threshold"
CONF_SEXUAL_BLOCK_THRESHOLD = "sexual_block_threshold"
CONF_DANGEROUS_BLOCK_THRESHOLD = "dangerous_block_threshold"
RECOMMENDED_HARM_BLOCK_THRESHOLD = "BLOCK_MEDIUM_AND_ABOVE"
+
+TIMEOUT_MILLIS = 10000
diff --git a/homeassistant/components/google_generative_ai_conversation/conversation.py b/homeassistant/components/google_generative_ai_conversation/conversation.py
index 8a6c5563601..e35346cc745 100644
--- a/homeassistant/components/google_generative_ai_conversation/conversation.py
+++ b/homeassistant/components/google_generative_ai_conversation/conversation.py
@@ -6,11 +6,18 @@ import codecs
from collections.abc import Callable
from typing import Any, Literal, cast
-from google.api_core.exceptions import GoogleAPIError
-import google.generativeai as genai
-from google.generativeai import protos
-import google.generativeai.types as genai_types
-from google.protobuf.json_format import MessageToDict
+from google.genai.errors import APIError
+from google.genai.types import (
+ AutomaticFunctionCallingConfig,
+ Content,
+ FunctionDeclaration,
+ GenerateContentConfig,
+ HarmCategory,
+ Part,
+ SafetySetting,
+ Schema,
+ Tool,
+)
from voluptuous_openapi import convert
from homeassistant.components import assist_pipeline, conversation
@@ -18,8 +25,8 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_LLM_HASS_API, MATCH_ALL
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers import chat_session, device_registry as dr, intent, llm
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers import device_registry as dr, intent, llm
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_CHAT_MODEL,
@@ -49,7 +56,7 @@ MAX_TOOL_ITERATIONS = 10
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up conversation entities."""
agent = GoogleGenerativeAIConversationEntity(config_entry)
@@ -57,21 +64,30 @@ async def async_setup_entry(
SUPPORTED_SCHEMA_KEYS = {
+ # Gemini API does not support all of the OpenAPI schema
+ # SoT: https://ai.google.dev/api/caching#Schema
"type",
"format",
"description",
"nullable",
"enum",
- "items",
+ "max_items",
+ "min_items",
"properties",
"required",
+ "items",
}
-def _format_schema(schema: dict[str, Any]) -> dict[str, Any]:
- """Format the schema to protobuf."""
- if (subschemas := schema.get("anyOf")) or (subschemas := schema.get("allOf")):
- for subschema in subschemas: # Gemini API does not support anyOf and allOf keys
+def _camel_to_snake(name: str) -> str:
+ """Convert camel case to snake case."""
+ return "".join(["_" + c.lower() if c.isupper() else c for c in name]).lstrip("_")
+
+
+def _format_schema(schema: dict[str, Any]) -> Schema:
+ """Format the schema to be compatible with Gemini API."""
+ if subschemas := schema.get("allOf"):
+ for subschema in subschemas: # Gemini API does not support allOf keys
if "type" in subschema: # Fallback to first subschema with 'type' field
return _format_schema(subschema)
return _format_schema(
@@ -80,42 +96,47 @@ def _format_schema(schema: dict[str, Any]) -> dict[str, Any]:
result = {}
for key, val in schema.items():
+ key = _camel_to_snake(key)
if key not in SUPPORTED_SCHEMA_KEYS:
continue
if key == "type":
- key = "type_"
val = val.upper()
elif key == "format":
- if schema.get("type") == "string" and val != "enum":
+ # Gemini API does not support all formats, see: https://ai.google.dev/api/caching#Schema
+ # formats that are not supported are ignored
+ if schema.get("type") == "string" and val not in ("enum", "date-time"):
continue
- if schema.get("type") not in ("number", "integer", "string"):
+ if schema.get("type") == "number" and val not in ("float", "double"):
+ continue
+ if schema.get("type") == "integer" and val not in ("int32", "int64"):
+ continue
+ if schema.get("type") not in ("string", "number", "integer"):
continue
- key = "format_"
elif key == "items":
val = _format_schema(val)
elif key == "properties":
val = {k: _format_schema(v) for k, v in val.items()}
result[key] = val
- if result.get("enum") and result.get("type_") != "STRING":
+ if result.get("enum") and result.get("type") != "STRING":
# enum is only allowed for STRING type. This is safe as long as the schema
# contains vol.Coerce for the respective type, for example:
# vol.All(vol.Coerce(int), vol.In([1, 2, 3]))
- result["type_"] = "STRING"
+ result["type"] = "STRING"
result["enum"] = [str(item) for item in result["enum"]]
- if result.get("type_") == "OBJECT" and not result.get("properties"):
+ if result.get("type") == "OBJECT" and not result.get("properties"):
# An object with undefined properties is not supported by Gemini API.
# Fallback to JSON string. This will probably fail for most tools that want it,
# but we don't have a better fallback strategy so far.
- result["properties"] = {"json": {"type_": "STRING"}}
+ result["properties"] = {"json": {"type": "STRING"}}
result["required"] = []
- return result
+ return cast(Schema, result)
def _format_tool(
tool: llm.Tool, custom_serializer: Callable[[Any], Any] | None
-) -> dict[str, Any]:
+) -> Tool:
"""Format tool specification."""
if tool.parameters.schema:
@@ -125,16 +146,14 @@ def _format_tool(
else:
parameters = None
- return protos.Tool(
- {
- "function_declarations": [
- {
- "name": tool.name,
- "description": tool.description,
- "parameters": parameters,
- }
- ]
- }
+ return Tool(
+ function_declarations=[
+ FunctionDeclaration(
+ name=tool.name,
+ description=tool.description,
+ parameters=parameters,
+ )
+ ]
)
@@ -151,14 +170,12 @@ def _escape_decode(value: Any) -> Any:
def _create_google_tool_response_content(
content: list[conversation.ToolResultContent],
-) -> protos.Content:
+) -> Content:
"""Create a Google tool response content."""
- return protos.Content(
+ return Content(
parts=[
- protos.Part(
- function_response=protos.FunctionResponse(
- name=tool_result.tool_name, response=tool_result.tool_result
- )
+ Part.from_function_response(
+ name=tool_result.tool_name, response=tool_result.tool_result
)
for tool_result in content
]
@@ -169,33 +186,36 @@ def _convert_content(
content: conversation.UserContent
| conversation.AssistantContent
| conversation.SystemContent,
-) -> genai_types.ContentDict:
+) -> Content:
"""Convert HA content to Google content."""
- if content.role != "assistant" or not content.tool_calls: # type: ignore[union-attr]
+ if content.role != "assistant" or not content.tool_calls:
role = "model" if content.role == "assistant" else content.role
- return {"role": role, "parts": content.content}
+ return Content(
+ role=role,
+ parts=[
+ Part.from_text(text=content.content if content.content else ""),
+ ],
+ )
# Handle the Assistant content with tool calls.
assert type(content) is conversation.AssistantContent
- parts = []
+ parts: list[Part] = []
if content.content:
- parts.append(protos.Part(text=content.content))
+ parts.append(Part.from_text(text=content.content))
if content.tool_calls:
parts.extend(
[
- protos.Part(
- function_call=protos.FunctionCall(
- name=tool_call.tool_name,
- args=_escape_decode(tool_call.tool_args),
- )
+ Part.from_function_call(
+ name=tool_call.tool_name,
+ args=_escape_decode(tool_call.tool_args),
)
for tool_call in content.tool_calls
]
)
- return protos.Content({"role": "model", "parts": parts})
+ return Content(role="model", parts=parts)
class GoogleGenerativeAIConversationEntity(
@@ -209,6 +229,7 @@ class GoogleGenerativeAIConversationEntity(
def __init__(self, entry: ConfigEntry) -> None:
"""Initialize the agent."""
self.entry = entry
+ self._genai_client = entry.runtime_data
self._attr_unique_id = entry.entry_id
self._attr_device_info = dr.DeviceInfo(
identifiers={(DOMAIN, entry.entry_id)},
@@ -243,17 +264,12 @@ class GoogleGenerativeAIConversationEntity(
conversation.async_unset_agent(self.hass, self.entry)
await super().async_will_remove_from_hass()
- async def async_process(
- self, user_input: conversation.ConversationInput
- ) -> conversation.ConversationResult:
- """Process a sentence."""
- with (
- chat_session.async_get_chat_session(
- self.hass, user_input.conversation_id
- ) as session,
- conversation.async_get_chat_log(self.hass, session, user_input) as chat_log,
- ):
- return await self._async_handle_message(user_input, chat_log)
+ def _fix_tool_name(self, tool_name: str) -> str:
+ """Fix tool name if needed."""
+ # The Gemini 2.0+ tokenizer seemingly has a issue with the HassListAddItem tool
+ # name. This makes sure when it incorrectly changes the name, that we change it
+ # back for HA to call.
+ return tool_name if tool_name != "HasListAddItem" else "HassListAddItem"
async def _async_handle_message(
self,
@@ -261,8 +277,6 @@ class GoogleGenerativeAIConversationEntity(
chat_log: conversation.ChatLog,
) -> conversation.ConversationResult:
"""Call the API."""
-
- assert user_input.agent_id
options = self.entry.options
try:
@@ -275,7 +289,7 @@ class GoogleGenerativeAIConversationEntity(
except conversation.ConverseError as err:
return err.as_conversation_result()
- tools: list[dict[str, Any]] | None = None
+ tools: list[Tool | Callable[..., Any]] | None = None
if chat_log.llm_api:
tools = [
_format_tool(tool, chat_log.llm_api.custom_serializer)
@@ -290,116 +304,138 @@ class GoogleGenerativeAIConversationEntity(
"gemini-1.0" not in model_name and "gemini-pro" not in model_name
)
- prompt = chat_log.content[0].content # type: ignore[union-attr]
- messages: list[genai_types.ContentDict] = []
+ prompt_content = cast(
+ conversation.SystemContent,
+ chat_log.content[0],
+ )
+
+ if prompt_content.content:
+ prompt = prompt_content.content
+ else:
+ raise HomeAssistantError("Invalid prompt content")
+
+ messages: list[Content] = []
# Google groups tool results, we do not. Group them before sending.
tool_results: list[conversation.ToolResultContent] = []
- for chat_content in chat_log.content[1:]:
+ for chat_content in chat_log.content[1:-1]:
if chat_content.role == "tool_result":
- # mypy doesn't like picking a type based on checking shared property 'role'
- tool_results.append(cast(conversation.ToolResultContent, chat_content))
+ tool_results.append(chat_content)
continue
if tool_results:
messages.append(_create_google_tool_response_content(tool_results))
tool_results.clear()
- messages.append(
- _convert_content(
- cast(
- conversation.UserContent
- | conversation.SystemContent
- | conversation.AssistantContent,
- chat_content,
- )
- )
- )
+ messages.append(_convert_content(chat_content))
if tool_results:
messages.append(_create_google_tool_response_content(tool_results))
-
- model = genai.GenerativeModel(
- model_name=model_name,
- generation_config={
- "temperature": self.entry.options.get(
- CONF_TEMPERATURE, RECOMMENDED_TEMPERATURE
+ generateContentConfig = GenerateContentConfig(
+ temperature=self.entry.options.get(
+ CONF_TEMPERATURE, RECOMMENDED_TEMPERATURE
+ ),
+ top_k=self.entry.options.get(CONF_TOP_K, RECOMMENDED_TOP_K),
+ top_p=self.entry.options.get(CONF_TOP_P, RECOMMENDED_TOP_P),
+ max_output_tokens=self.entry.options.get(
+ CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS
+ ),
+ safety_settings=[
+ SafetySetting(
+ category=HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+ threshold=self.entry.options.get(
+ CONF_HATE_BLOCK_THRESHOLD, RECOMMENDED_HARM_BLOCK_THRESHOLD
+ ),
),
- "top_p": self.entry.options.get(CONF_TOP_P, RECOMMENDED_TOP_P),
- "top_k": self.entry.options.get(CONF_TOP_K, RECOMMENDED_TOP_K),
- "max_output_tokens": self.entry.options.get(
- CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS
+ SafetySetting(
+ category=HarmCategory.HARM_CATEGORY_HARASSMENT,
+ threshold=self.entry.options.get(
+ CONF_HARASSMENT_BLOCK_THRESHOLD,
+ RECOMMENDED_HARM_BLOCK_THRESHOLD,
+ ),
),
- },
- safety_settings={
- "HARASSMENT": self.entry.options.get(
- CONF_HARASSMENT_BLOCK_THRESHOLD, RECOMMENDED_HARM_BLOCK_THRESHOLD
+ SafetySetting(
+ category=HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+ threshold=self.entry.options.get(
+ CONF_DANGEROUS_BLOCK_THRESHOLD, RECOMMENDED_HARM_BLOCK_THRESHOLD
+ ),
),
- "HATE": self.entry.options.get(
- CONF_HATE_BLOCK_THRESHOLD, RECOMMENDED_HARM_BLOCK_THRESHOLD
+ SafetySetting(
+ category=HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+ threshold=self.entry.options.get(
+ CONF_SEXUAL_BLOCK_THRESHOLD, RECOMMENDED_HARM_BLOCK_THRESHOLD
+ ),
),
- "SEXUAL": self.entry.options.get(
- CONF_SEXUAL_BLOCK_THRESHOLD, RECOMMENDED_HARM_BLOCK_THRESHOLD
- ),
- "DANGEROUS": self.entry.options.get(
- CONF_DANGEROUS_BLOCK_THRESHOLD, RECOMMENDED_HARM_BLOCK_THRESHOLD
- ),
- },
+ ],
tools=tools or None,
system_instruction=prompt if supports_system_instruction else None,
+ automatic_function_calling=AutomaticFunctionCallingConfig(
+ disable=True, maximum_remote_calls=None
+ ),
)
if not supports_system_instruction:
messages = [
- {"role": "user", "parts": prompt},
- {"role": "model", "parts": "Ok"},
+ Content(role="user", parts=[Part.from_text(text=prompt)]),
+ Content(role="model", parts=[Part.from_text(text="Ok")]),
*messages,
]
-
- chat = model.start_chat(history=messages)
- chat_request = user_input.text
+ chat = self._genai_client.aio.chats.create(
+ model=model_name, history=messages, config=generateContentConfig
+ )
+ chat_request: str | Content = user_input.text
# To prevent infinite loops, we limit the number of iterations
for _iteration in range(MAX_TOOL_ITERATIONS):
try:
- chat_response = await chat.send_message_async(chat_request)
- except (
- GoogleAPIError,
- ValueError,
- genai_types.BlockedPromptException,
- genai_types.StopCandidateException,
- ) as err:
- LOGGER.error("Error sending message: %s %s", type(err), err)
+ chat_response = await chat.send_message(message=chat_request)
- if isinstance(
- err, genai_types.StopCandidateException
- ) and "finish_reason: SAFETY\n" in str(err):
- error = "The message got blocked by your safety settings"
- else:
- error = (
- f"Sorry, I had a problem talking to Google Generative AI: {err}"
+ if chat_response.prompt_feedback:
+ raise HomeAssistantError(
+ f"The message got blocked due to content violations, reason: {chat_response.prompt_feedback.block_reason_message}"
)
+ except (
+ APIError,
+ ValueError,
+ ) as err:
+ LOGGER.error("Error sending message: %s %s", type(err), err)
+ error = f"Sorry, I had a problem talking to Google Generative AI: {err}"
raise HomeAssistantError(error) from err
- LOGGER.debug("Response: %s", chat_response.parts)
- if not chat_response.parts:
+ if (usage_metadata := chat_response.usage_metadata) is not None:
+ chat_log.async_trace(
+ {
+ "stats": {
+ "input_tokens": usage_metadata.prompt_token_count,
+ "cached_input_tokens": usage_metadata.cached_content_token_count
+ or 0,
+ "output_tokens": usage_metadata.candidates_token_count,
+ }
+ }
+ )
+
+ response_parts = chat_response.candidates[0].content.parts
+ if not response_parts:
raise HomeAssistantError(
"Sorry, I had a problem getting a response from Google Generative AI."
)
content = " ".join(
- [part.text.strip() for part in chat_response.parts if part.text]
+ [part.text.strip() for part in response_parts if part.text]
)
tool_calls = []
- for part in chat_response.parts:
+ for part in response_parts:
if not part.function_call:
continue
- tool_call = MessageToDict(part.function_call._pb) # noqa: SLF001
- tool_name = tool_call["name"]
- tool_args = _escape_decode(tool_call["args"])
+ tool_call = part.function_call
+ tool_name = tool_call.name
+ tool_args = _escape_decode(tool_call.args)
tool_calls.append(
- llm.ToolInput(tool_name=tool_name, tool_args=tool_args)
+ llm.ToolInput(
+ tool_name=self._fix_tool_name(tool_name),
+ tool_args=tool_args,
+ )
)
chat_request = _create_google_tool_response_content(
@@ -420,10 +456,12 @@ class GoogleGenerativeAIConversationEntity(
response = intent.IntentResponse(language=user_input.language)
response.async_set_speech(
- " ".join([part.text.strip() for part in chat_response.parts if part.text])
+ " ".join([part.text.strip() for part in response_parts if part.text])
)
return conversation.ConversationResult(
- response=response, conversation_id=chat_log.conversation_id
+ response=response,
+ conversation_id=chat_log.conversation_id,
+ continue_conversation=chat_log.continue_conversation,
)
async def _async_entry_update_listener(
diff --git a/homeassistant/components/google_generative_ai_conversation/manifest.json b/homeassistant/components/google_generative_ai_conversation/manifest.json
index 7b687b7da6f..ed215970d7f 100644
--- a/homeassistant/components/google_generative_ai_conversation/manifest.json
+++ b/homeassistant/components/google_generative_ai_conversation/manifest.json
@@ -2,11 +2,11 @@
"domain": "google_generative_ai_conversation",
"name": "Google Generative AI",
"after_dependencies": ["assist_pipeline", "intent"],
- "codeowners": ["@tronikos"],
+ "codeowners": ["@tronikos", "@ivanlh"],
"config_flow": true,
"dependencies": ["conversation"],
"documentation": "https://www.home-assistant.io/integrations/google_generative_ai_conversation",
"integration_type": "service",
"iot_class": "cloud_polling",
- "requirements": ["google-generativeai==0.8.2"]
+ "requirements": ["google-genai==1.1.0"]
}
diff --git a/homeassistant/components/google_generative_ai_conversation/services.yaml b/homeassistant/components/google_generative_ai_conversation/services.yaml
index f35697b89f8..82190d64540 100644
--- a/homeassistant/components/google_generative_ai_conversation/services.yaml
+++ b/homeassistant/components/google_generative_ai_conversation/services.yaml
@@ -9,3 +9,8 @@ generate_content:
required: false
selector:
object:
+ filenames:
+ required: false
+ selector:
+ text:
+ multiple: true
diff --git a/homeassistant/components/google_generative_ai_conversation/strings.json b/homeassistant/components/google_generative_ai_conversation/strings.json
index 9fea4805d38..7bf1831a34b 100644
--- a/homeassistant/components/google_generative_ai_conversation/strings.json
+++ b/homeassistant/components/google_generative_ai_conversation/strings.json
@@ -56,10 +56,21 @@
},
"image_filename": {
"name": "Image filename",
- "description": "Images",
+ "description": "Deprecated. Use filenames instead.",
+ "example": "/config/www/image.jpg"
+ },
+ "filenames": {
+ "name": "Attachment filenames",
+ "description": "Attachments to add to the prompt (images, PDFs, etc)",
"example": "/config/www/image.jpg"
}
}
}
+ },
+ "issues": {
+ "deprecated_image_filename_parameter": {
+ "title": "Deprecated 'image_filename' parameter",
+ "description": "The 'image_filename' parameter in Google Generative AI actions is deprecated. Please edit scripts and automations to use 'filenames' instead."
+ }
}
}
diff --git a/homeassistant/components/google_mail/__init__.py b/homeassistant/components/google_mail/__init__.py
index 7fae5f18da5..8ef978568dc 100644
--- a/homeassistant/components/google_mail/__init__.py
+++ b/homeassistant/components/google_mail/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, discovery
@@ -59,12 +59,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: GoogleMailConfigEntry) -
async def async_unload_entry(hass: HomeAssistant, entry: GoogleMailConfigEntry) -> bool:
"""Unload a config entry."""
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
for service_name in hass.services.async_services_for_domain(DOMAIN):
hass.services.async_remove(DOMAIN, service_name)
diff --git a/homeassistant/components/google_mail/sensor.py b/homeassistant/components/google_mail/sensor.py
index c832104d719..781ea9192f0 100644
--- a/homeassistant/components/google_mail/sensor.py
+++ b/homeassistant/components/google_mail/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GoogleMailConfigEntry
from .entity import GoogleMailEntity
@@ -29,7 +29,7 @@ SENSOR_TYPE = SensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: GoogleMailConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Google Mail sensor."""
async_add_entities([GoogleMailSensor(entry.runtime_data, SENSOR_TYPE)], True)
diff --git a/homeassistant/components/google_photos/__init__.py b/homeassistant/components/google_photos/__init__.py
index 2a7109d8189..40de02554ae 100644
--- a/homeassistant/components/google_photos/__init__.py
+++ b/homeassistant/components/google_photos/__init__.py
@@ -12,9 +12,8 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from . import api
from .const import DOMAIN
-from .coordinator import GooglePhotosUpdateCoordinator
+from .coordinator import GooglePhotosConfigEntry, GooglePhotosUpdateCoordinator
from .services import async_register_services
-from .types import GooglePhotosConfigEntry
__all__ = [
"DOMAIN",
@@ -43,7 +42,9 @@ async def async_setup_entry(
raise ConfigEntryNotReady from err
except ClientError as err:
raise ConfigEntryNotReady from err
- coordinator = GooglePhotosUpdateCoordinator(hass, GooglePhotosLibraryApi(auth))
+ coordinator = GooglePhotosUpdateCoordinator(
+ hass, entry, GooglePhotosLibraryApi(auth)
+ )
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/google_photos/coordinator.py b/homeassistant/components/google_photos/coordinator.py
index 3ba5a8124d6..215d40d7864 100644
--- a/homeassistant/components/google_photos/coordinator.py
+++ b/homeassistant/components/google_photos/coordinator.py
@@ -15,6 +15,7 @@ from google_photos_library_api.api import GooglePhotosLibraryApi
from google_photos_library_api.exceptions import GooglePhotosApiError
from google_photos_library_api.model import Album, NewAlbum
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -23,6 +24,8 @@ _LOGGER = logging.getLogger(__name__)
UPDATE_INTERVAL: Final = datetime.timedelta(hours=24)
ALBUM_PAGE_SIZE = 50
+type GooglePhotosConfigEntry = ConfigEntry[GooglePhotosUpdateCoordinator]
+
class GooglePhotosUpdateCoordinator(DataUpdateCoordinator[dict[str, str]]):
"""Coordinator for fetching Google Photos albums.
@@ -30,11 +33,19 @@ class GooglePhotosUpdateCoordinator(DataUpdateCoordinator[dict[str, str]]):
The `data` object is a dict from Album ID to Album title.
"""
- def __init__(self, hass: HomeAssistant, client: GooglePhotosLibraryApi) -> None:
+ config_entry: GooglePhotosConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: GooglePhotosConfigEntry,
+ client: GooglePhotosLibraryApi,
+ ) -> None:
"""Initialize TaskUpdateCoordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Google Photos",
update_interval=UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/google_photos/media_source.py b/homeassistant/components/google_photos/media_source.py
index 7ee81b51bc0..c0a87e46fbc 100644
--- a/homeassistant/components/google_photos/media_source.py
+++ b/homeassistant/components/google_photos/media_source.py
@@ -20,8 +20,8 @@ from homeassistant.components.media_source import (
)
from homeassistant.core import HomeAssistant
-from . import GooglePhotosConfigEntry
from .const import DOMAIN, READ_SCOPE
+from .coordinator import GooglePhotosConfigEntry
_LOGGER = logging.getLogger(__name__)
diff --git a/homeassistant/components/google_photos/services.py b/homeassistant/components/google_photos/services.py
index 22d3cc7deb0..8042df8f811 100644
--- a/homeassistant/components/google_photos/services.py
+++ b/homeassistant/components/google_photos/services.py
@@ -21,7 +21,7 @@ from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import config_validation as cv
from .const import DOMAIN, UPLOAD_SCOPE
-from .types import GooglePhotosConfigEntry
+from .coordinator import GooglePhotosConfigEntry
CONF_CONFIG_ENTRY_ID = "config_entry_id"
CONF_ALBUM = "album"
diff --git a/homeassistant/components/google_photos/types.py b/homeassistant/components/google_photos/types.py
deleted file mode 100644
index 4f4cc1845e4..00000000000
--- a/homeassistant/components/google_photos/types.py
+++ /dev/null
@@ -1,7 +0,0 @@
-"""Google Photos types."""
-
-from homeassistant.config_entries import ConfigEntry
-
-from .coordinator import GooglePhotosUpdateCoordinator
-
-type GooglePhotosConfigEntry = ConfigEntry[GooglePhotosUpdateCoordinator]
diff --git a/homeassistant/components/google_pubsub/manifest.json b/homeassistant/components/google_pubsub/manifest.json
index 9ea747898b2..b96f4e9ebc0 100644
--- a/homeassistant/components/google_pubsub/manifest.json
+++ b/homeassistant/components/google_pubsub/manifest.json
@@ -5,5 +5,5 @@
"documentation": "https://www.home-assistant.io/integrations/google_pubsub",
"iot_class": "cloud_push",
"quality_scale": "legacy",
- "requirements": ["google-cloud-pubsub==2.23.0"]
+ "requirements": ["google-cloud-pubsub==2.29.0"]
}
diff --git a/homeassistant/components/google_sheets/__init__.py b/homeassistant/components/google_sheets/__init__.py
index 942db675b5a..afafce816a9 100644
--- a/homeassistant/components/google_sheets/__init__.py
+++ b/homeassistant/components/google_sheets/__init__.py
@@ -12,7 +12,7 @@ from gspread.exceptions import APIError
from gspread.utils import ValueInputOption
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_TOKEN
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.exceptions import (
@@ -39,7 +39,7 @@ SERVICE_APPEND_SHEET = "append_sheet"
SHEET_SERVICE_SCHEMA = vol.All(
{
- vol.Required(DATA_CONFIG_ENTRY): ConfigEntrySelector(),
+ vol.Required(DATA_CONFIG_ENTRY): ConfigEntrySelector({"integration": DOMAIN}),
vol.Optional(WORKSHEET): cv.string,
vol.Required(DATA): vol.Any(cv.ensure_list, [dict]),
},
@@ -81,12 +81,7 @@ async def async_unload_entry(
hass: HomeAssistant, entry: GoogleSheetsConfigEntry
) -> bool:
"""Unload a config entry."""
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
for service_name in hass.services.async_services_for_domain(DOMAIN):
hass.services.async_remove(DOMAIN, service_name)
diff --git a/homeassistant/components/google_tasks/__init__.py b/homeassistant/components/google_tasks/__init__.py
index 45ad1777aa0..2d570854ad4 100644
--- a/homeassistant/components/google_tasks/__init__.py
+++ b/homeassistant/components/google_tasks/__init__.py
@@ -13,9 +13,8 @@ from homeassistant.helpers import config_entry_oauth2_flow
from . import api
from .const import DOMAIN
-from .coordinator import TaskUpdateCoordinator
+from .coordinator import GoogleTasksConfigEntry, TaskUpdateCoordinator
from .exceptions import GoogleTasksApiError
-from .types import GoogleTasksConfigEntry
__all__ = [
"DOMAIN",
@@ -52,6 +51,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: GoogleTasksConfigEntry)
coordinators = [
TaskUpdateCoordinator(
hass,
+ entry,
auth,
task_list["id"],
task_list["title"],
diff --git a/homeassistant/components/google_tasks/coordinator.py b/homeassistant/components/google_tasks/coordinator.py
index a06faf00a91..b61fe1c30db 100644
--- a/homeassistant/components/google_tasks/coordinator.py
+++ b/homeassistant/components/google_tasks/coordinator.py
@@ -5,6 +5,7 @@ import datetime
import logging
from typing import Any, Final
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -15,13 +16,18 @@ _LOGGER = logging.getLogger(__name__)
UPDATE_INTERVAL: Final = datetime.timedelta(minutes=30)
TIMEOUT = 10
+type GoogleTasksConfigEntry = ConfigEntry[list[TaskUpdateCoordinator]]
+
class TaskUpdateCoordinator(DataUpdateCoordinator[list[dict[str, Any]]]):
"""Coordinator for fetching Google Tasks for a Task List form the API."""
+ config_entry: GoogleTasksConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: GoogleTasksConfigEntry,
api: AsyncConfigEntryAuth,
task_list_id: str,
task_list_title: str,
@@ -30,6 +36,7 @@ class TaskUpdateCoordinator(DataUpdateCoordinator[list[dict[str, Any]]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"Google Tasks {task_list_id}",
update_interval=UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/google_tasks/todo.py b/homeassistant/components/google_tasks/todo.py
index 1df5e5fc2e9..d8e3c64bad8 100644
--- a/homeassistant/components/google_tasks/todo.py
+++ b/homeassistant/components/google_tasks/todo.py
@@ -12,12 +12,11 @@ from homeassistant.components.todo import (
TodoListEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
-from .coordinator import TaskUpdateCoordinator
-from .types import GoogleTasksConfigEntry
+from .coordinator import GoogleTasksConfigEntry, TaskUpdateCoordinator
PARALLEL_UPDATES = 0
@@ -69,7 +68,7 @@ def _convert_api_item(item: dict[str, str]) -> TodoItem:
async def async_setup_entry(
hass: HomeAssistant,
entry: GoogleTasksConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Google Tasks todo platform."""
async_add_entities(
diff --git a/homeassistant/components/google_tasks/types.py b/homeassistant/components/google_tasks/types.py
deleted file mode 100644
index 21500d11eb8..00000000000
--- a/homeassistant/components/google_tasks/types.py
+++ /dev/null
@@ -1,7 +0,0 @@
-"""Types for the Google Tasks integration."""
-
-from homeassistant.config_entries import ConfigEntry
-
-from .coordinator import TaskUpdateCoordinator
-
-type GoogleTasksConfigEntry = ConfigEntry[list[TaskUpdateCoordinator]]
diff --git a/homeassistant/components/google_translate/tts.py b/homeassistant/components/google_translate/tts.py
index 13e0ca4c273..201300d95b4 100644
--- a/homeassistant/components/google_translate/tts.py
+++ b/homeassistant/components/google_translate/tts.py
@@ -19,7 +19,7 @@ from homeassistant.components.tts import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import (
@@ -55,7 +55,7 @@ async def async_get_engine(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Google Translate speech platform via config entry."""
default_language = config_entry.data[CONF_LANG]
diff --git a/homeassistant/components/google_travel_time/sensor.py b/homeassistant/components/google_travel_time/sensor.py
index a3f9c236136..cac792dca53 100644
--- a/homeassistant/components/google_travel_time/sensor.py
+++ b/homeassistant/components/google_travel_time/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
)
from homeassistant.core import CoreState, HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.location import find_coordinates
from homeassistant.util import dt as dt_util
@@ -55,7 +55,7 @@ def convert_time_to_utc(timestr):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Google travel time sensor entry."""
api_key = config_entry.data[CONF_API_KEY]
diff --git a/homeassistant/components/govee_ble/binary_sensor.py b/homeassistant/components/govee_ble/binary_sensor.py
index 7b7a1fb5a50..c3c71714e90 100644
--- a/homeassistant/components/govee_ble/binary_sensor.py
+++ b/homeassistant/components/govee_ble/binary_sensor.py
@@ -20,7 +20,7 @@ from homeassistant.components.bluetooth.passive_update_processor import (
PassiveBluetoothProcessorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .coordinator import GoveeBLEPassiveBluetoothDataProcessor
@@ -76,7 +76,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the govee-ble BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/govee_ble/event.py b/homeassistant/components/govee_ble/event.py
index 5e5aa6354be..03f74f37f6a 100644
--- a/homeassistant/components/govee_ble/event.py
+++ b/homeassistant/components/govee_ble/event.py
@@ -16,7 +16,7 @@ from homeassistant.components.event import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import GoveeBLEConfigEntry, format_event_dispatcher_name
@@ -90,7 +90,7 @@ class GoveeBluetoothEventEntity(EventEntity):
async def async_setup_entry(
hass: HomeAssistant,
entry: GoveeBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a govee ble event."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/govee_ble/manifest.json b/homeassistant/components/govee_ble/manifest.json
index 1c61ae31010..b06dab243af 100644
--- a/homeassistant/components/govee_ble/manifest.json
+++ b/homeassistant/components/govee_ble/manifest.json
@@ -135,5 +135,5 @@
"dependencies": ["bluetooth_adapters"],
"documentation": "https://www.home-assistant.io/integrations/govee_ble",
"iot_class": "local_push",
- "requirements": ["govee-ble==0.43.0"]
+ "requirements": ["govee-ble==0.43.1"]
}
diff --git a/homeassistant/components/govee_ble/sensor.py b/homeassistant/components/govee_ble/sensor.py
index 383f50e5c46..fa0b828176c 100644
--- a/homeassistant/components/govee_ble/sensor.py
+++ b/homeassistant/components/govee_ble/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .coordinator import GoveeBLEConfigEntry, GoveeBLEPassiveBluetoothDataProcessor
@@ -105,7 +105,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: GoveeBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Govee BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/govee_light_local/__init__.py b/homeassistant/components/govee_light_local/__init__.py
index 44dbc825665..ee04dd81088 100644
--- a/homeassistant/components/govee_light_local/__init__.py
+++ b/homeassistant/components/govee_light_local/__init__.py
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(hass: HomeAssistant, entry: GoveeLocalConfigEntry) -> bool:
"""Set up Govee light local from a config entry."""
- coordinator = GoveeLocalApiCoordinator(hass=hass)
+ coordinator = GoveeLocalApiCoordinator(hass, entry)
async def await_cleanup():
cleanup_complete: asyncio.Event = coordinator.cleanup()
diff --git a/homeassistant/components/govee_light_local/coordinator.py b/homeassistant/components/govee_light_local/coordinator.py
index 240313a34b8..530ade1f743 100644
--- a/homeassistant/components/govee_light_local/coordinator.py
+++ b/homeassistant/components/govee_light_local/coordinator.py
@@ -26,11 +26,16 @@ type GoveeLocalConfigEntry = ConfigEntry[GoveeLocalApiCoordinator]
class GoveeLocalApiCoordinator(DataUpdateCoordinator[list[GoveeDevice]]):
"""Govee light local coordinator."""
- def __init__(self, hass: HomeAssistant) -> None:
+ config_entry: GoveeLocalConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: GoveeLocalConfigEntry
+ ) -> None:
"""Initialize my coordinator."""
super().__init__(
hass=hass,
logger=_LOGGER,
+ config_entry=config_entry,
name="GoveeLightLocalApi",
update_interval=SCAN_INTERVAL,
)
@@ -84,6 +89,10 @@ class GoveeLocalApiCoordinator(DataUpdateCoordinator[list[GoveeDevice]]):
"""Set light color in kelvin."""
await device.set_temperature(temperature)
+ async def set_scene(self, device: GoveeController, scene: str) -> None:
+ """Set light scene."""
+ await device.set_scene(scene)
+
@property
def devices(self) -> list[GoveeDevice]:
"""Return a list of discovered Govee devices."""
diff --git a/homeassistant/components/govee_light_local/light.py b/homeassistant/components/govee_light_local/light.py
index c7799a7ffc4..984654477e9 100644
--- a/homeassistant/components/govee_light_local/light.py
+++ b/homeassistant/components/govee_light_local/light.py
@@ -10,14 +10,16 @@ from govee_local_api import GoveeDevice, GoveeLightFeatures
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_COLOR_TEMP_KELVIN,
+ ATTR_EFFECT,
ATTR_RGB_COLOR,
ColorMode,
LightEntity,
+ LightEntityFeature,
filter_supported_color_modes,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER
@@ -25,11 +27,13 @@ from .coordinator import GoveeLocalApiCoordinator, GoveeLocalConfigEntry
_LOGGER = logging.getLogger(__name__)
+_NONE_SCENE = "none"
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: GoveeLocalConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Govee light setup."""
@@ -50,10 +54,22 @@ async def async_setup_entry(
class GoveeLight(CoordinatorEntity[GoveeLocalApiCoordinator], LightEntity):
"""Govee Light."""
+ _attr_translation_key = "govee_light"
_attr_has_entity_name = True
_attr_name = None
_attr_supported_color_modes: set[ColorMode]
_fixed_color_mode: ColorMode | None = None
+ _attr_effect_list: list[str] | None = None
+ _attr_effect: str | None = None
+ _attr_supported_features: LightEntityFeature = LightEntityFeature(0)
+ _last_color_state: (
+ tuple[
+ ColorMode | str | None,
+ int | None,
+ tuple[int, int, int] | tuple[int | None] | None,
+ ]
+ | None
+ ) = None
def __init__(
self,
@@ -80,6 +96,13 @@ class GoveeLight(CoordinatorEntity[GoveeLocalApiCoordinator], LightEntity):
if GoveeLightFeatures.BRIGHTNESS & capabilities.features:
color_modes.add(ColorMode.BRIGHTNESS)
+ if (
+ GoveeLightFeatures.SCENES & capabilities.features
+ and capabilities.scenes
+ ):
+ self._attr_supported_features = LightEntityFeature.EFFECT
+ self._attr_effect_list = [_NONE_SCENE, *capabilities.scenes.keys()]
+
self._attr_supported_color_modes = filter_supported_color_modes(color_modes)
if len(self._attr_supported_color_modes) == 1:
# If the light supports only a single color mode, set it now
@@ -143,12 +166,27 @@ class GoveeLight(CoordinatorEntity[GoveeLocalApiCoordinator], LightEntity):
if ATTR_RGB_COLOR in kwargs:
self._attr_color_mode = ColorMode.RGB
+ self._attr_effect = None
+ self._last_color_state = None
red, green, blue = kwargs[ATTR_RGB_COLOR]
await self.coordinator.set_rgb_color(self._device, red, green, blue)
elif ATTR_COLOR_TEMP_KELVIN in kwargs:
self._attr_color_mode = ColorMode.COLOR_TEMP
+ self._attr_effect = None
+ self._last_color_state = None
temperature: float = kwargs[ATTR_COLOR_TEMP_KELVIN]
await self.coordinator.set_temperature(self._device, int(temperature))
+ elif ATTR_EFFECT in kwargs:
+ effect = kwargs[ATTR_EFFECT]
+ if effect and self._attr_effect_list and effect in self._attr_effect_list:
+ if effect == _NONE_SCENE:
+ self._attr_effect = None
+ await self._restore_last_color_state()
+ else:
+ self._attr_effect = effect
+ self._save_last_color_state()
+ await self.coordinator.set_scene(self._device, effect)
+
self.async_write_ha_state()
async def async_turn_off(self, **kwargs: Any) -> None:
@@ -159,3 +197,27 @@ class GoveeLight(CoordinatorEntity[GoveeLocalApiCoordinator], LightEntity):
@callback
def _update_callback(self, device: GoveeDevice) -> None:
self.async_write_ha_state()
+
+ def _save_last_color_state(self) -> None:
+ color_mode = self.color_mode
+ self._last_color_state = (
+ color_mode,
+ self.brightness,
+ (self.color_temp_kelvin,)
+ if color_mode == ColorMode.COLOR_TEMP
+ else self.rgb_color,
+ )
+
+ async def _restore_last_color_state(self) -> None:
+ if self._last_color_state:
+ color_mode, brightness, color = self._last_color_state
+ if color:
+ if color_mode == ColorMode.RGB:
+ await self.coordinator.set_rgb_color(self._device, *color)
+ elif color_mode == ColorMode.COLOR_TEMP:
+ await self.coordinator.set_temperature(self._device, *color)
+ if brightness:
+ await self.coordinator.set_brightness(
+ self._device, int((float(brightness) / 255.0) * 100.0)
+ )
+ self._last_color_state = None
diff --git a/homeassistant/components/govee_light_local/manifest.json b/homeassistant/components/govee_light_local/manifest.json
index cba341cd482..55a6b9e8578 100644
--- a/homeassistant/components/govee_light_local/manifest.json
+++ b/homeassistant/components/govee_light_local/manifest.json
@@ -6,5 +6,5 @@
"dependencies": ["network"],
"documentation": "https://www.home-assistant.io/integrations/govee_light_local",
"iot_class": "local_push",
- "requirements": ["govee-local-api==2.0.1"]
+ "requirements": ["govee-local-api==2.1.0"]
}
diff --git a/homeassistant/components/govee_light_local/strings.json b/homeassistant/components/govee_light_local/strings.json
index ad8f0f41ae7..49f3a2cbeb9 100644
--- a/homeassistant/components/govee_light_local/strings.json
+++ b/homeassistant/components/govee_light_local/strings.json
@@ -9,5 +9,29 @@
"single_instance_allowed": "[%key:common::config_flow::abort::single_instance_allowed%]",
"no_devices_found": "[%key:common::config_flow::abort::no_devices_found%]"
}
+ },
+ "entity": {
+ "light": {
+ "govee_light": {
+ "state_attributes": {
+ "effect": {
+ "state": {
+ "none": "None",
+ "sunrise": "Sunrise",
+ "sunset": "Sunset",
+ "movie": "Movie",
+ "dating": "Dating",
+ "romantic": "Romantic",
+ "twinkle": "Twinkle",
+ "candlelight": "Candlelight",
+ "snowflake": "Snowflake",
+ "energetic": "Energetic",
+ "breathe": "Breathe",
+ "crossing": "Crossing"
+ }
+ }
+ }
+ }
+ }
}
}
diff --git a/homeassistant/components/gpsd/sensor.py b/homeassistant/components/gpsd/sensor.py
index 86d3ab7cc04..cc2257c88f7 100644
--- a/homeassistant/components/gpsd/sensor.py
+++ b/homeassistant/components/gpsd/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -156,7 +156,7 @@ SENSOR_TYPES: tuple[GpsdSensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: GPSDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the GPSD component."""
async_add_entities(
diff --git a/homeassistant/components/gpslogger/device_tracker.py b/homeassistant/components/gpslogger/device_tracker.py
index 3ed68ed1b06..be38382098d 100644
--- a/homeassistant/components/gpslogger/device_tracker.py
+++ b/homeassistant/components/gpslogger/device_tracker.py
@@ -12,7 +12,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import DOMAIN as GPL_DOMAIN, TRACKER_UPDATE
@@ -26,7 +26,9 @@ from .const import (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure a dispatcher connection based on a config entry."""
diff --git a/homeassistant/components/gree/__init__.py b/homeassistant/components/gree/__init__.py
index c385ce45262..7cb4f0f0921 100644
--- a/homeassistant/components/gree/__init__.py
+++ b/homeassistant/components/gree/__init__.py
@@ -26,7 +26,7 @@ PLATFORMS = [Platform.CLIMATE, Platform.SWITCH]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Gree Climate from a config entry."""
hass.data.setdefault(DOMAIN, {})
- gree_discovery = DiscoveryService(hass)
+ gree_discovery = DiscoveryService(hass, entry)
hass.data[DATA_DISCOVERY_SERVICE] = gree_discovery
async def _async_scan_update(_=None):
diff --git a/homeassistant/components/gree/climate.py b/homeassistant/components/gree/climate.py
index f197f21a4e1..f703ded1ea2 100644
--- a/homeassistant/components/gree/climate.py
+++ b/homeassistant/components/gree/climate.py
@@ -40,7 +40,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
COORDINATORS,
@@ -88,7 +88,7 @@ SWING_MODES = [SWING_OFF, SWING_VERTICAL, SWING_HORIZONTAL, SWING_BOTH]
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Gree HVAC device from a config entry."""
diff --git a/homeassistant/components/gree/const.py b/homeassistant/components/gree/const.py
index f926eb1c53e..14236f09fa2 100644
--- a/homeassistant/components/gree/const.py
+++ b/homeassistant/components/gree/const.py
@@ -20,3 +20,4 @@ MAX_ERRORS = 2
TARGET_TEMPERATURE_STEP = 1
UPDATE_INTERVAL = 60
+MAX_EXPECTED_RESPONSE_TIME_INTERVAL = UPDATE_INTERVAL * 2
diff --git a/homeassistant/components/gree/coordinator.py b/homeassistant/components/gree/coordinator.py
index 42d6734a6b2..c8b4e6cff54 100644
--- a/homeassistant/components/gree/coordinator.py
+++ b/homeassistant/components/gree/coordinator.py
@@ -2,6 +2,7 @@
from __future__ import annotations
+import copy
from datetime import datetime, timedelta
import logging
from typing import Any
@@ -11,6 +12,7 @@ from greeclimate.discovery import Discovery, Listener
from greeclimate.exceptions import DeviceNotBoundError, DeviceTimeoutError
from greeclimate.network import Response
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.json import json_dumps
@@ -23,6 +25,7 @@ from .const import (
DISPATCH_DEVICE_DISCOVERED,
DOMAIN,
MAX_ERRORS,
+ MAX_EXPECTED_RESPONSE_TIME_INTERVAL,
UPDATE_INTERVAL,
)
@@ -32,18 +35,21 @@ _LOGGER = logging.getLogger(__name__)
class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Manages polling for state changes from the device."""
- def __init__(self, hass: HomeAssistant, device: Device) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, device: Device
+ ) -> None:
"""Initialize the data update coordinator."""
- DataUpdateCoordinator.__init__(
- self,
+ super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"{DOMAIN}-{device.device_info.name}",
update_interval=timedelta(seconds=UPDATE_INTERVAL),
always_update=False,
)
self.device = device
- self.device.add_handler(Response.DATA, self.device_state_updated)
self.device.add_handler(Response.RESULT, self.device_state_updated)
self._error_count: int = 0
@@ -83,7 +89,9 @@ class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
# raise update failed if time for more than MAX_ERRORS has passed since last update
now = utcnow()
elapsed_success = now - self._last_response_time
- if self.update_interval and elapsed_success >= self.update_interval:
+ if self.update_interval and elapsed_success >= timedelta(
+ seconds=MAX_EXPECTED_RESPONSE_TIME_INTERVAL
+ ):
if not self._last_error_time or (
(now - self.update_interval) >= self._last_error_time
):
@@ -91,16 +99,19 @@ class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
self._error_count += 1
_LOGGER.warning(
- "Device %s is unresponsive for %s seconds",
+ "Device %s took an unusually long time to respond, %s seconds",
self.name,
elapsed_success,
)
+ else:
+ self._error_count = 0
if self.last_update_success and self._error_count >= MAX_ERRORS:
raise UpdateFailed(
f"Device {self.name} is unresponsive for too long and now unavailable"
)
- return self.device.raw_properties
+ self._last_response_time = utcnow()
+ return copy.deepcopy(self.device.raw_properties)
async def push_state_update(self):
"""Send state updates to the physical device."""
@@ -117,10 +128,11 @@ class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class DiscoveryService(Listener):
"""Discovery event handler for gree devices."""
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize discovery service."""
super().__init__()
self.hass = hass
+ self.entry = entry
self.discovery = Discovery(DISCOVERY_TIMEOUT)
self.discovery.add_listener(self)
@@ -144,7 +156,7 @@ class DiscoveryService(Listener):
device.device_info.ip,
device.device_info.port,
)
- coordo = DeviceDataUpdateCoordinator(self.hass, device)
+ coordo = DeviceDataUpdateCoordinator(self.hass, self.entry, device)
self.hass.data[DOMAIN][COORDINATORS].append(coordo)
await coordo.async_refresh()
diff --git a/homeassistant/components/gree/switch.py b/homeassistant/components/gree/switch.py
index c1612ce99de..67dc10138d1 100644
--- a/homeassistant/components/gree/switch.py
+++ b/homeassistant/components/gree/switch.py
@@ -16,7 +16,7 @@ from homeassistant.components.switch import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import COORDINATORS, DISPATCH_DEVICE_DISCOVERED, DOMAIN
from .entity import GreeEntity
@@ -93,7 +93,7 @@ GREE_SWITCHES: tuple[GreeSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Gree HVAC device from a config entry."""
diff --git a/homeassistant/components/group/binary_sensor.py b/homeassistant/components/group/binary_sensor.py
index 06c810c2643..fa1777d5510 100644
--- a/homeassistant/components/group/binary_sensor.py
+++ b/homeassistant/components/group/binary_sensor.py
@@ -26,7 +26,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -70,7 +73,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Binary Sensor Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/button.py b/homeassistant/components/group/button.py
index a18e074b775..c96d60067a1 100644
--- a/homeassistant/components/group/button.py
+++ b/homeassistant/components/group/button.py
@@ -22,7 +22,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -62,7 +65,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize button group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/cover.py b/homeassistant/components/group/cover.py
index b2e5c6eef37..64baba6d1e8 100644
--- a/homeassistant/components/group/cover.py
+++ b/homeassistant/components/group/cover.py
@@ -37,7 +37,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -80,7 +83,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Cover Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/event.py b/homeassistant/components/group/event.py
index e7f7938edf3..4009c788362 100644
--- a/homeassistant/components/group/event.py
+++ b/homeassistant/components/group/event.py
@@ -27,7 +27,10 @@ from homeassistant.const import (
)
from homeassistant.core import Event, EventStateChangedData, HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -68,7 +71,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize event group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/fan.py b/homeassistant/components/group/fan.py
index 87d9cb281f4..78745cb74c6 100644
--- a/homeassistant/components/group/fan.py
+++ b/homeassistant/components/group/fan.py
@@ -37,7 +37,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -82,7 +85,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Fan Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/light.py b/homeassistant/components/group/light.py
index 228645df974..259832d6152 100644
--- a/homeassistant/components/group/light.py
+++ b/homeassistant/components/group/light.py
@@ -48,7 +48,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -98,7 +101,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Light Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/lock.py b/homeassistant/components/group/lock.py
index e22e1ecd85c..7b460aa4632 100644
--- a/homeassistant/components/group/lock.py
+++ b/homeassistant/components/group/lock.py
@@ -28,7 +28,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -70,7 +73,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Lock Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/media_player.py b/homeassistant/components/group/media_player.py
index ab8ee64b3e1..3371e56b1dc 100644
--- a/homeassistant/components/group/media_player.py
+++ b/homeassistant/components/group/media_player.py
@@ -54,7 +54,10 @@ from homeassistant.core import (
callback,
)
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -99,7 +102,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize MediaPlayer Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/notify.py b/homeassistant/components/group/notify.py
index d6a9a6fd3c7..e710485c46f 100644
--- a/homeassistant/components/group/notify.py
+++ b/homeassistant/components/group/notify.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -129,7 +129,7 @@ class GroupNotifyPlatform(BaseNotificationService):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Notify Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/sensor.py b/homeassistant/components/group/sensor.py
index 4a3e191e511..9f0cc64ecf0 100644
--- a/homeassistant/components/group/sensor.py
+++ b/homeassistant/components/group/sensor.py
@@ -44,7 +44,10 @@ from homeassistant.helpers.entity import (
get_device_class,
get_unit_of_measurement,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.issue_registry import (
IssueSeverity,
async_create_issue,
@@ -130,7 +133,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Switch Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/group/switch.py b/homeassistant/components/group/switch.py
index 101c42d354f..29e625ca8e3 100644
--- a/homeassistant/components/group/switch.py
+++ b/homeassistant/components/group/switch.py
@@ -26,7 +26,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .entity import GroupEntity
@@ -71,7 +74,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Switch Group config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/growatt_server/sensor/__init__.py b/homeassistant/components/growatt_server/sensor/__init__.py
index e77660e6a3a..2794403811d 100644
--- a/homeassistant/components/growatt_server/sensor/__init__.py
+++ b/homeassistant/components/growatt_server/sensor/__init__.py
@@ -14,7 +14,7 @@ from homeassistant.const import CONF_NAME, CONF_PASSWORD, CONF_URL, CONF_USERNAM
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle, dt as dt_util
from ..const import (
@@ -61,7 +61,7 @@ def get_device_list(api, config):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Growatt sensor."""
config = {**config_entry.data}
diff --git a/homeassistant/components/growatt_server/sensor/total.py b/homeassistant/components/growatt_server/sensor/total.py
index 8111728d1e9..578745c8610 100644
--- a/homeassistant/components/growatt_server/sensor/total.py
+++ b/homeassistant/components/growatt_server/sensor/total.py
@@ -26,6 +26,7 @@ TOTAL_SENSOR_TYPES: tuple[GrowattSensorEntityDescription, ...] = (
api_key="todayEnergy",
native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
),
GrowattSensorEntityDescription(
key="total_output_power",
@@ -33,6 +34,7 @@ TOTAL_SENSOR_TYPES: tuple[GrowattSensorEntityDescription, ...] = (
api_key="invTodayPpv",
native_unit_of_measurement=UnitOfPower.WATT,
device_class=SensorDeviceClass.POWER,
+ state_class=SensorStateClass.MEASUREMENT,
),
GrowattSensorEntityDescription(
key="total_energy_output",
diff --git a/homeassistant/components/growatt_server/strings.json b/homeassistant/components/growatt_server/strings.json
index 9a985d98034..758428d7a55 100644
--- a/homeassistant/components/growatt_server/strings.json
+++ b/homeassistant/components/growatt_server/strings.json
@@ -38,28 +38,28 @@
"name": "Input 1 voltage"
},
"inverter_amperage_input_1": {
- "name": "Input 1 Amperage"
+ "name": "Input 1 amperage"
},
"inverter_wattage_input_1": {
- "name": "Input 1 Wattage"
+ "name": "Input 1 wattage"
},
"inverter_voltage_input_2": {
"name": "Input 2 voltage"
},
"inverter_amperage_input_2": {
- "name": "Input 2 Amperage"
+ "name": "Input 2 amperage"
},
"inverter_wattage_input_2": {
- "name": "Input 2 Wattage"
+ "name": "Input 2 wattage"
},
"inverter_voltage_input_3": {
"name": "Input 3 voltage"
},
"inverter_amperage_input_3": {
- "name": "Input 3 Amperage"
+ "name": "Input 3 amperage"
},
"inverter_wattage_input_3": {
- "name": "Input 3 Wattage"
+ "name": "Input 3 wattage"
},
"inverter_internal_wattage": {
"name": "Internal wattage"
@@ -137,13 +137,13 @@
"name": "Load consumption"
},
"mix_wattage_pv_1": {
- "name": "PV1 Wattage"
+ "name": "PV1 wattage"
},
"mix_wattage_pv_2": {
- "name": "PV2 Wattage"
+ "name": "PV2 wattage"
},
"mix_wattage_pv_all": {
- "name": "All PV Wattage"
+ "name": "All PV wattage"
},
"mix_export_to_grid": {
"name": "Export to grid"
@@ -182,7 +182,7 @@
"name": "Storage production today"
},
"storage_storage_production_lifetime": {
- "name": "Lifetime Storage production"
+ "name": "Lifetime storage production"
},
"storage_grid_discharge_today": {
"name": "Grid discharged today"
@@ -224,7 +224,7 @@
"name": "Storage charging/ discharging(-ve)"
},
"storage_load_consumption_solar_storage": {
- "name": "Load consumption (Solar + Storage)"
+ "name": "Load consumption (solar + storage)"
},
"storage_charge_today": {
"name": "Charge today"
@@ -257,7 +257,7 @@
"name": "Output voltage"
},
"storage_ac_output_frequency": {
- "name": "Ac output frequency"
+ "name": "AC output frequency"
},
"storage_current_pv": {
"name": "Solar charge current"
@@ -290,7 +290,7 @@
"name": "Lifetime total energy input 1"
},
"tlx_energy_today_input_1": {
- "name": "Energy Today Input 1"
+ "name": "Energy today input 1"
},
"tlx_voltage_input_1": {
"name": "[%key:component::growatt_server::entity::sensor::inverter_voltage_input_1::name%]"
@@ -305,7 +305,7 @@
"name": "Lifetime total energy input 2"
},
"tlx_energy_today_input_2": {
- "name": "Energy Today Input 2"
+ "name": "Energy today input 2"
},
"tlx_voltage_input_2": {
"name": "[%key:component::growatt_server::entity::sensor::inverter_voltage_input_2::name%]"
@@ -320,7 +320,7 @@
"name": "Lifetime total energy input 3"
},
"tlx_energy_today_input_3": {
- "name": "Energy Today Input 3"
+ "name": "Energy today input 3"
},
"tlx_voltage_input_3": {
"name": "[%key:component::growatt_server::entity::sensor::inverter_voltage_input_3::name%]"
@@ -335,16 +335,16 @@
"name": "Lifetime total energy input 4"
},
"tlx_energy_today_input_4": {
- "name": "Energy Today Input 4"
+ "name": "Energy today input 4"
},
"tlx_voltage_input_4": {
"name": "Input 4 voltage"
},
"tlx_amperage_input_4": {
- "name": "Input 4 Amperage"
+ "name": "Input 4 amperage"
},
"tlx_wattage_input_4": {
- "name": "Input 4 Wattage"
+ "name": "Input 4 wattage"
},
"tlx_solar_generation_total": {
"name": "Lifetime total solar energy"
@@ -434,10 +434,10 @@
"name": "Money lifetime"
},
"total_energy_today": {
- "name": "Energy Today"
+ "name": "Energy today"
},
"total_output_power": {
- "name": "Output Power"
+ "name": "Output power"
},
"total_energy_output": {
"name": "[%key:component::growatt_server::entity::sensor::inverter_energy_total::name%]"
diff --git a/homeassistant/components/gtfs/sensor.py b/homeassistant/components/gtfs/sensor.py
index 2637a55f772..8c624e2cdd6 100644
--- a/homeassistant/components/gtfs/sensor.py
+++ b/homeassistant/components/gtfs/sensor.py
@@ -341,7 +341,7 @@ def get_next_departure(
{tomorrow_order}
origin_stop_time.departure_time
LIMIT :limit
- """
+ """ # noqa: S608
result = schedule.engine.connect().execute(
text(sql_query),
{
diff --git a/homeassistant/components/guardian/__init__.py b/homeassistant/components/guardian/__init__.py
index c1cbb4c0e5a..075c388c4e4 100644
--- a/homeassistant/components/guardian/__init__.py
+++ b/homeassistant/components/guardian/__init__.py
@@ -11,7 +11,7 @@ from aioguardian import Client
from aioguardian.errors import GuardianError
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ATTR_DEVICE_ID,
CONF_DEVICE_ID,
@@ -247,12 +247,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
if unload_ok:
hass.data[DOMAIN].pop(entry.entry_id)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
# If this is the last loaded instance of Guardian, deregister any services
# defined during integration setup:
for service_name in SERVICES:
diff --git a/homeassistant/components/guardian/binary_sensor.py b/homeassistant/components/guardian/binary_sensor.py
index 84bb61da0e5..7d5f97bdb65 100644
--- a/homeassistant/components/guardian/binary_sensor.py
+++ b/homeassistant/components/guardian/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GuardianData
from .const import (
@@ -86,7 +86,9 @@ VALVE_CONTROLLER_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Guardian switches based on a config entry."""
data: GuardianData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/guardian/button.py b/homeassistant/components/guardian/button.py
index f4881a9d94b..01bac63c6e3 100644
--- a/homeassistant/components/guardian/button.py
+++ b/homeassistant/components/guardian/button.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_send
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GuardianData
from .const import API_SYSTEM_DIAGNOSTICS, DOMAIN
@@ -68,7 +68,9 @@ BUTTON_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Guardian buttons based on a config entry."""
data: GuardianData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/guardian/coordinator.py b/homeassistant/components/guardian/coordinator.py
index 849cec8063c..500b7c10784 100644
--- a/homeassistant/components/guardian/coordinator.py
+++ b/homeassistant/components/guardian/coordinator.py
@@ -42,6 +42,7 @@ class GuardianDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
LOGGER,
+ config_entry=entry,
name=f"{valve_controller_uid}_{api_name}",
update_interval=DEFAULT_UPDATE_INTERVAL,
)
@@ -50,7 +51,6 @@ class GuardianDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
self._api_lock = api_lock
self._client = client
- self.config_entry = entry
self.signal_reboot_requested = SIGNAL_REBOOT_REQUESTED.format(
self.config_entry.entry_id
)
diff --git a/homeassistant/components/guardian/sensor.py b/homeassistant/components/guardian/sensor.py
index 3f9547e652a..13dd8e01296 100644
--- a/homeassistant/components/guardian/sensor.py
+++ b/homeassistant/components/guardian/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import GuardianData
@@ -137,7 +137,9 @@ VALVE_CONTROLLER_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Guardian switches based on a config entry."""
data: GuardianData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/guardian/switch.py b/homeassistant/components/guardian/switch.py
index fccf4f55a1f..a2c9ca282be 100644
--- a/homeassistant/components/guardian/switch.py
+++ b/homeassistant/components/guardian/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GuardianData
from .const import API_VALVE_STATUS, API_WIFI_STATUS, DOMAIN
@@ -110,7 +110,9 @@ VALVE_CONTROLLER_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Guardian switches based on a config entry."""
data: GuardianData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/guardian/valve.py b/homeassistant/components/guardian/valve.py
index 8c9749958bf..6847b3211c5 100644
--- a/homeassistant/components/guardian/valve.py
+++ b/homeassistant/components/guardian/valve.py
@@ -17,7 +17,7 @@ from homeassistant.components.valve import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import GuardianData
from .const import API_VALVE_STATUS, DOMAIN
@@ -109,7 +109,9 @@ VALVE_CONTROLLER_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Guardian switches based on a config entry."""
data: GuardianData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/habitica/__init__.py b/homeassistant/components/habitica/__init__.py
index 1972e89c58a..217b5e739d1 100644
--- a/homeassistant/components/habitica/__init__.py
+++ b/homeassistant/components/habitica/__init__.py
@@ -9,9 +9,8 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.typing import ConfigType
from .const import CONF_API_USER, DOMAIN, X_CLIENT
-from .coordinator import HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry, HabiticaDataUpdateCoordinator
from .services import async_setup_services
-from .types import HabiticaConfigEntry
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
@@ -51,7 +50,7 @@ async def async_setup_entry(
x_client=X_CLIENT,
)
- coordinator = HabiticaDataUpdateCoordinator(hass, api)
+ coordinator = HabiticaDataUpdateCoordinator(hass, config_entry, api)
await coordinator.async_config_entry_first_refresh()
config_entry.runtime_data = coordinator
diff --git a/homeassistant/components/habitica/binary_sensor.py b/homeassistant/components/habitica/binary_sensor.py
index 5e3040e0606..c6f7ee0fb83 100644
--- a/homeassistant/components/habitica/binary_sensor.py
+++ b/homeassistant/components/habitica/binary_sensor.py
@@ -13,11 +13,11 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ASSETS_URL
+from .coordinator import HabiticaConfigEntry
from .entity import HabiticaBase
-from .types import HabiticaConfigEntry
PARALLEL_UPDATES = 1
@@ -56,7 +56,7 @@ BINARY_SENSOR_DESCRIPTIONS: tuple[HabiticaBinarySensorEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HabiticaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the habitica binary sensors."""
diff --git a/homeassistant/components/habitica/button.py b/homeassistant/components/habitica/button.py
index 450a5cdcf20..c57ba39fb6a 100644
--- a/homeassistant/components/habitica/button.py
+++ b/homeassistant/components/habitica/button.py
@@ -25,12 +25,15 @@ from homeassistant.components.button import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ASSETS_URL, DOMAIN
-from .coordinator import HabiticaData, HabiticaDataUpdateCoordinator
+from .coordinator import (
+ HabiticaConfigEntry,
+ HabiticaData,
+ HabiticaDataUpdateCoordinator,
+)
from .entity import HabiticaBase
-from .types import HabiticaConfigEntry
PARALLEL_UPDATES = 1
@@ -277,7 +280,7 @@ CLASS_SKILLS: tuple[HabiticaButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: HabiticaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up buttons from a config entry."""
diff --git a/homeassistant/components/habitica/calendar.py b/homeassistant/components/habitica/calendar.py
index f33f3c3c12f..b87a49670b0 100644
--- a/homeassistant/components/habitica/calendar.py
+++ b/homeassistant/components/habitica/calendar.py
@@ -18,11 +18,10 @@ from homeassistant.components.calendar import (
CalendarEvent,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
-from . import HabiticaConfigEntry
-from .coordinator import HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry, HabiticaDataUpdateCoordinator
from .entity import HabiticaBase
from .util import build_rrule, get_recurrence_rule
@@ -41,7 +40,7 @@ class HabiticaCalendar(StrEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HabiticaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the calendar platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/habitica/config_flow.py b/homeassistant/components/habitica/config_flow.py
index 7a7f369cb09..91a13bd7918 100644
--- a/homeassistant/components/habitica/config_flow.py
+++ b/homeassistant/components/habitica/config_flow.py
@@ -33,7 +33,6 @@ from homeassistant.helpers.selector import (
TextSelectorType,
)
-from . import HabiticaConfigEntry
from .const import (
CONF_API_USER,
DEFAULT_URL,
@@ -47,6 +46,7 @@ from .const import (
SITE_DATA_URL,
X_CLIENT,
)
+from .coordinator import HabiticaConfigEntry
STEP_ADVANCED_DATA_SCHEMA = vol.Schema(
{
diff --git a/homeassistant/components/habitica/const.py b/homeassistant/components/habitica/const.py
index 5eb616142e5..8b745ff2b99 100644
--- a/homeassistant/components/habitica/const.py
+++ b/homeassistant/components/habitica/const.py
@@ -35,6 +35,28 @@ ATTR_TYPE = "type"
ATTR_PRIORITY = "priority"
ATTR_TAG = "tag"
ATTR_KEYWORD = "keyword"
+ATTR_REMOVE_TAG = "remove_tag"
+ATTR_ALIAS = "alias"
+ATTR_PRIORITY = "priority"
+ATTR_COST = "cost"
+ATTR_NOTES = "notes"
+ATTR_UP_DOWN = "up_down"
+ATTR_FREQUENCY = "frequency"
+ATTR_COUNTER_UP = "counter_up"
+ATTR_COUNTER_DOWN = "counter_down"
+ATTR_ADD_CHECKLIST_ITEM = "add_checklist_item"
+ATTR_REMOVE_CHECKLIST_ITEM = "remove_checklist_item"
+ATTR_SCORE_CHECKLIST_ITEM = "score_checklist_item"
+ATTR_UNSCORE_CHECKLIST_ITEM = "unscore_checklist_item"
+ATTR_REMINDER = "reminder"
+ATTR_REMOVE_REMINDER = "remove_reminder"
+ATTR_CLEAR_REMINDER = "clear_reminder"
+ATTR_CLEAR_DATE = "clear_date"
+ATTR_REPEAT = "repeat"
+ATTR_INTERVAL = "every_x"
+ATTR_START_DATE = "start_date"
+ATTR_REPEAT_MONTHLY = "repeat_monthly"
+ATTR_STREAK = "streak"
SERVICE_CAST_SKILL = "cast_skill"
SERVICE_START_QUEST = "start_quest"
@@ -50,6 +72,13 @@ SERVICE_SCORE_REWARD = "score_reward"
SERVICE_TRANSFORMATION = "transformation"
+SERVICE_UPDATE_REWARD = "update_reward"
+SERVICE_CREATE_REWARD = "create_reward"
+SERVICE_UPDATE_HABIT = "update_habit"
+SERVICE_CREATE_HABIT = "create_habit"
+SERVICE_UPDATE_TODO = "update_todo"
+SERVICE_CREATE_TODO = "create_todo"
+SERVICE_UPDATE_DAILY = "update_daily"
DEVELOPER_ID = "4c4ca53f-c059-4ffa-966e-9d29dd405daf"
X_CLIENT = f"{DEVELOPER_ID} - {APPLICATION_NAME} {__version__}"
@@ -57,3 +86,5 @@ X_CLIENT = f"{DEVELOPER_ID} - {APPLICATION_NAME} {__version__}"
SECTION_REAUTH_LOGIN = "reauth_login"
SECTION_REAUTH_API_KEY = "reauth_api_key"
SECTION_DANGER_ZONE = "danger_zone"
+
+WEEK_DAYS = ["m", "t", "w", "th", "f", "s", "su"]
diff --git a/homeassistant/components/habitica/coordinator.py b/homeassistant/components/habitica/coordinator.py
index f97b98410bb..3c3a16f591a 100644
--- a/homeassistant/components/habitica/coordinator.py
+++ b/homeassistant/components/habitica/coordinator.py
@@ -11,6 +11,7 @@ from typing import Any
from aiohttp import ClientError
from habiticalib import (
+ Avatar,
ContentData,
Habitica,
HabiticaException,
@@ -19,7 +20,6 @@ from habiticalib import (
TaskFilter,
TooManyRequestsError,
UserData,
- UserStyles,
)
from homeassistant.config_entries import ConfigEntry
@@ -46,16 +46,22 @@ class HabiticaData:
tasks: list[TaskData]
+type HabiticaConfigEntry = ConfigEntry[HabiticaDataUpdateCoordinator]
+
+
class HabiticaDataUpdateCoordinator(DataUpdateCoordinator[HabiticaData]):
"""Habitica Data Update Coordinator."""
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, habitica: Habitica) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, habitica: Habitica
+ ) -> None:
"""Initialize the Habitica data coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=60),
request_refresh_debouncer=Debouncer(
@@ -159,12 +165,10 @@ class HabiticaDataUpdateCoordinator(DataUpdateCoordinator[HabiticaData]):
else:
await self.async_request_refresh()
- async def generate_avatar(self, user_styles: UserStyles) -> bytes:
+ async def generate_avatar(self, avatar: Avatar) -> bytes:
"""Generate Avatar."""
- avatar = BytesIO()
- await self.habitica.generate_avatar(
- fp=avatar, user_styles=user_styles, fmt="PNG"
- )
+ png = BytesIO()
+ await self.habitica.generate_avatar(fp=png, avatar=avatar, fmt="PNG")
- return avatar.getvalue()
+ return png.getvalue()
diff --git a/homeassistant/components/habitica/diagnostics.py b/homeassistant/components/habitica/diagnostics.py
index abfa0f35c4b..40a6d75b366 100644
--- a/homeassistant/components/habitica/diagnostics.py
+++ b/homeassistant/components/habitica/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.const import CONF_URL
from homeassistant.core import HomeAssistant
from .const import CONF_API_USER
-from .types import HabiticaConfigEntry
+from .coordinator import HabiticaConfigEntry
async def async_get_config_entry_diagnostics(
@@ -23,5 +23,5 @@ async def async_get_config_entry_diagnostics(
CONF_URL: config_entry.data[CONF_URL],
CONF_API_USER: config_entry.data[CONF_API_USER],
},
- "habitica_data": habitica_data.to_dict()["data"],
+ "habitica_data": habitica_data.to_dict(omit_none=False)["data"],
}
diff --git a/homeassistant/components/habitica/entity.py b/homeassistant/components/habitica/entity.py
index 932fec69f83..692ea5e5ac1 100644
--- a/homeassistant/components/habitica/entity.py
+++ b/homeassistant/components/habitica/entity.py
@@ -4,6 +4,8 @@ from __future__ import annotations
from typing import TYPE_CHECKING
+from yarl import URL
+
from homeassistant.const import CONF_NAME, CONF_URL
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity import EntityDescription
@@ -36,6 +38,10 @@ class HabiticaBase(CoordinatorEntity[HabiticaDataUpdateCoordinator]):
manufacturer=MANUFACTURER,
model=NAME,
name=coordinator.config_entry.data[CONF_NAME],
- configuration_url=coordinator.config_entry.data[CONF_URL],
+ configuration_url=(
+ URL(coordinator.config_entry.data[CONF_URL])
+ / "profile"
+ / coordinator.config_entry.unique_id
+ ),
identifiers={(DOMAIN, coordinator.config_entry.unique_id)},
)
diff --git a/homeassistant/components/habitica/icons.json b/homeassistant/components/habitica/icons.json
index 6ae6ebd728b..fcb9ec56fa7 100644
--- a/homeassistant/components/habitica/icons.json
+++ b/homeassistant/components/habitica/icons.json
@@ -217,6 +217,59 @@
"sections": {
"filter": "mdi:calendar-filter"
}
+ },
+ "update_reward": {
+ "service": "mdi:treasure-chest",
+ "sections": {
+ "tag_options": "mdi:tag",
+ "developer_options": "mdi:test-tube"
+ }
+ },
+ "create_reward": {
+ "service": "mdi:treasure-chest-outline",
+ "sections": {
+ "developer_options": "mdi:test-tube"
+ }
+ },
+ "update_habit": {
+ "service": "mdi:contrast-box",
+ "sections": {
+ "tag_options": "mdi:tag",
+ "developer_options": "mdi:test-tube"
+ }
+ },
+ "create_habit": {
+ "service": "mdi:contrast-box",
+ "sections": {
+ "developer_options": "mdi:test-tube"
+ }
+ },
+ "update_todo": {
+ "service": "mdi:pencil-box-outline",
+ "sections": {
+ "checklist_options": "mdi:format-list-checks",
+ "tag_options": "mdi:tag",
+ "developer_options": "mdi:test-tube",
+ "duedate_options": "mdi:calendar-blank",
+ "reminder_options": "mdi:reminder"
+ }
+ },
+ "create_todo": {
+ "service": "mdi:pencil-box-outline",
+ "sections": {
+ "developer_options": "mdi:test-tube"
+ }
+ },
+ "update_daily": {
+ "service": "mdi:calendar-month",
+ "sections": {
+ "checklist_options": "mdi:format-list-checks",
+ "tag_options": "mdi:tag",
+ "developer_options": "mdi:test-tube",
+ "reminder_options": "mdi:reminder",
+ "repeat_weekly_options": "mdi:calendar-refresh",
+ "repeat_monthly_options": "mdi:calendar-refresh"
+ }
}
}
}
diff --git a/homeassistant/components/habitica/image.py b/homeassistant/components/habitica/image.py
index f1dbbc64d41..1669f124bc7 100644
--- a/homeassistant/components/habitica/image.py
+++ b/homeassistant/components/habitica/image.py
@@ -2,18 +2,16 @@
from __future__ import annotations
-from dataclasses import asdict
from enum import StrEnum
-from habiticalib import UserStyles
+from habiticalib import Avatar, extract_avatar
from homeassistant.components.image import ImageEntity, ImageEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
-from . import HabiticaConfigEntry
-from .coordinator import HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry, HabiticaDataUpdateCoordinator
from .entity import HabiticaBase
PARALLEL_UPDATES = 1
@@ -28,7 +26,7 @@ class HabiticaImageEntity(StrEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HabiticaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the habitica image platform."""
@@ -45,7 +43,7 @@ class HabiticaImage(HabiticaBase, ImageEntity):
translation_key=HabiticaImageEntity.AVATAR,
)
_attr_content_type = "image/png"
- _current_appearance: UserStyles | None = None
+ _avatar: Avatar | None = None
_cache: bytes | None = None
def __init__(
@@ -57,13 +55,13 @@ class HabiticaImage(HabiticaBase, ImageEntity):
super().__init__(coordinator, self.entity_description)
ImageEntity.__init__(self, hass)
self._attr_image_last_updated = dt_util.utcnow()
+ self._avatar = extract_avatar(self.coordinator.data.user)
def _handle_coordinator_update(self) -> None:
"""Check if equipped gear and other things have changed since last avatar image generation."""
- new_appearance = UserStyles.from_dict(asdict(self.coordinator.data.user))
- if self._current_appearance != new_appearance:
- self._current_appearance = new_appearance
+ if self._avatar != self.coordinator.data.user:
+ self._avatar = extract_avatar(self.coordinator.data.user)
self._attr_image_last_updated = dt_util.utcnow()
self._cache = None
@@ -71,8 +69,6 @@ class HabiticaImage(HabiticaBase, ImageEntity):
async def async_image(self) -> bytes | None:
"""Return cached bytes, otherwise generate new avatar."""
- if not self._cache and self._current_appearance:
- self._cache = await self.coordinator.generate_avatar(
- self._current_appearance
- )
+ if not self._cache and self._avatar:
+ self._cache = await self.coordinator.generate_avatar(self._avatar)
return self._cache
diff --git a/homeassistant/components/habitica/manifest.json b/homeassistant/components/habitica/manifest.json
index 9ea346a0dcb..48b6997239e 100644
--- a/homeassistant/components/habitica/manifest.json
+++ b/homeassistant/components/habitica/manifest.json
@@ -6,5 +6,6 @@
"documentation": "https://www.home-assistant.io/integrations/habitica",
"iot_class": "cloud_polling",
"loggers": ["habiticalib"],
- "requirements": ["habiticalib==0.3.5"]
+ "quality_scale": "platinum",
+ "requirements": ["habiticalib==0.3.7"]
}
diff --git a/homeassistant/components/habitica/quality_scale.yaml b/homeassistant/components/habitica/quality_scale.yaml
index 9eadba496f2..1752e67cf46 100644
--- a/homeassistant/components/habitica/quality_scale.yaml
+++ b/homeassistant/components/habitica/quality_scale.yaml
@@ -51,7 +51,7 @@ rules:
status: exempt
comment: No supportable devices.
docs-supported-functions: done
- docs-troubleshooting: todo
+ docs-troubleshooting: done
docs-use-cases: done
dynamic-devices:
status: exempt
diff --git a/homeassistant/components/habitica/sensor.py b/homeassistant/components/habitica/sensor.py
index 57c391f5c12..e715dd6d07b 100644
--- a/homeassistant/components/habitica/sensor.py
+++ b/homeassistant/components/habitica/sensor.py
@@ -28,18 +28,18 @@ from homeassistant.components.sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import (
IssueSeverity,
async_create_issue,
async_delete_issue,
)
from homeassistant.helpers.typing import StateType
+from homeassistant.util import dt as dt_util
from .const import ASSETS_URL, DOMAIN
-from .coordinator import HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry, HabiticaDataUpdateCoordinator
from .entity import HabiticaBase
-from .types import HabiticaConfigEntry
from .util import get_attribute_points, get_attributes_total, inventory_list
_LOGGER = logging.getLogger(__name__)
@@ -106,6 +106,20 @@ SENSOR_DESCRIPTIONS: tuple[HabiticaSensorEntityDescription, ...] = (
key=HabiticaSensorEntity.DISPLAY_NAME,
translation_key=HabiticaSensorEntity.DISPLAY_NAME,
value_fn=lambda user, _: user.profile.name,
+ attributes_fn=lambda user, _: {
+ "blurb": user.profile.blurb,
+ "joined": (
+ dt_util.as_local(joined).date()
+ if (joined := user.auth.timestamps.created)
+ else None
+ ),
+ "last_login": (
+ dt_util.as_local(last).date()
+ if (last := user.auth.timestamps.loggedin)
+ else None
+ ),
+ "total_logins": user.loginIncentives,
+ },
),
HabiticaSensorEntityDescription(
key=HabiticaSensorEntity.HEALTH,
@@ -306,7 +320,7 @@ def entity_used_in(hass: HomeAssistant, entity_id: str) -> list[str]:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HabiticaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the habitica sensors."""
@@ -394,6 +408,11 @@ class HabiticaSensor(HabiticaBase, SensorEntity):
):
return SVG_CLASS[_class]
+ if self.entity_description.key is HabiticaSensorEntity.DISPLAY_NAME and (
+ img_url := self.coordinator.data.user.profile.imageUrl
+ ):
+ return img_url
+
if entity_picture := self.entity_description.entity_picture:
return (
entity_picture
diff --git a/homeassistant/components/habitica/services.py b/homeassistant/components/habitica/services.py
index ed4a6444ea2..9fb0b0b7537 100644
--- a/homeassistant/components/habitica/services.py
+++ b/homeassistant/components/habitica/services.py
@@ -3,16 +3,23 @@
from __future__ import annotations
from dataclasses import asdict
+from datetime import UTC, date, datetime, time
import logging
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, cast
+from uuid import UUID, uuid4
from aiohttp import ClientError
from habiticalib import (
+ Checklist,
Direction,
+ Frequency,
HabiticaException,
NotAuthorizedError,
NotFoundError,
+ Reminders,
+ Repeat,
Skill,
+ Task,
TaskData,
TaskPriority,
TaskType,
@@ -20,8 +27,9 @@ from habiticalib import (
)
import voluptuous as vol
+from homeassistant.components.todo import ATTR_RENAME
from homeassistant.config_entries import ConfigEntryState
-from homeassistant.const import ATTR_NAME, CONF_NAME
+from homeassistant.const import ATTR_DATE, ATTR_NAME, CONF_NAME
from homeassistant.core import (
HomeAssistant,
ServiceCall,
@@ -32,21 +40,43 @@ from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.selector import ConfigEntrySelector
+from homeassistant.util import dt as dt_util
from .const import (
+ ATTR_ADD_CHECKLIST_ITEM,
+ ATTR_ALIAS,
ATTR_ARGS,
+ ATTR_CLEAR_DATE,
+ ATTR_CLEAR_REMINDER,
ATTR_CONFIG_ENTRY,
+ ATTR_COST,
+ ATTR_COUNTER_DOWN,
+ ATTR_COUNTER_UP,
ATTR_DATA,
ATTR_DIRECTION,
+ ATTR_FREQUENCY,
+ ATTR_INTERVAL,
ATTR_ITEM,
ATTR_KEYWORD,
+ ATTR_NOTES,
ATTR_PATH,
ATTR_PRIORITY,
+ ATTR_REMINDER,
+ ATTR_REMOVE_CHECKLIST_ITEM,
+ ATTR_REMOVE_REMINDER,
+ ATTR_REMOVE_TAG,
+ ATTR_REPEAT,
+ ATTR_REPEAT_MONTHLY,
+ ATTR_SCORE_CHECKLIST_ITEM,
ATTR_SKILL,
+ ATTR_START_DATE,
+ ATTR_STREAK,
ATTR_TAG,
ATTR_TARGET,
ATTR_TASK,
ATTR_TYPE,
+ ATTR_UNSCORE_CHECKLIST_ITEM,
+ ATTR_UP_DOWN,
DOMAIN,
EVENT_API_CALL_SUCCESS,
SERVICE_ABORT_QUEST,
@@ -54,6 +84,9 @@ from .const import (
SERVICE_API_CALL,
SERVICE_CANCEL_QUEST,
SERVICE_CAST_SKILL,
+ SERVICE_CREATE_HABIT,
+ SERVICE_CREATE_REWARD,
+ SERVICE_CREATE_TODO,
SERVICE_GET_TASKS,
SERVICE_LEAVE_QUEST,
SERVICE_REJECT_QUEST,
@@ -61,8 +94,13 @@ from .const import (
SERVICE_SCORE_REWARD,
SERVICE_START_QUEST,
SERVICE_TRANSFORMATION,
+ SERVICE_UPDATE_DAILY,
+ SERVICE_UPDATE_HABIT,
+ SERVICE_UPDATE_REWARD,
+ SERVICE_UPDATE_TODO,
+ WEEK_DAYS,
)
-from .types import HabiticaConfigEntry
+from .coordinator import HabiticaConfigEntry
_LOGGER = logging.getLogger(__name__)
@@ -77,7 +115,7 @@ SERVICE_API_CALL_SCHEMA = vol.Schema(
SERVICE_CAST_SKILL_SCHEMA = vol.Schema(
{
- vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector(),
+ vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector({"integration": DOMAIN}),
vol.Required(ATTR_SKILL): cv.string,
vol.Optional(ATTR_TASK): cv.string,
}
@@ -85,12 +123,12 @@ SERVICE_CAST_SKILL_SCHEMA = vol.Schema(
SERVICE_MANAGE_QUEST_SCHEMA = vol.Schema(
{
- vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector(),
+ vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector({"integration": DOMAIN}),
}
)
SERVICE_SCORE_TASK_SCHEMA = vol.Schema(
{
- vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector(),
+ vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector({"integration": DOMAIN}),
vol.Required(ATTR_TASK): cv.string,
vol.Optional(ATTR_DIRECTION): cv.string,
}
@@ -98,15 +136,74 @@ SERVICE_SCORE_TASK_SCHEMA = vol.Schema(
SERVICE_TRANSFORMATION_SCHEMA = vol.Schema(
{
- vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector(),
+ vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector({"integration": DOMAIN}),
vol.Required(ATTR_ITEM): cv.string,
vol.Required(ATTR_TARGET): cv.string,
}
)
-SERVICE_GET_TASKS_SCHEMA = vol.Schema(
+BASE_TASK_SCHEMA = vol.Schema(
{
vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector(),
+ vol.Optional(ATTR_RENAME): cv.string,
+ vol.Optional(ATTR_NOTES): cv.string,
+ vol.Optional(ATTR_TAG): vol.All(cv.ensure_list, [str]),
+ vol.Optional(ATTR_ALIAS): vol.All(
+ cv.string, cv.matches_regex("^[a-zA-Z0-9-_]*$")
+ ),
+ vol.Optional(ATTR_COST): vol.All(vol.Coerce(float), vol.Range(0)),
+ vol.Optional(ATTR_PRIORITY): vol.All(
+ vol.Upper, vol.In(TaskPriority._member_names_)
+ ),
+ vol.Optional(ATTR_UP_DOWN): vol.All(cv.ensure_list, [str]),
+ vol.Optional(ATTR_COUNTER_UP): vol.All(int, vol.Range(0)),
+ vol.Optional(ATTR_COUNTER_DOWN): vol.All(int, vol.Range(0)),
+ vol.Optional(ATTR_FREQUENCY): vol.Coerce(Frequency),
+ vol.Optional(ATTR_DATE): cv.date,
+ vol.Optional(ATTR_CLEAR_DATE): cv.boolean,
+ vol.Optional(ATTR_REMINDER): vol.All(
+ cv.ensure_list, [vol.Any(cv.datetime, cv.time)]
+ ),
+ vol.Optional(ATTR_REMOVE_REMINDER): vol.All(
+ cv.ensure_list, [vol.Any(cv.datetime, cv.time)]
+ ),
+ vol.Optional(ATTR_CLEAR_REMINDER): cv.boolean,
+ vol.Optional(ATTR_ADD_CHECKLIST_ITEM): vol.All(cv.ensure_list, [str]),
+ vol.Optional(ATTR_REMOVE_CHECKLIST_ITEM): vol.All(cv.ensure_list, [str]),
+ vol.Optional(ATTR_SCORE_CHECKLIST_ITEM): vol.All(cv.ensure_list, [str]),
+ vol.Optional(ATTR_UNSCORE_CHECKLIST_ITEM): vol.All(cv.ensure_list, [str]),
+ vol.Optional(ATTR_START_DATE): cv.date,
+ vol.Optional(ATTR_INTERVAL): vol.All(int, vol.Range(0)),
+ vol.Optional(ATTR_REPEAT): vol.All(cv.ensure_list, [vol.In(WEEK_DAYS)]),
+ vol.Optional(ATTR_REPEAT_MONTHLY): vol.All(
+ cv.string, vol.In({"day_of_month", "day_of_week"})
+ ),
+ vol.Optional(ATTR_STREAK): vol.All(int, vol.Range(0)),
+ }
+)
+
+SERVICE_UPDATE_TASK_SCHEMA = BASE_TASK_SCHEMA.extend(
+ {
+ vol.Required(ATTR_TASK): cv.string,
+ vol.Optional(ATTR_REMOVE_TAG): vol.All(cv.ensure_list, [str]),
+ }
+)
+
+SERVICE_CREATE_TASK_SCHEMA = BASE_TASK_SCHEMA.extend(
+ {
+ vol.Required(ATTR_NAME): cv.string,
+ }
+)
+
+SERVICE_DAILY_SCHEMA = {
+ vol.Optional(ATTR_REMINDER): vol.All(cv.ensure_list, [cv.time]),
+ vol.Optional(ATTR_REMOVE_REMINDER): vol.All(cv.ensure_list, [cv.time]),
+}
+
+
+SERVICE_GET_TASKS_SCHEMA = vol.Schema(
+ {
+ vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector({"integration": DOMAIN}),
vol.Optional(ATTR_TYPE): vol.All(
cv.ensure_list, [vol.All(vol.Upper, vol.In({x.name for x in TaskType}))]
),
@@ -138,6 +235,16 @@ ITEMID_MAP = {
"shiny_seed": Skill.SHINY_SEED,
}
+SERVICE_TASK_TYPE_MAP = {
+ SERVICE_UPDATE_REWARD: TaskType.REWARD,
+ SERVICE_CREATE_REWARD: TaskType.REWARD,
+ SERVICE_UPDATE_HABIT: TaskType.HABIT,
+ SERVICE_CREATE_HABIT: TaskType.HABIT,
+ SERVICE_UPDATE_TODO: TaskType.TODO,
+ SERVICE_CREATE_TODO: TaskType.TODO,
+ SERVICE_UPDATE_DAILY: TaskType.DAILY,
+}
+
def get_config_entry(hass: HomeAssistant, entry_id: str) -> HabiticaConfigEntry:
"""Return config entry or raise if not found or not loaded."""
@@ -510,10 +617,310 @@ def async_setup_services(hass: HomeAssistant) -> None: # noqa: C901
or (task.notes and keyword in task.notes.lower())
or any(keyword in item.text.lower() for item in task.checklist)
]
- result: dict[str, Any] = {"tasks": [task.to_dict() for task in response]}
+ result: dict[str, Any] = {
+ "tasks": [task.to_dict(omit_none=False) for task in response]
+ }
return result
+ async def create_or_update_task(call: ServiceCall) -> ServiceResponse: # noqa: C901
+ """Create or update task action."""
+ entry = get_config_entry(hass, call.data[ATTR_CONFIG_ENTRY])
+ coordinator = entry.runtime_data
+ await coordinator.async_refresh()
+ is_update = call.service in (
+ SERVICE_UPDATE_HABIT,
+ SERVICE_UPDATE_REWARD,
+ SERVICE_UPDATE_TODO,
+ SERVICE_UPDATE_DAILY,
+ )
+ task_type = SERVICE_TASK_TYPE_MAP[call.service]
+ current_task = None
+
+ if is_update:
+ try:
+ current_task = next(
+ task
+ for task in coordinator.data.tasks
+ if call.data[ATTR_TASK] in (str(task.id), task.alias, task.text)
+ and task.Type is task_type
+ )
+ except StopIteration as e:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="task_not_found",
+ translation_placeholders={"task": f"'{call.data[ATTR_TASK]}'"},
+ ) from e
+
+ data = Task()
+
+ if not is_update:
+ data["type"] = task_type
+
+ if (text := call.data.get(ATTR_RENAME)) or (text := call.data.get(ATTR_NAME)):
+ data["text"] = text
+
+ if (notes := call.data.get(ATTR_NOTES)) is not None:
+ data["notes"] = notes
+
+ tags = cast(list[str], call.data.get(ATTR_TAG))
+ remove_tags = cast(list[str], call.data.get(ATTR_REMOVE_TAG))
+
+ if tags or remove_tags:
+ update_tags = set(current_task.tags) if current_task else set()
+ user_tags = {
+ tag.name.lower(): tag.id
+ for tag in coordinator.data.user.tags
+ if tag.id and tag.name
+ }
+
+ if tags:
+ # Creates new tag if it doesn't exist
+ async def create_tag(tag_name: str) -> UUID:
+ tag_id = (await coordinator.habitica.create_tag(tag_name)).data.id
+ if TYPE_CHECKING:
+ assert tag_id
+ return tag_id
+
+ try:
+ update_tags.update(
+ {
+ user_tags.get(tag_name.lower())
+ or (await create_tag(tag_name))
+ for tag_name in tags
+ }
+ )
+ except TooManyRequestsError as e:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="setup_rate_limit_exception",
+ translation_placeholders={"retry_after": str(e.retry_after)},
+ ) from e
+ except HabiticaException as e:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="service_call_exception",
+ translation_placeholders={"reason": str(e.error.message)},
+ ) from e
+ except ClientError as e:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="service_call_exception",
+ translation_placeholders={"reason": str(e)},
+ ) from e
+
+ if remove_tags:
+ update_tags.difference_update(
+ {
+ user_tags[tag_name.lower()]
+ for tag_name in remove_tags
+ if tag_name.lower() in user_tags
+ }
+ )
+
+ data["tags"] = list(update_tags)
+
+ if (alias := call.data.get(ATTR_ALIAS)) is not None:
+ data["alias"] = alias
+
+ if (cost := call.data.get(ATTR_COST)) is not None:
+ data["value"] = cost
+
+ if priority := call.data.get(ATTR_PRIORITY):
+ data["priority"] = TaskPriority[priority]
+
+ if frequency := call.data.get(ATTR_FREQUENCY):
+ data["frequency"] = frequency
+ else:
+ frequency = current_task.frequency if current_task else Frequency.WEEKLY
+
+ if up_down := call.data.get(ATTR_UP_DOWN):
+ data["up"] = "up" in up_down
+ data["down"] = "down" in up_down
+
+ if counter_up := call.data.get(ATTR_COUNTER_UP):
+ data["counterUp"] = counter_up
+
+ if counter_down := call.data.get(ATTR_COUNTER_DOWN):
+ data["counterDown"] = counter_down
+
+ if due_date := call.data.get(ATTR_DATE):
+ data["date"] = datetime.combine(due_date, time())
+
+ if call.data.get(ATTR_CLEAR_DATE):
+ data["date"] = None
+
+ checklist = current_task.checklist if current_task else []
+
+ if add_checklist_item := call.data.get(ATTR_ADD_CHECKLIST_ITEM):
+ checklist.extend(
+ Checklist(completed=False, id=uuid4(), text=item)
+ for item in add_checklist_item
+ if not any(i.text == item for i in checklist)
+ )
+ if remove_checklist_item := call.data.get(ATTR_REMOVE_CHECKLIST_ITEM):
+ checklist = [
+ item for item in checklist if item.text not in remove_checklist_item
+ ]
+
+ if score_checklist_item := call.data.get(ATTR_SCORE_CHECKLIST_ITEM):
+ for item in checklist:
+ if item.text in score_checklist_item:
+ item.completed = True
+
+ if unscore_checklist_item := call.data.get(ATTR_UNSCORE_CHECKLIST_ITEM):
+ for item in checklist:
+ if item.text in unscore_checklist_item:
+ item.completed = False
+ if (
+ add_checklist_item
+ or remove_checklist_item
+ or score_checklist_item
+ or unscore_checklist_item
+ ):
+ data["checklist"] = checklist
+
+ reminders = current_task.reminders if current_task else []
+
+ if add_reminders := call.data.get(ATTR_REMINDER):
+ if task_type is TaskType.TODO:
+ existing_reminder_datetimes = {
+ r.time.replace(tzinfo=None) for r in reminders
+ }
+
+ reminders.extend(
+ Reminders(id=uuid4(), time=r)
+ for r in add_reminders
+ if r not in existing_reminder_datetimes
+ )
+ if task_type is TaskType.DAILY:
+ existing_reminder_times = {
+ r.time.time().replace(microsecond=0, second=0) for r in reminders
+ }
+
+ reminders.extend(
+ Reminders(
+ id=uuid4(),
+ time=datetime.combine(date.today(), r, tzinfo=UTC),
+ )
+ for r in add_reminders
+ if r not in existing_reminder_times
+ )
+
+ if remove_reminder := call.data.get(ATTR_REMOVE_REMINDER):
+ if task_type is TaskType.TODO:
+ reminders = list(
+ filter(
+ lambda r: r.time.replace(tzinfo=None) not in remove_reminder,
+ reminders,
+ )
+ )
+ if task_type is TaskType.DAILY:
+ reminders = list(
+ filter(
+ lambda r: r.time.time().replace(second=0, microsecond=0)
+ not in remove_reminder,
+ reminders,
+ )
+ )
+
+ if clear_reminders := call.data.get(ATTR_CLEAR_REMINDER):
+ reminders = []
+
+ if add_reminders or remove_reminder or clear_reminders:
+ data["reminders"] = reminders
+
+ if start_date := call.data.get(ATTR_START_DATE):
+ data["startDate"] = datetime.combine(start_date, time())
+ else:
+ start_date = (
+ current_task.startDate
+ if current_task and current_task.startDate
+ else dt_util.start_of_local_day()
+ )
+ if repeat := call.data.get(ATTR_REPEAT):
+ if frequency is Frequency.WEEKLY:
+ data["repeat"] = Repeat(**{d: d in repeat for d in WEEK_DAYS})
+ else:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="frequency_not_weekly",
+ )
+ if repeat_monthly := call.data.get(ATTR_REPEAT_MONTHLY):
+ if frequency is not Frequency.MONTHLY:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="frequency_not_monthly",
+ )
+
+ if repeat_monthly == "day_of_week":
+ weekday = start_date.weekday()
+ data["weeksOfMonth"] = [(start_date.day - 1) // 7]
+ data["repeat"] = Repeat(
+ **{day: i == weekday for i, day in enumerate(WEEK_DAYS)}
+ )
+ data["daysOfMonth"] = []
+
+ else:
+ data["daysOfMonth"] = [start_date.day]
+ data["weeksOfMonth"] = []
+
+ if interval := call.data.get(ATTR_INTERVAL):
+ data["everyX"] = interval
+
+ if streak := call.data.get(ATTR_STREAK):
+ data["streak"] = streak
+
+ try:
+ if is_update:
+ if TYPE_CHECKING:
+ assert current_task
+ assert current_task.id
+ response = await coordinator.habitica.update_task(current_task.id, data)
+ else:
+ response = await coordinator.habitica.create_task(data)
+ except TooManyRequestsError as e:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="setup_rate_limit_exception",
+ translation_placeholders={"retry_after": str(e.retry_after)},
+ ) from e
+ except HabiticaException as e:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="service_call_exception",
+ translation_placeholders={"reason": str(e.error.message)},
+ ) from e
+ except ClientError as e:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="service_call_exception",
+ translation_placeholders={"reason": str(e)},
+ ) from e
+ else:
+ return response.data.to_dict(omit_none=True)
+
+ for service in (
+ SERVICE_UPDATE_DAILY,
+ SERVICE_UPDATE_HABIT,
+ SERVICE_UPDATE_REWARD,
+ SERVICE_UPDATE_TODO,
+ ):
+ hass.services.async_register(
+ DOMAIN,
+ service,
+ create_or_update_task,
+ schema=SERVICE_UPDATE_TASK_SCHEMA,
+ supports_response=SupportsResponse.ONLY,
+ )
+ for service in (SERVICE_CREATE_HABIT, SERVICE_CREATE_REWARD, SERVICE_CREATE_TODO):
+ hass.services.async_register(
+ DOMAIN,
+ service,
+ create_or_update_task,
+ schema=SERVICE_CREATE_TASK_SCHEMA,
+ supports_response=SupportsResponse.ONLY,
+ )
hass.services.async_register(
DOMAIN,
SERVICE_API_CALL,
diff --git a/homeassistant/components/habitica/services.yaml b/homeassistant/components/habitica/services.yaml
index f3095518290..46b3211790e 100644
--- a/homeassistant/components/habitica/services.yaml
+++ b/homeassistant/components/habitica/services.yaml
@@ -140,3 +140,292 @@ get_tasks:
required: false
selector:
text:
+update_reward:
+ fields:
+ config_entry: *config_entry
+ task: *task
+ rename: &rename
+ selector:
+ text:
+ notes: ¬es
+ required: false
+ selector:
+ text:
+ multiline: true
+ cost:
+ required: false
+ selector: &cost_selector
+ number:
+ min: 0
+ step: 0.01
+ unit_of_measurement: "🪙"
+ mode: box
+ tag_options: &tag_options
+ collapsed: true
+ fields:
+ tag: &tag
+ required: false
+ selector:
+ text:
+ multiple: true
+ remove_tag:
+ required: false
+ selector:
+ text:
+ multiple: true
+ developer_options: &developer_options
+ collapsed: true
+ fields:
+ alias: &alias
+ required: false
+ selector:
+ text:
+create_reward:
+ fields:
+ config_entry: *config_entry
+ name: &name
+ required: true
+ selector:
+ text:
+ notes: *notes
+ cost:
+ required: true
+ selector: *cost_selector
+ tag: *tag
+ developer_options: *developer_options
+update_habit:
+ fields:
+ config_entry: *config_entry
+ task: *task
+ rename: *rename
+ notes: *notes
+ up_down: &up_down
+ required: false
+ selector:
+ select:
+ options:
+ - value: up
+ label: "➕"
+ - value: down
+ label: "➖"
+ multiple: true
+ mode: list
+ priority: &priority
+ required: false
+ selector:
+ select:
+ options:
+ - "trivial"
+ - "easy"
+ - "medium"
+ - "hard"
+ mode: dropdown
+ translation_key: "priority"
+ frequency: &frequency
+ required: false
+ selector:
+ select:
+ options:
+ - "daily"
+ - "weekly"
+ - "monthly"
+ translation_key: "frequency"
+ mode: dropdown
+ tag_options: *tag_options
+ developer_options:
+ collapsed: true
+ fields:
+ counter_up:
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ unit_of_measurement: "➕"
+ mode: box
+ counter_down:
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ unit_of_measurement: "➖"
+ mode: box
+ alias: *alias
+create_habit:
+ fields:
+ config_entry: *config_entry
+ name: *name
+ notes: *notes
+ up_down: *up_down
+ priority: *priority
+ frequency: *frequency
+ tag: *tag
+ developer_options: *developer_options
+update_todo:
+ fields:
+ config_entry: *config_entry
+ task: *task
+ rename: *rename
+ notes: *notes
+ checklist_options: &checklist_options
+ collapsed: true
+ fields:
+ add_checklist_item: &add_checklist_item
+ required: false
+ selector:
+ text:
+ multiple: true
+ remove_checklist_item:
+ required: false
+ selector:
+ text:
+ multiple: true
+ score_checklist_item:
+ required: false
+ selector:
+ text:
+ multiple: true
+ unscore_checklist_item:
+ required: false
+ selector:
+ text:
+ multiple: true
+ priority: *priority
+ duedate_options:
+ collapsed: true
+ fields:
+ date: &due_date
+ required: false
+ selector:
+ date:
+ clear_date:
+ required: false
+ selector:
+ constant:
+ value: true
+ label: "🗑️"
+ reminder_options:
+ collapsed: true
+ fields:
+ reminder: &reminder
+ required: false
+ selector:
+ text:
+ type: datetime-local
+ multiple: true
+ remove_reminder:
+ required: false
+ selector:
+ text:
+ type: datetime-local
+ multiple: true
+ clear_reminder: &clear_reminder
+ required: false
+ selector:
+ constant:
+ value: true
+ label: "🗑️"
+ tag_options: *tag_options
+ developer_options: *developer_options
+create_todo:
+ fields:
+ config_entry: *config_entry
+ name: *name
+ notes: *notes
+ add_checklist_item: *add_checklist_item
+ priority: *priority
+ date: *due_date
+ reminder: *reminder
+ tag: *tag
+ developer_options: *developer_options
+update_daily:
+ fields:
+ config_entry: *config_entry
+ task: *task
+ rename: *rename
+ notes: *notes
+ checklist_options: *checklist_options
+ priority: *priority
+ start_date:
+ required: false
+ selector:
+ date:
+ frequency:
+ required: false
+ selector:
+ select:
+ options:
+ - "daily"
+ - "weekly"
+ - "monthly"
+ - "yearly"
+ translation_key: "frequency"
+ mode: dropdown
+ every_x:
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ unit_of_measurement: "🔃"
+ mode: box
+ repeat_weekly_options:
+ collapsed: true
+ fields:
+ repeat:
+ required: false
+ selector:
+ select:
+ options:
+ - "m"
+ - "t"
+ - "w"
+ - "th"
+ - "f"
+ - "s"
+ - "su"
+ mode: list
+ translation_key: repeat
+ multiple: true
+ repeat_monthly_options:
+ collapsed: true
+ fields:
+ repeat_monthly:
+ required: false
+ selector:
+ select:
+ options:
+ - "day_of_month"
+ - "day_of_week"
+ translation_key: repeat_monthly
+ mode: list
+ reminder_options:
+ collapsed: true
+ fields:
+ reminder:
+ required: false
+ selector:
+ text:
+ type: time
+ multiple: true
+ remove_reminder:
+ required: false
+ selector:
+ text:
+ type: time
+ multiple: true
+ clear_reminder: *clear_reminder
+ tag_options: *tag_options
+ developer_options:
+ collapsed: true
+ fields:
+ streak:
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ unit_of_measurement: "▶▶"
+ mode: box
+ alias: *alias
diff --git a/homeassistant/components/habitica/strings.json b/homeassistant/components/habitica/strings.json
index 4d353cec40e..fac0fdf3868 100644
--- a/homeassistant/components/habitica/strings.json
+++ b/homeassistant/components/habitica/strings.json
@@ -7,7 +7,52 @@
"unit_tasks": "tasks",
"unit_health_points": "HP",
"unit_mana_points": "MP",
- "unit_experience_points": "XP"
+ "unit_experience_points": "XP",
+ "config_entry_description": "Select the Habitica account to update a task.",
+ "task_description": "The name (or task ID) of the task you want to update.",
+ "rename_name": "Rename",
+ "rename_description": "The title for the Habitica task.",
+ "notes_name": "Notes",
+ "notes_description": "The notes for the Habitica task.",
+ "tag_name": "Add tags",
+ "tag_description": "Add tags to the Habitica task. If a tag does not already exist, a new one will be created.",
+ "remove_tag_name": "Remove tags",
+ "remove_tag_description": "Remove tags from the Habitica task.",
+ "alias_name": "Task alias",
+ "alias_description": "A task alias can be used instead of the name or task ID. Only dashes, underscores, and alphanumeric characters are supported. The task alias must be unique among all your tasks.",
+ "developer_options_name": "Advanced settings",
+ "developer_options_description": "Additional features available in developer mode.",
+ "tag_options_name": "Tags",
+ "tag_options_description": "Add or remove tags from a task.",
+ "name_description": "The title for the Habitica task.",
+ "cost_name": "Cost",
+ "priority_name": "Difficulty",
+ "priority_description": "The difficulty of the task.",
+ "frequency_name": "Counter reset",
+ "frequency_description": "The frequency at which the habit's counter resets: daily at the start of a new day, weekly after Sunday night, or monthly at the beginning of a new month.",
+ "up_down_name": "Rewards or losses",
+ "up_down_description": "Whether the habit is good and rewarding (positive), bad and penalizing (negative), or both.",
+ "add_checklist_item_name": "Add checklist items",
+ "add_checklist_item_description": "The items to add to a task's checklist.",
+ "remove_checklist_item_name": "Delete items",
+ "remove_checklist_item_description": "Remove items from a task's checklist.",
+ "score_checklist_item_name": "Complete items",
+ "score_checklist_item_description": "Mark items from a task's checklist as completed.",
+ "unscore_checklist_item_name": "Uncomplete items",
+ "unscore_checklist_item_description": "Undo completion of items of a task's checklist.",
+ "checklist_options_name": "Checklist",
+ "checklist_options_description": "Add, remove, or update status of an item on a task's checklist.",
+ "reminder_name": "Add reminders",
+ "reminder_description": "Add reminders to a Habitica task.",
+ "remove_reminder_name": "Remove reminders",
+ "remove_reminder_description": "Remove specific reminders from a Habitica task.",
+ "clear_reminder_name": "Clear all reminders",
+ "clear_reminder_description": "Remove all reminders from a Habitica task.",
+ "reminder_options_name": "Reminders",
+ "reminder_options_description": "Add, remove or clear reminders of a Habitica task.",
+ "date_name": "Due date",
+ "date_description": "The to-do's due date.",
+ "repeat_name": "Repeat on"
},
"config": {
"abort": {
@@ -199,7 +244,21 @@
},
"sensor": {
"display_name": {
- "name": "Display name"
+ "name": "Display name",
+ "state_attributes": {
+ "blurb": {
+ "name": "About"
+ },
+ "joined": {
+ "name": "Joined"
+ },
+ "last_login": {
+ "name": "Last login"
+ },
+ "total_logins": {
+ "name": "Total logins"
+ }
+ }
},
"health": {
"name": "Health",
@@ -443,6 +502,12 @@
},
"authentication_failed": {
"message": "Authentication failed. It looks like your API token has been reset. Please re-authenticate using your new token"
+ },
+ "frequency_not_weekly": {
+ "message": "Unable to update task, weekly repeat settings apply only to weekly recurring dailies."
+ },
+ "frequency_not_monthly": {
+ "message": "Unable to update task, monthly repeat settings apply only to monthly recurring dailies."
}
},
"issues": {
@@ -534,7 +599,7 @@
},
"cancel_quest": {
"name": "Cancel a pending quest",
- "description": "Cancels a quest that has not yet startet. All accepted and pending invitations will be canceled and the quest roll returned to the owner's inventory. Only quest leader or group leader can perform this action.",
+ "description": "Cancels a quest that has not yet started. All accepted and pending invitations will be canceled and the quest roll returned to the owner's inventory. Only quest leader or group leader can perform this action.",
"fields": {
"config_entry": {
"name": "[%key:component::habitica::common::config_entry_name%]",
@@ -615,7 +680,7 @@
"description": "Filter tasks by type."
},
"priority": {
- "name": "Difficulty",
+ "name": "[%key:component::habitica::common::priority_name%]",
"description": "Filter tasks by difficulty."
},
"task": {
@@ -637,6 +702,458 @@
"description": "Use the optional filters to narrow the returned tasks."
}
}
+ },
+ "update_reward": {
+ "name": "Update a reward",
+ "description": "Updates a specific reward for the selected Habitica character",
+ "fields": {
+ "config_entry": {
+ "name": "[%key:component::habitica::common::config_entry_name%]",
+ "description": "Select the Habitica account to update a reward."
+ },
+ "task": {
+ "name": "[%key:component::habitica::common::task_name%]",
+ "description": "[%key:component::habitica::common::task_description%]"
+ },
+ "rename": {
+ "name": "[%key:component::habitica::common::rename_name%]",
+ "description": "[%key:component::habitica::common::rename_description%]"
+ },
+ "notes": {
+ "name": "[%key:component::habitica::common::notes_name%]",
+ "description": "[%key:component::habitica::common::notes_description%]"
+ },
+ "tag": {
+ "name": "[%key:component::habitica::common::tag_name%]",
+ "description": "[%key:component::habitica::common::tag_description%]"
+ },
+ "remove_tag": {
+ "name": "[%key:component::habitica::common::remove_tag_name%]",
+ "description": "[%key:component::habitica::common::remove_tag_description%]"
+ },
+ "alias": {
+ "name": "[%key:component::habitica::common::alias_name%]",
+ "description": "[%key:component::habitica::common::alias_description%]"
+ },
+ "cost": {
+ "name": "[%key:component::habitica::common::cost_name%]",
+ "description": "Update the cost of a reward."
+ }
+ },
+ "sections": {
+ "tag_options": {
+ "name": "[%key:component::habitica::common::tag_options_name%]",
+ "description": "[%key:component::habitica::common::tag_options_description%]"
+ },
+ "developer_options": {
+ "name": "[%key:component::habitica::common::developer_options_name%]",
+ "description": "[%key:component::habitica::common::developer_options_description%]"
+ }
+ }
+ },
+ "create_reward": {
+ "name": "Create reward",
+ "description": "Adds a new custom reward.",
+ "fields": {
+ "config_entry": {
+ "name": "[%key:component::habitica::common::config_entry_name%]",
+ "description": "Select the Habitica account to create a reward."
+ },
+ "name": {
+ "name": "[%key:component::habitica::common::task_name%]",
+ "description": "[%key:component::habitica::common::name_description%]"
+ },
+ "notes": {
+ "name": "[%key:component::habitica::common::notes_name%]",
+ "description": "[%key:component::habitica::common::notes_description%]"
+ },
+ "tag": {
+ "name": "[%key:component::habitica::common::tag_options_name%]",
+ "description": "[%key:component::habitica::common::tag_description%]"
+ },
+ "alias": {
+ "name": "[%key:component::habitica::common::alias_name%]",
+ "description": "[%key:component::habitica::common::alias_description%]"
+ },
+ "cost": {
+ "name": "[%key:component::habitica::common::cost_name%]",
+ "description": "The cost of the reward."
+ }
+ },
+ "sections": {
+ "developer_options": {
+ "name": "[%key:component::habitica::common::developer_options_name%]",
+ "description": "[%key:component::habitica::common::developer_options_description%]"
+ }
+ }
+ },
+ "update_habit": {
+ "name": "Update a habit",
+ "description": "Updates a specific habit for the selected Habitica character",
+ "fields": {
+ "config_entry": {
+ "name": "[%key:component::habitica::common::config_entry_name%]",
+ "description": "Select the Habitica account to update a habit."
+ },
+ "task": {
+ "name": "[%key:component::habitica::common::task_name%]",
+ "description": "[%key:component::habitica::common::task_description%]"
+ },
+ "rename": {
+ "name": "[%key:component::habitica::common::rename_name%]",
+ "description": "[%key:component::habitica::common::rename_description%]"
+ },
+ "notes": {
+ "name": "[%key:component::habitica::common::notes_name%]",
+ "description": "[%key:component::habitica::common::notes_description%]"
+ },
+ "tag": {
+ "name": "[%key:component::habitica::common::tag_name%]",
+ "description": "[%key:component::habitica::common::tag_description%]"
+ },
+ "remove_tag": {
+ "name": "[%key:component::habitica::common::remove_tag_name%]",
+ "description": "[%key:component::habitica::common::remove_tag_description%]"
+ },
+ "alias": {
+ "name": "[%key:component::habitica::common::alias_name%]",
+ "description": "[%key:component::habitica::common::alias_description%]"
+ },
+ "priority": {
+ "name": "[%key:component::habitica::common::priority_name%]",
+ "description": "[%key:component::habitica::common::priority_description%]"
+ },
+ "frequency": {
+ "name": "[%key:component::habitica::common::frequency_name%]",
+ "description": "[%key:component::habitica::common::frequency_description%]"
+ },
+ "up_down": {
+ "name": "[%key:component::habitica::common::up_down_name%]",
+ "description": "[%key:component::habitica::common::up_down_description%]"
+ },
+ "counter_up": {
+ "name": "Adjust positive counter",
+ "description": "Update the up counter of a positive habit."
+ },
+ "counter_down": {
+ "name": "Adjust negative counter",
+ "description": "Update the down counter of a negative habit."
+ }
+ },
+ "sections": {
+ "tag_options": {
+ "name": "[%key:component::habitica::common::tag_options_name%]",
+ "description": "[%key:component::habitica::common::tag_options_description%]"
+ },
+ "developer_options": {
+ "name": "[%key:component::habitica::common::developer_options_name%]",
+ "description": "[%key:component::habitica::common::developer_options_description%]"
+ }
+ }
+ },
+ "create_habit": {
+ "name": "Create habit",
+ "description": "Adds a new habit.",
+ "fields": {
+ "config_entry": {
+ "name": "[%key:component::habitica::common::config_entry_name%]",
+ "description": "Select the Habitica account to create a habit."
+ },
+ "name": {
+ "name": "[%key:component::habitica::common::task_name%]",
+ "description": "[%key:component::habitica::common::name_description%]"
+ },
+ "notes": {
+ "name": "[%key:component::habitica::common::notes_name%]",
+ "description": "[%key:component::habitica::common::notes_description%]"
+ },
+ "tag": {
+ "name": "[%key:component::habitica::common::tag_options_name%]",
+ "description": "[%key:component::habitica::common::tag_description%]"
+ },
+ "alias": {
+ "name": "[%key:component::habitica::common::alias_name%]",
+ "description": "[%key:component::habitica::common::alias_description%]"
+ },
+ "priority": {
+ "name": "[%key:component::habitica::common::priority_name%]",
+ "description": "[%key:component::habitica::common::priority_description%]"
+ },
+ "frequency": {
+ "name": "[%key:component::habitica::common::frequency_name%]",
+ "description": "[%key:component::habitica::common::frequency_description%]"
+ },
+ "up_down": {
+ "name": "[%key:component::habitica::common::up_down_name%]",
+ "description": "[%key:component::habitica::common::up_down_description%]"
+ }
+ },
+ "sections": {
+ "developer_options": {
+ "name": "[%key:component::habitica::common::developer_options_name%]",
+ "description": "[%key:component::habitica::common::developer_options_description%]"
+ }
+ }
+ },
+ "update_todo": {
+ "name": "Update a to-do",
+ "description": "Updates a specific to-do for a selected Habitica character",
+ "fields": {
+ "config_entry": {
+ "name": "[%key:component::habitica::common::config_entry_name%]",
+ "description": "[%key:component::habitica::common::config_entry_description%]"
+ },
+ "task": {
+ "name": "[%key:component::habitica::common::task_name%]",
+ "description": "The name (or task ID) of the to-do you want to update."
+ },
+ "rename": {
+ "name": "[%key:component::habitica::common::rename_name%]",
+ "description": "[%key:component::habitica::common::rename_description%]"
+ },
+ "notes": {
+ "name": "[%key:component::habitica::common::notes_name%]",
+ "description": "[%key:component::habitica::common::notes_description%]"
+ },
+ "tag": {
+ "name": "[%key:component::habitica::common::tag_name%]",
+ "description": "[%key:component::habitica::common::tag_description%]"
+ },
+ "remove_tag": {
+ "name": "[%key:component::habitica::common::remove_tag_name%]",
+ "description": "[%key:component::habitica::common::remove_tag_description%]"
+ },
+ "alias": {
+ "name": "[%key:component::habitica::common::alias_name%]",
+ "description": "[%key:component::habitica::common::alias_description%]"
+ },
+ "priority": {
+ "name": "[%key:component::habitica::common::priority_name%]",
+ "description": "[%key:component::habitica::common::priority_description%]"
+ },
+ "date": {
+ "name": "[%key:component::habitica::common::date_name%]",
+ "description": "[%key:component::habitica::common::date_description%]"
+ },
+ "clear_date": {
+ "name": "Clear due date",
+ "description": "Remove the due date from the to-do."
+ },
+ "reminder": {
+ "name": "[%key:component::habitica::common::reminder_name%]",
+ "description": "[%key:component::habitica::common::reminder_description%]"
+ },
+ "remove_reminder": {
+ "name": "[%key:component::habitica::common::remove_reminder_name%]",
+ "description": "[%key:component::habitica::common::remove_reminder_description%]"
+ },
+ "clear_reminder": {
+ "name": "[%key:component::habitica::common::clear_reminder_name%]",
+ "description": "[%key:component::habitica::common::clear_reminder_description%]"
+ },
+ "add_checklist_item": {
+ "name": "[%key:component::habitica::common::add_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::add_checklist_item_description%]"
+ },
+ "remove_checklist_item": {
+ "name": "[%key:component::habitica::common::remove_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::remove_checklist_item_description%]"
+ },
+ "score_checklist_item": {
+ "name": "[%key:component::habitica::common::score_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::score_checklist_item_description%]"
+ },
+ "unscore_checklist_item": {
+ "name": "[%key:component::habitica::common::unscore_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::unscore_checklist_item_description%]"
+ }
+ },
+ "sections": {
+ "checklist_options": {
+ "name": "[%key:component::habitica::common::checklist_options_name%]",
+ "description": "[%key:component::habitica::common::checklist_options_description%]"
+ },
+ "duedate_options": {
+ "name": "[%key:component::habitica::common::date_name%]",
+ "description": "Set, update or remove due dates of a to-do."
+ },
+ "reminder_options": {
+ "name": "[%key:component::habitica::common::reminder_options_name%]",
+ "description": "[%key:component::habitica::common::reminder_options_description%]"
+ },
+ "tag_options": {
+ "name": "[%key:component::habitica::common::tag_options_name%]",
+ "description": "[%key:component::habitica::common::tag_options_description%]"
+ },
+ "developer_options": {
+ "name": "[%key:component::habitica::common::developer_options_name%]",
+ "description": "[%key:component::habitica::common::developer_options_description%]"
+ }
+ }
+ },
+ "create_todo": {
+ "name": "Create to-do",
+ "description": "Adds a new to-do.",
+ "fields": {
+ "config_entry": {
+ "name": "[%key:component::habitica::common::config_entry_name%]",
+ "description": "Select the Habitica account to create a to-do."
+ },
+ "name": {
+ "name": "[%key:component::habitica::common::task_name%]",
+ "description": "[%key:component::habitica::common::name_description%]"
+ },
+ "notes": {
+ "name": "[%key:component::habitica::common::notes_name%]",
+ "description": "[%key:component::habitica::common::notes_description%]"
+ },
+ "tag": {
+ "name": "[%key:component::habitica::common::tag_options_name%]",
+ "description": "[%key:component::habitica::common::tag_description%]"
+ },
+ "alias": {
+ "name": "[%key:component::habitica::common::alias_name%]",
+ "description": "[%key:component::habitica::common::alias_description%]"
+ },
+ "priority": {
+ "name": "[%key:component::habitica::common::priority_name%]",
+ "description": "[%key:component::habitica::common::priority_description%]"
+ },
+ "date": {
+ "name": "[%key:component::habitica::common::date_name%]",
+ "description": "[%key:component::habitica::common::date_description%]"
+ },
+ "reminder": {
+ "name": "[%key:component::habitica::common::reminder_options_name%]",
+ "description": "[%key:component::habitica::common::reminder_description%]"
+ },
+ "add_checklist_item": {
+ "name": "[%key:component::habitica::common::checklist_options_name%]",
+ "description": "[%key:component::habitica::common::add_checklist_item_description%]"
+ }
+ },
+ "sections": {
+ "developer_options": {
+ "name": "[%key:component::habitica::common::developer_options_name%]",
+ "description": "[%key:component::habitica::common::developer_options_description%]"
+ }
+ }
+ },
+ "update_daily": {
+ "name": "Update a daily",
+ "description": "Updates a specific daily for a selected Habitica character",
+ "fields": {
+ "config_entry": {
+ "name": "[%key:component::habitica::common::config_entry_name%]",
+ "description": "[%key:component::habitica::common::config_entry_description%]"
+ },
+ "task": {
+ "name": "[%key:component::habitica::common::task_name%]",
+ "description": "The name (or task ID) of the daily you want to update."
+ },
+ "rename": {
+ "name": "[%key:component::habitica::common::rename_name%]",
+ "description": "[%key:component::habitica::common::rename_description%]"
+ },
+ "notes": {
+ "name": "[%key:component::habitica::common::notes_name%]",
+ "description": "[%key:component::habitica::common::notes_description%]"
+ },
+ "tag": {
+ "name": "[%key:component::habitica::common::tag_name%]",
+ "description": "[%key:component::habitica::common::tag_description%]"
+ },
+ "remove_tag": {
+ "name": "[%key:component::habitica::common::remove_tag_name%]",
+ "description": "[%key:component::habitica::common::remove_tag_description%]"
+ },
+ "alias": {
+ "name": "[%key:component::habitica::common::alias_name%]",
+ "description": "[%key:component::habitica::common::alias_description%]"
+ },
+ "priority": {
+ "name": "[%key:component::habitica::common::priority_name%]",
+ "description": "[%key:component::habitica::common::priority_description%]"
+ },
+ "start_date": {
+ "name": "Start date",
+ "description": "Defines when the daily task becomes active and specifies the exact weekday or day of the month it repeats on."
+ },
+ "frequency": {
+ "name": "Repeat interval",
+ "description": "The repetition interval of a daily."
+ },
+ "every_x": {
+ "name": "Repeat every X",
+ "description": "The number of intervals (days, weeks, months, or years) after which the daily repeats, based on the chosen repetition interval. A value of 0 makes the daily inactive ('Grey Daily')."
+ },
+ "repeat": {
+ "name": "[%key:component::habitica::common::repeat_name%]",
+ "description": "The days of the week the daily repeats."
+ },
+ "repeat_monthly": {
+ "name": "[%key:component::habitica::common::repeat_name%]",
+ "description": "Whether a monthly recurring task repeats on the same calendar day each month or on the same weekday and week of the month, based on the start date."
+ },
+ "add_checklist_item": {
+ "name": "[%key:component::habitica::common::add_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::add_checklist_item_description%]"
+ },
+ "remove_checklist_item": {
+ "name": "[%key:component::habitica::common::remove_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::remove_checklist_item_description%]"
+ },
+ "score_checklist_item": {
+ "name": "[%key:component::habitica::common::score_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::score_checklist_item_description%]"
+ },
+ "unscore_checklist_item": {
+ "name": "[%key:component::habitica::common::unscore_checklist_item_name%]",
+ "description": "[%key:component::habitica::common::unscore_checklist_item_description%]"
+ },
+ "streak": {
+ "name": "Adjust streak",
+ "description": "Adjust or reset the streak counter of the daily."
+ },
+ "reminder": {
+ "name": "[%key:component::habitica::common::reminder_name%]",
+ "description": "[%key:component::habitica::common::reminder_description%]"
+ },
+ "remove_reminder": {
+ "name": "[%key:component::habitica::common::remove_reminder_name%]",
+ "description": "[%key:component::habitica::common::remove_reminder_description%]"
+ },
+ "clear_reminder": {
+ "name": "[%key:component::habitica::common::clear_reminder_name%]",
+ "description": "[%key:component::habitica::common::clear_reminder_description%]"
+ }
+ },
+ "sections": {
+ "checklist_options": {
+ "name": "[%key:component::habitica::common::checklist_options_name%]",
+ "description": "[%key:component::habitica::common::checklist_options_description%]"
+ },
+ "repeat_weekly_options": {
+ "name": "Weekly repeat days",
+ "description": "Options related to weekly repetition, applicable when the repetition interval is set to weekly."
+ },
+ "repeat_monthly_options": {
+ "name": "Monthly repeat day",
+ "description": "Options related to monthly repetition, applicable when the repetition interval is set to monthly."
+ },
+ "tag_options": {
+ "name": "[%key:component::habitica::common::tag_options_name%]",
+ "description": "[%key:component::habitica::common::tag_options_description%]"
+ },
+ "developer_options": {
+ "name": "[%key:component::habitica::common::developer_options_name%]",
+ "description": "[%key:component::habitica::common::developer_options_description%]"
+ },
+ "reminder_options": {
+ "name": "[%key:component::habitica::common::reminder_options_name%]",
+ "description": "[%key:component::habitica::common::reminder_options_description%]"
+ }
+ }
}
},
"selector": {
@@ -671,6 +1188,31 @@
"medium": "Medium",
"hard": "Hard"
}
+ },
+ "frequency": {
+ "options": {
+ "daily": "Daily",
+ "weekly": "Weekly",
+ "monthly": "Monthly",
+ "yearly": "Yearly"
+ }
+ },
+ "repeat": {
+ "options": {
+ "m": "[%key:common::time::monday%]",
+ "t": "[%key:common::time::tuesday%]",
+ "w": "[%key:common::time::wednesday%]",
+ "th": "[%key:common::time::thursday%]",
+ "f": "[%key:common::time::friday%]",
+ "s": "[%key:common::time::saturday%]",
+ "su": "[%key:common::time::sunday%]"
+ }
+ },
+ "repeat_monthly": {
+ "options": {
+ "day_of_month": "Day of the month",
+ "day_of_week": "Day of the week"
+ }
}
}
}
diff --git a/homeassistant/components/habitica/switch.py b/homeassistant/components/habitica/switch.py
index ddc0db27108..fb98460f7e5 100644
--- a/homeassistant/components/habitica/switch.py
+++ b/homeassistant/components/habitica/switch.py
@@ -13,11 +13,14 @@ from homeassistant.components.switch import (
SwitchEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .coordinator import HabiticaData, HabiticaDataUpdateCoordinator
+from .coordinator import (
+ HabiticaConfigEntry,
+ HabiticaData,
+ HabiticaDataUpdateCoordinator,
+)
from .entity import HabiticaBase
-from .types import HabiticaConfigEntry
PARALLEL_UPDATES = 1
@@ -52,7 +55,7 @@ SWTICH_DESCRIPTIONS: tuple[HabiticaSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: HabiticaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches from a config entry."""
diff --git a/homeassistant/components/habitica/todo.py b/homeassistant/components/habitica/todo.py
index c1786059300..71ba8e60e06 100644
--- a/homeassistant/components/habitica/todo.py
+++ b/homeassistant/components/habitica/todo.py
@@ -26,13 +26,12 @@ from homeassistant.components.todo import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import ASSETS_URL, DOMAIN
-from .coordinator import HabiticaDataUpdateCoordinator
+from .coordinator import HabiticaConfigEntry, HabiticaDataUpdateCoordinator
from .entity import HabiticaBase
-from .types import HabiticaConfigEntry
from .util import next_due_date
_LOGGER = logging.getLogger(__name__)
@@ -52,7 +51,7 @@ class HabiticaTodoList(StrEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HabiticaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor from a config entry created in the integrations UI."""
coordinator = config_entry.runtime_data
@@ -118,19 +117,24 @@ class BaseHabiticaListEntity(HabiticaBase, TodoListEntity):
"""Move an item in the To-do list."""
if TYPE_CHECKING:
assert self.todo_items
+ tasks_order = (
+ self.coordinator.data.user.tasksOrder.todos
+ if self.entity_description.key is HabiticaTodoList.TODOS
+ else self.coordinator.data.user.tasksOrder.dailys
+ )
if previous_uid:
- pos = (
- self.todo_items.index(
- next(item for item in self.todo_items if item.uid == previous_uid)
- )
- + 1
- )
+ pos = tasks_order.index(UUID(previous_uid))
+ if pos < tasks_order.index(UUID(uid)):
+ pos += 1
+
else:
pos = 0
try:
- await self.coordinator.habitica.reorder_task(UUID(uid), pos)
+ tasks_order[:] = (
+ await self.coordinator.habitica.reorder_task(UUID(uid), pos)
+ ).data
except TooManyRequestsError as e:
raise HomeAssistantError(
translation_domain=DOMAIN,
@@ -144,20 +148,6 @@ class BaseHabiticaListEntity(HabiticaBase, TodoListEntity):
translation_key=f"move_{self.entity_description.key}_item_failed",
translation_placeholders={"pos": str(pos)},
) from e
- else:
- # move tasks in the coordinator until we have fresh data
- tasks = self.coordinator.data.tasks
- new_pos = (
- tasks.index(
- next(task for task in tasks if task.id == UUID(previous_uid))
- )
- + 1
- if previous_uid
- else 0
- )
- old_pos = tasks.index(next(task for task in tasks if task.id == UUID(uid)))
- tasks.insert(new_pos, tasks.pop(old_pos))
- await self.coordinator.async_request_refresh()
async def async_update_todo_item(self, item: TodoItem) -> None:
"""Update a Habitica todo."""
@@ -271,7 +261,7 @@ class HabiticaTodosListEntity(BaseHabiticaListEntity):
def todo_items(self) -> list[TodoItem]:
"""Return the todo items."""
- return [
+ tasks = [
*(
TodoItem(
uid=str(task.id),
@@ -288,6 +278,15 @@ class HabiticaTodosListEntity(BaseHabiticaListEntity):
if task.Type is TaskType.TODO
),
]
+ return sorted(
+ tasks,
+ key=lambda task: (
+ float("inf")
+ if (uid := UUID(task.uid))
+ not in (tasks_order := self.coordinator.data.user.tasksOrder.todos)
+ else tasks_order.index(uid)
+ ),
+ )
async def async_create_todo_item(self, item: TodoItem) -> None:
"""Create a Habitica todo."""
@@ -348,7 +347,7 @@ class HabiticaDailiesListEntity(BaseHabiticaListEntity):
if TYPE_CHECKING:
assert self.coordinator.data.user.lastCron
- return [
+ tasks = [
*(
TodoItem(
uid=str(task.id),
@@ -365,3 +364,12 @@ class HabiticaDailiesListEntity(BaseHabiticaListEntity):
if task.Type is TaskType.DAILY
)
]
+ return sorted(
+ tasks,
+ key=lambda task: (
+ float("inf")
+ if (uid := UUID(task.uid))
+ not in (tasks_order := self.coordinator.data.user.tasksOrder.dailys)
+ else tasks_order.index(uid)
+ ),
+ )
diff --git a/homeassistant/components/habitica/types.py b/homeassistant/components/habitica/types.py
deleted file mode 100644
index 9789a65dc40..00000000000
--- a/homeassistant/components/habitica/types.py
+++ /dev/null
@@ -1,18 +0,0 @@
-"""Types for Habitica integration."""
-
-from enum import StrEnum
-
-from homeassistant.config_entries import ConfigEntry
-
-from .coordinator import HabiticaDataUpdateCoordinator
-
-type HabiticaConfigEntry = ConfigEntry[HabiticaDataUpdateCoordinator]
-
-
-class HabiticaTaskType(StrEnum):
- """Habitica Entities."""
-
- HABIT = "habit"
- DAILY = "daily"
- TODO = "todo"
- REWARD = "reward"
diff --git a/homeassistant/components/harmony/manifest.json b/homeassistant/components/harmony/manifest.json
index aab4f51b09a..f67eb4db5aa 100644
--- a/homeassistant/components/harmony/manifest.json
+++ b/homeassistant/components/harmony/manifest.json
@@ -7,7 +7,7 @@
"documentation": "https://www.home-assistant.io/integrations/harmony",
"iot_class": "local_push",
"loggers": ["aioharmony", "slixmpp"],
- "requirements": ["aioharmony==0.4.1"],
+ "requirements": ["aioharmony==0.5.2"],
"ssdp": [
{
"manufacturer": "Logitech",
diff --git a/homeassistant/components/harmony/remote.py b/homeassistant/components/harmony/remote.py
index 43bf0a348c0..d09dc3ff7e8 100644
--- a/homeassistant/components/harmony/remote.py
+++ b/homeassistant/components/harmony/remote.py
@@ -22,7 +22,7 @@ from homeassistant.components.remote import (
from homeassistant.core import HassJob, HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import VolDictType
@@ -56,7 +56,7 @@ HARMONY_CHANGE_CHANNEL_SCHEMA: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
entry: HarmonyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Harmony config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/harmony/select.py b/homeassistant/components/harmony/select.py
index 731b6836386..3f45a23e26e 100644
--- a/homeassistant/components/harmony/select.py
+++ b/homeassistant/components/harmony/select.py
@@ -6,7 +6,7 @@ import logging
from homeassistant.components.select import SelectEntity
from homeassistant.core import HassJob, HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ACTIVITY_POWER_OFF, DOMAIN
from .data import HarmonyConfigEntry, HarmonyData
@@ -21,7 +21,7 @@ TRANSLATABLE_POWER_OFF = "power_off"
async def async_setup_entry(
hass: HomeAssistant,
entry: HarmonyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up harmony activities select."""
async_add_entities([HarmonyActivitySelect(entry.runtime_data)])
diff --git a/homeassistant/components/hassio/backup.py b/homeassistant/components/hassio/backup.py
index ddaa821587f..20f1ec82a7a 100644
--- a/homeassistant/components/hassio/backup.py
+++ b/homeassistant/components/hassio/backup.py
@@ -4,6 +4,7 @@ from __future__ import annotations
import asyncio
from collections.abc import AsyncIterator, Callable, Coroutine, Mapping
+from contextlib import suppress
import logging
import os
from pathlib import Path, PurePath
@@ -27,11 +28,13 @@ from homeassistant.components.backup import (
AddonInfo,
AgentBackup,
BackupAgent,
+ BackupConfig,
BackupManagerError,
BackupNotFound,
BackupReaderWriter,
BackupReaderWriterError,
CreateBackupEvent,
+ CreateBackupParametersDict,
CreateBackupStage,
CreateBackupState,
Folder,
@@ -43,13 +46,13 @@ from homeassistant.components.backup import (
RestoreBackupStage,
RestoreBackupState,
WrittenBackup,
- async_get_manager as async_get_backup_manager,
suggested_filename as suggested_backup_filename,
suggested_filename_from_name_date,
)
from homeassistant.const import __version__ as HAVERSION
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.backup import async_get_manager as async_get_backup_manager
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.util import dt as dt_util
from homeassistant.util.enum import try_parse_enum
@@ -171,7 +174,7 @@ class SupervisorBackupAgent(BackupAgent):
),
)
except SupervisorNotFoundError as err:
- raise BackupNotFound from err
+ raise BackupNotFound(f"Backup {backup_id} not found") from err
async def async_upload_backup(
self,
@@ -184,13 +187,14 @@ class SupervisorBackupAgent(BackupAgent):
The upload will be skipped if the backup already exists in the agent's location.
"""
- if await self.async_get_backup(backup.backup_id):
- _LOGGER.debug(
- "Backup %s already exists in location %s",
- backup.backup_id,
- self.location,
- )
- return
+ with suppress(BackupNotFound):
+ if await self.async_get_backup(backup.backup_id):
+ _LOGGER.debug(
+ "Backup %s already exists in location %s",
+ backup.backup_id,
+ self.location,
+ )
+ return
stream = await open_stream()
upload_options = supervisor_backups.UploadBackupOptions(
location={self.location},
@@ -216,14 +220,14 @@ class SupervisorBackupAgent(BackupAgent):
self,
backup_id: str,
**kwargs: Any,
- ) -> AgentBackup | None:
+ ) -> AgentBackup:
"""Return a backup."""
try:
details = await self._client.backups.backup_info(backup_id)
- except SupervisorNotFoundError:
- return None
+ except SupervisorNotFoundError as err:
+ raise BackupNotFound(f"Backup {backup_id} not found") from err
if self.location not in details.location_attributes:
- return None
+ raise BackupNotFound(f"Backup {backup_id} not found")
return _backup_details_to_agent_backup(details, self.location)
async def async_delete_backup(self, backup_id: str, **kwargs: Any) -> None:
@@ -235,8 +239,8 @@ class SupervisorBackupAgent(BackupAgent):
location={self.location}
),
)
- except SupervisorNotFoundError:
- _LOGGER.debug("Backup %s does not exist", backup_id)
+ except SupervisorNotFoundError as err:
+ raise BackupNotFound(f"Backup {backup_id} not found") from err
class SupervisorBackupReaderWriter(BackupReaderWriter):
@@ -490,10 +494,12 @@ class SupervisorBackupReaderWriter(BackupReaderWriter):
) -> None:
"""Restore a backup."""
manager = self._hass.data[DATA_MANAGER]
- # The backup manager has already checked that the backup exists so we don't need to
- # check that here.
+ # The backup manager has already checked that the backup exists so we don't
+ # need to catch BackupNotFound here.
backup = await manager.backup_agents[agent_id].async_get_backup(backup_id)
if (
+ # Check for None to be backwards compatible with the old BackupAgent API,
+ # this can be removed in HA Core 2025.10
backup
and restore_homeassistant
and restore_database != backup.database_included
@@ -633,6 +639,27 @@ class SupervisorBackupReaderWriter(BackupReaderWriter):
_LOGGER.debug("Could not get restore job %s: %s", restore_job_id, err)
unsub()
+ async def async_validate_config(self, *, config: BackupConfig) -> None:
+ """Validate backup config.
+
+ Replace the core backup agent with the hassio default agent.
+ """
+ core_agent_id = "backup.local"
+ create_backup = config.data.create_backup
+ if core_agent_id not in create_backup.agent_ids:
+ _LOGGER.debug("Backup settings don't need to be adjusted")
+ return
+
+ default_agent = await _default_agent(self._client)
+ _LOGGER.info("Adjusting backup settings to not include core backup location")
+ automatic_agents = [
+ agent_id if agent_id != core_agent_id else default_agent
+ for agent_id in create_backup.agent_ids
+ ]
+ config.update(
+ create_backup=CreateBackupParametersDict(agent_ids=automatic_agents)
+ )
+
@callback
def _async_listen_job_events(
self, job_id: UUID, on_event: Callable[[Mapping[str, Any]], None]
@@ -728,7 +755,7 @@ async def backup_addon_before_update(
async def backup_core_before_update(hass: HomeAssistant) -> None:
"""Prepare for updating core."""
- backup_manager = async_get_backup_manager(hass)
+ backup_manager = await async_get_backup_manager(hass)
client = get_supervisor_client(hass)
try:
diff --git a/homeassistant/components/hassio/binary_sensor.py b/homeassistant/components/hassio/binary_sensor.py
index 9d6e2ba19da..e7c7427d728 100644
--- a/homeassistant/components/hassio/binary_sensor.py
+++ b/homeassistant/components/hassio/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ADDONS_COORDINATOR, ATTR_STARTED, ATTR_STATE, DATA_KEY_ADDONS
from .entity import HassioAddonEntity
@@ -38,7 +38,7 @@ ADDON_ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Binary sensor set up for Hass.io config entry."""
coordinator = hass.data[ADDONS_COORDINATOR]
diff --git a/homeassistant/components/hassio/sensor.py b/homeassistant/components/hassio/sensor.py
index 039bf483682..9b62faaabcf 100644
--- a/homeassistant/components/hassio/sensor.py
+++ b/homeassistant/components/hassio/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ADDONS_COORDINATOR,
@@ -111,7 +111,7 @@ HOST_ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Sensor set up for Hass.io config entry."""
coordinator = hass.data[ADDONS_COORDINATOR]
diff --git a/homeassistant/components/hassio/strings.json b/homeassistant/components/hassio/strings.json
index 799067b8215..a543dbc7f89 100644
--- a/homeassistant/components/hassio/strings.json
+++ b/homeassistant/components/hassio/strings.json
@@ -152,7 +152,7 @@
},
"unsupported_connectivity_check": {
"title": "Unsupported system - Connectivity check disabled",
- "description": "System is unsupported because Home Assistant cannot determine when an internet connection is available. Use the link to learn more and how to fix this."
+ "description": "System is unsupported because Home Assistant cannot determine when an Internet connection is available. Use the link to learn more and how to fix this."
},
"unsupported_content_trust": {
"title": "Unsupported system - Content-trust check disabled",
@@ -216,7 +216,7 @@
},
"unsupported_systemd_journal": {
"title": "Unsupported system - Systemd Journal issues",
- "description": "System is unsupported because Systemd Journal and/or the gateway service is missing, inactive or misconfigured . Use the link to learn more and how to fix this."
+ "description": "System is unsupported because Systemd Journal and/or the gateway service is missing, inactive or misconfigured. Use the link to learn more and how to fix this."
},
"unsupported_systemd_resolved": {
"title": "Unsupported system - Systemd-Resolved issues",
@@ -348,7 +348,7 @@
},
"homeassistant_exclude_database": {
"name": "Home Assistant exclude database",
- "description": "Exclude the Home Assistant database file from backup"
+ "description": "Exclude the Home Assistant database file from the backup."
}
}
},
@@ -385,8 +385,8 @@
"description": "[%key:component::hassio::services::backup_full::fields::location::description%]"
},
"homeassistant_exclude_database": {
- "name": "Home Assistant exclude database",
- "description": "Exclude the Home Assistant database file from backup"
+ "name": "[%key:component::hassio::services::backup_full::fields::homeassistant_exclude_database::name%]",
+ "description": "[%key:component::hassio::services::backup_full::fields::homeassistant_exclude_database::description%]"
}
}
},
diff --git a/homeassistant/components/hassio/update.py b/homeassistant/components/hassio/update.py
index 8e0585892f5..4ea703e87c3 100644
--- a/homeassistant/components/hassio/update.py
+++ b/homeassistant/components/hassio/update.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_ICON, ATTR_NAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ADDONS_COORDINATOR,
@@ -47,7 +47,7 @@ ENTITY_DESCRIPTION = UpdateEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Supervisor update based on a config entry."""
coordinator = hass.data[ADDONS_COORDINATOR]
diff --git a/homeassistant/components/heos/__init__.py b/homeassistant/components/heos/__init__.py
index 0c268b612ea..4df1a2fa0e1 100644
--- a/homeassistant/components/heos/__init__.py
+++ b/homeassistant/components/heos/__init__.py
@@ -37,24 +37,24 @@ async def async_setup_entry(hass: HomeAssistant, entry: HeosConfigEntry) -> bool
for device in device_registry.devices.get_devices_for_config_entry_id(
entry.entry_id
):
- for domain, player_id in device.identifiers:
- if domain == DOMAIN and not isinstance(player_id, str):
- # Create set of identifiers excluding this integration
- identifiers = { # type: ignore[unreachable]
- (domain, identifier)
- for domain, identifier in device.identifiers
- if domain != DOMAIN
- }
- migrated_identifiers = {(DOMAIN, str(player_id))}
- # Add migrated if not already present in another device, which occurs if the user downgraded and then upgraded
- if not device_registry.async_get_device(migrated_identifiers):
- identifiers.update(migrated_identifiers)
- if len(identifiers) > 0:
- device_registry.async_update_device(
- device.id, new_identifiers=identifiers
- )
- else:
- device_registry.async_remove_device(device.id)
+ for ident in device.identifiers:
+ if ident[0] != DOMAIN or isinstance(ident[1], str):
+ continue
+
+ player_id = int(ident[1]) # type: ignore[unreachable]
+
+ # Create set of identifiers excluding this integration
+ identifiers = {ident for ident in device.identifiers if ident[0] != DOMAIN}
+ migrated_identifiers = {(DOMAIN, str(player_id))}
+ # Add migrated if not already present in another device, which occurs if the user downgraded and then upgraded
+ if not device_registry.async_get_device(migrated_identifiers):
+ identifiers.update(migrated_identifiers)
+ if len(identifiers) > 0:
+ device_registry.async_update_device(
+ device.id, new_identifiers=identifiers
+ )
+ else:
+ device_registry.async_remove_device(device.id)
break
coordinator = HeosCoordinator(hass, entry)
@@ -69,3 +69,14 @@ async def async_setup_entry(hass: HomeAssistant, entry: HeosConfigEntry) -> bool
async def async_unload_entry(hass: HomeAssistant, entry: HeosConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+
+
+async def async_remove_config_entry_device(
+ hass: HomeAssistant, entry: HeosConfigEntry, device: dr.DeviceEntry
+) -> bool:
+ """Remove config entry from device if no longer present."""
+ return not any(
+ (domain, key)
+ for domain, key in device.identifiers
+ if domain == DOMAIN and int(key) in entry.runtime_data.heos.players
+ )
diff --git a/homeassistant/components/heos/config_flow.py b/homeassistant/components/heos/config_flow.py
index db2abee559c..e2d3e2522dc 100644
--- a/homeassistant/components/heos/config_flow.py
+++ b/homeassistant/components/heos/config_flow.py
@@ -5,11 +5,17 @@ import logging
from typing import TYPE_CHECKING, Any
from urllib.parse import urlparse
-from pyheos import CommandAuthenticationError, Heos, HeosError, HeosOptions
+from pyheos import (
+ CommandAuthenticationError,
+ ConnectionState,
+ Heos,
+ HeosError,
+ HeosOptions,
+)
import voluptuous as vol
from homeassistant.config_entries import (
- ConfigEntryState,
+ SOURCE_IGNORE,
ConfigFlow,
ConfigFlowResult,
OptionsFlow,
@@ -17,12 +23,9 @@ from homeassistant.config_entries import (
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import callback
from homeassistant.helpers import selector
-from homeassistant.helpers.service_info.ssdp import (
- ATTR_UPNP_FRIENDLY_NAME,
- SsdpServiceInfo,
-)
+from homeassistant.helpers.service_info.ssdp import SsdpServiceInfo
-from .const import DOMAIN
+from .const import DOMAIN, ENTRY_TITLE
from .coordinator import HeosConfigEntry
_LOGGER = logging.getLogger(__name__)
@@ -37,11 +40,6 @@ AUTH_SCHEMA = vol.Schema(
)
-def format_title(host: str) -> str:
- """Format the title for config entries."""
- return f"HEOS System (via {host})"
-
-
async def _validate_host(host: str, errors: dict[str, str]) -> bool:
"""Validate host is reachable, return True, otherwise populate errors and return False."""
heos = Heos(HeosOptions(host, events=False, heart_beat=False))
@@ -56,13 +54,19 @@ async def _validate_host(host: str, errors: dict[str, str]) -> bool:
async def _validate_auth(
- user_input: dict[str, str], heos: Heos, errors: dict[str, str]
+ user_input: dict[str, str], entry: HeosConfigEntry, errors: dict[str, str]
) -> bool:
"""Validate authentication by signing in or out, otherwise populate errors if needed."""
+ can_validate = (
+ hasattr(entry, "runtime_data")
+ and entry.runtime_data.heos.connection_state is ConnectionState.CONNECTED
+ )
if not user_input:
# Log out (neither username nor password provided)
+ if not can_validate:
+ return True
try:
- await heos.sign_out()
+ await entry.runtime_data.heos.sign_out()
except HeosError:
errors["base"] = "unknown"
_LOGGER.exception("Unexpected error occurred during sign-out")
@@ -81,8 +85,12 @@ async def _validate_auth(
return False
# Attempt to login (both username and password provided)
+ if not can_validate:
+ return True
try:
- await heos.sign_in(user_input[CONF_USERNAME], user_input[CONF_PASSWORD])
+ await entry.runtime_data.heos.sign_in(
+ user_input[CONF_USERNAME], user_input[CONF_PASSWORD]
+ )
except CommandAuthenticationError as err:
errors["base"] = "invalid_auth"
_LOGGER.warning("Failed to sign-in to HEOS Account: %s", err)
@@ -94,16 +102,32 @@ async def _validate_auth(
else:
_LOGGER.debug(
"Successfully signed-in to HEOS Account: %s",
- heos.signed_in_username,
+ entry.runtime_data.heos.signed_in_username,
)
return True
+def _get_current_hosts(entry: HeosConfigEntry) -> set[str]:
+ """Get a set of current hosts from the entry."""
+ hosts = set(entry.data[CONF_HOST])
+ if hasattr(entry, "runtime_data"):
+ hosts.update(
+ player.ip_address
+ for player in entry.runtime_data.heos.players.values()
+ if player.ip_address is not None
+ )
+ return hosts
+
+
class HeosFlowHandler(ConfigFlow, domain=DOMAIN):
"""Define a flow for HEOS."""
VERSION = 1
+ def __init__(self) -> None:
+ """Initialize the HEOS flow."""
+ self._discovered_host: str | None = None
+
@staticmethod
@callback
def async_get_options_flow(config_entry: HeosConfigEntry) -> OptionsFlow:
@@ -117,40 +141,86 @@ class HeosFlowHandler(ConfigFlow, domain=DOMAIN):
# Store discovered host
if TYPE_CHECKING:
assert discovery_info.ssdp_location
+
+ entry: HeosConfigEntry | None = await self.async_set_unique_id(DOMAIN)
hostname = urlparse(discovery_info.ssdp_location).hostname
- friendly_name = f"{discovery_info.upnp[ATTR_UPNP_FRIENDLY_NAME]} ({hostname})"
- self.hass.data.setdefault(DOMAIN, {})
- self.hass.data[DOMAIN][friendly_name] = hostname
- await self.async_set_unique_id(DOMAIN)
- # Show selection form
- return self.async_show_form(step_id="user")
+ assert hostname is not None
+
+ # Abort early when discovery is ignored or host is part of the current system
+ if entry and (
+ entry.source == SOURCE_IGNORE or hostname in _get_current_hosts(entry)
+ ):
+ return self.async_abort(reason="single_instance_allowed")
+
+ # Connect to discovered host and get system information
+ heos = Heos(HeosOptions(hostname, events=False, heart_beat=False))
+ try:
+ await heos.connect()
+ system_info = await heos.get_system_info()
+ except HeosError as error:
+ _LOGGER.debug(
+ "Failed to retrieve system information from discovered HEOS device %s",
+ hostname,
+ exc_info=error,
+ )
+ return self.async_abort(reason="cannot_connect")
+ finally:
+ await heos.disconnect()
+
+ # Select the preferred host, if available
+ if system_info.preferred_hosts:
+ hostname = system_info.preferred_hosts[0].ip_address
+
+ # Move to confirmation when not configured
+ if entry is None:
+ self._discovered_host = hostname
+ return await self.async_step_confirm_discovery()
+
+ # Only update if the configured host isn't part of the discovered hosts to ensure new players that come online don't trigger a reload
+ if entry.data[CONF_HOST] not in [host.ip_address for host in system_info.hosts]:
+ _LOGGER.debug(
+ "Updated host %s to discovered host %s", entry.data[CONF_HOST], hostname
+ )
+ return self.async_update_reload_and_abort(
+ entry,
+ data_updates={CONF_HOST: hostname},
+ reason="reconfigure_successful",
+ )
+ return self.async_abort(reason="single_instance_allowed")
+
+ async def async_step_confirm_discovery(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Confirm discovered HEOS system."""
+ if user_input is not None:
+ assert self._discovered_host is not None
+ return self.async_create_entry(
+ title=ENTRY_TITLE, data={CONF_HOST: self._discovered_host}
+ )
+
+ self._set_confirm_only()
+ return self.async_show_form(step_id="confirm_discovery")
async def async_step_user(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Obtain host and validate connection."""
- self.hass.data.setdefault(DOMAIN, {})
- await self.async_set_unique_id(DOMAIN)
+ await self.async_set_unique_id(DOMAIN, raise_on_progress=False)
+ self._abort_if_unique_id_configured(error="single_instance_allowed")
# Try connecting to host if provided
errors: dict[str, str] = {}
host = None
if user_input is not None:
host = user_input[CONF_HOST]
- # Map host from friendly name if in discovered hosts
- host = self.hass.data[DOMAIN].get(host, host)
if await _validate_host(host, errors):
- self.hass.data.pop(DOMAIN) # Remove discovery data
return self.async_create_entry(
- title=format_title(host), data={CONF_HOST: host}
+ title=ENTRY_TITLE, data={CONF_HOST: host}
)
# Return form
- host_type = (
- str if not self.hass.data[DOMAIN] else vol.In(list(self.hass.data[DOMAIN]))
- )
return self.async_show_form(
step_id="user",
- data_schema=vol.Schema({vol.Required(CONF_HOST, default=host): host_type}),
+ data_schema=vol.Schema({vol.Required(CONF_HOST, default=host): str}),
errors=errors,
)
@@ -186,8 +256,7 @@ class HeosFlowHandler(ConfigFlow, domain=DOMAIN):
errors: dict[str, str] = {}
entry: HeosConfigEntry = self._get_reauth_entry()
if user_input is not None:
- assert entry.state is ConfigEntryState.LOADED
- if await _validate_auth(user_input, entry.runtime_data.heos, errors):
+ if await _validate_auth(user_input, entry, errors):
return self.async_update_reload_and_abort(entry, options=user_input)
return self.async_show_form(
@@ -208,8 +277,7 @@ class HeosOptionsFlowHandler(OptionsFlow):
"""Manage the options."""
errors: dict[str, str] = {}
if user_input is not None:
- entry: HeosConfigEntry = self.config_entry
- if await _validate_auth(user_input, entry.runtime_data.heos, errors):
+ if await _validate_auth(user_input, self.config_entry, errors):
return self.async_create_entry(data=user_input)
return self.async_show_form(
diff --git a/homeassistant/components/heos/const.py b/homeassistant/components/heos/const.py
index 7f03fa11e79..6d603f7ad30 100644
--- a/homeassistant/components/heos/const.py
+++ b/homeassistant/components/heos/const.py
@@ -3,5 +3,9 @@
ATTR_PASSWORD = "password"
ATTR_USERNAME = "username"
DOMAIN = "heos"
+ENTRY_TITLE = "HEOS System"
+SERVICE_GROUP_VOLUME_SET = "group_volume_set"
+SERVICE_GROUP_VOLUME_DOWN = "group_volume_down"
+SERVICE_GROUP_VOLUME_UP = "group_volume_up"
SERVICE_SIGN_IN = "sign_in"
SERVICE_SIGN_OUT = "sign_out"
diff --git a/homeassistant/components/heos/coordinator.py b/homeassistant/components/heos/coordinator.py
index 94aa4ad0ab5..5e72eb1427e 100644
--- a/homeassistant/components/heos/coordinator.py
+++ b/homeassistant/components/heos/coordinator.py
@@ -6,7 +6,6 @@ entities to update. Entities subscribe to entity-specific updates within the ent
"""
from collections.abc import Callable, Sequence
-from datetime import datetime, timedelta
import logging
from typing import Any
@@ -16,6 +15,7 @@ from pyheos import (
HeosError,
HeosNowPlayingMedia,
HeosOptions,
+ HeosPlayer,
MediaItem,
MediaType,
PlayerUpdateResult,
@@ -24,10 +24,10 @@ from pyheos import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME, Platform
-from homeassistant.core import CALLBACK_TYPE, HassJob, HomeAssistant, callback
+from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr, entity_registry as er
-from homeassistant.helpers.event import async_call_later
+from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN
@@ -42,7 +42,6 @@ class HeosCoordinator(DataUpdateCoordinator[None]):
def __init__(self, hass: HomeAssistant, config_entry: HeosConfigEntry) -> None:
"""Set up the coordinator and set in config_entry."""
- self.host: str = config_entry.data[CONF_HOST]
credentials: Credentials | None = None
if config_entry.options:
credentials = Credentials(
@@ -52,18 +51,31 @@ class HeosCoordinator(DataUpdateCoordinator[None]):
# media position update upon start of playback or when media changes
self.heos = Heos(
HeosOptions(
- self.host,
+ config_entry.data[CONF_HOST],
all_progress_events=False,
auto_reconnect=True,
+ auto_failover=True,
credentials=credentials,
)
)
- self._update_sources_pending: bool = False
+ self._platform_callbacks: list[Callable[[Sequence[HeosPlayer]], None]] = []
+ self._update_sources_debouncer = Debouncer(
+ hass,
+ _LOGGER,
+ immediate=True,
+ cooldown=2.0,
+ function=self._async_update_sources,
+ )
self._source_list: list[str] = []
self._favorites: dict[int, MediaItem] = {}
self._inputs: Sequence[MediaItem] = []
super().__init__(hass, _LOGGER, config_entry=config_entry, name=DOMAIN)
+ @property
+ def host(self) -> str:
+ """Get the host address of the device."""
+ return self.heos.current_host
+
@property
def inputs(self) -> Sequence[MediaItem]:
"""Get input sources across all devices."""
@@ -124,6 +136,27 @@ class HeosCoordinator(DataUpdateCoordinator[None]):
self.async_update_listeners()
return remove_listener
+ def async_add_platform_callback(
+ self, add_entities_callback: Callable[[Sequence[HeosPlayer]], None]
+ ) -> None:
+ """Add a callback to add entities for a platform."""
+ self._platform_callbacks.append(add_entities_callback)
+
+ def _async_handle_player_update_result(
+ self, update_result: PlayerUpdateResult
+ ) -> None:
+ """Handle a player update result."""
+ if update_result.added_player_ids and self._platform_callbacks:
+ new_players = [
+ self.heos.players[player_id]
+ for player_id in update_result.added_player_ids
+ ]
+ for add_entities_callback in self._platform_callbacks:
+ add_entities_callback(new_players)
+
+ if update_result.updated_player_ids:
+ self._async_update_player_ids(update_result.updated_player_ids)
+
async def _async_on_auth_failure(self) -> None:
"""Handle when the user credentials are no longer valid."""
assert self.config_entry is not None
@@ -136,44 +169,27 @@ class HeosCoordinator(DataUpdateCoordinator[None]):
async def _async_on_reconnected(self) -> None:
"""Handle when reconnected so resources are updated and entities marked available."""
- await self._async_update_players()
+ assert self.config_entry is not None
+ if self.host != self.config_entry.data[CONF_HOST]:
+ self.hass.config_entries.async_update_entry(
+ self.config_entry, data={CONF_HOST: self.host}
+ )
+ _LOGGER.warning("Successfully failed over to HEOS host %s", self.host)
+ else:
+ _LOGGER.warning("Successfully reconnected to HEOS host %s", self.host)
await self._async_update_sources()
- _LOGGER.warning("Successfully reconnected to HEOS host %s", self.host)
self.async_update_listeners()
async def _async_on_controller_event(
- self, event: str, data: PlayerUpdateResult | None
+ self, event: str, data: PlayerUpdateResult | None = None
) -> None:
"""Handle a controller event, such as players or groups changed."""
if event == const.EVENT_PLAYERS_CHANGED:
assert data is not None
- if data.updated_player_ids:
- self._async_update_player_ids(data.updated_player_ids)
- elif (
- event in (const.EVENT_SOURCES_CHANGED, const.EVENT_USER_CHANGED)
- and not self._update_sources_pending
- ):
- # Update the sources after a brief delay as we may have received multiple qualifying
- # events at once and devices cannot handle immediately attempting to refresh sources.
- self._update_sources_pending = True
-
- async def update_sources_job(_: datetime | None = None) -> None:
- await self._async_update_sources()
- self._update_sources_pending = False
- self.async_update_listeners()
-
- assert self.config_entry is not None
- self.config_entry.async_on_unload(
- async_call_later(
- self.hass,
- timedelta(seconds=1),
- HassJob(
- update_sources_job,
- "heos_update_sources",
- cancel_on_shutdown=True,
- ),
- )
- )
+ self._async_handle_player_update_result(data)
+ elif event in (const.EVENT_SOURCES_CHANGED, const.EVENT_USER_CHANGED):
+ # Debounce because we may have received multiple qualifying events in rapid succession.
+ await self._update_sources_debouncer.async_call()
self.async_update_listeners()
def _async_update_player_ids(self, updated_player_ids: dict[int, int]) -> None:
@@ -235,17 +251,6 @@ class HeosCoordinator(DataUpdateCoordinator[None]):
else:
self._source_list.extend([source.name for source in self._inputs])
- async def _async_update_players(self) -> None:
- """Update players after reconnection."""
- try:
- player_updates = await self.heos.load_players()
- except HeosError as error:
- _LOGGER.error("Unable to refresh players: %s", error)
- return
- # After reconnecting, player_id may have changed
- if player_updates.updated_player_ids:
- self._async_update_player_ids(player_updates.updated_player_ids)
-
@callback
def async_get_source_list(self) -> list[str]:
"""Return the list of sources for players."""
diff --git a/homeassistant/components/heos/icons.json b/homeassistant/components/heos/icons.json
index 23c2c8faeaf..d7a998b6aec 100644
--- a/homeassistant/components/heos/icons.json
+++ b/homeassistant/components/heos/icons.json
@@ -1,5 +1,14 @@
{
"services": {
+ "group_volume_set": {
+ "service": "mdi:volume-medium"
+ },
+ "group_volume_down": {
+ "service": "mdi:volume-low"
+ },
+ "group_volume_up": {
+ "service": "mdi:volume-high"
+ },
"sign_in": {
"service": "mdi:login"
},
diff --git a/homeassistant/components/heos/manifest.json b/homeassistant/components/heos/manifest.json
index 22dbbf4da28..cbac9f20574 100644
--- a/homeassistant/components/heos/manifest.json
+++ b/homeassistant/components/heos/manifest.json
@@ -7,9 +7,8 @@
"integration_type": "hub",
"iot_class": "local_push",
"loggers": ["pyheos"],
- "quality_scale": "silver",
- "requirements": ["pyheos==1.0.1"],
- "single_config_entry": true,
+ "quality_scale": "platinum",
+ "requirements": ["pyheos==1.0.4"],
"ssdp": [
{
"st": "urn:schemas-denon-com:device:ACT-Denon:1"
diff --git a/homeassistant/components/heos/media_player.py b/homeassistant/components/heos/media_player.py
index b53cb94d8e7..311190ccb74 100644
--- a/homeassistant/components/heos/media_player.py
+++ b/homeassistant/components/heos/media_player.py
@@ -2,26 +2,36 @@
from __future__ import annotations
-from collections.abc import Awaitable, Callable, Coroutine
+from collections.abc import Awaitable, Callable, Coroutine, Sequence
+from contextlib import suppress
from datetime import datetime
from functools import reduce, wraps
+import logging
from operator import ior
-from typing import Any
+from typing import Any, Final
from pyheos import (
AddCriteriaType,
ControlType,
HeosError,
HeosPlayer,
+ MediaItem,
+ MediaMusicSource,
+ MediaType as HeosMediaType,
PlayState,
RepeatType,
const as heos_const,
)
+from pyheos.util import mediauri as heos_source
+import voluptuous as vol
from homeassistant.components import media_source
from homeassistant.components.media_player import (
ATTR_MEDIA_ENQUEUE,
+ ATTR_MEDIA_VOLUME_LEVEL,
+ BrowseError,
BrowseMedia,
+ MediaClass,
MediaPlayerEnqueue,
MediaPlayerEntity,
MediaPlayerEntityFeature,
@@ -30,20 +40,32 @@ from homeassistant.components.media_player import (
RepeatMode,
async_process_play_media_url,
)
+from homeassistant.components.media_source import BrowseMediaSource
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
-from homeassistant.helpers import entity_registry as er
+from homeassistant.helpers import (
+ config_validation as cv,
+ entity_platform,
+ entity_registry as er,
+)
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.dt import utcnow
-from .const import DOMAIN as HEOS_DOMAIN
+from .const import (
+ DOMAIN as HEOS_DOMAIN,
+ SERVICE_GROUP_VOLUME_DOWN,
+ SERVICE_GROUP_VOLUME_SET,
+ SERVICE_GROUP_VOLUME_UP,
+)
from .coordinator import HeosConfigEntry, HeosCoordinator
PARALLEL_UPDATES = 0
+BROWSE_ROOT: Final = "heos://media"
+
BASE_SUPPORTED_FEATURES = (
MediaPlayerEntityFeature.VOLUME_MUTE
| MediaPlayerEntityFeature.VOLUME_SET
@@ -86,16 +108,51 @@ HEOS_HA_REPEAT_TYPE_MAP = {
}
HA_HEOS_REPEAT_TYPE_MAP = {v: k for k, v in HEOS_HA_REPEAT_TYPE_MAP.items()}
+HEOS_MEDIA_TYPE_TO_MEDIA_CLASS = {
+ HeosMediaType.ALBUM: MediaClass.ALBUM,
+ HeosMediaType.ARTIST: MediaClass.ARTIST,
+ HeosMediaType.CONTAINER: MediaClass.DIRECTORY,
+ HeosMediaType.GENRE: MediaClass.GENRE,
+ HeosMediaType.HEOS_SERVER: MediaClass.DIRECTORY,
+ HeosMediaType.HEOS_SERVICE: MediaClass.DIRECTORY,
+ HeosMediaType.MUSIC_SERVICE: MediaClass.DIRECTORY,
+ HeosMediaType.PLAYLIST: MediaClass.PLAYLIST,
+ HeosMediaType.SONG: MediaClass.TRACK,
+ HeosMediaType.STATION: MediaClass.TRACK,
+}
+
+_LOGGER = logging.getLogger(__name__)
+
async def async_setup_entry(
- hass: HomeAssistant, entry: HeosConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: HeosConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add media players for a config entry."""
- devices = [
- HeosMediaPlayer(entry.runtime_data, player)
- for player in entry.runtime_data.heos.players.values()
- ]
- async_add_entities(devices)
+ # Register custom entity services
+ platform = entity_platform.async_get_current_platform()
+ platform.async_register_entity_service(
+ SERVICE_GROUP_VOLUME_SET,
+ {vol.Required(ATTR_MEDIA_VOLUME_LEVEL): cv.small_float},
+ "async_set_group_volume_level",
+ )
+ platform.async_register_entity_service(
+ SERVICE_GROUP_VOLUME_DOWN, None, "async_group_volume_down"
+ )
+ platform.async_register_entity_service(
+ SERVICE_GROUP_VOLUME_UP, None, "async_group_volume_up"
+ )
+
+ def add_entities_callback(players: Sequence[HeosPlayer]) -> None:
+ """Add entities for each player."""
+ async_add_entities(
+ [HeosMediaPlayer(entry.runtime_data, player) for player in players]
+ )
+
+ coordinator = entry.runtime_data
+ coordinator.async_add_platform_callback(add_entities_callback)
+ add_entities_callback(list(coordinator.heos.players.values()))
type _FuncType[**_P] = Callable[_P, Awaitable[Any]]
@@ -251,6 +308,16 @@ class HeosMediaPlayer(CoordinatorEntity[HeosCoordinator], MediaPlayerEntity):
self, media_type: MediaType | str, media_id: str, **kwargs: Any
) -> None:
"""Play a piece of media."""
+ if heos_source.is_media_uri(media_id):
+ media, data = heos_source.from_media_uri(media_id)
+ if not isinstance(media, MediaItem):
+ raise ValueError(f"Invalid media id '{media_id}'")
+ await self._player.play_media(
+ media,
+ HA_HEOS_ENQUEUE_MAP[kwargs.get(ATTR_MEDIA_ENQUEUE)],
+ )
+ return
+
if media_source.is_media_source_id(media_id):
media_type = MediaType.URL
play_item = await media_source.async_resolve_media(
@@ -339,6 +406,41 @@ class HeosMediaPlayer(CoordinatorEntity[HeosCoordinator], MediaPlayerEntity):
"""Set volume level, range 0..1."""
await self._player.set_volume(int(volume * 100))
+ @catch_action_error("set group volume level")
+ async def async_set_group_volume_level(self, volume_level: float) -> None:
+ """Set group volume level."""
+ if self._player.group_id is None:
+ raise ServiceValidationError(
+ translation_domain=HEOS_DOMAIN,
+ translation_key="entity_not_grouped",
+ translation_placeholders={"entity_id": self.entity_id},
+ )
+ await self.coordinator.heos.set_group_volume(
+ self._player.group_id, int(volume_level * 100)
+ )
+
+ @catch_action_error("group volume down")
+ async def async_group_volume_down(self) -> None:
+ """Turn group volume down for media player."""
+ if self._player.group_id is None:
+ raise ServiceValidationError(
+ translation_domain=HEOS_DOMAIN,
+ translation_key="entity_not_grouped",
+ translation_placeholders={"entity_id": self.entity_id},
+ )
+ await self.coordinator.heos.group_volume_down(self._player.group_id)
+
+ @catch_action_error("group volume up")
+ async def async_group_volume_up(self) -> None:
+ """Turn group volume up for media player."""
+ if self._player.group_id is None:
+ raise ServiceValidationError(
+ translation_domain=HEOS_DOMAIN,
+ translation_key="entity_not_grouped",
+ translation_placeholders={"entity_id": self.entity_id},
+ )
+ await self.coordinator.heos.group_volume_up(self._player.group_id)
+
@catch_action_error("join players")
async def async_join_players(self, group_members: list[str]) -> None:
"""Join `group_members` as a player group with the current player."""
@@ -468,14 +570,103 @@ class HeosMediaPlayer(CoordinatorEntity[HeosCoordinator], MediaPlayerEntity):
"""Volume level of the media player (0..1)."""
return self._player.volume / 100
+ async def _async_browse_media_root(self) -> BrowseMedia:
+ """Return media browsing root."""
+ if not self.coordinator.heos.music_sources:
+ try:
+ await self.coordinator.heos.get_music_sources()
+ except HeosError as error:
+ _LOGGER.debug("Unable to load music sources: %s", error)
+ children: list[BrowseMedia] = [
+ _media_to_browse_media(source)
+ for source in self.coordinator.heos.music_sources.values()
+ if source.available or source.source_id == heos_const.MUSIC_SOURCE_TUNEIN
+ ]
+ root = BrowseMedia(
+ title="Music Sources",
+ media_class=MediaClass.DIRECTORY,
+ children_media_class=MediaClass.DIRECTORY,
+ media_content_type="",
+ media_content_id=BROWSE_ROOT,
+ can_expand=True,
+ can_play=False,
+ children=children,
+ )
+ # Append media source items
+ with suppress(BrowseError):
+ browse = await self._async_browse_media_source()
+ # If domain is None, it's an overview of available sources
+ if browse.domain is None and browse.children:
+ children.extend(browse.children)
+ else:
+ children.append(browse)
+ return root
+
+ async def _async_browse_heos_media(self, media_content_id: str) -> BrowseMedia:
+ """Browse a HEOS media item."""
+ media, data = heos_source.from_media_uri(media_content_id)
+ browse_media = _media_to_browse_media(media)
+ try:
+ browse_result = await self.coordinator.heos.browse_media(media)
+ except HeosError as error:
+ _LOGGER.debug("Unable to browse media %s: %s", media, error)
+ else:
+ browse_media.children = [
+ _media_to_browse_media(item)
+ for item in browse_result.items
+ if item.browsable or item.playable
+ ]
+ return browse_media
+
+ async def _async_browse_media_source(
+ self, media_content_id: str | None = None
+ ) -> BrowseMediaSource:
+ """Browse a media source item."""
+ return await media_source.async_browse_media(
+ self.hass,
+ media_content_id,
+ content_filter=lambda item: item.media_content_type.startswith("audio/"),
+ )
+
async def async_browse_media(
self,
media_content_type: MediaType | str | None = None,
media_content_id: str | None = None,
) -> BrowseMedia:
"""Implement the websocket media browsing helper."""
- return await media_source.async_browse_media(
- self.hass,
- media_content_id,
- content_filter=lambda item: item.media_content_type.startswith("audio/"),
+ if media_content_id in (None, BROWSE_ROOT):
+ return await self._async_browse_media_root()
+ assert media_content_id is not None
+ if heos_source.is_media_uri(media_content_id):
+ return await self._async_browse_heos_media(media_content_id)
+ if media_source.is_media_source_id(media_content_id):
+ return await self._async_browse_media_source(media_content_id)
+ raise ServiceValidationError(
+ translation_domain=HEOS_DOMAIN,
+ translation_key="unsupported_media_content_id",
+ translation_placeholders={"media_content_id": media_content_id},
)
+
+
+def _media_to_browse_media(media: MediaItem | MediaMusicSource) -> BrowseMedia:
+ """Convert a HEOS media item to a browse media item."""
+ can_expand = False
+ can_play = False
+
+ if isinstance(media, MediaMusicSource):
+ can_expand = (
+ media.source_id == heos_const.MUSIC_SOURCE_TUNEIN or media.available
+ )
+ else:
+ can_expand = media.browsable
+ can_play = media.playable
+
+ return BrowseMedia(
+ can_expand=can_expand,
+ can_play=can_play,
+ media_content_id=heos_source.to_media_uri(media),
+ media_content_type="",
+ media_class=HEOS_MEDIA_TYPE_TO_MEDIA_CLASS[media.type],
+ title=media.name,
+ thumbnail=media.image_url,
+ )
diff --git a/homeassistant/components/heos/quality_scale.yaml b/homeassistant/components/heos/quality_scale.yaml
index 67022ec492c..5f5062b6a82 100644
--- a/homeassistant/components/heos/quality_scale.yaml
+++ b/homeassistant/components/heos/quality_scale.yaml
@@ -38,9 +38,7 @@ rules:
# Gold
devices: done
diagnostics: done
- discovery-update-info:
- status: todo
- comment: Explore if this is possible.
+ discovery-update-info: done
discovery: done
docs-data-update: done
docs-examples: done
@@ -49,7 +47,7 @@ rules:
docs-supported-functions: done
docs-troubleshooting: done
docs-use-cases: done
- dynamic-devices: todo
+ dynamic-devices: done
entity-category: done
entity-device-class: done
entity-disabled-by-default: done
@@ -57,8 +55,8 @@ rules:
exception-translations: done
icon-translations: done
reconfiguration-flow: done
- repair-issues: todo
- stale-devices: todo
+ repair-issues: done
+ stale-devices: done
# Platinum
async-dependency: done
inject-websession:
diff --git a/homeassistant/components/heos/services.yaml b/homeassistant/components/heos/services.yaml
index 8dc222d65ba..8f3a43421f6 100644
--- a/homeassistant/components/heos/services.yaml
+++ b/homeassistant/components/heos/services.yaml
@@ -1,3 +1,29 @@
+group_volume_set:
+ target:
+ entity:
+ integration: heos
+ domain: media_player
+ fields:
+ volume_level:
+ required: true
+ selector:
+ number:
+ min: 0
+ max: 1
+ step: 0.01
+
+group_volume_down:
+ target:
+ entity:
+ integration: heos
+ domain: media_player
+
+group_volume_up:
+ target:
+ entity:
+ integration: heos
+ domain: media_player
+
sign_in:
fields:
username:
diff --git a/homeassistant/components/heos/strings.json b/homeassistant/components/heos/strings.json
index 53e20a032b5..593c437accc 100644
--- a/homeassistant/components/heos/strings.json
+++ b/homeassistant/components/heos/strings.json
@@ -8,9 +8,13 @@
"host": "[%key:common::config_flow::data::host%]"
},
"data_description": {
- "host": "Host name or IP address of a HEOS-capable product (preferrably one connected via wire to the network)."
+ "host": "Host name or IP address of a HEOS-capable product (preferably one connected via wire to the network)."
}
},
+ "confirm_discovery": {
+ "title": "Discovered HEOS System",
+ "description": "Do you want to add your HEOS devices to Home Assistant?"
+ },
"reconfigure": {
"title": "Reconfigure HEOS",
"description": "Change the host name or IP address of the HEOS-capable product used to access your HEOS System.",
@@ -43,6 +47,7 @@
},
"abort": {
"already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]",
+ "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
"reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",
"reconfigure_successful": "[%key:common::config_flow::abort::reconfigure_successful%]",
"single_instance_allowed": "[%key:common::config_flow::abort::single_instance_allowed%]"
@@ -71,6 +76,24 @@
}
},
"services": {
+ "group_volume_set": {
+ "name": "Set group volume",
+ "description": "Sets the group's volume while preserving member volume ratios.",
+ "fields": {
+ "volume_level": {
+ "name": "Level",
+ "description": "The volume. 0 is inaudible, 1 is the maximum volume."
+ }
+ }
+ },
+ "group_volume_down": {
+ "name": "Turn down group volume",
+ "description": "Turns down the group volume."
+ },
+ "group_volume_up": {
+ "name": "Turn up group volume",
+ "description": "Turns up the group volume."
+ },
"sign_in": {
"name": "Sign in",
"description": "Signs in to a HEOS account.",
@@ -94,6 +117,9 @@
"action_error": {
"message": "Unable to {action}: {error}"
},
+ "entity_not_grouped": {
+ "message": "Entity {entity_id} is not joined to a group"
+ },
"entity_not_found": {
"message": "Entity {entity_id} was not found"
},
@@ -120,6 +146,9 @@
},
"unknown_source": {
"message": "Unknown source: {source}"
+ },
+ "unsupported_media_content_id": {
+ "message": "Unsupported media_content_id: {media_content_id}"
}
},
"issues": {
diff --git a/homeassistant/components/here_travel_time/__init__.py b/homeassistant/components/here_travel_time/__init__.py
index 1b99ba64827..132b12de4ce 100644
--- a/homeassistant/components/here_travel_time/__init__.py
+++ b/homeassistant/components/here_travel_time/__init__.py
@@ -56,7 +56,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
else:
cls = HERERoutingDataUpdateCoordinator
- data_coordinator = cls(hass, api_key, here_travel_time_config)
+ data_coordinator = cls(hass, config_entry, api_key, here_travel_time_config)
hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = data_coordinator
async def _async_update_at_start(_: HomeAssistant) -> None:
diff --git a/homeassistant/components/here_travel_time/coordinator.py b/homeassistant/components/here_travel_time/coordinator.py
index 65e1305e44e..a3345e78e4e 100644
--- a/homeassistant/components/here_travel_time/coordinator.py
+++ b/homeassistant/components/here_travel_time/coordinator.py
@@ -25,6 +25,7 @@ from here_transit import (
)
import voluptuous as vol
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfLength
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
@@ -44,9 +45,12 @@ _LOGGER = logging.getLogger(__name__)
class HERERoutingDataUpdateCoordinator(DataUpdateCoordinator[HERETravelTimeData]):
"""here_routing DataUpdateCoordinator."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
api_key: str,
config: HERETravelTimeConfig,
) -> None:
@@ -54,6 +58,7 @@ class HERERoutingDataUpdateCoordinator(DataUpdateCoordinator[HERETravelTimeData]
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL),
)
@@ -168,9 +173,12 @@ class HERETransitDataUpdateCoordinator(
):
"""HERETravelTime DataUpdateCoordinator."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
api_key: str,
config: HERETravelTimeConfig,
) -> None:
@@ -178,6 +186,7 @@ class HERETransitDataUpdateCoordinator(
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL),
)
diff --git a/homeassistant/components/here_travel_time/sensor.py b/homeassistant/components/here_travel_time/sensor.py
index 4d7566ef2e2..0f0cbb7d3cb 100644
--- a/homeassistant/components/here_travel_time/sensor.py
+++ b/homeassistant/components/here_travel_time/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -78,7 +78,7 @@ def sensor_descriptions(travel_mode: str) -> tuple[SensorEntityDescription, ...]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add HERE travel time entities from a config_entry."""
diff --git a/homeassistant/components/hisense_aehw4a1/climate.py b/homeassistant/components/hisense_aehw4a1/climate.py
index 1dc1eaabcaa..cd9f3666e08 100644
--- a/homeassistant/components/hisense_aehw4a1/climate.py
+++ b/homeassistant/components/hisense_aehw4a1/climate.py
@@ -28,7 +28,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import CONF_IP_ADDRESS, DOMAIN
@@ -121,7 +121,7 @@ def _build_entity(device):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the AEH-W4A1 climate platform."""
# Priority 1: manual config
diff --git a/homeassistant/components/history_stats/sensor.py b/homeassistant/components/history_stats/sensor.py
index b25daf56598..6935b13bc3d 100644
--- a/homeassistant/components/history_stats/sensor.py
+++ b/homeassistant/components/history_stats/sensor.py
@@ -27,7 +27,10 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.device import async_device_info_to_link_from_entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.reload import async_setup_reload_service
from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -116,7 +119,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: HistoryStatsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the History stats sensor entry."""
diff --git a/homeassistant/components/history_stats/strings.json b/homeassistant/components/history_stats/strings.json
index aff2ac50bef..e10a72f6742 100644
--- a/homeassistant/components/history_stats/strings.json
+++ b/homeassistant/components/history_stats/strings.json
@@ -24,7 +24,7 @@
}
},
"options": {
- "description": "Read the documention for further details on how to configure the history stats sensor using these options.",
+ "description": "Read the documentation for further details on how to configure the history stats sensor using these options.",
"data": {
"start": "Start",
"end": "End",
diff --git a/homeassistant/components/hive/alarm_control_panel.py b/homeassistant/components/hive/alarm_control_panel.py
index 2b196ce820b..c2fe47642a0 100644
--- a/homeassistant/components/hive/alarm_control_panel.py
+++ b/homeassistant/components/hive/alarm_control_panel.py
@@ -11,7 +11,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HiveEntity
@@ -27,7 +27,9 @@ HIVETOHA = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hive thermostat based on a config entry."""
diff --git a/homeassistant/components/hive/binary_sensor.py b/homeassistant/components/hive/binary_sensor.py
index d2938896f92..2076d592a7c 100644
--- a/homeassistant/components/hive/binary_sensor.py
+++ b/homeassistant/components/hive/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HiveEntity
@@ -68,7 +68,9 @@ SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hive thermostat based on a config entry."""
diff --git a/homeassistant/components/hive/climate.py b/homeassistant/components/hive/climate.py
index c76379cf940..bd7553faa1a 100644
--- a/homeassistant/components/hive/climate.py
+++ b/homeassistant/components/hive/climate.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import refresh_system
from .const import (
@@ -58,7 +58,9 @@ _LOGGER = logging.getLogger()
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hive thermostat based on a config entry."""
diff --git a/homeassistant/components/hive/light.py b/homeassistant/components/hive/light.py
index e941087c6fb..80a81583429 100644
--- a/homeassistant/components/hive/light.py
+++ b/homeassistant/components/hive/light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from . import refresh_system
@@ -29,7 +29,9 @@ SCAN_INTERVAL = timedelta(seconds=15)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hive thermostat based on a config entry."""
diff --git a/homeassistant/components/hive/manifest.json b/homeassistant/components/hive/manifest.json
index f68478516ab..712ccf09cae 100644
--- a/homeassistant/components/hive/manifest.json
+++ b/homeassistant/components/hive/manifest.json
@@ -9,5 +9,5 @@
},
"iot_class": "cloud_polling",
"loggers": ["apyhiveapi"],
- "requirements": ["pyhive-integration==1.0.1"]
+ "requirements": ["pyhive-integration==1.0.2"]
}
diff --git a/homeassistant/components/hive/sensor.py b/homeassistant/components/hive/sensor.py
index 00a2116e268..0609e43c4a9 100644
--- a/homeassistant/components/hive/sensor.py
+++ b/homeassistant/components/hive/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN
@@ -89,7 +89,9 @@ SENSOR_TYPES: tuple[HiveSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hive thermostat based on a config entry."""
hive = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/hive/strings.json b/homeassistant/components/hive/strings.json
index 219776ad7e6..6323a2eecbf 100644
--- a/homeassistant/components/hive/strings.json
+++ b/homeassistant/components/hive/strings.json
@@ -2,27 +2,27 @@
"config": {
"step": {
"user": {
- "title": "Hive Login",
+ "title": "Hive login",
"description": "Enter your Hive login information.",
"data": {
"username": "[%key:common::config_flow::data::username%]",
"password": "[%key:common::config_flow::data::password%]",
- "scan_interval": "Scan Interval (seconds)"
+ "scan_interval": "Scan interval (seconds)"
}
},
"2fa": {
- "title": "Hive Two-factor Authentication.",
- "description": "Enter your Hive authentication code. \n \n Please enter code 0000 to request another code.",
+ "title": "Hive two-factor authentication",
+ "description": "Enter your Hive authentication code.\n\nPlease enter code 0000 to request another code.",
"data": {
"2fa": "Two-factor code"
}
},
"configuration": {
+ "title": "Hive configuration",
+ "description": "Enter your Hive configuration.",
"data": {
- "device_name": "Device Name"
- },
- "description": "Enter your Hive configuration",
- "title": "Hive Configuration."
+ "device_name": "Device name"
+ }
},
"reauth": {
"title": "[%key:component::hive::config::step::user::title%]",
@@ -37,7 +37,7 @@
"invalid_username": "Failed to sign into Hive. Your email address is not recognised.",
"invalid_password": "Failed to sign into Hive. Incorrect password, please try again.",
"invalid_code": "Failed to sign into Hive. Your two-factor authentication code was incorrect.",
- "no_internet_available": "An internet connection is required to connect to Hive.",
+ "no_internet_available": "An Internet connection is required to connect to Hive.",
"unknown": "[%key:common::config_flow::error::unknown%]"
},
"abort": {
diff --git a/homeassistant/components/hive/switch.py b/homeassistant/components/hive/switch.py
index 1421616db57..d4fefea5a56 100644
--- a/homeassistant/components/hive/switch.py
+++ b/homeassistant/components/hive/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import refresh_system
from .const import ATTR_MODE, DOMAIN
@@ -33,7 +33,9 @@ SWITCH_TYPES: tuple[SwitchEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hive thermostat based on a config entry."""
diff --git a/homeassistant/components/hive/water_heater.py b/homeassistant/components/hive/water_heater.py
index b038739d2ad..5f0a3d0f3fa 100644
--- a/homeassistant/components/hive/water_heater.py
+++ b/homeassistant/components/hive/water_heater.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_OFF, STATE_ON, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import refresh_system
from .const import (
@@ -45,7 +45,9 @@ SUPPORT_WATER_HEATER = [STATE_ECO, STATE_ON, STATE_OFF]
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hive thermostat based on a config entry."""
diff --git a/homeassistant/components/hko/__init__.py b/homeassistant/components/hko/__init__.py
index 5b06644580e..b7e21f731d8 100644
--- a/homeassistant/components/hko/__init__.py
+++ b/homeassistant/components/hko/__init__.py
@@ -25,7 +25,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
)[KEY_DISTRICT]
websession = async_get_clientsession(hass)
- coordinator = HKOUpdateCoordinator(hass, websession, district, location)
+ coordinator = HKOUpdateCoordinator(hass, entry, websession, district, location)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/hko/coordinator.py b/homeassistant/components/hko/coordinator.py
index 566ba5dcf5e..aede960e702 100644
--- a/homeassistant/components/hko/coordinator.py
+++ b/homeassistant/components/hko/coordinator.py
@@ -11,7 +11,6 @@ from hko import HKO, HKOError
from homeassistant.components.weather import (
ATTR_CONDITION_CLOUDY,
ATTR_CONDITION_FOG,
- ATTR_CONDITION_HAIL,
ATTR_CONDITION_LIGHTNING_RAINY,
ATTR_CONDITION_PARTLYCLOUDY,
ATTR_CONDITION_POURING,
@@ -26,6 +25,7 @@ from homeassistant.components.weather import (
ATTR_FORECAST_TEMP_LOW,
ATTR_FORECAST_TIME,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -69,8 +69,15 @@ _LOGGER = logging.getLogger(__name__)
class HKOUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""HKO Update Coordinator."""
+ config_entry: ConfigEntry
+
def __init__(
- self, hass: HomeAssistant, session: ClientSession, district: str, location: str
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ session: ClientSession,
+ district: str,
+ location: str,
) -> None:
"""Update data via library."""
self.location = location
@@ -80,6 +87,7 @@ class HKOUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=15),
)
@@ -136,7 +144,7 @@ class HKOUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Return the condition corresponding to the weather info."""
info = info.lower()
if WEATHER_INFO_RAIN in info:
- return ATTR_CONDITION_HAIL
+ return ATTR_CONDITION_RAINY
if WEATHER_INFO_SNOW in info and WEATHER_INFO_RAIN in info:
return ATTR_CONDITION_SNOWY_RAINY
if WEATHER_INFO_SNOW in info:
diff --git a/homeassistant/components/hko/weather.py b/homeassistant/components/hko/weather.py
index 6d3d12d8ab4..e746d4304d3 100644
--- a/homeassistant/components/hko/weather.py
+++ b/homeassistant/components/hko/weather.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -28,7 +28,7 @@ from .coordinator import HKOUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a HKO weather entity from a config_entry."""
assert config_entry.unique_id is not None
diff --git a/homeassistant/components/hlk_sw16/entity.py b/homeassistant/components/hlk_sw16/entity.py
index fdef5f6764b..91510760968 100644
--- a/homeassistant/components/hlk_sw16/entity.py
+++ b/homeassistant/components/hlk_sw16/entity.py
@@ -35,7 +35,7 @@ class SW16Entity(Entity):
self.async_write_ha_state()
@property
- def available(self):
+ def available(self) -> bool:
"""Return True if entity is available."""
return bool(self._client.is_connected)
@@ -44,7 +44,7 @@ class SW16Entity(Entity):
"""Update availability state."""
self.async_write_ha_state()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register update callback."""
self._client.register_status_callback(
self.handle_event_callback, self._device_port
diff --git a/homeassistant/components/hlk_sw16/switch.py b/homeassistant/components/hlk_sw16/switch.py
index 3911dd6eab9..c6e6f7f5201 100644
--- a/homeassistant/components/hlk_sw16/switch.py
+++ b/homeassistant/components/hlk_sw16/switch.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DATA_DEVICE_REGISTER
from .const import DOMAIN
@@ -26,7 +26,9 @@ def devices_from_entities(hass, entry):
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HLK-SW16 platform."""
async_add_entities(devices_from_entities(hass, entry))
diff --git a/homeassistant/components/holiday/calendar.py b/homeassistant/components/holiday/calendar.py
index 6dccd972164..1c01319129b 100644
--- a/homeassistant/components/holiday/calendar.py
+++ b/homeassistant/components/holiday/calendar.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_COUNTRY
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_point_in_utc_time
from homeassistant.util import dt as dt_util
@@ -69,7 +69,7 @@ def _get_obj_holidays_and_language(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Holiday Calendar config entry."""
country: str = config_entry.data[CONF_COUNTRY]
diff --git a/homeassistant/components/holiday/manifest.json b/homeassistant/components/holiday/manifest.json
index 6952d48ef32..ec47b222370 100644
--- a/homeassistant/components/holiday/manifest.json
+++ b/homeassistant/components/holiday/manifest.json
@@ -5,5 +5,5 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/holiday",
"iot_class": "local_polling",
- "requirements": ["holidays==0.66", "babel==2.15.0"]
+ "requirements": ["holidays==0.68", "babel==2.15.0"]
}
diff --git a/homeassistant/components/home_connect/__init__.py b/homeassistant/components/home_connect/__init__.py
index becc78cef90..fe01a3e9564 100644
--- a/homeassistant/components/home_connect/__init__.py
+++ b/homeassistant/components/home_connect/__init__.py
@@ -3,100 +3,31 @@
from __future__ import annotations
import logging
-from typing import Any, cast
+from typing import Any
from aiohomeconnect.client import Client as HomeConnectClient
-from aiohomeconnect.model import CommandKey, Option, OptionKey, ProgramKey, SettingKey
-from aiohomeconnect.model.error import HomeConnectError
-import voluptuous as vol
+import aiohttp
-from homeassistant.const import ATTR_DEVICE_ID, Platform
-from homeassistant.core import HomeAssistant, ServiceCall, callback
-from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
-from homeassistant.helpers import (
- config_entry_oauth2_flow,
- config_validation as cv,
- device_registry as dr,
-)
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
+from homeassistant.helpers import config_entry_oauth2_flow, config_validation as cv
from homeassistant.helpers.entity_registry import RegistryEntry, async_migrate_entries
+from homeassistant.helpers.issue_registry import async_delete_issue
from homeassistant.helpers.typing import ConfigType
from .api import AsyncConfigEntryAuth
-from .const import (
- ATTR_KEY,
- ATTR_PROGRAM,
- ATTR_UNIT,
- ATTR_VALUE,
- DOMAIN,
- OLD_NEW_UNIQUE_ID_SUFFIX_MAP,
- SERVICE_OPTION_ACTIVE,
- SERVICE_OPTION_SELECTED,
- SERVICE_PAUSE_PROGRAM,
- SERVICE_RESUME_PROGRAM,
- SERVICE_SELECT_PROGRAM,
- SERVICE_SETTING,
- SERVICE_START_PROGRAM,
- SVE_TRANSLATION_PLACEHOLDER_KEY,
- SVE_TRANSLATION_PLACEHOLDER_PROGRAM,
- SVE_TRANSLATION_PLACEHOLDER_VALUE,
-)
+from .const import DOMAIN, OLD_NEW_UNIQUE_ID_SUFFIX_MAP
from .coordinator import HomeConnectConfigEntry, HomeConnectCoordinator
-from .utils import get_dict_from_home_connect_error
+from .services import register_actions
_LOGGER = logging.getLogger(__name__)
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
-SERVICE_SETTING_SCHEMA = vol.Schema(
- {
- vol.Required(ATTR_DEVICE_ID): str,
- vol.Required(ATTR_KEY): vol.All(
- vol.Coerce(SettingKey),
- vol.NotIn([SettingKey.UNKNOWN]),
- ),
- vol.Required(ATTR_VALUE): vol.Any(str, int, bool),
- }
-)
-
-SERVICE_OPTION_SCHEMA = vol.Schema(
- {
- vol.Required(ATTR_DEVICE_ID): str,
- vol.Required(ATTR_KEY): vol.All(
- vol.Coerce(OptionKey),
- vol.NotIn([OptionKey.UNKNOWN]),
- ),
- vol.Required(ATTR_VALUE): vol.Any(str, int, bool),
- vol.Optional(ATTR_UNIT): str,
- }
-)
-
-SERVICE_PROGRAM_SCHEMA = vol.Any(
- {
- vol.Required(ATTR_DEVICE_ID): str,
- vol.Required(ATTR_PROGRAM): vol.All(
- vol.Coerce(ProgramKey),
- vol.NotIn([ProgramKey.UNKNOWN]),
- ),
- vol.Required(ATTR_KEY): vol.All(
- vol.Coerce(OptionKey),
- vol.NotIn([OptionKey.UNKNOWN]),
- ),
- vol.Required(ATTR_VALUE): vol.Any(int, str),
- vol.Optional(ATTR_UNIT): str,
- },
- {
- vol.Required(ATTR_DEVICE_ID): str,
- vol.Required(ATTR_PROGRAM): vol.All(
- vol.Coerce(ProgramKey),
- vol.NotIn([ProgramKey.UNKNOWN]),
- ),
- },
-)
-
-SERVICE_COMMAND_SCHEMA = vol.Schema({vol.Required(ATTR_DEVICE_ID): str})
-
PLATFORMS = [
Platform.BINARY_SENSOR,
+ Platform.BUTTON,
Platform.LIGHT,
Platform.NUMBER,
Platform.SELECT,
@@ -106,216 +37,9 @@ PLATFORMS = [
]
-async def _get_client_and_ha_id(
- hass: HomeAssistant, device_id: str
-) -> tuple[HomeConnectClient, str]:
- device_registry = dr.async_get(hass)
- device_entry = device_registry.async_get(device_id)
- if device_entry is None:
- raise ServiceValidationError("Device entry not found for device id")
- entry: HomeConnectConfigEntry | None = None
- for entry_id in device_entry.config_entries:
- _entry = hass.config_entries.async_get_entry(entry_id)
- assert _entry
- if _entry.domain == DOMAIN:
- entry = cast(HomeConnectConfigEntry, _entry)
- break
- if entry is None:
- raise ServiceValidationError(
- "Home Connect config entry not found for that device id"
- )
-
- ha_id = next(
- (
- identifier[1]
- for identifier in device_entry.identifiers
- if identifier[0] == DOMAIN
- ),
- None,
- )
- if ha_id is None:
- raise ServiceValidationError(
- translation_domain=DOMAIN,
- translation_key="appliance_not_found",
- translation_placeholders={
- "device_id": device_id,
- },
- )
- return entry.runtime_data.client, ha_id
-
-
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up Home Connect component."""
-
- async def _async_service_program(call: ServiceCall, start: bool):
- """Execute calls to services taking a program."""
- program = call.data[ATTR_PROGRAM]
- client, ha_id = await _get_client_and_ha_id(hass, call.data[ATTR_DEVICE_ID])
-
- option_key = call.data.get(ATTR_KEY)
- options = (
- [
- Option(
- option_key,
- call.data[ATTR_VALUE],
- unit=call.data.get(ATTR_UNIT),
- )
- ]
- if option_key is not None
- else None
- )
-
- try:
- if start:
- await client.start_program(ha_id, program_key=program, options=options)
- else:
- await client.set_selected_program(
- ha_id, program_key=program, options=options
- )
- except HomeConnectError as err:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="start_program" if start else "select_program",
- translation_placeholders={
- **get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_PROGRAM: program,
- },
- ) from err
-
- async def _async_service_set_program_options(call: ServiceCall, active: bool):
- """Execute calls to services taking a program."""
- option_key = call.data[ATTR_KEY]
- value = call.data[ATTR_VALUE]
- unit = call.data.get(ATTR_UNIT)
- client, ha_id = await _get_client_and_ha_id(hass, call.data[ATTR_DEVICE_ID])
-
- try:
- if active:
- await client.set_active_program_option(
- ha_id,
- option_key=option_key,
- value=value,
- unit=unit,
- )
- else:
- await client.set_selected_program_option(
- ha_id,
- option_key=option_key,
- value=value,
- unit=unit,
- )
- except HomeConnectError as err:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="set_options_active_program"
- if active
- else "set_options_selected_program",
- translation_placeholders={
- **get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_KEY: option_key,
- SVE_TRANSLATION_PLACEHOLDER_VALUE: str(value),
- },
- ) from err
-
- async def _async_service_command(call: ServiceCall, command_key: CommandKey):
- """Execute calls to services executing a command."""
- client, ha_id = await _get_client_and_ha_id(hass, call.data[ATTR_DEVICE_ID])
-
- try:
- await client.put_command(ha_id, command_key=command_key, value=True)
- except HomeConnectError as err:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="execute_command",
- translation_placeholders={
- **get_dict_from_home_connect_error(err),
- "command": command_key.value,
- },
- ) from err
-
- async def async_service_option_active(call: ServiceCall):
- """Service for setting an option for an active program."""
- await _async_service_set_program_options(call, True)
-
- async def async_service_option_selected(call: ServiceCall):
- """Service for setting an option for a selected program."""
- await _async_service_set_program_options(call, False)
-
- async def async_service_setting(call: ServiceCall):
- """Service for changing a setting."""
- key = call.data[ATTR_KEY]
- value = call.data[ATTR_VALUE]
- client, ha_id = await _get_client_and_ha_id(hass, call.data[ATTR_DEVICE_ID])
-
- try:
- await client.set_setting(ha_id, setting_key=key, value=value)
- except HomeConnectError as err:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="set_setting",
- translation_placeholders={
- **get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_KEY: key,
- SVE_TRANSLATION_PLACEHOLDER_VALUE: str(value),
- },
- ) from err
-
- async def async_service_pause_program(call: ServiceCall):
- """Service for pausing a program."""
- await _async_service_command(call, CommandKey.BSH_COMMON_PAUSE_PROGRAM)
-
- async def async_service_resume_program(call: ServiceCall):
- """Service for resuming a paused program."""
- await _async_service_command(call, CommandKey.BSH_COMMON_RESUME_PROGRAM)
-
- async def async_service_select_program(call: ServiceCall):
- """Service for selecting a program."""
- await _async_service_program(call, False)
-
- async def async_service_start_program(call: ServiceCall):
- """Service for starting a program."""
- await _async_service_program(call, True)
-
- hass.services.async_register(
- DOMAIN,
- SERVICE_OPTION_ACTIVE,
- async_service_option_active,
- schema=SERVICE_OPTION_SCHEMA,
- )
- hass.services.async_register(
- DOMAIN,
- SERVICE_OPTION_SELECTED,
- async_service_option_selected,
- schema=SERVICE_OPTION_SCHEMA,
- )
- hass.services.async_register(
- DOMAIN, SERVICE_SETTING, async_service_setting, schema=SERVICE_SETTING_SCHEMA
- )
- hass.services.async_register(
- DOMAIN,
- SERVICE_PAUSE_PROGRAM,
- async_service_pause_program,
- schema=SERVICE_COMMAND_SCHEMA,
- )
- hass.services.async_register(
- DOMAIN,
- SERVICE_RESUME_PROGRAM,
- async_service_resume_program,
- schema=SERVICE_COMMAND_SCHEMA,
- )
- hass.services.async_register(
- DOMAIN,
- SERVICE_SELECT_PROGRAM,
- async_service_select_program,
- schema=SERVICE_PROGRAM_SCHEMA,
- )
- hass.services.async_register(
- DOMAIN,
- SERVICE_START_PROGRAM,
- async_service_start_program,
- schema=SERVICE_PROGRAM_SCHEMA,
- )
-
+ register_actions(hass)
return True
@@ -330,18 +54,31 @@ async def async_setup_entry(hass: HomeAssistant, entry: HomeConnectConfigEntry)
session = config_entry_oauth2_flow.OAuth2Session(hass, entry, implementation)
config_entry_auth = AsyncConfigEntryAuth(hass, session)
+ try:
+ await config_entry_auth.async_get_access_token()
+ except aiohttp.ClientResponseError as err:
+ if 400 <= err.status < 500:
+ raise ConfigEntryAuthFailed from err
+ raise ConfigEntryNotReady from err
+ except aiohttp.ClientError as err:
+ raise ConfigEntryNotReady from err
home_connect_client = HomeConnectClient(config_entry_auth)
coordinator = HomeConnectCoordinator(hass, entry, home_connect_client)
- await coordinator.async_config_entry_first_refresh()
-
+ await coordinator.async_setup()
entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
entry.runtime_data.start_event_listener()
+ entry.async_create_background_task(
+ hass,
+ coordinator.async_refresh(),
+ f"home_connect-initial-full-refresh-{entry.entry_id}",
+ )
+
return True
@@ -349,6 +86,8 @@ async def async_unload_entry(
hass: HomeAssistant, entry: HomeConnectConfigEntry
) -> bool:
"""Unload a config entry."""
+ async_delete_issue(hass, DOMAIN, "deprecated_set_program_and_option_actions")
+ async_delete_issue(hass, DOMAIN, "deprecated_command_actions")
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/home_connect/api.py b/homeassistant/components/home_connect/api.py
index 5d711dae032..b66236c367d 100644
--- a/homeassistant/components/home_connect/api.py
+++ b/homeassistant/components/home_connect/api.py
@@ -1,5 +1,7 @@
"""API for Home Connect bound to HASS OAuth."""
+from typing import cast
+
from aiohomeconnect.client import AbstractAuth
from aiohomeconnect.const import API_ENDPOINT
@@ -25,4 +27,4 @@ class AsyncConfigEntryAuth(AbstractAuth):
"""Return a valid access token."""
await self.session.async_ensure_token_valid()
- return self.session.token["access_token"]
+ return cast(str, self.session.token["access_token"])
diff --git a/homeassistant/components/home_connect/binary_sensor.py b/homeassistant/components/home_connect/binary_sensor.py
index 67e3d56e713..b7b7e50047e 100644
--- a/homeassistant/components/home_connect/binary_sensor.py
+++ b/homeassistant/components/home_connect/binary_sensor.py
@@ -3,7 +3,7 @@
from dataclasses import dataclass
from typing import cast
-from aiohomeconnect.model import StatusKey
+from aiohomeconnect.model import EventKey, StatusKey
from homeassistant.components.automation import automations_with_entity
from homeassistant.components.binary_sensor import (
@@ -12,9 +12,10 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.components.script import scripts_with_entity
+from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import (
IssueSeverity,
async_create_issue,
@@ -37,6 +38,8 @@ from .coordinator import (
)
from .entity import HomeConnectEntity
+PARALLEL_UPDATES = 0
+
REFRIGERATION_DOOR_BOOLEAN_MAP = {
REFRIGERATION_STATUS_DOOR_CLOSED: False,
REFRIGERATION_STATUS_DOOR_OPEN: True,
@@ -93,12 +96,42 @@ BINARY_SENSORS = (
key=StatusKey.CONSUMER_PRODUCTS_CLEANING_ROBOT_LOST,
translation_key="lost",
),
+ HomeConnectBinarySensorEntityDescription(
+ key=StatusKey.REFRIGERATION_COMMON_DOOR_BOTTLE_COOLER,
+ boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
+ device_class=BinarySensorDeviceClass.DOOR,
+ translation_key="bottle_cooler_door",
+ ),
HomeConnectBinarySensorEntityDescription(
key=StatusKey.REFRIGERATION_COMMON_DOOR_CHILLER_COMMON,
boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
device_class=BinarySensorDeviceClass.DOOR,
+ translation_key="common_chiller_door",
+ ),
+ HomeConnectBinarySensorEntityDescription(
+ key=StatusKey.REFRIGERATION_COMMON_DOOR_CHILLER,
+ boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
+ device_class=BinarySensorDeviceClass.DOOR,
translation_key="chiller_door",
),
+ HomeConnectBinarySensorEntityDescription(
+ key=StatusKey.REFRIGERATION_COMMON_DOOR_CHILLER_LEFT,
+ boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
+ device_class=BinarySensorDeviceClass.DOOR,
+ translation_key="left_chiller_door",
+ ),
+ HomeConnectBinarySensorEntityDescription(
+ key=StatusKey.REFRIGERATION_COMMON_DOOR_CHILLER_RIGHT,
+ boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
+ device_class=BinarySensorDeviceClass.DOOR,
+ translation_key="right_chiller_door",
+ ),
+ HomeConnectBinarySensorEntityDescription(
+ key=StatusKey.REFRIGERATION_COMMON_DOOR_FLEX_COMPARTMENT,
+ boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
+ device_class=BinarySensorDeviceClass.DOOR,
+ translation_key="flex_compartment_door",
+ ),
HomeConnectBinarySensorEntityDescription(
key=StatusKey.REFRIGERATION_COMMON_DOOR_FREEZER,
boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
@@ -111,6 +144,17 @@ BINARY_SENSORS = (
device_class=BinarySensorDeviceClass.DOOR,
translation_key="refrigerator_door",
),
+ HomeConnectBinarySensorEntityDescription(
+ key=StatusKey.REFRIGERATION_COMMON_DOOR_WINE_COMPARTMENT,
+ boolean_map=REFRIGERATION_DOOR_BOOLEAN_MAP,
+ device_class=BinarySensorDeviceClass.DOOR,
+ translation_key="wine_compartment_door",
+ ),
+)
+
+CONNECTED_BINARY_ENTITY_DESCRIPTION = BinarySensorEntityDescription(
+ key=EventKey.BSH_COMMON_APPLIANCE_CONNECTED,
+ device_class=BinarySensorDeviceClass.CONNECTIVITY,
)
@@ -119,7 +163,11 @@ def _get_entities_for_appliance(
appliance: HomeConnectApplianceData,
) -> list[HomeConnectEntity]:
"""Get a list of entities."""
- entities: list[HomeConnectEntity] = []
+ entities: list[HomeConnectEntity] = [
+ HomeConnectConnectivityBinarySensor(
+ entry.runtime_data, appliance, CONNECTED_BINARY_ENTITY_DESCRIPTION
+ )
+ ]
entities.extend(
HomeConnectBinarySensor(entry.runtime_data, appliance, description)
for description in BINARY_SENSORS
@@ -133,7 +181,7 @@ def _get_entities_for_appliance(
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeConnectConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Connect binary sensor."""
setup_home_connect_entry(
@@ -159,6 +207,21 @@ class HomeConnectBinarySensor(HomeConnectEntity, BinarySensorEntity):
self._attr_is_on = None
+class HomeConnectConnectivityBinarySensor(HomeConnectEntity, BinarySensorEntity):
+ """Binary sensor for Home Connect appliance's connection status."""
+
+ _attr_entity_category = EntityCategory.DIAGNOSTIC
+
+ def update_native_value(self) -> None:
+ """Set the native value of the binary sensor."""
+ self._attr_is_on = self.appliance.info.connected
+
+ @property
+ def available(self) -> bool:
+ """Return the availability."""
+ return self.coordinator.last_update_success
+
+
class HomeConnectDoorBinarySensor(HomeConnectBinarySensor):
"""Binary sensor for Home Connect Generic Door."""
diff --git a/homeassistant/components/home_connect/button.py b/homeassistant/components/home_connect/button.py
new file mode 100644
index 00000000000..0bd31c6b7c9
--- /dev/null
+++ b/homeassistant/components/home_connect/button.py
@@ -0,0 +1,156 @@
+"""Provides button entities for Home Connect."""
+
+from aiohomeconnect.model import CommandKey
+from aiohomeconnect.model.error import HomeConnectError
+
+from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .common import setup_home_connect_entry
+from .const import APPLIANCES_WITH_PROGRAMS, DOMAIN
+from .coordinator import (
+ HomeConnectApplianceData,
+ HomeConnectConfigEntry,
+ HomeConnectCoordinator,
+)
+from .entity import HomeConnectEntity
+from .utils import get_dict_from_home_connect_error
+
+PARALLEL_UPDATES = 1
+
+
+class HomeConnectCommandButtonEntityDescription(ButtonEntityDescription):
+ """Describes Home Connect button entity."""
+
+ key: CommandKey
+
+
+COMMAND_BUTTONS = (
+ HomeConnectCommandButtonEntityDescription(
+ key=CommandKey.BSH_COMMON_OPEN_DOOR,
+ translation_key="open_door",
+ ),
+ HomeConnectCommandButtonEntityDescription(
+ key=CommandKey.BSH_COMMON_PARTLY_OPEN_DOOR,
+ translation_key="partly_open_door",
+ ),
+ HomeConnectCommandButtonEntityDescription(
+ key=CommandKey.BSH_COMMON_PAUSE_PROGRAM,
+ translation_key="pause_program",
+ ),
+ HomeConnectCommandButtonEntityDescription(
+ key=CommandKey.BSH_COMMON_RESUME_PROGRAM,
+ translation_key="resume_program",
+ ),
+)
+
+
+def _get_entities_for_appliance(
+ entry: HomeConnectConfigEntry,
+ appliance: HomeConnectApplianceData,
+) -> list[HomeConnectEntity]:
+ """Get a list of entities."""
+ entities: list[HomeConnectEntity] = []
+ entities.extend(
+ HomeConnectCommandButtonEntity(entry.runtime_data, appliance, description)
+ for description in COMMAND_BUTTONS
+ if description.key in appliance.commands
+ )
+ if appliance.info.type in APPLIANCES_WITH_PROGRAMS:
+ entities.append(
+ HomeConnectStopProgramButtonEntity(entry.runtime_data, appliance)
+ )
+
+ return entities
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: HomeConnectConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the Home Connect button entities."""
+ setup_home_connect_entry(
+ entry,
+ _get_entities_for_appliance,
+ async_add_entities,
+ )
+
+
+class HomeConnectButtonEntity(HomeConnectEntity, ButtonEntity):
+ """Describes Home Connect button entity."""
+
+ entity_description: ButtonEntityDescription
+
+ def __init__(
+ self,
+ coordinator: HomeConnectCoordinator,
+ appliance: HomeConnectApplianceData,
+ desc: ButtonEntityDescription,
+ ) -> None:
+ """Initialize the entity."""
+ super().__init__(
+ coordinator,
+ appliance,
+ desc,
+ (appliance.info.ha_id,),
+ )
+
+ def update_native_value(self) -> None:
+ """Set the value of the entity."""
+
+
+class HomeConnectCommandButtonEntity(HomeConnectButtonEntity):
+ """Button entity for Home Connect commands."""
+
+ entity_description: HomeConnectCommandButtonEntityDescription
+
+ async def async_press(self) -> None:
+ """Press the button."""
+ try:
+ await self.coordinator.client.put_command(
+ self.appliance.info.ha_id,
+ command_key=self.entity_description.key,
+ value=True,
+ )
+ except HomeConnectError as error:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="execute_command",
+ translation_placeholders={
+ **get_dict_from_home_connect_error(error),
+ "command": self.entity_description.key,
+ },
+ ) from error
+
+
+class HomeConnectStopProgramButtonEntity(HomeConnectButtonEntity):
+ """Button entity for stopping a program."""
+
+ def __init__(
+ self,
+ coordinator: HomeConnectCoordinator,
+ appliance: HomeConnectApplianceData,
+ ) -> None:
+ """Initialize the entity."""
+ super().__init__(
+ coordinator,
+ appliance,
+ ButtonEntityDescription(
+ key="StopProgram",
+ translation_key="stop_program",
+ ),
+ )
+
+ async def async_press(self) -> None:
+ """Press the button."""
+ try:
+ await self.coordinator.client.stop_program(self.appliance.info.ha_id)
+ except HomeConnectError as error:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="stop_program",
+ translation_placeholders=get_dict_from_home_connect_error(error),
+ ) from error
diff --git a/homeassistant/components/home_connect/common.py b/homeassistant/components/home_connect/common.py
index 6bd098a76fc..cd3fefad80c 100644
--- a/homeassistant/components/home_connect/common.py
+++ b/homeassistant/components/home_connect/common.py
@@ -1,15 +1,41 @@
"""Common callbacks for all Home Connect platforms."""
+from collections import defaultdict
from collections.abc import Callable
from functools import partial
from typing import cast
from aiohomeconnect.model import EventKey
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import HomeConnectApplianceData, HomeConnectConfigEntry
-from .entity import HomeConnectEntity
+from .entity import HomeConnectEntity, HomeConnectOptionEntity
+
+
+def _create_option_entities(
+ entry: HomeConnectConfigEntry,
+ appliance: HomeConnectApplianceData,
+ known_entity_unique_ids: dict[str, str],
+ get_option_entities_for_appliance: Callable[
+ [HomeConnectConfigEntry, HomeConnectApplianceData],
+ list[HomeConnectOptionEntity],
+ ],
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Create the required option entities for the appliances."""
+ option_entities_to_add = [
+ entity
+ for entity in get_option_entities_for_appliance(entry, appliance)
+ if entity.unique_id not in known_entity_unique_ids
+ ]
+ known_entity_unique_ids.update(
+ {
+ cast(str, entity.unique_id): appliance.info.ha_id
+ for entity in option_entities_to_add
+ }
+ )
+ async_add_entities(option_entities_to_add)
def _handle_paired_or_connected_appliance(
@@ -18,7 +44,13 @@ def _handle_paired_or_connected_appliance(
get_entities_for_appliance: Callable[
[HomeConnectConfigEntry, HomeConnectApplianceData], list[HomeConnectEntity]
],
- async_add_entities: AddEntitiesCallback,
+ get_option_entities_for_appliance: Callable[
+ [HomeConnectConfigEntry, HomeConnectApplianceData],
+ list[HomeConnectOptionEntity],
+ ]
+ | None,
+ changed_options_listener_remove_callbacks: dict[str, list[Callable[[], None]]],
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Handle a new paired appliance or an appliance that has been connected.
@@ -34,6 +66,33 @@ def _handle_paired_or_connected_appliance(
for entity in get_entities_for_appliance(entry, appliance)
if entity.unique_id not in known_entity_unique_ids
]
+ if get_option_entities_for_appliance:
+ entities_to_add.extend(
+ entity
+ for entity in get_option_entities_for_appliance(entry, appliance)
+ if entity.unique_id not in known_entity_unique_ids
+ )
+ for event_key in (
+ EventKey.BSH_COMMON_ROOT_ACTIVE_PROGRAM,
+ EventKey.BSH_COMMON_ROOT_SELECTED_PROGRAM,
+ ):
+ changed_options_listener_remove_callback = (
+ entry.runtime_data.async_add_listener(
+ partial(
+ _create_option_entities,
+ entry,
+ appliance,
+ known_entity_unique_ids,
+ get_option_entities_for_appliance,
+ async_add_entities,
+ ),
+ (appliance.info.ha_id, event_key),
+ )
+ )
+ entry.async_on_unload(changed_options_listener_remove_callback)
+ changed_options_listener_remove_callbacks[appliance.info.ha_id].append(
+ changed_options_listener_remove_callback
+ )
known_entity_unique_ids.update(
{
cast(str, entity.unique_id): appliance.info.ha_id
@@ -47,11 +106,17 @@ def _handle_paired_or_connected_appliance(
def _handle_depaired_appliance(
entry: HomeConnectConfigEntry,
known_entity_unique_ids: dict[str, str],
+ changed_options_listener_remove_callbacks: dict[str, list[Callable[[], None]]],
) -> None:
"""Handle a removed appliance."""
for entity_unique_id, appliance_id in known_entity_unique_ids.copy().items():
if appliance_id not in entry.runtime_data.data:
known_entity_unique_ids.pop(entity_unique_id, None)
+ if appliance_id in changed_options_listener_remove_callbacks:
+ for listener in changed_options_listener_remove_callbacks.pop(
+ appliance_id
+ ):
+ listener()
def setup_home_connect_entry(
@@ -59,22 +124,18 @@ def setup_home_connect_entry(
get_entities_for_appliance: Callable[
[HomeConnectConfigEntry, HomeConnectApplianceData], list[HomeConnectEntity]
],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+ get_option_entities_for_appliance: Callable[
+ [HomeConnectConfigEntry, HomeConnectApplianceData],
+ list[HomeConnectOptionEntity],
+ ]
+ | None = None,
) -> None:
"""Set up the callbacks for paired and depaired appliances."""
known_entity_unique_ids: dict[str, str] = {}
-
- entities: list[HomeConnectEntity] = []
- for appliance in entry.runtime_data.data.values():
- entities_to_add = get_entities_for_appliance(entry, appliance)
- known_entity_unique_ids.update(
- {
- cast(str, entity.unique_id): appliance.info.ha_id
- for entity in entities_to_add
- }
- )
- entities.extend(entities_to_add)
- async_add_entities(entities)
+ changed_options_listener_remove_callbacks: dict[str, list[Callable[[], None]]] = (
+ defaultdict(list)
+ )
entry.async_on_unload(
entry.runtime_data.async_add_special_listener(
@@ -83,6 +144,8 @@ def setup_home_connect_entry(
entry,
known_entity_unique_ids,
get_entities_for_appliance,
+ get_option_entities_for_appliance,
+ changed_options_listener_remove_callbacks,
async_add_entities,
),
(
@@ -93,7 +156,12 @@ def setup_home_connect_entry(
)
entry.async_on_unload(
entry.runtime_data.async_add_special_listener(
- partial(_handle_depaired_appliance, entry, known_entity_unique_ids),
+ partial(
+ _handle_depaired_appliance,
+ entry,
+ known_entity_unique_ids,
+ changed_options_listener_remove_callbacks,
+ ),
(EventKey.BSH_COMMON_APPLIANCE_DEPAIRED,),
)
)
diff --git a/homeassistant/components/home_connect/config_flow.py b/homeassistant/components/home_connect/config_flow.py
index 444ea24cb6b..02a3ca29335 100644
--- a/homeassistant/components/home_connect/config_flow.py
+++ b/homeassistant/components/home_connect/config_flow.py
@@ -1,7 +1,12 @@
"""Config flow for Home Connect."""
+from collections.abc import Mapping
import logging
+from typing import Any
+import voluptuous as vol
+
+from homeassistant.config_entries import SOURCE_REAUTH, ConfigFlowResult
from homeassistant.helpers import config_entry_oauth2_flow
from .const import DOMAIN
@@ -20,3 +25,29 @@ class OAuth2FlowHandler(
def logger(self) -> logging.Logger:
"""Return logger."""
return logging.getLogger(__name__)
+
+ async def async_step_reauth(
+ self, entry_data: Mapping[str, Any]
+ ) -> ConfigFlowResult:
+ """Perform reauth upon an API authentication error."""
+ return await self.async_step_reauth_confirm()
+
+ async def async_step_reauth_confirm(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Dialog that informs the user that reauth is required."""
+ if user_input is None:
+ return self.async_show_form(
+ step_id="reauth_confirm",
+ data_schema=vol.Schema({}),
+ )
+ return await self.async_step_user()
+
+ async def async_oauth_create_entry(self, data: dict) -> ConfigFlowResult:
+ """Create an oauth config entry or update existing entry for reauth."""
+ if self.source == SOURCE_REAUTH:
+ return self.async_update_reload_and_abort(
+ self._get_reauth_entry(),
+ data_updates=data,
+ )
+ return await super().async_oauth_create_entry(data)
diff --git a/homeassistant/components/home_connect/const.py b/homeassistant/components/home_connect/const.py
index 127aa1ffe92..64bf4af29a4 100644
--- a/homeassistant/components/home_connect/const.py
+++ b/homeassistant/components/home_connect/const.py
@@ -1,9 +1,16 @@
"""Constants for the Home Connect integration."""
-from aiohomeconnect.model import EventKey, SettingKey, StatusKey
+from typing import cast
+
+from aiohomeconnect.model import EventKey, OptionKey, ProgramKey, SettingKey, StatusKey
+
+from homeassistant.const import UnitOfTemperature, UnitOfTime, UnitOfVolume
+
+from .utils import bsh_key_to_translation_key
DOMAIN = "home_connect"
+API_DEFAULT_RETRY_AFTER = 60
APPLIANCES_WITH_PROGRAMS = (
"CleaningRobot",
@@ -17,6 +24,13 @@ APPLIANCES_WITH_PROGRAMS = (
"WasherDryer",
)
+UNIT_MAP = {
+ "seconds": UnitOfTime.SECONDS,
+ "ml": UnitOfVolume.MILLILITERS,
+ "°C": UnitOfTemperature.CELSIUS,
+ "°F": UnitOfTemperature.FAHRENHEIT,
+}
+
BSH_POWER_ON = "BSH.Common.EnumType.PowerState.On"
BSH_POWER_OFF = "BSH.Common.EnumType.PowerState.Off"
@@ -52,22 +66,283 @@ SERVICE_OPTION_SELECTED = "set_option_selected"
SERVICE_PAUSE_PROGRAM = "pause_program"
SERVICE_RESUME_PROGRAM = "resume_program"
SERVICE_SELECT_PROGRAM = "select_program"
+SERVICE_SET_PROGRAM_AND_OPTIONS = "set_program_and_options"
SERVICE_SETTING = "change_setting"
SERVICE_START_PROGRAM = "start_program"
-
+ATTR_AFFECTS_TO = "affects_to"
ATTR_KEY = "key"
ATTR_PROGRAM = "program"
ATTR_UNIT = "unit"
ATTR_VALUE = "value"
+AFFECTS_TO_ACTIVE_PROGRAM = "active_program"
+AFFECTS_TO_SELECTED_PROGRAM = "selected_program"
-SVE_TRANSLATION_KEY_SET_SETTING = "set_setting_entity"
-SVE_TRANSLATION_PLACEHOLDER_APPLIANCE_NAME = "appliance_name"
-SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID = "entity_id"
-SVE_TRANSLATION_PLACEHOLDER_PROGRAM = "program"
-SVE_TRANSLATION_PLACEHOLDER_KEY = "key"
-SVE_TRANSLATION_PLACEHOLDER_VALUE = "value"
+
+TRANSLATION_KEYS_PROGRAMS_MAP = {
+ bsh_key_to_translation_key(program.value): cast(ProgramKey, program)
+ for program in ProgramKey
+ if program != ProgramKey.UNKNOWN
+}
+
+PROGRAMS_TRANSLATION_KEYS_MAP = {
+ value: key for key, value in TRANSLATION_KEYS_PROGRAMS_MAP.items()
+}
+
+AVAILABLE_MAPS_ENUM = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CleaningRobot.EnumType.AvailableMaps.TempMap",
+ "ConsumerProducts.CleaningRobot.EnumType.AvailableMaps.Map1",
+ "ConsumerProducts.CleaningRobot.EnumType.AvailableMaps.Map2",
+ "ConsumerProducts.CleaningRobot.EnumType.AvailableMaps.Map3",
+ )
+}
+
+CLEANING_MODE_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CleaningRobot.EnumType.CleaningModes.Silent",
+ "ConsumerProducts.CleaningRobot.EnumType.CleaningModes.Standard",
+ "ConsumerProducts.CleaningRobot.EnumType.CleaningModes.Power",
+ )
+}
+
+BEAN_AMOUNT_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.VeryMild",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.Mild",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.MildPlus",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.Normal",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.NormalPlus",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.Strong",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.StrongPlus",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.VeryStrong",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.VeryStrongPlus",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.ExtraStrong",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.DoubleShot",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.DoubleShotPlus",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.DoubleShotPlusPlus",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.TripleShot",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.TripleShotPlus",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanAmount.CoffeeGround",
+ )
+}
+
+COFFEE_TEMPERATURE_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeTemperature.88C",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeTemperature.90C",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeTemperature.92C",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeTemperature.94C",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeTemperature.95C",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeTemperature.96C",
+ )
+}
+
+BEAN_CONTAINER_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanContainerSelection.Right",
+ "ConsumerProducts.CoffeeMaker.EnumType.BeanContainerSelection.Left",
+ )
+}
+
+FLOW_RATE_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CoffeeMaker.EnumType.FlowRate.Normal",
+ "ConsumerProducts.CoffeeMaker.EnumType.FlowRate.Intense",
+ "ConsumerProducts.CoffeeMaker.EnumType.FlowRate.IntensePlus",
+ )
+}
+
+COFFEE_MILK_RATIO_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.10Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.20Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.25Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.30Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.40Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.50Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.55Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.60Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.65Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.67Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.70Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.75Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.80Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.85Percent",
+ "ConsumerProducts.CoffeeMaker.EnumType.CoffeeMilkRatio.90Percent",
+ )
+}
+
+HOT_WATER_TEMPERATURE_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.WhiteTea",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.GreenTea",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.BlackTea",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.50C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.55C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.60C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.65C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.70C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.75C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.80C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.85C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.90C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.95C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.97C",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.122F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.131F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.140F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.149F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.158F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.167F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.176F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.185F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.194F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.203F",
+ "ConsumerProducts.CoffeeMaker.EnumType.HotWaterTemperature.Max",
+ )
+}
+
+DRYING_TARGET_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "LaundryCare.Dryer.EnumType.DryingTarget.IronDry",
+ "LaundryCare.Dryer.EnumType.DryingTarget.GentleDry",
+ "LaundryCare.Dryer.EnumType.DryingTarget.CupboardDry",
+ "LaundryCare.Dryer.EnumType.DryingTarget.CupboardDryPlus",
+ "LaundryCare.Dryer.EnumType.DryingTarget.ExtraDry",
+ )
+}
+
+VENTING_LEVEL_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "Cooking.Hood.EnumType.Stage.FanOff",
+ "Cooking.Hood.EnumType.Stage.FanStage01",
+ "Cooking.Hood.EnumType.Stage.FanStage02",
+ "Cooking.Hood.EnumType.Stage.FanStage03",
+ "Cooking.Hood.EnumType.Stage.FanStage04",
+ "Cooking.Hood.EnumType.Stage.FanStage05",
+ )
+}
+
+INTENSIVE_LEVEL_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "Cooking.Hood.EnumType.IntensiveStage.IntensiveStageOff",
+ "Cooking.Hood.EnumType.IntensiveStage.IntensiveStage1",
+ "Cooking.Hood.EnumType.IntensiveStage.IntensiveStage2",
+ )
+}
+
+WARMING_LEVEL_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "Cooking.Oven.EnumType.WarmingLevel.Low",
+ "Cooking.Oven.EnumType.WarmingLevel.Medium",
+ "Cooking.Oven.EnumType.WarmingLevel.High",
+ )
+}
+
+TEMPERATURE_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "LaundryCare.Washer.EnumType.Temperature.Cold",
+ "LaundryCare.Washer.EnumType.Temperature.GC20",
+ "LaundryCare.Washer.EnumType.Temperature.GC30",
+ "LaundryCare.Washer.EnumType.Temperature.GC40",
+ "LaundryCare.Washer.EnumType.Temperature.GC50",
+ "LaundryCare.Washer.EnumType.Temperature.GC60",
+ "LaundryCare.Washer.EnumType.Temperature.GC70",
+ "LaundryCare.Washer.EnumType.Temperature.GC80",
+ "LaundryCare.Washer.EnumType.Temperature.GC90",
+ "LaundryCare.Washer.EnumType.Temperature.UlCold",
+ "LaundryCare.Washer.EnumType.Temperature.UlWarm",
+ "LaundryCare.Washer.EnumType.Temperature.UlHot",
+ "LaundryCare.Washer.EnumType.Temperature.UlExtraHot",
+ )
+}
+
+SPIN_SPEED_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "LaundryCare.Washer.EnumType.SpinSpeed.Off",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM400",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM600",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM700",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM800",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM900",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM1000",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM1200",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM1400",
+ "LaundryCare.Washer.EnumType.SpinSpeed.RPM1600",
+ "LaundryCare.Washer.EnumType.SpinSpeed.UlOff",
+ "LaundryCare.Washer.EnumType.SpinSpeed.UlLow",
+ "LaundryCare.Washer.EnumType.SpinSpeed.UlMedium",
+ "LaundryCare.Washer.EnumType.SpinSpeed.UlHigh",
+ )
+}
+
+VARIO_PERFECT_OPTIONS = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "LaundryCare.Common.EnumType.VarioPerfect.Off",
+ "LaundryCare.Common.EnumType.VarioPerfect.EcoPerfect",
+ "LaundryCare.Common.EnumType.VarioPerfect.SpeedPerfect",
+ )
+}
+
+
+PROGRAM_ENUM_OPTIONS = {
+ bsh_key_to_translation_key(option_key): (
+ option_key,
+ options,
+ )
+ for option_key, options in (
+ (
+ OptionKey.CONSUMER_PRODUCTS_CLEANING_ROBOT_REFERENCE_MAP_ID,
+ AVAILABLE_MAPS_ENUM,
+ ),
+ (
+ OptionKey.CONSUMER_PRODUCTS_CLEANING_ROBOT_CLEANING_MODE,
+ CLEANING_MODE_OPTIONS,
+ ),
+ (OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEAN_AMOUNT, BEAN_AMOUNT_OPTIONS),
+ (
+ OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_COFFEE_TEMPERATURE,
+ COFFEE_TEMPERATURE_OPTIONS,
+ ),
+ (
+ OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEAN_CONTAINER_SELECTION,
+ BEAN_CONTAINER_OPTIONS,
+ ),
+ (OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_FLOW_RATE, FLOW_RATE_OPTIONS),
+ (
+ OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_COFFEE_MILK_RATIO,
+ COFFEE_MILK_RATIO_OPTIONS,
+ ),
+ (
+ OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_HOT_WATER_TEMPERATURE,
+ HOT_WATER_TEMPERATURE_OPTIONS,
+ ),
+ (OptionKey.LAUNDRY_CARE_DRYER_DRYING_TARGET, DRYING_TARGET_OPTIONS),
+ (OptionKey.COOKING_COMMON_HOOD_VENTING_LEVEL, VENTING_LEVEL_OPTIONS),
+ (OptionKey.COOKING_COMMON_HOOD_INTENSIVE_LEVEL, INTENSIVE_LEVEL_OPTIONS),
+ (OptionKey.COOKING_OVEN_WARMING_LEVEL, WARMING_LEVEL_OPTIONS),
+ (OptionKey.LAUNDRY_CARE_WASHER_TEMPERATURE, TEMPERATURE_OPTIONS),
+ (OptionKey.LAUNDRY_CARE_WASHER_SPIN_SPEED, SPIN_SPEED_OPTIONS),
+ (OptionKey.LAUNDRY_CARE_COMMON_VARIO_PERFECT, VARIO_PERFECT_OPTIONS),
+ )
+}
OLD_NEW_UNIQUE_ID_SUFFIX_MAP = {
diff --git a/homeassistant/components/home_connect/coordinator.py b/homeassistant/components/home_connect/coordinator.py
index 68652936872..495b4efab32 100644
--- a/homeassistant/components/home_connect/coordinator.py
+++ b/homeassistant/components/home_connect/coordinator.py
@@ -1,20 +1,25 @@
"""Coordinator for Home Connect."""
-import asyncio
+from __future__ import annotations
+
+from asyncio import sleep as asyncio_sleep
from collections import defaultdict
from collections.abc import Callable
from dataclasses import dataclass
import logging
-from typing import Any
+from typing import Any, cast
from aiohomeconnect.client import Client as HomeConnectClient
from aiohomeconnect.model import (
+ CommandKey,
Event,
EventKey,
EventMessage,
EventType,
GetSetting,
HomeAppliance,
+ OptionKey,
+ ProgramKey,
SettingKey,
Status,
StatusKey,
@@ -24,39 +29,45 @@ from aiohomeconnect.model.error import (
HomeConnectApiError,
HomeConnectError,
HomeConnectRequestError,
+ TooManyRequestsError,
+ UnauthorizedError,
)
-from aiohomeconnect.model.program import EnumerateProgram
+from aiohomeconnect.model.program import EnumerateProgram, ProgramDefinitionOption
from propcache.api import cached_property
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
+from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import APPLIANCES_WITH_PROGRAMS, DOMAIN
+from .const import API_DEFAULT_RETRY_AFTER, APPLIANCES_WITH_PROGRAMS, DOMAIN
from .utils import get_dict_from_home_connect_error
_LOGGER = logging.getLogger(__name__)
type HomeConnectConfigEntry = ConfigEntry[HomeConnectCoordinator]
-EVENT_STREAM_RECONNECT_DELAY = 30
-
@dataclass(frozen=True, kw_only=True)
class HomeConnectApplianceData:
"""Class to hold Home Connect appliance data."""
+ commands: set[CommandKey]
events: dict[EventKey, Event]
info: HomeAppliance
+ options: dict[OptionKey, ProgramDefinitionOption]
programs: list[EnumerateProgram]
settings: dict[SettingKey, GetSetting]
status: dict[StatusKey, Status]
- def update(self, other: "HomeConnectApplianceData") -> None:
+ def update(self, other: HomeConnectApplianceData) -> None:
"""Update data with data from other instance."""
+ self.commands.update(other.commands)
self.events.update(other.events)
self.info.connected = other.info.connected
+ self.options.clear()
+ self.options.update(other.options)
self.programs.clear()
self.programs.extend(other.programs)
self.settings.update(other.settings)
@@ -88,6 +99,7 @@ class HomeConnectCoordinator(
CALLBACK_TYPE, tuple[CALLBACK_TYPE, tuple[EventKey, ...]]
] = {}
self.device_registry = dr.async_get(self.hass)
+ self.data = {}
@cached_property
def context_listeners(self) -> dict[tuple[str, EventKey], list[CALLBACK_TYPE]]:
@@ -143,12 +155,22 @@ class HomeConnectCoordinator(
f"home_connect-events_listener_task-{self.config_entry.entry_id}",
)
- async def _event_listener(self) -> None:
+ async def _event_listener(self) -> None: # noqa: C901
"""Match event with listener for event type."""
+ retry_time = 10
while True:
try:
async for event_message in self.client.stream_all_events():
+ retry_time = 10
event_message_ha_id = event_message.ha_id
+ if (
+ event_message_ha_id in self.data
+ and not self.data[event_message_ha_id].info.connected
+ ):
+ self.data[event_message_ha_id].info.connected = True
+ self._call_all_event_listeners_for_appliance(
+ event_message_ha_id
+ )
match event_message.type:
case EventType.STATUS:
statuses = self.data[event_message_ha_id].status
@@ -168,8 +190,9 @@ class HomeConnectCoordinator(
settings = self.data[event_message_ha_id].settings
events = self.data[event_message_ha_id].events
for event in event_message.data.items:
- if event.key in SettingKey:
- setting_key = SettingKey(event.key)
+ event_key = event.key
+ if event_key in SettingKey:
+ setting_key = SettingKey(event_key)
if setting_key in settings:
settings[setting_key].value = event.value
else:
@@ -179,7 +202,16 @@ class HomeConnectCoordinator(
value=event.value,
)
else:
- events[event.key] = event
+ if event_key in (
+ EventKey.BSH_COMMON_ROOT_ACTIVE_PROGRAM,
+ EventKey.BSH_COMMON_ROOT_SELECTED_PROGRAM,
+ ):
+ await self.update_options(
+ event_message_ha_id,
+ event_key,
+ ProgramKey(cast(str, event.value)),
+ )
+ events[event_key] = event
self._call_event_listener(event_message)
case EventType.EVENT:
@@ -200,15 +232,15 @@ class HomeConnectCoordinator(
self.data[event_message_ha_id].update(appliance_data)
else:
self.data[event_message_ha_id] = appliance_data
- for listener, context in list(
- self._special_listeners.values()
- ) + list(self._listeners.values()):
- assert isinstance(context, tuple)
+ for listener, context in self._special_listeners.values():
if (
EventKey.BSH_COMMON_APPLIANCE_DEPAIRED
not in context
):
listener()
+ self._call_all_event_listeners_for_appliance(
+ event_message_ha_id
+ )
case EventType.DISCONNECTED:
self.data[event_message_ha_id].info.connected = False
@@ -234,23 +266,28 @@ class HomeConnectCoordinator(
except (EventStreamInterruptedError, HomeConnectRequestError) as error:
_LOGGER.debug(
"Non-breaking error (%s) while listening for events,"
- " continuing in 30 seconds",
- type(error).__name__,
+ " continuing in %s seconds",
+ error,
+ retry_time,
)
- await asyncio.sleep(EVENT_STREAM_RECONNECT_DELAY)
+ await asyncio_sleep(retry_time)
+ retry_time = min(retry_time * 2, 3600)
except HomeConnectApiError as error:
_LOGGER.error("Error while listening for events: %s", error)
self.hass.config_entries.async_schedule_reload(
self.config_entry.entry_id
)
break
- # if there was a non-breaking error, we continue listening
- # but we need to refresh the data to get the possible changes
- # that happened while the event stream was interrupted
- await self.async_refresh()
+
+ # Trigger to delete the possible depaired device entities
+ # from known_entities variable at common.py
+ for listener, context in self._special_listeners.values():
+ assert isinstance(context, tuple)
+ if EventKey.BSH_COMMON_APPLIANCE_DEPAIRED in context:
+ listener()
@callback
- def _call_event_listener(self, event_message: EventMessage):
+ def _call_event_listener(self, event_message: EventMessage) -> None:
"""Call listener for event."""
for event in event_message.data.items:
for listener in self.context_listeners.get(
@@ -259,28 +296,98 @@ class HomeConnectCoordinator(
listener()
@callback
- def _call_all_event_listeners_for_appliance(self, ha_id: str):
+ def _call_all_event_listeners_for_appliance(self, ha_id: str) -> None:
for listener, context in self._listeners.values():
if isinstance(context, tuple) and context[0] == ha_id:
listener()
async def _async_update_data(self) -> dict[str, HomeConnectApplianceData]:
"""Fetch data from Home Connect."""
+ await self._async_setup()
+
+ for appliance_data in self.data.values():
+ appliance = appliance_data.info
+ ha_id = appliance.ha_id
+ while True:
+ try:
+ self.data[ha_id] = await self._get_appliance_data(
+ appliance, self.data.get(ha_id)
+ )
+ except TooManyRequestsError as err:
+ _LOGGER.debug(
+ "Rate limit exceeded on initial fetch: %s",
+ err,
+ )
+ await asyncio_sleep(err.retry_after or API_DEFAULT_RETRY_AFTER)
+ else:
+ break
+
+ for listener, context in self._special_listeners.values():
+ assert isinstance(context, tuple)
+ if EventKey.BSH_COMMON_APPLIANCE_PAIRED in context:
+ listener()
+
+ return self.data
+
+ async def async_setup(self) -> None:
+ """Set up the devices."""
+ try:
+ await self._async_setup()
+ except UpdateFailed as err:
+ raise ConfigEntryNotReady from err
+
+ async def _async_setup(self) -> None:
+ """Set up the devices."""
+ old_appliances = set(self.data.keys())
try:
appliances = await self.client.get_home_appliances()
+ except UnauthorizedError as error:
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN,
+ translation_key="auth_error",
+ translation_placeholders=get_dict_from_home_connect_error(error),
+ ) from error
except HomeConnectError as error:
+ for appliance_data in self.data.values():
+ appliance_data.info.connected = False
raise UpdateFailed(
translation_domain=DOMAIN,
translation_key="fetch_api_error",
translation_placeholders=get_dict_from_home_connect_error(error),
) from error
- return {
- appliance.ha_id: await self._get_appliance_data(
- appliance, self.data.get(appliance.ha_id) if self.data else None
+ for appliance in appliances.homeappliances:
+ self.device_registry.async_get_or_create(
+ config_entry_id=self.config_entry.entry_id,
+ identifiers={(DOMAIN, appliance.ha_id)},
+ manufacturer=appliance.brand,
+ name=appliance.name,
+ model=appliance.vib,
)
- for appliance in appliances.homeappliances
- }
+ if appliance.ha_id not in self.data:
+ self.data[appliance.ha_id] = HomeConnectApplianceData(
+ commands=set(),
+ events={},
+ info=appliance,
+ options={},
+ programs=[],
+ settings={},
+ status={},
+ )
+ else:
+ self.data[appliance.ha_id].info.connected = appliance.connected
+ old_appliances.remove(appliance.ha_id)
+
+ for ha_id in old_appliances:
+ self.data.pop(ha_id, None)
+ device = self.device_registry.async_get_device(
+ identifiers={(DOMAIN, ha_id)}
+ )
+ if device:
+ self.device_registry.async_update_device(
+ device_id=device.id,
+ remove_config_entry_id=self.config_entry.entry_id,
+ )
async def _get_appliance_data(
self,
@@ -302,13 +409,13 @@ class HomeConnectCoordinator(
await self.client.get_settings(appliance.ha_id)
).settings
}
+ except TooManyRequestsError:
+ raise
except HomeConnectError as error:
_LOGGER.debug(
"Error fetching settings for %s: %s",
appliance.ha_id,
- error
- if isinstance(error, HomeConnectApiError)
- else type(error).__name__,
+ error,
)
settings = {}
try:
@@ -316,40 +423,43 @@ class HomeConnectCoordinator(
status.key: status
for status in (await self.client.get_status(appliance.ha_id)).status
}
+ except TooManyRequestsError:
+ raise
except HomeConnectError as error:
_LOGGER.debug(
"Error fetching status for %s: %s",
appliance.ha_id,
- error
- if isinstance(error, HomeConnectApiError)
- else type(error).__name__,
+ error,
)
status = {}
programs = []
events = {}
+ options = {}
if appliance.type in APPLIANCES_WITH_PROGRAMS:
try:
all_programs = await self.client.get_all_programs(appliance.ha_id)
+ except TooManyRequestsError:
+ raise
except HomeConnectError as error:
_LOGGER.debug(
"Error fetching programs for %s: %s",
appliance.ha_id,
- error
- if isinstance(error, HomeConnectApiError)
- else type(error).__name__,
+ error,
)
else:
programs.extend(all_programs.programs)
+ current_program_key = None
+ program_options = None
for program, event_key in (
- (
- all_programs.active,
- EventKey.BSH_COMMON_ROOT_ACTIVE_PROGRAM,
- ),
(
all_programs.selected,
EventKey.BSH_COMMON_ROOT_SELECTED_PROGRAM,
),
+ (
+ all_programs.active,
+ EventKey.BSH_COMMON_ROOT_ACTIVE_PROGRAM,
+ ),
):
if program and program.key:
events[event_key] = Event(
@@ -360,10 +470,43 @@ class HomeConnectCoordinator(
"",
program.key,
)
+ current_program_key = program.key
+ program_options = program.options
+ if current_program_key:
+ options = await self.get_options_definitions(
+ appliance.ha_id, current_program_key
+ )
+ for option in program_options or []:
+ option_event_key = EventKey(option.key)
+ events[option_event_key] = Event(
+ option_event_key,
+ option.key,
+ 0,
+ "",
+ "",
+ option.value,
+ option.name,
+ display_value=option.display_value,
+ unit=option.unit,
+ )
+
+ try:
+ commands = {
+ command.key
+ for command in (
+ await self.client.get_available_commands(appliance.ha_id)
+ ).commands
+ }
+ except TooManyRequestsError:
+ raise
+ except HomeConnectError:
+ commands = set()
appliance_data = HomeConnectApplianceData(
+ commands=commands,
events=events,
info=appliance,
+ options=options,
programs=programs,
settings=settings,
status=status,
@@ -373,3 +516,61 @@ class HomeConnectCoordinator(
appliance_data = appliance_data_to_update
return appliance_data
+
+ async def get_options_definitions(
+ self, ha_id: str, program_key: ProgramKey
+ ) -> dict[OptionKey, ProgramDefinitionOption]:
+ """Get options with constraints for appliance."""
+ if program_key is ProgramKey.UNKNOWN:
+ return {}
+ try:
+ return {
+ option.key: option
+ for option in (
+ await self.client.get_available_program(
+ ha_id, program_key=program_key
+ )
+ ).options
+ or []
+ }
+ except TooManyRequestsError:
+ raise
+ except HomeConnectError as error:
+ _LOGGER.debug(
+ "Error fetching options for %s: %s",
+ ha_id,
+ error,
+ )
+ return {}
+
+ async def update_options(
+ self, ha_id: str, event_key: EventKey, program_key: ProgramKey
+ ) -> None:
+ """Update options for appliance."""
+ options = self.data[ha_id].options
+ events = self.data[ha_id].events
+ options_to_notify = options.copy()
+ options.clear()
+ options.update(await self.get_options_definitions(ha_id, program_key))
+
+ for option in options.values():
+ option_value = option.constraints.default if option.constraints else None
+ if option_value is not None:
+ option_event_key = EventKey(option.key)
+ events[option_event_key] = Event(
+ option_event_key,
+ option.key.value,
+ 0,
+ "",
+ "",
+ option_value,
+ option.name,
+ unit=option.unit,
+ )
+ options_to_notify.update(options)
+ for option_key in options_to_notify:
+ for listener in self.context_listeners.get(
+ (ha_id, EventKey(option_key)),
+ [],
+ ):
+ listener()
diff --git a/homeassistant/components/home_connect/entity.py b/homeassistant/components/home_connect/entity.py
index 8eb9d757f14..facb3b14a9b 100644
--- a/homeassistant/components/home_connect/entity.py
+++ b/homeassistant/components/home_connect/entity.py
@@ -1,17 +1,30 @@
"""Home Connect entity base class."""
from abc import abstractmethod
+from collections.abc import Callable, Coroutine
+import contextlib
+from datetime import datetime
import logging
+from typing import Any, Concatenate, cast
-from aiohomeconnect.model import EventKey
+from aiohomeconnect.model import EventKey, OptionKey
+from aiohomeconnect.model.error import (
+ ActiveProgramNotSetError,
+ HomeConnectError,
+ TooManyRequestsError,
+)
+from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.core import callback
+from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import EntityDescription
+from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .const import DOMAIN
+from .const import API_DEFAULT_RETRY_AFTER, DOMAIN
from .coordinator import HomeConnectApplianceData, HomeConnectCoordinator
+from .utils import get_dict_from_home_connect_error
_LOGGER = logging.getLogger(__name__)
@@ -27,9 +40,13 @@ class HomeConnectEntity(CoordinatorEntity[HomeConnectCoordinator]):
coordinator: HomeConnectCoordinator,
appliance: HomeConnectApplianceData,
desc: EntityDescription,
+ context_override: Any | None = None,
) -> None:
"""Initialize the entity."""
- super().__init__(coordinator, (appliance.info.ha_id, EventKey(desc.key)))
+ context = (appliance.info.ha_id, EventKey(desc.key))
+ if context_override is not None:
+ context = context_override
+ super().__init__(coordinator, context)
self.appliance = appliance
self.entity_description = desc
self._attr_unique_id = f"{appliance.info.ha_id}-{desc.key}"
@@ -46,17 +63,109 @@ class HomeConnectEntity(CoordinatorEntity[HomeConnectCoordinator]):
def _handle_coordinator_update(self) -> None:
"""Handle updated data from the coordinator."""
self.update_native_value()
+ available = self._attr_available = self.appliance.info.connected
self.async_write_ha_state()
- _LOGGER.debug("Updated %s, new state: %s", self.entity_id, self.state)
+ state = STATE_UNAVAILABLE if not available else self.state
+ _LOGGER.debug("Updated %s, new state: %s", self.entity_id, state)
@property
def bsh_key(self) -> str:
"""Return the BSH key."""
return self.entity_description.key
+ @property
+ def available(self) -> bool:
+ """Return True if entity is available.
+
+ Do not use self.last_update_success for available state
+ as event updates should take precedence over the coordinator
+ refresh.
+ """
+ return self._attr_available
+
+
+class HomeConnectOptionEntity(HomeConnectEntity):
+ """Class for entities that represents program options."""
+
@property
def available(self) -> bool:
"""Return True if entity is available."""
- return (
- self.appliance.info.connected and self._attr_available and super().available
- )
+ return super().available and self.bsh_key in self.appliance.options
+
+ @property
+ def option_value(self) -> str | int | float | bool | None:
+ """Return the state of the entity."""
+ if event := self.appliance.events.get(EventKey(self.bsh_key)):
+ return event.value
+ return None
+
+ async def async_set_option(self, value: str | float | bool) -> None:
+ """Set an option for the entity."""
+ try:
+ # We try to set the active program option first,
+ # if it fails we try to set the selected program option
+ with contextlib.suppress(ActiveProgramNotSetError):
+ await self.coordinator.client.set_active_program_option(
+ self.appliance.info.ha_id,
+ option_key=self.bsh_key,
+ value=value,
+ )
+ _LOGGER.debug(
+ "Updated %s for the active program, new state: %s",
+ self.entity_id,
+ self.state,
+ )
+ return
+
+ await self.coordinator.client.set_selected_program_option(
+ self.appliance.info.ha_id,
+ option_key=self.bsh_key,
+ value=value,
+ )
+ _LOGGER.debug(
+ "Updated %s for the selected program, new state: %s",
+ self.entity_id,
+ self.state,
+ )
+ except HomeConnectError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_option",
+ translation_placeholders=get_dict_from_home_connect_error(err),
+ ) from err
+
+ @property
+ def bsh_key(self) -> OptionKey:
+ """Return the BSH key."""
+ return cast(OptionKey, self.entity_description.key)
+
+
+def constraint_fetcher[_EntityT: HomeConnectEntity, **_P](
+ func: Callable[Concatenate[_EntityT, _P], Coroutine[Any, Any, Any]],
+) -> Callable[Concatenate[_EntityT, _P], Coroutine[Any, Any, None]]:
+ """Decorate the function to catch Home Connect too many requests error and retry later.
+
+ If it needs to be called later, it will call async_write_ha_state function
+ """
+
+ async def handler_to_return(
+ self: _EntityT, *args: _P.args, **kwargs: _P.kwargs
+ ) -> None:
+ async def handler(_datetime: datetime | None = None) -> None:
+ try:
+ await func(self, *args, **kwargs)
+ except TooManyRequestsError as err:
+ if (retry_after := err.retry_after) is None:
+ retry_after = API_DEFAULT_RETRY_AFTER
+ async_call_later(self.hass, retry_after, handler)
+ except HomeConnectError as err:
+ _LOGGER.error(
+ "Error fetching constraints for %s: %s", self.entity_id, err
+ )
+ else:
+ if _datetime is not None:
+ self.async_write_ha_state()
+
+ await handler()
+
+ return handler_to_return
diff --git a/homeassistant/components/home_connect/icons.json b/homeassistant/components/home_connect/icons.json
index 166b2fe2c34..9b4c9276998 100644
--- a/homeassistant/components/home_connect/icons.json
+++ b/homeassistant/components/home_connect/icons.json
@@ -18,6 +18,9 @@
"set_option_selected": {
"service": "mdi:gesture-tap"
},
+ "set_program_and_options": {
+ "service": "mdi:form-select"
+ },
"change_setting": {
"service": "mdi:cog"
}
@@ -46,6 +49,23 @@
"default": "mdi:map-marker-remove-variant"
}
},
+ "button": {
+ "open_door": {
+ "default": "mdi:door-open"
+ },
+ "partly_open_door": {
+ "default": "mdi:door-open"
+ },
+ "pause_program": {
+ "default": "mdi:pause"
+ },
+ "resume_program": {
+ "default": "mdi:play"
+ },
+ "stop_program": {
+ "default": "mdi:stop"
+ }
+ },
"sensor": {
"operation_state": {
"default": "mdi:state-machine",
@@ -93,7 +113,7 @@
"milk_counter": {
"default": "mdi:cup"
},
- "coffee_and_milk": {
+ "coffee_and_milk_counter": {
"default": "mdi:coffee"
},
"ristretto_espresso_counter": {
@@ -205,6 +225,39 @@
},
"door-assistant_freezer": {
"default": "mdi:door"
+ },
+ "silence_on_demand": {
+ "default": "mdi:volume-mute",
+ "state": {
+ "on": "mdi:volume-mute",
+ "off": "mdi:volume-high"
+ }
+ },
+ "half_load": {
+ "default": "mdi:fraction-one-half"
+ },
+ "hygiene_plus": {
+ "default": "mdi:silverware-clean"
+ },
+ "eco_dry": {
+ "default": "mdi:sprout"
+ },
+ "fast_pre_heat": {
+ "default": "mdi:fire"
+ },
+ "i_dos_1_active": {
+ "default": "mdi:numeric-1-circle"
+ },
+ "i_dos_2_active": {
+ "default": "mdi:numeric-2-circle"
+ }
+ },
+ "time": {
+ "start_in_relative": {
+ "default": "mdi:progress-clock"
+ },
+ "finish_in_relative": {
+ "default": "mdi:progress-clock"
}
}
}
diff --git a/homeassistant/components/home_connect/light.py b/homeassistant/components/home_connect/light.py
index 05c154d9153..707620f099a 100644
--- a/homeassistant/components/home_connect/light.py
+++ b/homeassistant/components/home_connect/light.py
@@ -17,15 +17,11 @@ from homeassistant.components.light import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .common import setup_home_connect_entry
-from .const import (
- BSH_AMBIENT_LIGHT_COLOR_CUSTOM_COLOR,
- DOMAIN,
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID,
-)
+from .const import BSH_AMBIENT_LIGHT_COLOR_CUSTOM_COLOR, DOMAIN
from .coordinator import (
HomeConnectApplianceData,
HomeConnectConfigEntry,
@@ -36,6 +32,8 @@ from .utils import get_dict_from_home_connect_error
_LOGGER = logging.getLogger(__name__)
+PARALLEL_UPDATES = 1
+
@dataclass(frozen=True, kw_only=True)
class HomeConnectLightEntityDescription(LightEntityDescription):
@@ -94,7 +92,7 @@ def _get_entities_for_appliance(
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeConnectConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Connect light."""
setup_home_connect_entry(
@@ -162,7 +160,7 @@ class HomeConnectLight(HomeConnectEntity, LightEntity):
translation_key="turn_on_light",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
+ "entity_id": self.entity_id,
},
) from err
if self._color_key and self._custom_color_key:
@@ -181,7 +179,7 @@ class HomeConnectLight(HomeConnectEntity, LightEntity):
translation_key="select_light_custom_color",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
+ "entity_id": self.entity_id,
},
) from err
@@ -199,7 +197,7 @@ class HomeConnectLight(HomeConnectEntity, LightEntity):
translation_key="set_light_color",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
+ "entity_id": self.entity_id,
},
) from err
return
@@ -229,7 +227,7 @@ class HomeConnectLight(HomeConnectEntity, LightEntity):
translation_key="set_light_color",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
+ "entity_id": self.entity_id,
},
) from err
return
@@ -252,7 +250,7 @@ class HomeConnectLight(HomeConnectEntity, LightEntity):
translation_key="set_light_brightness",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
+ "entity_id": self.entity_id,
},
) from err
@@ -270,7 +268,7 @@ class HomeConnectLight(HomeConnectEntity, LightEntity):
translation_key="turn_off_light",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
+ "entity_id": self.entity_id,
},
) from err
diff --git a/homeassistant/components/home_connect/manifest.json b/homeassistant/components/home_connect/manifest.json
index 41d359446fa..62892e7c85b 100644
--- a/homeassistant/components/home_connect/manifest.json
+++ b/homeassistant/components/home_connect/manifest.json
@@ -3,9 +3,10 @@
"name": "Home Connect",
"codeowners": ["@DavidMStraub", "@Diegorro98", "@MartinHjelmare"],
"config_flow": true,
- "dependencies": ["application_credentials"],
+ "dependencies": ["application_credentials", "repairs"],
"documentation": "https://www.home-assistant.io/integrations/home_connect",
"iot_class": "cloud_push",
"loggers": ["aiohomeconnect"],
- "requirements": ["aiohomeconnect==0.12.3"]
+ "requirements": ["aiohomeconnect==0.16.3"],
+ "single_config_entry": true
}
diff --git a/homeassistant/components/home_connect/number.py b/homeassistant/components/home_connect/number.py
index aa0c4e4ae3f..99fe6c17296 100644
--- a/homeassistant/components/home_connect/number.py
+++ b/homeassistant/components/home_connect/number.py
@@ -3,7 +3,7 @@
import logging
from typing import cast
-from aiohomeconnect.model import GetSetting, SettingKey
+from aiohomeconnect.model import GetSetting, OptionKey, SettingKey
from aiohomeconnect.model.error import HomeConnectError
from homeassistant.components.number import (
@@ -13,22 +13,17 @@ from homeassistant.components.number import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import setup_home_connect_entry
-from .const import (
- DOMAIN,
- SVE_TRANSLATION_KEY_SET_SETTING,
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID,
- SVE_TRANSLATION_PLACEHOLDER_KEY,
- SVE_TRANSLATION_PLACEHOLDER_VALUE,
-)
+from .const import DOMAIN, UNIT_MAP
from .coordinator import HomeConnectApplianceData, HomeConnectConfigEntry
-from .entity import HomeConnectEntity
+from .entity import HomeConnectEntity, HomeConnectOptionEntity, constraint_fetcher
from .utils import get_dict_from_home_connect_error
_LOGGER = logging.getLogger(__name__)
+PARALLEL_UPDATES = 1
NUMBERS = (
NumberEntityDescription(
@@ -76,6 +71,47 @@ NUMBERS = (
device_class=NumberDeviceClass.TEMPERATURE,
translation_key="wine_compartment_3_setpoint_temperature",
),
+ NumberEntityDescription(
+ key=SettingKey.COOKING_HOOD_COLOR_TEMPERATURE_PERCENT,
+ translation_key="color_temperature_percent",
+ native_unit_of_measurement="%",
+ ),
+ NumberEntityDescription(
+ key=SettingKey.LAUNDRY_CARE_WASHER_I_DOS_1_BASE_LEVEL,
+ device_class=NumberDeviceClass.VOLUME,
+ translation_key="washer_i_dos_1_base_level",
+ ),
+ NumberEntityDescription(
+ key=SettingKey.LAUNDRY_CARE_WASHER_I_DOS_2_BASE_LEVEL,
+ device_class=NumberDeviceClass.VOLUME,
+ translation_key="washer_i_dos_2_base_level",
+ ),
+)
+
+NUMBER_OPTIONS = (
+ NumberEntityDescription(
+ key=OptionKey.BSH_COMMON_DURATION,
+ translation_key="duration",
+ ),
+ NumberEntityDescription(
+ key=OptionKey.BSH_COMMON_FINISH_IN_RELATIVE,
+ translation_key="finish_in_relative",
+ ),
+ NumberEntityDescription(
+ key=OptionKey.BSH_COMMON_START_IN_RELATIVE,
+ translation_key="start_in_relative",
+ ),
+ NumberEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_FILL_QUANTITY,
+ translation_key="fill_quantity",
+ device_class=NumberDeviceClass.VOLUME,
+ native_step=1,
+ ),
+ NumberEntityDescription(
+ key=OptionKey.COOKING_OVEN_SETPOINT_TEMPERATURE,
+ translation_key="setpoint_temperature",
+ device_class=NumberDeviceClass.TEMPERATURE,
+ ),
)
@@ -91,16 +127,29 @@ def _get_entities_for_appliance(
]
+def _get_option_entities_for_appliance(
+ entry: HomeConnectConfigEntry,
+ appliance: HomeConnectApplianceData,
+) -> list[HomeConnectOptionEntity]:
+ """Get a list of currently available option entities."""
+ return [
+ HomeConnectOptionNumberEntity(entry.runtime_data, appliance, description)
+ for description in NUMBER_OPTIONS
+ if description.key in appliance.options
+ ]
+
+
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeConnectConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Connect number."""
setup_home_connect_entry(
entry,
_get_entities_for_appliance,
async_add_entities,
+ _get_option_entities_for_appliance,
)
@@ -124,28 +173,34 @@ class HomeConnectNumberEntity(HomeConnectEntity, NumberEntity):
except HomeConnectError as err:
raise HomeAssistantError(
translation_domain=DOMAIN,
- translation_key=SVE_TRANSLATION_KEY_SET_SETTING,
+ translation_key="set_setting_entity",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
- SVE_TRANSLATION_PLACEHOLDER_KEY: self.bsh_key,
- SVE_TRANSLATION_PLACEHOLDER_VALUE: str(value),
+ "entity_id": self.entity_id,
+ "key": self.bsh_key,
+ "value": str(value),
},
) from err
+ @constraint_fetcher
async def async_fetch_constraints(self) -> None:
"""Fetch the max and min values and step for the number entity."""
- try:
+ setting_key = cast(SettingKey, self.bsh_key)
+ data = self.appliance.settings.get(setting_key)
+ if not data or not data.unit or not data.constraints:
data = await self.coordinator.client.get_setting(
- self.appliance.info.ha_id, setting_key=SettingKey(self.bsh_key)
+ self.appliance.info.ha_id, setting_key=setting_key
)
- except HomeConnectError as err:
- _LOGGER.error("An error occurred: %s", err)
- else:
+ if data.unit:
+ self._attr_native_unit_of_measurement = data.unit
self.set_constraints(data)
def set_constraints(self, setting: GetSetting) -> None:
"""Set constraints for the number entity."""
+ if setting.unit:
+ self._attr_native_unit_of_measurement = UNIT_MAP.get(
+ setting.unit, setting.unit
+ )
if not (constraints := setting.constraints):
return
if constraints.max:
@@ -166,11 +221,51 @@ class HomeConnectNumberEntity(HomeConnectEntity, NumberEntity):
"""When entity is added to hass."""
await super().async_added_to_hass()
data = self.appliance.settings[cast(SettingKey, self.bsh_key)]
- self._attr_native_unit_of_measurement = data.unit
self.set_constraints(data)
if (
- not hasattr(self, "_attr_native_min_value")
+ not hasattr(self, "_attr_native_unit_of_measurement")
+ or not hasattr(self, "_attr_native_min_value")
or not hasattr(self, "_attr_native_max_value")
or not hasattr(self, "_attr_native_step")
):
await self.async_fetch_constraints()
+
+
+class HomeConnectOptionNumberEntity(HomeConnectOptionEntity, NumberEntity):
+ """Number option class for Home Connect."""
+
+ async def async_set_native_value(self, value: float) -> None:
+ """Set the native value of the entity."""
+ await self.async_set_option(value)
+
+ def update_native_value(self) -> None:
+ """Set the value of the entity."""
+ self._attr_native_value = cast(float | None, self.option_value)
+ option_definition = self.appliance.options.get(self.bsh_key)
+ if option_definition:
+ if option_definition.unit:
+ candidate_unit = UNIT_MAP.get(
+ option_definition.unit, option_definition.unit
+ )
+ if (
+ not hasattr(self, "_attr_native_unit_of_measurement")
+ or candidate_unit != self._attr_native_unit_of_measurement
+ ):
+ self._attr_native_unit_of_measurement = candidate_unit
+ option_constraints = option_definition.constraints
+ if option_constraints:
+ if (
+ not hasattr(self, "_attr_native_min_value")
+ or self._attr_native_min_value != option_constraints.min
+ ) and option_constraints.min:
+ self._attr_native_min_value = option_constraints.min
+ if (
+ not hasattr(self, "_attr_native_max_value")
+ or self._attr_native_max_value != option_constraints.max
+ ) and option_constraints.max:
+ self._attr_native_max_value = option_constraints.max
+ if (
+ not hasattr(self, "_attr_native_step")
+ or self._attr_native_step != option_constraints.step_size
+ ) and option_constraints.step_size:
+ self._attr_native_step = option_constraints.step_size
diff --git a/homeassistant/components/home_connect/select.py b/homeassistant/components/home_connect/select.py
index 13518c5dea2..c82e0686cb5 100644
--- a/homeassistant/components/home_connect/select.py
+++ b/homeassistant/components/home_connect/select.py
@@ -2,36 +2,74 @@
from collections.abc import Callable, Coroutine
from dataclasses import dataclass
+import logging
from typing import Any, cast
from aiohomeconnect.client import Client as HomeConnectClient
-from aiohomeconnect.model import EventKey, ProgramKey
+from aiohomeconnect.model import EventKey, OptionKey, ProgramKey, SettingKey
from aiohomeconnect.model.error import HomeConnectError
from aiohomeconnect.model.program import Execution
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import setup_home_connect_entry
-from .const import APPLIANCES_WITH_PROGRAMS, DOMAIN, SVE_TRANSLATION_PLACEHOLDER_PROGRAM
+from .const import (
+ APPLIANCES_WITH_PROGRAMS,
+ AVAILABLE_MAPS_ENUM,
+ BEAN_AMOUNT_OPTIONS,
+ BEAN_CONTAINER_OPTIONS,
+ CLEANING_MODE_OPTIONS,
+ COFFEE_MILK_RATIO_OPTIONS,
+ COFFEE_TEMPERATURE_OPTIONS,
+ DOMAIN,
+ DRYING_TARGET_OPTIONS,
+ FLOW_RATE_OPTIONS,
+ HOT_WATER_TEMPERATURE_OPTIONS,
+ INTENSIVE_LEVEL_OPTIONS,
+ PROGRAMS_TRANSLATION_KEYS_MAP,
+ SPIN_SPEED_OPTIONS,
+ TEMPERATURE_OPTIONS,
+ TRANSLATION_KEYS_PROGRAMS_MAP,
+ VARIO_PERFECT_OPTIONS,
+ VENTING_LEVEL_OPTIONS,
+ WARMING_LEVEL_OPTIONS,
+)
from .coordinator import (
HomeConnectApplianceData,
HomeConnectConfigEntry,
HomeConnectCoordinator,
)
-from .entity import HomeConnectEntity
+from .entity import HomeConnectEntity, HomeConnectOptionEntity, constraint_fetcher
from .utils import bsh_key_to_translation_key, get_dict_from_home_connect_error
-TRANSLATION_KEYS_PROGRAMS_MAP = {
- bsh_key_to_translation_key(program.value): cast(ProgramKey, program)
- for program in ProgramKey
- if program != ProgramKey.UNKNOWN
+_LOGGER = logging.getLogger(__name__)
+
+PARALLEL_UPDATES = 1
+
+FUNCTIONAL_LIGHT_COLOR_TEMPERATURE_ENUM = {
+ bsh_key_to_translation_key(option): option
+ for option in (
+ "Cooking.Hood.EnumType.ColorTemperature.custom",
+ "Cooking.Hood.EnumType.ColorTemperature.warm",
+ "Cooking.Hood.EnumType.ColorTemperature.warmToNeutral",
+ "Cooking.Hood.EnumType.ColorTemperature.neutral",
+ "Cooking.Hood.EnumType.ColorTemperature.neutralToCold",
+ "Cooking.Hood.EnumType.ColorTemperature.cold",
+ )
}
-PROGRAMS_TRANSLATION_KEYS_MAP = {
- value: key for key, value in TRANSLATION_KEYS_PROGRAMS_MAP.items()
+AMBIENT_LIGHT_COLOR_TEMPERATURE_ENUM = {
+ **{
+ bsh_key_to_translation_key(option): option
+ for option in ("BSH.Common.EnumType.AmbientLightColor.CustomColor",)
+ },
+ **{
+ str(option): f"BSH.Common.EnumType.AmbientLightColor.Color{option}"
+ for option in range(1, 100)
+ },
}
@@ -48,6 +86,14 @@ class HomeConnectProgramSelectEntityDescription(
error_translation_key: str
+@dataclass(frozen=True, kw_only=True)
+class HomeConnectSelectEntityDescription(SelectEntityDescription):
+ """Entity Description class for settings and options that have enumeration values."""
+
+ translation_key_values: dict[str, str]
+ values_translation_key: dict[str, str]
+
+
PROGRAM_SELECT_ENTITY_DESCRIPTIONS = (
HomeConnectProgramSelectEntityDescription(
key=EventKey.BSH_COMMON_ROOT_ACTIVE_PROGRAM,
@@ -69,32 +115,238 @@ PROGRAM_SELECT_ENTITY_DESCRIPTIONS = (
),
)
+SELECT_ENTITY_DESCRIPTIONS = (
+ HomeConnectSelectEntityDescription(
+ key=SettingKey.CONSUMER_PRODUCTS_CLEANING_ROBOT_CURRENT_MAP,
+ translation_key="current_map",
+ options=list(AVAILABLE_MAPS_ENUM),
+ translation_key_values=AVAILABLE_MAPS_ENUM,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in AVAILABLE_MAPS_ENUM.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=SettingKey.COOKING_HOOD_COLOR_TEMPERATURE,
+ translation_key="functional_light_color_temperature",
+ options=list(FUNCTIONAL_LIGHT_COLOR_TEMPERATURE_ENUM),
+ translation_key_values=FUNCTIONAL_LIGHT_COLOR_TEMPERATURE_ENUM,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in FUNCTIONAL_LIGHT_COLOR_TEMPERATURE_ENUM.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=SettingKey.BSH_COMMON_AMBIENT_LIGHT_COLOR,
+ translation_key="ambient_light_color",
+ options=list(AMBIENT_LIGHT_COLOR_TEMPERATURE_ENUM),
+ translation_key_values=AMBIENT_LIGHT_COLOR_TEMPERATURE_ENUM,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in AMBIENT_LIGHT_COLOR_TEMPERATURE_ENUM.items()
+ },
+ ),
+)
+
+PROGRAM_SELECT_OPTION_ENTITY_DESCRIPTIONS = (
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_CLEANING_ROBOT_REFERENCE_MAP_ID,
+ translation_key="reference_map_id",
+ options=list(AVAILABLE_MAPS_ENUM),
+ translation_key_values=AVAILABLE_MAPS_ENUM,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in AVAILABLE_MAPS_ENUM.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_CLEANING_ROBOT_CLEANING_MODE,
+ translation_key="cleaning_mode",
+ options=list(CLEANING_MODE_OPTIONS),
+ translation_key_values=CLEANING_MODE_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in CLEANING_MODE_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEAN_AMOUNT,
+ translation_key="bean_amount",
+ options=list(BEAN_AMOUNT_OPTIONS),
+ translation_key_values=BEAN_AMOUNT_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in BEAN_AMOUNT_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_COFFEE_TEMPERATURE,
+ translation_key="coffee_temperature",
+ options=list(COFFEE_TEMPERATURE_OPTIONS),
+ translation_key_values=COFFEE_TEMPERATURE_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in COFFEE_TEMPERATURE_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEAN_CONTAINER_SELECTION,
+ translation_key="bean_container",
+ options=list(BEAN_CONTAINER_OPTIONS),
+ translation_key_values=BEAN_CONTAINER_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in BEAN_CONTAINER_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_FLOW_RATE,
+ translation_key="flow_rate",
+ options=list(FLOW_RATE_OPTIONS),
+ translation_key_values=FLOW_RATE_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in FLOW_RATE_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_COFFEE_MILK_RATIO,
+ translation_key="coffee_milk_ratio",
+ options=list(COFFEE_MILK_RATIO_OPTIONS),
+ translation_key_values=COFFEE_MILK_RATIO_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in FLOW_RATE_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_HOT_WATER_TEMPERATURE,
+ translation_key="hot_water_temperature",
+ options=list(HOT_WATER_TEMPERATURE_OPTIONS),
+ translation_key_values=HOT_WATER_TEMPERATURE_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in HOT_WATER_TEMPERATURE_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.LAUNDRY_CARE_DRYER_DRYING_TARGET,
+ translation_key="drying_target",
+ options=list(DRYING_TARGET_OPTIONS),
+ translation_key_values=DRYING_TARGET_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in DRYING_TARGET_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.COOKING_COMMON_HOOD_VENTING_LEVEL,
+ translation_key="venting_level",
+ options=list(VENTING_LEVEL_OPTIONS),
+ translation_key_values=VENTING_LEVEL_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in VENTING_LEVEL_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.COOKING_COMMON_HOOD_INTENSIVE_LEVEL,
+ translation_key="intensive_level",
+ options=list(INTENSIVE_LEVEL_OPTIONS),
+ translation_key_values=INTENSIVE_LEVEL_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in INTENSIVE_LEVEL_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.COOKING_OVEN_WARMING_LEVEL,
+ translation_key="warming_level",
+ options=list(WARMING_LEVEL_OPTIONS),
+ translation_key_values=WARMING_LEVEL_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in WARMING_LEVEL_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.LAUNDRY_CARE_WASHER_TEMPERATURE,
+ translation_key="washer_temperature",
+ options=list(TEMPERATURE_OPTIONS),
+ translation_key_values=TEMPERATURE_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in TEMPERATURE_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.LAUNDRY_CARE_WASHER_SPIN_SPEED,
+ translation_key="spin_speed",
+ options=list(SPIN_SPEED_OPTIONS),
+ translation_key_values=SPIN_SPEED_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in SPIN_SPEED_OPTIONS.items()
+ },
+ ),
+ HomeConnectSelectEntityDescription(
+ key=OptionKey.LAUNDRY_CARE_COMMON_VARIO_PERFECT,
+ translation_key="vario_perfect",
+ options=list(VARIO_PERFECT_OPTIONS),
+ translation_key_values=VARIO_PERFECT_OPTIONS,
+ values_translation_key={
+ value: translation_key
+ for translation_key, value in VARIO_PERFECT_OPTIONS.items()
+ },
+ ),
+)
+
def _get_entities_for_appliance(
entry: HomeConnectConfigEntry,
appliance: HomeConnectApplianceData,
) -> list[HomeConnectEntity]:
"""Get a list of entities."""
- return (
- [
- HomeConnectProgramSelectEntity(entry.runtime_data, appliance, desc)
- for desc in PROGRAM_SELECT_ENTITY_DESCRIPTIONS
- ]
- if appliance.info.type in APPLIANCES_WITH_PROGRAMS
- else []
- )
+ return [
+ *(
+ [
+ HomeConnectProgramSelectEntity(entry.runtime_data, appliance, desc)
+ for desc in PROGRAM_SELECT_ENTITY_DESCRIPTIONS
+ ]
+ if appliance.info.type in APPLIANCES_WITH_PROGRAMS
+ else []
+ ),
+ *[
+ HomeConnectSelectEntity(entry.runtime_data, appliance, desc)
+ for desc in SELECT_ENTITY_DESCRIPTIONS
+ if desc.key in appliance.settings
+ ],
+ ]
+
+
+def _get_option_entities_for_appliance(
+ entry: HomeConnectConfigEntry,
+ appliance: HomeConnectApplianceData,
+) -> list[HomeConnectOptionEntity]:
+ """Get a list of entities."""
+ return [
+ HomeConnectSelectOptionEntity(entry.runtime_data, appliance, desc)
+ for desc in PROGRAM_SELECT_OPTION_ENTITY_DESCRIPTIONS
+ if desc.key in appliance.options
+ ]
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeConnectConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Connect select entities."""
setup_home_connect_entry(
entry,
_get_entities_for_appliance,
async_add_entities,
+ _get_option_entities_for_appliance,
)
@@ -149,6 +401,133 @@ class HomeConnectProgramSelectEntity(HomeConnectEntity, SelectEntity):
translation_key=self.entity_description.error_translation_key,
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_PROGRAM: program_key.value,
+ "program": program_key.value,
},
) from err
+
+
+class HomeConnectSelectEntity(HomeConnectEntity, SelectEntity):
+ """Select setting class for Home Connect."""
+
+ entity_description: HomeConnectSelectEntityDescription
+ _original_option_keys: set[str | None]
+
+ def __init__(
+ self,
+ coordinator: HomeConnectCoordinator,
+ appliance: HomeConnectApplianceData,
+ desc: HomeConnectSelectEntityDescription,
+ ) -> None:
+ """Initialize the entity."""
+ self._original_option_keys = set(desc.values_translation_key)
+ super().__init__(
+ coordinator,
+ appliance,
+ desc,
+ )
+
+ async def async_select_option(self, option: str) -> None:
+ """Select new option."""
+ value = self.entity_description.translation_key_values[option]
+ try:
+ await self.coordinator.client.set_setting(
+ self.appliance.info.ha_id,
+ setting_key=cast(SettingKey, self.bsh_key),
+ value=value,
+ )
+ except HomeConnectError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_setting_entity",
+ translation_placeholders={
+ **get_dict_from_home_connect_error(err),
+ "entity_id": self.entity_id,
+ "key": self.bsh_key,
+ "value": value,
+ },
+ ) from err
+
+ def update_native_value(self) -> None:
+ """Set the value of the entity."""
+ data = self.appliance.settings[cast(SettingKey, self.bsh_key)]
+ self._attr_current_option = self.entity_description.values_translation_key.get(
+ data.value
+ )
+
+ async def async_added_to_hass(self) -> None:
+ """When entity is added to hass."""
+ await super().async_added_to_hass()
+ await self.async_fetch_options()
+
+ @constraint_fetcher
+ async def async_fetch_options(self) -> None:
+ """Fetch options from the API."""
+ setting = self.appliance.settings.get(cast(SettingKey, self.bsh_key))
+ if (
+ not setting
+ or not setting.constraints
+ or not setting.constraints.allowed_values
+ ):
+ setting = await self.coordinator.client.get_setting(
+ self.appliance.info.ha_id,
+ setting_key=cast(SettingKey, self.bsh_key),
+ )
+
+ if setting and setting.constraints and setting.constraints.allowed_values:
+ self._original_option_keys = set(setting.constraints.allowed_values)
+ self._attr_options = [
+ self.entity_description.values_translation_key[option]
+ for option in self._original_option_keys
+ if option is not None
+ and option in self.entity_description.values_translation_key
+ ]
+
+
+class HomeConnectSelectOptionEntity(HomeConnectOptionEntity, SelectEntity):
+ """Select option class for Home Connect."""
+
+ entity_description: HomeConnectSelectEntityDescription
+ _original_option_keys: set[str | None]
+
+ def __init__(
+ self,
+ coordinator: HomeConnectCoordinator,
+ appliance: HomeConnectApplianceData,
+ desc: HomeConnectSelectEntityDescription,
+ ) -> None:
+ """Initialize the entity."""
+ self._original_option_keys = set(desc.values_translation_key)
+ super().__init__(
+ coordinator,
+ appliance,
+ desc,
+ )
+
+ async def async_select_option(self, option: str) -> None:
+ """Select new option."""
+ await self.async_set_option(
+ self.entity_description.translation_key_values[option]
+ )
+
+ def update_native_value(self) -> None:
+ """Set the value of the entity."""
+ self._attr_current_option = (
+ self.entity_description.values_translation_key.get(
+ cast(str, self.option_value), None
+ )
+ if self.option_value is not None
+ else None
+ )
+ if (
+ (option_definition := self.appliance.options.get(self.bsh_key))
+ and (option_constraints := option_definition.constraints)
+ and option_constraints.allowed_values
+ and self._original_option_keys != set(option_constraints.allowed_values)
+ ):
+ self._original_option_keys = set(option_constraints.allowed_values)
+ self._attr_options = [
+ self.entity_description.values_translation_key[option]
+ for option in self._original_option_keys
+ if option is not None
+ and option in self.entity_description.values_translation_key
+ ]
diff --git a/homeassistant/components/home_connect/sensor.py b/homeassistant/components/home_connect/sensor.py
index 545df1d68b6..796af8260fc 100644
--- a/homeassistant/components/home_connect/sensor.py
+++ b/homeassistant/components/home_connect/sensor.py
@@ -2,6 +2,7 @@
from dataclasses import dataclass
from datetime import timedelta
+import logging
from typing import cast
from aiohomeconnect.model import EventKey, StatusKey
@@ -12,9 +13,9 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.const import PERCENTAGE, UnitOfTime, UnitOfVolume
+from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfVolume
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util, slugify
from .common import setup_home_connect_entry
@@ -23,9 +24,14 @@ from .const import (
BSH_OPERATION_STATE_FINISHED,
BSH_OPERATION_STATE_PAUSE,
BSH_OPERATION_STATE_RUN,
+ UNIT_MAP,
)
from .coordinator import HomeConnectApplianceData, HomeConnectConfigEntry
-from .entity import HomeConnectEntity
+from .entity import HomeConnectEntity, constraint_fetcher
+
+_LOGGER = logging.getLogger(__name__)
+
+PARALLEL_UPDATES = 0
EVENT_OPTIONS = ["confirmed", "off", "present"]
@@ -38,6 +44,7 @@ class HomeConnectSensorEntityDescription(
default_value: str | None = None
appliance_types: tuple[str, ...] | None = None
+ fetch_unit: bool = False
BSH_PROGRAM_SENSORS = (
@@ -56,12 +63,6 @@ BSH_PROGRAM_SENSORS = (
"WasherDryer",
),
),
- HomeConnectSensorEntityDescription(
- key=EventKey.BSH_COMMON_OPTION_DURATION,
- device_class=SensorDeviceClass.DURATION,
- native_unit_of_measurement=UnitOfTime.SECONDS,
- appliance_types=("Oven",),
- ),
HomeConnectSensorEntityDescription(
key=EventKey.BSH_COMMON_OPTION_PROGRAM_PROGRESS,
native_unit_of_measurement=PERCENTAGE,
@@ -99,16 +100,19 @@ SENSORS = (
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_COFFEE,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="coffee_counter",
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_POWDER_COFFEE,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="powder_coffee_counter",
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_HOT_WATER,
+ entity_category=EntityCategory.DIAGNOSTIC,
native_unit_of_measurement=UnitOfVolume.MILLILITERS,
device_class=SensorDeviceClass.VOLUME,
state_class=SensorStateClass.TOTAL_INCREASING,
@@ -116,31 +120,37 @@ SENSORS = (
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_HOT_WATER_CUPS,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="hot_water_cups_counter",
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_HOT_MILK,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="hot_milk_counter",
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_FROTHY_MILK,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="frothy_milk_counter",
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_MILK,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="milk_counter",
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_COFFEE_AND_MILK,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="coffee_and_milk_counter",
),
HomeConnectSensorEntityDescription(
key=StatusKey.CONSUMER_PRODUCTS_COFFEE_MAKER_BEVERAGE_COUNTER_RISTRETTO_ESPRESSO,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.TOTAL_INCREASING,
translation_key="ristretto_espresso_counter",
),
@@ -174,6 +184,13 @@ SENSORS = (
],
translation_key="last_selected_map",
),
+ HomeConnectSensorEntityDescription(
+ key=StatusKey.COOKING_OVEN_CURRENT_CAVITY_TEMPERATURE,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ translation_key="oven_current_cavity_temperature",
+ fetch_unit=True,
+ ),
)
EVENT_SENSORS = (
@@ -272,7 +289,7 @@ def _get_entities_for_appliance(
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeConnectConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Connect sensor."""
setup_home_connect_entry(
@@ -307,6 +324,27 @@ class HomeConnectSensor(HomeConnectEntity, SensorEntity):
case _:
self._attr_native_value = status
+ async def async_added_to_hass(self) -> None:
+ """When entity is added to hass."""
+ await super().async_added_to_hass()
+ if self.entity_description.fetch_unit:
+ data = self.appliance.status[cast(StatusKey, self.bsh_key)]
+ if data.unit:
+ self._attr_native_unit_of_measurement = UNIT_MAP.get(
+ data.unit, data.unit
+ )
+ else:
+ await self.fetch_unit()
+
+ @constraint_fetcher
+ async def fetch_unit(self) -> None:
+ """Fetch the unit of measurement."""
+ data = await self.coordinator.client.get_status_value(
+ self.appliance.info.ha_id, status_key=cast(StatusKey, self.bsh_key)
+ )
+ if data.unit:
+ self._attr_native_unit_of_measurement = UNIT_MAP.get(data.unit, data.unit)
+
class HomeConnectProgramSensor(HomeConnectSensor):
"""Sensor class for Home Connect sensors that reports information related to the running program."""
@@ -347,6 +385,13 @@ class HomeConnectProgramSensor(HomeConnectSensor):
def update_native_value(self) -> None:
"""Update the program sensor's status."""
+ self.program_running = (
+ status := self.appliance.status.get(StatusKey.BSH_COMMON_OPERATION_STATE)
+ ) is not None and status.value in [
+ BSH_OPERATION_STATE_RUN,
+ BSH_OPERATION_STATE_PAUSE,
+ BSH_OPERATION_STATE_FINISHED,
+ ]
event = self.appliance.events.get(cast(EventKey, self.bsh_key))
if event:
self._update_native_value(event.value)
diff --git a/homeassistant/components/home_connect/services.py b/homeassistant/components/home_connect/services.py
new file mode 100644
index 00000000000..fac1c5fe1a9
--- /dev/null
+++ b/homeassistant/components/home_connect/services.py
@@ -0,0 +1,572 @@
+"""Custom actions (previously known as services) for the Home Connect integration."""
+
+from __future__ import annotations
+
+from collections.abc import Awaitable
+from typing import Any, cast
+
+from aiohomeconnect.client import Client as HomeConnectClient
+from aiohomeconnect.model import (
+ ArrayOfOptions,
+ CommandKey,
+ Option,
+ OptionKey,
+ ProgramKey,
+ SettingKey,
+)
+from aiohomeconnect.model.error import HomeConnectError
+import voluptuous as vol
+
+from homeassistant.const import ATTR_DEVICE_ID
+from homeassistant.core import HomeAssistant, ServiceCall
+from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
+from homeassistant.helpers import config_validation as cv, device_registry as dr
+from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
+
+from .const import (
+ AFFECTS_TO_ACTIVE_PROGRAM,
+ AFFECTS_TO_SELECTED_PROGRAM,
+ ATTR_AFFECTS_TO,
+ ATTR_KEY,
+ ATTR_PROGRAM,
+ ATTR_UNIT,
+ ATTR_VALUE,
+ DOMAIN,
+ PROGRAM_ENUM_OPTIONS,
+ SERVICE_OPTION_ACTIVE,
+ SERVICE_OPTION_SELECTED,
+ SERVICE_PAUSE_PROGRAM,
+ SERVICE_RESUME_PROGRAM,
+ SERVICE_SELECT_PROGRAM,
+ SERVICE_SET_PROGRAM_AND_OPTIONS,
+ SERVICE_SETTING,
+ SERVICE_START_PROGRAM,
+ TRANSLATION_KEYS_PROGRAMS_MAP,
+)
+from .coordinator import HomeConnectConfigEntry
+from .utils import bsh_key_to_translation_key, get_dict_from_home_connect_error
+
+CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
+
+
+PROGRAM_OPTIONS = {
+ bsh_key_to_translation_key(key): (
+ key,
+ value,
+ )
+ for key, value in {
+ OptionKey.BSH_COMMON_DURATION: int,
+ OptionKey.BSH_COMMON_START_IN_RELATIVE: int,
+ OptionKey.BSH_COMMON_FINISH_IN_RELATIVE: int,
+ OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_FILL_QUANTITY: int,
+ OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_MULTIPLE_BEVERAGES: bool,
+ OptionKey.DISHCARE_DISHWASHER_INTENSIV_ZONE: bool,
+ OptionKey.DISHCARE_DISHWASHER_BRILLIANCE_DRY: bool,
+ OptionKey.DISHCARE_DISHWASHER_VARIO_SPEED_PLUS: bool,
+ OptionKey.DISHCARE_DISHWASHER_SILENCE_ON_DEMAND: bool,
+ OptionKey.DISHCARE_DISHWASHER_HALF_LOAD: bool,
+ OptionKey.DISHCARE_DISHWASHER_EXTRA_DRY: bool,
+ OptionKey.DISHCARE_DISHWASHER_HYGIENE_PLUS: bool,
+ OptionKey.DISHCARE_DISHWASHER_ECO_DRY: bool,
+ OptionKey.DISHCARE_DISHWASHER_ZEOLITE_DRY: bool,
+ OptionKey.COOKING_OVEN_SETPOINT_TEMPERATURE: int,
+ OptionKey.COOKING_OVEN_FAST_PRE_HEAT: bool,
+ OptionKey.LAUNDRY_CARE_WASHER_I_DOS_1_ACTIVE: bool,
+ OptionKey.LAUNDRY_CARE_WASHER_I_DOS_2_ACTIVE: bool,
+ }.items()
+}
+
+
+SERVICE_SETTING_SCHEMA = vol.Schema(
+ {
+ vol.Required(ATTR_DEVICE_ID): str,
+ vol.Required(ATTR_KEY): vol.All(
+ vol.Coerce(SettingKey),
+ vol.NotIn([SettingKey.UNKNOWN]),
+ ),
+ vol.Required(ATTR_VALUE): vol.Any(str, int, bool),
+ }
+)
+
+# DEPRECATED: Remove in 2025.9.0
+SERVICE_OPTION_SCHEMA = vol.Schema(
+ {
+ vol.Required(ATTR_DEVICE_ID): str,
+ vol.Required(ATTR_KEY): vol.All(
+ vol.Coerce(OptionKey),
+ vol.NotIn([OptionKey.UNKNOWN]),
+ ),
+ vol.Required(ATTR_VALUE): vol.Any(str, int, bool),
+ vol.Optional(ATTR_UNIT): str,
+ }
+)
+
+# DEPRECATED: Remove in 2025.9.0
+SERVICE_PROGRAM_SCHEMA = vol.Any(
+ {
+ vol.Required(ATTR_DEVICE_ID): str,
+ vol.Required(ATTR_PROGRAM): vol.All(
+ vol.Coerce(ProgramKey),
+ vol.NotIn([ProgramKey.UNKNOWN]),
+ ),
+ vol.Required(ATTR_KEY): vol.All(
+ vol.Coerce(OptionKey),
+ vol.NotIn([OptionKey.UNKNOWN]),
+ ),
+ vol.Required(ATTR_VALUE): vol.Any(int, str),
+ vol.Optional(ATTR_UNIT): str,
+ },
+ {
+ vol.Required(ATTR_DEVICE_ID): str,
+ vol.Required(ATTR_PROGRAM): vol.All(
+ vol.Coerce(ProgramKey),
+ vol.NotIn([ProgramKey.UNKNOWN]),
+ ),
+ },
+)
+
+
+def _require_program_or_at_least_one_option(data: dict) -> dict:
+ if ATTR_PROGRAM not in data and not any(
+ option_key in data for option_key in (PROGRAM_ENUM_OPTIONS | PROGRAM_OPTIONS)
+ ):
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="required_program_or_one_option_at_least",
+ )
+ return data
+
+
+SERVICE_PROGRAM_AND_OPTIONS_SCHEMA = vol.All(
+ vol.Schema(
+ {
+ vol.Required(ATTR_DEVICE_ID): str,
+ vol.Required(ATTR_AFFECTS_TO): vol.In(
+ [AFFECTS_TO_ACTIVE_PROGRAM, AFFECTS_TO_SELECTED_PROGRAM]
+ ),
+ vol.Optional(ATTR_PROGRAM): vol.In(TRANSLATION_KEYS_PROGRAMS_MAP.keys()),
+ }
+ )
+ .extend(
+ {
+ vol.Optional(translation_key): vol.In(allowed_values.keys())
+ for translation_key, (
+ key,
+ allowed_values,
+ ) in PROGRAM_ENUM_OPTIONS.items()
+ }
+ )
+ .extend(
+ {
+ vol.Optional(translation_key): schema
+ for translation_key, (key, schema) in PROGRAM_OPTIONS.items()
+ }
+ ),
+ _require_program_or_at_least_one_option,
+)
+
+SERVICE_COMMAND_SCHEMA = vol.Schema({vol.Required(ATTR_DEVICE_ID): str})
+
+
+async def _get_client_and_ha_id(
+ hass: HomeAssistant, device_id: str
+) -> tuple[HomeConnectClient, str]:
+ device_registry = dr.async_get(hass)
+ device_entry = device_registry.async_get(device_id)
+ if device_entry is None:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="device_entry_not_found",
+ translation_placeholders={
+ "device_id": device_id,
+ },
+ )
+ entry: HomeConnectConfigEntry | None = None
+ for entry_id in device_entry.config_entries:
+ _entry = hass.config_entries.async_get_entry(entry_id)
+ assert _entry
+ if _entry.domain == DOMAIN:
+ entry = cast(HomeConnectConfigEntry, _entry)
+ break
+ if entry is None:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="config_entry_not_found",
+ translation_placeholders={
+ "device_id": device_id,
+ },
+ )
+
+ ha_id = next(
+ (
+ identifier[1]
+ for identifier in device_entry.identifiers
+ if identifier[0] == DOMAIN
+ ),
+ None,
+ )
+ if ha_id is None:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="appliance_not_found",
+ translation_placeholders={
+ "device_id": device_id,
+ },
+ )
+ return entry.runtime_data.client, ha_id
+
+
+async def _async_service_program(call: ServiceCall, start: bool) -> None:
+ """Execute calls to services taking a program."""
+ program = call.data[ATTR_PROGRAM]
+ client, ha_id = await _get_client_and_ha_id(call.hass, call.data[ATTR_DEVICE_ID])
+
+ option_key = call.data.get(ATTR_KEY)
+ options = (
+ [
+ Option(
+ option_key,
+ call.data[ATTR_VALUE],
+ unit=call.data.get(ATTR_UNIT),
+ )
+ ]
+ if option_key is not None
+ else None
+ )
+
+ async_create_issue(
+ call.hass,
+ DOMAIN,
+ "deprecated_set_program_and_option_actions",
+ breaks_in_ha_version="2025.9.0",
+ is_fixable=True,
+ is_persistent=True,
+ severity=IssueSeverity.WARNING,
+ translation_key="deprecated_set_program_and_option_actions",
+ translation_placeholders={
+ "new_action_key": SERVICE_SET_PROGRAM_AND_OPTIONS,
+ "remove_release": "2025.9.0",
+ "deprecated_action_yaml": "\n".join(
+ [
+ "```yaml",
+ f"action: {DOMAIN}.{SERVICE_START_PROGRAM if start else SERVICE_SELECT_PROGRAM}",
+ "data:",
+ f" {ATTR_DEVICE_ID}: DEVICE_ID",
+ f" {ATTR_PROGRAM}: {program}",
+ *([f" {ATTR_KEY}: {options[0].key}"] if options else []),
+ *([f" {ATTR_VALUE}: {options[0].value}"] if options else []),
+ *(
+ [f" {ATTR_UNIT}: {options[0].unit}"]
+ if options and options[0].unit
+ else []
+ ),
+ "```",
+ ]
+ ),
+ "new_action_yaml": "\n ".join(
+ [
+ "```yaml",
+ f"action: {DOMAIN}.{SERVICE_SET_PROGRAM_AND_OPTIONS}",
+ "data:",
+ f" {ATTR_DEVICE_ID}: DEVICE_ID",
+ f" {ATTR_AFFECTS_TO}: {AFFECTS_TO_ACTIVE_PROGRAM if start else AFFECTS_TO_SELECTED_PROGRAM}",
+ f" {ATTR_PROGRAM}: {bsh_key_to_translation_key(program.value)}",
+ *(
+ [
+ f" {bsh_key_to_translation_key(options[0].key)}: {options[0].value}"
+ ]
+ if options
+ else []
+ ),
+ "```",
+ ]
+ ),
+ "repo_link": "[aiohomeconnect](https://github.com/MartinHjelmare/aiohomeconnect)",
+ },
+ )
+
+ try:
+ if start:
+ await client.start_program(ha_id, program_key=program, options=options)
+ else:
+ await client.set_selected_program(
+ ha_id, program_key=program, options=options
+ )
+ except HomeConnectError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="start_program" if start else "select_program",
+ translation_placeholders={
+ **get_dict_from_home_connect_error(err),
+ "program": program,
+ },
+ ) from err
+
+
+async def _async_service_set_program_options(call: ServiceCall, active: bool) -> None:
+ """Execute calls to services taking a program."""
+ option_key = call.data[ATTR_KEY]
+ value = call.data[ATTR_VALUE]
+ unit = call.data.get(ATTR_UNIT)
+ client, ha_id = await _get_client_and_ha_id(call.hass, call.data[ATTR_DEVICE_ID])
+
+ async_create_issue(
+ call.hass,
+ DOMAIN,
+ "deprecated_set_program_and_option_actions",
+ breaks_in_ha_version="2025.9.0",
+ is_fixable=True,
+ is_persistent=True,
+ severity=IssueSeverity.WARNING,
+ translation_key="deprecated_set_program_and_option_actions",
+ translation_placeholders={
+ "new_action_key": SERVICE_SET_PROGRAM_AND_OPTIONS,
+ "remove_release": "2025.9.0",
+ "deprecated_action_yaml": "\n".join(
+ [
+ "```yaml",
+ f"action: {DOMAIN}.{SERVICE_OPTION_ACTIVE if active else SERVICE_OPTION_SELECTED}",
+ "data:",
+ f" {ATTR_DEVICE_ID}: DEVICE_ID",
+ f" {ATTR_KEY}: {option_key}",
+ f" {ATTR_VALUE}: {value}",
+ *([f" {ATTR_UNIT}: {unit}"] if unit else []),
+ "```",
+ ]
+ ),
+ "new_action_yaml": "\n ".join(
+ [
+ "```yaml",
+ f"action: {DOMAIN}.{SERVICE_SET_PROGRAM_AND_OPTIONS}",
+ "data:",
+ f" {ATTR_DEVICE_ID}: DEVICE_ID",
+ f" {ATTR_AFFECTS_TO}: {AFFECTS_TO_ACTIVE_PROGRAM if active else AFFECTS_TO_SELECTED_PROGRAM}",
+ f" {bsh_key_to_translation_key(option_key)}: {value}",
+ "```",
+ ]
+ ),
+ "repo_link": "[aiohomeconnect](https://github.com/MartinHjelmare/aiohomeconnect)",
+ },
+ )
+ try:
+ if active:
+ await client.set_active_program_option(
+ ha_id,
+ option_key=option_key,
+ value=value,
+ unit=unit,
+ )
+ else:
+ await client.set_selected_program_option(
+ ha_id,
+ option_key=option_key,
+ value=value,
+ unit=unit,
+ )
+ except HomeConnectError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_options_active_program"
+ if active
+ else "set_options_selected_program",
+ translation_placeholders={
+ **get_dict_from_home_connect_error(err),
+ "key": option_key,
+ "value": str(value),
+ },
+ ) from err
+
+
+async def _async_service_command(call: ServiceCall, command_key: CommandKey) -> None:
+ """Execute calls to services executing a command."""
+ client, ha_id = await _get_client_and_ha_id(call.hass, call.data[ATTR_DEVICE_ID])
+
+ async_create_issue(
+ call.hass,
+ DOMAIN,
+ "deprecated_command_actions",
+ breaks_in_ha_version="2025.9.0",
+ is_fixable=True,
+ is_persistent=True,
+ severity=IssueSeverity.WARNING,
+ translation_key="deprecated_command_actions",
+ )
+
+ try:
+ await client.put_command(ha_id, command_key=command_key, value=True)
+ except HomeConnectError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="execute_command",
+ translation_placeholders={
+ **get_dict_from_home_connect_error(err),
+ "command": command_key.value,
+ },
+ ) from err
+
+
+async def async_service_option_active(call: ServiceCall) -> None:
+ """Service for setting an option for an active program."""
+ await _async_service_set_program_options(call, True)
+
+
+async def async_service_option_selected(call: ServiceCall) -> None:
+ """Service for setting an option for a selected program."""
+ await _async_service_set_program_options(call, False)
+
+
+async def async_service_setting(call: ServiceCall) -> None:
+ """Service for changing a setting."""
+ key = call.data[ATTR_KEY]
+ value = call.data[ATTR_VALUE]
+ client, ha_id = await _get_client_and_ha_id(call.hass, call.data[ATTR_DEVICE_ID])
+
+ try:
+ await client.set_setting(ha_id, setting_key=key, value=value)
+ except HomeConnectError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="set_setting",
+ translation_placeholders={
+ **get_dict_from_home_connect_error(err),
+ "key": key,
+ "value": str(value),
+ },
+ ) from err
+
+
+async def async_service_pause_program(call: ServiceCall) -> None:
+ """Service for pausing a program."""
+ await _async_service_command(call, CommandKey.BSH_COMMON_PAUSE_PROGRAM)
+
+
+async def async_service_resume_program(call: ServiceCall) -> None:
+ """Service for resuming a paused program."""
+ await _async_service_command(call, CommandKey.BSH_COMMON_RESUME_PROGRAM)
+
+
+async def async_service_select_program(call: ServiceCall) -> None:
+ """Service for selecting a program."""
+ await _async_service_program(call, False)
+
+
+async def async_service_set_program_and_options(call: ServiceCall) -> None:
+ """Service for setting a program and options."""
+ data = dict(call.data)
+ program = data.pop(ATTR_PROGRAM, None)
+ affects_to = data.pop(ATTR_AFFECTS_TO)
+ client, ha_id = await _get_client_and_ha_id(call.hass, data.pop(ATTR_DEVICE_ID))
+
+ options: list[Option] = []
+
+ for option, value in data.items():
+ if option in PROGRAM_ENUM_OPTIONS:
+ options.append(
+ Option(
+ PROGRAM_ENUM_OPTIONS[option][0],
+ PROGRAM_ENUM_OPTIONS[option][1][value],
+ )
+ )
+ elif option in PROGRAM_OPTIONS:
+ option_key = PROGRAM_OPTIONS[option][0]
+ options.append(Option(option_key, value))
+
+ method_call: Awaitable[Any]
+ exception_translation_key: str
+ if program:
+ program = (
+ program
+ if isinstance(program, ProgramKey)
+ else TRANSLATION_KEYS_PROGRAMS_MAP[program]
+ )
+
+ if affects_to == AFFECTS_TO_ACTIVE_PROGRAM:
+ method_call = client.start_program(
+ ha_id, program_key=program, options=options
+ )
+ exception_translation_key = "start_program"
+ elif affects_to == AFFECTS_TO_SELECTED_PROGRAM:
+ method_call = client.set_selected_program(
+ ha_id, program_key=program, options=options
+ )
+ exception_translation_key = "select_program"
+ else:
+ array_of_options = ArrayOfOptions(options)
+ if affects_to == AFFECTS_TO_ACTIVE_PROGRAM:
+ method_call = client.set_active_program_options(
+ ha_id, array_of_options=array_of_options
+ )
+ exception_translation_key = "set_options_active_program"
+ else:
+ # affects_to is AFFECTS_TO_SELECTED_PROGRAM
+ method_call = client.set_selected_program_options(
+ ha_id, array_of_options=array_of_options
+ )
+ exception_translation_key = "set_options_selected_program"
+
+ try:
+ await method_call
+ except HomeConnectError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key=exception_translation_key,
+ translation_placeholders={
+ **get_dict_from_home_connect_error(err),
+ **({"program": program} if program else {}),
+ },
+ ) from err
+
+
+async def async_service_start_program(call: ServiceCall) -> None:
+ """Service for starting a program."""
+ await _async_service_program(call, True)
+
+
+def register_actions(hass: HomeAssistant) -> None:
+ """Register custom actions."""
+
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_OPTION_ACTIVE,
+ async_service_option_active,
+ schema=SERVICE_OPTION_SCHEMA,
+ )
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_OPTION_SELECTED,
+ async_service_option_selected,
+ schema=SERVICE_OPTION_SCHEMA,
+ )
+ hass.services.async_register(
+ DOMAIN, SERVICE_SETTING, async_service_setting, schema=SERVICE_SETTING_SCHEMA
+ )
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_PAUSE_PROGRAM,
+ async_service_pause_program,
+ schema=SERVICE_COMMAND_SCHEMA,
+ )
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_RESUME_PROGRAM,
+ async_service_resume_program,
+ schema=SERVICE_COMMAND_SCHEMA,
+ )
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_SELECT_PROGRAM,
+ async_service_select_program,
+ schema=SERVICE_PROGRAM_SCHEMA,
+ )
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_START_PROGRAM,
+ async_service_start_program,
+ schema=SERVICE_PROGRAM_SCHEMA,
+ )
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_SET_PROGRAM_AND_OPTIONS,
+ async_service_set_program_and_options,
+ schema=SERVICE_PROGRAM_AND_OPTIONS_SCHEMA,
+ )
diff --git a/homeassistant/components/home_connect/services.yaml b/homeassistant/components/home_connect/services.yaml
index 0738b58595a..2b53090fd34 100644
--- a/homeassistant/components/home_connect/services.yaml
+++ b/homeassistant/components/home_connect/services.yaml
@@ -46,6 +46,560 @@ select_program:
example: "seconds"
selector:
text:
+set_program_and_options:
+ fields:
+ device_id:
+ required: true
+ selector:
+ device:
+ integration: home_connect
+ affects_to:
+ example: active_program
+ required: true
+ selector:
+ select:
+ translation_key: affects_to
+ options:
+ - active_program
+ - selected_program
+ program:
+ example: dishcare_dishwasher_program_auto2
+ required: true
+ selector:
+ select:
+ mode: dropdown
+ custom_value: false
+ translation_key: programs
+ options:
+ - consumer_products_cleaning_robot_program_cleaning_clean_all
+ - consumer_products_cleaning_robot_program_cleaning_clean_map
+ - consumer_products_cleaning_robot_program_basic_go_home
+ - consumer_products_coffee_maker_program_beverage_ristretto
+ - consumer_products_coffee_maker_program_beverage_espresso
+ - consumer_products_coffee_maker_program_beverage_espresso_doppio
+ - consumer_products_coffee_maker_program_beverage_coffee
+ - consumer_products_coffee_maker_program_beverage_x_l_coffee
+ - consumer_products_coffee_maker_program_beverage_caffe_grande
+ - consumer_products_coffee_maker_program_beverage_espresso_macchiato
+ - consumer_products_coffee_maker_program_beverage_cappuccino
+ - consumer_products_coffee_maker_program_beverage_latte_macchiato
+ - consumer_products_coffee_maker_program_beverage_caffe_latte
+ - consumer_products_coffee_maker_program_beverage_milk_froth
+ - consumer_products_coffee_maker_program_beverage_warm_milk
+ - consumer_products_coffee_maker_program_coffee_world_kleiner_brauner
+ - consumer_products_coffee_maker_program_coffee_world_grosser_brauner
+ - consumer_products_coffee_maker_program_coffee_world_verlaengerter
+ - consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun
+ - consumer_products_coffee_maker_program_coffee_world_wiener_melange
+ - consumer_products_coffee_maker_program_coffee_world_flat_white
+ - consumer_products_coffee_maker_program_coffee_world_cortado
+ - consumer_products_coffee_maker_program_coffee_world_cafe_cortado
+ - consumer_products_coffee_maker_program_coffee_world_cafe_con_leche
+ - consumer_products_coffee_maker_program_coffee_world_cafe_au_lait
+ - consumer_products_coffee_maker_program_coffee_world_doppio
+ - consumer_products_coffee_maker_program_coffee_world_kaapi
+ - consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd
+ - consumer_products_coffee_maker_program_coffee_world_galao
+ - consumer_products_coffee_maker_program_coffee_world_garoto
+ - consumer_products_coffee_maker_program_coffee_world_americano
+ - consumer_products_coffee_maker_program_coffee_world_red_eye
+ - consumer_products_coffee_maker_program_coffee_world_black_eye
+ - consumer_products_coffee_maker_program_coffee_world_dead_eye
+ - consumer_products_coffee_maker_program_beverage_hot_water
+ - dishcare_dishwasher_program_pre_rinse
+ - dishcare_dishwasher_program_auto_1
+ - dishcare_dishwasher_program_auto_2
+ - dishcare_dishwasher_program_auto_3
+ - dishcare_dishwasher_program_eco_50
+ - dishcare_dishwasher_program_quick_45
+ - dishcare_dishwasher_program_intensiv_70
+ - dishcare_dishwasher_program_normal_65
+ - dishcare_dishwasher_program_glas_40
+ - dishcare_dishwasher_program_glass_care
+ - dishcare_dishwasher_program_night_wash
+ - dishcare_dishwasher_program_quick_65
+ - dishcare_dishwasher_program_normal_45
+ - dishcare_dishwasher_program_intensiv_45
+ - dishcare_dishwasher_program_auto_half_load
+ - dishcare_dishwasher_program_intensiv_power
+ - dishcare_dishwasher_program_magic_daily
+ - dishcare_dishwasher_program_super_60
+ - dishcare_dishwasher_program_kurz_60
+ - dishcare_dishwasher_program_express_sparkle_65
+ - dishcare_dishwasher_program_machine_care
+ - dishcare_dishwasher_program_steam_fresh
+ - dishcare_dishwasher_program_maximum_cleaning
+ - dishcare_dishwasher_program_mixed_load
+ - laundry_care_dryer_program_cotton
+ - laundry_care_dryer_program_synthetic
+ - laundry_care_dryer_program_mix
+ - laundry_care_dryer_program_blankets
+ - laundry_care_dryer_program_business_shirts
+ - laundry_care_dryer_program_down_feathers
+ - laundry_care_dryer_program_hygiene
+ - laundry_care_dryer_program_jeans
+ - laundry_care_dryer_program_outdoor
+ - laundry_care_dryer_program_synthetic_refresh
+ - laundry_care_dryer_program_towels
+ - laundry_care_dryer_program_delicates
+ - laundry_care_dryer_program_super_40
+ - laundry_care_dryer_program_shirts_15
+ - laundry_care_dryer_program_pillow
+ - laundry_care_dryer_program_anti_shrink
+ - laundry_care_dryer_program_my_time_my_drying_time
+ - laundry_care_dryer_program_time_cold
+ - laundry_care_dryer_program_time_warm
+ - laundry_care_dryer_program_in_basket
+ - laundry_care_dryer_program_time_cold_fix_time_cold_20
+ - laundry_care_dryer_program_time_cold_fix_time_cold_30
+ - laundry_care_dryer_program_time_cold_fix_time_cold_60
+ - laundry_care_dryer_program_time_warm_fix_time_warm_30
+ - laundry_care_dryer_program_time_warm_fix_time_warm_40
+ - laundry_care_dryer_program_time_warm_fix_time_warm_60
+ - laundry_care_dryer_program_dessous
+ - cooking_common_program_hood_automatic
+ - cooking_common_program_hood_venting
+ - cooking_common_program_hood_delayed_shut_off
+ - cooking_oven_program_heating_mode_pre_heating
+ - cooking_oven_program_heating_mode_hot_air
+ - cooking_oven_program_heating_mode_hot_air_eco
+ - cooking_oven_program_heating_mode_hot_air_grilling
+ - cooking_oven_program_heating_mode_top_bottom_heating
+ - cooking_oven_program_heating_mode_top_bottom_heating_eco
+ - cooking_oven_program_heating_mode_bottom_heating
+ - cooking_oven_program_heating_mode_pizza_setting
+ - cooking_oven_program_heating_mode_slow_cook
+ - cooking_oven_program_heating_mode_intensive_heat
+ - cooking_oven_program_heating_mode_keep_warm
+ - cooking_oven_program_heating_mode_preheat_ovenware
+ - cooking_oven_program_heating_mode_frozen_heatup_special
+ - cooking_oven_program_heating_mode_desiccation
+ - cooking_oven_program_heating_mode_defrost
+ - cooking_oven_program_heating_mode_proof
+ - cooking_oven_program_heating_mode_hot_air_30_steam
+ - cooking_oven_program_heating_mode_hot_air_60_steam
+ - cooking_oven_program_heating_mode_hot_air_80_steam
+ - cooking_oven_program_heating_mode_hot_air_100_steam
+ - cooking_oven_program_heating_mode_sabbath_programme
+ - cooking_oven_program_microwave_90_watt
+ - cooking_oven_program_microwave_180_watt
+ - cooking_oven_program_microwave_360_watt
+ - cooking_oven_program_microwave_600_watt
+ - cooking_oven_program_microwave_900_watt
+ - cooking_oven_program_microwave_1000_watt
+ - cooking_oven_program_microwave_max
+ - cooking_oven_program_heating_mode_warming_drawer
+ - laundry_care_washer_program_cotton
+ - laundry_care_washer_program_cotton_cotton_eco
+ - laundry_care_washer_program_cotton_eco_4060
+ - laundry_care_washer_program_cotton_colour
+ - laundry_care_washer_program_easy_care
+ - laundry_care_washer_program_mix
+ - laundry_care_washer_program_mix_night_wash
+ - laundry_care_washer_program_delicates_silk
+ - laundry_care_washer_program_wool
+ - laundry_care_washer_program_sensitive
+ - laundry_care_washer_program_auto_30
+ - laundry_care_washer_program_auto_40
+ - laundry_care_washer_program_auto_60
+ - laundry_care_washer_program_chiffon
+ - laundry_care_washer_program_curtains
+ - laundry_care_washer_program_dark_wash
+ - laundry_care_washer_program_dessous
+ - laundry_care_washer_program_monsoon
+ - laundry_care_washer_program_outdoor
+ - laundry_care_washer_program_plush_toy
+ - laundry_care_washer_program_shirts_blouses
+ - laundry_care_washer_program_sport_fitness
+ - laundry_care_washer_program_towels
+ - laundry_care_washer_program_water_proof
+ - laundry_care_washer_program_power_speed_59
+ - laundry_care_washer_program_super_153045_super_15
+ - laundry_care_washer_program_super_153045_super_1530
+ - laundry_care_washer_program_down_duvet_duvet
+ - laundry_care_washer_program_rinse_rinse_spin_drain
+ - laundry_care_washer_program_drum_clean
+ - laundry_care_washer_dryer_program_cotton
+ - laundry_care_washer_dryer_program_cotton_eco_4060
+ - laundry_care_washer_dryer_program_mix
+ - laundry_care_washer_dryer_program_easy_care
+ - laundry_care_washer_dryer_program_wash_and_dry_60
+ - laundry_care_washer_dryer_program_wash_and_dry_90
+ cleaning_robot_options:
+ collapsed: true
+ fields:
+ consumer_products_cleaning_robot_option_reference_map_id:
+ example: consumer_products_cleaning_robot_enum_type_available_maps_map1
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: available_maps
+ options:
+ - consumer_products_cleaning_robot_enum_type_available_maps_temp_map
+ - consumer_products_cleaning_robot_enum_type_available_maps_map1
+ - consumer_products_cleaning_robot_enum_type_available_maps_map2
+ - consumer_products_cleaning_robot_enum_type_available_maps_map3
+ consumer_products_cleaning_robot_option_cleaning_mode:
+ example: consumer_products_cleaning_robot_enum_type_cleaning_modes_standard
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: cleaning_mode
+ options:
+ - consumer_products_cleaning_robot_enum_type_cleaning_modes_silent
+ - consumer_products_cleaning_robot_enum_type_cleaning_modes_standard
+ - consumer_products_cleaning_robot_enum_type_cleaning_modes_power
+ coffee_maker_options:
+ collapsed: true
+ fields:
+ consumer_products_coffee_maker_option_bean_amount:
+ example: consumer_products_coffee_maker_enum_type_bean_amount_normal
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: bean_amount
+ options:
+ - consumer_products_coffee_maker_enum_type_bean_amount_very_mild
+ - consumer_products_coffee_maker_enum_type_bean_amount_mild
+ - consumer_products_coffee_maker_enum_type_bean_amount_mild_plus
+ - consumer_products_coffee_maker_enum_type_bean_amount_normal
+ - consumer_products_coffee_maker_enum_type_bean_amount_normal_plus
+ - consumer_products_coffee_maker_enum_type_bean_amount_strong
+ - consumer_products_coffee_maker_enum_type_bean_amount_strong_plus
+ - consumer_products_coffee_maker_enum_type_bean_amount_very_strong
+ - consumer_products_coffee_maker_enum_type_bean_amount_very_strong_plus
+ - consumer_products_coffee_maker_enum_type_bean_amount_extra_strong
+ - consumer_products_coffee_maker_enum_type_bean_amount_double_shot
+ - consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus
+ - consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus_plus
+ - consumer_products_coffee_maker_enum_type_bean_amount_triple_shot
+ - consumer_products_coffee_maker_enum_type_bean_amount_triple_shot_plus
+ - consumer_products_coffee_maker_enum_type_bean_amount_coffee_ground
+ consumer_products_coffee_maker_option_fill_quantity:
+ example: 60
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ mode: box
+ unit_of_measurement: ml
+ consumer_products_coffee_maker_option_coffee_temperature:
+ example: consumer_products_coffee_maker_enum_type_coffee_temperature_88_c
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: coffee_temperature
+ options:
+ - consumer_products_coffee_maker_enum_type_coffee_temperature_88_c
+ - consumer_products_coffee_maker_enum_type_coffee_temperature_90_c
+ - consumer_products_coffee_maker_enum_type_coffee_temperature_92_c
+ - consumer_products_coffee_maker_enum_type_coffee_temperature_94_c
+ - consumer_products_coffee_maker_enum_type_coffee_temperature_95_c
+ - consumer_products_coffee_maker_enum_type_coffee_temperature_96_c
+ consumer_products_coffee_maker_option_bean_container:
+ example: consumer_products_coffee_maker_enum_type_bean_container_selection_right
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: bean_container
+ options:
+ - consumer_products_coffee_maker_enum_type_bean_container_selection_right
+ - consumer_products_coffee_maker_enum_type_bean_container_selection_left
+ consumer_products_coffee_maker_option_flow_rate:
+ example: consumer_products_coffee_maker_enum_type_flow_rate_normal
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: flow_rate
+ options:
+ - consumer_products_coffee_maker_enum_type_flow_rate_normal
+ - consumer_products_coffee_maker_enum_type_flow_rate_intense
+ - consumer_products_coffee_maker_enum_type_flow_rate_intense_plus
+ consumer_products_coffee_maker_option_multiple_beverages:
+ example: false
+ required: false
+ selector:
+ boolean:
+ consumer_products_coffee_maker_option_coffee_milk_ratio:
+ example: consumer_products_coffee_maker_enum_type_coffee_milk_ratio_50_percent
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: coffee_milk_ratio
+ options:
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_10_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_20_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_25_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_30_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_40_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_50_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_55_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_60_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_65_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_67_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_70_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_75_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_80_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_85_percent
+ - consumer_products_coffee_maker_enum_type_coffee_milk_ratio_90_percent
+ consumer_products_coffee_maker_option_hot_water_temperature:
+ example: consumer_products_coffee_maker_enum_type_hot_water_temperature_50_c
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: hot_water_temperature
+ options:
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_white_tea
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_green_tea
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_black_tea
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_50_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_55_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_60_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_65_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_70_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_75_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_80_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_85_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_90_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_95_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_97_c
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_122_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_131_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_140_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_149_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_158_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_167_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_176_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_185_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_194_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_203_f
+ - consumer_products_coffee_maker_enum_type_hot_water_temperature_max
+ dish_washer_options:
+ collapsed: true
+ fields:
+ b_s_h_common_option_start_in_relative:
+ example: 3600
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ mode: box
+ unit_of_measurement: s
+ dishcare_dishwasher_option_intensiv_zone:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_brilliance_dry:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_vario_speed_plus:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_silence_on_demand:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_half_load:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_extra_dry:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_hygiene_plus:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_eco_dry:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dishcare_dishwasher_option_zeolite_dry:
+ example: false
+ required: false
+ selector:
+ boolean:
+ dryer_options:
+ collapsed: true
+ fields:
+ laundry_care_dryer_option_drying_target:
+ example: laundry_care_dryer_enum_type_drying_target_iron_dry
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: drying_target
+ options:
+ - laundry_care_dryer_enum_type_drying_target_iron_dry
+ - laundry_care_dryer_enum_type_drying_target_gentle_dry
+ - laundry_care_dryer_enum_type_drying_target_cupboard_dry
+ - laundry_care_dryer_enum_type_drying_target_cupboard_dry_plus
+ - laundry_care_dryer_enum_type_drying_target_extra_dry
+ hood_options:
+ collapsed: true
+ fields:
+ cooking_hood_option_venting_level:
+ example: cooking_hood_enum_type_stage_fan_stage01
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: venting_level
+ options:
+ - cooking_hood_enum_type_stage_fan_off
+ - cooking_hood_enum_type_stage_fan_stage_01
+ - cooking_hood_enum_type_stage_fan_stage_02
+ - cooking_hood_enum_type_stage_fan_stage_03
+ - cooking_hood_enum_type_stage_fan_stage_04
+ - cooking_hood_enum_type_stage_fan_stage_05
+ cooking_hood_option_intensive_level:
+ example: cooking_hood_enum_type_intensive_stage_intensive_stage1
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: intensive_level
+ options:
+ - cooking_hood_enum_type_intensive_stage_intensive_stage_off
+ - cooking_hood_enum_type_intensive_stage_intensive_stage1
+ - cooking_hood_enum_type_intensive_stage_intensive_stage2
+ oven_options:
+ collapsed: true
+ fields:
+ cooking_oven_option_setpoint_temperature:
+ example: 180
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ mode: box
+ unit_of_measurement: °C/°F
+ b_s_h_common_option_duration:
+ example: 900
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ mode: box
+ unit_of_measurement: s
+ cooking_oven_option_fast_pre_heat:
+ example: false
+ required: false
+ selector:
+ boolean:
+ warming_drawer_options:
+ collapsed: true
+ fields:
+ cooking_oven_option_warming_level:
+ example: cooking_oven_enum_type_warming_level_medium
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: warming_level
+ options:
+ - cooking_oven_enum_type_warming_level_low
+ - cooking_oven_enum_type_warming_level_medium
+ - cooking_oven_enum_type_warming_level_high
+ washer_options:
+ collapsed: true
+ fields:
+ laundry_care_washer_option_temperature:
+ example: laundry_care_washer_enum_type_temperature_g_c_40
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: washer_temperature
+ options:
+ - laundry_care_washer_enum_type_temperature_cold
+ - laundry_care_washer_enum_type_temperature_g_c_20
+ - laundry_care_washer_enum_type_temperature_g_c_30
+ - laundry_care_washer_enum_type_temperature_g_c_40
+ - laundry_care_washer_enum_type_temperature_g_c_50
+ - laundry_care_washer_enum_type_temperature_g_c_60
+ - laundry_care_washer_enum_type_temperature_g_c_70
+ - laundry_care_washer_enum_type_temperature_g_c_80
+ - laundry_care_washer_enum_type_temperature_g_c_90
+ - laundry_care_washer_enum_type_temperature_ul_cold
+ - laundry_care_washer_enum_type_temperature_ul_warm
+ - laundry_care_washer_enum_type_temperature_ul_hot
+ - laundry_care_washer_enum_type_temperature_ul_extra_hot
+ laundry_care_washer_option_spin_speed:
+ example: laundry_care_washer_enum_type_spin_speed_r_p_m800
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: spin_speed
+ options:
+ - laundry_care_washer_enum_type_spin_speed_off
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_400
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_600
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_700
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_800
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_900
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_1000
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_1200
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_1400
+ - laundry_care_washer_enum_type_spin_speed_r_p_m_1600
+ - laundry_care_washer_enum_type_spin_speed_ul_off
+ - laundry_care_washer_enum_type_spin_speed_ul_low
+ - laundry_care_washer_enum_type_spin_speed_ul_medium
+ - laundry_care_washer_enum_type_spin_speed_ul_high
+ b_s_h_common_option_finish_in_relative:
+ example: 3600
+ required: false
+ selector:
+ number:
+ min: 0
+ step: 1
+ mode: box
+ unit_of_measurement: s
+ laundry_care_washer_option_i_dos1_active:
+ example: false
+ required: false
+ selector:
+ boolean:
+ laundry_care_washer_option_i_dos2_active:
+ example: false
+ required: false
+ selector:
+ boolean:
+ laundry_care_washer_option_vario_perfect:
+ example: laundry_care_common_enum_type_vario_perfect_eco_perfect
+ required: false
+ selector:
+ select:
+ mode: dropdown
+ translation_key: vario_perfect
+ options:
+ - laundry_care_common_enum_type_vario_perfect_off
+ - laundry_care_common_enum_type_vario_perfect_eco_perfect
+ - laundry_care_common_enum_type_vario_perfect_speed_perfect
pause_program:
fields:
device_id:
diff --git a/homeassistant/components/home_connect/strings.json b/homeassistant/components/home_connect/strings.json
index d163d04a6f7..00ab29affd8 100644
--- a/homeassistant/components/home_connect/strings.json
+++ b/homeassistant/components/home_connect/strings.json
@@ -7,9 +7,14 @@
"step": {
"pick_implementation": {
"title": "[%key:common::config_flow::title::oauth2_pick_implementation%]"
+ },
+ "reauth_confirm": {
+ "title": "[%key:common::config_flow::title::reauth%]",
+ "description": "The Home Connect integration needs to re-authenticate your account"
}
},
"abort": {
+ "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",
"missing_configuration": "[%key:common::config_flow::abort::oauth2_missing_configuration%]",
"no_url_available": "[%key:common::config_flow::abort::oauth2_no_url_available%]",
"oauth_error": "[%key:common::config_flow::abort::oauth2_error%]",
@@ -22,9 +27,18 @@
}
},
"exceptions": {
+ "auth_error": {
+ "message": "Authentication error: {error}. Please, re-authenticate your account"
+ },
"appliance_not_found": {
"message": "Appliance for device ID {device_id} not found"
},
+ "device_entry_not_found": {
+ "message": "Device entry for device ID {device_id} not found"
+ },
+ "config_entry_not_found": {
+ "message": "Config entry for device ID {device_id} not found"
+ },
"turn_on_light": {
"message": "Error turning on {entity_id}: {error}"
},
@@ -87,6 +101,12 @@
},
"fetch_api_error": {
"message": "Error obtaining data from the API: {error}"
+ },
+ "required_program_or_one_option_at_least": {
+ "message": "A program or at least one of the possible options for a program should be specified"
+ },
+ "set_option": {
+ "message": "Error setting the option for the program: {error}"
}
},
"issues": {
@@ -94,9 +114,371 @@
"title": "Deprecated binary door sensor detected in some automations or scripts",
"description": "The binary door sensor `{entity}`, which is deprecated, is used in the following automations or scripts:\n{items}\n\nA sensor entity with additional possible states is available and should be used going forward; Please use it on the above automations or scripts to fix this issue."
},
+ "deprecated_command_actions": {
+ "title": "The command related actions are deprecated in favor of the new buttons",
+ "description": "The `pause_program` and `resume_program` actions have been deprecated in favor of new button entities, if the command is available for your appliance. Please update your automations, scripts and panels that use this action to use the button entities instead, and press on submit to fix the issue."
+ },
"deprecated_program_switch": {
"title": "Deprecated program switch detected in some automations or scripts",
- "description": "Program switch are deprecated and {entity_id} is used in the following automations or scripts:\n{items}\n\nYou can use active program select entity to run the program without any additional option and get the current running program on the above automations or scripts to fix this issue."
+ "description": "Program switches are deprecated and {entity_id} is used in the following automations or scripts:\n{items}\n\nYou can use the active program select entity to run the program without any additional options and get the current running program on the above automations or scripts to fix this issue."
+ },
+ "deprecated_set_program_and_option_actions": {
+ "title": "The executed action is deprecated",
+ "fix_flow": {
+ "step": {
+ "confirm": {
+ "title": "[%key:component::home_connect::issues::deprecated_set_program_and_option_actions::title%]",
+ "description": "`start_program`, `select_program`, `set_option_active`, and `set_option_selected` actions are deprecated and will be removed in the {remove_release} release, please use the `{new_action_key}` action instead. For the executed action:\n{deprecated_action_yaml}\nyou can do the following transformation using the recognized options:\n {new_action_yaml}\nIf the option is not in the recognized options, please submit an issue or a pull request requesting the addition of the option at {repo_link}."
+ }
+ }
+ }
+ }
+ },
+ "selector": {
+ "affects_to": {
+ "options": {
+ "active_program": "Active program",
+ "selected_program": "Selected program"
+ }
+ },
+ "programs": {
+ "options": {
+ "consumer_products_cleaning_robot_program_cleaning_clean_all": "Clean all",
+ "consumer_products_cleaning_robot_program_cleaning_clean_map": "Clean map",
+ "consumer_products_cleaning_robot_program_basic_go_home": "Go home",
+ "consumer_products_coffee_maker_program_beverage_ristretto": "Ristretto",
+ "consumer_products_coffee_maker_program_beverage_espresso": "Espresso",
+ "consumer_products_coffee_maker_program_beverage_espresso_doppio": "Espresso doppio",
+ "consumer_products_coffee_maker_program_beverage_coffee": "Coffee",
+ "consumer_products_coffee_maker_program_beverage_x_l_coffee": "XL coffee",
+ "consumer_products_coffee_maker_program_beverage_caffe_grande": "Caffe grande",
+ "consumer_products_coffee_maker_program_beverage_espresso_macchiato": "Espresso macchiato",
+ "consumer_products_coffee_maker_program_beverage_cappuccino": "Cappuccino",
+ "consumer_products_coffee_maker_program_beverage_latte_macchiato": "Latte macchiato",
+ "consumer_products_coffee_maker_program_beverage_caffe_latte": "Caffe latte",
+ "consumer_products_coffee_maker_program_beverage_milk_froth": "Milk froth",
+ "consumer_products_coffee_maker_program_beverage_warm_milk": "Warm milk",
+ "consumer_products_coffee_maker_program_coffee_world_kleiner_brauner": "Kleiner brauner",
+ "consumer_products_coffee_maker_program_coffee_world_grosser_brauner": "Grosser brauner",
+ "consumer_products_coffee_maker_program_coffee_world_verlaengerter": "Verlaengerter",
+ "consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun": "Verlaengerter braun",
+ "consumer_products_coffee_maker_program_coffee_world_wiener_melange": "Wiener melange",
+ "consumer_products_coffee_maker_program_coffee_world_flat_white": "Flat white",
+ "consumer_products_coffee_maker_program_coffee_world_cortado": "Cortado",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_cortado": "Cafe cortado",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_con_leche": "Cafe con leche",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_au_lait": "Cafe au lait",
+ "consumer_products_coffee_maker_program_coffee_world_doppio": "Doppio",
+ "consumer_products_coffee_maker_program_coffee_world_kaapi": "Kaapi",
+ "consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd": "Koffie verkeerd",
+ "consumer_products_coffee_maker_program_coffee_world_galao": "Galao",
+ "consumer_products_coffee_maker_program_coffee_world_garoto": "Garoto",
+ "consumer_products_coffee_maker_program_coffee_world_americano": "Americano",
+ "consumer_products_coffee_maker_program_coffee_world_red_eye": "Red eye",
+ "consumer_products_coffee_maker_program_coffee_world_black_eye": "Black eye",
+ "consumer_products_coffee_maker_program_coffee_world_dead_eye": "Dead eye",
+ "consumer_products_coffee_maker_program_beverage_hot_water": "Hot water",
+ "dishcare_dishwasher_program_pre_rinse": "Pre_rinse",
+ "dishcare_dishwasher_program_auto_1": "Auto 1",
+ "dishcare_dishwasher_program_auto_2": "Auto 2",
+ "dishcare_dishwasher_program_auto_3": "Auto 3",
+ "dishcare_dishwasher_program_eco_50": "Eco 50ºC",
+ "dishcare_dishwasher_program_quick_45": "Quick 45ºC",
+ "dishcare_dishwasher_program_intensiv_70": "Intensive 70ºC",
+ "dishcare_dishwasher_program_normal_65": "Normal 65ºC",
+ "dishcare_dishwasher_program_glas_40": "Glass 40ºC",
+ "dishcare_dishwasher_program_glass_care": "Glass care",
+ "dishcare_dishwasher_program_night_wash": "Night wash",
+ "dishcare_dishwasher_program_quick_65": "Quick 65ºC",
+ "dishcare_dishwasher_program_normal_45": "Normal 45ºC",
+ "dishcare_dishwasher_program_intensiv_45": "Intensive 45ºC",
+ "dishcare_dishwasher_program_auto_half_load": "Auto half load",
+ "dishcare_dishwasher_program_intensiv_power": "Intensive power",
+ "dishcare_dishwasher_program_magic_daily": "Magic daily",
+ "dishcare_dishwasher_program_super_60": "Super 60ºC",
+ "dishcare_dishwasher_program_kurz_60": "Kurz 60ºC",
+ "dishcare_dishwasher_program_express_sparkle_65": "Express sparkle 65ºC",
+ "dishcare_dishwasher_program_machine_care": "Machine care",
+ "dishcare_dishwasher_program_steam_fresh": "Steam fresh",
+ "dishcare_dishwasher_program_maximum_cleaning": "Maximum cleaning",
+ "dishcare_dishwasher_program_mixed_load": "Mixed load",
+ "laundry_care_dryer_program_cotton": "Cotton",
+ "laundry_care_dryer_program_synthetic": "Synthetic",
+ "laundry_care_dryer_program_mix": "Mix",
+ "laundry_care_dryer_program_blankets": "Blankets",
+ "laundry_care_dryer_program_business_shirts": "Business shirts",
+ "laundry_care_dryer_program_down_feathers": "Down feathers",
+ "laundry_care_dryer_program_hygiene": "Hygiene",
+ "laundry_care_dryer_program_jeans": "Jeans",
+ "laundry_care_dryer_program_outdoor": "Outdoor",
+ "laundry_care_dryer_program_synthetic_refresh": "Synthetic refresh",
+ "laundry_care_dryer_program_towels": "Towels",
+ "laundry_care_dryer_program_delicates": "Delicates",
+ "laundry_care_dryer_program_super_40": "Super 40ºC",
+ "laundry_care_dryer_program_shirts_15": "Shirts 15ºC",
+ "laundry_care_dryer_program_pillow": "Pillow",
+ "laundry_care_dryer_program_anti_shrink": "Anti shrink",
+ "laundry_care_dryer_program_my_time_my_drying_time": "My drying time",
+ "laundry_care_dryer_program_time_cold": "Cold (variable time)",
+ "laundry_care_dryer_program_time_warm": "Warm (variable time)",
+ "laundry_care_dryer_program_in_basket": "In basket",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_20": "Cold (20 min)",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_30": "Cold (30 min)",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_60": "Cold (60 min)",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_30": "Warm (30 min)",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_40": "Warm (40 min)",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_60": "Warm (60 min)",
+ "laundry_care_dryer_program_dessous": "Dessous",
+ "cooking_common_program_hood_automatic": "Automatic",
+ "cooking_common_program_hood_venting": "Venting",
+ "cooking_common_program_hood_delayed_shut_off": "Delayed shut off",
+ "cooking_oven_program_heating_mode_pre_heating": "Pre-heating",
+ "cooking_oven_program_heating_mode_hot_air": "Hot air",
+ "cooking_oven_program_heating_mode_hot_air_eco": "Hot air eco",
+ "cooking_oven_program_heating_mode_hot_air_grilling": "Hot air grilling",
+ "cooking_oven_program_heating_mode_top_bottom_heating": "Top bottom heating",
+ "cooking_oven_program_heating_mode_top_bottom_heating_eco": "Top bottom heating eco",
+ "cooking_oven_program_heating_mode_bottom_heating": "Bottom heating",
+ "cooking_oven_program_heating_mode_pizza_setting": "Pizza setting",
+ "cooking_oven_program_heating_mode_slow_cook": "Slow cook",
+ "cooking_oven_program_heating_mode_intensive_heat": "Intensive heat",
+ "cooking_oven_program_heating_mode_keep_warm": "Keep warm",
+ "cooking_oven_program_heating_mode_preheat_ovenware": "Preheat ovenware",
+ "cooking_oven_program_heating_mode_frozen_heatup_special": "Special Heat-Up for frozen products",
+ "cooking_oven_program_heating_mode_desiccation": "Desiccation",
+ "cooking_oven_program_heating_mode_defrost": "Defrost",
+ "cooking_oven_program_heating_mode_proof": "Proof",
+ "cooking_oven_program_heating_mode_hot_air_30_steam": "Hot air + 30 RH",
+ "cooking_oven_program_heating_mode_hot_air_60_steam": "Hot air + 60 RH",
+ "cooking_oven_program_heating_mode_hot_air_80_steam": "Hot air + 80 RH",
+ "cooking_oven_program_heating_mode_hot_air_100_steam": "Hot air + 100 RH",
+ "cooking_oven_program_heating_mode_sabbath_programme": "Sabbath programme",
+ "cooking_oven_program_microwave_90_watt": "90 Watt",
+ "cooking_oven_program_microwave_180_watt": "180 Watt",
+ "cooking_oven_program_microwave_360_watt": "360 Watt",
+ "cooking_oven_program_microwave_600_watt": "600 Watt",
+ "cooking_oven_program_microwave_900_watt": "900 Watt",
+ "cooking_oven_program_microwave_1000_watt": "1000 Watt",
+ "cooking_oven_program_microwave_max": "Max",
+ "cooking_oven_program_heating_mode_warming_drawer": "Warming drawer",
+ "laundry_care_washer_program_cotton": "Cotton",
+ "laundry_care_washer_program_cotton_cotton_eco": "Cotton eco",
+ "laundry_care_washer_program_cotton_eco_4060": "Cotton eco 40/60ºC",
+ "laundry_care_washer_program_cotton_colour": "Cotton color",
+ "laundry_care_washer_program_easy_care": "Easy care",
+ "laundry_care_washer_program_mix": "Mix",
+ "laundry_care_washer_program_mix_night_wash": "Mix night wash",
+ "laundry_care_washer_program_delicates_silk": "Delicates silk",
+ "laundry_care_washer_program_wool": "Wool",
+ "laundry_care_washer_program_sensitive": "Sensitive",
+ "laundry_care_washer_program_auto_30": "Auto 30ºC",
+ "laundry_care_washer_program_auto_40": "Auto 40ºC",
+ "laundry_care_washer_program_auto_60": "Auto 60ºC",
+ "laundry_care_washer_program_chiffon": "Chiffon",
+ "laundry_care_washer_program_curtains": "Curtains",
+ "laundry_care_washer_program_dark_wash": "Dark wash",
+ "laundry_care_washer_program_dessous": "Dessous",
+ "laundry_care_washer_program_monsoon": "Monsoon",
+ "laundry_care_washer_program_outdoor": "Outdoor",
+ "laundry_care_washer_program_plush_toy": "Plush toy",
+ "laundry_care_washer_program_shirts_blouses": "Shirts blouses",
+ "laundry_care_washer_program_sport_fitness": "Sport fitness",
+ "laundry_care_washer_program_towels": "Towels",
+ "laundry_care_washer_program_water_proof": "Water proof",
+ "laundry_care_washer_program_power_speed_59": "Power speed <59 min",
+ "laundry_care_washer_program_super_153045_super_15": "Super 15 min",
+ "laundry_care_washer_program_super_153045_super_1530": "Super 15/30 min",
+ "laundry_care_washer_program_down_duvet_duvet": "Down duvet",
+ "laundry_care_washer_program_rinse_rinse_spin_drain": "Rinse spin drain",
+ "laundry_care_washer_program_drum_clean": "Drum clean",
+ "laundry_care_washer_dryer_program_cotton": "Cotton",
+ "laundry_care_washer_dryer_program_cotton_eco_4060": "Cotton eco 40/60ºC",
+ "laundry_care_washer_dryer_program_mix": "Mix",
+ "laundry_care_washer_dryer_program_easy_care": "Easy care",
+ "laundry_care_washer_dryer_program_wash_and_dry_60": "Wash and dry (60 min)",
+ "laundry_care_washer_dryer_program_wash_and_dry_90": "Wash and dry (90 min)"
+ }
+ },
+ "available_maps": {
+ "options": {
+ "consumer_products_cleaning_robot_enum_type_available_maps_temp_map": "Temporary map",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map1": "Map 1",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map2": "Map 2",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map3": "Map 3"
+ }
+ },
+ "cleaning_mode": {
+ "options": {
+ "consumer_products_cleaning_robot_enum_type_cleaning_modes_silent": "Silent",
+ "consumer_products_cleaning_robot_enum_type_cleaning_modes_standard": "Standard",
+ "consumer_products_cleaning_robot_enum_type_cleaning_modes_power": "Power"
+ }
+ },
+ "bean_amount": {
+ "options": {
+ "consumer_products_coffee_maker_enum_type_bean_amount_very_mild": "Very mild",
+ "consumer_products_coffee_maker_enum_type_bean_amount_mild": "Mild",
+ "consumer_products_coffee_maker_enum_type_bean_amount_mild_plus": "Mild +",
+ "consumer_products_coffee_maker_enum_type_bean_amount_normal": "Normal",
+ "consumer_products_coffee_maker_enum_type_bean_amount_normal_plus": "Normal +",
+ "consumer_products_coffee_maker_enum_type_bean_amount_strong": "Strong",
+ "consumer_products_coffee_maker_enum_type_bean_amount_strong_plus": "Strong +",
+ "consumer_products_coffee_maker_enum_type_bean_amount_very_strong": "Very strong",
+ "consumer_products_coffee_maker_enum_type_bean_amount_very_strong_plus": "Very strong +",
+ "consumer_products_coffee_maker_enum_type_bean_amount_extra_strong": "Extra strong",
+ "consumer_products_coffee_maker_enum_type_bean_amount_double_shot": "Double shot",
+ "consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus": "Double shot +",
+ "consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus_plus": "Double shot ++",
+ "consumer_products_coffee_maker_enum_type_bean_amount_triple_shot": "Triple shot",
+ "consumer_products_coffee_maker_enum_type_bean_amount_triple_shot_plus": "Triple shot +",
+ "consumer_products_coffee_maker_enum_type_bean_amount_coffee_ground": "Coffee ground"
+ }
+ },
+ "coffee_temperature": {
+ "options": {
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_88_c": "88ºC",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_90_c": "90ºC",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_92_c": "92ºC",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_94_c": "94ºC",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_95_c": "95ºC",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_96_c": "96ºC"
+ }
+ },
+ "bean_container": {
+ "options": {
+ "consumer_products_coffee_maker_enum_type_bean_container_selection_right": "Right",
+ "consumer_products_coffee_maker_enum_type_bean_container_selection_left": "Left"
+ }
+ },
+ "flow_rate": {
+ "options": {
+ "consumer_products_coffee_maker_enum_type_flow_rate_normal": "Normal",
+ "consumer_products_coffee_maker_enum_type_flow_rate_intense": "Intense",
+ "consumer_products_coffee_maker_enum_type_flow_rate_intense_plus": "Intense +"
+ }
+ },
+ "coffee_milk_ratio": {
+ "options": {
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_10_percent": "10%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_20_percent": "20%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_25_percent": "25%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_30_percent": "30%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_40_percent": "40%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_50_percent": "50%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_55_percent": "55%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_60_percent": "60%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_65_percent": "65%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_67_percent": "67%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_70_percent": "70%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_75_percent": "75%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_80_percent": "80%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_85_percent": "85%",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_90_percent": "90%"
+ }
+ },
+ "hot_water_temperature": {
+ "options": {
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_white_tea": "White tea",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_green_tea": "Green tea",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_black_tea": "Black tea",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_50_c": "50ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_55_c": "55ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_60_c": "60ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_65_c": "65ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_70_c": "70ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_75_c": "75ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_80_c": "80ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_85_c": "85ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_90_c": "90ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_95_c": "95ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_97_c": "97ºC",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_122_f": "122ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_131_f": "131ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_140_f": "140ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_149_f": "149ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_158_f": "158ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_167_f": "167ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_176_f": "176ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_185_f": "185ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_194_f": "194ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_203_f": "203ºF",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_max": "Max"
+ }
+ },
+ "drying_target": {
+ "options": {
+ "laundry_care_dryer_enum_type_drying_target_iron_dry": "Iron dry",
+ "laundry_care_dryer_enum_type_drying_target_gentle_dry": "Gentle dry",
+ "laundry_care_dryer_enum_type_drying_target_cupboard_dry": "Cupboard dry",
+ "laundry_care_dryer_enum_type_drying_target_cupboard_dry_plus": "Cupboard dry +",
+ "laundry_care_dryer_enum_type_drying_target_extra_dry": "Extra dry"
+ }
+ },
+ "venting_level": {
+ "options": {
+ "cooking_hood_enum_type_stage_fan_off": "Fan off",
+ "cooking_hood_enum_type_stage_fan_stage_01": "Fan stage 1",
+ "cooking_hood_enum_type_stage_fan_stage_02": "Fan stage 2",
+ "cooking_hood_enum_type_stage_fan_stage_03": "Fan stage 3",
+ "cooking_hood_enum_type_stage_fan_stage_04": "Fan stage 4",
+ "cooking_hood_enum_type_stage_fan_stage_05": "Fan stage 5"
+ }
+ },
+ "intensive_level": {
+ "options": {
+ "cooking_hood_enum_type_intensive_stage_intensive_stage_off": "Intensive stage off",
+ "cooking_hood_enum_type_intensive_stage_intensive_stage1": "Intensive stage 1",
+ "cooking_hood_enum_type_intensive_stage_intensive_stage2": "Intensive stage 2"
+ }
+ },
+ "warming_level": {
+ "options": {
+ "cooking_oven_enum_type_warming_level_low": "Low",
+ "cooking_oven_enum_type_warming_level_medium": "Medium",
+ "cooking_oven_enum_type_warming_level_high": "High"
+ }
+ },
+ "washer_temperature": {
+ "options": {
+ "laundry_care_washer_enum_type_temperature_cold": "Cold",
+ "laundry_care_washer_enum_type_temperature_g_c_20": "20ºC clothes",
+ "laundry_care_washer_enum_type_temperature_g_c_30": "30ºC clothes",
+ "laundry_care_washer_enum_type_temperature_g_c_40": "40ºC clothes",
+ "laundry_care_washer_enum_type_temperature_g_c_50": "50ºC clothes",
+ "laundry_care_washer_enum_type_temperature_g_c_60": "60ºC clothes",
+ "laundry_care_washer_enum_type_temperature_g_c_70": "70ºC clothes",
+ "laundry_care_washer_enum_type_temperature_g_c_80": "80ºC clothes",
+ "laundry_care_washer_enum_type_temperature_g_c_90": "90ºC clothes",
+ "laundry_care_washer_enum_type_temperature_ul_cold": "Cold",
+ "laundry_care_washer_enum_type_temperature_ul_warm": "Warm",
+ "laundry_care_washer_enum_type_temperature_ul_hot": "Hot",
+ "laundry_care_washer_enum_type_temperature_ul_extra_hot": "Extra hot"
+ }
+ },
+ "spin_speed": {
+ "options": {
+ "laundry_care_washer_enum_type_spin_speed_off": "Off",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_400": "400 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_600": "600 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_700": "700 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_800": "800 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_900": "900 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1000": "1000 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1200": "1200 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1400": "1400 rpm",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1600": "1600 rpm",
+ "laundry_care_washer_enum_type_spin_speed_ul_off": "Off",
+ "laundry_care_washer_enum_type_spin_speed_ul_low": "Low",
+ "laundry_care_washer_enum_type_spin_speed_ul_medium": "Medium",
+ "laundry_care_washer_enum_type_spin_speed_ul_high": "High"
+ }
+ },
+ "vario_perfect": {
+ "options": {
+ "laundry_care_common_enum_type_vario_perfect_off": "Off",
+ "laundry_care_common_enum_type_vario_perfect_eco_perfect": "Eco perfect",
+ "laundry_care_common_enum_type_vario_perfect_speed_perfect": "Speed perfect"
+ }
}
},
"services": {
@@ -105,8 +487,8 @@
"description": "Selects a program and starts it.",
"fields": {
"device_id": {
- "name": "Device ID",
- "description": "ID of the device."
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::name%]",
+ "description": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::description%]"
},
"program": { "name": "Program", "description": "Program to select." },
"key": { "name": "Option key", "description": "Key of the option." },
@@ -122,8 +504,8 @@
"description": "Selects a program without starting it.",
"fields": {
"device_id": {
- "name": "Device ID",
- "description": "[%key:component::home_connect::services::start_program::fields::device_id::description%]"
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::name%]",
+ "description": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::description%]"
},
"program": {
"name": "[%key:component::home_connect::services::start_program::fields::program::name%]",
@@ -143,13 +525,197 @@
}
}
},
+ "set_program_and_options": {
+ "name": "Set program and options",
+ "description": "Starts or selects a program with options or sets the options for the active or the selected program.",
+ "fields": {
+ "device_id": {
+ "name": "Device ID",
+ "description": "ID of the device."
+ },
+ "affects_to": {
+ "name": "Affects to",
+ "description": "Selects if the program affected by the action should be the active or the selected program."
+ },
+ "program": {
+ "name": "Program",
+ "description": "Program to select"
+ },
+ "consumer_products_cleaning_robot_option_reference_map_id": {
+ "name": "Reference map ID",
+ "description": "Defines which reference map is to be used."
+ },
+ "consumer_products_cleaning_robot_option_cleaning_mode": {
+ "name": "Cleaning mode",
+ "description": "Defines the favoured cleaning mode."
+ },
+ "consumer_products_coffee_maker_option_bean_amount": {
+ "name": "Bean amount",
+ "description": "Describes the amount of coffee beans used in a coffee machine program."
+ },
+ "consumer_products_coffee_maker_option_fill_quantity": {
+ "name": "Fill quantity",
+ "description": "Describes the amount of water (in ml) used in a coffee machine program."
+ },
+ "consumer_products_coffee_maker_option_coffee_temperature": {
+ "name": "Coffee Temperature",
+ "description": "Describes the coffee temperature used in a coffee machine program."
+ },
+ "consumer_products_coffee_maker_option_bean_container": {
+ "name": "Bean container",
+ "description": "Defines the preferred bean container."
+ },
+ "consumer_products_coffee_maker_option_flow_rate": {
+ "name": "Flow rate",
+ "description": "Defines the water-coffee contact time. The duration extends to coffee intensity."
+ },
+ "consumer_products_coffee_maker_option_multiple_beverages": {
+ "name": "Multiple beverages",
+ "description": "Defines if double dispensing is enabled."
+ },
+ "consumer_products_coffee_maker_option_coffee_milk_ratio": {
+ "name": "Coffee milk ratio",
+ "description": "Defines the amount of milk."
+ },
+ "consumer_products_coffee_maker_option_hot_water_temperature": {
+ "name": "Hot water temperature",
+ "description": "Defines the temperature suitable for the type of tea."
+ },
+ "b_s_h_common_option_start_in_relative": {
+ "name": "Start in relative",
+ "description": "Defines in how many time the program should start."
+ },
+ "dishcare_dishwasher_option_intensiv_zone": {
+ "name": "Intensive zone",
+ "description": "Defines if the cleaning is done with higher spray pressure on the lower basket for very dirty pots and pans."
+ },
+ "dishcare_dishwasher_option_brilliance_dry": {
+ "name": "Brilliance dry",
+ "description": "Defines if the program sequence is optimized with a special drying cycle to ensure more shine on glasses and plastic items."
+ },
+ "dishcare_dishwasher_option_vario_speed_plus": {
+ "name": "Vario speed +",
+ "description": "Defines if the program run time is reduced by up to 66% with the usual optimum cleaning and drying."
+ },
+ "dishcare_dishwasher_option_silence_on_demand": {
+ "name": "Silence on demand",
+ "description": "Defines if the extra silent mode is activated for a selected period of time."
+ },
+ "dishcare_dishwasher_option_half_load": {
+ "name": "Half load",
+ "description": "Defines if economical cleaning is enabled for smaller loads. This reduces energy and water consumption and also saves time. The utensils can be placed in the upper and lower baskets."
+ },
+ "dishcare_dishwasher_option_extra_dry": {
+ "name": "Extra dry",
+ "description": "Defines if improved drying for glasses and plasticware is enabled."
+ },
+ "dishcare_dishwasher_option_hygiene_plus": {
+ "name": "Hygiene +",
+ "description": "Defines if the cleaning is done with increased temperature. This ensures maximum hygienic cleanliness for regular use."
+ },
+ "dishcare_dishwasher_option_eco_dry": {
+ "name": "Eco dry",
+ "description": "Defines if the door is opened automatically for extra energy efficient and effective drying."
+ },
+ "dishcare_dishwasher_option_zeolite_dry": {
+ "name": "Zeolite dry",
+ "description": "Defines if the program sequence is optimized with special drying cycle ensures improved drying for glasses, plates and plasticware."
+ },
+ "laundry_care_dryer_option_drying_target": {
+ "name": "Drying target",
+ "description": "Describes the drying target for a dryer program."
+ },
+ "cooking_hood_option_venting_level": {
+ "name": "Venting level",
+ "description": "Defines the required fan setting."
+ },
+ "cooking_hood_option_intensive_level": {
+ "name": "Intensive level",
+ "description": "Defines the intensive setting."
+ },
+ "cooking_oven_option_setpoint_temperature": {
+ "name": "Setpoint temperature",
+ "description": "Defines the target cavity temperature, which will be hold by the oven."
+ },
+ "b_s_h_common_option_duration": {
+ "name": "Duration",
+ "description": "Defines the run-time of the program. Afterwards, the appliance is stopped."
+ },
+ "cooking_oven_option_fast_pre_heat": {
+ "name": "Fast pre-heat",
+ "description": "Defines if the cooking compartment is heated up quickly. Please note that the setpoint temperature has to be equal to or higher than 100 °C or 212 °F. Otherwise, the fast pre-heat option is not activated."
+ },
+ "cooking_oven_option_warming_level": {
+ "name": "Warming level",
+ "description": "Defines the level of the warming drawer."
+ },
+ "laundry_care_washer_option_temperature": {
+ "name": "Temperature",
+ "description": "Defines the temperature of the washing program."
+ },
+ "laundry_care_washer_option_spin_speed": {
+ "name": "Spin speed",
+ "description": "Defines the spin speed of a washer program."
+ },
+ "b_s_h_common_option_finish_in_relative": {
+ "name": "Finish in relative",
+ "description": "Defines when the program should end in seconds."
+ },
+ "laundry_care_washer_option_i_dos1_active": {
+ "name": "i-Dos 1 Active",
+ "description": "Defines if the detergent feed is activated / deactivated. (i-Dos content 1)"
+ },
+ "laundry_care_washer_option_i_dos2_active": {
+ "name": "i-Dos 2 Active",
+ "description": "Defines if the detergent feed is activated / deactivated. (i-Dos content 2)"
+ },
+ "laundry_care_washer_option_vario_perfect": {
+ "name": "Vario perfect",
+ "description": "Defines if a cycle saves energy (Eco Perfect) or time (Speed Perfect)."
+ }
+ },
+ "sections": {
+ "cleaning_robot_options": {
+ "name": "Cleaning robot options",
+ "description": "Options for cleaning robots."
+ },
+ "coffee_maker_options": {
+ "name": "Coffee maker options",
+ "description": "Options for coffee makers."
+ },
+ "dish_washer_options": {
+ "name": "Dishwasher options",
+ "description": "Options for dishwashers."
+ },
+ "dryer_options": {
+ "name": "Dryer options",
+ "description": "Options for dryers (and washer dryers)."
+ },
+ "hood_options": {
+ "name": "Hood options",
+ "description": "Options for hoods."
+ },
+ "oven_options": {
+ "name": "Oven options",
+ "description": "Options for ovens."
+ },
+ "warming_drawer_options": {
+ "name": "Warming drawer options",
+ "description": "Options for warming drawers."
+ },
+ "washer_options": {
+ "name": "Washer options",
+ "description": "Options for washers (and washer dryers)."
+ }
+ }
+ },
"pause_program": {
"name": "Pause program",
"description": "Pauses the current running program.",
"fields": {
"device_id": {
- "name": "Device ID",
- "description": "[%key:component::home_connect::services::start_program::fields::device_id::description%]"
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::name%]",
+ "description": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::description%]"
}
}
},
@@ -158,8 +724,8 @@
"description": "Resumes a paused program.",
"fields": {
"device_id": {
- "name": "Device ID",
- "description": "[%key:component::home_connect::services::start_program::fields::device_id::description%]"
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::name%]",
+ "description": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::description%]"
}
}
},
@@ -168,8 +734,8 @@
"description": "Sets an option for the active program.",
"fields": {
"device_id": {
- "name": "Device ID",
- "description": "[%key:component::home_connect::services::start_program::fields::device_id::description%]"
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::name%]",
+ "description": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::description%]"
},
"key": {
"name": "Key",
@@ -183,18 +749,18 @@
},
"set_option_selected": {
"name": "Set selected program option",
- "description": "Sets an option for the selected program.",
+ "description": "Sets options for the selected program.",
"fields": {
"device_id": {
- "name": "Device ID",
- "description": "[%key:component::home_connect::services::start_program::fields::device_id::description%]"
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::name%]",
+ "description": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::description%]"
},
"key": {
- "name": "Key",
+ "name": "[%key:component::home_connect::services::start_program::fields::key::name%]",
"description": "[%key:component::home_connect::services::start_program::fields::key::description%]"
},
"value": {
- "name": "Value",
+ "name": "[%key:component::home_connect::services::start_program::fields::value::name%]",
"description": "[%key:component::home_connect::services::start_program::fields::value::description%]"
}
}
@@ -204,8 +770,8 @@
"description": "Changes a setting.",
"fields": {
"device_id": {
- "name": "Device ID",
- "description": "[%key:component::home_connect::services::start_program::fields::device_id::description%]"
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::name%]",
+ "description": "[%key:component::home_connect::services::set_program_and_options::fields::device_id::description%]"
},
"key": { "name": "Key", "description": "Key of the setting." },
"value": { "name": "Value", "description": "Value of the setting." }
@@ -242,14 +808,49 @@
"lost": {
"name": "Lost"
},
+ "bottle_cooler_door": {
+ "name": "Bottle cooler door"
+ },
+ "common_chiller_door": {
+ "name": "Common chiller door"
+ },
"chiller_door": {
"name": "Chiller door"
},
+ "left_chiller_door": {
+ "name": "Left chiller door"
+ },
+ "right_chiller_door": {
+ "name": "Right chiller door"
+ },
+ "flex_compartment_door": {
+ "name": "Flex compartment door"
+ },
"freezer_door": {
"name": "Freezer door"
},
"refrigerator_door": {
"name": "Refrigerator door"
+ },
+ "wine_compartment_door": {
+ "name": "Wine compartment door"
+ }
+ },
+ "button": {
+ "open_door": {
+ "name": "Open door"
+ },
+ "partly_open_door": {
+ "name": "Partly open door"
+ },
+ "pause_program": {
+ "name": "Pause program"
+ },
+ "resume_program": {
+ "name": "Resume program"
+ },
+ "stop_program": {
+ "name": "Stop program"
}
},
"light": {
@@ -293,325 +894,571 @@
},
"wine_compartment_3_setpoint_temperature": {
"name": "Wine compartment 3 temperature"
+ },
+ "color_temperature_percent": {
+ "name": "Functional light color temperature percent"
+ },
+ "washer_i_dos_1_base_level": {
+ "name": "i-Dos 1 base level"
+ },
+ "washer_i_dos_2_base_level": {
+ "name": "i-Dos 2 base level"
+ },
+ "duration": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::b_s_h_common_option_duration::name%]"
+ },
+ "start_in_relative": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::b_s_h_common_option_start_in_relative::name%]"
+ },
+ "finish_in_relative": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::b_s_h_common_option_finish_in_relative::name%]"
+ },
+ "fill_quantity": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_fill_quantity::name%]"
+ },
+ "setpoint_temperature": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::cooking_oven_option_setpoint_temperature::name%]"
}
},
"select": {
"selected_program": {
"name": "Selected program",
"state": {
- "consumer_products_cleaning_robot_program_cleaning_clean_all": "Clean all",
- "consumer_products_cleaning_robot_program_cleaning_clean_map": "Clean map",
- "consumer_products_cleaning_robot_program_basic_go_home": "Go home",
- "consumer_products_coffee_maker_program_beverage_ristretto": "Ristretto",
- "consumer_products_coffee_maker_program_beverage_espresso": "Espresso",
- "consumer_products_coffee_maker_program_beverage_espresso_doppio": "Espresso doppio",
- "consumer_products_coffee_maker_program_beverage_coffee": "Coffee",
- "consumer_products_coffee_maker_program_beverage_x_l_coffee": "XL coffee",
- "consumer_products_coffee_maker_program_beverage_caffe_grande": "Caffe grande",
- "consumer_products_coffee_maker_program_beverage_espresso_macchiato": "Espresso macchiato",
- "consumer_products_coffee_maker_program_beverage_cappuccino": "Cappuccino",
- "consumer_products_coffee_maker_program_beverage_latte_macchiato": "Latte macchiato",
- "consumer_products_coffee_maker_program_beverage_caffe_latte": "Caffe latte",
- "consumer_products_coffee_maker_program_beverage_milk_froth": "Milk froth",
- "consumer_products_coffee_maker_program_beverage_warm_milk": "Warm milk",
- "consumer_products_coffee_maker_program_coffee_world_kleiner_brauner": "Kleiner brauner",
- "consumer_products_coffee_maker_program_coffee_world_grosser_brauner": "Grosser brauner",
- "consumer_products_coffee_maker_program_coffee_world_verlaengerter": "Verlaengerter",
- "consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun": "Verlaengerter braun",
- "consumer_products_coffee_maker_program_coffee_world_wiener_melange": "Wiener melange",
- "consumer_products_coffee_maker_program_coffee_world_flat_white": "Flat white",
- "consumer_products_coffee_maker_program_coffee_world_cortado": "Cortado",
- "consumer_products_coffee_maker_program_coffee_world_cafe_cortado": "Cafe cortado",
- "consumer_products_coffee_maker_program_coffee_world_cafe_con_leche": "Cafe con leche",
- "consumer_products_coffee_maker_program_coffee_world_cafe_au_lait": "Cafe au lait",
- "consumer_products_coffee_maker_program_coffee_world_doppio": "Doppio",
- "consumer_products_coffee_maker_program_coffee_world_kaapi": "Kaapi",
- "consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd": "Koffie verkeerd",
- "consumer_products_coffee_maker_program_coffee_world_galao": "Galao",
- "consumer_products_coffee_maker_program_coffee_world_garoto": "Garoto",
- "consumer_products_coffee_maker_program_coffee_world_americano": "Americano",
- "consumer_products_coffee_maker_program_coffee_world_red_eye": "Red eye",
- "consumer_products_coffee_maker_program_coffee_world_black_eye": "Black eye",
- "consumer_products_coffee_maker_program_coffee_world_dead_eye": "Dead eye",
- "consumer_products_coffee_maker_program_beverage_hot_water": "Hot water",
- "dishcare_dishwasher_program_pre_rinse": "Pre_rinse",
- "dishcare_dishwasher_program_auto_1": "Auto 1",
- "dishcare_dishwasher_program_auto_2": "Auto 2",
- "dishcare_dishwasher_program_auto_3": "Auto 3",
- "dishcare_dishwasher_program_eco_50": "Eco 50ºC",
- "dishcare_dishwasher_program_quick_45": "Quick 45ºC",
- "dishcare_dishwasher_program_intensiv_70": "Intensive 70ºC",
- "dishcare_dishwasher_program_normal_65": "Normal 65ºC",
- "dishcare_dishwasher_program_glas_40": "Glass 40ºC",
- "dishcare_dishwasher_program_glass_care": "Glass care",
- "dishcare_dishwasher_program_night_wash": "Night wash",
- "dishcare_dishwasher_program_quick_65": "Quick 65ºC",
- "dishcare_dishwasher_program_normal_45": "Normal 45ºC",
- "dishcare_dishwasher_program_intensiv_45": "Intensive 45ºC",
- "dishcare_dishwasher_program_auto_half_load": "Auto half load",
- "dishcare_dishwasher_program_intensiv_power": "Intensive power",
- "dishcare_dishwasher_program_magic_daily": "Magic daily",
- "dishcare_dishwasher_program_super_60": "Super 60ºC",
- "dishcare_dishwasher_program_kurz_60": "Kurz 60ºC",
- "dishcare_dishwasher_program_express_sparkle_65": "Express sparkle 65ºC",
- "dishcare_dishwasher_program_machine_care": "Machine care",
- "dishcare_dishwasher_program_steam_fresh": "Steam fresh",
- "dishcare_dishwasher_program_maximum_cleaning": "Maximum cleaning",
- "dishcare_dishwasher_program_mixed_load": "Mixed load",
- "laundry_care_dryer_program_cotton": "Cotton",
- "laundry_care_dryer_program_synthetic": "Synthetic",
- "laundry_care_dryer_program_mix": "Mix",
- "laundry_care_dryer_program_blankets": "Blankets",
- "laundry_care_dryer_program_business_shirts": "Business shirts",
- "laundry_care_dryer_program_down_feathers": "Down feathers",
- "laundry_care_dryer_program_hygiene": "Hygiene",
- "laundry_care_dryer_program_jeans": "Jeans",
- "laundry_care_dryer_program_outdoor": "Outdoor",
- "laundry_care_dryer_program_synthetic_refresh": "Synthetic refresh",
- "laundry_care_dryer_program_towels": "Towels",
- "laundry_care_dryer_program_delicates": "Delicates",
- "laundry_care_dryer_program_super_40": "Super 40ºC",
- "laundry_care_dryer_program_shirts_15": "Shirts 15ºC",
- "laundry_care_dryer_program_pillow": "Pillow",
- "laundry_care_dryer_program_anti_shrink": "Anti shrink",
- "laundry_care_dryer_program_my_time_my_drying_time": "My drying time",
- "laundry_care_dryer_program_time_cold": "Cold (variable time)",
- "laundry_care_dryer_program_time_warm": "Warm (variable time)",
- "laundry_care_dryer_program_in_basket": "In basket",
- "laundry_care_dryer_program_time_cold_fix_time_cold_20": "Cold (20 min)",
- "laundry_care_dryer_program_time_cold_fix_time_cold_30": "Cold (30 min)",
- "laundry_care_dryer_program_time_cold_fix_time_cold_60": "Cold (60 min)",
- "laundry_care_dryer_program_time_warm_fix_time_warm_30": "Warm (30 min)",
- "laundry_care_dryer_program_time_warm_fix_time_warm_40": "Warm (40 min)",
- "laundry_care_dryer_program_time_warm_fix_time_warm_60": "Warm (60 min)",
- "laundry_care_dryer_program_dessous": "Dessous",
- "cooking_common_program_hood_automatic": "Automatic",
- "cooking_common_program_hood_venting": "Venting",
- "cooking_common_program_hood_delayed_shut_off": "Delayed shut off",
- "cooking_oven_program_heating_mode_pre_heating": "Pre-heating",
- "cooking_oven_program_heating_mode_hot_air": "Hot air",
- "cooking_oven_program_heating_mode_hot_air_eco": "Hot air eco",
- "cooking_oven_program_heating_mode_hot_air_grilling": "Hot air grilling",
- "cooking_oven_program_heating_mode_top_bottom_heating": "Top bottom heating",
- "cooking_oven_program_heating_mode_top_bottom_heating_eco": "Top bottom heating eco",
- "cooking_oven_program_heating_mode_bottom_heating": "Bottom heating",
- "cooking_oven_program_heating_mode_pizza_setting": "Pizza setting",
- "cooking_oven_program_heating_mode_slow_cook": "Slow cook",
- "cooking_oven_program_heating_mode_intensive_heat": "Intensive heat",
- "cooking_oven_program_heating_mode_keep_warm": "Keep warm",
- "cooking_oven_program_heating_mode_preheat_ovenware": "Preheat ovenware",
- "cooking_oven_program_heating_mode_frozen_heatup_special": "Special Heat-Up for frozen products",
- "cooking_oven_program_heating_mode_desiccation": "Desiccation",
- "cooking_oven_program_heating_mode_defrost": "Defrost",
- "cooking_oven_program_heating_mode_proof": "Proof",
- "cooking_oven_program_heating_mode_hot_air_30_steam": "Hot air + 30 RH",
- "cooking_oven_program_heating_mode_hot_air_60_steam": "Hot air + 60 RH",
- "cooking_oven_program_heating_mode_hot_air_80_steam": "Hot air + 80 RH",
- "cooking_oven_program_heating_mode_hot_air_100_steam": "Hot air + 100 RH",
- "cooking_oven_program_heating_mode_sabbath_programme": "Sabbath programme",
- "cooking_oven_program_microwave_90_watt": "90 Watt",
- "cooking_oven_program_microwave_180_watt": "180 Watt",
- "cooking_oven_program_microwave_360_watt": "360 Watt",
- "cooking_oven_program_microwave_600_watt": "600 Watt",
- "cooking_oven_program_microwave_900_watt": "900 Watt",
- "cooking_oven_program_microwave_1000_watt": "1000 Watt",
- "cooking_oven_program_microwave_max": "Max",
- "cooking_oven_program_heating_mode_warming_drawer": "Warming drawer",
- "laundry_care_washer_program_cotton": "Cotton",
- "laundry_care_washer_program_cotton_cotton_eco": "Cotton eco",
- "laundry_care_washer_program_cotton_eco_4060": "Cotton eco 40/60ºC",
- "laundry_care_washer_program_cotton_colour": "Cotton color",
- "laundry_care_washer_program_easy_care": "Easy care",
- "laundry_care_washer_program_mix": "Mix",
- "laundry_care_washer_program_mix_night_wash": "Mix night wash",
- "laundry_care_washer_program_delicates_silk": "Delicates silk",
- "laundry_care_washer_program_wool": "Wool",
- "laundry_care_washer_program_sensitive": "Sensitive",
- "laundry_care_washer_program_auto_30": "Auto 30ºC",
- "laundry_care_washer_program_auto_40": "Auto 40ºC",
- "laundry_care_washer_program_auto_60": "Auto 60ºC",
- "laundry_care_washer_program_chiffon": "Chiffon",
- "laundry_care_washer_program_curtains": "Curtains",
- "laundry_care_washer_program_dark_wash": "Dark wash",
- "laundry_care_washer_program_dessous": "Dessous",
- "laundry_care_washer_program_monsoon": "Monsoon",
- "laundry_care_washer_program_outdoor": "Outdoor",
- "laundry_care_washer_program_plush_toy": "Plush toy",
- "laundry_care_washer_program_shirts_blouses": "Shirts blouses",
- "laundry_care_washer_program_sport_fitness": "Sport fitness",
- "laundry_care_washer_program_towels": "Towels",
- "laundry_care_washer_program_water_proof": "Water proof",
- "laundry_care_washer_program_power_speed_59": "Power speed <60 min",
- "laundry_care_washer_program_super_153045_super_15": "Super 15 min",
- "laundry_care_washer_program_super_153045_super_1530": "Super 15/30 min",
- "laundry_care_washer_program_down_duvet_duvet": "Down duvet",
- "laundry_care_washer_program_rinse_rinse_spin_drain": "Rinse spin drain",
- "laundry_care_washer_program_drum_clean": "Drum clean",
- "laundry_care_washer_dryer_program_cotton": "Cotton",
- "laundry_care_washer_dryer_program_cotton_eco_4060": "Cotton eco 40/60 ºC",
- "laundry_care_washer_dryer_program_mix": "Mix",
- "laundry_care_washer_dryer_program_easy_care": "Easy care",
- "laundry_care_washer_dryer_program_wash_and_dry_60": "Wash and dry (60 min)",
- "laundry_care_washer_dryer_program_wash_and_dry_90": "Wash and dry (90 min)"
+ "consumer_products_cleaning_robot_program_cleaning_clean_all": "[%key:component::home_connect::selector::programs::options::consumer_products_cleaning_robot_program_cleaning_clean_all%]",
+ "consumer_products_cleaning_robot_program_cleaning_clean_map": "[%key:component::home_connect::selector::programs::options::consumer_products_cleaning_robot_program_cleaning_clean_map%]",
+ "consumer_products_cleaning_robot_program_basic_go_home": "[%key:component::home_connect::selector::programs::options::consumer_products_cleaning_robot_program_basic_go_home%]",
+ "consumer_products_coffee_maker_program_beverage_ristretto": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_ristretto%]",
+ "consumer_products_coffee_maker_program_beverage_espresso": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_espresso%]",
+ "consumer_products_coffee_maker_program_beverage_espresso_doppio": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_espresso_doppio%]",
+ "consumer_products_coffee_maker_program_beverage_coffee": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_coffee%]",
+ "consumer_products_coffee_maker_program_beverage_x_l_coffee": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_x_l_coffee%]",
+ "consumer_products_coffee_maker_program_beverage_caffe_grande": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_caffe_grande%]",
+ "consumer_products_coffee_maker_program_beverage_espresso_macchiato": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_espresso_macchiato%]",
+ "consumer_products_coffee_maker_program_beverage_cappuccino": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_cappuccino%]",
+ "consumer_products_coffee_maker_program_beverage_latte_macchiato": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_latte_macchiato%]",
+ "consumer_products_coffee_maker_program_beverage_caffe_latte": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_caffe_latte%]",
+ "consumer_products_coffee_maker_program_beverage_milk_froth": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_milk_froth%]",
+ "consumer_products_coffee_maker_program_beverage_warm_milk": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_warm_milk%]",
+ "consumer_products_coffee_maker_program_coffee_world_kleiner_brauner": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_kleiner_brauner%]",
+ "consumer_products_coffee_maker_program_coffee_world_grosser_brauner": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_grosser_brauner%]",
+ "consumer_products_coffee_maker_program_coffee_world_verlaengerter": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_verlaengerter%]",
+ "consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun%]",
+ "consumer_products_coffee_maker_program_coffee_world_wiener_melange": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_wiener_melange%]",
+ "consumer_products_coffee_maker_program_coffee_world_flat_white": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_flat_white%]",
+ "consumer_products_coffee_maker_program_coffee_world_cortado": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cortado%]",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_cortado": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cafe_cortado%]",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_con_leche": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cafe_con_leche%]",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_au_lait": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cafe_au_lait%]",
+ "consumer_products_coffee_maker_program_coffee_world_doppio": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_doppio%]",
+ "consumer_products_coffee_maker_program_coffee_world_kaapi": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_kaapi%]",
+ "consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd%]",
+ "consumer_products_coffee_maker_program_coffee_world_galao": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_galao%]",
+ "consumer_products_coffee_maker_program_coffee_world_garoto": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_garoto%]",
+ "consumer_products_coffee_maker_program_coffee_world_americano": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_americano%]",
+ "consumer_products_coffee_maker_program_coffee_world_red_eye": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_red_eye%]",
+ "consumer_products_coffee_maker_program_coffee_world_black_eye": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_black_eye%]",
+ "consumer_products_coffee_maker_program_coffee_world_dead_eye": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_dead_eye%]",
+ "consumer_products_coffee_maker_program_beverage_hot_water": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_hot_water%]",
+ "dishcare_dishwasher_program_pre_rinse": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_pre_rinse%]",
+ "dishcare_dishwasher_program_auto_1": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_1%]",
+ "dishcare_dishwasher_program_auto_2": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_2%]",
+ "dishcare_dishwasher_program_auto_3": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_3%]",
+ "dishcare_dishwasher_program_eco_50": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_eco_50%]",
+ "dishcare_dishwasher_program_quick_45": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_quick_45%]",
+ "dishcare_dishwasher_program_intensiv_70": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_intensiv_70%]",
+ "dishcare_dishwasher_program_normal_65": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_normal_65%]",
+ "dishcare_dishwasher_program_glas_40": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_glas_40%]",
+ "dishcare_dishwasher_program_glass_care": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_glass_care%]",
+ "dishcare_dishwasher_program_night_wash": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_night_wash%]",
+ "dishcare_dishwasher_program_quick_65": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_quick_65%]",
+ "dishcare_dishwasher_program_normal_45": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_normal_45%]",
+ "dishcare_dishwasher_program_intensiv_45": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_intensiv_45%]",
+ "dishcare_dishwasher_program_auto_half_load": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_half_load%]",
+ "dishcare_dishwasher_program_intensiv_power": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_intensiv_power%]",
+ "dishcare_dishwasher_program_magic_daily": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_magic_daily%]",
+ "dishcare_dishwasher_program_super_60": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_super_60%]",
+ "dishcare_dishwasher_program_kurz_60": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_kurz_60%]",
+ "dishcare_dishwasher_program_express_sparkle_65": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_express_sparkle_65%]",
+ "dishcare_dishwasher_program_machine_care": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_machine_care%]",
+ "dishcare_dishwasher_program_steam_fresh": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_steam_fresh%]",
+ "dishcare_dishwasher_program_maximum_cleaning": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_maximum_cleaning%]",
+ "dishcare_dishwasher_program_mixed_load": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_mixed_load%]",
+ "laundry_care_dryer_program_cotton": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_cotton%]",
+ "laundry_care_dryer_program_synthetic": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_synthetic%]",
+ "laundry_care_dryer_program_mix": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_mix%]",
+ "laundry_care_dryer_program_blankets": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_blankets%]",
+ "laundry_care_dryer_program_business_shirts": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_business_shirts%]",
+ "laundry_care_dryer_program_down_feathers": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_down_feathers%]",
+ "laundry_care_dryer_program_hygiene": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_hygiene%]",
+ "laundry_care_dryer_program_jeans": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_jeans%]",
+ "laundry_care_dryer_program_outdoor": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_outdoor%]",
+ "laundry_care_dryer_program_synthetic_refresh": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_synthetic_refresh%]",
+ "laundry_care_dryer_program_towels": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_towels%]",
+ "laundry_care_dryer_program_delicates": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_delicates%]",
+ "laundry_care_dryer_program_super_40": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_super_40%]",
+ "laundry_care_dryer_program_shirts_15": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_shirts_15%]",
+ "laundry_care_dryer_program_pillow": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_pillow%]",
+ "laundry_care_dryer_program_anti_shrink": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_anti_shrink%]",
+ "laundry_care_dryer_program_my_time_my_drying_time": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_my_time_my_drying_time%]",
+ "laundry_care_dryer_program_time_cold": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold%]",
+ "laundry_care_dryer_program_time_warm": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm%]",
+ "laundry_care_dryer_program_in_basket": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_in_basket%]",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_20": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold_fix_time_cold_20%]",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_30": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold_fix_time_cold_30%]",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_60": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold_fix_time_cold_60%]",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_30": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm_fix_time_warm_30%]",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_40": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm_fix_time_warm_40%]",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_60": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm_fix_time_warm_60%]",
+ "laundry_care_dryer_program_dessous": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_dessous%]",
+ "cooking_common_program_hood_automatic": "[%key:component::home_connect::selector::programs::options::cooking_common_program_hood_automatic%]",
+ "cooking_common_program_hood_venting": "[%key:component::home_connect::selector::programs::options::cooking_common_program_hood_venting%]",
+ "cooking_common_program_hood_delayed_shut_off": "[%key:component::home_connect::selector::programs::options::cooking_common_program_hood_delayed_shut_off%]",
+ "cooking_oven_program_heating_mode_pre_heating": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_pre_heating%]",
+ "cooking_oven_program_heating_mode_hot_air": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air%]",
+ "cooking_oven_program_heating_mode_hot_air_eco": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_eco%]",
+ "cooking_oven_program_heating_mode_hot_air_grilling": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_grilling%]",
+ "cooking_oven_program_heating_mode_top_bottom_heating": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_top_bottom_heating%]",
+ "cooking_oven_program_heating_mode_top_bottom_heating_eco": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_top_bottom_heating_eco%]",
+ "cooking_oven_program_heating_mode_bottom_heating": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_bottom_heating%]",
+ "cooking_oven_program_heating_mode_pizza_setting": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_pizza_setting%]",
+ "cooking_oven_program_heating_mode_slow_cook": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_slow_cook%]",
+ "cooking_oven_program_heating_mode_intensive_heat": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_intensive_heat%]",
+ "cooking_oven_program_heating_mode_keep_warm": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_keep_warm%]",
+ "cooking_oven_program_heating_mode_preheat_ovenware": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_preheat_ovenware%]",
+ "cooking_oven_program_heating_mode_frozen_heatup_special": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_frozen_heatup_special%]",
+ "cooking_oven_program_heating_mode_desiccation": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_desiccation%]",
+ "cooking_oven_program_heating_mode_defrost": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_defrost%]",
+ "cooking_oven_program_heating_mode_proof": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_proof%]",
+ "cooking_oven_program_heating_mode_hot_air_30_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_30_steam%]",
+ "cooking_oven_program_heating_mode_hot_air_60_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_60_steam%]",
+ "cooking_oven_program_heating_mode_hot_air_80_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_80_steam%]",
+ "cooking_oven_program_heating_mode_hot_air_100_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_100_steam%]",
+ "cooking_oven_program_heating_mode_sabbath_programme": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_sabbath_programme%]",
+ "cooking_oven_program_microwave_90_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_90_watt%]",
+ "cooking_oven_program_microwave_180_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_180_watt%]",
+ "cooking_oven_program_microwave_360_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_360_watt%]",
+ "cooking_oven_program_microwave_600_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_600_watt%]",
+ "cooking_oven_program_microwave_900_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_900_watt%]",
+ "cooking_oven_program_microwave_1000_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_1000_watt%]",
+ "cooking_oven_program_microwave_max": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_max%]",
+ "cooking_oven_program_heating_mode_warming_drawer": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_warming_drawer%]",
+ "laundry_care_washer_program_cotton": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton%]",
+ "laundry_care_washer_program_cotton_cotton_eco": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton_cotton_eco%]",
+ "laundry_care_washer_program_cotton_eco_4060": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton_eco_4060%]",
+ "laundry_care_washer_program_cotton_colour": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton_colour%]",
+ "laundry_care_washer_program_easy_care": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_easy_care%]",
+ "laundry_care_washer_program_mix": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_mix%]",
+ "laundry_care_washer_program_mix_night_wash": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_mix_night_wash%]",
+ "laundry_care_washer_program_delicates_silk": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_delicates_silk%]",
+ "laundry_care_washer_program_wool": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_wool%]",
+ "laundry_care_washer_program_sensitive": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_sensitive%]",
+ "laundry_care_washer_program_auto_30": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_auto_30%]",
+ "laundry_care_washer_program_auto_40": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_auto_40%]",
+ "laundry_care_washer_program_auto_60": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_auto_60%]",
+ "laundry_care_washer_program_chiffon": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_chiffon%]",
+ "laundry_care_washer_program_curtains": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_curtains%]",
+ "laundry_care_washer_program_dark_wash": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_dark_wash%]",
+ "laundry_care_washer_program_dessous": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_dessous%]",
+ "laundry_care_washer_program_monsoon": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_monsoon%]",
+ "laundry_care_washer_program_outdoor": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_outdoor%]",
+ "laundry_care_washer_program_plush_toy": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_plush_toy%]",
+ "laundry_care_washer_program_shirts_blouses": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_shirts_blouses%]",
+ "laundry_care_washer_program_sport_fitness": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_sport_fitness%]",
+ "laundry_care_washer_program_towels": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_towels%]",
+ "laundry_care_washer_program_water_proof": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_water_proof%]",
+ "laundry_care_washer_program_power_speed_59": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_power_speed_59%]",
+ "laundry_care_washer_program_super_153045_super_15": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_super_153045_super_15%]",
+ "laundry_care_washer_program_super_153045_super_1530": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_super_153045_super_1530%]",
+ "laundry_care_washer_program_down_duvet_duvet": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_down_duvet_duvet%]",
+ "laundry_care_washer_program_rinse_rinse_spin_drain": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_rinse_rinse_spin_drain%]",
+ "laundry_care_washer_program_drum_clean": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_drum_clean%]",
+ "laundry_care_washer_dryer_program_cotton": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_cotton%]",
+ "laundry_care_washer_dryer_program_cotton_eco_4060": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_cotton_eco_4060%]",
+ "laundry_care_washer_dryer_program_mix": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_mix%]",
+ "laundry_care_washer_dryer_program_easy_care": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_easy_care%]",
+ "laundry_care_washer_dryer_program_wash_and_dry_60": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_wash_and_dry_60%]",
+ "laundry_care_washer_dryer_program_wash_and_dry_90": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_wash_and_dry_90%]"
}
},
"active_program": {
"name": "Active program",
"state": {
- "consumer_products_cleaning_robot_program_cleaning_clean_all": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_cleaning_robot_program_cleaning_clean_all%]",
- "consumer_products_cleaning_robot_program_cleaning_clean_map": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_cleaning_robot_program_cleaning_clean_map%]",
- "consumer_products_cleaning_robot_program_basic_go_home": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_cleaning_robot_program_basic_go_home%]",
- "consumer_products_coffee_maker_program_beverage_ristretto": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_ristretto%]",
- "consumer_products_coffee_maker_program_beverage_espresso": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_espresso%]",
- "consumer_products_coffee_maker_program_beverage_espresso_doppio": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_espresso_doppio%]",
- "consumer_products_coffee_maker_program_beverage_coffee": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_coffee%]",
- "consumer_products_coffee_maker_program_beverage_x_l_coffee": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_x_l_coffee%]",
- "consumer_products_coffee_maker_program_beverage_caffe_grande": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_caffe_grande%]",
- "consumer_products_coffee_maker_program_beverage_espresso_macchiato": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_espresso_macchiato%]",
- "consumer_products_coffee_maker_program_beverage_cappuccino": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_cappuccino%]",
- "consumer_products_coffee_maker_program_beverage_latte_macchiato": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_latte_macchiato%]",
- "consumer_products_coffee_maker_program_beverage_caffe_latte": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_caffe_latte%]",
- "consumer_products_coffee_maker_program_beverage_milk_froth": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_milk_froth%]",
- "consumer_products_coffee_maker_program_beverage_warm_milk": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_warm_milk%]",
- "consumer_products_coffee_maker_program_coffee_world_kleiner_brauner": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_kleiner_brauner%]",
- "consumer_products_coffee_maker_program_coffee_world_grosser_brauner": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_grosser_brauner%]",
- "consumer_products_coffee_maker_program_coffee_world_verlaengerter": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_verlaengerter%]",
- "consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun%]",
- "consumer_products_coffee_maker_program_coffee_world_wiener_melange": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_wiener_melange%]",
- "consumer_products_coffee_maker_program_coffee_world_flat_white": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_flat_white%]",
- "consumer_products_coffee_maker_program_coffee_world_cortado": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_cortado%]",
- "consumer_products_coffee_maker_program_coffee_world_cafe_cortado": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_cafe_cortado%]",
- "consumer_products_coffee_maker_program_coffee_world_cafe_con_leche": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_cafe_con_leche%]",
- "consumer_products_coffee_maker_program_coffee_world_cafe_au_lait": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_cafe_au_lait%]",
- "consumer_products_coffee_maker_program_coffee_world_doppio": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_doppio%]",
- "consumer_products_coffee_maker_program_coffee_world_kaapi": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_kaapi%]",
- "consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd%]",
- "consumer_products_coffee_maker_program_coffee_world_galao": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_galao%]",
- "consumer_products_coffee_maker_program_coffee_world_garoto": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_garoto%]",
- "consumer_products_coffee_maker_program_coffee_world_americano": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_americano%]",
- "consumer_products_coffee_maker_program_coffee_world_red_eye": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_red_eye%]",
- "consumer_products_coffee_maker_program_coffee_world_black_eye": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_black_eye%]",
- "consumer_products_coffee_maker_program_coffee_world_dead_eye": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_coffee_world_dead_eye%]",
- "consumer_products_coffee_maker_program_beverage_hot_water": "[%key:component::home_connect::entity::select::selected_program::state::consumer_products_coffee_maker_program_beverage_hot_water%]",
- "dishcare_dishwasher_program_pre_rinse": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_pre_rinse%]",
- "dishcare_dishwasher_program_auto_1": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_auto_1%]",
- "dishcare_dishwasher_program_auto_2": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_auto_2%]",
- "dishcare_dishwasher_program_auto_3": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_auto_3%]",
- "dishcare_dishwasher_program_eco_50": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_eco_50%]",
- "dishcare_dishwasher_program_quick_45": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_quick_45%]",
- "dishcare_dishwasher_program_intensiv_70": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_intensiv_70%]",
- "dishcare_dishwasher_program_normal_65": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_normal_65%]",
- "dishcare_dishwasher_program_glas_40": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_glas_40%]",
- "dishcare_dishwasher_program_glass_care": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_glass_care%]",
- "dishcare_dishwasher_program_night_wash": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_night_wash%]",
- "dishcare_dishwasher_program_quick_65": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_quick_65%]",
- "dishcare_dishwasher_program_normal_45": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_normal_45%]",
- "dishcare_dishwasher_program_intensiv_45": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_intensiv_45%]",
- "dishcare_dishwasher_program_auto_half_load": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_auto_half_load%]",
- "dishcare_dishwasher_program_intensiv_power": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_intensiv_power%]",
- "dishcare_dishwasher_program_magic_daily": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_magic_daily%]",
- "dishcare_dishwasher_program_super_60": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_super_60%]",
- "dishcare_dishwasher_program_kurz_60": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_kurz_60%]",
- "dishcare_dishwasher_program_express_sparkle_65": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_express_sparkle_65%]",
- "dishcare_dishwasher_program_machine_care": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_machine_care%]",
- "dishcare_dishwasher_program_steam_fresh": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_steam_fresh%]",
- "dishcare_dishwasher_program_maximum_cleaning": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_maximum_cleaning%]",
- "dishcare_dishwasher_program_mixed_load": "[%key:component::home_connect::entity::select::selected_program::state::dishcare_dishwasher_program_mixed_load%]",
- "laundry_care_dryer_program_cotton": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_cotton%]",
- "laundry_care_dryer_program_synthetic": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_synthetic%]",
- "laundry_care_dryer_program_mix": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_mix%]",
- "laundry_care_dryer_program_blankets": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_blankets%]",
- "laundry_care_dryer_program_business_shirts": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_business_shirts%]",
- "laundry_care_dryer_program_down_feathers": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_down_feathers%]",
- "laundry_care_dryer_program_hygiene": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_hygiene%]",
- "laundry_care_dryer_program_jeans": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_jeans%]",
- "laundry_care_dryer_program_outdoor": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_outdoor%]",
- "laundry_care_dryer_program_synthetic_refresh": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_synthetic_refresh%]",
- "laundry_care_dryer_program_towels": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_towels%]",
- "laundry_care_dryer_program_delicates": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_delicates%]",
- "laundry_care_dryer_program_super_40": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_super_40%]",
- "laundry_care_dryer_program_shirts_15": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_shirts_15%]",
- "laundry_care_dryer_program_pillow": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_pillow%]",
- "laundry_care_dryer_program_anti_shrink": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_anti_shrink%]",
- "laundry_care_dryer_program_my_time_my_drying_time": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_my_time_my_drying_time%]",
- "laundry_care_dryer_program_time_cold": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_cold%]",
- "laundry_care_dryer_program_time_warm": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_warm%]",
- "laundry_care_dryer_program_in_basket": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_in_basket%]",
- "laundry_care_dryer_program_time_cold_fix_time_cold_20": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_cold_fix_time_cold_20%]",
- "laundry_care_dryer_program_time_cold_fix_time_cold_30": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_cold_fix_time_cold_30%]",
- "laundry_care_dryer_program_time_cold_fix_time_cold_60": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_cold_fix_time_cold_60%]",
- "laundry_care_dryer_program_time_warm_fix_time_warm_30": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_warm_fix_time_warm_30%]",
- "laundry_care_dryer_program_time_warm_fix_time_warm_40": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_warm_fix_time_warm_40%]",
- "laundry_care_dryer_program_time_warm_fix_time_warm_60": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_time_warm_fix_time_warm_60%]",
- "laundry_care_dryer_program_dessous": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_dryer_program_dessous%]",
- "cooking_common_program_hood_automatic": "[%key:component::home_connect::entity::select::selected_program::state::cooking_common_program_hood_automatic%]",
- "cooking_common_program_hood_venting": "[%key:component::home_connect::entity::select::selected_program::state::cooking_common_program_hood_venting%]",
- "cooking_common_program_hood_delayed_shut_off": "[%key:component::home_connect::entity::select::selected_program::state::cooking_common_program_hood_delayed_shut_off%]",
- "cooking_oven_program_heating_mode_pre_heating": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_pre_heating%]",
- "cooking_oven_program_heating_mode_hot_air": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_hot_air%]",
- "cooking_oven_program_heating_mode_hot_air_eco": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_hot_air_eco%]",
- "cooking_oven_program_heating_mode_hot_air_grilling": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_hot_air_grilling%]",
- "cooking_oven_program_heating_mode_top_bottom_heating": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_top_bottom_heating%]",
- "cooking_oven_program_heating_mode_top_bottom_heating_eco": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_top_bottom_heating_eco%]",
- "cooking_oven_program_heating_mode_bottom_heating": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_bottom_heating%]",
- "cooking_oven_program_heating_mode_pizza_setting": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_pizza_setting%]",
- "cooking_oven_program_heating_mode_slow_cook": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_slow_cook%]",
- "cooking_oven_program_heating_mode_intensive_heat": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_intensive_heat%]",
- "cooking_oven_program_heating_mode_keep_warm": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_keep_warm%]",
- "cooking_oven_program_heating_mode_preheat_ovenware": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_preheat_ovenware%]",
- "cooking_oven_program_heating_mode_frozen_heatup_special": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_frozen_heatup_special%]",
- "cooking_oven_program_heating_mode_desiccation": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_desiccation%]",
- "cooking_oven_program_heating_mode_defrost": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_defrost%]",
- "cooking_oven_program_heating_mode_proof": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_proof%]",
- "cooking_oven_program_heating_mode_hot_air_30_steam": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_hot_air_30_steam%]",
- "cooking_oven_program_heating_mode_hot_air_60_steam": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_hot_air_60_steam%]",
- "cooking_oven_program_heating_mode_hot_air_80_steam": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_hot_air_80_steam%]",
- "cooking_oven_program_heating_mode_hot_air_100_steam": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_hot_air_100_steam%]",
- "cooking_oven_program_heating_mode_sabbath_programme": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_sabbath_programme%]",
- "cooking_oven_program_microwave_90_watt": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_microwave_90_watt%]",
- "cooking_oven_program_microwave_180_watt": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_microwave_180_watt%]",
- "cooking_oven_program_microwave_360_watt": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_microwave_360_watt%]",
- "cooking_oven_program_microwave_600_watt": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_microwave_600_watt%]",
- "cooking_oven_program_microwave_900_watt": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_microwave_900_watt%]",
- "cooking_oven_program_microwave_1000_watt": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_microwave_1000_watt%]",
- "cooking_oven_program_microwave_max": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_microwave_max%]",
- "cooking_oven_program_heating_mode_warming_drawer": "[%key:component::home_connect::entity::select::selected_program::state::cooking_oven_program_heating_mode_warming_drawer%]",
- "laundry_care_washer_program_cotton": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_cotton%]",
- "laundry_care_washer_program_cotton_cotton_eco": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_cotton_cotton_eco%]",
- "laundry_care_washer_program_cotton_eco_4060": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_cotton_eco_4060%]",
- "laundry_care_washer_program_cotton_colour": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_cotton_colour%]",
- "laundry_care_washer_program_easy_care": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_easy_care%]",
- "laundry_care_washer_program_mix": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_mix%]",
- "laundry_care_washer_program_mix_night_wash": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_mix_night_wash%]",
- "laundry_care_washer_program_delicates_silk": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_delicates_silk%]",
- "laundry_care_washer_program_wool": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_wool%]",
- "laundry_care_washer_program_sensitive": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_sensitive%]",
- "laundry_care_washer_program_auto_30": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_auto_30%]",
- "laundry_care_washer_program_auto_40": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_auto_40%]",
- "laundry_care_washer_program_auto_60": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_auto_60%]",
- "laundry_care_washer_program_chiffon": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_chiffon%]",
- "laundry_care_washer_program_curtains": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_curtains%]",
- "laundry_care_washer_program_dark_wash": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_dark_wash%]",
- "laundry_care_washer_program_dessous": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_dessous%]",
- "laundry_care_washer_program_monsoon": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_monsoon%]",
- "laundry_care_washer_program_outdoor": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_outdoor%]",
- "laundry_care_washer_program_plush_toy": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_plush_toy%]",
- "laundry_care_washer_program_shirts_blouses": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_shirts_blouses%]",
- "laundry_care_washer_program_sport_fitness": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_sport_fitness%]",
- "laundry_care_washer_program_towels": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_towels%]",
- "laundry_care_washer_program_water_proof": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_water_proof%]",
- "laundry_care_washer_program_power_speed_59": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_power_speed_59%]",
- "laundry_care_washer_program_super_153045_super_15": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_super_153045_super_15%]",
- "laundry_care_washer_program_super_153045_super_1530": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_super_153045_super_1530%]",
- "laundry_care_washer_program_down_duvet_duvet": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_down_duvet_duvet%]",
- "laundry_care_washer_program_rinse_rinse_spin_drain": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_rinse_rinse_spin_drain%]",
- "laundry_care_washer_program_drum_clean": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_program_drum_clean%]",
- "laundry_care_washer_dryer_program_cotton": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_dryer_program_cotton%]",
- "laundry_care_washer_dryer_program_cotton_eco_4060": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_dryer_program_cotton_eco_4060%]",
- "laundry_care_washer_dryer_program_mix": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_dryer_program_mix%]",
- "laundry_care_washer_dryer_program_easy_care": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_dryer_program_easy_care%]",
- "laundry_care_washer_dryer_program_wash_and_dry_60": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_dryer_program_wash_and_dry_60%]",
- "laundry_care_washer_dryer_program_wash_and_dry_90": "[%key:component::home_connect::entity::select::selected_program::state::laundry_care_washer_dryer_program_wash_and_dry_90%]"
+ "consumer_products_cleaning_robot_program_cleaning_clean_all": "[%key:component::home_connect::selector::programs::options::consumer_products_cleaning_robot_program_cleaning_clean_all%]",
+ "consumer_products_cleaning_robot_program_cleaning_clean_map": "[%key:component::home_connect::selector::programs::options::consumer_products_cleaning_robot_program_cleaning_clean_map%]",
+ "consumer_products_cleaning_robot_program_basic_go_home": "[%key:component::home_connect::selector::programs::options::consumer_products_cleaning_robot_program_basic_go_home%]",
+ "consumer_products_coffee_maker_program_beverage_ristretto": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_ristretto%]",
+ "consumer_products_coffee_maker_program_beverage_espresso": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_espresso%]",
+ "consumer_products_coffee_maker_program_beverage_espresso_doppio": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_espresso_doppio%]",
+ "consumer_products_coffee_maker_program_beverage_coffee": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_coffee%]",
+ "consumer_products_coffee_maker_program_beverage_x_l_coffee": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_x_l_coffee%]",
+ "consumer_products_coffee_maker_program_beverage_caffe_grande": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_caffe_grande%]",
+ "consumer_products_coffee_maker_program_beverage_espresso_macchiato": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_espresso_macchiato%]",
+ "consumer_products_coffee_maker_program_beverage_cappuccino": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_cappuccino%]",
+ "consumer_products_coffee_maker_program_beverage_latte_macchiato": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_latte_macchiato%]",
+ "consumer_products_coffee_maker_program_beverage_caffe_latte": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_caffe_latte%]",
+ "consumer_products_coffee_maker_program_beverage_milk_froth": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_milk_froth%]",
+ "consumer_products_coffee_maker_program_beverage_warm_milk": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_warm_milk%]",
+ "consumer_products_coffee_maker_program_coffee_world_kleiner_brauner": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_kleiner_brauner%]",
+ "consumer_products_coffee_maker_program_coffee_world_grosser_brauner": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_grosser_brauner%]",
+ "consumer_products_coffee_maker_program_coffee_world_verlaengerter": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_verlaengerter%]",
+ "consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_verlaengerter_braun%]",
+ "consumer_products_coffee_maker_program_coffee_world_wiener_melange": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_wiener_melange%]",
+ "consumer_products_coffee_maker_program_coffee_world_flat_white": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_flat_white%]",
+ "consumer_products_coffee_maker_program_coffee_world_cortado": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cortado%]",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_cortado": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cafe_cortado%]",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_con_leche": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cafe_con_leche%]",
+ "consumer_products_coffee_maker_program_coffee_world_cafe_au_lait": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_cafe_au_lait%]",
+ "consumer_products_coffee_maker_program_coffee_world_doppio": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_doppio%]",
+ "consumer_products_coffee_maker_program_coffee_world_kaapi": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_kaapi%]",
+ "consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_koffie_verkeerd%]",
+ "consumer_products_coffee_maker_program_coffee_world_galao": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_galao%]",
+ "consumer_products_coffee_maker_program_coffee_world_garoto": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_garoto%]",
+ "consumer_products_coffee_maker_program_coffee_world_americano": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_americano%]",
+ "consumer_products_coffee_maker_program_coffee_world_red_eye": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_red_eye%]",
+ "consumer_products_coffee_maker_program_coffee_world_black_eye": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_black_eye%]",
+ "consumer_products_coffee_maker_program_coffee_world_dead_eye": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_coffee_world_dead_eye%]",
+ "consumer_products_coffee_maker_program_beverage_hot_water": "[%key:component::home_connect::selector::programs::options::consumer_products_coffee_maker_program_beverage_hot_water%]",
+ "dishcare_dishwasher_program_pre_rinse": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_pre_rinse%]",
+ "dishcare_dishwasher_program_auto_1": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_1%]",
+ "dishcare_dishwasher_program_auto_2": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_2%]",
+ "dishcare_dishwasher_program_auto_3": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_3%]",
+ "dishcare_dishwasher_program_eco_50": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_eco_50%]",
+ "dishcare_dishwasher_program_quick_45": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_quick_45%]",
+ "dishcare_dishwasher_program_intensiv_70": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_intensiv_70%]",
+ "dishcare_dishwasher_program_normal_65": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_normal_65%]",
+ "dishcare_dishwasher_program_glas_40": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_glas_40%]",
+ "dishcare_dishwasher_program_glass_care": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_glass_care%]",
+ "dishcare_dishwasher_program_night_wash": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_night_wash%]",
+ "dishcare_dishwasher_program_quick_65": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_quick_65%]",
+ "dishcare_dishwasher_program_normal_45": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_normal_45%]",
+ "dishcare_dishwasher_program_intensiv_45": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_intensiv_45%]",
+ "dishcare_dishwasher_program_auto_half_load": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_auto_half_load%]",
+ "dishcare_dishwasher_program_intensiv_power": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_intensiv_power%]",
+ "dishcare_dishwasher_program_magic_daily": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_magic_daily%]",
+ "dishcare_dishwasher_program_super_60": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_super_60%]",
+ "dishcare_dishwasher_program_kurz_60": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_kurz_60%]",
+ "dishcare_dishwasher_program_express_sparkle_65": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_express_sparkle_65%]",
+ "dishcare_dishwasher_program_machine_care": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_machine_care%]",
+ "dishcare_dishwasher_program_steam_fresh": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_steam_fresh%]",
+ "dishcare_dishwasher_program_maximum_cleaning": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_maximum_cleaning%]",
+ "dishcare_dishwasher_program_mixed_load": "[%key:component::home_connect::selector::programs::options::dishcare_dishwasher_program_mixed_load%]",
+ "laundry_care_dryer_program_cotton": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_cotton%]",
+ "laundry_care_dryer_program_synthetic": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_synthetic%]",
+ "laundry_care_dryer_program_mix": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_mix%]",
+ "laundry_care_dryer_program_blankets": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_blankets%]",
+ "laundry_care_dryer_program_business_shirts": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_business_shirts%]",
+ "laundry_care_dryer_program_down_feathers": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_down_feathers%]",
+ "laundry_care_dryer_program_hygiene": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_hygiene%]",
+ "laundry_care_dryer_program_jeans": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_jeans%]",
+ "laundry_care_dryer_program_outdoor": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_outdoor%]",
+ "laundry_care_dryer_program_synthetic_refresh": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_synthetic_refresh%]",
+ "laundry_care_dryer_program_towels": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_towels%]",
+ "laundry_care_dryer_program_delicates": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_delicates%]",
+ "laundry_care_dryer_program_super_40": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_super_40%]",
+ "laundry_care_dryer_program_shirts_15": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_shirts_15%]",
+ "laundry_care_dryer_program_pillow": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_pillow%]",
+ "laundry_care_dryer_program_anti_shrink": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_anti_shrink%]",
+ "laundry_care_dryer_program_my_time_my_drying_time": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_my_time_my_drying_time%]",
+ "laundry_care_dryer_program_time_cold": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold%]",
+ "laundry_care_dryer_program_time_warm": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm%]",
+ "laundry_care_dryer_program_in_basket": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_in_basket%]",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_20": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold_fix_time_cold_20%]",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_30": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold_fix_time_cold_30%]",
+ "laundry_care_dryer_program_time_cold_fix_time_cold_60": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_cold_fix_time_cold_60%]",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_30": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm_fix_time_warm_30%]",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_40": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm_fix_time_warm_40%]",
+ "laundry_care_dryer_program_time_warm_fix_time_warm_60": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_time_warm_fix_time_warm_60%]",
+ "laundry_care_dryer_program_dessous": "[%key:component::home_connect::selector::programs::options::laundry_care_dryer_program_dessous%]",
+ "cooking_common_program_hood_automatic": "[%key:component::home_connect::selector::programs::options::cooking_common_program_hood_automatic%]",
+ "cooking_common_program_hood_venting": "[%key:component::home_connect::selector::programs::options::cooking_common_program_hood_venting%]",
+ "cooking_common_program_hood_delayed_shut_off": "[%key:component::home_connect::selector::programs::options::cooking_common_program_hood_delayed_shut_off%]",
+ "cooking_oven_program_heating_mode_pre_heating": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_pre_heating%]",
+ "cooking_oven_program_heating_mode_hot_air": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air%]",
+ "cooking_oven_program_heating_mode_hot_air_eco": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_eco%]",
+ "cooking_oven_program_heating_mode_hot_air_grilling": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_grilling%]",
+ "cooking_oven_program_heating_mode_top_bottom_heating": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_top_bottom_heating%]",
+ "cooking_oven_program_heating_mode_top_bottom_heating_eco": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_top_bottom_heating_eco%]",
+ "cooking_oven_program_heating_mode_bottom_heating": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_bottom_heating%]",
+ "cooking_oven_program_heating_mode_pizza_setting": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_pizza_setting%]",
+ "cooking_oven_program_heating_mode_slow_cook": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_slow_cook%]",
+ "cooking_oven_program_heating_mode_intensive_heat": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_intensive_heat%]",
+ "cooking_oven_program_heating_mode_keep_warm": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_keep_warm%]",
+ "cooking_oven_program_heating_mode_preheat_ovenware": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_preheat_ovenware%]",
+ "cooking_oven_program_heating_mode_frozen_heatup_special": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_frozen_heatup_special%]",
+ "cooking_oven_program_heating_mode_desiccation": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_desiccation%]",
+ "cooking_oven_program_heating_mode_defrost": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_defrost%]",
+ "cooking_oven_program_heating_mode_proof": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_proof%]",
+ "cooking_oven_program_heating_mode_hot_air_30_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_30_steam%]",
+ "cooking_oven_program_heating_mode_hot_air_60_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_60_steam%]",
+ "cooking_oven_program_heating_mode_hot_air_80_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_80_steam%]",
+ "cooking_oven_program_heating_mode_hot_air_100_steam": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_hot_air_100_steam%]",
+ "cooking_oven_program_heating_mode_sabbath_programme": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_sabbath_programme%]",
+ "cooking_oven_program_microwave_90_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_90_watt%]",
+ "cooking_oven_program_microwave_180_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_180_watt%]",
+ "cooking_oven_program_microwave_360_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_360_watt%]",
+ "cooking_oven_program_microwave_600_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_600_watt%]",
+ "cooking_oven_program_microwave_900_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_900_watt%]",
+ "cooking_oven_program_microwave_1000_watt": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_1000_watt%]",
+ "cooking_oven_program_microwave_max": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_microwave_max%]",
+ "cooking_oven_program_heating_mode_warming_drawer": "[%key:component::home_connect::selector::programs::options::cooking_oven_program_heating_mode_warming_drawer%]",
+ "laundry_care_washer_program_cotton": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton%]",
+ "laundry_care_washer_program_cotton_cotton_eco": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton_cotton_eco%]",
+ "laundry_care_washer_program_cotton_eco_4060": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton_eco_4060%]",
+ "laundry_care_washer_program_cotton_colour": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_cotton_colour%]",
+ "laundry_care_washer_program_easy_care": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_easy_care%]",
+ "laundry_care_washer_program_mix": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_mix%]",
+ "laundry_care_washer_program_mix_night_wash": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_mix_night_wash%]",
+ "laundry_care_washer_program_delicates_silk": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_delicates_silk%]",
+ "laundry_care_washer_program_wool": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_wool%]",
+ "laundry_care_washer_program_sensitive": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_sensitive%]",
+ "laundry_care_washer_program_auto_30": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_auto_30%]",
+ "laundry_care_washer_program_auto_40": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_auto_40%]",
+ "laundry_care_washer_program_auto_60": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_auto_60%]",
+ "laundry_care_washer_program_chiffon": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_chiffon%]",
+ "laundry_care_washer_program_curtains": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_curtains%]",
+ "laundry_care_washer_program_dark_wash": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_dark_wash%]",
+ "laundry_care_washer_program_dessous": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_dessous%]",
+ "laundry_care_washer_program_monsoon": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_monsoon%]",
+ "laundry_care_washer_program_outdoor": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_outdoor%]",
+ "laundry_care_washer_program_plush_toy": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_plush_toy%]",
+ "laundry_care_washer_program_shirts_blouses": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_shirts_blouses%]",
+ "laundry_care_washer_program_sport_fitness": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_sport_fitness%]",
+ "laundry_care_washer_program_towels": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_towels%]",
+ "laundry_care_washer_program_water_proof": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_water_proof%]",
+ "laundry_care_washer_program_power_speed_59": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_power_speed_59%]",
+ "laundry_care_washer_program_super_153045_super_15": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_super_153045_super_15%]",
+ "laundry_care_washer_program_super_153045_super_1530": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_super_153045_super_1530%]",
+ "laundry_care_washer_program_down_duvet_duvet": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_down_duvet_duvet%]",
+ "laundry_care_washer_program_rinse_rinse_spin_drain": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_rinse_rinse_spin_drain%]",
+ "laundry_care_washer_program_drum_clean": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_program_drum_clean%]",
+ "laundry_care_washer_dryer_program_cotton": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_cotton%]",
+ "laundry_care_washer_dryer_program_cotton_eco_4060": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_cotton_eco_4060%]",
+ "laundry_care_washer_dryer_program_mix": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_mix%]",
+ "laundry_care_washer_dryer_program_easy_care": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_easy_care%]",
+ "laundry_care_washer_dryer_program_wash_and_dry_60": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_wash_and_dry_60%]",
+ "laundry_care_washer_dryer_program_wash_and_dry_90": "[%key:component::home_connect::selector::programs::options::laundry_care_washer_dryer_program_wash_and_dry_90%]"
+ }
+ },
+ "current_map": {
+ "name": "Current map",
+ "state": {
+ "consumer_products_cleaning_robot_enum_type_available_maps_temp_map": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_temp_map%]",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map1": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_map1%]",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map2": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_map2%]",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map3": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_map3%]"
+ }
+ },
+ "functional_light_color_temperature": {
+ "name": "Functional light color temperature",
+ "state": {
+ "cooking_hood_enum_type_color_temperature_custom": "Custom",
+ "cooking_hood_enum_type_color_temperature_warm": "Warm",
+ "cooking_hood_enum_type_color_temperature_warm_to_neutral": "Warm to Neutral",
+ "cooking_hood_enum_type_color_temperature_neutral": "Neutral",
+ "cooking_hood_enum_type_color_temperature_neutral_to_cold": "Neutral to Cold",
+ "cooking_hood_enum_type_color_temperature_cold": "Cold"
+ }
+ },
+ "ambient_light_color": {
+ "name": "Ambient light color",
+ "state": {
+ "b_s_h_common_enum_type_ambient_light_color_custom_color": "Custom"
+ }
+ },
+ "reference_map_id": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_cleaning_robot_option_reference_map_id::name%]",
+ "state": {
+ "consumer_products_cleaning_robot_enum_type_available_maps_temp_map": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_temp_map%]",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map1": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_map1%]",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map2": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_map2%]",
+ "consumer_products_cleaning_robot_enum_type_available_maps_map3": "[%key:component::home_connect::selector::available_maps::options::consumer_products_cleaning_robot_enum_type_available_maps_map3%]"
+ }
+ },
+ "cleaning_mode": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_cleaning_robot_option_cleaning_mode::name%]",
+ "state": {
+ "consumer_products_cleaning_robot_enum_type_cleaning_modes_silent": "[%key:component::home_connect::selector::cleaning_mode::options::consumer_products_cleaning_robot_enum_type_cleaning_modes_silent%]",
+ "consumer_products_cleaning_robot_enum_type_cleaning_modes_standard": "[%key:component::home_connect::selector::cleaning_mode::options::consumer_products_cleaning_robot_enum_type_cleaning_modes_standard%]",
+ "consumer_products_cleaning_robot_enum_type_cleaning_modes_power": "[%key:component::home_connect::selector::cleaning_mode::options::consumer_products_cleaning_robot_enum_type_cleaning_modes_power%]"
+ }
+ },
+ "bean_amount": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_bean_amount::name%]",
+ "state": {
+ "consumer_products_coffee_maker_enum_type_bean_amount_very_mild": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_very_mild%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_mild": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_mild%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_mild_plus": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_mild_plus%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_normal": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_normal%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_normal_plus": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_normal_plus%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_strong": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_strong%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_strong_plus": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_strong_plus%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_very_strong": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_very_strong%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_very_strong_plus": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_very_strong_plus%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_extra_strong": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_extra_strong%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_double_shot": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_double_shot%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus_plus": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_double_shot_plus_plus%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_triple_shot": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_triple_shot%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_triple_shot_plus": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_triple_shot_plus%]",
+ "consumer_products_coffee_maker_enum_type_bean_amount_coffee_ground": "[%key:component::home_connect::selector::bean_amount::options::consumer_products_coffee_maker_enum_type_bean_amount_coffee_ground%]"
+ }
+ },
+ "coffee_temperature": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_coffee_temperature::name%]",
+ "state": {
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_88_c": "[%key:component::home_connect::selector::coffee_temperature::options::consumer_products_coffee_maker_enum_type_coffee_temperature_88_c%]",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_90_c": "[%key:component::home_connect::selector::coffee_temperature::options::consumer_products_coffee_maker_enum_type_coffee_temperature_90_c%]",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_92_c": "[%key:component::home_connect::selector::coffee_temperature::options::consumer_products_coffee_maker_enum_type_coffee_temperature_92_c%]",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_94_c": "[%key:component::home_connect::selector::coffee_temperature::options::consumer_products_coffee_maker_enum_type_coffee_temperature_94_c%]",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_95_c": "[%key:component::home_connect::selector::coffee_temperature::options::consumer_products_coffee_maker_enum_type_coffee_temperature_95_c%]",
+ "consumer_products_coffee_maker_enum_type_coffee_temperature_96_c": "[%key:component::home_connect::selector::coffee_temperature::options::consumer_products_coffee_maker_enum_type_coffee_temperature_96_c%]"
+ }
+ },
+ "bean_container": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_bean_container::name%]",
+ "state": {
+ "consumer_products_coffee_maker_enum_type_bean_container_selection_right": "[%key:component::home_connect::selector::bean_container::options::consumer_products_coffee_maker_enum_type_bean_container_selection_right%]",
+ "consumer_products_coffee_maker_enum_type_bean_container_selection_left": "[%key:component::home_connect::selector::bean_container::options::consumer_products_coffee_maker_enum_type_bean_container_selection_left%]"
+ }
+ },
+ "flow_rate": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_flow_rate::name%]",
+ "state": {
+ "consumer_products_coffee_maker_enum_type_flow_rate_normal": "[%key:component::home_connect::selector::flow_rate::options::consumer_products_coffee_maker_enum_type_flow_rate_normal%]",
+ "consumer_products_coffee_maker_enum_type_flow_rate_intense": "[%key:component::home_connect::selector::flow_rate::options::consumer_products_coffee_maker_enum_type_flow_rate_intense%]",
+ "consumer_products_coffee_maker_enum_type_flow_rate_intense_plus": "[%key:component::home_connect::selector::flow_rate::options::consumer_products_coffee_maker_enum_type_flow_rate_intense_plus%]"
+ }
+ },
+ "coffee_milk_ratio": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_coffee_milk_ratio::name%]",
+ "state": {
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_10_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_10_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_20_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_20_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_25_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_25_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_30_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_30_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_40_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_40_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_50_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_50_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_55_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_55_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_60_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_60_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_65_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_65_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_67_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_67_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_70_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_70_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_75_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_75_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_80_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_80_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_85_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_85_percent%]",
+ "consumer_products_coffee_maker_enum_type_coffee_milk_ratio_90_percent": "[%key:component::home_connect::selector::coffee_milk_ratio::options::consumer_products_coffee_maker_enum_type_coffee_milk_ratio_90_percent%]"
+ }
+ },
+ "hot_water_temperature": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_hot_water_temperature::name%]",
+ "state": {
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_white_tea": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_white_tea%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_green_tea": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_green_tea%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_black_tea": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_black_tea%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_50_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_50_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_55_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_55_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_60_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_60_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_65_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_65_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_70_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_70_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_75_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_75_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_80_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_80_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_85_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_85_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_90_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_90_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_95_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_95_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_97_c": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_97_c%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_122_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_122_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_131_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_131_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_140_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_140_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_149_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_149_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_158_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_158_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_167_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_167_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_176_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_176_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_185_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_185_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_194_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_194_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_203_f": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_203_f%]",
+ "consumer_products_coffee_maker_enum_type_hot_water_temperature_max": "[%key:component::home_connect::selector::hot_water_temperature::options::consumer_products_coffee_maker_enum_type_hot_water_temperature_max%]"
+ }
+ },
+ "drying_target": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::laundry_care_dryer_option_drying_target::name%]",
+ "state": {
+ "laundry_care_dryer_enum_type_drying_target_iron_dry": "[%key:component::home_connect::selector::drying_target::options::laundry_care_dryer_enum_type_drying_target_iron_dry%]",
+ "laundry_care_dryer_enum_type_drying_target_gentle_dry": "[%key:component::home_connect::selector::drying_target::options::laundry_care_dryer_enum_type_drying_target_gentle_dry%]",
+ "laundry_care_dryer_enum_type_drying_target_cupboard_dry": "[%key:component::home_connect::selector::drying_target::options::laundry_care_dryer_enum_type_drying_target_cupboard_dry%]",
+ "laundry_care_dryer_enum_type_drying_target_cupboard_dry_plus": "[%key:component::home_connect::selector::drying_target::options::laundry_care_dryer_enum_type_drying_target_cupboard_dry_plus%]",
+ "laundry_care_dryer_enum_type_drying_target_extra_dry": "[%key:component::home_connect::selector::drying_target::options::laundry_care_dryer_enum_type_drying_target_extra_dry%]"
+ }
+ },
+ "venting_level": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::cooking_hood_option_venting_level::name%]",
+ "state": {
+ "cooking_hood_enum_type_stage_fan_off": "[%key:component::home_connect::selector::venting_level::options::cooking_hood_enum_type_stage_fan_off%]",
+ "cooking_hood_enum_type_stage_fan_stage_01": "[%key:component::home_connect::selector::venting_level::options::cooking_hood_enum_type_stage_fan_stage_01%]",
+ "cooking_hood_enum_type_stage_fan_stage_02": "[%key:component::home_connect::selector::venting_level::options::cooking_hood_enum_type_stage_fan_stage_02%]",
+ "cooking_hood_enum_type_stage_fan_stage_03": "[%key:component::home_connect::selector::venting_level::options::cooking_hood_enum_type_stage_fan_stage_03%]",
+ "cooking_hood_enum_type_stage_fan_stage_04": "[%key:component::home_connect::selector::venting_level::options::cooking_hood_enum_type_stage_fan_stage_04%]",
+ "cooking_hood_enum_type_stage_fan_stage_05": "[%key:component::home_connect::selector::venting_level::options::cooking_hood_enum_type_stage_fan_stage_05%]"
+ }
+ },
+ "intensive_level": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::cooking_hood_option_intensive_level::name%]",
+ "state": {
+ "cooking_hood_enum_type_intensive_stage_intensive_stage_off": "[%key:component::home_connect::selector::intensive_level::options::cooking_hood_enum_type_intensive_stage_intensive_stage_off%]",
+ "cooking_hood_enum_type_intensive_stage_intensive_stage1": "[%key:component::home_connect::selector::intensive_level::options::cooking_hood_enum_type_intensive_stage_intensive_stage1%]",
+ "cooking_hood_enum_type_intensive_stage_intensive_stage2": "[%key:component::home_connect::selector::intensive_level::options::cooking_hood_enum_type_intensive_stage_intensive_stage2%]"
+ }
+ },
+ "warming_level": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::cooking_oven_option_warming_level::name%]",
+ "state": {
+ "cooking_oven_enum_type_warming_level_low": "[%key:component::home_connect::selector::warming_level::options::cooking_oven_enum_type_warming_level_low%]",
+ "cooking_oven_enum_type_warming_level_medium": "[%key:component::home_connect::selector::warming_level::options::cooking_oven_enum_type_warming_level_medium%]",
+ "cooking_oven_enum_type_warming_level_high": "[%key:component::home_connect::selector::warming_level::options::cooking_oven_enum_type_warming_level_high%]"
+ }
+ },
+ "washer_temperature": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::laundry_care_washer_option_temperature::name%]",
+ "state": {
+ "laundry_care_washer_enum_type_temperature_cold": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_cold%]",
+ "laundry_care_washer_enum_type_temperature_g_c_20": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_20%]",
+ "laundry_care_washer_enum_type_temperature_g_c_30": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_30%]",
+ "laundry_care_washer_enum_type_temperature_g_c_40": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_40%]",
+ "laundry_care_washer_enum_type_temperature_g_c_50": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_50%]",
+ "laundry_care_washer_enum_type_temperature_g_c_60": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_60%]",
+ "laundry_care_washer_enum_type_temperature_g_c_70": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_70%]",
+ "laundry_care_washer_enum_type_temperature_g_c_80": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_80%]",
+ "laundry_care_washer_enum_type_temperature_g_c_90": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_g_c_90%]",
+ "laundry_care_washer_enum_type_temperature_ul_cold": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_ul_cold%]",
+ "laundry_care_washer_enum_type_temperature_ul_warm": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_ul_warm%]",
+ "laundry_care_washer_enum_type_temperature_ul_hot": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_ul_hot%]",
+ "laundry_care_washer_enum_type_temperature_ul_extra_hot": "[%key:component::home_connect::selector::washer_temperature::options::laundry_care_washer_enum_type_temperature_ul_extra_hot%]"
+ }
+ },
+ "spin_speed": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::laundry_care_washer_option_spin_speed::name%]",
+ "state": {
+ "laundry_care_washer_enum_type_spin_speed_off": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_off%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_400": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_400%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_600": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_600%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_700": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_700%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_800": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_800%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_900": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_900%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1000": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_1000%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1200": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_1200%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1400": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_1400%]",
+ "laundry_care_washer_enum_type_spin_speed_r_p_m_1600": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_r_p_m_1600%]",
+ "laundry_care_washer_enum_type_spin_speed_ul_off": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_ul_off%]",
+ "laundry_care_washer_enum_type_spin_speed_ul_low": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_ul_low%]",
+ "laundry_care_washer_enum_type_spin_speed_ul_medium": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_ul_medium%]",
+ "laundry_care_washer_enum_type_spin_speed_ul_high": "[%key:component::home_connect::selector::spin_speed::options::laundry_care_washer_enum_type_spin_speed_ul_high%]"
+ }
+ },
+ "vario_perfect": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::laundry_care_washer_option_vario_perfect::name%]",
+ "state": {
+ "laundry_care_common_enum_type_vario_perfect_off": "[%key:component::home_connect::selector::vario_perfect::options::laundry_care_common_enum_type_vario_perfect_off%]",
+ "laundry_care_common_enum_type_vario_perfect_eco_perfect": "[%key:component::home_connect::selector::vario_perfect::options::laundry_care_common_enum_type_vario_perfect_eco_perfect%]",
+ "laundry_care_common_enum_type_vario_perfect_speed_perfect": "[%key:component::home_connect::selector::vario_perfect::options::laundry_care_common_enum_type_vario_perfect_speed_perfect%]"
}
}
},
@@ -628,7 +1475,7 @@
"inactive": "Inactive",
"ready": "Ready",
"delayedstart": "Delayed start",
- "run": "Run",
+ "run": "Running",
"pause": "[%key:common::state::paused%]",
"actionrequired": "Action required",
"finished": "Finished",
@@ -695,6 +1542,9 @@
"map3": "Map 3"
}
},
+ "oven_current_cavity_temperature": {
+ "name": "Current oven cavity temperature"
+ },
"freezer_door_alarm": {
"name": "Freezer door alarm",
"state": {
@@ -799,6 +1649,45 @@
},
"door_assistant_freezer": {
"name": "Freezer door assistant"
+ },
+ "multiple_beverages": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::consumer_products_coffee_maker_option_multiple_beverages::name%]"
+ },
+ "intensiv_zone": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_intensiv_zone::name%]"
+ },
+ "brilliance_dry": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_brilliance_dry::name%]"
+ },
+ "vario_speed_plus": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_vario_speed_plus::name%]"
+ },
+ "silence_on_demand": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_silence_on_demand::name%]"
+ },
+ "half_load": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_half_load::name%]"
+ },
+ "extra_dry": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_extra_dry::name%]"
+ },
+ "hygiene_plus": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_hygiene_plus::name%]"
+ },
+ "eco_dry": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_eco_dry::name%]"
+ },
+ "zeolite_dry": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::dishcare_dishwasher_option_zeolite_dry::name%]"
+ },
+ "fast_pre_heat": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::cooking_oven_option_fast_pre_heat::name%]"
+ },
+ "i_dos1_active": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::laundry_care_washer_option_i_dos1_active::name%]"
+ },
+ "i_dos2_active": {
+ "name": "[%key:component::home_connect::services::set_program_and_options::fields::laundry_care_washer_option_i_dos2_active::name%]"
}
},
"time": {
diff --git a/homeassistant/components/home_connect/switch.py b/homeassistant/components/home_connect/switch.py
index e7fcd29e191..33e30f184b7 100644
--- a/homeassistant/components/home_connect/switch.py
+++ b/homeassistant/components/home_connect/switch.py
@@ -3,7 +3,7 @@
import logging
from typing import Any, cast
-from aiohomeconnect.model import EventKey, ProgramKey, SettingKey
+from aiohomeconnect.model import EventKey, OptionKey, ProgramKey, SettingKey
from aiohomeconnect.model.error import HomeConnectError
from aiohomeconnect.model.program import EnumerateProgram
@@ -13,7 +13,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import (
IssueSeverity,
async_create_issue,
@@ -22,26 +22,18 @@ from homeassistant.helpers.issue_registry import (
from homeassistant.helpers.typing import UNDEFINED, UndefinedType
from .common import setup_home_connect_entry
-from .const import (
- BSH_POWER_OFF,
- BSH_POWER_ON,
- BSH_POWER_STANDBY,
- DOMAIN,
- SVE_TRANSLATION_PLACEHOLDER_APPLIANCE_NAME,
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID,
- SVE_TRANSLATION_PLACEHOLDER_KEY,
- SVE_TRANSLATION_PLACEHOLDER_VALUE,
-)
+from .const import BSH_POWER_OFF, BSH_POWER_ON, BSH_POWER_STANDBY, DOMAIN
from .coordinator import (
HomeConnectApplianceData,
HomeConnectConfigEntry,
HomeConnectCoordinator,
)
-from .entity import HomeConnectEntity
+from .entity import HomeConnectEntity, HomeConnectOptionEntity
from .utils import get_dict_from_home_connect_error
_LOGGER = logging.getLogger(__name__)
+PARALLEL_UPDATES = 1
SWITCHES = (
SwitchEntityDescription(
@@ -100,6 +92,61 @@ POWER_SWITCH_DESCRIPTION = SwitchEntityDescription(
translation_key="power",
)
+SWITCH_OPTIONS = (
+ SwitchEntityDescription(
+ key=OptionKey.CONSUMER_PRODUCTS_COFFEE_MAKER_MULTIPLE_BEVERAGES,
+ translation_key="multiple_beverages",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_INTENSIV_ZONE,
+ translation_key="intensiv_zone",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_BRILLIANCE_DRY,
+ translation_key="brilliance_dry",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_VARIO_SPEED_PLUS,
+ translation_key="vario_speed_plus",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_SILENCE_ON_DEMAND,
+ translation_key="silence_on_demand",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_HALF_LOAD,
+ translation_key="half_load",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_EXTRA_DRY,
+ translation_key="extra_dry",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_HYGIENE_PLUS,
+ translation_key="hygiene_plus",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_ECO_DRY,
+ translation_key="eco_dry",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.DISHCARE_DISHWASHER_ZEOLITE_DRY,
+ translation_key="zeolite_dry",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.COOKING_OVEN_FAST_PRE_HEAT,
+ translation_key="fast_pre_heat",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.LAUNDRY_CARE_WASHER_I_DOS_1_ACTIVE,
+ translation_key="i_dos1_active",
+ ),
+ SwitchEntityDescription(
+ key=OptionKey.LAUNDRY_CARE_WASHER_I_DOS_2_ACTIVE,
+ translation_key="i_dos2_active",
+ ),
+)
+
def _get_entities_for_appliance(
entry: HomeConnectConfigEntry,
@@ -123,20 +170,32 @@ def _get_entities_for_appliance(
for description in SWITCHES
if description.key in appliance.settings
)
-
return entities
+def _get_option_entities_for_appliance(
+ entry: HomeConnectConfigEntry,
+ appliance: HomeConnectApplianceData,
+) -> list[HomeConnectOptionEntity]:
+ """Get a list of currently available option entities."""
+ return [
+ HomeConnectSwitchOptionEntity(entry.runtime_data, appliance, description)
+ for description in SWITCH_OPTIONS
+ if description.key in appliance.options
+ ]
+
+
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeConnectConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Connect switch."""
setup_home_connect_entry(
entry,
_get_entities_for_appliance,
async_add_entities,
+ _get_option_entities_for_appliance,
)
@@ -158,8 +217,8 @@ class HomeConnectSwitch(HomeConnectEntity, SwitchEntity):
translation_key="turn_on",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
- SVE_TRANSLATION_PLACEHOLDER_KEY: self.bsh_key,
+ "entity_id": self.entity_id,
+ "key": self.bsh_key,
},
) from err
@@ -178,8 +237,8 @@ class HomeConnectSwitch(HomeConnectEntity, SwitchEntity):
translation_key="turn_off",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
- SVE_TRANSLATION_PLACEHOLDER_KEY: self.bsh_key,
+ "entity_id": self.entity_id,
+ "key": self.bsh_key,
},
) from err
@@ -317,7 +376,7 @@ class HomeConnectPowerSwitch(HomeConnectEntity, SwitchEntity):
translation_key="power_on",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_APPLIANCE_NAME: self.appliance.info.name,
+ "appliance_name": self.appliance.info.name,
},
) from err
@@ -330,7 +389,7 @@ class HomeConnectPowerSwitch(HomeConnectEntity, SwitchEntity):
translation_domain=DOMAIN,
translation_key="unable_to_retrieve_turn_off",
translation_placeholders={
- SVE_TRANSLATION_PLACEHOLDER_APPLIANCE_NAME: self.appliance.info.name
+ "appliance_name": self.appliance.info.name
},
)
@@ -338,9 +397,7 @@ class HomeConnectPowerSwitch(HomeConnectEntity, SwitchEntity):
raise HomeAssistantError(
translation_domain=DOMAIN,
translation_key="turn_off_not_supported",
- translation_placeholders={
- SVE_TRANSLATION_PLACEHOLDER_APPLIANCE_NAME: self.appliance.info.name
- },
+ translation_placeholders={"appliance_name": self.appliance.info.name},
)
try:
await self.coordinator.client.set_setting(
@@ -355,8 +412,8 @@ class HomeConnectPowerSwitch(HomeConnectEntity, SwitchEntity):
translation_key="power_off",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_APPLIANCE_NAME: self.appliance.info.name,
- SVE_TRANSLATION_PLACEHOLDER_VALUE: self.power_off_state,
+ "appliance_name": self.appliance.info.name,
+ "value": self.power_off_state,
},
) from err
@@ -403,3 +460,19 @@ class HomeConnectPowerSwitch(HomeConnectEntity, SwitchEntity):
self.power_off_state = BSH_POWER_STANDBY
else:
self.power_off_state = None
+
+
+class HomeConnectSwitchOptionEntity(HomeConnectOptionEntity, SwitchEntity):
+ """Switch option class for Home Connect."""
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn on the option."""
+ await self.async_set_option(True)
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn off the option."""
+ await self.async_set_option(False)
+
+ def update_native_value(self) -> None:
+ """Set the value of the entity."""
+ self._attr_is_on = cast(bool | None, self.option_value)
diff --git a/homeassistant/components/home_connect/time.py b/homeassistant/components/home_connect/time.py
index 48f651857d2..7cfa0a7d3e4 100644
--- a/homeassistant/components/home_connect/time.py
+++ b/homeassistant/components/home_connect/time.py
@@ -9,20 +9,16 @@ from aiohomeconnect.model.error import HomeConnectError
from homeassistant.components.time import TimeEntity, TimeEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import setup_home_connect_entry
-from .const import (
- DOMAIN,
- SVE_TRANSLATION_KEY_SET_SETTING,
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID,
- SVE_TRANSLATION_PLACEHOLDER_KEY,
- SVE_TRANSLATION_PLACEHOLDER_VALUE,
-)
+from .const import DOMAIN
from .coordinator import HomeConnectApplianceData, HomeConnectConfigEntry
from .entity import HomeConnectEntity
from .utils import get_dict_from_home_connect_error
+PARALLEL_UPDATES = 1
+
TIME_ENTITIES = (
TimeEntityDescription(
key=SettingKey.BSH_COMMON_ALARM_CLOCK,
@@ -46,7 +42,7 @@ def _get_entities_for_appliance(
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeConnectConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Home Connect switch."""
setup_home_connect_entry(
@@ -82,12 +78,12 @@ class HomeConnectTimeEntity(HomeConnectEntity, TimeEntity):
except HomeConnectError as err:
raise HomeAssistantError(
translation_domain=DOMAIN,
- translation_key=SVE_TRANSLATION_KEY_SET_SETTING,
+ translation_key="set_setting_entity",
translation_placeholders={
**get_dict_from_home_connect_error(err),
- SVE_TRANSLATION_PLACEHOLDER_ENTITY_ID: self.entity_id,
- SVE_TRANSLATION_PLACEHOLDER_KEY: self.bsh_key,
- SVE_TRANSLATION_PLACEHOLDER_VALUE: str(value),
+ "entity_id": self.entity_id,
+ "key": self.bsh_key,
+ "value": str(value),
},
) from err
diff --git a/homeassistant/components/home_connect/utils.py b/homeassistant/components/home_connect/utils.py
index 108465072e1..ee5febb3cf7 100644
--- a/homeassistant/components/home_connect/utils.py
+++ b/homeassistant/components/home_connect/utils.py
@@ -2,7 +2,7 @@
import re
-from aiohomeconnect.model.error import HomeConnectApiError, HomeConnectError
+from aiohomeconnect.model.error import HomeConnectError
RE_CAMEL_CASE = re.compile(r"(? dict[str, str]:
"""Return a translation string from a Home Connect error."""
- return {
- "error": str(err)
- if isinstance(err, HomeConnectApiError)
- else type(err).__name__
- }
+ return {"error": str(err)}
def bsh_key_to_translation_key(bsh_key: str) -> str:
diff --git a/homeassistant/components/homeassistant/exposed_entities.py b/homeassistant/components/homeassistant/exposed_entities.py
index 7bd9f9ab7bc..b7e420dedde 100644
--- a/homeassistant/components/homeassistant/exposed_entities.py
+++ b/homeassistant/components/homeassistant/exposed_entities.py
@@ -437,18 +437,21 @@ def ws_expose_entity(
def ws_list_exposed_entities(
hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]
) -> None:
- """Expose an entity to an assistant."""
+ """List entities which are exposed to assistants."""
result: dict[str, Any] = {}
exposed_entities = hass.data[DATA_EXPOSED_ENTITIES]
entity_registry = er.async_get(hass)
for entity_id in chain(exposed_entities.entities, entity_registry.entities):
- result[entity_id] = {}
+ exposed_to = {}
entity_settings = async_get_entity_settings(hass, entity_id)
for assistant, settings in entity_settings.items():
- if "should_expose" not in settings:
+ if "should_expose" not in settings or not settings["should_expose"]:
continue
- result[entity_id][assistant] = settings["should_expose"]
+ exposed_to[assistant] = True
+ if not exposed_to:
+ continue
+ result[entity_id] = exposed_to
connection.send_result(msg["id"], {"exposed_entities": result})
diff --git a/homeassistant/components/homeassistant/strings.json b/homeassistant/components/homeassistant/strings.json
index 3283d480fdd..b8b5f77cf52 100644
--- a/homeassistant/components/homeassistant/strings.json
+++ b/homeassistant/components/homeassistant/strings.json
@@ -12,7 +12,7 @@
},
"imperial_unit_system": {
"title": "The imperial unit system is deprecated",
- "description": "The imperial unit system is deprecated and your system is currently using us customary. Please update your configuration to use the us customary unit system and reload the core configuration to fix this issue."
+ "description": "The imperial unit system is deprecated and your system is currently using US customary. Please update your configuration to use the US customary unit system and reload the Core configuration to fix this issue."
},
"deprecated_yaml": {
"title": "The {integration_title} YAML configuration is being removed",
@@ -44,7 +44,7 @@
},
"no_platform_setup": {
"title": "Unused YAML configuration for the {platform} integration",
- "description": "It's not possible to configure {platform} {domain} by adding `{platform_key}` to the {domain} configuration. Please check the documentation for more information on how to set up this integration.\n\nTo resolve this:\n1. Remove `{platform_key}` occurences from the `{domain}:` configuration in your YAML configuration file.\n2. Restart Home Assistant.\n\nExample that should be removed:\n{yaml_example}"
+ "description": "It's not possible to configure {platform} {domain} by adding `{platform_key}` to the {domain} configuration. Please check the documentation for more information on how to set up this integration.\n\nTo resolve this:\n1. Remove `{platform_key}` occurrences from the `{domain}:` configuration in your YAML configuration file.\n2. Restart Home Assistant.\n\nExample that should be removed:\n{yaml_example}"
},
"storage_corruption": {
"title": "Storage corruption detected for {storage_key}",
@@ -111,8 +111,8 @@
"description": "Checks the Home Assistant YAML-configuration files for errors. Errors will be shown in the Home Assistant logs."
},
"reload_core_config": {
- "name": "Reload core configuration",
- "description": "Reloads the core configuration from the YAML-configuration."
+ "name": "Reload Core configuration",
+ "description": "Reloads the Core configuration from the YAML-configuration."
},
"restart": {
"name": "[%key:common::action::restart%]",
@@ -160,7 +160,7 @@
},
"update_entity": {
"name": "Update entity",
- "description": "Forces one or more entities to update its data.",
+ "description": "Forces one or more entities to update their data.",
"fields": {
"entity_id": {
"name": "Entities to update",
@@ -188,7 +188,7 @@
},
"reload_all": {
"name": "Reload all",
- "description": "Reload all YAML configuration that can be reloaded without restarting Home Assistant."
+ "description": "Reloads all YAML configuration that can be reloaded without restarting Home Assistant."
}
},
"exceptions": {
diff --git a/homeassistant/components/homeassistant_alerts/coordinator.py b/homeassistant/components/homeassistant_alerts/coordinator.py
index a81824d2376..542ebf857df 100644
--- a/homeassistant/components/homeassistant_alerts/coordinator.py
+++ b/homeassistant/components/homeassistant_alerts/coordinator.py
@@ -40,6 +40,7 @@ class AlertUpdateCoordinator(DataUpdateCoordinator[dict[str, IntegrationAlert]])
super().__init__(
hass,
_LOGGER,
+ config_entry=None,
name=DOMAIN,
update_interval=UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/homeassistant_hardware/coordinator.py b/homeassistant/components/homeassistant_hardware/coordinator.py
new file mode 100644
index 00000000000..9eb900b13fd
--- /dev/null
+++ b/homeassistant/components/homeassistant_hardware/coordinator.py
@@ -0,0 +1,47 @@
+"""Home Assistant hardware firmware update coordinator."""
+
+from __future__ import annotations
+
+from datetime import timedelta
+import logging
+
+from aiohttp import ClientSession
+from ha_silabs_firmware_client import (
+ FirmwareManifest,
+ FirmwareUpdateClient,
+ ManifestMissing,
+)
+
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+_LOGGER = logging.getLogger(__name__)
+
+
+FIRMWARE_REFRESH_INTERVAL = timedelta(hours=8)
+
+
+class FirmwareUpdateCoordinator(DataUpdateCoordinator[FirmwareManifest]):
+ """Coordinator to manage firmware updates."""
+
+ def __init__(self, hass: HomeAssistant, session: ClientSession, url: str) -> None:
+ """Initialize the firmware update coordinator."""
+ super().__init__(
+ hass,
+ _LOGGER,
+ name="firmware update coordinator",
+ update_interval=FIRMWARE_REFRESH_INTERVAL,
+ always_update=False,
+ )
+ self.hass = hass
+ self.session = session
+
+ self.client = FirmwareUpdateClient(url, session)
+
+ async def _async_update_data(self) -> FirmwareManifest:
+ try:
+ return await self.client.async_update_data()
+ except ManifestMissing as err:
+ raise UpdateFailed(
+ "GitHub release assets haven't been uploaded yet"
+ ) from err
diff --git a/homeassistant/components/homeassistant_hardware/firmware_config_flow.py b/homeassistant/components/homeassistant_hardware/firmware_config_flow.py
index 8d7a302e786..83031587712 100644
--- a/homeassistant/components/homeassistant_hardware/firmware_config_flow.py
+++ b/homeassistant/components/homeassistant_hardware/firmware_config_flow.py
@@ -28,12 +28,13 @@ from . import silabs_multiprotocol_addon
from .const import OTBR_DOMAIN, ZHA_DOMAIN
from .util import (
ApplicationType,
+ FirmwareInfo,
OwningAddon,
OwningIntegration,
get_otbr_addon_manager,
get_zigbee_flasher_addon_manager,
guess_hardware_owners,
- probe_silabs_firmware_type,
+ probe_silabs_firmware_info,
)
_LOGGER = logging.getLogger(__name__)
@@ -52,7 +53,7 @@ class BaseFirmwareInstallFlow(ConfigEntryBaseFlow, ABC):
"""Instantiate base flow."""
super().__init__(*args, **kwargs)
- self._probed_firmware_type: ApplicationType | None = None
+ self._probed_firmware_info: FirmwareInfo | None = None
self._device: str | None = None # To be set in a subclass
self._hardware_name: str = "unknown" # To be set in a subclass
@@ -64,8 +65,8 @@ class BaseFirmwareInstallFlow(ConfigEntryBaseFlow, ABC):
"""Shared translation placeholders."""
placeholders = {
"firmware_type": (
- self._probed_firmware_type.value
- if self._probed_firmware_type is not None
+ self._probed_firmware_info.firmware_type.value
+ if self._probed_firmware_info is not None
else "unknown"
),
"model": self._hardware_name,
@@ -120,39 +121,49 @@ class BaseFirmwareInstallFlow(ConfigEntryBaseFlow, ABC):
description_placeholders=self._get_translation_placeholders(),
)
- async def _probe_firmware_type(self) -> bool:
- """Probe the firmware currently on the device."""
- assert self._device is not None
-
- self._probed_firmware_type = await probe_silabs_firmware_type(
- self._device,
- probe_methods=(
- # We probe in order of frequency: Zigbee, Thread, then multi-PAN
- ApplicationType.GECKO_BOOTLOADER,
- ApplicationType.EZSP,
- ApplicationType.SPINEL,
- ApplicationType.CPC,
- ),
- )
-
- return self._probed_firmware_type in (
+ async def _probe_firmware_info(
+ self,
+ probe_methods: tuple[ApplicationType, ...] = (
+ # We probe in order of frequency: Zigbee, Thread, then multi-PAN
+ ApplicationType.GECKO_BOOTLOADER,
ApplicationType.EZSP,
ApplicationType.SPINEL,
ApplicationType.CPC,
+ ),
+ ) -> bool:
+ """Probe the firmware currently on the device."""
+ assert self._device is not None
+
+ self._probed_firmware_info = await probe_silabs_firmware_info(
+ self._device,
+ probe_methods=probe_methods,
+ )
+
+ return (
+ self._probed_firmware_info is not None
+ and self._probed_firmware_info.firmware_type
+ in (
+ ApplicationType.EZSP,
+ ApplicationType.SPINEL,
+ ApplicationType.CPC,
+ )
)
async def async_step_pick_firmware_zigbee(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Pick Zigbee firmware."""
- if not await self._probe_firmware_type():
+ if not await self._probe_firmware_info():
return self.async_abort(
reason="unsupported_firmware",
description_placeholders=self._get_translation_placeholders(),
)
# Allow the stick to be used with ZHA without flashing
- if self._probed_firmware_type == ApplicationType.EZSP:
+ if (
+ self._probed_firmware_info is not None
+ and self._probed_firmware_info.firmware_type == ApplicationType.EZSP
+ ):
return await self.async_step_confirm_zigbee()
if not is_hassio(self.hass):
@@ -338,7 +349,12 @@ class BaseFirmwareInstallFlow(ConfigEntryBaseFlow, ABC):
"""Confirm Zigbee setup."""
assert self._device is not None
assert self._hardware_name is not None
- self._probed_firmware_type = ApplicationType.EZSP
+
+ if not await self._probe_firmware_info(probe_methods=(ApplicationType.EZSP,)):
+ return self.async_abort(
+ reason="unsupported_firmware",
+ description_placeholders=self._get_translation_placeholders(),
+ )
if user_input is not None:
await self.hass.config_entries.flow.async_init(
@@ -366,7 +382,7 @@ class BaseFirmwareInstallFlow(ConfigEntryBaseFlow, ABC):
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Pick Thread firmware."""
- if not await self._probe_firmware_type():
+ if not await self._probe_firmware_info():
return self.async_abort(
reason="unsupported_firmware",
description_placeholders=self._get_translation_placeholders(),
@@ -458,7 +474,11 @@ class BaseFirmwareInstallFlow(ConfigEntryBaseFlow, ABC):
"""Confirm OTBR setup."""
assert self._device is not None
- self._probed_firmware_type = ApplicationType.SPINEL
+ if not await self._probe_firmware_info(probe_methods=(ApplicationType.SPINEL,)):
+ return self.async_abort(
+ reason="unsupported_firmware",
+ description_placeholders=self._get_translation_placeholders(),
+ )
if user_input is not None:
# OTBR discovery is done automatically via hassio
@@ -497,14 +517,14 @@ class BaseFirmwareConfigFlow(BaseFirmwareInstallFlow, ConfigFlow):
class BaseFirmwareOptionsFlow(BaseFirmwareInstallFlow, OptionsFlow):
"""Zigbee and Thread options flow handlers."""
+ _probed_firmware_info: FirmwareInfo
+
def __init__(self, config_entry: ConfigEntry, *args: Any, **kwargs: Any) -> None:
"""Instantiate options flow."""
super().__init__(*args, **kwargs)
self._config_entry = config_entry
- self._probed_firmware_type = ApplicationType(self.config_entry.data["firmware"])
-
# Make `context` a regular dictionary
self.context = {}
diff --git a/homeassistant/components/homeassistant_hardware/manifest.json b/homeassistant/components/homeassistant_hardware/manifest.json
index 2efa12ccfda..f3a02185b83 100644
--- a/homeassistant/components/homeassistant_hardware/manifest.json
+++ b/homeassistant/components/homeassistant_hardware/manifest.json
@@ -5,5 +5,8 @@
"codeowners": ["@home-assistant/core"],
"documentation": "https://www.home-assistant.io/integrations/homeassistant_hardware",
"integration_type": "system",
- "requirements": ["universal-silabs-flasher==0.0.25"]
+ "requirements": [
+ "universal-silabs-flasher==0.0.30",
+ "ha-silabs-firmware-client==0.2.0"
+ ]
}
diff --git a/homeassistant/components/homeassistant_hardware/strings.json b/homeassistant/components/homeassistant_hardware/strings.json
index de328a54bb7..6dda01561f1 100644
--- a/homeassistant/components/homeassistant_hardware/strings.json
+++ b/homeassistant/components/homeassistant_hardware/strings.json
@@ -39,8 +39,8 @@
"description": "The OpenThread Border Router (OTBR) add-on is now starting."
},
"otbr_failed": {
- "title": "Failed to setup OpenThread Border Router",
- "description": "The OpenThread Border Router add-on installation was unsuccessful. Ensure no other software is trying to communicate with the {model}, you have access to the internet and can install other add-ons, and try again. Check the Supervisor logs if the problem persists."
+ "title": "Failed to set up OpenThread Border Router",
+ "description": "The OpenThread Border Router add-on installation was unsuccessful. Ensure no other software is trying to communicate with the {model}, you have access to the Internet and can install other add-ons, and try again. Check the Supervisor logs if the problem persists."
},
"confirm_otbr": {
"title": "OpenThread Border Router setup complete",
@@ -48,16 +48,16 @@
}
},
"abort": {
- "not_hassio_thread": "The OpenThread Border Router addon can only be installed with Home Assistant OS. If you would like to use the {model} as an Thread border router, please flash the firmware manually using the [web flasher]({docs_web_flasher_url}) and set up OpenThread Border Router to communicate with it.",
+ "not_hassio_thread": "The OpenThread Border Router add-on can only be installed with Home Assistant OS. If you would like to use the {model} as a Thread border router, please flash the firmware manually using the [web flasher]({docs_web_flasher_url}) and set up OpenThread Border Router to communicate with it.",
"otbr_addon_already_running": "The OpenThread Border Router add-on is already running, it cannot be installed again.",
"zha_still_using_stick": "This {model} is in use by the Zigbee Home Automation integration. Please migrate your Zigbee network to another adapter or delete the integration and try again.",
"otbr_still_using_stick": "This {model} is in use by the OpenThread Border Router add-on. If you use the Thread network, make sure you have alternative border routers. Uninstall the add-on and try again.",
- "unsupported_firmware": "The radio firmware on your {model} could not be determined. Make sure that no other integration or addon is currently trying to communicate with the device. If you are running Home Assistant OS in a virtual machine or in Docker, please make sure that permissions are set correctly for the device."
+ "unsupported_firmware": "The radio firmware on your {model} could not be determined. Make sure that no other integration or add-on is currently trying to communicate with the device. If you are running Home Assistant OS in a virtual machine or in Docker, please make sure that permissions are set correctly for the device."
},
"progress": {
- "install_zigbee_flasher_addon": "The Silicon Labs Flasher addon is installed, this may take a few minutes.",
+ "install_zigbee_flasher_addon": "The Silicon Labs Flasher add-on is installed, this may take a few minutes.",
"run_zigbee_flasher_addon": "Please wait while Zigbee firmware is installed to your {model}, this will take a few minutes. Do not make any changes to your hardware or software until this finishes.",
- "uninstall_zigbee_flasher_addon": "The Silicon Labs Flasher addon is being removed."
+ "uninstall_zigbee_flasher_addon": "The Silicon Labs Flasher add-on is being removed."
}
}
},
diff --git a/homeassistant/components/homeassistant_hardware/update.py b/homeassistant/components/homeassistant_hardware/update.py
new file mode 100644
index 00000000000..e835286238f
--- /dev/null
+++ b/homeassistant/components/homeassistant_hardware/update.py
@@ -0,0 +1,331 @@
+"""Home Assistant Hardware base firmware update entity."""
+
+from __future__ import annotations
+
+from collections.abc import AsyncIterator, Callable
+from contextlib import AsyncExitStack, asynccontextmanager
+from dataclasses import dataclass
+import logging
+from typing import Any, cast
+
+from ha_silabs_firmware_client import FirmwareManifest, FirmwareMetadata
+from universal_silabs_flasher.firmware import parse_firmware_image
+from universal_silabs_flasher.flasher import Flasher
+from yarl import URL
+
+from homeassistant.components.update import (
+ UpdateEntity,
+ UpdateEntityDescription,
+ UpdateEntityFeature,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import CALLBACK_TYPE, callback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.restore_state import ExtraStoredData
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .coordinator import FirmwareUpdateCoordinator
+from .helpers import async_register_firmware_info_callback
+from .util import (
+ ApplicationType,
+ FirmwareInfo,
+ guess_firmware_info,
+ probe_silabs_firmware_info,
+)
+
+_LOGGER = logging.getLogger(__name__)
+
+type FirmwareChangeCallbackType = Callable[
+ [ApplicationType | None, ApplicationType | None], None
+]
+
+
+@dataclass(kw_only=True, frozen=True)
+class FirmwareUpdateEntityDescription(UpdateEntityDescription):
+ """Describes Home Assistant Hardware firmware update entity."""
+
+ version_parser: Callable[[str], str]
+ fw_type: str | None
+ version_key: str | None
+ expected_firmware_type: ApplicationType | None
+ firmware_name: str | None
+
+
+@dataclass
+class FirmwareUpdateExtraStoredData(ExtraStoredData):
+ """Extra stored data for Home Assistant Hardware firmware update entity."""
+
+ firmware_manifest: FirmwareManifest | None = None
+
+ def as_dict(self) -> dict[str, Any]:
+ """Return a dict representation of the extra data."""
+ return {
+ "firmware_manifest": (
+ self.firmware_manifest.as_dict()
+ if self.firmware_manifest is not None
+ else None
+ )
+ }
+
+ @classmethod
+ def from_dict(cls, data: dict[str, Any]) -> FirmwareUpdateExtraStoredData:
+ """Initialize the extra data from a dict."""
+ if data["firmware_manifest"] is None:
+ return cls(firmware_manifest=None)
+
+ return cls(
+ FirmwareManifest.from_json(
+ data["firmware_manifest"],
+ # This data is not technically part of the manifest and is loaded externally
+ url=URL(data["firmware_manifest"]["url"]),
+ html_url=URL(data["firmware_manifest"]["html_url"]),
+ )
+ )
+
+
+class BaseFirmwareUpdateEntity(
+ CoordinatorEntity[FirmwareUpdateCoordinator], UpdateEntity
+):
+ """Base Home Assistant Hardware firmware update entity."""
+
+ # Subclasses provide the mapping between firmware types and entity descriptions
+ entity_description: FirmwareUpdateEntityDescription
+ bootloader_reset_type: str | None = None
+
+ _attr_supported_features = (
+ UpdateEntityFeature.INSTALL | UpdateEntityFeature.PROGRESS
+ )
+ # Until this entity can be associated with a device, we must manually name it
+ _attr_has_entity_name = False
+
+ def __init__(
+ self,
+ device: str,
+ config_entry: ConfigEntry,
+ update_coordinator: FirmwareUpdateCoordinator,
+ entity_description: FirmwareUpdateEntityDescription,
+ ) -> None:
+ """Initialize the Hardware firmware update entity."""
+ super().__init__(update_coordinator)
+
+ self.entity_description = entity_description
+ self._current_device = device
+ self._config_entry = config_entry
+ self._current_firmware_info: FirmwareInfo | None = None
+ self._firmware_type_change_callbacks: set[FirmwareChangeCallbackType] = set()
+
+ self._latest_manifest: FirmwareManifest | None = None
+ self._latest_firmware: FirmwareMetadata | None = None
+
+ def add_firmware_type_changed_callback(
+ self,
+ change_callback: FirmwareChangeCallbackType,
+ ) -> CALLBACK_TYPE:
+ """Add a callback for when the firmware type changes."""
+ self._firmware_type_change_callbacks.add(change_callback)
+
+ @callback
+ def remove_callback() -> None:
+ self._firmware_type_change_callbacks.discard(change_callback)
+
+ return remove_callback
+
+ async def async_added_to_hass(self) -> None:
+ """Handle entity which will be added."""
+ await super().async_added_to_hass()
+
+ self.async_on_remove(
+ async_register_firmware_info_callback(
+ self.hass,
+ self._current_device,
+ self._firmware_info_callback,
+ )
+ )
+
+ self.async_on_remove(
+ self._config_entry.async_on_state_change(self._on_config_entry_change)
+ )
+
+ if (extra_data := await self.async_get_last_extra_data()) and (
+ hardware_extra_data := FirmwareUpdateExtraStoredData.from_dict(
+ extra_data.as_dict()
+ )
+ ):
+ self._latest_manifest = hardware_extra_data.firmware_manifest
+
+ self._update_attributes()
+
+ @property
+ def extra_restore_state_data(self) -> FirmwareUpdateExtraStoredData:
+ """Return state data to be restored."""
+ return FirmwareUpdateExtraStoredData(firmware_manifest=self._latest_manifest)
+
+ @callback
+ def _on_config_entry_change(self) -> None:
+ """Handle config entry changes."""
+ self._update_attributes()
+ self.async_write_ha_state()
+
+ @callback
+ def _firmware_info_callback(self, firmware_info: FirmwareInfo) -> None:
+ """Handle updated firmware info being pushed by an integration."""
+ self._current_firmware_info = firmware_info
+
+ # If the firmware type does not change, we can just update the attributes
+ if (
+ self._current_firmware_info.firmware_type
+ == self.entity_description.expected_firmware_type
+ ):
+ self._update_attributes()
+ self.async_write_ha_state()
+ return
+
+ # Otherwise, fire the firmware type change callbacks. They are expected to
+ # replace the entity so there is no purpose in firing other callbacks.
+ for change_callback in self._firmware_type_change_callbacks.copy():
+ try:
+ change_callback(
+ self.entity_description.expected_firmware_type,
+ self._current_firmware_info.firmware_type,
+ )
+ except Exception: # noqa: BLE001
+ _LOGGER.warning(
+ "Failed to call firmware type changed callback", exc_info=True
+ )
+
+ def _update_attributes(self) -> None:
+ """Recompute the attributes of the entity."""
+
+ # This entity is not currently associated with a device so we must manually
+ # give it a name
+ self._attr_name = f"{self._config_entry.title} Update"
+ self._attr_title = self.entity_description.firmware_name or "unknown"
+
+ if (
+ self._current_firmware_info is None
+ or self._current_firmware_info.firmware_version is None
+ ):
+ self._attr_installed_version = None
+ else:
+ self._attr_installed_version = self.entity_description.version_parser(
+ self._current_firmware_info.firmware_version
+ )
+
+ self._latest_firmware = None
+ self._attr_latest_version = None
+ self._attr_release_summary = None
+ self._attr_release_url = None
+
+ if (
+ self._latest_manifest is None
+ or self.entity_description.fw_type is None
+ or self.entity_description.version_key is None
+ ):
+ return
+
+ try:
+ self._latest_firmware = next(
+ f
+ for f in self._latest_manifest.firmwares
+ if f.filename.startswith(self.entity_description.fw_type)
+ )
+ except StopIteration:
+ pass
+ else:
+ version = cast(
+ str, self._latest_firmware.metadata[self.entity_description.version_key]
+ )
+ self._attr_latest_version = self.entity_description.version_parser(version)
+ self._attr_release_summary = self._latest_firmware.release_notes
+ self._attr_release_url = str(self._latest_manifest.html_url)
+
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
+ self._latest_manifest = self.coordinator.data
+ self._update_attributes()
+ self.async_write_ha_state()
+
+ def _update_progress(self, offset: int, total_size: int) -> None:
+ """Handle update progress."""
+
+ # Firmware updates in ~30s so we still get responsive update progress even
+ # without decimal places
+ self._attr_update_percentage = round((offset * 100) / total_size)
+ self.async_write_ha_state()
+
+ @asynccontextmanager
+ async def _temporarily_stop_hardware_owners(
+ self, device: str
+ ) -> AsyncIterator[None]:
+ """Temporarily stop addons and integrations communicating with the device."""
+ firmware_info = await guess_firmware_info(self.hass, device)
+ _LOGGER.debug("Identified firmware info: %s", firmware_info)
+
+ async with AsyncExitStack() as stack:
+ for owner in firmware_info.owners:
+ await stack.enter_async_context(owner.temporarily_stop(self.hass))
+
+ yield
+
+ async def async_install(
+ self, version: str | None, backup: bool, **kwargs: Any
+ ) -> None:
+ """Install an update."""
+ assert self._latest_firmware is not None
+ assert self.entity_description.expected_firmware_type is not None
+
+ # Start off by setting the progress bar to an indeterminate state
+ self._attr_in_progress = True
+ self._attr_update_percentage = None
+ self.async_write_ha_state()
+
+ fw_data = await self.coordinator.client.async_fetch_firmware(
+ self._latest_firmware
+ )
+ fw_image = await self.hass.async_add_executor_job(parse_firmware_image, fw_data)
+
+ device = self._current_device
+
+ flasher = Flasher(
+ device=device,
+ probe_methods=(
+ ApplicationType.GECKO_BOOTLOADER.as_flasher_application_type(),
+ ApplicationType.EZSP.as_flasher_application_type(),
+ ApplicationType.SPINEL.as_flasher_application_type(),
+ ApplicationType.CPC.as_flasher_application_type(),
+ ),
+ bootloader_reset=self.bootloader_reset_type,
+ )
+
+ async with self._temporarily_stop_hardware_owners(device):
+ try:
+ try:
+ # Enter the bootloader with indeterminate progress
+ await flasher.enter_bootloader()
+
+ # Flash the firmware, with progress
+ await flasher.flash_firmware(
+ fw_image, progress_callback=self._update_progress
+ )
+ except Exception as err:
+ raise HomeAssistantError("Failed to flash firmware") from err
+
+ # Probe the running application type with indeterminate progress
+ self._attr_update_percentage = None
+ self.async_write_ha_state()
+
+ firmware_info = await probe_silabs_firmware_info(
+ device,
+ probe_methods=(self.entity_description.expected_firmware_type,),
+ )
+
+ if firmware_info is None:
+ raise HomeAssistantError(
+ "Failed to probe the firmware after flashing"
+ )
+
+ self._firmware_info_callback(firmware_info)
+ finally:
+ self._attr_in_progress = False
+ self.async_write_ha_state()
diff --git a/homeassistant/components/homeassistant_hardware/util.py b/homeassistant/components/homeassistant_hardware/util.py
index 53cbcbae5d4..64f363e4f23 100644
--- a/homeassistant/components/homeassistant_hardware/util.py
+++ b/homeassistant/components/homeassistant_hardware/util.py
@@ -4,7 +4,8 @@ from __future__ import annotations
import asyncio
from collections import defaultdict
-from collections.abc import Iterable
+from collections.abc import AsyncIterator, Iterable
+from contextlib import asynccontextmanager
from dataclasses import dataclass
from enum import StrEnum
import logging
@@ -12,7 +13,7 @@ import logging
from universal_silabs_flasher.const import ApplicationType as FlasherApplicationType
from universal_silabs_flasher.flasher import Flasher
-from homeassistant.components.hassio import AddonError, AddonState
+from homeassistant.components.hassio import AddonError, AddonManager, AddonState
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.hassio import is_hassio
@@ -42,6 +43,7 @@ class ApplicationType(StrEnum):
CPC = "cpc"
EZSP = "ezsp"
SPINEL = "spinel"
+ ROUTER = "router"
@classmethod
def from_flasher_application_type(
@@ -104,6 +106,28 @@ class OwningAddon:
else:
return addon_info.state == AddonState.RUNNING
+ @asynccontextmanager
+ async def temporarily_stop(self, hass: HomeAssistant) -> AsyncIterator[None]:
+ """Temporarily stop the add-on, restarting it after completion."""
+ addon_manager = self._get_addon_manager(hass)
+
+ try:
+ addon_info = await addon_manager.async_get_addon_info()
+ except AddonError:
+ yield
+ return
+
+ if addon_info.state != AddonState.RUNNING:
+ yield
+ return
+
+ try:
+ await addon_manager.async_stop_addon()
+ await addon_manager.async_wait_until_addon_state(AddonState.NOT_RUNNING)
+ yield
+ finally:
+ await addon_manager.async_start_addon_waiting()
+
@dataclass(kw_only=True)
class OwningIntegration:
@@ -122,6 +146,23 @@ class OwningIntegration:
ConfigEntryState.SETUP_IN_PROGRESS,
)
+ @asynccontextmanager
+ async def temporarily_stop(self, hass: HomeAssistant) -> AsyncIterator[None]:
+ """Temporarily stop the integration, restarting it after completion."""
+ if (entry := hass.config_entries.async_get_entry(self.config_entry_id)) is None:
+ yield
+ return
+
+ if entry.state != ConfigEntryState.LOADED:
+ yield
+ return
+
+ try:
+ await hass.config_entries.async_unload(entry.entry_id)
+ yield
+ finally:
+ await hass.config_entries.async_setup(entry.entry_id)
+
@dataclass(kw_only=True)
class FirmwareInfo:
@@ -143,6 +184,31 @@ class FirmwareInfo:
return all(states)
+async def get_otbr_addon_firmware_info(
+ hass: HomeAssistant, otbr_addon_manager: AddonManager
+) -> FirmwareInfo | None:
+ """Get firmware info from the OTBR add-on."""
+ try:
+ otbr_addon_info = await otbr_addon_manager.async_get_addon_info()
+ except AddonError:
+ return None
+
+ if otbr_addon_info.state == AddonState.NOT_INSTALLED:
+ return None
+
+ if (otbr_path := otbr_addon_info.options.get("device")) is None:
+ return None
+
+ # Only create a new entry if there are no existing OTBR ones
+ return FirmwareInfo(
+ device=otbr_path,
+ firmware_type=ApplicationType.SPINEL,
+ firmware_version=None,
+ source="otbr",
+ owners=[OwningAddon(slug=otbr_addon_manager.addon_slug)],
+ )
+
+
async def guess_hardware_owners(
hass: HomeAssistant, device_path: str
) -> list[FirmwareInfo]:
@@ -155,28 +221,19 @@ async def guess_hardware_owners(
# It may be possible for the OTBR addon to be present without the integration
if is_hassio(hass):
otbr_addon_manager = get_otbr_addon_manager(hass)
+ otbr_addon_fw_info = await get_otbr_addon_firmware_info(
+ hass, otbr_addon_manager
+ )
+ otbr_path = (
+ otbr_addon_fw_info.device if otbr_addon_fw_info is not None else None
+ )
- try:
- otbr_addon_info = await otbr_addon_manager.async_get_addon_info()
- except AddonError:
- pass
- else:
- if otbr_addon_info.state != AddonState.NOT_INSTALLED:
- otbr_path = otbr_addon_info.options.get("device")
-
- # Only create a new entry if there are no existing OTBR ones
- if otbr_path is not None and not any(
- info.source == "otbr" for info in device_guesses[otbr_path]
- ):
- device_guesses[otbr_path].append(
- FirmwareInfo(
- device=otbr_path,
- firmware_type=ApplicationType.SPINEL,
- firmware_version=None,
- source="otbr",
- owners=[OwningAddon(slug=otbr_addon_manager.addon_slug)],
- )
- )
+ # Only create a new entry if there are no existing OTBR ones
+ if otbr_path is not None and not any(
+ info.source == "otbr" for info in device_guesses[otbr_path]
+ ):
+ assert otbr_addon_fw_info is not None
+ device_guesses[otbr_path].append(otbr_addon_fw_info)
if is_hassio(hass):
multipan_addon_manager = await get_multiprotocol_addon_manager(hass)
@@ -232,10 +289,10 @@ async def guess_firmware_info(hass: HomeAssistant, device_path: str) -> Firmware
return guesses[-1][0]
-async def probe_silabs_firmware_type(
+async def probe_silabs_firmware_info(
device: str, *, probe_methods: Iterable[ApplicationType] | None = None
-) -> ApplicationType | None:
- """Probe the running firmware on a Silabs device."""
+) -> FirmwareInfo | None:
+ """Probe the running firmware on a SiLabs device."""
flasher = Flasher(
device=device,
**(
@@ -253,4 +310,26 @@ async def probe_silabs_firmware_type(
if flasher.app_type is None:
return None
- return ApplicationType.from_flasher_application_type(flasher.app_type)
+ return FirmwareInfo(
+ device=device,
+ firmware_type=ApplicationType.from_flasher_application_type(flasher.app_type),
+ firmware_version=(
+ flasher.app_version.orig_version
+ if flasher.app_version is not None
+ else None
+ ),
+ source="probe",
+ owners=[],
+ )
+
+
+async def probe_silabs_firmware_type(
+ device: str, *, probe_methods: Iterable[ApplicationType] | None = None
+) -> ApplicationType | None:
+ """Probe the running firmware type on a SiLabs device."""
+
+ fw_info = await probe_silabs_firmware_info(device, probe_methods=probe_methods)
+ if fw_info is None:
+ return None
+
+ return fw_info.firmware_type
diff --git a/homeassistant/components/homeassistant_sky_connect/__init__.py b/homeassistant/components/homeassistant_sky_connect/__init__.py
index 758f0c1e1ef..b3af47df61d 100644
--- a/homeassistant/components/homeassistant_sky_connect/__init__.py
+++ b/homeassistant/components/homeassistant_sky_connect/__init__.py
@@ -8,11 +8,16 @@ from homeassistant.components.homeassistant_hardware.util import guess_firmware_
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
+from .const import DESCRIPTION, DEVICE, FIRMWARE, FIRMWARE_VERSION, PRODUCT
+
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up a Home Assistant SkyConnect config entry."""
+
+ await hass.config_entries.async_forward_entry_setups(entry, ["update"])
+
return True
@@ -33,15 +38,13 @@ async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
# Add-on startup with type service get started before Core, always (e.g. the
# Multi-Protocol add-on). Probing the firmware would interfere with the add-on,
# so we can't safely probe here. Instead, we must make an educated guess!
- firmware_guess = await guess_firmware_info(
- hass, config_entry.data["device"]
- )
+ firmware_guess = await guess_firmware_info(hass, config_entry.data[DEVICE])
new_data = {**config_entry.data}
- new_data["firmware"] = firmware_guess.firmware_type.value
+ new_data[FIRMWARE] = firmware_guess.firmware_type.value
# Copy `description` to `product`
- new_data["product"] = new_data["description"]
+ new_data[PRODUCT] = new_data[DESCRIPTION]
hass.config_entries.async_update_entry(
config_entry,
@@ -50,6 +53,18 @@ async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
minor_version=2,
)
+ if config_entry.minor_version == 2:
+ # Add a `firmware_version` key
+ hass.config_entries.async_update_entry(
+ config_entry,
+ data={
+ **config_entry.data,
+ FIRMWARE_VERSION: None,
+ },
+ version=1,
+ minor_version=3,
+ )
+
_LOGGER.debug(
"Migration to version %s.%s successful",
config_entry.version,
diff --git a/homeassistant/components/homeassistant_sky_connect/config_flow.py b/homeassistant/components/homeassistant_sky_connect/config_flow.py
index b3b4f68ba96..d28d74a681c 100644
--- a/homeassistant/components/homeassistant_sky_connect/config_flow.py
+++ b/homeassistant/components/homeassistant_sky_connect/config_flow.py
@@ -10,7 +10,10 @@ from homeassistant.components.homeassistant_hardware import (
firmware_config_flow,
silabs_multiprotocol_addon,
)
-from homeassistant.components.homeassistant_hardware.util import ApplicationType
+from homeassistant.components.homeassistant_hardware.util import (
+ ApplicationType,
+ FirmwareInfo,
+)
from homeassistant.config_entries import (
ConfigEntry,
ConfigEntryBaseFlow,
@@ -21,7 +24,20 @@ from homeassistant.config_entries import (
from homeassistant.core import callback
from homeassistant.helpers.service_info.usb import UsbServiceInfo
-from .const import DOCS_WEB_FLASHER_URL, DOMAIN, HardwareVariant
+from .const import (
+ DESCRIPTION,
+ DEVICE,
+ DOCS_WEB_FLASHER_URL,
+ DOMAIN,
+ FIRMWARE,
+ FIRMWARE_VERSION,
+ MANUFACTURER,
+ PID,
+ PRODUCT,
+ SERIAL_NUMBER,
+ VID,
+ HardwareVariant,
+)
from .util import get_hardware_variant, get_usb_service_info
_LOGGER = logging.getLogger(__name__)
@@ -34,6 +50,7 @@ if TYPE_CHECKING:
def _get_translation_placeholders(self) -> dict[str, str]:
return {}
+
else:
# Multiple inheritance with `Protocol` seems to break
TranslationPlaceholderProtocol = object
@@ -64,7 +81,7 @@ class HomeAssistantSkyConnectConfigFlow(
"""Handle a config flow for Home Assistant SkyConnect."""
VERSION = 1
- MINOR_VERSION = 2
+ MINOR_VERSION = 3
def __init__(self, *args: Any, **kwargs: Any) -> None:
"""Initialize the config flow."""
@@ -79,7 +96,7 @@ class HomeAssistantSkyConnectConfigFlow(
config_entry: ConfigEntry,
) -> OptionsFlow:
"""Return the options flow."""
- firmware_type = ApplicationType(config_entry.data["firmware"])
+ firmware_type = ApplicationType(config_entry.data[FIRMWARE])
if firmware_type is ApplicationType.CPC:
return HomeAssistantSkyConnectMultiPanOptionsFlowHandler(config_entry)
@@ -97,7 +114,7 @@ class HomeAssistantSkyConnectConfigFlow(
unique_id = f"{vid}:{pid}_{serial_number}_{manufacturer}_{description}"
if await self.async_set_unique_id(unique_id):
- self._abort_if_unique_id_configured(updates={"device": device})
+ self._abort_if_unique_id_configured(updates={DEVICE: device})
discovery_info.device = await self.hass.async_add_executor_job(
usb.get_serial_by_id, discovery_info.device
@@ -118,19 +135,20 @@ class HomeAssistantSkyConnectConfigFlow(
"""Create the config entry."""
assert self._usb_info is not None
assert self._hw_variant is not None
- assert self._probed_firmware_type is not None
+ assert self._probed_firmware_info is not None
return self.async_create_entry(
title=self._hw_variant.full_name,
data={
- "vid": self._usb_info.vid,
- "pid": self._usb_info.pid,
- "serial_number": self._usb_info.serial_number,
- "manufacturer": self._usb_info.manufacturer,
- "description": self._usb_info.description, # For backwards compatibility
- "product": self._usb_info.description,
- "device": self._usb_info.device,
- "firmware": self._probed_firmware_type.value,
+ VID: self._usb_info.vid,
+ PID: self._usb_info.pid,
+ SERIAL_NUMBER: self._usb_info.serial_number,
+ MANUFACTURER: self._usb_info.manufacturer,
+ DESCRIPTION: self._usb_info.description, # For backwards compatibility
+ PRODUCT: self._usb_info.description,
+ DEVICE: self._usb_info.device,
+ FIRMWARE: self._probed_firmware_info.firmware_type.value,
+ FIRMWARE_VERSION: self._probed_firmware_info.firmware_version,
},
)
@@ -145,7 +163,7 @@ class HomeAssistantSkyConnectMultiPanOptionsFlowHandler(
) -> silabs_multiprotocol_addon.SerialPortSettings:
"""Return the radio serial port settings."""
return silabs_multiprotocol_addon.SerialPortSettings(
- device=self.config_entry.data["device"],
+ device=self.config_entry.data[DEVICE],
baudrate="115200",
flow_control=True,
)
@@ -179,7 +197,8 @@ class HomeAssistantSkyConnectMultiPanOptionsFlowHandler(
entry=self.config_entry,
data={
**self.config_entry.data,
- "firmware": ApplicationType.EZSP.value,
+ FIRMWARE: ApplicationType.EZSP.value,
+ FIRMWARE_VERSION: None,
},
options=self.config_entry.options,
)
@@ -198,23 +217,32 @@ class HomeAssistantSkyConnectOptionsFlowHandler(
self._usb_info = get_usb_service_info(self.config_entry)
self._hw_variant = HardwareVariant.from_usb_product_name(
- self.config_entry.data["product"]
+ self.config_entry.data[PRODUCT]
)
self._hardware_name = self._hw_variant.full_name
self._device = self._usb_info.device
+ self._probed_firmware_info = FirmwareInfo(
+ device=self._device,
+ firmware_type=ApplicationType(self.config_entry.data[FIRMWARE]),
+ firmware_version=self.config_entry.data[FIRMWARE_VERSION],
+ source="guess",
+ owners=[],
+ )
+
# Regenerate the translation placeholders
self._get_translation_placeholders()
def _async_flow_finished(self) -> ConfigFlowResult:
"""Create the config entry."""
- assert self._probed_firmware_type is not None
+ assert self._probed_firmware_info is not None
self.hass.config_entries.async_update_entry(
entry=self.config_entry,
data={
**self.config_entry.data,
- "firmware": self._probed_firmware_type.value,
+ FIRMWARE: self._probed_firmware_info.firmware_type.value,
+ FIRMWARE_VERSION: self._probed_firmware_info.firmware_version,
},
options=self.config_entry.options,
)
diff --git a/homeassistant/components/homeassistant_sky_connect/const.py b/homeassistant/components/homeassistant_sky_connect/const.py
index cae0b98a25b..70ff047366d 100644
--- a/homeassistant/components/homeassistant_sky_connect/const.py
+++ b/homeassistant/components/homeassistant_sky_connect/const.py
@@ -7,6 +7,20 @@ from typing import Self
DOMAIN = "homeassistant_sky_connect"
DOCS_WEB_FLASHER_URL = "https://skyconnect.home-assistant.io/firmware-update/"
+NABU_CASA_FIRMWARE_RELEASES_URL = (
+ "https://api.github.com/repos/NabuCasa/silabs-firmware-builder/releases/latest"
+)
+
+FIRMWARE = "firmware"
+FIRMWARE_VERSION = "firmware_version"
+SERIAL_NUMBER = "serial_number"
+MANUFACTURER = "manufacturer"
+PRODUCT = "product"
+DESCRIPTION = "description"
+PID = "pid"
+VID = "vid"
+DEVICE = "device"
+
@dataclasses.dataclass(frozen=True)
class VariantInfo:
diff --git a/homeassistant/components/homeassistant_sky_connect/update.py b/homeassistant/components/homeassistant_sky_connect/update.py
new file mode 100644
index 00000000000..43e3f1ca255
--- /dev/null
+++ b/homeassistant/components/homeassistant_sky_connect/update.py
@@ -0,0 +1,169 @@
+"""Home Assistant SkyConnect firmware update entity."""
+
+from __future__ import annotations
+
+import logging
+
+import aiohttp
+
+from homeassistant.components.homeassistant_hardware.coordinator import (
+ FirmwareUpdateCoordinator,
+)
+from homeassistant.components.homeassistant_hardware.update import (
+ BaseFirmwareUpdateEntity,
+ FirmwareUpdateEntityDescription,
+)
+from homeassistant.components.homeassistant_hardware.util import (
+ ApplicationType,
+ FirmwareInfo,
+)
+from homeassistant.components.update import UpdateDeviceClass
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers import entity_registry as er
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .const import FIRMWARE, FIRMWARE_VERSION, NABU_CASA_FIRMWARE_RELEASES_URL
+
+_LOGGER = logging.getLogger(__name__)
+
+
+FIRMWARE_ENTITY_DESCRIPTIONS: dict[
+ ApplicationType | None, FirmwareUpdateEntityDescription
+] = {
+ ApplicationType.EZSP: FirmwareUpdateEntityDescription(
+ key="firmware",
+ display_precision=0,
+ device_class=UpdateDeviceClass.FIRMWARE,
+ entity_category=EntityCategory.CONFIG,
+ version_parser=lambda fw: fw.split(" ", 1)[0],
+ fw_type="skyconnect_zigbee_ncp",
+ version_key="ezsp_version",
+ expected_firmware_type=ApplicationType.EZSP,
+ firmware_name="EmberZNet",
+ ),
+ ApplicationType.SPINEL: FirmwareUpdateEntityDescription(
+ key="firmware",
+ display_precision=0,
+ device_class=UpdateDeviceClass.FIRMWARE,
+ entity_category=EntityCategory.CONFIG,
+ version_parser=lambda fw: fw.split("/", 1)[1].split("_", 1)[0],
+ fw_type="skyconnect_openthread_rcp",
+ version_key="ot_rcp_version",
+ expected_firmware_type=ApplicationType.SPINEL,
+ firmware_name="OpenThread RCP",
+ ),
+ None: FirmwareUpdateEntityDescription(
+ key="firmware",
+ display_precision=0,
+ device_class=UpdateDeviceClass.FIRMWARE,
+ entity_category=EntityCategory.CONFIG,
+ version_parser=lambda fw: fw,
+ fw_type=None,
+ version_key=None,
+ expected_firmware_type=None,
+ firmware_name=None,
+ ),
+}
+
+
+def _async_create_update_entity(
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ session: aiohttp.ClientSession,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> FirmwareUpdateEntity:
+ """Create an update entity that handles firmware type changes."""
+ firmware_type = config_entry.data[FIRMWARE]
+ entity_description = FIRMWARE_ENTITY_DESCRIPTIONS[
+ ApplicationType(firmware_type) if firmware_type is not None else None
+ ]
+
+ entity = FirmwareUpdateEntity(
+ device=config_entry.data["device"],
+ config_entry=config_entry,
+ update_coordinator=FirmwareUpdateCoordinator(
+ hass,
+ session,
+ NABU_CASA_FIRMWARE_RELEASES_URL,
+ ),
+ entity_description=entity_description,
+ )
+
+ def firmware_type_changed(
+ old_type: ApplicationType | None, new_type: ApplicationType | None
+ ) -> None:
+ """Replace the current entity when the firmware type changes."""
+ er.async_get(hass).async_remove(entity.entity_id)
+ async_add_entities(
+ [
+ _async_create_update_entity(
+ hass, config_entry, session, async_add_entities
+ )
+ ]
+ )
+
+ entity.async_on_remove(
+ entity.add_firmware_type_changed_callback(firmware_type_changed)
+ )
+
+ return entity
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the firmware update config entry."""
+ session = async_get_clientsession(hass)
+ entity = _async_create_update_entity(
+ hass, config_entry, session, async_add_entities
+ )
+
+ async_add_entities([entity])
+
+
+class FirmwareUpdateEntity(BaseFirmwareUpdateEntity):
+ """SkyConnect firmware update entity."""
+
+ bootloader_reset_type = None
+
+ def __init__(
+ self,
+ device: str,
+ config_entry: ConfigEntry,
+ update_coordinator: FirmwareUpdateCoordinator,
+ entity_description: FirmwareUpdateEntityDescription,
+ ) -> None:
+ """Initialize the SkyConnect firmware update entity."""
+ super().__init__(device, config_entry, update_coordinator, entity_description)
+
+ self._attr_unique_id = (
+ f"{self._config_entry.data['serial_number']}_{self.entity_description.key}"
+ )
+
+ # Use the cached firmware info if it exists
+ if self._config_entry.data[FIRMWARE] is not None:
+ self._current_firmware_info = FirmwareInfo(
+ device=device,
+ firmware_type=ApplicationType(self._config_entry.data[FIRMWARE]),
+ firmware_version=self._config_entry.data[FIRMWARE_VERSION],
+ owners=[],
+ source="homeassistant_sky_connect",
+ )
+
+ @callback
+ def _firmware_info_callback(self, firmware_info: FirmwareInfo) -> None:
+ """Handle updated firmware info being pushed by an integration."""
+ self.hass.config_entries.async_update_entry(
+ self._config_entry,
+ data={
+ **self._config_entry.data,
+ FIRMWARE: firmware_info.firmware_type,
+ FIRMWARE_VERSION: firmware_info.firmware_version,
+ },
+ )
+ super()._firmware_info_callback(firmware_info)
diff --git a/homeassistant/components/homeassistant_yellow/__init__.py b/homeassistant/components/homeassistant_yellow/__init__.py
index b0837eeedbe..06f908ab61e 100644
--- a/homeassistant/components/homeassistant_yellow/__init__.py
+++ b/homeassistant/components/homeassistant_yellow/__init__.py
@@ -18,7 +18,7 @@ from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
from homeassistant.helpers import discovery_flow
from homeassistant.helpers.hassio import is_hassio
-from .const import FIRMWARE, RADIO_DEVICE, ZHA_HW_DISCOVERY_DATA
+from .const import FIRMWARE, FIRMWARE_VERSION, RADIO_DEVICE, ZHA_HW_DISCOVERY_DATA
_LOGGER = logging.getLogger(__name__)
@@ -55,6 +55,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
data=ZHA_HW_DISCOVERY_DATA,
)
+ await hass.config_entries.async_forward_entry_setups(entry, ["update"])
+
return True
@@ -87,6 +89,18 @@ async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
minor_version=2,
)
+ if config_entry.minor_version == 2:
+ # Add a `firmware_version` key
+ hass.config_entries.async_update_entry(
+ config_entry,
+ data={
+ **config_entry.data,
+ FIRMWARE_VERSION: None,
+ },
+ version=1,
+ minor_version=3,
+ )
+
_LOGGER.debug(
"Migration to version %s.%s successful",
config_entry.version,
diff --git a/homeassistant/components/homeassistant_yellow/config_flow.py b/homeassistant/components/homeassistant_yellow/config_flow.py
index 502a20db07c..5472c346e94 100644
--- a/homeassistant/components/homeassistant_yellow/config_flow.py
+++ b/homeassistant/components/homeassistant_yellow/config_flow.py
@@ -24,7 +24,10 @@ from homeassistant.components.homeassistant_hardware.silabs_multiprotocol_addon
OptionsFlowHandler as MultiprotocolOptionsFlowHandler,
SerialPortSettings as MultiprotocolSerialPortSettings,
)
-from homeassistant.components.homeassistant_hardware.util import ApplicationType
+from homeassistant.components.homeassistant_hardware.util import (
+ ApplicationType,
+ FirmwareInfo,
+)
from homeassistant.config_entries import (
SOURCE_HARDWARE,
ConfigEntry,
@@ -34,7 +37,14 @@ from homeassistant.config_entries import (
from homeassistant.core import HomeAssistant, async_get_hass, callback
from homeassistant.helpers import discovery_flow, selector
-from .const import DOMAIN, FIRMWARE, RADIO_DEVICE, ZHA_DOMAIN, ZHA_HW_DISCOVERY_DATA
+from .const import (
+ DOMAIN,
+ FIRMWARE,
+ FIRMWARE_VERSION,
+ RADIO_DEVICE,
+ ZHA_DOMAIN,
+ ZHA_HW_DISCOVERY_DATA,
+)
from .hardware import BOARD_NAME
_LOGGER = logging.getLogger(__name__)
@@ -52,7 +62,7 @@ class HomeAssistantYellowConfigFlow(BaseFirmwareConfigFlow, domain=DOMAIN):
"""Handle a config flow for Home Assistant Yellow."""
VERSION = 1
- MINOR_VERSION = 2
+ MINOR_VERSION = 3
def __init__(self, *args: Any, **kwargs: Any) -> None:
"""Instantiate config flow."""
@@ -79,10 +89,13 @@ class HomeAssistantYellowConfigFlow(BaseFirmwareConfigFlow, domain=DOMAIN):
) -> ConfigFlowResult:
"""Handle the initial step."""
# We do not actually use any portion of `BaseFirmwareConfigFlow` beyond this
- await self._probe_firmware_type()
+ await self._probe_firmware_info()
# Kick off ZHA hardware discovery automatically if Zigbee firmware is running
- if self._probed_firmware_type is ApplicationType.EZSP:
+ if (
+ self._probed_firmware_info is not None
+ and self._probed_firmware_info.firmware_type is ApplicationType.EZSP
+ ):
discovery_flow.async_create_flow(
self.hass,
ZHA_DOMAIN,
@@ -98,7 +111,11 @@ class HomeAssistantYellowConfigFlow(BaseFirmwareConfigFlow, domain=DOMAIN):
title=BOARD_NAME,
data={
# Assume the firmware type is EZSP if we cannot probe it
- FIRMWARE: (self._probed_firmware_type or ApplicationType.EZSP).value,
+ FIRMWARE: (
+ self._probed_firmware_info.firmware_type
+ if self._probed_firmware_info is not None
+ else ApplicationType.EZSP
+ ).value,
},
)
@@ -264,6 +281,14 @@ class HomeAssistantYellowOptionsFlowHandler(
self._hardware_name = BOARD_NAME
self._device = RADIO_DEVICE
+ self._probed_firmware_info = FirmwareInfo(
+ device=self._device,
+ firmware_type=ApplicationType(self.config_entry.data["firmware"]),
+ firmware_version=None,
+ source="guess",
+ owners=[],
+ )
+
# Regenerate the translation placeholders
self._get_translation_placeholders()
@@ -285,13 +310,14 @@ class HomeAssistantYellowOptionsFlowHandler(
def _async_flow_finished(self) -> ConfigFlowResult:
"""Create the config entry."""
- assert self._probed_firmware_type is not None
+ assert self._probed_firmware_info is not None
self.hass.config_entries.async_update_entry(
entry=self.config_entry,
data={
**self.config_entry.data,
- FIRMWARE: self._probed_firmware_type.value,
+ FIRMWARE: self._probed_firmware_info.firmware_type.value,
+ FIRMWARE_VERSION: self._probed_firmware_info.firmware_version,
},
)
diff --git a/homeassistant/components/homeassistant_yellow/const.py b/homeassistant/components/homeassistant_yellow/const.py
index 79753ae9b9e..b98b1133d01 100644
--- a/homeassistant/components/homeassistant_yellow/const.py
+++ b/homeassistant/components/homeassistant_yellow/const.py
@@ -2,7 +2,10 @@
DOMAIN = "homeassistant_yellow"
+RADIO_MODEL = "Home Assistant Yellow"
+RADIO_MANUFACTURER = "Nabu Casa"
RADIO_DEVICE = "/dev/ttyAMA1"
+
ZHA_HW_DISCOVERY_DATA = {
"name": "Yellow",
"port": {
@@ -14,4 +17,9 @@ ZHA_HW_DISCOVERY_DATA = {
}
FIRMWARE = "firmware"
+FIRMWARE_VERSION = "firmware_version"
ZHA_DOMAIN = "zha"
+
+NABU_CASA_FIRMWARE_RELEASES_URL = (
+ "https://api.github.com/repos/NabuCasa/silabs-firmware-builder/releases/latest"
+)
diff --git a/homeassistant/components/homeassistant_yellow/update.py b/homeassistant/components/homeassistant_yellow/update.py
new file mode 100644
index 00000000000..88d4f2912d3
--- /dev/null
+++ b/homeassistant/components/homeassistant_yellow/update.py
@@ -0,0 +1,172 @@
+"""Home Assistant Yellow firmware update entity."""
+
+from __future__ import annotations
+
+import logging
+
+import aiohttp
+
+from homeassistant.components.homeassistant_hardware.coordinator import (
+ FirmwareUpdateCoordinator,
+)
+from homeassistant.components.homeassistant_hardware.update import (
+ BaseFirmwareUpdateEntity,
+ FirmwareUpdateEntityDescription,
+)
+from homeassistant.components.homeassistant_hardware.util import (
+ ApplicationType,
+ FirmwareInfo,
+)
+from homeassistant.components.update import UpdateDeviceClass
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers import entity_registry as er
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .const import (
+ FIRMWARE,
+ FIRMWARE_VERSION,
+ NABU_CASA_FIRMWARE_RELEASES_URL,
+ RADIO_DEVICE,
+)
+
+_LOGGER = logging.getLogger(__name__)
+
+
+FIRMWARE_ENTITY_DESCRIPTIONS: dict[
+ ApplicationType | None, FirmwareUpdateEntityDescription
+] = {
+ ApplicationType.EZSP: FirmwareUpdateEntityDescription(
+ key="firmware",
+ display_precision=0,
+ device_class=UpdateDeviceClass.FIRMWARE,
+ entity_category=EntityCategory.CONFIG,
+ version_parser=lambda fw: fw.split(" ", 1)[0],
+ fw_type="yellow_zigbee_ncp",
+ version_key="ezsp_version",
+ expected_firmware_type=ApplicationType.EZSP,
+ firmware_name="EmberZNet",
+ ),
+ ApplicationType.SPINEL: FirmwareUpdateEntityDescription(
+ key="firmware",
+ display_precision=0,
+ device_class=UpdateDeviceClass.FIRMWARE,
+ entity_category=EntityCategory.CONFIG,
+ version_parser=lambda fw: fw.split("/", 1)[1].split("_", 1)[0],
+ fw_type="yellow_openthread_rcp",
+ version_key="ot_rcp_version",
+ expected_firmware_type=ApplicationType.SPINEL,
+ firmware_name="OpenThread RCP",
+ ),
+ None: FirmwareUpdateEntityDescription(
+ key="firmware",
+ display_precision=0,
+ device_class=UpdateDeviceClass.FIRMWARE,
+ entity_category=EntityCategory.CONFIG,
+ version_parser=lambda fw: fw,
+ fw_type=None,
+ version_key=None,
+ expected_firmware_type=None,
+ firmware_name=None,
+ ),
+}
+
+
+def _async_create_update_entity(
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ session: aiohttp.ClientSession,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> FirmwareUpdateEntity:
+ """Create an update entity that handles firmware type changes."""
+ firmware_type = config_entry.data[FIRMWARE]
+ entity_description = FIRMWARE_ENTITY_DESCRIPTIONS[
+ ApplicationType(firmware_type) if firmware_type is not None else None
+ ]
+
+ entity = FirmwareUpdateEntity(
+ device=RADIO_DEVICE,
+ config_entry=config_entry,
+ update_coordinator=FirmwareUpdateCoordinator(
+ hass,
+ session,
+ NABU_CASA_FIRMWARE_RELEASES_URL,
+ ),
+ entity_description=entity_description,
+ )
+
+ def firmware_type_changed(
+ old_type: ApplicationType | None, new_type: ApplicationType | None
+ ) -> None:
+ """Replace the current entity when the firmware type changes."""
+ er.async_get(hass).async_remove(entity.entity_id)
+ async_add_entities(
+ [
+ _async_create_update_entity(
+ hass, config_entry, session, async_add_entities
+ )
+ ]
+ )
+
+ entity.async_on_remove(
+ entity.add_firmware_type_changed_callback(firmware_type_changed)
+ )
+
+ return entity
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the firmware update config entry."""
+ session = async_get_clientsession(hass)
+ entity = _async_create_update_entity(
+ hass, config_entry, session, async_add_entities
+ )
+
+ async_add_entities([entity])
+
+
+class FirmwareUpdateEntity(BaseFirmwareUpdateEntity):
+ """Yellow firmware update entity."""
+
+ bootloader_reset_type = "yellow" # Triggers a GPIO reset
+
+ def __init__(
+ self,
+ device: str,
+ config_entry: ConfigEntry,
+ update_coordinator: FirmwareUpdateCoordinator,
+ entity_description: FirmwareUpdateEntityDescription,
+ ) -> None:
+ """Initialize the Yellow firmware update entity."""
+ super().__init__(device, config_entry, update_coordinator, entity_description)
+
+ self._attr_unique_id = self.entity_description.key
+
+ # Use the cached firmware info if it exists
+ if self._config_entry.data[FIRMWARE] is not None:
+ self._current_firmware_info = FirmwareInfo(
+ device=device,
+ firmware_type=ApplicationType(self._config_entry.data[FIRMWARE]),
+ firmware_version=self._config_entry.data[FIRMWARE_VERSION],
+ owners=[],
+ source="homeassistant_yellow",
+ )
+
+ @callback
+ def _firmware_info_callback(self, firmware_info: FirmwareInfo) -> None:
+ """Handle updated firmware info being pushed by an integration."""
+ self.hass.config_entries.async_update_entry(
+ self._config_entry,
+ data={
+ **self._config_entry.data,
+ FIRMWARE: firmware_info.firmware_type,
+ FIRMWARE_VERSION: firmware_info.firmware_version,
+ },
+ )
+ super()._firmware_info_callback(firmware_info)
diff --git a/homeassistant/components/homee/__init__.py b/homeassistant/components/homee/__init__.py
index 9837d6094ff..6158a699302 100644
--- a/homeassistant/components/homee/__init__.py
+++ b/homeassistant/components/homee/__init__.py
@@ -14,7 +14,17 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
-PLATFORMS = [Platform.COVER, Platform.SENSOR]
+PLATFORMS = [
+ Platform.BINARY_SENSOR,
+ Platform.BUTTON,
+ Platform.COVER,
+ Platform.LIGHT,
+ Platform.NUMBER,
+ Platform.SELECT,
+ Platform.SENSOR,
+ Platform.SWITCH,
+ Platform.VALVE,
+]
type HomeeConfigEntry = ConfigEntry[Homee]
diff --git a/homeassistant/components/homee/binary_sensor.py b/homeassistant/components/homee/binary_sensor.py
new file mode 100644
index 00000000000..3f5f5c46a29
--- /dev/null
+++ b/homeassistant/components/homee/binary_sensor.py
@@ -0,0 +1,190 @@
+"""The Homee binary sensor platform."""
+
+from pyHomee.const import AttributeType
+from pyHomee.model import HomeeAttribute
+
+from homeassistant.components.binary_sensor import (
+ BinarySensorDeviceClass,
+ BinarySensorEntity,
+ BinarySensorEntityDescription,
+)
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import HomeeConfigEntry
+from .entity import HomeeEntity
+
+PARALLEL_UPDATES = 0
+
+BINARY_SENSOR_DESCRIPTIONS: dict[AttributeType, BinarySensorEntityDescription] = {
+ AttributeType.BATTERY_LOW_ALARM: BinarySensorEntityDescription(
+ key="battery",
+ device_class=BinarySensorDeviceClass.BATTERY,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.BLACKOUT_ALARM: BinarySensorEntityDescription(
+ key="blackout_alarm",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.COALARM: BinarySensorEntityDescription(
+ key="carbon_monoxide", device_class=BinarySensorDeviceClass.CO
+ ),
+ AttributeType.CO2ALARM: BinarySensorEntityDescription(
+ key="carbon_dioxide", device_class=BinarySensorDeviceClass.PROBLEM
+ ),
+ AttributeType.FLOOD_ALARM: BinarySensorEntityDescription(
+ key="flood",
+ device_class=BinarySensorDeviceClass.MOISTURE,
+ ),
+ AttributeType.HIGH_TEMPERATURE_ALARM: BinarySensorEntityDescription(
+ key="high_temperature",
+ device_class=BinarySensorDeviceClass.HEAT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.LEAK_ALARM: BinarySensorEntityDescription(
+ key="leak_alarm",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ ),
+ AttributeType.LOAD_ALARM: BinarySensorEntityDescription(
+ key="load_alarm",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.LOCK_STATE: BinarySensorEntityDescription(
+ key="lock",
+ device_class=BinarySensorDeviceClass.LOCK,
+ ),
+ AttributeType.LOW_TEMPERATURE_ALARM: BinarySensorEntityDescription(
+ key="low_temperature",
+ device_class=BinarySensorDeviceClass.COLD,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.MALFUNCTION_ALARM: BinarySensorEntityDescription(
+ key="malfunction",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.MAXIMUM_ALARM: BinarySensorEntityDescription(
+ key="maximum",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.MINIMUM_ALARM: BinarySensorEntityDescription(
+ key="minimum",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.MOTION_ALARM: BinarySensorEntityDescription(
+ key="motion",
+ device_class=BinarySensorDeviceClass.MOTION,
+ ),
+ AttributeType.MOTOR_BLOCKED_ALARM: BinarySensorEntityDescription(
+ key="motor_blocked",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.ON_OFF: BinarySensorEntityDescription(
+ key="plug",
+ device_class=BinarySensorDeviceClass.PLUG,
+ ),
+ AttributeType.OPEN_CLOSE: BinarySensorEntityDescription(
+ key="opening",
+ device_class=BinarySensorDeviceClass.OPENING,
+ ),
+ AttributeType.OVER_CURRENT_ALARM: BinarySensorEntityDescription(
+ key="overcurrent",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.OVERLOAD_ALARM: BinarySensorEntityDescription(
+ key="overload",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.PRESENCE_ALARM: BinarySensorEntityDescription(
+ key="presence",
+ device_class=BinarySensorDeviceClass.PRESENCE,
+ ),
+ AttributeType.POWER_SUPPLY_ALARM: BinarySensorEntityDescription(
+ key="power",
+ device_class=BinarySensorDeviceClass.POWER,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.RAIN_FALL: BinarySensorEntityDescription(
+ key="rain",
+ device_class=BinarySensorDeviceClass.MOISTURE,
+ ),
+ AttributeType.REPLACE_FILTER_ALARM: BinarySensorEntityDescription(
+ key="replace_filter",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.SMOKE_ALARM: BinarySensorEntityDescription(
+ key="smoke",
+ device_class=BinarySensorDeviceClass.SMOKE,
+ ),
+ AttributeType.STORAGE_ALARM: BinarySensorEntityDescription(
+ key="storage",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.SURGE_ALARM: BinarySensorEntityDescription(
+ key="surge",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.TAMPER_ALARM: BinarySensorEntityDescription(
+ key="tamper",
+ device_class=BinarySensorDeviceClass.TAMPER,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.VOLTAGE_DROP_ALARM: BinarySensorEntityDescription(
+ key="voltage_drop",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ AttributeType.WATER_ALARM: BinarySensorEntityDescription(
+ key="water",
+ device_class=BinarySensorDeviceClass.MOISTURE,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: HomeeConfigEntry,
+ async_add_devices: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add the Homee platform for the binary sensor component."""
+
+ async_add_devices(
+ HomeeBinarySensor(
+ attribute, config_entry, BINARY_SENSOR_DESCRIPTIONS[attribute.type]
+ )
+ for node in config_entry.runtime_data.nodes
+ for attribute in node.attributes
+ if attribute.type in BINARY_SENSOR_DESCRIPTIONS and not attribute.editable
+ )
+
+
+class HomeeBinarySensor(HomeeEntity, BinarySensorEntity):
+ """Representation of a Homee binary sensor."""
+
+ def __init__(
+ self,
+ attribute: HomeeAttribute,
+ entry: HomeeConfigEntry,
+ description: BinarySensorEntityDescription,
+ ) -> None:
+ """Initialize a Homee binary sensor entity."""
+ super().__init__(attribute, entry)
+
+ self.entity_description = description
+ self._attr_translation_key = description.key
+
+ @property
+ def is_on(self) -> bool:
+ """Return true if the binary sensor is on."""
+ return bool(self._attribute.current_value)
diff --git a/homeassistant/components/homee/button.py b/homeassistant/components/homee/button.py
new file mode 100644
index 00000000000..33a8b5f23c8
--- /dev/null
+++ b/homeassistant/components/homee/button.py
@@ -0,0 +1,80 @@
+"""The homee button platform."""
+
+from pyHomee.const import AttributeType
+from pyHomee.model import HomeeAttribute
+
+from homeassistant.components.button import (
+ ButtonDeviceClass,
+ ButtonEntity,
+ ButtonEntityDescription,
+)
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import HomeeConfigEntry
+from .entity import HomeeEntity
+
+PARALLEL_UPDATES = 0
+
+BUTTON_DESCRIPTIONS: dict[AttributeType, ButtonEntityDescription] = {
+ AttributeType.AUTOMATIC_MODE_IMPULSE: ButtonEntityDescription(key="automatic_mode"),
+ AttributeType.BRIEFLY_OPEN_IMPULSE: ButtonEntityDescription(key="briefly_open"),
+ AttributeType.IDENTIFICATION_MODE: ButtonEntityDescription(
+ key="identification_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=ButtonDeviceClass.IDENTIFY,
+ ),
+ AttributeType.IMPULSE: ButtonEntityDescription(key="impulse"),
+ AttributeType.LIGHT_IMPULSE: ButtonEntityDescription(key="light"),
+ AttributeType.OPEN_PARTIAL_IMPULSE: ButtonEntityDescription(key="open_partial"),
+ AttributeType.PERMANENTLY_OPEN_IMPULSE: ButtonEntityDescription(
+ key="permanently_open"
+ ),
+ AttributeType.RESET_METER: ButtonEntityDescription(
+ key="reset_meter",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.VENTILATE_IMPULSE: ButtonEntityDescription(key="ventilate"),
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: HomeeConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add the Homee platform for the button component."""
+
+ async_add_entities(
+ HomeeButton(attribute, config_entry, BUTTON_DESCRIPTIONS[attribute.type])
+ for node in config_entry.runtime_data.nodes
+ for attribute in node.attributes
+ if attribute.type in BUTTON_DESCRIPTIONS and attribute.editable
+ )
+
+
+class HomeeButton(HomeeEntity, ButtonEntity):
+ """Representation of a Homee button."""
+
+ def __init__(
+ self,
+ attribute: HomeeAttribute,
+ entry: HomeeConfigEntry,
+ description: ButtonEntityDescription,
+ ) -> None:
+ """Initialize a Homee button entity."""
+ super().__init__(attribute, entry)
+ self.entity_description = description
+ if attribute.instance == 0:
+ if attribute.type == AttributeType.IMPULSE:
+ self._attr_name = None
+ else:
+ self._attr_translation_key = description.key
+ else:
+ self._attr_translation_key = f"{description.key}_instance"
+ self._attr_translation_placeholders = {"instance": str(attribute.instance)}
+
+ async def async_press(self) -> None:
+ """Handle the button press."""
+ await self.async_set_homee_value(1)
diff --git a/homeassistant/components/homee/const.py b/homeassistant/components/homee/const.py
index 1d7ce27335f..2c614d3f5eb 100644
--- a/homeassistant/components/homee/const.py
+++ b/homeassistant/components/homee/const.py
@@ -1,5 +1,7 @@
"""Constants for the homee integration."""
+from pyHomee.const import NodeProfile
+
from homeassistant.const import (
DEGREE,
LIGHT_LUX,
@@ -62,3 +64,34 @@ WINDOW_MAP = {
2.0: "tilted",
}
WINDOW_MAP_REVERSED = {0.0: "open", 1.0: "closed", 2.0: "tilted"}
+
+# Profile Groups
+CLIMATE_PROFILES = [
+ NodeProfile.COSI_THERM_CHANNEL,
+ NodeProfile.HEATING_SYSTEM,
+ NodeProfile.RADIATOR_THERMOSTAT,
+ NodeProfile.ROOM_THERMOSTAT,
+ NodeProfile.ROOM_THERMOSTAT_WITH_HUMIDITY_SENSOR,
+ NodeProfile.THERMOSTAT_WITH_HEATING_AND_COOLING,
+ NodeProfile.WIFI_RADIATOR_THERMOSTAT,
+ NodeProfile.WIFI_ROOM_THERMOSTAT,
+]
+
+LIGHT_PROFILES = [
+ NodeProfile.DIMMABLE_COLOR_LIGHT,
+ NodeProfile.DIMMABLE_COLOR_METERING_PLUG,
+ NodeProfile.DIMMABLE_COLOR_TEMPERATURE_LIGHT,
+ NodeProfile.DIMMABLE_EXTENDED_COLOR_LIGHT,
+ NodeProfile.DIMMABLE_LIGHT,
+ NodeProfile.DIMMABLE_LIGHT_WITH_BRIGHTNESS_SENSOR,
+ NodeProfile.DIMMABLE_LIGHT_WITH_BRIGHTNESS_AND_PRESENCE_SENSOR,
+ NodeProfile.DIMMABLE_LIGHT_WITH_PRESENCE_SENSOR,
+ NodeProfile.DIMMABLE_METERING_SWITCH,
+ NodeProfile.DIMMABLE_METERING_PLUG,
+ NodeProfile.DIMMABLE_PLUG,
+ NodeProfile.DIMMABLE_RGBWLIGHT,
+ NodeProfile.DIMMABLE_SWITCH,
+ NodeProfile.WIFI_DIMMABLE_RGBWLIGHT,
+ NodeProfile.WIFI_DIMMABLE_LIGHT,
+ NodeProfile.WIFI_ON_OFF_DIMMABLE_METERING_SWITCH,
+]
diff --git a/homeassistant/components/homee/cover.py b/homeassistant/components/homee/cover.py
index 2e6f7babaff..79a9b00ffba 100644
--- a/homeassistant/components/homee/cover.py
+++ b/homeassistant/components/homee/cover.py
@@ -14,13 +14,15 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import HomeeConfigEntry
from .entity import HomeeNodeEntity
_LOGGER = logging.getLogger(__name__)
+PARALLEL_UPDATES = 0
+
OPEN_CLOSE_ATTRIBUTES = [
AttributeType.OPEN_CLOSE,
AttributeType.SLAT_ROTATION_IMPULSE,
@@ -78,7 +80,7 @@ def get_device_class(node: HomeeNode) -> CoverDeviceClass | None:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HomeeConfigEntry,
- async_add_devices: AddEntitiesCallback,
+ async_add_devices: AddConfigEntryEntitiesCallback,
) -> None:
"""Add the homee platform for the cover integration."""
@@ -205,17 +207,17 @@ class HomeeCover(HomeeNodeEntity, CoverEntity):
"""Open the cover."""
assert self._open_close_attribute is not None
if not self._open_close_attribute.is_reversed:
- await self.async_set_value(self._open_close_attribute, 0)
+ await self.async_set_homee_value(self._open_close_attribute, 0)
else:
- await self.async_set_value(self._open_close_attribute, 1)
+ await self.async_set_homee_value(self._open_close_attribute, 1)
async def async_close_cover(self, **kwargs: Any) -> None:
"""Close cover."""
assert self._open_close_attribute is not None
if not self._open_close_attribute.is_reversed:
- await self.async_set_value(self._open_close_attribute, 1)
+ await self.async_set_homee_value(self._open_close_attribute, 1)
else:
- await self.async_set_value(self._open_close_attribute, 0)
+ await self.async_set_homee_value(self._open_close_attribute, 0)
async def async_set_cover_position(self, **kwargs: Any) -> None:
"""Move the cover to a specific position."""
@@ -230,12 +232,12 @@ class HomeeCover(HomeeNodeEntity, CoverEntity):
homee_max = attribute.maximum
homee_position = (position / 100) * (homee_max - homee_min) + homee_min
- await self.async_set_value(attribute, homee_position)
+ await self.async_set_homee_value(attribute, homee_position)
async def async_stop_cover(self, **kwargs: Any) -> None:
"""Stop the cover."""
if self._open_close_attribute is not None:
- await self.async_set_value(self._open_close_attribute, 2)
+ await self.async_set_homee_value(self._open_close_attribute, 2)
async def async_open_cover_tilt(self, **kwargs: Any) -> None:
"""Open the cover tilt."""
@@ -245,9 +247,9 @@ class HomeeCover(HomeeNodeEntity, CoverEntity):
)
) is not None:
if not slat_attribute.is_reversed:
- await self.async_set_value(slat_attribute, 2)
+ await self.async_set_homee_value(slat_attribute, 2)
else:
- await self.async_set_value(slat_attribute, 1)
+ await self.async_set_homee_value(slat_attribute, 1)
async def async_close_cover_tilt(self, **kwargs: Any) -> None:
"""Close the cover tilt."""
@@ -257,9 +259,9 @@ class HomeeCover(HomeeNodeEntity, CoverEntity):
)
) is not None:
if not slat_attribute.is_reversed:
- await self.async_set_value(slat_attribute, 1)
+ await self.async_set_homee_value(slat_attribute, 1)
else:
- await self.async_set_value(slat_attribute, 2)
+ await self.async_set_homee_value(slat_attribute, 2)
async def async_set_cover_tilt_position(self, **kwargs: Any) -> None:
"""Move the cover tilt to a specific position."""
@@ -276,4 +278,4 @@ class HomeeCover(HomeeNodeEntity, CoverEntity):
homee_max = attribute.maximum
homee_position = (position / 100) * (homee_max - homee_min) + homee_min
- await self.async_set_value(attribute, homee_position)
+ await self.async_set_homee_value(attribute, homee_position)
diff --git a/homeassistant/components/homee/entity.py b/homeassistant/components/homee/entity.py
index 5a46b366d3e..165a655d82b 100644
--- a/homeassistant/components/homee/entity.py
+++ b/homeassistant/components/homee/entity.py
@@ -26,10 +26,14 @@ class HomeeEntity(Entity):
f"{entry.runtime_data.settings.uid}-{attribute.node_id}-{attribute.id}"
)
self._entry = entry
+ node = entry.runtime_data.get_node_by_id(attribute.node_id)
self._attr_device_info = DeviceInfo(
identifiers={
(DOMAIN, f"{entry.runtime_data.settings.uid}-{attribute.node_id}")
- }
+ },
+ name=node.name,
+ model=get_name_for_enum(NodeProfile, node.profile),
+ via_device=(DOMAIN, entry.runtime_data.settings.uid),
)
self._host_connected = entry.runtime_data.connected
@@ -50,6 +54,17 @@ class HomeeEntity(Entity):
"""Return the availability of the underlying node."""
return (self._attribute.state == AttributeState.NORMAL) and self._host_connected
+ async def async_set_homee_value(self, value: float) -> None:
+ """Set an attribute value on the homee node."""
+ homee = self._entry.runtime_data
+ try:
+ await homee.set_value(self._attribute.node_id, self._attribute.id, value)
+ except ConnectionClosed as exception:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="connection_closed",
+ ) from exception
+
async def async_update(self) -> None:
"""Update entity from homee."""
homee = self._entry.runtime_data
@@ -129,14 +144,9 @@ class HomeeNodeEntity(Entity):
return None
- def has_attribute(self, attribute_type: AttributeType) -> bool:
- """Check if an attribute of the given type exists."""
- if self._node.attribute_map is None:
- return False
-
- return attribute_type in self._node.attribute_map
-
- async def async_set_value(self, attribute: HomeeAttribute, value: float) -> None:
+ async def async_set_homee_value(
+ self, attribute: HomeeAttribute, value: float
+ ) -> None:
"""Set an attribute value on the homee node."""
homee = self._entry.runtime_data
try:
diff --git a/homeassistant/components/homee/icons.json b/homeassistant/components/homee/icons.json
index 3b1ee17b89c..b4ad8871568 100644
--- a/homeassistant/components/homee/icons.json
+++ b/homeassistant/components/homee/icons.json
@@ -1,12 +1,26 @@
{
"entity": {
"sensor": {
+ "brightness": {
+ "default": "mdi:brightness-5"
+ },
+ "brightness_instance": {
+ "default": "mdi:brightness-5"
+ },
"link_quality": {
"default": "mdi:signal"
},
"window_position": {
"default": "mdi:window-closed"
}
+ },
+ "switch": {
+ "watchdog": {
+ "default": "mdi:dog"
+ },
+ "manual_operation": {
+ "default": "mdi:hand-back-left"
+ }
}
}
}
diff --git a/homeassistant/components/homee/light.py b/homeassistant/components/homee/light.py
new file mode 100644
index 00000000000..9c66764760e
--- /dev/null
+++ b/homeassistant/components/homee/light.py
@@ -0,0 +1,217 @@
+"""The Homee light platform."""
+
+from typing import Any
+
+from pyHomee.const import AttributeType
+from pyHomee.model import HomeeAttribute, HomeeNode
+
+from homeassistant.components.light import (
+ ATTR_BRIGHTNESS,
+ ATTR_COLOR_TEMP_KELVIN,
+ ATTR_HS_COLOR,
+ ColorMode,
+ LightEntity,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.util.color import (
+ brightness_to_value,
+ color_hs_to_RGB,
+ color_RGB_to_hs,
+ value_to_brightness,
+)
+
+from . import HomeeConfigEntry
+from .const import LIGHT_PROFILES
+from .entity import HomeeNodeEntity
+
+LIGHT_ATTRIBUTES = [
+ AttributeType.COLOR,
+ AttributeType.COLOR_MODE,
+ AttributeType.COLOR_TEMPERATURE,
+ AttributeType.DIMMING_LEVEL,
+]
+
+PARALLEL_UPDATES = 0
+
+
+def is_light_node(node: HomeeNode) -> bool:
+ """Determine if a node is controllable as a homee light based on its profile and attributes."""
+ assert node.attribute_map is not None
+ return node.profile in LIGHT_PROFILES and AttributeType.ON_OFF in node.attribute_map
+
+
+def get_color_mode(supported_modes: set[ColorMode]) -> ColorMode:
+ """Determine the color mode from the supported modes."""
+ if ColorMode.HS in supported_modes:
+ return ColorMode.HS
+ if ColorMode.COLOR_TEMP in supported_modes:
+ return ColorMode.COLOR_TEMP
+ if ColorMode.BRIGHTNESS in supported_modes:
+ return ColorMode.BRIGHTNESS
+
+ return ColorMode.ONOFF
+
+
+def get_light_attribute_sets(
+ node: HomeeNode,
+) -> list[dict[AttributeType, HomeeAttribute]]:
+ """Return the lights with their attributes as found in the node."""
+ lights: list[dict[AttributeType, HomeeAttribute]] = []
+ on_off_attributes = [
+ i for i in node.attributes if i.type == AttributeType.ON_OFF and i.editable
+ ]
+ for a in on_off_attributes:
+ attribute_dict: dict[AttributeType, HomeeAttribute] = {a.type: a}
+ for attribute in node.attributes:
+ if attribute.instance == a.instance and attribute.type in LIGHT_ATTRIBUTES:
+ attribute_dict[attribute.type] = attribute
+ lights.append(attribute_dict)
+
+ return lights
+
+
+def rgb_list_to_decimal(color: tuple[int, int, int]) -> int:
+ """Convert an rgb color from list to decimal representation."""
+ return int(int(color[0]) << 16) + (int(color[1]) << 8) + (int(color[2]))
+
+
+def decimal_to_rgb_list(color: float) -> list[int]:
+ """Convert an rgb color from decimal to list representation."""
+ return [
+ (int(color) & 0xFF0000) >> 16,
+ (int(color) & 0x00FF00) >> 8,
+ (int(color) & 0x0000FF),
+ ]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: HomeeConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add the Homee platform for the light entity."""
+
+ async_add_entities(
+ HomeeLight(node, light, config_entry)
+ for node in config_entry.runtime_data.nodes
+ for light in get_light_attribute_sets(node)
+ if is_light_node(node)
+ )
+
+
+class HomeeLight(HomeeNodeEntity, LightEntity):
+ """Representation of a Homee light."""
+
+ def __init__(
+ self,
+ node: HomeeNode,
+ light: dict[AttributeType, HomeeAttribute],
+ entry: HomeeConfigEntry,
+ ) -> None:
+ """Initialize a Homee light."""
+ super().__init__(node, entry)
+
+ self._on_off_attr: HomeeAttribute = light[AttributeType.ON_OFF]
+ self._dimmer_attr: HomeeAttribute | None = light.get(
+ AttributeType.DIMMING_LEVEL
+ )
+ self._col_attr: HomeeAttribute | None = light.get(AttributeType.COLOR)
+ self._temp_attr: HomeeAttribute | None = light.get(
+ AttributeType.COLOR_TEMPERATURE
+ )
+ self._mode_attr: HomeeAttribute | None = light.get(AttributeType.COLOR_MODE)
+
+ self._attr_supported_color_modes = self._get_supported_color_modes()
+ self._attr_color_mode = get_color_mode(self._attr_supported_color_modes)
+
+ if self._temp_attr is not None:
+ self._attr_min_color_temp_kelvin = int(self._temp_attr.minimum)
+ self._attr_max_color_temp_kelvin = int(self._temp_attr.maximum)
+
+ if self._on_off_attr.instance > 0:
+ self._attr_translation_key = "light_instance"
+ self._attr_translation_placeholders = {
+ "instance": str(self._on_off_attr.instance)
+ }
+ else:
+ # If a device has only one light, it will get its name.
+ self._attr_name = None
+ self._attr_unique_id = (
+ f"{entry.runtime_data.settings.uid}-{self._node.id}-{self._on_off_attr.id}"
+ )
+
+ @property
+ def brightness(self) -> int:
+ """Return the brightness of the light."""
+ assert self._dimmer_attr is not None
+ return value_to_brightness(
+ (self._dimmer_attr.minimum + 1, self._dimmer_attr.maximum),
+ self._dimmer_attr.current_value,
+ )
+
+ @property
+ def hs_color(self) -> tuple[float, float] | None:
+ """Return the color of the light."""
+ assert self._col_attr is not None
+ rgb = decimal_to_rgb_list(self._col_attr.current_value)
+ return color_RGB_to_hs(*rgb)
+
+ @property
+ def color_temp_kelvin(self) -> int:
+ """Return the color temperature of the light."""
+ assert self._temp_attr is not None
+ return int(self._temp_attr.current_value)
+
+ @property
+ def is_on(self) -> bool:
+ """Return true if light is on."""
+ return bool(self._on_off_attr.current_value)
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Instruct the light to turn on."""
+ if ATTR_BRIGHTNESS in kwargs and self._dimmer_attr is not None:
+ target_value = round(
+ brightness_to_value(
+ (self._dimmer_attr.minimum, self._dimmer_attr.maximum),
+ kwargs[ATTR_BRIGHTNESS],
+ )
+ )
+ await self.async_set_homee_value(self._dimmer_attr, target_value)
+ else:
+ # If no brightness value is given, just turn on.
+ await self.async_set_homee_value(self._on_off_attr, 1)
+
+ if ATTR_COLOR_TEMP_KELVIN in kwargs and self._temp_attr is not None:
+ await self.async_set_homee_value(
+ self._temp_attr, kwargs[ATTR_COLOR_TEMP_KELVIN]
+ )
+ if ATTR_HS_COLOR in kwargs:
+ color = kwargs[ATTR_HS_COLOR]
+ if self._col_attr is not None:
+ await self.async_set_homee_value(
+ self._col_attr,
+ rgb_list_to_decimal(color_hs_to_RGB(*color)),
+ )
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Instruct the light to turn off."""
+ await self.async_set_homee_value(self._on_off_attr, 0)
+
+ def _get_supported_color_modes(self) -> set[ColorMode]:
+ """Determine the supported color modes from the available attributes."""
+ color_modes: set[ColorMode] = set()
+
+ if self._temp_attr is not None and self._temp_attr.editable:
+ color_modes.add(ColorMode.COLOR_TEMP)
+ if self._col_attr is not None:
+ color_modes.add(ColorMode.HS)
+
+ # If no other color modes are available, set one of those.
+ if len(color_modes) == 0:
+ if self._dimmer_attr is not None:
+ color_modes.add(ColorMode.BRIGHTNESS)
+ else:
+ color_modes.add(ColorMode.ONOFF)
+
+ return color_modes
diff --git a/homeassistant/components/homee/manifest.json b/homeassistant/components/homee/manifest.json
index d85ba25b6e7..3c2a99c30dc 100644
--- a/homeassistant/components/homee/manifest.json
+++ b/homeassistant/components/homee/manifest.json
@@ -8,5 +8,5 @@
"iot_class": "local_push",
"loggers": ["homee"],
"quality_scale": "bronze",
- "requirements": ["pyHomee==1.2.5"]
+ "requirements": ["pyHomee==1.2.8"]
}
diff --git a/homeassistant/components/homee/number.py b/homeassistant/components/homee/number.py
new file mode 100644
index 00000000000..5f76b826fcf
--- /dev/null
+++ b/homeassistant/components/homee/number.py
@@ -0,0 +1,132 @@
+"""The Homee number platform."""
+
+from pyHomee.const import AttributeType
+from pyHomee.model import HomeeAttribute
+
+from homeassistant.components.number import (
+ NumberDeviceClass,
+ NumberEntity,
+ NumberEntityDescription,
+)
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import HomeeConfigEntry
+from .const import HOMEE_UNIT_TO_HA_UNIT
+from .entity import HomeeEntity
+
+PARALLEL_UPDATES = 0
+
+NUMBER_DESCRIPTIONS = {
+ AttributeType.DOWN_POSITION: NumberEntityDescription(
+ key="down_position",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.DOWN_SLAT_POSITION: NumberEntityDescription(
+ key="down_slat_position",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.DOWN_TIME: NumberEntityDescription(
+ key="down_time",
+ device_class=NumberDeviceClass.DURATION,
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.ENDPOSITION_CONFIGURATION: NumberEntityDescription(
+ key="endposition_configuration",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.MOTION_ALARM_CANCELATION_DELAY: NumberEntityDescription(
+ key="motion_alarm_cancelation_delay",
+ device_class=NumberDeviceClass.DURATION,
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.OPEN_WINDOW_DETECTION_SENSIBILITY: NumberEntityDescription(
+ key="open_window_detection_sensibility",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.POLLING_INTERVAL: NumberEntityDescription(
+ key="polling_interval",
+ device_class=NumberDeviceClass.DURATION,
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.SHUTTER_SLAT_TIME: NumberEntityDescription(
+ key="shutter_slat_time",
+ device_class=NumberDeviceClass.DURATION,
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.SLAT_MAX_ANGLE: NumberEntityDescription(
+ key="slat_max_angle",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.SLAT_MIN_ANGLE: NumberEntityDescription(
+ key="slat_min_angle",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.SLAT_STEPS: NumberEntityDescription(
+ key="slat_steps",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.TEMPERATURE_OFFSET: NumberEntityDescription(
+ key="temperature_offset",
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.UP_TIME: NumberEntityDescription(
+ key="up_time",
+ device_class=NumberDeviceClass.DURATION,
+ entity_category=EntityCategory.CONFIG,
+ ),
+ AttributeType.WAKE_UP_INTERVAL: NumberEntityDescription(
+ key="wake_up_interval",
+ device_class=NumberDeviceClass.DURATION,
+ entity_category=EntityCategory.CONFIG,
+ ),
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: HomeeConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add the Homee platform for the number component."""
+
+ async_add_entities(
+ HomeeNumber(attribute, config_entry, NUMBER_DESCRIPTIONS[attribute.type])
+ for node in config_entry.runtime_data.nodes
+ for attribute in node.attributes
+ if attribute.type in NUMBER_DESCRIPTIONS and attribute.data != "fixed_value"
+ )
+
+
+class HomeeNumber(HomeeEntity, NumberEntity):
+ """Representation of a Homee number."""
+
+ def __init__(
+ self,
+ attribute: HomeeAttribute,
+ entry: HomeeConfigEntry,
+ description: NumberEntityDescription,
+ ) -> None:
+ """Initialize a Homee number entity."""
+ super().__init__(attribute, entry)
+ self.entity_description = description
+ self._attr_translation_key = description.key
+ self._attr_native_unit_of_measurement = HOMEE_UNIT_TO_HA_UNIT[attribute.unit]
+ self._attr_native_min_value = attribute.minimum
+ self._attr_native_max_value = attribute.maximum
+ self._attr_native_step = attribute.step_value
+
+ @property
+ def available(self) -> bool:
+ """Return the availability of the entity."""
+ return super().available and self._attribute.editable
+
+ @property
+ def native_value(self) -> int:
+ """Return the native value of the number."""
+ return int(self._attribute.current_value)
+
+ async def async_set_native_value(self, value: float) -> None:
+ """Set the selected value."""
+ await self.async_set_homee_value(value)
diff --git a/homeassistant/components/homee/quality_scale.yaml b/homeassistant/components/homee/quality_scale.yaml
index ff99d177018..906218cf823 100644
--- a/homeassistant/components/homee/quality_scale.yaml
+++ b/homeassistant/components/homee/quality_scale.yaml
@@ -35,7 +35,7 @@ rules:
entity-unavailable: done
integration-owner: done
log-when-unavailable: done
- parallel-updates: todo
+ parallel-updates: done
reauthentication-flow: todo
test-coverage: todo
diff --git a/homeassistant/components/homee/select.py b/homeassistant/components/homee/select.py
new file mode 100644
index 00000000000..70c7972bbda
--- /dev/null
+++ b/homeassistant/components/homee/select.py
@@ -0,0 +1,63 @@
+"""The Homee select platform."""
+
+from pyHomee.const import AttributeType
+from pyHomee.model import HomeeAttribute
+
+from homeassistant.components.select import SelectEntity, SelectEntityDescription
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import HomeeConfigEntry
+from .entity import HomeeEntity
+
+PARALLEL_UPDATES = 0
+
+SELECT_DESCRIPTIONS: dict[AttributeType, SelectEntityDescription] = {
+ AttributeType.REPEATER_MODE: SelectEntityDescription(
+ key="repeater_mode",
+ options=["off", "level1", "level2"],
+ entity_category=EntityCategory.CONFIG,
+ ),
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: HomeeConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add the Homee platform for the select component."""
+
+ async_add_entities(
+ HomeeSelect(attribute, config_entry, SELECT_DESCRIPTIONS[attribute.type])
+ for node in config_entry.runtime_data.nodes
+ for attribute in node.attributes
+ if attribute.type in SELECT_DESCRIPTIONS and attribute.editable
+ )
+
+
+class HomeeSelect(HomeeEntity, SelectEntity):
+ """Representation of a Homee select entity."""
+
+ def __init__(
+ self,
+ attribute: HomeeAttribute,
+ entry: HomeeConfigEntry,
+ description: SelectEntityDescription,
+ ) -> None:
+ """Initialize a Homee select entity."""
+ super().__init__(attribute, entry)
+ self.entity_description = description
+ assert description.options is not None
+ self._attr_options = description.options
+ self._attr_translation_key = description.key
+
+ @property
+ def current_option(self) -> str:
+ """Return the current selected option."""
+ return self.options[int(self._attribute.current_value)]
+
+ async def async_select_option(self, option: str) -> None:
+ """Change the selected option."""
+ await self.async_set_homee_value(self.options.index(option))
diff --git a/homeassistant/components/homee/sensor.py b/homeassistant/components/homee/sensor.py
index da01c2aa5b9..e65b73b4a67 100644
--- a/homeassistant/components/homee/sensor.py
+++ b/homeassistant/components/homee/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import HomeeConfigEntry
from .const import (
@@ -27,6 +27,8 @@ from .const import (
from .entity import HomeeEntity, HomeeNodeEntity
from .helpers import get_name_for_enum
+PARALLEL_UPDATES = 0
+
def get_open_close_value(attribute: HomeeAttribute) -> str | None:
"""Return the open/close value."""
@@ -40,10 +42,22 @@ def get_window_value(attribute: HomeeAttribute) -> str | None:
return vals.get(attribute.current_value)
+def get_brightness_device_class(
+ attribute: HomeeAttribute, device_class: SensorDeviceClass | None
+) -> SensorDeviceClass | None:
+ """Return the device class for a brightness sensor."""
+ if attribute.unit == "%":
+ return None
+ return device_class
+
+
@dataclass(frozen=True, kw_only=True)
class HomeeSensorEntityDescription(SensorEntityDescription):
"""A class that describes Homee sensor entities."""
+ device_class_fn: Callable[
+ [HomeeAttribute, SensorDeviceClass | None], SensorDeviceClass | None
+ ] = lambda attribute, device_class: device_class
value_fn: Callable[[HomeeAttribute], str | float | None] = (
lambda value: value.current_value
)
@@ -67,6 +81,7 @@ SENSOR_DESCRIPTIONS: dict[AttributeType, HomeeSensorEntityDescription] = {
AttributeType.BRIGHTNESS: HomeeSensorEntityDescription(
key="brightness",
device_class=SensorDeviceClass.ILLUMINANCE,
+ device_class_fn=get_brightness_device_class,
state_class=SensorStateClass.MEASUREMENT,
value_fn=(
lambda attribute: attribute.current_value * 1000
@@ -157,7 +172,7 @@ SENSOR_DESCRIPTIONS: dict[AttributeType, HomeeSensorEntityDescription] = {
AttributeType.RAIN_FALL_TODAY: HomeeSensorEntityDescription(
key="rainfall_day",
device_class=SensorDeviceClass.PRECIPITATION,
- state_class=SensorStateClass.MEASUREMENT,
+ state_class=SensorStateClass.TOTAL_INCREASING,
),
AttributeType.RELATIVE_HUMIDITY: HomeeSensorEntityDescription(
key="humidity",
@@ -262,7 +277,7 @@ NODE_SENSOR_DESCRIPTIONS: tuple[HomeeNodeSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HomeeConfigEntry,
- async_add_devices: AddEntitiesCallback,
+ async_add_devices: AddConfigEntryEntitiesCallback,
) -> None:
"""Add the homee platform for the sensor components."""
@@ -303,6 +318,9 @@ class HomeeSensor(HomeeEntity, SensorEntity):
if attribute.instance > 0:
self._attr_translation_key = f"{self._attr_translation_key}_instance"
self._attr_translation_placeholders = {"instance": str(attribute.instance)}
+ self._attr_device_class = description.device_class_fn(
+ attribute, description.device_class
+ )
@property
def native_value(self) -> float | str | None:
diff --git a/homeassistant/components/homee/strings.json b/homeassistant/components/homee/strings.json
index 025d8df21d6..da8357d16bc 100644
--- a/homeassistant/components/homee/strings.json
+++ b/homeassistant/components/homee/strings.json
@@ -26,7 +26,174 @@
}
},
"entity": {
+ "binary_sensor": {
+ "blackout_alarm": {
+ "name": "Blackout"
+ },
+ "carbon_dioxide": {
+ "name": "Carbon dioxide"
+ },
+ "flood": {
+ "name": "Flood"
+ },
+ "high_temperature": {
+ "name": "High temperature"
+ },
+ "leak_alarm": {
+ "name": "Leak"
+ },
+ "load_alarm": {
+ "name": "Load",
+ "state": {
+ "off": "Normal",
+ "on": "Overload"
+ }
+ },
+ "low_temperature": {
+ "name": "Low temperature"
+ },
+ "malfunction": {
+ "name": "Malfunction"
+ },
+ "maximum": {
+ "name": "Maximum level"
+ },
+ "minimum": {
+ "name": "Minimum level"
+ },
+ "motor_blocked": {
+ "name": "Motor blocked"
+ },
+ "overcurrent": {
+ "name": "Overcurrent"
+ },
+ "overload": {
+ "name": "Overload"
+ },
+ "rain": {
+ "name": "Rain"
+ },
+ "replace_filter": {
+ "name": "Replace filter",
+ "state": {
+ "on": "Replace"
+ }
+ },
+ "storage": {
+ "name": "Storage",
+ "state": {
+ "off": "Space available",
+ "on": "Storage full"
+ }
+ },
+ "surge": {
+ "name": "Surge"
+ },
+ "voltage_drop": {
+ "name": "Voltage drop"
+ },
+ "water": {
+ "name": "Water"
+ }
+ },
+ "button": {
+ "automatic_mode": {
+ "name": "Automatic mode"
+ },
+ "briefly_open": {
+ "name": "Briefly open"
+ },
+ "identification_mode": {
+ "name": "Identification mode"
+ },
+ "impulse_instance": {
+ "name": "Impulse {instance}"
+ },
+ "light": {
+ "name": "Light"
+ },
+ "light_instance": {
+ "name": "Light {instance}"
+ },
+ "open_partial": {
+ "name": "Open partially"
+ },
+ "permanently_open": {
+ "name": "Open permanently"
+ },
+ "reset_meter": {
+ "name": "Reset meter"
+ },
+ "reset_meter_instance": {
+ "name": "Reset meter {instance}"
+ },
+ "ventilate": {
+ "name": "Ventilate"
+ }
+ },
+ "light": {
+ "light_instance": {
+ "name": "Light {instance}"
+ }
+ },
+ "number": {
+ "down_position": {
+ "name": "Down position"
+ },
+ "down_slat_position": {
+ "name": "Down slat position"
+ },
+ "down_time": {
+ "name": "Down-movement duration"
+ },
+ "endposition_configuration": {
+ "name": "End position"
+ },
+ "motion_alarm_cancelation_delay": {
+ "name": "Motion alarm delay"
+ },
+ "open_window_detection_sensibility": {
+ "name": "Window open sensibility"
+ },
+ "polling_interval": {
+ "name": "Polling interval"
+ },
+ "shutter_slat_time": {
+ "name": "Slat turn duration"
+ },
+ "slat_max_angle": {
+ "name": "Maximum slat angle"
+ },
+ "slat_min_angle": {
+ "name": "Minimum slat angle"
+ },
+ "slat_steps": {
+ "name": "Slat steps"
+ },
+ "temperature_offset": {
+ "name": "Temperature offset"
+ },
+ "up_time": {
+ "name": "Up-movement duration"
+ },
+ "wake_up_interval": {
+ "name": "Wake-up interval"
+ }
+ },
+ "select": {
+ "repeater_mode": {
+ "name": "Repeater mode",
+ "state": {
+ "off": "[%key:common::state::off%]",
+ "level1": "Level 1",
+ "level2": "Level 2"
+ }
+ }
+ },
"sensor": {
+ "brightness": {
+ "name": "Illuminance"
+ },
"brightness_instance": {
"name": "Illuminance {instance}"
},
@@ -151,11 +318,30 @@
"tilted": "Tilted"
}
}
+ },
+ "switch": {
+ "external_binary_input": {
+ "name": "Child lock"
+ },
+ "manual_operation": {
+ "name": "Manual operation"
+ },
+ "on_off_instance": {
+ "name": "Switch {instance}"
+ },
+ "watchdog": {
+ "name": "Watchdog"
+ }
+ },
+ "valve": {
+ "valve_position": {
+ "name": "Valve position"
+ }
}
},
"exceptions": {
"connection_closed": {
- "message": "Could not connect to Homee while setting attribute"
+ "message": "Could not connect to Homee while setting attribute."
}
}
}
diff --git a/homeassistant/components/homee/switch.py b/homeassistant/components/homee/switch.py
new file mode 100644
index 00000000000..041b96963f1
--- /dev/null
+++ b/homeassistant/components/homee/switch.py
@@ -0,0 +1,129 @@
+"""The homee switch platform."""
+
+from collections.abc import Callable
+from dataclasses import dataclass
+from typing import Any
+
+from pyHomee.const import AttributeType, NodeProfile
+from pyHomee.model import HomeeAttribute
+
+from homeassistant.components.switch import (
+ SwitchDeviceClass,
+ SwitchEntity,
+ SwitchEntityDescription,
+)
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import HomeeConfigEntry
+from .const import CLIMATE_PROFILES, LIGHT_PROFILES
+from .entity import HomeeEntity
+
+PARALLEL_UPDATES = 0
+
+
+def get_device_class(
+ attribute: HomeeAttribute, config_entry: HomeeConfigEntry
+) -> SwitchDeviceClass:
+ """Check device class of Switch according to node profile."""
+ node = config_entry.runtime_data.get_node_by_id(attribute.node_id)
+ if node.profile in [
+ NodeProfile.ON_OFF_PLUG,
+ NodeProfile.METERING_PLUG,
+ NodeProfile.DOUBLE_ON_OFF_PLUG,
+ NodeProfile.IMPULSE_PLUG,
+ ]:
+ return SwitchDeviceClass.OUTLET
+
+ return SwitchDeviceClass.SWITCH
+
+
+@dataclass(frozen=True, kw_only=True)
+class HomeeSwitchEntityDescription(SwitchEntityDescription):
+ """A class that describes Homee switch entity."""
+
+ device_class_fn: Callable[[HomeeAttribute, HomeeConfigEntry], SwitchDeviceClass] = (
+ lambda attribute, entry: SwitchDeviceClass.SWITCH
+ )
+
+
+SWITCH_DESCRIPTIONS: dict[AttributeType, HomeeSwitchEntityDescription] = {
+ AttributeType.EXTERNAL_BINARY_INPUT: HomeeSwitchEntityDescription(
+ key="external_binary_input", entity_category=EntityCategory.CONFIG
+ ),
+ AttributeType.MANUAL_OPERATION: HomeeSwitchEntityDescription(
+ key="manual_operation"
+ ),
+ AttributeType.ON_OFF: HomeeSwitchEntityDescription(
+ key="on_off", device_class_fn=get_device_class, name=None
+ ),
+ AttributeType.WATCHDOG_ON_OFF: HomeeSwitchEntityDescription(
+ key="watchdog", entity_category=EntityCategory.CONFIG
+ ),
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: HomeeConfigEntry,
+ async_add_devices: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the switch platform for the Homee component."""
+
+ for node in config_entry.runtime_data.nodes:
+ async_add_devices(
+ HomeeSwitch(attribute, config_entry, SWITCH_DESCRIPTIONS[attribute.type])
+ for attribute in node.attributes
+ if (attribute.type in SWITCH_DESCRIPTIONS and attribute.editable)
+ and not (
+ attribute.type == AttributeType.ON_OFF
+ and node.profile in LIGHT_PROFILES
+ )
+ and not (
+ attribute.type == AttributeType.MANUAL_OPERATION
+ and node.profile in CLIMATE_PROFILES
+ )
+ )
+
+
+class HomeeSwitch(HomeeEntity, SwitchEntity):
+ """Representation of a Homee switch."""
+
+ entity_description: HomeeSwitchEntityDescription
+
+ def __init__(
+ self,
+ attribute: HomeeAttribute,
+ entry: HomeeConfigEntry,
+ description: HomeeSwitchEntityDescription,
+ ) -> None:
+ """Initialize a Homee switch entity."""
+ super().__init__(attribute, entry)
+ self.entity_description = description
+ if attribute.instance == 0:
+ if attribute.type == AttributeType.ON_OFF:
+ self._attr_name = None
+ else:
+ self._attr_translation_key = description.key
+ else:
+ self._attr_translation_key = f"{description.key}_instance"
+ self._attr_translation_placeholders = {"instance": str(attribute.instance)}
+
+ @property
+ def is_on(self) -> bool:
+ """Return True if entity is on."""
+ return bool(self._attribute.current_value)
+
+ @property
+ def device_class(self) -> SwitchDeviceClass:
+ """Return the device class of the switch."""
+ return self.entity_description.device_class_fn(self._attribute, self._entry)
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the switch on."""
+ await self.async_set_homee_value(1)
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the switch off."""
+ await self.async_set_homee_value(0)
diff --git a/homeassistant/components/homee/valve.py b/homeassistant/components/homee/valve.py
new file mode 100644
index 00000000000..995716d7ef8
--- /dev/null
+++ b/homeassistant/components/homee/valve.py
@@ -0,0 +1,83 @@
+"""The Homee valve platform."""
+
+from pyHomee.const import AttributeType
+from pyHomee.model import HomeeAttribute
+
+from homeassistant.components.valve import (
+ ValveDeviceClass,
+ ValveEntity,
+ ValveEntityDescription,
+ ValveEntityFeature,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import HomeeConfigEntry
+from .entity import HomeeEntity
+
+PARALLEL_UPDATES = 0
+
+VALVE_DESCRIPTIONS = {
+ AttributeType.CURRENT_VALVE_POSITION: ValveEntityDescription(
+ key="valve_position",
+ device_class=ValveDeviceClass.WATER,
+ )
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: HomeeConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add the Homee platform for the valve component."""
+
+ async_add_entities(
+ HomeeValve(attribute, config_entry, VALVE_DESCRIPTIONS[attribute.type])
+ for node in config_entry.runtime_data.nodes
+ for attribute in node.attributes
+ if attribute.type in VALVE_DESCRIPTIONS
+ )
+
+
+class HomeeValve(HomeeEntity, ValveEntity):
+ """Representation of a Homee valve."""
+
+ _attr_reports_position = True
+
+ def __init__(
+ self,
+ attribute: HomeeAttribute,
+ entry: HomeeConfigEntry,
+ description: ValveEntityDescription,
+ ) -> None:
+ """Initialize a Homee valve entity."""
+ super().__init__(attribute, entry)
+ self.entity_description = description
+ self._attr_translation_key = description.key
+
+ @property
+ def supported_features(self) -> ValveEntityFeature:
+ """Return the supported features."""
+ if self._attribute.editable:
+ return ValveEntityFeature.SET_POSITION
+ return ValveEntityFeature(0)
+
+ @property
+ def current_valve_position(self) -> int | None:
+ """Return the current valve position."""
+ return int(self._attribute.current_value)
+
+ @property
+ def is_closing(self) -> bool:
+ """Return if the valve is closing."""
+ return self._attribute.target_value < self._attribute.current_value
+
+ @property
+ def is_opening(self) -> bool:
+ """Return if the valve is opening."""
+ return self._attribute.target_value > self._attribute.current_value
+
+ async def async_set_valve_position(self, position: int) -> None:
+ """Move the valve to a specific position."""
+ await self.async_set_homee_value(position)
diff --git a/homeassistant/components/homekit/__init__.py b/homeassistant/components/homekit/__init__.py
index 97fb17d7db5..9bd5711832c 100644
--- a/homeassistant/components/homekit/__init__.py
+++ b/homeassistant/components/homekit/__init__.py
@@ -221,6 +221,34 @@ UNPAIR_SERVICE_SCHEMA = vol.All(
)
+@callback
+def _async_update_entries_from_yaml(
+ hass: HomeAssistant, config: ConfigType, start_import_flow: bool
+) -> None:
+ current_entries = hass.config_entries.async_entries(DOMAIN)
+ entries_by_name, entries_by_port = _async_get_imported_entries_indices(
+ current_entries
+ )
+ hk_config: list[dict[str, Any]] = config[DOMAIN]
+
+ for index, conf in enumerate(hk_config):
+ if _async_update_config_entry_from_yaml(
+ hass, entries_by_name, entries_by_port, conf
+ ):
+ continue
+
+ if start_import_flow:
+ conf[CONF_ENTRY_INDEX] = index
+ hass.async_create_task(
+ hass.config_entries.flow.async_init(
+ DOMAIN,
+ context={"source": SOURCE_IMPORT},
+ data=conf,
+ ),
+ eager_start=True,
+ )
+
+
def _async_all_homekit_instances(hass: HomeAssistant) -> list[HomeKit]:
"""All active HomeKit instances."""
hk_data: HomeKitEntryData | None
@@ -258,31 +286,10 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
await hass.async_add_executor_job(get_loader)
_async_register_events_and_services(hass)
-
if DOMAIN not in config:
return True
- current_entries = hass.config_entries.async_entries(DOMAIN)
- entries_by_name, entries_by_port = _async_get_imported_entries_indices(
- current_entries
- )
-
- for index, conf in enumerate(config[DOMAIN]):
- if _async_update_config_entry_from_yaml(
- hass, entries_by_name, entries_by_port, conf
- ):
- continue
-
- conf[CONF_ENTRY_INDEX] = index
- hass.async_create_task(
- hass.config_entries.flow.async_init(
- DOMAIN,
- context={"source": SOURCE_IMPORT},
- data=conf,
- ),
- eager_start=True,
- )
-
+ _async_update_entries_from_yaml(hass, config, start_import_flow=True)
return True
@@ -326,13 +333,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: HomeKitConfigEntry) -> b
conf = entry.data
options = entry.options
- name = conf[CONF_NAME]
- port = conf[CONF_PORT]
- _LOGGER.debug("Begin setup HomeKit for %s", name)
-
+ name: str = conf[CONF_NAME]
+ port: int = conf[CONF_PORT]
# ip_address and advertise_ip are yaml only
- ip_address = conf.get(CONF_IP_ADDRESS, _DEFAULT_BIND)
- advertise_ips: list[str] = conf.get(
+ ip_address: str | list[str] | None = conf.get(CONF_IP_ADDRESS, _DEFAULT_BIND)
+ advertise_ips: list[str]
+ advertise_ips = conf.get(
CONF_ADVERTISE_IP
) or await network.async_get_announce_addresses(hass)
@@ -344,13 +350,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: HomeKitConfigEntry) -> b
# with users who have not migrated yet we do not do exclude
# these entities by default as we cannot migrate automatically
# since it requires a re-pairing.
- exclude_accessory_mode = conf.get(
+ exclude_accessory_mode: bool = conf.get(
CONF_EXCLUDE_ACCESSORY_MODE, DEFAULT_EXCLUDE_ACCESSORY_MODE
)
- homekit_mode = options.get(CONF_HOMEKIT_MODE, DEFAULT_HOMEKIT_MODE)
- entity_config = options.get(CONF_ENTITY_CONFIG, {}).copy()
- entity_filter = FILTER_SCHEMA(options.get(CONF_FILTER, {}))
- devices = options.get(CONF_DEVICES, [])
+ homekit_mode: str = options.get(CONF_HOMEKIT_MODE, DEFAULT_HOMEKIT_MODE)
+ entity_config: dict[str, Any] = options.get(CONF_ENTITY_CONFIG, {}).copy()
+ entity_filter: EntityFilter = FILTER_SCHEMA(options.get(CONF_FILTER, {}))
+ devices: list[str] = options.get(CONF_DEVICES, [])
homekit = HomeKit(
hass,
@@ -500,26 +506,15 @@ def _async_register_events_and_services(hass: HomeAssistant) -> None:
async def _handle_homekit_reload(service: ServiceCall) -> None:
"""Handle start HomeKit service call."""
config = await async_integration_yaml_config(hass, DOMAIN)
-
if not config or DOMAIN not in config:
return
-
- current_entries = hass.config_entries.async_entries(DOMAIN)
- entries_by_name, entries_by_port = _async_get_imported_entries_indices(
- current_entries
- )
-
- for conf in config[DOMAIN]:
- _async_update_config_entry_from_yaml(
- hass, entries_by_name, entries_by_port, conf
+ _async_update_entries_from_yaml(hass, config, start_import_flow=False)
+ await asyncio.gather(
+ *(
+ create_eager_task(hass.config_entries.async_reload(entry.entry_id))
+ for entry in hass.config_entries.async_entries(DOMAIN)
)
-
- reload_tasks = [
- create_eager_task(hass.config_entries.async_reload(entry.entry_id))
- for entry in current_entries
- ]
-
- await asyncio.gather(*reload_tasks)
+ )
async_register_admin_service(
hass,
@@ -537,7 +532,7 @@ class HomeKit:
hass: HomeAssistant,
name: str,
port: int,
- ip_address: str | None,
+ ip_address: list[str] | str | None,
entity_filter: EntityFilter,
exclude_accessory_mode: bool,
entity_config: dict[str, Any],
diff --git a/homeassistant/components/homekit/accessories.py b/homeassistant/components/homekit/accessories.py
index 8d10387e239..0d810d6986d 100644
--- a/homeassistant/components/homekit/accessories.py
+++ b/homeassistant/components/homekit/accessories.py
@@ -15,6 +15,7 @@ from pyhap.service import Service
from pyhap.util import callback as pyhap_callback
from homeassistant.components.cover import CoverDeviceClass, CoverEntityFeature
+from homeassistant.components.lawn_mower import LawnMowerEntityFeature
from homeassistant.components.media_player import MediaPlayerDeviceClass
from homeassistant.components.remote import RemoteEntityFeature
from homeassistant.components.sensor import SensorDeviceClass
@@ -250,6 +251,13 @@ def get_accessory( # noqa: C901
elif state.domain == "vacuum":
a_type = "Vacuum"
+ elif (
+ state.domain == "lawn_mower"
+ and features & LawnMowerEntityFeature.DOCK
+ and features & LawnMowerEntityFeature.START_MOWING
+ ):
+ a_type = "LawnMower"
+
elif state.domain == "remote" and features & RemoteEntityFeature.ACTIVITY:
a_type = "ActivityRemote"
diff --git a/homeassistant/components/homekit/config_flow.py b/homeassistant/components/homekit/config_flow.py
index 53db7774821..0ef2e8563bc 100644
--- a/homeassistant/components/homekit/config_flow.py
+++ b/homeassistant/components/homekit/config_flow.py
@@ -106,6 +106,7 @@ SUPPORTED_DOMAINS = [
"sensor",
"switch",
"vacuum",
+ "lawn_mower",
"water_heater",
VALVE_DOMAIN,
]
@@ -123,6 +124,7 @@ DEFAULT_DOMAINS = [
REMOTE_DOMAIN,
"switch",
"vacuum",
+ "lawn_mower",
"water_heater",
]
diff --git a/homeassistant/components/homekit/manifest.json b/homeassistant/components/homekit/manifest.json
index d7ea293b5dc..4ae2e43dfb2 100644
--- a/homeassistant/components/homekit/manifest.json
+++ b/homeassistant/components/homekit/manifest.json
@@ -10,7 +10,7 @@
"loggers": ["pyhap"],
"requirements": [
"HAP-python==4.9.2",
- "fnv-hash-fast==1.2.2",
+ "fnv-hash-fast==1.4.0",
"PyQRCode==1.2.1",
"base36==0.1.1"
],
diff --git a/homeassistant/components/homekit/strings.json b/homeassistant/components/homekit/strings.json
index 92b836d5ec6..dcdf6892dc2 100644
--- a/homeassistant/components/homekit/strings.json
+++ b/homeassistant/components/homekit/strings.json
@@ -2,13 +2,13 @@
"options": {
"step": {
"yaml": {
- "title": "Adjust HomeKit Options",
+ "title": "Adjust HomeKit options",
"description": "This entry is controlled via YAML"
},
"init": {
"data": {
- "mode": "HomeKit Mode",
- "include_exclude_mode": "Inclusion Mode",
+ "mode": "HomeKit mode",
+ "include_exclude_mode": "Inclusion mode",
"domains": "[%key:component::homekit::config::step::user::data::include_domains%]"
},
"description": "HomeKit can be configured expose a bridge or a single accessory. In accessory mode, only a single entity can be used. Accessory mode is required for media players with the TV or RECEIVER device class to function properly. Entities in the \u201cDomains to include\u201d will be included to HomeKit. You will be able to select which entities to include or exclude from this list on the next screen.",
@@ -40,14 +40,14 @@
"camera_audio": "Cameras that support audio"
},
"description": "Check all cameras that support native H.264 streams. If the camera does not output a H.264 stream, the system will transcode the video to H.264 for HomeKit. Transcoding requires a performant CPU and is unlikely to work on single board computers.",
- "title": "Camera Configuration"
+ "title": "Camera configuration"
},
"advanced": {
"data": {
"devices": "Devices (Triggers)"
},
"description": "Programmable switches are created for each selected device. When a device trigger fires, HomeKit can be configured to run an automation or scene.",
- "title": "Advanced Configuration"
+ "title": "Advanced configuration"
}
}
},
@@ -72,7 +72,7 @@
"services": {
"reload": {
"name": "[%key:common::action::reload%]",
- "description": "Reloads homekit and re-process YAML-configuration."
+ "description": "Reloads HomeKit and re-processes the YAML-configuration."
},
"reset_accessory": {
"name": "Reset accessory",
diff --git a/homeassistant/components/homekit/type_switches.py b/homeassistant/components/homekit/type_switches.py
index 0482a5956ac..8c6fc1ed672 100644
--- a/homeassistant/components/homekit/type_switches.py
+++ b/homeassistant/components/homekit/type_switches.py
@@ -16,6 +16,12 @@ from pyhap.const import (
from homeassistant.components import button, input_button
from homeassistant.components.input_select import ATTR_OPTIONS, SERVICE_SELECT_OPTION
+from homeassistant.components.lawn_mower import (
+ DOMAIN as LAWN_MOWER_DOMAIN,
+ SERVICE_DOCK,
+ SERVICE_START_MOWING,
+ LawnMowerActivity,
+)
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.components.vacuum import (
DOMAIN as VACUUM_DOMAIN,
@@ -218,6 +224,29 @@ class Vacuum(Switch):
self.char_on.set_value(current_state)
+@TYPES.register("LawnMower")
+class LawnMower(Switch):
+ """Generate a Switch accessory."""
+
+ def set_state(self, value: bool) -> None:
+ """Move switch state to value if call came from HomeKit."""
+ _LOGGER.debug("%s: Set switch state to %s", self.entity_id, value)
+ state = self.hass.states.get(self.entity_id)
+ assert state
+
+ service = SERVICE_START_MOWING if value else SERVICE_DOCK
+ self.async_call_service(
+ LAWN_MOWER_DOMAIN, service, {ATTR_ENTITY_ID: self.entity_id}
+ )
+
+ @callback
+ def async_update_state(self, new_state: State) -> None:
+ """Update switch state after state changed."""
+ current_state = new_state.state in (LawnMowerActivity.MOWING, STATE_ON)
+ _LOGGER.debug("%s: Set current state to %s", self.entity_id, current_state)
+ self.char_on.set_value(current_state)
+
+
class ValveBase(HomeAccessory):
"""Valve base class."""
diff --git a/homeassistant/components/homekit_controller/alarm_control_panel.py b/homeassistant/components/homekit_controller/alarm_control_panel.py
index b17f122dfa5..a0342203e4a 100644
--- a/homeassistant/components/homekit_controller/alarm_control_panel.py
+++ b/homeassistant/components/homekit_controller/alarm_control_panel.py
@@ -15,7 +15,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_BATTERY_LEVEL, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KNOWN_DEVICES
from .connection import HKDevice
@@ -40,7 +40,7 @@ TARGET_STATE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit alarm control panel."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/binary_sensor.py b/homeassistant/components/homekit_controller/binary_sensor.py
index 26e19c8944a..1c80da3cc9c 100644
--- a/homeassistant/components/homekit_controller/binary_sensor.py
+++ b/homeassistant/components/homekit_controller/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KNOWN_DEVICES
from .connection import HKDevice
@@ -156,7 +156,7 @@ REJECT_CHAR_BY_TYPE = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit lighting."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/button.py b/homeassistant/components/homekit_controller/button.py
index ac2133f61ca..730b3c8425d 100644
--- a/homeassistant/components/homekit_controller/button.py
+++ b/homeassistant/components/homekit_controller/button.py
@@ -20,7 +20,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNOWN_DEVICES
@@ -66,7 +66,7 @@ BUTTON_ENTITIES: dict[str, HomeKitButtonEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit buttons."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/camera.py b/homeassistant/components/homekit_controller/camera.py
index 4332032867a..36bf30e5bab 100644
--- a/homeassistant/components/homekit_controller/camera.py
+++ b/homeassistant/components/homekit_controller/camera.py
@@ -9,7 +9,7 @@ from homeassistant.components.camera import Camera
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KNOWN_DEVICES
from .connection import HKDevice
@@ -39,7 +39,7 @@ class HomeKitCamera(AccessoryEntity, Camera):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit sensors."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/climate.py b/homeassistant/components/homekit_controller/climate.py
index cbf4ad61c2f..7341bbd3a4a 100644
--- a/homeassistant/components/homekit_controller/climate.py
+++ b/homeassistant/components/homekit_controller/climate.py
@@ -41,7 +41,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -111,7 +111,7 @@ HASS_FAN_MODE_TO_HOMEKIT_ROTATION = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit climate."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/connection.py b/homeassistant/components/homekit_controller/connection.py
index 211aec2c2d5..43cbdec67fa 100644
--- a/homeassistant/components/homekit_controller/connection.py
+++ b/homeassistant/components/homekit_controller/connection.py
@@ -154,7 +154,6 @@ class HKDevice:
self._pending_subscribes: set[tuple[int, int]] = set()
self._subscribe_timer: CALLBACK_TYPE | None = None
self._load_platforms_lock = asyncio.Lock()
- self._full_update_requested: bool = False
@property
def entity_map(self) -> Accessories:
@@ -841,48 +840,11 @@ class HKDevice:
async def async_request_update(self, now: datetime | None = None) -> None:
"""Request an debounced update from the accessory."""
- self._full_update_requested = True
await self._debounced_update.async_call()
async def async_update(self, now: datetime | None = None) -> None:
"""Poll state of all entities attached to this bridge/accessory."""
to_poll = self.pollable_characteristics
- accessories = self.entity_map.accessories
-
- if (
- not self._full_update_requested
- and len(accessories) == 1
- and self.available
- and not (to_poll - self.watchable_characteristics)
- and self.pairing.is_available
- and await self.pairing.controller.async_reachable(
- self.unique_id, timeout=5.0
- )
- ):
- # If its a single accessory and all chars are watchable,
- # only poll the firmware version to keep the connection alive
- # https://github.com/home-assistant/core/issues/123412
- #
- # Firmware revision is used here since iOS does this to keep camera
- # connections alive, and the goal is to not regress
- # https://github.com/home-assistant/core/issues/116143
- # by polling characteristics that are not normally polled frequently
- # and may not be tested by the device vendor.
- #
- _LOGGER.debug(
- "Accessory is reachable, limiting poll to firmware version: %s",
- self.unique_id,
- )
- first_accessory = accessories[0]
- accessory_info = first_accessory.services.first(
- service_type=ServicesTypes.ACCESSORY_INFORMATION
- )
- assert accessory_info is not None
- firmware_iid = accessory_info[CharacteristicsTypes.FIRMWARE_REVISION].iid
- to_poll = {(first_accessory.aid, firmware_iid)}
-
- self._full_update_requested = False
-
if not to_poll:
self.async_update_available_state()
_LOGGER.debug(
diff --git a/homeassistant/components/homekit_controller/cover.py b/homeassistant/components/homekit_controller/cover.py
index 4fff32002e2..5ea990f55e6 100644
--- a/homeassistant/components/homekit_controller/cover.py
+++ b/homeassistant/components/homekit_controller/cover.py
@@ -19,7 +19,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KNOWN_DEVICES
from .connection import HKDevice
@@ -51,7 +51,7 @@ CURRENT_WINDOW_STATE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit covers."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/event.py b/homeassistant/components/homekit_controller/event.py
index 890c12c9bab..b90d561d60d 100644
--- a/homeassistant/components/homekit_controller/event.py
+++ b/homeassistant/components/homekit_controller/event.py
@@ -14,7 +14,7 @@ from homeassistant.components.event import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KNOWN_DEVICES
from .connection import HKDevice
@@ -86,7 +86,7 @@ class HomeKitEventEntity(BaseCharacteristicEntity, EventEntity):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit event."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/fan.py b/homeassistant/components/homekit_controller/fan.py
index b7f1842392b..9ba476a0ef3 100644
--- a/homeassistant/components/homekit_controller/fan.py
+++ b/homeassistant/components/homekit_controller/fan.py
@@ -17,7 +17,7 @@ from homeassistant.components.fan import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -210,7 +210,7 @@ ENTITY_TYPES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit fans."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/humidifier.py b/homeassistant/components/homekit_controller/humidifier.py
index b2b0e0b1026..7906d5ec52b 100644
--- a/homeassistant/components/homekit_controller/humidifier.py
+++ b/homeassistant/components/homekit_controller/humidifier.py
@@ -20,7 +20,7 @@ from homeassistant.components.humidifier import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNOWN_DEVICES
@@ -165,7 +165,7 @@ class HomeKitDehumidifier(HomeKitBaseHumidifier):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit humidifer."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/light.py b/homeassistant/components/homekit_controller/light.py
index 04c75731731..5409df7c1a8 100644
--- a/homeassistant/components/homekit_controller/light.py
+++ b/homeassistant/components/homekit_controller/light.py
@@ -20,7 +20,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from . import KNOWN_DEVICES
@@ -31,7 +31,7 @@ from .entity import HomeKitEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit lightbulb."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/lock.py b/homeassistant/components/homekit_controller/lock.py
index 98974c4a514..06b8382c8af 100644
--- a/homeassistant/components/homekit_controller/lock.py
+++ b/homeassistant/components/homekit_controller/lock.py
@@ -11,7 +11,7 @@ from homeassistant.components.lock import LockEntity, LockState
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_BATTERY_LEVEL, STATE_UNKNOWN, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KNOWN_DEVICES
from .connection import HKDevice
@@ -32,7 +32,7 @@ REVERSED_TARGET_STATE_MAP = {v: k for k, v in TARGET_STATE_MAP.items()}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit lock."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/manifest.json b/homeassistant/components/homekit_controller/manifest.json
index b7c82b9fd51..98db9a397d3 100644
--- a/homeassistant/components/homekit_controller/manifest.json
+++ b/homeassistant/components/homekit_controller/manifest.json
@@ -14,6 +14,6 @@
"documentation": "https://www.home-assistant.io/integrations/homekit_controller",
"iot_class": "local_push",
"loggers": ["aiohomekit", "commentjson"],
- "requirements": ["aiohomekit==3.2.7"],
+ "requirements": ["aiohomekit==3.2.8"],
"zeroconf": ["_hap._tcp.local.", "_hap._udp.local."]
}
diff --git a/homeassistant/components/homekit_controller/media_player.py b/homeassistant/components/homekit_controller/media_player.py
index 4232d1b7649..e3b4a760680 100644
--- a/homeassistant/components/homekit_controller/media_player.py
+++ b/homeassistant/components/homekit_controller/media_player.py
@@ -22,7 +22,7 @@ from homeassistant.components.media_player import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KNOWN_DEVICES
from .connection import HKDevice
@@ -41,7 +41,7 @@ HK_TO_HA_STATE = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit television."""
hkid: str = config_entry.data["AccessoryPairingID"]
@@ -83,7 +83,7 @@ class HomeKitTelevision(HomeKitEntity, MediaPlayerEntity):
@property
def supported_features(self) -> MediaPlayerEntityFeature:
"""Flag media player features that are supported."""
- features = MediaPlayerEntityFeature(0)
+ features = MediaPlayerEntityFeature.TURN_OFF | MediaPlayerEntityFeature.TURN_ON
if self.service.has(CharacteristicsTypes.ACTIVE_IDENTIFIER):
features |= MediaPlayerEntityFeature.SELECT_SOURCE
@@ -177,6 +177,14 @@ class HomeKitTelevision(HomeKitEntity, MediaPlayerEntity):
return MediaPlayerState.ON
+ async def async_turn_on(self) -> None:
+ """Turn the tv on."""
+ await self.async_put_characteristics({CharacteristicsTypes.ACTIVE: 1})
+
+ async def async_turn_off(self) -> None:
+ """Turn the tv off."""
+ await self.async_put_characteristics({CharacteristicsTypes.ACTIVE: 0})
+
async def async_media_play(self) -> None:
"""Send play command."""
if self.state == MediaPlayerState.PLAYING:
diff --git a/homeassistant/components/homekit_controller/number.py b/homeassistant/components/homekit_controller/number.py
index 340d31c91ae..96d6707d8eb 100644
--- a/homeassistant/components/homekit_controller/number.py
+++ b/homeassistant/components/homekit_controller/number.py
@@ -18,7 +18,7 @@ from homeassistant.components.number import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNOWN_DEVICES
@@ -68,7 +68,7 @@ NUMBER_ENTITIES: dict[str, NumberEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit numbers."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/select.py b/homeassistant/components/homekit_controller/select.py
index f672f293122..f174743b12f 100644
--- a/homeassistant/components/homekit_controller/select.py
+++ b/homeassistant/components/homekit_controller/select.py
@@ -15,7 +15,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNOWN_DEVICES
@@ -148,7 +148,7 @@ class EcobeeModeSelect(BaseHomeKitSelect):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit select entities."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/sensor.py b/homeassistant/components/homekit_controller/sensor.py
index 059be5bad99..c97b45152e0 100644
--- a/homeassistant/components/homekit_controller/sensor.py
+++ b/homeassistant/components/homekit_controller/sensor.py
@@ -43,7 +43,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNOWN_DEVICES
@@ -640,7 +640,7 @@ class RSSISensor(HomeKitEntity, SensorEntity):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit sensors."""
hkid = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homekit_controller/switch.py b/homeassistant/components/homekit_controller/switch.py
index 5abed2a5c79..c24a4edf545 100644
--- a/homeassistant/components/homekit_controller/switch.py
+++ b/homeassistant/components/homekit_controller/switch.py
@@ -17,7 +17,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNOWN_DEVICES
@@ -224,7 +224,7 @@ ENTITY_TYPES: dict[str, type[HomeKitSwitch | HomeKitFaucet | HomeKitValve]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homekit switches."""
hkid: str = config_entry.data["AccessoryPairingID"]
diff --git a/homeassistant/components/homematic/entity.py b/homeassistant/components/homematic/entity.py
index 5a5b2a3b8c8..44e95e98f38 100644
--- a/homeassistant/components/homematic/entity.py
+++ b/homeassistant/components/homematic/entity.py
@@ -62,7 +62,7 @@ class HMDevice(Entity):
if self._state:
self._state = self._state.upper()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Load data init callbacks."""
self._subscribe_homematic_events()
@@ -77,7 +77,7 @@ class HMDevice(Entity):
return self._name
@property
- def available(self):
+ def available(self) -> bool:
"""Return true if device is available."""
return self._available
diff --git a/homeassistant/components/homematic/sensor.py b/homeassistant/components/homematic/sensor.py
index b33a725db0f..24172e196c1 100644
--- a/homeassistant/components/homematic/sensor.py
+++ b/homeassistant/components/homematic/sensor.py
@@ -177,6 +177,7 @@ SENSOR_DESCRIPTIONS: dict[str, SensorEntityDescription] = {
"WIND_DIRECTION": SensorEntityDescription(
key="WIND_DIRECTION",
native_unit_of_measurement=DEGREE,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
"WIND_DIRECTION_RANGE": SensorEntityDescription(
key="WIND_DIRECTION_RANGE",
diff --git a/homeassistant/components/homematicip_cloud/alarm_control_panel.py b/homeassistant/components/homematicip_cloud/alarm_control_panel.py
index 4241316c2a4..d5b084644e3 100644
--- a/homeassistant/components/homematicip_cloud/alarm_control_panel.py
+++ b/homeassistant/components/homematicip_cloud/alarm_control_panel.py
@@ -14,7 +14,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .hap import AsyncHome, HomematicipHAP
@@ -27,7 +27,7 @@ CONST_ALARM_CONTROL_PANEL_NAME = "HmIP Alarm Control Panel"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP alrm control panel from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/binary_sensor.py b/homeassistant/components/homematicip_cloud/binary_sensor.py
index 38590e4505b..f0cd3732718 100644
--- a/homeassistant/components/homematicip_cloud/binary_sensor.py
+++ b/homeassistant/components/homematicip_cloud/binary_sensor.py
@@ -37,7 +37,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -76,7 +76,7 @@ SAM_DEVICE_ATTRIBUTES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP Cloud binary sensor from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/button.py b/homeassistant/components/homematicip_cloud/button.py
index 244be47d7f6..fedc271714c 100644
--- a/homeassistant/components/homematicip_cloud/button.py
+++ b/homeassistant/components/homematicip_cloud/button.py
@@ -7,7 +7,7 @@ from homematicip.aio.device import AsyncWallMountedGarageDoorController
from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -17,7 +17,7 @@ from .hap import HomematicipHAP
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP button from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/climate.py b/homeassistant/components/homematicip_cloud/climate.py
index e7132fac83c..35bd18ff438 100644
--- a/homeassistant/components/homematicip_cloud/climate.py
+++ b/homeassistant/components/homematicip_cloud/climate.py
@@ -29,7 +29,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -57,7 +57,7 @@ HMIP_ECO_CM = "ECO"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP climate from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/cover.py b/homeassistant/components/homematicip_cloud/cover.py
index 1db536afd4f..27a84abb572 100644
--- a/homeassistant/components/homematicip_cloud/cover.py
+++ b/homeassistant/components/homematicip_cloud/cover.py
@@ -23,7 +23,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -38,7 +38,7 @@ HMIP_SLATS_CLOSED = 1
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP cover from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/event.py b/homeassistant/components/homematicip_cloud/event.py
index 8fb558b2b34..654f56bb47f 100644
--- a/homeassistant/components/homematicip_cloud/event.py
+++ b/homeassistant/components/homematicip_cloud/event.py
@@ -12,7 +12,7 @@ from homeassistant.components.event import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -37,7 +37,7 @@ EVENT_DESCRIPTIONS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP cover from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/light.py b/homeassistant/components/homematicip_cloud/light.py
index cf051103a10..ad946809fd4 100644
--- a/homeassistant/components/homematicip_cloud/light.py
+++ b/homeassistant/components/homematicip_cloud/light.py
@@ -30,7 +30,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -40,7 +40,7 @@ from .hap import HomematicipHAP
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP Cloud lights from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/lock.py b/homeassistant/components/homematicip_cloud/lock.py
index b00f42fc844..a054e95a80d 100644
--- a/homeassistant/components/homematicip_cloud/lock.py
+++ b/homeassistant/components/homematicip_cloud/lock.py
@@ -11,7 +11,7 @@ from homematicip.base.enums import LockState, MotorState
from homeassistant.components.lock import LockEntity, LockEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -37,7 +37,7 @@ DEVICE_DLD_ATTRIBUTES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP locks from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/sensor.py b/homeassistant/components/homematicip_cloud/sensor.py
index 9ed9b33d7c7..0280f5bc7d5 100644
--- a/homeassistant/components/homematicip_cloud/sensor.py
+++ b/homeassistant/components/homematicip_cloud/sensor.py
@@ -57,7 +57,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN
@@ -96,7 +96,7 @@ ILLUMINATION_DEVICE_ATTRIBUTES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP Cloud sensors from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/strings.json b/homeassistant/components/homematicip_cloud/strings.json
index 37deace7ebf..7b1b08ac4e2 100644
--- a/homeassistant/components/homematicip_cloud/strings.json
+++ b/homeassistant/components/homematicip_cloud/strings.json
@@ -3,6 +3,7 @@
"step": {
"init": {
"title": "Pick Homematic IP access point",
+ "description": "If you are about to register a **Homematic IP HCU1**, please press the button on top of the device before you continue.\n\nThe registration process must be completed within 5 minutes.",
"data": {
"hapid": "Access point ID (SGTIN)",
"pin": "[%key:common::config_flow::data::pin%]",
@@ -34,7 +35,7 @@
"services": {
"activate_eco_mode_with_duration": {
"name": "Activate eco mode with duration",
- "description": "Activates eco mode with period.",
+ "description": "Activates the eco mode for a specified duration.",
"fields": {
"duration": {
"name": "Duration",
@@ -48,7 +49,7 @@
},
"activate_eco_mode_with_period": {
"name": "Activate eco more with period",
- "description": "[%key:component::homematicip_cloud::services::activate_eco_mode_with_duration::description%]",
+ "description": "Activates the eco mode until a given time.",
"fields": {
"endtime": {
"name": "Endtime",
@@ -62,7 +63,7 @@
},
"activate_vacation": {
"name": "Activate vacation",
- "description": "Activates the vacation mode until the given time.",
+ "description": "Activates the vacation mode until a given time.",
"fields": {
"endtime": {
"name": "[%key:component::homematicip_cloud::services::activate_eco_mode_with_period::fields::endtime::name%]",
diff --git a/homeassistant/components/homematicip_cloud/switch.py b/homeassistant/components/homematicip_cloud/switch.py
index 70bf14631cb..a9aa1c664d7 100644
--- a/homeassistant/components/homematicip_cloud/switch.py
+++ b/homeassistant/components/homematicip_cloud/switch.py
@@ -25,7 +25,7 @@ from homematicip.aio.group import AsyncExtendedLinkedSwitchingGroup, AsyncSwitch
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import ATTR_GROUP_MEMBER_UNREACHABLE, HomematicipGenericEntity
@@ -35,7 +35,7 @@ from .hap import HomematicipHAP
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP switch from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homematicip_cloud/weather.py b/homeassistant/components/homematicip_cloud/weather.py
index cbe7c2845b8..1125c73f8d4 100644
--- a/homeassistant/components/homematicip_cloud/weather.py
+++ b/homeassistant/components/homematicip_cloud/weather.py
@@ -25,7 +25,7 @@ from homeassistant.components.weather import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfSpeed, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import HomematicipGenericEntity
@@ -53,7 +53,7 @@ HOME_WEATHER_CONDITION = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the HomematicIP weather sensor from a config entry."""
hap = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/homewizard/__init__.py b/homeassistant/components/homewizard/__init__.py
index 36c9681dcd2..3831146aed8 100644
--- a/homeassistant/components/homewizard/__init__.py
+++ b/homeassistant/components/homewizard/__init__.py
@@ -7,7 +7,7 @@ from homewizard_energy import (
has_v2_api,
)
-from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntry
+from homeassistant.config_entries import SOURCE_REAUTH
from homeassistant.const import CONF_IP_ADDRESS, CONF_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@@ -15,9 +15,7 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from .const import DOMAIN, PLATFORMS
-from .coordinator import HWEnergyDeviceUpdateCoordinator
-
-type HomeWizardConfigEntry = ConfigEntry[HWEnergyDeviceUpdateCoordinator]
+from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: HomeWizardConfigEntry) -> bool:
@@ -42,7 +40,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: HomeWizardConfigEntry) -
if is_battery:
await async_check_v2_support_and_create_issue(hass, entry)
- coordinator = HWEnergyDeviceUpdateCoordinator(hass, api)
+ coordinator = HWEnergyDeviceUpdateCoordinator(hass, entry, api)
try:
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/homewizard/button.py b/homeassistant/components/homewizard/button.py
index b86f797ec2d..5a36cdb71f2 100644
--- a/homeassistant/components/homewizard/button.py
+++ b/homeassistant/components/homewizard/button.py
@@ -3,10 +3,9 @@
from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import HomeWizardConfigEntry
-from .coordinator import HWEnergyDeviceUpdateCoordinator
+from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
from .helpers import homewizard_exception_handler
@@ -16,7 +15,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeWizardConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Identify button."""
if entry.runtime_data.data.device.supports_identify():
diff --git a/homeassistant/components/homewizard/config_flow.py b/homeassistant/components/homewizard/config_flow.py
index 6bcc51f939e..68dc54aef0e 100644
--- a/homeassistant/components/homewizard/config_flow.py
+++ b/homeassistant/components/homewizard/config_flow.py
@@ -23,8 +23,10 @@ import voluptuous as vol
from homeassistant.components import onboarding
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_IP_ADDRESS, CONF_TOKEN
+from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import AbortFlow
from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers import instance_id
from homeassistant.helpers.selector import TextSelector
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
@@ -88,7 +90,7 @@ class HomeWizardConfigFlow(ConfigFlow, domain=DOMAIN):
# Tell device we want a token, user must now press the button within 30 seconds
# The first attempt will always fail, but this opens the window to press the button
- token = await async_request_token(self.ip_address)
+ token = await async_request_token(self.hass, self.ip_address)
errors: dict[str, str] | None = None
if token is None:
@@ -250,7 +252,7 @@ class HomeWizardConfigFlow(ConfigFlow, domain=DOMAIN):
errors: dict[str, str] | None = None
- token = await async_request_token(self.ip_address)
+ token = await async_request_token(self.hass, self.ip_address)
if user_input is not None:
if token is None:
@@ -353,7 +355,7 @@ async def async_try_connect(ip_address: str, token: str | None = None) -> Device
await energy_api.close()
-async def async_request_token(ip_address: str) -> str | None:
+async def async_request_token(hass: HomeAssistant, ip_address: str) -> str | None:
"""Try to request a token from the device.
This method is used to request a token from the device,
@@ -362,8 +364,12 @@ async def async_request_token(ip_address: str) -> str | None:
api = HomeWizardEnergyV2(ip_address)
+ # Get a part of the unique id to make the token unique
+ # This is to prevent token conflicts when multiple HA instances are used
+ uuid = await instance_id.async_get(hass)
+
try:
- return await api.get_token("home-assistant")
+ return await api.get_token(f"home-assistant#{uuid[:6]}")
except DisabledError:
return None
finally:
diff --git a/homeassistant/components/homewizard/coordinator.py b/homeassistant/components/homewizard/coordinator.py
index 92beb99ad2c..e87381c5fa9 100644
--- a/homeassistant/components/homewizard/coordinator.py
+++ b/homeassistant/components/homewizard/coordinator.py
@@ -13,6 +13,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN, LOGGER, UPDATE_INTERVAL
+type HomeWizardConfigEntry = ConfigEntry[HWEnergyDeviceUpdateCoordinator]
+
class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator[DeviceResponseEntry]):
"""Gather data for the energy device."""
@@ -20,11 +22,22 @@ class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator[DeviceResponseEntry]
api: HomeWizardEnergy
api_disabled: bool = False
- config_entry: ConfigEntry
+ config_entry: HomeWizardConfigEntry
- def __init__(self, hass: HomeAssistant, api: HomeWizardEnergy) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: HomeWizardConfigEntry,
+ api: HomeWizardEnergy,
+ ) -> None:
"""Initialize update coordinator."""
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=UPDATE_INTERVAL,
+ )
self.api = api
async def _async_update_data(self) -> DeviceResponseEntry:
diff --git a/homeassistant/components/homewizard/diagnostics.py b/homeassistant/components/homewizard/diagnostics.py
index 12bd25671e0..a3ae2555173 100644
--- a/homeassistant/components/homewizard/diagnostics.py
+++ b/homeassistant/components/homewizard/diagnostics.py
@@ -9,7 +9,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_IP_ADDRESS
from homeassistant.core import HomeAssistant
-from . import HomeWizardConfigEntry
+from .coordinator import HomeWizardConfigEntry
TO_REDACT = {
CONF_IP_ADDRESS,
diff --git a/homeassistant/components/homewizard/number.py b/homeassistant/components/homewizard/number.py
index 5806295fc81..a703043a63b 100644
--- a/homeassistant/components/homewizard/number.py
+++ b/homeassistant/components/homewizard/number.py
@@ -5,10 +5,9 @@ from __future__ import annotations
from homeassistant.components.number import NumberEntity
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import HomeWizardConfigEntry
-from .coordinator import HWEnergyDeviceUpdateCoordinator
+from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
from .helpers import homewizard_exception_handler
@@ -18,7 +17,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeWizardConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up numbers for device."""
if entry.runtime_data.data.device.supports_state():
diff --git a/homeassistant/components/homewizard/repairs.py b/homeassistant/components/homewizard/repairs.py
index 4c9a03b493f..60790202032 100644
--- a/homeassistant/components/homewizard/repairs.py
+++ b/homeassistant/components/homewizard/repairs.py
@@ -47,7 +47,7 @@ class MigrateToV2ApiRepairFlow(RepairsFlow):
# Tell device we want a token, user must now press the button within 30 seconds
# The first attempt will always fail, but this opens the window to press the button
- token = await async_request_token(ip_address)
+ token = await async_request_token(self.hass, ip_address)
errors: dict[str, str] | None = None
if token is None:
diff --git a/homeassistant/components/homewizard/sensor.py b/homeassistant/components/homewizard/sensor.py
index f6f5588956c..dd557532240 100644
--- a/homeassistant/components/homewizard/sensor.py
+++ b/homeassistant/components/homewizard/sensor.py
@@ -33,13 +33,12 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
-from . import HomeWizardConfigEntry
from .const import DOMAIN
-from .coordinator import HWEnergyDeviceUpdateCoordinator
+from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
PARALLEL_UPDATES = 1
@@ -691,7 +690,7 @@ EXTERNAL_SENSORS = {
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeWizardConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize sensors."""
diff --git a/homeassistant/components/homewizard/switch.py b/homeassistant/components/homewizard/switch.py
index 8ebb56433b1..1930b40583d 100644
--- a/homeassistant/components/homewizard/switch.py
+++ b/homeassistant/components/homewizard/switch.py
@@ -16,10 +16,9 @@ from homeassistant.components.switch import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import HomeWizardConfigEntry
-from .coordinator import HWEnergyDeviceUpdateCoordinator
+from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
from .helpers import homewizard_exception_handler
@@ -70,7 +69,7 @@ SWITCHES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: HomeWizardConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches."""
async_add_entities(
diff --git a/homeassistant/components/homeworks/binary_sensor.py b/homeassistant/components/homeworks/binary_sensor.py
index f1ba3c02835..9bdea75479d 100644
--- a/homeassistant/components/homeworks/binary_sensor.py
+++ b/homeassistant/components/homeworks/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import HomeworksData, HomeworksKeypad
from .const import (
@@ -31,7 +31,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homeworks binary sensors."""
data: HomeworksData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/homeworks/button.py b/homeassistant/components/homeworks/button.py
index 6a13573ac88..d76c18985e9 100644
--- a/homeassistant/components/homeworks/button.py
+++ b/homeassistant/components/homeworks/button.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import HomeworksData
from .const import (
@@ -27,7 +27,9 @@ from .entity import HomeworksEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homeworks buttons."""
data: HomeworksData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/homeworks/light.py b/homeassistant/components/homeworks/light.py
index ac52c1f4974..f07758bbace 100644
--- a/homeassistant/components/homeworks/light.py
+++ b/homeassistant/components/homeworks/light.py
@@ -13,7 +13,7 @@ from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import HomeworksData
from .const import CONF_ADDR, CONF_CONTROLLER_ID, CONF_DIMMERS, CONF_RATE, DOMAIN
@@ -23,7 +23,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Homeworks lights."""
data: HomeworksData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/homeworks/strings.json b/homeassistant/components/homeworks/strings.json
index 10cc2e61fb9..1a144615e89 100644
--- a/homeassistant/components/homeworks/strings.json
+++ b/homeassistant/components/homeworks/strings.json
@@ -14,7 +14,7 @@
},
"step": {
"import_finish": {
- "description": "The existing YAML configuration has succesfully been imported.\n\nYou can now remove the `homeworks` configuration from your configuration.yaml file."
+ "description": "The existing YAML configuration has successfully been imported.\n\nYou can now remove the `homeworks` configuration from your configuration.yaml file."
},
"import_controller_name": {
"description": "Lutron Homeworks is no longer configured through configuration.yaml.\n\nPlease fill in the form to import the existing configuration to the UI.",
diff --git a/homeassistant/components/honeywell/climate.py b/homeassistant/components/honeywell/climate.py
index 1df5eb9601b..5fe84aadd75 100644
--- a/homeassistant/components/honeywell/climate.py
+++ b/homeassistant/components/honeywell/climate.py
@@ -36,7 +36,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import TemperatureConverter
from . import HoneywellConfigEntry, HoneywellData
@@ -98,7 +98,7 @@ SCAN_INTERVAL = datetime.timedelta(seconds=30)
async def async_setup_entry(
hass: HomeAssistant,
entry: HoneywellConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Honeywell thermostat."""
cool_away_temp = entry.options.get(CONF_COOL_AWAY_TEMPERATURE)
diff --git a/homeassistant/components/honeywell/humidifier.py b/homeassistant/components/honeywell/humidifier.py
index e94ba465c30..77776f84a2e 100644
--- a/homeassistant/components/honeywell/humidifier.py
+++ b/homeassistant/components/honeywell/humidifier.py
@@ -15,7 +15,7 @@ from homeassistant.components.humidifier import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import HoneywellConfigEntry
from .const import DOMAIN
@@ -73,7 +73,7 @@ HUMIDIFIERS: dict[str, HoneywellHumidifierEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HoneywellConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Honeywell (de)humidifier dynamically."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/honeywell/sensor.py b/homeassistant/components/honeywell/sensor.py
index a9109d5d557..75ac6b1b6d3 100644
--- a/homeassistant/components/honeywell/sensor.py
+++ b/homeassistant/components/honeywell/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import HoneywellConfigEntry
@@ -81,7 +81,7 @@ SENSOR_TYPES: tuple[HoneywellSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HoneywellConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Honeywell thermostat."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/honeywell/switch.py b/homeassistant/components/honeywell/switch.py
index 3602dd1ba10..06c79bf4b1d 100644
--- a/homeassistant/components/honeywell/switch.py
+++ b/homeassistant/components/honeywell/switch.py
@@ -15,7 +15,7 @@ from homeassistant.components.switch import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import HoneywellConfigEntry, HoneywellData
from .const import DOMAIN
@@ -34,7 +34,7 @@ SWITCH_TYPES: tuple[SwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: HoneywellConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Honeywell switches."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/huawei_lte/binary_sensor.py b/homeassistant/components/huawei_lte/binary_sensor.py
index 06b859cea84..c3434dd0b64 100644
--- a/homeassistant/components/huawei_lte/binary_sensor.py
+++ b/homeassistant/components/huawei_lte/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
@@ -30,7 +30,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up from config entry."""
router = hass.data[DOMAIN].routers[config_entry.entry_id]
diff --git a/homeassistant/components/huawei_lte/button.py b/homeassistant/components/huawei_lte/button.py
index 55b009d25bf..44b35d51dd4 100644
--- a/homeassistant/components/huawei_lte/button.py
+++ b/homeassistant/components/huawei_lte/button.py
@@ -25,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: entity_platform.AddEntitiesCallback,
+ async_add_entities: entity_platform.AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Huawei LTE buttons."""
router = hass.data[DOMAIN].routers[config_entry.entry_id]
diff --git a/homeassistant/components/huawei_lte/device_tracker.py b/homeassistant/components/huawei_lte/device_tracker.py
index df849d4f712..83e82bf17ff 100644
--- a/homeassistant/components/huawei_lte/device_tracker.py
+++ b/homeassistant/components/huawei_lte/device_tracker.py
@@ -17,7 +17,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Router
from .const import (
@@ -53,7 +53,7 @@ def _get_hosts(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up from config entry."""
@@ -128,7 +128,7 @@ def _is_us(host: _HostType) -> bool:
@callback
def async_add_new_entities(
router: Router,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
tracked: set[str],
) -> None:
"""Add new entities that are not already being tracked."""
diff --git a/homeassistant/components/huawei_lte/select.py b/homeassistant/components/huawei_lte/select.py
index d8a16ae2f79..3df6fa53320 100644
--- a/homeassistant/components/huawei_lte/select.py
+++ b/homeassistant/components/huawei_lte/select.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UNDEFINED
from . import Router
@@ -38,7 +38,7 @@ class HuaweiSelectEntityDescription(SelectEntityDescription):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up from config entry."""
router = hass.data[DOMAIN].routers[config_entry.entry_id]
diff --git a/homeassistant/components/huawei_lte/sensor.py b/homeassistant/components/huawei_lte/sensor.py
index 86965e89dd0..3543433ca45 100644
--- a/homeassistant/components/huawei_lte/sensor.py
+++ b/homeassistant/components/huawei_lte/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import Router
@@ -754,7 +754,7 @@ SENSOR_META: dict[str, HuaweiSensorGroup] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up from config entry."""
router = hass.data[DOMAIN].routers[config_entry.entry_id]
diff --git a/homeassistant/components/huawei_lte/switch.py b/homeassistant/components/huawei_lte/switch.py
index 07fd89d0b6c..ac8bca4234c 100644
--- a/homeassistant/components/huawei_lte/switch.py
+++ b/homeassistant/components/huawei_lte/switch.py
@@ -13,7 +13,7 @@ from homeassistant.components.switch import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
@@ -28,7 +28,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up from config entry."""
router = hass.data[DOMAIN].routers[config_entry.entry_id]
diff --git a/homeassistant/components/hue/binary_sensor.py b/homeassistant/components/hue/binary_sensor.py
index 2cb8e8b5d90..ecaa6576775 100644
--- a/homeassistant/components/hue/binary_sensor.py
+++ b/homeassistant/components/hue/binary_sensor.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import HueBridge
from .const import DOMAIN
@@ -15,7 +15,7 @@ from .v2.binary_sensor import async_setup_entry as setup_entry_v2
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor entities."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/event.py b/homeassistant/components/hue/event.py
index 64f3ccba9f9..249f81687c0 100644
--- a/homeassistant/components/hue/event.py
+++ b/homeassistant/components/hue/event.py
@@ -16,7 +16,7 @@ from homeassistant.components.event import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import HueBridge
from .const import DEFAULT_BUTTON_EVENT_TYPES, DEVICE_SPECIFIC_EVENT_TYPES, DOMAIN
@@ -26,7 +26,7 @@ from .v2.entity import HueBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up event platform from Hue button resources."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/light.py b/homeassistant/components/hue/light.py
index c3168b5c8c1..9906c9bffa4 100644
--- a/homeassistant/components/hue/light.py
+++ b/homeassistant/components/hue/light.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import HueBridge
from .const import DOMAIN
@@ -16,7 +16,7 @@ from .v2.light import async_setup_entry as setup_entry_v2
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up light entities."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/manifest.json b/homeassistant/components/hue/manifest.json
index 22f1d3991e7..8bc3d84bd50 100644
--- a/homeassistant/components/hue/manifest.json
+++ b/homeassistant/components/hue/manifest.json
@@ -10,6 +10,6 @@
"integration_type": "hub",
"iot_class": "local_push",
"loggers": ["aiohue"],
- "requirements": ["aiohue==4.7.3"],
+ "requirements": ["aiohue==4.7.4"],
"zeroconf": ["_hue._tcp.local."]
}
diff --git a/homeassistant/components/hue/scene.py b/homeassistant/components/hue/scene.py
index 1d83804820d..0b9eb4efbd6 100644
--- a/homeassistant/components/hue/scene.py
+++ b/homeassistant/components/hue/scene.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
@@ -34,7 +34,7 @@ ATTR_BRIGHTNESS = "brightness"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up scene platform from Hue group scenes."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/sensor.py b/homeassistant/components/hue/sensor.py
index 45cff053aef..227742fdbab 100644
--- a/homeassistant/components/hue/sensor.py
+++ b/homeassistant/components/hue/sensor.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import HueBridge
from .const import DOMAIN
@@ -15,7 +15,7 @@ from .v2.sensor import async_setup_entry as setup_entry_v2
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor entities."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/switch.py b/homeassistant/components/hue/switch.py
index b4bc57acf2d..b6b21686d25 100644
--- a/homeassistant/components/hue/switch.py
+++ b/homeassistant/components/hue/switch.py
@@ -22,7 +22,7 @@ from homeassistant.components.switch import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import HueBridge
from .const import DOMAIN
@@ -32,7 +32,7 @@ from .v2.entity import HueBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hue switch platform from Hue resources."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/v2/binary_sensor.py b/homeassistant/components/hue/v2/binary_sensor.py
index 5054ab6e817..6e4c7f98973 100644
--- a/homeassistant/components/hue/v2/binary_sensor.py
+++ b/homeassistant/components/hue/v2/binary_sensor.py
@@ -30,7 +30,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from ..bridge import HueBridge
from ..const import DOMAIN
@@ -49,7 +49,7 @@ type ControllerType = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hue Sensors from Config Entry."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/v2/device.py b/homeassistant/components/hue/v2/device.py
index 25a027f9ebe..7bb3d28e962 100644
--- a/homeassistant/components/hue/v2/device.py
+++ b/homeassistant/components/hue/v2/device.py
@@ -94,7 +94,12 @@ async def async_setup_devices(bridge: HueBridge):
add_device(hue_resource)
# create/update all current devices found in controllers
- known_devices = [add_device(hue_device) for hue_device in dev_controller]
+ # sort the devices to ensure bridges are added first
+ hue_devices = list(dev_controller)
+ hue_devices.sort(
+ key=lambda dev: dev.metadata.archetype != DeviceArchetypes.BRIDGE_V2
+ )
+ known_devices = [add_device(hue_device) for hue_device in hue_devices]
known_devices += [add_device(hue_room) for hue_room in api.groups.room]
known_devices += [add_device(hue_zone) for hue_zone in api.groups.zone]
diff --git a/homeassistant/components/hue/v2/group.py b/homeassistant/components/hue/v2/group.py
index 17cd20b55aa..2f9f195df97 100644
--- a/homeassistant/components/hue/v2/group.py
+++ b/homeassistant/components/hue/v2/group.py
@@ -26,7 +26,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from ..bridge import HueBridge
@@ -42,7 +42,7 @@ from .helpers import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hue groups on light platform."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hue/v2/light.py b/homeassistant/components/hue/v2/light.py
index 86d8cc93e54..4b00299bc9d 100644
--- a/homeassistant/components/hue/v2/light.py
+++ b/homeassistant/components/hue/v2/light.py
@@ -27,7 +27,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from ..bridge import HueBridge
@@ -48,7 +48,7 @@ FALLBACK_KELVIN = 5800 # halfway
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hue Light from Config Entry."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
@@ -107,7 +107,9 @@ class HueLight(HueBaseEntity, LightEntity):
self._attr_effect_list = []
if effects := resource.effects:
self._attr_effect_list = [
- x.value for x in effects.status_values if x != EffectStatus.NO_EFFECT
+ x.value
+ for x in effects.status_values
+ if x not in (EffectStatus.NO_EFFECT, EffectStatus.UNKNOWN)
]
if timed_effects := resource.timed_effects:
self._attr_effect_list += [
diff --git a/homeassistant/components/hue/v2/sensor.py b/homeassistant/components/hue/v2/sensor.py
index bdf1db6df2e..ae6e456a8b4 100644
--- a/homeassistant/components/hue/v2/sensor.py
+++ b/homeassistant/components/hue/v2/sensor.py
@@ -28,7 +28,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import LIGHT_LUX, PERCENTAGE, EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from ..bridge import HueBridge
from ..const import DOMAIN
@@ -46,7 +46,7 @@ type ControllerType = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Hue Sensors from Config Entry."""
bridge: HueBridge = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/huisbaasje/sensor.py b/homeassistant/components/huisbaasje/sensor.py
index c024e3030fa..91c953b2182 100644
--- a/homeassistant/components/huisbaasje/sensor.py
+++ b/homeassistant/components/huisbaasje/sensor.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -219,7 +219,7 @@ SENSORS_INFO = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
coordinator: DataUpdateCoordinator[dict[str, dict[str, Any]]] = hass.data[DOMAIN][
diff --git a/homeassistant/components/humidifier/strings.json b/homeassistant/components/humidifier/strings.json
index 753368dc572..436f7df8312 100644
--- a/homeassistant/components/humidifier/strings.json
+++ b/homeassistant/components/humidifier/strings.json
@@ -89,7 +89,7 @@
"fields": {
"mode": {
"name": "Mode",
- "description": "Operation mode. For example, _normal_, _eco_, or _away_. For a list of possible values, refer to the integration documentation."
+ "description": "Operation mode. For example, \"normal\", \"eco\", or \"away\". For a list of possible values, refer to the integration documentation."
}
}
},
diff --git a/homeassistant/components/hunterdouglas_powerview/__init__.py b/homeassistant/components/hunterdouglas_powerview/__init__.py
index b4bbc37b1e8..3e9ff8727ce 100644
--- a/homeassistant/components/hunterdouglas_powerview/__init__.py
+++ b/homeassistant/components/hunterdouglas_powerview/__init__.py
@@ -91,7 +91,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: PowerviewConfigEntry) ->
entry, unique_id=device_info.serial_number
)
- coordinator = PowerviewShadeUpdateCoordinator(hass, shades, hub)
+ coordinator = PowerviewShadeUpdateCoordinator(hass, entry, shades, hub)
coordinator.async_set_updated_data(PowerviewShadeData())
# populate raw shade data into the coordinator for diagnostics
coordinator.data.store_group_data(shade_data)
diff --git a/homeassistant/components/hunterdouglas_powerview/button.py b/homeassistant/components/hunterdouglas_powerview/button.py
index adb3e177a8e..c0bcac3a7df 100644
--- a/homeassistant/components/hunterdouglas_powerview/button.py
+++ b/homeassistant/components/hunterdouglas_powerview/button.py
@@ -22,7 +22,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PowerviewShadeUpdateCoordinator
from .entity import ShadeEntity
@@ -74,7 +74,7 @@ BUTTONS_SHADE: Final = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerviewConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the hunter douglas advanced feature buttons."""
pv_entry = entry.runtime_data
diff --git a/homeassistant/components/hunterdouglas_powerview/coordinator.py b/homeassistant/components/hunterdouglas_powerview/coordinator.py
index f074b06b2bc..2ff1914079a 100644
--- a/homeassistant/components/hunterdouglas_powerview/coordinator.py
+++ b/homeassistant/components/hunterdouglas_powerview/coordinator.py
@@ -10,6 +10,7 @@ from aiopvapi.helpers.aiorequest import PvApiMaintenance
from aiopvapi.hub import Hub
from aiopvapi.shades import Shades
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -22,7 +23,11 @@ _LOGGER = logging.getLogger(__name__)
class PowerviewShadeUpdateCoordinator(DataUpdateCoordinator[PowerviewShadeData]):
"""DataUpdateCoordinator to gather data from a powerview hub."""
- def __init__(self, hass: HomeAssistant, shades: Shades, hub: Hub) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, shades: Shades, hub: Hub
+ ) -> None:
"""Initialize DataUpdateCoordinator to gather data for specific Powerview Hub."""
self.shades = shades
self.hub = hub
@@ -33,6 +38,7 @@ class PowerviewShadeUpdateCoordinator(DataUpdateCoordinator[PowerviewShadeData])
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"powerview hub {hub.hub_address}",
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/hunterdouglas_powerview/cover.py b/homeassistant/components/hunterdouglas_powerview/cover.py
index 197fb4e6223..3f36a57891c 100644
--- a/homeassistant/components/hunterdouglas_powerview/cover.py
+++ b/homeassistant/components/hunterdouglas_powerview/cover.py
@@ -26,7 +26,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from .const import STATE_ATTRIBUTE_ROOM_NAME
@@ -50,7 +50,7 @@ SCAN_INTERVAL = timedelta(minutes=10)
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerviewConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the hunter douglas shades."""
pv_entry = entry.runtime_data
diff --git a/homeassistant/components/hunterdouglas_powerview/number.py b/homeassistant/components/hunterdouglas_powerview/number.py
index fb8c9f76d79..216cdb2642a 100644
--- a/homeassistant/components/hunterdouglas_powerview/number.py
+++ b/homeassistant/components/hunterdouglas_powerview/number.py
@@ -14,7 +14,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PowerviewShadeUpdateCoordinator
from .entity import ShadeEntity
@@ -54,7 +54,7 @@ NUMBERS: Final = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerviewConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the hunter douglas number entities."""
pv_entry = entry.runtime_data
diff --git a/homeassistant/components/hunterdouglas_powerview/scene.py b/homeassistant/components/hunterdouglas_powerview/scene.py
index 2aaa255c5ab..5016b590f91 100644
--- a/homeassistant/components/hunterdouglas_powerview/scene.py
+++ b/homeassistant/components/hunterdouglas_powerview/scene.py
@@ -10,7 +10,7 @@ from aiopvapi.resources.scene import Scene as PvScene
from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import STATE_ATTRIBUTE_ROOM_NAME
from .coordinator import PowerviewShadeUpdateCoordinator
@@ -25,7 +25,7 @@ RESYNC_DELAY = 60
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerviewConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up powerview scene entries."""
pv_entry = entry.runtime_data
diff --git a/homeassistant/components/hunterdouglas_powerview/select.py b/homeassistant/components/hunterdouglas_powerview/select.py
index db850a0ddbf..932ff3ce3bd 100644
--- a/homeassistant/components/hunterdouglas_powerview/select.py
+++ b/homeassistant/components/hunterdouglas_powerview/select.py
@@ -12,7 +12,7 @@ from aiopvapi.resources.shade import BaseShade
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PowerviewShadeUpdateCoordinator
from .entity import ShadeEntity
@@ -54,7 +54,7 @@ DROPDOWNS: Final = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerviewConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the hunter douglas select entities."""
pv_entry = entry.runtime_data
diff --git a/homeassistant/components/hunterdouglas_powerview/sensor.py b/homeassistant/components/hunterdouglas_powerview/sensor.py
index f5e3ddd5e12..6ebf8e2b278 100644
--- a/homeassistant/components/hunterdouglas_powerview/sensor.py
+++ b/homeassistant/components/hunterdouglas_powerview/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, SIGNAL_STRENGTH_DECIBELS, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PowerviewShadeUpdateCoordinator
from .entity import ShadeEntity
@@ -79,7 +79,7 @@ SENSORS: Final = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerviewConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the hunter douglas sensor entities."""
pv_entry = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/__init__.py b/homeassistant/components/husqvarna_automower/__init__.py
index a08256fb0b5..1945647a706 100644
--- a/homeassistant/components/husqvarna_automower/__init__.py
+++ b/homeassistant/components/husqvarna_automower/__init__.py
@@ -5,7 +5,6 @@ import logging
from aioautomower.session import AutomowerSession
from aiohttp import ClientResponseError
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -13,7 +12,7 @@ from homeassistant.helpers import aiohttp_client, config_entry_oauth2_flow
from homeassistant.util import dt as dt_util
from . import api
-from .coordinator import AutomowerDataUpdateCoordinator
+from .coordinator import AutomowerConfigEntry, AutomowerDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -29,8 +28,6 @@ PLATFORMS: list[Platform] = [
Platform.SWITCH,
]
-type AutomowerConfigEntry = ConfigEntry[AutomowerDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: AutomowerConfigEntry) -> bool:
"""Set up this integration using UI."""
@@ -61,7 +58,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: AutomowerConfigEntry) ->
# without the scope. So only polling would be possible.
raise ConfigEntryAuthFailed
- coordinator = AutomowerDataUpdateCoordinator(hass, automower_api)
+ coordinator = AutomowerDataUpdateCoordinator(hass, entry, automower_api)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/husqvarna_automower/binary_sensor.py b/homeassistant/components/husqvarna_automower/binary_sensor.py
index 907d34e812a..1e5b9fac990 100644
--- a/homeassistant/components/husqvarna_automower/binary_sensor.py
+++ b/homeassistant/components/husqvarna_automower/binary_sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.components.script import scripts_with_entity
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import (
IssueSeverity,
async_create_issue,
@@ -71,7 +71,7 @@ MOWER_BINARY_SENSOR_TYPES: tuple[AutomowerBinarySensorEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/button.py b/homeassistant/components/husqvarna_automower/button.py
index 7e6e581cdf1..1f7ed7127e0 100644
--- a/homeassistant/components/husqvarna_automower/button.py
+++ b/homeassistant/components/husqvarna_automower/button.py
@@ -10,7 +10,7 @@ from aioautomower.session import AutomowerSession
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AutomowerConfigEntry
from .coordinator import AutomowerDataUpdateCoordinator
@@ -54,7 +54,7 @@ MOWER_BUTTON_TYPES: tuple[AutomowerButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/calendar.py b/homeassistant/components/husqvarna_automower/calendar.py
index 9e2ea037afb..26e939ec7d9 100644
--- a/homeassistant/components/husqvarna_automower/calendar.py
+++ b/homeassistant/components/husqvarna_automower/calendar.py
@@ -7,7 +7,7 @@ from aioautomower.model import make_name_string
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import AutomowerConfigEntry
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up lawn mower platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/coordinator.py b/homeassistant/components/husqvarna_automower/coordinator.py
index a587b4f3821..9456074596a 100644
--- a/homeassistant/components/husqvarna_automower/coordinator.py
+++ b/homeassistant/components/husqvarna_automower/coordinator.py
@@ -6,7 +6,6 @@ import asyncio
from collections.abc import Callable
from datetime import timedelta
import logging
-from typing import TYPE_CHECKING
from aioautomower.exceptions import (
ApiError,
@@ -14,9 +13,10 @@ from aioautomower.exceptions import (
HusqvarnaTimeoutError,
HusqvarnaWSServerHandshakeError,
)
-from aioautomower.model import MowerAttributes
+from aioautomower.model import MowerDictionary
from aioautomower.session import AutomowerSession
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers import device_registry as dr, entity_registry as er
@@ -24,25 +24,30 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN
-if TYPE_CHECKING:
- from . import AutomowerConfigEntry
-
_LOGGER = logging.getLogger(__name__)
MAX_WS_RECONNECT_TIME = 600
SCAN_INTERVAL = timedelta(minutes=8)
DEFAULT_RECONNECT_TIME = 2 # Define a default reconnect time
+type AutomowerConfigEntry = ConfigEntry[AutomowerDataUpdateCoordinator]
-class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttributes]]):
+
+class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[MowerDictionary]):
"""Class to manage fetching Husqvarna data."""
config_entry: AutomowerConfigEntry
- def __init__(self, hass: HomeAssistant, api: AutomowerSession) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: AutomowerConfigEntry,
+ api: AutomowerSession,
+ ) -> None:
"""Initialize data updater."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
@@ -56,7 +61,7 @@ class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttrib
self._zones_last_update: dict[str, set[str]] = {}
self._areas_last_update: dict[str, set[int]] = {}
- async def _async_update_data(self) -> dict[str, MowerAttributes]:
+ async def _async_update_data(self) -> MowerDictionary:
"""Subscribe for websocket and poll data from the API."""
if not self.ws_connected:
await self.api.connect()
@@ -79,7 +84,7 @@ class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttrib
return data
@callback
- def callback(self, ws_data: dict[str, MowerAttributes]) -> None:
+ def callback(self, ws_data: MowerDictionary) -> None:
"""Process websocket callbacks and write them to the DataUpdateCoordinator."""
self.async_set_updated_data(ws_data)
@@ -114,7 +119,7 @@ class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttrib
"reconnect_task",
)
- def _async_add_remove_devices(self, data: dict[str, MowerAttributes]) -> None:
+ def _async_add_remove_devices(self, data: MowerDictionary) -> None:
"""Add new device, remove non-existing device."""
current_devices = set(data)
@@ -154,9 +159,7 @@ class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttrib
for mower_callback in self.new_devices_callbacks:
mower_callback(new_devices)
- def _async_add_remove_stay_out_zones(
- self, data: dict[str, MowerAttributes]
- ) -> None:
+ def _async_add_remove_stay_out_zones(self, data: MowerDictionary) -> None:
"""Add new stay-out zones, remove non-existing stay-out zones."""
current_zones = {
mower_id: set(mower_data.stay_out_zones.zones)
@@ -202,7 +205,7 @@ class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttrib
return current_zones
- def _async_add_remove_work_areas(self, data: dict[str, MowerAttributes]) -> None:
+ def _async_add_remove_work_areas(self, data: MowerDictionary) -> None:
"""Add new work areas, remove non-existing work areas."""
current_areas = {
mower_id: set(mower_data.work_areas)
diff --git a/homeassistant/components/husqvarna_automower/device_tracker.py b/homeassistant/components/husqvarna_automower/device_tracker.py
index 2fd59b63014..78fad7c3610 100644
--- a/homeassistant/components/husqvarna_automower/device_tracker.py
+++ b/homeassistant/components/husqvarna_automower/device_tracker.py
@@ -2,7 +2,7 @@
from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AutomowerConfigEntry
from .coordinator import AutomowerDataUpdateCoordinator
@@ -15,7 +15,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/lawn_mower.py b/homeassistant/components/husqvarna_automower/lawn_mower.py
index dd75a8b9bc4..ee6007f089b 100644
--- a/homeassistant/components/husqvarna_automower/lawn_mower.py
+++ b/homeassistant/components/husqvarna_automower/lawn_mower.py
@@ -15,7 +15,7 @@ from homeassistant.components.lawn_mower import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AutomowerConfigEntry
from .const import DOMAIN
@@ -49,7 +49,7 @@ OVERRIDE_MODES = [MOW, PARK]
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up lawn mower platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/manifest.json b/homeassistant/components/husqvarna_automower/manifest.json
index 0eabf5ec0d6..45d4df95a04 100644
--- a/homeassistant/components/husqvarna_automower/manifest.json
+++ b/homeassistant/components/husqvarna_automower/manifest.json
@@ -8,5 +8,5 @@
"iot_class": "cloud_push",
"loggers": ["aioautomower"],
"quality_scale": "silver",
- "requirements": ["aioautomower==2025.1.1"]
+ "requirements": ["aioautomower==2025.3.1"]
}
diff --git a/homeassistant/components/husqvarna_automower/number.py b/homeassistant/components/husqvarna_automower/number.py
index d3666494646..cdcf4b45a2d 100644
--- a/homeassistant/components/husqvarna_automower/number.py
+++ b/homeassistant/components/husqvarna_automower/number.py
@@ -11,7 +11,7 @@ from aioautomower.session import AutomowerSession
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AutomowerConfigEntry
from .coordinator import AutomowerDataUpdateCoordinator
@@ -107,7 +107,7 @@ WORK_AREA_NUMBER_TYPES: tuple[WorkAreaNumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/select.py b/homeassistant/components/husqvarna_automower/select.py
index 03b1ac02587..9124a0705e1 100644
--- a/homeassistant/components/husqvarna_automower/select.py
+++ b/homeassistant/components/husqvarna_automower/select.py
@@ -8,7 +8,7 @@ from aioautomower.model import HeadlightModes
from homeassistant.components.select import SelectEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AutomowerConfigEntry
from .coordinator import AutomowerDataUpdateCoordinator
@@ -29,7 +29,7 @@ HEADLIGHT_MODES: list = [
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/sensor.py b/homeassistant/components/husqvarna_automower/sensor.py
index a2f4b5f4bab..2e1d4041e5a 100644
--- a/homeassistant/components/husqvarna_automower/sensor.py
+++ b/homeassistant/components/husqvarna_automower/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfLength, UnitOfTime
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import AutomowerConfigEntry
@@ -430,7 +430,7 @@ WORK_AREA_SENSOR_TYPES: tuple[WorkAreaSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower/switch.py b/homeassistant/components/husqvarna_automower/switch.py
index d55d51b42fe..69a3e670eda 100644
--- a/homeassistant/components/husqvarna_automower/switch.py
+++ b/homeassistant/components/husqvarna_automower/switch.py
@@ -7,7 +7,7 @@ from aioautomower.model import MowerModes, StayOutZones, Zone
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import AutomowerConfigEntry
from .coordinator import AutomowerDataUpdateCoordinator
@@ -26,7 +26,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: AutomowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/husqvarna_automower_ble/__init__.py b/homeassistant/components/husqvarna_automower_ble/__init__.py
index 2025ba64cf1..ca07d1ab8d2 100644
--- a/homeassistant/components/husqvarna_automower_ble/__init__.py
+++ b/homeassistant/components/husqvarna_automower_ble/__init__.py
@@ -45,7 +45,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
model = await mower.get_model()
LOGGER.debug("Connected to Automower: %s", model)
- coordinator = HusqvarnaCoordinator(hass, mower, address, channel_id, model)
+ coordinator = HusqvarnaCoordinator(hass, entry, mower, address, channel_id, model)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/husqvarna_automower_ble/coordinator.py b/homeassistant/components/husqvarna_automower_ble/coordinator.py
index c577ccd9196..dde3462c081 100644
--- a/homeassistant/components/husqvarna_automower_ble/coordinator.py
+++ b/homeassistant/components/husqvarna_automower_ble/coordinator.py
@@ -9,6 +9,7 @@ from bleak import BleakError
from bleak_retry_connector import close_stale_connections_by_address
from homeassistant.components import bluetooth
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -20,9 +21,12 @@ SCAN_INTERVAL = timedelta(seconds=60)
class HusqvarnaCoordinator(DataUpdateCoordinator[dict[str, bytes]]):
"""Class to manage fetching data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
mower: Mower,
address: str,
channel_id: str,
@@ -32,6 +36,7 @@ class HusqvarnaCoordinator(DataUpdateCoordinator[dict[str, bytes]]):
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/husqvarna_automower_ble/lawn_mower.py b/homeassistant/components/husqvarna_automower_ble/lawn_mower.py
index 980efc6f069..4b239394c2d 100644
--- a/homeassistant/components/husqvarna_automower_ble/lawn_mower.py
+++ b/homeassistant/components/husqvarna_automower_ble/lawn_mower.py
@@ -12,7 +12,7 @@ from homeassistant.components.lawn_mower import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import LOGGER
from .coordinator import HusqvarnaCoordinator
@@ -22,7 +22,7 @@ from .entity import HusqvarnaAutomowerBleEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up AutomowerLawnMower integration from a config entry."""
coordinator: HusqvarnaCoordinator = config_entry.runtime_data
diff --git a/homeassistant/components/huum/climate.py b/homeassistant/components/huum/climate.py
index 7e0e4ce5ef1..84173260d04 100644
--- a/homeassistant/components/huum/climate.py
+++ b/homeassistant/components/huum/climate.py
@@ -20,7 +20,7 @@ from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTempera
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -30,7 +30,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Huum sauna with config flow."""
huum_handler = hass.data.setdefault(DOMAIN, {})[entry.entry_id]
diff --git a/homeassistant/components/hvv_departures/binary_sensor.py b/homeassistant/components/hvv_departures/binary_sensor.py
index 913c61f91b4..622a8436e04 100644
--- a/homeassistant/components/hvv_departures/binary_sensor.py
+++ b/homeassistant/components/hvv_departures/binary_sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -30,7 +30,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary_sensor platform."""
hub = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/hvv_departures/sensor.py b/homeassistant/components/hvv_departures/sensor.py
index 6ad61295d04..667893db8f2 100644
--- a/homeassistant/components/hvv_departures/sensor.py
+++ b/homeassistant/components/hvv_departures/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.const import ATTR_ID, CONF_OFFSET
from homeassistant.core import HomeAssistant
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle
from homeassistant.util.dt import get_time_zone, utcnow
@@ -42,7 +42,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
hub = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hydrawise/__init__.py b/homeassistant/components/hydrawise/__init__.py
index ee5a8a66610..ce4d7a8f8c2 100644
--- a/homeassistant/components/hydrawise/__init__.py
+++ b/homeassistant/components/hydrawise/__init__.py
@@ -39,10 +39,10 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
app_id=APP_ID,
)
- main_coordinator = HydrawiseMainDataUpdateCoordinator(hass, hydrawise)
+ main_coordinator = HydrawiseMainDataUpdateCoordinator(hass, config_entry, hydrawise)
await main_coordinator.async_config_entry_first_refresh()
water_use_coordinator = HydrawiseWaterUseDataUpdateCoordinator(
- hass, hydrawise, main_coordinator
+ hass, config_entry, hydrawise, main_coordinator
)
await water_use_coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = (
diff --git a/homeassistant/components/hydrawise/binary_sensor.py b/homeassistant/components/hydrawise/binary_sensor.py
index 83e8a8325f9..b2862930933 100644
--- a/homeassistant/components/hydrawise/binary_sensor.py
+++ b/homeassistant/components/hydrawise/binary_sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from .const import DOMAIN, SERVICE_RESUME, SERVICE_START_WATERING, SERVICE_SUSPEND
@@ -78,7 +78,7 @@ SCHEMA_SUSPEND: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Hydrawise binary_sensor platform."""
coordinators: HydrawiseUpdateCoordinators = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hydrawise/coordinator.py b/homeassistant/components/hydrawise/coordinator.py
index 4721a9fb154..35d816b341b 100644
--- a/homeassistant/components/hydrawise/coordinator.py
+++ b/homeassistant/components/hydrawise/coordinator.py
@@ -7,6 +7,7 @@ from dataclasses import dataclass, field
from pydrawise import HydrawiseBase
from pydrawise.schema import Controller, ControllerWaterUseSummary, Sensor, User, Zone
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from homeassistant.util.dt import now
@@ -39,6 +40,7 @@ class HydrawiseDataUpdateCoordinator(DataUpdateCoordinator[HydrawiseData]):
"""Base class for Hydrawise Data Update Coordinators."""
api: HydrawiseBase
+ config_entry: ConfigEntry
class HydrawiseMainDataUpdateCoordinator(HydrawiseDataUpdateCoordinator):
@@ -49,9 +51,17 @@ class HydrawiseMainDataUpdateCoordinator(HydrawiseDataUpdateCoordinator):
integration are updated in a timely manner.
"""
- def __init__(self, hass: HomeAssistant, api: HydrawiseBase) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: HydrawiseBase
+ ) -> None:
"""Initialize HydrawiseDataUpdateCoordinator."""
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=MAIN_SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=MAIN_SCAN_INTERVAL,
+ )
self.api = api
async def _async_update_data(self) -> HydrawiseData:
@@ -82,6 +92,7 @@ class HydrawiseWaterUseDataUpdateCoordinator(HydrawiseDataUpdateCoordinator):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
api: HydrawiseBase,
main_coordinator: HydrawiseMainDataUpdateCoordinator,
) -> None:
@@ -89,6 +100,7 @@ class HydrawiseWaterUseDataUpdateCoordinator(HydrawiseDataUpdateCoordinator):
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=f"{DOMAIN} water use",
update_interval=WATER_USE_SCAN_INTERVAL,
)
diff --git a/homeassistant/components/hydrawise/manifest.json b/homeassistant/components/hydrawise/manifest.json
index de45eb061d5..0c355c34a71 100644
--- a/homeassistant/components/hydrawise/manifest.json
+++ b/homeassistant/components/hydrawise/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/hydrawise",
"iot_class": "cloud_polling",
"loggers": ["pydrawise"],
- "requirements": ["pydrawise==2025.1.0"]
+ "requirements": ["pydrawise==2025.3.0"]
}
diff --git a/homeassistant/components/hydrawise/sensor.py b/homeassistant/components/hydrawise/sensor.py
index 96cc16832da..60bc1d7dc63 100644
--- a/homeassistant/components/hydrawise/sensor.py
+++ b/homeassistant/components/hydrawise/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTime, UnitOfVolume
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN
@@ -131,7 +131,7 @@ FLOW_MEASUREMENT_KEYS = [x.key for x in FLOW_CONTROLLER_SENSORS]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Hydrawise sensor platform."""
coordinators: HydrawiseUpdateCoordinators = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hydrawise/switch.py b/homeassistant/components/hydrawise/switch.py
index 62cd81a0481..bc6b31e6d2e 100644
--- a/homeassistant/components/hydrawise/switch.py
+++ b/homeassistant/components/hydrawise/switch.py
@@ -16,7 +16,7 @@ from homeassistant.components.switch import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DEFAULT_WATERING_TIME, DOMAIN
@@ -63,7 +63,7 @@ SWITCH_KEYS: list[str] = [desc.key for desc in SWITCH_TYPES]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Hydrawise switch platform."""
coordinators: HydrawiseUpdateCoordinators = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hydrawise/valve.py b/homeassistant/components/hydrawise/valve.py
index 37f196bc054..13aff22ccbf 100644
--- a/homeassistant/components/hydrawise/valve.py
+++ b/homeassistant/components/hydrawise/valve.py
@@ -14,7 +14,7 @@ from homeassistant.components.valve import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import HydrawiseUpdateCoordinators
@@ -31,7 +31,7 @@ VALVE_TYPES: tuple[ValveEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Hydrawise valve platform."""
coordinators: HydrawiseUpdateCoordinators = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hyperion/camera.py b/homeassistant/components/hyperion/camera.py
index 23ce2715140..1260be20eb2 100644
--- a/homeassistant/components/hyperion/camera.py
+++ b/homeassistant/components/hyperion/camera.py
@@ -32,7 +32,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import (
get_hyperion_device_id,
@@ -54,7 +54,7 @@ IMAGE_STREAM_JPG_SENTINEL = "data:image/jpg;base64,"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Hyperion platform from config entry."""
entry_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hyperion/light.py b/homeassistant/components/hyperion/light.py
index 40d093430a5..f8932a682ab 100644
--- a/homeassistant/components/hyperion/light.py
+++ b/homeassistant/components/hyperion/light.py
@@ -25,7 +25,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from . import (
@@ -76,7 +76,7 @@ ICON_EFFECT = "mdi:lava-lamp"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Hyperion platform from config entry."""
diff --git a/homeassistant/components/hyperion/sensor.py b/homeassistant/components/hyperion/sensor.py
index ad972806ae5..42b41acea96 100644
--- a/homeassistant/components/hyperion/sensor.py
+++ b/homeassistant/components/hyperion/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import (
get_hyperion_device_id,
@@ -63,7 +63,7 @@ def _sensor_unique_id(server_id: str, instance_num: int, suffix: str) -> str:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Hyperion platform from config entry."""
entry_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/hyperion/switch.py b/homeassistant/components/hyperion/switch.py
index 94cbf2aba29..8b66783e889 100644
--- a/homeassistant/components/hyperion/switch.py
+++ b/homeassistant/components/hyperion/switch.py
@@ -34,7 +34,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import slugify
from . import (
@@ -90,7 +90,7 @@ def _component_to_translation_key(component: str) -> str:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Hyperion platform from config entry."""
entry_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/ialarm/__init__.py b/homeassistant/components/ialarm/__init__.py
index 95c62b87a19..2484a46f906 100644
--- a/homeassistant/components/ialarm/__init__.py
+++ b/homeassistant/components/ialarm/__init__.py
@@ -29,7 +29,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except (TimeoutError, ConnectionError) as ex:
raise ConfigEntryNotReady from ex
- coordinator = IAlarmDataUpdateCoordinator(hass, ialarm, mac)
+ coordinator = IAlarmDataUpdateCoordinator(hass, entry, ialarm, mac)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})
diff --git a/homeassistant/components/ialarm/alarm_control_panel.py b/homeassistant/components/ialarm/alarm_control_panel.py
index 4ae3787dc1d..e203f892c35 100644
--- a/homeassistant/components/ialarm/alarm_control_panel.py
+++ b/homeassistant/components/ialarm/alarm_control_panel.py
@@ -10,7 +10,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DATA_COORDINATOR, DOMAIN
@@ -18,7 +18,9 @@ from .coordinator import IAlarmDataUpdateCoordinator
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a iAlarm alarm control panel based on a config entry."""
coordinator: IAlarmDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
diff --git a/homeassistant/components/ialarm/coordinator.py b/homeassistant/components/ialarm/coordinator.py
index ad0f2298a3b..61e87c36796 100644
--- a/homeassistant/components/ialarm/coordinator.py
+++ b/homeassistant/components/ialarm/coordinator.py
@@ -11,6 +11,7 @@ from homeassistant.components.alarm_control_panel import (
SCAN_INTERVAL,
AlarmControlPanelState,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -22,7 +23,11 @@ _LOGGER = logging.getLogger(__name__)
class IAlarmDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Class to manage fetching iAlarm data."""
- def __init__(self, hass: HomeAssistant, ialarm: IAlarm, mac: str) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, ialarm: IAlarm, mac: str
+ ) -> None:
"""Initialize global iAlarm data updater."""
self.ialarm = ialarm
self.state: AlarmControlPanelState | None = None
@@ -32,6 +37,7 @@ class IAlarmDataUpdateCoordinator(DataUpdateCoordinator[None]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/iaqualink/binary_sensor.py b/homeassistant/components/iaqualink/binary_sensor.py
index 9e173dc36e0..8fe9d77fbe8 100644
--- a/homeassistant/components/iaqualink/binary_sensor.py
+++ b/homeassistant/components/iaqualink/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN as AQUALINK_DOMAIN
from .entity import AqualinkEntity
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up discovered binary sensors."""
async_add_entities(
diff --git a/homeassistant/components/iaqualink/climate.py b/homeassistant/components/iaqualink/climate.py
index 53d1bce80de..d30700898c8 100644
--- a/homeassistant/components/iaqualink/climate.py
+++ b/homeassistant/components/iaqualink/climate.py
@@ -18,7 +18,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import refresh_system
from .const import DOMAIN as AQUALINK_DOMAIN
@@ -33,7 +33,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up discovered switches."""
async_add_entities(
diff --git a/homeassistant/components/iaqualink/light.py b/homeassistant/components/iaqualink/light.py
index 59172c13576..e515c482158 100644
--- a/homeassistant/components/iaqualink/light.py
+++ b/homeassistant/components/iaqualink/light.py
@@ -16,7 +16,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import refresh_system
from .const import DOMAIN as AQUALINK_DOMAIN
@@ -29,7 +29,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up discovered lights."""
async_add_entities(
diff --git a/homeassistant/components/iaqualink/sensor.py b/homeassistant/components/iaqualink/sensor.py
index 881adb420bf..1b453f28d8f 100644
--- a/homeassistant/components/iaqualink/sensor.py
+++ b/homeassistant/components/iaqualink/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN as AQUALINK_DOMAIN
from .entity import AqualinkEntity
@@ -23,7 +23,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up discovered sensors."""
async_add_entities(
diff --git a/homeassistant/components/iaqualink/switch.py b/homeassistant/components/iaqualink/switch.py
index 601c5701a4a..e746cbb4f4b 100644
--- a/homeassistant/components/iaqualink/switch.py
+++ b/homeassistant/components/iaqualink/switch.py
@@ -9,7 +9,7 @@ from iaqualink.device import AqualinkSwitch
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import refresh_system
from .const import DOMAIN as AQUALINK_DOMAIN
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up discovered switches."""
async_add_entities(
diff --git a/homeassistant/components/ibeacon/device_tracker.py b/homeassistant/components/ibeacon/device_tracker.py
index d002cb10f44..0d2ee0137cc 100644
--- a/homeassistant/components/ibeacon/device_tracker.py
+++ b/homeassistant/components/ibeacon/device_tracker.py
@@ -9,7 +9,7 @@ from homeassistant.components.device_tracker.config_entry import BaseTrackerEnti
from homeassistant.const import STATE_HOME, STATE_NOT_HOME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IBeaconConfigEntry
from .const import SIGNAL_IBEACON_DEVICE_NEW
@@ -20,7 +20,7 @@ from .entity import IBeaconEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: IBeaconConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for iBeacon Tracker component."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ibeacon/sensor.py b/homeassistant/components/ibeacon/sensor.py
index f73aef4b803..7e1fd371128 100644
--- a/homeassistant/components/ibeacon/sensor.py
+++ b/homeassistant/components/ibeacon/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfLength
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IBeaconConfigEntry
from .const import SIGNAL_IBEACON_DEVICE_NEW
@@ -69,7 +69,7 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: IBeaconConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for iBeacon Tracker component."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/icloud/device_tracker.py b/homeassistant/components/icloud/device_tracker.py
index 11a18a10020..ca194143852 100644
--- a/homeassistant/components/icloud/device_tracker.py
+++ b/homeassistant/components/icloud/device_tracker.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .account import IcloudAccount, IcloudDevice
from .const import (
@@ -21,7 +21,9 @@ from .const import (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for iCloud component."""
account: IcloudAccount = hass.data[DOMAIN][entry.unique_id]
diff --git a/homeassistant/components/icloud/sensor.py b/homeassistant/components/icloud/sensor.py
index 53c9765f6cc..533605b8c7b 100644
--- a/homeassistant/components/icloud/sensor.py
+++ b/homeassistant/components/icloud/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.const import PERCENTAGE
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level
from .account import IcloudAccount, IcloudDevice
@@ -18,7 +18,9 @@ from .const import DOMAIN
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for iCloud component."""
account: IcloudAccount = hass.data[DOMAIN][entry.unique_id]
diff --git a/homeassistant/components/icloud/strings.json b/homeassistant/components/icloud/strings.json
index adc96043d66..fc78e8c2ba6 100644
--- a/homeassistant/components/icloud/strings.json
+++ b/homeassistant/components/icloud/strings.json
@@ -46,7 +46,7 @@
"services": {
"update": {
"name": "Update",
- "description": "Updates iCloud devices.",
+ "description": "Asks for a state update of all devices linked to an iCloud account.",
"fields": {
"account": {
"name": "Account",
diff --git a/homeassistant/components/idasen_desk/__init__.py b/homeassistant/components/idasen_desk/__init__.py
index 671319e46eb..1ea0efeef72 100644
--- a/homeassistant/components/idasen_desk/__init__.py
+++ b/homeassistant/components/idasen_desk/__init__.py
@@ -9,25 +9,22 @@ from idasen_ha.errors import AuthFailedError
from homeassistant.components import bluetooth
from homeassistant.components.bluetooth.match import ADDRESS, BluetoothCallbackMatcher
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ADDRESS, EVENT_HOMEASSISTANT_STOP, Platform
from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryNotReady
-from .coordinator import IdasenDeskCoordinator
+from .coordinator import IdasenDeskConfigEntry, IdasenDeskCoordinator
PLATFORMS: list[Platform] = [Platform.BUTTON, Platform.COVER, Platform.SENSOR]
_LOGGER = logging.getLogger(__name__)
-type IdasenDeskConfigEntry = ConfigEntry[IdasenDeskCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: IdasenDeskConfigEntry) -> bool:
"""Set up IKEA Idasen from a config entry."""
address: str = entry.data[CONF_ADDRESS].upper()
- coordinator = IdasenDeskCoordinator(hass, entry.title, address)
+ coordinator = IdasenDeskCoordinator(hass, entry, address)
entry.runtime_data = coordinator
try:
diff --git a/homeassistant/components/idasen_desk/button.py b/homeassistant/components/idasen_desk/button.py
index cd7553da1ac..2e63e5e7cb8 100644
--- a/homeassistant/components/idasen_desk/button.py
+++ b/homeassistant/components/idasen_desk/button.py
@@ -8,9 +8,9 @@ from typing import Any, Final
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import IdasenDeskConfigEntry, IdasenDeskCoordinator
+from .coordinator import IdasenDeskConfigEntry, IdasenDeskCoordinator
from .entity import IdasenDeskEntity
_LOGGER = logging.getLogger(__name__)
@@ -44,7 +44,7 @@ BUTTONS: Final = [
async def async_setup_entry(
hass: HomeAssistant,
entry: IdasenDeskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set buttons for device."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/idasen_desk/coordinator.py b/homeassistant/components/idasen_desk/coordinator.py
index d9e90cfe5ea..5da3d57cf9a 100644
--- a/homeassistant/components/idasen_desk/coordinator.py
+++ b/homeassistant/components/idasen_desk/coordinator.py
@@ -7,24 +7,31 @@ import logging
from idasen_ha import Desk
from homeassistant.components import bluetooth
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
+type IdasenDeskConfigEntry = ConfigEntry[IdasenDeskCoordinator]
+
class IdasenDeskCoordinator(DataUpdateCoordinator[int | None]):
"""Class to manage updates for the Idasen Desk."""
+ config_entry: IdasenDeskConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- name: str,
+ config_entry: IdasenDeskConfigEntry,
address: str,
) -> None:
"""Init IdasenDeskCoordinator."""
- super().__init__(hass, _LOGGER, name=name)
+ super().__init__(
+ hass, _LOGGER, config_entry=config_entry, name=config_entry.title
+ )
self.address = address
self._expected_connected = False
diff --git a/homeassistant/components/idasen_desk/cover.py b/homeassistant/components/idasen_desk/cover.py
index a8ba0983e99..b451f4d0156 100644
--- a/homeassistant/components/idasen_desk/cover.py
+++ b/homeassistant/components/idasen_desk/cover.py
@@ -14,16 +14,16 @@ from homeassistant.components.cover import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import IdasenDeskConfigEntry, IdasenDeskCoordinator
+from .coordinator import IdasenDeskConfigEntry, IdasenDeskCoordinator
from .entity import IdasenDeskEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: IdasenDeskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the cover platform for Idasen Desk."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/idasen_desk/entity.py b/homeassistant/components/idasen_desk/entity.py
index bda7afd528c..46730ee13fe 100644
--- a/homeassistant/components/idasen_desk/entity.py
+++ b/homeassistant/components/idasen_desk/entity.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import IdasenDeskCoordinator
+from .coordinator import IdasenDeskCoordinator
class IdasenDeskEntity(CoordinatorEntity[IdasenDeskCoordinator]):
diff --git a/homeassistant/components/idasen_desk/sensor.py b/homeassistant/components/idasen_desk/sensor.py
index 4613d316a52..22680b4fa7f 100644
--- a/homeassistant/components/idasen_desk/sensor.py
+++ b/homeassistant/components/idasen_desk/sensor.py
@@ -13,9 +13,9 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfLength
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import IdasenDeskConfigEntry, IdasenDeskCoordinator
+from .coordinator import IdasenDeskConfigEntry, IdasenDeskCoordinator
from .entity import IdasenDeskEntity
@@ -43,7 +43,7 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: IdasenDeskConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Idasen Desk sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/igloohome/__init__.py b/homeassistant/components/igloohome/__init__.py
index 5e5e21452cf..a3907fcbcf3 100644
--- a/homeassistant/components/igloohome/__init__.py
+++ b/homeassistant/components/igloohome/__init__.py
@@ -19,7 +19,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryError, ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-PLATFORMS: list[Platform] = [Platform.SENSOR]
+PLATFORMS: list[Platform] = [Platform.LOCK, Platform.SENSOR]
@dataclass
@@ -35,7 +35,6 @@ type IgloohomeConfigEntry = ConfigEntry[IgloohomeRuntimeData]
async def async_setup_entry(hass: HomeAssistant, entry: IgloohomeConfigEntry) -> bool:
"""Set up igloohome from a config entry."""
-
authentication = IgloohomeAuth(
session=async_get_clientsession(hass),
client_id=entry.data[CONF_CLIENT_ID],
diff --git a/homeassistant/components/igloohome/lock.py b/homeassistant/components/igloohome/lock.py
new file mode 100644
index 00000000000..b434c055145
--- /dev/null
+++ b/homeassistant/components/igloohome/lock.py
@@ -0,0 +1,91 @@
+"""Implementation of the lock platform."""
+
+from datetime import timedelta
+
+from aiohttp import ClientError
+from igloohome_api import (
+ BRIDGE_JOB_LOCK,
+ BRIDGE_JOB_UNLOCK,
+ DEVICE_TYPE_LOCK,
+ Api as IgloohomeApi,
+ ApiException,
+ GetDeviceInfoResponse,
+)
+
+from homeassistant.components.lock import LockEntity, LockEntityFeature
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import IgloohomeConfigEntry
+from .entity import IgloohomeBaseEntity
+from .utils import get_linked_bridge
+
+# Scan interval set to allow Lock entity update the bridge linked to it.
+SCAN_INTERVAL = timedelta(hours=1)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: IgloohomeConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up lock entities."""
+ async_add_entities(
+ IgloohomeLockEntity(
+ api_device_info=device,
+ api=entry.runtime_data.api,
+ bridge_id=str(bridge),
+ )
+ for device in entry.runtime_data.devices
+ if device.type == DEVICE_TYPE_LOCK
+ and (bridge := get_linked_bridge(device.deviceId, entry.runtime_data.devices))
+ is not None
+ )
+
+
+class IgloohomeLockEntity(IgloohomeBaseEntity, LockEntity):
+ """Implementation of a device that has locking capabilities."""
+
+ # Operating on assumed state because there is no API to query the state.
+ _attr_assumed_state = True
+ _attr_supported_features = LockEntityFeature.OPEN
+ _attr_name = None
+
+ def __init__(
+ self, api_device_info: GetDeviceInfoResponse, api: IgloohomeApi, bridge_id: str
+ ) -> None:
+ """Initialize the class."""
+ super().__init__(
+ api_device_info=api_device_info,
+ api=api,
+ unique_key="lock",
+ )
+ self.bridge_id = bridge_id
+
+ async def async_lock(self, **kwargs):
+ """Lock this lock."""
+ try:
+ await self.api.create_bridge_proxied_job(
+ self.api_device_info.deviceId, self.bridge_id, BRIDGE_JOB_LOCK
+ )
+ except (ApiException, ClientError) as err:
+ raise HomeAssistantError from err
+
+ async def async_unlock(self, **kwargs):
+ """Unlock this lock."""
+ try:
+ await self.api.create_bridge_proxied_job(
+ self.api_device_info.deviceId, self.bridge_id, BRIDGE_JOB_UNLOCK
+ )
+ except (ApiException, ClientError) as err:
+ raise HomeAssistantError from err
+
+ async def async_open(self, **kwargs):
+ """Open (unlatch) this lock."""
+ try:
+ await self.api.create_bridge_proxied_job(
+ self.api_device_info.deviceId, self.bridge_id, BRIDGE_JOB_UNLOCK
+ )
+ except (ApiException, ClientError) as err:
+ raise HomeAssistantError from err
diff --git a/homeassistant/components/igloohome/sensor.py b/homeassistant/components/igloohome/sensor.py
index 7f25798e454..10a9ece6771 100644
--- a/homeassistant/components/igloohome/sensor.py
+++ b/homeassistant/components/igloohome/sensor.py
@@ -8,7 +8,7 @@ from igloohome_api import Api as IgloohomeApi, ApiException, GetDeviceInfoRespon
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IgloohomeConfigEntry
from .entity import IgloohomeBaseEntity
@@ -22,7 +22,7 @@ SCAN_INTERVAL = timedelta(hours=1)
async def async_setup_entry(
hass: HomeAssistant,
entry: IgloohomeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor entities."""
diff --git a/homeassistant/components/igloohome/utils.py b/homeassistant/components/igloohome/utils.py
new file mode 100644
index 00000000000..be17912b8b8
--- /dev/null
+++ b/homeassistant/components/igloohome/utils.py
@@ -0,0 +1,16 @@
+"""House utility functions."""
+
+from igloohome_api import DEVICE_TYPE_BRIDGE, GetDeviceInfoResponse
+
+
+def get_linked_bridge(
+ device_id: str, devices: list[GetDeviceInfoResponse]
+) -> str | None:
+ """Return the ID of the bridge that is linked to the device. None if no bridge is linked."""
+ bridges = (bridge for bridge in devices if bridge.type == DEVICE_TYPE_BRIDGE)
+ for bridge in bridges:
+ if device_id in (
+ linked_device.deviceId for linked_device in bridge.linkedDevices
+ ):
+ return bridge.deviceId
+ return None
diff --git a/homeassistant/components/ihc/entity.py b/homeassistant/components/ihc/entity.py
index f90b2ee943c..8847ffc9f49 100644
--- a/homeassistant/components/ihc/entity.py
+++ b/homeassistant/components/ihc/entity.py
@@ -54,7 +54,7 @@ class IHCEntity(Entity):
self.ihc_note = ""
self.ihc_position = ""
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Add callback for IHC changes."""
_LOGGER.debug("Adding IHC entity notify event: %s", self.ihc_id)
self.ihc_controller.add_notify_event(self.ihc_id, self.on_ihc_change, True)
diff --git a/homeassistant/components/imap/coordinator.py b/homeassistant/components/imap/coordinator.py
index 1df107196ff..34d3f43eb69 100644
--- a/homeassistant/components/imap/coordinator.py
+++ b/homeassistant/components/imap/coordinator.py
@@ -241,6 +241,7 @@ class ImapDataUpdateCoordinator(DataUpdateCoordinator[int | None]):
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=DOMAIN,
update_interval=update_interval,
)
@@ -279,7 +280,7 @@ class ImapDataUpdateCoordinator(DataUpdateCoordinator[int | None]):
if self.custom_event_template is not None:
try:
data["custom"] = self.custom_event_template.async_render(
- data, parse_result=True
+ data | {"text": message.text}, parse_result=True
)
_LOGGER.debug(
"IMAP custom template (%s) for msguid %s (%s) rendered to: %s, initial: %s",
diff --git a/homeassistant/components/imap/sensor.py b/homeassistant/components/imap/sensor.py
index 60892388252..01009e3d17b 100644
--- a/homeassistant/components/imap/sensor.py
+++ b/homeassistant/components/imap/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import CONF_USERNAME, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import ImapConfigEntry
@@ -30,7 +30,9 @@ IMAP_MAIL_COUNT_DESCRIPTION = SensorEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ImapConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ImapConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Imap sensor."""
diff --git a/homeassistant/components/imgw_pib/__init__.py b/homeassistant/components/imgw_pib/__init__.py
index eb12e1a2bb4..4bceee51f8e 100644
--- a/homeassistant/components/imgw_pib/__init__.py
+++ b/homeassistant/components/imgw_pib/__init__.py
@@ -2,7 +2,6 @@
from __future__ import annotations
-from dataclasses import dataclass
import logging
from aiohttp import ClientError
@@ -10,7 +9,6 @@ from imgw_pib import ImgwPib
from imgw_pib.exceptions import ApiError
from homeassistant.components.binary_sensor import DOMAIN as BINARY_SENSOR_PLATFORM
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@@ -18,21 +16,12 @@ from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import CONF_STATION_ID, DOMAIN
-from .coordinator import ImgwPibDataUpdateCoordinator
+from .coordinator import ImgwPibConfigEntry, ImgwPibData, ImgwPibDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
_LOGGER = logging.getLogger(__name__)
-type ImgwPibConfigEntry = ConfigEntry[ImgwPibData]
-
-
-@dataclass
-class ImgwPibData:
- """Data for the IMGW-PIB integration."""
-
- coordinator: ImgwPibDataUpdateCoordinator
-
async def async_setup_entry(hass: HomeAssistant, entry: ImgwPibConfigEntry) -> bool:
"""Set up IMGW-PIB from a config entry."""
@@ -49,9 +38,16 @@ async def async_setup_entry(hass: HomeAssistant, entry: ImgwPibConfigEntry) -> b
hydrological_details=False,
)
except (ClientError, TimeoutError, ApiError) as err:
- raise ConfigEntryNotReady from err
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="cannot_connect",
+ translation_placeholders={
+ "entry": entry.title,
+ "error": repr(err),
+ },
+ ) from err
- coordinator = ImgwPibDataUpdateCoordinator(hass, imgwpib, station_id)
+ coordinator = ImgwPibDataUpdateCoordinator(hass, entry, imgwpib, station_id)
await coordinator.async_config_entry_first_refresh()
# Remove binary_sensor entities for which the endpoint has been blocked by IMGW-PIB API
diff --git a/homeassistant/components/imgw_pib/coordinator.py b/homeassistant/components/imgw_pib/coordinator.py
index 77a58001a6f..f74878d672c 100644
--- a/homeassistant/components/imgw_pib/coordinator.py
+++ b/homeassistant/components/imgw_pib/coordinator.py
@@ -1,9 +1,13 @@
"""Data Update Coordinator for IMGW-PIB integration."""
+from __future__ import annotations
+
+from dataclasses import dataclass
import logging
from imgw_pib import ApiError, HydrologicalData, ImgwPib
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -13,12 +17,25 @@ from .const import DOMAIN, UPDATE_INTERVAL
_LOGGER = logging.getLogger(__name__)
+@dataclass
+class ImgwPibData:
+ """Data for the IMGW-PIB integration."""
+
+ coordinator: ImgwPibDataUpdateCoordinator
+
+
+type ImgwPibConfigEntry = ConfigEntry[ImgwPibData]
+
+
class ImgwPibDataUpdateCoordinator(DataUpdateCoordinator[HydrologicalData]):
"""Class to manage fetching IMGW-PIB data API."""
+ config_entry: ImgwPibConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ImgwPibConfigEntry,
imgwpib: ImgwPib,
station_id: str,
) -> None:
@@ -33,11 +50,24 @@ class ImgwPibDataUpdateCoordinator(DataUpdateCoordinator[HydrologicalData]):
configuration_url=f"https://hydro.imgw.pl/#/station/hydro/{station_id}",
)
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=UPDATE_INTERVAL,
+ )
async def _async_update_data(self) -> HydrologicalData:
"""Update data via internal method."""
try:
return await self.imgwpib.get_hydrological_data()
except ApiError as err:
- raise UpdateFailed(err) from err
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="update_error",
+ translation_placeholders={
+ "entry": self.config_entry.title,
+ "error": repr(err),
+ },
+ ) from err
diff --git a/homeassistant/components/imgw_pib/diagnostics.py b/homeassistant/components/imgw_pib/diagnostics.py
index d135208115f..ce9cb3f9e95 100644
--- a/homeassistant/components/imgw_pib/diagnostics.py
+++ b/homeassistant/components/imgw_pib/diagnostics.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.core import HomeAssistant
-from . import ImgwPibConfigEntry
+from .coordinator import ImgwPibConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/imgw_pib/sensor.py b/homeassistant/components/imgw_pib/sensor.py
index 15043af2015..33b82bbb43b 100644
--- a/homeassistant/components/imgw_pib/sensor.py
+++ b/homeassistant/components/imgw_pib/sensor.py
@@ -17,12 +17,11 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfLength, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import ImgwPibConfigEntry
from .const import DOMAIN
-from .coordinator import ImgwPibDataUpdateCoordinator
+from .coordinator import ImgwPibConfigEntry, ImgwPibDataUpdateCoordinator
from .entity import ImgwPibEntity
PARALLEL_UPDATES = 1
@@ -60,7 +59,7 @@ SENSOR_TYPES: tuple[ImgwPibSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ImgwPibConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a IMGW-PIB sensor entity from a config_entry."""
coordinator = entry.runtime_data.coordinator
diff --git a/homeassistant/components/imgw_pib/strings.json b/homeassistant/components/imgw_pib/strings.json
index 9a17dcf7087..89be0661c6f 100644
--- a/homeassistant/components/imgw_pib/strings.json
+++ b/homeassistant/components/imgw_pib/strings.json
@@ -25,5 +25,13 @@
"name": "Water temperature"
}
}
+ },
+ "exceptions": {
+ "cannot_connect": {
+ "message": "An error occurred while connecting to the IMGW-PIB API for {entry}: {error}"
+ },
+ "update_error": {
+ "message": "An error occurred while retrieving data from the IMGW-PIB API for {entry}: {error}"
+ }
}
}
diff --git a/homeassistant/components/improv_ble/__init__.py b/homeassistant/components/improv_ble/__init__.py
index 985684cb5b8..ff40b65a8d0 100644
--- a/homeassistant/components/improv_ble/__init__.py
+++ b/homeassistant/components/improv_ble/__init__.py
@@ -1 +1,11 @@
"""The Improv BLE integration."""
+
+from __future__ import annotations
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+ """Set up improv_ble from a config entry."""
+ raise NotImplementedError
diff --git a/homeassistant/components/improv_ble/config_flow.py b/homeassistant/components/improv_ble/config_flow.py
index 22f2bf3623c..0dcefba6428 100644
--- a/homeassistant/components/improv_ble/config_flow.py
+++ b/homeassistant/components/improv_ble/config_flow.py
@@ -83,12 +83,9 @@ class ImprovBLEConfigFlow(ConfigFlow, domain=DOMAIN):
self._discovery_info = self._discovered_devices[address]
return await self.async_step_start_improv()
- current_addresses = self._async_current_ids()
for discovery in bluetooth.async_discovered_service_info(self.hass):
- if (
- discovery.address in current_addresses
- or discovery.address in self._discovered_devices
- or not device_filter(discovery.advertisement)
+ if discovery.address in self._discovered_devices or not device_filter(
+ discovery.advertisement
):
continue
self._discovered_devices[discovery.address] = discovery
@@ -364,6 +361,18 @@ class ImprovBLEConfigFlow(ConfigFlow, domain=DOMAIN):
assert self._provision_result is not None
result = self._provision_result
+ if result["type"] == "abort" and result["reason"] in (
+ "provision_successful",
+ "provision_successful_url",
+ ):
+ # Delete ignored config entry, if it exists
+ address = self.context["unique_id"]
+ current_entries = self._async_current_entries(include_ignore=True)
+ for entry in current_entries:
+ if entry.unique_id == address:
+ _LOGGER.debug("Removing ignored entry: %s", entry)
+ await self.hass.config_entries.async_remove(entry.entry_id)
+ break
self._provision_result = None
return result
diff --git a/homeassistant/components/incomfort/__init__.py b/homeassistant/components/incomfort/__init__.py
index 4d05a57bcfa..307ff09206f 100644
--- a/homeassistant/components/incomfort/__init__.py
+++ b/homeassistant/components/incomfort/__init__.py
@@ -5,14 +5,18 @@ from __future__ import annotations
from aiohttp import ClientResponseError
from incomfortclient import InvalidGateway, InvalidHeaterList
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers import device_registry as dr
from .const import DOMAIN
-from .coordinator import InComfortData, InComfortDataCoordinator, async_connect_gateway
+from .coordinator import (
+ InComfortConfigEntry,
+ InComfortData,
+ InComfortDataCoordinator,
+ async_connect_gateway,
+)
from .errors import InComfortTimeout, InComfortUnknownError, NoHeaters, NotFound
PLATFORMS = (
@@ -24,8 +28,6 @@ PLATFORMS = (
INTEGRATION_TITLE = "Intergas InComfort/Intouch Lan2RF gateway"
-type InComfortConfigEntry = ConfigEntry[InComfortDataCoordinator]
-
@callback
def async_cleanup_stale_devices(
@@ -93,7 +95,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: InComfortConfigEntry) ->
name="RFGateway",
)
async_cleanup_stale_devices(hass, entry, data, gateway_device)
- coordinator = InComfortDataCoordinator(hass, data, entry.entry_id)
+ coordinator = InComfortDataCoordinator(hass, entry, data)
entry.runtime_data = coordinator
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/incomfort/binary_sensor.py b/homeassistant/components/incomfort/binary_sensor.py
index e4353e457a5..356cee82e57 100644
--- a/homeassistant/components/incomfort/binary_sensor.py
+++ b/homeassistant/components/incomfort/binary_sensor.py
@@ -15,10 +15,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import InComfortConfigEntry
-from .coordinator import InComfortDataCoordinator
+from .coordinator import InComfortConfigEntry, InComfortDataCoordinator
from .entity import IncomfortBoilerEntity
PARALLEL_UPDATES = 0
@@ -71,7 +70,7 @@ SENSOR_TYPES: tuple[IncomfortBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: InComfortConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an InComfort/InTouch binary_sensor entity."""
incomfort_coordinator = entry.runtime_data
diff --git a/homeassistant/components/incomfort/climate.py b/homeassistant/components/incomfort/climate.py
index f814b1fb1f5..d44ba15507e 100644
--- a/homeassistant/components/incomfort/climate.py
+++ b/homeassistant/components/incomfort/climate.py
@@ -15,11 +15,10 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import InComfortConfigEntry
from .const import CONF_LEGACY_SETPOINT_STATUS, DOMAIN
-from .coordinator import InComfortDataCoordinator
+from .coordinator import InComfortConfigEntry, InComfortDataCoordinator
from .entity import IncomfortEntity
PARALLEL_UPDATES = 1
@@ -28,7 +27,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: InComfortConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up InComfort/InTouch climate devices."""
incomfort_coordinator = entry.runtime_data
@@ -74,7 +73,10 @@ class InComfortClimate(IncomfortEntity, ClimateEntity):
name=f"Thermostat {room.room_no}",
)
if coordinator.unique_id:
- self._attr_device_info["via_device"] = (DOMAIN, coordinator.unique_id)
+ self._attr_device_info["via_device"] = (
+ DOMAIN,
+ coordinator.config_entry.entry_id,
+ )
@property
def extra_state_attributes(self) -> dict[str, Any]:
diff --git a/homeassistant/components/incomfort/config_flow.py b/homeassistant/components/incomfort/config_flow.py
index 8e4a5f72619..875bc25bd2f 100644
--- a/homeassistant/components/incomfort/config_flow.py
+++ b/homeassistant/components/incomfort/config_flow.py
@@ -28,9 +28,8 @@ from homeassistant.helpers.selector import (
)
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
-from . import InComfortConfigEntry
from .const import CONF_LEGACY_SETPOINT_STATUS, DOMAIN
-from .coordinator import async_connect_gateway
+from .coordinator import InComfortConfigEntry, async_connect_gateway
TITLE = "Intergas InComfort/Intouch Lan2RF gateway"
diff --git a/homeassistant/components/incomfort/coordinator.py b/homeassistant/components/incomfort/coordinator.py
index 3436d40298a..5c72b9daa06 100644
--- a/homeassistant/components/incomfort/coordinator.py
+++ b/homeassistant/components/incomfort/coordinator.py
@@ -12,12 +12,15 @@ from incomfortclient import (
InvalidHeaterList,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryError
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+type InComfortConfigEntry = ConfigEntry[InComfortDataCoordinator]
+
_LOGGER = logging.getLogger(__name__)
UPDATE_INTERVAL = 30
@@ -50,14 +53,20 @@ async def async_connect_gateway(
class InComfortDataCoordinator(DataUpdateCoordinator[InComfortData]):
"""Data coordinator for InComfort entities."""
+ config_entry: InComfortConfigEntry
+
def __init__(
- self, hass: HomeAssistant, incomfort_data: InComfortData, unique_id: str | None
+ self,
+ hass: HomeAssistant,
+ config_entry: InComfortConfigEntry,
+ incomfort_data: InComfortData,
) -> None:
"""Initialize coordinator."""
- self.unique_id = unique_id
+ self.unique_id = config_entry.unique_id
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="InComfort datacoordinator",
update_interval=timedelta(seconds=UPDATE_INTERVAL),
)
diff --git a/homeassistant/components/incomfort/diagnostics.py b/homeassistant/components/incomfort/diagnostics.py
index a2f89a94f58..4d7af14eac7 100644
--- a/homeassistant/components/incomfort/diagnostics.py
+++ b/homeassistant/components/incomfort/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_PASSWORD
from homeassistant.core import HomeAssistant, callback
-from . import InComfortConfigEntry
+from .coordinator import InComfortConfigEntry
REDACT_CONFIG = {CONF_PASSWORD}
diff --git a/homeassistant/components/incomfort/entity.py b/homeassistant/components/incomfort/entity.py
index 1924c91376b..1e0d357e8e0 100644
--- a/homeassistant/components/incomfort/entity.py
+++ b/homeassistant/components/incomfort/entity.py
@@ -29,4 +29,7 @@ class IncomfortBoilerEntity(IncomfortEntity):
serial_number=heater.serial_no,
)
if coordinator.unique_id:
- self._attr_device_info["via_device"] = (DOMAIN, coordinator.unique_id)
+ self._attr_device_info["via_device"] = (
+ DOMAIN,
+ coordinator.config_entry.entry_id,
+ )
diff --git a/homeassistant/components/incomfort/manifest.json b/homeassistant/components/incomfort/manifest.json
index d02b1d27554..825f198dd30 100644
--- a/homeassistant/components/incomfort/manifest.json
+++ b/homeassistant/components/incomfort/manifest.json
@@ -10,5 +10,6 @@
"documentation": "https://www.home-assistant.io/integrations/incomfort",
"iot_class": "local_polling",
"loggers": ["incomfortclient"],
+ "quality_scale": "platinum",
"requirements": ["incomfort-client==0.6.7"]
}
diff --git a/homeassistant/components/incomfort/quality_scale.yaml b/homeassistant/components/incomfort/quality_scale.yaml
new file mode 100644
index 00000000000..f5af3c9d061
--- /dev/null
+++ b/homeassistant/components/incomfort/quality_scale.yaml
@@ -0,0 +1,77 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: |
+ No actions implemented.
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: |
+ No actions implemented.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: done
+ comment: |
+ Entities are set up dand updated through the datacoordimator.
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+ # Silver
+ config-entry-unloading: done
+ log-when-unavailable: done
+ entity-unavailable: done
+ action-exceptions: done
+ reauthentication-flow: done
+ parallel-updates: done
+ test-coverage: done
+ integration-owner: done
+ docs-installation-parameters: done
+ docs-configuration-parameters: done
+
+ # Gold
+ entity-translations: done
+ entity-device-class: done
+ devices: done
+ entity-category: done
+ entity-disabled-by-default: done
+ discovery: done
+ stale-devices:
+ status: exempt
+ comment: >
+ There is a maximum of 3 heaters that can be discovered by the gateway.
+ The user must remove manually any heeater devices that have been replaced.
+ diagnostics: done
+ exception-translations: done
+ icon-translations: done
+ reconfiguration-flow: done
+ dynamic-devices: done
+ discovery-update-info: done
+ repair-issues:
+ status: exempt
+ comment: |
+ No current issues to repair.
+ docs-use-cases: done
+ docs-supported-devices: done
+ docs-supported-functions: done
+ docs-data-update: done
+ docs-known-limitations:
+ status: done
+ comment: There are no known limmitations,
+ docs-troubleshooting: done
+ docs-examples: done
+
+ # Platinum
+ async-dependency: done
+ inject-websession: done
+ strict-typing: done
diff --git a/homeassistant/components/incomfort/sensor.py b/homeassistant/components/incomfort/sensor.py
index e3f3fc785b2..e344fb01aae 100644
--- a/homeassistant/components/incomfort/sensor.py
+++ b/homeassistant/components/incomfort/sensor.py
@@ -15,11 +15,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory, UnitOfPressure, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import InComfortConfigEntry
-from .coordinator import InComfortDataCoordinator
+from .coordinator import InComfortConfigEntry, InComfortDataCoordinator
from .entity import IncomfortBoilerEntity
PARALLEL_UPDATES = 0
@@ -68,7 +67,7 @@ SENSOR_TYPES: tuple[IncomfortSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: InComfortConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up InComfort/InTouch sensor entities."""
incomfort_coordinator = entry.runtime_data
diff --git a/homeassistant/components/incomfort/strings.json b/homeassistant/components/incomfort/strings.json
index 15e28b6e0b9..73ba88078a8 100644
--- a/homeassistant/components/incomfort/strings.json
+++ b/homeassistant/components/incomfort/strings.json
@@ -2,7 +2,7 @@
"config": {
"step": {
"user": {
- "description": "Set up new Intergas gateway, some older systems might not need credentials to be set up. For newer devices authentication is required.",
+ "description": "Set up new Intergas gateway. Note that some older systems might not accept credentials to be set up. For newer devices authentication is required.",
"data": {
"host": "[%key:common::config_flow::data::host%]",
"username": "[%key:common::config_flow::data::username%]",
diff --git a/homeassistant/components/incomfort/water_heater.py b/homeassistant/components/incomfort/water_heater.py
index 0ab4a6a06b8..2a2c7cc47da 100644
--- a/homeassistant/components/incomfort/water_heater.py
+++ b/homeassistant/components/incomfort/water_heater.py
@@ -10,10 +10,9 @@ from incomfortclient import Heater as InComfortHeater
from homeassistant.components.water_heater import WaterHeaterEntity
from homeassistant.const import EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import InComfortConfigEntry
-from .coordinator import InComfortDataCoordinator
+from .coordinator import InComfortConfigEntry, InComfortDataCoordinator
from .entity import IncomfortBoilerEntity
_LOGGER = logging.getLogger(__name__)
@@ -26,7 +25,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: InComfortConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an InComfort/InTouch water_heater device."""
incomfort_coordinator = entry.runtime_data
diff --git a/homeassistant/components/inkbird/__init__.py b/homeassistant/components/inkbird/__init__.py
index c715c64599a..9dd058e841a 100644
--- a/homeassistant/components/inkbird/__init__.py
+++ b/homeassistant/components/inkbird/__init__.py
@@ -4,17 +4,20 @@ from __future__ import annotations
import logging
-from inkbird_ble import INKBIRDBluetoothDeviceData
+from inkbird_ble import INKBIRDBluetoothDeviceData, SensorUpdate
-from homeassistant.components.bluetooth import BluetoothScanningMode
+from homeassistant.components.bluetooth import (
+ BluetoothScanningMode,
+ BluetoothServiceInfo,
+)
from homeassistant.components.bluetooth.passive_update_processor import (
PassiveBluetoothProcessorCoordinator,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, callback
-from .const import DOMAIN
+from .const import CONF_DEVICE_TYPE, DOMAIN
PLATFORMS: list[Platform] = [Platform.SENSOR]
@@ -25,20 +28,33 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up INKBIRD BLE device from a config entry."""
address = entry.unique_id
assert address is not None
- data = INKBIRDBluetoothDeviceData()
- coordinator = hass.data.setdefault(DOMAIN, {})[entry.entry_id] = (
- PassiveBluetoothProcessorCoordinator(
- hass,
- _LOGGER,
- address=address,
- mode=BluetoothScanningMode.ACTIVE,
- update_method=data.update,
- )
+ device_type: str | None = entry.data.get(CONF_DEVICE_TYPE)
+ data = INKBIRDBluetoothDeviceData(device_type)
+
+ @callback
+ def _async_on_update(service_info: BluetoothServiceInfo) -> SensorUpdate:
+ """Handle update callback from the passive BLE processor."""
+ nonlocal device_type
+ update = data.update(service_info)
+ if device_type is None and data.device_type is not None:
+ device_type_str = str(data.device_type)
+ hass.config_entries.async_update_entry(
+ entry, data={**entry.data, CONF_DEVICE_TYPE: device_type_str}
+ )
+ device_type = device_type_str
+ return update
+
+ coordinator = PassiveBluetoothProcessorCoordinator(
+ hass,
+ _LOGGER,
+ address=address,
+ mode=BluetoothScanningMode.ACTIVE,
+ update_method=_async_on_update,
)
+ hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
- entry.async_on_unload(
- coordinator.async_start()
- ) # only start after all platforms have had a chance to subscribe
+ # only start after all platforms have had a chance to subscribe
+ entry.async_on_unload(coordinator.async_start())
return True
diff --git a/homeassistant/components/inkbird/const.py b/homeassistant/components/inkbird/const.py
index 9d0e1638958..93fdcc7519c 100644
--- a/homeassistant/components/inkbird/const.py
+++ b/homeassistant/components/inkbird/const.py
@@ -1,3 +1,5 @@
"""Constants for the INKBIRD Bluetooth integration."""
DOMAIN = "inkbird"
+
+CONF_DEVICE_TYPE = "device_type"
diff --git a/homeassistant/components/inkbird/manifest.json b/homeassistant/components/inkbird/manifest.json
index c1922004317..aaa9c4b3473 100644
--- a/homeassistant/components/inkbird/manifest.json
+++ b/homeassistant/components/inkbird/manifest.json
@@ -21,6 +21,18 @@
{
"local_name": "tps",
"connectable": false
+ },
+ {
+ "local_name": "ITH-11-B",
+ "connectable": false
+ },
+ {
+ "local_name": "ITH-13-B",
+ "connectable": false
+ },
+ {
+ "local_name": "ITH-21-B",
+ "connectable": false
}
],
"codeowners": ["@bdraco"],
@@ -28,5 +40,5 @@
"dependencies": ["bluetooth_adapters"],
"documentation": "https://www.home-assistant.io/integrations/inkbird",
"iot_class": "local_push",
- "requirements": ["inkbird-ble==0.5.8"]
+ "requirements": ["inkbird-ble==0.9.0"]
}
diff --git a/homeassistant/components/inkbird/sensor.py b/homeassistant/components/inkbird/sensor.py
index 05b2ebbafa0..efda28b110d 100644
--- a/homeassistant/components/inkbird/sensor.py
+++ b/homeassistant/components/inkbird/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -98,7 +98,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the INKBIRD BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/input_select/strings.json b/homeassistant/components/input_select/strings.json
index faa47c979a1..72fd50f7ec7 100644
--- a/homeassistant/components/input_select/strings.json
+++ b/homeassistant/components/input_select/strings.json
@@ -20,7 +20,7 @@
"services": {
"select_next": {
"name": "Next",
- "description": "Select the next option.",
+ "description": "Selects the next option.",
"fields": {
"cycle": {
"name": "Cycle",
@@ -44,7 +44,7 @@
"fields": {
"cycle": {
"name": "[%key:component::input_select::services::select_next::fields::cycle::name%]",
- "description": "[%key:component::input_select::services::select_next::fields::cycle::description%]"
+ "description": "If the option should cycle from the first to the last option on the list."
}
}
},
diff --git a/homeassistant/components/insteon/binary_sensor.py b/homeassistant/components/insteon/binary_sensor.py
index abb26b7f8e8..887c8fb64a3 100644
--- a/homeassistant/components/insteon/binary_sensor.py
+++ b/homeassistant/components/insteon/binary_sensor.py
@@ -22,7 +22,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_ADD_ENTITIES
from .entity import InsteonEntity
@@ -46,7 +46,7 @@ SENSOR_TYPES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Insteon binary sensors from a config entry."""
diff --git a/homeassistant/components/insteon/climate.py b/homeassistant/components/insteon/climate.py
index 506841e7efb..eb33e3ab88c 100644
--- a/homeassistant/components/insteon/climate.py
+++ b/homeassistant/components/insteon/climate.py
@@ -20,7 +20,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_ADD_ENTITIES
from .entity import InsteonEntity
@@ -55,7 +55,7 @@ FAN_MODES = {4: FAN_AUTO, 8: FAN_ONLY}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Insteon climate entities from a config entry."""
diff --git a/homeassistant/components/insteon/cover.py b/homeassistant/components/insteon/cover.py
index fe4f484798d..679ee67a1de 100644
--- a/homeassistant/components/insteon/cover.py
+++ b/homeassistant/components/insteon/cover.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_ADD_ENTITIES
from .entity import InsteonEntity
@@ -22,7 +22,7 @@ from .utils import async_add_insteon_devices, async_add_insteon_entities
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Insteon covers from a config entry."""
diff --git a/homeassistant/components/insteon/entity.py b/homeassistant/components/insteon/entity.py
index 79e5c18a934..b7886723fdf 100644
--- a/homeassistant/components/insteon/entity.py
+++ b/homeassistant/components/insteon/entity.py
@@ -109,7 +109,7 @@ class InsteonEntity(Entity):
)
self.async_write_ha_state()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register INSTEON update events."""
_LOGGER.debug(
"Tracking updates for device %s group %d name %s",
@@ -137,7 +137,7 @@ class InsteonEntity(Entity):
)
)
- async def async_will_remove_from_hass(self):
+ async def async_will_remove_from_hass(self) -> None:
"""Unsubscribe to INSTEON update events."""
_LOGGER.debug(
"Remove tracking updates for device %s group %d name %s",
diff --git a/homeassistant/components/insteon/fan.py b/homeassistant/components/insteon/fan.py
index 0f1c70b9ea8..f4e0abf3d54 100644
--- a/homeassistant/components/insteon/fan.py
+++ b/homeassistant/components/insteon/fan.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -26,7 +26,7 @@ SPEED_RANGE = (1, 255) # off is not included
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Insteon fans from a config entry."""
diff --git a/homeassistant/components/insteon/light.py b/homeassistant/components/insteon/light.py
index d19f3cca34a..e4f09fe5689 100644
--- a/homeassistant/components/insteon/light.py
+++ b/homeassistant/components/insteon/light.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_ADD_ENTITIES
from .entity import InsteonEntity
@@ -22,7 +22,7 @@ MAX_BRIGHTNESS = 255
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Insteon lights from a config entry."""
diff --git a/homeassistant/components/insteon/lock.py b/homeassistant/components/insteon/lock.py
index d5f30eacbac..787b0c583cc 100644
--- a/homeassistant/components/insteon/lock.py
+++ b/homeassistant/components/insteon/lock.py
@@ -7,7 +7,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_ADD_ENTITIES
from .entity import InsteonEntity
@@ -17,7 +17,7 @@ from .utils import async_add_insteon_devices, async_add_insteon_entities
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Insteon locks from a config entry."""
diff --git a/homeassistant/components/insteon/strings.json b/homeassistant/components/insteon/strings.json
index 4a8aadb70db..3a15d667ca7 100644
--- a/homeassistant/components/insteon/strings.json
+++ b/homeassistant/components/insteon/strings.json
@@ -111,7 +111,7 @@
},
"services": {
"add_all_link": {
- "name": "Add all link",
+ "name": "Add All-Link",
"description": "Tells the Insteon Modem (IM) start All-Linking mode. Once the IM is in All-Linking mode, press the link button on the device to complete All-Linking.",
"fields": {
"group": {
@@ -120,13 +120,13 @@
},
"mode": {
"name": "[%key:common::config_flow::data::mode%]",
- "description": "Linking mode controller - IM is controller responder - IM is responder."
+ "description": "Linking mode of the Insteon Modem."
}
}
},
"delete_all_link": {
- "name": "Delete all link",
- "description": "Tells the Insteon Modem (IM) to remove an All-Link record from the All-Link Database of the IM and a device. Once the IM is set to delete the link, press the link button on the corresponding device to complete the process.",
+ "name": "Delete All-Link",
+ "description": "Tells the Insteon Modem (IM) to remove an All-Link record from the All-Link database of the IM and a device. Once the IM is set to delete the link, press the link button on the corresponding device to complete the process.",
"fields": {
"group": {
"name": "Group",
@@ -135,8 +135,8 @@
}
},
"load_all_link_database": {
- "name": "Load all link database",
- "description": "Load the All-Link Database for a device. WARNING - Loading a device All-LInk database is very time consuming and inconsistent. This may take a LONG time and may need to be repeated to obtain all records.",
+ "name": "Load All-Link database",
+ "description": "Loads the All-Link database for a device. WARNING - Loading a device All-Link database is very time consuming and inconsistent. This may take a LONG time and may need to be repeated to obtain all records.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -144,13 +144,13 @@
},
"reload": {
"name": "[%key:common::action::reload%]",
- "description": "Reloads all records. If true the current records are cleared from memory (does not effect the device) and the records are reloaded. If false the existing records are left in place and only missing records are added. Default is false."
+ "description": "If enabled, all current records are cleared from memory (does not effect the device) and reloaded. Otherwise the existing records are left in place and only missing records are added."
}
}
},
"print_all_link_database": {
- "name": "Print all link database",
- "description": "Prints the All-Link Database for a device. Requires that the All-Link Database is loaded into memory.",
+ "name": "Print All-Link database",
+ "description": "Prints the All-Link database for a device. Requires that the All-Link database is loaded into memory.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -159,8 +159,8 @@
}
},
"print_im_all_link_database": {
- "name": "Print IM all link database",
- "description": "Prints the All-Link Database for the INSTEON Modem (IM)."
+ "name": "Print IM All-Link database",
+ "description": "Prints the All-Link database for the INSTEON Modem (IM)."
},
"x10_all_units_off": {
"name": "X10 all units off",
diff --git a/homeassistant/components/insteon/switch.py b/homeassistant/components/insteon/switch.py
index 67ce5fa8c0d..e3f7cf3d7a9 100644
--- a/homeassistant/components/insteon/switch.py
+++ b/homeassistant/components/insteon/switch.py
@@ -7,7 +7,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_ADD_ENTITIES
from .entity import InsteonEntity
@@ -17,7 +17,7 @@ from .utils import async_add_insteon_devices, async_add_insteon_entities
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Insteon switches from a config entry."""
diff --git a/homeassistant/components/insteon/utils.py b/homeassistant/components/insteon/utils.py
index 5b1d6379328..4ee859934d2 100644
--- a/homeassistant/components/insteon/utils.py
+++ b/homeassistant/components/insteon/utils.py
@@ -43,7 +43,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_send,
dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_CAT,
@@ -415,7 +415,7 @@ def async_add_insteon_entities(
hass: HomeAssistant,
platform: Platform,
entity_type: type[InsteonEntity],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
discovery_info: dict[str, Any],
) -> None:
"""Add an Insteon group to a platform."""
@@ -432,7 +432,7 @@ def async_add_insteon_devices(
hass: HomeAssistant,
platform: Platform,
entity_type: type[InsteonEntity],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add all entities to a platform."""
for address in devices:
diff --git a/homeassistant/components/integration/sensor.py b/homeassistant/components/integration/sensor.py
index 27aa74d0785..df5342111a7 100644
--- a/homeassistant/components/integration/sensor.py
+++ b/homeassistant/components/integration/sensor.py
@@ -42,7 +42,10 @@ from homeassistant.core import (
from homeassistant.helpers import config_validation as cv, entity_registry as er
from homeassistant.helpers.device import async_device_info_to_link_from_entity
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import (
async_call_later,
async_track_state_change_event,
@@ -234,7 +237,7 @@ class IntegrationSensorExtraStoredData(SensorExtraStoredData):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Integration - Riemann sum integral config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/intellifire/__init__.py b/homeassistant/components/intellifire/__init__.py
index ce78f1a6fa3..cda30820a2f 100644
--- a/homeassistant/components/intellifire/__init__.py
+++ b/homeassistant/components/intellifire/__init__.py
@@ -128,9 +128,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
) from err
# Construct coordinator
- data_update_coordinator = IntellifireDataUpdateCoordinator(
- hass=hass, fireplace=fireplace
- )
+ data_update_coordinator = IntellifireDataUpdateCoordinator(hass, entry, fireplace)
LOGGER.debug("Fireplace to Initialized - Awaiting first refresh")
await data_update_coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/intellifire/binary_sensor.py b/homeassistant/components/intellifire/binary_sensor.py
index 7d00bdfc26d..3da1d2e3dc0 100644
--- a/homeassistant/components/intellifire/binary_sensor.py
+++ b/homeassistant/components/intellifire/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IntellifireDataUpdateCoordinator
from .const import DOMAIN
@@ -152,7 +152,7 @@ INTELLIFIRE_BINARY_SENSORS: tuple[IntellifireBinarySensorEntityDescription, ...]
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a IntelliFire On/Off Sensor."""
coordinator: IntellifireDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/intellifire/climate.py b/homeassistant/components/intellifire/climate.py
index f72df254424..f067f2a849d 100644
--- a/homeassistant/components/intellifire/climate.py
+++ b/homeassistant/components/intellifire/climate.py
@@ -13,7 +13,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IntellifireDataUpdateCoordinator
from .const import DEFAULT_THERMOSTAT_TEMP, DOMAIN, LOGGER
@@ -27,7 +27,7 @@ INTELLIFIRE_CLIMATES: tuple[ClimateEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure the fan entry.."""
coordinator: IntellifireDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/intellifire/coordinator.py b/homeassistant/components/intellifire/coordinator.py
index b4f03f4b5c8..6a23e7438db 100644
--- a/homeassistant/components/intellifire/coordinator.py
+++ b/homeassistant/components/intellifire/coordinator.py
@@ -9,6 +9,7 @@ from intellifire4py.control import IntelliFireController
from intellifire4py.model import IntelliFirePollData
from intellifire4py.read import IntelliFireDataProvider
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -19,15 +20,19 @@ from .const import DOMAIN, LOGGER
class IntellifireDataUpdateCoordinator(DataUpdateCoordinator[IntelliFirePollData]):
"""Class to manage the polling of the fireplace API."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
fireplace: UnifiedFireplace,
) -> None:
"""Initialize the Coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=15),
)
diff --git a/homeassistant/components/intellifire/fan.py b/homeassistant/components/intellifire/fan.py
index c5bec07faaa..174d964d357 100644
--- a/homeassistant/components/intellifire/fan.py
+++ b/homeassistant/components/intellifire/fan.py
@@ -17,7 +17,7 @@ from homeassistant.components.fan import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -58,7 +58,7 @@ INTELLIFIRE_FANS: tuple[IntellifireFanEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the fans."""
coordinator: IntellifireDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/intellifire/light.py b/homeassistant/components/intellifire/light.py
index 5f25b5de823..0cf5c7774ed 100644
--- a/homeassistant/components/intellifire/light.py
+++ b/homeassistant/components/intellifire/light.py
@@ -17,7 +17,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, LOGGER
from .coordinator import IntellifireDataUpdateCoordinator
@@ -85,7 +85,7 @@ class IntellifireLight(IntellifireEntity, LightEntity):
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the fans."""
coordinator: IntellifireDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/intellifire/number.py b/homeassistant/components/intellifire/number.py
index 17ed3b7bd27..0776835833e 100644
--- a/homeassistant/components/intellifire/number.py
+++ b/homeassistant/components/intellifire/number.py
@@ -11,7 +11,7 @@ from homeassistant.components.number import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, LOGGER
from .coordinator import IntellifireDataUpdateCoordinator
@@ -21,7 +21,7 @@ from .entity import IntellifireEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the fans."""
coordinator: IntellifireDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/intellifire/sensor.py b/homeassistant/components/intellifire/sensor.py
index eaff89d08e7..7763fb1b9b2 100644
--- a/homeassistant/components/intellifire/sensor.py
+++ b/homeassistant/components/intellifire/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from .const import DOMAIN
@@ -141,7 +141,9 @@ INTELLIFIRE_SENSORS: tuple[IntellifireSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Define setup entry call."""
diff --git a/homeassistant/components/intellifire/switch.py b/homeassistant/components/intellifire/switch.py
index ac6096497b6..2185ad47cae 100644
--- a/homeassistant/components/intellifire/switch.py
+++ b/homeassistant/components/intellifire/switch.py
@@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IntellifireDataUpdateCoordinator
from .const import DOMAIN
@@ -53,7 +53,7 @@ INTELLIFIRE_SWITCHES: tuple[IntellifireSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure switch entities."""
coordinator: IntellifireDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/intent/__init__.py b/homeassistant/components/intent/__init__.py
index a1451f8fcca..922fa376903 100644
--- a/homeassistant/components/intent/__init__.py
+++ b/homeassistant/components/intent/__init__.py
@@ -2,13 +2,15 @@
from __future__ import annotations
+from collections.abc import Collection
import logging
from typing import Any, Protocol
from aiohttp import web
import voluptuous as vol
-from homeassistant.components import http
+from homeassistant.components import http, sensor
+from homeassistant.components.climate import DOMAIN as CLIMATE_DOMAIN
from homeassistant.components.cover import (
ATTR_POSITION,
DOMAIN as COVER_DOMAIN,
@@ -39,7 +41,12 @@ from homeassistant.const import (
SERVICE_TURN_ON,
)
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant, State
-from homeassistant.helpers import config_validation as cv, integration_platform, intent
+from homeassistant.helpers import (
+ area_registry as ar,
+ config_validation as cv,
+ integration_platform,
+ intent,
+)
from homeassistant.helpers.typing import ConfigType
from homeassistant.util import dt as dt_util
@@ -140,6 +147,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
intent.async_register(hass, GetCurrentDateIntentHandler())
intent.async_register(hass, GetCurrentTimeIntentHandler())
intent.async_register(hass, RespondIntentHandler())
+ intent.async_register(hass, GetTemperatureIntent())
return True
@@ -444,6 +452,109 @@ class RespondIntentHandler(intent.IntentHandler):
return response
+class GetTemperatureIntent(intent.IntentHandler):
+ """Handle GetTemperature intents."""
+
+ intent_type = intent.INTENT_GET_TEMPERATURE
+ description = "Gets the current temperature of a climate device or entity"
+ slot_schema = {
+ vol.Optional("area"): intent.non_empty_string,
+ vol.Optional("name"): intent.non_empty_string,
+ vol.Optional("floor"): intent.non_empty_string,
+ vol.Optional("preferred_area_id"): cv.string,
+ vol.Optional("preferred_floor_id"): cv.string,
+ }
+ platforms = {CLIMATE_DOMAIN}
+
+ async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse:
+ """Handle the intent."""
+ hass = intent_obj.hass
+ slots = self.async_validate_slots(intent_obj.slots)
+
+ name: str | None = None
+ if "name" in slots:
+ name = slots["name"]["value"]
+
+ area: str | None = None
+ if "area" in slots:
+ area = slots["area"]["value"]
+
+ floor_name: str | None = None
+ if "floor" in slots:
+ floor_name = slots["floor"]["value"]
+
+ match_preferences = intent.MatchTargetsPreferences(
+ area_id=slots.get("preferred_area_id", {}).get("value"),
+ floor_id=slots.get("preferred_floor_id", {}).get("value"),
+ )
+
+ if (not name) and (area or match_preferences.area_id):
+ # Look for temperature sensors assigned to an area
+ area_registry = ar.async_get(hass)
+ area_temperature_ids: dict[str, str] = {}
+
+ # Keep candidates that are registered as area temperature sensors
+ def area_candidate_filter(
+ candidate: intent.MatchTargetsCandidate,
+ possible_area_ids: Collection[str],
+ ) -> bool:
+ for area_id in possible_area_ids:
+ temperature_id = area_temperature_ids.get(area_id)
+ if (temperature_id is None) and (
+ area_entry := area_registry.async_get_area(area_id)
+ ):
+ temperature_id = area_entry.temperature_entity_id or ""
+ area_temperature_ids[area_id] = temperature_id
+
+ if candidate.state.entity_id == temperature_id:
+ return True
+
+ return False
+
+ match_constraints = intent.MatchTargetsConstraints(
+ area_name=area,
+ floor_name=floor_name,
+ domains=[sensor.DOMAIN],
+ device_classes=[sensor.SensorDeviceClass.TEMPERATURE],
+ assistant=intent_obj.assistant,
+ single_target=True,
+ )
+ match_result = intent.async_match_targets(
+ hass,
+ match_constraints,
+ match_preferences,
+ area_candidate_filter=area_candidate_filter,
+ )
+ if match_result.is_match:
+ # Found temperature sensor
+ response = intent_obj.create_response()
+ response.response_type = intent.IntentResponseType.QUERY_ANSWER
+ response.async_set_states(matched_states=match_result.states)
+ return response
+
+ # Look for climate devices
+ match_constraints = intent.MatchTargetsConstraints(
+ name=name,
+ area_name=area,
+ floor_name=floor_name,
+ domains=[CLIMATE_DOMAIN],
+ assistant=intent_obj.assistant,
+ single_target=True,
+ )
+ match_result = intent.async_match_targets(
+ hass, match_constraints, match_preferences
+ )
+ if not match_result.is_match:
+ raise intent.MatchFailedError(
+ result=match_result, constraints=match_constraints
+ )
+
+ response = intent_obj.create_response()
+ response.response_type = intent.IntentResponseType.QUERY_ANSWER
+ response.async_set_states(matched_states=match_result.states)
+ return response
+
+
async def _async_process_intent(
hass: HomeAssistant, domain: str, platform: IntentPlatformProtocol
) -> None:
diff --git a/homeassistant/components/iometer/__init__.py b/homeassistant/components/iometer/__init__.py
index 5106d449fed..bbf046e70e9 100644
--- a/homeassistant/components/iometer/__init__.py
+++ b/homeassistant/components/iometer/__init__.py
@@ -26,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: IOmeterConfigEntry) -> b
except IOmeterConnectionError as err:
raise ConfigEntryNotReady from err
- coordinator = IOMeterCoordinator(hass, client)
+ coordinator = IOMeterCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/iometer/coordinator.py b/homeassistant/components/iometer/coordinator.py
index 3321b032e4b..4050341151b 100644
--- a/homeassistant/components/iometer/coordinator.py
+++ b/homeassistant/components/iometer/coordinator.py
@@ -8,6 +8,7 @@ from iometer import IOmeterClient, IOmeterConnectionError, Reading, Status
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
+from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN
@@ -31,18 +32,25 @@ class IOMeterCoordinator(DataUpdateCoordinator[IOmeterData]):
config_entry: IOmeterConfigEntry
client: IOmeterClient
+ current_fw_version: str = ""
- def __init__(self, hass: HomeAssistant, client: IOmeterClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: IOmeterConfigEntry,
+ client: IOmeterClient,
+ ) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=DEFAULT_SCAN_INTERVAL,
)
self.client = client
- self.identifier = self.config_entry.entry_id
+ self.identifier = config_entry.entry_id
async def _async_update_data(self) -> IOmeterData:
"""Update data async."""
@@ -52,4 +60,17 @@ class IOMeterCoordinator(DataUpdateCoordinator[IOmeterData]):
except IOmeterConnectionError as error:
raise UpdateFailed(f"Error communicating with IOmeter: {error}") from error
+ fw_version = f"{status.device.core.version}/{status.device.bridge.version}"
+ if self.current_fw_version and fw_version != self.current_fw_version:
+ device_registry = dr.async_get(self.hass)
+ device_entry = device_registry.async_get_device(
+ identifiers={(DOMAIN, status.device.id)}
+ )
+ assert device_entry
+ device_registry.async_update_device(
+ device_entry.id,
+ sw_version=fw_version,
+ )
+ self.current_fw_version = fw_version
+
return IOmeterData(reading=reading, status=status)
diff --git a/homeassistant/components/iometer/entity.py b/homeassistant/components/iometer/entity.py
index 86494857e18..a52ef1c66ed 100644
--- a/homeassistant/components/iometer/entity.py
+++ b/homeassistant/components/iometer/entity.py
@@ -20,5 +20,5 @@ class IOmeterEntity(CoordinatorEntity[IOMeterCoordinator]):
identifiers={(DOMAIN, status.device.id)},
manufacturer="IOmeter GmbH",
model="IOmeter",
- sw_version=f"{status.device.core.version}/{status.device.bridge.version}",
+ sw_version=coordinator.current_fw_version,
)
diff --git a/homeassistant/components/iometer/sensor.py b/homeassistant/components/iometer/sensor.py
index 7d4c1155e8b..3dff3cc6ea9 100644
--- a/homeassistant/components/iometer/sensor.py
+++ b/homeassistant/components/iometer/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import IOMeterCoordinator, IOmeterData
@@ -111,7 +111,7 @@ SENSOR_TYPES: list[IOmeterEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Sensors."""
coordinator: IOMeterCoordinator = config_entry.runtime_data
diff --git a/homeassistant/components/ios/sensor.py b/homeassistant/components/ios/sensor.py
index a97c2145919..a3c9876a884 100644
--- a/homeassistant/components/ios/sensor.py
+++ b/homeassistant/components/ios/sensor.py
@@ -14,7 +14,10 @@ from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.icon import icon_for_battery_level
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -64,7 +67,7 @@ def setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up iOS from a config entry."""
async_add_entities(
diff --git a/homeassistant/components/iotawatt/coordinator.py b/homeassistant/components/iotawatt/coordinator.py
index 4f9ac1f94b7..13802ebdd76 100644
--- a/homeassistant/components/iotawatt/coordinator.py
+++ b/homeassistant/components/iotawatt/coordinator.py
@@ -26,13 +26,14 @@ class IotawattUpdater(DataUpdateCoordinator):
"""Class to manage fetching update data from the IoTaWatt Energy Device."""
api: Iotawatt | None = None
+ config_entry: ConfigEntry
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize IotaWattUpdater object."""
- self.entry = entry
super().__init__(
hass=hass,
logger=_LOGGER,
+ config_entry=entry,
name=entry.title,
update_interval=timedelta(seconds=30),
request_refresh_debouncer=Debouncer(
@@ -57,11 +58,11 @@ class IotawattUpdater(DataUpdateCoordinator):
"""Fetch sensors from IoTaWatt device."""
if self.api is None:
api = Iotawatt(
- self.entry.title,
- self.entry.data[CONF_HOST],
+ self.config_entry.title,
+ self.config_entry.data[CONF_HOST],
httpx_client.get_async_client(self.hass),
- self.entry.data.get(CONF_USERNAME),
- self.entry.data.get(CONF_PASSWORD),
+ self.config_entry.data.get(CONF_USERNAME),
+ self.config_entry.data.get(CONF_PASSWORD),
integratedInterval="d",
includeNonTotalSensors=False,
)
diff --git a/homeassistant/components/iotawatt/sensor.py b/homeassistant/components/iotawatt/sensor.py
index c9af588c160..f5210f7fbba 100644
--- a/homeassistant/components/iotawatt/sensor.py
+++ b/homeassistant/components/iotawatt/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -114,7 +114,7 @@ ENTITY_DESCRIPTION_KEY_MAP: dict[str, IotaWattSensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add sensors for passed config_entry in HA."""
coordinator: IotawattUpdater = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/iotty/__init__.py b/homeassistant/components/iotty/__init__.py
index 804f3f40196..c9eb2639348 100644
--- a/homeassistant/components/iotty/__init__.py
+++ b/homeassistant/components/iotty/__init__.py
@@ -2,12 +2,8 @@
from __future__ import annotations
-from dataclasses import dataclass
import logging
-from iottycloud.device import Device
-
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.config_entry_oauth2_flow import (
@@ -15,22 +11,16 @@ from homeassistant.helpers.config_entry_oauth2_flow import (
async_get_config_entry_implementation,
)
-from . import coordinator
+from .coordinator import (
+ IottyConfigEntry,
+ IottyConfigEntryData,
+ IottyDataUpdateCoordinator,
+)
_LOGGER = logging.getLogger(__name__)
PLATFORMS: list[Platform] = [Platform.COVER, Platform.SWITCH]
-type IottyConfigEntry = ConfigEntry[IottyConfigEntryData]
-
-
-@dataclass
-class IottyConfigEntryData:
- """Contains config entry data for iotty."""
-
- known_devices: set[Device]
- coordinator: coordinator.IottyDataUpdateCoordinator
-
async def async_setup_entry(hass: HomeAssistant, entry: IottyConfigEntry) -> bool:
"""Set up iotty from a config entry."""
@@ -39,9 +29,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: IottyConfigEntry) -> boo
implementation = await async_get_config_entry_implementation(hass, entry)
session = OAuth2Session(hass, entry, implementation)
- data_update_coordinator = coordinator.IottyDataUpdateCoordinator(
- hass, entry, session
- )
+ data_update_coordinator = IottyDataUpdateCoordinator(hass, entry, session)
entry.runtime_data = IottyConfigEntryData(set(), data_update_coordinator)
@@ -51,6 +39,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: IottyConfigEntry) -> boo
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: IottyConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/iotty/coordinator.py b/homeassistant/components/iotty/coordinator.py
index 420248f7724..af870c347bd 100644
--- a/homeassistant/components/iotty/coordinator.py
+++ b/homeassistant/components/iotty/coordinator.py
@@ -32,16 +32,27 @@ class IottyData:
devices: list[Device]
+@dataclass
+class IottyConfigEntryData:
+ """Contains config entry data for iotty."""
+
+ known_devices: set[Device]
+ coordinator: IottyDataUpdateCoordinator
+
+
+type IottyConfigEntry = ConfigEntry[IottyConfigEntryData]
+
+
class IottyDataUpdateCoordinator(DataUpdateCoordinator[IottyData]):
"""Class to manage fetching Iotty data."""
- config_entry: ConfigEntry
+ config_entry: IottyConfigEntry
_entities: dict[str, Entity]
_devices: list[Device]
_device_registry: dr.DeviceRegistry
def __init__(
- self, hass: HomeAssistant, entry: ConfigEntry, session: OAuth2Session
+ self, hass: HomeAssistant, entry: IottyConfigEntry, session: OAuth2Session
) -> None:
"""Initialize the coordinator."""
_LOGGER.debug("Initializing iotty data update coordinator")
@@ -49,11 +60,11 @@ class IottyDataUpdateCoordinator(DataUpdateCoordinator[IottyData]):
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=f"{DOMAIN}_coordinator",
update_interval=UPDATE_INTERVAL,
)
- self.config_entry = entry
self._entities = {}
self._devices = []
self.iotty = api.IottyProxy(
diff --git a/homeassistant/components/iotty/cover.py b/homeassistant/components/iotty/cover.py
index 50a4a1deeba..d8b11131f4f 100644
--- a/homeassistant/components/iotty/cover.py
+++ b/homeassistant/components/iotty/cover.py
@@ -16,11 +16,10 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import IottyConfigEntry
from .api import IottyProxy
-from .coordinator import IottyDataUpdateCoordinator
+from .coordinator import IottyConfigEntry, IottyDataUpdateCoordinator
from .entity import IottyEntity
_LOGGER = logging.getLogger(__name__)
@@ -29,7 +28,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: IottyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Activate the iotty Shutter component."""
_LOGGER.debug("Setup COVER entry id is %s", config_entry.entry_id)
diff --git a/homeassistant/components/iotty/switch.py b/homeassistant/components/iotty/switch.py
index b06e3ea308d..113a4439e85 100644
--- a/homeassistant/components/iotty/switch.py
+++ b/homeassistant/components/iotty/switch.py
@@ -20,11 +20,10 @@ from homeassistant.components.switch import (
SwitchEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import IottyConfigEntry
from .api import IottyProxy
-from .coordinator import IottyDataUpdateCoordinator
+from .coordinator import IottyConfigEntry, IottyDataUpdateCoordinator
from .entity import IottyEntity
_LOGGER = logging.getLogger(__name__)
@@ -46,7 +45,7 @@ ENTITIES: dict[str, SwitchEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: IottyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Activate the iotty Switch component."""
_LOGGER.debug("Setup SWITCH entry id is %s", config_entry.entry_id)
diff --git a/homeassistant/components/ipma/manifest.json b/homeassistant/components/ipma/manifest.json
index 1abd7807213..971525e013f 100644
--- a/homeassistant/components/ipma/manifest.json
+++ b/homeassistant/components/ipma/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/ipma",
"iot_class": "cloud_polling",
"loggers": ["geopy", "pyipma"],
- "requirements": ["pyipma==3.0.8"]
+ "requirements": ["pyipma==3.0.9"]
}
diff --git a/homeassistant/components/ipma/sensor.py b/homeassistant/components/ipma/sensor.py
index 2a921cdbb04..78fd018cf9a 100644
--- a/homeassistant/components/ipma/sensor.py
+++ b/homeassistant/components/ipma/sensor.py
@@ -16,7 +16,7 @@ from pyipma.uv import UV
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle
from .const import DATA_API, DATA_LOCATION, DOMAIN, MIN_TIME_BETWEEN_UPDATES
@@ -86,7 +86,9 @@ SENSOR_TYPES: tuple[IPMASensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the IPMA sensor platform."""
api = hass.data[DOMAIN][entry.entry_id][DATA_API]
diff --git a/homeassistant/components/ipma/weather.py b/homeassistant/components/ipma/weather.py
index 855587eee2e..d285f9e1ad3 100644
--- a/homeassistant/components/ipma/weather.py
+++ b/homeassistant/components/ipma/weather.py
@@ -31,7 +31,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sun import is_up
from homeassistant.util import Throttle
@@ -51,7 +51,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
api = hass.data[DOMAIN][config_entry.entry_id][DATA_API]
diff --git a/homeassistant/components/ipp/__init__.py b/homeassistant/components/ipp/__init__.py
index 0a94795613b..99332dca0e2 100644
--- a/homeassistant/components/ipp/__init__.py
+++ b/homeassistant/components/ipp/__init__.py
@@ -2,39 +2,17 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import (
- CONF_HOST,
- CONF_PORT,
- CONF_SSL,
- CONF_VERIFY_SSL,
- Platform,
-)
+from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .const import CONF_BASE_PATH
-from .coordinator import IPPDataUpdateCoordinator
+from .coordinator import IPPConfigEntry, IPPDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
-type IPPConfigEntry = ConfigEntry[IPPDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: IPPConfigEntry) -> bool:
"""Set up IPP from a config entry."""
- # config flow sets this to either UUID, serial number or None
- if (device_id := entry.unique_id) is None:
- device_id = entry.entry_id
-
- coordinator = IPPDataUpdateCoordinator(
- hass,
- host=entry.data[CONF_HOST],
- port=entry.data[CONF_PORT],
- base_path=entry.data[CONF_BASE_PATH],
- tls=entry.data[CONF_SSL],
- verify_ssl=entry.data[CONF_VERIFY_SSL],
- device_id=device_id,
- )
+ coordinator = IPPDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
@@ -44,6 +22,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: IPPConfigEntry) -> bool:
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: IPPConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/ipp/coordinator.py b/homeassistant/components/ipp/coordinator.py
index 535b18bcaf0..1c3dc4d0a03 100644
--- a/homeassistant/components/ipp/coordinator.py
+++ b/homeassistant/components/ipp/coordinator.py
@@ -7,45 +7,42 @@ import logging
from pyipp import IPP, IPPError, Printer as IPPPrinter
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, CONF_PORT, CONF_SSL, CONF_VERIFY_SSL
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import DOMAIN
+from .const import CONF_BASE_PATH, DOMAIN
SCAN_INTERVAL = timedelta(seconds=60)
_LOGGER = logging.getLogger(__name__)
+type IPPConfigEntry = ConfigEntry[IPPDataUpdateCoordinator]
+
class IPPDataUpdateCoordinator(DataUpdateCoordinator[IPPPrinter]):
"""Class to manage fetching IPP data from single endpoint."""
- def __init__(
- self,
- hass: HomeAssistant,
- *,
- host: str,
- port: int,
- base_path: str,
- tls: bool,
- verify_ssl: bool,
- device_id: str,
- ) -> None:
+ config_entry: IPPConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: IPPConfigEntry) -> None:
"""Initialize global IPP data updater."""
- self.device_id = device_id
+ self.device_id = config_entry.unique_id or config_entry.entry_id
self.ipp = IPP(
- host=host,
- port=port,
- base_path=base_path,
- tls=tls,
- verify_ssl=verify_ssl,
- session=async_get_clientsession(hass, verify_ssl),
+ host=config_entry.data[CONF_HOST],
+ port=config_entry.data[CONF_PORT],
+ base_path=config_entry.data[CONF_BASE_PATH],
+ tls=config_entry.data[CONF_SSL],
+ verify_ssl=config_entry.data[CONF_VERIFY_SSL],
+ session=async_get_clientsession(hass, config_entry.data[CONF_VERIFY_SSL]),
)
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/ipp/diagnostics.py b/homeassistant/components/ipp/diagnostics.py
index 9b10dc68966..cd136e78373 100644
--- a/homeassistant/components/ipp/diagnostics.py
+++ b/homeassistant/components/ipp/diagnostics.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.core import HomeAssistant
-from . import IPPConfigEntry
+from .coordinator import IPPConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/ipp/sensor.py b/homeassistant/components/ipp/sensor.py
index a2792c7749b..e16819a54ff 100644
--- a/homeassistant/components/ipp/sensor.py
+++ b/homeassistant/components/ipp/sensor.py
@@ -17,10 +17,9 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import ATTR_LOCATION, PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import IPPConfigEntry
from .const import (
ATTR_COMMAND_SET,
ATTR_INFO,
@@ -32,6 +31,7 @@ from .const import (
ATTR_STATE_REASON,
ATTR_URI_SUPPORTED,
)
+from .coordinator import IPPConfigEntry
from .entity import IPPEntity
@@ -87,7 +87,7 @@ PRINTER_SENSORS: tuple[IPPSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: IPPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up IPP sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/iqvia/sensor.py b/homeassistant/components/iqvia/sensor.py
index d04e0885454..64492c634e9 100644
--- a/homeassistant/components/iqvia/sensor.py
+++ b/homeassistant/components/iqvia/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_STATE
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
@@ -127,7 +127,9 @@ INDEX_SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up IQVIA sensors based on a config entry."""
sensors: list[ForecastSensor | IndexSensor] = [
diff --git a/homeassistant/components/iqvia/strings.json b/homeassistant/components/iqvia/strings.json
index 5dc0dea53d5..a0697a6c210 100644
--- a/homeassistant/components/iqvia/strings.json
+++ b/homeassistant/components/iqvia/strings.json
@@ -4,7 +4,7 @@
"user": {
"description": "Fill out your U.S. or Canadian ZIP code.",
"data": {
- "zip_code": "ZIP Code"
+ "zip_code": "ZIP code"
}
}
},
diff --git a/homeassistant/components/iron_os/__init__.py b/homeassistant/components/iron_os/__init__.py
index 6af6abb1436..77099e48b41 100644
--- a/homeassistant/components/iron_os/__init__.py
+++ b/homeassistant/components/iron_os/__init__.py
@@ -8,7 +8,6 @@ from typing import TYPE_CHECKING
from pynecil import IronOSUpdate, Pynecil
from homeassistant.components import bluetooth
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@@ -19,6 +18,7 @@ from homeassistant.util.hass_dict import HassKey
from .const import DOMAIN
from .coordinator import (
+ IronOSConfigEntry,
IronOSCoordinators,
IronOSFirmwareUpdateCoordinator,
IronOSLiveDataCoordinator,
@@ -39,7 +39,6 @@ PLATFORMS: list[Platform] = [
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
-type IronOSConfigEntry = ConfigEntry[IronOSCoordinators]
IRON_OS_KEY: HassKey[IronOSFirmwareUpdateCoordinator] = HassKey(DOMAIN)
@@ -73,10 +72,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bo
device = Pynecil(ble_device)
- live_data = IronOSLiveDataCoordinator(hass, device)
+ live_data = IronOSLiveDataCoordinator(hass, entry, device)
await live_data.async_config_entry_first_refresh()
- settings = IronOSSettingsCoordinator(hass, device)
+ settings = IronOSSettingsCoordinator(hass, entry, device)
await settings.async_config_entry_first_refresh()
entry.runtime_data = IronOSCoordinators(
diff --git a/homeassistant/components/iron_os/binary_sensor.py b/homeassistant/components/iron_os/binary_sensor.py
index 81ba0e08c95..66e642c7aaa 100644
--- a/homeassistant/components/iron_os/binary_sensor.py
+++ b/homeassistant/components/iron_os/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IronOSConfigEntry
from .coordinator import IronOSLiveDataCoordinator
@@ -29,7 +29,7 @@ class PinecilBinarySensor(StrEnum):
async def async_setup_entry(
hass: HomeAssistant,
entry: IronOSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors from a config entry."""
coordinator = entry.runtime_data.live_data
diff --git a/homeassistant/components/iron_os/button.py b/homeassistant/components/iron_os/button.py
index be16148a656..e069ddb1d9f 100644
--- a/homeassistant/components/iron_os/button.py
+++ b/homeassistant/components/iron_os/button.py
@@ -10,7 +10,7 @@ from pynecil import CharSetting
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IronOSConfigEntry
from .coordinator import IronOSCoordinators
@@ -53,7 +53,7 @@ BUTTON_DESCRIPTIONS: tuple[IronOSButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: IronOSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button entities from a config entry."""
coordinators = entry.runtime_data
diff --git a/homeassistant/components/iron_os/config_flow.py b/homeassistant/components/iron_os/config_flow.py
index 444db79c926..8509577114f 100644
--- a/homeassistant/components/iron_os/config_flow.py
+++ b/homeassistant/components/iron_os/config_flow.py
@@ -61,7 +61,7 @@ class IronOSConfigFlow(ConfigFlow, domain=DOMAIN):
self._abort_if_unique_id_configured()
return self.async_create_entry(title=title, data={})
- current_addresses = self._async_current_ids()
+ current_addresses = self._async_current_ids(include_ignore=False)
for discovery_info in async_discovered_service_info(self.hass, True):
address = discovery_info.address
if (
diff --git a/homeassistant/components/iron_os/coordinator.py b/homeassistant/components/iron_os/coordinator.py
index 080fee20762..84c9b895766 100644
--- a/homeassistant/components/iron_os/coordinator.py
+++ b/homeassistant/components/iron_os/coordinator.py
@@ -8,6 +8,7 @@ from enum import Enum
import logging
from typing import cast
+from awesomeversion import AwesomeVersion
from pynecil import (
CharSetting,
CommunicationError,
@@ -34,6 +35,8 @@ SCAN_INTERVAL = timedelta(seconds=5)
SCAN_INTERVAL_GITHUB = timedelta(hours=3)
SCAN_INTERVAL_SETTINGS = timedelta(seconds=60)
+V223 = AwesomeVersion("v2.23")
+
@dataclass
class IronOSCoordinators:
@@ -43,15 +46,19 @@ class IronOSCoordinators:
settings: IronOSSettingsCoordinator
+type IronOSConfigEntry = ConfigEntry[IronOSCoordinators]
+
+
class IronOSBaseCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
"""IronOS base coordinator."""
device_info: DeviceInfoResponse
- config_entry: ConfigEntry
+ config_entry: IronOSConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: IronOSConfigEntry,
device: Pynecil,
update_interval: timedelta,
) -> None:
@@ -60,6 +67,7 @@ class IronOSBaseCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=update_interval,
request_refresh_debouncer=Debouncer(
@@ -67,6 +75,7 @@ class IronOSBaseCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
),
)
self.device = device
+ self.v223_features = False
async def _async_setup(self) -> None:
"""Set up the coordinator."""
@@ -76,13 +85,17 @@ class IronOSBaseCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
except CommunicationError as e:
raise UpdateFailed("Cannot connect to device") from e
+ self.v223_features = AwesomeVersion(self.device_info.build) >= V223
+
class IronOSLiveDataCoordinator(IronOSBaseCoordinator[LiveDataResponse]):
"""IronOS coordinator."""
- def __init__(self, hass: HomeAssistant, device: Pynecil) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: IronOSConfigEntry, device: Pynecil
+ ) -> None:
"""Initialize IronOS coordinator."""
- super().__init__(hass, device=device, update_interval=SCAN_INTERVAL)
+ super().__init__(hass, config_entry, device, SCAN_INTERVAL)
async def _async_update_data(self) -> LiveDataResponse:
"""Fetch data from Device."""
@@ -109,35 +122,14 @@ class IronOSLiveDataCoordinator(IronOSBaseCoordinator[LiveDataResponse]):
return False
-class IronOSFirmwareUpdateCoordinator(DataUpdateCoordinator[LatestRelease]):
- """IronOS coordinator for retrieving update information from github."""
-
- def __init__(self, hass: HomeAssistant, github: IronOSUpdate) -> None:
- """Initialize IronOS coordinator."""
- super().__init__(
- hass,
- _LOGGER,
- config_entry=None,
- name=DOMAIN,
- update_interval=SCAN_INTERVAL_GITHUB,
- )
- self.github = github
-
- async def _async_update_data(self) -> LatestRelease:
- """Fetch data from Github."""
-
- try:
- return await self.github.latest_release()
- except UpdateException as e:
- raise UpdateFailed("Failed to check for latest IronOS update") from e
-
-
class IronOSSettingsCoordinator(IronOSBaseCoordinator[SettingsDataResponse]):
"""IronOS coordinator."""
- def __init__(self, hass: HomeAssistant, device: Pynecil) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: IronOSConfigEntry, device: Pynecil
+ ) -> None:
"""Initialize IronOS coordinator."""
- super().__init__(hass, device=device, update_interval=SCAN_INTERVAL_SETTINGS)
+ super().__init__(hass, config_entry, device, SCAN_INTERVAL_SETTINGS)
async def _async_update_data(self) -> SettingsDataResponse:
"""Fetch data from Device."""
@@ -173,3 +165,26 @@ class IronOSSettingsCoordinator(IronOSBaseCoordinator[SettingsDataResponse]):
)
self.async_update_listeners()
await self.async_request_refresh()
+
+
+class IronOSFirmwareUpdateCoordinator(DataUpdateCoordinator[LatestRelease]):
+ """IronOS coordinator for retrieving update information from github."""
+
+ def __init__(self, hass: HomeAssistant, github: IronOSUpdate) -> None:
+ """Initialize IronOS coordinator."""
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=None,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL_GITHUB,
+ )
+ self.github = github
+
+ async def _async_update_data(self) -> LatestRelease:
+ """Fetch data from Github."""
+
+ try:
+ return await self.github.latest_release()
+ except UpdateException as e:
+ raise UpdateFailed("Failed to check for latest IronOS update") from e
diff --git a/homeassistant/components/iron_os/icons.json b/homeassistant/components/iron_os/icons.json
index 6410c561b9d..695b9d16849 100644
--- a/homeassistant/components/iron_os/icons.json
+++ b/homeassistant/components/iron_os/icons.json
@@ -73,6 +73,9 @@
},
"power_limit": {
"default": "mdi:flash-alert"
+ },
+ "hall_effect_sleep_time": {
+ "default": "mdi:timer-sand"
}
},
"select": {
@@ -105,6 +108,9 @@
},
"usb_pd_mode": {
"default": "mdi:meter-electric-outline"
+ },
+ "tip_type": {
+ "default": "mdi:pencil-outline"
}
},
"sensor": {
@@ -154,7 +160,16 @@
"soldering": "mdi:soldering-iron",
"sleeping": "mdi:sleep",
"settings": "mdi:menu-open",
- "debug": "mdi:bug-play"
+ "debug": "mdi:bug-play",
+ "soldering_profile": "mdi:chart-box-outline",
+ "temperature_adjust": "mdi:thermostat-box",
+ "usb_pd_debug": "mdi:bug-play",
+ "thermal_runaway": "mdi:fire-alert",
+ "startup_logo": "mdi:dots-circle",
+ "cjc_calibration": "mdi:tune-vertical",
+ "startup_warnings": "mdi:alert",
+ "initialisation_done": "mdi:check-circle",
+ "hibernating": "mdi:sleep"
}
},
"estimated_power": {
diff --git a/homeassistant/components/iron_os/manifest.json b/homeassistant/components/iron_os/manifest.json
index 462e75c5b6e..58cbdaa3bc6 100644
--- a/homeassistant/components/iron_os/manifest.json
+++ b/homeassistant/components/iron_os/manifest.json
@@ -13,5 +13,6 @@
"documentation": "https://www.home-assistant.io/integrations/iron_os",
"iot_class": "local_polling",
"loggers": ["pynecil"],
- "requirements": ["pynecil==4.0.1"]
+ "quality_scale": "platinum",
+ "requirements": ["pynecil==4.1.0"]
}
diff --git a/homeassistant/components/iron_os/number.py b/homeassistant/components/iron_os/number.py
index 518c11372c4..6ad5947cb6f 100644
--- a/homeassistant/components/iron_os/number.py
+++ b/homeassistant/components/iron_os/number.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IronOSConfigEntry
from .const import MAX_TEMP, MIN_TEMP
@@ -65,6 +65,7 @@ class PinecilNumber(StrEnum):
VOLTAGE_DIV = "voltage_div"
TEMP_INCREMENT_SHORT = "temp_increment_short"
TEMP_INCREMENT_LONG = "temp_increment_long"
+ HALL_EFFECT_SLEEP_TIME = "hall_effect_sleep_time"
def multiply(value: float | None, multiplier: float) -> float | None:
@@ -323,18 +324,38 @@ PINECIL_NUMBER_DESCRIPTIONS: tuple[IronOSNumberEntityDescription, ...] = (
),
)
+PINECIL_NUMBER_DESCRIPTIONS_V223: tuple[IronOSNumberEntityDescription, ...] = (
+ IronOSNumberEntityDescription(
+ key=PinecilNumber.HALL_EFFECT_SLEEP_TIME,
+ translation_key=PinecilNumber.HALL_EFFECT_SLEEP_TIME,
+ value_fn=(lambda _, settings: settings.get("hall_sleep_time")),
+ characteristic=CharSetting.HALL_SLEEP_TIME,
+ raw_value_fn=lambda value: value,
+ mode=NumberMode.BOX,
+ native_min_value=0,
+ native_max_value=60,
+ native_step=5,
+ entity_category=EntityCategory.CONFIG,
+ native_unit_of_measurement=UnitOfTime.SECONDS,
+ entity_registry_enabled_default=False,
+ ),
+)
+
async def async_setup_entry(
hass: HomeAssistant,
entry: IronOSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number entities from a config entry."""
coordinators = entry.runtime_data
+ descriptions = PINECIL_NUMBER_DESCRIPTIONS
+
+ if coordinators.live_data.v223_features:
+ descriptions += PINECIL_NUMBER_DESCRIPTIONS_V223
async_add_entities(
- IronOSNumberEntity(coordinators, description)
- for description in PINECIL_NUMBER_DESCRIPTIONS
+ IronOSNumberEntity(coordinators, description) for description in descriptions
)
diff --git a/homeassistant/components/iron_os/quality_scale.yaml b/homeassistant/components/iron_os/quality_scale.yaml
index c80b8b5adfe..8f7eb5ff36a 100644
--- a/homeassistant/components/iron_os/quality_scale.yaml
+++ b/homeassistant/components/iron_os/quality_scale.yaml
@@ -21,8 +21,10 @@ rules:
entity-unique-id: done
has-entity-name: done
runtime-data: done
- test-before-configure: todo
- test-before-setup: todo
+ test-before-configure:
+ status: exempt
+ comment: Device is set up from a Bluetooth discovery
+ test-before-setup: done
unique-config-entry: done
# Silver
@@ -70,7 +72,9 @@ rules:
repair-issues:
status: exempt
comment: no repairs/issues
- stale-devices: todo
+ stale-devices:
+ status: exempt
+ comment: Stale devices are removed with the config entry as there is only one device per entry
# Platinum
async-dependency: done
diff --git a/homeassistant/components/iron_os/select.py b/homeassistant/components/iron_os/select.py
index e9c7f81c208..32652829531 100644
--- a/homeassistant/components/iron_os/select.py
+++ b/homeassistant/components/iron_os/select.py
@@ -17,13 +17,14 @@ from pynecil import (
ScrollSpeed,
SettingsDataResponse,
TempUnit,
+ TipType,
USBPDMode,
)
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IronOSConfigEntry
from .coordinator import IronOSCoordinators
@@ -53,6 +54,7 @@ class PinecilSelect(StrEnum):
LOCKING_MODE = "locking_mode"
LOGO_DURATION = "logo_duration"
USB_PD_MODE = "usb_pd_mode"
+ TIP_TYPE = "tip_type"
def enum_to_str(enum: Enum | None) -> str | None:
@@ -138,6 +140,8 @@ PINECIL_SELECT_DESCRIPTIONS: tuple[IronOSSelectEntityDescription, ...] = (
entity_category=EntityCategory.CONFIG,
entity_registry_enabled_default=False,
),
+)
+PINECIL_SELECT_DESCRIPTIONS_V222: tuple[IronOSSelectEntityDescription, ...] = (
IronOSSelectEntityDescription(
key=PinecilSelect.USB_PD_MODE,
translation_key=PinecilSelect.USB_PD_MODE,
@@ -149,19 +153,46 @@ PINECIL_SELECT_DESCRIPTIONS: tuple[IronOSSelectEntityDescription, ...] = (
entity_registry_enabled_default=False,
),
)
+PINECIL_SELECT_DESCRIPTIONS_V223: tuple[IronOSSelectEntityDescription, ...] = (
+ IronOSSelectEntityDescription(
+ key=PinecilSelect.USB_PD_MODE,
+ translation_key=PinecilSelect.USB_PD_MODE,
+ characteristic=CharSetting.USB_PD_MODE,
+ value_fn=lambda x: enum_to_str(x.get("usb_pd_mode")),
+ raw_value_fn=lambda value: USBPDMode[value.upper()],
+ options=[x.name.lower() for x in USBPDMode],
+ entity_category=EntityCategory.CONFIG,
+ entity_registry_enabled_default=False,
+ ),
+ IronOSSelectEntityDescription(
+ key=PinecilSelect.TIP_TYPE,
+ translation_key=PinecilSelect.TIP_TYPE,
+ characteristic=CharSetting.TIP_TYPE,
+ value_fn=lambda x: enum_to_str(x.get("tip_type")),
+ raw_value_fn=lambda value: TipType[value.upper()],
+ options=[x.name.lower() for x in TipType],
+ entity_category=EntityCategory.CONFIG,
+ ),
+)
async def async_setup_entry(
hass: HomeAssistant,
entry: IronOSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select entities from a config entry."""
- coordinator = entry.runtime_data
+ coordinators = entry.runtime_data
+ descriptions = PINECIL_SELECT_DESCRIPTIONS
+
+ descriptions += (
+ PINECIL_SELECT_DESCRIPTIONS_V223
+ if coordinators.live_data.v223_features
+ else PINECIL_SELECT_DESCRIPTIONS_V222
+ )
async_add_entities(
- IronOSSelectEntity(coordinator, description)
- for description in PINECIL_SELECT_DESCRIPTIONS
+ IronOSSelectEntity(coordinators, description) for description in descriptions
)
diff --git a/homeassistant/components/iron_os/sensor.py b/homeassistant/components/iron_os/sensor.py
index d178b46723f..79f1e54a6f4 100644
--- a/homeassistant/components/iron_os/sensor.py
+++ b/homeassistant/components/iron_os/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import IronOSConfigEntry
@@ -184,7 +184,7 @@ PINECIL_SENSOR_DESCRIPTIONS: tuple[IronOSSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: IronOSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors from a config entry."""
coordinator = entry.runtime_data.live_data
diff --git a/homeassistant/components/iron_os/strings.json b/homeassistant/components/iron_os/strings.json
index 60168699427..ddae9a3020f 100644
--- a/homeassistant/components/iron_os/strings.json
+++ b/homeassistant/components/iron_os/strings.json
@@ -94,6 +94,9 @@
},
"temp_increment_long": {
"name": "Long-press temperature step"
+ },
+ "hall_effect_sleep_time": {
+ "name": "Hall sensor sleep timeout"
}
},
"select": {
@@ -173,6 +176,15 @@
"off": "[%key:common::state::off%]",
"on": "[%key:common::state::on%]"
}
+ },
+ "tip_type": {
+ "name": "Soldering tip type",
+ "state": {
+ "auto": "Auto sense",
+ "ts100_long": "TS100 long/Hakko T12 tip",
+ "pine_short": "Pinecil short tip",
+ "pts200": "PTS200 short tip"
+ }
}
},
"sensor": {
@@ -223,7 +235,16 @@
"sleeping": "Sleeping",
"settings": "Settings",
"debug": "Debug",
- "boost": "Boost"
+ "boost": "Boost",
+ "soldering_profile": "Soldering profile",
+ "temperature_adjust": "Temperature adjust",
+ "usb_pd_debug": "USB PD debug",
+ "thermal_runaway": "Thermal runaway",
+ "startup_logo": "Booting",
+ "cjc_calibration": "CJC calibration",
+ "startup_warnings": "Startup warnings",
+ "initialisation_done": "Initialisation done",
+ "hibernating": "Hibernating"
}
},
"estimated_power": {
diff --git a/homeassistant/components/iron_os/switch.py b/homeassistant/components/iron_os/switch.py
index d88e8cfdcb5..124b670048a 100644
--- a/homeassistant/components/iron_os/switch.py
+++ b/homeassistant/components/iron_os/switch.py
@@ -12,7 +12,7 @@ from pynecil import CharSetting, SettingsDataResponse
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IronOSConfigEntry
from .coordinator import IronOSCoordinators
@@ -100,7 +100,7 @@ SWITCH_DESCRIPTIONS: tuple[IronOSSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: IronOSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches from a config entry."""
diff --git a/homeassistant/components/iron_os/update.py b/homeassistant/components/iron_os/update.py
index b431d321f24..4ec626ffc2a 100644
--- a/homeassistant/components/iron_os/update.py
+++ b/homeassistant/components/iron_os/update.py
@@ -9,7 +9,7 @@ from homeassistant.components.update import (
UpdateEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IRON_OS_KEY, IronOSConfigEntry, IronOSLiveDataCoordinator
from .coordinator import IronOSFirmwareUpdateCoordinator
@@ -26,7 +26,7 @@ UPDATE_DESCRIPTION = UpdateEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: IronOSConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up IronOS update platform."""
diff --git a/homeassistant/components/iskra/__init__.py b/homeassistant/components/iskra/__init__.py
index b841da9df26..21c60db20fe 100644
--- a/homeassistant/components/iskra/__init__.py
+++ b/homeassistant/components/iskra/__init__.py
@@ -6,7 +6,6 @@ from pyiskra.adapters import Modbus, RestAPI
from pyiskra.devices import Device
from pyiskra.exceptions import DeviceConnectionError, DeviceNotSupported, NotAuthorised
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_ADDRESS,
CONF_HOST,
@@ -21,14 +20,11 @@ from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
from .const import DOMAIN, MANUFACTURER
-from .coordinator import IskraDataUpdateCoordinator
+from .coordinator import IskraConfigEntry, IskraDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type IskraConfigEntry = ConfigEntry[list[IskraDataUpdateCoordinator]]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: IskraConfigEntry) -> bool:
"""Set up iskra device from a config entry."""
conf = entry.data
@@ -79,11 +75,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: IskraConfigEntry) -> boo
)
coordinators = [
- IskraDataUpdateCoordinator(hass, child_device)
+ IskraDataUpdateCoordinator(hass, entry, child_device)
for child_device in base_device.get_child_devices()
]
else:
- coordinators = [IskraDataUpdateCoordinator(hass, base_device)]
+ coordinators = [IskraDataUpdateCoordinator(hass, entry, base_device)]
for coordinator in coordinators:
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/iskra/coordinator.py b/homeassistant/components/iskra/coordinator.py
index 175d8ed4c86..d476556e96d 100644
--- a/homeassistant/components/iskra/coordinator.py
+++ b/homeassistant/components/iskra/coordinator.py
@@ -11,6 +11,7 @@ from pyiskra.exceptions import (
NotAuthorised,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -18,21 +19,26 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
+type IskraConfigEntry = ConfigEntry[list[IskraDataUpdateCoordinator]]
+
class IskraDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Class to manage fetching Iskra data."""
- def __init__(self, hass: HomeAssistant, device: Device) -> None:
+ config_entry: IskraConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: IskraConfigEntry, device: Device
+ ) -> None:
"""Initialize."""
self.device = device
- update_interval = timedelta(seconds=60)
-
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
- update_interval=update_interval,
+ update_interval=timedelta(seconds=60),
)
async def _async_update_data(self) -> None:
diff --git a/homeassistant/components/iskra/sensor.py b/homeassistant/components/iskra/sensor.py
index df9e3ec53f9..10aa5555249 100644
--- a/homeassistant/components/iskra/sensor.py
+++ b/homeassistant/components/iskra/sensor.py
@@ -24,9 +24,8 @@ from homeassistant.const import (
UnitOfReactivePower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import IskraConfigEntry
from .const import (
ATTR_FREQUENCY,
ATTR_NON_RESETTABLE_COUNTER,
@@ -44,7 +43,7 @@ from .const import (
ATTR_TOTAL_APPARENT_POWER,
ATTR_TOTAL_REACTIVE_POWER,
)
-from .coordinator import IskraDataUpdateCoordinator
+from .coordinator import IskraConfigEntry, IskraDataUpdateCoordinator
from .entity import IskraEntity
@@ -208,7 +207,7 @@ def get_counter_entity_description(
async def async_setup_entry(
hass: HomeAssistant,
entry: IskraConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Iskra sensors based on config_entry."""
diff --git a/homeassistant/components/islamic_prayer_times/__init__.py b/homeassistant/components/islamic_prayer_times/__init__.py
index d61eba343ac..731d1324c71 100644
--- a/homeassistant/components/islamic_prayer_times/__init__.py
+++ b/homeassistant/components/islamic_prayer_times/__init__.py
@@ -4,20 +4,20 @@ from __future__ import annotations
import logging
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from .coordinator import IslamicPrayerDataUpdateCoordinator
+from .coordinator import (
+ IslamicPrayerDataUpdateCoordinator,
+ IslamicPrayerTimesConfigEntry,
+)
PLATFORMS = [Platform.SENSOR]
_LOGGER = logging.getLogger(__name__)
-type IslamicPrayerTimesConfigEntry = ConfigEntry[IslamicPrayerDataUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, config_entry: IslamicPrayerTimesConfigEntry
@@ -36,7 +36,7 @@ async def async_setup_entry(
await er.async_migrate_entries(hass, config_entry.entry_id, update_unique_id)
- coordinator = IslamicPrayerDataUpdateCoordinator(hass)
+ coordinator = IslamicPrayerDataUpdateCoordinator(hass, config_entry)
await coordinator.async_config_entry_first_refresh()
config_entry.runtime_data = coordinator
@@ -48,7 +48,9 @@ async def async_setup_entry(
return True
-async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, config_entry: IslamicPrayerTimesConfigEntry
+) -> bool:
"""Migrate old entry."""
_LOGGER.debug("Migrating from version %s", config_entry.version)
diff --git a/homeassistant/components/islamic_prayer_times/coordinator.py b/homeassistant/components/islamic_prayer_times/coordinator.py
index 35903afa393..a6cd3fb151e 100644
--- a/homeassistant/components/islamic_prayer_times/coordinator.py
+++ b/homeassistant/components/islamic_prayer_times/coordinator.py
@@ -29,21 +29,26 @@ from .const import (
_LOGGER = logging.getLogger(__name__)
+type IslamicPrayerTimesConfigEntry = ConfigEntry[IslamicPrayerDataUpdateCoordinator]
+
class IslamicPrayerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, datetime]]):
"""Islamic Prayer Client Object."""
- config_entry: ConfigEntry
+ config_entry: IslamicPrayerTimesConfigEntry
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: IslamicPrayerTimesConfigEntry
+ ) -> None:
"""Initialize the Islamic Prayer client."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
)
- self.latitude = self.config_entry.data[CONF_LATITUDE]
- self.longitude = self.config_entry.data[CONF_LONGITUDE]
+ self.latitude = config_entry.data[CONF_LATITUDE]
+ self.longitude = config_entry.data[CONF_LONGITUDE]
self.event_unsub: CALLBACK_TYPE | None = None
@property
diff --git a/homeassistant/components/islamic_prayer_times/sensor.py b/homeassistant/components/islamic_prayer_times/sensor.py
index c46b629d2d8..3d35786186e 100644
--- a/homeassistant/components/islamic_prayer_times/sensor.py
+++ b/homeassistant/components/islamic_prayer_times/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import IslamicPrayerTimesConfigEntry
@@ -51,7 +51,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: IslamicPrayerTimesConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Islamic prayer times sensor platform."""
diff --git a/homeassistant/components/israel_rail/__init__.py b/homeassistant/components/israel_rail/__init__.py
index 3c33a159a63..ed800f559d4 100644
--- a/homeassistant/components/israel_rail/__init__.py
+++ b/homeassistant/components/israel_rail/__init__.py
@@ -4,13 +4,12 @@ import logging
from israelrailapi import TrainSchedule
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from .const import CONF_DESTINATION, CONF_START, DOMAIN
-from .coordinator import IsraelRailDataUpdateCoordinator
+from .coordinator import IsraelRailConfigEntry, IsraelRailDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -18,9 +17,6 @@ _LOGGER = logging.getLogger(__name__)
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type IsraelRailConfigEntry = ConfigEntry[IsraelRailDataUpdateCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: IsraelRailConfigEntry) -> bool:
"""Set up Israel rail from a config entry."""
config = entry.data
@@ -43,7 +39,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: IsraelRailConfigEntry) -
) from e
israel_rail_coordinator = IsraelRailDataUpdateCoordinator(
- hass, train_schedule, start, destination
+ hass, entry, train_schedule, start, destination
)
await israel_rail_coordinator.async_config_entry_first_refresh()
entry.runtime_data = israel_rail_coordinator
diff --git a/homeassistant/components/israel_rail/coordinator.py b/homeassistant/components/israel_rail/coordinator.py
index b022e3fd790..190ed938790 100644
--- a/homeassistant/components/israel_rail/coordinator.py
+++ b/homeassistant/components/israel_rail/coordinator.py
@@ -38,14 +38,18 @@ def departure_time(train_route: TrainRoute) -> datetime | None:
return start_datetime.astimezone() if start_datetime else None
+type IsraelRailConfigEntry = ConfigEntry[IsraelRailDataUpdateCoordinator]
+
+
class IsraelRailDataUpdateCoordinator(DataUpdateCoordinator[list[DataConnection]]):
"""A IsraelRail Data Update Coordinator."""
- config_entry: ConfigEntry
+ config_entry: IsraelRailConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: IsraelRailConfigEntry,
train_schedule: TrainSchedule,
start: str,
destination: str,
@@ -54,6 +58,7 @@ class IsraelRailDataUpdateCoordinator(DataUpdateCoordinator[list[DataConnection]
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=DEFAULT_SCAN_INTERVAL,
)
diff --git a/homeassistant/components/israel_rail/sensor.py b/homeassistant/components/israel_rail/sensor.py
index 132a9a74826..86b2e135cfb 100644
--- a/homeassistant/components/israel_rail/sensor.py
+++ b/homeassistant/components/israel_rail/sensor.py
@@ -15,13 +15,16 @@ from homeassistant.components.sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import IsraelRailConfigEntry
from .const import ATTRIBUTION, DEPARTURES_COUNT, DOMAIN
-from .coordinator import DataConnection, IsraelRailDataUpdateCoordinator
+from .coordinator import (
+ DataConnection,
+ IsraelRailConfigEntry,
+ IsraelRailDataUpdateCoordinator,
+)
_LOGGER = logging.getLogger(__name__)
@@ -70,7 +73,7 @@ SENSORS: tuple[IsraelRailSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: IsraelRailConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor from a config entry created in the integrations UI."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/iss/sensor.py b/homeassistant/components/iss/sensor.py
index f4f91f0099e..b6e98e07f8a 100644
--- a/homeassistant/components/iss/sensor.py
+++ b/homeassistant/components/iss/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE, CONF_SHOW_ON_MAP
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -25,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
coordinator: DataUpdateCoordinator[IssData] = hass.data[DOMAIN]
diff --git a/homeassistant/components/ista_ecotrend/__init__.py b/homeassistant/components/ista_ecotrend/__init__.py
index 76ef8d13fd4..4262b354acb 100644
--- a/homeassistant/components/ista_ecotrend/__init__.py
+++ b/homeassistant/components/ista_ecotrend/__init__.py
@@ -6,20 +6,17 @@ import logging
from pyecotrend_ista import KeycloakError, LoginError, PyEcotrendIsta, ServerError
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from .const import DOMAIN
-from .coordinator import IstaCoordinator
+from .coordinator import IstaConfigEntry, IstaCoordinator
_LOGGER = logging.getLogger(__name__)
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type IstaConfigEntry = ConfigEntry[IstaCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: IstaConfigEntry) -> bool:
"""Set up ista EcoTrend from a config entry."""
@@ -42,7 +39,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: IstaConfigEntry) -> bool
translation_placeholders={CONF_EMAIL: entry.data[CONF_EMAIL]},
) from e
- coordinator = IstaCoordinator(hass, ista)
+ coordinator = IstaCoordinator(hass, entry, ista)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/ista_ecotrend/coordinator.py b/homeassistant/components/ista_ecotrend/coordinator.py
index 0f14cd06fe3..53ef4a46d20 100644
--- a/homeassistant/components/ista_ecotrend/coordinator.py
+++ b/homeassistant/components/ista_ecotrend/coordinator.py
@@ -18,17 +18,22 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
+type IstaConfigEntry = ConfigEntry[IstaCoordinator]
+
class IstaCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Ista EcoTrend data update coordinator."""
- config_entry: ConfigEntry
+ config_entry: IstaConfigEntry
- def __init__(self, hass: HomeAssistant, ista: PyEcotrendIsta) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: IstaConfigEntry, ista: PyEcotrendIsta
+ ) -> None:
"""Initialize ista EcoTrend data update coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(days=1),
)
diff --git a/homeassistant/components/ista_ecotrend/sensor.py b/homeassistant/components/ista_ecotrend/sensor.py
index e96ac103741..ee54e502c26 100644
--- a/homeassistant/components/ista_ecotrend/sensor.py
+++ b/homeassistant/components/ista_ecotrend/sensor.py
@@ -30,13 +30,12 @@ from homeassistant.helpers.device_registry import (
DeviceEntryType,
DeviceInfo,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import IstaConfigEntry
from .const import DOMAIN
-from .coordinator import IstaCoordinator
+from .coordinator import IstaConfigEntry, IstaCoordinator
from .util import IstaConsumptionType, IstaValueType, get_native_value, get_statistics
_LOGGER = logging.getLogger(__name__)
@@ -153,7 +152,7 @@ SENSOR_DESCRIPTIONS: tuple[IstaSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: IstaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ista EcoTrend sensors."""
diff --git a/homeassistant/components/isy994/binary_sensor.py b/homeassistant/components/isy994/binary_sensor.py
index 179944ad35f..8c9ce7dcc12 100644
--- a/homeassistant/components/isy994/binary_sensor.py
+++ b/homeassistant/components/isy994/binary_sensor.py
@@ -23,7 +23,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_ON, Platform
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.restore_state import RestoreEntity
@@ -54,7 +54,9 @@ DEVICE_PARENT_REQUIRED = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY binary sensor platform."""
entities: list[
diff --git a/homeassistant/components/isy994/button.py b/homeassistant/components/isy994/button.py
index b3b6aa40503..a895312c45a 100644
--- a/homeassistant/components/isy994/button.py
+++ b/homeassistant/components/isy994/button.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_NETWORK, DOMAIN
from .models import IsyData
@@ -28,7 +28,7 @@ from .models import IsyData
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ISY/IoX button from config entry."""
isy_data: IsyData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/isy994/climate.py b/homeassistant/components/isy994/climate.py
index d5deba56284..57c1b6aa79d 100644
--- a/homeassistant/components/isy994/climate.py
+++ b/homeassistant/components/isy994/climate.py
@@ -37,7 +37,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.enum import try_parse_enum
from .const import (
@@ -61,7 +61,9 @@ from .models import IsyData
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY thermostat platform."""
diff --git a/homeassistant/components/isy994/cover.py b/homeassistant/components/isy994/cover.py
index b9d7ec44d27..6a660aaaf6f 100644
--- a/homeassistant/components/isy994/cover.py
+++ b/homeassistant/components/isy994/cover.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import _LOGGER, DOMAIN, UOM_8_BIT_RANGE
from .entity import ISYNodeEntity, ISYProgramEntity
@@ -23,7 +23,9 @@ from .models import IsyData
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY cover platform."""
isy_data: IsyData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/isy994/entity.py b/homeassistant/components/isy994/entity.py
index 893b33644fe..1da727fdee8 100644
--- a/homeassistant/components/isy994/entity.py
+++ b/homeassistant/components/isy994/entity.py
@@ -106,7 +106,7 @@ class ISYNodeEntity(ISYEntity):
return getattr(self._node, TAG_ENABLED, True)
@property
- def extra_state_attributes(self) -> dict:
+ def extra_state_attributes(self) -> dict[str, Any]:
"""Get the state attributes for the device.
The 'aux_properties' in the pyisy Node class are combined with the
@@ -189,7 +189,7 @@ class ISYProgramEntity(ISYEntity):
self._actions = actions
@property
- def extra_state_attributes(self) -> dict:
+ def extra_state_attributes(self) -> dict[str, Any]:
"""Get the state attributes for the device."""
attr = {}
if self._actions:
diff --git a/homeassistant/components/isy994/fan.py b/homeassistant/components/isy994/fan.py
index fc0406e2d5f..aa6059abf49 100644
--- a/homeassistant/components/isy994/fan.py
+++ b/homeassistant/components/isy994/fan.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -27,7 +27,9 @@ SPEED_RANGE = (1, 255) # off is not included
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY fan platform."""
isy_data: IsyData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/isy994/light.py b/homeassistant/components/isy994/light.py
index b9b269d9ca3..29df8398f97 100644
--- a/homeassistant/components/isy994/light.py
+++ b/homeassistant/components/isy994/light.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import _LOGGER, CONF_RESTORE_LIGHT_STATE, DOMAIN, UOM_PERCENTAGE
@@ -24,7 +24,9 @@ ATTR_LAST_BRIGHTNESS = "last_brightness"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY light platform."""
isy_data: IsyData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/isy994/lock.py b/homeassistant/components/isy994/lock.py
index dc2da2a6ee2..d6866a8e00c 100644
--- a/homeassistant/components/isy994/lock.py
+++ b/homeassistant/components/isy994/lock.py
@@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
@@ -48,7 +48,9 @@ def async_setup_lock_services(hass: HomeAssistant) -> None:
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY lock platform."""
isy_data: IsyData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/isy994/number.py b/homeassistant/components/isy994/number.py
index c8feba1bf8d..fc30e6296d4 100644
--- a/homeassistant/components/isy994/number.py
+++ b/homeassistant/components/isy994/number.py
@@ -38,7 +38,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -80,7 +80,7 @@ BACKLIGHT_MEMORY_FILTER = {"memory": DEV_BL_ADDR, "cmd1": DEV_CMD_MEMORY_WRITE}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ISY/IoX number entities from config entry."""
isy_data: IsyData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/isy994/select.py b/homeassistant/components/isy994/select.py
index 8befcf024d1..868c96375bb 100644
--- a/homeassistant/components/isy994/select.py
+++ b/homeassistant/components/isy994/select.py
@@ -34,7 +34,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import _LOGGER, DOMAIN, UOM_INDEX
@@ -56,7 +56,7 @@ BACKLIGHT_MEMORY_FILTER = {"memory": DEV_BL_ADDR, "cmd1": DEV_CMD_MEMORY_WRITE}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ISY/IoX select entities from config entry."""
isy_data: IsyData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/isy994/sensor.py b/homeassistant/components/isy994/sensor.py
index 58ba3171bc8..2d27f4602c6 100644
--- a/homeassistant/components/isy994/sensor.py
+++ b/homeassistant/components/isy994/sensor.py
@@ -33,7 +33,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
_LOGGER,
@@ -97,9 +97,9 @@ ISY_CONTROL_TO_DEVICE_CLASS = {
"WEIGHT": SensorDeviceClass.WEIGHT,
"WINDCH": SensorDeviceClass.TEMPERATURE,
}
-ISY_CONTROL_TO_STATE_CLASS = {
- control: SensorStateClass.MEASUREMENT for control in ISY_CONTROL_TO_DEVICE_CLASS
-}
+ISY_CONTROL_TO_STATE_CLASS = dict.fromkeys(
+ ISY_CONTROL_TO_DEVICE_CLASS, SensorStateClass.MEASUREMENT
+)
ISY_CONTROL_TO_ENTITY_CATEGORY = {
PROP_RAMP_RATE: EntityCategory.DIAGNOSTIC,
PROP_ON_LEVEL: EntityCategory.DIAGNOSTIC,
@@ -108,7 +108,9 @@ ISY_CONTROL_TO_ENTITY_CATEGORY = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY sensor platform."""
isy_data: IsyData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/isy994/strings.json b/homeassistant/components/isy994/strings.json
index 86a1f14ff91..8872226daba 100644
--- a/homeassistant/components/isy994/strings.json
+++ b/homeassistant/components/isy994/strings.json
@@ -58,7 +58,7 @@
"services": {
"send_raw_node_command": {
"name": "Send raw node command",
- "description": "[%key:component::isy994::options::step::init::description%]",
+ "description": "Sends a “raw” (e.g., DON, DOF) ISY REST device command to a node using its Home Assistant entity ID. This is useful for devices that aren’t fully supported in Home Assistant yet, such as controls for many NodeServer nodes.",
"fields": {
"command": {
"name": "Command",
diff --git a/homeassistant/components/isy994/switch.py b/homeassistant/components/isy994/switch.py
index c05bd2ddbbb..946feddcd10 100644
--- a/homeassistant/components/isy994/switch.py
+++ b/homeassistant/components/isy994/switch.py
@@ -25,7 +25,7 @@ from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import ISYAuxControlEntity, ISYNodeEntity, ISYProgramEntity
@@ -42,7 +42,9 @@ class ISYSwitchEntityDescription(SwitchEntityDescription):
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ISY switch platform."""
isy_data: IsyData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/ituran/device_tracker.py b/homeassistant/components/ituran/device_tracker.py
index 37796570c61..5f816709864 100644
--- a/homeassistant/components/ituran/device_tracker.py
+++ b/homeassistant/components/ituran/device_tracker.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import IturanConfigEntry
from .coordinator import IturanDataUpdateCoordinator
@@ -14,7 +14,7 @@ from .entity import IturanBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: IturanConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ituran tracker from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/ituran/sensor.py b/homeassistant/components/ituran/sensor.py
index e962f5bd561..a115b2be89c 100644
--- a/homeassistant/components/ituran/sensor.py
+++ b/homeassistant/components/ituran/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfSpeed,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import IturanConfigEntry
@@ -87,7 +87,7 @@ SENSOR_TYPES: list[IturanSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: IturanConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ituran sensors from config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/izone/climate.py b/homeassistant/components/izone/climate.py
index e61917c825b..80c3a0384c1 100644
--- a/homeassistant/components/izone/climate.py
+++ b/homeassistant/components/izone/climate.py
@@ -33,7 +33,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.temperature import display_temp as show_temp
from homeassistant.helpers.typing import ConfigType, VolDictType
@@ -73,7 +73,9 @@ IZONE_SERVICE_AIRFLOW_SCHEMA: VolDictType = {
async def async_setup_entry(
- hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize an IZone Controller."""
disco = hass.data[DATA_DISCOVERY_SERVICE]
diff --git a/homeassistant/components/jellyfin/__init__.py b/homeassistant/components/jellyfin/__init__.py
index 4f0886dfa22..1cb6219ada0 100644
--- a/homeassistant/components/jellyfin/__init__.py
+++ b/homeassistant/components/jellyfin/__init__.py
@@ -2,16 +2,13 @@
from typing import Any
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
from .client_wrapper import CannotConnect, InvalidAuth, create_client, validate_input
from .const import CONF_CLIENT_DEVICE_ID, DOMAIN, PLATFORMS
-from .coordinator import JellyfinDataUpdateCoordinator
-
-type JellyfinConfigEntry = ConfigEntry[JellyfinDataUpdateCoordinator]
+from .coordinator import JellyfinConfigEntry, JellyfinDataUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: JellyfinConfigEntry) -> bool:
@@ -35,7 +32,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: JellyfinConfigEntry) ->
server_info: dict[str, Any] = connect_result["Servers"][0]
- coordinator = JellyfinDataUpdateCoordinator(hass, client, server_info, user_id)
+ coordinator = JellyfinDataUpdateCoordinator(
+ hass, entry, client, server_info, user_id
+ )
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/jellyfin/config_flow.py b/homeassistant/components/jellyfin/config_flow.py
index 0c170d2485f..03c637a989f 100644
--- a/homeassistant/components/jellyfin/config_flow.py
+++ b/homeassistant/components/jellyfin/config_flow.py
@@ -13,9 +13,9 @@ from homeassistant.const import CONF_PASSWORD, CONF_URL, CONF_USERNAME
from homeassistant.core import callback
from homeassistant.util.uuid import random_uuid_hex
-from . import JellyfinConfigEntry
from .client_wrapper import CannotConnect, InvalidAuth, create_client, validate_input
from .const import CONF_CLIENT_DEVICE_ID, DOMAIN, SUPPORTED_AUDIO_CODECS
+from .coordinator import JellyfinConfigEntry
_LOGGER = logging.getLogger(__name__)
diff --git a/homeassistant/components/jellyfin/coordinator.py b/homeassistant/components/jellyfin/coordinator.py
index 20428250254..cd22ad4ab39 100644
--- a/homeassistant/components/jellyfin/coordinator.py
+++ b/homeassistant/components/jellyfin/coordinator.py
@@ -13,15 +13,18 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import CONF_CLIENT_DEVICE_ID, DOMAIN, LOGGER, USER_APP_NAME
+type JellyfinConfigEntry = ConfigEntry[JellyfinDataUpdateCoordinator]
+
class JellyfinDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict[str, Any]]]):
"""Data update coordinator for the Jellyfin integration."""
- config_entry: ConfigEntry
+ config_entry: JellyfinConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: JellyfinConfigEntry,
api_client: JellyfinClient,
system_info: dict[str, Any],
user_id: str,
@@ -30,6 +33,7 @@ class JellyfinDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict[str, An
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=10),
)
@@ -37,7 +41,7 @@ class JellyfinDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict[str, An
self.server_id: str = system_info["Id"]
self.server_name: str = system_info["Name"]
self.server_version: str | None = system_info.get("Version")
- self.client_device_id: str = self.config_entry.data[CONF_CLIENT_DEVICE_ID]
+ self.client_device_id: str = config_entry.data[CONF_CLIENT_DEVICE_ID]
self.user_id: str = user_id
self.session_ids: set[str] = set()
diff --git a/homeassistant/components/jellyfin/diagnostics.py b/homeassistant/components/jellyfin/diagnostics.py
index 8042d588d1b..721e0ae654e 100644
--- a/homeassistant/components/jellyfin/diagnostics.py
+++ b/homeassistant/components/jellyfin/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_PASSWORD
from homeassistant.core import HomeAssistant
-from . import JellyfinConfigEntry
+from .coordinator import JellyfinConfigEntry
TO_REDACT = {CONF_PASSWORD}
diff --git a/homeassistant/components/jellyfin/media_player.py b/homeassistant/components/jellyfin/media_player.py
index bf6e95c0c96..e0fcc8a559b 100644
--- a/homeassistant/components/jellyfin/media_player.py
+++ b/homeassistant/components/jellyfin/media_player.py
@@ -12,21 +12,20 @@ from homeassistant.components.media_player import (
MediaType,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import parse_datetime
-from . import JellyfinConfigEntry
from .browse_media import build_item_response, build_root_response
from .client_wrapper import get_artwork_url
-from .const import CONTENT_TYPE_MAP, LOGGER
-from .coordinator import JellyfinDataUpdateCoordinator
+from .const import CONTENT_TYPE_MAP, LOGGER, MAX_IMAGE_WIDTH
+from .coordinator import JellyfinConfigEntry, JellyfinDataUpdateCoordinator
from .entity import JellyfinClientEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: JellyfinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Jellyfin media_player from a config entry."""
coordinator = entry.runtime_data
@@ -170,7 +169,9 @@ class JellyfinMediaPlayer(JellyfinClientEntity, MediaPlayerEntity):
if self.now_playing is None:
return None
- return get_artwork_url(self.coordinator.api_client, self.now_playing, 150)
+ return get_artwork_url(
+ self.coordinator.api_client, self.now_playing, MAX_IMAGE_WIDTH
+ )
@property
def supported_features(self) -> MediaPlayerEntityFeature:
diff --git a/homeassistant/components/jellyfin/media_source.py b/homeassistant/components/jellyfin/media_source.py
index a061118dd0a..a4d08d8d024 100644
--- a/homeassistant/components/jellyfin/media_source.py
+++ b/homeassistant/components/jellyfin/media_source.py
@@ -19,7 +19,6 @@ from homeassistant.components.media_source import (
)
from homeassistant.core import HomeAssistant
-from . import JellyfinConfigEntry
from .const import (
COLLECTION_TYPE_MOVIES,
COLLECTION_TYPE_MUSIC,
@@ -48,6 +47,7 @@ from .const import (
PLAYABLE_ITEM_TYPES,
SUPPORTED_COLLECTION_TYPES,
)
+from .coordinator import JellyfinConfigEntry
_LOGGER = logging.getLogger(__name__)
diff --git a/homeassistant/components/jellyfin/remote.py b/homeassistant/components/jellyfin/remote.py
index ae33d58cc0c..27a0b131ca0 100644
--- a/homeassistant/components/jellyfin/remote.py
+++ b/homeassistant/components/jellyfin/remote.py
@@ -14,18 +14,17 @@ from homeassistant.components.remote import (
RemoteEntity,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import JellyfinConfigEntry
from .const import LOGGER
-from .coordinator import JellyfinDataUpdateCoordinator
+from .coordinator import JellyfinConfigEntry, JellyfinDataUpdateCoordinator
from .entity import JellyfinClientEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: JellyfinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Jellyfin remote from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/jellyfin/sensor.py b/homeassistant/components/jellyfin/sensor.py
index 5c519f661ee..e1100a9f43b 100644
--- a/homeassistant/components/jellyfin/sensor.py
+++ b/homeassistant/components/jellyfin/sensor.py
@@ -8,10 +8,10 @@ from typing import Any
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import JellyfinConfigEntry, JellyfinDataUpdateCoordinator
+from .coordinator import JellyfinConfigEntry, JellyfinDataUpdateCoordinator
from .entity import JellyfinServerEntity
@@ -43,7 +43,7 @@ SENSOR_TYPES: tuple[JellyfinSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: JellyfinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Jellyfin sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/jewish_calendar/__init__.py b/homeassistant/components/jewish_calendar/__init__.py
index 823e9bd59be..9f7ec6ba976 100644
--- a/homeassistant/components/jewish_calendar/__init__.py
+++ b/homeassistant/components/jewish_calendar/__init__.py
@@ -3,6 +3,7 @@
from __future__ import annotations
from functools import partial
+import logging
from hdate import Location
@@ -14,7 +15,8 @@ from homeassistant.const import (
CONF_TIME_ZONE,
Platform,
)
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers import entity_registry as er
from .const import (
CONF_CANDLE_LIGHT_MINUTES,
@@ -27,6 +29,7 @@ from .const import (
)
from .entity import JewishCalendarConfigEntry, JewishCalendarData
+_LOGGER = logging.getLogger(__name__)
PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SENSOR]
@@ -80,3 +83,49 @@ async def async_unload_entry(
) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
+
+
+async def async_migrate_entry(
+ hass: HomeAssistant, config_entry: JewishCalendarConfigEntry
+) -> bool:
+ """Migrate old entry."""
+
+ _LOGGER.debug("Migrating from version %s", config_entry.version)
+
+ @callback
+ def update_unique_id(
+ entity_entry: er.RegistryEntry,
+ ) -> dict[str, str] | None:
+ """Update unique ID of entity entry."""
+ key_translations = {
+ "first_light": "alot_hashachar",
+ "talit": "talit_and_tefillin",
+ "sunrise": "netz_hachama",
+ "gra_end_shma": "sof_zman_shema_gra",
+ "mga_end_shma": "sof_zman_shema_mga",
+ "gra_end_tfila": "sof_zman_tfilla_gra",
+ "mga_end_tfila": "sof_zman_tfilla_mga",
+ "midday": "chatzot_hayom",
+ "big_mincha": "mincha_gedola",
+ "small_mincha": "mincha_ketana",
+ "plag_mincha": "plag_hamincha",
+ "sunset": "shkia",
+ "first_stars": "tset_hakohavim_tsom",
+ "three_stars": "tset_hakohavim_shabbat",
+ }
+ new_keys = tuple(key_translations.values())
+ if not entity_entry.unique_id.endswith(new_keys):
+ old_key = entity_entry.unique_id.split("-")[1]
+ new_unique_id = f"{config_entry.entry_id}-{key_translations[old_key]}"
+ return {"new_unique_id": new_unique_id}
+ return None
+
+ if config_entry.version > 1:
+ # This means the user has downgraded from a future version
+ return False
+
+ if config_entry.version == 1:
+ await er.async_migrate_entries(hass, config_entry.entry_id, update_unique_id)
+ hass.config_entries.async_update_entry(config_entry, version=2)
+
+ return True
diff --git a/homeassistant/components/jewish_calendar/binary_sensor.py b/homeassistant/components/jewish_calendar/binary_sensor.py
index 85519bf37b0..f33d79a01f5 100644
--- a/homeassistant/components/jewish_calendar/binary_sensor.py
+++ b/homeassistant/components/jewish_calendar/binary_sensor.py
@@ -5,9 +5,7 @@ from __future__ import annotations
from collections.abc import Callable
from dataclasses import dataclass
import datetime as dt
-from datetime import datetime
-import hdate
from hdate.zmanim import Zmanim
from homeassistant.components.binary_sensor import (
@@ -17,7 +15,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers import event
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .entity import JewishCalendarConfigEntry, JewishCalendarEntity
@@ -27,7 +25,7 @@ from .entity import JewishCalendarConfigEntry, JewishCalendarEntity
class JewishCalendarBinarySensorMixIns(BinarySensorEntityDescription):
"""Binary Sensor description mixin class for Jewish Calendar."""
- is_on: Callable[[Zmanim], bool] = lambda _: False
+ is_on: Callable[[Zmanim, dt.datetime], bool] = lambda _, __: False
@dataclass(frozen=True)
@@ -42,18 +40,18 @@ BINARY_SENSORS: tuple[JewishCalendarBinarySensorEntityDescription, ...] = (
key="issur_melacha_in_effect",
name="Issur Melacha in Effect",
icon="mdi:power-plug-off",
- is_on=lambda state: bool(state.issur_melacha_in_effect),
+ is_on=lambda state, now: bool(state.issur_melacha_in_effect(now)),
),
JewishCalendarBinarySensorEntityDescription(
key="erev_shabbat_hag",
name="Erev Shabbat/Hag",
- is_on=lambda state: bool(state.erev_shabbat_chag),
+ is_on=lambda state, now: bool(state.erev_shabbat_chag(now)),
entity_registry_enabled_default=False,
),
JewishCalendarBinarySensorEntityDescription(
key="motzei_shabbat_hag",
name="Motzei Shabbat/Hag",
- is_on=lambda state: bool(state.motzei_shabbat_chag),
+ is_on=lambda state, now: bool(state.motzei_shabbat_chag(now)),
entity_registry_enabled_default=False,
),
)
@@ -62,7 +60,7 @@ BINARY_SENSORS: tuple[JewishCalendarBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: JewishCalendarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Jewish Calendar binary sensors."""
async_add_entities(
@@ -84,16 +82,16 @@ class JewishCalendarBinarySensor(JewishCalendarEntity, BinarySensorEntity):
def is_on(self) -> bool:
"""Return true if sensor is on."""
zmanim = self._get_zmanim()
- return self.entity_description.is_on(zmanim)
+ return self.entity_description.is_on(zmanim, dt_util.now())
def _get_zmanim(self) -> Zmanim:
"""Return the Zmanim object for now()."""
- return hdate.Zmanim(
- date=dt_util.now(),
+ return Zmanim(
+ date=dt.date.today(),
location=self._location,
candle_lighting_offset=self._candle_lighting_offset,
havdalah_offset=self._havdalah_offset,
- hebrew=self._hebrew,
+ language=self._language,
)
async def async_added_to_hass(self) -> None:
@@ -109,7 +107,7 @@ class JewishCalendarBinarySensor(JewishCalendarEntity, BinarySensorEntity):
return await super().async_will_remove_from_hass()
@callback
- def _update(self, now: datetime | None = None) -> None:
+ def _update(self, now: dt.datetime | None = None) -> None:
"""Update the state of the sensor."""
self._update_unsub = None
self._schedule_update()
@@ -119,7 +117,7 @@ class JewishCalendarBinarySensor(JewishCalendarEntity, BinarySensorEntity):
"""Schedule the next update of the sensor."""
now = dt_util.now()
zmanim = self._get_zmanim()
- update = zmanim.zmanim["sunrise"] + dt.timedelta(days=1)
+ update = zmanim.netz_hachama.local + dt.timedelta(days=1)
candle_lighting = zmanim.candle_lighting
if candle_lighting is not None and now < candle_lighting < update:
update = candle_lighting
diff --git a/homeassistant/components/jewish_calendar/config_flow.py b/homeassistant/components/jewish_calendar/config_flow.py
index a2eadbf57bd..23bcb23435b 100644
--- a/homeassistant/components/jewish_calendar/config_flow.py
+++ b/homeassistant/components/jewish_calendar/config_flow.py
@@ -86,7 +86,7 @@ def _get_data_schema(hass: HomeAssistant) -> vol.Schema:
class JewishCalendarConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a config flow for Jewish calendar."""
- VERSION = 1
+ VERSION = 2
@staticmethod
@callback
diff --git a/homeassistant/components/jewish_calendar/entity.py b/homeassistant/components/jewish_calendar/entity.py
index 1d2a6e45c0a..2c031f0d160 100644
--- a/homeassistant/components/jewish_calendar/entity.py
+++ b/homeassistant/components/jewish_calendar/entity.py
@@ -3,6 +3,7 @@
from dataclasses import dataclass
from hdate import Location
+from hdate.translator import Language
from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
@@ -17,7 +18,7 @@ type JewishCalendarConfigEntry = ConfigEntry[JewishCalendarData]
class JewishCalendarData:
"""Jewish Calendar runtime dataclass."""
- language: str
+ language: Language
diaspora: bool
location: Location
candle_lighting_offset: int
@@ -43,7 +44,6 @@ class JewishCalendarEntity(Entity):
)
data = config_entry.runtime_data
self._location = data.location
- self._hebrew = data.language == "hebrew"
self._language = data.language
self._candle_lighting_offset = data.candle_lighting_offset
self._havdalah_offset = data.havdalah_offset
diff --git a/homeassistant/components/jewish_calendar/manifest.json b/homeassistant/components/jewish_calendar/manifest.json
index aca45320002..877c4cf9a99 100644
--- a/homeassistant/components/jewish_calendar/manifest.json
+++ b/homeassistant/components/jewish_calendar/manifest.json
@@ -6,6 +6,6 @@
"documentation": "https://www.home-assistant.io/integrations/jewish_calendar",
"iot_class": "calculated",
"loggers": ["hdate"],
- "requirements": ["hdate==0.11.1"],
+ "requirements": ["hdate[astral]==1.0.3"],
"single_config_entry": true
}
diff --git a/homeassistant/components/jewish_calendar/sensor.py b/homeassistant/components/jewish_calendar/sensor.py
index 5e02435ed06..7cb281b3af4 100644
--- a/homeassistant/components/jewish_calendar/sensor.py
+++ b/homeassistant/components/jewish_calendar/sensor.py
@@ -2,12 +2,13 @@
from __future__ import annotations
-from datetime import date as Date
+import datetime as dt
import logging
-from typing import Any, cast
+from typing import Any
-from hdate import HDate, HebrewDate, htables
-from hdate.zmanim import Zmanim
+from hdate import HDateInfo, Zmanim
+from hdate.holidays import HolidayDatabase
+from hdate.parasha import Parasha
from homeassistant.components.sensor import (
SensorDeviceClass,
@@ -16,7 +17,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import SUN_EVENT_SUNSET, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sun import get_astral_event_date
from homeassistant.util import dt as dt_util
@@ -59,83 +60,83 @@ INFO_SENSORS: tuple[SensorEntityDescription, ...] = (
TIME_SENSORS: tuple[SensorEntityDescription, ...] = (
SensorEntityDescription(
- key="first_light",
+ key="alot_hashachar",
name="Alot Hashachar", # codespell:ignore alot
icon="mdi:weather-sunset-up",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="talit",
+ key="talit_and_tefillin",
name="Talit and Tefillin",
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="sunrise",
+ key="netz_hachama",
name="Hanetz Hachama",
icon="mdi:calendar-clock",
),
SensorEntityDescription(
- key="gra_end_shma",
+ key="sof_zman_shema_gra",
name='Latest time for Shma Gr"a',
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="mga_end_shma",
+ key="sof_zman_shema_mga",
name='Latest time for Shma MG"A',
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="gra_end_tfila",
+ key="sof_zman_tfilla_gra",
name='Latest time for Tefilla Gr"a',
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="mga_end_tfila",
+ key="sof_zman_tfilla_mga",
name='Latest time for Tefilla MG"A',
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="midday",
+ key="chatzot_hayom",
name="Chatzot Hayom",
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="big_mincha",
+ key="mincha_gedola",
name="Mincha Gedola",
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="small_mincha",
+ key="mincha_ketana",
name="Mincha Ketana",
icon="mdi:calendar-clock",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="plag_mincha",
+ key="plag_hamincha",
name="Plag Hamincha",
icon="mdi:weather-sunset-down",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="sunset",
+ key="shkia",
name="Shkia",
icon="mdi:weather-sunset",
),
SensorEntityDescription(
- key="first_stars",
+ key="tset_hakohavim_tsom",
name="T'set Hakochavim",
icon="mdi:weather-night",
entity_registry_enabled_default=False,
),
SensorEntityDescription(
- key="three_stars",
+ key="tset_hakohavim_shabbat",
name="T'set Hakochavim, 3 stars",
icon="mdi:weather-night",
entity_registry_enabled_default=False,
@@ -168,7 +169,7 @@ TIME_SENSORS: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: JewishCalendarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Jewish calendar sensors ."""
sensors = [
@@ -212,7 +213,9 @@ class JewishCalendarSensor(JewishCalendarEntity, SensorEntity):
_LOGGER.debug("Now: %s Sunset: %s", now, sunset)
- daytime_date = HDate(today, diaspora=self._diaspora, hebrew=self._hebrew)
+ daytime_date = HDateInfo(
+ today, diaspora=self._diaspora, language=self._language
+ )
# The Jewish day starts after darkness (called "tzais") and finishes at
# sunset ("shkia"). The time in between is a gray area
@@ -238,14 +241,14 @@ class JewishCalendarSensor(JewishCalendarEntity, SensorEntity):
"New value for %s: %s", self.entity_description.key, self._attr_native_value
)
- def make_zmanim(self, date: Date) -> Zmanim:
+ def make_zmanim(self, date: dt.date) -> Zmanim:
"""Create a Zmanim object."""
return Zmanim(
date=date,
location=self._location,
candle_lighting_offset=self._candle_lighting_offset,
havdalah_offset=self._havdalah_offset,
- hebrew=self._hebrew,
+ language=self._language,
)
@property
@@ -254,43 +257,40 @@ class JewishCalendarSensor(JewishCalendarEntity, SensorEntity):
return self._attrs
def get_state(
- self, daytime_date: HDate, after_shkia_date: HDate, after_tzais_date: HDate
+ self,
+ daytime_date: HDateInfo,
+ after_shkia_date: HDateInfo,
+ after_tzais_date: HDateInfo,
) -> Any | None:
"""For a given type of sensor, return the state."""
# Terminology note: by convention in py-libhdate library, "upcoming"
# refers to "current" or "upcoming" dates.
if self.entity_description.key == "date":
- hdate = cast(HebrewDate, after_shkia_date.hdate)
- month = htables.MONTHS[hdate.month.value - 1]
+ hdate = after_shkia_date.hdate
+ hdate.month.set_language(self._language)
self._attrs = {
- "hebrew_year": hdate.year,
- "hebrew_month_name": month.hebrew if self._hebrew else month.english,
- "hebrew_day": hdate.day,
+ "hebrew_year": str(hdate.year),
+ "hebrew_month_name": str(hdate.month),
+ "hebrew_day": str(hdate.day),
}
- return after_shkia_date.hebrew_date
+ return after_shkia_date.hdate
if self.entity_description.key == "weekly_portion":
- self._attr_options = [
- (p.hebrew if self._hebrew else p.english) for p in htables.PARASHAOT
- ]
+ self._attr_options = list(Parasha)
# Compute the weekly portion based on the upcoming shabbat.
return after_tzais_date.upcoming_shabbat.parasha
if self.entity_description.key == "holiday":
- _id = _type = _type_id = ""
- _holiday_type = after_shkia_date.holiday_type
- if isinstance(_holiday_type, list):
- _id = ", ".join(after_shkia_date.holiday_name)
- _type = ", ".join([_htype.name for _htype in _holiday_type])
- _type_id = ", ".join([str(_htype.value) for _htype in _holiday_type])
- else:
- _id = after_shkia_date.holiday_name
- _type = _holiday_type.name
- _type_id = _holiday_type.value
- self._attrs = {"id": _id, "type": _type, "type_id": _type_id}
- self._attr_options = htables.get_all_holidays(self._language)
-
- return after_shkia_date.holiday_description
+ _holidays = after_shkia_date.holidays
+ _id = ", ".join(holiday.name for holiday in _holidays)
+ _type = ", ".join(
+ dict.fromkeys(_holiday.type.name for _holiday in _holidays)
+ )
+ self._attrs = {"id": _id, "type": _type}
+ self._attr_options = HolidayDatabase(self._diaspora).get_all_names(
+ self._language
+ )
+ return ", ".join(str(holiday) for holiday in _holidays) if _holidays else ""
if self.entity_description.key == "omer_count":
- return after_shkia_date.omer_day
+ return after_shkia_date.omer.total_days if after_shkia_date.omer else 0
if self.entity_description.key == "daf_yomi":
return daytime_date.daf_yomi
@@ -303,7 +303,10 @@ class JewishCalendarTimeSensor(JewishCalendarSensor):
_attr_device_class = SensorDeviceClass.TIMESTAMP
def get_state(
- self, daytime_date: HDate, after_shkia_date: HDate, after_tzais_date: HDate
+ self,
+ daytime_date: HDateInfo,
+ after_shkia_date: HDateInfo,
+ after_tzais_date: HDateInfo,
) -> Any | None:
"""For a given type of sensor, return the state."""
if self.entity_description.key == "upcoming_shabbat_candle_lighting":
@@ -325,5 +328,5 @@ class JewishCalendarTimeSensor(JewishCalendarSensor):
)
return times.havdalah
- times = self.make_zmanim(dt_util.now()).zmanim
- return times[self.entity_description.key]
+ times = self.make_zmanim(dt_util.now().date())
+ return times.zmanim[self.entity_description.key].local
diff --git a/homeassistant/components/juicenet/number.py b/homeassistant/components/juicenet/number.py
index 383d0d590c4..69323884f61 100644
--- a/homeassistant/components/juicenet/number.py
+++ b/homeassistant/components/juicenet/number.py
@@ -13,7 +13,7 @@ from homeassistant.components.number import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN, JUICENET_API, JUICENET_COORDINATOR
@@ -43,7 +43,7 @@ NUMBER_TYPES: tuple[JuiceNetNumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JuiceNet Numbers."""
juicenet_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/juicenet/sensor.py b/homeassistant/components/juicenet/sensor.py
index 1f0b815cd97..7bf0639f5d0 100644
--- a/homeassistant/components/juicenet/sensor.py
+++ b/homeassistant/components/juicenet/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, JUICENET_API, JUICENET_COORDINATOR
from .entity import JuiceNetDevice
@@ -70,7 +70,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JuiceNet Sensors."""
juicenet_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/juicenet/switch.py b/homeassistant/components/juicenet/switch.py
index d800ac58c2c..9f34b7afdb3 100644
--- a/homeassistant/components/juicenet/switch.py
+++ b/homeassistant/components/juicenet/switch.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, JUICENET_API, JUICENET_COORDINATOR
from .entity import JuiceNetDevice
@@ -14,7 +14,7 @@ from .entity import JuiceNetDevice
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JuiceNet switches."""
juicenet_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/justnimbus/__init__.py b/homeassistant/components/justnimbus/__init__.py
index 101a2086962..123807d887c 100644
--- a/homeassistant/components/justnimbus/__init__.py
+++ b/homeassistant/components/justnimbus/__init__.py
@@ -13,7 +13,7 @@ from .coordinator import JustNimbusCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up JustNimbus from a config entry."""
if "zip_code" in entry.data:
- coordinator = JustNimbusCoordinator(hass=hass, entry=entry)
+ coordinator = JustNimbusCoordinator(hass, entry)
else:
raise ConfigEntryAuthFailed
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/justnimbus/coordinator.py b/homeassistant/components/justnimbus/coordinator.py
index 4031ad86fdf..a6945c45417 100644
--- a/homeassistant/components/justnimbus/coordinator.py
+++ b/homeassistant/components/justnimbus/coordinator.py
@@ -20,16 +20,20 @@ _LOGGER = logging.getLogger(__name__)
class JustNimbusCoordinator(DataUpdateCoordinator[justnimbus.JustNimbusModel]):
"""Data update coordinator."""
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=1),
)
self._client = justnimbus.JustNimbusClient(
- client_id=entry.data[CONF_CLIENT_ID], zip_code=entry.data[CONF_ZIP_CODE]
+ client_id=config_entry.data[CONF_CLIENT_ID],
+ zip_code=config_entry.data[CONF_ZIP_CODE],
)
async def _async_update_data(self) -> justnimbus.JustNimbusModel:
diff --git a/homeassistant/components/justnimbus/sensor.py b/homeassistant/components/justnimbus/sensor.py
index c2c22307371..1e288e272cd 100644
--- a/homeassistant/components/justnimbus/sensor.py
+++ b/homeassistant/components/justnimbus/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import JustNimbusCoordinator
@@ -101,7 +101,9 @@ SENSOR_TYPES = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JustNimbus sensor."""
coordinator: JustNimbusCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/jvc_projector/__init__.py b/homeassistant/components/jvc_projector/__init__.py
index 09e93127e40..ad7e333ca13 100644
--- a/homeassistant/components/jvc_projector/__init__.py
+++ b/homeassistant/components/jvc_projector/__init__.py
@@ -4,7 +4,6 @@ from __future__ import annotations
from jvcprojector import JvcProjector, JvcProjectorAuthError, JvcProjectorConnectError
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_HOST,
CONF_PASSWORD,
@@ -15,9 +14,7 @@ from homeassistant.const import (
from homeassistant.core import Event, HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
-from .coordinator import JvcProjectorDataUpdateCoordinator
-
-type JVCConfigEntry = ConfigEntry[JvcProjectorDataUpdateCoordinator]
+from .coordinator import JVCConfigEntry, JvcProjectorDataUpdateCoordinator
PLATFORMS = [Platform.BINARY_SENSOR, Platform.REMOTE, Platform.SELECT, Platform.SENSOR]
@@ -41,7 +38,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: JVCConfigEntry) -> bool:
await device.disconnect()
raise ConfigEntryAuthFailed("Password authentication failed") from err
- coordinator = JvcProjectorDataUpdateCoordinator(hass, device)
+ coordinator = JvcProjectorDataUpdateCoordinator(hass, entry, device)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/jvc_projector/binary_sensor.py b/homeassistant/components/jvc_projector/binary_sensor.py
index 6dfac63892b..7ae76298839 100644
--- a/homeassistant/components/jvc_projector/binary_sensor.py
+++ b/homeassistant/components/jvc_projector/binary_sensor.py
@@ -6,16 +6,18 @@ from jvcprojector import const
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import JVCConfigEntry, JvcProjectorDataUpdateCoordinator
+from .coordinator import JVCConfigEntry, JvcProjectorDataUpdateCoordinator
from .entity import JvcProjectorEntity
ON_STATUS = (const.ON, const.WARMING)
async def async_setup_entry(
- hass: HomeAssistant, entry: JVCConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: JVCConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JVC Projector platform from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/jvc_projector/coordinator.py b/homeassistant/components/jvc_projector/coordinator.py
index a2ecfa8eb52..db97b05f980 100644
--- a/homeassistant/components/jvc_projector/coordinator.py
+++ b/homeassistant/components/jvc_projector/coordinator.py
@@ -13,6 +13,7 @@ from jvcprojector import (
const,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.device_registry import format_mac
@@ -25,15 +26,22 @@ _LOGGER = logging.getLogger(__name__)
INTERVAL_SLOW = timedelta(seconds=10)
INTERVAL_FAST = timedelta(seconds=5)
+type JVCConfigEntry = ConfigEntry[JvcProjectorDataUpdateCoordinator]
+
class JvcProjectorDataUpdateCoordinator(DataUpdateCoordinator[dict[str, str]]):
"""Data update coordinator for the JVC Projector integration."""
- def __init__(self, hass: HomeAssistant, device: JvcProjector) -> None:
+ config_entry: JVCConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: JVCConfigEntry, device: JvcProjector
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=_LOGGER,
+ config_entry=config_entry,
name=NAME,
update_interval=INTERVAL_SLOW,
)
diff --git a/homeassistant/components/jvc_projector/remote.py b/homeassistant/components/jvc_projector/remote.py
index f90a2816363..e1aff2fbb4c 100644
--- a/homeassistant/components/jvc_projector/remote.py
+++ b/homeassistant/components/jvc_projector/remote.py
@@ -12,9 +12,9 @@ from jvcprojector import const
from homeassistant.components.remote import RemoteEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import JVCConfigEntry
+from .coordinator import JVCConfigEntry
from .entity import JvcProjectorEntity
COMMANDS = {
@@ -54,7 +54,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: JVCConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: JVCConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JVC Projector platform from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/jvc_projector/select.py b/homeassistant/components/jvc_projector/select.py
index 60c80f98fc0..b83695609cb 100644
--- a/homeassistant/components/jvc_projector/select.py
+++ b/homeassistant/components/jvc_projector/select.py
@@ -10,9 +10,9 @@ from jvcprojector import JvcProjector, const
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import JVCConfigEntry, JvcProjectorDataUpdateCoordinator
+from .coordinator import JVCConfigEntry, JvcProjectorDataUpdateCoordinator
from .entity import JvcProjectorEntity
@@ -40,7 +40,7 @@ SELECTS: Final[list[JvcProjectorSelectDescription]] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: JVCConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JVC Projector platform from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/jvc_projector/sensor.py b/homeassistant/components/jvc_projector/sensor.py
index 3edf51e4316..7a7799bc4ee 100644
--- a/homeassistant/components/jvc_projector/sensor.py
+++ b/homeassistant/components/jvc_projector/sensor.py
@@ -11,9 +11,9 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import JVCConfigEntry, JvcProjectorDataUpdateCoordinator
+from .coordinator import JVCConfigEntry, JvcProjectorDataUpdateCoordinator
from .entity import JvcProjectorEntity
JVC_SENSORS = (
@@ -34,7 +34,9 @@ JVC_SENSORS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: JVCConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: JVCConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the JVC Projector platform from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/kaleidescape/media_player.py b/homeassistant/components/kaleidescape/media_player.py
index 33acb899728..88e2e16bef2 100644
--- a/homeassistant/components/kaleidescape/media_player.py
+++ b/homeassistant/components/kaleidescape/media_player.py
@@ -22,7 +22,7 @@ if TYPE_CHECKING:
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
- from homeassistant.helpers.entity_platform import AddEntitiesCallback
+ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
KALEIDESCAPE_PLAYING_STATES = [
@@ -38,7 +38,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from a config entry."""
entities = [KaleidescapeMediaPlayer(hass.data[KALEIDESCAPE_DOMAIN][entry.entry_id])]
diff --git a/homeassistant/components/kaleidescape/remote.py b/homeassistant/components/kaleidescape/remote.py
index 2d35ad2787f..ddafd52f220 100644
--- a/homeassistant/components/kaleidescape/remote.py
+++ b/homeassistant/components/kaleidescape/remote.py
@@ -18,11 +18,13 @@ if TYPE_CHECKING:
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
- from homeassistant.helpers.entity_platform import AddEntitiesCallback
+ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from a config entry."""
entities = [KaleidescapeRemote(hass.data[KALEIDESCAPE_DOMAIN][entry.entry_id])]
diff --git a/homeassistant/components/kaleidescape/sensor.py b/homeassistant/components/kaleidescape/sensor.py
index 5520943e683..8bff5df2e70 100644
--- a/homeassistant/components/kaleidescape/sensor.py
+++ b/homeassistant/components/kaleidescape/sensor.py
@@ -18,7 +18,7 @@ if TYPE_CHECKING:
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
- from homeassistant.helpers.entity_platform import AddEntitiesCallback
+ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
@@ -131,7 +131,9 @@ SENSOR_TYPES: tuple[KaleidescapeSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from a config entry."""
device: KaleidescapeDevice = hass.data[KALEIDESCAPE_DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/keenetic_ndms2/binary_sensor.py b/homeassistant/components/keenetic_ndms2/binary_sensor.py
index cb7d83b9238..4d1b5da3552 100644
--- a/homeassistant/components/keenetic_ndms2/binary_sensor.py
+++ b/homeassistant/components/keenetic_ndms2/binary_sensor.py
@@ -7,7 +7,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KeeneticRouter
from .const import DOMAIN, ROUTER
@@ -16,7 +16,7 @@ from .const import DOMAIN, ROUTER
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Keenetic NDMS2 component."""
router: KeeneticRouter = hass.data[DOMAIN][config_entry.entry_id][ROUTER]
diff --git a/homeassistant/components/keenetic_ndms2/device_tracker.py b/homeassistant/components/keenetic_ndms2/device_tracker.py
index 0f5166e16dd..4143611d6af 100644
--- a/homeassistant/components/keenetic_ndms2/device_tracker.py
+++ b/homeassistant/components/keenetic_ndms2/device_tracker.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN, ROUTER
@@ -26,7 +26,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Keenetic NDMS2 component."""
router: KeeneticRouter = hass.data[DOMAIN][config_entry.entry_id][ROUTER]
diff --git a/homeassistant/components/keenetic_ndms2/strings.json b/homeassistant/components/keenetic_ndms2/strings.json
index 765a3fc4d47..739846de0a8 100644
--- a/homeassistant/components/keenetic_ndms2/strings.json
+++ b/homeassistant/components/keenetic_ndms2/strings.json
@@ -21,7 +21,7 @@
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_account%]",
"no_udn": "SSDP discovery info has no UDN",
- "not_keenetic_ndms2": "Discovered item is not a Keenetic router"
+ "not_keenetic_ndms2": "Discovered device is not a Keenetic router"
}
},
"options": {
@@ -33,7 +33,7 @@
"interfaces": "Choose interfaces to scan",
"try_hotspot": "Use 'ip hotspot' data (most accurate)",
"include_arp": "Use ARP data (ignored if hotspot data used)",
- "include_associated": "Use WiFi AP associations data (ignored if hotspot data used)"
+ "include_associated": "Use Wi-Fi AP associations data (ignored if hotspot data used)"
}
}
}
diff --git a/homeassistant/components/kegtron/sensor.py b/homeassistant/components/kegtron/sensor.py
index e0638fccea0..602c61f96ff 100644
--- a/homeassistant/components/kegtron/sensor.py
+++ b/homeassistant/components/kegtron/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -110,7 +110,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Kegtron BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/keymitt_ble/strings.json b/homeassistant/components/keymitt_ble/strings.json
index 2a1f428603e..5e7e895d222 100644
--- a/homeassistant/components/keymitt_ble/strings.json
+++ b/homeassistant/components/keymitt_ble/strings.json
@@ -34,7 +34,7 @@
"services": {
"calibrate": {
"name": "Calibrate",
- "description": "Calibration - Set depth, press & hold duration, and operation mode. Warning - this will send a push command to the device.",
+ "description": "Sets the depth, press or release duration, and operation mode. Warning - this will send a push command to the device.",
"fields": {
"entity_id": {
"name": "Entity",
@@ -42,15 +42,15 @@
},
"depth": {
"name": "Depth",
- "description": "Depth in percent."
+ "description": "How far to extend the push arm."
},
"duration": {
"name": "Duration",
- "description": "Duration in seconds."
+ "description": "How long to press or release."
},
"mode": {
"name": "[%key:common::config_flow::data::mode%]",
- "description": "Normal | invert | toggle."
+ "description": "The operation mode of the arm."
}
}
}
diff --git a/homeassistant/components/keymitt_ble/switch.py b/homeassistant/components/keymitt_ble/switch.py
index ca458c5020f..57d3af98062 100644
--- a/homeassistant/components/keymitt_ble/switch.py
+++ b/homeassistant/components/keymitt_ble/switch.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.typing import VolDictType
@@ -29,7 +29,9 @@ CALIBRATE_SCHEMA: VolDictType = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MicroBot based on a config entry."""
coordinator: MicroBotDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/kitchen_sink/__init__.py b/homeassistant/components/kitchen_sink/__init__.py
index 09a72fc529c..de8e521f0e8 100644
--- a/homeassistant/components/kitchen_sink/__init__.py
+++ b/homeassistant/components/kitchen_sink/__init__.py
@@ -70,11 +70,11 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
return True
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set the config entry up."""
# Set up demo platforms with config entry
await hass.config_entries.async_forward_entry_setups(
- config_entry, COMPONENTS_WITH_DEMO_PLATFORM
+ entry, COMPONENTS_WITH_DEMO_PLATFORM
)
# Create issues
@@ -85,7 +85,9 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
await _insert_statistics(hass)
# Start a reauth flow
- config_entry.async_start_reauth(hass)
+ entry.async_start_reauth(hass)
+
+ entry.async_on_unload(entry.add_update_listener(_async_update_listener))
# Notify backup listeners
hass.async_create_task(_notify_backup_listeners(hass), eager_start=False)
@@ -93,6 +95,11 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
return True
+async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Handle update."""
+ await hass.config_entries.async_reload(entry.entry_id)
+
+
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload config entry."""
# Notify backup listeners
@@ -289,7 +296,7 @@ async def _insert_statistics(hass: HomeAssistant) -> None:
metadata = {
"source": RECORDER_DOMAIN,
"name": None,
- "statistic_id": "sensor.statistics_issue_1",
+ "statistic_id": "sensor.statistics_issues_issue_1",
"unit_of_measurement": UnitOfVolume.CUBIC_METERS,
"has_mean": True,
"has_sum": False,
@@ -301,7 +308,7 @@ async def _insert_statistics(hass: HomeAssistant) -> None:
metadata = {
"source": RECORDER_DOMAIN,
"name": None,
- "statistic_id": "sensor.statistics_issue_2",
+ "statistic_id": "sensor.statistics_issues_issue_2",
"unit_of_measurement": "cats",
"has_mean": True,
"has_sum": False,
@@ -313,7 +320,7 @@ async def _insert_statistics(hass: HomeAssistant) -> None:
metadata = {
"source": RECORDER_DOMAIN,
"name": None,
- "statistic_id": "sensor.statistics_issue_3",
+ "statistic_id": "sensor.statistics_issues_issue_3",
"unit_of_measurement": UnitOfVolume.CUBIC_METERS,
"has_mean": True,
"has_sum": False,
@@ -325,7 +332,7 @@ async def _insert_statistics(hass: HomeAssistant) -> None:
metadata = {
"source": RECORDER_DOMAIN,
"name": None,
- "statistic_id": "sensor.statistics_issue_4",
+ "statistic_id": "sensor.statistics_issues_issue_4",
"unit_of_measurement": UnitOfVolume.CUBIC_METERS,
"has_mean": True,
"has_sum": False,
diff --git a/homeassistant/components/kitchen_sink/backup.py b/homeassistant/components/kitchen_sink/backup.py
index 44ac0456105..46b204845ad 100644
--- a/homeassistant/components/kitchen_sink/backup.py
+++ b/homeassistant/components/kitchen_sink/backup.py
@@ -7,7 +7,13 @@ from collections.abc import AsyncIterator, Callable, Coroutine
import logging
from typing import Any
-from homeassistant.components.backup import AddonInfo, AgentBackup, BackupAgent, Folder
+from homeassistant.components.backup import (
+ AddonInfo,
+ AgentBackup,
+ BackupAgent,
+ BackupNotFound,
+ Folder,
+)
from homeassistant.core import HomeAssistant, callback
from . import DATA_BACKUP_AGENT_LISTENERS, DOMAIN
@@ -110,9 +116,9 @@ class KitchenSinkBackupAgent(BackupAgent):
self,
backup_id: str,
**kwargs: Any,
- ) -> AgentBackup | None:
+ ) -> AgentBackup:
"""Return a backup."""
for backup in self._uploads:
if backup.backup_id == backup_id:
return backup
- return None
+ raise BackupNotFound(f"Backup {backup_id} not found")
diff --git a/homeassistant/components/kitchen_sink/button.py b/homeassistant/components/kitchen_sink/button.py
index 5c62c4b32d1..1ee9bd78095 100644
--- a/homeassistant/components/kitchen_sink/button.py
+++ b/homeassistant/components/kitchen_sink/button.py
@@ -7,7 +7,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -15,7 +15,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo button platform."""
async_add_entities(
diff --git a/homeassistant/components/kitchen_sink/config_flow.py b/homeassistant/components/kitchen_sink/config_flow.py
index 019d1dddcad..e1ffe334038 100644
--- a/homeassistant/components/kitchen_sink/config_flow.py
+++ b/homeassistant/components/kitchen_sink/config_flow.py
@@ -12,7 +12,9 @@ from homeassistant.config_entries import (
ConfigEntry,
ConfigFlow,
ConfigFlowResult,
+ ConfigSubentryFlow,
OptionsFlow,
+ SubentryFlowResult,
)
from homeassistant.core import callback
@@ -35,6 +37,14 @@ class KitchenSinkConfigFlow(ConfigFlow, domain=DOMAIN):
"""Get the options flow for this handler."""
return OptionsFlowHandler()
+ @classmethod
+ @callback
+ def async_get_supported_subentry_types(
+ cls, config_entry: ConfigEntry
+ ) -> dict[str, type[ConfigSubentryFlow]]:
+ """Return subentries supported by this handler."""
+ return {"entity": SubentryFlowHandler}
+
async def async_step_import(self, import_data: dict[str, Any]) -> ConfigFlowResult:
"""Set the config entry up from yaml."""
return self.async_create_entry(title="Kitchen Sink", data=import_data)
@@ -94,3 +104,60 @@ class OptionsFlowHandler(OptionsFlow):
}
),
)
+
+
+class SubentryFlowHandler(ConfigSubentryFlow):
+ """Handle subentry flow."""
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """User flow to create a sensor subentry."""
+ return await self.async_step_add_sensor()
+
+ async def async_step_add_sensor(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Add a new sensor."""
+ if user_input is not None:
+ title = user_input.pop("name")
+ return self.async_create_entry(data=user_input, title=title)
+
+ return self.async_show_form(
+ step_id="add_sensor",
+ data_schema=vol.Schema(
+ {
+ vol.Required("name"): str,
+ vol.Required("state"): int,
+ }
+ ),
+ )
+
+ async def async_step_reconfigure(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Reconfigure a sensor subentry."""
+ return await self.async_step_reconfigure_sensor()
+
+ async def async_step_reconfigure_sensor(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Reconfigure a sensor."""
+ if user_input is not None:
+ title = user_input.pop("name")
+ return self.async_update_and_abort(
+ self._get_reconfigure_entry(),
+ self._get_reconfigure_subentry(),
+ data=user_input,
+ title=title,
+ )
+
+ return self.async_show_form(
+ step_id="reconfigure_sensor",
+ data_schema=vol.Schema(
+ {
+ vol.Required("name"): str,
+ vol.Required("state"): int,
+ }
+ ),
+ )
diff --git a/homeassistant/components/kitchen_sink/image.py b/homeassistant/components/kitchen_sink/image.py
index 504b36464f5..130317f4bc5 100644
--- a/homeassistant/components/kitchen_sink/image.py
+++ b/homeassistant/components/kitchen_sink/image.py
@@ -7,7 +7,10 @@ from pathlib import Path
from homeassistant.components.image import ImageEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
@@ -35,7 +38,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Everything but the Kitchen Sink config entry."""
await async_setup_platform(hass, {}, async_add_entities)
diff --git a/homeassistant/components/kitchen_sink/lawn_mower.py b/homeassistant/components/kitchen_sink/lawn_mower.py
index 51814fb262d..18a3f3dee77 100644
--- a/homeassistant/components/kitchen_sink/lawn_mower.py
+++ b/homeassistant/components/kitchen_sink/lawn_mower.py
@@ -9,7 +9,10 @@ from homeassistant.components.lawn_mower import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -71,7 +74,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Everything but the Kitchen Sink config entry."""
await async_setup_platform(hass, {}, async_add_entities)
diff --git a/homeassistant/components/kitchen_sink/lock.py b/homeassistant/components/kitchen_sink/lock.py
index 80ecc57d0d9..63566482cdf 100644
--- a/homeassistant/components/kitchen_sink/lock.py
+++ b/homeassistant/components/kitchen_sink/lock.py
@@ -7,7 +7,10 @@ from typing import Any
from homeassistant.components.lock import LockEntity, LockEntityFeature, LockState
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -49,7 +52,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Everything but the Kitchen Sink config entry."""
await async_setup_platform(hass, {}, async_add_entities)
diff --git a/homeassistant/components/kitchen_sink/notify.py b/homeassistant/components/kitchen_sink/notify.py
index fb34a36f0b7..be5bad58109 100644
--- a/homeassistant/components/kitchen_sink/notify.py
+++ b/homeassistant/components/kitchen_sink/notify.py
@@ -7,7 +7,7 @@ from homeassistant.components.notify import NotifyEntity, NotifyEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -15,7 +15,7 @@ from . import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo notify entity platform."""
async_add_entities(
diff --git a/homeassistant/components/kitchen_sink/sensor.py b/homeassistant/components/kitchen_sink/sensor.py
index 95e56c276e4..19d1b31aeab 100644
--- a/homeassistant/components/kitchen_sink/sensor.py
+++ b/homeassistant/components/kitchen_sink/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UNDEFINED, StateType, UndefinedType
from . import DOMAIN
@@ -21,7 +21,7 @@ from .device import async_create_device
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Everything but the Kitchen Sink config entry."""
async_create_device(
@@ -90,6 +90,23 @@ async def async_setup_entry(
]
)
+ for subentry_id, subentry in config_entry.subentries.items():
+ async_add_entities(
+ [
+ DemoSensor(
+ device_unique_id=subentry_id,
+ unique_id=subentry_id,
+ device_name=subentry.title,
+ entity_name=None,
+ state=subentry.data["state"],
+ device_class=None,
+ state_class=None,
+ unit_of_measurement=None,
+ )
+ ],
+ config_subentry_id=subentry_id,
+ )
+
class DemoSensor(SensorEntity):
"""Representation of a Demo sensor."""
diff --git a/homeassistant/components/kitchen_sink/strings.json b/homeassistant/components/kitchen_sink/strings.json
index c03f909e617..e0cdf75b707 100644
--- a/homeassistant/components/kitchen_sink/strings.json
+++ b/homeassistant/components/kitchen_sink/strings.json
@@ -9,6 +9,31 @@
}
}
},
+ "config_subentries": {
+ "entity": {
+ "step": {
+ "add_sensor": {
+ "description": "Configure the new sensor",
+ "data": {
+ "name": "[%key:common::config_flow::data::name%]",
+ "state": "Initial state"
+ }
+ },
+ "reconfigure_sensor": {
+ "description": "Reconfigure the sensor",
+ "data": {
+ "name": "[%key:component::kitchen_sink::config_subentries::entity::step::reconfigure_sensor::data::state%]",
+ "state": "Initial state"
+ }
+ }
+ },
+ "initiate_flow": {
+ "user": "Add sensor",
+ "reconfigure": "Reconfigure sensor"
+ },
+ "entry_type": "Sensor"
+ }
+ },
"options": {
"step": {
"init": {
diff --git a/homeassistant/components/kitchen_sink/switch.py b/homeassistant/components/kitchen_sink/switch.py
index 68a8312b496..45d3cb14eca 100644
--- a/homeassistant/components/kitchen_sink/switch.py
+++ b/homeassistant/components/kitchen_sink/switch.py
@@ -8,7 +8,7 @@ from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
from .device import async_create_device
@@ -17,7 +17,7 @@ from .device import async_create_device
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the demo switch platform."""
async_create_device(
diff --git a/homeassistant/components/kitchen_sink/weather.py b/homeassistant/components/kitchen_sink/weather.py
index e94e823c692..a6b7cc69d05 100644
--- a/homeassistant/components/kitchen_sink/weather.py
+++ b/homeassistant/components/kitchen_sink/weather.py
@@ -26,7 +26,7 @@ from homeassistant.components.weather import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfPressure, UnitOfSpeed, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_interval
from homeassistant.util import dt as dt_util
@@ -56,7 +56,7 @@ CONDITION_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Demo config entry."""
async_add_entities(
diff --git a/homeassistant/components/kmtronic/switch.py b/homeassistant/components/kmtronic/switch.py
index f00ecf8623c..b32f78b0e98 100644
--- a/homeassistant/components/kmtronic/switch.py
+++ b/homeassistant/components/kmtronic/switch.py
@@ -7,14 +7,16 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_REVERSE, DATA_COORDINATOR, DATA_HUB, DOMAIN, MANUFACTURER
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Config entry example."""
coordinator = hass.data[DOMAIN][entry.entry_id][DATA_COORDINATOR]
diff --git a/homeassistant/components/knocki/__init__.py b/homeassistant/components/knocki/__init__.py
index dfdf060e3b5..966f1dbf309 100644
--- a/homeassistant/components/knocki/__init__.py
+++ b/homeassistant/components/knocki/__init__.py
@@ -4,17 +4,14 @@ from __future__ import annotations
from knocki import Event, EventType, KnockiClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TOKEN, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .coordinator import KnockiCoordinator
+from .coordinator import KnockiConfigEntry, KnockiCoordinator
PLATFORMS: list[Platform] = [Platform.EVENT]
-type KnockiConfigEntry = ConfigEntry[KnockiCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: KnockiConfigEntry) -> bool:
"""Set up Knocki from a config entry."""
@@ -22,7 +19,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: KnockiConfigEntry) -> bo
session=async_get_clientsession(hass), token=entry.data[CONF_TOKEN]
)
- coordinator = KnockiCoordinator(hass, client)
+ coordinator = KnockiCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/knocki/coordinator.py b/homeassistant/components/knocki/coordinator.py
index c1e32b817e1..f5cc373f5c1 100644
--- a/homeassistant/components/knocki/coordinator.py
+++ b/homeassistant/components/knocki/coordinator.py
@@ -3,21 +3,29 @@
from knocki import Event, KnockiClient, KnockiConnectionError, Trigger
from homeassistant.components.event import DOMAIN as EVENT_DOMAIN
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN, LOGGER
+type KnockiConfigEntry = ConfigEntry[KnockiCoordinator]
+
class KnockiCoordinator(DataUpdateCoordinator[dict[int, Trigger]]):
"""The Knocki coordinator."""
- def __init__(self, hass: HomeAssistant, client: KnockiClient) -> None:
+ config_entry: KnockiConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: KnockiConfigEntry, client: KnockiClient
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
)
self.client = client
diff --git a/homeassistant/components/knocki/event.py b/homeassistant/components/knocki/event.py
index 74dc5a0f64c..1b148f03f1c 100644
--- a/homeassistant/components/knocki/event.py
+++ b/homeassistant/components/knocki/event.py
@@ -5,7 +5,7 @@ from knocki import Event, EventType, KnockiClient, Trigger
from homeassistant.components.event import EventEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import KnockiConfigEntry
from .const import DOMAIN
@@ -14,7 +14,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
entry: KnockiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Knocki from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/knx/__init__.py b/homeassistant/components/knx/__init__.py
index fa3439b02f4..8ad16642e45 100644
--- a/homeassistant/components/knx/__init__.py
+++ b/homeassistant/components/knx/__init__.py
@@ -486,7 +486,7 @@ class KNXModule:
transcoder := DPTBase.parse_transcoder(dpt)
):
self._address_filter_transcoder.update(
- {_filter: transcoder for _filter in _filters}
+ dict.fromkeys(_filters, transcoder)
)
return self.xknx.telegram_queue.register_telegram_received_cb(
diff --git a/homeassistant/components/knx/binary_sensor.py b/homeassistant/components/knx/binary_sensor.py
index c629860351c..c11612f79bf 100644
--- a/homeassistant/components/knx/binary_sensor.py
+++ b/homeassistant/components/knx/binary_sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.restore_state import RestoreEntity
@@ -45,7 +45,7 @@ from .storage.const import CONF_ENTITY, CONF_GA_PASSIVE, CONF_GA_SENSOR, CONF_GA
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the KNX binary sensor platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/button.py b/homeassistant/components/knx/button.py
index 5a2add5dcd7..538299a0556 100644
--- a/homeassistant/components/knx/button.py
+++ b/homeassistant/components/knx/button.py
@@ -8,7 +8,7 @@ from homeassistant import config_entries
from homeassistant.components.button import ButtonEntity
from homeassistant.const import CONF_ENTITY_CATEGORY, CONF_NAME, CONF_PAYLOAD, Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNXModule
@@ -19,7 +19,7 @@ from .entity import KnxYamlEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the KNX binary sensor platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/climate.py b/homeassistant/components/knx/climate.py
index e3bb63581e7..fdce5e0c470 100644
--- a/homeassistant/components/knx/climate.py
+++ b/homeassistant/components/knx/climate.py
@@ -34,7 +34,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNXModule
@@ -49,7 +49,7 @@ CONTROLLER_MODES_INV = {value: key for key, value in CONTROLLER_MODES.items()}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/cover.py b/homeassistant/components/knx/cover.py
index 2d38426a687..3c5752b990c 100644
--- a/homeassistant/components/knx/cover.py
+++ b/homeassistant/components/knx/cover.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNXModule
@@ -34,7 +34,7 @@ from .schema import CoverSchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up cover(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/date.py b/homeassistant/components/knx/date.py
index 8f65ac8a952..7980e6a2bc3 100644
--- a/homeassistant/components/knx/date.py
+++ b/homeassistant/components/knx/date.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType
@@ -36,7 +36,7 @@ from .entity import KnxYamlEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/datetime.py b/homeassistant/components/knx/datetime.py
index b75e1a14f67..7701597a8ef 100644
--- a/homeassistant/components/knx/datetime.py
+++ b/homeassistant/components/knx/datetime.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType
from homeassistant.util import dt as dt_util
@@ -37,7 +37,7 @@ from .entity import KnxYamlEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/expose.py b/homeassistant/components/knx/expose.py
index 6585b848d8a..461e6f25879 100644
--- a/homeassistant/components/knx/expose.py
+++ b/homeassistant/components/knx/expose.py
@@ -30,6 +30,7 @@ from homeassistant.exceptions import TemplateError
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, StateType
+from homeassistant.util import dt as dt_util
from .const import CONF_RESPOND_TO_READ, KNX_ADDRESS
from .schema import ExposeSchema
@@ -217,7 +218,7 @@ class KNXExposeTime:
self.device = xknx_device_cls(
self.xknx,
name=expose_type.capitalize(),
- localtime=True,
+ localtime=dt_util.get_default_time_zone(),
group_address=config[KNX_ADDRESS],
)
diff --git a/homeassistant/components/knx/fan.py b/homeassistant/components/knx/fan.py
index 75d91e48048..926b6458706 100644
--- a/homeassistant/components/knx/fan.py
+++ b/homeassistant/components/knx/fan.py
@@ -11,7 +11,7 @@ from homeassistant import config_entries
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.const import CONF_ENTITY_CATEGORY, CONF_NAME, Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from homeassistant.util.percentage import (
percentage_to_ranged_value,
@@ -30,7 +30,7 @@ DEFAULT_PERCENTAGE: Final = 50
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up fan(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/light.py b/homeassistant/components/knx/light.py
index 33edc19fb1c..865cfdc6e25 100644
--- a/homeassistant/components/knx/light.py
+++ b/homeassistant/components/knx/light.py
@@ -22,7 +22,7 @@ from homeassistant.components.light import (
from homeassistant.const import CONF_ENTITY_CATEGORY, CONF_NAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.typing import ConfigType
@@ -61,7 +61,7 @@ from .storage.entity_store_schema import LightColorMode
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up light(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/manifest.json b/homeassistant/components/knx/manifest.json
index 86c050443e3..bde6dfa226f 100644
--- a/homeassistant/components/knx/manifest.json
+++ b/homeassistant/components/knx/manifest.json
@@ -10,9 +10,9 @@
"iot_class": "local_push",
"loggers": ["xknx", "xknxproject"],
"requirements": [
- "xknx==3.5.0",
- "xknxproject==3.8.1",
- "knx-frontend==2025.1.30.194235"
+ "xknx==3.6.0",
+ "xknxproject==3.8.2",
+ "knx-frontend==2025.3.8.214559"
],
"single_config_entry": true
}
diff --git a/homeassistant/components/knx/notify.py b/homeassistant/components/knx/notify.py
index 245de2e937e..97980ab3d36 100644
--- a/homeassistant/components/knx/notify.py
+++ b/homeassistant/components/knx/notify.py
@@ -9,7 +9,7 @@ from homeassistant import config_entries
from homeassistant.components.notify import NotifyEntity
from homeassistant.const import CONF_ENTITY_CATEGORY, CONF_NAME, CONF_TYPE, Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNXModule
@@ -20,7 +20,7 @@ from .entity import KnxYamlEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up notify(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/number.py b/homeassistant/components/knx/number.py
index 27e4ff743ab..67e8778accc 100644
--- a/homeassistant/components/knx/number.py
+++ b/homeassistant/components/knx/number.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNXModule
@@ -31,7 +31,7 @@ from .schema import NumberSchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/scene.py b/homeassistant/components/knx/scene.py
index dfd226d72b1..f5361a6e7da 100644
--- a/homeassistant/components/knx/scene.py
+++ b/homeassistant/components/knx/scene.py
@@ -10,7 +10,7 @@ from homeassistant import config_entries
from homeassistant.components.scene import Scene
from homeassistant.const import CONF_ENTITY_CATEGORY, CONF_NAME, Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNXModule
@@ -22,7 +22,7 @@ from .schema import SceneSchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up scene(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/select.py b/homeassistant/components/knx/select.py
index b499e3c601d..e80fa66f9d4 100644
--- a/homeassistant/components/knx/select.py
+++ b/homeassistant/components/knx/select.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType
@@ -36,7 +36,7 @@ from .schema import SelectSchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/sensor.py b/homeassistant/components/knx/sensor.py
index fa4911aa4b7..8e537ea234e 100644
--- a/homeassistant/components/knx/sensor.py
+++ b/homeassistant/components/knx/sensor.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, StateType
from homeassistant.util.enum import try_parse_enum
@@ -112,7 +112,7 @@ SYSTEM_ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/services.py b/homeassistant/components/knx/services.py
index f0f760180f4..fc28e0850ed 100644
--- a/homeassistant/components/knx/services.py
+++ b/homeassistant/components/knx/services.py
@@ -126,7 +126,7 @@ async def service_event_register_modify(call: ServiceCall) -> None:
transcoder := DPTBase.parse_transcoder(dpt)
):
knx_module.group_address_transcoder.update(
- {_address: transcoder for _address in group_addresses}
+ dict.fromkeys(group_addresses, transcoder)
)
for group_address in group_addresses:
if group_address in knx_module.knx_event_callback.group_addresses:
diff --git a/homeassistant/components/knx/storage/entity_store_schema.py b/homeassistant/components/knx/storage/entity_store_schema.py
index d99ffa86f52..cde18a181ec 100644
--- a/homeassistant/components/knx/storage/entity_store_schema.py
+++ b/homeassistant/components/knx/storage/entity_store_schema.py
@@ -114,7 +114,7 @@ BINARY_SENSOR_SCHEMA = vol.Schema(
),
vol.Optional(CONF_RESET_AFTER): selector.NumberSelector(
selector.NumberSelectorConfig(
- min=0, max=10, step=0.1, unit_of_measurement="s"
+ min=0, max=600, step=0.1, unit_of_measurement="s"
)
),
},
diff --git a/homeassistant/components/knx/strings.json b/homeassistant/components/knx/strings.json
index dadc8e84796..10730d87ed1 100644
--- a/homeassistant/components/knx/strings.json
+++ b/homeassistant/components/knx/strings.json
@@ -397,7 +397,7 @@
},
"response": {
"name": "Send as Response",
- "description": "If set to `True`, the telegram will be sent as a `GroupValueResponse` instead of a `GroupValueWrite`."
+ "description": "Whether the telegram should be sent as a `GroupValueResponse` instead of a `GroupValueWrite`."
}
}
},
@@ -425,7 +425,7 @@
},
"remove": {
"name": "Remove event registration",
- "description": "If `True` the group address(es) will be removed."
+ "description": "Whether the group address(es) will be removed."
}
}
},
@@ -455,7 +455,7 @@
},
"remove": {
"name": "Remove exposure",
- "description": "If `True` the exposure will be removed. Only `address` is required for removal."
+ "description": "Whether the exposure should be removed. Only the 'Address' field is required for removal."
}
}
},
diff --git a/homeassistant/components/knx/switch.py b/homeassistant/components/knx/switch.py
index 725468cd6a9..730c5b788ff 100644
--- a/homeassistant/components/knx/switch.py
+++ b/homeassistant/components/knx/switch.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.restore_state import RestoreEntity
@@ -48,7 +48,7 @@ from .storage.const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch(es) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/text.py b/homeassistant/components/knx/text.py
index 2256afadbd9..9c2bb88f92b 100644
--- a/homeassistant/components/knx/text.py
+++ b/homeassistant/components/knx/text.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType
@@ -30,7 +30,7 @@ from .entity import KnxYamlEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor(s) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/time.py b/homeassistant/components/knx/time.py
index 1e82c324502..2c74ab18af3 100644
--- a/homeassistant/components/knx/time.py
+++ b/homeassistant/components/knx/time.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType
@@ -36,7 +36,7 @@ from .entity import KnxYamlEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/knx/weather.py b/homeassistant/components/knx/weather.py
index a1e5c0efe48..342ab445611 100644
--- a/homeassistant/components/knx/weather.py
+++ b/homeassistant/components/knx/weather.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import KNXModule
@@ -28,7 +28,7 @@ from .schema import WeatherSchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch(es) for KNX platform."""
knx_module = hass.data[KNX_MODULE_KEY]
diff --git a/homeassistant/components/kodi/media_player.py b/homeassistant/components/kodi/media_player.py
index bbddbd9f348..c4a2436548a 100644
--- a/homeassistant/components/kodi/media_player.py
+++ b/homeassistant/components/kodi/media_player.py
@@ -46,7 +46,10 @@ from homeassistant.helpers import (
entity_platform,
)
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_time_interval
from homeassistant.helpers.network import is_internal_request
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, VolDictType
@@ -206,7 +209,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Kodi media player platform."""
platform = entity_platform.async_get_current_platform()
diff --git a/homeassistant/components/konnected/binary_sensor.py b/homeassistant/components/konnected/binary_sensor.py
index 75c381c53f2..3f1a27302d8 100644
--- a/homeassistant/components/konnected/binary_sensor.py
+++ b/homeassistant/components/konnected/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN as KONNECTED_DOMAIN
@@ -21,7 +21,7 @@ from .const import DOMAIN as KONNECTED_DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors attached to a Konnected device from a config entry."""
data = hass.data[KONNECTED_DOMAIN]
diff --git a/homeassistant/components/konnected/sensor.py b/homeassistant/components/konnected/sensor.py
index 6191f98f179..cd36c217627 100644
--- a/homeassistant/components/konnected/sensor.py
+++ b/homeassistant/components/konnected/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN as KONNECTED_DOMAIN, SIGNAL_DS18B20_NEW
@@ -43,7 +43,7 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors attached to a Konnected device from a config entry."""
data = hass.data[KONNECTED_DOMAIN]
diff --git a/homeassistant/components/konnected/switch.py b/homeassistant/components/konnected/switch.py
index 65b99d623f1..58311502cbe 100644
--- a/homeassistant/components/konnected/switch.py
+++ b/homeassistant/components/konnected/switch.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_ACTIVATION,
@@ -33,7 +33,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches attached to a Konnected device from a config entry."""
data = hass.data[KONNECTED_DOMAIN]
diff --git a/homeassistant/components/kostal_plenticore/coordinator.py b/homeassistant/components/kostal_plenticore/coordinator.py
index 5f4393146f0..a404a997663 100644
--- a/homeassistant/components/kostal_plenticore/coordinator.py
+++ b/homeassistant/components/kostal_plenticore/coordinator.py
@@ -16,6 +16,7 @@ from pykoplenti import (
ExtendedApiClient,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PASSWORD, EVENT_HOMEASSISTANT_STOP
from homeassistant.core import CALLBACK_TYPE, HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@@ -162,9 +163,12 @@ class DataUpdateCoordinatorMixin:
class PlenticoreUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
"""Base implementation of DataUpdateCoordinator for Plenticore data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
logger: logging.Logger,
name: str,
update_inverval: timedelta,
@@ -174,6 +178,7 @@ class PlenticoreUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
super().__init__(
hass=hass,
logger=logger,
+ config_entry=config_entry,
name=name,
update_interval=update_inverval,
)
@@ -240,9 +245,12 @@ class SettingDataUpdateCoordinator(
class PlenticoreSelectUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
"""Base implementation of DataUpdateCoordinator for Plenticore data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
logger: logging.Logger,
name: str,
update_inverval: timedelta,
@@ -252,6 +260,7 @@ class PlenticoreSelectUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
super().__init__(
hass=hass,
logger=logger,
+ config_entry=config_entry,
name=name,
update_interval=update_inverval,
)
diff --git a/homeassistant/components/kostal_plenticore/number.py b/homeassistant/components/kostal_plenticore/number.py
index 8afe69a7749..7efb00cf8f4 100644
--- a/homeassistant/components/kostal_plenticore/number.py
+++ b/homeassistant/components/kostal_plenticore/number.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -73,7 +73,9 @@ NUMBER_SETTINGS_DATA = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Kostal Plenticore Number entities."""
plenticore = hass.data[DOMAIN][entry.entry_id]
@@ -82,11 +84,7 @@ async def async_setup_entry(
available_settings_data = await plenticore.client.get_settings()
settings_data_update_coordinator = SettingDataUpdateCoordinator(
- hass,
- _LOGGER,
- "Settings Data",
- timedelta(seconds=30),
- plenticore,
+ hass, entry, _LOGGER, "Settings Data", timedelta(seconds=30), plenticore
)
for description in NUMBER_SETTINGS_DATA:
diff --git a/homeassistant/components/kostal_plenticore/select.py b/homeassistant/components/kostal_plenticore/select.py
index 73f3f94eda8..61929b9fadc 100644
--- a/homeassistant/components/kostal_plenticore/select.py
+++ b/homeassistant/components/kostal_plenticore/select.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -42,18 +42,16 @@ SELECT_SETTINGS_DATA = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add kostal plenticore Select widget."""
plenticore: Plenticore = hass.data[DOMAIN][entry.entry_id]
available_settings_data = await plenticore.client.get_settings()
select_data_update_coordinator = SelectDataUpdateCoordinator(
- hass,
- _LOGGER,
- "Settings Data",
- timedelta(seconds=30),
- plenticore,
+ hass, entry, _LOGGER, "Settings Data", timedelta(seconds=30), plenticore
)
entities = []
diff --git a/homeassistant/components/kostal_plenticore/sensor.py b/homeassistant/components/kostal_plenticore/sensor.py
index 67de34f2fce..1be7fb06e7b 100644
--- a/homeassistant/components/kostal_plenticore/sensor.py
+++ b/homeassistant/components/kostal_plenticore/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -807,7 +807,9 @@ SENSOR_PROCESS_DATA = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add kostal plenticore Sensors."""
plenticore = hass.data[DOMAIN][entry.entry_id]
@@ -816,11 +818,7 @@ async def async_setup_entry(
available_process_data = await plenticore.client.get_process_data()
process_data_update_coordinator = ProcessDataUpdateCoordinator(
- hass,
- _LOGGER,
- "Process Data",
- timedelta(seconds=10),
- plenticore,
+ hass, entry, _LOGGER, "Process Data", timedelta(seconds=10), plenticore
)
for description in SENSOR_PROCESS_DATA:
module_id = description.module_id
diff --git a/homeassistant/components/kostal_plenticore/switch.py b/homeassistant/components/kostal_plenticore/switch.py
index 7ce2d468c88..e3d5f830c78 100644
--- a/homeassistant/components/kostal_plenticore/switch.py
+++ b/homeassistant/components/kostal_plenticore/switch.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -48,7 +48,9 @@ SWITCH_SETTINGS_DATA = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add kostal plenticore Switch."""
plenticore = hass.data[DOMAIN][entry.entry_id]
@@ -57,11 +59,7 @@ async def async_setup_entry(
available_settings_data = await plenticore.client.get_settings()
settings_data_update_coordinator = SettingDataUpdateCoordinator(
- hass,
- _LOGGER,
- "Settings Data",
- timedelta(seconds=30),
- plenticore,
+ hass, entry, _LOGGER, "Settings Data", timedelta(seconds=30), plenticore
)
for description in SWITCH_SETTINGS_DATA:
if (
diff --git a/homeassistant/components/kraken/sensor.py b/homeassistant/components/kraken/sensor.py
index 37fee795783..1d3f36d29e4 100644
--- a/homeassistant/components/kraken/sensor.py
+++ b/homeassistant/components/kraken/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -139,7 +139,7 @@ SENSOR_TYPES: tuple[KrakenSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add kraken entities from a config_entry."""
diff --git a/homeassistant/components/kulersky/light.py b/homeassistant/components/kulersky/light.py
index 552507ef50b..bcc3f32dceb 100644
--- a/homeassistant/components/kulersky/light.py
+++ b/homeassistant/components/kulersky/light.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_interval
from .const import DATA_ADDRESSES, DATA_DISCOVERY_SUBSCRIPTION, DOMAIN
@@ -31,7 +31,7 @@ DISCOVERY_INTERVAL = timedelta(seconds=60)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Kuler sky light devices."""
diff --git a/homeassistant/components/lacrosse_view/__init__.py b/homeassistant/components/lacrosse_view/__init__.py
index d977af418a2..e98d1d421be 100644
--- a/homeassistant/components/lacrosse_view/__init__.py
+++ b/homeassistant/components/lacrosse_view/__init__.py
@@ -30,7 +30,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except LoginError as error:
raise ConfigEntryAuthFailed from error
- coordinator = LaCrosseUpdateCoordinator(hass, api, entry)
+ coordinator = LaCrosseUpdateCoordinator(hass, entry, api)
_LOGGER.debug("First refresh")
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/lacrosse_view/coordinator.py b/homeassistant/components/lacrosse_view/coordinator.py
index 5ec02a86709..16d7e8b2bb8 100644
--- a/homeassistant/components/lacrosse_view/coordinator.py
+++ b/homeassistant/components/lacrosse_view/coordinator.py
@@ -10,10 +10,10 @@ from lacrosse_view import HTTPError, LaCrosse, Location, LoginError, Sensor
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
+from homeassistant.exceptions import ConfigEntryAuthFailed
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import SCAN_INTERVAL
+from .const import DOMAIN, SCAN_INTERVAL
_LOGGER = logging.getLogger(__name__)
@@ -26,12 +26,14 @@ class LaCrosseUpdateCoordinator(DataUpdateCoordinator[list[Sensor]]):
name: str
id: str
hass: HomeAssistant
+ devices: list[Sensor] | None = None
+ config_entry: ConfigEntry
def __init__(
self,
hass: HomeAssistant,
- api: LaCrosse,
entry: ConfigEntry,
+ api: LaCrosse,
) -> None:
"""Initialize DataUpdateCoordinator for LaCrosse View."""
self.api = api
@@ -44,6 +46,7 @@ class LaCrosseUpdateCoordinator(DataUpdateCoordinator[list[Sensor]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name="LaCrosse View",
update_interval=timedelta(seconds=SCAN_INTERVAL),
)
@@ -60,24 +63,46 @@ class LaCrosseUpdateCoordinator(DataUpdateCoordinator[list[Sensor]]):
except LoginError as error:
raise ConfigEntryAuthFailed from error
+ if self.devices is None:
+ _LOGGER.debug("Getting devices")
+ try:
+ self.devices = await self.api.get_devices(
+ location=Location(id=self.id, name=self.name),
+ )
+ except HTTPError as error:
+ raise UpdateFailed from error
+
try:
# Fetch last hour of data
- sensors = await self.api.get_sensors(
- location=Location(id=self.id, name=self.name),
- tz=self.hass.config.time_zone,
- start=str(now - 3600),
- end=str(now),
- )
- except HTTPError as error:
- raise ConfigEntryNotReady from error
+ for sensor in self.devices:
+ data = await self.api.get_sensor_status(
+ sensor=sensor,
+ tz=self.hass.config.time_zone,
+ )
+ _LOGGER.debug("Got data: %s", data)
- _LOGGER.debug("Got data: %s", sensors)
+ if data_error := data.get("error"):
+ if data_error == "no_readings":
+ sensor.data = None
+ _LOGGER.debug("No readings for %s", sensor.name)
+ continue
+ _LOGGER.debug("Error: %s", data_error)
+ raise UpdateFailed(
+ translation_domain=DOMAIN, translation_key="update_error"
+ )
+
+ sensor.data = data["data"]["current"]
+
+ except HTTPError as error:
+ raise UpdateFailed(
+ translation_domain=DOMAIN, translation_key="update_error"
+ ) from error
# Verify that we have permission to read the sensors
- for sensor in sensors:
+ for sensor in self.devices:
if not sensor.permissions.get("read", False):
raise ConfigEntryAuthFailed(
f"This account does not have permission to read {sensor.name}"
)
- return sensors
+ return self.devices
diff --git a/homeassistant/components/lacrosse_view/sensor.py b/homeassistant/components/lacrosse_view/sensor.py
index fceddeb9b2c..667fcbb8dcc 100644
--- a/homeassistant/components/lacrosse_view/sensor.py
+++ b/homeassistant/components/lacrosse_view/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -48,7 +48,7 @@ def get_value(sensor: Sensor, field: str) -> float | int | str | None:
field_data = sensor.data.get(field) if sensor.data is not None else None
if field_data is None:
return None
- value = field_data["values"][-1]["s"]
+ value = field_data["spot"]["value"]
try:
value = float(value)
except ValueError:
@@ -64,6 +64,7 @@ SENSOR_DESCRIPTIONS = {
state_class=SensorStateClass.MEASUREMENT,
value_fn=get_value,
native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+ suggested_display_precision=2,
),
"Humidity": LaCrosseSensorEntityDescription(
key="Humidity",
@@ -71,6 +72,7 @@ SENSOR_DESCRIPTIONS = {
state_class=SensorStateClass.MEASUREMENT,
value_fn=get_value,
native_unit_of_measurement=PERCENTAGE,
+ suggested_display_precision=2,
),
"HeatIndex": LaCrosseSensorEntityDescription(
key="HeatIndex",
@@ -79,6 +81,7 @@ SENSOR_DESCRIPTIONS = {
state_class=SensorStateClass.MEASUREMENT,
value_fn=get_value,
native_unit_of_measurement=UnitOfTemperature.FAHRENHEIT,
+ suggested_display_precision=2,
),
"WindSpeed": LaCrosseSensorEntityDescription(
key="WindSpeed",
@@ -86,6 +89,7 @@ SENSOR_DESCRIPTIONS = {
value_fn=get_value,
native_unit_of_measurement=UnitOfSpeed.KILOMETERS_PER_HOUR,
device_class=SensorDeviceClass.WIND_SPEED,
+ suggested_display_precision=2,
),
"Rain": LaCrosseSensorEntityDescription(
key="Rain",
@@ -93,12 +97,15 @@ SENSOR_DESCRIPTIONS = {
value_fn=get_value,
native_unit_of_measurement=UnitOfPrecipitationDepth.MILLIMETERS,
device_class=SensorDeviceClass.PRECIPITATION,
+ suggested_display_precision=2,
),
"WindHeading": LaCrosseSensorEntityDescription(
key="WindHeading",
translation_key="wind_heading",
value_fn=get_value,
native_unit_of_measurement=DEGREE,
+ suggested_display_precision=2,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
"WetDry": LaCrosseSensorEntityDescription(
key="WetDry",
@@ -117,6 +124,7 @@ SENSOR_DESCRIPTIONS = {
value_fn=get_value,
device_class=SensorDeviceClass.ATMOSPHERIC_PRESSURE,
native_unit_of_measurement=UnitOfPressure.HPA,
+ suggested_display_precision=2,
),
"FeelsLike": LaCrosseSensorEntityDescription(
key="FeelsLike",
@@ -125,6 +133,7 @@ SENSOR_DESCRIPTIONS = {
value_fn=get_value,
device_class=SensorDeviceClass.TEMPERATURE,
native_unit_of_measurement=UnitOfTemperature.FAHRENHEIT,
+ suggested_display_precision=2,
),
"WindChill": LaCrosseSensorEntityDescription(
key="WindChill",
@@ -133,6 +142,7 @@ SENSOR_DESCRIPTIONS = {
value_fn=get_value,
device_class=SensorDeviceClass.TEMPERATURE,
native_unit_of_measurement=UnitOfTemperature.FAHRENHEIT,
+ suggested_display_precision=2,
),
}
# map of API returned unit of measurement strings to their corresponding unit of measurement
@@ -149,7 +159,7 @@ UNIT_OF_MEASUREMENT_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LaCrosse View from a config entry."""
coordinator: DataUpdateCoordinator[list[Sensor]] = hass.data[DOMAIN][
@@ -226,7 +236,6 @@ class LaCrosseViewSensor(
name=sensor.name,
manufacturer="LaCrosse Technology",
model=sensor.model,
- via_device=(DOMAIN, sensor.location.id),
)
self.index = index
diff --git a/homeassistant/components/lacrosse_view/strings.json b/homeassistant/components/lacrosse_view/strings.json
index 8dc27ba259e..c5d9a11e49a 100644
--- a/homeassistant/components/lacrosse_view/strings.json
+++ b/homeassistant/components/lacrosse_view/strings.json
@@ -42,5 +42,10 @@
"name": "Wind chill"
}
}
+ },
+ "exceptions": {
+ "update_error": {
+ "message": "Error updating data"
+ }
}
}
diff --git a/homeassistant/components/lamarzocco/__init__.py b/homeassistant/components/lamarzocco/__init__.py
index d20616e1940..25c8fd1091e 100644
--- a/homeassistant/components/lamarzocco/__init__.py
+++ b/homeassistant/components/lamarzocco/__init__.py
@@ -61,6 +61,42 @@ async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry) -
client=client,
)
+ # initialize the firmware update coordinator early to check the firmware version
+ firmware_device = LaMarzoccoMachine(
+ model=entry.data[CONF_MODEL],
+ serial_number=entry.unique_id,
+ name=entry.data[CONF_NAME],
+ cloud_client=cloud_client,
+ )
+
+ firmware_coordinator = LaMarzoccoFirmwareUpdateCoordinator(
+ hass, entry, firmware_device
+ )
+ await firmware_coordinator.async_config_entry_first_refresh()
+ gateway_version = version.parse(
+ firmware_device.firmware[FirmwareType.GATEWAY].current_version
+ )
+
+ if gateway_version >= version.parse("v5.0.9"):
+ # remove host from config entry, it is not supported anymore
+ data = {k: v for k, v in entry.data.items() if k != CONF_HOST}
+ hass.config_entries.async_update_entry(
+ entry,
+ data=data,
+ )
+
+ elif gateway_version < version.parse("v3.4-rc5"):
+ # incompatible gateway firmware, create an issue
+ ir.async_create_issue(
+ hass,
+ DOMAIN,
+ "unsupported_gateway_firmware",
+ is_fixable=False,
+ severity=ir.IssueSeverity.ERROR,
+ translation_key="unsupported_gateway_firmware",
+ translation_placeholders={"gateway_version": str(gateway_version)},
+ )
+
# initialize local API
local_client: LaMarzoccoLocalClient | None = None
if (host := entry.data.get(CONF_HOST)) is not None:
@@ -117,30 +153,16 @@ async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry) -
coordinators = LaMarzoccoRuntimeData(
LaMarzoccoConfigUpdateCoordinator(hass, entry, device, local_client),
- LaMarzoccoFirmwareUpdateCoordinator(hass, entry, device),
+ firmware_coordinator,
LaMarzoccoStatisticsUpdateCoordinator(hass, entry, device),
)
# API does not like concurrent requests, so no asyncio.gather here
await coordinators.config_coordinator.async_config_entry_first_refresh()
- await coordinators.firmware_coordinator.async_config_entry_first_refresh()
await coordinators.statistics_coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinators
- gateway_version = device.firmware[FirmwareType.GATEWAY].current_version
- if version.parse(gateway_version) < version.parse("v3.4-rc5"):
- # incompatible gateway firmware, create an issue
- ir.async_create_issue(
- hass,
- DOMAIN,
- "unsupported_gateway_firmware",
- is_fixable=False,
- severity=ir.IssueSeverity.ERROR,
- translation_key="unsupported_gateway_firmware",
- translation_placeholders={"gateway_version": gateway_version},
- )
-
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
async def update_listener(
diff --git a/homeassistant/components/lamarzocco/binary_sensor.py b/homeassistant/components/lamarzocco/binary_sensor.py
index e36b53bc993..a98cddcda9c 100644
--- a/homeassistant/components/lamarzocco/binary_sensor.py
+++ b/homeassistant/components/lamarzocco/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LaMarzoccoConfigEntry
from .entity import LaMarzoccoEntity, LaMarzoccoEntityDescription, LaMarzoccScaleEntity
@@ -71,7 +71,7 @@ SCALE_ENTITIES: tuple[LaMarzoccoBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor entities."""
coordinator = entry.runtime_data.config_coordinator
@@ -83,7 +83,7 @@ async def async_setup_entry(
]
if (
- coordinator.device.model == MachineModel.LINEA_MINI
+ coordinator.device.model in (MachineModel.LINEA_MINI, MachineModel.LINEA_MINI_R)
and coordinator.device.config.scale
):
entities.extend(
diff --git a/homeassistant/components/lamarzocco/button.py b/homeassistant/components/lamarzocco/button.py
index 22e92f656ff..db51d610949 100644
--- a/homeassistant/components/lamarzocco/button.py
+++ b/homeassistant/components/lamarzocco/button.py
@@ -10,7 +10,7 @@ from pylamarzocco.exceptions import RequestNotSuccessful
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMarzoccoConfigEntry, LaMarzoccoUpdateCoordinator
@@ -53,7 +53,7 @@ ENTITIES: tuple[LaMarzoccoButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button entities."""
diff --git a/homeassistant/components/lamarzocco/calendar.py b/homeassistant/components/lamarzocco/calendar.py
index 1dcc7c324ac..4365bf56b2d 100644
--- a/homeassistant/components/lamarzocco/calendar.py
+++ b/homeassistant/components/lamarzocco/calendar.py
@@ -7,7 +7,7 @@ from pylamarzocco.models import LaMarzoccoWakeUpSleepEntry
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .coordinator import LaMarzoccoConfigEntry, LaMarzoccoUpdateCoordinator
@@ -32,7 +32,7 @@ DAY_OF_WEEK = [
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch entities and services."""
diff --git a/homeassistant/components/lamarzocco/manifest.json b/homeassistant/components/lamarzocco/manifest.json
index afd367b0f6e..73f00b2bdd0 100644
--- a/homeassistant/components/lamarzocco/manifest.json
+++ b/homeassistant/components/lamarzocco/manifest.json
@@ -37,5 +37,5 @@
"iot_class": "cloud_polling",
"loggers": ["pylamarzocco"],
"quality_scale": "platinum",
- "requirements": ["pylamarzocco==1.4.6"]
+ "requirements": ["pylamarzocco==1.4.9"]
}
diff --git a/homeassistant/components/lamarzocco/number.py b/homeassistant/components/lamarzocco/number.py
index 44b582fbf1a..08e9ad7e590 100644
--- a/homeassistant/components/lamarzocco/number.py
+++ b/homeassistant/components/lamarzocco/number.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMarzoccoConfigEntry, LaMarzoccoUpdateCoordinator
@@ -144,9 +144,12 @@ KEY_ENTITIES: tuple[LaMarzoccoKeyNumberEntityDescription, ...] = (
set_value_fn=lambda machine, value, key: machine.set_prebrew_time(
prebrew_off_time=value, key=key
),
- native_value_fn=lambda config, key: config.prebrew_configuration[key].off_time,
+ native_value_fn=lambda config, key: config.prebrew_configuration[key][
+ 0
+ ].off_time,
available_fn=lambda device: len(device.config.prebrew_configuration) > 0
- and device.config.prebrew_mode == PrebrewMode.PREBREW,
+ and device.config.prebrew_mode
+ in (PrebrewMode.PREBREW, PrebrewMode.PREBREW_ENABLED),
supported_fn=lambda coordinator: coordinator.device.model
!= MachineModel.GS3_MP,
),
@@ -162,9 +165,12 @@ KEY_ENTITIES: tuple[LaMarzoccoKeyNumberEntityDescription, ...] = (
set_value_fn=lambda machine, value, key: machine.set_prebrew_time(
prebrew_on_time=value, key=key
),
- native_value_fn=lambda config, key: config.prebrew_configuration[key].off_time,
+ native_value_fn=lambda config, key: config.prebrew_configuration[key][
+ 0
+ ].off_time,
available_fn=lambda device: len(device.config.prebrew_configuration) > 0
- and device.config.prebrew_mode == PrebrewMode.PREBREW,
+ and device.config.prebrew_mode
+ in (PrebrewMode.PREBREW, PrebrewMode.PREBREW_ENABLED),
supported_fn=lambda coordinator: coordinator.device.model
!= MachineModel.GS3_MP,
),
@@ -180,8 +186,8 @@ KEY_ENTITIES: tuple[LaMarzoccoKeyNumberEntityDescription, ...] = (
set_value_fn=lambda machine, value, key: machine.set_preinfusion_time(
preinfusion_time=value, key=key
),
- native_value_fn=lambda config, key: config.prebrew_configuration[
- key
+ native_value_fn=lambda config, key: config.prebrew_configuration[key][
+ 1
].preinfusion_time,
available_fn=lambda device: len(device.config.prebrew_configuration) > 0
and device.config.prebrew_mode == PrebrewMode.PREINFUSION,
@@ -220,7 +226,8 @@ SCALE_KEY_ENTITIES: tuple[LaMarzoccoKeyNumberEntityDescription, ...] = (
config.bbw_settings.doses[key] if config.bbw_settings else None
),
supported_fn=(
- lambda coordinator: coordinator.device.model == MachineModel.LINEA_MINI
+ lambda coordinator: coordinator.device.model
+ in (MachineModel.LINEA_MINI, MachineModel.LINEA_MINI_R)
and coordinator.device.config.scale is not None
),
),
@@ -230,7 +237,7 @@ SCALE_KEY_ENTITIES: tuple[LaMarzoccoKeyNumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number entities."""
coordinator = entry.runtime_data.config_coordinator
diff --git a/homeassistant/components/lamarzocco/select.py b/homeassistant/components/lamarzocco/select.py
index 7acb654f0d2..5ebe2d7b9da 100644
--- a/homeassistant/components/lamarzocco/select.py
+++ b/homeassistant/components/lamarzocco/select.py
@@ -19,7 +19,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMarzoccoConfigEntry
@@ -38,6 +38,7 @@ STEAM_LEVEL_LM_TO_HA = {value: key for key, value in STEAM_LEVEL_HA_TO_LM.items(
PREBREW_MODE_HA_TO_LM = {
"disabled": PrebrewMode.DISABLED,
"prebrew": PrebrewMode.PREBREW,
+ "prebrew_enabled": PrebrewMode.PREBREW_ENABLED,
"preinfusion": PrebrewMode.PREINFUSION,
}
@@ -88,6 +89,7 @@ ENTITIES: tuple[LaMarzoccoSelectEntityDescription, ...] = (
MachineModel.GS3_AV,
MachineModel.LINEA_MICRA,
MachineModel.LINEA_MINI,
+ MachineModel.LINEA_MINI_R,
),
),
LaMarzoccoSelectEntityDescription(
@@ -126,7 +128,7 @@ SCALE_ENTITIES: tuple[LaMarzoccoSelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select entities."""
coordinator = entry.runtime_data.config_coordinator
@@ -138,7 +140,7 @@ async def async_setup_entry(
]
if (
- coordinator.device.model == MachineModel.LINEA_MINI
+ coordinator.device.model in (MachineModel.LINEA_MINI, MachineModel.LINEA_MINI_R)
and coordinator.device.config.scale
):
entities.extend(
diff --git a/homeassistant/components/lamarzocco/sensor.py b/homeassistant/components/lamarzocco/sensor.py
index a2d6143daa5..0d4a5e53ebe 100644
--- a/homeassistant/components/lamarzocco/sensor.py
+++ b/homeassistant/components/lamarzocco/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LaMarzoccoConfigEntry, LaMarzoccoUpdateCoordinator
from .entity import LaMarzoccoEntity, LaMarzoccoEntityDescription, LaMarzoccScaleEntity
@@ -80,7 +80,7 @@ ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...] = (
BoilerType.STEAM
].current_temperature,
supported_fn=lambda coordinator: coordinator.device.model
- != MachineModel.LINEA_MINI,
+ not in (MachineModel.LINEA_MINI, MachineModel.LINEA_MINI_R),
),
)
@@ -125,7 +125,8 @@ SCALE_ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...] = (
device.config.scale.battery if device.config.scale else 0
),
supported_fn=(
- lambda coordinator: coordinator.device.model == MachineModel.LINEA_MINI
+ lambda coordinator: coordinator.device.model
+ in (MachineModel.LINEA_MINI, MachineModel.LINEA_MINI_R)
),
),
)
@@ -134,7 +135,7 @@ SCALE_ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor entities."""
config_coordinator = entry.runtime_data.config_coordinator
@@ -148,7 +149,8 @@ async def async_setup_entry(
]
if (
- config_coordinator.device.model == MachineModel.LINEA_MINI
+ config_coordinator.device.model
+ in (MachineModel.LINEA_MINI, MachineModel.LINEA_MINI_R)
and config_coordinator.device.config.scale
):
entities.extend(
diff --git a/homeassistant/components/lamarzocco/strings.json b/homeassistant/components/lamarzocco/strings.json
index 62050685c27..04853b8d0ca 100644
--- a/homeassistant/components/lamarzocco/strings.json
+++ b/homeassistant/components/lamarzocco/strings.json
@@ -148,6 +148,7 @@
"state": {
"disabled": "Disabled",
"prebrew": "Prebrew",
+ "prebrew_enabled": "Prebrew",
"preinfusion": "Preinfusion"
}
},
diff --git a/homeassistant/components/lamarzocco/switch.py b/homeassistant/components/lamarzocco/switch.py
index 54bd1ac2aed..ee03ba421d4 100644
--- a/homeassistant/components/lamarzocco/switch.py
+++ b/homeassistant/components/lamarzocco/switch.py
@@ -13,7 +13,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMarzoccoConfigEntry, LaMarzoccoUpdateCoordinator
@@ -64,7 +64,7 @@ ENTITIES: tuple[LaMarzoccoSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch entities and services."""
diff --git a/homeassistant/components/lamarzocco/update.py b/homeassistant/components/lamarzocco/update.py
index 0833ee6e249..37960d26e95 100644
--- a/homeassistant/components/lamarzocco/update.py
+++ b/homeassistant/components/lamarzocco/update.py
@@ -15,7 +15,7 @@ from homeassistant.components.update import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMarzoccoConfigEntry
@@ -55,7 +55,7 @@ ENTITIES: tuple[LaMarzoccoUpdateEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LaMarzoccoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create update entities."""
diff --git a/homeassistant/components/lametric/button.py b/homeassistant/components/lametric/button.py
index f0a452f2d02..3c7d754fa0b 100644
--- a/homeassistant/components/lametric/button.py
+++ b/homeassistant/components/lametric/button.py
@@ -12,7 +12,7 @@ from homeassistant.components.button import ButtonEntity, ButtonEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMetricDataUpdateCoordinator
@@ -58,7 +58,7 @@ BUTTONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LaMetric button based on a config entry."""
coordinator: LaMetricDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lametric/coordinator.py b/homeassistant/components/lametric/coordinator.py
index 6655b035740..c292b2971b6 100644
--- a/homeassistant/components/lametric/coordinator.py
+++ b/homeassistant/components/lametric/coordinator.py
@@ -21,14 +21,15 @@ class LaMetricDataUpdateCoordinator(DataUpdateCoordinator[Device]):
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize the LaMatric coordinator."""
- self.config_entry = entry
self.lametric = LaMetricDevice(
host=entry.data[CONF_HOST],
api_key=entry.data[CONF_API_KEY],
session=async_get_clientsession(hass),
)
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass, LOGGER, config_entry=entry, name=DOMAIN, update_interval=SCAN_INTERVAL
+ )
async def _async_update_data(self) -> Device:
"""Fetch device information of the LaMetric device."""
diff --git a/homeassistant/components/lametric/number.py b/homeassistant/components/lametric/number.py
index ccfd48a3abf..7f356741d76 100644
--- a/homeassistant/components/lametric/number.py
+++ b/homeassistant/components/lametric/number.py
@@ -12,7 +12,7 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMetricDataUpdateCoordinator
@@ -58,7 +58,7 @@ NUMBERS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LaMetric number based on a config entry."""
coordinator: LaMetricDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lametric/select.py b/homeassistant/components/lametric/select.py
index bf9872f2791..eab7cd5997c 100644
--- a/homeassistant/components/lametric/select.py
+++ b/homeassistant/components/lametric/select.py
@@ -12,7 +12,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMetricDataUpdateCoordinator
@@ -43,7 +43,7 @@ SELECTS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LaMetric select based on a config entry."""
coordinator: LaMetricDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lametric/sensor.py b/homeassistant/components/lametric/sensor.py
index f202a77b530..a5d5da3c046 100644
--- a/homeassistant/components/lametric/sensor.py
+++ b/homeassistant/components/lametric/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMetricDataUpdateCoordinator
@@ -45,7 +45,7 @@ SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LaMetric sensor based on a config entry."""
coordinator: LaMetricDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lametric/switch.py b/homeassistant/components/lametric/switch.py
index 3aabfaf17e1..85e61164639 100644
--- a/homeassistant/components/lametric/switch.py
+++ b/homeassistant/components/lametric/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LaMetricDataUpdateCoordinator
@@ -48,7 +48,7 @@ SWITCHES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LaMetric switch based on a config entry."""
coordinator: LaMetricDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/landisgyr_heat_meter/__init__.py b/homeassistant/components/landisgyr_heat_meter/__init__.py
index 5cbdc593100..7e7ebe61eb7 100644
--- a/homeassistant/components/landisgyr_heat_meter/__init__.py
+++ b/homeassistant/components/landisgyr_heat_meter/__init__.py
@@ -27,7 +27,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
reader = ultraheat_api.UltraheatReader(entry.data[CONF_DEVICE])
api = ultraheat_api.HeatMeterService(reader)
- coordinator = UltraheatCoordinator(hass, api)
+ coordinator = UltraheatCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/landisgyr_heat_meter/coordinator.py b/homeassistant/components/landisgyr_heat_meter/coordinator.py
index db265449f37..4214fa1db3e 100644
--- a/homeassistant/components/landisgyr_heat_meter/coordinator.py
+++ b/homeassistant/components/landisgyr_heat_meter/coordinator.py
@@ -7,6 +7,7 @@ import serial
from ultraheat_api.response import HeatMeterResponse
from ultraheat_api.service import HeatMeterService
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -18,11 +19,16 @@ _LOGGER = logging.getLogger(__name__)
class UltraheatCoordinator(DataUpdateCoordinator[HeatMeterResponse]):
"""Coordinator for getting data from the ultraheat api."""
- def __init__(self, hass: HomeAssistant, api: HeatMeterService) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: HeatMeterService
+ ) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="ultraheat",
update_interval=POLLING_INTERVAL,
)
diff --git a/homeassistant/components/landisgyr_heat_meter/sensor.py b/homeassistant/components/landisgyr_heat_meter/sensor.py
index dd76d3e53cc..9bb4af572fd 100644
--- a/homeassistant/components/landisgyr_heat_meter/sensor.py
+++ b/homeassistant/components/landisgyr_heat_meter/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
@@ -269,7 +269,9 @@ HEAT_METER_SENSOR_TYPES = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
unique_id = entry.entry_id
diff --git a/homeassistant/components/lastfm/__init__.py b/homeassistant/components/lastfm/__init__.py
index ebcc929c39c..8611d06eee1 100644
--- a/homeassistant/components/lastfm/__init__.py
+++ b/homeassistant/components/lastfm/__init__.py
@@ -12,7 +12,7 @@ from .coordinator import LastFMDataUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up lastfm from a config entry."""
- coordinator = LastFMDataUpdateCoordinator(hass)
+ coordinator = LastFMDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/lastfm/coordinator.py b/homeassistant/components/lastfm/coordinator.py
index 18473745450..ae89e103b80 100644
--- a/homeassistant/components/lastfm/coordinator.py
+++ b/homeassistant/components/lastfm/coordinator.py
@@ -39,15 +39,16 @@ class LastFMDataUpdateCoordinator(DataUpdateCoordinator[dict[str, LastFMUserData
config_entry: ConfigEntry
_client: LastFMNetwork
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize the LastFM data coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
- self._client = LastFMNetwork(api_key=self.config_entry.options[CONF_API_KEY])
+ self._client = LastFMNetwork(api_key=config_entry.options[CONF_API_KEY])
async def _async_update_data(self) -> dict[str, LastFMUserData]:
res = {}
diff --git a/homeassistant/components/lastfm/sensor.py b/homeassistant/components/lastfm/sensor.py
index 48770113a80..89025583e92 100644
--- a/homeassistant/components/lastfm/sensor.py
+++ b/homeassistant/components/lastfm/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -27,7 +27,7 @@ from .coordinator import LastFMDataUpdateCoordinator, LastFMUserData
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize the entries."""
diff --git a/homeassistant/components/launch_library/sensor.py b/homeassistant/components/launch_library/sensor.py
index 7d3b2bd97b6..201b4c8f037 100644
--- a/homeassistant/components/launch_library/sensor.py
+++ b/homeassistant/components/launch_library/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, PERCENTAGE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -122,7 +122,7 @@ SENSOR_DESCRIPTIONS: tuple[LaunchLibrarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
name = entry.data.get(CONF_NAME, DEFAULT_NEXT_LAUNCH_NAME)
diff --git a/homeassistant/components/laundrify/__init__.py b/homeassistant/components/laundrify/__init__.py
index b08624b6d23..7e3dd848348 100644
--- a/homeassistant/components/laundrify/__init__.py
+++ b/homeassistant/components/laundrify/__init__.py
@@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .const import DEFAULT_POLL_INTERVAL, DOMAIN
+from .const import DOMAIN
from .coordinator import LaundrifyUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -34,7 +34,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except ApiConnectionException as err:
raise ConfigEntryNotReady("Cannot reach laundrify API") from err
- coordinator = LaundrifyUpdateCoordinator(hass, api_client, DEFAULT_POLL_INTERVAL)
+ coordinator = LaundrifyUpdateCoordinator(hass, entry, api_client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/laundrify/binary_sensor.py b/homeassistant/components/laundrify/binary_sensor.py
index cee6aa6c754..82f4f7609dc 100644
--- a/homeassistant/components/laundrify/binary_sensor.py
+++ b/homeassistant/components/laundrify/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER, MODELS
@@ -23,7 +23,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors from a config entry created in the integrations UI."""
diff --git a/homeassistant/components/laundrify/coordinator.py b/homeassistant/components/laundrify/coordinator.py
index 22f68a7c5ae..928e30a9ed5 100644
--- a/homeassistant/components/laundrify/coordinator.py
+++ b/homeassistant/components/laundrify/coordinator.py
@@ -7,11 +7,12 @@ import logging
from laundrify_aio import LaundrifyAPI, LaundrifyDevice
from laundrify_aio.exceptions import ApiConnectionException, UnauthorizedException
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import DOMAIN, REQUEST_TIMEOUT
+from .const import DEFAULT_POLL_INTERVAL, DOMAIN, REQUEST_TIMEOUT
_LOGGER = logging.getLogger(__name__)
@@ -19,15 +20,21 @@ _LOGGER = logging.getLogger(__name__)
class LaundrifyUpdateCoordinator(DataUpdateCoordinator[dict[str, LaundrifyDevice]]):
"""Class to manage fetching laundrify API data."""
+ config_entry: ConfigEntry
+
def __init__(
- self, hass: HomeAssistant, laundrify_api: LaundrifyAPI, poll_interval: int
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ laundrify_api: LaundrifyAPI,
) -> None:
"""Initialize laundrify coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
- update_interval=timedelta(seconds=poll_interval),
+ update_interval=timedelta(seconds=DEFAULT_POLL_INTERVAL),
)
self.laundrify_api = laundrify_api
diff --git a/homeassistant/components/laundrify/sensor.py b/homeassistant/components/laundrify/sensor.py
index 98169f95fce..3c343861b0a 100644
--- a/homeassistant/components/laundrify/sensor.py
+++ b/homeassistant/components/laundrify/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -24,7 +24,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add power sensor for passed config_entry in HA."""
diff --git a/homeassistant/components/lawn_mower/__init__.py b/homeassistant/components/lawn_mower/__init__.py
index 0680bfc9d71..f8c3e0cd67d 100644
--- a/homeassistant/components/lawn_mower/__init__.py
+++ b/homeassistant/components/lawn_mower/__init__.py
@@ -28,6 +28,7 @@ from .const import (
_LOGGER = logging.getLogger(__name__)
DATA_COMPONENT: HassKey[EntityComponent[LawnMowerEntity]] = HassKey(DOMAIN)
+ENTITY_ID_FORMAT = DOMAIN + ".{}"
PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA
PLATFORM_SCHEMA_BASE = cv.PLATFORM_SCHEMA_BASE
SCAN_INTERVAL = timedelta(seconds=60)
diff --git a/homeassistant/components/lcn/__init__.py b/homeassistant/components/lcn/__init__.py
index 58924413c56..256e132b30d 100644
--- a/homeassistant/components/lcn/__init__.py
+++ b/homeassistant/components/lcn/__init__.py
@@ -49,6 +49,7 @@ from .helpers import (
InputType,
async_update_config_entry,
generate_unique_id,
+ purge_device_registry,
register_lcn_address_devices,
register_lcn_host_device,
)
@@ -120,6 +121,9 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
register_lcn_host_device(hass, config_entry)
register_lcn_address_devices(hass, config_entry)
+ # clean up orphaned devices
+ purge_device_registry(hass, config_entry.entry_id, {**config_entry.data})
+
# forward config_entry to components
await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
diff --git a/homeassistant/components/lcn/binary_sensor.py b/homeassistant/components/lcn/binary_sensor.py
index d0ce4815f19..65afae56f22 100644
--- a/homeassistant/components/lcn/binary_sensor.py
+++ b/homeassistant/components/lcn/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.script import scripts_with_entity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DOMAIN, CONF_ENTITIES, CONF_SOURCE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import (
IssueSeverity,
async_create_issue,
@@ -35,7 +35,7 @@ from .helpers import InputType
def add_lcn_entities(
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
entity_configs: Iterable[ConfigType],
) -> None:
"""Add entities for this domain."""
@@ -54,7 +54,7 @@ def add_lcn_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LCN switch entities from a config entry."""
add_entities = partial(
diff --git a/homeassistant/components/lcn/climate.py b/homeassistant/components/lcn/climate.py
index 1dff15c4f22..e91ae723714 100644
--- a/homeassistant/components/lcn/climate.py
+++ b/homeassistant/components/lcn/climate.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .const import (
@@ -43,7 +43,7 @@ PARALLEL_UPDATES = 0
def add_lcn_entities(
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
entity_configs: Iterable[ConfigType],
) -> None:
"""Add entities for this domain."""
@@ -57,7 +57,7 @@ def add_lcn_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LCN switch entities from a config entry."""
add_entities = partial(
diff --git a/homeassistant/components/lcn/cover.py b/homeassistant/components/lcn/cover.py
index 042461b6af2..be713871aae 100644
--- a/homeassistant/components/lcn/cover.py
+++ b/homeassistant/components/lcn/cover.py
@@ -10,7 +10,7 @@ from homeassistant.components.cover import DOMAIN as DOMAIN_COVER, CoverEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DOMAIN, CONF_ENTITIES
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .const import (
@@ -28,7 +28,7 @@ PARALLEL_UPDATES = 0
def add_lcn_entities(
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
entity_configs: Iterable[ConfigType],
) -> None:
"""Add entities for this domain."""
@@ -45,7 +45,7 @@ def add_lcn_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LCN cover entities from a config entry."""
add_entities = partial(
diff --git a/homeassistant/components/lcn/entity.py b/homeassistant/components/lcn/entity.py
index 12d8f966801..ffb680c4237 100644
--- a/homeassistant/components/lcn/entity.py
+++ b/homeassistant/components/lcn/entity.py
@@ -3,19 +3,18 @@
from collections.abc import Callable
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_ADDRESS, CONF_DOMAIN, CONF_NAME, CONF_RESOURCE
+from homeassistant.const import CONF_ADDRESS, CONF_NAME, CONF_RESOURCE
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.typing import ConfigType
-from .const import CONF_DOMAIN_DATA, DOMAIN
+from .const import DOMAIN
from .helpers import (
AddressType,
DeviceConnectionType,
InputType,
generate_unique_id,
get_device_connection,
- get_device_model,
)
@@ -36,6 +35,14 @@ class LcnEntity(Entity):
self.address: AddressType = config[CONF_ADDRESS]
self._unregister_for_inputs: Callable | None = None
self._name: str = config[CONF_NAME]
+ self._attr_device_info = DeviceInfo(
+ identifiers={
+ (
+ DOMAIN,
+ generate_unique_id(self.config_entry.entry_id, self.address),
+ )
+ },
+ )
@property
def unique_id(self) -> str:
@@ -44,28 +51,6 @@ class LcnEntity(Entity):
self.config_entry.entry_id, self.address, self.config[CONF_RESOURCE]
)
- @property
- def device_info(self) -> DeviceInfo | None:
- """Return device specific attributes."""
- address = f"{'g' if self.address[2] else 'm'}{self.address[0]:03d}{self.address[1]:03d}"
- model = (
- "LCN resource"
- f" ({get_device_model(self.config[CONF_DOMAIN], self.config[CONF_DOMAIN_DATA])})"
- )
-
- return DeviceInfo(
- identifiers={(DOMAIN, self.unique_id)},
- name=f"{address}.{self.config[CONF_RESOURCE]}",
- model=model,
- manufacturer="Issendorff",
- via_device=(
- DOMAIN,
- generate_unique_id(
- self.config_entry.entry_id, self.config[CONF_ADDRESS]
- ),
- ),
- )
-
async def async_added_to_hass(self) -> None:
"""Run when entity about to be added to hass."""
self.device_connection = get_device_connection(
diff --git a/homeassistant/components/lcn/helpers.py b/homeassistant/components/lcn/helpers.py
index b999c6f3770..2176c669251 100644
--- a/homeassistant/components/lcn/helpers.py
+++ b/homeassistant/components/lcn/helpers.py
@@ -4,7 +4,6 @@ from __future__ import annotations
import asyncio
from copy import deepcopy
-from itertools import chain
import re
from typing import cast
@@ -22,7 +21,6 @@ from homeassistant.const import (
CONF_NAME,
CONF_RESOURCE,
CONF_SENSORS,
- CONF_SOURCE,
CONF_SWITCHES,
)
from homeassistant.core import HomeAssistant
@@ -30,23 +28,14 @@ from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.typing import ConfigType
from .const import (
- BINSENSOR_PORTS,
CONF_CLIMATES,
CONF_HARDWARE_SERIAL,
CONF_HARDWARE_TYPE,
- CONF_OUTPUT,
CONF_SCENES,
CONF_SOFTWARE_SERIAL,
CONNECTION,
DEVICE_CONNECTIONS,
DOMAIN,
- LED_PORTS,
- LOGICOP_PORTS,
- OUTPUT_PORTS,
- S0_INPUTS,
- SETPOINTS,
- THRESHOLDS,
- VARIABLES,
)
# typing
@@ -96,31 +85,6 @@ def get_resource(domain_name: str, domain_data: ConfigType) -> str:
raise ValueError("Unknown domain")
-def get_device_model(domain_name: str, domain_data: ConfigType) -> str:
- """Return the model for the specified domain_data."""
- if domain_name in ("switch", "light"):
- return "Output" if domain_data[CONF_OUTPUT] in OUTPUT_PORTS else "Relay"
- if domain_name in ("binary_sensor", "sensor"):
- if domain_data[CONF_SOURCE] in BINSENSOR_PORTS:
- return "Binary Sensor"
- if domain_data[CONF_SOURCE] in chain(
- VARIABLES, SETPOINTS, THRESHOLDS, S0_INPUTS
- ):
- return "Variable"
- if domain_data[CONF_SOURCE] in LED_PORTS:
- return "Led"
- if domain_data[CONF_SOURCE] in LOGICOP_PORTS:
- return "Logical Operation"
- return "Key"
- if domain_name == "cover":
- return "Motor"
- if domain_name == "climate":
- return "Regulator"
- if domain_name == "scene":
- return "Scene"
- raise ValueError("Unknown domain")
-
-
def generate_unique_id(
entry_id: str,
address: AddressType,
@@ -169,13 +133,6 @@ def purge_device_registry(
) -> None:
"""Remove orphans from device registry which are not in entry data."""
device_registry = dr.async_get(hass)
- entity_registry = er.async_get(hass)
-
- # Find all devices that are referenced in the entity registry.
- references_entities = {
- entry.device_id
- for entry in entity_registry.entities.get_entries_for_config_entry_id(entry_id)
- }
# Find device that references the host.
references_host = set()
@@ -198,7 +155,6 @@ def purge_device_registry(
entry.id
for entry in dr.async_entries_for_config_entry(device_registry, entry_id)
}
- - references_entities
- references_host
- references_entry_data
)
diff --git a/homeassistant/components/lcn/light.py b/homeassistant/components/lcn/light.py
index 9ec660325c8..cba7c0888b7 100644
--- a/homeassistant/components/lcn/light.py
+++ b/homeassistant/components/lcn/light.py
@@ -17,7 +17,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DOMAIN, CONF_ENTITIES
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .const import (
@@ -37,7 +37,7 @@ PARALLEL_UPDATES = 0
def add_lcn_entities(
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
entity_configs: Iterable[ConfigType],
) -> None:
"""Add entities for this domain."""
@@ -54,7 +54,7 @@ def add_lcn_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LCN light entities from a config entry."""
add_entities = partial(
diff --git a/homeassistant/components/lcn/scene.py b/homeassistant/components/lcn/scene.py
index 0f40926cf17..072d0a20757 100644
--- a/homeassistant/components/lcn/scene.py
+++ b/homeassistant/components/lcn/scene.py
@@ -10,7 +10,7 @@ from homeassistant.components.scene import DOMAIN as DOMAIN_SCENE, Scene
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DOMAIN, CONF_ENTITIES, CONF_SCENE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .const import (
@@ -29,7 +29,7 @@ PARALLEL_UPDATES = 0
def add_lcn_entities(
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
entity_configs: Iterable[ConfigType],
) -> None:
"""Add entities for this domain."""
@@ -43,7 +43,7 @@ def add_lcn_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LCN switch entities from a config entry."""
add_entities = partial(
diff --git a/homeassistant/components/lcn/sensor.py b/homeassistant/components/lcn/sensor.py
index ada0857742c..7783df8679a 100644
--- a/homeassistant/components/lcn/sensor.py
+++ b/homeassistant/components/lcn/sensor.py
@@ -3,7 +3,6 @@
from collections.abc import Iterable
from functools import partial
from itertools import chain
-from typing import cast
import pypck
@@ -18,9 +17,14 @@ from homeassistant.const import (
CONF_ENTITIES,
CONF_SOURCE,
CONF_UNIT_OF_MEASUREMENT,
+ LIGHT_LUX,
+ UnitOfElectricCurrent,
+ UnitOfElectricPotential,
+ UnitOfSpeed,
+ UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .const import (
@@ -47,10 +51,21 @@ DEVICE_CLASS_MAPPING = {
pypck.lcn_defs.VarUnit.AMPERE: SensorDeviceClass.CURRENT,
}
+UNIT_OF_MEASUREMENT_MAPPING = {
+ pypck.lcn_defs.VarUnit.CELSIUS: UnitOfTemperature.CELSIUS,
+ pypck.lcn_defs.VarUnit.KELVIN: UnitOfTemperature.KELVIN,
+ pypck.lcn_defs.VarUnit.FAHRENHEIT: UnitOfTemperature.FAHRENHEIT,
+ pypck.lcn_defs.VarUnit.LUX_T: LIGHT_LUX,
+ pypck.lcn_defs.VarUnit.LUX_I: LIGHT_LUX,
+ pypck.lcn_defs.VarUnit.METERPERSECOND: UnitOfSpeed.METERS_PER_SECOND,
+ pypck.lcn_defs.VarUnit.VOLT: UnitOfElectricPotential.VOLT,
+ pypck.lcn_defs.VarUnit.AMPERE: UnitOfElectricCurrent.AMPERE,
+}
+
def add_lcn_entities(
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
entity_configs: Iterable[ConfigType],
) -> None:
"""Add entities for this domain."""
@@ -69,7 +84,7 @@ def add_lcn_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LCN switch entities from a config entry."""
add_entities = partial(
@@ -103,8 +118,10 @@ class LcnVariableSensor(LcnEntity, SensorEntity):
config[CONF_DOMAIN_DATA][CONF_UNIT_OF_MEASUREMENT]
)
- self._attr_native_unit_of_measurement = cast(str, self.unit.value)
- self._attr_device_class = DEVICE_CLASS_MAPPING.get(self.unit, None)
+ self._attr_native_unit_of_measurement = UNIT_OF_MEASUREMENT_MAPPING.get(
+ self.unit
+ )
+ self._attr_device_class = DEVICE_CLASS_MAPPING.get(self.unit)
async def async_added_to_hass(self) -> None:
"""Run when entity about to be added to hass."""
diff --git a/homeassistant/components/lcn/strings.json b/homeassistant/components/lcn/strings.json
index 0bdd85a3678..0a8112d997a 100644
--- a/homeassistant/components/lcn/strings.json
+++ b/homeassistant/components/lcn/strings.json
@@ -396,19 +396,19 @@
},
"address_to_device_id": {
"name": "Address to device ID",
- "description": "Convert LCN address to device ID.",
+ "description": "Converts an LCN address into a device ID.",
"fields": {
"id": {
"name": "Module or group ID",
- "description": "Target module or group ID."
+ "description": "Module or group number of the target."
},
"segment_id": {
"name": "Segment ID",
- "description": "Target segment ID."
+ "description": "Segment number of the target."
},
"type": {
"name": "Type",
- "description": "Target type."
+ "description": "Module type of the target."
},
"host": {
"name": "Host name",
diff --git a/homeassistant/components/lcn/switch.py b/homeassistant/components/lcn/switch.py
index dd940bd38b3..6267a081bc9 100644
--- a/homeassistant/components/lcn/switch.py
+++ b/homeassistant/components/lcn/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import DOMAIN as DOMAIN_SWITCH, SwitchEntit
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DOMAIN, CONF_ENTITIES
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .const import (
@@ -30,7 +30,7 @@ PARALLEL_UPDATES = 0
def add_lcn_switch_entities(
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
entity_configs: Iterable[ConfigType],
) -> None:
"""Add entities for this domain."""
@@ -53,7 +53,7 @@ def add_lcn_switch_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LCN switch entities from a config entry."""
add_entities = partial(
diff --git a/homeassistant/components/ld2410_ble/__init__.py b/homeassistant/components/ld2410_ble/__init__.py
index 57e3dfa4617..db67010823d 100644
--- a/homeassistant/components/ld2410_ble/__init__.py
+++ b/homeassistant/components/ld2410_ble/__init__.py
@@ -41,7 +41,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
ld2410_ble = LD2410BLE(ble_device)
- coordinator = LD2410BLECoordinator(hass, ld2410_ble)
+ coordinator = LD2410BLECoordinator(hass, entry, ld2410_ble)
try:
await ld2410_ble.initialise()
diff --git a/homeassistant/components/ld2410_ble/binary_sensor.py b/homeassistant/components/ld2410_ble/binary_sensor.py
index c52bc34b699..3ba43e0d6dc 100644
--- a/homeassistant/components/ld2410_ble/binary_sensor.py
+++ b/homeassistant/components/ld2410_ble/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import LD2410BLE, LD2410BLECoordinator
@@ -31,7 +31,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform for LD2410BLE."""
data: LD2410BLEData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/ld2410_ble/coordinator.py b/homeassistant/components/ld2410_ble/coordinator.py
index 2f0fd079773..b318542e798 100644
--- a/homeassistant/components/ld2410_ble/coordinator.py
+++ b/homeassistant/components/ld2410_ble/coordinator.py
@@ -6,6 +6,7 @@ import time
from ld2410_ble import LD2410BLE, LD2410BLEState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import CALLBACK_TYPE, HassJob, HomeAssistant, callback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -21,11 +22,16 @@ DEBOUNCE_SECONDS = 1.0
class LD2410BLECoordinator(DataUpdateCoordinator[None]):
"""Data coordinator for receiving LD2410B updates."""
- def __init__(self, hass: HomeAssistant, ld2410_ble: LD2410BLE) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, ld2410_ble: LD2410BLE
+ ) -> None:
"""Initialise the coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
)
self._ld2410_ble = ld2410_ble
diff --git a/homeassistant/components/ld2410_ble/manifest.json b/homeassistant/components/ld2410_ble/manifest.json
index 36d0150642e..1896f2109a7 100644
--- a/homeassistant/components/ld2410_ble/manifest.json
+++ b/homeassistant/components/ld2410_ble/manifest.json
@@ -20,5 +20,5 @@
"documentation": "https://www.home-assistant.io/integrations/ld2410_ble",
"integration_type": "device",
"iot_class": "local_push",
- "requirements": ["bluetooth-data-tools==1.23.4", "ld2410-ble==0.1.1"]
+ "requirements": ["bluetooth-data-tools==1.26.1", "ld2410-ble==0.1.1"]
}
diff --git a/homeassistant/components/ld2410_ble/sensor.py b/homeassistant/components/ld2410_ble/sensor.py
index 6daa1397161..db4e42580c4 100644
--- a/homeassistant/components/ld2410_ble/sensor.py
+++ b/homeassistant/components/ld2410_ble/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.const import EntityCategory, UnitOfLength
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import LD2410BLE, LD2410BLECoordinator
@@ -122,7 +122,7 @@ SENSOR_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform for LD2410BLE."""
data: LD2410BLEData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/leaone/sensor.py b/homeassistant/components/leaone/sensor.py
index 62948868870..c815a0964e0 100644
--- a/homeassistant/components/leaone/sensor.py
+++ b/homeassistant/components/leaone/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfMass,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -107,7 +107,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Leaone BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/led_ble/light.py b/homeassistant/components/led_ble/light.py
index 3bca7269eba..14f2f228e13 100644
--- a/homeassistant/components/led_ble/light.py
+++ b/homeassistant/components/led_ble/light.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -32,7 +32,7 @@ from .models import LEDBLEData
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the light platform for LEDBLE."""
data: LEDBLEData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/led_ble/manifest.json b/homeassistant/components/led_ble/manifest.json
index 309399e6958..270495c8770 100644
--- a/homeassistant/components/led_ble/manifest.json
+++ b/homeassistant/components/led_ble/manifest.json
@@ -35,5 +35,5 @@
"dependencies": ["bluetooth_adapters"],
"documentation": "https://www.home-assistant.io/integrations/led_ble",
"iot_class": "local_polling",
- "requirements": ["bluetooth-data-tools==1.23.4", "led-ble==1.1.6"]
+ "requirements": ["bluetooth-data-tools==1.26.1", "led-ble==1.1.6"]
}
diff --git a/homeassistant/components/lektrico/__init__.py b/homeassistant/components/lektrico/__init__.py
index 475b6132541..0a6675237dd 100644
--- a/homeassistant/components/lektrico/__init__.py
+++ b/homeassistant/components/lektrico/__init__.py
@@ -4,11 +4,10 @@ from __future__ import annotations
from lektricowifi import Device
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import ATTR_SERIAL_NUMBER, CONF_TYPE, Platform
+from homeassistant.const import CONF_TYPE, Platform
from homeassistant.core import HomeAssistant
-from .coordinator import LektricoDeviceDataUpdateCoordinator
+from .coordinator import LektricoConfigEntry, LektricoDeviceDataUpdateCoordinator
# List the platforms that charger supports.
CHARGERS_PLATFORMS: list[Platform] = [
@@ -26,15 +25,10 @@ LB_DEVICES_PLATFORMS: list[Platform] = [
Platform.SENSOR,
]
-type LektricoConfigEntry = ConfigEntry[LektricoDeviceDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: LektricoConfigEntry) -> bool:
"""Set up Lektrico Charging Station from a config entry."""
- coordinator = LektricoDeviceDataUpdateCoordinator(
- hass,
- f"{entry.data[CONF_TYPE]}_{entry.data[ATTR_SERIAL_NUMBER]}",
- )
+ coordinator = LektricoDeviceDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
@@ -45,7 +39,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: LektricoConfigEntry) ->
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: LektricoConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(
@@ -53,7 +47,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
)
-def _get_platforms(entry: ConfigEntry) -> list[Platform]:
+def _get_platforms(entry: LektricoConfigEntry) -> list[Platform]:
"""Return the platforms for this type of device."""
_device_type: str = entry.data[CONF_TYPE]
if _device_type in (Device.TYPE_1P7K, Device.TYPE_3P22K):
diff --git a/homeassistant/components/lektrico/binary_sensor.py b/homeassistant/components/lektrico/binary_sensor.py
index d0a3e39690c..37e55ade798 100644
--- a/homeassistant/components/lektrico/binary_sensor.py
+++ b/homeassistant/components/lektrico/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import ATTR_SERIAL_NUMBER, CONF_TYPE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LektricoConfigEntry, LektricoDeviceDataUpdateCoordinator
from .entity import LektricoEntity
@@ -101,7 +101,7 @@ BINARY_SENSORS: tuple[LektricoBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LektricoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lektrico binary sensor entities based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/lektrico/button.py b/homeassistant/components/lektrico/button.py
index 62aef12ff53..e598773321d 100644
--- a/homeassistant/components/lektrico/button.py
+++ b/homeassistant/components/lektrico/button.py
@@ -13,7 +13,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import ATTR_SERIAL_NUMBER, CONF_TYPE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LektricoConfigEntry, LektricoDeviceDataUpdateCoordinator
from .entity import LektricoEntity
@@ -60,7 +60,7 @@ BUTTONS_FOR_LB_DEVICES: tuple[LektricoButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LektricoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lektrico charger based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/lektrico/coordinator.py b/homeassistant/components/lektrico/coordinator.py
index 7c72a00e2d3..aa96cf49e07 100644
--- a/homeassistant/components/lektrico/coordinator.py
+++ b/homeassistant/components/lektrico/coordinator.py
@@ -22,18 +22,21 @@ from .const import LOGGER
SCAN_INTERVAL = timedelta(seconds=10)
+type LektricoConfigEntry = ConfigEntry[LektricoDeviceDataUpdateCoordinator]
+
class LektricoDeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Data update coordinator for Lektrico device."""
- config_entry: ConfigEntry
+ config_entry: LektricoConfigEntry
- def __init__(self, hass: HomeAssistant, device_name: str) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: LektricoConfigEntry) -> None:
"""Initialize a Lektrico Device."""
super().__init__(
hass,
LOGGER,
- name=device_name,
+ config_entry=config_entry,
+ name=f"{config_entry.data[CONF_TYPE]}_{config_entry.data[ATTR_SERIAL_NUMBER]}",
update_interval=SCAN_INTERVAL,
)
self.device = Device(
diff --git a/homeassistant/components/lektrico/number.py b/homeassistant/components/lektrico/number.py
index 8054ba8afe5..c54ee938607 100644
--- a/homeassistant/components/lektrico/number.py
+++ b/homeassistant/components/lektrico/number.py
@@ -15,7 +15,7 @@ from homeassistant.const import (
UnitOfElectricCurrent,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LektricoConfigEntry, LektricoDeviceDataUpdateCoordinator
from .entity import LektricoEntity
@@ -58,7 +58,7 @@ NUMBERS: tuple[LektricoNumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LektricoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lektrico number entities based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/lektrico/select.py b/homeassistant/components/lektrico/select.py
index ef45d97d697..513a82365af 100644
--- a/homeassistant/components/lektrico/select.py
+++ b/homeassistant/components/lektrico/select.py
@@ -9,7 +9,7 @@ from lektricowifi import Device
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import ATTR_SERIAL_NUMBER, CONF_TYPE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LektricoConfigEntry, LektricoDeviceDataUpdateCoordinator
from .entity import LektricoEntity
@@ -46,7 +46,7 @@ SELECTS: tuple[LektricoSelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LektricoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lektrico select entities based on a config entry."""
diff --git a/homeassistant/components/lektrico/sensor.py b/homeassistant/components/lektrico/sensor.py
index d55d91c4cd4..927011459b0 100644
--- a/homeassistant/components/lektrico/sensor.py
+++ b/homeassistant/components/lektrico/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import IntegrationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import LektricoConfigEntry, LektricoDeviceDataUpdateCoordinator
@@ -283,7 +283,7 @@ SENSORS_FOR_LB_3_PHASE: tuple[LektricoSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LektricoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lektrico charger based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/lektrico/strings.json b/homeassistant/components/lektrico/strings.json
index e24700c9b09..eb0203e0661 100644
--- a/homeassistant/components/lektrico/strings.json
+++ b/homeassistant/components/lektrico/strings.json
@@ -24,7 +24,7 @@
"entity": {
"binary_sensor": {
"state_e_activated": {
- "name": "Ev error"
+ "name": "EV error"
},
"overtemp": {
"name": "Thermal throttling"
@@ -45,10 +45,10 @@
"name": "Overvoltage"
},
"rcd_error": {
- "name": "Rcd error"
+ "name": "RCD error"
},
"cp_diode_failure": {
- "name": "Ev diode short"
+ "name": "EV diode short"
},
"contactor_failure": {
"name": "Relay contacts welded"
@@ -64,7 +64,7 @@
},
"number": {
"led_max_brightness": {
- "name": "Led brightness"
+ "name": "LED brightness"
},
"dynamic_limit": {
"name": "Dynamic limit"
@@ -86,7 +86,7 @@
"name": "State",
"state": {
"available": "Available",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"connected": "Connected",
"error": "Error",
"locked": "Locked",
diff --git a/homeassistant/components/lektrico/switch.py b/homeassistant/components/lektrico/switch.py
index 0fdfbd2ad41..065e96f84b8 100644
--- a/homeassistant/components/lektrico/switch.py
+++ b/homeassistant/components/lektrico/switch.py
@@ -9,7 +9,7 @@ from lektricowifi import Device
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import ATTR_SERIAL_NUMBER, CONF_TYPE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LektricoConfigEntry, LektricoDeviceDataUpdateCoordinator
from .entity import LektricoEntity
@@ -59,7 +59,7 @@ SWITCHS_FOR_3_PHASE_CHARGERS: tuple[LektricoSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LektricoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lektrico switch entities based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/letpot/__init__.py b/homeassistant/components/letpot/__init__.py
index 5bfc4edfc0c..50c73f949a3 100644
--- a/homeassistant/components/letpot/__init__.py
+++ b/homeassistant/components/letpot/__init__.py
@@ -9,7 +9,6 @@ from letpot.converters import CONVERTERS
from letpot.exceptions import LetPotAuthenticationException, LetPotException
from letpot.models import AuthenticationInfo
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_EMAIL, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -21,11 +20,14 @@ from .const import (
CONF_REFRESH_TOKEN_EXPIRES,
CONF_USER_ID,
)
-from .coordinator import LetPotDeviceCoordinator
+from .coordinator import LetPotConfigEntry, LetPotDeviceCoordinator
-PLATFORMS: list[Platform] = [Platform.SWITCH, Platform.TIME]
-
-type LetPotConfigEntry = ConfigEntry[list[LetPotDeviceCoordinator]]
+PLATFORMS: list[Platform] = [
+ Platform.BINARY_SENSOR,
+ Platform.SENSOR,
+ Platform.SWITCH,
+ Platform.TIME,
+]
async def async_setup_entry(hass: HomeAssistant, entry: LetPotConfigEntry) -> bool:
@@ -67,7 +69,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: LetPotConfigEntry) -> bo
raise ConfigEntryNotReady from exc
coordinators: list[LetPotDeviceCoordinator] = [
- LetPotDeviceCoordinator(hass, auth, device)
+ LetPotDeviceCoordinator(hass, entry, auth, device)
for device in devices
if any(converter.supports_type(device.device_type) for converter in CONVERTERS)
]
diff --git a/homeassistant/components/letpot/binary_sensor.py b/homeassistant/components/letpot/binary_sensor.py
new file mode 100644
index 00000000000..bfc7a5ab4a7
--- /dev/null
+++ b/homeassistant/components/letpot/binary_sensor.py
@@ -0,0 +1,122 @@
+"""Support for LetPot binary sensor entities."""
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from letpot.models import DeviceFeature, LetPotDeviceStatus
+
+from homeassistant.components.binary_sensor import (
+ BinarySensorDeviceClass,
+ BinarySensorEntity,
+ BinarySensorEntityDescription,
+)
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import LetPotConfigEntry, LetPotDeviceCoordinator
+from .entity import LetPotEntity, LetPotEntityDescription
+
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
+
+@dataclass(frozen=True, kw_only=True)
+class LetPotBinarySensorEntityDescription(
+ LetPotEntityDescription, BinarySensorEntityDescription
+):
+ """Describes a LetPot binary sensor entity."""
+
+ is_on_fn: Callable[[LetPotDeviceStatus], bool]
+
+
+BINARY_SENSORS: tuple[LetPotBinarySensorEntityDescription, ...] = (
+ LetPotBinarySensorEntityDescription(
+ key="low_nutrients",
+ translation_key="low_nutrients",
+ is_on_fn=lambda status: bool(status.errors.low_nutrients),
+ entity_registry_enabled_default=False,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ supported_fn=(
+ lambda coordinator: coordinator.data.errors.low_nutrients is not None
+ ),
+ ),
+ LetPotBinarySensorEntityDescription(
+ key="low_water",
+ translation_key="low_water",
+ is_on_fn=lambda status: bool(status.errors.low_water),
+ entity_registry_enabled_default=False,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ supported_fn=lambda coordinator: coordinator.data.errors.low_water is not None,
+ ),
+ LetPotBinarySensorEntityDescription(
+ key="pump",
+ translation_key="pump",
+ is_on_fn=lambda status: status.pump_status == 1,
+ device_class=BinarySensorDeviceClass.RUNNING,
+ supported_fn=(
+ lambda coordinator: DeviceFeature.PUMP_STATUS
+ in coordinator.device_client.device_features
+ ),
+ ),
+ LetPotBinarySensorEntityDescription(
+ key="pump_error",
+ translation_key="pump_error",
+ is_on_fn=lambda status: bool(status.errors.pump_malfunction),
+ entity_registry_enabled_default=False,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ supported_fn=(
+ lambda coordinator: coordinator.data.errors.pump_malfunction is not None
+ ),
+ ),
+ LetPotBinarySensorEntityDescription(
+ key="refill_error",
+ translation_key="refill_error",
+ is_on_fn=lambda status: bool(status.errors.refill_error),
+ entity_registry_enabled_default=False,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ supported_fn=(
+ lambda coordinator: coordinator.data.errors.refill_error is not None
+ ),
+ ),
+)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: LetPotConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up LetPot binary sensor entities based on a config entry and device status/features."""
+ coordinators = entry.runtime_data
+ async_add_entities(
+ LetPotBinarySensorEntity(coordinator, description)
+ for description in BINARY_SENSORS
+ for coordinator in coordinators
+ if description.supported_fn(coordinator)
+ )
+
+
+class LetPotBinarySensorEntity(LetPotEntity, BinarySensorEntity):
+ """Defines a LetPot binary sensor entity."""
+
+ entity_description: LetPotBinarySensorEntityDescription
+
+ def __init__(
+ self,
+ coordinator: LetPotDeviceCoordinator,
+ description: LetPotBinarySensorEntityDescription,
+ ) -> None:
+ """Initialize LetPot binary sensor entity."""
+ super().__init__(coordinator)
+ self.entity_description = description
+ self._attr_unique_id = f"{coordinator.config_entry.unique_id}_{coordinator.device.serial_number}_{description.key}"
+
+ @property
+ def is_on(self) -> bool:
+ """Return if the binary sensor is on."""
+ return self.entity_description.is_on_fn(self.coordinator.data)
diff --git a/homeassistant/components/letpot/coordinator.py b/homeassistant/components/letpot/coordinator.py
index a2a35d566c6..bd787157482 100644
--- a/homeassistant/components/letpot/coordinator.py
+++ b/homeassistant/components/letpot/coordinator.py
@@ -4,23 +4,22 @@ from __future__ import annotations
import asyncio
import logging
-from typing import TYPE_CHECKING
from letpot.deviceclient import LetPotDeviceClient
from letpot.exceptions import LetPotAuthenticationException, LetPotException
from letpot.models import AuthenticationInfo, LetPotDevice, LetPotDeviceStatus
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import REQUEST_UPDATE_TIMEOUT
-if TYPE_CHECKING:
- from . import LetPotConfigEntry
-
_LOGGER = logging.getLogger(__name__)
+type LetPotConfigEntry = ConfigEntry[list[LetPotDeviceCoordinator]]
+
class LetPotDeviceCoordinator(DataUpdateCoordinator[LetPotDeviceStatus]):
"""Class to handle data updates for a specific garden."""
@@ -31,12 +30,17 @@ class LetPotDeviceCoordinator(DataUpdateCoordinator[LetPotDeviceStatus]):
device_client: LetPotDeviceClient
def __init__(
- self, hass: HomeAssistant, info: AuthenticationInfo, device: LetPotDevice
+ self,
+ hass: HomeAssistant,
+ config_entry: LetPotConfigEntry,
+ info: AuthenticationInfo,
+ device: LetPotDevice,
) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"LetPot {device.serial_number}",
)
self._info = info
diff --git a/homeassistant/components/letpot/entity.py b/homeassistant/components/letpot/entity.py
index b4d505f4092..5e2c46fee84 100644
--- a/homeassistant/components/letpot/entity.py
+++ b/homeassistant/components/letpot/entity.py
@@ -1,18 +1,27 @@
"""Base class for LetPot entities."""
from collections.abc import Callable, Coroutine
+from dataclasses import dataclass
from typing import Any, Concatenate
from letpot.exceptions import LetPotConnectionException, LetPotException
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
from .coordinator import LetPotDeviceCoordinator
+@dataclass(frozen=True, kw_only=True)
+class LetPotEntityDescription(EntityDescription):
+ """Description for all LetPot entities."""
+
+ supported_fn: Callable[[LetPotDeviceCoordinator], bool] = lambda _: True
+
+
class LetPotEntity(CoordinatorEntity[LetPotDeviceCoordinator]):
"""Defines a base LetPot entity."""
diff --git a/homeassistant/components/letpot/icons.json b/homeassistant/components/letpot/icons.json
index 2a2b727adcd..43541b57150 100644
--- a/homeassistant/components/letpot/icons.json
+++ b/homeassistant/components/letpot/icons.json
@@ -1,5 +1,30 @@
{
"entity": {
+ "binary_sensor": {
+ "low_nutrients": {
+ "default": "mdi:beaker-alert",
+ "state": {
+ "off": "mdi:beaker"
+ }
+ },
+ "low_water": {
+ "default": "mdi:water-percent-alert",
+ "state": {
+ "off": "mdi:water-percent"
+ }
+ },
+ "pump": {
+ "default": "mdi:pump",
+ "state": {
+ "off": "mdi:pump-off"
+ }
+ }
+ },
+ "sensor": {
+ "water_level": {
+ "default": "mdi:water-percent"
+ }
+ },
"switch": {
"alarm_sound": {
"default": "mdi:bell-ring",
diff --git a/homeassistant/components/letpot/quality_scale.yaml b/homeassistant/components/letpot/quality_scale.yaml
index 0eda413a461..9804a5ec3a4 100644
--- a/homeassistant/components/letpot/quality_scale.yaml
+++ b/homeassistant/components/letpot/quality_scale.yaml
@@ -44,7 +44,7 @@ rules:
log-when-unavailable: todo
parallel-updates: done
reauthentication-flow: done
- test-coverage: todo
+ test-coverage: done
# Gold
devices: done
@@ -59,9 +59,9 @@ rules:
docs-troubleshooting: todo
docs-use-cases: todo
dynamic-devices: todo
- entity-category: todo
- entity-device-class: todo
- entity-disabled-by-default: todo
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default: done
entity-translations: done
exception-translations: done
icon-translations: done
diff --git a/homeassistant/components/letpot/sensor.py b/homeassistant/components/letpot/sensor.py
new file mode 100644
index 00000000000..b0b113eb063
--- /dev/null
+++ b/homeassistant/components/letpot/sensor.py
@@ -0,0 +1,110 @@
+"""Support for LetPot sensor entities."""
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from letpot.models import DeviceFeature, LetPotDeviceStatus, TemperatureUnit
+
+from homeassistant.components.sensor import (
+ SensorDeviceClass,
+ SensorEntity,
+ SensorEntityDescription,
+ SensorStateClass,
+)
+from homeassistant.const import PERCENTAGE, UnitOfTemperature
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.typing import StateType
+
+from .coordinator import LetPotConfigEntry, LetPotDeviceCoordinator
+from .entity import LetPotEntity, LetPotEntityDescription
+
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
+
+LETPOT_TEMPERATURE_UNIT_HA_UNIT = {
+ TemperatureUnit.CELSIUS: UnitOfTemperature.CELSIUS,
+ TemperatureUnit.FAHRENHEIT: UnitOfTemperature.FAHRENHEIT,
+}
+
+
+@dataclass(frozen=True, kw_only=True)
+class LetPotSensorEntityDescription(LetPotEntityDescription, SensorEntityDescription):
+ """Describes a LetPot sensor entity."""
+
+ native_unit_of_measurement_fn: Callable[[LetPotDeviceStatus], str | None]
+ value_fn: Callable[[LetPotDeviceStatus], StateType]
+
+
+SENSORS: tuple[LetPotSensorEntityDescription, ...] = (
+ LetPotSensorEntityDescription(
+ key="temperature",
+ value_fn=lambda status: status.temperature_value,
+ native_unit_of_measurement_fn=(
+ lambda status: LETPOT_TEMPERATURE_UNIT_HA_UNIT[
+ status.temperature_unit or TemperatureUnit.CELSIUS
+ ]
+ ),
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ supported_fn=(
+ lambda coordinator: DeviceFeature.TEMPERATURE
+ in coordinator.device_client.device_features
+ ),
+ ),
+ LetPotSensorEntityDescription(
+ key="water_level",
+ translation_key="water_level",
+ value_fn=lambda status: status.water_level,
+ native_unit_of_measurement_fn=lambda _: PERCENTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ supported_fn=(
+ lambda coordinator: DeviceFeature.WATER_LEVEL
+ in coordinator.device_client.device_features
+ ),
+ ),
+)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: LetPotConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up LetPot sensor entities based on a device features."""
+ coordinators = entry.runtime_data
+ async_add_entities(
+ LetPotSensorEntity(coordinator, description)
+ for description in SENSORS
+ for coordinator in coordinators
+ if description.supported_fn(coordinator)
+ )
+
+
+class LetPotSensorEntity(LetPotEntity, SensorEntity):
+ """Defines a LetPot sensor entity."""
+
+ entity_description: LetPotSensorEntityDescription
+
+ def __init__(
+ self,
+ coordinator: LetPotDeviceCoordinator,
+ description: LetPotSensorEntityDescription,
+ ) -> None:
+ """Initialize LetPot sensor entity."""
+ super().__init__(coordinator)
+ self.entity_description = description
+ self._attr_unique_id = f"{coordinator.config_entry.unique_id}_{coordinator.device.serial_number}_{description.key}"
+
+ @property
+ def native_unit_of_measurement(self) -> str | None:
+ """Return the native unit of measurement."""
+ return self.entity_description.native_unit_of_measurement_fn(
+ self.coordinator.data
+ )
+
+ @property
+ def native_value(self) -> StateType:
+ """Return the state of the sensor."""
+ return self.entity_description.value_fn(self.coordinator.data)
diff --git a/homeassistant/components/letpot/strings.json b/homeassistant/components/letpot/strings.json
index 12913085644..cdc5a36a15f 100644
--- a/homeassistant/components/letpot/strings.json
+++ b/homeassistant/components/letpot/strings.json
@@ -32,6 +32,28 @@
}
},
"entity": {
+ "binary_sensor": {
+ "low_nutrients": {
+ "name": "Low nutrients"
+ },
+ "low_water": {
+ "name": "Low water"
+ },
+ "pump": {
+ "name": "Pump"
+ },
+ "pump_error": {
+ "name": "Pump error"
+ },
+ "refill_error": {
+ "name": "Refill error"
+ }
+ },
+ "sensor": {
+ "water_level": {
+ "name": "Water level"
+ }
+ },
"switch": {
"alarm_sound": {
"name": "Alarm sound"
diff --git a/homeassistant/components/letpot/switch.py b/homeassistant/components/letpot/switch.py
index 36d07276c48..0b00318c53b 100644
--- a/homeassistant/components/letpot/switch.py
+++ b/homeassistant/components/letpot/switch.py
@@ -10,11 +10,10 @@ from letpot.models import DeviceFeature, LetPotDeviceStatus
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import LetPotConfigEntry
-from .coordinator import LetPotDeviceCoordinator
-from .entity import LetPotEntity, exception_handler
+from .coordinator import LetPotConfigEntry, LetPotDeviceCoordinator
+from .entity import LetPotEntity, LetPotEntityDescription, exception_handler
# Each change pushes a 'full' device status with the change. The library will cache
# pending changes to avoid overwriting, but try to avoid a lot of parallelism.
@@ -22,14 +21,33 @@ PARALLEL_UPDATES = 1
@dataclass(frozen=True, kw_only=True)
-class LetPotSwitchEntityDescription(SwitchEntityDescription):
+class LetPotSwitchEntityDescription(LetPotEntityDescription, SwitchEntityDescription):
"""Describes a LetPot switch entity."""
value_fn: Callable[[LetPotDeviceStatus], bool | None]
set_value_fn: Callable[[LetPotDeviceClient, bool], Coroutine[Any, Any, None]]
-BASE_SWITCHES: tuple[LetPotSwitchEntityDescription, ...] = (
+SWITCHES: tuple[LetPotSwitchEntityDescription, ...] = (
+ LetPotSwitchEntityDescription(
+ key="alarm_sound",
+ translation_key="alarm_sound",
+ value_fn=lambda status: status.system_sound,
+ set_value_fn=lambda device_client, value: device_client.set_sound(value),
+ entity_category=EntityCategory.CONFIG,
+ supported_fn=lambda coordinator: coordinator.data.system_sound is not None,
+ ),
+ LetPotSwitchEntityDescription(
+ key="auto_mode",
+ translation_key="auto_mode",
+ value_fn=lambda status: status.water_mode == 1,
+ set_value_fn=lambda device_client, value: device_client.set_water_mode(value),
+ entity_category=EntityCategory.CONFIG,
+ supported_fn=(
+ lambda coordinator: DeviceFeature.PUMP_AUTO
+ in coordinator.device_client.device_features
+ ),
+ ),
LetPotSwitchEntityDescription(
key="power",
translation_key="power",
@@ -45,44 +63,21 @@ BASE_SWITCHES: tuple[LetPotSwitchEntityDescription, ...] = (
entity_category=EntityCategory.CONFIG,
),
)
-ALARM_SWITCH: LetPotSwitchEntityDescription = LetPotSwitchEntityDescription(
- key="alarm_sound",
- translation_key="alarm_sound",
- value_fn=lambda status: status.system_sound,
- set_value_fn=lambda device_client, value: device_client.set_sound(value),
- entity_category=EntityCategory.CONFIG,
-)
-AUTO_MODE_SWITCH: LetPotSwitchEntityDescription = LetPotSwitchEntityDescription(
- key="auto_mode",
- translation_key="auto_mode",
- value_fn=lambda status: status.water_mode == 1,
- set_value_fn=lambda device_client, value: device_client.set_water_mode(value),
- entity_category=EntityCategory.CONFIG,
-)
async def async_setup_entry(
hass: HomeAssistant,
entry: LetPotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LetPot switch entities based on a config entry and device status/features."""
coordinators = entry.runtime_data
entities: list[SwitchEntity] = [
LetPotSwitchEntity(coordinator, description)
- for description in BASE_SWITCHES
+ for description in SWITCHES
for coordinator in coordinators
+ if description.supported_fn(coordinator)
]
- entities.extend(
- LetPotSwitchEntity(coordinator, ALARM_SWITCH)
- for coordinator in coordinators
- if coordinator.data.system_sound is not None
- )
- entities.extend(
- LetPotSwitchEntity(coordinator, AUTO_MODE_SWITCH)
- for coordinator in coordinators
- if DeviceFeature.PUMP_AUTO in coordinator.device_client.device_features
- )
async_add_entities(entities)
diff --git a/homeassistant/components/letpot/time.py b/homeassistant/components/letpot/time.py
index 80ce9743d8c..bae61df6a28 100644
--- a/homeassistant/components/letpot/time.py
+++ b/homeassistant/components/letpot/time.py
@@ -11,10 +11,9 @@ from letpot.models import LetPotDeviceStatus
from homeassistant.components.time import TimeEntity, TimeEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import LetPotConfigEntry
-from .coordinator import LetPotDeviceCoordinator
+from .coordinator import LetPotConfigEntry, LetPotDeviceCoordinator
from .entity import LetPotEntity, exception_handler
# Each change pushes a 'full' device status with the change. The library will cache
@@ -55,7 +54,7 @@ TIME_SENSORS: tuple[LetPotTimeEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: LetPotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LetPot time entities based on a config entry."""
coordinators = entry.runtime_data
diff --git a/homeassistant/components/lg_netcast/media_player.py b/homeassistant/components/lg_netcast/media_player.py
index b3f8f8e0437..de652eeef08 100644
--- a/homeassistant/components/lg_netcast/media_player.py
+++ b/homeassistant/components/lg_netcast/media_player.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_HOST, CONF_MODEL, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.trigger import PluggableAction
from .const import ATTR_MANUFACTURER, DOMAIN
@@ -47,7 +47,7 @@ SUPPORT_LGTV = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a LG Netcast Media Player from a config_entry."""
diff --git a/homeassistant/components/lg_soundbar/media_player.py b/homeassistant/components/lg_soundbar/media_player.py
index cebe1d33728..c3ea22ee08f 100644
--- a/homeassistant/components/lg_soundbar/media_player.py
+++ b/homeassistant/components/lg_soundbar/media_player.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -23,7 +23,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up media_player from a config entry created in the integrations UI."""
async_add_entities(
diff --git a/homeassistant/components/lg_thinq/__init__.py b/homeassistant/components/lg_thinq/__init__.py
index 657524f0ef5..f83cbadf925 100644
--- a/homeassistant/components/lg_thinq/__init__.py
+++ b/homeassistant/components/lg_thinq/__init__.py
@@ -47,6 +47,7 @@ PLATFORMS = [
Platform.SENSOR,
Platform.SWITCH,
Platform.VACUUM,
+ Platform.WATER_HEATER,
]
_LOGGER = logging.getLogger(__name__)
@@ -100,7 +101,7 @@ async def async_setup_coordinators(
# Setup coordinator per device.
task_list = [
- hass.async_create_task(async_setup_device_coordinator(hass, bridge))
+ hass.async_create_task(async_setup_device_coordinator(hass, entry, bridge))
for bridge in bridge_list
]
task_result = await asyncio.gather(*task_list)
diff --git a/homeassistant/components/lg_thinq/binary_sensor.py b/homeassistant/components/lg_thinq/binary_sensor.py
index 845bf8c3079..61b600037a7 100644
--- a/homeassistant/components/lg_thinq/binary_sensor.py
+++ b/homeassistant/components/lg_thinq/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ThinqConfigEntry
from .entity import ThinQEntity
@@ -76,7 +76,8 @@ BINARY_SENSOR_DESC: dict[ThinQProperty, ThinQBinarySensorEntityDescription] = {
),
ThinQProperty.WATER_HEATER_OPERATION_MODE: ThinQBinarySensorEntityDescription(
key=ThinQProperty.WATER_HEATER_OPERATION_MODE,
- translation_key="operation_mode",
+ device_class=BinarySensorDeviceClass.POWER,
+ translation_key=ThinQProperty.WATER_HEATER_OPERATION_MODE,
on_key="power_on",
),
ThinQProperty.ONE_TOUCH_FILTER: ThinQBinarySensorEntityDescription(
@@ -136,7 +137,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for binary sensor platform."""
entities: list[ThinQBinarySensorEntity] = []
diff --git a/homeassistant/components/lg_thinq/climate.py b/homeassistant/components/lg_thinq/climate.py
index 5cf9ccbd442..98a86a8d355 100644
--- a/homeassistant/components/lg_thinq/climate.py
+++ b/homeassistant/components/lg_thinq/climate.py
@@ -2,7 +2,6 @@
from __future__ import annotations
-from dataclasses import dataclass
import logging
from typing import Any
@@ -10,8 +9,11 @@ from thinqconnect import DeviceType
from thinqconnect.integration import ExtendedProperty
from homeassistant.components.climate import (
+ ATTR_HVAC_MODE,
ATTR_TARGET_TEMP_HIGH,
ATTR_TARGET_TEMP_LOW,
+ SWING_OFF,
+ SWING_ON,
ClimateEntity,
ClimateEntityDescription,
ClimateEntityFeature,
@@ -19,38 +21,26 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.temperature import display_temp
from . import ThinqConfigEntry
from .coordinator import DeviceDataUpdateCoordinator
from .entity import ThinQEntity
-
-@dataclass(frozen=True, kw_only=True)
-class ThinQClimateEntityDescription(ClimateEntityDescription):
- """Describes ThinQ climate entity."""
-
- min_temp: float | None = None
- max_temp: float | None = None
- step: float | None = None
-
-
-DEVICE_TYPE_CLIMATE_MAP: dict[DeviceType, tuple[ThinQClimateEntityDescription, ...]] = {
+DEVICE_TYPE_CLIMATE_MAP: dict[DeviceType, tuple[ClimateEntityDescription, ...]] = {
DeviceType.AIR_CONDITIONER: (
- ThinQClimateEntityDescription(
+ ClimateEntityDescription(
key=ExtendedProperty.CLIMATE_AIR_CONDITIONER,
name=None,
translation_key=ExtendedProperty.CLIMATE_AIR_CONDITIONER,
),
),
DeviceType.SYSTEM_BOILER: (
- ThinQClimateEntityDescription(
+ ClimateEntityDescription(
key=ExtendedProperty.CLIMATE_SYSTEM_BOILER,
name=None,
- min_temp=16,
- max_temp=30,
- step=1,
+ translation_key=ExtendedProperty.CLIMATE_SYSTEM_BOILER,
),
),
}
@@ -63,23 +53,24 @@ STR_TO_HVAC: dict[str, HVACMode] = {
"heat": HVACMode.HEAT,
}
-HVAC_TO_STR: dict[HVACMode, str] = {
- HVACMode.AUTO: "auto",
- HVACMode.COOL: "cool",
- HVACMode.DRY: "air_dry",
- HVACMode.FAN_ONLY: "fan",
- HVACMode.HEAT: "heat",
-}
+HVAC_TO_STR = {v: k for k, v in STR_TO_HVAC.items()}
THINQ_PRESET_MODE: list[str] = ["air_clean", "aroma", "energy_saving"]
+STR_TO_SWING = {
+ "true": SWING_ON,
+ "false": SWING_OFF,
+}
+
+SWING_TO_STR = {v: k for k, v in STR_TO_SWING.items()}
+
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for climate platform."""
entities: list[ThinQClimateEntity] = []
@@ -102,12 +93,10 @@ async def async_setup_entry(
class ThinQClimateEntity(ThinQEntity, ClimateEntity):
"""Represent a thinq climate platform."""
- entity_description: ThinQClimateEntityDescription
-
def __init__(
self,
coordinator: DeviceDataUpdateCoordinator,
- entity_description: ThinQClimateEntityDescription,
+ entity_description: ClimateEntityDescription,
property_id: str,
) -> None:
"""Initialize a climate entity."""
@@ -121,7 +110,9 @@ class ThinQClimateEntity(ThinQEntity, ClimateEntity):
self._attr_hvac_modes = [HVACMode.OFF]
self._attr_hvac_mode = HVACMode.OFF
self._attr_preset_modes = []
- self._attr_temperature_unit = UnitOfTemperature.CELSIUS
+ self._attr_temperature_unit = (
+ self._get_unit_of_measurement(self.data.unit) or UnitOfTemperature.CELSIUS
+ )
self._requested_hvac_mode: str | None = None
# Set up HVAC modes.
@@ -142,6 +133,14 @@ class ThinQClimateEntity(ThinQEntity, ClimateEntity):
self._attr_supported_features |= (
ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
)
+ # Supports swing mode.
+ if self.data.swing_modes:
+ self._attr_swing_modes = [SWING_ON, SWING_OFF]
+ self._attr_supported_features |= ClimateEntityFeature.SWING_MODE
+
+ if self.data.swing_horizontal_modes:
+ self._attr_swing_horizontal_modes = [SWING_ON, SWING_OFF]
+ self._attr_supported_features |= ClimateEntityFeature.SWING_HORIZONTAL_MODE
def _update_status(self) -> None:
"""Update status itself."""
@@ -150,6 +149,13 @@ class ThinQClimateEntity(ThinQEntity, ClimateEntity):
# Update fan, hvac and preset mode.
if self.supported_features & ClimateEntityFeature.FAN_MODE:
self._attr_fan_mode = self.data.fan_mode
+ if self.supported_features & ClimateEntityFeature.SWING_MODE:
+ self._attr_swing_mode = STR_TO_SWING.get(self.data.swing_mode)
+ if self.supported_features & ClimateEntityFeature.SWING_HORIZONTAL_MODE:
+ self._attr_swing_horizontal_mode = STR_TO_SWING.get(
+ self.data.swing_horizontal_mode
+ )
+
if self.data.is_on:
hvac_mode = self._requested_hvac_mode or self.data.hvac_mode
if hvac_mode in STR_TO_HVAC:
@@ -166,24 +172,23 @@ class ThinQClimateEntity(ThinQEntity, ClimateEntity):
self._attr_current_temperature = self.data.current_temp
# Update min, max and step.
- if (max_temp := self.entity_description.max_temp) is not None or (
- max_temp := self.data.max
- ) is not None:
- self._attr_max_temp = max_temp
- if (min_temp := self.entity_description.min_temp) is not None or (
- min_temp := self.data.min
- ) is not None:
- self._attr_min_temp = min_temp
- if (step := self.entity_description.step) is not None or (
- step := self.data.step
- ) is not None:
- self._attr_target_temperature_step = step
+ if self.data.max is not None:
+ self._attr_max_temp = self.data.max
+ if self.data.min is not None:
+ self._attr_min_temp = self.data.min
+
+ self._attr_target_temperature_step = self.data.step
# Update target temperatures.
self._attr_target_temperature = self.data.target_temp
self._attr_target_temperature_high = self.data.target_temp_high
self._attr_target_temperature_low = self.data.target_temp_low
+ # Update unit.
+ self._attr_temperature_unit = (
+ self._get_unit_of_measurement(self.data.unit) or UnitOfTemperature.CELSIUS
+ )
+
_LOGGER.debug(
"[%s:%s] update status: c:%s, t:%s, l:%s, h:%s, hvac:%s, unit:%s, step:%s",
self.coordinator.device_name,
@@ -268,6 +273,34 @@ class ThinQClimateEntity(ThinQEntity, ClimateEntity):
self.coordinator.api.async_set_fan_mode(self.property_id, fan_mode)
)
+ async def async_set_swing_mode(self, swing_mode: str) -> None:
+ """Set new swing mode."""
+ _LOGGER.debug(
+ "[%s:%s] async_set_swing_mode: %s",
+ self.coordinator.device_name,
+ self.property_id,
+ swing_mode,
+ )
+ await self.async_call_api(
+ self.coordinator.api.async_set_swing_mode(
+ self.property_id, SWING_TO_STR.get(swing_mode)
+ )
+ )
+
+ async def async_set_swing_horizontal_mode(self, swing_horizontal_mode: str) -> None:
+ """Set new swing horizontal mode."""
+ _LOGGER.debug(
+ "[%s:%s] async_set_swing_horizontal_mode: %s",
+ self.coordinator.device_name,
+ self.property_id,
+ swing_horizontal_mode,
+ )
+ await self.async_call_api(
+ self.coordinator.api.async_set_swing_horizontal_mode(
+ self.property_id, SWING_TO_STR.get(swing_horizontal_mode)
+ )
+ )
+
def _round_by_step(self, temperature: float) -> float:
"""Round the value by step."""
if (
@@ -290,6 +323,10 @@ class ThinQClimateEntity(ThinQEntity, ClimateEntity):
self.property_id,
kwargs,
)
+ if hvac_mode := kwargs.get(ATTR_HVAC_MODE):
+ await self.async_set_hvac_mode(HVACMode(hvac_mode))
+ if hvac_mode == HVACMode.OFF:
+ return
if (temperature := kwargs.get(ATTR_TEMPERATURE)) is not None:
if (
diff --git a/homeassistant/components/lg_thinq/const.py b/homeassistant/components/lg_thinq/const.py
index a65dee715db..20c6455241a 100644
--- a/homeassistant/components/lg_thinq/const.py
+++ b/homeassistant/components/lg_thinq/const.py
@@ -3,6 +3,8 @@
from datetime import timedelta
from typing import Final
+from homeassistant.const import UnitOfTemperature
+
# Config flow
DOMAIN = "lg_thinq"
COMPANY = "LGE"
@@ -18,3 +20,10 @@ MQTT_SUBSCRIPTION_INTERVAL: Final = timedelta(days=1)
# MQTT: Message types
DEVICE_PUSH_MESSAGE: Final = "DEVICE_PUSH"
DEVICE_STATUS_MESSAGE: Final = "DEVICE_STATUS"
+
+# Unit conversion map
+DEVICE_UNIT_TO_HA: dict[str, str] = {
+ "F": UnitOfTemperature.FAHRENHEIT,
+ "C": UnitOfTemperature.CELSIUS,
+}
+REVERSE_DEVICE_UNIT_TO_HA = {v: k for k, v in DEVICE_UNIT_TO_HA.items()}
diff --git a/homeassistant/components/lg_thinq/coordinator.py b/homeassistant/components/lg_thinq/coordinator.py
index 9f317dc21d9..513cd27a7b2 100644
--- a/homeassistant/components/lg_thinq/coordinator.py
+++ b/homeassistant/components/lg_thinq/coordinator.py
@@ -2,16 +2,21 @@
from __future__ import annotations
+from collections.abc import Mapping
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
from thinqconnect import ThinQAPIException
from thinqconnect.integration import HABridge
-from homeassistant.core import HomeAssistant
+from homeassistant.const import EVENT_CORE_CONFIG_UPDATE
+from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import DOMAIN
+if TYPE_CHECKING:
+ from . import ThinqConfigEntry
+
+from .const import DOMAIN, REVERSE_DEVICE_UNIT_TO_HA
_LOGGER = logging.getLogger(__name__)
@@ -19,11 +24,16 @@ _LOGGER = logging.getLogger(__name__)
class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""LG Device's Data Update Coordinator."""
- def __init__(self, hass: HomeAssistant, ha_bridge: HABridge) -> None:
+ config_entry: ThinqConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ThinqConfigEntry, ha_bridge: HABridge
+ ) -> None:
"""Initialize data coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"{DOMAIN}_{ha_bridge.device.device_id}",
)
@@ -46,6 +56,40 @@ class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
f"{self.device_id}_{self.sub_id}" if self.sub_id else self.device_id
)
+ # Set your preferred temperature unit. This will allow us to retrieve
+ # temperature values from the API in a converted value corresponding to
+ # preferred unit.
+ self._update_preferred_temperature_unit()
+
+ # Add a callback to handle core config update.
+ self.unit_system: str | None = None
+ self.hass.bus.async_listen(
+ event_type=EVENT_CORE_CONFIG_UPDATE,
+ listener=self._handle_update_config,
+ event_filter=self.async_config_update_filter,
+ )
+
+ async def _handle_update_config(self, _: Event) -> None:
+ """Handle update core config."""
+ self._update_preferred_temperature_unit()
+
+ await self.async_refresh()
+
+ @callback
+ def async_config_update_filter(self, event_data: Mapping[str, Any]) -> bool:
+ """Filter out unwanted events."""
+ if (unit_system := event_data.get("unit_system")) != self.unit_system:
+ self.unit_system = unit_system
+ return True
+
+ return False
+
+ def _update_preferred_temperature_unit(self) -> None:
+ """Update preferred temperature unit."""
+ self.api.set_preferred_temperature_unit(
+ REVERSE_DEVICE_UNIT_TO_HA.get(self.hass.config.units.temperature_unit)
+ )
+
async def _async_update_data(self) -> dict[str, Any]:
"""Request to the server to update the status from full response data."""
try:
@@ -71,10 +115,10 @@ class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
async def async_setup_device_coordinator(
- hass: HomeAssistant, ha_bridge: HABridge
+ hass: HomeAssistant, config_entry: ThinqConfigEntry, ha_bridge: HABridge
) -> DeviceDataUpdateCoordinator:
"""Create DeviceDataUpdateCoordinator and device_api per device."""
- coordinator = DeviceDataUpdateCoordinator(hass, ha_bridge)
+ coordinator = DeviceDataUpdateCoordinator(hass, config_entry, ha_bridge)
await coordinator.async_refresh()
_LOGGER.debug(
diff --git a/homeassistant/components/lg_thinq/entity.py b/homeassistant/components/lg_thinq/entity.py
index 7856506559b..61d8199f321 100644
--- a/homeassistant/components/lg_thinq/entity.py
+++ b/homeassistant/components/lg_thinq/entity.py
@@ -10,25 +10,19 @@ from thinqconnect import ThinQAPIException
from thinqconnect.devices.const import Location
from thinqconnect.integration import PropertyState
-from homeassistant.const import UnitOfTemperature
from homeassistant.core import callback
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .const import COMPANY, DOMAIN
+from .const import COMPANY, DEVICE_UNIT_TO_HA, DOMAIN
from .coordinator import DeviceDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
EMPTY_STATE = PropertyState()
-UNIT_CONVERSION_MAP: dict[str, str] = {
- "F": UnitOfTemperature.FAHRENHEIT,
- "C": UnitOfTemperature.CELSIUS,
-}
-
class ThinQEntity(CoordinatorEntity[DeviceDataUpdateCoordinator]):
"""The base implementation of all lg thinq entities."""
@@ -75,7 +69,7 @@ class ThinQEntity(CoordinatorEntity[DeviceDataUpdateCoordinator]):
if unit is None:
return None
- return UNIT_CONVERSION_MAP.get(unit)
+ return DEVICE_UNIT_TO_HA.get(unit)
def _update_status(self) -> None:
"""Update status itself.
diff --git a/homeassistant/components/lg_thinq/event.py b/homeassistant/components/lg_thinq/event.py
index b963cba37cc..f9baadf7a05 100644
--- a/homeassistant/components/lg_thinq/event.py
+++ b/homeassistant/components/lg_thinq/event.py
@@ -9,7 +9,7 @@ from thinqconnect.integration import ActiveMode, ThinQPropertyEx
from homeassistant.components.event import EventEntity, EventEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ThinqConfigEntry
from .coordinator import DeviceDataUpdateCoordinator
@@ -57,7 +57,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for event platform."""
entities: list[ThinQEventEntity] = []
diff --git a/homeassistant/components/lg_thinq/fan.py b/homeassistant/components/lg_thinq/fan.py
index edcadf2598a..6d07c98744a 100644
--- a/homeassistant/components/lg_thinq/fan.py
+++ b/homeassistant/components/lg_thinq/fan.py
@@ -14,7 +14,7 @@ from homeassistant.components.fan import (
FanEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -41,7 +41,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for fan platform."""
entities: list[ThinQFanEntity] = []
diff --git a/homeassistant/components/lg_thinq/icons.json b/homeassistant/components/lg_thinq/icons.json
index 42ae5746f24..787b50167c1 100644
--- a/homeassistant/components/lg_thinq/icons.json
+++ b/homeassistant/components/lg_thinq/icons.json
@@ -80,6 +80,9 @@
},
"one_touch_filter": {
"default": "mdi:air-filter"
+ },
+ "water_heater_operation_mode": {
+ "default": "mdi:power"
}
},
"climate": {
@@ -407,6 +410,12 @@
},
"power_level_for_location": {
"default": "mdi:radiator"
+ },
+ "cycle_count": {
+ "default": "mdi:counter"
+ },
+ "cycle_count_for_location": {
+ "default": "mdi:counter"
}
}
}
diff --git a/homeassistant/components/lg_thinq/manifest.json b/homeassistant/components/lg_thinq/manifest.json
index b00d28c1d4f..cffc61cb1c4 100644
--- a/homeassistant/components/lg_thinq/manifest.json
+++ b/homeassistant/components/lg_thinq/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/lg_thinq",
"iot_class": "cloud_push",
"loggers": ["thinqconnect"],
- "requirements": ["thinqconnect==1.0.4"]
+ "requirements": ["thinqconnect==1.0.5"]
}
diff --git a/homeassistant/components/lg_thinq/number.py b/homeassistant/components/lg_thinq/number.py
index 634c1a8fe84..7003519e0ce 100644
--- a/homeassistant/components/lg_thinq/number.py
+++ b/homeassistant/components/lg_thinq/number.py
@@ -16,7 +16,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import PERCENTAGE, UnitOfTemperature, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ThinqConfigEntry
from .entity import ThinQEntity
@@ -118,16 +118,7 @@ DEVICE_TYPE_NUMBER_MAP: dict[DeviceType, tuple[NumberEntityDescription, ...]] =
DeviceType.WASHTOWER_DRYER: WASHER_NUMBERS,
DeviceType.WASHTOWER: WASHER_NUMBERS,
DeviceType.WASHTOWER_WASHER: WASHER_NUMBERS,
- DeviceType.WATER_HEATER: (
- NumberEntityDescription(
- key=ThinQProperty.TARGET_TEMPERATURE,
- native_max_value=60,
- native_min_value=35,
- native_step=1,
- native_unit_of_measurement=UnitOfTemperature.CELSIUS,
- translation_key=ThinQProperty.TARGET_TEMPERATURE,
- ),
- ),
+ DeviceType.WATER_HEATER: (NUMBER_DESC[ThinQProperty.TARGET_TEMPERATURE],),
DeviceType.WINE_CELLAR: (
NUMBER_DESC[ThinQProperty.LIGHT_STATUS],
NUMBER_DESC[ThinQProperty.TARGET_TEMPERATURE],
@@ -140,7 +131,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for number platform."""
entities: list[ThinQNumberEntity] = []
@@ -179,7 +170,7 @@ class ThinQNumberEntity(ThinQEntity, NumberEntity):
) is not None:
self._attr_native_unit_of_measurement = unit_of_measurement
- # Undate range.
+ # Update range.
if (
self.entity_description.native_min_value is None
and (min_value := self.data.min) is not None
diff --git a/homeassistant/components/lg_thinq/select.py b/homeassistant/components/lg_thinq/select.py
index e555d616ca3..929fa0b1d28 100644
--- a/homeassistant/components/lg_thinq/select.py
+++ b/homeassistant/components/lg_thinq/select.py
@@ -10,7 +10,7 @@ from thinqconnect.integration import ActiveMode
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ThinqConfigEntry
from .coordinator import DeviceDataUpdateCoordinator
@@ -142,7 +142,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for select platform."""
entities: list[ThinQSelectEntity] = []
diff --git a/homeassistant/components/lg_thinq/sensor.py b/homeassistant/components/lg_thinq/sensor.py
index 7baaab52403..754b07cb2db 100644
--- a/homeassistant/components/lg_thinq/sensor.py
+++ b/homeassistant/components/lg_thinq/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import ThinqConfigEntry
@@ -248,6 +248,24 @@ TEMPERATURE_SENSOR_DESC: dict[ThinQProperty, SensorEntityDescription] = {
state_class=SensorStateClass.MEASUREMENT,
translation_key=ThinQProperty.CURRENT_TEMPERATURE,
),
+ ThinQPropertyEx.ROOM_AIR_CURRENT_TEMPERATURE: SensorEntityDescription(
+ key=ThinQPropertyEx.ROOM_AIR_CURRENT_TEMPERATURE,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ translation_key=ThinQPropertyEx.ROOM_AIR_CURRENT_TEMPERATURE,
+ ),
+ ThinQPropertyEx.ROOM_IN_WATER_CURRENT_TEMPERATURE: SensorEntityDescription(
+ key=ThinQPropertyEx.ROOM_IN_WATER_CURRENT_TEMPERATURE,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ translation_key=ThinQPropertyEx.ROOM_IN_WATER_CURRENT_TEMPERATURE,
+ ),
+ ThinQPropertyEx.ROOM_OUT_WATER_CURRENT_TEMPERATURE: SensorEntityDescription(
+ key=ThinQPropertyEx.ROOM_OUT_WATER_CURRENT_TEMPERATURE,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ translation_key=ThinQPropertyEx.ROOM_OUT_WATER_CURRENT_TEMPERATURE,
+ ),
}
WATER_FILTER_INFO_SENSOR_DESC: dict[ThinQProperty, SensorEntityDescription] = {
ThinQProperty.USED_TIME: SensorEntityDescription(
@@ -341,6 +359,10 @@ TIMER_SENSOR_DESC: dict[ThinQProperty, SensorEntityDescription] = {
}
WASHER_SENSORS: tuple[SensorEntityDescription, ...] = (
+ SensorEntityDescription(
+ key=ThinQProperty.CYCLE_COUNT,
+ translation_key=ThinQProperty.CYCLE_COUNT,
+ ),
RUN_STATE_SENSOR_DESC[ThinQProperty.CURRENT_STATE],
TIMER_SENSOR_DESC[TimerProperty.TOTAL],
TIMER_SENSOR_DESC[TimerProperty.RELATIVE_TO_START_WM],
@@ -470,6 +492,11 @@ DEVICE_TYPE_SENSOR_MAP: dict[DeviceType, tuple[SensorEntityDescription, ...]] =
RUN_STATE_SENSOR_DESC[ThinQProperty.CURRENT_STATE],
),
DeviceType.STYLER: WASHER_SENSORS,
+ DeviceType.SYSTEM_BOILER: (
+ TEMPERATURE_SENSOR_DESC[ThinQPropertyEx.ROOM_AIR_CURRENT_TEMPERATURE],
+ TEMPERATURE_SENSOR_DESC[ThinQPropertyEx.ROOM_IN_WATER_CURRENT_TEMPERATURE],
+ TEMPERATURE_SENSOR_DESC[ThinQPropertyEx.ROOM_OUT_WATER_CURRENT_TEMPERATURE],
+ ),
DeviceType.WASHCOMBO_MAIN: WASHER_SENSORS,
DeviceType.WASHCOMBO_MINI: WASHER_SENSORS,
DeviceType.WASHER: WASHER_SENSORS,
@@ -492,7 +519,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for sensor platform."""
entities: list[ThinQSensorEntity] = []
@@ -554,36 +581,44 @@ class ThinQSensorEntity(ThinQEntity, SensorEntity):
local_now = datetime.now(
tz=dt_util.get_time_zone(self.coordinator.hass.config.time_zone)
)
- if value in [0, None, time.min]:
- # Reset to None
+ self._device_state = (
+ self.coordinator.data[self._device_state_id].value
+ if self._device_state_id in self.coordinator.data
+ else None
+ )
+ if value in [0, None, time.min] or (
+ self._device_state == "power_off"
+ and self.entity_description.key
+ in [TimerProperty.REMAIN, TimerProperty.TOTAL]
+ ):
+ # Reset to None when power_off
value = None
elif self.entity_description.device_class == SensorDeviceClass.TIMESTAMP:
if self.entity_description.key in TIME_SENSOR_DESC:
- # Set timestamp for time
+ # Set timestamp for absolute time
value = local_now.replace(hour=value.hour, minute=value.minute)
else:
# Set timestamp for delta
- new_state = (
- self.coordinator.data[self._device_state_id].value
- if self._device_state_id in self.coordinator.data
- else None
- )
- if (
- self.native_value is not None
- and self._device_state == new_state
- ):
- # Skip update when same state
- return
-
- self._device_state = new_state
- time_delta = timedelta(
+ event_data = timedelta(
hours=value.hour, minutes=value.minute, seconds=value.second
)
- value = (
- (local_now - time_delta)
+ new_time = (
+ (local_now - event_data)
if self.entity_description.key == TimerProperty.RUNNING
- else (local_now + time_delta)
+ else (local_now + event_data)
)
+ # The remain_time may change during the wash/dry operation depending on various reasons.
+ # If there is a diff of more than 60sec, the new timestamp is used
+ if (
+ parse_native_value := dt_util.parse_datetime(
+ str(self.native_value)
+ )
+ ) is None or abs(new_time - parse_native_value) > timedelta(
+ seconds=60
+ ):
+ value = new_time
+ else:
+ value = self.native_value
elif self.entity_description.device_class == SensorDeviceClass.DURATION:
# Set duration
value = self._get_duration(
diff --git a/homeassistant/components/lg_thinq/strings.json b/homeassistant/components/lg_thinq/strings.json
index dee2d21e05a..e1d3779f44b 100644
--- a/homeassistant/components/lg_thinq/strings.json
+++ b/homeassistant/components/lg_thinq/strings.json
@@ -105,6 +105,12 @@
},
"one_touch_filter": {
"name": "Fresh air filter"
+ },
+ "water_heater_operation_mode": {
+ "name": "[%key:component::binary_sensor::entity_component::power::name%]",
+ "state": {
+ "off": "[%key:common::state::standby%]"
+ }
}
},
"climate": {
@@ -264,10 +270,10 @@
"name": "{location} schedule turn-on"
},
"relative_hour_to_start_wm": {
- "name": "Delay starts in"
+ "name": "Delayed start"
},
"relative_hour_to_start_wm_for_location": {
- "name": "{location} delay starts in"
+ "name": "{location} delayed start"
},
"relative_hour_to_stop": {
"name": "Schedule turn-off"
@@ -276,10 +282,10 @@
"name": "{location} schedule turn-off"
},
"relative_hour_to_stop_wm": {
- "name": "Delay ends in"
+ "name": "Delayed end"
},
"relative_hour_to_stop_wm_for_location": {
- "name": "{location} delay ends in"
+ "name": "{location} delayed end"
},
"sleep_timer_relative_hour_to_stop": {
"name": "Sleep timer"
@@ -305,6 +311,15 @@
"current_temperature": {
"name": "Current temperature"
},
+ "room_air_current_temperature": {
+ "name": "Indoor temperature"
+ },
+ "room_in_water_current_temperature": {
+ "name": "Inlet temperature"
+ },
+ "room_out_water_current_temperature": {
+ "name": "Outlet temperature"
+ },
"temperature": {
"name": "Temperature"
},
@@ -396,7 +411,7 @@
"cancel": "Cancel",
"carbonation": "Carbonation",
"change_condition": "Settings Change",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"charging_complete": "Charging completed",
"checking_turbidity": "Detecting soil level",
"cleaning": "Cleaning",
@@ -483,7 +498,7 @@
"cancel": "[%key:component::lg_thinq::entity::sensor::current_state::state::cancel%]",
"carbonation": "[%key:component::lg_thinq::entity::sensor::current_state::state::carbonation%]",
"change_condition": "[%key:component::lg_thinq::entity::sensor::current_state::state::change_condition%]",
- "charging": "[%key:component::lg_thinq::entity::sensor::current_state::state::charging%]",
+ "charging": "[%key:common::state::charging%]",
"charging_complete": "[%key:component::lg_thinq::entity::sensor::current_state::state::charging_complete%]",
"checking_turbidity": "[%key:component::lg_thinq::entity::sensor::current_state::state::checking_turbidity%]",
"cleaning": "[%key:component::lg_thinq::entity::sensor::current_state::state::cleaning%]",
@@ -848,6 +863,12 @@
},
"power_level_for_location": {
"name": "{location} power level"
+ },
+ "cycle_count": {
+ "name": "Cycles"
+ },
+ "cycle_count_for_location": {
+ "name": "{location} cycles"
}
},
"select": {
@@ -912,6 +933,7 @@
"state": {
"cancel": "[%key:component::lg_thinq::entity::sensor::current_state::state::cancel%]",
"power_off": "Power off",
+ "power_on": "Power on",
"preheating": "Preheating",
"start": "[%key:common::action::start%]",
"stop": "[%key:common::action::stop%]",
@@ -923,6 +945,7 @@
"state": {
"cancel": "[%key:component::lg_thinq::entity::sensor::current_state::state::cancel%]",
"power_off": "[%key:component::lg_thinq::entity::select::operation_mode::state::power_off%]",
+ "power_on": "[%key:component::lg_thinq::entity::select::operation_mode::state::power_on%]",
"preheating": "[%key:component::lg_thinq::entity::select::operation_mode::state::preheating%]",
"start": "[%key:common::action::start%]",
"stop": "[%key:common::action::stop%]",
diff --git a/homeassistant/components/lg_thinq/switch.py b/homeassistant/components/lg_thinq/switch.py
index 6d69ce9a314..06363140193 100644
--- a/homeassistant/components/lg_thinq/switch.py
+++ b/homeassistant/components/lg_thinq/switch.py
@@ -17,7 +17,7 @@ from homeassistant.components.switch import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ThinqConfigEntry
from .entity import ThinQEntity
@@ -172,7 +172,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for switch platform."""
entities: list[ThinQSwitchEntity] = []
diff --git a/homeassistant/components/lg_thinq/vacuum.py b/homeassistant/components/lg_thinq/vacuum.py
index 6cbb731869c..6cf2a9086b1 100644
--- a/homeassistant/components/lg_thinq/vacuum.py
+++ b/homeassistant/components/lg_thinq/vacuum.py
@@ -15,7 +15,7 @@ from homeassistant.components.vacuum import (
VacuumEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ThinqConfigEntry
from .entity import ThinQEntity
@@ -73,7 +73,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ThinqConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an entry for vacuum platform."""
entities: list[ThinQStateVacuumEntity] = []
diff --git a/homeassistant/components/lg_thinq/water_heater.py b/homeassistant/components/lg_thinq/water_heater.py
new file mode 100644
index 00000000000..5a5c8d024b6
--- /dev/null
+++ b/homeassistant/components/lg_thinq/water_heater.py
@@ -0,0 +1,201 @@
+"""Support for waterheater entities."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from thinqconnect import DeviceType
+from thinqconnect.integration import ExtendedProperty
+
+from homeassistant.components.water_heater import (
+ ATTR_OPERATION_MODE,
+ STATE_ECO,
+ STATE_HEAT_PUMP,
+ STATE_OFF,
+ STATE_PERFORMANCE,
+ WaterHeaterEntity,
+ WaterHeaterEntityDescription,
+ WaterHeaterEntityFeature,
+)
+from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import ThinqConfigEntry
+from .coordinator import DeviceDataUpdateCoordinator
+from .entity import ThinQEntity
+
+DEVICE_TYPE_WH_MAP: dict[DeviceType, WaterHeaterEntityDescription] = {
+ DeviceType.WATER_HEATER: WaterHeaterEntityDescription(
+ key=ExtendedProperty.WATER_HEATER,
+ name=None,
+ ),
+ DeviceType.SYSTEM_BOILER: WaterHeaterEntityDescription(
+ key=ExtendedProperty.WATER_BOILER,
+ name=None,
+ ),
+}
+
+# Mapping between device and HA operation modes
+DEVICE_OP_MODE_TO_HA = {
+ "auto": STATE_ECO,
+ "heat_pump": STATE_HEAT_PUMP,
+ "turbo": STATE_PERFORMANCE,
+ "vacation": STATE_OFF,
+}
+HA_STATE_TO_DEVICE_OP_MODE = {v: k for k, v in DEVICE_OP_MODE_TO_HA.items()}
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: ThinqConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up an entry for water_heater platform."""
+ entities: list[ThinQWaterHeaterEntity] = []
+ for coordinator in entry.runtime_data.coordinators.values():
+ if (
+ description := DEVICE_TYPE_WH_MAP.get(coordinator.api.device.device_type)
+ ) is not None:
+ if coordinator.api.device.device_type == DeviceType.WATER_HEATER:
+ entities.append(
+ ThinQWaterHeaterEntity(
+ coordinator, description, ExtendedProperty.WATER_HEATER
+ )
+ )
+ elif coordinator.api.device.device_type == DeviceType.SYSTEM_BOILER:
+ entities.append(
+ ThinQWaterBoilerEntity(
+ coordinator, description, ExtendedProperty.WATER_BOILER
+ )
+ )
+ if entities:
+ async_add_entities(entities)
+
+
+class ThinQWaterHeaterEntity(ThinQEntity, WaterHeaterEntity):
+ """Represent a ThinQ water heater entity."""
+
+ def __init__(
+ self,
+ coordinator: DeviceDataUpdateCoordinator,
+ entity_description: WaterHeaterEntityDescription,
+ property_id: str,
+ ) -> None:
+ """Initialize a water_heater entity."""
+ super().__init__(coordinator, entity_description, property_id)
+ self._attr_supported_features = (
+ WaterHeaterEntityFeature.TARGET_TEMPERATURE
+ | WaterHeaterEntityFeature.OPERATION_MODE
+ )
+ self._attr_temperature_unit = (
+ self._get_unit_of_measurement(self.data.unit) or UnitOfTemperature.CELSIUS
+ )
+ if modes := self.data.job_modes:
+ self._attr_operation_list = [
+ DEVICE_OP_MODE_TO_HA.get(mode, mode) for mode in modes
+ ]
+ else:
+ self._attr_operation_list = [STATE_HEAT_PUMP]
+
+ def _update_status(self) -> None:
+ """Update status itself."""
+ super()._update_status()
+ self._attr_current_temperature = self.data.current_temp
+ self._attr_target_temperature = self.data.target_temp
+
+ if self.data.max is not None:
+ self._attr_max_temp = self.data.max
+ if self.data.min is not None:
+ self._attr_min_temp = self.data.min
+ if self.data.step is not None:
+ self._attr_target_temperature_step = self.data.step
+
+ self._attr_temperature_unit = (
+ self._get_unit_of_measurement(self.data.unit) or UnitOfTemperature.CELSIUS
+ )
+ if self.data.is_on:
+ self._attr_current_operation = (
+ DEVICE_OP_MODE_TO_HA.get(job_mode, job_mode)
+ if (job_mode := self.data.job_mode) is not None
+ else STATE_HEAT_PUMP
+ )
+ else:
+ self._attr_current_operation = STATE_OFF
+
+ _LOGGER.debug(
+ "[%s:%s] update status: c:%s, t:%s, op_mode:%s, op_list:%s, is_on:%s",
+ self.coordinator.device_name,
+ self.property_id,
+ self.current_temperature,
+ self.target_temperature,
+ self.current_operation,
+ self.operation_list,
+ self.data.is_on,
+ )
+
+ async def async_set_temperature(self, **kwargs: Any) -> None:
+ """Set new target temperatures."""
+ _LOGGER.debug(
+ "[%s:%s] async_set_temperature: %s",
+ self.coordinator.device_name,
+ self.property_id,
+ kwargs,
+ )
+ if (operation_mode := kwargs.get(ATTR_OPERATION_MODE)) is not None:
+ await self.async_set_operation_mode(str(operation_mode))
+ if operation_mode == STATE_OFF:
+ return
+
+ if (
+ temperature := kwargs.get(ATTR_TEMPERATURE)
+ ) is not None and temperature != self.target_temperature:
+ await self.async_call_api(
+ self.coordinator.api.async_set_target_temperature(
+ self.property_id, temperature
+ )
+ )
+
+ async def async_set_operation_mode(self, operation_mode: str) -> None:
+ """Set new operation mode."""
+ mode = HA_STATE_TO_DEVICE_OP_MODE.get(operation_mode, operation_mode)
+ _LOGGER.debug(
+ "[%s:%s] async_set_operation_mode: %s",
+ self.coordinator.device_name,
+ self.property_id,
+ mode,
+ )
+ await self.async_call_api(
+ self.coordinator.api.async_set_job_mode(self.property_id, mode)
+ )
+
+
+class ThinQWaterBoilerEntity(ThinQWaterHeaterEntity):
+ """Represent a ThinQ water boiler entity."""
+
+ def __init__(
+ self,
+ coordinator: DeviceDataUpdateCoordinator,
+ entity_description: WaterHeaterEntityDescription,
+ property_id: str,
+ ) -> None:
+ """Initialize a water_heater entity."""
+ super().__init__(coordinator, entity_description, property_id)
+ self._attr_supported_features |= WaterHeaterEntityFeature.ON_OFF
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the entity on."""
+ _LOGGER.debug(
+ "[%s:%s] async_turn_on", self.coordinator.device_name, self.property_id
+ )
+ await self.async_call_api(self.coordinator.api.async_turn_on(self.property_id))
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the entity off."""
+ _LOGGER.debug(
+ "[%s:%s] async_turn_off", self.coordinator.device_name, self.property_id
+ )
+ await self.async_call_api(self.coordinator.api.async_turn_off(self.property_id))
diff --git a/homeassistant/components/lidarr/__init__.py b/homeassistant/components/lidarr/__init__.py
index a421a881b69..e3a5cf250b2 100644
--- a/homeassistant/components/lidarr/__init__.py
+++ b/homeassistant/components/lidarr/__init__.py
@@ -2,12 +2,11 @@
from __future__ import annotations
-from dataclasses import dataclass, fields
+from dataclasses import fields
from aiopyarr.lidarr_client import LidarrClient
from aiopyarr.models.host_configuration import PyArrHostConfiguration
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY, CONF_URL, CONF_VERIFY_SSL, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
@@ -18,27 +17,16 @@ from .const import DEFAULT_NAME, DOMAIN
from .coordinator import (
AlbumsDataUpdateCoordinator,
DiskSpaceDataUpdateCoordinator,
+ LidarrConfigEntry,
+ LidarrData,
QueueDataUpdateCoordinator,
StatusDataUpdateCoordinator,
WantedDataUpdateCoordinator,
)
-type LidarrConfigEntry = ConfigEntry[LidarrData]
-
PLATFORMS = [Platform.SENSOR]
-@dataclass(kw_only=True, slots=True)
-class LidarrData:
- """Lidarr data type."""
-
- disk_space: DiskSpaceDataUpdateCoordinator
- queue: QueueDataUpdateCoordinator
- status: StatusDataUpdateCoordinator
- wanted: WantedDataUpdateCoordinator
- albums: AlbumsDataUpdateCoordinator
-
-
async def async_setup_entry(hass: HomeAssistant, entry: LidarrConfigEntry) -> bool:
"""Set up Lidarr from a config entry."""
host_configuration = PyArrHostConfiguration(
@@ -52,11 +40,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: LidarrConfigEntry) -> bo
request_timeout=60,
)
data = LidarrData(
- disk_space=DiskSpaceDataUpdateCoordinator(hass, host_configuration, lidarr),
- queue=QueueDataUpdateCoordinator(hass, host_configuration, lidarr),
- status=StatusDataUpdateCoordinator(hass, host_configuration, lidarr),
- wanted=WantedDataUpdateCoordinator(hass, host_configuration, lidarr),
- albums=AlbumsDataUpdateCoordinator(hass, host_configuration, lidarr),
+ disk_space=DiskSpaceDataUpdateCoordinator(
+ hass, entry, host_configuration, lidarr
+ ),
+ queue=QueueDataUpdateCoordinator(hass, entry, host_configuration, lidarr),
+ status=StatusDataUpdateCoordinator(hass, entry, host_configuration, lidarr),
+ wanted=WantedDataUpdateCoordinator(hass, entry, host_configuration, lidarr),
+ albums=AlbumsDataUpdateCoordinator(hass, entry, host_configuration, lidarr),
)
for field in fields(data):
coordinator = getattr(data, field.name)
diff --git a/homeassistant/components/lidarr/coordinator.py b/homeassistant/components/lidarr/coordinator.py
index 1010f708748..3f9d2be4bec 100644
--- a/homeassistant/components/lidarr/coordinator.py
+++ b/homeassistant/components/lidarr/coordinator.py
@@ -3,6 +3,7 @@
from __future__ import annotations
from abc import ABC, abstractmethod
+from dataclasses import dataclass
from datetime import timedelta
from typing import Generic, TypeVar, cast
@@ -17,17 +18,32 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DEFAULT_MAX_RECORDS, DOMAIN, LOGGER
+
+@dataclass(kw_only=True, slots=True)
+class LidarrData:
+ """Lidarr data type."""
+
+ disk_space: DiskSpaceDataUpdateCoordinator
+ queue: QueueDataUpdateCoordinator
+ status: StatusDataUpdateCoordinator
+ wanted: WantedDataUpdateCoordinator
+ albums: AlbumsDataUpdateCoordinator
+
+
T = TypeVar("T", bound=list[LidarrRootFolder] | LidarrQueue | str | LidarrAlbum | int)
+type LidarrConfigEntry = ConfigEntry[LidarrData]
+
class LidarrDataUpdateCoordinator(DataUpdateCoordinator[T], Generic[T], ABC):
"""Data update coordinator for the Lidarr integration."""
- config_entry: ConfigEntry
+ config_entry: LidarrConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: LidarrConfigEntry,
host_configuration: PyArrHostConfiguration,
api_client: LidarrClient,
) -> None:
@@ -35,6 +51,7 @@ class LidarrDataUpdateCoordinator(DataUpdateCoordinator[T], Generic[T], ABC):
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/lidarr/sensor.py b/homeassistant/components/lidarr/sensor.py
index 805fcce53ad..81b2c570eab 100644
--- a/homeassistant/components/lidarr/sensor.py
+++ b/homeassistant/components/lidarr/sensor.py
@@ -16,11 +16,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import LidarrConfigEntry
from .const import BYTE_SIZES
-from .coordinator import LidarrDataUpdateCoordinator, T
+from .coordinator import LidarrConfigEntry, LidarrDataUpdateCoordinator, T
from .entity import LidarrEntity
@@ -115,7 +114,7 @@ SENSOR_TYPES: dict[str, LidarrSensorEntityDescription[Any]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: LidarrConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lidarr sensors based on a config entry."""
entities: list[LidarrSensor[Any]] = []
diff --git a/homeassistant/components/life360/__init__.py b/homeassistant/components/life360/__init__.py
index 5c2d62545d6..60c1ac753e6 100644
--- a/homeassistant/components/life360/__init__.py
+++ b/homeassistant/components/life360/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
@@ -26,11 +26,14 @@ async def async_setup_entry(hass: HomeAssistant, _: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
- """Unload config entry."""
- if all(
- config_entry.state is ConfigEntryState.NOT_LOADED
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.entry_id != entry.entry_id
- ):
- ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ """Unload a config entry."""
return True
+
+
+async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Remove a config entry."""
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
+ ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ # Remove any remaining disabled or ignored entries
+ for _entry in hass.config_entries.async_entries(DOMAIN):
+ hass.async_create_task(hass.config_entries.async_remove(_entry.entry_id))
diff --git a/homeassistant/components/lifx/__init__.py b/homeassistant/components/lifx/__init__.py
index 2847862029f..7a6d95549ff 100644
--- a/homeassistant/components/lifx/__init__.py
+++ b/homeassistant/components/lifx/__init__.py
@@ -211,7 +211,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except socket.gaierror as ex:
connection.async_stop()
raise ConfigEntryNotReady(f"Could not resolve {host}: {ex}") from ex
- coordinator = LIFXUpdateCoordinator(hass, connection, entry.title)
+ coordinator = LIFXUpdateCoordinator(hass, entry, connection)
coordinator.async_setup()
try:
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/lifx/binary_sensor.py b/homeassistant/components/lifx/binary_sensor.py
index 454561a6f4e..f5a974b4626 100644
--- a/homeassistant/components/lifx/binary_sensor.py
+++ b/homeassistant/components/lifx/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, HEV_CYCLE_STATE
from .coordinator import LIFXUpdateCoordinator
@@ -26,7 +26,9 @@ HEV_CYCLE_STATE_SENSOR = BinarySensorEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LIFX from a config entry."""
coordinator: LIFXUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lifx/button.py b/homeassistant/components/lifx/button.py
index 694c91b4c27..25ab61aebae 100644
--- a/homeassistant/components/lifx/button.py
+++ b/homeassistant/components/lifx/button.py
@@ -10,7 +10,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, IDENTIFY, RESTART
from .coordinator import LIFXUpdateCoordinator
@@ -32,7 +32,7 @@ IDENTIFY_BUTTON_DESCRIPTION = ButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LIFX from a config entry."""
domain_data = hass.data[DOMAIN]
diff --git a/homeassistant/components/lifx/coordinator.py b/homeassistant/components/lifx/coordinator.py
index eaaff7e6540..b77dbdc015a 100644
--- a/homeassistant/components/lifx/coordinator.py
+++ b/homeassistant/components/lifx/coordinator.py
@@ -23,6 +23,7 @@ from aiolifx_themes.themes import ThemeLibrary, ThemePainter
from awesomeversion import AwesomeVersion
from propcache.api import cached_property
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
SIGNAL_STRENGTH_DECIBELS,
SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
@@ -86,11 +87,13 @@ class SkyType(IntEnum):
class LIFXUpdateCoordinator(DataUpdateCoordinator[None]):
"""DataUpdateCoordinator to gather data for a specific lifx device."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
connection: LIFXConnection,
- title: str,
) -> None:
"""Initialize DataUpdateCoordinator."""
assert connection.device is not None
@@ -105,7 +108,8 @@ class LIFXUpdateCoordinator(DataUpdateCoordinator[None]):
super().__init__(
hass,
_LOGGER,
- name=f"{title} ({self.device.ip_addr})",
+ config_entry=config_entry,
+ name=f"{config_entry.title} ({self.device.ip_addr})",
update_interval=timedelta(seconds=LIGHT_UPDATE_INTERVAL),
# We don't want an immediate refresh since the device
# takes a moment to reflect the state change
diff --git a/homeassistant/components/lifx/light.py b/homeassistant/components/lifx/light.py
index 2a8031b3874..5641786eb61 100644
--- a/homeassistant/components/lifx/light.py
+++ b/homeassistant/components/lifx/light.py
@@ -22,7 +22,7 @@ from homeassistant.const import ATTR_ENTITY_ID, Platform
from homeassistant.core import CALLBACK_TYPE, HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.typing import VolDictType
@@ -79,7 +79,7 @@ HSBK_KELVIN = 3
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LIFX from a config entry."""
domain_data = hass.data[DOMAIN]
diff --git a/homeassistant/components/lifx/manifest.json b/homeassistant/components/lifx/manifest.json
index 8d460c25322..18b9457ebf4 100644
--- a/homeassistant/components/lifx/manifest.json
+++ b/homeassistant/components/lifx/manifest.json
@@ -51,7 +51,7 @@
"iot_class": "local_polling",
"loggers": ["aiolifx", "aiolifx_effects", "bitstring"],
"requirements": [
- "aiolifx==1.1.2",
+ "aiolifx==1.1.4",
"aiolifx-effects==0.3.2",
"aiolifx-themes==0.6.4"
]
diff --git a/homeassistant/components/lifx/select.py b/homeassistant/components/lifx/select.py
index de3a5b431a9..13b81e2a784 100644
--- a/homeassistant/components/lifx/select.py
+++ b/homeassistant/components/lifx/select.py
@@ -8,7 +8,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_THEME,
@@ -38,7 +38,9 @@ THEME_ENTITY = SelectEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LIFX from a config entry."""
coordinator: LIFXUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lifx/sensor.py b/homeassistant/components/lifx/sensor.py
index 68f354024e4..96feba633f4 100644
--- a/homeassistant/components/lifx/sensor.py
+++ b/homeassistant/components/lifx/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_RSSI, DOMAIN
from .coordinator import LIFXUpdateCoordinator
@@ -32,7 +32,9 @@ RSSI_SENSOR = SensorEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up LIFX sensor from config entry."""
coordinator: LIFXUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lifx/strings.json b/homeassistant/components/lifx/strings.json
index 39102d904d5..be0485c6dff 100644
--- a/homeassistant/components/lifx/strings.json
+++ b/homeassistant/components/lifx/strings.json
@@ -66,7 +66,7 @@
}
},
"set_state": {
- "name": "Set State",
+ "name": "Set state",
"description": "Sets a color/brightness and possibly turn the light on/off.",
"fields": {
"infrared": {
@@ -201,7 +201,7 @@
},
"effect_morph": {
"name": "Morph effect",
- "description": "Starts the firmware-based Morph effect on LIFX Tiles on Candle.",
+ "description": "Starts the firmware-based Morph effect on LIFX Tiles or Candle.",
"fields": {
"speed": {
"name": "Speed",
@@ -209,11 +209,11 @@
},
"palette": {
"name": "Palette",
- "description": "List of at least 2 and at most 16 colors as hue (0-360), saturation (0-100), brightness (0-100) and kelvin (1500-9000) values to use for this effect. Overrides the theme attribute."
+ "description": "List of at least 2 and at most 16 colors as hue (0-360), saturation (0-100), brightness (0-100) and Kelvin (1500-9000) values to use for this effect. Overrides the 'Theme' attribute."
},
"theme": {
"name": "[%key:component::lifx::entity::select::theme::name%]",
- "description": "Predefined color theme to use for the effect. Overridden by the palette attribute."
+ "description": "Predefined color theme to use for the effect. Overridden by the 'Palette' attribute."
},
"power_on": {
"name": "Power on",
@@ -223,27 +223,27 @@
},
"effect_sky": {
"name": "Sky effect",
- "description": "Starts the firmware-based Sky effect on LIFX Ceiling.",
+ "description": "Starts a firmware-based effect on LIFX Ceiling lights that animates a sky scene across the device.",
"fields": {
"speed": {
"name": "Speed",
- "description": "How long the Sunrise and Sunset sky types will take to complete. For the Cloud sky type, it is the speed of the clouds across the device."
+ "description": "How long the Sunrise and Sunset sky types will take to complete. For the Clouds sky type, it is the speed of the clouds across the device."
},
"sky_type": {
"name": "Sky type",
"description": "The style of sky that will be animated by the effect."
},
"cloud_saturation_min": {
- "name": "Cloud saturation Minimum",
- "description": "Minimum cloud saturation."
+ "name": "Cloud saturation minimum",
+ "description": "The minimum cloud saturation for the Clouds sky type."
},
"cloud_saturation_max": {
- "name": "Cloud Saturation maximum",
- "description": "Maximum cloud saturation."
+ "name": "Cloud saturation maximum",
+ "description": "The maximum cloud saturation for the Clouds sky type."
},
"palette": {
"name": "Palette",
- "description": "List of 1 to 6 colors as hue (0-360), saturation (0-100), brightness (0-100) and kelvin (1500-9000) values to use for this effect."
+ "description": "List of 1 to 6 colors as hue (0-360), saturation (0-100), brightness (0-100) and Kelvin (1500-9000) values to use for this effect."
},
"power_on": {
"name": "Power on",
@@ -256,16 +256,16 @@
"description": "Stops a running effect."
},
"paint_theme": {
- "name": "Paint Theme",
- "description": "Paint either a provided theme or custom palette across one or more LIFX lights.",
+ "name": "Paint theme",
+ "description": "Paints either a provided theme or custom palette across one or more LIFX lights.",
"fields": {
"palette": {
"name": "Palette",
- "description": "List of at least 2 and at most 16 colors as hue (0-360), saturation (0-100), brightness (0-100) and kelvin (1500-9000) values to paint across the target lights. Overrides the theme attribute."
+ "description": "List of at least 2 and at most 16 colors as hue (0-360), saturation (0-100), brightness (0-100) and Kelvin (1500-9000) values to paint across the target lights. Overrides the 'Theme' attribute."
},
"theme": {
"name": "[%key:component::lifx::entity::select::theme::name%]",
- "description": "Predefined color theme to paint. Overridden by the palette attribute."
+ "description": "Predefined color theme to paint. Overridden by the 'Palette' attribute."
},
"transition": {
"name": "Transition",
diff --git a/homeassistant/components/light/intent.py b/homeassistant/components/light/intent.py
index 83f2ee58b5e..250e1f5b2c1 100644
--- a/homeassistant/components/light/intent.py
+++ b/homeassistant/components/light/intent.py
@@ -28,13 +28,21 @@ async def async_setup_intents(hass: HomeAssistant) -> None:
DOMAIN,
SERVICE_TURN_ON,
optional_slots={
- ("color", ATTR_RGB_COLOR): color_util.color_name_to_rgb,
- ("temperature", ATTR_COLOR_TEMP_KELVIN): cv.positive_int,
- ("brightness", ATTR_BRIGHTNESS_PCT): vol.All(
- vol.Coerce(int), vol.Range(0, 100)
+ "color": intent.IntentSlotInfo(
+ service_data_name=ATTR_RGB_COLOR,
+ value_schema=color_util.color_name_to_rgb,
+ ),
+ "temperature": intent.IntentSlotInfo(
+ service_data_name=ATTR_COLOR_TEMP_KELVIN,
+ value_schema=cv.positive_int,
+ ),
+ "brightness": intent.IntentSlotInfo(
+ service_data_name=ATTR_BRIGHTNESS_PCT,
+ description="The brightness percentage of the light between 0 and 100, where 0 is off and 100 is fully lit",
+ value_schema=vol.All(vol.Coerce(int), vol.Range(0, 100)),
),
},
- description="Sets the brightness or color of a light",
+ description="Sets the brightness percentage or color of a light",
platforms={DOMAIN},
),
)
diff --git a/homeassistant/components/light/strings.json b/homeassistant/components/light/strings.json
index b874e48406e..c0f658c3a44 100644
--- a/homeassistant/components/light/strings.json
+++ b/homeassistant/components/light/strings.json
@@ -285,7 +285,7 @@
"services": {
"turn_on": {
"name": "[%key:common::action::turn_on%]",
- "description": "Turn on one or more lights and adjust properties of the light, even when they are turned on already.",
+ "description": "Turns on one or more lights and adjusts their properties, even when they are turned on already.",
"fields": {
"transition": {
"name": "[%key:component::light::common::field_transition_name%]",
@@ -364,7 +364,7 @@
},
"turn_off": {
"name": "[%key:common::action::turn_off%]",
- "description": "Turn off one or more lights.",
+ "description": "Turns off one or more lights.",
"fields": {
"transition": {
"name": "[%key:component::light::common::field_transition_name%]",
@@ -383,7 +383,7 @@
},
"toggle": {
"name": "[%key:common::action::toggle%]",
- "description": "Toggles one or more lights, from on to off, or, off to on, based on their current state.",
+ "description": "Toggles one or more lights, from on to off, or off to on, based on their current state.",
"fields": {
"transition": {
"name": "[%key:component::light::common::field_transition_name%]",
diff --git a/homeassistant/components/linak/__init__.py b/homeassistant/components/linak/__init__.py
new file mode 100644
index 00000000000..4e3c37807ba
--- /dev/null
+++ b/homeassistant/components/linak/__init__.py
@@ -0,0 +1 @@
+"""LINAK virtual integration."""
diff --git a/homeassistant/components/linak/manifest.json b/homeassistant/components/linak/manifest.json
new file mode 100644
index 00000000000..db1ddd67bda
--- /dev/null
+++ b/homeassistant/components/linak/manifest.json
@@ -0,0 +1,6 @@
+{
+ "domain": "linak",
+ "name": "LINAK",
+ "integration_type": "virtual",
+ "supported_by": "idasen_desk"
+}
diff --git a/homeassistant/components/linear_garage_door/__init__.py b/homeassistant/components/linear_garage_door/__init__.py
index e4aa30c98df..c2a6c6a7ed1 100644
--- a/homeassistant/components/linear_garage_door/__init__.py
+++ b/homeassistant/components/linear_garage_door/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
@@ -31,7 +31,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
},
)
- coordinator = LinearUpdateCoordinator(hass)
+ coordinator = LinearUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
@@ -43,14 +43,16 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
- if all(
- config_entry.state is ConfigEntryState.NOT_LOADED
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.entry_id != entry.entry_id
- ):
- ir.async_delete_issue(hass, DOMAIN, DOMAIN)
-
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
hass.data[DOMAIN].pop(entry.entry_id)
return unload_ok
+
+
+async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Remove a config entry."""
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
+ ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ # Remove any remaining disabled or ignored entries
+ for _entry in hass.config_entries.async_entries(DOMAIN):
+ hass.async_create_task(hass.config_entries.async_remove(_entry.entry_id))
diff --git a/homeassistant/components/linear_garage_door/coordinator.py b/homeassistant/components/linear_garage_door/coordinator.py
index 38b1306ec38..b55affe92e7 100644
--- a/homeassistant/components/linear_garage_door/coordinator.py
+++ b/homeassistant/components/linear_garage_door/coordinator.py
@@ -34,15 +34,16 @@ class LinearUpdateCoordinator(DataUpdateCoordinator[dict[str, LinearDevice]]):
_devices: list[dict[str, Any]] | None = None
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize DataUpdateCoordinator for Linear."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Linear Garage Door",
update_interval=timedelta(seconds=60),
)
- self.site_id = self.config_entry.data["site_id"]
+ self.site_id = config_entry.data["site_id"]
async def _async_update_data(self) -> dict[str, LinearDevice]:
"""Get the data for Linear."""
diff --git a/homeassistant/components/linear_garage_door/cover.py b/homeassistant/components/linear_garage_door/cover.py
index 1f7ae7ce114..7b0510f00d1 100644
--- a/homeassistant/components/linear_garage_door/cover.py
+++ b/homeassistant/components/linear_garage_door/cover.py
@@ -10,7 +10,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LinearUpdateCoordinator
@@ -24,7 +24,7 @@ SCAN_INTERVAL = timedelta(seconds=10)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Linear Garage Door cover."""
coordinator: LinearUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/linear_garage_door/light.py b/homeassistant/components/linear_garage_door/light.py
index 3679491712f..ac03894d446 100644
--- a/homeassistant/components/linear_garage_door/light.py
+++ b/homeassistant/components/linear_garage_door/light.py
@@ -7,7 +7,7 @@ from linear_garage_door import Linear
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LinearUpdateCoordinator
@@ -19,7 +19,7 @@ SUPPORTED_SUBDEVICES = ["Light"]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Linear Garage Door cover."""
coordinator: LinearUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/linkedgo/__init__.py b/homeassistant/components/linkedgo/__init__.py
new file mode 100644
index 00000000000..e26fefa6b96
--- /dev/null
+++ b/homeassistant/components/linkedgo/__init__.py
@@ -0,0 +1 @@
+"""LinkedGo virtual integration."""
diff --git a/homeassistant/components/linkedgo/manifest.json b/homeassistant/components/linkedgo/manifest.json
new file mode 100644
index 00000000000..03c650cac08
--- /dev/null
+++ b/homeassistant/components/linkedgo/manifest.json
@@ -0,0 +1,6 @@
+{
+ "domain": "linkedgo",
+ "name": "LinkedGo",
+ "integration_type": "virtual",
+ "supported_by": "shelly"
+}
diff --git a/homeassistant/components/linkplay/button.py b/homeassistant/components/linkplay/button.py
index 1c93ebcdc3e..8865cf00aa5 100644
--- a/homeassistant/components/linkplay/button.py
+++ b/homeassistant/components/linkplay/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LinkPlayConfigEntry
from .entity import LinkPlayBaseEntity, exception_wrap
@@ -50,7 +50,7 @@ BUTTON_TYPES: tuple[LinkPlayButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: LinkPlayConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the LinkPlay buttons from config entry."""
diff --git a/homeassistant/components/linkplay/manifest.json b/homeassistant/components/linkplay/manifest.json
index ec9a8759a30..0fceed1f691 100644
--- a/homeassistant/components/linkplay/manifest.json
+++ b/homeassistant/components/linkplay/manifest.json
@@ -7,6 +7,6 @@
"integration_type": "hub",
"iot_class": "local_polling",
"loggers": ["linkplay"],
- "requirements": ["python-linkplay==0.1.3"],
+ "requirements": ["python-linkplay==0.2.0"],
"zeroconf": ["_linkplay._tcp.local."]
}
diff --git a/homeassistant/components/linkplay/media_player.py b/homeassistant/components/linkplay/media_player.py
index 456fbf23289..16b0d5f75f1 100644
--- a/homeassistant/components/linkplay/media_player.py
+++ b/homeassistant/components/linkplay/media_player.py
@@ -30,7 +30,7 @@ from homeassistant.helpers import (
entity_platform,
entity_registry as er,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from . import LinkPlayConfigEntry, LinkPlayData
@@ -86,16 +86,10 @@ REPEAT_MAP: dict[LoopMode, RepeatMode] = {
REPEAT_MAP_INV: dict[RepeatMode, LoopMode] = {v: k for k, v in REPEAT_MAP.items()}
-EQUALIZER_MAP: dict[EqualizerMode, str] = {
- EqualizerMode.NONE: "None",
- EqualizerMode.CLASSIC: "Classic",
- EqualizerMode.POP: "Pop",
- EqualizerMode.JAZZ: "Jazz",
- EqualizerMode.VOCAL: "Vocal",
+EQUALIZER_MAP_INV: dict[str, EqualizerMode] = {
+ mode.value: mode for mode in EqualizerMode
}
-EQUALIZER_MAP_INV: dict[str, EqualizerMode] = {v: k for k, v in EQUALIZER_MAP.items()}
-
DEFAULT_FEATURES: MediaPlayerEntityFeature = (
MediaPlayerEntityFeature.PLAY
| MediaPlayerEntityFeature.PLAY_MEDIA
@@ -125,11 +119,13 @@ SERVICE_PLAY_PRESET_SCHEMA = cv.make_entity_service_schema(
}
)
+RETRY_POLL_MAXIMUM = 3
+
async def async_setup_entry(
hass: HomeAssistant,
entry: LinkPlayConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a media player from a config entry."""
@@ -146,7 +142,6 @@ async def async_setup_entry(
class LinkPlayMediaPlayerEntity(LinkPlayBaseEntity, MediaPlayerEntity):
"""Representation of a LinkPlay media player."""
- _attr_sound_mode_list = list(EQUALIZER_MAP.values())
_attr_device_class = MediaPlayerDeviceClass.RECEIVER
_attr_media_content_type = MediaType.MUSIC
_attr_name = None
@@ -156,19 +151,26 @@ class LinkPlayMediaPlayerEntity(LinkPlayBaseEntity, MediaPlayerEntity):
super().__init__(bridge)
self._attr_unique_id = bridge.device.uuid
+ self._retry_count = 0
self._attr_source_list = [
SOURCE_MAP[playing_mode] for playing_mode in bridge.device.playmode_support
]
+ self._attr_sound_mode_list = [
+ mode.value for mode in bridge.player.available_equalizer_modes
+ ]
@exception_wrap
async def async_update(self) -> None:
"""Update the state of the media player."""
try:
await self._bridge.player.update_status()
+ self._retry_count = 0
self._update_properties()
except LinkPlayRequestException:
- self._attr_available = False
+ self._retry_count += 1
+ if self._retry_count >= RETRY_POLL_MAXIMUM:
+ self._attr_available = False
@exception_wrap
async def async_select_source(self, source: str) -> None:
@@ -342,7 +344,7 @@ class LinkPlayMediaPlayerEntity(LinkPlayBaseEntity, MediaPlayerEntity):
self._attr_is_volume_muted = self._bridge.player.muted
self._attr_repeat = REPEAT_MAP[self._bridge.player.loop_mode]
self._attr_shuffle = self._bridge.player.loop_mode == LoopMode.RANDOM_PLAYBACK
- self._attr_sound_mode = EQUALIZER_MAP[self._bridge.player.equalizer_mode]
+ self._attr_sound_mode = self._bridge.player.equalizer_mode.value
self._attr_supported_features = DEFAULT_FEATURES
if self._bridge.player.status == PlayingStatus.PLAYING:
diff --git a/homeassistant/components/linkplay/strings.json b/homeassistant/components/linkplay/strings.json
index 31b4649e131..5d68754879c 100644
--- a/homeassistant/components/linkplay/strings.json
+++ b/homeassistant/components/linkplay/strings.json
@@ -11,7 +11,7 @@
}
},
"discovery_confirm": {
- "description": "Do you want to setup {name}?"
+ "description": "Do you want to set up {name}?"
}
},
"abort": {
@@ -26,11 +26,11 @@
"services": {
"play_preset": {
"name": "Play preset",
- "description": "Play the preset number on the device.",
+ "description": "Plays a preset on the device.",
"fields": {
"preset_number": {
"name": "Preset number",
- "description": "The preset number on the device to play."
+ "description": "The number of the preset to play."
}
}
}
@@ -44,7 +44,7 @@
},
"exceptions": {
"invalid_grouping_entity": {
- "message": "Entity with id {entity_id} can't be added to the LinkPlay multiroom. Is the entity a LinkPlay mediaplayer?"
+ "message": "Entity with ID {entity_id} can't be added to the LinkPlay multiroom. Is the entity a LinkPlay media player?"
}
}
}
diff --git a/homeassistant/components/linkplay/utils.py b/homeassistant/components/linkplay/utils.py
index 00bb691362b..7151ed1537a 100644
--- a/homeassistant/components/linkplay/utils.py
+++ b/homeassistant/components/linkplay/utils.py
@@ -25,10 +25,12 @@ MODELS_ARYLIC_A30: Final[str] = "A30"
MODELS_ARYLIC_A50: Final[str] = "A50"
MODELS_ARYLIC_A50S: Final[str] = "A50+"
MODELS_ARYLIC_UP2STREAM_AMP: Final[str] = "Up2Stream Amp 2.0"
+MODELS_ARYLIC_UP2STREAM_AMP_2P1: Final[str] = "Up2Stream Amp 2.1"
MODELS_ARYLIC_UP2STREAM_AMP_V3: Final[str] = "Up2Stream Amp v3"
MODELS_ARYLIC_UP2STREAM_AMP_V4: Final[str] = "Up2Stream Amp v4"
MODELS_ARYLIC_UP2STREAM_PRO: Final[str] = "Up2Stream Pro v1"
MODELS_ARYLIC_UP2STREAM_PRO_V3: Final[str] = "Up2Stream Pro v3"
+MODELS_ARYLIC_S10P: Final[str] = "Arylic S10+"
MODELS_ARYLIC_UP2STREAM_PLATE_AMP: Final[str] = "Up2Stream Plate Amp"
MODELS_IEAST_AUDIOCAST_M5: Final[str] = "AudioCast M5"
MODELS_WIIM_AMP: Final[str] = "WiiM Amp"
@@ -49,9 +51,10 @@ PROJECTID_LOOKUP: Final[dict[str, tuple[str, str]]] = {
"UP2STREAM_AMP_V3": (MANUFACTURER_ARYLIC, MODELS_ARYLIC_UP2STREAM_AMP_V3),
"UP2STREAM_AMP_V4": (MANUFACTURER_ARYLIC, MODELS_ARYLIC_UP2STREAM_AMP_V4),
"UP2STREAM_PRO_V3": (MANUFACTURER_ARYLIC, MODELS_ARYLIC_UP2STREAM_PRO_V3),
+ "S10P_WIFI": (MANUFACTURER_ARYLIC, MODELS_ARYLIC_S10P),
"ARYLIC_V20": (MANUFACTURER_ARYLIC, MODELS_ARYLIC_UP2STREAM_PLATE_AMP),
"UP2STREAM_MINI_V3": (MANUFACTURER_ARYLIC, MODELS_GENERIC),
- "UP2STREAM_AMP_2P1": (MANUFACTURER_ARYLIC, MODELS_GENERIC),
+ "UP2STREAM_AMP_2P1": (MANUFACTURER_ARYLIC, MODELS_ARYLIC_UP2STREAM_AMP_2P1),
"RP0014_A50C_S": (MANUFACTURER_ARYLIC, MODELS_GENERIC),
"ARYLIC_A30": (MANUFACTURER_ARYLIC, MODELS_GENERIC),
"ARYLIC_SUBWOOFER": (MANUFACTURER_ARYLIC, MODELS_GENERIC),
diff --git a/homeassistant/components/litejet/light.py b/homeassistant/components/litejet/light.py
index f2b9af9adb4..95870927072 100644
--- a/homeassistant/components/litejet/light.py
+++ b/homeassistant/components/litejet/light.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_DEFAULT_TRANSITION, DOMAIN
@@ -27,7 +27,7 @@ ATTR_NUMBER = "number"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
diff --git a/homeassistant/components/litejet/scene.py b/homeassistant/components/litejet/scene.py
index 712e223aa3e..dd96b5accb6 100644
--- a/homeassistant/components/litejet/scene.py
+++ b/homeassistant/components/litejet/scene.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -22,7 +22,7 @@ ATTR_NUMBER = "number"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
diff --git a/homeassistant/components/litejet/switch.py b/homeassistant/components/litejet/switch.py
index 28f751f3ec1..1b46ba360c3 100644
--- a/homeassistant/components/litejet/switch.py
+++ b/homeassistant/components/litejet/switch.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -19,7 +19,7 @@ ATTR_NUMBER = "number"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
diff --git a/homeassistant/components/litterrobot/binary_sensor.py b/homeassistant/components/litterrobot/binary_sensor.py
index 700985d285f..ca9af22f1e9 100644
--- a/homeassistant/components/litterrobot/binary_sensor.py
+++ b/homeassistant/components/litterrobot/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LitterRobotConfigEntry
from .entity import LitterRobotEntity, _WhiskerEntityT
@@ -63,7 +63,7 @@ BINARY_SENSOR_MAP: dict[type[Robot], tuple[RobotBinarySensorEntityDescription, .
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot binary sensors using config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/litterrobot/button.py b/homeassistant/components/litterrobot/button.py
index 758548b3a67..da6ac53ccec 100644
--- a/homeassistant/components/litterrobot/button.py
+++ b/homeassistant/components/litterrobot/button.py
@@ -11,7 +11,7 @@ from pylitterbot import FeederRobot, LitterRobot3, LitterRobot4, Robot
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LitterRobotConfigEntry
from .entity import LitterRobotEntity, _WhiskerEntityT
@@ -48,7 +48,7 @@ ROBOT_BUTTON_MAP: dict[type[Robot], RobotButtonEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot cleaner using config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/litterrobot/select.py b/homeassistant/components/litterrobot/select.py
index f6e3781f3df..be3a9915940 100644
--- a/homeassistant/components/litterrobot/select.py
+++ b/homeassistant/components/litterrobot/select.py
@@ -12,7 +12,7 @@ from pylitterbot.robot.litterrobot4 import BrightnessLevel
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LitterRobotConfigEntry, LitterRobotDataUpdateCoordinator
from .entity import LitterRobotEntity, _WhiskerEntityT
@@ -68,7 +68,7 @@ ROBOT_SELECT_MAP: dict[type[Robot], RobotSelectEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot selects using config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/litterrobot/sensor.py b/homeassistant/components/litterrobot/sensor.py
index 3e25a0556c6..a638f24cf2a 100644
--- a/homeassistant/components/litterrobot/sensor.py
+++ b/homeassistant/components/litterrobot/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfMass
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LitterRobotConfigEntry
from .entity import LitterRobotEntity, _WhiskerEntityT
@@ -160,7 +160,7 @@ PET_SENSORS: list[RobotSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot sensors using config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/litterrobot/strings.json b/homeassistant/components/litterrobot/strings.json
index 19b007de068..052427f3032 100644
--- a/homeassistant/components/litterrobot/strings.json
+++ b/homeassistant/components/litterrobot/strings.json
@@ -77,31 +77,31 @@
"status_code": {
"name": "Status code",
"state": {
- "br": "Bonnet Removed",
- "ccc": "Clean Cycle Complete",
- "ccp": "Clean Cycle In Progress",
- "cd": "Cat Detected",
- "csf": "Cat Sensor Fault",
- "csi": "Cat Sensor Interrupted",
- "cst": "Cat Sensor Timing",
- "df1": "Drawer Almost Full - 2 Cycles Left",
- "df2": "Drawer Almost Full - 1 Cycle Left",
- "dfs": "Drawer Full",
- "dhf": "Dump + Home Position Fault",
- "dpf": "Dump Position Fault",
- "ec": "Empty Cycle",
- "hpf": "Home Position Fault",
+ "br": "Bonnet removed",
+ "ccc": "Clean cycle complete",
+ "ccp": "Clean cycle in progress",
+ "cd": "Cat detected",
+ "csf": "Cat sensor fault",
+ "csi": "Cat sensor interrupted",
+ "cst": "Cat sensor timing",
+ "df1": "Drawer almost full - 2 cycles left",
+ "df2": "Drawer almost full - 1 cycle left",
+ "dfs": "Drawer full",
+ "dhf": "Dump + home position fault",
+ "dpf": "Dump position fault",
+ "ec": "Empty cycle",
+ "hpf": "Home position fault",
"off": "[%key:common::state::off%]",
"offline": "Offline",
- "otf": "Over Torque Fault",
+ "otf": "Over torque fault",
"p": "[%key:common::state::paused%]",
- "pd": "Pinch Detect",
- "pwrd": "Powering Down",
- "pwru": "Powering Up",
+ "pd": "Pinch detect",
+ "pwrd": "Powering down",
+ "pwru": "Powering up",
"rdy": "Ready",
- "scf": "Cat Sensor Fault At Startup",
- "sdf": "Drawer Full At Startup",
- "spf": "Pinch Detect At Startup"
+ "scf": "Cat sensor fault at startup",
+ "sdf": "Drawer full at startup",
+ "spf": "Pinch detect at startup"
}
},
"waste_drawer": {
diff --git a/homeassistant/components/litterrobot/switch.py b/homeassistant/components/litterrobot/switch.py
index 4839748c068..5924f8f094a 100644
--- a/homeassistant/components/litterrobot/switch.py
+++ b/homeassistant/components/litterrobot/switch.py
@@ -11,7 +11,7 @@ from pylitterbot import FeederRobot, LitterRobot
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LitterRobotConfigEntry
from .entity import LitterRobotEntity, _WhiskerEntityT
@@ -45,7 +45,7 @@ ROBOT_SWITCHES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot switches using config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/litterrobot/time.py b/homeassistant/components/litterrobot/time.py
index 69d81d63eae..3573418613b 100644
--- a/homeassistant/components/litterrobot/time.py
+++ b/homeassistant/components/litterrobot/time.py
@@ -12,7 +12,7 @@ from pylitterbot import LitterRobot3
from homeassistant.components.time import TimeEntity, TimeEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .coordinator import LitterRobotConfigEntry
@@ -49,7 +49,7 @@ LITTER_ROBOT_3_SLEEP_START = RobotTimeEntityDescription[LitterRobot3](
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot cleaner using config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/litterrobot/update.py b/homeassistant/components/litterrobot/update.py
index 53ab23e9db8..4d9dfe5074d 100644
--- a/homeassistant/components/litterrobot/update.py
+++ b/homeassistant/components/litterrobot/update.py
@@ -15,7 +15,7 @@ from homeassistant.components.update import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import LitterRobotConfigEntry
from .entity import LitterRobotEntity
@@ -31,7 +31,7 @@ FIRMWARE_UPDATE_ENTITY = UpdateEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot update platform."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/litterrobot/vacuum.py b/homeassistant/components/litterrobot/vacuum.py
index 314fab6a621..9989c306b51 100644
--- a/homeassistant/components/litterrobot/vacuum.py
+++ b/homeassistant/components/litterrobot/vacuum.py
@@ -17,7 +17,7 @@ from homeassistant.components.vacuum import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .coordinator import LitterRobotConfigEntry
@@ -46,7 +46,7 @@ LITTER_BOX_ENTITY = StateVacuumEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: LitterRobotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Litter-Robot cleaner using config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/livisi/binary_sensor.py b/homeassistant/components/livisi/binary_sensor.py
index d4edd59f2d7..50eb4cd28b9 100644
--- a/homeassistant/components/livisi/binary_sensor.py
+++ b/homeassistant/components/livisi/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, LIVISI_STATE_CHANGE, LOGGER, WDS_DEVICE_TYPE
from .coordinator import LivisiDataUpdateCoordinator
@@ -21,7 +21,7 @@ from .entity import LivisiEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary_sensor device."""
coordinator: LivisiDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/livisi/climate.py b/homeassistant/components/livisi/climate.py
index 3ecdcb486c0..1f5e3360c7d 100644
--- a/homeassistant/components/livisi/climate.py
+++ b/homeassistant/components/livisi/climate.py
@@ -16,7 +16,7 @@ from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
@@ -33,7 +33,7 @@ from .entity import LivisiEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate device."""
coordinator: LivisiDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/livisi/coordinator.py b/homeassistant/components/livisi/coordinator.py
index b8b282c2829..6557416ed3a 100644
--- a/homeassistant/components/livisi/coordinator.py
+++ b/homeassistant/components/livisi/coordinator.py
@@ -39,10 +39,10 @@ class LivisiDataUpdateCoordinator(DataUpdateCoordinator[list[dict[str, Any]]]):
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Livisi devices",
update_interval=timedelta(seconds=DEVICE_POLLING_DELAY),
)
- self.config_entry = config_entry
self.hass = hass
self.aiolivisi = aiolivisi
self.websocket = Websocket(aiolivisi)
diff --git a/homeassistant/components/livisi/switch.py b/homeassistant/components/livisi/switch.py
index fa604c5fc87..5599a4af0d4 100644
--- a/homeassistant/components/livisi/switch.py
+++ b/homeassistant/components/livisi/switch.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, LIVISI_STATE_CHANGE, LOGGER, SWITCH_DEVICE_TYPES
from .coordinator import LivisiDataUpdateCoordinator
@@ -19,7 +19,7 @@ from .entity import LivisiEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch device."""
coordinator: LivisiDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/local_calendar/calendar.py b/homeassistant/components/local_calendar/calendar.py
index eb7b0c20d91..df6f994a46c 100644
--- a/homeassistant/components/local_calendar/calendar.py
+++ b/homeassistant/components/local_calendar/calendar.py
@@ -26,7 +26,7 @@ from homeassistant.components.calendar import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import CONF_CALENDAR_NAME, DOMAIN
@@ -40,7 +40,7 @@ PRODID = "-//homeassistant.io//local_calendar 1.0//EN"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the local calendar platform."""
store = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/local_calendar/manifest.json b/homeassistant/components/local_calendar/manifest.json
index 21a4134a8b6..fc6d0bc00c7 100644
--- a/homeassistant/components/local_calendar/manifest.json
+++ b/homeassistant/components/local_calendar/manifest.json
@@ -7,5 +7,5 @@
"documentation": "https://www.home-assistant.io/integrations/local_calendar",
"iot_class": "local_polling",
"loggers": ["ical"],
- "requirements": ["ical==8.3.0"]
+ "requirements": ["ical==9.0.1"]
}
diff --git a/homeassistant/components/local_file/camera.py b/homeassistant/components/local_file/camera.py
index db421bbce1d..8be0389678d 100644
--- a/homeassistant/components/local_file/camera.py
+++ b/homeassistant/components/local_file/camera.py
@@ -20,7 +20,10 @@ from homeassistant.helpers import (
entity_platform,
issue_registry as ir,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import slugify
@@ -40,7 +43,7 @@ PLATFORM_SCHEMA = CAMERA_PLATFORM_SCHEMA.extend(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Camera for local file from a config entry."""
diff --git a/homeassistant/components/local_ip/sensor.py b/homeassistant/components/local_ip/sensor.py
index 7f855220563..a4cb9f2d60e 100644
--- a/homeassistant/components/local_ip/sensor.py
+++ b/homeassistant/components/local_ip/sensor.py
@@ -5,7 +5,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SENSOR
@@ -13,7 +13,7 @@ from .const import SENSOR
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from config_entry."""
name = entry.data.get(CONF_NAME) or "Local IP"
diff --git a/homeassistant/components/local_todo/manifest.json b/homeassistant/components/local_todo/manifest.json
index 68154f10885..27d3ccce4a7 100644
--- a/homeassistant/components/local_todo/manifest.json
+++ b/homeassistant/components/local_todo/manifest.json
@@ -5,5 +5,5 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/local_todo",
"iot_class": "local_polling",
- "requirements": ["ical==8.3.0"]
+ "requirements": ["ical==9.0.1"]
}
diff --git a/homeassistant/components/local_todo/strings.json b/homeassistant/components/local_todo/strings.json
index 2403fae60a5..ebf7810494c 100644
--- a/homeassistant/components/local_todo/strings.json
+++ b/homeassistant/components/local_todo/strings.json
@@ -6,7 +6,8 @@
"description": "Please choose a name for your new To-do list",
"data": {
"todo_list_name": "To-do list name"
- }
+ },
+ "submit": "Create"
}
},
"abort": {
diff --git a/homeassistant/components/local_todo/todo.py b/homeassistant/components/local_todo/todo.py
index c496fd6b6ba..30df24ea854 100644
--- a/homeassistant/components/local_todo/todo.py
+++ b/homeassistant/components/local_todo/todo.py
@@ -17,7 +17,7 @@ from homeassistant.components.todo import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.setup import SetupPhases, async_pause_setup
from homeassistant.util import dt as dt_util
@@ -65,7 +65,7 @@ def _migrate_calendar(calendar: Calendar) -> bool:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: LocalTodoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the local_todo todo platform."""
diff --git a/homeassistant/components/locative/device_tracker.py b/homeassistant/components/locative/device_tracker.py
index 47a498331eb..f7ae9039729 100644
--- a/homeassistant/components/locative/device_tracker.py
+++ b/homeassistant/components/locative/device_tracker.py
@@ -4,13 +4,15 @@ from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as LT_DOMAIN, TRACKER_UPDATE
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure a dispatcher connection based on a config entry."""
diff --git a/homeassistant/components/logbook/strings.json b/homeassistant/components/logbook/strings.json
index 27ad49b0e3a..5a38b57a9b7 100644
--- a/homeassistant/components/logbook/strings.json
+++ b/homeassistant/components/logbook/strings.json
@@ -7,7 +7,7 @@
"fields": {
"name": {
"name": "[%key:common::config_flow::data::name%]",
- "description": "Custom name for an entity, can be referenced using an `entity_id`."
+ "description": "Custom name for an entity, can be referenced using the 'Entity ID' field."
},
"message": {
"name": "Message",
diff --git a/homeassistant/components/logger/helpers.py b/homeassistant/components/logger/helpers.py
index 034266428a3..00cea7e8aa5 100644
--- a/homeassistant/components/logger/helpers.py
+++ b/homeassistant/components/logger/helpers.py
@@ -203,7 +203,7 @@ class LoggerSettings:
else:
loggers = {domain}
- combined_logs = {logger: LOGSEVERITY[settings.level] for logger in loggers}
+ combined_logs = dict.fromkeys(loggers, LOGSEVERITY[settings.level])
# Don't override the log levels with the ones from YAML
# since we want whatever the user is asking for to be honored.
diff --git a/homeassistant/components/london_air/sensor.py b/homeassistant/components/london_air/sensor.py
index 81133433d05..a4d34fcb2d6 100644
--- a/homeassistant/components/london_air/sensor.py
+++ b/homeassistant/components/london_air/sensor.py
@@ -37,10 +37,12 @@ AUTHORITIES = [
"Enfield",
"Greenwich",
"Hackney",
+ "Hammersmith and Fulham",
"Haringey",
"Harrow",
"Havering",
"Hillingdon",
+ "Hounslow",
"Islington",
"Kensington and Chelsea",
"Kingston",
diff --git a/homeassistant/components/lookin/__init__.py b/homeassistant/components/lookin/__init__.py
index a0e529bc189..247282309e4 100644
--- a/homeassistant/components/lookin/__init__.py
+++ b/homeassistant/components/lookin/__init__.py
@@ -19,7 +19,7 @@ from aiolookin import (
)
from aiolookin.models import UDPCommandType, UDPEvent
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryNotReady
@@ -122,6 +122,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
coordinator_class = LookinDataUpdateCoordinator[MeteoSensor]
meteo_coordinator = coordinator_class(
hass,
+ entry,
push_coordinator,
name=entry.title,
update_method=lookin_protocol.get_meteo_sensor,
@@ -140,6 +141,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
updater = _async_remote_updater(lookin_protocol, uuid)
coordinator = LookinDataUpdateCoordinator(
hass,
+ entry,
push_coordinator,
name=f"{entry.title} {uuid}",
update_method=updater,
@@ -190,12 +192,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
hass.data[DOMAIN].pop(entry.entry_id)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
manager: LookinUDPManager = hass.data[DOMAIN][UDP_MANAGER]
await manager.async_stop()
return unload_ok
diff --git a/homeassistant/components/lookin/climate.py b/homeassistant/components/lookin/climate.py
index 051a18c9a32..9cef56bcf9f 100644
--- a/homeassistant/components/lookin/climate.py
+++ b/homeassistant/components/lookin/climate.py
@@ -28,7 +28,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TYPE_TO_PLATFORM
from .coordinator import LookinDataUpdateCoordinator
@@ -65,7 +65,7 @@ LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the climate platform for lookin from a config entry."""
lookin_data: LookinData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/lookin/coordinator.py b/homeassistant/components/lookin/coordinator.py
index d9834bd1d94..a74cd0e4861 100644
--- a/homeassistant/components/lookin/coordinator.py
+++ b/homeassistant/components/lookin/coordinator.py
@@ -7,6 +7,7 @@ from datetime import timedelta
import logging
import time
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -43,9 +44,12 @@ class LookinPushCoordinator:
class LookinDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
"""DataUpdateCoordinator to gather data for a specific lookin devices."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
push_coordinator: LookinPushCoordinator,
name: str,
update_interval: timedelta | None = None,
@@ -56,6 +60,7 @@ class LookinDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=name,
update_interval=update_interval,
update_method=update_method,
diff --git a/homeassistant/components/lookin/light.py b/homeassistant/components/lookin/light.py
index 804d0ebef01..d46cb96d6c0 100644
--- a/homeassistant/components/lookin/light.py
+++ b/homeassistant/components/lookin/light.py
@@ -9,7 +9,7 @@ from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TYPE_TO_PLATFORM
from .entity import LookinPowerPushRemoteEntity
@@ -21,7 +21,7 @@ LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the light platform for lookin from a config entry."""
lookin_data: LookinData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/lookin/media_player.py b/homeassistant/components/lookin/media_player.py
index b3dda9c9e0c..a3568d9f155 100644
--- a/homeassistant/components/lookin/media_player.py
+++ b/homeassistant/components/lookin/media_player.py
@@ -15,7 +15,7 @@ from homeassistant.components.media_player import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TYPE_TO_PLATFORM
from .coordinator import LookinDataUpdateCoordinator
@@ -44,7 +44,7 @@ _FUNCTION_NAME_TO_FEATURE = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the media_player platform for lookin from a config entry."""
lookin_data: LookinData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/lookin/sensor.py b/homeassistant/components/lookin/sensor.py
index cae4f7782a8..89e1ed6aa69 100644
--- a/homeassistant/components/lookin/sensor.py
+++ b/homeassistant/components/lookin/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import LookinDeviceCoordinatorEntity
@@ -43,7 +43,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up lookin sensors from the config entry."""
lookin_data: LookinData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/loqed/__init__.py b/homeassistant/components/loqed/__init__.py
index b6408880c96..b308e2c0f1d 100644
--- a/homeassistant/components/loqed/__init__.py
+++ b/homeassistant/components/loqed/__init__.py
@@ -44,7 +44,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
aiohttp.ClientError,
) as ex:
raise ConfigEntryNotReady(f"Unable to connect to bridge at {host}") from ex
- coordinator = LoqedDataCoordinator(hass, api, lock, entry)
+ coordinator = LoqedDataCoordinator(hass, entry, api, lock)
await coordinator.ensure_webhooks()
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/loqed/coordinator.py b/homeassistant/components/loqed/coordinator.py
index 1447934103e..7b60385a759 100644
--- a/homeassistant/components/loqed/coordinator.py
+++ b/homeassistant/components/loqed/coordinator.py
@@ -71,19 +71,20 @@ class StatusMessage(TypedDict):
class LoqedDataCoordinator(DataUpdateCoordinator[StatusMessage]):
"""Data update coordinator for the loqed platform."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
api: loqed.LoqedAPI,
lock: loqed.Lock,
- entry: ConfigEntry,
) -> None:
"""Initialize the Loqed Data Update coordinator."""
- super().__init__(hass, _LOGGER, name="Loqed sensors")
+ super().__init__(hass, _LOGGER, config_entry=config_entry, name="Loqed sensors")
self._api = api
- self._entry = entry
self.lock = lock
- self.device_name = self._entry.data[CONF_NAME]
+ self.device_name = config_entry.data[CONF_NAME]
async def _async_update_data(self) -> StatusMessage:
"""Fetch data from API endpoint."""
@@ -110,17 +111,19 @@ class LoqedDataCoordinator(DataUpdateCoordinator[StatusMessage]):
async def ensure_webhooks(self) -> None:
"""Register webhook on LOQED bridge."""
- webhook_id = self._entry.data[CONF_WEBHOOK_ID]
+ webhook_id = self.config_entry.data[CONF_WEBHOOK_ID]
webhook.async_register(
self.hass, DOMAIN, "Loqed", webhook_id, self._handle_webhook
)
if cloud.async_active_subscription(self.hass):
- webhook_url = await async_cloudhook_generate_url(self.hass, self._entry)
+ webhook_url = await async_cloudhook_generate_url(
+ self.hass, self.config_entry
+ )
else:
webhook_url = webhook.async_generate_url(
- self.hass, self._entry.data[CONF_WEBHOOK_ID]
+ self.hass, self.config_entry.data[CONF_WEBHOOK_ID]
)
_LOGGER.debug("Webhook URL: %s", webhook_url)
@@ -140,10 +143,10 @@ class LoqedDataCoordinator(DataUpdateCoordinator[StatusMessage]):
async def remove_webhooks(self) -> None:
"""Remove webhook from LOQED bridge."""
- webhook_id = self._entry.data[CONF_WEBHOOK_ID]
+ webhook_id = self.config_entry.data[CONF_WEBHOOK_ID]
- if CONF_CLOUDHOOK_URL in self._entry.data:
- webhook_url = self._entry.data[CONF_CLOUDHOOK_URL]
+ if CONF_CLOUDHOOK_URL in self.config_entry.data:
+ webhook_url = self.config_entry.data[CONF_CLOUDHOOK_URL]
else:
webhook_url = webhook.async_generate_url(self.hass, webhook_id)
diff --git a/homeassistant/components/loqed/lock.py b/homeassistant/components/loqed/lock.py
index be6b39176d6..2064537df52 100644
--- a/homeassistant/components/loqed/lock.py
+++ b/homeassistant/components/loqed/lock.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.lock import LockEntity, LockEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LoqedDataCoordinator
from .const import DOMAIN
@@ -20,7 +20,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Loqed lock platform."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/loqed/sensor.py b/homeassistant/components/loqed/sensor.py
index 1d4595db8e9..c28b55b4f98 100644
--- a/homeassistant/components/loqed/sensor.py
+++ b/homeassistant/components/loqed/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.const import (
EntityCategory,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import LoqedDataCoordinator, StatusMessage
@@ -42,7 +42,9 @@ SENSORS: Final[tuple[SensorEntityDescription, ...]] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Loqed lock platform."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/luftdaten/sensor.py b/homeassistant/components/luftdaten/sensor.py
index 8b9def63fda..2189386a4bb 100644
--- a/homeassistant/components/luftdaten/sensor.py
+++ b/homeassistant/components/luftdaten/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -72,7 +72,9 @@ SENSORS: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Sensor.Community sensor based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lupusec/alarm_control_panel.py b/homeassistant/components/lupusec/alarm_control_panel.py
index 4b3d12ad743..03feabae0dc 100644
--- a/homeassistant/components/lupusec/alarm_control_panel.py
+++ b/homeassistant/components/lupusec/alarm_control_panel.py
@@ -14,7 +14,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
from .entity import LupusecDevice
@@ -25,7 +25,7 @@ SCAN_INTERVAL = timedelta(seconds=2)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an alarm control panel for a Lupusec device."""
data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/lupusec/binary_sensor.py b/homeassistant/components/lupusec/binary_sensor.py
index b2413e2b462..bcd21adc1aa 100644
--- a/homeassistant/components/lupusec/binary_sensor.py
+++ b/homeassistant/components/lupusec/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
from .entity import LupusecBaseSensor
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a binary sensors for a Lupusec device."""
diff --git a/homeassistant/components/lupusec/entity.py b/homeassistant/components/lupusec/entity.py
index dc0dac89dc8..8cfb559b84f 100644
--- a/homeassistant/components/lupusec/entity.py
+++ b/homeassistant/components/lupusec/entity.py
@@ -18,7 +18,7 @@ class LupusecDevice(Entity):
self._device = device
self._attr_unique_id = device.device_id
- def update(self):
+ def update(self) -> None:
"""Update automation state."""
self._device.refresh()
diff --git a/homeassistant/components/lupusec/switch.py b/homeassistant/components/lupusec/switch.py
index 23f3c927880..a70df90f8e7 100644
--- a/homeassistant/components/lupusec/switch.py
+++ b/homeassistant/components/lupusec/switch.py
@@ -11,7 +11,7 @@ import lupupy.constants as CONST
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
from .entity import LupusecBaseSensor
@@ -22,7 +22,7 @@ SCAN_INTERVAL = timedelta(seconds=2)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lupusec switch devices."""
diff --git a/homeassistant/components/lutron/binary_sensor.py b/homeassistant/components/lutron/binary_sensor.py
index c33b545413d..5bed760e1ac 100644
--- a/homeassistant/components/lutron/binary_sensor.py
+++ b/homeassistant/components/lutron/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, LutronData
from .entity import LutronDevice
@@ -25,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron binary_sensor platform.
diff --git a/homeassistant/components/lutron/config_flow.py b/homeassistant/components/lutron/config_flow.py
index 6a48e0d4b67..3f55a2b131b 100644
--- a/homeassistant/components/lutron/config_flow.py
+++ b/homeassistant/components/lutron/config_flow.py
@@ -9,10 +9,21 @@ from urllib.error import HTTPError
from pylutron import Lutron
import voluptuous as vol
-from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.config_entries import (
+ ConfigEntry,
+ ConfigFlow,
+ ConfigFlowResult,
+ OptionsFlow,
+)
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
+from homeassistant.core import callback
+from homeassistant.helpers.selector import (
+ NumberSelector,
+ NumberSelectorConfig,
+ NumberSelectorMode,
+)
-from .const import DOMAIN
+from .const import CONF_DEFAULT_DIMMER_LEVEL, DEFAULT_DIMMER_LEVEL, DOMAIN
_LOGGER = logging.getLogger(__name__)
@@ -68,3 +79,36 @@ class LutronConfigFlow(ConfigFlow, domain=DOMAIN):
),
errors=errors,
)
+
+ @staticmethod
+ @callback
+ def async_get_options_flow(
+ config_entry: ConfigEntry,
+ ) -> OptionsFlowHandler:
+ """Get the options flow for this handler."""
+ return OptionsFlowHandler()
+
+
+class OptionsFlowHandler(OptionsFlow):
+ """Handle option flow for lutron."""
+
+ async def async_step_init(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle options flow."""
+ if user_input is not None:
+ return self.async_create_entry(title="", data=user_input)
+
+ data_schema = vol.Schema(
+ {
+ vol.Required(
+ CONF_DEFAULT_DIMMER_LEVEL,
+ default=self.config_entry.options.get(
+ CONF_DEFAULT_DIMMER_LEVEL, DEFAULT_DIMMER_LEVEL
+ ),
+ ): NumberSelector(
+ NumberSelectorConfig(min=1, max=255, mode=NumberSelectorMode.SLIDER)
+ )
+ }
+ )
+ return self.async_show_form(step_id="init", data_schema=data_schema)
diff --git a/homeassistant/components/lutron/const.py b/homeassistant/components/lutron/const.py
index 3862f7eb1d8..b69e35f38ba 100644
--- a/homeassistant/components/lutron/const.py
+++ b/homeassistant/components/lutron/const.py
@@ -1,3 +1,7 @@
"""Lutron constants."""
DOMAIN = "lutron"
+
+CONF_DEFAULT_DIMMER_LEVEL = "default_dimmer_level"
+
+DEFAULT_DIMMER_LEVEL = 255 / 2
diff --git a/homeassistant/components/lutron/cover.py b/homeassistant/components/lutron/cover.py
index 2f80798aee4..e8f3ad09879 100644
--- a/homeassistant/components/lutron/cover.py
+++ b/homeassistant/components/lutron/cover.py
@@ -15,7 +15,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, LutronData
from .entity import LutronDevice
@@ -26,7 +26,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron cover platform.
diff --git a/homeassistant/components/lutron/event.py b/homeassistant/components/lutron/event.py
index 7b1b9e65137..942e165b97f 100644
--- a/homeassistant/components/lutron/event.py
+++ b/homeassistant/components/lutron/event.py
@@ -8,7 +8,7 @@ from homeassistant.components.event import EventEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_ID
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import slugify
from . import ATTR_ACTION, ATTR_FULL_ID, ATTR_UUID, DOMAIN, LutronData
@@ -33,7 +33,7 @@ LEGACY_EVENT_TYPES: dict[LutronEventType, str] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron event platform."""
entry_data: LutronData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/lutron/fan.py b/homeassistant/components/lutron/fan.py
index 7db8b12c8d0..5928c3c2da3 100644
--- a/homeassistant/components/lutron/fan.py
+++ b/homeassistant/components/lutron/fan.py
@@ -10,7 +10,7 @@ from pylutron import Output
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, LutronData
from .entity import LutronDevice
@@ -21,7 +21,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron fan platform.
diff --git a/homeassistant/components/lutron/light.py b/homeassistant/components/lutron/light.py
index 7e8829b231c..a7489e13b7b 100644
--- a/homeassistant/components/lutron/light.py
+++ b/homeassistant/components/lutron/light.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from collections.abc import Mapping
from typing import Any
-from pylutron import Output
+from pylutron import Lutron, LutronEntity, Output
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
@@ -17,16 +17,17 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, LutronData
+from .const import CONF_DEFAULT_DIMMER_LEVEL, DEFAULT_DIMMER_LEVEL
from .entity import LutronDevice
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron light platform.
@@ -37,7 +38,7 @@ async def async_setup_entry(
async_add_entities(
(
- LutronLight(area_name, device, entry_data.client)
+ LutronLight(area_name, device, entry_data.client, config_entry)
for area_name, device in entry_data.lights
),
True,
@@ -64,6 +65,17 @@ class LutronLight(LutronDevice, LightEntity):
_prev_brightness: int | None = None
_attr_name = None
+ def __init__(
+ self,
+ area_name: str,
+ lutron_device: LutronEntity,
+ controller: Lutron,
+ config_entry: ConfigEntry,
+ ) -> None:
+ """Initialize the device."""
+ super().__init__(area_name, lutron_device, controller)
+ self._config_entry = config_entry
+
def turn_on(self, **kwargs: Any) -> None:
"""Turn the light on."""
if flash := kwargs.get(ATTR_FLASH):
@@ -72,7 +84,9 @@ class LutronLight(LutronDevice, LightEntity):
if ATTR_BRIGHTNESS in kwargs and self._lutron_device.is_dimmable:
brightness = kwargs[ATTR_BRIGHTNESS]
elif self._prev_brightness == 0:
- brightness = 255 / 2
+ brightness = self._config_entry.options.get(
+ CONF_DEFAULT_DIMMER_LEVEL, DEFAULT_DIMMER_LEVEL
+ )
else:
brightness = self._prev_brightness
self._prev_brightness = brightness
diff --git a/homeassistant/components/lutron/scene.py b/homeassistant/components/lutron/scene.py
index 9e8070713a9..4889f9056ac 100644
--- a/homeassistant/components/lutron/scene.py
+++ b/homeassistant/components/lutron/scene.py
@@ -9,7 +9,7 @@ from pylutron import Button, Keypad, Lutron
from homeassistant.components.scene import Scene
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, LutronData
from .entity import LutronKeypad
@@ -18,7 +18,7 @@ from .entity import LutronKeypad
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron scene platform.
diff --git a/homeassistant/components/lutron/strings.json b/homeassistant/components/lutron/strings.json
index b73e0bd15ed..37db509e294 100644
--- a/homeassistant/components/lutron/strings.json
+++ b/homeassistant/components/lutron/strings.json
@@ -19,6 +19,15 @@
}
}
},
+ "options": {
+ "step": {
+ "init": {
+ "data": {
+ "default_dimmer_level": "Default light level when first turning on a light from Home Assistant"
+ }
+ }
+ }
+ },
"entity": {
"event": {
"button": {
diff --git a/homeassistant/components/lutron/switch.py b/homeassistant/components/lutron/switch.py
index c8b93dd7398..e1e97d1774a 100644
--- a/homeassistant/components/lutron/switch.py
+++ b/homeassistant/components/lutron/switch.py
@@ -10,7 +10,7 @@ from pylutron import Button, Keypad, Led, Lutron, Output
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, LutronData
from .entity import LutronDevice, LutronKeypad
@@ -19,7 +19,7 @@ from .entity import LutronDevice, LutronKeypad
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron switch platform.
diff --git a/homeassistant/components/lutron_caseta/__init__.py b/homeassistant/components/lutron_caseta/__init__.py
index d697d6244b5..b489fe9dba7 100644
--- a/homeassistant/components/lutron_caseta/__init__.py
+++ b/homeassistant/components/lutron_caseta/__init__.py
@@ -3,7 +3,6 @@
from __future__ import annotations
import asyncio
-import contextlib
from itertools import chain
import logging
import ssl
@@ -37,11 +36,12 @@ from .const import (
ATTR_SERIAL,
ATTR_TYPE,
BRIDGE_DEVICE_ID,
- BRIDGE_TIMEOUT,
CONF_CA_CERTS,
CONF_CERTFILE,
CONF_KEYFILE,
CONF_SUBTYPE,
+ CONFIGURE_TIMEOUT,
+ CONNECT_TIMEOUT,
DOMAIN,
LUTRON_CASETA_BUTTON_EVENT,
MANUFACTURER,
@@ -161,28 +161,40 @@ async def async_setup_entry(
keyfile = hass.config.path(entry.data[CONF_KEYFILE])
certfile = hass.config.path(entry.data[CONF_CERTFILE])
ca_certs = hass.config.path(entry.data[CONF_CA_CERTS])
- bridge = None
+ connected_future: asyncio.Future[None] = hass.loop.create_future()
+
+ def _on_connect() -> None:
+ nonlocal connected_future
+ if not connected_future.done():
+ connected_future.set_result(None)
try:
bridge = Smartbridge.create_tls(
- hostname=host, keyfile=keyfile, certfile=certfile, ca_certs=ca_certs
+ hostname=host,
+ keyfile=keyfile,
+ certfile=certfile,
+ ca_certs=ca_certs,
+ on_connect_callback=_on_connect,
)
except ssl.SSLError:
_LOGGER.error("Invalid certificate used to connect to bridge at %s", host)
return False
- timed_out = True
- with contextlib.suppress(TimeoutError):
- async with asyncio.timeout(BRIDGE_TIMEOUT):
- await bridge.connect()
- timed_out = False
+ connect_task = hass.async_create_task(bridge.connect())
+ for future, name, timeout in (
+ (connected_future, "connect", CONNECT_TIMEOUT),
+ (connect_task, "configure", CONFIGURE_TIMEOUT),
+ ):
+ try:
+ async with asyncio.timeout(timeout):
+ await future
+ except TimeoutError as ex:
+ connect_task.cancel()
+ await bridge.close()
+ raise ConfigEntryNotReady(f"Timed out on {name} for {host}") from ex
- if timed_out or not bridge.is_connected():
- await bridge.close()
- if timed_out:
- raise ConfigEntryNotReady(f"Timed out while trying to connect to {host}")
- if not bridge.is_connected():
- raise ConfigEntryNotReady(f"Cannot connect to {host}")
+ if not bridge.is_connected():
+ raise ConfigEntryNotReady(f"Connection failed to {host}")
_LOGGER.debug("Connected to Lutron Caseta bridge via LEAP at %s", host)
await _async_migrate_unique_ids(hass, entry)
diff --git a/homeassistant/components/lutron_caseta/binary_sensor.py b/homeassistant/components/lutron_caseta/binary_sensor.py
index b51756692c1..cb0f0da5227 100644
--- a/homeassistant/components/lutron_caseta/binary_sensor.py
+++ b/homeassistant/components/lutron_caseta/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import ATTR_SUGGESTED_AREA
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as CASETA_DOMAIN
from .const import CONFIG_URL, MANUFACTURER, UNASSIGNED_AREA
@@ -21,7 +21,7 @@ from .util import area_name_from_id
async def async_setup_entry(
hass: HomeAssistant,
config_entry: LutronCasetaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron Caseta binary_sensor platform.
diff --git a/homeassistant/components/lutron_caseta/button.py b/homeassistant/components/lutron_caseta/button.py
index e56758b0af6..f2da502d346 100644
--- a/homeassistant/components/lutron_caseta/button.py
+++ b/homeassistant/components/lutron_caseta/button.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.button import ButtonEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .device_trigger import LEAP_TO_DEVICE_TYPE_SUBTYPE_MAP
from .entity import LutronCasetaEntity
@@ -17,7 +17,7 @@ from .models import LutronCasetaConfigEntry, LutronCasetaData
async def async_setup_entry(
hass: HomeAssistant,
config_entry: LutronCasetaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Lutron pico and keypad buttons."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/lutron_caseta/config_flow.py b/homeassistant/components/lutron_caseta/config_flow.py
index 767c3d2f2b7..45e7a04bdc9 100644
--- a/homeassistant/components/lutron_caseta/config_flow.py
+++ b/homeassistant/components/lutron_caseta/config_flow.py
@@ -20,10 +20,11 @@ from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from .const import (
ABORT_REASON_CANNOT_CONNECT,
BRIDGE_DEVICE_ID,
- BRIDGE_TIMEOUT,
CONF_CA_CERTS,
CONF_CERTFILE,
CONF_KEYFILE,
+ CONFIGURE_TIMEOUT,
+ CONNECT_TIMEOUT,
DOMAIN,
ERROR_CANNOT_CONNECT,
STEP_IMPORT_FAILED,
@@ -232,7 +233,7 @@ class LutronCasetaFlowHandler(ConfigFlow, domain=DOMAIN):
return None
try:
- async with asyncio.timeout(BRIDGE_TIMEOUT):
+ async with asyncio.timeout(CONNECT_TIMEOUT + CONFIGURE_TIMEOUT):
await bridge.connect()
except TimeoutError:
_LOGGER.error(
diff --git a/homeassistant/components/lutron_caseta/const.py b/homeassistant/components/lutron_caseta/const.py
index 809b9e8d007..26a83de6f4b 100644
--- a/homeassistant/components/lutron_caseta/const.py
+++ b/homeassistant/components/lutron_caseta/const.py
@@ -34,7 +34,8 @@ ACTION_RELEASE = "release"
CONF_SUBTYPE = "subtype"
-BRIDGE_TIMEOUT = 35
+CONNECT_TIMEOUT = 9
+CONFIGURE_TIMEOUT = 50
UNASSIGNED_AREA = "Unassigned"
diff --git a/homeassistant/components/lutron_caseta/cover.py b/homeassistant/components/lutron_caseta/cover.py
index d8fac38ce2b..e05fddb996f 100644
--- a/homeassistant/components/lutron_caseta/cover.py
+++ b/homeassistant/components/lutron_caseta/cover.py
@@ -11,7 +11,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import LutronCasetaUpdatableEntity
from .models import LutronCasetaConfigEntry
@@ -108,13 +108,14 @@ PYLUTRON_TYPE_TO_CLASSES = {
"QsWirelessHorizontalSheerBlind": LutronCasetaShade,
"Shade": LutronCasetaShade,
"PalladiomWireFreeShade": LutronCasetaShade,
+ "SerenaEssentialsRollerShade": LutronCasetaShade,
}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: LutronCasetaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron Caseta cover platform.
diff --git a/homeassistant/components/lutron_caseta/device_trigger.py b/homeassistant/components/lutron_caseta/device_trigger.py
index 0b432f88045..31c9a0e171d 100644
--- a/homeassistant/components/lutron_caseta/device_trigger.py
+++ b/homeassistant/components/lutron_caseta/device_trigger.py
@@ -277,6 +277,21 @@ FOUR_GROUP_REMOTE_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
}
)
+# See mappings at https://github.com/home-assistant/core/issues/137548#issuecomment-2643440119
+PADDLE_SWITCH_PICO_BUTTON_TYPES_TO_LIP = {
+ "on": 2, # 'Number': 2 in LIP
+ "off": 4, # 'Number': 4 in LIP
+}
+PADDLE_SWITCH_PICO_BUTTON_TYPES_TO_LEAP = {
+ "on": 0, # 'ButtonNumber': 0 in LEAP
+ "off": 2, # 'ButtonNumber': 2 in LEAP
+}
+PADDLE_SWITCH_PICO_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
+ {
+ vol.Required(CONF_SUBTYPE): vol.In(PADDLE_SWITCH_PICO_BUTTON_TYPES_TO_LIP),
+ }
+)
+
DEVICE_TYPE_SCHEMA_MAP = {
"Pico2Button": PICO_2_BUTTON_TRIGGER_SCHEMA,
@@ -288,6 +303,7 @@ DEVICE_TYPE_SCHEMA_MAP = {
"Pico4ButtonZone": PICO_4_BUTTON_ZONE_TRIGGER_SCHEMA,
"Pico4Button2Group": PICO_4_BUTTON_2_GROUP_TRIGGER_SCHEMA,
"FourGroupRemote": FOUR_GROUP_REMOTE_TRIGGER_SCHEMA,
+ "PaddleSwitchPico": PADDLE_SWITCH_PICO_TRIGGER_SCHEMA,
}
DEVICE_TYPE_SUBTYPE_MAP_TO_LIP = {
@@ -300,6 +316,7 @@ DEVICE_TYPE_SUBTYPE_MAP_TO_LIP = {
"Pico4ButtonZone": PICO_4_BUTTON_ZONE_BUTTON_TYPES_TO_LIP,
"Pico4Button2Group": PICO_4_BUTTON_2_GROUP_BUTTON_TYPES_TO_LIP,
"FourGroupRemote": FOUR_GROUP_REMOTE_BUTTON_TYPES_TO_LIP,
+ "PaddleSwitchPico": PADDLE_SWITCH_PICO_BUTTON_TYPES_TO_LIP,
}
DEVICE_TYPE_SUBTYPE_MAP_TO_LEAP = {
@@ -312,6 +329,7 @@ DEVICE_TYPE_SUBTYPE_MAP_TO_LEAP = {
"Pico4ButtonZone": PICO_4_BUTTON_ZONE_BUTTON_TYPES_TO_LEAP,
"Pico4Button2Group": PICO_4_BUTTON_2_GROUP_BUTTON_TYPES_TO_LEAP,
"FourGroupRemote": FOUR_GROUP_REMOTE_BUTTON_TYPES_TO_LEAP,
+ "PaddleSwitchPico": PADDLE_SWITCH_PICO_BUTTON_TYPES_TO_LEAP,
}
LEAP_TO_DEVICE_TYPE_SUBTYPE_MAP: dict[str, dict[int, str]] = {
@@ -326,6 +344,7 @@ TRIGGER_SCHEMA = vol.Any(
PICO_4_BUTTON_ZONE_TRIGGER_SCHEMA,
PICO_4_BUTTON_2_GROUP_TRIGGER_SCHEMA,
FOUR_GROUP_REMOTE_TRIGGER_SCHEMA,
+ PADDLE_SWITCH_PICO_TRIGGER_SCHEMA,
)
diff --git a/homeassistant/components/lutron_caseta/entity.py b/homeassistant/components/lutron_caseta/entity.py
index f954be74f1d..5ab211ed87b 100644
--- a/homeassistant/components/lutron_caseta/entity.py
+++ b/homeassistant/components/lutron_caseta/entity.py
@@ -63,7 +63,7 @@ class LutronCasetaEntity(Entity):
info[ATTR_SUGGESTED_AREA] = area
self._attr_device_info = info
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register callbacks."""
self._smartbridge.add_subscriber(self.device_id, self.async_write_ha_state)
diff --git a/homeassistant/components/lutron_caseta/fan.py b/homeassistant/components/lutron_caseta/fan.py
index 69167929e14..1e7fe07b8ba 100644
--- a/homeassistant/components/lutron_caseta/fan.py
+++ b/homeassistant/components/lutron_caseta/fan.py
@@ -12,7 +12,7 @@ from homeassistant.components.fan import (
FanEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -28,7 +28,7 @@ ORDERED_NAMED_FAN_SPEEDS = [FAN_LOW, FAN_MEDIUM, FAN_MEDIUM_HIGH, FAN_HIGH]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: LutronCasetaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron Caseta fan platform.
diff --git a/homeassistant/components/lutron_caseta/light.py b/homeassistant/components/lutron_caseta/light.py
index 722c9a15d91..b920a95e435 100644
--- a/homeassistant/components/lutron_caseta/light.py
+++ b/homeassistant/components/lutron_caseta/light.py
@@ -22,7 +22,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DEVICE_TYPE_COLOR_TUNE,
@@ -66,7 +66,7 @@ def to_hass_level(level):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron Caseta light platform.
diff --git a/homeassistant/components/lutron_caseta/manifest.json b/homeassistant/components/lutron_caseta/manifest.json
index bbb6df41a89..96b00a1f392 100644
--- a/homeassistant/components/lutron_caseta/manifest.json
+++ b/homeassistant/components/lutron_caseta/manifest.json
@@ -9,7 +9,7 @@
},
"iot_class": "local_push",
"loggers": ["pylutron_caseta"],
- "requirements": ["pylutron-caseta==0.23.0"],
+ "requirements": ["pylutron-caseta==0.24.0"],
"zeroconf": [
{
"type": "_lutron._tcp.local.",
diff --git a/homeassistant/components/lutron_caseta/scene.py b/homeassistant/components/lutron_caseta/scene.py
index db4423495a4..671df82d8e0 100644
--- a/homeassistant/components/lutron_caseta/scene.py
+++ b/homeassistant/components/lutron_caseta/scene.py
@@ -8,7 +8,7 @@ from homeassistant.components.scene import Scene
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN as CASETA_DOMAIN
from .util import serial_to_unique_id
@@ -17,7 +17,7 @@ from .util import serial_to_unique_id
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron Caseta scene platform.
diff --git a/homeassistant/components/lutron_caseta/switch.py b/homeassistant/components/lutron_caseta/switch.py
index 66f23926fbf..b71ccf4bfa8 100644
--- a/homeassistant/components/lutron_caseta/switch.py
+++ b/homeassistant/components/lutron_caseta/switch.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import LutronCasetaUpdatableEntity
@@ -13,7 +13,7 @@ from .entity import LutronCasetaUpdatableEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Lutron Caseta switch platform.
diff --git a/homeassistant/components/lyric/climate.py b/homeassistant/components/lyric/climate.py
index c5d17cfb176..ffcf08b927a 100644
--- a/homeassistant/components/lyric/climate.py
+++ b/homeassistant/components/lyric/climate.py
@@ -35,7 +35,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -121,7 +121,9 @@ SCHEMA_HOLD_TIME: VolDictType = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Honeywell Lyric climate platform based on a config entry."""
coordinator: DataUpdateCoordinator[Lyric] = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lyric/sensor.py b/homeassistant/components/lyric/sensor.py
index 38cb895a110..065ee0fba9d 100644
--- a/homeassistant/components/lyric/sensor.py
+++ b/homeassistant/components/lyric/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from homeassistant.util import dt as dt_util
@@ -159,7 +159,9 @@ def get_datetime_from_future_time(time_str: str) -> datetime:
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Honeywell Lyric sensor platform based on a config entry."""
coordinator: DataUpdateCoordinator[Lyric] = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/lyric/strings.json b/homeassistant/components/lyric/strings.json
index 83c65359643..bc48a791e70 100644
--- a/homeassistant/components/lyric/strings.json
+++ b/homeassistant/components/lyric/strings.json
@@ -53,12 +53,12 @@
},
"services": {
"set_hold_time": {
- "name": "Set Hold Time",
- "description": "Sets the time to hold until.",
+ "name": "Set hold time",
+ "description": "Sets the time period to keep the temperature and override the schedule.",
"fields": {
"time_period": {
- "name": "Time Period",
- "description": "Time to hold until."
+ "name": "Time period",
+ "description": "Duration for which to override the schedule."
}
}
}
diff --git a/homeassistant/components/madvr/binary_sensor.py b/homeassistant/components/madvr/binary_sensor.py
index b6820f94fea..45c915aba8c 100644
--- a/homeassistant/components/madvr/binary_sensor.py
+++ b/homeassistant/components/madvr/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import MadVRConfigEntry, MadVRCoordinator
from .entity import MadVREntity
@@ -55,7 +55,7 @@ BINARY_SENSORS: tuple[MadvrBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: MadVRConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary sensor entities."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/madvr/remote.py b/homeassistant/components/madvr/remote.py
index 032a1d718f5..23e969e56e3 100644
--- a/homeassistant/components/madvr/remote.py
+++ b/homeassistant/components/madvr/remote.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.remote import RemoteEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import MadVRConfigEntry, MadVRCoordinator
from .entity import MadVREntity
@@ -19,7 +19,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: MadVRConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the madVR remote."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/madvr/sensor.py b/homeassistant/components/madvr/sensor.py
index e54e9dca476..783004f3b84 100644
--- a/homeassistant/components/madvr/sensor.py
+++ b/homeassistant/components/madvr/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -253,7 +253,7 @@ SENSORS: tuple[MadvrSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: MadVRConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor entities."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/madvr/strings.json b/homeassistant/components/madvr/strings.json
index 19f23afddaf..38b949ee5d6 100644
--- a/homeassistant/components/madvr/strings.json
+++ b/homeassistant/components/madvr/strings.json
@@ -3,7 +3,7 @@
"step": {
"user": {
"title": "Set up madVR Envy",
- "description": "Your device needs to be on in order to add the integation.",
+ "description": "Your device needs to be on in order to add the integration.",
"data": {
"host": "[%key:common::config_flow::data::host%]",
"port": "[%key:common::config_flow::data::port%]"
@@ -15,7 +15,7 @@
},
"reconfigure": {
"title": "Reconfigure madVR Envy",
- "description": "Your device needs to be on in order to reconfigure the integation.",
+ "description": "Your device needs to be on in order to reconfigure the integration.",
"data": {
"host": "[%key:common::config_flow::data::host%]",
"port": "[%key:common::config_flow::data::port%]"
diff --git a/homeassistant/components/mastodon/__init__.py b/homeassistant/components/mastodon/__init__.py
index 2f713a97dfe..17b8614a2e9 100644
--- a/homeassistant/components/mastodon/__init__.py
+++ b/homeassistant/components/mastodon/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from mastodon.Mastodon import Mastodon, MastodonError
+from mastodon.Mastodon import Account, Instance, InstanceV2, Mastodon, MastodonError
from homeassistant.const import (
CONF_ACCESS_TOKEN,
@@ -47,7 +47,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: MastodonConfigEntry) ->
assert entry.unique_id
- coordinator = MastodonCoordinator(hass, client)
+ coordinator = MastodonCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
@@ -107,7 +107,9 @@ async def async_migrate_entry(hass: HomeAssistant, entry: MastodonConfigEntry) -
return True
-def setup_mastodon(entry: MastodonConfigEntry) -> tuple[Mastodon, dict, dict]:
+def setup_mastodon(
+ entry: MastodonConfigEntry,
+) -> tuple[Mastodon, InstanceV2 | Instance, Account]:
"""Get mastodon details."""
client = create_mastodon_client(
entry.data[CONF_BASE_URL],
diff --git a/homeassistant/components/mastodon/config_flow.py b/homeassistant/components/mastodon/config_flow.py
index 1b93cbecd98..1ae1e6b229e 100644
--- a/homeassistant/components/mastodon/config_flow.py
+++ b/homeassistant/components/mastodon/config_flow.py
@@ -4,7 +4,12 @@ from __future__ import annotations
from typing import Any
-from mastodon.Mastodon import MastodonNetworkError, MastodonUnauthorizedError
+from mastodon.Mastodon import (
+ Account,
+ Instance,
+ MastodonNetworkError,
+ MastodonUnauthorizedError,
+)
import voluptuous as vol
from yarl import URL
@@ -56,8 +61,8 @@ class MastodonConfigFlow(ConfigFlow, domain=DOMAIN):
client_secret: str,
access_token: str,
) -> tuple[
- dict[str, str] | None,
- dict[str, str] | None,
+ Instance | None,
+ Account | None,
dict[str, str],
]:
"""Check connection to the Mastodon instance."""
diff --git a/homeassistant/components/mastodon/const.py b/homeassistant/components/mastodon/const.py
index b7e86eaad5a..2efda329467 100644
--- a/homeassistant/components/mastodon/const.py
+++ b/homeassistant/components/mastodon/const.py
@@ -12,17 +12,10 @@ DATA_HASS_CONFIG = "mastodon_hass_config"
DEFAULT_URL: Final = "https://mastodon.social"
DEFAULT_NAME: Final = "Mastodon"
-INSTANCE_VERSION: Final = "version"
-INSTANCE_URI: Final = "uri"
-INSTANCE_DOMAIN: Final = "domain"
-ACCOUNT_USERNAME: Final = "username"
-ACCOUNT_FOLLOWERS_COUNT: Final = "followers_count"
-ACCOUNT_FOLLOWING_COUNT: Final = "following_count"
-ACCOUNT_STATUSES_COUNT: Final = "statuses_count"
-
ATTR_CONFIG_ENTRY_ID = "config_entry_id"
ATTR_STATUS = "status"
ATTR_VISIBILITY = "visibility"
ATTR_CONTENT_WARNING = "content_warning"
ATTR_MEDIA_WARNING = "media_warning"
ATTR_MEDIA = "media"
+ATTR_MEDIA_DESCRIPTION = "media_description"
diff --git a/homeassistant/components/mastodon/coordinator.py b/homeassistant/components/mastodon/coordinator.py
index 4c6fe6b1c88..99785eca80b 100644
--- a/homeassistant/components/mastodon/coordinator.py
+++ b/homeassistant/components/mastodon/coordinator.py
@@ -4,10 +4,9 @@ from __future__ import annotations
from dataclasses import dataclass
from datetime import timedelta
-from typing import Any
from mastodon import Mastodon
-from mastodon.Mastodon import MastodonError
+from mastodon.Mastodon import Account, Instance, InstanceV2, MastodonError
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
@@ -21,27 +20,35 @@ class MastodonData:
"""Mastodon data type."""
client: Mastodon
- instance: dict
- account: dict
+ instance: InstanceV2 | Instance
+ account: Account
coordinator: MastodonCoordinator
type MastodonConfigEntry = ConfigEntry[MastodonData]
-class MastodonCoordinator(DataUpdateCoordinator[dict[str, Any]]):
+class MastodonCoordinator(DataUpdateCoordinator[Account]):
"""Class to manage fetching Mastodon data."""
- def __init__(self, hass: HomeAssistant, client: Mastodon) -> None:
+ config_entry: MastodonConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: MastodonConfigEntry, client: Mastodon
+ ) -> None:
"""Initialize coordinator."""
super().__init__(
- hass, logger=LOGGER, name="Mastodon", update_interval=timedelta(hours=1)
+ hass,
+ logger=LOGGER,
+ config_entry=config_entry,
+ name="Mastodon",
+ update_interval=timedelta(hours=1),
)
self.client = client
- async def _async_update_data(self) -> dict[str, Any]:
+ async def _async_update_data(self) -> Account:
try:
- account: dict = await self.hass.async_add_executor_job(
+ account: Account = await self.hass.async_add_executor_job(
self.client.account_verify_credentials
)
except MastodonError as ex:
diff --git a/homeassistant/components/mastodon/diagnostics.py b/homeassistant/components/mastodon/diagnostics.py
index 7246ae9cf63..31444413dfd 100644
--- a/homeassistant/components/mastodon/diagnostics.py
+++ b/homeassistant/components/mastodon/diagnostics.py
@@ -4,9 +4,11 @@ from __future__ import annotations
from typing import Any
+from mastodon.Mastodon import Account, Instance
+
from homeassistant.core import HomeAssistant
-from . import MastodonConfigEntry
+from .coordinator import MastodonConfigEntry
async def async_get_config_entry_diagnostics(
@@ -25,7 +27,7 @@ async def async_get_config_entry_diagnostics(
}
-def get_diagnostics(config_entry: MastodonConfigEntry) -> tuple[dict, dict]:
+def get_diagnostics(config_entry: MastodonConfigEntry) -> tuple[Instance, Account]:
"""Get mastodon diagnostics."""
client = config_entry.runtime_data.client
diff --git a/homeassistant/components/mastodon/entity.py b/homeassistant/components/mastodon/entity.py
index 93d630627d7..60224e75e41 100644
--- a/homeassistant/components/mastodon/entity.py
+++ b/homeassistant/components/mastodon/entity.py
@@ -4,9 +4,8 @@ from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import MastodonConfigEntry
-from .const import DEFAULT_NAME, DOMAIN, INSTANCE_VERSION
-from .coordinator import MastodonCoordinator
+from .const import DEFAULT_NAME, DOMAIN
+from .coordinator import MastodonConfigEntry, MastodonCoordinator
from .utils import construct_mastodon_username
@@ -41,7 +40,7 @@ class MastodonEntity(CoordinatorEntity[MastodonCoordinator]):
manufacturer="Mastodon gGmbH",
model=full_account_name,
entry_type=DeviceEntryType.SERVICE,
- sw_version=data.runtime_data.instance[INSTANCE_VERSION],
+ sw_version=data.runtime_data.instance.version,
name=name,
)
diff --git a/homeassistant/components/mastodon/manifest.json b/homeassistant/components/mastodon/manifest.json
index 20c506e7766..d7b21ad3a0c 100644
--- a/homeassistant/components/mastodon/manifest.json
+++ b/homeassistant/components/mastodon/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "service",
"iot_class": "cloud_polling",
"loggers": ["mastodon"],
- "requirements": ["Mastodon.py==1.8.1"]
+ "requirements": ["Mastodon.py==2.0.1"]
}
diff --git a/homeassistant/components/mastodon/notify.py b/homeassistant/components/mastodon/notify.py
index 8e7e9dc1947..149ef1f6a48 100644
--- a/homeassistant/components/mastodon/notify.py
+++ b/homeassistant/components/mastodon/notify.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from typing import Any, cast
from mastodon import Mastodon
-from mastodon.Mastodon import MastodonAPIError
+from mastodon.Mastodon import MastodonAPIError, MediaAttachment
import voluptuous as vol
from homeassistant.components.notify import (
@@ -52,7 +52,7 @@ async def async_get_service(
if discovery_info is None:
return None
- client: Mastodon = discovery_info.get("client")
+ client = cast(Mastodon, discovery_info.get("client"))
return MastodonNotificationService(hass, client)
@@ -114,7 +114,7 @@ class MastodonNotificationService(BaseNotificationService):
message,
visibility=target,
spoiler_text=content_warning,
- media_ids=mediadata["id"],
+ media_ids=mediadata.id,
sensitive=sensitive,
)
except MastodonAPIError as err:
@@ -134,12 +134,14 @@ class MastodonNotificationService(BaseNotificationService):
translation_key="unable_to_send_message",
) from err
- def _upload_media(self, media_path: Any = None) -> Any:
+ def _upload_media(self, media_path: Any = None) -> MediaAttachment:
"""Upload media."""
with open(media_path, "rb"):
media_type = get_media_type(media_path)
try:
- mediadata = self.client.media_post(media_path, mime_type=media_type)
+ mediadata: MediaAttachment = self.client.media_post(
+ media_path, mime_type=media_type
+ )
except MastodonAPIError as err:
raise HomeAssistantError(
translation_domain=DOMAIN,
diff --git a/homeassistant/components/mastodon/quality_scale.yaml b/homeassistant/components/mastodon/quality_scale.yaml
index 43636ed6924..f07f7e0a8ad 100644
--- a/homeassistant/components/mastodon/quality_scale.yaml
+++ b/homeassistant/components/mastodon/quality_scale.yaml
@@ -93,7 +93,4 @@ rules:
# Platinum
async-dependency: todo
inject-websession: todo
- strict-typing:
- status: todo
- comment: |
- Requirement 'Mastodon.py==1.8.1' appears untyped
+ strict-typing: done
diff --git a/homeassistant/components/mastodon/sensor.py b/homeassistant/components/mastodon/sensor.py
index 1bb59ad7c05..bfdc9c90333 100644
--- a/homeassistant/components/mastodon/sensor.py
+++ b/homeassistant/components/mastodon/sensor.py
@@ -4,7 +4,8 @@ from __future__ import annotations
from collections.abc import Callable
from dataclasses import dataclass
-from typing import Any
+
+from mastodon.Mastodon import Account
from homeassistant.components.sensor import (
SensorEntity,
@@ -12,15 +13,10 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import MastodonConfigEntry
-from .const import (
- ACCOUNT_FOLLOWERS_COUNT,
- ACCOUNT_FOLLOWING_COUNT,
- ACCOUNT_STATUSES_COUNT,
-)
+from .coordinator import MastodonConfigEntry
from .entity import MastodonEntity
# Coordinator is used to centralize the data updates
@@ -31,7 +27,7 @@ PARALLEL_UPDATES = 0
class MastodonSensorEntityDescription(SensorEntityDescription):
"""Describes Mastodon sensor entity."""
- value_fn: Callable[[dict[str, Any]], StateType]
+ value_fn: Callable[[Account], StateType]
ENTITY_DESCRIPTIONS = (
@@ -39,19 +35,19 @@ ENTITY_DESCRIPTIONS = (
key="followers",
translation_key="followers",
state_class=SensorStateClass.TOTAL,
- value_fn=lambda data: data.get(ACCOUNT_FOLLOWERS_COUNT),
+ value_fn=lambda data: data.followers_count,
),
MastodonSensorEntityDescription(
key="following",
translation_key="following",
state_class=SensorStateClass.TOTAL,
- value_fn=lambda data: data.get(ACCOUNT_FOLLOWING_COUNT),
+ value_fn=lambda data: data.following_count,
),
MastodonSensorEntityDescription(
key="posts",
translation_key="posts",
state_class=SensorStateClass.TOTAL,
- value_fn=lambda data: data.get(ACCOUNT_STATUSES_COUNT),
+ value_fn=lambda data: data.statuses_count,
),
)
@@ -59,7 +55,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: MastodonConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform for entity."""
coordinator = entry.runtime_data.coordinator
diff --git a/homeassistant/components/mastodon/services.py b/homeassistant/components/mastodon/services.py
index ab3a89c0c4b..68e95e726a1 100644
--- a/homeassistant/components/mastodon/services.py
+++ b/homeassistant/components/mastodon/services.py
@@ -5,23 +5,24 @@ from functools import partial
from typing import Any, cast
from mastodon import Mastodon
-from mastodon.Mastodon import MastodonAPIError
+from mastodon.Mastodon import MastodonAPIError, MediaAttachment
import voluptuous as vol
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant, ServiceCall, ServiceResponse
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
-from . import MastodonConfigEntry
from .const import (
ATTR_CONFIG_ENTRY_ID,
ATTR_CONTENT_WARNING,
ATTR_MEDIA,
+ ATTR_MEDIA_DESCRIPTION,
ATTR_MEDIA_WARNING,
ATTR_STATUS,
ATTR_VISIBILITY,
DOMAIN,
)
+from .coordinator import MastodonConfigEntry
from .utils import get_media_type
@@ -42,6 +43,7 @@ SERVICE_POST_SCHEMA = vol.Schema(
vol.Optional(ATTR_VISIBILITY): vol.In([x.lower() for x in StatusVisibility]),
vol.Optional(ATTR_CONTENT_WARNING): str,
vol.Optional(ATTR_MEDIA): str,
+ vol.Optional(ATTR_MEDIA_DESCRIPTION): str,
vol.Optional(ATTR_MEDIA_WARNING): bool,
}
)
@@ -81,6 +83,7 @@ def setup_services(hass: HomeAssistant) -> None:
)
spoiler_text: str | None = call.data.get(ATTR_CONTENT_WARNING)
media_path: str | None = call.data.get(ATTR_MEDIA)
+ media_description: str | None = call.data.get(ATTR_MEDIA_DESCRIPTION)
media_warning: str | None = call.data.get(ATTR_MEDIA_WARNING)
await hass.async_add_executor_job(
@@ -91,6 +94,7 @@ def setup_services(hass: HomeAssistant) -> None:
visibility=visibility,
spoiler_text=spoiler_text,
media_path=media_path,
+ media_description=media_description,
sensitive=media_warning,
)
)
@@ -100,7 +104,7 @@ def setup_services(hass: HomeAssistant) -> None:
def _post(client: Mastodon, **kwargs: Any) -> None:
"""Post to Mastodon."""
- media_data: dict[str, Any] | None = None
+ media_data: MediaAttachment | None = None
media_path = kwargs.get("media_path")
if media_path:
@@ -112,9 +116,12 @@ def setup_services(hass: HomeAssistant) -> None:
)
media_type = get_media_type(media_path)
+ media_description = kwargs.get("media_description")
try:
media_data = client.media_post(
- media_file=media_path, mime_type=media_type
+ media_file=media_path,
+ mime_type=media_type,
+ description=media_description,
)
except MastodonAPIError as err:
@@ -125,11 +132,12 @@ def setup_services(hass: HomeAssistant) -> None:
) from err
kwargs.pop("media_path", None)
+ kwargs.pop("media_description", None)
try:
media_ids: str | None = None
if media_data:
- media_ids = media_data["id"]
+ media_ids = media_data.id
client.status_post(media_ids=media_ids, **kwargs)
except MastodonAPIError as err:
raise HomeAssistantError(
diff --git a/homeassistant/components/mastodon/services.yaml b/homeassistant/components/mastodon/services.yaml
index 161a0d152ca..206dc36c1a2 100644
--- a/homeassistant/components/mastodon/services.yaml
+++ b/homeassistant/components/mastodon/services.yaml
@@ -24,6 +24,10 @@ post:
media:
selector:
text:
+ media_description:
+ required: false
+ selector:
+ text:
media_warning:
required: true
selector:
diff --git a/homeassistant/components/mastodon/strings.json b/homeassistant/components/mastodon/strings.json
index 87858f768e4..9e6cf6db6bf 100644
--- a/homeassistant/components/mastodon/strings.json
+++ b/homeassistant/components/mastodon/strings.json
@@ -22,7 +22,7 @@
"error": {
"unauthorized_error": "The credentials are incorrect.",
"network_error": "The Mastodon instance was not found.",
- "unknown": "Unknown error occured when connecting to the Mastodon instance."
+ "unknown": "Unknown error occurred when connecting to the Mastodon instance."
}
},
"exceptions": {
@@ -89,6 +89,10 @@
"name": "Media",
"description": "Attach an image or video to the post."
},
+ "media_description": {
+ "name": "Media description",
+ "description": "If an image or video is attached, will add a description for this media for people with visual impairments."
+ },
"media_warning": {
"name": "Media warning",
"description": "If an image or video is attached, will mark the media as sensitive (default: no media warning)."
diff --git a/homeassistant/components/mastodon/utils.py b/homeassistant/components/mastodon/utils.py
index e9c2567b675..898578c931b 100644
--- a/homeassistant/components/mastodon/utils.py
+++ b/homeassistant/components/mastodon/utils.py
@@ -6,8 +6,9 @@ import mimetypes
from typing import Any
from mastodon import Mastodon
+from mastodon.Mastodon import Account, Instance, InstanceV2
-from .const import ACCOUNT_USERNAME, DEFAULT_NAME, INSTANCE_DOMAIN, INSTANCE_URI
+from .const import DEFAULT_NAME
def create_mastodon_client(
@@ -23,14 +24,13 @@ def create_mastodon_client(
def construct_mastodon_username(
- instance: dict[str, str] | None, account: dict[str, str] | None
+ instance: InstanceV2 | Instance | None, account: Account | None
) -> str:
"""Construct a mastodon username from the account and instance."""
if instance and account:
- return (
- f"@{account[ACCOUNT_USERNAME]}@"
- f"{instance.get(INSTANCE_URI, instance.get(INSTANCE_DOMAIN))}"
- )
+ if type(instance) is InstanceV2:
+ return f"@{account.username}@{instance.domain}"
+ return f"@{account.username}@{instance.uri}"
return DEFAULT_NAME
diff --git a/homeassistant/components/matter/binary_sensor.py b/homeassistant/components/matter/binary_sensor.py
index 484ed94fb90..b5665e5d47a 100644
--- a/homeassistant/components/matter/binary_sensor.py
+++ b/homeassistant/components/matter/binary_sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity, MatterEntityDescription
from .helpers import get_matter
@@ -28,7 +28,7 @@ from .models import MatterDiscoverySchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter binary sensor from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/button.py b/homeassistant/components/matter/button.py
index 634406d18eb..6a0a5fc5b1d 100644
--- a/homeassistant/components/matter/button.py
+++ b/homeassistant/components/matter/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity, MatterEntityDescription
from .helpers import get_matter
@@ -26,7 +26,7 @@ from .models import MatterDiscoverySchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter Button platform."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/climate.py b/homeassistant/components/matter/climate.py
index 25419c34e42..df57da4ded3 100644
--- a/homeassistant/components/matter/climate.py
+++ b/homeassistant/components/matter/climate.py
@@ -24,7 +24,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity
from .helpers import get_matter
@@ -174,7 +174,7 @@ class ThermostatRunningState(IntEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter climate platform from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/cover.py b/homeassistant/components/matter/cover.py
index 5b109d52189..2e2d4390b30 100644
--- a/homeassistant/components/matter/cover.py
+++ b/homeassistant/components/matter/cover.py
@@ -19,7 +19,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import LOGGER
from .entity import MatterEntity
@@ -48,7 +48,7 @@ class OperationalStatus(IntEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter Cover from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/event.py b/homeassistant/components/matter/event.py
index 3cb3fe385d4..6fa775fd1b9 100644
--- a/homeassistant/components/matter/event.py
+++ b/homeassistant/components/matter/event.py
@@ -16,7 +16,7 @@ from homeassistant.components.event import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity
from .helpers import get_matter
@@ -39,7 +39,7 @@ EVENT_TYPES_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter switches from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/fan.py b/homeassistant/components/matter/fan.py
index 8b8ebee619d..2c9e190d58a 100644
--- a/homeassistant/components/matter/fan.py
+++ b/homeassistant/components/matter/fan.py
@@ -16,7 +16,7 @@ from homeassistant.components.fan import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity
from .helpers import get_matter
@@ -45,7 +45,7 @@ PRESET_SLEEP_WIND = "sleep_wind"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter fan from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/light.py b/homeassistant/components/matter/light.py
index 5c20554f065..8ea804a8a7c 100644
--- a/homeassistant/components/matter/light.py
+++ b/homeassistant/components/matter/light.py
@@ -24,7 +24,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .const import LOGGER
@@ -77,7 +77,7 @@ TRANSITION_BLOCKLIST = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter Light from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/lock.py b/homeassistant/components/matter/lock.py
index 8524b39d584..81de7482d46 100644
--- a/homeassistant/components/matter/lock.py
+++ b/homeassistant/components/matter/lock.py
@@ -15,7 +15,7 @@ from homeassistant.components.lock import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_CODE, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import LOGGER
from .entity import MatterEntity
@@ -28,7 +28,7 @@ DoorLockFeature = clusters.DoorLock.Bitmaps.Feature
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter lock from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/manifest.json b/homeassistant/components/matter/manifest.json
index 669fa1af8c4..48f0bfa2e67 100644
--- a/homeassistant/components/matter/manifest.json
+++ b/homeassistant/components/matter/manifest.json
@@ -1,6 +1,6 @@
{
"domain": "matter",
- "name": "Matter (BETA)",
+ "name": "Matter",
"after_dependencies": ["hassio"],
"codeowners": ["@home-assistant/matter"],
"config_flow": true,
diff --git a/homeassistant/components/matter/number.py b/homeassistant/components/matter/number.py
index 93b6b8f75c9..44538f46856 100644
--- a/homeassistant/components/matter/number.py
+++ b/homeassistant/components/matter/number.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity, MatterEntityDescription
from .helpers import get_matter
@@ -32,7 +32,7 @@ from .models import MatterDiscoverySchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter Number Input from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/select.py b/homeassistant/components/matter/select.py
index b2d1c7f8ddb..e78c34391cd 100644
--- a/homeassistant/components/matter/select.py
+++ b/homeassistant/components/matter/select.py
@@ -14,7 +14,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity, MatterEntityDescription
from .helpers import get_matter
@@ -47,7 +47,7 @@ type SelectCluster = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter ModeSelect from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/sensor.py b/homeassistant/components/matter/sensor.py
index 3503e112db5..10f8db275f5 100644
--- a/homeassistant/components/matter/sensor.py
+++ b/homeassistant/components/matter/sensor.py
@@ -40,7 +40,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import slugify
from .entity import MatterEntity, MatterEntityDescription
@@ -81,7 +81,7 @@ OPERATIONAL_STATE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter sensors from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/strings.json b/homeassistant/components/matter/strings.json
index f299b5cb628..1404d0a9076 100644
--- a/homeassistant/components/matter/strings.json
+++ b/homeassistant/components/matter/strings.json
@@ -263,7 +263,7 @@
"paused": "[%key:common::state::paused%]",
"error": "Error",
"seeking_charger": "Seeking charger",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"docked": "Docked"
}
},
diff --git a/homeassistant/components/matter/switch.py b/homeassistant/components/matter/switch.py
index 890ca662295..af4803af9a1 100644
--- a/homeassistant/components/matter/switch.py
+++ b/homeassistant/components/matter/switch.py
@@ -16,7 +16,7 @@ from homeassistant.components.switch import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity, MatterEntityDescription
from .helpers import get_matter
@@ -26,7 +26,7 @@ from .models import MatterDiscoverySchema
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter switches from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/update.py b/homeassistant/components/matter/update.py
index 5ee9b2e5fa0..7c9ca991914 100644
--- a/homeassistant/components/matter/update.py
+++ b/homeassistant/components/matter/update.py
@@ -21,7 +21,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_ON, Platform
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.restore_state import ExtraStoredData
@@ -60,7 +60,7 @@ class MatterUpdateExtraStoredData(ExtraStoredData):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter lock from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/vacuum.py b/homeassistant/components/matter/vacuum.py
index de4a885d8fb..5ea1716a37d 100644
--- a/homeassistant/components/matter/vacuum.py
+++ b/homeassistant/components/matter/vacuum.py
@@ -17,7 +17,7 @@ from homeassistant.components.vacuum import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity
from .helpers import get_matter
@@ -50,7 +50,7 @@ class ModeTag(IntEnum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter vacuum platform from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/matter/valve.py b/homeassistant/components/matter/valve.py
index 29946621853..bea11468c6b 100644
--- a/homeassistant/components/matter/valve.py
+++ b/homeassistant/components/matter/valve.py
@@ -14,7 +14,7 @@ from homeassistant.components.valve import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import MatterEntity
from .helpers import get_matter
@@ -28,7 +28,7 @@ ValveStateEnum = ValveConfigurationAndControl.Enums.ValveStateEnum
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Matter valve platform from Config Entry."""
matter = get_matter(hass)
diff --git a/homeassistant/components/mazda/__init__.py b/homeassistant/components/mazda/__init__.py
index fd323060ac0..ccbb331573e 100644
--- a/homeassistant/components/mazda/__init__.py
+++ b/homeassistant/components/mazda/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
@@ -29,11 +29,13 @@ async def async_setup_entry(hass: HomeAssistant, _: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
- if all(
- config_entry.state is ConfigEntryState.NOT_LOADED
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.entry_id != entry.entry_id
- ):
- ir.async_delete_issue(hass, DOMAIN, DOMAIN)
-
return True
+
+
+async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Remove a config entry."""
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
+ ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ # Remove any remaining disabled or ignored entries
+ for _entry in hass.config_entries.async_entries(DOMAIN):
+ hass.async_create_task(hass.config_entries.async_remove(_entry.entry_id))
diff --git a/homeassistant/components/mcp/__init__.py b/homeassistant/components/mcp/__init__.py
index 4a2b4da990d..41b6a260d9f 100644
--- a/homeassistant/components/mcp/__init__.py
+++ b/homeassistant/components/mcp/__init__.py
@@ -39,7 +39,6 @@ async def async_setup_entry(
entry.async_on_unload(unsub)
entry.runtime_data = coordinator
- entry.async_on_unload(coordinator.close)
return True
diff --git a/homeassistant/components/mcp/coordinator.py b/homeassistant/components/mcp/coordinator.py
index a5c5ee55dbf..6e66036c548 100644
--- a/homeassistant/components/mcp/coordinator.py
+++ b/homeassistant/components/mcp/coordinator.py
@@ -40,6 +40,7 @@ async def mcp_client(url: str) -> AsyncGenerator[ClientSession]:
await session.initialize()
yield session
except ExceptionGroup as err:
+ _LOGGER.debug("Error creating MCP client: %s", err)
raise err.exceptions[0] from err
@@ -51,13 +52,13 @@ class ModelContextProtocolTool(llm.Tool):
name: str,
description: str | None,
parameters: vol.Schema,
- session: ClientSession,
+ server_url: str,
) -> None:
"""Initialize the tool."""
self.name = name
self.description = description
self.parameters = parameters
- self.session = session
+ self.server_url = server_url
async def async_call(
self,
@@ -67,10 +68,16 @@ class ModelContextProtocolTool(llm.Tool):
) -> JsonObjectType:
"""Call the tool."""
try:
- result = await self.session.call_tool(
- tool_input.tool_name, tool_input.tool_args
- )
+ async with asyncio.timeout(TIMEOUT):
+ async with mcp_client(self.server_url) as session:
+ result = await session.call_tool(
+ tool_input.tool_name, tool_input.tool_args
+ )
+ except TimeoutError as error:
+ _LOGGER.debug("Timeout when calling tool: %s", error)
+ raise HomeAssistantError(f"Timeout when calling tool: {error}") from error
except httpx.HTTPStatusError as error:
+ _LOGGER.debug("Error when calling tool: %s", error)
raise HomeAssistantError(f"Error when calling tool: {error}") from error
return result.model_dump(exclude_unset=True, exclude_none=True)
@@ -79,8 +86,6 @@ class ModelContextProtocolCoordinator(DataUpdateCoordinator[list[llm.Tool]]):
"""Define an object to hold MCP data."""
config_entry: ConfigEntry
- _session: ClientSession | None = None
- _setup_error: Exception | None = None
def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize ModelContextProtocolCoordinator."""
@@ -91,52 +96,6 @@ class ModelContextProtocolCoordinator(DataUpdateCoordinator[list[llm.Tool]]):
config_entry=config_entry,
update_interval=UPDATE_INTERVAL,
)
- self._stop = asyncio.Event()
-
- async def _async_setup(self) -> None:
- """Set up the client connection."""
- connected = asyncio.Event()
- stop = asyncio.Event()
- self.config_entry.async_create_background_task(
- self.hass, self._connect(connected, stop), "mcp-client"
- )
- try:
- async with asyncio.timeout(TIMEOUT):
- await connected.wait()
- self._stop = stop
- finally:
- if self._setup_error is not None:
- raise self._setup_error
-
- async def _connect(self, connected: asyncio.Event, stop: asyncio.Event) -> None:
- """Create a server-sent event MCP client."""
- url = self.config_entry.data[CONF_URL]
- try:
- async with (
- sse_client(url=url) as streams,
- ClientSession(*streams) as session,
- ):
- await session.initialize()
- self._session = session
- connected.set()
- await stop.wait()
- except httpx.HTTPStatusError as err:
- self._setup_error = err
- _LOGGER.debug("Error connecting to MCP server: %s", err)
- raise UpdateFailed(f"Error connecting to MCP server: {err}") from err
- except ExceptionGroup as err:
- self._setup_error = err.exceptions[0]
- _LOGGER.debug("Error connecting to MCP server: %s", err)
- raise UpdateFailed(
- "Error connecting to MCP server: {err.exceptions[0]}"
- ) from err.exceptions[0]
- finally:
- self._session = None
-
- async def close(self) -> None:
- """Close the client connection."""
- if self._stop is not None:
- self._stop.set()
async def _async_update_data(self) -> list[llm.Tool]:
"""Fetch data from API endpoint.
@@ -144,11 +103,15 @@ class ModelContextProtocolCoordinator(DataUpdateCoordinator[list[llm.Tool]]):
This is the place to pre-process the data to lookup tables
so entities can quickly look up their data.
"""
- if self._session is None:
- raise UpdateFailed("No session available")
try:
- result = await self._session.list_tools()
+ async with asyncio.timeout(TIMEOUT):
+ async with mcp_client(self.config_entry.data[CONF_URL]) as session:
+ result = await session.list_tools()
+ except TimeoutError as error:
+ _LOGGER.debug("Timeout when listing tools: %s", error)
+ raise UpdateFailed(f"Timeout when listing tools: {error}") from error
except httpx.HTTPError as err:
+ _LOGGER.debug("Error communicating with API: %s", err)
raise UpdateFailed(f"Error communicating with API: {err}") from err
_LOGGER.debug("Received tools: %s", result.tools)
@@ -165,7 +128,7 @@ class ModelContextProtocolCoordinator(DataUpdateCoordinator[list[llm.Tool]]):
tool.name,
tool.description,
parameters,
- self._session,
+ self.config_entry.data[CONF_URL],
)
)
return tools
diff --git a/homeassistant/components/mcp_server/__init__.py b/homeassistant/components/mcp_server/__init__.py
index 941eccbe528..e523f46228f 100644
--- a/homeassistant/components/mcp_server/__init__.py
+++ b/homeassistant/components/mcp_server/__init__.py
@@ -6,7 +6,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.typing import ConfigType
-from . import http, llm_api
+from . import http
from .const import DOMAIN
from .session import SessionManager
from .types import MCPServerConfigEntry
@@ -25,7 +25,6 @@ CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the Model Context Protocol component."""
http.async_register(hass)
- llm_api.async_register_api(hass)
return True
diff --git a/homeassistant/components/mcp_server/config_flow.py b/homeassistant/components/mcp_server/config_flow.py
index 8d8d311b874..e8df68de5e2 100644
--- a/homeassistant/components/mcp_server/config_flow.py
+++ b/homeassistant/components/mcp_server/config_flow.py
@@ -16,7 +16,7 @@ from homeassistant.helpers.selector import (
SelectSelectorConfig,
)
-from .const import DOMAIN, LLM_API, LLM_API_NAME
+from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
@@ -33,13 +33,6 @@ class ModelContextServerProtocolConfigFlow(ConfigFlow, domain=DOMAIN):
) -> ConfigFlowResult:
"""Handle the initial step."""
llm_apis = {api.id: api.name for api in llm.async_get_apis(self.hass)}
- if LLM_API not in llm_apis:
- # MCP server component is not loaded yet, so make the LLM API a choice.
- llm_apis = {
- LLM_API: LLM_API_NAME,
- **llm_apis,
- }
-
if user_input is not None:
return self.async_create_entry(
title=llm_apis[user_input[CONF_LLM_HASS_API]], data=user_input
diff --git a/homeassistant/components/mcp_server/const.py b/homeassistant/components/mcp_server/const.py
index 8958ac36616..3f2e12cbb6a 100644
--- a/homeassistant/components/mcp_server/const.py
+++ b/homeassistant/components/mcp_server/const.py
@@ -2,5 +2,6 @@
DOMAIN = "mcp_server"
TITLE = "Model Context Protocol Server"
-LLM_API = "stateless_assist"
-LLM_API_NAME = "Stateless Assist"
+# The Stateless API is no longer registered explicitly, but this name may still exist in the
+# users config entry.
+STATELESS_LLM_API = "stateless_assist"
diff --git a/homeassistant/components/mcp_server/http.py b/homeassistant/components/mcp_server/http.py
index 433d978cef7..bc8fdbd56c8 100644
--- a/homeassistant/components/mcp_server/http.py
+++ b/homeassistant/components/mcp_server/http.py
@@ -25,7 +25,6 @@ from mcp import types
from homeassistant.components import conversation
from homeassistant.components.http import KEY_HASS, HomeAssistantView
-from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import CONF_LLM_HASS_API
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import llm
@@ -56,11 +55,9 @@ def async_get_config_entry(hass: HomeAssistant) -> MCPServerConfigEntry:
Will raise an HTTP error if the expected configuration is not present.
"""
- config_entries: list[MCPServerConfigEntry] = [
- config_entry
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.state == ConfigEntryState.LOADED
- ]
+ config_entries: list[MCPServerConfigEntry] = (
+ hass.config_entries.async_loaded_entries(DOMAIN)
+ )
if not config_entries:
raise HTTPNotFound(text="Model Context Protocol server is not configured")
if len(config_entries) > 1:
diff --git a/homeassistant/components/mcp_server/llm_api.py b/homeassistant/components/mcp_server/llm_api.py
deleted file mode 100644
index 5c29b29153e..00000000000
--- a/homeassistant/components/mcp_server/llm_api.py
+++ /dev/null
@@ -1,48 +0,0 @@
-"""LLM API for MCP Server."""
-
-from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers import llm
-from homeassistant.util import yaml as yaml_util
-
-from .const import LLM_API, LLM_API_NAME
-
-EXPOSED_ENTITY_FIELDS = {"name", "domain", "description", "areas", "names"}
-
-
-def async_register_api(hass: HomeAssistant) -> None:
- """Register the LLM API."""
- llm.async_register_api(hass, StatelessAssistAPI(hass))
-
-
-class StatelessAssistAPI(llm.AssistAPI):
- """LLM API for MCP Server that provides the Assist API without state information in the prompt.
-
- Syncing the state information is possible, but may put unnecessary load on
- the system so we are instead providing the prompt without entity state. Since
- actions don't care about the current state, there is little quality loss.
- """
-
- def __init__(self, hass: HomeAssistant) -> None:
- """Initialize the StatelessAssistAPI."""
- super().__init__(hass)
- self.id = LLM_API
- self.name = LLM_API_NAME
-
- @callback
- def _async_get_exposed_entities_prompt(
- self, llm_context: llm.LLMContext, exposed_entities: dict | None
- ) -> list[str]:
- """Return the prompt for the exposed entities."""
- prompt = []
-
- if exposed_entities and exposed_entities["entities"]:
- prompt.append(
- "An overview of the areas and the devices in this smart home:"
- )
- entities = [
- {k: v for k, v in entity_info.items() if k in EXPOSED_ENTITY_FIELDS}
- for entity_info in exposed_entities["entities"].values()
- ]
- prompt.append(yaml_util.dump(list(entities)))
-
- return prompt
diff --git a/homeassistant/components/mcp_server/manifest.json b/homeassistant/components/mcp_server/manifest.json
index 18b2e5bc417..a3e00d13c4b 100644
--- a/homeassistant/components/mcp_server/manifest.json
+++ b/homeassistant/components/mcp_server/manifest.json
@@ -8,6 +8,6 @@
"integration_type": "service",
"iot_class": "local_push",
"quality_scale": "silver",
- "requirements": ["mcp==1.1.2", "aiohttp_sse==2.2.0", "anyio==4.8.0"],
+ "requirements": ["mcp==1.1.2", "aiohttp_sse==2.2.0", "anyio==4.9.0"],
"single_config_entry": true
}
diff --git a/homeassistant/components/mcp_server/server.py b/homeassistant/components/mcp_server/server.py
index ba21abd722c..88b179ae7c2 100644
--- a/homeassistant/components/mcp_server/server.py
+++ b/homeassistant/components/mcp_server/server.py
@@ -21,6 +21,8 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import llm
+from .const import STATELESS_LLM_API
+
_LOGGER = logging.getLogger(__name__)
@@ -47,16 +49,23 @@ async def create_server(
A Model Context Protocol Server object is associated with a single session.
The MCP SDK handles the details of the protocol.
"""
+ if llm_api_id == STATELESS_LLM_API:
+ llm_api_id = llm.LLM_API_ASSIST
server = Server("home-assistant")
+ async def get_api_instance() -> llm.APIInstance:
+ """Get the LLM API selected."""
+ # Backwards compatibility with old MCP Server config
+ return await llm.async_get_api(hass, llm_api_id, llm_context)
+
@server.list_prompts() # type: ignore[no-untyped-call, misc]
async def handle_list_prompts() -> list[types.Prompt]:
- llm_api = await llm.async_get_api(hass, llm_api_id, llm_context)
+ llm_api = await get_api_instance()
return [
types.Prompt(
name=llm_api.api.name,
- description=f"Default prompt for the Home Assistant LLM API {llm_api.api.name}",
+ description=f"Default prompt for Home Assistant {llm_api.api.name} API",
)
]
@@ -64,12 +73,12 @@ async def create_server(
async def handle_get_prompt(
name: str, arguments: dict[str, str] | None
) -> types.GetPromptResult:
- llm_api = await llm.async_get_api(hass, llm_api_id, llm_context)
+ llm_api = await get_api_instance()
if name != llm_api.api.name:
raise ValueError(f"Unknown prompt: {name}")
return types.GetPromptResult(
- description=f"Default prompt for the Home Assistant LLM API {llm_api.api.name}",
+ description=f"Default prompt for Home Assistant {llm_api.api.name} API",
messages=[
types.PromptMessage(
role="assistant",
@@ -84,13 +93,13 @@ async def create_server(
@server.list_tools() # type: ignore[no-untyped-call, misc]
async def list_tools() -> list[types.Tool]:
"""List available time tools."""
- llm_api = await llm.async_get_api(hass, llm_api_id, llm_context)
+ llm_api = await get_api_instance()
return [_format_tool(tool, llm_api.custom_serializer) for tool in llm_api.tools]
@server.call_tool() # type: ignore[no-untyped-call, misc]
async def call_tool(name: str, arguments: dict) -> Sequence[types.TextContent]:
"""Handle calling tools."""
- llm_api = await llm.async_get_api(hass, llm_api_id, llm_context)
+ llm_api = await get_api_instance()
tool_input = llm.ToolInput(tool_name=name, tool_args=arguments)
_LOGGER.debug("Tool call: %s(%s)", tool_input.tool_name, tool_input.tool_args)
diff --git a/homeassistant/components/mcp_server/strings.json b/homeassistant/components/mcp_server/strings.json
index fbd14038ddc..57f1baf183c 100644
--- a/homeassistant/components/mcp_server/strings.json
+++ b/homeassistant/components/mcp_server/strings.json
@@ -7,7 +7,7 @@
"llm_hass_api": "[%key:common::config_flow::data::llm_hass_api%]"
},
"data_description": {
- "llm_hass_api": "The method for controling Home Assistant to expose with the Model Context Protocol."
+ "llm_hass_api": "The method for controlling Home Assistant to expose with the Model Context Protocol."
}
}
},
diff --git a/homeassistant/components/mealie/__init__.py b/homeassistant/components/mealie/__init__.py
index 5e1523b939a..e019dae2c33 100644
--- a/homeassistant/components/mealie/__init__.py
+++ b/homeassistant/components/mealie/__init__.py
@@ -86,9 +86,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: MealieConfigEntry) -> bo
sw_version=about.version,
)
- mealplan_coordinator = MealieMealplanCoordinator(hass, client)
- shoppinglist_coordinator = MealieShoppingListCoordinator(hass, client)
- statistics_coordinator = MealieStatisticsCoordinator(hass, client)
+ mealplan_coordinator = MealieMealplanCoordinator(hass, entry, client)
+ shoppinglist_coordinator = MealieShoppingListCoordinator(hass, entry, client)
+ statistics_coordinator = MealieStatisticsCoordinator(hass, entry, client)
await mealplan_coordinator.async_config_entry_first_refresh()
await shoppinglist_coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/mealie/calendar.py b/homeassistant/components/mealie/calendar.py
index 729bc16c6fd..556ddede2e2 100644
--- a/homeassistant/components/mealie/calendar.py
+++ b/homeassistant/components/mealie/calendar.py
@@ -8,7 +8,7 @@ from aiomealie import Mealplan, MealplanEntryType
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import MealieConfigEntry, MealieMealplanCoordinator
from .entity import MealieEntity
@@ -19,7 +19,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: MealieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the calendar platform for entity."""
coordinator = entry.runtime_data.mealplan_coordinator
diff --git a/homeassistant/components/mealie/coordinator.py b/homeassistant/components/mealie/coordinator.py
index cf8dfb5bc90..ae5b9cd8c97 100644
--- a/homeassistant/components/mealie/coordinator.py
+++ b/homeassistant/components/mealie/coordinator.py
@@ -48,11 +48,14 @@ class MealieDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
_name: str
_update_interval: timedelta
- def __init__(self, hass: HomeAssistant, client: MealieClient) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: MealieConfigEntry, client: MealieClient
+ ) -> None:
"""Initialize the Mealie data coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=f"Mealie {self._name}",
update_interval=self._update_interval,
)
diff --git a/homeassistant/components/mealie/sensor.py b/homeassistant/components/mealie/sensor.py
index e4b1655a9d1..062a2646cab 100644
--- a/homeassistant/components/mealie/sensor.py
+++ b/homeassistant/components/mealie/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import MealieConfigEntry, MealieStatisticsCoordinator
@@ -59,7 +59,7 @@ SENSOR_TYPES: tuple[MealieStatisticsSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: MealieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Mealie sensors based on a config entry."""
coordinator = entry.runtime_data.statistics_coordinator
diff --git a/homeassistant/components/mealie/strings.json b/homeassistant/components/mealie/strings.json
index fa63252e837..186fc4c4ac0 100644
--- a/homeassistant/components/mealie/strings.json
+++ b/homeassistant/components/mealie/strings.json
@@ -146,11 +146,11 @@
"services": {
"get_mealplan": {
"name": "Get mealplan",
- "description": "Get mealplan from Mealie",
+ "description": "Gets a mealplan from Mealie",
"fields": {
"config_entry_id": {
"name": "Mealie instance",
- "description": "Select the Mealie instance to get mealplan from"
+ "description": "The Mealie instance to use for this action."
},
"start_date": {
"name": "Start date",
@@ -164,7 +164,7 @@
},
"get_recipe": {
"name": "Get recipe",
- "description": "Get recipe from Mealie",
+ "description": "Gets a recipe from Mealie",
"fields": {
"config_entry_id": {
"name": "[%key:component::mealie::services::get_mealplan::fields::config_entry_id::name%]",
@@ -178,7 +178,7 @@
},
"import_recipe": {
"name": "Import recipe",
- "description": "Import recipe from an URL",
+ "description": "Imports a recipe from an URL",
"fields": {
"config_entry_id": {
"name": "[%key:component::mealie::services::get_mealplan::fields::config_entry_id::name%]",
@@ -196,7 +196,7 @@
},
"set_random_mealplan": {
"name": "Set random mealplan",
- "description": "Set a random mealplan for a specific date",
+ "description": "Sets a random mealplan for a specific date",
"fields": {
"config_entry_id": {
"name": "[%key:component::mealie::services::get_mealplan::fields::config_entry_id::name%]",
@@ -214,7 +214,7 @@
},
"set_mealplan": {
"name": "Set a mealplan",
- "description": "Set a mealplan for a specific date",
+ "description": "Sets a mealplan for a specific date",
"fields": {
"config_entry_id": {
"name": "[%key:component::mealie::services::get_mealplan::fields::config_entry_id::name%]",
diff --git a/homeassistant/components/mealie/todo.py b/homeassistant/components/mealie/todo.py
index be04b00113e..d42c9033922 100644
--- a/homeassistant/components/mealie/todo.py
+++ b/homeassistant/components/mealie/todo.py
@@ -14,7 +14,7 @@ from homeassistant.components.todo import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MealieConfigEntry, MealieShoppingListCoordinator
@@ -46,7 +46,7 @@ def _convert_api_item(item: ShoppingItem) -> TodoItem:
async def async_setup_entry(
hass: HomeAssistant,
entry: MealieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the todo platform for entity."""
coordinator = entry.runtime_data.shoppinglist_coordinator
diff --git a/homeassistant/components/meater/sensor.py b/homeassistant/components/meater/sensor.py
index 2a26d848ac2..00fc28b8718 100644
--- a/homeassistant/components/meater/sensor.py
+++ b/homeassistant/components/meater/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -136,7 +136,9 @@ SENSOR_TYPES = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the entry."""
coordinator: DataUpdateCoordinator[dict[str, MeaterProbe]] = hass.data[DOMAIN][
diff --git a/homeassistant/components/medcom_ble/sensor.py b/homeassistant/components/medcom_ble/sensor.py
index b5cb29be845..f837620c829 100644
--- a/homeassistant/components/medcom_ble/sensor.py
+++ b/homeassistant/components/medcom_ble/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -37,7 +37,7 @@ SENSORS_MAPPING_TEMPLATE: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Medcom BLE radiation monitor sensors."""
diff --git a/homeassistant/components/media_extractor/manifest.json b/homeassistant/components/media_extractor/manifest.json
index f0f8ee03ad0..575c0fa878d 100644
--- a/homeassistant/components/media_extractor/manifest.json
+++ b/homeassistant/components/media_extractor/manifest.json
@@ -8,6 +8,6 @@
"iot_class": "calculated",
"loggers": ["yt_dlp"],
"quality_scale": "internal",
- "requirements": ["yt-dlp[default]==2025.01.26"],
+ "requirements": ["yt-dlp[default]==2025.02.19"],
"single_config_entry": true
}
diff --git a/homeassistant/components/media_extractor/strings.json b/homeassistant/components/media_extractor/strings.json
index 125aa08337a..11b5a884e4d 100644
--- a/homeassistant/components/media_extractor/strings.json
+++ b/homeassistant/components/media_extractor/strings.json
@@ -17,12 +17,12 @@
},
"media_content_type": {
"name": "Media content type",
- "description": "The type of the content to play. Must be one of MUSIC, TVSHOW, VIDEO, EPISODE, CHANNEL or PLAYLIST MUSIC."
+ "description": "The type of the content to play."
}
}
},
"extract_media_url": {
- "name": "Get Media URL",
+ "name": "Get media URL",
"description": "Extract media URL from a service.",
"fields": {
"url": {
diff --git a/homeassistant/components/media_player/__init__.py b/homeassistant/components/media_player/__init__.py
index e109b0418c9..45d08bea7ce 100644
--- a/homeassistant/components/media_player/__init__.py
+++ b/homeassistant/components/media_player/__init__.py
@@ -52,7 +52,7 @@ from homeassistant.const import ( # noqa: F401
STATE_PLAYING,
STATE_STANDBY,
)
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, SupportsResponse
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.deprecation import (
@@ -124,6 +124,7 @@ from .const import ( # noqa: F401
CONTENT_AUTH_EXPIRY_TIME,
DOMAIN,
REPEAT_MODES,
+ SERVICE_BROWSE_MEDIA,
SERVICE_CLEAR_PLAYLIST,
SERVICE_JOIN,
SERVICE_PLAY_MEDIA,
@@ -201,6 +202,12 @@ MEDIA_PLAYER_PLAY_MEDIA_SCHEMA = {
vol.Optional(ATTR_MEDIA_EXTRA, default={}): dict,
}
+MEDIA_PLAYER_BROWSE_MEDIA_SCHEMA = {
+ vol.Optional(ATTR_MEDIA_CONTENT_TYPE): cv.string,
+ vol.Optional(ATTR_MEDIA_CONTENT_ID): cv.string,
+}
+
+
ATTR_TO_PROPERTY = [
ATTR_MEDIA_VOLUME_LEVEL,
ATTR_MEDIA_VOLUME_MUTED,
@@ -431,6 +438,15 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"async_play_media",
[MediaPlayerEntityFeature.PLAY_MEDIA],
)
+ component.async_register_entity_service(
+ SERVICE_BROWSE_MEDIA,
+ {
+ vol.Optional(ATTR_MEDIA_CONTENT_TYPE): cv.string,
+ vol.Optional(ATTR_MEDIA_CONTENT_ID): cv.string,
+ },
+ "async_browse_media",
+ supports_response=SupportsResponse.ONLY,
+ )
component.async_register_entity_service(
SERVICE_SHUFFLE_SET,
{vol.Required(ATTR_MEDIA_SHUFFLE): cv.boolean},
@@ -1015,7 +1031,6 @@ class MediaPlayerEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
if self.state in {
MediaPlayerState.OFF,
- MediaPlayerState.IDLE,
MediaPlayerState.STANDBY,
}:
await self.async_turn_on()
diff --git a/homeassistant/components/media_player/const.py b/homeassistant/components/media_player/const.py
index ca2f3307846..387fdb05401 100644
--- a/homeassistant/components/media_player/const.py
+++ b/homeassistant/components/media_player/const.py
@@ -173,6 +173,7 @@ _DEPRECATED_MEDIA_TYPE_VIDEO = DeprecatedConstantEnum(MediaType.VIDEO, "2025.10"
SERVICE_CLEAR_PLAYLIST = "clear_playlist"
SERVICE_JOIN = "join"
SERVICE_PLAY_MEDIA = "play_media"
+SERVICE_BROWSE_MEDIA = "browse_media"
SERVICE_SELECT_SOUND_MODE = "select_sound_mode"
SERVICE_SELECT_SOURCE = "select_source"
SERVICE_UNJOIN = "unjoin"
diff --git a/homeassistant/components/media_player/icons.json b/homeassistant/components/media_player/icons.json
index c11211c38ec..5008ea62d2e 100644
--- a/homeassistant/components/media_player/icons.json
+++ b/homeassistant/components/media_player/icons.json
@@ -32,6 +32,9 @@
}
},
"services": {
+ "browse_media": {
+ "service": "mdi:folder-search"
+ },
"clear_playlist": {
"service": "mdi:playlist-remove"
},
diff --git a/homeassistant/components/media_player/intent.py b/homeassistant/components/media_player/intent.py
index edfab2a668f..af37c0d68bb 100644
--- a/homeassistant/components/media_player/intent.py
+++ b/homeassistant/components/media_player/intent.py
@@ -96,11 +96,16 @@ async def async_setup_intents(hass: HomeAssistant) -> None:
required_states={MediaPlayerState.PLAYING},
required_features=MediaPlayerEntityFeature.VOLUME_SET,
required_slots={
- ATTR_MEDIA_VOLUME_LEVEL: vol.All(
- vol.Coerce(int), vol.Range(min=0, max=100), lambda val: val / 100
- )
+ ATTR_MEDIA_VOLUME_LEVEL: intent.IntentSlotInfo(
+ description="The volume percentage of the media player",
+ value_schema=vol.All(
+ vol.Coerce(int),
+ vol.Range(min=0, max=100),
+ lambda val: val / 100,
+ ),
+ ),
},
- description="Sets the volume of a media player",
+ description="Sets the volume percentage of a media player",
platforms={DOMAIN},
device_classes={MediaPlayerDeviceClass},
),
diff --git a/homeassistant/components/media_player/services.yaml b/homeassistant/components/media_player/services.yaml
index 7338747b545..6b13a6b9c09 100644
--- a/homeassistant/components/media_player/services.yaml
+++ b/homeassistant/components/media_player/services.yaml
@@ -165,6 +165,22 @@ play_media:
selector:
boolean:
+browse_media:
+ target:
+ entity:
+ domain: media_player
+ fields:
+ media_content_type:
+ required: false
+ example: "music"
+ selector:
+ text:
+ media_content_id:
+ required: false
+ example: "A:ALBUMARTIST/Beatles"
+ selector:
+ text:
+
select_source:
target:
entity:
diff --git a/homeassistant/components/media_player/strings.json b/homeassistant/components/media_player/strings.json
index be06ae22cdc..87b5ec692af 100644
--- a/homeassistant/components/media_player/strings.json
+++ b/homeassistant/components/media_player/strings.json
@@ -248,7 +248,7 @@
},
"media_content_type": {
"name": "Content type",
- "description": "The type of the content to play. Such as image, music, tv show, video, episode, channel, or playlist."
+ "description": "The type of the content to play, such as image, music, tv show, video, episode, channel, or playlist."
},
"enqueue": {
"name": "Enqueue",
@@ -260,6 +260,20 @@
}
}
},
+ "browse_media": {
+ "name": "Browse media",
+ "description": "Browses the available media.",
+ "fields": {
+ "media_content_id": {
+ "name": "Content ID",
+ "description": "The ID of the content to browse. Integration dependent."
+ },
+ "media_content_type": {
+ "name": "Content type",
+ "description": "The type of the content to browse, such as image, music, tv show, video, episode, channel, or playlist."
+ }
+ }
+ },
"select_source": {
"name": "Select source",
"description": "Sends the media player the command to change input source.",
@@ -285,22 +299,22 @@
"description": "Removes all items from the playlist."
},
"shuffle_set": {
- "name": "Shuffle",
- "description": "Playback mode that selects the media in randomized order.",
+ "name": "Set shuffle",
+ "description": "Enables or disables the shuffle mode.",
"fields": {
"shuffle": {
- "name": "Shuffle",
- "description": "Whether or not shuffle mode is enabled."
+ "name": "Shuffle mode",
+ "description": "Whether the media should be played in randomized order or not."
}
}
},
"repeat_set": {
- "name": "Repeat",
- "description": "Playback mode that plays the media in a loop.",
+ "name": "Set repeat",
+ "description": "Sets the repeat mode.",
"fields": {
"repeat": {
"name": "Repeat mode",
- "description": "Repeat mode to set."
+ "description": "Whether the media (one or all) should be played in a loop or not."
}
}
},
diff --git a/homeassistant/components/melcloud/climate.py b/homeassistant/components/melcloud/climate.py
index 4defd47bc39..9c2ee60b12c 100644
--- a/homeassistant/components/melcloud/climate.py
+++ b/homeassistant/components/melcloud/climate.py
@@ -28,7 +28,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MelCloudDevice
from .const import (
@@ -76,7 +76,9 @@ ATW_ZONE_HVAC_ACTION_LOOKUP = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MelCloud device climate based on config_entry."""
mel_devices = hass.data[DOMAIN][entry.entry_id]
@@ -150,6 +152,14 @@ class AtaDeviceClimate(MelCloudClimate):
self._attr_unique_id = f"{self.api.device.serial}-{self.api.device.mac}"
self._attr_device_info = self.api.device_info
+ async def async_added_to_hass(self) -> None:
+ """When entity is added to hass."""
+ await super().async_added_to_hass()
+
+ # We can only check for vane_horizontal once we fetch the device data from the cloud
+ if self._device.vane_horizontal:
+ self._attr_supported_features |= ClimateEntityFeature.SWING_HORIZONTAL_MODE
+
@property
def extra_state_attributes(self) -> dict[str, Any] | None:
"""Return the optional state attributes with device specific additions."""
@@ -272,15 +282,29 @@ class AtaDeviceClimate(MelCloudClimate):
"""Return vertical vane position or mode."""
return self._device.vane_vertical
+ @property
+ def swing_horizontal_mode(self) -> str | None:
+ """Return horizontal vane position or mode."""
+ return self._device.vane_horizontal
+
async def async_set_swing_mode(self, swing_mode: str) -> None:
"""Set vertical vane position or mode."""
await self.async_set_vane_vertical(swing_mode)
+ async def async_set_swing_horizontal_mode(self, swing_horizontal_mode: str) -> None:
+ """Set horizontal vane position or mode."""
+ await self.async_set_vane_horizontal(swing_horizontal_mode)
+
@property
def swing_modes(self) -> list[str] | None:
"""Return a list of available vertical vane positions and modes."""
return self._device.vane_vertical_positions
+ @property
+ def swing_horizontal_modes(self) -> list[str] | None:
+ """Return a list of available horizontal vane positions and modes."""
+ return self._device.vane_horizontal_positions
+
async def async_turn_on(self) -> None:
"""Turn the entity on."""
await self._device.set({"power": True})
diff --git a/homeassistant/components/melcloud/sensor.py b/homeassistant/components/melcloud/sensor.py
index 84585c556ca..51a026e717a 100644
--- a/homeassistant/components/melcloud/sensor.py
+++ b/homeassistant/components/melcloud/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfEnergy, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MelCloudDevice
from .const import DOMAIN
@@ -104,7 +104,9 @@ ATW_ZONE_SENSORS: tuple[MelcloudSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MELCloud device sensors based on config_entry."""
mel_devices = hass.data[DOMAIN].get(entry.entry_id)
diff --git a/homeassistant/components/melcloud/strings.json b/homeassistant/components/melcloud/strings.json
index 19ef0b76aad..8c168295e88 100644
--- a/homeassistant/components/melcloud/strings.json
+++ b/homeassistant/components/melcloud/strings.json
@@ -11,20 +11,20 @@
},
"reauth_confirm": {
"title": "[%key:common::config_flow::title::reauth%]",
- "description": "The Melcloud integration needs to re-authenticate your connection details",
+ "description": "The MELCloud integration needs to re-authenticate your connection details",
"data": {
"username": "[%key:common::config_flow::data::email%]",
"password": "[%key:common::config_flow::data::password%]"
}
},
"reconfigure": {
- "title": "Reconfigure your MelCloud",
+ "title": "Reconfigure your MELCloud",
"description": "Reconfigure the entry to obtain a new token, for your account: `{username}`.",
"data": {
"password": "[%key:common::config_flow::data::password%]"
},
"data_description": {
- "password": "Enter the (new) password for MelCloud."
+ "password": "Enter the (new) password for MELCloud."
}
}
},
@@ -70,7 +70,7 @@
},
"deprecated_yaml_import_issue_cannot_connect": {
"title": "The MELCloud YAML configuration import failed",
- "description": "Configuring MELCloud using YAML is being removed but there was an connection error importing your YAML configuration.\n\nEnsure connection to MELCloud works and restart Home Assistant to try again or remove the MELCloud YAML configuration from your configuration.yaml file and continue to [set up the integration](/config/integrations/dashboard/add?domain=melcoud) manually."
+ "description": "Configuring MELCloud using YAML is being removed but there was a connection error importing your YAML configuration.\n\nEnsure connection to MELCloud works and restart Home Assistant to try again or remove the MELCloud YAML configuration from your configuration.yaml file and continue to [set up the integration](/config/integrations/dashboard/add?domain=melcoud) manually."
}
},
"entity": {
diff --git a/homeassistant/components/melcloud/water_heater.py b/homeassistant/components/melcloud/water_heater.py
index 8de1ac53311..76fbad41575 100644
--- a/homeassistant/components/melcloud/water_heater.py
+++ b/homeassistant/components/melcloud/water_heater.py
@@ -20,14 +20,16 @@ from homeassistant.components.water_heater import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, MelCloudDevice
from .const import ATTR_STATUS
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MelCloud device climate based on config_entry."""
mel_devices = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/melnor/__init__.py b/homeassistant/components/melnor/__init__.py
index afaf8eb95f8..6ab725d747c 100644
--- a/homeassistant/components/melnor/__init__.py
+++ b/homeassistant/components/melnor/__init__.py
@@ -57,7 +57,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
bluetooth.BluetoothScanningMode.PASSIVE,
)
- coordinator = MelnorDataUpdateCoordinator(hass, device)
+ coordinator = MelnorDataUpdateCoordinator(hass, entry, device)
await coordinator.async_config_entry_first_refresh()
hass.data[DOMAIN][entry.entry_id] = coordinator
diff --git a/homeassistant/components/melnor/coordinator.py b/homeassistant/components/melnor/coordinator.py
index 669fe916082..52662fd0c4c 100644
--- a/homeassistant/components/melnor/coordinator.py
+++ b/homeassistant/components/melnor/coordinator.py
@@ -5,6 +5,7 @@ import logging
from melnor_bluetooth.device import Device
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -14,13 +15,17 @@ _LOGGER = logging.getLogger(__name__)
class MelnorDataUpdateCoordinator(DataUpdateCoordinator[Device]):
"""Melnor data update coordinator."""
+ config_entry: ConfigEntry
_device: Device
- def __init__(self, hass: HomeAssistant, device: Device) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, device: Device
+ ) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Melnor Bluetooth",
update_interval=timedelta(seconds=5),
)
diff --git a/homeassistant/components/melnor/number.py b/homeassistant/components/melnor/number.py
index 15c47008346..42c22ae5a43 100644
--- a/homeassistant/components/melnor/number.py
+++ b/homeassistant/components/melnor/number.py
@@ -16,7 +16,7 @@ from homeassistant.components.number import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MelnorDataUpdateCoordinator
@@ -68,7 +68,7 @@ ZONE_ENTITY_DESCRIPTIONS: list[MelnorZoneNumberEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the number platform."""
diff --git a/homeassistant/components/melnor/sensor.py b/homeassistant/components/melnor/sensor.py
index bbb3416dcc9..525a29dc6cf 100644
--- a/homeassistant/components/melnor/sensor.py
+++ b/homeassistant/components/melnor/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
EntityCategory,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -105,7 +105,7 @@ ZONE_ENTITY_DESCRIPTIONS: list[MelnorZoneSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
diff --git a/homeassistant/components/melnor/switch.py b/homeassistant/components/melnor/switch.py
index d7fb96739b3..cc5abe8f6f3 100644
--- a/homeassistant/components/melnor/switch.py
+++ b/homeassistant/components/melnor/switch.py
@@ -15,7 +15,7 @@ from homeassistant.components.switch import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MelnorDataUpdateCoordinator
@@ -52,7 +52,7 @@ ZONE_ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switch platform."""
diff --git a/homeassistant/components/melnor/time.py b/homeassistant/components/melnor/time.py
index 08de7e054de..277eb6e36eb 100644
--- a/homeassistant/components/melnor/time.py
+++ b/homeassistant/components/melnor/time.py
@@ -13,7 +13,7 @@ from homeassistant.components.time import TimeEntity, TimeEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MelnorDataUpdateCoordinator
@@ -42,7 +42,7 @@ ZONE_ENTITY_DESCRIPTIONS: list[MelnorZoneTimeEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the number platform."""
diff --git a/homeassistant/components/met/__init__.py b/homeassistant/components/met/__init__.py
index 1cd7a4bde57..17fc411bf20 100644
--- a/homeassistant/components/met/__init__.py
+++ b/homeassistant/components/met/__init__.py
@@ -4,7 +4,6 @@ from __future__ import annotations
import logging
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
@@ -15,14 +14,12 @@ from .const import (
DEFAULT_HOME_LONGITUDE,
DOMAIN,
)
-from .coordinator import MetDataUpdateCoordinator
+from .coordinator import MetDataUpdateCoordinator, MetWeatherConfigEntry
PLATFORMS = [Platform.WEATHER]
_LOGGER = logging.getLogger(__name__)
-type MetWeatherConfigEntry = ConfigEntry[MetDataUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, config_entry: MetWeatherConfigEntry
diff --git a/homeassistant/components/met/coordinator.py b/homeassistant/components/met/coordinator.py
index 3887a29f83c..de27da7a07f 100644
--- a/homeassistant/components/met/coordinator.py
+++ b/homeassistant/components/met/coordinator.py
@@ -31,6 +31,8 @@ URL = "https://aa015h6buqvih86i1.api.met.no/weatherapi/locationforecast/2.0/comp
_LOGGER = logging.getLogger(__name__)
+type MetWeatherConfigEntry = ConfigEntry[MetDataUpdateCoordinator]
+
class CannotConnect(HomeAssistantError):
"""Unable to connect to the web site."""
@@ -89,7 +91,11 @@ class MetWeatherData:
class MetDataUpdateCoordinator(DataUpdateCoordinator[MetWeatherData]):
"""Class to manage fetching Met data."""
- def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
+ config_entry: MetWeatherConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: MetWeatherConfigEntry
+ ) -> None:
"""Initialize global Met data updater."""
self._unsub_track_home: Callable[[], None] | None = None
self.weather = MetWeatherData(hass, config_entry.data)
@@ -97,7 +103,13 @@ class MetDataUpdateCoordinator(DataUpdateCoordinator[MetWeatherData]):
update_interval = timedelta(minutes=randrange(55, 65))
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=update_interval)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=update_interval,
+ )
async def _async_update_data(self) -> MetWeatherData:
"""Fetch data from Met."""
diff --git a/homeassistant/components/met/weather.py b/homeassistant/components/met/weather.py
index 7b95567366b..c4f9c8e6885 100644
--- a/homeassistant/components/met/weather.py
+++ b/homeassistant/components/met/weather.py
@@ -34,10 +34,9 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er, sun
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_system import METRIC_SYSTEM
-from . import MetWeatherConfigEntry
from .const import (
ATTR_CONDITION_CLEAR_NIGHT,
ATTR_CONDITION_SUNNY,
@@ -47,7 +46,7 @@ from .const import (
DOMAIN,
FORECAST_MAP,
)
-from .coordinator import MetDataUpdateCoordinator
+from .coordinator import MetDataUpdateCoordinator, MetWeatherConfigEntry
DEFAULT_NAME = "Met.no"
@@ -55,7 +54,7 @@ DEFAULT_NAME = "Met.no"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MetWeatherConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/met_eireann/weather.py b/homeassistant/components/met_eireann/weather.py
index 404ef5d8393..72706ccb70f 100644
--- a/homeassistant/components/met_eireann/weather.py
+++ b/homeassistant/components/met_eireann/weather.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from homeassistant.util import dt as dt_util
@@ -47,7 +47,7 @@ def format_condition(condition: str | None) -> str | None:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
coordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/meteo_france/sensor.py b/homeassistant/components/meteo_france/sensor.py
index 826716f1679..c29cc1ceda9 100644
--- a/homeassistant/components/meteo_france/sensor.py
+++ b/homeassistant/components/meteo_france/sensor.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -182,7 +182,9 @@ SENSOR_TYPES_PROBABILITY: tuple[MeteoFranceSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Meteo-France sensor platform."""
data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/meteo_france/weather.py b/homeassistant/components/meteo_france/weather.py
index 8305547afd3..67a56271c2b 100644
--- a/homeassistant/components/meteo_france/weather.py
+++ b/homeassistant/components/meteo_france/weather.py
@@ -28,7 +28,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -55,7 +55,9 @@ def format_condition(condition: str):
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Meteo-France weather platform."""
coordinator: DataUpdateCoordinator[MeteoFranceForecast] = hass.data[DOMAIN][
diff --git a/homeassistant/components/meteoclimatic/sensor.py b/homeassistant/components/meteoclimatic/sensor.py
index 2194f82e43e..169da7a0a18 100644
--- a/homeassistant/components/meteoclimatic/sensor.py
+++ b/homeassistant/components/meteoclimatic/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -101,6 +101,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
name="Wind Bearing",
native_unit_of_measurement=DEGREE,
icon="mdi:weather-windy",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
SensorEntityDescription(
key="rain",
@@ -112,7 +113,9 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Meteoclimatic sensor platform."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/meteoclimatic/weather.py b/homeassistant/components/meteoclimatic/weather.py
index 75a93689efa..fa3b3c92288 100644
--- a/homeassistant/components/meteoclimatic/weather.py
+++ b/homeassistant/components/meteoclimatic/weather.py
@@ -7,7 +7,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfPressure, UnitOfSpeed, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -26,7 +26,9 @@ def format_condition(condition):
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Meteoclimatic weather platform."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/metoffice/sensor.py b/homeassistant/components/metoffice/sensor.py
index 61f825abdc3..5a256144d11 100644
--- a/homeassistant/components/metoffice/sensor.py
+++ b/homeassistant/components/metoffice/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
@@ -142,7 +142,9 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Met Office weather sensor platform."""
hass_data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/metoffice/weather.py b/homeassistant/components/metoffice/weather.py
index 5eeddee8dd4..d3f1320c47e 100644
--- a/homeassistant/components/metoffice/weather.py
+++ b/homeassistant/components/metoffice/weather.py
@@ -21,7 +21,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfPressure, UnitOfSpeed, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import TimestampDataUpdateCoordinator
from . import get_device_info
@@ -39,7 +39,9 @@ from .data import MetOfficeData
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Met Office weather sensor platform."""
entity_registry = er.async_get(hass)
diff --git a/homeassistant/components/microbees/__init__.py b/homeassistant/components/microbees/__init__.py
index 488988ab593..12c536121da 100644
--- a/homeassistant/components/microbees/__init__.py
+++ b/homeassistant/components/microbees/__init__.py
@@ -45,7 +45,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
raise ConfigEntryAuthFailed("Token not valid, trigger renewal") from ex
raise ConfigEntryNotReady from ex
microbees = MicroBees(token=session.token[CONF_ACCESS_TOKEN])
- coordinator = MicroBeesUpdateCoordinator(hass, microbees)
+ coordinator = MicroBeesUpdateCoordinator(hass, entry, microbees)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = HomeAssistantMicroBeesData(
connector=microbees,
diff --git a/homeassistant/components/microbees/binary_sensor.py b/homeassistant/components/microbees/binary_sensor.py
index 551f68ba354..1dc2a8d9702 100644
--- a/homeassistant/components/microbees/binary_sensor.py
+++ b/homeassistant/components/microbees/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MicroBeesUpdateCoordinator
@@ -36,7 +36,9 @@ BINARYSENSOR_TYPES = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the microBees binary sensor platform."""
coordinator: MicroBeesUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/microbees/button.py b/homeassistant/components/microbees/button.py
index f449fa9afee..ca3a76753a7 100644
--- a/homeassistant/components/microbees/button.py
+++ b/homeassistant/components/microbees/button.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MicroBeesUpdateCoordinator
@@ -15,7 +15,9 @@ BUTTON_TRANSLATIONS = {51: "button_gate", 91: "button_panic"}
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the microBees button platform."""
coordinator: MicroBeesUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/microbees/climate.py b/homeassistant/components/microbees/climate.py
index 077048ee352..554ca3b32cc 100644
--- a/homeassistant/components/microbees/climate.py
+++ b/homeassistant/components/microbees/climate.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MicroBeesUpdateCoordinator
@@ -26,7 +26,9 @@ THERMOVALVE_SENSOR_ID = 782
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the microBees climate platform."""
coordinator: MicroBeesUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/microbees/coordinator.py b/homeassistant/components/microbees/coordinator.py
index af207507e77..0094dc33e81 100644
--- a/homeassistant/components/microbees/coordinator.py
+++ b/homeassistant/components/microbees/coordinator.py
@@ -9,6 +9,7 @@ import logging
import aiohttp
from microBeesPy import Actuator, Bee, MicroBees, MicroBeesException, Sensor
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -28,11 +29,16 @@ class MicroBeesCoordinatorData:
class MicroBeesUpdateCoordinator(DataUpdateCoordinator[MicroBeesCoordinatorData]):
"""MicroBees coordinator."""
- def __init__(self, hass: HomeAssistant, microbees: MicroBees) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, microbees: MicroBees
+ ) -> None:
"""Initialize microBees coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="microBees Coordinator",
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/microbees/cover.py b/homeassistant/components/microbees/cover.py
index b6d5d366d89..fe87fcddd62 100644
--- a/homeassistant/components/microbees/cover.py
+++ b/homeassistant/components/microbees/cover.py
@@ -12,7 +12,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from .const import DOMAIN
@@ -23,7 +23,9 @@ COVER_IDS = {47: "roller_shutter"}
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the microBees cover platform."""
coordinator: MicroBeesUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/microbees/light.py b/homeassistant/components/microbees/light.py
index 654cdc37182..a7ff60dc64a 100644
--- a/homeassistant/components/microbees/light.py
+++ b/homeassistant/components/microbees/light.py
@@ -6,7 +6,7 @@ from homeassistant.components.light import ATTR_RGBW_COLOR, ColorMode, LightEnti
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MicroBeesUpdateCoordinator
@@ -14,7 +14,9 @@ from .entity import MicroBeesActuatorEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Config entry."""
coordinator: MicroBeesUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/microbees/sensor.py b/homeassistant/components/microbees/sensor.py
index 360422de735..e4be463ab10 100644
--- a/homeassistant/components/microbees/sensor.py
+++ b/homeassistant/components/microbees/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MicroBeesUpdateCoordinator
@@ -63,7 +63,9 @@ SENSOR_TYPES = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id].coordinator
diff --git a/homeassistant/components/microbees/switch.py b/homeassistant/components/microbees/switch.py
index 1d668d041e1..deda2d78d09 100644
--- a/homeassistant/components/microbees/switch.py
+++ b/homeassistant/components/microbees/switch.py
@@ -6,7 +6,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MicroBeesUpdateCoordinator
@@ -17,7 +17,9 @@ SWITCH_PRODUCT_IDS = {25, 26, 27, 35, 38, 46, 63, 64, 65, 86}
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id].coordinator
diff --git a/homeassistant/components/mikrotik/__init__.py b/homeassistant/components/mikrotik/__init__.py
index cecf96a6c3e..4e17653c05a 100644
--- a/homeassistant/components/mikrotik/__init__.py
+++ b/homeassistant/components/mikrotik/__init__.py
@@ -1,19 +1,16 @@
"""The Mikrotik component."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
from .const import ATTR_MANUFACTURER, DOMAIN
-from .coordinator import MikrotikDataUpdateCoordinator, get_api
+from .coordinator import MikrotikConfigEntry, MikrotikDataUpdateCoordinator, get_api
from .errors import CannotConnect, LoginError
PLATFORMS = [Platform.DEVICE_TRACKER]
-type MikrotikConfigEntry = ConfigEntry[MikrotikDataUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, config_entry: MikrotikConfigEntry
@@ -47,6 +44,8 @@ async def async_setup_entry(
return True
-async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, config_entry: MikrotikConfigEntry
+) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
diff --git a/homeassistant/components/mikrotik/coordinator.py b/homeassistant/components/mikrotik/coordinator.py
index 6cb36d58fbe..c68b13eeca8 100644
--- a/homeassistant/components/mikrotik/coordinator.py
+++ b/homeassistant/components/mikrotik/coordinator.py
@@ -45,6 +45,8 @@ from .errors import CannotConnect, LoginError
_LOGGER = logging.getLogger(__name__)
+type MikrotikConfigEntry = ConfigEntry[MikrotikDataUpdateCoordinator]
+
class MikrotikData:
"""Handle all communication with the Mikrotik API."""
@@ -246,17 +248,21 @@ class MikrotikData:
class MikrotikDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Mikrotik Hub Object."""
+ config_entry: MikrotikConfigEntry
+
def __init__(
- self, hass: HomeAssistant, config_entry: ConfigEntry, api: librouteros.Api
+ self,
+ hass: HomeAssistant,
+ config_entry: MikrotikConfigEntry,
+ api: librouteros.Api,
) -> None:
"""Initialize the Mikrotik Client."""
- self.hass = hass
- self.config_entry: ConfigEntry = config_entry
- self._mk_data = MikrotikData(self.hass, self.config_entry, api)
+ self._mk_data = MikrotikData(hass, config_entry, api)
super().__init__(
- self.hass,
+ hass,
_LOGGER,
- name=f"{DOMAIN} - {self.host}",
+ config_entry=config_entry,
+ name=f"{DOMAIN} - {config_entry.data[CONF_HOST]}",
update_interval=timedelta(seconds=10),
)
diff --git a/homeassistant/components/mikrotik/device_tracker.py b/homeassistant/components/mikrotik/device_tracker.py
index 19d5c789c09..f7bc10e31d4 100644
--- a/homeassistant/components/mikrotik/device_tracker.py
+++ b/homeassistant/components/mikrotik/device_tracker.py
@@ -10,18 +10,17 @@ from homeassistant.components.device_tracker import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
-from . import MikrotikConfigEntry
-from .coordinator import Device, MikrotikDataUpdateCoordinator
+from .coordinator import Device, MikrotikConfigEntry, MikrotikDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MikrotikConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Mikrotik component."""
coordinator = config_entry.runtime_data
@@ -55,7 +54,7 @@ async def async_setup_entry(
@callback
def update_items(
coordinator: MikrotikDataUpdateCoordinator,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
tracked: dict[str, MikrotikDataUpdateCoordinatorTracker],
) -> None:
"""Update tracked device state from the hub."""
diff --git a/homeassistant/components/mill/__init__.py b/homeassistant/components/mill/__init__.py
index 116b3ef0341..2fcf2033930 100644
--- a/homeassistant/components/mill/__init__.py
+++ b/homeassistant/components/mill/__init__.py
@@ -47,9 +47,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except TimeoutError as error:
raise ConfigEntryNotReady from error
data_coordinator = MillDataUpdateCoordinator(
- hass,
- mill_data_connection=mill_data_connection,
- update_interval=update_interval,
+ hass, entry, mill_data_connection, update_interval
)
await data_coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/mill/climate.py b/homeassistant/components/mill/climate.py
index 3cd9247c63a..ba496923a30 100644
--- a/homeassistant/components/mill/climate.py
+++ b/homeassistant/components/mill/climate.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -54,7 +54,9 @@ SET_ROOM_TEMP_SCHEMA = vol.Schema(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Mill climate."""
if entry.data.get(CONNECTION_TYPE) == LOCAL:
diff --git a/homeassistant/components/mill/coordinator.py b/homeassistant/components/mill/coordinator.py
index 9821519ca84..ae527f8cce5 100644
--- a/homeassistant/components/mill/coordinator.py
+++ b/homeassistant/components/mill/coordinator.py
@@ -8,6 +8,7 @@ import logging
from mill import Mill
from mill_local import Mill as MillLocal
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -19,12 +20,14 @@ _LOGGER = logging.getLogger(__name__)
class MillDataUpdateCoordinator(DataUpdateCoordinator):
"""Class to manage fetching Mill data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- update_interval: timedelta | None = None,
- *,
+ config_entry: ConfigEntry,
mill_data_connection: Mill | MillLocal,
+ update_interval: timedelta,
) -> None:
"""Initialize global Mill data updater."""
self.mill_data_connection = mill_data_connection
@@ -32,6 +35,7 @@ class MillDataUpdateCoordinator(DataUpdateCoordinator):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_method=mill_data_connection.fetch_heater_and_sensor_data,
update_interval=update_interval,
diff --git a/homeassistant/components/mill/number.py b/homeassistant/components/mill/number.py
index b4ef7bdd2c2..8433a9853c6 100644
--- a/homeassistant/components/mill/number.py
+++ b/homeassistant/components/mill/number.py
@@ -8,7 +8,7 @@ from homeassistant.components.number import NumberDeviceClass, NumberEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_USERNAME, UnitOfPower
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CLOUD, CONNECTION_TYPE, DOMAIN
from .coordinator import MillDataUpdateCoordinator
@@ -16,7 +16,9 @@ from .entity import MillBaseEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Mill Number."""
if entry.data.get(CONNECTION_TYPE) == CLOUD:
diff --git a/homeassistant/components/mill/sensor.py b/homeassistant/components/mill/sensor.py
index 57eead9be18..3a47cb427d2 100644
--- a/homeassistant/components/mill/sensor.py
+++ b/homeassistant/components/mill/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -146,7 +146,9 @@ SOCKET_SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Mill sensor."""
if entry.data.get(CONNECTION_TYPE) == LOCAL:
diff --git a/homeassistant/components/min_max/sensor.py b/homeassistant/components/min_max/sensor.py
index 89252a58864..9039c3e9e24 100644
--- a/homeassistant/components/min_max/sensor.py
+++ b/homeassistant/components/min_max/sensor.py
@@ -25,7 +25,10 @@ from homeassistant.const import (
)
from homeassistant.core import Event, EventStateChangedData, HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.reload import async_setup_reload_service
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, StateType
@@ -75,7 +78,7 @@ PLATFORM_SCHEMA = SENSOR_PLATFORM_SCHEMA.extend(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize min/max/mean config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/minecraft_server/__init__.py b/homeassistant/components/minecraft_server/__init__.py
index f1392ea488a..d8f60380a6c 100644
--- a/homeassistant/components/minecraft_server/__init__.py
+++ b/homeassistant/components/minecraft_server/__init__.py
@@ -9,22 +9,13 @@ import dns.rdata
import dns.rdataclass
import dns.rdatatype
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import (
- CONF_ADDRESS,
- CONF_HOST,
- CONF_NAME,
- CONF_PORT,
- CONF_TYPE,
- Platform,
-)
+from homeassistant.const import CONF_ADDRESS, CONF_HOST, CONF_PORT, Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr, entity_registry as er
from .api import MinecraftServer, MinecraftServerAddressError, MinecraftServerType
from .const import DOMAIN, KEY_LATENCY, KEY_MOTD
-from .coordinator import MinecraftServerCoordinator
+from .coordinator import MinecraftServerConfigEntry, MinecraftServerCoordinator
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR]
@@ -38,32 +29,18 @@ def load_dnspython_rdata_classes() -> None:
dns.rdata.get_rdata_class(dns.rdataclass.IN, rdtype) # type: ignore[no-untyped-call]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(
+ hass: HomeAssistant, entry: MinecraftServerConfigEntry
+) -> bool:
"""Set up Minecraft Server from a config entry."""
# Workaround to avoid blocking imports from dnspython (https://github.com/rthalley/dnspython/issues/1083)
await hass.async_add_executor_job(load_dnspython_rdata_classes)
- # Create API instance.
- api = MinecraftServer(
- hass,
- entry.data.get(CONF_TYPE, MinecraftServerType.JAVA_EDITION),
- entry.data[CONF_ADDRESS],
- )
-
- # Initialize API instance.
- try:
- await api.async_initialize()
- except MinecraftServerAddressError as error:
- raise ConfigEntryNotReady(f"Initialization failed: {error}") from error
-
- # Create coordinator instance.
- coordinator = MinecraftServerCoordinator(hass, entry.data[CONF_NAME], api)
+ # Create coordinator instance and store it.
+ coordinator = MinecraftServerCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
-
- # Store coordinator instance.
- domain_data = hass.data.setdefault(DOMAIN, {})
- domain_data[entry.entry_id] = coordinator
+ entry.runtime_data = coordinator
# Set up platforms.
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
@@ -71,21 +48,16 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True
-async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, config_entry: MinecraftServerConfigEntry
+) -> bool:
"""Unload Minecraft Server config entry."""
-
- # Unload platforms.
- unload_ok = await hass.config_entries.async_unload_platforms(
- config_entry, PLATFORMS
- )
-
- # Clean up.
- hass.data[DOMAIN].pop(config_entry.entry_id)
-
- return unload_ok
+ return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
-async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, config_entry: MinecraftServerConfigEntry
+) -> bool:
"""Migrate old config entry to a new format."""
# 1 --> 2: Use config entry ID as base for unique IDs.
@@ -159,7 +131,9 @@ async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
async def _async_migrate_device_identifiers(
- hass: HomeAssistant, config_entry: ConfigEntry, old_unique_id: str | None
+ hass: HomeAssistant,
+ config_entry: MinecraftServerConfigEntry,
+ old_unique_id: str | None,
) -> None:
"""Migrate the device identifiers to the new format."""
device_registry = dr.async_get(hass)
diff --git a/homeassistant/components/minecraft_server/binary_sensor.py b/homeassistant/components/minecraft_server/binary_sensor.py
index 60f2e00da0e..a7279040a6d 100644
--- a/homeassistant/components/minecraft_server/binary_sensor.py
+++ b/homeassistant/components/minecraft_server/binary_sensor.py
@@ -5,12 +5,10 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import MinecraftServerCoordinator
+from .coordinator import MinecraftServerConfigEntry, MinecraftServerCoordinator
from .entity import MinecraftServerEntity
KEY_STATUS = "status"
@@ -24,14 +22,17 @@ BINARY_SENSOR_DESCRIPTIONS = [
),
]
+# Coordinator is used to centralize the data updates.
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: MinecraftServerConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Minecraft Server binary sensor platform."""
- coordinator = hass.data[DOMAIN][config_entry.entry_id]
+ coordinator = config_entry.runtime_data
# Add binary sensor entities.
async_add_entities(
@@ -49,7 +50,7 @@ class MinecraftServerBinarySensorEntity(MinecraftServerEntity, BinarySensorEntit
self,
coordinator: MinecraftServerCoordinator,
description: BinarySensorEntityDescription,
- config_entry: ConfigEntry,
+ config_entry: MinecraftServerConfigEntry,
) -> None:
"""Initialize binary sensor base entity."""
super().__init__(coordinator, config_entry)
diff --git a/homeassistant/components/minecraft_server/config_flow.py b/homeassistant/components/minecraft_server/config_flow.py
index 3ffdc33f3b2..d0f7cf5a8fb 100644
--- a/homeassistant/components/minecraft_server/config_flow.py
+++ b/homeassistant/components/minecraft_server/config_flow.py
@@ -8,10 +8,10 @@ from typing import Any
import voluptuous as vol
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
-from homeassistant.const import CONF_ADDRESS, CONF_NAME, CONF_TYPE
+from homeassistant.const import CONF_ADDRESS, CONF_TYPE
from .api import MinecraftServer, MinecraftServerAddressError, MinecraftServerType
-from .const import DEFAULT_NAME, DOMAIN
+from .const import DOMAIN
DEFAULT_ADDRESS = "localhost:25565"
@@ -37,7 +37,6 @@ class MinecraftServerConfigFlow(ConfigFlow, domain=DOMAIN):
# Prepare config entry data.
config_data = {
- CONF_NAME: user_input[CONF_NAME],
CONF_ADDRESS: address,
}
@@ -78,9 +77,6 @@ class MinecraftServerConfigFlow(ConfigFlow, domain=DOMAIN):
step_id="user",
data_schema=vol.Schema(
{
- vol.Required(
- CONF_NAME, default=user_input.get(CONF_NAME, DEFAULT_NAME)
- ): str,
vol.Required(
CONF_ADDRESS,
default=user_input.get(CONF_ADDRESS, DEFAULT_ADDRESS),
diff --git a/homeassistant/components/minecraft_server/const.py b/homeassistant/components/minecraft_server/const.py
index e7a58741696..35a1c0dd5a5 100644
--- a/homeassistant/components/minecraft_server/const.py
+++ b/homeassistant/components/minecraft_server/const.py
@@ -1,7 +1,5 @@
"""Constants for the Minecraft Server integration."""
-DEFAULT_NAME = "Minecraft Server"
-
DOMAIN = "minecraft_server"
KEY_LATENCY = "latency"
diff --git a/homeassistant/components/minecraft_server/coordinator.py b/homeassistant/components/minecraft_server/coordinator.py
index 37eeb9f2ac2..457b0700535 100644
--- a/homeassistant/components/minecraft_server/coordinator.py
+++ b/homeassistant/components/minecraft_server/coordinator.py
@@ -5,16 +5,23 @@ from __future__ import annotations
from datetime import timedelta
import logging
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_ADDRESS, CONF_TYPE
from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .api import (
MinecraftServer,
+ MinecraftServerAddressError,
MinecraftServerConnectionError,
MinecraftServerData,
MinecraftServerNotInitializedError,
+ MinecraftServerType,
)
+type MinecraftServerConfigEntry = ConfigEntry[MinecraftServerCoordinator]
+
SCAN_INTERVAL = timedelta(seconds=60)
_LOGGER = logging.getLogger(__name__)
@@ -23,17 +30,40 @@ _LOGGER = logging.getLogger(__name__)
class MinecraftServerCoordinator(DataUpdateCoordinator[MinecraftServerData]):
"""Minecraft Server data update coordinator."""
- def __init__(self, hass: HomeAssistant, name: str, api: MinecraftServer) -> None:
+ config_entry: MinecraftServerConfigEntry
+ _api: MinecraftServer
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: MinecraftServerConfigEntry,
+ ) -> None:
"""Initialize coordinator instance."""
- self._api = api
super().__init__(
hass=hass,
- name=name,
+ name=config_entry.title,
+ config_entry=config_entry,
logger=_LOGGER,
update_interval=SCAN_INTERVAL,
)
+ async def _async_setup(self) -> None:
+ """Set up the Minecraft Server data coordinator."""
+
+ # Create API instance.
+ self._api = MinecraftServer(
+ self.hass,
+ self.config_entry.data.get(CONF_TYPE, MinecraftServerType.JAVA_EDITION),
+ self.config_entry.data[CONF_ADDRESS],
+ )
+
+ # Initialize API instance.
+ try:
+ await self._api.async_initialize()
+ except MinecraftServerAddressError as error:
+ raise ConfigEntryNotReady(f"Initialization failed: {error}") from error
+
async def _async_update_data(self) -> MinecraftServerData:
"""Get updated data from the server."""
try:
diff --git a/homeassistant/components/minecraft_server/diagnostics.py b/homeassistant/components/minecraft_server/diagnostics.py
index 0bcffe1434a..dd94411b969 100644
--- a/homeassistant/components/minecraft_server/diagnostics.py
+++ b/homeassistant/components/minecraft_server/diagnostics.py
@@ -5,20 +5,19 @@ from dataclasses import asdict
from typing import Any
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_ADDRESS, CONF_NAME
+from homeassistant.const import CONF_ADDRESS
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
+from .coordinator import MinecraftServerConfigEntry
-TO_REDACT: Iterable[Any] = {CONF_ADDRESS, CONF_NAME, "players_list"}
+TO_REDACT: Iterable[Any] = {CONF_ADDRESS, "players_list"}
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, config_entry: ConfigEntry
+ hass: HomeAssistant, config_entry: MinecraftServerConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
- coordinator = hass.data[DOMAIN][config_entry.entry_id]
+ coordinator = config_entry.runtime_data
return {
"config_entry": {
diff --git a/homeassistant/components/minecraft_server/manifest.json b/homeassistant/components/minecraft_server/manifest.json
index d6ade4853c9..be399a3c8dc 100644
--- a/homeassistant/components/minecraft_server/manifest.json
+++ b/homeassistant/components/minecraft_server/manifest.json
@@ -6,5 +6,6 @@
"documentation": "https://www.home-assistant.io/integrations/minecraft_server",
"iot_class": "local_polling",
"loggers": ["dnspython", "mcstatus"],
+ "quality_scale": "silver",
"requirements": ["mcstatus==11.1.1"]
}
diff --git a/homeassistant/components/minecraft_server/quality_scale.yaml b/homeassistant/components/minecraft_server/quality_scale.yaml
index fc3db3b3075..288e58fad39 100644
--- a/homeassistant/components/minecraft_server/quality_scale.yaml
+++ b/homeassistant/components/minecraft_server/quality_scale.yaml
@@ -6,22 +6,15 @@ rules:
appropriate-polling: done
brands: done
common-modules: done
- config-flow:
- status: todo
- comment: Check removal and replacement of name in config flow with the title (server address).
- config-flow-test-coverage:
- status: todo
- comment: |
- Merge test_show_config_form with full flow test.
- Move full flow test to the top of all tests.
- All test cases should end in either CREATE_ENTRY or ABORT.
+ config-flow: done
+ config-flow-test-coverage: done
dependency-transparency: done
docs-actions:
status: exempt
comment: Integration doesn't provide any service actions.
docs-high-level-description: done
docs-installation-instructions: done
- docs-removal-instructions: todo
+ docs-removal-instructions: done
entity-event-setup:
status: done
comment: Handled by coordinator.
@@ -29,7 +22,7 @@ rules:
status: done
comment: Using confid entry ID as the dependency mcstatus doesn't provide a unique information.
has-entity-name: done
- runtime-data: todo
+ runtime-data: done
test-before-configure: done
test-before-setup:
status: done
@@ -58,11 +51,7 @@ rules:
log-when-unavailable:
status: done
comment: Handled by coordinator.
- parallel-updates:
- status: todo
- comment: |
- Although this is handled by the coordinator and no service actions are provided,
- PARALLEL_UPDATES should still be set to 0 in binary_sensor and sensor according to the rule.
+ parallel-updates: done
reauthentication-flow:
status: exempt
comment: No authentication is required for the integration.
diff --git a/homeassistant/components/minecraft_server/sensor.py b/homeassistant/components/minecraft_server/sensor.py
index fae004a015e..cfc16c7724d 100644
--- a/homeassistant/components/minecraft_server/sensor.py
+++ b/homeassistant/components/minecraft_server/sensor.py
@@ -7,15 +7,14 @@ from dataclasses import dataclass
from typing import Any
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TYPE, EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .api import MinecraftServerData, MinecraftServerType
-from .const import DOMAIN, KEY_LATENCY, KEY_MOTD
-from .coordinator import MinecraftServerCoordinator
+from .const import KEY_LATENCY, KEY_MOTD
+from .coordinator import MinecraftServerConfigEntry, MinecraftServerCoordinator
from .entity import MinecraftServerEntity
ATTR_PLAYERS_LIST = "players_list"
@@ -31,6 +30,9 @@ KEY_VERSION = "version"
UNIT_PLAYERS_MAX = "players"
UNIT_PLAYERS_ONLINE = "players"
+# Coordinator is used to centralize the data updates.
+PARALLEL_UPDATES = 0
+
@dataclass(frozen=True, kw_only=True)
class MinecraftServerSensorEntityDescription(SensorEntityDescription):
@@ -158,11 +160,11 @@ SENSOR_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: MinecraftServerConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Minecraft Server sensor platform."""
- coordinator = hass.data[DOMAIN][config_entry.entry_id]
+ coordinator = config_entry.runtime_data
# Add sensor entities.
async_add_entities(
@@ -184,7 +186,7 @@ class MinecraftServerSensorEntity(MinecraftServerEntity, SensorEntity):
self,
coordinator: MinecraftServerCoordinator,
description: MinecraftServerSensorEntityDescription,
- config_entry: ConfigEntry,
+ config_entry: MinecraftServerConfigEntry,
) -> None:
"""Initialize sensor base entity."""
super().__init__(coordinator, config_entry)
diff --git a/homeassistant/components/minecraft_server/strings.json b/homeassistant/components/minecraft_server/strings.json
index c084c9e6df0..cb4670dcac4 100644
--- a/homeassistant/components/minecraft_server/strings.json
+++ b/homeassistant/components/minecraft_server/strings.json
@@ -2,12 +2,14 @@
"config": {
"step": {
"user": {
- "title": "Link your Minecraft Server",
- "description": "Set up your Minecraft Server instance to allow monitoring.",
"data": {
- "name": "[%key:common::config_flow::data::name%]",
"address": "Server address"
- }
+ },
+ "data_description": {
+ "address": "The hostname, IP address or SRV record of your Minecraft server, optionally including the port."
+ },
+ "title": "Link your Minecraft Server",
+ "description": "Set up your Minecraft Server instance to allow monitoring."
}
},
"abort": {
diff --git a/homeassistant/components/mjpeg/camera.py b/homeassistant/components/mjpeg/camera.py
index dcb2eff2fd6..c60f1c4d760 100644
--- a/homeassistant/components/mjpeg/camera.py
+++ b/homeassistant/components/mjpeg/camera.py
@@ -27,7 +27,7 @@ from homeassistant.helpers.aiohttp_client import (
async_get_clientsession,
)
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.httpx_client import get_async_client
from .const import CONF_MJPEG_URL, CONF_STILL_IMAGE_URL, DOMAIN, LOGGER
@@ -39,7 +39,7 @@ BUFFER_SIZE = 102400
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a MJPEG IP Camera based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/moat/sensor.py b/homeassistant/components/moat/sensor.py
index 66edfbe91f2..e968577d789 100644
--- a/homeassistant/components/moat/sensor.py
+++ b/homeassistant/components/moat/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -105,7 +105,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Moat BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/mobile_app/binary_sensor.py b/homeassistant/components/mobile_app/binary_sensor.py
index e19e00b1277..8f8b8d97295 100644
--- a/homeassistant/components/mobile_app/binary_sensor.py
+++ b/homeassistant/components/mobile_app/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.const import CONF_WEBHOOK_ID, STATE_ON
from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_SENSOR_ATTRIBUTES,
@@ -28,7 +28,7 @@ from .entity import MobileAppEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up mobile app binary sensor from a config entry."""
entities = []
diff --git a/homeassistant/components/mobile_app/device_tracker.py b/homeassistant/components/mobile_app/device_tracker.py
index 7e84930e2e9..7e5a0a291b6 100644
--- a/homeassistant/components/mobile_app/device_tracker.py
+++ b/homeassistant/components/mobile_app/device_tracker.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import (
@@ -33,7 +33,9 @@ ATTR_KEYS = (ATTR_ALTITUDE, ATTR_COURSE, ATTR_SPEED, ATTR_VERTICAL_ACCURACY)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Mobile app based off an entry."""
entity = MobileAppEntity(entry)
diff --git a/homeassistant/components/mobile_app/sensor.py b/homeassistant/components/mobile_app/sensor.py
index 06ab924aba2..8200ad1fccd 100644
--- a/homeassistant/components/mobile_app/sensor.py
+++ b/homeassistant/components/mobile_app/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.const import CONF_WEBHOOK_ID, STATE_UNKNOWN, UnitOfTemperatur
from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -36,7 +36,7 @@ from .webhook import _extract_sensor_unique_id
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up mobile app sensor from a config entry."""
entities = []
diff --git a/homeassistant/components/modbus/__init__.py b/homeassistant/components/modbus/__init__.py
index 61df7206402..52642cc32e3 100644
--- a/homeassistant/components/modbus/__init__.py
+++ b/homeassistant/components/modbus/__init__.py
@@ -79,6 +79,16 @@ from .const import (
CONF_FAN_MODE_TOP,
CONF_FAN_MODE_VALUES,
CONF_FANS,
+ CONF_HVAC_ACTION_COOLING,
+ CONF_HVAC_ACTION_DEFROSTING,
+ CONF_HVAC_ACTION_DRYING,
+ CONF_HVAC_ACTION_FAN,
+ CONF_HVAC_ACTION_HEATING,
+ CONF_HVAC_ACTION_IDLE,
+ CONF_HVAC_ACTION_OFF,
+ CONF_HVAC_ACTION_PREHEATING,
+ CONF_HVAC_ACTION_REGISTER,
+ CONF_HVAC_ACTION_VALUES,
CONF_HVAC_MODE_AUTO,
CONF_HVAC_MODE_COOL,
CONF_HVAC_MODE_DRY,
@@ -297,6 +307,45 @@ CLIMATE_SCHEMA = vol.All(
vol.Optional(CONF_WRITE_REGISTERS, default=False): cv.boolean,
}
),
+ vol.Optional(CONF_HVAC_ACTION_REGISTER): vol.Maybe(
+ {
+ CONF_ADDRESS: cv.positive_int,
+ CONF_HVAC_ACTION_VALUES: {
+ vol.Optional(CONF_HVAC_ACTION_COOLING): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ vol.Optional(CONF_HVAC_ACTION_DEFROSTING): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ vol.Optional(CONF_HVAC_ACTION_DRYING): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ vol.Optional(CONF_HVAC_ACTION_FAN): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ vol.Optional(CONF_HVAC_ACTION_HEATING): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ vol.Optional(CONF_HVAC_ACTION_IDLE): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ vol.Optional(CONF_HVAC_ACTION_OFF): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ vol.Optional(CONF_HVAC_ACTION_PREHEATING): vol.Any(
+ cv.positive_int, [cv.positive_int]
+ ),
+ },
+ vol.Optional(
+ CONF_INPUT_TYPE, default=CALL_TYPE_REGISTER_HOLDING
+ ): vol.In(
+ [
+ CALL_TYPE_REGISTER_HOLDING,
+ CALL_TYPE_REGISTER_INPUT,
+ ]
+ ),
+ }
+ ),
vol.Optional(CONF_FAN_MODE_REGISTER): vol.Maybe(
vol.All(
{
diff --git a/homeassistant/components/modbus/climate.py b/homeassistant/components/modbus/climate.py
index fca1b94611a..be10a9495c6 100644
--- a/homeassistant/components/modbus/climate.py
+++ b/homeassistant/components/modbus/climate.py
@@ -24,6 +24,7 @@ from homeassistant.components.climate import (
SWING_VERTICAL,
ClimateEntity,
ClimateEntityFeature,
+ HVACAction,
HVACMode,
)
from homeassistant.const import (
@@ -61,6 +62,16 @@ from .const import (
CONF_FAN_MODE_REGISTER,
CONF_FAN_MODE_TOP,
CONF_FAN_MODE_VALUES,
+ CONF_HVAC_ACTION_COOLING,
+ CONF_HVAC_ACTION_DEFROSTING,
+ CONF_HVAC_ACTION_DRYING,
+ CONF_HVAC_ACTION_FAN,
+ CONF_HVAC_ACTION_HEATING,
+ CONF_HVAC_ACTION_IDLE,
+ CONF_HVAC_ACTION_OFF,
+ CONF_HVAC_ACTION_PREHEATING,
+ CONF_HVAC_ACTION_REGISTER,
+ CONF_HVAC_ACTION_VALUES,
CONF_HVAC_MODE_AUTO,
CONF_HVAC_MODE_COOL,
CONF_HVAC_MODE_DRY,
@@ -74,6 +85,7 @@ from .const import (
CONF_HVAC_ON_VALUE,
CONF_HVAC_ONOFF_COIL,
CONF_HVAC_ONOFF_REGISTER,
+ CONF_INPUT_TYPE,
CONF_MAX_TEMP,
CONF_MIN_TEMP,
CONF_STEP,
@@ -188,6 +200,34 @@ class ModbusThermostat(BaseStructPlatform, RestoreEntity, ClimateEntity):
self._attr_hvac_mode = HVACMode.AUTO
self._attr_hvac_modes = [HVACMode.AUTO]
+ if CONF_HVAC_ACTION_REGISTER in config:
+ action_config = config[CONF_HVAC_ACTION_REGISTER]
+ self._hvac_action_register = action_config[CONF_ADDRESS]
+ self._hvac_action_type = action_config[CONF_INPUT_TYPE]
+
+ self._attr_hvac_action = None
+ self._hvac_action_mapping: list[tuple[int, HVACAction]] = []
+ action_value_config = action_config[CONF_HVAC_ACTION_VALUES]
+
+ for hvac_action_kw, hvac_action in (
+ (CONF_HVAC_ACTION_COOLING, HVACAction.COOLING),
+ (CONF_HVAC_ACTION_DEFROSTING, HVACAction.DEFROSTING),
+ (CONF_HVAC_ACTION_DRYING, HVACAction.DRYING),
+ (CONF_HVAC_ACTION_FAN, HVACAction.FAN),
+ (CONF_HVAC_ACTION_HEATING, HVACAction.HEATING),
+ (CONF_HVAC_ACTION_IDLE, HVACAction.IDLE),
+ (CONF_HVAC_ACTION_OFF, HVACAction.OFF),
+ (CONF_HVAC_ACTION_PREHEATING, HVACAction.PREHEATING),
+ ):
+ if hvac_action_kw in action_value_config:
+ values = action_value_config[hvac_action_kw]
+ if not isinstance(values, list):
+ values = [values]
+ for value in values:
+ self._hvac_action_mapping.append((value, hvac_action))
+ else:
+ self._hvac_action_register = None
+
if CONF_FAN_MODE_REGISTER in config:
self._attr_supported_features = (
self._attr_supported_features | ClimateEntityFeature.FAN_MODE
@@ -216,7 +256,6 @@ class ModbusThermostat(BaseStructPlatform, RestoreEntity, ClimateEntity):
self._fan_mode_mapping_from_modbus[value] = fan_mode
self._fan_mode_mapping_to_modbus[fan_mode] = value
self._attr_fan_modes.append(fan_mode)
-
else:
# No FAN modes defined
self._fan_mode_register = None
@@ -457,6 +496,20 @@ class ModbusThermostat(BaseStructPlatform, RestoreEntity, ClimateEntity):
self._attr_hvac_mode = mode
break
+ # Read the HVAC action register if defined
+ if self._hvac_action_register is not None:
+ hvac_action = await self._async_read_register(
+ self._hvac_action_type, self._hvac_action_register, raw=True
+ )
+
+ # Translate the value received
+ if hvac_action is not None:
+ self._attr_hvac_action = None
+ for value, action in self._hvac_action_mapping:
+ if hvac_action == value:
+ self._attr_hvac_action = action
+ break
+
# Read the Fan mode register if defined
if self._fan_mode_register is not None:
fan_mode = await self._async_read_register(
diff --git a/homeassistant/components/modbus/const.py b/homeassistant/components/modbus/const.py
index 5926569040d..634637a6b08 100644
--- a/homeassistant/components/modbus/const.py
+++ b/homeassistant/components/modbus/const.py
@@ -63,6 +63,16 @@ CONF_HVAC_ONOFF_REGISTER = "hvac_onoff_register"
CONF_HVAC_ON_VALUE = "hvac_on_value"
CONF_HVAC_OFF_VALUE = "hvac_off_value"
CONF_HVAC_ONOFF_COIL = "hvac_onoff_coil"
+CONF_HVAC_ACTION_REGISTER = "hvac_action_register"
+CONF_HVAC_ACTION_COOLING = "action_cooling"
+CONF_HVAC_ACTION_DEFROSTING = "action_defrosting"
+CONF_HVAC_ACTION_DRYING = "action_drying"
+CONF_HVAC_ACTION_FAN = "action_fan"
+CONF_HVAC_ACTION_HEATING = "action_heating"
+CONF_HVAC_ACTION_IDLE = "action_idle"
+CONF_HVAC_ACTION_OFF = "action_off"
+CONF_HVAC_ACTION_PREHEATING = "action_preheating"
+CONF_HVAC_ACTION_VALUES = "values"
CONF_HVAC_MODE_OFF = "state_off"
CONF_HVAC_MODE_HEAT = "state_heat"
CONF_HVAC_MODE_COOL = "state_cool"
diff --git a/homeassistant/components/modbus/modbus.py b/homeassistant/components/modbus/modbus.py
index 81cfc3127d1..006ef504590 100644
--- a/homeassistant/components/modbus/modbus.py
+++ b/homeassistant/components/modbus/modbus.py
@@ -384,6 +384,11 @@ class ModbusHub:
{ATTR_SLAVE: slave} if slave is not None else {ATTR_SLAVE: 1}
)
entry = self._pb_request[use_call]
+
+ if use_call in {"write_registers", "write_coils"}:
+ if not isinstance(value, list):
+ value = [value]
+
kwargs[entry.value_attr_name] = value
try:
result: ModbusPDU = await entry.func(address, **kwargs)
diff --git a/homeassistant/components/modbus/strings.json b/homeassistant/components/modbus/strings.json
index 7b55022645e..347549dc837 100644
--- a/homeassistant/components/modbus/strings.json
+++ b/homeassistant/components/modbus/strings.json
@@ -2,11 +2,11 @@
"services": {
"reload": {
"name": "[%key:common::action::reload%]",
- "description": "Reloads all modbus entities."
+ "description": "Reloads all Modbus entities."
},
"write_coil": {
"name": "Write coil",
- "description": "Writes to a modbus coil.",
+ "description": "Writes to a Modbus coil.",
"fields": {
"address": {
"name": "Address",
@@ -17,8 +17,8 @@
"description": "State to write."
},
"slave": {
- "name": "Slave",
- "description": "Address of the modbus unit/slave."
+ "name": "Server",
+ "description": "Address of the Modbus unit/server."
},
"hub": {
"name": "Hub",
@@ -28,7 +28,7 @@
},
"write_register": {
"name": "Write register",
- "description": "Writes to a modbus holding register.",
+ "description": "Writes to a Modbus holding register.",
"fields": {
"address": {
"name": "[%key:component::modbus::services::write_coil::fields::address::name%]",
diff --git a/homeassistant/components/modem_callerid/button.py b/homeassistant/components/modem_callerid/button.py
index 3cad9062be9..954a638818d 100644
--- a/homeassistant/components/modem_callerid/button.py
+++ b/homeassistant/components/modem_callerid/button.py
@@ -9,13 +9,15 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_KEY_API, DOMAIN
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Modem Caller ID sensor."""
api = hass.data[DOMAIN][entry.entry_id][DATA_KEY_API]
diff --git a/homeassistant/components/modem_callerid/sensor.py b/homeassistant/components/modem_callerid/sensor.py
index 00c821f3511..de8e4b2f73c 100644
--- a/homeassistant/components/modem_callerid/sensor.py
+++ b/homeassistant/components/modem_callerid/sensor.py
@@ -9,13 +9,15 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EVENT_HOMEASSISTANT_STOP, STATE_IDLE
from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CID, DATA_KEY_API, DOMAIN
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Modem Caller ID sensor."""
api = hass.data[DOMAIN][entry.entry_id][DATA_KEY_API]
diff --git a/homeassistant/components/modern_forms/__init__.py b/homeassistant/components/modern_forms/__init__.py
index ef2bbad70ce..901e3f431a1 100644
--- a/homeassistant/components/modern_forms/__init__.py
+++ b/homeassistant/components/modern_forms/__init__.py
@@ -9,7 +9,7 @@ from typing import Any, Concatenate
from aiomodernforms import ModernFormsConnectionError, ModernFormsError
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_HOST, Platform
+from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from .const import DOMAIN
@@ -30,7 +30,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up a Modern Forms device from a config entry."""
# Create Modern Forms instance for this entry
- coordinator = ModernFormsDataUpdateCoordinator(hass, host=entry.data[CONF_HOST])
+ coordinator = ModernFormsDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})
diff --git a/homeassistant/components/modern_forms/binary_sensor.py b/homeassistant/components/modern_forms/binary_sensor.py
index ea903c580a4..2bba85f54d7 100644
--- a/homeassistant/components/modern_forms/binary_sensor.py
+++ b/homeassistant/components/modern_forms/binary_sensor.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import CLEAR_TIMER, DOMAIN
@@ -16,7 +16,7 @@ from .entity import ModernFormsDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Modern Forms binary sensors."""
coordinator: ModernFormsDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/modern_forms/coordinator.py b/homeassistant/components/modern_forms/coordinator.py
index ecd928aa922..203ba54380d 100644
--- a/homeassistant/components/modern_forms/coordinator.py
+++ b/homeassistant/components/modern_forms/coordinator.py
@@ -8,6 +8,8 @@ import logging
from aiomodernforms import ModernFormsDevice, ModernFormsError
from aiomodernforms.models import Device as ModernFormsDeviceState
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -21,20 +23,22 @@ _LOGGER = logging.getLogger(__name__)
class ModernFormsDataUpdateCoordinator(DataUpdateCoordinator[ModernFormsDeviceState]):
"""Class to manage fetching Modern Forms data from single endpoint."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- *,
- host: str,
+ config_entry: ConfigEntry,
) -> None:
"""Initialize global Modern Forms data updater."""
self.modern_forms = ModernFormsDevice(
- host, session=async_get_clientsession(hass)
+ config_entry.data[CONF_HOST], session=async_get_clientsession(hass)
)
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/modern_forms/fan.py b/homeassistant/components/modern_forms/fan.py
index 988edcb60e5..26c69b28a5c 100644
--- a/homeassistant/components/modern_forms/fan.py
+++ b/homeassistant/components/modern_forms/fan.py
@@ -11,7 +11,7 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -35,7 +35,7 @@ from .entity import ModernFormsDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Modern Forms platform from config entry."""
diff --git a/homeassistant/components/modern_forms/light.py b/homeassistant/components/modern_forms/light.py
index 2b53a414cea..6216efe3ff4 100644
--- a/homeassistant/components/modern_forms/light.py
+++ b/homeassistant/components/modern_forms/light.py
@@ -11,7 +11,7 @@ from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEnti
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -36,7 +36,7 @@ BRIGHTNESS_RANGE = (1, 255)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Modern Forms platform from config entry."""
diff --git a/homeassistant/components/modern_forms/sensor.py b/homeassistant/components/modern_forms/sensor.py
index 0f1e90cbe52..aa7d163cfdc 100644
--- a/homeassistant/components/modern_forms/sensor.py
+++ b/homeassistant/components/modern_forms/sensor.py
@@ -7,7 +7,7 @@ from datetime import datetime
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -19,7 +19,7 @@ from .entity import ModernFormsDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Modern Forms sensor based on a config entry."""
coordinator: ModernFormsDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/modern_forms/switch.py b/homeassistant/components/modern_forms/switch.py
index f2e8b1b705c..89a5b779d74 100644
--- a/homeassistant/components/modern_forms/switch.py
+++ b/homeassistant/components/modern_forms/switch.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import modernforms_exception_handler
from .const import DOMAIN
@@ -18,7 +18,7 @@ from .entity import ModernFormsDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Modern Forms switch based on a config entry."""
coordinator: ModernFormsDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/moehlenhoff_alpha2/__init__.py b/homeassistant/components/moehlenhoff_alpha2/__init__.py
index 244e3bc701b..b015f9a09dd 100644
--- a/homeassistant/components/moehlenhoff_alpha2/__init__.py
+++ b/homeassistant/components/moehlenhoff_alpha2/__init__.py
@@ -17,7 +17,7 @@ PLATFORMS = [Platform.BINARY_SENSOR, Platform.BUTTON, Platform.CLIMATE, Platform
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up a config entry."""
base = Alpha2Base(entry.data[CONF_HOST])
- coordinator = Alpha2BaseCoordinator(hass, base)
+ coordinator = Alpha2BaseCoordinator(hass, entry, base)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/moehlenhoff_alpha2/binary_sensor.py b/homeassistant/components/moehlenhoff_alpha2/binary_sensor.py
index 1e7018ff1c7..a7479aef5e8 100644
--- a/homeassistant/components/moehlenhoff_alpha2/binary_sensor.py
+++ b/homeassistant/components/moehlenhoff_alpha2/binary_sensor.py
@@ -7,7 +7,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -17,7 +17,7 @@ from .coordinator import Alpha2BaseCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Alpha2 sensor entities from a config_entry."""
diff --git a/homeassistant/components/moehlenhoff_alpha2/button.py b/homeassistant/components/moehlenhoff_alpha2/button.py
index c7ac574724a..57f9d0e31a2 100644
--- a/homeassistant/components/moehlenhoff_alpha2/button.py
+++ b/homeassistant/components/moehlenhoff_alpha2/button.py
@@ -4,7 +4,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -15,7 +15,7 @@ from .coordinator import Alpha2BaseCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Alpha2 button entities."""
diff --git a/homeassistant/components/moehlenhoff_alpha2/climate.py b/homeassistant/components/moehlenhoff_alpha2/climate.py
index 7c24dad4469..85d5939049e 100644
--- a/homeassistant/components/moehlenhoff_alpha2/climate.py
+++ b/homeassistant/components/moehlenhoff_alpha2/climate.py
@@ -12,7 +12,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, PRESET_AUTO, PRESET_DAY, PRESET_NIGHT
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Alpha2Climate entities from a config_entry."""
diff --git a/homeassistant/components/moehlenhoff_alpha2/coordinator.py b/homeassistant/components/moehlenhoff_alpha2/coordinator.py
index 2bac4b49575..50c2f9a5297 100644
--- a/homeassistant/components/moehlenhoff_alpha2/coordinator.py
+++ b/homeassistant/components/moehlenhoff_alpha2/coordinator.py
@@ -8,6 +8,7 @@ import logging
import aiohttp
from moehlenhoff_alpha2 import Alpha2Base
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -20,12 +21,17 @@ UPDATE_INTERVAL = timedelta(seconds=60)
class Alpha2BaseCoordinator(DataUpdateCoordinator[dict[str, dict]]):
"""Keep the base instance in one place and centralize the update."""
- def __init__(self, hass: HomeAssistant, base: Alpha2Base) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, base: Alpha2Base
+ ) -> None:
"""Initialize Alpha2Base data updater."""
self.base = base
super().__init__(
hass=hass,
logger=_LOGGER,
+ config_entry=config_entry,
name="alpha2_base",
update_interval=UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/moehlenhoff_alpha2/manifest.json b/homeassistant/components/moehlenhoff_alpha2/manifest.json
index 14f40991a84..45b7f8c9565 100644
--- a/homeassistant/components/moehlenhoff_alpha2/manifest.json
+++ b/homeassistant/components/moehlenhoff_alpha2/manifest.json
@@ -5,5 +5,5 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/moehlenhoff_alpha2",
"iot_class": "local_push",
- "requirements": ["moehlenhoff-alpha2==1.3.1"]
+ "requirements": ["moehlenhoff-alpha2==1.4.0"]
}
diff --git a/homeassistant/components/moehlenhoff_alpha2/sensor.py b/homeassistant/components/moehlenhoff_alpha2/sensor.py
index 5286257ff61..306e80e54d3 100644
--- a/homeassistant/components/moehlenhoff_alpha2/sensor.py
+++ b/homeassistant/components/moehlenhoff_alpha2/sensor.py
@@ -4,7 +4,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -14,7 +14,7 @@ from .coordinator import Alpha2BaseCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Alpha2 sensor entities from a config_entry."""
diff --git a/homeassistant/components/mold_indicator/sensor.py b/homeassistant/components/mold_indicator/sensor.py
index 750ddce8513..451cc65fb55 100644
--- a/homeassistant/components/mold_indicator/sensor.py
+++ b/homeassistant/components/mold_indicator/sensor.py
@@ -36,7 +36,10 @@ from homeassistant.core import (
)
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.device import async_device_info_to_link_from_entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util.unit_conversion import TemperatureConverter
@@ -105,7 +108,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Mold indicator sensor entry."""
name: str = entry.options[CONF_NAME]
diff --git a/homeassistant/components/monarch_money/__init__.py b/homeassistant/components/monarch_money/__init__.py
index 5f9aba7dd07..8b7cfa6aa5b 100644
--- a/homeassistant/components/monarch_money/__init__.py
+++ b/homeassistant/components/monarch_money/__init__.py
@@ -4,13 +4,10 @@ from __future__ import annotations
from typedmonarchmoney import TypedMonarchMoney
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TOKEN, Platform
from homeassistant.core import HomeAssistant
-from .coordinator import MonarchMoneyDataUpdateCoordinator
-
-type MonarchMoneyConfigEntry = ConfigEntry[MonarchMoneyDataUpdateCoordinator]
+from .coordinator import MonarchMoneyConfigEntry, MonarchMoneyDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
@@ -21,7 +18,7 @@ async def async_setup_entry(
"""Set up Monarch Money from a config entry."""
monarch_client = TypedMonarchMoney(token=entry.data.get(CONF_TOKEN))
- mm_coordinator = MonarchMoneyDataUpdateCoordinator(hass, monarch_client)
+ mm_coordinator = MonarchMoneyDataUpdateCoordinator(hass, entry, monarch_client)
await mm_coordinator.async_config_entry_first_refresh()
entry.runtime_data = mm_coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/monarch_money/coordinator.py b/homeassistant/components/monarch_money/coordinator.py
index 3e689c48e91..7f3dac9419f 100644
--- a/homeassistant/components/monarch_money/coordinator.py
+++ b/homeassistant/components/monarch_money/coordinator.py
@@ -30,21 +30,26 @@ class MonarchData:
cashflow_summary: MonarchCashflowSummary
+type MonarchMoneyConfigEntry = ConfigEntry[MonarchMoneyDataUpdateCoordinator]
+
+
class MonarchMoneyDataUpdateCoordinator(DataUpdateCoordinator[MonarchData]):
"""Data update coordinator for Monarch Money."""
- config_entry: ConfigEntry
+ config_entry: MonarchMoneyConfigEntry
subscription_id: str
def __init__(
self,
hass: HomeAssistant,
+ config_entry: MonarchMoneyConfigEntry,
client: TypedMonarchMoney,
) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name="monarchmoney",
update_interval=timedelta(hours=4),
)
diff --git a/homeassistant/components/monarch_money/sensor.py b/homeassistant/components/monarch_money/sensor.py
index fe7c728cf41..1597d9820a1 100644
--- a/homeassistant/components/monarch_money/sensor.py
+++ b/homeassistant/components/monarch_money/sensor.py
@@ -14,10 +14,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import CURRENCY_DOLLAR, PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import MonarchMoneyConfigEntry
+from .coordinator import MonarchMoneyConfigEntry
from .entity import MonarchMoneyAccountEntity, MonarchMoneyCashFlowEntity
@@ -110,7 +110,7 @@ MONARCH_CASHFLOW_SENSORS: tuple[MonarchMoneyCashflowSensorEntityDescription, ...
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MonarchMoneyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Monarch Money sensors for config entries."""
mm_coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/monoprice/media_player.py b/homeassistant/components/monoprice/media_player.py
index 2dde0832440..9d678c16874 100644
--- a/homeassistant/components/monoprice/media_player.py
+++ b/homeassistant/components/monoprice/media_player.py
@@ -16,7 +16,7 @@ from homeassistant.const import CONF_PORT
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform, service
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_SOURCES,
@@ -58,7 +58,7 @@ def _get_sources(config_entry):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Monoprice 6-zone amplifier platform."""
port = config_entry.data[CONF_PORT]
diff --git a/homeassistant/components/monzo/__init__.py b/homeassistant/components/monzo/__init__.py
index a88082b2ce6..662cfecd2e9 100644
--- a/homeassistant/components/monzo/__init__.py
+++ b/homeassistant/components/monzo/__init__.py
@@ -26,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
external_api = AuthenticatedMonzoAPI(async_get_clientsession(hass), session)
- coordinator = MonzoCoordinator(hass, external_api)
+ coordinator = MonzoCoordinator(hass, entry, external_api)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/monzo/coordinator.py b/homeassistant/components/monzo/coordinator.py
index caac551f986..06c751a23e0 100644
--- a/homeassistant/components/monzo/coordinator.py
+++ b/homeassistant/components/monzo/coordinator.py
@@ -8,6 +8,7 @@ from typing import Any
from monzopy import AuthorisationExpiredError, InvalidMonzoAPIResponseError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -29,11 +30,16 @@ class MonzoData:
class MonzoCoordinator(DataUpdateCoordinator[MonzoData]):
"""Class to manage fetching Monzo data from the API."""
- def __init__(self, hass: HomeAssistant, api: AuthenticatedMonzoAPI) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: AuthenticatedMonzoAPI
+ ) -> None:
"""Initialize."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=1),
)
diff --git a/homeassistant/components/monzo/sensor.py b/homeassistant/components/monzo/sensor.py
index 41b97d90452..0b6ab2b70a5 100644
--- a/homeassistant/components/monzo/sensor.py
+++ b/homeassistant/components/monzo/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import MonzoCoordinator
@@ -65,7 +65,7 @@ MODEL_POT = "Pot"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinator: MonzoCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/moon/sensor.py b/homeassistant/components/moon/sensor.py
index 09048579859..12d0ff3ed41 100644
--- a/homeassistant/components/moon/sensor.py
+++ b/homeassistant/components/moon/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN
@@ -26,7 +26,7 @@ STATE_WAXING_GIBBOUS = "waxing_gibbous"
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from config_entry."""
async_add_entities([MoonSensorEntity(entry)], True)
diff --git a/homeassistant/components/mopeka/sensor.py b/homeassistant/components/mopeka/sensor.py
index 0f67efaea1e..53c93f771f2 100644
--- a/homeassistant/components/mopeka/sensor.py
+++ b/homeassistant/components/mopeka/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from . import MopekaConfigEntry
@@ -115,7 +115,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: MopekaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Mopeka BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/motion_blinds/__init__.py b/homeassistant/components/motion_blinds/__init__.py
index 182ea310029..2abcc273e23 100644
--- a/homeassistant/components/motion_blinds/__init__.py
+++ b/homeassistant/components/motion_blinds/__init__.py
@@ -1,18 +1,18 @@
"""The motion_blinds component."""
import asyncio
-from datetime import timedelta
import logging
from typing import TYPE_CHECKING
from motionblinds import AsyncMotionMulticast
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY, CONF_HOST, EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from .const import (
+ CONF_BLIND_TYPE_LIST,
CONF_INTERFACE,
CONF_WAIT_FOR_PUSH,
DEFAULT_INTERFACE,
@@ -25,7 +25,6 @@ from .const import (
KEY_SETUP_LOCK,
KEY_UNSUB_STOP,
PLATFORMS,
- UPDATE_INTERVAL,
)
from .coordinator import DataUpdateCoordinatorMotionBlinds
from .gateway import ConnectMotionGateway
@@ -41,6 +40,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
key = entry.data[CONF_API_KEY]
multicast_interface = entry.data.get(CONF_INTERFACE, DEFAULT_INTERFACE)
wait_for_push = entry.options.get(CONF_WAIT_FOR_PUSH, DEFAULT_WAIT_FOR_PUSH)
+ blind_type_list = entry.data.get(CONF_BLIND_TYPE_LIST)
# Create multicast Listener
async with setup_lock:
@@ -83,7 +83,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
# Connect to motion gateway
multicast = hass.data[DOMAIN][KEY_MULTICAST_LISTENER]
connect_gateway_class = ConnectMotionGateway(hass, multicast)
- if not await connect_gateway_class.async_connect_gateway(host, key):
+ if not await connect_gateway_class.async_connect_gateway(
+ host, key, blind_type_list
+ ):
raise ConfigEntryNotReady
motion_gateway = connect_gateway_class.gateway_device
api_lock = asyncio.Lock()
@@ -94,15 +96,17 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
}
coordinator = DataUpdateCoordinatorMotionBlinds(
- hass,
- _LOGGER,
- coordinator_info,
- # Name of the data. For logging purposes.
- name=entry.title,
- # Polling interval. Will only be polled if there are subscribers.
- update_interval=timedelta(seconds=UPDATE_INTERVAL),
+ hass, entry, _LOGGER, coordinator_info
)
+ # store blind type list for next time
+ if entry.data.get(CONF_BLIND_TYPE_LIST) != motion_gateway.blind_type_list:
+ data = {
+ **entry.data,
+ CONF_BLIND_TYPE_LIST: motion_gateway.blind_type_list,
+ }
+ hass.config_entries.async_update_entry(entry, data=data)
+
# Fetch initial data so we have data when entities subscribe
await coordinator.async_config_entry_first_refresh()
@@ -132,12 +136,7 @@ async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
multicast.Unregister_motion_gateway(config_entry.data[CONF_HOST])
hass.data[DOMAIN].pop(config_entry.entry_id)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
# No motion gateways left, stop Motion multicast
unsub_stop = hass.data[DOMAIN].pop(KEY_UNSUB_STOP)
unsub_stop()
diff --git a/homeassistant/components/motion_blinds/button.py b/homeassistant/components/motion_blinds/button.py
index 89841bf8fd4..09f29e09c70 100644
--- a/homeassistant/components/motion_blinds/button.py
+++ b/homeassistant/components/motion_blinds/button.py
@@ -8,7 +8,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, KEY_COORDINATOR, KEY_GATEWAY
from .coordinator import DataUpdateCoordinatorMotionBlinds
@@ -18,7 +18,7 @@ from .entity import MotionCoordinatorEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Motionblinds."""
entities: list[ButtonEntity] = []
diff --git a/homeassistant/components/motion_blinds/config_flow.py b/homeassistant/components/motion_blinds/config_flow.py
index d8d1e7c21f1..a7bb34af1e6 100644
--- a/homeassistant/components/motion_blinds/config_flow.py
+++ b/homeassistant/components/motion_blinds/config_flow.py
@@ -156,6 +156,7 @@ class MotionBlindsFlowHandler(ConfigFlow, domain=DOMAIN):
errors: dict[str, str] = {}
if user_input is not None:
key = user_input[CONF_API_KEY]
+ assert self._host
connect_gateway_class = ConnectMotionGateway(self.hass)
if not await connect_gateway_class.async_connect_gateway(self._host, key):
diff --git a/homeassistant/components/motion_blinds/const.py b/homeassistant/components/motion_blinds/const.py
index 96067d7ceb0..950fa3ab4c7 100644
--- a/homeassistant/components/motion_blinds/const.py
+++ b/homeassistant/components/motion_blinds/const.py
@@ -8,6 +8,7 @@ DEFAULT_GATEWAY_NAME = "Motionblinds Gateway"
PLATFORMS = [Platform.BUTTON, Platform.COVER, Platform.SENSOR]
+CONF_BLIND_TYPE_LIST = "blind_type_list"
CONF_WAIT_FOR_PUSH = "wait_for_push"
CONF_INTERFACE = "interface"
DEFAULT_WAIT_FOR_PUSH = False
diff --git a/homeassistant/components/motion_blinds/coordinator.py b/homeassistant/components/motion_blinds/coordinator.py
index b2abd205ce5..79e26e5aed4 100644
--- a/homeassistant/components/motion_blinds/coordinator.py
+++ b/homeassistant/components/motion_blinds/coordinator.py
@@ -7,6 +7,7 @@ from typing import Any
from motionblinds import DEVICE_TYPES_WIFI, ParseException
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -25,21 +26,22 @@ _LOGGER = logging.getLogger(__name__)
class DataUpdateCoordinatorMotionBlinds(DataUpdateCoordinator):
"""Class to manage fetching data from single endpoint."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
logger: logging.Logger,
coordinator_info: dict[str, Any],
- *,
- name: str,
- update_interval: timedelta,
) -> None:
"""Initialize global data updater."""
super().__init__(
hass,
logger,
- name=name,
- update_interval=update_interval,
+ config_entry=config_entry,
+ name=config_entry.title,
+ update_interval=timedelta(seconds=UPDATE_INTERVAL),
)
self.api_lock = coordinator_info[KEY_API_LOCK]
diff --git a/homeassistant/components/motion_blinds/cover.py b/homeassistant/components/motion_blinds/cover.py
index 1ea3a6ed9d6..dbf43e3d30f 100644
--- a/homeassistant/components/motion_blinds/cover.py
+++ b/homeassistant/components/motion_blinds/cover.py
@@ -18,7 +18,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from .const import (
@@ -83,7 +83,7 @@ SET_ABSOLUTE_POSITION_SCHEMA: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Motion Blind from a config entry."""
entities: list[MotionBaseDevice] = []
diff --git a/homeassistant/components/motion_blinds/gateway.py b/homeassistant/components/motion_blinds/gateway.py
index 44f7caa74b2..9826557919c 100644
--- a/homeassistant/components/motion_blinds/gateway.py
+++ b/homeassistant/components/motion_blinds/gateway.py
@@ -42,11 +42,16 @@ class ConnectMotionGateway:
for blind in self.gateway_device.device_list.values():
blind.Update_from_cache()
- async def async_connect_gateway(self, host, key):
+ async def async_connect_gateway(
+ self,
+ host: str,
+ key: str,
+ blind_type_list: dict[str, int] | None = None,
+ ) -> bool:
"""Connect to the Motion Gateway."""
_LOGGER.debug("Initializing with host %s (key %s)", host, key[:3])
self._gateway_device = MotionGateway(
- ip=host, key=key, multicast=self._multicast
+ ip=host, key=key, multicast=self._multicast, blind_type_list=blind_type_list
)
try:
# update device info and get the connected sub devices
diff --git a/homeassistant/components/motion_blinds/manifest.json b/homeassistant/components/motion_blinds/manifest.json
index b327c146300..1654d5b5937 100644
--- a/homeassistant/components/motion_blinds/manifest.json
+++ b/homeassistant/components/motion_blinds/manifest.json
@@ -21,5 +21,5 @@
"documentation": "https://www.home-assistant.io/integrations/motion_blinds",
"iot_class": "local_push",
"loggers": ["motionblinds"],
- "requirements": ["motionblinds==0.6.25"]
+ "requirements": ["motionblinds==0.6.26"]
}
diff --git a/homeassistant/components/motion_blinds/sensor.py b/homeassistant/components/motion_blinds/sensor.py
index 6418cebda0c..60d283aa0b6 100644
--- a/homeassistant/components/motion_blinds/sensor.py
+++ b/homeassistant/components/motion_blinds/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.const import (
EntityCategory,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, KEY_COORDINATOR, KEY_GATEWAY
from .entity import MotionCoordinatorEntity
@@ -26,7 +26,7 @@ ATTR_BATTERY_VOLTAGE = "battery_voltage"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Motionblinds."""
entities: list[SensorEntity] = []
diff --git a/homeassistant/components/motionblinds_ble/button.py b/homeassistant/components/motionblinds_ble/button.py
index a099276cd85..12fb6c7a513 100644
--- a/homeassistant/components/motionblinds_ble/button.py
+++ b/homeassistant/components/motionblinds_ble/button.py
@@ -13,7 +13,7 @@ from homeassistant.components.button import ButtonEntity, ButtonEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_CONNECT, ATTR_DISCONNECT, ATTR_FAVORITE, CONF_MAC_CODE, DOMAIN
from .entity import MotionblindsBLEEntity
@@ -53,7 +53,9 @@ BUTTON_TYPES: list[MotionblindsBLEButtonEntityDescription] = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button entities based on a config entry."""
diff --git a/homeassistant/components/motionblinds_ble/cover.py b/homeassistant/components/motionblinds_ble/cover.py
index afeeb5b0d70..beaee8598b5 100644
--- a/homeassistant/components/motionblinds_ble/cover.py
+++ b/homeassistant/components/motionblinds_ble/cover.py
@@ -19,7 +19,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_BLIND_TYPE, CONF_MAC_CODE, DOMAIN, ICON_VERTICAL_BLIND
from .entity import MotionblindsBLEEntity
@@ -61,7 +61,9 @@ BLIND_TYPE_TO_ENTITY_DESCRIPTION: dict[str, MotionblindsBLECoverEntityDescriptio
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up cover entity based on a config entry."""
diff --git a/homeassistant/components/motionblinds_ble/select.py b/homeassistant/components/motionblinds_ble/select.py
index c297c887910..976f51a0a0f 100644
--- a/homeassistant/components/motionblinds_ble/select.py
+++ b/homeassistant/components/motionblinds_ble/select.py
@@ -11,7 +11,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_SPEED, CONF_MAC_CODE, DOMAIN
from .entity import MotionblindsBLEEntity
@@ -32,7 +32,9 @@ SELECT_TYPES: dict[str, SelectEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select entities based on a config entry."""
diff --git a/homeassistant/components/motionblinds_ble/sensor.py b/homeassistant/components/motionblinds_ble/sensor.py
index 740a0509a9e..8993a3b1cd5 100644
--- a/homeassistant/components/motionblinds_ble/sensor.py
+++ b/homeassistant/components/motionblinds_ble/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
EntityCategory,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -92,7 +92,9 @@ SENSORS: tuple[MotionblindsBLESensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor entities based on a config entry."""
diff --git a/homeassistant/components/motioneye/camera.py b/homeassistant/components/motioneye/camera.py
index df4c321037e..159956277a8 100644
--- a/homeassistant/components/motioneye/camera.py
+++ b/homeassistant/components/motioneye/camera.py
@@ -42,7 +42,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from . import get_camera_from_cameras, is_acceptable_camera, listen_for_new_cameras
@@ -93,7 +93,9 @@ SCHEMA_SERVICE_SET_TEXT = vol.Schema(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up motionEye from a config entry."""
entry_data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/motioneye/sensor.py b/homeassistant/components/motioneye/sensor.py
index e0113544848..c160b77c16a 100644
--- a/homeassistant/components/motioneye/sensor.py
+++ b/homeassistant/components/motioneye/sensor.py
@@ -12,7 +12,7 @@ from motioneye_client.const import KEY_ACTIONS
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -24,7 +24,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up motionEye from a config entry."""
entry_data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/motioneye/switch.py b/homeassistant/components/motioneye/switch.py
index 9d704f17740..89d3b8a8727 100644
--- a/homeassistant/components/motioneye/switch.py
+++ b/homeassistant/components/motioneye/switch.py
@@ -19,7 +19,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from . import get_camera_from_cameras, listen_for_new_cameras
@@ -67,7 +67,9 @@ MOTIONEYE_SWITCHES = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up motionEye from a config entry."""
entry_data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/motionmount/binary_sensor.py b/homeassistant/components/motionmount/binary_sensor.py
index f19af67e198..4bb880311f9 100644
--- a/homeassistant/components/motionmount/binary_sensor.py
+++ b/homeassistant/components/motionmount/binary_sensor.py
@@ -6,17 +6,20 @@ from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
BinarySensorEntity,
)
+from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MotionMountConfigEntry
from .entity import MotionMountEntity
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
entry: MotionMountConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Vogel's MotionMount from a config entry."""
mm = entry.runtime_data
@@ -29,6 +32,8 @@ class MotionMountMovingSensor(MotionMountEntity, BinarySensorEntity):
_attr_device_class = BinarySensorDeviceClass.MOVING
_attr_translation_key = "motionmount_is_moving"
+ _attr_entity_category = EntityCategory.DIAGNOSTIC
+ _attr_entity_registry_enabled_default = False
def __init__(
self, mm: motionmount.MotionMount, config_entry: MotionMountConfigEntry
diff --git a/homeassistant/components/motionmount/icons.json b/homeassistant/components/motionmount/icons.json
new file mode 100644
index 00000000000..8d6d867f4d0
--- /dev/null
+++ b/homeassistant/components/motionmount/icons.json
@@ -0,0 +1,12 @@
+{
+ "entity": {
+ "sensor": {
+ "motionmount_error_status": {
+ "default": "mdi:alert-circle-outline",
+ "state": {
+ "none": "mdi:check-circle-outline"
+ }
+ }
+ }
+ }
+}
diff --git a/homeassistant/components/motionmount/manifest.json b/homeassistant/components/motionmount/manifest.json
index 422be417006..337ce776b33 100644
--- a/homeassistant/components/motionmount/manifest.json
+++ b/homeassistant/components/motionmount/manifest.json
@@ -1,11 +1,12 @@
{
"domain": "motionmount",
"name": "Vogel's MotionMount",
- "codeowners": ["@RJPoelstra"],
+ "codeowners": ["@laiho-vogels"],
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/motionmount",
"integration_type": "device",
"iot_class": "local_push",
+ "quality_scale": "bronze",
"requirements": ["python-MotionMount==2.3.0"],
"zeroconf": ["_tvm._tcp.local."]
}
diff --git a/homeassistant/components/motionmount/number.py b/homeassistant/components/motionmount/number.py
index 6305820174f..3e2c1b067aa 100644
--- a/homeassistant/components/motionmount/number.py
+++ b/homeassistant/components/motionmount/number.py
@@ -8,17 +8,19 @@ from homeassistant.components.number import NumberEntity
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MotionMountConfigEntry
from .const import DOMAIN
from .entity import MotionMountEntity
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
entry: MotionMountConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Vogel's MotionMount from a config entry."""
mm = entry.runtime_data
diff --git a/homeassistant/components/motionmount/quality_scale.yaml b/homeassistant/components/motionmount/quality_scale.yaml
new file mode 100644
index 00000000000..8b210931eaf
--- /dev/null
+++ b/homeassistant/components/motionmount/quality_scale.yaml
@@ -0,0 +1,78 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: Integration does not have actions
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: done
+ comment: Integration does register actions aside from entity actions
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: exempt
+ comment: Integration does not register events.
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions:
+ status: exempt
+ comment: Integration does not have actions
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: Integration has no options flow
+ docs-installation-parameters: done
+ entity-unavailable: done
+ integration-owner: done
+ log-when-unavailable: done
+ parallel-updates: done
+ reauthentication-flow: done
+ test-coverage: todo
+
+ # Gold
+ devices: done
+ diagnostics: todo
+ discovery-update-info: done
+ discovery: done
+ docs-data-update: done
+ docs-examples: todo
+ docs-known-limitations: done
+ docs-supported-devices: done
+ docs-supported-functions: done
+ docs-troubleshooting: done
+ docs-use-cases: done
+ dynamic-devices:
+ status: exempt
+ comment: Single device per config entry
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default: done
+ entity-translations: done
+ exception-translations: done
+ icon-translations: done
+ reconfiguration-flow: todo
+ repair-issues:
+ status: exempt
+ comment: Integration does not need user intervention
+ stale-devices:
+ status: exempt
+ comment: Integration does not support dynamic devices
+
+ # Platinum
+ async-dependency: done
+ inject-websession:
+ status: exempt
+ comment: Device doesn't make http requests.
+ strict-typing: done
diff --git a/homeassistant/components/motionmount/select.py b/homeassistant/components/motionmount/select.py
index 31c5056b91f..a8fcc84f2ec 100644
--- a/homeassistant/components/motionmount/select.py
+++ b/homeassistant/components/motionmount/select.py
@@ -9,7 +9,7 @@ import motionmount
from homeassistant.components.select import SelectEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MotionMountConfigEntry
from .const import DOMAIN, WALL_PRESET_NAME
@@ -17,12 +17,13 @@ from .entity import MotionMountEntity
_LOGGER = logging.getLogger(__name__)
SCAN_INTERVAL = timedelta(seconds=60)
+PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: MotionMountConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Vogel's MotionMount from a config entry."""
mm = entry.runtime_data
diff --git a/homeassistant/components/motionmount/sensor.py b/homeassistant/components/motionmount/sensor.py
index 685c3ebf932..28fe921d9ac 100644
--- a/homeassistant/components/motionmount/sensor.py
+++ b/homeassistant/components/motionmount/sensor.py
@@ -6,12 +6,15 @@ import motionmount
from motionmount import MotionMountSystemError
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
+from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MotionMountConfigEntry
from .entity import MotionMountEntity
+PARALLEL_UPDATES = 0
+
ERROR_MESSAGES: Final = {
MotionMountSystemError.MotorError: "motor",
MotionMountSystemError.ObstructionDetected: "obstruction",
@@ -24,7 +27,7 @@ ERROR_MESSAGES: Final = {
async def async_setup_entry(
hass: HomeAssistant,
entry: MotionMountConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Vogel's MotionMount from a config entry."""
mm = entry.runtime_data
@@ -45,6 +48,8 @@ class MotionMountErrorStatusSensor(MotionMountEntity, SensorEntity):
"internal",
]
_attr_translation_key = "motionmount_error_status"
+ _attr_entity_category = EntityCategory.DIAGNOSTIC
+ _attr_entity_registry_enabled_default = False
def __init__(
self, mm: motionmount.MotionMount, config_entry: MotionMountConfigEntry
diff --git a/homeassistant/components/mpd/media_player.py b/homeassistant/components/mpd/media_player.py
index db3901016f7..14b69e941b7 100644
--- a/homeassistant/components/mpd/media_player.py
+++ b/homeassistant/components/mpd/media_player.py
@@ -31,7 +31,7 @@ from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, CONF_PORT
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle, dt as dt_util
from .const import DOMAIN, LOGGER
@@ -68,7 +68,9 @@ PLATFORM_SCHEMA = MEDIA_PLAYER_PLATFORM_SCHEMA.extend(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up media player from config_entry."""
diff --git a/homeassistant/components/mqtt/__init__.py b/homeassistant/components/mqtt/__init__.py
index 6656afe2c8a..ae010bf18c9 100644
--- a/homeassistant/components/mqtt/__init__.py
+++ b/homeassistant/components/mqtt/__init__.py
@@ -13,7 +13,7 @@ import voluptuous as vol
from homeassistant import config as conf_util
from homeassistant.components import websocket_api
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_DISCOVERY, SERVICE_RELOAD
+from homeassistant.const import CONF_DISCOVERY, CONF_PLATFORM, SERVICE_RELOAD
from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.exceptions import (
ConfigValidationError,
@@ -81,6 +81,7 @@ from .const import (
ENTRY_OPTION_FIELDS,
MQTT_CONNECTION_STATE,
TEMPLATE_ERRORS,
+ Platform,
)
from .models import (
DATA_MQTT,
@@ -293,6 +294,21 @@ async def async_check_config_schema(
) from exc
+def _platforms_in_use(hass: HomeAssistant, entry: ConfigEntry) -> set[str | Platform]:
+ """Return a set of platforms in use."""
+ domains: set[str | Platform] = {
+ entry.domain
+ for entry in er.async_entries_for_config_entry(
+ er.async_get(hass), entry.entry_id
+ )
+ }
+ # Update with domains from subentries
+ for subentry in entry.subentries.values():
+ components = subentry.data["components"].values()
+ domains.update(component[CONF_PLATFORM] for component in components)
+ return domains
+
+
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the actions and websocket API for the MQTT component."""
@@ -434,12 +450,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
mqtt_data, conf = await _setup_client()
platforms_used = platforms_from_config(mqtt_data.config)
- platforms_used.update(
- entry.domain
- for entry in er.async_entries_for_config_entry(
- er.async_get(hass), entry.entry_id
- )
- )
+ platforms_used.update(_platforms_in_use(hass, entry))
integration = async_get_loaded_integration(hass, DOMAIN)
# Preload platforms we know we are going to use so
# discovery can setup each platform synchronously
diff --git a/homeassistant/components/mqtt/abbreviations.py b/homeassistant/components/mqtt/abbreviations.py
index 584b238b3a8..a9037a5f247 100644
--- a/homeassistant/components/mqtt/abbreviations.py
+++ b/homeassistant/components/mqtt/abbreviations.py
@@ -150,6 +150,7 @@ ABBREVIATIONS = {
"pl_rst_pct": "payload_reset_percentage",
"pl_rst_pr_mode": "payload_reset_preset_mode",
"pl_stop": "payload_stop",
+ "pl_stop_tilt": "payload_stop_tilt",
"pl_strt": "payload_start",
"pl_ret": "payload_return_to_base",
"pl_toff": "payload_turn_off",
@@ -218,10 +219,16 @@ ABBREVIATIONS = {
"sup_vol": "support_volume_set",
"sup_feat": "supported_features",
"sup_clrm": "supported_color_modes",
+ "swing_h_mode_cmd_tpl": "swing_horizontal_mode_command_template",
+ "swing_h_mode_cmd_t": "swing_horizontal_mode_command_topic",
+ "swing_h_mode_stat_tpl": "swing_horizontal_mode_state_template",
+ "swing_h_mode_stat_t": "swing_horizontal_mode_state_topic",
+ "swing_h_modes": "swing_horizontal_modes",
"swing_mode_cmd_tpl": "swing_mode_command_template",
"swing_mode_cmd_t": "swing_mode_command_topic",
"swing_mode_stat_tpl": "swing_mode_state_template",
"swing_mode_stat_t": "swing_mode_state_topic",
+ "swing_modes": "swing_modes",
"temp_cmd_tpl": "temperature_command_template",
"temp_cmd_t": "temperature_command_topic",
"temp_hi_cmd_tpl": "temperature_high_command_template",
diff --git a/homeassistant/components/mqtt/alarm_control_panel.py b/homeassistant/components/mqtt/alarm_control_panel.py
index 7bdc13d0522..64b1a6b05fa 100644
--- a/homeassistant/components/mqtt/alarm_control_panel.py
+++ b/homeassistant/components/mqtt/alarm_control_panel.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_CODE, CONF_NAME, CONF_VALUE_TEMPLATE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from . import subscription
@@ -115,7 +115,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT alarm control panel through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/async_client.py b/homeassistant/components/mqtt/async_client.py
index 5f90136df44..0467eb3a289 100644
--- a/homeassistant/components/mqtt/async_client.py
+++ b/homeassistant/components/mqtt/async_client.py
@@ -6,7 +6,14 @@ from functools import lru_cache
from types import TracebackType
from typing import Self
-from paho.mqtt.client import Client as MQTTClient
+from paho.mqtt.client import (
+ CallbackOnConnect_v2,
+ CallbackOnDisconnect_v2,
+ CallbackOnPublish_v2,
+ CallbackOnSubscribe_v2,
+ CallbackOnUnsubscribe_v2,
+ Client as MQTTClient,
+)
_MQTT_LOCK_COUNT = 7
@@ -44,6 +51,12 @@ class AsyncMQTTClient(MQTTClient):
that is not needed since we are running in an async event loop.
"""
+ on_connect: CallbackOnConnect_v2
+ on_disconnect: CallbackOnDisconnect_v2
+ on_publish: CallbackOnPublish_v2
+ on_subscribe: CallbackOnSubscribe_v2
+ on_unsubscribe: CallbackOnUnsubscribe_v2
+
def setup(self) -> None:
"""Set up the client.
diff --git a/homeassistant/components/mqtt/binary_sensor.py b/homeassistant/components/mqtt/binary_sensor.py
index d736123eae8..a1e146d4e36 100644
--- a/homeassistant/components/mqtt/binary_sensor.py
+++ b/homeassistant/components/mqtt/binary_sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, event as evt
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType
@@ -69,7 +69,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT binary sensor through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/button.py b/homeassistant/components/mqtt/button.py
index b6056c2efd9..5b2bcc8920f 100644
--- a/homeassistant/components/mqtt/button.py
+++ b/homeassistant/components/mqtt/button.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE_CLASS, CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .config import DEFAULT_RETAIN, MQTT_BASE_SCHEMA
@@ -43,7 +43,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT button through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/camera.py b/homeassistant/components/mqtt/camera.py
index 88fabad0446..d3615edcbba 100644
--- a/homeassistant/components/mqtt/camera.py
+++ b/homeassistant/components/mqtt/camera.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import subscription
@@ -60,7 +60,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_BASE.extend({}, extra=vol.REMOVE_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT camera through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/client.py b/homeassistant/components/mqtt/client.py
index 3aca566dbfc..f6f53599363 100644
--- a/homeassistant/components/mqtt/client.py
+++ b/homeassistant/components/mqtt/client.py
@@ -15,6 +15,7 @@ import socket
import ssl
import time
from typing import TYPE_CHECKING, Any
+from uuid import uuid4
import certifi
@@ -292,27 +293,44 @@ class MqttClientSetup:
"""
# We don't import on the top because some integrations
# should be able to optionally rely on MQTT.
- import paho.mqtt.client as mqtt # pylint: disable=import-outside-toplevel
+ from paho.mqtt import client as mqtt # pylint: disable=import-outside-toplevel
# pylint: disable-next=import-outside-toplevel
from .async_client import AsyncMQTTClient
config = self._config
+ clean_session: bool | None = None
if (protocol := config.get(CONF_PROTOCOL, DEFAULT_PROTOCOL)) == PROTOCOL_31:
proto = mqtt.MQTTv31
+ clean_session = True
elif protocol == PROTOCOL_5:
proto = mqtt.MQTTv5
else:
proto = mqtt.MQTTv311
+ clean_session = True
if (client_id := config.get(CONF_CLIENT_ID)) is None:
- # PAHO MQTT relies on the MQTT server to generate random client IDs.
- # However, that feature is not mandatory so we generate our own.
- client_id = None
+ # PAHO MQTT relies on the MQTT server to generate random client ID
+ # for protocol version 3.1, however, that feature is not mandatory
+ # so we generate our own.
+ client_id = mqtt._base62(uuid4().int, padding=22) # noqa: SLF001
transport: str = config.get(CONF_TRANSPORT, DEFAULT_TRANSPORT)
self._client = AsyncMQTTClient(
- mqtt.CallbackAPIVersion.VERSION1,
- client_id,
+ callback_api_version=mqtt.CallbackAPIVersion.VERSION2,
+ client_id=client_id,
+ # See: https://eclipse.dev/paho/files/paho.mqtt.python/html/client.html
+ # clean_session (bool defaults to None)
+ # a boolean that determines the client type.
+ # If True, the broker will remove all information about this client when it
+ # disconnects. If False, the client is a persistent client and subscription
+ # information and queued messages will be retained when the client
+ # disconnects. Note that a client will never discard its own outgoing
+ # messages on disconnect. Calling connect() or reconnect() will cause the
+ # messages to be resent. Use reinitialise() to reset a client to its
+ # original state. The clean_session argument only applies to MQTT versions
+ # v3.1.1 and v3.1. It is not accepted if the MQTT version is v5.0 - use the
+ # clean_start argument on connect() instead.
+ clean_session=clean_session,
protocol=proto,
transport=transport, # type: ignore[arg-type]
reconnect_on_failure=False,
@@ -371,6 +389,7 @@ class MQTT:
self.loop = hass.loop
self.config_entry = config_entry
self.conf = conf
+ self.is_mqttv5 = conf.get(CONF_PROTOCOL, DEFAULT_PROTOCOL) == PROTOCOL_5
self._simple_subscriptions: defaultdict[str, set[Subscription]] = defaultdict(
set
@@ -476,9 +495,9 @@ class MQTT:
mqttc.on_connect = self._async_mqtt_on_connect
mqttc.on_disconnect = self._async_mqtt_on_disconnect
mqttc.on_message = self._async_mqtt_on_message
- mqttc.on_publish = self._async_mqtt_on_callback
- mqttc.on_subscribe = self._async_mqtt_on_callback
- mqttc.on_unsubscribe = self._async_mqtt_on_callback
+ mqttc.on_publish = self._async_mqtt_on_publish
+ mqttc.on_subscribe = self._async_mqtt_on_subscribe_unsubscribe
+ mqttc.on_unsubscribe = self._async_mqtt_on_subscribe_unsubscribe
# suppress exceptions at callback
mqttc.suppress_exceptions = True
@@ -498,7 +517,7 @@ class MQTT:
def _async_reader_callback(self, client: mqtt.Client) -> None:
"""Handle reading data from the socket."""
if (status := client.loop_read(MAX_PACKETS_TO_READ)) != 0:
- self._async_on_disconnect(status)
+ self._async_handle_callback_exception(status)
@callback
def _async_start_misc_periodic(self) -> None:
@@ -593,7 +612,7 @@ class MQTT:
def _async_writer_callback(self, client: mqtt.Client) -> None:
"""Handle writing data to the socket."""
if (status := client.loop_write()) != 0:
- self._async_on_disconnect(status)
+ self._async_handle_callback_exception(status)
def _on_socket_register_write(
self, client: mqtt.Client, userdata: Any, sock: SocketType
@@ -652,14 +671,25 @@ class MQTT:
result: int | None = None
self._available_future = client_available
self._should_reconnect = True
+ connect_partial = partial(
+ self._mqttc.connect,
+ host=self.conf[CONF_BROKER],
+ port=self.conf.get(CONF_PORT, DEFAULT_PORT),
+ keepalive=self.conf.get(CONF_KEEPALIVE, DEFAULT_KEEPALIVE),
+ # See:
+ # https://eclipse.dev/paho/files/paho.mqtt.python/html/client.html
+ # `clean_start` (bool) – (MQTT v5.0 only) `True`, `False` or
+ # `MQTT_CLEAN_START_FIRST_ONLY`. Sets the MQTT v5.0 clean_start flag
+ # always, never or on the first successful connect only,
+ # respectively. MQTT session data (such as outstanding messages and
+ # subscriptions) is cleared on successful connect when the
+ # clean_start flag is set. For MQTT v3.1.1, the clean_session
+ # argument of Client should be used for similar result.
+ clean_start=True if self.is_mqttv5 else mqtt.MQTT_CLEAN_START_FIRST_ONLY,
+ )
try:
async with self._connection_lock, self._async_connect_in_executor():
- result = await self.hass.async_add_executor_job(
- self._mqttc.connect,
- self.conf[CONF_BROKER],
- self.conf.get(CONF_PORT, DEFAULT_PORT),
- self.conf.get(CONF_KEEPALIVE, DEFAULT_KEEPALIVE),
- )
+ result = await self.hass.async_add_executor_job(connect_partial)
except (OSError, mqtt.WebsocketConnectionError) as err:
_LOGGER.error("Failed to connect to MQTT server due to exception: %s", err)
self._async_connection_result(False)
@@ -983,29 +1013,28 @@ class MQTT:
self,
_mqttc: mqtt.Client,
_userdata: None,
- _flags: dict[str, int],
- result_code: int,
- properties: mqtt.Properties | None = None,
+ _connect_flags: mqtt.ConnectFlags,
+ reason_code: mqtt.ReasonCode,
+ _properties: mqtt.Properties | None = None,
) -> None:
"""On connect callback.
Resubscribe to all topics we were subscribed to and publish birth
message.
"""
- # pylint: disable-next=import-outside-toplevel
- import paho.mqtt.client as mqtt
-
- if result_code != mqtt.CONNACK_ACCEPTED:
- if result_code in (
- mqtt.CONNACK_REFUSED_BAD_USERNAME_PASSWORD,
- mqtt.CONNACK_REFUSED_NOT_AUTHORIZED,
- ):
+ if reason_code.is_failure:
+ # 24: Continue authentication
+ # 25: Re-authenticate
+ # 134: Bad user name or password
+ # 135: Not authorized
+ # 140: Bad authentication method
+ if reason_code.value in (24, 25, 134, 135, 140):
self._should_reconnect = False
self.hass.async_create_task(self.async_disconnect())
self.config_entry.async_start_reauth(self.hass)
_LOGGER.error(
"Unable to connect to the MQTT broker: %s",
- mqtt.connack_string(result_code),
+ reason_code.getName(), # type: ignore[no-untyped-call]
)
self._async_connection_result(False)
return
@@ -1016,7 +1045,7 @@ class MQTT:
"Connected to MQTT server %s:%s (%s)",
self.conf[CONF_BROKER],
self.conf.get(CONF_PORT, DEFAULT_PORT),
- result_code,
+ reason_code,
)
birth: dict[str, Any]
@@ -1153,18 +1182,32 @@ class MQTT:
self._mqtt_data.state_write_requests.process_write_state_requests(msg)
@callback
- def _async_mqtt_on_callback(
+ def _async_mqtt_on_publish(
self,
_mqttc: mqtt.Client,
_userdata: None,
mid: int,
- _granted_qos_reason: tuple[int, ...] | mqtt.ReasonCodes | None = None,
- _properties_reason: mqtt.ReasonCodes | None = None,
+ _reason_code: mqtt.ReasonCode,
+ _properties: mqtt.Properties | None,
) -> None:
+ """Publish callback."""
+ self._async_mqtt_on_callback(mid)
+
+ @callback
+ def _async_mqtt_on_subscribe_unsubscribe(
+ self,
+ _mqttc: mqtt.Client,
+ _userdata: None,
+ mid: int,
+ _reason_code: list[mqtt.ReasonCode],
+ _properties: mqtt.Properties | None,
+ ) -> None:
+ """Subscribe / Unsubscribe callback."""
+ self._async_mqtt_on_callback(mid)
+
+ @callback
+ def _async_mqtt_on_callback(self, mid: int) -> None:
"""Publish / Subscribe / Unsubscribe callback."""
- # The callback signature for on_unsubscribe is different from on_subscribe
- # see https://github.com/eclipse/paho.mqtt.python/issues/687
- # properties and reason codes are not used in Home Assistant
future = self._async_get_mid_future(mid)
if future.done() and (future.cancelled() or future.exception()):
# Timed out or cancelled
@@ -1180,19 +1223,28 @@ class MQTT:
self._pending_operations[mid] = future
return future
+ @callback
+ def _async_handle_callback_exception(self, status: mqtt.MQTTErrorCode) -> None:
+ """Handle a callback exception."""
+ # We don't import on the top because some integrations
+ # should be able to optionally rely on MQTT.
+ import paho.mqtt.client as mqtt # pylint: disable=import-outside-toplevel
+
+ _LOGGER.warning(
+ "Error returned from MQTT server: %s",
+ mqtt.error_string(status),
+ )
+
@callback
def _async_mqtt_on_disconnect(
self,
_mqttc: mqtt.Client,
_userdata: None,
- result_code: int,
+ _disconnect_flags: mqtt.DisconnectFlags,
+ reason_code: mqtt.ReasonCode,
properties: mqtt.Properties | None = None,
) -> None:
"""Disconnected callback."""
- self._async_on_disconnect(result_code)
-
- @callback
- def _async_on_disconnect(self, result_code: int) -> None:
if not self.connected:
# This function is re-entrant and may be called multiple times
# when there is a broken pipe error.
@@ -1203,11 +1255,11 @@ class MQTT:
self.connected = False
async_dispatcher_send(self.hass, MQTT_CONNECTION_STATE, False)
_LOGGER.log(
- logging.INFO if result_code == 0 else logging.DEBUG,
+ logging.INFO if reason_code == 0 else logging.DEBUG,
"Disconnected from MQTT server %s:%s (%s)",
self.conf[CONF_BROKER],
self.conf.get(CONF_PORT, DEFAULT_PORT),
- result_code,
+ reason_code,
)
@callback
diff --git a/homeassistant/components/mqtt/climate.py b/homeassistant/components/mqtt/climate.py
index 12619609f64..931a57a71cc 100644
--- a/homeassistant/components/mqtt/climate.py
+++ b/homeassistant/components/mqtt/climate.py
@@ -45,7 +45,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -113,11 +113,19 @@ CONF_PRESET_MODE_COMMAND_TOPIC = "preset_mode_command_topic"
CONF_PRESET_MODE_VALUE_TEMPLATE = "preset_mode_value_template"
CONF_PRESET_MODE_COMMAND_TEMPLATE = "preset_mode_command_template"
CONF_PRESET_MODES_LIST = "preset_modes"
+
+CONF_SWING_HORIZONTAL_MODE_COMMAND_TEMPLATE = "swing_horizontal_mode_command_template"
+CONF_SWING_HORIZONTAL_MODE_COMMAND_TOPIC = "swing_horizontal_mode_command_topic"
+CONF_SWING_HORIZONTAL_MODE_LIST = "swing_horizontal_modes"
+CONF_SWING_HORIZONTAL_MODE_STATE_TEMPLATE = "swing_horizontal_mode_state_template"
+CONF_SWING_HORIZONTAL_MODE_STATE_TOPIC = "swing_horizontal_mode_state_topic"
+
CONF_SWING_MODE_COMMAND_TEMPLATE = "swing_mode_command_template"
CONF_SWING_MODE_COMMAND_TOPIC = "swing_mode_command_topic"
CONF_SWING_MODE_LIST = "swing_modes"
CONF_SWING_MODE_STATE_TEMPLATE = "swing_mode_state_template"
CONF_SWING_MODE_STATE_TOPIC = "swing_mode_state_topic"
+
CONF_TEMP_HIGH_COMMAND_TEMPLATE = "temperature_high_command_template"
CONF_TEMP_HIGH_COMMAND_TOPIC = "temperature_high_command_topic"
CONF_TEMP_HIGH_STATE_TEMPLATE = "temperature_high_state_template"
@@ -145,6 +153,8 @@ MQTT_CLIMATE_ATTRIBUTES_BLOCKED = frozenset(
climate.ATTR_MIN_TEMP,
climate.ATTR_PRESET_MODE,
climate.ATTR_PRESET_MODES,
+ climate.ATTR_SWING_HORIZONTAL_MODE,
+ climate.ATTR_SWING_HORIZONTAL_MODES,
climate.ATTR_SWING_MODE,
climate.ATTR_SWING_MODES,
climate.ATTR_TARGET_TEMP_HIGH,
@@ -162,6 +172,7 @@ VALUE_TEMPLATE_KEYS = (
CONF_MODE_STATE_TEMPLATE,
CONF_ACTION_TEMPLATE,
CONF_PRESET_MODE_VALUE_TEMPLATE,
+ CONF_SWING_HORIZONTAL_MODE_STATE_TEMPLATE,
CONF_SWING_MODE_STATE_TEMPLATE,
CONF_TEMP_HIGH_STATE_TEMPLATE,
CONF_TEMP_LOW_STATE_TEMPLATE,
@@ -174,6 +185,7 @@ COMMAND_TEMPLATE_KEYS = {
CONF_MODE_COMMAND_TEMPLATE,
CONF_POWER_COMMAND_TEMPLATE,
CONF_PRESET_MODE_COMMAND_TEMPLATE,
+ CONF_SWING_HORIZONTAL_MODE_COMMAND_TEMPLATE,
CONF_SWING_MODE_COMMAND_TEMPLATE,
CONF_TEMP_COMMAND_TEMPLATE,
CONF_TEMP_HIGH_COMMAND_TEMPLATE,
@@ -194,6 +206,8 @@ TOPIC_KEYS = (
CONF_POWER_COMMAND_TOPIC,
CONF_PRESET_MODE_COMMAND_TOPIC,
CONF_PRESET_MODE_STATE_TOPIC,
+ CONF_SWING_HORIZONTAL_MODE_COMMAND_TOPIC,
+ CONF_SWING_HORIZONTAL_MODE_STATE_TOPIC,
CONF_SWING_MODE_COMMAND_TOPIC,
CONF_SWING_MODE_STATE_TOPIC,
CONF_TEMP_COMMAND_TOPIC,
@@ -302,6 +316,13 @@ _PLATFORM_SCHEMA_BASE = MQTT_BASE_SCHEMA.extend(
vol.Optional(CONF_PRESET_MODE_COMMAND_TEMPLATE): cv.template,
vol.Optional(CONF_PRESET_MODE_STATE_TOPIC): valid_subscribe_topic,
vol.Optional(CONF_PRESET_MODE_VALUE_TEMPLATE): cv.template,
+ vol.Optional(CONF_SWING_HORIZONTAL_MODE_COMMAND_TEMPLATE): cv.template,
+ vol.Optional(CONF_SWING_HORIZONTAL_MODE_COMMAND_TOPIC): valid_publish_topic,
+ vol.Optional(
+ CONF_SWING_HORIZONTAL_MODE_LIST, default=[SWING_ON, SWING_OFF]
+ ): cv.ensure_list,
+ vol.Optional(CONF_SWING_HORIZONTAL_MODE_STATE_TEMPLATE): cv.template,
+ vol.Optional(CONF_SWING_HORIZONTAL_MODE_STATE_TOPIC): valid_subscribe_topic,
vol.Optional(CONF_SWING_MODE_COMMAND_TEMPLATE): cv.template,
vol.Optional(CONF_SWING_MODE_COMMAND_TOPIC): valid_publish_topic,
vol.Optional(
@@ -350,7 +371,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT climate through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
@@ -515,6 +536,7 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
_attr_fan_mode: str | None = None
_attr_hvac_mode: HVACMode | None = None
+ _attr_swing_horizontal_mode: str | None = None
_attr_swing_mode: str | None = None
_default_name = DEFAULT_NAME
_entity_id_format = climate.ENTITY_ID_FORMAT
@@ -543,6 +565,7 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
if (precision := config.get(CONF_PRECISION)) is not None:
self._attr_precision = precision
self._attr_fan_modes = config[CONF_FAN_MODE_LIST]
+ self._attr_swing_horizontal_modes = config[CONF_SWING_HORIZONTAL_MODE_LIST]
self._attr_swing_modes = config[CONF_SWING_MODE_LIST]
self._attr_target_temperature_step = config[CONF_TEMP_STEP]
@@ -568,6 +591,11 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
if self._topic[CONF_FAN_MODE_STATE_TOPIC] is None or self._optimistic:
self._attr_fan_mode = FAN_LOW
+ if (
+ self._topic[CONF_SWING_HORIZONTAL_MODE_STATE_TOPIC] is None
+ or self._optimistic
+ ):
+ self._attr_swing_horizontal_mode = SWING_OFF
if self._topic[CONF_SWING_MODE_STATE_TOPIC] is None or self._optimistic:
self._attr_swing_mode = SWING_OFF
if self._topic[CONF_MODE_STATE_TOPIC] is None or self._optimistic:
@@ -629,6 +657,11 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
):
support |= ClimateEntityFeature.FAN_MODE
+ if (self._topic[CONF_SWING_HORIZONTAL_MODE_STATE_TOPIC] is not None) or (
+ self._topic[CONF_SWING_HORIZONTAL_MODE_COMMAND_TOPIC] is not None
+ ):
+ support |= ClimateEntityFeature.SWING_HORIZONTAL_MODE
+
if (self._topic[CONF_SWING_MODE_STATE_TOPIC] is not None) or (
self._topic[CONF_SWING_MODE_COMMAND_TOPIC] is not None
):
@@ -744,6 +777,16 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
),
{"_attr_fan_mode"},
)
+ self.add_subscription(
+ CONF_SWING_HORIZONTAL_MODE_STATE_TOPIC,
+ partial(
+ self._handle_mode_received,
+ CONF_SWING_HORIZONTAL_MODE_STATE_TEMPLATE,
+ "_attr_swing_horizontal_mode",
+ CONF_SWING_HORIZONTAL_MODE_LIST,
+ ),
+ {"_attr_swing_horizontal_mode"},
+ )
self.add_subscription(
CONF_SWING_MODE_STATE_TOPIC,
partial(
@@ -782,6 +825,20 @@ class MqttClimate(MqttTemperatureControlEntity, ClimateEntity):
self.async_write_ha_state()
+ async def async_set_swing_horizontal_mode(self, swing_horizontal_mode: str) -> None:
+ """Set new swing horizontal mode."""
+ payload = self._command_templates[CONF_SWING_HORIZONTAL_MODE_COMMAND_TEMPLATE](
+ swing_horizontal_mode
+ )
+ await self._publish(CONF_SWING_HORIZONTAL_MODE_COMMAND_TOPIC, payload)
+
+ if (
+ self._optimistic
+ or self._topic[CONF_SWING_HORIZONTAL_MODE_STATE_TOPIC] is None
+ ):
+ self._attr_swing_horizontal_mode = swing_horizontal_mode
+ self.async_write_ha_state()
+
async def async_set_swing_mode(self, swing_mode: str) -> None:
"""Set new swing mode."""
payload = self._command_templates[CONF_SWING_MODE_COMMAND_TEMPLATE](swing_mode)
diff --git a/homeassistant/components/mqtt/config_flow.py b/homeassistant/components/mqtt/config_flow.py
index a9d417fc783..8dfccbb6b2a 100644
--- a/homeassistant/components/mqtt/config_flow.py
+++ b/homeassistant/components/mqtt/config_flow.py
@@ -5,14 +5,24 @@ from __future__ import annotations
import asyncio
from collections import OrderedDict
from collections.abc import Callable, Mapping
+from copy import deepcopy
+from dataclasses import dataclass
+from enum import IntEnum
import logging
import queue
from ssl import PROTOCOL_TLS_CLIENT, SSLContext, SSLError
from types import MappingProxyType
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, cast
+from uuid import uuid4
-from cryptography.hazmat.primitives.serialization import load_pem_private_key
-from cryptography.x509 import load_pem_x509_certificate
+from cryptography.hazmat.primitives.serialization import (
+ Encoding,
+ NoEncryption,
+ PrivateFormat,
+ load_der_private_key,
+ load_pem_private_key,
+)
+from cryptography.x509 import load_der_x509_certificate, load_pem_x509_certificate
import voluptuous as vol
from homeassistant.components.file_upload import process_uploaded_file
@@ -22,21 +32,32 @@ from homeassistant.config_entries import (
ConfigEntry,
ConfigFlow,
ConfigFlowResult,
+ ConfigSubentryFlow,
OptionsFlow,
+ SubentryFlowResult,
)
from homeassistant.const import (
+ ATTR_CONFIGURATION_URL,
+ ATTR_HW_VERSION,
+ ATTR_MODEL,
+ ATTR_MODEL_ID,
+ ATTR_NAME,
+ ATTR_SW_VERSION,
CONF_CLIENT_ID,
+ CONF_DEVICE,
CONF_DISCOVERY,
CONF_HOST,
+ CONF_NAME,
CONF_PASSWORD,
CONF_PAYLOAD,
+ CONF_PLATFORM,
CONF_PORT,
CONF_PROTOCOL,
CONF_USERNAME,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.data_entry_flow import AbortFlow
-from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers import config_validation as cv, entity_registry as er
from homeassistant.helpers.hassio import is_hassio
from homeassistant.helpers.json import json_dumps
from homeassistant.helpers.selector import (
@@ -47,9 +68,12 @@ from homeassistant.helpers.selector import (
NumberSelectorConfig,
NumberSelectorMode,
SelectOptionDict,
+ Selector,
SelectSelector,
SelectSelectorConfig,
SelectSelectorMode,
+ TemplateSelector,
+ TemplateSelectorConfig,
TextSelector,
TextSelectorConfig,
TextSelectorType,
@@ -64,13 +88,22 @@ from .const import (
ATTR_QOS,
ATTR_RETAIN,
ATTR_TOPIC,
+ CONF_AVAILABILITY_TEMPLATE,
+ CONF_AVAILABILITY_TOPIC,
CONF_BIRTH_MESSAGE,
CONF_BROKER,
CONF_CERTIFICATE,
CONF_CLIENT_CERT,
CONF_CLIENT_KEY,
+ CONF_COMMAND_TEMPLATE,
+ CONF_COMMAND_TOPIC,
CONF_DISCOVERY_PREFIX,
+ CONF_ENTITY_PICTURE,
CONF_KEEPALIVE,
+ CONF_PAYLOAD_AVAILABLE,
+ CONF_PAYLOAD_NOT_AVAILABLE,
+ CONF_QOS,
+ CONF_RETAIN,
CONF_TLS_INSECURE,
CONF_TRANSPORT,
CONF_WILL_MESSAGE,
@@ -82,6 +115,8 @@ from .const import (
DEFAULT_DISCOVERY,
DEFAULT_ENCODING,
DEFAULT_KEEPALIVE,
+ DEFAULT_PAYLOAD_AVAILABLE,
+ DEFAULT_PAYLOAD_NOT_AVAILABLE,
DEFAULT_PORT,
DEFAULT_PREFIX,
DEFAULT_PROTOCOL,
@@ -92,12 +127,17 @@ from .const import (
SUPPORTED_PROTOCOLS,
TRANSPORT_TCP,
TRANSPORT_WEBSOCKETS,
+ Platform,
)
+from .models import MqttAvailabilityData, MqttDeviceData, MqttSubentryData
from .util import (
async_create_certificate_temp_files,
get_file_path,
valid_birth_will,
valid_publish_topic,
+ valid_qos_schema,
+ valid_subscribe_topic,
+ valid_subscribe_topic_template,
)
_LOGGER = logging.getLogger(__name__)
@@ -105,6 +145,8 @@ _LOGGER = logging.getLogger(__name__)
ADDON_SETUP_TIMEOUT = 5
ADDON_SETUP_TIMEOUT_ROUNDS = 5
+CONF_CLIENT_KEY_PASSWORD = "client_key_password"
+
MQTT_TIMEOUT = 5
ADVANCED_OPTIONS = "advanced_options"
@@ -119,10 +161,10 @@ PORT_SELECTOR = vol.All(
vol.Coerce(int),
)
PASSWORD_SELECTOR = TextSelector(TextSelectorConfig(type=TextSelectorType.PASSWORD))
-QOS_SELECTOR = vol.All(
- NumberSelector(NumberSelectorConfig(mode=NumberSelectorMode.BOX, min=0, max=2)),
- vol.Coerce(int),
+QOS_SELECTOR = NumberSelector(
+ NumberSelectorConfig(mode=NumberSelectorMode.BOX, min=0, max=2)
)
+QOS_DATA_SCHEMA = vol.All(QOS_SELECTOR, valid_qos_schema)
KEEPALIVE_SELECTOR = vol.All(
NumberSelector(
NumberSelectorConfig(
@@ -165,12 +207,86 @@ BROKER_VERIFICATION_SELECTOR = SelectSelector(
# mime configuration from https://pki-tutorial.readthedocs.io/en/latest/mime.html
CA_CERT_UPLOAD_SELECTOR = FileSelector(
- FileSelectorConfig(accept=".crt,application/x-x509-ca-cert")
+ FileSelectorConfig(accept=".pem,.crt,.cer,.der,application/x-x509-ca-cert")
)
CERT_UPLOAD_SELECTOR = FileSelector(
- FileSelectorConfig(accept=".crt,application/x-x509-user-cert")
+ FileSelectorConfig(accept=".pem,.crt,.cer,.der,application/x-x509-user-cert")
+)
+KEY_UPLOAD_SELECTOR = FileSelector(
+ FileSelectorConfig(accept=".pem,.key,.der,.pk8,application/pkcs8")
+)
+
+# Subentry selectors
+SUBENTRY_PLATFORMS = [Platform.NOTIFY]
+SUBENTRY_PLATFORM_SELECTOR = SelectSelector(
+ SelectSelectorConfig(
+ options=[platform.value for platform in SUBENTRY_PLATFORMS],
+ mode=SelectSelectorMode.DROPDOWN,
+ translation_key=CONF_PLATFORM,
+ )
+)
+
+TEMPLATE_SELECTOR = TemplateSelector(TemplateSelectorConfig())
+
+SUBENTRY_AVAILABILITY_SCHEMA = vol.Schema(
+ {
+ vol.Optional(CONF_AVAILABILITY_TOPIC): TEXT_SELECTOR,
+ vol.Optional(CONF_AVAILABILITY_TEMPLATE): TEMPLATE_SELECTOR,
+ vol.Optional(
+ CONF_PAYLOAD_AVAILABLE, default=DEFAULT_PAYLOAD_AVAILABLE
+ ): TEXT_SELECTOR,
+ vol.Optional(
+ CONF_PAYLOAD_NOT_AVAILABLE, default=DEFAULT_PAYLOAD_NOT_AVAILABLE
+ ): TEXT_SELECTOR,
+ }
+)
+
+
+@dataclass(frozen=True)
+class PlatformField:
+ """Stores a platform config field schema, required flag and validator."""
+
+ selector: Selector
+ required: bool
+ validator: Callable[..., Any]
+ error: str | None = None
+ default: str | int | vol.Undefined = vol.UNDEFINED
+ exclude_from_reconfig: bool = False
+
+
+COMMON_ENTITY_FIELDS = {
+ CONF_PLATFORM: PlatformField(
+ SUBENTRY_PLATFORM_SELECTOR, True, str, exclude_from_reconfig=True
+ ),
+ CONF_NAME: PlatformField(TEXT_SELECTOR, False, str, exclude_from_reconfig=True),
+ CONF_ENTITY_PICTURE: PlatformField(TEXT_SELECTOR, False, cv.url, "invalid_url"),
+}
+
+COMMON_MQTT_FIELDS = {
+ CONF_QOS: PlatformField(QOS_SELECTOR, False, valid_qos_schema, default=0),
+ CONF_RETAIN: PlatformField(BOOLEAN_SELECTOR, False, bool),
+}
+PLATFORM_MQTT_FIELDS = {
+ Platform.NOTIFY.value: {
+ CONF_COMMAND_TOPIC: PlatformField(
+ TEXT_SELECTOR, True, valid_publish_topic, "invalid_publish_topic"
+ ),
+ CONF_COMMAND_TEMPLATE: PlatformField(
+ TEMPLATE_SELECTOR, False, cv.template, "invalid_template"
+ ),
+ },
+}
+
+MQTT_DEVICE_SCHEMA = vol.Schema(
+ {
+ vol.Required(ATTR_NAME): TEXT_SELECTOR,
+ vol.Optional(ATTR_SW_VERSION): TEXT_SELECTOR,
+ vol.Optional(ATTR_HW_VERSION): TEXT_SELECTOR,
+ vol.Optional(ATTR_MODEL): TEXT_SELECTOR,
+ vol.Optional(ATTR_MODEL_ID): TEXT_SELECTOR,
+ vol.Optional(ATTR_CONFIGURATION_URL): TEXT_SELECTOR,
+ }
)
-KEY_UPLOAD_SELECTOR = FileSelector(FileSelectorConfig(accept=".key,application/pkcs8"))
REAUTH_SCHEMA = vol.Schema(
{
@@ -204,6 +320,57 @@ def update_password_from_user_input(
return substituted_used_data
+@callback
+def validate_field(
+ field: str,
+ validator: Callable[..., Any],
+ user_input: dict[str, Any] | None,
+ errors: dict[str, str],
+ error: str,
+) -> None:
+ """Validate a single field."""
+ if user_input is None or field not in user_input:
+ return
+ try:
+ validator(user_input[field])
+ except (ValueError, vol.Invalid):
+ errors[field] = error
+
+
+@callback
+def validate_user_input(
+ user_input: dict[str, Any],
+ data_schema_fields: dict[str, PlatformField],
+ errors: dict[str, str],
+) -> None:
+ """Validate user input."""
+ for field, value in user_input.items():
+ validator = data_schema_fields[field].validator
+ try:
+ validator(value)
+ except (ValueError, vol.Invalid):
+ errors[field] = data_schema_fields[field].error or "invalid_input"
+
+
+@callback
+def data_schema_from_fields(
+ data_schema_fields: dict[str, PlatformField],
+ reconfig: bool,
+) -> vol.Schema:
+ """Generate data schema from platform fields."""
+ return vol.Schema(
+ {
+ vol.Required(field_name, default=field_details.default)
+ if field_details.required
+ else vol.Optional(
+ field_name, default=field_details.default
+ ): field_details.selector
+ for field_name, field_details in data_schema_fields.items()
+ if not field_details.exclude_from_reconfig or not reconfig
+ }
+ )
+
+
class FlowHandler(ConfigFlow, domain=DOMAIN):
"""Handle a config flow."""
@@ -219,6 +386,14 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
self.install_task: asyncio.Task | None = None
self.start_task: asyncio.Task | None = None
+ @classmethod
+ @callback
+ def async_get_supported_subentry_types(
+ cls, config_entry: ConfigEntry
+ ) -> dict[str, type[ConfigSubentryFlow]]:
+ """Return subentries supported by this handler."""
+ return {CONF_DEVICE: MQTTSubentryFlowHandler}
+
@staticmethod
@callback
def async_get_options_flow(
@@ -674,7 +849,7 @@ class MQTTOptionsFlowHandler(OptionsFlow):
"birth_payload", description={"suggested_value": birth[CONF_PAYLOAD]}
)
] = TEXT_SELECTOR
- fields[vol.Optional("birth_qos", default=birth[ATTR_QOS])] = QOS_SELECTOR
+ fields[vol.Optional("birth_qos", default=birth[ATTR_QOS])] = QOS_DATA_SCHEMA
fields[vol.Optional("birth_retain", default=birth[ATTR_RETAIN])] = (
BOOLEAN_SELECTOR
)
@@ -697,7 +872,7 @@ class MQTTOptionsFlowHandler(OptionsFlow):
"will_payload", description={"suggested_value": will[CONF_PAYLOAD]}
)
] = TEXT_SELECTOR
- fields[vol.Optional("will_qos", default=will[ATTR_QOS])] = QOS_SELECTOR
+ fields[vol.Optional("will_qos", default=will[ATTR_QOS])] = QOS_DATA_SCHEMA
fields[vol.Optional("will_retain", default=will[ATTR_RETAIN])] = (
BOOLEAN_SELECTOR
)
@@ -710,17 +885,408 @@ class MQTTOptionsFlowHandler(OptionsFlow):
)
-async def _get_uploaded_file(hass: HomeAssistant, id: str) -> str:
- """Get file content from uploaded file."""
+class MQTTSubentryFlowHandler(ConfigSubentryFlow):
+ """Handle MQTT subentry flow."""
- def _proces_uploaded_file() -> str:
+ _subentry_data: MqttSubentryData
+ _component_id: str | None = None
+
+ @callback
+ def update_component_fields(
+ self, data_schema: vol.Schema, user_input: dict[str, Any]
+ ) -> None:
+ """Update the componment fields."""
+ if TYPE_CHECKING:
+ assert self._component_id is not None
+ component_data = self._subentry_data["components"][self._component_id]
+ # Remove the fields from the component data if they are not in the user input
+ for field in [
+ form_field
+ for form_field in data_schema.schema
+ if form_field in component_data and form_field not in user_input
+ ]:
+ component_data.pop(field)
+ component_data.update(user_input)
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Add a subentry."""
+ self._subentry_data = MqttSubentryData(device=MqttDeviceData(), components={})
+ return await self.async_step_device()
+
+ async def async_step_reconfigure(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Reconfigure a subentry."""
+ reconfigure_subentry = self._get_reconfigure_subentry()
+ self._subentry_data = cast(
+ MqttSubentryData, deepcopy(dict(reconfigure_subentry.data))
+ )
+ return await self.async_step_summary_menu()
+
+ async def async_step_device(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Add a new MQTT device."""
+ errors: dict[str, str] = {}
+ validate_field("configuration_url", cv.url, user_input, errors, "invalid_url")
+ if not errors and user_input is not None:
+ self._subentry_data[CONF_DEVICE] = cast(MqttDeviceData, user_input)
+ if self.source == SOURCE_RECONFIGURE:
+ return await self.async_step_summary_menu()
+ return await self.async_step_entity()
+
+ data_schema = self.add_suggested_values_to_schema(
+ MQTT_DEVICE_SCHEMA,
+ self._subentry_data[CONF_DEVICE] if user_input is None else user_input,
+ )
+ return self.async_show_form(
+ step_id=CONF_DEVICE,
+ data_schema=data_schema,
+ errors=errors,
+ last_step=False,
+ )
+
+ async def async_step_entity(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Add or edit an mqtt entity."""
+ errors: dict[str, str] = {}
+ data_schema_fields = COMMON_ENTITY_FIELDS
+ entity_name_label: str = ""
+ platform_label: str = ""
+ if reconfig := (self._component_id is not None):
+ name: str | None = self._subentry_data["components"][
+ self._component_id
+ ].get(CONF_NAME)
+ platform_label = f"{self._subentry_data['components'][self._component_id][CONF_PLATFORM]} "
+ entity_name_label = f" ({name})" if name is not None else ""
+ data_schema = data_schema_from_fields(data_schema_fields, reconfig=reconfig)
+ if user_input is not None:
+ validate_user_input(user_input, data_schema_fields, errors)
+ if not errors:
+ if self._component_id is None:
+ self._component_id = uuid4().hex
+ self._subentry_data["components"].setdefault(self._component_id, {})
+ self.update_component_fields(data_schema, user_input)
+ return await self.async_step_mqtt_platform_config()
+ data_schema = self.add_suggested_values_to_schema(data_schema, user_input)
+ elif self.source == SOURCE_RECONFIGURE and self._component_id is not None:
+ data_schema = self.add_suggested_values_to_schema(
+ data_schema, self._subentry_data["components"][self._component_id]
+ )
+ device_name = self._subentry_data[CONF_DEVICE][CONF_NAME]
+ return self.async_show_form(
+ step_id="entity",
+ data_schema=data_schema,
+ description_placeholders={
+ "mqtt_device": device_name,
+ "entity_name_label": entity_name_label,
+ "platform_label": platform_label,
+ },
+ errors=errors,
+ last_step=False,
+ )
+
+ def _show_update_or_delete_form(self, step_id: str) -> SubentryFlowResult:
+ """Help selecting an entity to update or delete."""
+ device_name = self._subentry_data[CONF_DEVICE][CONF_NAME]
+ entities = [
+ SelectOptionDict(
+ value=key, label=f"{device_name} {component.get(CONF_NAME, '-')}"
+ )
+ for key, component in self._subentry_data["components"].items()
+ ]
+ data_schema = vol.Schema(
+ {
+ vol.Required("component"): SelectSelector(
+ SelectSelectorConfig(
+ options=entities,
+ mode=SelectSelectorMode.LIST,
+ )
+ )
+ }
+ )
+ return self.async_show_form(
+ step_id=step_id, data_schema=data_schema, last_step=False
+ )
+
+ async def async_step_update_entity(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Select the entity to update."""
+ if user_input:
+ self._component_id = user_input["component"]
+ return await self.async_step_entity()
+ if len(self._subentry_data["components"]) == 1:
+ # Return first key
+ self._component_id = next(iter(self._subentry_data["components"]))
+ return await self.async_step_entity()
+ return self._show_update_or_delete_form("update_entity")
+
+ async def async_step_delete_entity(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Select the entity to delete."""
+ if user_input:
+ del self._subentry_data["components"][user_input["component"]]
+ return await self.async_step_summary_menu()
+ return self._show_update_or_delete_form("delete_entity")
+
+ async def async_step_mqtt_platform_config(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Configure entity platform MQTT details."""
+ errors: dict[str, str] = {}
+ if TYPE_CHECKING:
+ assert self._component_id is not None
+ platform = self._subentry_data["components"][self._component_id][CONF_PLATFORM]
+ data_schema_fields = PLATFORM_MQTT_FIELDS[platform] | COMMON_MQTT_FIELDS
+ data_schema = data_schema_from_fields(
+ data_schema_fields, reconfig=self._component_id is not None
+ )
+ if user_input is not None:
+ # Test entity fields against the validator
+ validate_user_input(user_input, data_schema_fields, errors)
+ if not errors:
+ self.update_component_fields(data_schema, user_input)
+ self._component_id = None
+ if self.source == SOURCE_RECONFIGURE:
+ return await self.async_step_summary_menu()
+ return self._async_create_subentry()
+
+ data_schema = self.add_suggested_values_to_schema(data_schema, user_input)
+ else:
+ data_schema = self.add_suggested_values_to_schema(
+ data_schema, self._subentry_data["components"][self._component_id]
+ )
+ device_name = self._subentry_data[CONF_DEVICE][CONF_NAME]
+ entity_name: str | None
+ if entity_name := self._subentry_data["components"][self._component_id].get(
+ CONF_NAME
+ ):
+ full_entity_name: str = f"{device_name} {entity_name}"
+ else:
+ full_entity_name = device_name
+ return self.async_show_form(
+ step_id="mqtt_platform_config",
+ data_schema=data_schema,
+ description_placeholders={
+ "mqtt_device": device_name,
+ CONF_PLATFORM: platform,
+ "entity": full_entity_name,
+ },
+ errors=errors,
+ last_step=False,
+ )
+
+ @callback
+ def _async_create_subentry(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Create a subentry for a new MQTT device."""
+ device_name = self._subentry_data[CONF_DEVICE][CONF_NAME]
+ component: dict[str, Any] = next(
+ iter(self._subentry_data["components"].values())
+ )
+ platform = component[CONF_PLATFORM]
+ entity_name: str | None
+ if entity_name := component.get(CONF_NAME):
+ full_entity_name: str = f"{device_name} {entity_name}"
+ else:
+ full_entity_name = device_name
+
+ return self.async_create_entry(
+ data=self._subentry_data,
+ title=self._subentry_data[CONF_DEVICE][CONF_NAME],
+ description_placeholders={
+ "entity": full_entity_name,
+ CONF_PLATFORM: platform,
+ },
+ )
+
+ async def async_step_availability(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Configure availability options."""
+ errors: dict[str, str] = {}
+ validate_field(
+ "availability_topic",
+ valid_subscribe_topic,
+ user_input,
+ errors,
+ "invalid_subscribe_topic",
+ )
+ validate_field(
+ "availability_template",
+ valid_subscribe_topic_template,
+ user_input,
+ errors,
+ "invalid_template",
+ )
+ if not errors and user_input is not None:
+ self._subentry_data.setdefault("availability", MqttAvailabilityData())
+ self._subentry_data["availability"] = cast(MqttAvailabilityData, user_input)
+ return await self.async_step_summary_menu()
+
+ data_schema = SUBENTRY_AVAILABILITY_SCHEMA
+ data_schema = self.add_suggested_values_to_schema(
+ data_schema,
+ dict(self._subentry_data.setdefault("availability", {}))
+ if self.source == SOURCE_RECONFIGURE
+ else user_input,
+ )
+ return self.async_show_form(
+ step_id="availability",
+ data_schema=data_schema,
+ errors=errors,
+ last_step=False,
+ )
+
+ async def async_step_summary_menu(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Show summary menu and decide to add more entities or to finish the flow."""
+ self._component_id = None
+ mqtt_device = self._subentry_data[CONF_DEVICE][CONF_NAME]
+ mqtt_items = ", ".join(
+ f"{mqtt_device} {component.get(CONF_NAME, '-')}"
+ for component in self._subentry_data["components"].values()
+ )
+ menu_options = [
+ "entity",
+ "update_entity",
+ ]
+ if len(self._subentry_data["components"]) > 1:
+ menu_options.append("delete_entity")
+ menu_options.extend(["device", "availability"])
+ if self._subentry_data != self._get_reconfigure_subentry().data:
+ menu_options.append("save_changes")
+ return self.async_show_menu(
+ step_id="summary_menu",
+ menu_options=menu_options,
+ description_placeholders={
+ "mqtt_device": mqtt_device,
+ "mqtt_items": mqtt_items,
+ },
+ )
+
+ async def async_step_save_changes(
+ self, user_input: dict[str, Any] | None = None
+ ) -> SubentryFlowResult:
+ """Save the changes made to the subentry."""
+ entry = self._get_reconfigure_entry()
+ subentry = self._get_reconfigure_subentry()
+ entity_registry = er.async_get(self.hass)
+
+ # When a component is removed from the MQTT device,
+ # And we save the changes to the subentry,
+ # we need to clean up stale entity registry entries.
+ # The component id is used as a part of the unique id of the entity.
+ for unique_id, platform in [
+ (
+ f"{subentry.subentry_id}_{component_id}",
+ subentry.data["components"][component_id][CONF_PLATFORM],
+ )
+ for component_id in subentry.data["components"]
+ if component_id not in self._subentry_data["components"]
+ ]:
+ if entity_id := entity_registry.async_get_entity_id(
+ platform, DOMAIN, unique_id
+ ):
+ entity_registry.async_remove(entity_id)
+
+ return self.async_update_and_abort(
+ entry,
+ subentry,
+ data=self._subentry_data,
+ title=self._subentry_data[CONF_DEVICE][CONF_NAME],
+ )
+
+
+@callback
+def async_is_pem_data(data: bytes) -> bool:
+ """Return True if data is in PEM format."""
+ return (
+ b"-----BEGIN CERTIFICATE-----" in data
+ or b"-----BEGIN PRIVATE KEY-----" in data
+ or b"-----BEGIN RSA PRIVATE KEY-----" in data
+ or b"-----BEGIN ENCRYPTED PRIVATE KEY-----" in data
+ )
+
+
+class PEMType(IntEnum):
+ """Type of PEM data."""
+
+ CERTIFICATE = 1
+ PRIVATE_KEY = 2
+
+
+@callback
+def async_convert_to_pem(
+ data: bytes, pem_type: PEMType, password: str | None = None
+) -> str | None:
+ """Convert data to PEM format."""
+ try:
+ if async_is_pem_data(data):
+ if not password:
+ # Assume unencrypted PEM encoded private key
+ return data.decode(DEFAULT_ENCODING)
+ # Return decrypted PEM encoded private key
+ return (
+ load_pem_private_key(data, password=password.encode(DEFAULT_ENCODING))
+ .private_bytes(
+ encoding=Encoding.PEM,
+ format=PrivateFormat.TraditionalOpenSSL,
+ encryption_algorithm=NoEncryption(),
+ )
+ .decode(DEFAULT_ENCODING)
+ )
+ # Convert from DER encoding to PEM
+ if pem_type == PEMType.CERTIFICATE:
+ return (
+ load_der_x509_certificate(data)
+ .public_bytes(
+ encoding=Encoding.PEM,
+ )
+ .decode(DEFAULT_ENCODING)
+ )
+ # Assume DER encoded private key
+ pem_key_data: bytes = load_der_private_key(
+ data, password.encode(DEFAULT_ENCODING) if password else None
+ ).private_bytes(
+ encoding=Encoding.PEM,
+ format=PrivateFormat.TraditionalOpenSSL,
+ encryption_algorithm=NoEncryption(),
+ )
+ return pem_key_data.decode("utf-8")
+ except (TypeError, ValueError, SSLError):
+ _LOGGER.exception("Error converting %s file data to PEM format", pem_type.name)
+ return None
+
+
+async def _get_uploaded_file(hass: HomeAssistant, id: str) -> bytes:
+ """Get file content from uploaded certificate or key file."""
+
+ def _proces_uploaded_file() -> bytes:
with process_uploaded_file(hass, id) as file_path:
- return file_path.read_text(encoding=DEFAULT_ENCODING)
+ return file_path.read_bytes()
return await hass.async_add_executor_job(_proces_uploaded_file)
-async def async_get_broker_settings(
+def _validate_pki_file(
+ file_id: str | None, pem_data: str | None, errors: dict[str, str], error: str
+) -> bool:
+ """Return False if uploaded file could not be converted to PEM format."""
+ if file_id and not pem_data:
+ errors["base"] = error
+ return False
+ return True
+
+
+async def async_get_broker_settings( # noqa: C901
flow: ConfigFlow | OptionsFlow,
fields: OrderedDict[Any, Any],
entry_config: MappingProxyType[str, Any] | None,
@@ -768,6 +1334,10 @@ async def async_get_broker_settings(
validated_user_input.update(user_input)
client_certificate_id: str | None = user_input.get(CONF_CLIENT_CERT)
client_key_id: str | None = user_input.get(CONF_CLIENT_KEY)
+ # We do not store the private key password in the entry data
+ client_key_password: str | None = validated_user_input.pop(
+ CONF_CLIENT_KEY_PASSWORD, None
+ )
if (client_certificate_id and not client_key_id) or (
not client_certificate_id and client_key_id
):
@@ -775,7 +1345,14 @@ async def async_get_broker_settings(
return False
certificate_id: str | None = user_input.get(CONF_CERTIFICATE)
if certificate_id:
- certificate = await _get_uploaded_file(hass, certificate_id)
+ certificate_data_raw = await _get_uploaded_file(hass, certificate_id)
+ certificate = async_convert_to_pem(
+ certificate_data_raw, PEMType.CERTIFICATE
+ )
+ if not _validate_pki_file(
+ certificate_id, certificate, errors, "bad_certificate"
+ ):
+ return False
# Return to form for file upload CA cert or client cert and key
if (
@@ -797,9 +1374,26 @@ async def async_get_broker_settings(
return False
if client_certificate_id:
- client_certificate = await _get_uploaded_file(hass, client_certificate_id)
+ client_certificate_data = await _get_uploaded_file(
+ hass, client_certificate_id
+ )
+ client_certificate = async_convert_to_pem(
+ client_certificate_data, PEMType.CERTIFICATE
+ )
+ if not _validate_pki_file(
+ client_certificate_id, client_certificate, errors, "bad_client_cert"
+ ):
+ return False
+
if client_key_id:
- client_key = await _get_uploaded_file(hass, client_key_id)
+ client_key_data = await _get_uploaded_file(hass, client_key_id)
+ client_key = async_convert_to_pem(
+ client_key_data, PEMType.PRIVATE_KEY, password=client_key_password
+ )
+ if not _validate_pki_file(
+ client_key_id, client_key, errors, "client_key_error"
+ ):
+ return False
certificate_data: dict[str, Any] = {}
if certificate:
@@ -956,6 +1550,14 @@ async def async_get_broker_settings(
description={"suggested_value": user_input_basic.get(CONF_CLIENT_KEY)},
)
] = KEY_UPLOAD_SELECTOR
+ fields[
+ vol.Optional(
+ CONF_CLIENT_KEY_PASSWORD,
+ description={
+ "suggested_value": user_input_basic.get(CONF_CLIENT_KEY_PASSWORD)
+ },
+ )
+ ] = PASSWORD_SELECTOR
verification_mode = current_config.get(SET_CA_CERT) or (
"off"
if current_ca_certificate is None
@@ -1023,14 +1625,14 @@ def try_connection(
result: queue.Queue[bool] = queue.Queue(maxsize=1)
def on_connect(
- client_: mqtt.Client,
- userdata: None,
- flags: dict[str, Any],
- result_code: int,
- properties: mqtt.Properties | None = None,
+ _mqttc: mqtt.Client,
+ _userdata: None,
+ _connect_flags: mqtt.ConnectFlags,
+ reason_code: mqtt.ReasonCode,
+ _properties: mqtt.Properties | None = None,
) -> None:
"""Handle connection result."""
- result.put(result_code == mqtt.CONNACK_ACCEPTED)
+ result.put(not reason_code.is_failure)
client.on_connect = on_connect
@@ -1060,7 +1662,7 @@ def check_certicate_chain() -> str | None:
with open(private_key, "rb") as client_key_file:
load_pem_private_key(client_key_file.read(), password=None)
except (TypeError, ValueError):
- return "bad_client_key"
+ return "client_key_error"
# Check the certificate chain
context = SSLContext(PROTOCOL_TLS_CLIENT)
if client_certificate and private_key:
diff --git a/homeassistant/components/mqtt/cover.py b/homeassistant/components/mqtt/cover.py
index 626e0cef64a..428c4d0e205 100644
--- a/homeassistant/components/mqtt/cover.py
+++ b/homeassistant/components/mqtt/cover.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from homeassistant.util.json import JSON_DECODE_EXCEPTIONS, json_loads
@@ -81,6 +81,7 @@ CONF_TILT_STATUS_TOPIC = "tilt_status_topic"
CONF_TILT_STATUS_TEMPLATE = "tilt_status_template"
CONF_STATE_STOPPED = "state_stopped"
+CONF_PAYLOAD_STOP_TILT = "payload_stop_tilt"
CONF_TILT_CLOSED_POSITION = "tilt_closed_value"
CONF_TILT_MAX = "tilt_max"
CONF_TILT_MIN = "tilt_min"
@@ -203,6 +204,9 @@ _PLATFORM_SCHEMA_BASE = MQTT_BASE_SCHEMA.extend(
vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
vol.Optional(CONF_GET_POSITION_TEMPLATE): cv.template,
vol.Optional(CONF_TILT_COMMAND_TEMPLATE): cv.template,
+ vol.Optional(CONF_PAYLOAD_STOP_TILT, default=DEFAULT_PAYLOAD_STOP): vol.Any(
+ cv.string, None
+ ),
}
).extend(MQTT_ENTITY_COMMON_SCHEMA.schema)
@@ -220,7 +224,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT cover through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
@@ -592,6 +596,12 @@ class MqttCover(MqttEntity, CoverEntity):
self._attr_current_cover_tilt_position = tilt_percentage
self.async_write_ha_state()
+ async def async_stop_cover_tilt(self, **kwargs: Any) -> None:
+ """Stop moving the cover tilt."""
+ await self.async_publish_with_config(
+ self._config[CONF_TILT_COMMAND_TOPIC], self._config[CONF_PAYLOAD_STOP_TILT]
+ )
+
async def async_set_cover_position(self, **kwargs: Any) -> None:
"""Move the cover to a specific position."""
position_percentage = kwargs[ATTR_POSITION]
diff --git a/homeassistant/components/mqtt/device_tracker.py b/homeassistant/components/mqtt/device_tracker.py
index d3ad57ef43d..4017245cf51 100644
--- a/homeassistant/components/mqtt/device_tracker.py
+++ b/homeassistant/components/mqtt/device_tracker.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -79,7 +79,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT event through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/entity.py b/homeassistant/components/mqtt/entity.py
index fb047cc8d5e..0b4f65fab47 100644
--- a/homeassistant/components/mqtt/entity.py
+++ b/homeassistant/components/mqtt/entity.py
@@ -43,7 +43,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_send,
)
from homeassistant.helpers.entity import Entity, async_generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import (
async_track_device_registry_updated_event,
async_track_entity_registry_updated_event,
@@ -111,6 +111,7 @@ from .discovery import (
from .models import (
DATA_MQTT,
MessageCallbackType,
+ MqttSubentryData,
MqttValueTemplate,
MqttValueTemplateException,
PublishPayloadType,
@@ -238,7 +239,7 @@ def async_setup_entity_entry_helper(
entry: ConfigEntry,
entity_class: type[MqttEntity] | None,
domain: str,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
discovery_schema: VolSchemaType,
platform_schema_modern: VolSchemaType,
schema_class_mapping: dict[str, type[MqttEntity]] | None = None,
@@ -282,11 +283,10 @@ def async_setup_entity_entry_helper(
@callback
def _async_setup_entities() -> None:
- """Set up MQTT items from configuration.yaml."""
+ """Set up MQTT items from subentries and configuration.yaml."""
nonlocal entity_class
mqtt_data = hass.data[DATA_MQTT]
- if not (config_yaml := mqtt_data.config):
- return
+ config_yaml = mqtt_data.config
yaml_configs: list[ConfigType] = [
config
for config_item in config_yaml
@@ -294,6 +294,43 @@ def async_setup_entity_entry_helper(
for config in configs
if config_domain == domain
]
+ # process subentry entity setup
+ for config_subentry_id, subentry in entry.subentries.items():
+ subentry_data = cast(MqttSubentryData, subentry.data)
+ availability_config = subentry_data.get("availability", {})
+ subentry_entities: list[Entity] = []
+ device_config = subentry_data["device"].copy()
+ device_config["identifiers"] = config_subentry_id
+ for component_id, component_data in subentry_data["components"].items():
+ if component_data["platform"] != domain:
+ continue
+ component_config: dict[str, Any] = component_data.copy()
+ component_config[CONF_UNIQUE_ID] = (
+ f"{config_subentry_id}_{component_id}"
+ )
+ component_config[CONF_DEVICE] = device_config
+ component_config.pop("platform")
+ component_config.update(availability_config)
+
+ try:
+ config = platform_schema_modern(component_config)
+ if schema_class_mapping is not None:
+ entity_class = schema_class_mapping[config[CONF_SCHEMA]]
+ if TYPE_CHECKING:
+ assert entity_class is not None
+ subentry_entities.append(entity_class(hass, config, entry, None))
+ except vol.Invalid as exc:
+ _LOGGER.error(
+ "Schema violation occurred when trying to set up "
+ "entity from subentry %s %s %s: %s",
+ config_subentry_id,
+ subentry.title,
+ subentry.data,
+ exc,
+ )
+
+ async_add_entities(subentry_entities, config_subentry_id=config_subentry_id)
+
entities: list[Entity] = []
for yaml_config in yaml_configs:
try:
diff --git a/homeassistant/components/mqtt/event.py b/homeassistant/components/mqtt/event.py
index 5855f94dad7..aef21838d59 100644
--- a/homeassistant/components/mqtt/event.py
+++ b/homeassistant/components/mqtt/event.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE_CLASS, CONF_NAME, CONF_VALUE_TEMPLATE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from homeassistant.util.json import JSON_DECODE_EXCEPTIONS, json_loads_object
@@ -73,7 +73,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT event through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/fan.py b/homeassistant/components/mqtt/fan.py
index d8e96eb2734..3fac4d4ffe0 100644
--- a/homeassistant/components/mqtt/fan.py
+++ b/homeassistant/components/mqtt/fan.py
@@ -28,7 +28,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -190,7 +190,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT fan through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/humidifier.py b/homeassistant/components/mqtt/humidifier.py
index bffe0ec1420..07ddcddb13a 100644
--- a/homeassistant/components/mqtt/humidifier.py
+++ b/homeassistant/components/mqtt/humidifier.py
@@ -31,7 +31,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -183,7 +183,7 @@ TOPICS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT humidifier through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/image.py b/homeassistant/components/mqtt/image.py
index 4b7b2d783d2..a668608dd55 100644
--- a/homeassistant/components/mqtt/image.py
+++ b/homeassistant/components/mqtt/image.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.httpx_client import get_async_client
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, VolSchemaType
@@ -82,7 +82,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT image through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/lawn_mower.py b/homeassistant/components/mqtt/lawn_mower.py
index 87577c4b4d9..1917c56f209 100644
--- a/homeassistant/components/mqtt/lawn_mower.py
+++ b/homeassistant/components/mqtt/lawn_mower.py
@@ -10,6 +10,7 @@ import voluptuous as vol
from homeassistant.components import lawn_mower
from homeassistant.components.lawn_mower import (
+ ENTITY_ID_FORMAT,
LawnMowerActivity,
LawnMowerEntity,
LawnMowerEntityFeature,
@@ -18,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, CONF_OPTIMISTIC
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -50,7 +51,6 @@ CONF_START_MOWING_COMMAND_TOPIC = "start_mowing_command_topic"
CONF_START_MOWING_COMMAND_TEMPLATE = "start_mowing_command_template"
DEFAULT_NAME = "MQTT Lawn Mower"
-ENTITY_ID_FORMAT = lawn_mower.DOMAIN + ".{}"
MQTT_LAWN_MOWER_ATTRIBUTES_BLOCKED: frozenset[str] = frozenset()
@@ -80,7 +80,7 @@ DISCOVERY_SCHEMA = vol.All(PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EX
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT lawn mower through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/light/__init__.py b/homeassistant/components/mqtt/light/__init__.py
index 328f80cb5ea..3ffad9226be 100644
--- a/homeassistant/components/mqtt/light/__init__.py
+++ b/homeassistant/components/mqtt/light/__init__.py
@@ -9,7 +9,7 @@ import voluptuous as vol
from homeassistant.components import light
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from ..entity import async_setup_entity_entry_helper
@@ -69,7 +69,7 @@ PLATFORM_SCHEMA_MODERN = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT lights through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/light/schema_json.py b/homeassistant/components/mqtt/light/schema_json.py
index 14e21e61d48..d18da9e917a 100644
--- a/homeassistant/components/mqtt/light/schema_json.py
+++ b/homeassistant/components/mqtt/light/schema_json.py
@@ -31,7 +31,6 @@ from homeassistant.components.light import (
LightEntity,
LightEntityFeature,
brightness_supported,
- color_supported,
valid_supported_color_modes,
)
from homeassistant.const import (
@@ -217,6 +216,10 @@ class MqttLightJson(MqttEntity, LightEntity, RestoreEntity):
self._attr_color_mode = next(iter(self.supported_color_modes))
else:
self._attr_color_mode = ColorMode.UNKNOWN
+ elif config.get(CONF_BRIGHTNESS):
+ # Brightness is supported and no supported_color_modes are set,
+ # so set brightness as the supported color mode.
+ self._attr_supported_color_modes = {ColorMode.BRIGHTNESS}
def _update_color(self, values: dict[str, Any]) -> None:
color_mode: str = values["color_mode"]
@@ -289,7 +292,7 @@ class MqttLightJson(MqttEntity, LightEntity, RestoreEntity):
elif values["state"] is None:
self._attr_is_on = None
- if color_supported(self.supported_color_modes) and "color_mode" in values:
+ if "color_mode" in values:
self._update_color(values)
if brightness_supported(self.supported_color_modes):
diff --git a/homeassistant/components/mqtt/lock.py b/homeassistant/components/mqtt/lock.py
index 895bfba3560..727e689798e 100644
--- a/homeassistant/components/mqtt/lock.py
+++ b/homeassistant/components/mqtt/lock.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, TemplateVarsType
@@ -116,7 +116,7 @@ STATE_CONFIG_KEYS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT lock through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/models.py b/homeassistant/components/mqtt/models.py
index 34c1f304944..bcfe94bbd58 100644
--- a/homeassistant/components/mqtt/models.py
+++ b/homeassistant/components/mqtt/models.py
@@ -420,5 +420,34 @@ class MqttComponentConfig:
discovery_payload: MQTTDiscoveryPayload
+class MqttDeviceData(TypedDict, total=False):
+ """Hold the data for an MQTT device."""
+
+ name: str
+ identifiers: str
+ configuration_url: str
+ sw_version: str
+ hw_version: str
+ model: str
+ model_id: str
+
+
+class MqttAvailabilityData(TypedDict, total=False):
+ """Hold the availability configuration for a device."""
+
+ availability_topic: str
+ availability_template: str
+ payload_available: str
+ payload_not_available: str
+
+
+class MqttSubentryData(TypedDict, total=False):
+ """Hold the data for a MQTT subentry."""
+
+ device: MqttDeviceData
+ components: dict[str, dict[str, Any]]
+ availability: MqttAvailabilityData
+
+
DATA_MQTT: HassKey[MqttData] = HassKey("mqtt")
DATA_MQTT_AVAILABLE: HassKey[asyncio.Future[bool]] = HassKey("mqtt_client_available")
diff --git a/homeassistant/components/mqtt/notify.py b/homeassistant/components/mqtt/notify.py
index 7e0a7fd4dd8..0b6dbce38b4 100644
--- a/homeassistant/components/mqtt/notify.py
+++ b/homeassistant/components/mqtt/notify.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .config import DEFAULT_RETAIN, MQTT_BASE_SCHEMA
@@ -39,7 +39,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT notify through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/number.py b/homeassistant/components/mqtt/number.py
index 9b47a3ad23a..5ee93cfba07 100644
--- a/homeassistant/components/mqtt/number.py
+++ b/homeassistant/components/mqtt/number.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -109,7 +109,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT number through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/scene.py b/homeassistant/components/mqtt/scene.py
index c6651510a36..12f680b6e12 100644
--- a/homeassistant/components/mqtt/scene.py
+++ b/homeassistant/components/mqtt/scene.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, CONF_PAYLOAD_ON
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType
from .config import MQTT_BASE_SCHEMA
@@ -43,7 +43,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT scene through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/select.py b/homeassistant/components/mqtt/select.py
index 55d56ecd774..1b3ea1a7c44 100644
--- a/homeassistant/components/mqtt/select.py
+++ b/homeassistant/components/mqtt/select.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, CONF_OPTIMISTIC, CONF_VALUE_TEMPLATE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -63,7 +63,7 @@ DISCOVERY_SCHEMA = vol.All(PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EX
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT select through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/sensor.py b/homeassistant/components/mqtt/sensor.py
index ad84ebb09a3..4d67b0d56e6 100644
--- a/homeassistant/components/mqtt/sensor.py
+++ b/homeassistant/components/mqtt/sensor.py
@@ -11,6 +11,7 @@ import voluptuous as vol
from homeassistant.components import sensor
from homeassistant.components.sensor import (
CONF_STATE_CLASS,
+ DEVICE_CLASS_UNITS,
DEVICE_CLASSES_SCHEMA,
ENTITY_ID_FORMAT,
STATE_CLASSES_SCHEMA,
@@ -31,15 +32,16 @@ from homeassistant.const import (
)
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, State, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
+from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from homeassistant.util import dt as dt_util
from . import subscription
from .config import MQTT_RO_SCHEMA
-from .const import CONF_OPTIONS, CONF_STATE_TOPIC, PAYLOAD_NONE
+from .const import CONF_OPTIONS, CONF_STATE_TOPIC, DOMAIN, PAYLOAD_NONE
from .entity import MqttAvailabilityMixin, MqttEntity, async_setup_entity_entry_helper
from .models import MqttValueTemplate, PayloadSentinel, ReceiveMessage
from .schemas import MQTT_ENTITY_COMMON_SCHEMA
@@ -63,6 +65,10 @@ MQTT_SENSOR_ATTRIBUTES_BLOCKED = frozenset(
DEFAULT_NAME = "MQTT Sensor"
DEFAULT_FORCE_UPDATE = False
+URL_DOCS_SUPPORTED_SENSOR_UOM = (
+ "https://www.home-assistant.io/integrations/sensor/#device-class"
+)
+
_PLATFORM_SCHEMA_BASE = MQTT_RO_SCHEMA.extend(
{
vol.Optional(CONF_DEVICE_CLASS): vol.Any(DEVICE_CLASSES_SCHEMA, None),
@@ -107,6 +113,23 @@ def validate_sensor_state_and_device_class_config(config: ConfigType) -> ConfigT
f"got `{CONF_DEVICE_CLASS}` '{device_class}'"
)
+ if (device_class := config.get(CONF_DEVICE_CLASS)) is None or (
+ unit_of_measurement := config.get(CONF_UNIT_OF_MEASUREMENT)
+ ) is None:
+ return config
+
+ if (
+ device_class in DEVICE_CLASS_UNITS
+ and unit_of_measurement not in DEVICE_CLASS_UNITS[device_class]
+ ):
+ _LOGGER.warning(
+ "The unit of measurement `%s` is not valid "
+ "together with device class `%s`. "
+ "this will stop working in HA Core 2025.7.0",
+ unit_of_measurement,
+ device_class,
+ )
+
return config
@@ -124,7 +147,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT sensor through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
@@ -155,8 +178,40 @@ class MqttSensor(MqttEntity, RestoreSensor):
None
)
+ @callback
+ def async_check_uom(self) -> None:
+ """Check if the unit of measurement is valid with the device class."""
+ if (
+ self._discovery_data is not None
+ or self.device_class is None
+ or self.native_unit_of_measurement is None
+ ):
+ return
+ if (
+ self.device_class in DEVICE_CLASS_UNITS
+ and self.native_unit_of_measurement
+ not in DEVICE_CLASS_UNITS[self.device_class]
+ ):
+ async_create_issue(
+ self.hass,
+ DOMAIN,
+ self.entity_id,
+ issue_domain=sensor.DOMAIN,
+ is_fixable=False,
+ severity=IssueSeverity.WARNING,
+ learn_more_url=URL_DOCS_SUPPORTED_SENSOR_UOM,
+ translation_placeholders={
+ "uom": self.native_unit_of_measurement,
+ "device_class": self.device_class.value,
+ "entity_id": self.entity_id,
+ },
+ translation_key="invalid_unit_of_measurement",
+ breaks_in_ha_version="2025.7.0",
+ )
+
async def mqtt_async_added_to_hass(self) -> None:
"""Restore state for entities with expire_after set."""
+ self.async_check_uom()
last_state: State | None
last_sensor_data: SensorExtraStoredData | None
if (
diff --git a/homeassistant/components/mqtt/siren.py b/homeassistant/components/mqtt/siren.py
index 5e3ca76e722..48ab4676dea 100644
--- a/homeassistant/components/mqtt/siren.py
+++ b/homeassistant/components/mqtt/siren.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.json import json_dumps
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.template import Template
@@ -113,7 +113,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT siren through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/strings.json b/homeassistant/components/mqtt/strings.json
index fc316306d56..f0112097f4e 100644
--- a/homeassistant/components/mqtt/strings.json
+++ b/homeassistant/components/mqtt/strings.json
@@ -3,6 +3,10 @@
"invalid_platform_config": {
"title": "Invalid config found for mqtt {domain} item",
"description": "Home Assistant detected an invalid config for a manually configured item.\n\nPlatform domain: **{domain}**\nConfiguration file: **{config_file}**\nNear line: **{line}**\nConfiguration found:\n```yaml\n{config}\n```\nError: **{error}**.\n\nMake sure the configuration is valid and [reload](/developer-tools/yaml) the manually configured MQTT items or restart Home Assistant to fix this issue."
+ },
+ "invalid_unit_of_measurement": {
+ "title": "Sensor with invalid unit of measurement",
+ "description": "Manual configured Sensor entity **{entity_id}** has a configured unit of measurement **{uom}** which is not valid with configured device class **{device_class}**. Make sure a valid unit of measurement is configured or remove the device class, and [reload](/developer-tools/yaml) the manually configured MQTT items or restart Home Assistant to fix this issue."
}
},
"config": {
@@ -26,6 +30,7 @@
"client_id": "Client ID (leave empty to randomly generated one)",
"client_cert": "Upload client certificate file",
"client_key": "Upload private key file",
+ "client_key_password": "[%key:common::config_flow::data::password%]",
"keepalive": "The time between sending keep alive messages",
"tls_insecure": "Ignore broker certificate validation",
"protocol": "MQTT protocol",
@@ -45,6 +50,7 @@
"client_id": "The unique ID to identify the Home Assistant MQTT API as MQTT client. It is recommended to leave this option blank.",
"client_cert": "The client certificate to authenticate against your MQTT broker.",
"client_key": "The private key file that belongs to your client certificate.",
+ "client_key_password": "The password for the private key file (if set).",
"keepalive": "A value less than 90 seconds is advised.",
"tls_insecure": "Option to ignore validation of your MQTT broker's certificate.",
"protocol": "The MQTT protocol your broker operates at. For example 3.1.1.",
@@ -93,8 +99,8 @@
"bad_will": "Invalid will topic",
"bad_discovery_prefix": "Invalid discovery prefix",
"bad_certificate": "The CA certificate is invalid",
- "bad_client_cert": "Invalid client certificate, ensure a PEM coded file is supplied",
- "bad_client_key": "Invalid private key, ensure a PEM coded file is supplied without password",
+ "bad_client_cert": "Invalid client certificate, ensure a valid file is supplied",
+ "client_key_error": "Invalid private key file or invalid password supplied",
"bad_client_cert_key": "Client certificate and private key are not a valid pair",
"bad_ws_headers": "Supply valid HTTP headers as a JSON object",
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
@@ -102,6 +108,127 @@
"invalid_inclusion": "The client certificate and private key must be configured together"
}
},
+ "config_subentries": {
+ "device": {
+ "initiate_flow": {
+ "user": "Add MQTT Device",
+ "reconfigure": "Reconfigure MQTT Device"
+ },
+ "entry_type": "MQTT Device",
+ "step": {
+ "availability": {
+ "title": "Availability options",
+ "description": "The availability feature allows a device to report it's availability.",
+ "data": {
+ "availability_topic": "Availability topic",
+ "availability_template": "Availability template",
+ "payload_available": "Payload available",
+ "payload_not_available": "Payload not available"
+ },
+ "data_description": {
+ "availability_topic": "Topic to receive the availabillity payload on",
+ "availability_template": "A [template](https://www.home-assistant.io/docs/configuration/templating/#using-templates-with-the-mqtt-integration) to render the availability payload received on the availability topic",
+ "payload_available": "The payload that indicates the device is available (defaults to 'online')",
+ "payload_not_available": "The payload that indicates the device is not available (defaults to 'offline')"
+ }
+ },
+ "device": {
+ "title": "Configure MQTT device details",
+ "description": "Enter the MQTT device details:",
+ "data": {
+ "name": "[%key:common::config_flow::data::name%]",
+ "configuration_url": "Configuration URL",
+ "sw_version": "Software version",
+ "hw_version": "Hardware version",
+ "model": "Model",
+ "model_id": "Model ID"
+ },
+ "data_description": {
+ "name": "The name of the manually added MQTT device.",
+ "configuration_url": "A link to the webpage that can manage the configuration of this device. Can be either a 'http://', 'https://' or an internal 'homeassistant://' URL.",
+ "sw_version": "The software version of the device. E.g. '2025.1.0'.",
+ "hw_version": "The hardware version of the device. E.g. 'v1.0 rev a'.",
+ "model": "E.g. 'Cleanmaster Pro'.",
+ "model_id": "E.g. '123NK2PRO'."
+ }
+ },
+ "summary_menu": {
+ "title": "Reconfigure \"{mqtt_device}\"",
+ "description": "Entities set up:\n{mqtt_items}\n\nDecide what to do next:",
+ "menu_options": {
+ "entity": "Add another entity to \"{mqtt_device}\"",
+ "update_entity": "Update entity properties",
+ "delete_entity": "Delete an entity",
+ "availability": "Configure availability",
+ "device": "Update device properties",
+ "save_changes": "Save changes"
+ }
+ },
+ "entity": {
+ "title": "Configure MQTT device \"{mqtt_device}\"",
+ "description": "Configure the basic {platform_label}entity settings{entity_name_label}",
+ "data": {
+ "platform": "Type of entity",
+ "name": "Entity name",
+ "entity_picture": "Entity picture"
+ },
+ "data_description": {
+ "platform": "The type of the entity to configure.",
+ "name": "The name of the entity. Leave empty to set it to `None` to [mark it as main feature of the MQTT device](https://www.home-assistant.io/integrations/mqtt/#naming-of-mqtt-entities).",
+ "entity_picture": "An URL to a picture to be assigned."
+ }
+ },
+ "delete_entity": {
+ "title": "Delete entity",
+ "description": "Delete an entity. The entity will be removed from the device. Removing an entity will break any automations or scripts that depend on it.",
+ "data": {
+ "component": "Entity"
+ },
+ "data_description": {
+ "component": "Select the entity you want to delete. Minimal one entity is required."
+ }
+ },
+ "update_entity": {
+ "title": "Select entity",
+ "description": "Select the entity you want to update",
+ "data": {
+ "component": "Entity"
+ },
+ "data_description": {
+ "component": "Select the entity you want to update."
+ }
+ },
+ "mqtt_platform_config": {
+ "title": "Configure MQTT device \"{mqtt_device}\"",
+ "description": "Please configure MQTT specific details for {platform} entity \"{entity}\":",
+ "data": {
+ "command_topic": "Command topic",
+ "command_template": "Command template",
+ "retain": "Retain",
+ "qos": "QoS"
+ },
+ "data_description": {
+ "command_topic": "The publishing topic that will be used to control the {platform} entity.",
+ "command_template": "A [template](https://www.home-assistant.io/docs/configuration/templating/#using-command-templates-with-mqtt) to render the payload to be published at the command topic.",
+ "retain": "Select if values published by the {platform} entity should be retained at the MQTT broker.",
+ "qos": "The QoS value {platform} entity should use."
+ }
+ }
+ },
+ "abort": {
+ "reconfigure_successful": "[%key:common::config_flow::abort::reconfigure_successful%]"
+ },
+ "create_entry": {
+ "default": "MQTT device with {platform} entity \"{entity}\" was set up successfully.\n\nNote that you can reconfigure the MQTT device at any time, e.g. to add more entities."
+ },
+ "error": {
+ "invalid_input": "Invalid value",
+ "invalid_subscribe_topic": "Invalid subscribe topic",
+ "invalid_template": "Invalid template",
+ "invalid_url": "Invalid URL"
+ }
+ }
+ },
"device_automation": {
"trigger_type": {
"button_short_press": "\"{subtype}\" pressed",
@@ -207,7 +334,7 @@
"bad_discovery_prefix": "[%key:component::mqtt::config::error::bad_discovery_prefix%]",
"bad_certificate": "[%key:component::mqtt::config::error::bad_certificate%]",
"bad_client_cert": "[%key:component::mqtt::config::error::bad_client_cert%]",
- "bad_client_key": "[%key:component::mqtt::config::error::bad_client_key%]",
+ "client_key_error": "[%key:component::mqtt::config::error::client_key_error%]",
"bad_client_cert_key": "[%key:component::mqtt::config::error::bad_client_cert_key%]",
"bad_ws_headers": "[%key:component::mqtt::config::error::bad_ws_headers%]",
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
@@ -215,6 +342,11 @@
}
},
"selector": {
+ "platform": {
+ "options": {
+ "notify": "Notify"
+ }
+ },
"set_ca_cert": {
"options": {
"off": "[%key:common::state::off%]",
@@ -230,7 +362,7 @@
"fields": {
"evaluate_payload": {
"name": "Evaluate payload",
- "description": "When `payload` is a Python bytes literal, evaluate the bytes literal and publish the raw data."
+ "description": "If 'Payload' is a Python bytes literal, evaluate the bytes literal and publish the raw data."
},
"topic": {
"name": "Topic",
diff --git a/homeassistant/components/mqtt/switch.py b/homeassistant/components/mqtt/switch.py
index a305fa83485..f6996fc77ce 100644
--- a/homeassistant/components/mqtt/switch.py
+++ b/homeassistant/components/mqtt/switch.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType
@@ -70,7 +70,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.REMOVE_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT switch through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/text.py b/homeassistant/components/mqtt/text.py
index b4ed33a7730..d306fc0819b 100644
--- a/homeassistant/components/mqtt/text.py
+++ b/homeassistant/components/mqtt/text.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.service_info.mqtt import ReceivePayloadType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
@@ -95,7 +95,7 @@ PLATFORM_SCHEMA_MODERN = vol.All(_PLATFORM_SCHEMA_BASE, valid_text_size_configur
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT text through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/update.py b/homeassistant/components/mqtt/update.py
index 59742d24b60..c4916b5010c 100644
--- a/homeassistant/components/mqtt/update.py
+++ b/homeassistant/components/mqtt/update.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE_CLASS, CONF_NAME, CONF_VALUE_TEMPLATE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from homeassistant.util.json import JSON_DECODE_EXCEPTIONS, json_loads
@@ -82,7 +82,7 @@ MQTT_JSON_UPDATE_SCHEMA = vol.Schema(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT update entity through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/vacuum.py b/homeassistant/components/mqtt/vacuum.py
index ae6b25eff14..f1d2eb34fe1 100644
--- a/homeassistant/components/mqtt/vacuum.py
+++ b/homeassistant/components/mqtt/vacuum.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_SUPPORTED_FEATURES, CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.json import json_dumps
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, VolSchemaType
from homeassistant.util.json import json_loads_object
@@ -175,7 +175,7 @@ DISCOVERY_SCHEMA = PLATFORM_SCHEMA_MODERN.extend({}, extra=vol.ALLOW_EXTRA)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT vacuum through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/valve.py b/homeassistant/components/mqtt/valve.py
index b380199332b..53f7d06429e 100644
--- a/homeassistant/components/mqtt/valve.py
+++ b/homeassistant/components/mqtt/valve.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from homeassistant.util.json import JSON_DECODE_EXCEPTIONS, json_loads
from homeassistant.util.percentage import (
@@ -136,7 +136,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT valve through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mqtt/water_heater.py b/homeassistant/components/mqtt/water_heater.py
index 967eceac326..31d4f0fe30e 100644
--- a/homeassistant/components/mqtt/water_heater.py
+++ b/homeassistant/components/mqtt/water_heater.py
@@ -36,7 +36,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from homeassistant.util.unit_conversion import TemperatureConverter
@@ -166,7 +166,7 @@ DISCOVERY_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MQTT water heater device through YAML and through MQTT discovery."""
async_setup_entity_entry_helper(
diff --git a/homeassistant/components/mullvad/binary_sensor.py b/homeassistant/components/mullvad/binary_sensor.py
index 2e649d9a586..ad488058025 100644
--- a/homeassistant/components/mullvad/binary_sensor.py
+++ b/homeassistant/components/mullvad/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -28,7 +28,7 @@ BINARY_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinator = hass.data[DOMAIN]
diff --git a/homeassistant/components/music_assistant/__init__.py b/homeassistant/components/music_assistant/__init__.py
index e569bb93a42..a2d2dae9e3f 100644
--- a/homeassistant/components/music_assistant/__init__.py
+++ b/homeassistant/components/music_assistant/__init__.py
@@ -9,7 +9,7 @@ from typing import TYPE_CHECKING
from music_assistant_client import MusicAssistantClient
from music_assistant_client.exceptions import CannotConnect, InvalidServerVersion
from music_assistant_models.enums import EventType
-from music_assistant_models.errors import MusicAssistantError
+from music_assistant_models.errors import ActionUnavailable, MusicAssistantError
from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import CONF_URL, EVENT_HOMEASSISTANT_STOP, Platform
@@ -23,7 +23,7 @@ from homeassistant.helpers.issue_registry import (
async_delete_issue,
)
-from .actions import register_actions
+from .actions import get_music_assistant_client, register_actions
from .const import DOMAIN, LOGGER
if TYPE_CHECKING:
@@ -137,6 +137,18 @@ async def async_setup_entry(
mass.subscribe(handle_player_removed, EventType.PLAYER_REMOVED)
)
+ # check if any playerconfigs have been removed while we were disconnected
+ all_player_configs = await mass.config.get_player_configs()
+ player_ids = {player.player_id for player in all_player_configs}
+ dev_reg = dr.async_get(hass)
+ dev_entries = dr.async_entries_for_config_entry(dev_reg, entry.entry_id)
+ for device in dev_entries:
+ for identifier in device.identifiers:
+ if identifier[0] == DOMAIN and identifier[1] not in player_ids:
+ dev_reg.async_update_device(
+ device.id, remove_config_entry_id=entry.entry_id
+ )
+
return True
@@ -174,3 +186,31 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await mass_entry_data.mass.disconnect()
return unload_ok
+
+
+async def async_remove_config_entry_device(
+ hass: HomeAssistant, config_entry: ConfigEntry, device_entry: dr.DeviceEntry
+) -> bool:
+ """Remove a config entry from a device."""
+ player_id = next(
+ (
+ identifier[1]
+ for identifier in device_entry.identifiers
+ if identifier[0] == DOMAIN
+ ),
+ None,
+ )
+ if player_id is None:
+ # this should not be possible at all, but guard it anyways
+ return False
+ mass = get_music_assistant_client(hass, config_entry.entry_id)
+ if mass.players.get(player_id) is None:
+ # player is already removed on the server, this is an orphaned device
+ return True
+ # try to remove the player from the server
+ try:
+ await mass.config.remove_player_config(player_id)
+ except ActionUnavailable:
+ return False
+ else:
+ return True
diff --git a/homeassistant/components/music_assistant/actions.py b/homeassistant/components/music_assistant/actions.py
index bcd33b7fd6c..031229d1544 100644
--- a/homeassistant/components/music_assistant/actions.py
+++ b/homeassistant/components/music_assistant/actions.py
@@ -23,6 +23,7 @@ from .const import (
ATTR_ALBUM_TYPE,
ATTR_ALBUMS,
ATTR_ARTISTS,
+ ATTR_AUDIOBOOKS,
ATTR_CONFIG_ENTRY_ID,
ATTR_FAVORITE,
ATTR_ITEMS,
@@ -32,6 +33,7 @@ from .const import (
ATTR_OFFSET,
ATTR_ORDER_BY,
ATTR_PLAYLISTS,
+ ATTR_PODCASTS,
ATTR_RADIO,
ATTR_SEARCH,
ATTR_SEARCH_ALBUM,
@@ -48,6 +50,15 @@ from .schemas import (
if TYPE_CHECKING:
from music_assistant_client import MusicAssistantClient
+ from music_assistant_models.media_items import (
+ Album,
+ Artist,
+ Audiobook,
+ Playlist,
+ Podcast,
+ Radio,
+ Track,
+ )
from . import MusicAssistantConfigEntry
@@ -154,6 +165,14 @@ async def handle_search(call: ServiceCall) -> ServiceResponse:
media_item_dict_from_mass_item(mass, item)
for item in search_results.radio
],
+ ATTR_AUDIOBOOKS: [
+ media_item_dict_from_mass_item(mass, item)
+ for item in search_results.audiobooks
+ ],
+ ATTR_PODCASTS: [
+ media_item_dict_from_mass_item(mass, item)
+ for item in search_results.podcasts
+ ],
}
)
return response
@@ -173,6 +192,15 @@ async def handle_get_library(call: ServiceCall) -> ServiceResponse:
"offset": offset,
"order_by": order_by,
}
+ library_result: (
+ list[Album]
+ | list[Artist]
+ | list[Track]
+ | list[Radio]
+ | list[Playlist]
+ | list[Audiobook]
+ | list[Podcast]
+ )
if media_type == MediaType.ALBUM:
library_result = await mass.music.get_library_albums(
**base_params,
@@ -181,7 +209,7 @@ async def handle_get_library(call: ServiceCall) -> ServiceResponse:
elif media_type == MediaType.ARTIST:
library_result = await mass.music.get_library_artists(
**base_params,
- album_artists_only=call.data.get(ATTR_ALBUM_ARTISTS_ONLY),
+ album_artists_only=bool(call.data.get(ATTR_ALBUM_ARTISTS_ONLY)),
)
elif media_type == MediaType.TRACK:
library_result = await mass.music.get_library_tracks(
@@ -195,6 +223,14 @@ async def handle_get_library(call: ServiceCall) -> ServiceResponse:
library_result = await mass.music.get_library_playlists(
**base_params,
)
+ elif media_type == MediaType.AUDIOBOOK:
+ library_result = await mass.music.get_library_audiobooks(
+ **base_params,
+ )
+ elif media_type == MediaType.PODCAST:
+ library_result = await mass.music.get_library_podcasts(
+ **base_params,
+ )
else:
raise ServiceValidationError(f"Unsupported media type {media_type}")
diff --git a/homeassistant/components/music_assistant/const.py b/homeassistant/components/music_assistant/const.py
index 1980c495278..d2ee1f75028 100644
--- a/homeassistant/components/music_assistant/const.py
+++ b/homeassistant/components/music_assistant/const.py
@@ -34,6 +34,8 @@ ATTR_ARTISTS = "artists"
ATTR_ALBUMS = "albums"
ATTR_TRACKS = "tracks"
ATTR_PLAYLISTS = "playlists"
+ATTR_AUDIOBOOKS = "audiobooks"
+ATTR_PODCASTS = "podcasts"
ATTR_RADIO = "radio"
ATTR_ITEMS = "items"
ATTR_RADIO_MODE = "radio_mode"
diff --git a/homeassistant/components/music_assistant/manifest.json b/homeassistant/components/music_assistant/manifest.json
index f5cdcf50673..fb8bb9c3ac2 100644
--- a/homeassistant/components/music_assistant/manifest.json
+++ b/homeassistant/components/music_assistant/manifest.json
@@ -7,6 +7,6 @@
"documentation": "https://www.home-assistant.io/integrations/music_assistant",
"iot_class": "local_push",
"loggers": ["music_assistant"],
- "requirements": ["music-assistant-client==1.0.8"],
+ "requirements": ["music-assistant-client==1.1.1"],
"zeroconf": ["_mass._tcp.local."]
}
diff --git a/homeassistant/components/music_assistant/media_browser.py b/homeassistant/components/music_assistant/media_browser.py
index e65d6d4a975..a926e2a0595 100644
--- a/homeassistant/components/music_assistant/media_browser.py
+++ b/homeassistant/components/music_assistant/media_browser.py
@@ -166,6 +166,8 @@ async def build_playlist_items_listing(
) -> BrowseMedia:
"""Build Playlist items browse listing."""
playlist = await mass.music.get_item_by_uri(identifier)
+ if TYPE_CHECKING:
+ assert playlist.uri is not None
return BrowseMedia(
media_class=MediaClass.PLAYLIST,
@@ -219,6 +221,9 @@ async def build_artist_items_listing(
artist = await mass.music.get_item_by_uri(identifier)
albums = await mass.music.get_artist_albums(artist.item_id, artist.provider)
+ if TYPE_CHECKING:
+ assert artist.uri is not None
+
return BrowseMedia(
media_class=MediaType.ARTIST,
media_content_id=artist.uri,
@@ -267,6 +272,9 @@ async def build_album_items_listing(
album = await mass.music.get_item_by_uri(identifier)
tracks = await mass.music.get_album_tracks(album.item_id, album.provider)
+ if TYPE_CHECKING:
+ assert album.uri is not None
+
return BrowseMedia(
media_class=MediaType.ALBUM,
media_content_id=album.uri,
@@ -340,6 +348,9 @@ def build_item(
title = item.name
img_url = mass.get_media_item_image_url(item)
+ if TYPE_CHECKING:
+ assert item.uri is not None
+
return BrowseMedia(
media_class=media_class or item.media_type.value,
media_content_id=item.uri,
diff --git a/homeassistant/components/music_assistant/media_player.py b/homeassistant/components/music_assistant/media_player.py
index 4a7e20046b2..56bde7bbae7 100644
--- a/homeassistant/components/music_assistant/media_player.py
+++ b/homeassistant/components/music_assistant/media_player.py
@@ -9,6 +9,7 @@ import functools
import os
from typing import TYPE_CHECKING, Any, Concatenate
+from music_assistant_models.constants import PLAYER_CONTROL_NONE
from music_assistant_models.enums import (
EventType,
MediaType,
@@ -20,6 +21,7 @@ from music_assistant_models.enums import (
from music_assistant_models.errors import MediaNotFoundError, MusicAssistantError
from music_assistant_models.event import MassEvent
from music_assistant_models.media_items import ItemMapping, MediaItemType, Track
+from music_assistant_models.player_queue import PlayerQueue
import voluptuous as vol
from homeassistant.components import media_source
@@ -41,7 +43,7 @@ from homeassistant.core import HomeAssistant, ServiceResponse, SupportsResponse
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_registry as er
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.util.dt import utc_from_timestamp
@@ -78,21 +80,15 @@ from .schemas import QUEUE_DETAILS_SCHEMA, queue_item_dict_from_mass_item
if TYPE_CHECKING:
from music_assistant_client import MusicAssistantClient
from music_assistant_models.player import Player
- from music_assistant_models.player_queue import PlayerQueue
-SUPPORTED_FEATURES = (
- MediaPlayerEntityFeature.PAUSE
- | MediaPlayerEntityFeature.VOLUME_SET
- | MediaPlayerEntityFeature.STOP
+SUPPORTED_FEATURES_BASE = (
+ MediaPlayerEntityFeature.STOP
| MediaPlayerEntityFeature.PREVIOUS_TRACK
| MediaPlayerEntityFeature.NEXT_TRACK
| MediaPlayerEntityFeature.SHUFFLE_SET
| MediaPlayerEntityFeature.REPEAT_SET
- | MediaPlayerEntityFeature.TURN_ON
- | MediaPlayerEntityFeature.TURN_OFF
| MediaPlayerEntityFeature.PLAY
| MediaPlayerEntityFeature.PLAY_MEDIA
- | MediaPlayerEntityFeature.VOLUME_STEP
| MediaPlayerEntityFeature.CLEAR_PLAYLIST
| MediaPlayerEntityFeature.BROWSE_MEDIA
| MediaPlayerEntityFeature.MEDIA_ENQUEUE
@@ -137,7 +133,7 @@ def catch_musicassistant_error[_R, **P](
async def async_setup_entry(
hass: HomeAssistant,
entry: MusicAssistantConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Music Assistant MediaPlayer(s) from Config Entry."""
mass = entry.runtime_data.mass
@@ -212,11 +208,7 @@ class MusicAssistantPlayer(MusicAssistantEntity, MediaPlayerEntity):
"""Initialize MediaPlayer entity."""
super().__init__(mass, player_id)
self._attr_icon = self.player.icon.replace("mdi-", "mdi:")
- self._attr_supported_features = SUPPORTED_FEATURES
- if PlayerFeature.SET_MEMBERS in self.player.supported_features:
- self._attr_supported_features |= MediaPlayerEntityFeature.GROUPING
- if PlayerFeature.VOLUME_MUTE in self.player.supported_features:
- self._attr_supported_features |= MediaPlayerEntityFeature.VOLUME_MUTE
+ self._set_supported_features()
self._attr_device_class = MediaPlayerDeviceClass.SPEAKER
self._prev_time: float = 0
@@ -241,6 +233,19 @@ class MusicAssistantPlayer(MusicAssistantEntity, MediaPlayerEntity):
)
)
+ # we subscribe to the player config changed event to update
+ # the supported features of the player
+ async def player_config_changed(event: MassEvent) -> None:
+ self._set_supported_features()
+ await self.async_on_update()
+ self.async_write_ha_state()
+
+ self.async_on_remove(
+ self.mass.subscribe(
+ player_config_changed, EventType.PLAYER_CONFIG_UPDATED, self.player_id
+ )
+ )
+
@property
def active_queue(self) -> PlayerQueue | None:
"""Return the active queue for this player (if any)."""
@@ -271,22 +276,26 @@ class MusicAssistantPlayer(MusicAssistantEntity, MediaPlayerEntity):
self._attr_state = MediaPlayerState(player.state.value)
else:
self._attr_state = MediaPlayerState(STATE_OFF)
- group_members_entity_ids: list[str] = []
+
+ group_members: list[str] = []
if player.group_childs:
- # translate MA group_childs to HA group_members as entity id's
- entity_registry = er.async_get(self.hass)
- group_members_entity_ids = [
- entity_id
- for child_id in player.group_childs
- if (
- entity_id := entity_registry.async_get_entity_id(
- self.platform.domain, DOMAIN, child_id
- )
+ group_members = player.group_childs
+ elif player.synced_to and (parent := self.mass.players.get(player.synced_to)):
+ group_members = parent.group_childs
+
+ # translate MA group_childs to HA group_members as entity id's
+ entity_registry = er.async_get(self.hass)
+ group_members_entity_ids: list[str] = [
+ entity_id
+ for child_id in group_members
+ if (
+ entity_id := entity_registry.async_get_entity_id(
+ self.platform.domain, DOMAIN, child_id
)
- ]
- # NOTE: we sort the group_members for now,
- # until the MA API returns them sorted (group_childs is now a set)
- self._attr_group_members = sorted(group_members_entity_ids)
+ )
+ ]
+
+ self._attr_group_members = group_members_entity_ids
self._attr_volume_level = (
player.volume_level / 100 if player.volume_level is not None else None
)
@@ -473,6 +482,8 @@ class MusicAssistantPlayer(MusicAssistantEntity, MediaPlayerEntity):
album=album,
media_type=MediaType(media_type) if media_type else None,
):
+ if TYPE_CHECKING:
+ assert item.uri is not None
media_uris.append(item.uri)
if not media_uris:
@@ -680,3 +691,20 @@ class MusicAssistantPlayer(MusicAssistantEntity, MediaPlayerEntity):
if isinstance(queue_option, MediaPlayerEnqueue):
queue_option = QUEUE_OPTION_MAP.get(queue_option)
return queue_option
+
+ def _set_supported_features(self) -> None:
+ """Set supported features based on player capabilities."""
+ supported_features = SUPPORTED_FEATURES_BASE
+ if PlayerFeature.SET_MEMBERS in self.player.supported_features:
+ supported_features |= MediaPlayerEntityFeature.GROUPING
+ if PlayerFeature.PAUSE in self.player.supported_features:
+ supported_features |= MediaPlayerEntityFeature.PAUSE
+ if self.player.mute_control != PLAYER_CONTROL_NONE:
+ supported_features |= MediaPlayerEntityFeature.VOLUME_MUTE
+ if self.player.volume_control != PLAYER_CONTROL_NONE:
+ supported_features |= MediaPlayerEntityFeature.VOLUME_STEP
+ supported_features |= MediaPlayerEntityFeature.VOLUME_SET
+ if self.player.power_control != PLAYER_CONTROL_NONE:
+ supported_features |= MediaPlayerEntityFeature.TURN_ON
+ supported_features |= MediaPlayerEntityFeature.TURN_OFF
+ self._attr_supported_features = supported_features
diff --git a/homeassistant/components/music_assistant/schemas.py b/homeassistant/components/music_assistant/schemas.py
index d8c4fe1649d..7501d3d2038 100644
--- a/homeassistant/components/music_assistant/schemas.py
+++ b/homeassistant/components/music_assistant/schemas.py
@@ -15,6 +15,7 @@ from .const import (
ATTR_ALBUM,
ATTR_ALBUMS,
ATTR_ARTISTS,
+ ATTR_AUDIOBOOKS,
ATTR_BIT_DEPTH,
ATTR_CONTENT_TYPE,
ATTR_CURRENT_INDEX,
@@ -31,6 +32,7 @@ from .const import (
ATTR_OFFSET,
ATTR_ORDER_BY,
ATTR_PLAYLISTS,
+ ATTR_PODCASTS,
ATTR_PROVIDER,
ATTR_QUEUE_ID,
ATTR_QUEUE_ITEM_ID,
@@ -65,20 +67,20 @@ MEDIA_ITEM_SCHEMA = vol.Schema(
def media_item_dict_from_mass_item(
mass: MusicAssistantClient,
- item: MediaItemType | ItemMapping | None,
-) -> dict[str, Any] | None:
+ item: MediaItemType | ItemMapping,
+) -> dict[str, Any]:
"""Parse a Music Assistant MediaItem."""
- if not item:
- return None
- base = {
+ base: dict[str, Any] = {
ATTR_MEDIA_TYPE: item.media_type,
ATTR_URI: item.uri,
ATTR_NAME: item.name,
ATTR_VERSION: item.version,
ATTR_IMAGE: mass.get_media_item_image_url(item),
}
+ artists: list[ItemMapping] | None
if artists := getattr(item, "artists", None):
base[ATTR_ARTISTS] = [media_item_dict_from_mass_item(mass, x) for x in artists]
+ album: ItemMapping | None
if album := getattr(item, "album", None):
base[ATTR_ALBUM] = media_item_dict_from_mass_item(mass, album)
return base
@@ -101,6 +103,12 @@ SEARCH_RESULT_SCHEMA = vol.Schema(
vol.Required(ATTR_RADIO): vol.All(
cv.ensure_list, [vol.Schema(MEDIA_ITEM_SCHEMA)]
),
+ vol.Required(ATTR_AUDIOBOOKS): vol.All(
+ cv.ensure_list, [vol.Schema(MEDIA_ITEM_SCHEMA)]
+ ),
+ vol.Required(ATTR_PODCASTS): vol.All(
+ cv.ensure_list, [vol.Schema(MEDIA_ITEM_SCHEMA)]
+ ),
},
)
@@ -151,7 +159,11 @@ def queue_item_dict_from_mass_item(
ATTR_QUEUE_ITEM_ID: item.queue_item_id,
ATTR_NAME: item.name,
ATTR_DURATION: item.duration,
- ATTR_MEDIA_ITEM: media_item_dict_from_mass_item(mass, item.media_item),
+ ATTR_MEDIA_ITEM: (
+ media_item_dict_from_mass_item(mass, item.media_item)
+ if item.media_item
+ else None
+ ),
}
if streamdetails := item.streamdetails:
base[ATTR_STREAM_TITLE] = streamdetails.stream_title
diff --git a/homeassistant/components/music_assistant/services.yaml b/homeassistant/components/music_assistant/services.yaml
index 73e8e2d7521..a3715ea2580 100644
--- a/homeassistant/components/music_assistant/services.yaml
+++ b/homeassistant/components/music_assistant/services.yaml
@@ -21,7 +21,10 @@ play_media:
options:
- artist
- album
+ - audiobook
+ - folder
- playlist
+ - podcast
- track
- radio
artist:
@@ -118,7 +121,9 @@ search:
options:
- artist
- album
+ - audiobook
- playlist
+ - podcast
- track
- radio
artist:
@@ -160,7 +165,9 @@ get_library:
options:
- artist
- album
+ - audiobook
- playlist
+ - podcast
- track
- radio
favorite:
diff --git a/homeassistant/components/music_assistant/strings.json b/homeassistant/components/music_assistant/strings.json
index 32b72088518..371ecdc3a86 100644
--- a/homeassistant/components/music_assistant/strings.json
+++ b/homeassistant/components/music_assistant/strings.json
@@ -139,8 +139,8 @@
}
},
"get_library": {
- "name": "Get Library items",
- "description": "Get items from a Music Assistant library.",
+ "name": "Get library items",
+ "description": "Retrieves items from a Music Assistant library.",
"fields": {
"config_entry_id": {
"name": "[%key:component::music_assistant::services::search::fields::config_entry_id::name%]",
@@ -167,7 +167,7 @@
"description": "Offset to start the list from."
},
"order_by": {
- "name": "Order By",
+ "name": "Order by",
"description": "Sort the list by this field."
},
"album_type": {
@@ -176,7 +176,7 @@
},
"album_artists_only": {
"name": "Enable album artists filter (only for artist library)",
- "description": "Only return Album Artists when listing the Artists library items."
+ "description": "Only return album artists when listing the artists library items."
}
}
}
@@ -195,8 +195,11 @@
"options": {
"artist": "Artist",
"album": "Album",
+ "audiobook": "Audiobook",
+ "folder": "Folder",
"track": "Track",
"playlist": "Playlist",
+ "podcast": "Podcast",
"radio": "Radio"
}
},
diff --git a/homeassistant/components/mutesync/binary_sensor.py b/homeassistant/components/mutesync/binary_sensor.py
index 87bf246f4e0..7a9025762ef 100644
--- a/homeassistant/components/mutesync/binary_sensor.py
+++ b/homeassistant/components/mutesync/binary_sensor.py
@@ -5,7 +5,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import update_coordinator
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -18,7 +18,7 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the mütesync button."""
coordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/myq/__init__.py b/homeassistant/components/myq/__init__.py
index 41b36a34c20..47629006887 100644
--- a/homeassistant/components/myq/__init__.py
+++ b/homeassistant/components/myq/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
@@ -29,11 +29,13 @@ async def async_setup_entry(hass: HomeAssistant, _: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
- if all(
- config_entry.state is ConfigEntryState.NOT_LOADED
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.entry_id != entry.entry_id
- ):
- ir.async_delete_issue(hass, DOMAIN, DOMAIN)
-
return True
+
+
+async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Remove a config entry."""
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
+ ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ # Remove any remaining disabled or ignored entries
+ for _entry in hass.config_entries.async_entries(DOMAIN):
+ hass.async_create_task(hass.config_entries.async_remove(_entry.entry_id))
diff --git a/homeassistant/components/mysensors/__init__.py b/homeassistant/components/mysensors/__init__.py
index 19dcce78446..e2aca8b9f01 100644
--- a/homeassistant/components/mysensors/__init__.py
+++ b/homeassistant/components/mysensors/__init__.py
@@ -17,7 +17,6 @@ from .const import (
DOMAIN,
MYSENSORS_DISCOVERED_NODES,
MYSENSORS_GATEWAYS,
- MYSENSORS_ON_UNLOAD,
PLATFORMS,
DevId,
DiscoveryInfo,
@@ -62,13 +61,6 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
if not unload_ok:
return False
- key = MYSENSORS_ON_UNLOAD.format(entry.entry_id)
- if key in hass.data[DOMAIN]:
- for fnct in hass.data[DOMAIN][key]:
- fnct()
-
- hass.data[DOMAIN].pop(key)
-
del hass.data[DOMAIN][MYSENSORS_GATEWAYS][entry.entry_id]
hass.data[DOMAIN].pop(MYSENSORS_DISCOVERED_NODES.format(entry.entry_id), None)
diff --git a/homeassistant/components/mysensors/binary_sensor.py b/homeassistant/components/mysensors/binary_sensor.py
index 54f7036b79c..e950f083b5b 100644
--- a/homeassistant/components/mysensors/binary_sensor.py
+++ b/homeassistant/components/mysensors/binary_sensor.py
@@ -15,12 +15,11 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo
from .entity import MySensorsChildEntity
-from .helpers import on_unload
@dataclass(frozen=True)
@@ -71,7 +70,7 @@ SENSORS: dict[str, MySensorsBinarySensorDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
@@ -86,9 +85,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.BINARY_SENSOR),
diff --git a/homeassistant/components/mysensors/climate.py b/homeassistant/components/mysensors/climate.py
index 23b7c47ebf3..eb54a76b8a8 100644
--- a/homeassistant/components/mysensors/climate.py
+++ b/homeassistant/components/mysensors/climate.py
@@ -15,13 +15,12 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_system import METRIC_SYSTEM
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo
from .entity import MySensorsChildEntity
-from .helpers import on_unload
DICT_HA_TO_MYS = {
HVACMode.AUTO: "AutoChangeOver",
@@ -43,7 +42,7 @@ OPERATION_LIST = [HVACMode.OFF, HVACMode.AUTO, HVACMode.COOL, HVACMode.HEAT]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
@@ -57,9 +56,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.CLIMATE),
@@ -85,7 +82,10 @@ class MySensorsHVAC(MySensorsChildEntity, ClimateEntity):
and set_req.V_HVAC_SETPOINT_HEAT in self._values
):
features = features | ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
- else:
+ elif (
+ set_req.V_HVAC_SETPOINT_COOL in self._values
+ or set_req.V_HVAC_SETPOINT_HEAT in self._values
+ ):
features = features | ClimateEntityFeature.TARGET_TEMPERATURE
return features
@@ -111,13 +111,11 @@ class MySensorsHVAC(MySensorsChildEntity, ClimateEntity):
@property
def target_temperature(self) -> float | None:
- """Return the temperature we try to reach."""
+ """Return the temperature we try to reach.
+
+ Either V_HVAC_SETPOINT_COOL or V_HVAC_SETPOINT_HEAT may be used.
+ """
set_req = self.gateway.const.SetReq
- if (
- set_req.V_HVAC_SETPOINT_COOL in self._values
- and set_req.V_HVAC_SETPOINT_HEAT in self._values
- ):
- return None
temp = self._values.get(set_req.V_HVAC_SETPOINT_COOL)
if temp is None:
temp = self._values.get(set_req.V_HVAC_SETPOINT_HEAT)
@@ -127,21 +125,13 @@ class MySensorsHVAC(MySensorsChildEntity, ClimateEntity):
def target_temperature_high(self) -> float | None:
"""Return the highbound target temperature we try to reach."""
set_req = self.gateway.const.SetReq
- if set_req.V_HVAC_SETPOINT_HEAT in self._values:
- temp = self._values.get(set_req.V_HVAC_SETPOINT_COOL)
- return float(temp) if temp is not None else None
-
- return None
+ return float(self._values[set_req.V_HVAC_SETPOINT_COOL])
@property
def target_temperature_low(self) -> float | None:
"""Return the lowbound target temperature we try to reach."""
set_req = self.gateway.const.SetReq
- if set_req.V_HVAC_SETPOINT_COOL in self._values:
- temp = self._values.get(set_req.V_HVAC_SETPOINT_HEAT)
- return float(temp) if temp is not None else None
-
- return None
+ return float(self._values[set_req.V_HVAC_SETPOINT_HEAT])
@property
def hvac_mode(self) -> HVACMode:
@@ -185,10 +175,6 @@ class MySensorsHVAC(MySensorsChildEntity, ClimateEntity):
self.gateway.set_child_value(
self.node_id, self.child_id, value_type, value, ack=1
)
- if self.assumed_state:
- # Optimistically assume that device has changed state
- self._values[value_type] = value
- self.async_write_ha_state()
async def async_set_fan_mode(self, fan_mode: str) -> None:
"""Set new target temperature."""
@@ -196,10 +182,6 @@ class MySensorsHVAC(MySensorsChildEntity, ClimateEntity):
self.gateway.set_child_value(
self.node_id, self.child_id, set_req.V_HVAC_SPEED, fan_mode, ack=1
)
- if self.assumed_state:
- # Optimistically assume that device has changed state
- self._values[set_req.V_HVAC_SPEED] = fan_mode
- self.async_write_ha_state()
async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
"""Set new target temperature."""
@@ -210,10 +192,6 @@ class MySensorsHVAC(MySensorsChildEntity, ClimateEntity):
DICT_HA_TO_MYS[hvac_mode],
ack=1,
)
- if self.assumed_state:
- # Optimistically assume that device has changed state
- self._values[self.value_type] = hvac_mode
- self.async_write_ha_state()
@callback
def _async_update(self) -> None:
diff --git a/homeassistant/components/mysensors/const.py b/homeassistant/components/mysensors/const.py
index a65b46616d3..a87b78b549e 100644
--- a/homeassistant/components/mysensors/const.py
+++ b/homeassistant/components/mysensors/const.py
@@ -34,7 +34,6 @@ CHILD_CALLBACK: str = "mysensors_child_callback_{}_{}_{}_{}"
NODE_CALLBACK: str = "mysensors_node_callback_{}_{}"
MYSENSORS_DISCOVERY: str = "mysensors_discovery_{}_{}"
MYSENSORS_NODE_DISCOVERY: str = "mysensors_node_discovery"
-MYSENSORS_ON_UNLOAD: str = "mysensors_on_unload_{}"
TYPE: Final = "type"
UPDATE_DELAY: float = 0.1
diff --git a/homeassistant/components/mysensors/cover.py b/homeassistant/components/mysensors/cover.py
index 808589b9022..84346a5d10a 100644
--- a/homeassistant/components/mysensors/cover.py
+++ b/homeassistant/components/mysensors/cover.py
@@ -7,15 +7,14 @@ from typing import Any
from homeassistant.components.cover import ATTR_POSITION, CoverEntity
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import STATE_OFF, STATE_ON, Platform
+from homeassistant.const import STATE_ON, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo
from .entity import MySensorsChildEntity
-from .helpers import on_unload
@unique
@@ -31,7 +30,7 @@ class CoverState(Enum):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
@@ -45,9 +44,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.COVER),
@@ -113,13 +110,6 @@ class MySensorsCover(MySensorsChildEntity, CoverEntity):
self.gateway.set_child_value(
self.node_id, self.child_id, set_req.V_UP, 1, ack=1
)
- if self.assumed_state:
- # Optimistically assume that cover has changed state.
- if set_req.V_DIMMER in self._values:
- self._values[set_req.V_DIMMER] = 100
- else:
- self._values[set_req.V_LIGHT] = STATE_ON
- self.async_write_ha_state()
async def async_close_cover(self, **kwargs: Any) -> None:
"""Move the cover down."""
@@ -127,13 +117,6 @@ class MySensorsCover(MySensorsChildEntity, CoverEntity):
self.gateway.set_child_value(
self.node_id, self.child_id, set_req.V_DOWN, 1, ack=1
)
- if self.assumed_state:
- # Optimistically assume that cover has changed state.
- if set_req.V_DIMMER in self._values:
- self._values[set_req.V_DIMMER] = 0
- else:
- self._values[set_req.V_LIGHT] = STATE_OFF
- self.async_write_ha_state()
async def async_set_cover_position(self, **kwargs: Any) -> None:
"""Move the cover to a specific position."""
@@ -142,10 +125,6 @@ class MySensorsCover(MySensorsChildEntity, CoverEntity):
self.gateway.set_child_value(
self.node_id, self.child_id, set_req.V_DIMMER, position, ack=1
)
- if self.assumed_state:
- # Optimistically assume that cover has changed state.
- self._values[set_req.V_DIMMER] = position
- self.async_write_ha_state()
async def async_stop_cover(self, **kwargs: Any) -> None:
"""Stop the device."""
diff --git a/homeassistant/components/mysensors/device_tracker.py b/homeassistant/components/mysensors/device_tracker.py
index 5abe6a64e2d..e6368b0b81d 100644
--- a/homeassistant/components/mysensors/device_tracker.py
+++ b/homeassistant/components/mysensors/device_tracker.py
@@ -7,18 +7,17 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo
from .entity import MySensorsChildEntity
-from .helpers import on_unload
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
@@ -33,9 +32,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.DEVICE_TRACKER),
diff --git a/homeassistant/components/mysensors/gateway.py b/homeassistant/components/mysensors/gateway.py
index bdc83f30b21..91453ea3306 100644
--- a/homeassistant/components/mysensors/gateway.py
+++ b/homeassistant/components/mysensors/gateway.py
@@ -47,7 +47,6 @@ from .handler import HANDLERS
from .helpers import (
discover_mysensors_node,
discover_mysensors_platform,
- on_unload,
validate_child,
validate_node,
)
@@ -293,9 +292,7 @@ async def _gw_start(
"""Stop the gateway."""
await gw_stop(hass, entry, gateway)
- on_unload(
- hass,
- entry.entry_id,
+ entry.async_on_unload(
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, stop_this_gw),
)
diff --git a/homeassistant/components/mysensors/helpers.py b/homeassistant/components/mysensors/helpers.py
index c96ad6cea8e..9ed41dfe4e9 100644
--- a/homeassistant/components/mysensors/helpers.py
+++ b/homeassistant/components/mysensors/helpers.py
@@ -27,7 +27,6 @@ from .const import (
MYSENSORS_DISCOVERED_NODES,
MYSENSORS_DISCOVERY,
MYSENSORS_NODE_DISCOVERY,
- MYSENSORS_ON_UNLOAD,
TYPE_TO_PLATFORMS,
DevId,
GatewayId,
@@ -41,18 +40,6 @@ SCHEMAS: Registry[
] = Registry()
-@callback
-def on_unload(hass: HomeAssistant, gateway_id: GatewayId, fnct: Callable) -> None:
- """Register a callback to be called when entry is unloaded.
-
- This function is used by platforms to cleanup after themselves.
- """
- key = MYSENSORS_ON_UNLOAD.format(gateway_id)
- if key not in hass.data[DOMAIN]:
- hass.data[DOMAIN][key] = []
- hass.data[DOMAIN][key].append(fnct)
-
-
@callback
def discover_mysensors_platform(
hass: HomeAssistant, gateway_id: GatewayId, platform: str, new_devices: list[DevId]
diff --git a/homeassistant/components/mysensors/light.py b/homeassistant/components/mysensors/light.py
index 87f60174cab..fa5e625c72b 100644
--- a/homeassistant/components/mysensors/light.py
+++ b/homeassistant/components/mysensors/light.py
@@ -12,22 +12,21 @@ from homeassistant.components.light import (
LightEntity,
)
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import STATE_OFF, STATE_ON, Platform
+from homeassistant.const import STATE_ON, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.color import rgb_hex_to_rgb_list
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo, SensorType
from .entity import MySensorsChildEntity
-from .helpers import on_unload
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
device_class_map: dict[SensorType, type[MySensorsChildEntity]] = {
@@ -46,9 +45,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.LIGHT),
@@ -80,11 +77,6 @@ class MySensorsLight(MySensorsChildEntity, LightEntity):
self.node_id, self.child_id, set_req.V_LIGHT, 1, ack=1
)
- if self.assumed_state:
- # optimistically assume that light has changed state
- self._state = True
- self._values[set_req.V_LIGHT] = STATE_ON
-
def _turn_on_dimmer(self, **kwargs: Any) -> None:
"""Turn on dimmer child device."""
set_req = self.gateway.const.SetReq
@@ -101,20 +93,10 @@ class MySensorsLight(MySensorsChildEntity, LightEntity):
self.node_id, self.child_id, set_req.V_DIMMER, percent, ack=1
)
- if self.assumed_state:
- # optimistically assume that light has changed state
- self._attr_brightness = brightness
- self._values[set_req.V_DIMMER] = percent
-
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the device off."""
value_type = self.gateway.const.SetReq.V_LIGHT
self.gateway.set_child_value(self.node_id, self.child_id, value_type, 0, ack=1)
- if self.assumed_state:
- # optimistically assume that light has changed state
- self._state = False
- self._values[value_type] = STATE_OFF
- self.async_write_ha_state()
@callback
def _async_update_light(self) -> None:
@@ -142,8 +124,6 @@ class MySensorsLightDimmer(MySensorsLight):
"""Turn the device on."""
self._turn_on_light()
self._turn_on_dimmer(**kwargs)
- if self.assumed_state:
- self.async_write_ha_state()
@callback
def _async_update(self) -> None:
@@ -164,8 +144,6 @@ class MySensorsLightRGB(MySensorsLight):
self._turn_on_light()
self._turn_on_dimmer(**kwargs)
self._turn_on_rgb(**kwargs)
- if self.assumed_state:
- self.async_write_ha_state()
def _turn_on_rgb(self, **kwargs: Any) -> None:
"""Turn on RGB child device."""
@@ -179,11 +157,6 @@ class MySensorsLightRGB(MySensorsLight):
self.node_id, self.child_id, self.value_type, hex_color, ack=1
)
- if self.assumed_state:
- # optimistically assume that light has changed state
- self._attr_rgb_color = new_rgb
- self._values[self.value_type] = hex_color
-
@callback
def _async_update(self) -> None:
"""Update the controller with the latest value from a sensor."""
@@ -212,8 +185,6 @@ class MySensorsLightRGBW(MySensorsLightRGB):
self._turn_on_light()
self._turn_on_dimmer(**kwargs)
self._turn_on_rgbw(**kwargs)
- if self.assumed_state:
- self.async_write_ha_state()
def _turn_on_rgbw(self, **kwargs: Any) -> None:
"""Turn on RGBW child device."""
@@ -227,11 +198,6 @@ class MySensorsLightRGBW(MySensorsLightRGB):
self.node_id, self.child_id, self.value_type, hex_color, ack=1
)
- if self.assumed_state:
- # optimistically assume that light has changed state
- self._attr_rgbw_color = new_rgbw
- self._values[self.value_type] = hex_color
-
@callback
def _async_update_rgb_or_w(self) -> None:
"""Update the controller with values from RGBW child."""
diff --git a/homeassistant/components/mysensors/remote.py b/homeassistant/components/mysensors/remote.py
index 1a4f6fdaa90..ccb67f78eba 100644
--- a/homeassistant/components/mysensors/remote.py
+++ b/homeassistant/components/mysensors/remote.py
@@ -14,18 +14,17 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo
from .entity import MySensorsChildEntity
-from .helpers import on_unload
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
@@ -40,9 +39,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.REMOTE),
diff --git a/homeassistant/components/mysensors/sensor.py b/homeassistant/components/mysensors/sensor.py
index eec3c6bcd79..3a7101e6b39 100644
--- a/homeassistant/components/mysensors/sensor.py
+++ b/homeassistant/components/mysensors/sensor.py
@@ -35,7 +35,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_system import METRIC_SYSTEM
from . import setup_mysensors_platform
@@ -50,7 +50,6 @@ from .const import (
NodeDiscoveryInfo,
)
from .entity import MySensorNodeEntity, MySensorsChildEntity
-from .helpers import on_unload
SENSORS: dict[str, SensorEntityDescription] = {
"V_TEMP": SensorEntityDescription(
@@ -102,6 +101,7 @@ SENSORS: dict[str, SensorEntityDescription] = {
key="V_DIRECTION",
native_unit_of_measurement=DEGREE,
icon="mdi:compass",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
"V_WEIGHT": SensorEntityDescription(
key="V_WEIGHT",
@@ -210,7 +210,7 @@ SENSORS: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
@@ -232,9 +232,7 @@ async def async_setup_entry(
gateway: BaseAsyncGateway = hass.data[DOMAIN][MYSENSORS_GATEWAYS][gateway_id]
async_add_entities([MyBatterySensor(gateway_id, gateway, node_id)])
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.SENSOR),
@@ -242,9 +240,7 @@ async def async_setup_entry(
),
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_NODE_DISCOVERY,
diff --git a/homeassistant/components/mysensors/switch.py b/homeassistant/components/mysensors/switch.py
index 4eabf6374f1..9b57102a94c 100644
--- a/homeassistant/components/mysensors/switch.py
+++ b/homeassistant/components/mysensors/switch.py
@@ -6,21 +6,20 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import STATE_OFF, STATE_ON, Platform
+from homeassistant.const import STATE_ON, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo, SensorType
from .entity import MySensorsChildEntity
-from .helpers import on_unload
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
device_class_map: dict[SensorType, type[MySensorsSwitch]] = {
@@ -48,9 +47,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.SWITCH),
@@ -72,17 +69,9 @@ class MySensorsSwitch(MySensorsChildEntity, SwitchEntity):
self.gateway.set_child_value(
self.node_id, self.child_id, self.value_type, 1, ack=1
)
- if self.assumed_state:
- # Optimistically assume that switch has changed state
- self._values[self.value_type] = STATE_ON
- self.async_write_ha_state()
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the switch off."""
self.gateway.set_child_value(
self.node_id, self.child_id, self.value_type, 0, ack=1
)
- if self.assumed_state:
- # Optimistically assume that switch has changed state
- self._values[self.value_type] = STATE_OFF
- self.async_write_ha_state()
diff --git a/homeassistant/components/mysensors/text.py b/homeassistant/components/mysensors/text.py
index 4edb5ccdbd8..9fdd9da5345 100644
--- a/homeassistant/components/mysensors/text.py
+++ b/homeassistant/components/mysensors/text.py
@@ -7,18 +7,17 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import setup_mysensors_platform
from .const import MYSENSORS_DISCOVERY, DiscoveryInfo
from .entity import MySensorsChildEntity
-from .helpers import on_unload
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up this platform for a specific ConfigEntry(==Gateway)."""
@@ -33,9 +32,7 @@ async def async_setup_entry(
async_add_entities=async_add_entities,
)
- on_unload(
- hass,
- config_entry.entry_id,
+ config_entry.async_on_unload(
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(config_entry.entry_id, Platform.TEXT),
diff --git a/homeassistant/components/mystrom/light.py b/homeassistant/components/mystrom/light.py
index 5dabb609437..3942f601a20 100644
--- a/homeassistant/components/mystrom/light.py
+++ b/homeassistant/components/mystrom/light.py
@@ -18,7 +18,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, MANUFACTURER
@@ -31,7 +31,9 @@ EFFECT_SUNRISE = "sunrise"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the myStrom entities."""
info = hass.data[DOMAIN][entry.entry_id].info
diff --git a/homeassistant/components/mystrom/sensor.py b/homeassistant/components/mystrom/sensor.py
index 2c35d35dad6..bd5c9b923a2 100644
--- a/homeassistant/components/mystrom/sensor.py
+++ b/homeassistant/components/mystrom/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfPower, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, MANUFACTURER
@@ -55,7 +55,9 @@ SENSOR_TYPES: tuple[MyStromSwitchSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the myStrom entities."""
device: MyStromSwitch = hass.data[DOMAIN][entry.entry_id].device
diff --git a/homeassistant/components/mystrom/switch.py b/homeassistant/components/mystrom/switch.py
index af135027aac..f626656a4e3 100644
--- a/homeassistant/components/mystrom/switch.py
+++ b/homeassistant/components/mystrom/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo, format_mac
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, MANUFACTURER
@@ -21,7 +21,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the myStrom entities."""
device = hass.data[DOMAIN][entry.entry_id].device
diff --git a/homeassistant/components/myuplink/__init__.py b/homeassistant/components/myuplink/__init__.py
index 5ad114e973e..407e4da5475 100644
--- a/homeassistant/components/myuplink/__init__.py
+++ b/homeassistant/components/myuplink/__init__.py
@@ -9,7 +9,6 @@ from aiohttp import ClientError, ClientResponseError
import jwt
from myuplink import MyUplinkAPI, get_manufacturer, get_model, get_system_name
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -22,7 +21,7 @@ from homeassistant.helpers.device_registry import DeviceEntry
from .api import AsyncConfigEntryAuth
from .const import DOMAIN, OAUTH2_SCOPES
-from .coordinator import MyUplinkDataCoordinator
+from .coordinator import MyUplinkConfigEntry, MyUplinkDataCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -35,8 +34,6 @@ PLATFORMS: list[Platform] = [
Platform.UPDATE,
]
-type MyUplinkConfigEntry = ConfigEntry[MyUplinkDataCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, config_entry: MyUplinkConfigEntry
@@ -77,7 +74,7 @@ async def async_setup_entry(
# Setup MyUplinkAPI and coordinator for data fetch
api = MyUplinkAPI(auth)
- coordinator = MyUplinkDataCoordinator(hass, api)
+ coordinator = MyUplinkDataCoordinator(hass, config_entry, api)
await coordinator.async_config_entry_first_refresh()
config_entry.runtime_data = coordinator
diff --git a/homeassistant/components/myuplink/binary_sensor.py b/homeassistant/components/myuplink/binary_sensor.py
index d903c7cbfae..785a7ff4532 100644
--- a/homeassistant/components/myuplink/binary_sensor.py
+++ b/homeassistant/components/myuplink/binary_sensor.py
@@ -9,10 +9,10 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .const import F_SERIES
+from .coordinator import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .entity import MyUplinkEntity, MyUplinkSystemEntity
from .helpers import find_matching_platform, transform_model_series
@@ -58,7 +58,7 @@ def get_description(device_point: DevicePoint) -> BinarySensorEntityDescription
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MyUplinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up myUplink binary_sensor."""
entities: list[BinarySensorEntity] = []
diff --git a/homeassistant/components/myuplink/coordinator.py b/homeassistant/components/myuplink/coordinator.py
index 211fd894ac5..6bf762ad642 100644
--- a/homeassistant/components/myuplink/coordinator.py
+++ b/homeassistant/components/myuplink/coordinator.py
@@ -7,6 +7,7 @@ import logging
from myuplink import Device, DevicePoint, MyUplinkAPI, System
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -23,14 +24,22 @@ class CoordinatorData:
time: datetime
+type MyUplinkConfigEntry = ConfigEntry[MyUplinkDataCoordinator]
+
+
class MyUplinkDataCoordinator(DataUpdateCoordinator[CoordinatorData]):
"""Coordinator for myUplink data."""
- def __init__(self, hass: HomeAssistant, api: MyUplinkAPI) -> None:
+ config_entry: MyUplinkConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: MyUplinkConfigEntry, api: MyUplinkAPI
+ ) -> None:
"""Initialize myUplink coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="myuplink",
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/myuplink/diagnostics.py b/homeassistant/components/myuplink/diagnostics.py
index 5e26cf273b4..61605a04fc8 100644
--- a/homeassistant/components/myuplink/diagnostics.py
+++ b/homeassistant/components/myuplink/diagnostics.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant
-from . import MyUplinkConfigEntry
+from .coordinator import MyUplinkConfigEntry
TO_REDACT = {"access_token", "refresh_token", "serialNumber"}
diff --git a/homeassistant/components/myuplink/number.py b/homeassistant/components/myuplink/number.py
index e1cbd393947..33100850837 100644
--- a/homeassistant/components/myuplink/number.py
+++ b/homeassistant/components/myuplink/number.py
@@ -7,10 +7,10 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescriptio
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .const import DOMAIN, F_SERIES
+from .coordinator import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .entity import MyUplinkEntity
from .helpers import find_matching_platform, skip_entity, transform_model_series
@@ -63,7 +63,7 @@ def get_description(device_point: DevicePoint) -> NumberEntityDescription | None
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MyUplinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up myUplink number."""
entities: list[NumberEntity] = []
diff --git a/homeassistant/components/myuplink/select.py b/homeassistant/components/myuplink/select.py
index 0074d1c75ff..36f9be63669 100644
--- a/homeassistant/components/myuplink/select.py
+++ b/homeassistant/components/myuplink/select.py
@@ -9,10 +9,10 @@ from homeassistant.components.select import SelectEntity
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .const import DOMAIN
+from .coordinator import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .entity import MyUplinkEntity
from .helpers import find_matching_platform, skip_entity
@@ -20,7 +20,7 @@ from .helpers import find_matching_platform, skip_entity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MyUplinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up myUplink select."""
entities: list[SelectEntity] = []
diff --git a/homeassistant/components/myuplink/sensor.py b/homeassistant/components/myuplink/sensor.py
index fa50e8a7001..3b14cdd4630 100644
--- a/homeassistant/components/myuplink/sensor.py
+++ b/homeassistant/components/myuplink/sensor.py
@@ -21,11 +21,11 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .const import F_SERIES
+from .coordinator import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .entity import MyUplinkEntity
from .helpers import find_matching_platform, skip_entity, transform_model_series
@@ -214,7 +214,7 @@ def get_description(device_point: DevicePoint) -> SensorEntityDescription | None
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MyUplinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up myUplink sensor."""
diff --git a/homeassistant/components/myuplink/switch.py b/homeassistant/components/myuplink/switch.py
index 3addc7ce6a9..2d3706f2bdb 100644
--- a/homeassistant/components/myuplink/switch.py
+++ b/homeassistant/components/myuplink/switch.py
@@ -9,10 +9,10 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .const import DOMAIN, F_SERIES
+from .coordinator import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .entity import MyUplinkEntity
from .helpers import find_matching_platform, skip_entity, transform_model_series
@@ -55,7 +55,7 @@ def get_description(device_point: DevicePoint) -> SwitchEntityDescription | None
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MyUplinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up myUplink switch."""
entities: list[SwitchEntity] = []
diff --git a/homeassistant/components/myuplink/update.py b/homeassistant/components/myuplink/update.py
index 9e94de0a503..ee259f5cbe8 100644
--- a/homeassistant/components/myuplink/update.py
+++ b/homeassistant/components/myuplink/update.py
@@ -6,9 +6,9 @@ from homeassistant.components.update import (
UpdateEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import MyUplinkConfigEntry, MyUplinkDataCoordinator
+from .coordinator import MyUplinkConfigEntry, MyUplinkDataCoordinator
from .entity import MyUplinkEntity
UPDATE_DESCRIPTION = UpdateEntityDescription(
@@ -20,7 +20,7 @@ UPDATE_DESCRIPTION = UpdateEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: MyUplinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up update entity."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/nam/__init__.py b/homeassistant/components/nam/__init__.py
index 624415adb12..6b4ca6ff324 100644
--- a/homeassistant/components/nam/__init__.py
+++ b/homeassistant/components/nam/__init__.py
@@ -3,7 +3,6 @@
from __future__ import annotations
import logging
-from typing import TYPE_CHECKING
from aiohttp.client_exceptions import ClientConnectorError, ClientError
from nettigo_air_monitor import (
@@ -14,7 +13,6 @@ from nettigo_air_monitor import (
)
from homeassistant.components.air_quality import DOMAIN as AIR_QUALITY_PLATFORM
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -22,14 +20,12 @@ from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import ATTR_SDS011, ATTR_SPS30, DOMAIN
-from .coordinator import NAMDataUpdateCoordinator
+from .coordinator import NAMConfigEntry, NAMDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.BUTTON, Platform.SENSOR]
-type NAMConfigEntry = ConfigEntry[NAMDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: NAMConfigEntry) -> bool:
"""Set up Nettigo as config entry."""
@@ -52,10 +48,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NAMConfigEntry) -> bool:
except AuthFailedError as err:
raise ConfigEntryAuthFailed from err
- if TYPE_CHECKING:
- assert entry.unique_id
-
- coordinator = NAMDataUpdateCoordinator(hass, nam, entry.unique_id)
+ coordinator = NAMDataUpdateCoordinator(hass, entry, nam)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/nam/button.py b/homeassistant/components/nam/button.py
index 8ac56f3d70e..60145e4fe27 100644
--- a/homeassistant/components/nam/button.py
+++ b/homeassistant/components/nam/button.py
@@ -11,10 +11,10 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import NAMConfigEntry, NAMDataUpdateCoordinator
+from .coordinator import NAMConfigEntry, NAMDataUpdateCoordinator
PARALLEL_UPDATES = 1
@@ -28,7 +28,9 @@ RESTART_BUTTON: ButtonEntityDescription = ButtonEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: NAMConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NAMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a Nettigo Air Monitor entities from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/nam/const.py b/homeassistant/components/nam/const.py
index 4b7b50b309a..2dedcf3c68a 100644
--- a/homeassistant/components/nam/const.py
+++ b/homeassistant/components/nam/const.py
@@ -11,6 +11,7 @@ SUFFIX_P1: Final = "_p1"
SUFFIX_P2: Final = "_p2"
SUFFIX_P4: Final = "_p4"
+ATTR_BH1750_ILLUMINANCE: Final = "bh1750_illuminance"
ATTR_BME280_HUMIDITY: Final = "bme280_humidity"
ATTR_BME280_PRESSURE: Final = "bme280_pressure"
ATTR_BME280_TEMPERATURE: Final = "bme280_temperature"
diff --git a/homeassistant/components/nam/coordinator.py b/homeassistant/components/nam/coordinator.py
index 5019f0e3a1d..3e2c9c24474 100644
--- a/homeassistant/components/nam/coordinator.py
+++ b/homeassistant/components/nam/coordinator.py
@@ -1,6 +1,7 @@
"""The Nettigo Air Monitor coordinator."""
import logging
+from typing import TYPE_CHECKING
from nettigo_air_monitor import (
ApiError,
@@ -10,6 +11,7 @@ from nettigo_air_monitor import (
)
from tenacity import RetryError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -18,20 +20,28 @@ from .const import DEFAULT_UPDATE_INTERVAL, DOMAIN, MANUFACTURER
_LOGGER = logging.getLogger(__name__)
+type NAMConfigEntry = ConfigEntry[NAMDataUpdateCoordinator]
+
class NAMDataUpdateCoordinator(DataUpdateCoordinator[NAMSensors]):
"""Class to manage fetching Nettigo Air Monitor data."""
+ config_entry: NAMConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: NAMConfigEntry,
nam: NettigoAirMonitor,
- unique_id: str,
) -> None:
"""Initialize."""
- self.unique_id = unique_id
+ if TYPE_CHECKING:
+ assert config_entry.unique_id
+
+ self.unique_id = config_entry.unique_id
+
self.device_info = DeviceInfo(
- connections={(CONNECTION_NETWORK_MAC, unique_id)},
+ connections={(CONNECTION_NETWORK_MAC, self.unique_id)},
name="Nettigo Air Monitor",
sw_version=nam.software_version,
manufacturer=MANUFACTURER,
@@ -40,7 +50,11 @@ class NAMDataUpdateCoordinator(DataUpdateCoordinator[NAMSensors]):
self.nam = nam
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=DEFAULT_UPDATE_INTERVAL
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=DEFAULT_UPDATE_INTERVAL,
)
async def _async_update_data(self) -> NAMSensors:
diff --git a/homeassistant/components/nam/diagnostics.py b/homeassistant/components/nam/diagnostics.py
index d29eb40ced7..905c1669496 100644
--- a/homeassistant/components/nam/diagnostics.py
+++ b/homeassistant/components/nam/diagnostics.py
@@ -9,7 +9,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
-from . import NAMConfigEntry
+from .coordinator import NAMConfigEntry
TO_REDACT = {CONF_PASSWORD, CONF_USERNAME}
diff --git a/homeassistant/components/nam/manifest.json b/homeassistant/components/nam/manifest.json
index c3a559de50b..1c3b9db7a86 100644
--- a/homeassistant/components/nam/manifest.json
+++ b/homeassistant/components/nam/manifest.json
@@ -7,7 +7,7 @@
"integration_type": "device",
"iot_class": "local_polling",
"loggers": ["nettigo_air_monitor"],
- "requirements": ["nettigo-air-monitor==4.0.0"],
+ "requirements": ["nettigo-air-monitor==4.1.0"],
"zeroconf": [
{
"type": "_http._tcp.local.",
diff --git a/homeassistant/components/nam/sensor.py b/homeassistant/components/nam/sensor.py
index 27fae62be8a..45cfd313e8f 100644
--- a/homeassistant/components/nam/sensor.py
+++ b/homeassistant/components/nam/sensor.py
@@ -19,6 +19,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import (
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
CONCENTRATION_PARTS_PER_MILLION,
+ LIGHT_LUX,
PERCENTAGE,
SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
EntityCategory,
@@ -27,13 +28,13 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.dt import utcnow
-from . import NAMConfigEntry, NAMDataUpdateCoordinator
from .const import (
+ ATTR_BH1750_ILLUMINANCE,
ATTR_BME280_HUMIDITY,
ATTR_BME280_PRESSURE,
ATTR_BME280_TEMPERATURE,
@@ -69,6 +70,7 @@ from .const import (
DOMAIN,
MIGRATION_SENSORS,
)
+from .coordinator import NAMConfigEntry, NAMDataUpdateCoordinator
PARALLEL_UPDATES = 1
@@ -83,6 +85,15 @@ class NAMSensorEntityDescription(SensorEntityDescription):
SENSORS: tuple[NAMSensorEntityDescription, ...] = (
+ NAMSensorEntityDescription(
+ key=ATTR_BH1750_ILLUMINANCE,
+ translation_key="bh1750_illuminance",
+ suggested_display_precision=0,
+ native_unit_of_measurement=LIGHT_LUX,
+ device_class=SensorDeviceClass.ILLUMINANCE,
+ state_class=SensorStateClass.MEASUREMENT,
+ value=lambda sensors: sensors.bh1750_illuminance,
+ ),
NAMSensorEntityDescription(
key=ATTR_BME280_HUMIDITY,
translation_key="bme280_humidity",
@@ -356,7 +367,9 @@ SENSORS: tuple[NAMSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: NAMConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NAMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a Nettigo Air Monitor entities from a config_entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/nam/strings.json b/homeassistant/components/nam/strings.json
index 2caa4d8bd97..be9fb1fbb07 100644
--- a/homeassistant/components/nam/strings.json
+++ b/homeassistant/components/nam/strings.json
@@ -54,6 +54,9 @@
},
"entity": {
"sensor": {
+ "bh1750_illuminance": {
+ "name": "BH1750 illuminance"
+ },
"bme280_humidity": {
"name": "BME280 humidity"
},
@@ -98,6 +101,17 @@
"medium": "Medium",
"high": "High",
"very_high": "Very high"
+ },
+ "state_attributes": {
+ "options": {
+ "state": {
+ "very_low": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_low%]",
+ "low": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::low%]",
+ "medium": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::medium%]",
+ "high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::high%]",
+ "very_high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_high%]"
+ }
+ }
}
},
"pmsx003_pm1": {
@@ -120,6 +134,17 @@
"medium": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::medium%]",
"high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::high%]",
"very_high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_high%]"
+ },
+ "state_attributes": {
+ "options": {
+ "state": {
+ "very_low": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_low%]",
+ "low": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::low%]",
+ "medium": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::medium%]",
+ "high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::high%]",
+ "very_high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_high%]"
+ }
+ }
}
},
"sds011_pm10": {
@@ -145,6 +170,17 @@
"medium": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::medium%]",
"high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::high%]",
"very_high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_high%]"
+ },
+ "state_attributes": {
+ "options": {
+ "state": {
+ "very_low": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_low%]",
+ "low": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::low%]",
+ "medium": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::medium%]",
+ "high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::high%]",
+ "very_high": "[%key:component::nam::entity::sensor::pmsx003_caqi_level::state::very_high%]"
+ }
+ }
}
},
"sps30_pm1": {
diff --git a/homeassistant/components/nanoleaf/__init__.py b/homeassistant/components/nanoleaf/__init__.py
index 4a34c2843aa..7ee1c14a9b1 100644
--- a/homeassistant/components/nanoleaf/__init__.py
+++ b/homeassistant/components/nanoleaf/__init__.py
@@ -8,7 +8,6 @@ import logging
from aionanoleaf import EffectsEvent, Nanoleaf, StateEvent, TouchEvent
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_DEVICE_ID,
CONF_HOST,
@@ -22,23 +21,20 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.dispatcher import async_dispatcher_send
from .const import DOMAIN, NANOLEAF_EVENT, TOUCH_GESTURE_TRIGGER_MAP, TOUCH_MODELS
-from .coordinator import NanoleafCoordinator
+from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
_LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.BUTTON, Platform.EVENT, Platform.LIGHT]
-type NanoleafConfigEntry = ConfigEntry[NanoleafCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: NanoleafConfigEntry) -> bool:
"""Set up Nanoleaf from a config entry."""
nanoleaf = Nanoleaf(
async_get_clientsession(hass), entry.data[CONF_HOST], entry.data[CONF_TOKEN]
)
- coordinator = NanoleafCoordinator(hass, nanoleaf)
+ coordinator = NanoleafCoordinator(hass, entry, nanoleaf)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/nanoleaf/button.py b/homeassistant/components/nanoleaf/button.py
index 34d0f4f5076..813d81ab571 100644
--- a/homeassistant/components/nanoleaf/button.py
+++ b/homeassistant/components/nanoleaf/button.py
@@ -3,17 +3,16 @@
from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NanoleafConfigEntry
-from .coordinator import NanoleafCoordinator
+from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
from .entity import NanoleafEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: NanoleafConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nanoleaf button."""
async_add_entities([NanoleafIdentifyButton(entry.runtime_data)])
diff --git a/homeassistant/components/nanoleaf/coordinator.py b/homeassistant/components/nanoleaf/coordinator.py
index e080afc492e..495a63b9164 100644
--- a/homeassistant/components/nanoleaf/coordinator.py
+++ b/homeassistant/components/nanoleaf/coordinator.py
@@ -5,20 +5,31 @@ import logging
from aionanoleaf import InvalidToken, Nanoleaf, Unavailable
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
_LOGGER = logging.getLogger(__name__)
+type NanoleafConfigEntry = ConfigEntry[NanoleafCoordinator]
+
class NanoleafCoordinator(DataUpdateCoordinator[None]):
"""Class to manage fetching Nanoleaf data."""
- def __init__(self, hass: HomeAssistant, nanoleaf: Nanoleaf) -> None:
+ config_entry: NanoleafConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: NanoleafConfigEntry, nanoleaf: Nanoleaf
+ ) -> None:
"""Initialize the Nanoleaf data coordinator."""
super().__init__(
- hass, _LOGGER, name="Nanoleaf", update_interval=timedelta(minutes=1)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name="Nanoleaf",
+ update_interval=timedelta(minutes=1),
)
self.nanoleaf = nanoleaf
diff --git a/homeassistant/components/nanoleaf/diagnostics.py b/homeassistant/components/nanoleaf/diagnostics.py
index 6f8691905ef..ce2045acf7b 100644
--- a/homeassistant/components/nanoleaf/diagnostics.py
+++ b/homeassistant/components/nanoleaf/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_TOKEN
from homeassistant.core import HomeAssistant
-from . import NanoleafConfigEntry
+from .coordinator import NanoleafConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/nanoleaf/entity.py b/homeassistant/components/nanoleaf/entity.py
index ffe4a098022..dd0b455fa0f 100644
--- a/homeassistant/components/nanoleaf/entity.py
+++ b/homeassistant/components/nanoleaf/entity.py
@@ -3,8 +3,8 @@
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import NanoleafCoordinator
from .const import DOMAIN
+from .coordinator import NanoleafCoordinator
class NanoleafEntity(CoordinatorEntity[NanoleafCoordinator]):
diff --git a/homeassistant/components/nanoleaf/event.py b/homeassistant/components/nanoleaf/event.py
index 5763c2aa595..78ff889bdc5 100644
--- a/homeassistant/components/nanoleaf/event.py
+++ b/homeassistant/components/nanoleaf/event.py
@@ -3,17 +3,17 @@
from homeassistant.components.event import EventEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NanoleafConfigEntry, NanoleafCoordinator
from .const import TOUCH_MODELS
+from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
from .entity import NanoleafEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: NanoleafConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nanoleaf event."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/nanoleaf/light.py b/homeassistant/components/nanoleaf/light.py
index 681053fa573..6d42110d53e 100644
--- a/homeassistant/components/nanoleaf/light.py
+++ b/homeassistant/components/nanoleaf/light.py
@@ -15,10 +15,9 @@ from homeassistant.components.light import (
LightEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NanoleafConfigEntry
-from .coordinator import NanoleafCoordinator
+from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
from .entity import NanoleafEntity
RESERVED_EFFECTS = ("*Solid*", "*Static*", "*Dynamic*")
@@ -28,7 +27,7 @@ DEFAULT_NAME = "Nanoleaf"
async def async_setup_entry(
hass: HomeAssistant,
entry: NanoleafConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nanoleaf light."""
async_add_entities([NanoleafLight(entry.runtime_data)])
diff --git a/homeassistant/components/nanoleaf/manifest.json b/homeassistant/components/nanoleaf/manifest.json
index 4b4c026260d..7af7465bbd0 100644
--- a/homeassistant/components/nanoleaf/manifest.json
+++ b/homeassistant/components/nanoleaf/manifest.json
@@ -5,7 +5,7 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/nanoleaf",
"homekit": {
- "models": ["NL29", "NL42", "NL47", "NL48", "NL52", "NL59"]
+ "models": ["NL29", "NL42", "NL47", "NL48", "NL52", "NL59", "NL69", "NL81"]
},
"iot_class": "local_push",
"loggers": ["aionanoleaf"],
@@ -22,6 +22,12 @@
},
{
"st": "nanoleaf:nl52"
+ },
+ {
+ "st": "nanoleaf:nl69"
+ },
+ {
+ "st": "inanoleaf:nl81"
}
],
"zeroconf": ["_nanoleafms._tcp.local.", "_nanoleafapi._tcp.local."]
diff --git a/homeassistant/components/nasweb/switch.py b/homeassistant/components/nasweb/switch.py
index c5a9e085b83..740db1ed1a1 100644
--- a/homeassistant/components/nasweb/switch.py
+++ b/homeassistant/components/nasweb/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import DOMAIN as DOMAIN_SWITCH, SwitchEntit
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import DiscoveryInfoType
from homeassistant.helpers.update_coordinator import (
BaseCoordinatorEntity,
@@ -38,7 +38,7 @@ def _get_output(coordinator: NASwebCoordinator, index: int) -> NASwebOutput | No
async def async_setup_entry(
hass: HomeAssistant,
config: NASwebConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up switch platform."""
diff --git a/homeassistant/components/neato/button.py b/homeassistant/components/neato/button.py
index 29114ce5188..8658dfd1b1b 100644
--- a/homeassistant/components/neato/button.py
+++ b/homeassistant/components/neato/button.py
@@ -8,14 +8,16 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import NEATO_ROBOTS
from .entity import NeatoEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Neato button from config entry."""
entities = [NeatoDismissAlertButton(robot) for robot in hass.data[NEATO_ROBOTS]]
diff --git a/homeassistant/components/neato/camera.py b/homeassistant/components/neato/camera.py
index e4d5f81f33a..42278a3a48f 100644
--- a/homeassistant/components/neato/camera.py
+++ b/homeassistant/components/neato/camera.py
@@ -13,7 +13,7 @@ from urllib3.response import HTTPResponse
from homeassistant.components.camera import Camera
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import NEATO_LOGIN, NEATO_MAP_DATA, NEATO_ROBOTS, SCAN_INTERVAL_MINUTES
from .entity import NeatoEntity
@@ -26,7 +26,9 @@ ATTR_GENERATED_AT = "generated_at"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Neato camera with config entry."""
neato: NeatoHub = hass.data[NEATO_LOGIN]
diff --git a/homeassistant/components/neato/manifest.json b/homeassistant/components/neato/manifest.json
index e4b471cb5ac..ef7cda52f19 100644
--- a/homeassistant/components/neato/manifest.json
+++ b/homeassistant/components/neato/manifest.json
@@ -7,5 +7,5 @@
"documentation": "https://www.home-assistant.io/integrations/neato",
"iot_class": "cloud_polling",
"loggers": ["pybotvac"],
- "requirements": ["pybotvac==0.0.25"]
+ "requirements": ["pybotvac==0.0.26"]
}
diff --git a/homeassistant/components/neato/sensor.py b/homeassistant/components/neato/sensor.py
index c247cc48493..4be02fe1ef7 100644
--- a/homeassistant/components/neato/sensor.py
+++ b/homeassistant/components/neato/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import NEATO_LOGIN, NEATO_ROBOTS, SCAN_INTERVAL_MINUTES
from .entity import NeatoEntity
@@ -27,7 +27,9 @@ BATTERY = "Battery"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Neato sensor using config entry."""
neato: NeatoHub = hass.data[NEATO_LOGIN]
diff --git a/homeassistant/components/neato/switch.py b/homeassistant/components/neato/switch.py
index 25da1c41df1..1ae06fef44c 100644
--- a/homeassistant/components/neato/switch.py
+++ b/homeassistant/components/neato/switch.py
@@ -13,7 +13,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import NEATO_LOGIN, NEATO_ROBOTS, SCAN_INTERVAL_MINUTES
from .entity import NeatoEntity
@@ -29,7 +29,9 @@ SWITCH_TYPES = {SWITCH_TYPE_SCHEDULE: ["Schedule"]}
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Neato switch with config entry."""
neato: NeatoHub = hass.data[NEATO_LOGIN]
diff --git a/homeassistant/components/neato/vacuum.py b/homeassistant/components/neato/vacuum.py
index 1a9285964a2..a1e1382eb04 100644
--- a/homeassistant/components/neato/vacuum.py
+++ b/homeassistant/components/neato/vacuum.py
@@ -21,7 +21,7 @@ from homeassistant.const import ATTR_MODE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ACTION,
@@ -58,7 +58,9 @@ ATTR_ZONE = "zone"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Neato vacuum with config entry."""
neato: NeatoHub = hass.data[NEATO_LOGIN]
diff --git a/homeassistant/components/nederlandse_spoorwegen/sensor.py b/homeassistant/components/nederlandse_spoorwegen/sensor.py
index ff3eea9252c..1e7fc54f4f7 100644
--- a/homeassistant/components/nederlandse_spoorwegen/sensor.py
+++ b/homeassistant/components/nederlandse_spoorwegen/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
-from homeassistant.util import Throttle
+from homeassistant.util import Throttle, dt as dt_util
_LOGGER = logging.getLogger(__name__)
@@ -119,6 +119,8 @@ class NSDepartureSensor(SensorEntity):
self._time = time
self._state = None
self._trips = None
+ self._first_trip = None
+ self._next_trip = None
@property
def name(self):
@@ -133,44 +135,44 @@ class NSDepartureSensor(SensorEntity):
@property
def extra_state_attributes(self):
"""Return the state attributes."""
- if not self._trips:
+ if not self._trips or self._first_trip is None:
return None
- if self._trips[0].trip_parts:
- route = [self._trips[0].departure]
- route.extend(k.destination for k in self._trips[0].trip_parts)
+ if self._first_trip.trip_parts:
+ route = [self._first_trip.departure]
+ route.extend(k.destination for k in self._first_trip.trip_parts)
# Static attributes
attributes = {
- "going": self._trips[0].going,
+ "going": self._first_trip.going,
"departure_time_planned": None,
"departure_time_actual": None,
"departure_delay": False,
- "departure_platform_planned": self._trips[0].departure_platform_planned,
- "departure_platform_actual": self._trips[0].departure_platform_actual,
+ "departure_platform_planned": self._first_trip.departure_platform_planned,
+ "departure_platform_actual": self._first_trip.departure_platform_actual,
"arrival_time_planned": None,
"arrival_time_actual": None,
"arrival_delay": False,
- "arrival_platform_planned": self._trips[0].arrival_platform_planned,
- "arrival_platform_actual": self._trips[0].arrival_platform_actual,
+ "arrival_platform_planned": self._first_trip.arrival_platform_planned,
+ "arrival_platform_actual": self._first_trip.arrival_platform_actual,
"next": None,
- "status": self._trips[0].status.lower(),
- "transfers": self._trips[0].nr_transfers,
+ "status": self._first_trip.status.lower(),
+ "transfers": self._first_trip.nr_transfers,
"route": route,
"remarks": None,
}
# Planned departure attributes
- if self._trips[0].departure_time_planned is not None:
- attributes["departure_time_planned"] = self._trips[
- 0
- ].departure_time_planned.strftime("%H:%M")
+ if self._first_trip.departure_time_planned is not None:
+ attributes["departure_time_planned"] = (
+ self._first_trip.departure_time_planned.strftime("%H:%M")
+ )
# Actual departure attributes
- if self._trips[0].departure_time_actual is not None:
- attributes["departure_time_actual"] = self._trips[
- 0
- ].departure_time_actual.strftime("%H:%M")
+ if self._first_trip.departure_time_actual is not None:
+ attributes["departure_time_actual"] = (
+ self._first_trip.departure_time_actual.strftime("%H:%M")
+ )
# Delay departure attributes
if (
@@ -182,16 +184,16 @@ class NSDepartureSensor(SensorEntity):
attributes["departure_delay"] = True
# Planned arrival attributes
- if self._trips[0].arrival_time_planned is not None:
- attributes["arrival_time_planned"] = self._trips[
- 0
- ].arrival_time_planned.strftime("%H:%M")
+ if self._first_trip.arrival_time_planned is not None:
+ attributes["arrival_time_planned"] = (
+ self._first_trip.arrival_time_planned.strftime("%H:%M")
+ )
# Actual arrival attributes
- if self._trips[0].arrival_time_actual is not None:
- attributes["arrival_time_actual"] = self._trips[
- 0
- ].arrival_time_actual.strftime("%H:%M")
+ if self._first_trip.arrival_time_actual is not None:
+ attributes["arrival_time_actual"] = (
+ self._first_trip.arrival_time_actual.strftime("%H:%M")
+ )
# Delay arrival attributes
if (
@@ -202,15 +204,14 @@ class NSDepartureSensor(SensorEntity):
attributes["arrival_delay"] = True
# Next attributes
- if len(self._trips) > 1:
- if self._trips[1].departure_time_actual is not None:
- attributes["next"] = self._trips[1].departure_time_actual.strftime(
- "%H:%M"
- )
- elif self._trips[1].departure_time_planned is not None:
- attributes["next"] = self._trips[1].departure_time_planned.strftime(
- "%H:%M"
- )
+ if self._next_trip.departure_time_actual is not None:
+ attributes["next"] = self._next_trip.departure_time_actual.strftime("%H:%M")
+ elif self._next_trip.departure_time_planned is not None:
+ attributes["next"] = self._next_trip.departure_time_planned.strftime(
+ "%H:%M"
+ )
+ else:
+ attributes["next"] = None
return attributes
@@ -225,6 +226,7 @@ class NSDepartureSensor(SensorEntity):
):
self._state = None
self._trips = None
+ self._first_trip = None
return
# Set the search parameter to search from a specific trip time
@@ -236,19 +238,51 @@ class NSDepartureSensor(SensorEntity):
.strftime("%d-%m-%Y %H:%M")
)
else:
- trip_time = datetime.now().strftime("%d-%m-%Y %H:%M")
+ trip_time = dt_util.now().strftime("%d-%m-%Y %H:%M")
try:
self._trips = self._nsapi.get_trips(
trip_time, self._departure, self._via, self._heading, True, 0, 2
)
if self._trips:
- if self._trips[0].departure_time_actual is None:
- planned_time = self._trips[0].departure_time_planned
- self._state = planned_time.strftime("%H:%M")
+ all_times = []
+
+ # If a train is delayed we can observe this through departure_time_actual.
+ for trip in self._trips:
+ if trip.departure_time_actual is None:
+ all_times.append(trip.departure_time_planned)
+ else:
+ all_times.append(trip.departure_time_actual)
+
+ # Remove all trains that already left.
+ filtered_times = [
+ (i, time)
+ for i, time in enumerate(all_times)
+ if time > dt_util.now()
+ ]
+
+ if len(filtered_times) > 0:
+ sorted_times = sorted(filtered_times, key=lambda x: x[1])
+ self._first_trip = self._trips[sorted_times[0][0]]
+ self._state = sorted_times[0][1].strftime("%H:%M")
+
+ # Filter again to remove trains that leave at the exact same time.
+ filtered_times = [
+ (i, time)
+ for i, time in enumerate(all_times)
+ if time > sorted_times[0][1]
+ ]
+
+ if len(filtered_times) > 0:
+ sorted_times = sorted(filtered_times, key=lambda x: x[1])
+ self._next_trip = self._trips[sorted_times[0][0]]
+ else:
+ self._next_trip = None
+
else:
- actual_time = self._trips[0].departure_time_actual
- self._state = actual_time.strftime("%H:%M")
+ self._first_trip = None
+ self._state = None
+
except (
requests.exceptions.ConnectionError,
requests.exceptions.HTTPError,
diff --git a/homeassistant/components/ness_alarm/services.yaml b/homeassistant/components/ness_alarm/services.yaml
index b02d5e36805..aed1e1836bd 100644
--- a/homeassistant/components/ness_alarm/services.yaml
+++ b/homeassistant/components/ness_alarm/services.yaml
@@ -7,7 +7,7 @@ aux:
selector:
number:
min: 1
- max: 4
+ max: 8
state:
default: true
selector:
diff --git a/homeassistant/components/ness_alarm/strings.json b/homeassistant/components/ness_alarm/strings.json
index ec4e39a6128..f4490ac98db 100644
--- a/homeassistant/components/ness_alarm/strings.json
+++ b/homeassistant/components/ness_alarm/strings.json
@@ -2,7 +2,7 @@
"services": {
"aux": {
"name": "Aux",
- "description": "Trigger an aux output.",
+ "description": "Changes the state of an aux output.",
"fields": {
"output_id": {
"name": "Output ID",
@@ -10,17 +10,17 @@
},
"state": {
"name": "State",
- "description": "The On/Off State. If P14xE 8E is enabled then a value of true will pulse output x for the time specified in P14(x+4)E."
+ "description": "The on/off state of the output. If P14xE 8E is enabled then turning on will pulse the output for the time specified in P14(x+4)E."
}
}
},
"panic": {
"name": "Panic",
- "description": "Triggers a panic.",
+ "description": "Triggers a panic alarm.",
"fields": {
"code": {
"name": "Code",
- "description": "The user code to use to trigger the panic."
+ "description": "The user code to use to trigger the panic alarm."
}
}
}
diff --git a/homeassistant/components/nest/__init__.py b/homeassistant/components/nest/__init__.py
index 8adc0e4f714..af85f1fc5ae 100644
--- a/homeassistant/components/nest/__init__.py
+++ b/homeassistant/components/nest/__init__.py
@@ -8,7 +8,7 @@ from collections.abc import Awaitable, Callable
from http import HTTPStatus
import logging
-from aiohttp import web
+from aiohttp import ClientError, ClientResponseError, web
from google_nest_sdm.camera_traits import CameraClipPreviewTrait
from google_nest_sdm.device import Device
from google_nest_sdm.event import EventMessage
@@ -198,7 +198,17 @@ async def async_setup_entry(hass: HomeAssistant, entry: NestConfigEntry) -> bool
entry, unique_id=entry.data[CONF_PROJECT_ID]
)
- subscriber = await api.new_subscriber(hass, entry)
+ auth = await api.new_auth(hass, entry)
+ try:
+ await auth.async_get_access_token()
+ except ClientResponseError as err:
+ if 400 <= err.status < 500:
+ raise ConfigEntryAuthFailed from err
+ raise ConfigEntryNotReady from err
+ except ClientError as err:
+ raise ConfigEntryNotReady from err
+
+ subscriber = await api.new_subscriber(hass, entry, auth)
if not subscriber:
return False
# Keep media for last N events in memory
diff --git a/homeassistant/components/nest/api.py b/homeassistant/components/nest/api.py
index e86e326b1c2..d55826f7ed0 100644
--- a/homeassistant/components/nest/api.py
+++ b/homeassistant/components/nest/api.py
@@ -50,13 +50,14 @@ class AsyncConfigEntryAuth(AbstractAuth):
return cast(str, self._oauth_session.token["access_token"])
async def async_get_creds(self) -> Credentials:
- """Return an OAuth credential for Pub/Sub Subscriber."""
- # We don't have a way for Home Assistant to refresh creds on behalf
- # of the google pub/sub subscriber. Instead, build a full
- # Credentials object with enough information for the subscriber to
- # handle this on its own. We purposely don't refresh the token here
- # even when it is expired to fully hand off this responsibility and
- # know it is working at startup (then if not, fail loudly).
+ """Return an OAuth credential for Pub/Sub Subscriber.
+
+ The subscriber will call this when connecting to the stream to refresh
+ the token. We construct a credentials object using the underlying
+ OAuth2Session since the subscriber may expect the expiry fields to
+ be present.
+ """
+ await self.async_get_access_token()
token = self._oauth_session.token
creds = Credentials( # type: ignore[no-untyped-call]
token=token["access_token"],
@@ -101,9 +102,7 @@ class AccessTokenAuthImpl(AbstractAuth):
)
-async def new_subscriber(
- hass: HomeAssistant, entry: NestConfigEntry
-) -> GoogleNestSubscriber | None:
+async def new_auth(hass: HomeAssistant, entry: NestConfigEntry) -> AbstractAuth:
"""Create a GoogleNestSubscriber."""
implementation = (
await config_entry_oauth2_flow.async_get_config_entry_implementation(
@@ -114,14 +113,22 @@ async def new_subscriber(
implementation, config_entry_oauth2_flow.LocalOAuth2Implementation
):
raise TypeError(f"Unexpected auth implementation {implementation}")
- if (subscription_name := entry.data.get(CONF_SUBSCRIPTION_NAME)) is None:
- subscription_name = entry.data[CONF_SUBSCRIBER_ID]
- auth = AsyncConfigEntryAuth(
+ return AsyncConfigEntryAuth(
aiohttp_client.async_get_clientsession(hass),
config_entry_oauth2_flow.OAuth2Session(hass, entry, implementation),
implementation.client_id,
implementation.client_secret,
)
+
+
+async def new_subscriber(
+ hass: HomeAssistant,
+ entry: NestConfigEntry,
+ auth: AbstractAuth,
+) -> GoogleNestSubscriber:
+ """Create a GoogleNestSubscriber."""
+ if (subscription_name := entry.data.get(CONF_SUBSCRIPTION_NAME)) is None:
+ subscription_name = entry.data[CONF_SUBSCRIBER_ID]
return GoogleNestSubscriber(auth, entry.data[CONF_PROJECT_ID], subscription_name)
diff --git a/homeassistant/components/nest/camera.py b/homeassistant/components/nest/camera.py
index df02f17444f..f5985da9ff8 100644
--- a/homeassistant/components/nest/camera.py
+++ b/homeassistant/components/nest/camera.py
@@ -30,7 +30,7 @@ from homeassistant.components.camera import (
from homeassistant.components.stream import CONF_EXTRA_PART_WAIT_TIME
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_point_in_utc_time
from homeassistant.util.dt import utcnow
@@ -51,7 +51,9 @@ BACKOFF_MULTIPLIER = 1.5
async def async_setup_entry(
- hass: HomeAssistant, entry: NestConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NestConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the cameras."""
diff --git a/homeassistant/components/nest/climate.py b/homeassistant/components/nest/climate.py
index 3193d592120..f5eff664f83 100644
--- a/homeassistant/components/nest/climate.py
+++ b/homeassistant/components/nest/climate.py
@@ -30,7 +30,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .device_info import NestDeviceInfo
from .types import NestConfigEntry
@@ -76,7 +76,9 @@ MIN_TEMP_RANGE = 1.66667
async def async_setup_entry(
- hass: HomeAssistant, entry: NestConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NestConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the client entities."""
diff --git a/homeassistant/components/nest/device_info.py b/homeassistant/components/nest/device_info.py
index facd429b139..8241b8aa5f8 100644
--- a/homeassistant/components/nest/device_info.py
+++ b/homeassistant/components/nest/device_info.py
@@ -7,7 +7,6 @@ from collections.abc import Mapping
from google_nest_sdm.device import Device
from google_nest_sdm.device_traits import ConnectivityTrait, InfoTrait
-from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
@@ -84,8 +83,7 @@ def async_nest_devices(hass: HomeAssistant) -> Mapping[str, Device]:
"""Return a mapping of all nest devices for all config entries."""
return {
device.name: device
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.state == ConfigEntryState.LOADED
+ for config_entry in hass.config_entries.async_loaded_entries(DOMAIN)
for device in config_entry.runtime_data.device_manager.devices.values()
}
diff --git a/homeassistant/components/nest/event.py b/homeassistant/components/nest/event.py
index 1a2c0317496..9bb041fce6c 100644
--- a/homeassistant/components/nest/event.py
+++ b/homeassistant/components/nest/event.py
@@ -13,7 +13,7 @@ from homeassistant.components.event import (
EventEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .device_info import NestDeviceInfo
from .events import (
@@ -66,7 +66,9 @@ ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
- hass: HomeAssistant, entry: NestConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NestConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
async_add_entities(
diff --git a/homeassistant/components/nest/manifest.json b/homeassistant/components/nest/manifest.json
index a0d8bc06640..d9383533300 100644
--- a/homeassistant/components/nest/manifest.json
+++ b/homeassistant/components/nest/manifest.json
@@ -19,5 +19,5 @@
"documentation": "https://www.home-assistant.io/integrations/nest",
"iot_class": "cloud_push",
"loggers": ["google_nest_sdm"],
- "requirements": ["google-nest-sdm==7.1.3"]
+ "requirements": ["google-nest-sdm==7.1.4"]
}
diff --git a/homeassistant/components/nest/sensor.py b/homeassistant/components/nest/sensor.py
index 02a0e305813..a6fda48fe87 100644
--- a/homeassistant/components/nest/sensor.py
+++ b/homeassistant/components/nest/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .device_info import NestDeviceInfo
from .types import NestConfigEntry
@@ -31,7 +31,9 @@ DEVICE_TYPE_MAP = {
async def async_setup_entry(
- hass: HomeAssistant, entry: NestConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NestConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
diff --git a/homeassistant/components/nest/strings.json b/homeassistant/components/nest/strings.json
index 23da524ab7e..54f543aa845 100644
--- a/homeassistant/components/nest/strings.json
+++ b/homeassistant/components/nest/strings.json
@@ -58,6 +58,7 @@
},
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_account%]",
+ "already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]",
"missing_configuration": "[%key:common::config_flow::abort::oauth2_missing_configuration%]",
"missing_credentials": "[%key:common::config_flow::abort::oauth2_missing_credentials%]",
"authorize_url_timeout": "[%key:common::config_flow::abort::oauth2_authorize_url_timeout%]",
diff --git a/homeassistant/components/netatmo/binary_sensor.py b/homeassistant/components/netatmo/binary_sensor.py
index c478525753a..d35bfa7e8a6 100644
--- a/homeassistant/components/netatmo/binary_sensor.py
+++ b/homeassistant/components/netatmo/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import NETATMO_CREATE_WEATHER_SENSOR
from .data_handler import NetatmoDevice
@@ -23,7 +23,9 @@ BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Netatmo binary sensors based on a config entry."""
diff --git a/homeassistant/components/netatmo/button.py b/homeassistant/components/netatmo/button.py
index 7b2899c84aa..e77b5188067 100644
--- a/homeassistant/components/netatmo/button.py
+++ b/homeassistant/components/netatmo/button.py
@@ -10,7 +10,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_URL_CONTROL, NETATMO_CREATE_BUTTON
from .data_handler import HOME, SIGNAL_NAME, NetatmoDevice
@@ -22,7 +22,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo button platform."""
diff --git a/homeassistant/components/netatmo/camera.py b/homeassistant/components/netatmo/camera.py
index 3bd7bcd859d..f21998bbac8 100644
--- a/homeassistant/components/netatmo/camera.py
+++ b/homeassistant/components/netatmo/camera.py
@@ -16,7 +16,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_CAMERA_LIGHT_MODE,
@@ -48,7 +48,9 @@ DEFAULT_QUALITY = "high"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo camera platform."""
diff --git a/homeassistant/components/netatmo/climate.py b/homeassistant/components/netatmo/climate.py
index 02c955beac3..2e3d8c6bcb8 100644
--- a/homeassistant/components/netatmo/climate.py
+++ b/homeassistant/components/netatmo/climate.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import (
@@ -118,7 +118,9 @@ NA_VALVE = DeviceType.NRV
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo energy platform."""
diff --git a/homeassistant/components/netatmo/config_flow.py b/homeassistant/components/netatmo/config_flow.py
index d853694ffea..02d9c2fa3a6 100644
--- a/homeassistant/components/netatmo/config_flow.py
+++ b/homeassistant/components/netatmo/config_flow.py
@@ -135,7 +135,7 @@ class NetatmoOptionsFlowHandler(OptionsFlow):
vol.Optional(
CONF_WEATHER_AREAS,
default=weather_areas,
- ): cv.multi_select({wa: None for wa in weather_areas}),
+ ): cv.multi_select(dict.fromkeys(weather_areas)),
vol.Optional(CONF_NEW_AREA): str,
}
)
diff --git a/homeassistant/components/netatmo/cover.py b/homeassistant/components/netatmo/cover.py
index c34b3a1b47b..a599aacd719 100644
--- a/homeassistant/components/netatmo/cover.py
+++ b/homeassistant/components/netatmo/cover.py
@@ -16,7 +16,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_URL_CONTROL, NETATMO_CREATE_COVER
from .data_handler import HOME, SIGNAL_NAME, NetatmoDevice
@@ -28,7 +28,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo cover platform."""
diff --git a/homeassistant/components/netatmo/fan.py b/homeassistant/components/netatmo/fan.py
index 9f3fe7174ff..b0dc74c2b58 100644
--- a/homeassistant/components/netatmo/fan.py
+++ b/homeassistant/components/netatmo/fan.py
@@ -11,7 +11,7 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_URL_CONTROL, NETATMO_CREATE_FAN
from .data_handler import HOME, SIGNAL_NAME, NetatmoDevice
@@ -28,7 +28,7 @@ PRESETS = {v: k for k, v in PRESET_MAPPING.items()}
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo fan platform."""
diff --git a/homeassistant/components/netatmo/light.py b/homeassistant/components/netatmo/light.py
index fe30dc0eaa4..ce28c455dea 100644
--- a/homeassistant/components/netatmo/light.py
+++ b/homeassistant/components/netatmo/light.py
@@ -11,7 +11,7 @@ from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEnti
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_URL_CONTROL,
@@ -30,7 +30,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo camera light platform."""
diff --git a/homeassistant/components/netatmo/select.py b/homeassistant/components/netatmo/select.py
index 92568b73e80..e8637c90584 100644
--- a/homeassistant/components/netatmo/select.py
+++ b/homeassistant/components/netatmo/select.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_URL_ENERGY,
@@ -26,7 +26,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo energy platform schedule selector."""
diff --git a/homeassistant/components/netatmo/sensor.py b/homeassistant/components/netatmo/sensor.py
index cc233dcc0ce..5f8084d542c 100644
--- a/homeassistant/components/netatmo/sensor.py
+++ b/homeassistant/components/netatmo/sensor.py
@@ -38,7 +38,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -385,7 +385,9 @@ BATTERY_SENSOR_DESCRIPTION = NetatmoSensorEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo sensor platform."""
diff --git a/homeassistant/components/netatmo/services.yaml b/homeassistant/components/netatmo/services.yaml
index cab0528199d..c130d8e96e3 100644
--- a/homeassistant/components/netatmo/services.yaml
+++ b/homeassistant/components/netatmo/services.yaml
@@ -39,7 +39,7 @@ set_preset_mode_with_end_datetime:
select:
options:
- "away"
- - "Frost Guard"
+ - "frost_guard"
end_datetime:
required: true
example: '"2019-04-20 05:04:20"'
diff --git a/homeassistant/components/netatmo/switch.py b/homeassistant/components/netatmo/switch.py
index 6ba4628a358..9ee37c11528 100644
--- a/homeassistant/components/netatmo/switch.py
+++ b/homeassistant/components/netatmo/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_URL_CONTROL, NETATMO_CREATE_SWITCH
from .data_handler import HOME, SIGNAL_NAME, NetatmoDevice
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netatmo switch platform."""
diff --git a/homeassistant/components/netgear/button.py b/homeassistant/components/netgear/button.py
index e5b9ec209c7..726c1b2296d 100644
--- a/homeassistant/components/netgear/button.py
+++ b/homeassistant/components/netgear/button.py
@@ -12,7 +12,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN, KEY_COORDINATOR, KEY_ROUTER
@@ -38,7 +38,9 @@ BUTTONS = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button for Netgear component."""
router = hass.data[DOMAIN][entry.entry_id][KEY_ROUTER]
diff --git a/homeassistant/components/netgear/const.py b/homeassistant/components/netgear/const.py
index f7a683326d3..c8ecd8e7e1d 100644
--- a/homeassistant/components/netgear/const.py
+++ b/homeassistant/components/netgear/const.py
@@ -62,6 +62,7 @@ MODELS_V2 = [
"RBR",
"RBS",
"RBW",
+ "RS",
"LBK",
"LBR",
"CBK",
diff --git a/homeassistant/components/netgear/device_tracker.py b/homeassistant/components/netgear/device_tracker.py
index b17430d2abb..56f4ecac14f 100644
--- a/homeassistant/components/netgear/device_tracker.py
+++ b/homeassistant/components/netgear/device_tracker.py
@@ -7,7 +7,7 @@ import logging
from homeassistant.components.device_tracker import ScannerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DEVICE_ICONS, DOMAIN, KEY_COORDINATOR, KEY_ROUTER
@@ -18,7 +18,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Netgear component."""
router = hass.data[DOMAIN][entry.entry_id][KEY_ROUTER]
diff --git a/homeassistant/components/netgear/sensor.py b/homeassistant/components/netgear/sensor.py
index d807f7aed0a..521e18098eb 100644
--- a/homeassistant/components/netgear/sensor.py
+++ b/homeassistant/components/netgear/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -274,7 +274,9 @@ SENSOR_LINK_TYPES = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Netgear component."""
router = hass.data[DOMAIN][entry.entry_id][KEY_ROUTER]
diff --git a/homeassistant/components/netgear/switch.py b/homeassistant/components/netgear/switch.py
index 85f214d784a..dd8468df099 100644
--- a/homeassistant/components/netgear/switch.py
+++ b/homeassistant/components/netgear/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN, KEY_COORDINATOR, KEY_ROUTER
@@ -99,7 +99,9 @@ ROUTER_SWITCH_TYPES = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches for Netgear component."""
router = hass.data[DOMAIN][entry.entry_id][KEY_ROUTER]
diff --git a/homeassistant/components/netgear/update.py b/homeassistant/components/netgear/update.py
index 1fbfee3d892..388ad8bff4f 100644
--- a/homeassistant/components/netgear/update.py
+++ b/homeassistant/components/netgear/update.py
@@ -12,7 +12,7 @@ from homeassistant.components.update import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN, KEY_COORDINATOR_FIRMWARE, KEY_ROUTER
@@ -23,7 +23,9 @@ LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up update entities for Netgear component."""
router = hass.data[DOMAIN][entry.entry_id][KEY_ROUTER]
diff --git a/homeassistant/components/netgear_lte/__init__.py b/homeassistant/components/netgear_lte/__init__.py
index 1846d1f7992..47a39a39be0 100644
--- a/homeassistant/components/netgear_lte/__init__.py
+++ b/homeassistant/components/netgear_lte/__init__.py
@@ -6,7 +6,6 @@ from aiohttp.cookiejar import CookieJar
import eternalegypt
from eternalegypt.eternalegypt import SMS
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@@ -23,7 +22,7 @@ from .const import (
DATA_SESSION,
DOMAIN,
)
-from .coordinator import NetgearLTEDataUpdateCoordinator
+from .coordinator import NetgearLTEConfigEntry, NetgearLTEDataUpdateCoordinator
from .services import async_setup_services
EVENT_SMS = "netgear_lte_sms"
@@ -55,7 +54,6 @@ PLATFORMS = [
Platform.NOTIFY,
Platform.SENSOR,
]
-type NetgearLTEConfigEntry = ConfigEntry[NetgearLTEDataUpdateCoordinator]
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
@@ -94,7 +92,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NetgearLTEConfigEntry) -
await modem.add_sms_listener(fire_sms_event)
- coordinator = NetgearLTEDataUpdateCoordinator(hass, modem)
+ coordinator = NetgearLTEDataUpdateCoordinator(hass, entry, modem)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
@@ -118,12 +116,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NetgearLTEConfigEntry) -
async def async_unload_entry(hass: HomeAssistant, entry: NetgearLTEConfigEntry) -> bool:
"""Unload a config entry."""
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
hass.data.pop(DOMAIN, None)
for service_name in hass.services.async_services()[DOMAIN]:
hass.services.async_remove(DOMAIN, service_name)
diff --git a/homeassistant/components/netgear_lte/binary_sensor.py b/homeassistant/components/netgear_lte/binary_sensor.py
index 280d240b90f..890bcb37443 100644
--- a/homeassistant/components/netgear_lte/binary_sensor.py
+++ b/homeassistant/components/netgear_lte/binary_sensor.py
@@ -9,9 +9,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NetgearLTEConfigEntry
+from .coordinator import NetgearLTEConfigEntry
from .entity import LTEEntity
BINARY_SENSORS: tuple[BinarySensorEntityDescription, ...] = (
@@ -39,7 +39,7 @@ BINARY_SENSORS: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: NetgearLTEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netgear LTE binary sensor."""
async_add_entities(
diff --git a/homeassistant/components/netgear_lte/coordinator.py b/homeassistant/components/netgear_lte/coordinator.py
index afd0cb743bf..7bcefca6403 100644
--- a/homeassistant/components/netgear_lte/coordinator.py
+++ b/homeassistant/components/netgear_lte/coordinator.py
@@ -3,17 +3,16 @@
from __future__ import annotations
from datetime import timedelta
-from typing import TYPE_CHECKING
from eternalegypt.eternalegypt import Error, Information, Modem
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN, LOGGER
-if TYPE_CHECKING:
- from . import NetgearLTEConfigEntry
+type NetgearLTEConfigEntry = ConfigEntry[NetgearLTEDataUpdateCoordinator]
class NetgearLTEDataUpdateCoordinator(DataUpdateCoordinator[Information]):
@@ -24,12 +23,14 @@ class NetgearLTEDataUpdateCoordinator(DataUpdateCoordinator[Information]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: NetgearLTEConfigEntry,
modem: Modem,
) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=10),
)
diff --git a/homeassistant/components/netgear_lte/entity.py b/homeassistant/components/netgear_lte/entity.py
index 3353da6dc77..9d56605b7d2 100644
--- a/homeassistant/components/netgear_lte/entity.py
+++ b/homeassistant/components/netgear_lte/entity.py
@@ -5,9 +5,8 @@ from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import NetgearLTEConfigEntry
from .const import DOMAIN, MANUFACTURER
-from .coordinator import NetgearLTEDataUpdateCoordinator
+from .coordinator import NetgearLTEConfigEntry, NetgearLTEDataUpdateCoordinator
class LTEEntity(CoordinatorEntity[NetgearLTEDataUpdateCoordinator]):
diff --git a/homeassistant/components/netgear_lte/sensor.py b/homeassistant/components/netgear_lte/sensor.py
index 73e5de7eaeb..49301267d9d 100644
--- a/homeassistant/components/netgear_lte/sensor.py
+++ b/homeassistant/components/netgear_lte/sensor.py
@@ -19,10 +19,10 @@ from homeassistant.const import (
UnitOfInformation,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import NetgearLTEConfigEntry
+from .coordinator import NetgearLTEConfigEntry
from .entity import LTEEntity
@@ -127,7 +127,7 @@ SENSORS: tuple[NetgearLTESensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: NetgearLTEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Netgear LTE sensor."""
async_add_entities(NetgearLTESensor(entry, description) for description in SENSORS)
diff --git a/homeassistant/components/network/__init__.py b/homeassistant/components/network/__init__.py
index 10046f75127..200cce86997 100644
--- a/homeassistant/components/network/__init__.py
+++ b/homeassistant/components/network/__init__.py
@@ -20,7 +20,7 @@ from .const import (
PUBLIC_TARGET_IP,
)
from .models import Adapter
-from .network import Network, async_get_network
+from .network import Network, async_get_loaded_network, async_get_network
_LOGGER = logging.getLogger(__name__)
@@ -34,6 +34,12 @@ async def async_get_adapters(hass: HomeAssistant) -> list[Adapter]:
return network.adapters
+@callback
+def async_get_loaded_adapters(hass: HomeAssistant) -> list[Adapter]:
+ """Get the network adapter configuration."""
+ return async_get_loaded_network(hass).adapters
+
+
@bind_hass
async def async_get_source_ip(
hass: HomeAssistant, target_ip: str | UndefinedType = UNDEFINED
@@ -74,7 +80,14 @@ async def async_get_enabled_source_ips(
hass: HomeAssistant,
) -> list[IPv4Address | IPv6Address]:
"""Build the list of enabled source ips."""
- adapters = await async_get_adapters(hass)
+ return async_get_enabled_source_ips_from_adapters(await async_get_adapters(hass))
+
+
+@callback
+def async_get_enabled_source_ips_from_adapters(
+ adapters: list[Adapter],
+) -> list[IPv4Address | IPv6Address]:
+ """Build the list of enabled source ips."""
sources: list[IPv4Address | IPv6Address] = []
for adapter in adapters:
if not adapter["enabled"]:
@@ -151,5 +164,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
async_register_websocket_commands,
)
+ await async_get_network(hass)
+
async_register_websocket_commands(hass)
return True
diff --git a/homeassistant/components/network/const.py b/homeassistant/components/network/const.py
index 120ae9dfd7c..d8c8858be72 100644
--- a/homeassistant/components/network/const.py
+++ b/homeassistant/components/network/const.py
@@ -12,8 +12,6 @@ DOMAIN: Final = "network"
STORAGE_KEY: Final = "core.network"
STORAGE_VERSION: Final = 1
-DATA_NETWORK: Final = "network"
-
ATTR_ADAPTERS: Final = "adapters"
ATTR_CONFIGURED_ADAPTERS: Final = "configured_adapters"
DEFAULT_CONFIGURED_ADAPTERS: list[str] = []
diff --git a/homeassistant/components/network/network.py b/homeassistant/components/network/network.py
index 4158307bb1a..db25bedcaea 100644
--- a/homeassistant/components/network/network.py
+++ b/homeassistant/components/network/network.py
@@ -9,11 +9,12 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.singleton import singleton
from homeassistant.helpers.storage import Store
from homeassistant.util.async_ import create_eager_task
+from homeassistant.util.hass_dict import HassKey
from .const import (
ATTR_CONFIGURED_ADAPTERS,
- DATA_NETWORK,
DEFAULT_CONFIGURED_ADAPTERS,
+ DOMAIN,
STORAGE_KEY,
STORAGE_VERSION,
)
@@ -22,8 +23,16 @@ from .util import async_load_adapters, enable_adapters, enable_auto_detected_ada
_LOGGER = logging.getLogger(__name__)
+DATA_NETWORK: HassKey[Network] = HassKey(DOMAIN)
-@singleton(DATA_NETWORK)
+
+@callback
+def async_get_loaded_network(hass: HomeAssistant) -> Network:
+ """Get network singleton."""
+ return hass.data[DATA_NETWORK]
+
+
+@singleton(DOMAIN)
async def async_get_network(hass: HomeAssistant) -> Network:
"""Get network singleton."""
network = Network(hass)
diff --git a/homeassistant/components/nexia/__init__.py b/homeassistant/components/nexia/__init__.py
index 66a8ec5bdb8..8d0d509f8a2 100644
--- a/homeassistant/components/nexia/__init__.py
+++ b/homeassistant/components/nexia/__init__.py
@@ -58,7 +58,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NexiaConfigEntry) -> boo
f"Error connecting to Nexia service: {os_error}"
) from os_error
- coordinator = NexiaDataUpdateCoordinator(hass, nexia_home)
+ coordinator = NexiaDataUpdateCoordinator(hass, entry, nexia_home)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/nexia/binary_sensor.py b/homeassistant/components/nexia/binary_sensor.py
index 204d84ed975..224836c81e6 100644
--- a/homeassistant/components/nexia/binary_sensor.py
+++ b/homeassistant/components/nexia/binary_sensor.py
@@ -2,7 +2,7 @@
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import NexiaThermostatEntity
from .types import NexiaConfigEntry
@@ -11,7 +11,7 @@ from .types import NexiaConfigEntry
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NexiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for a Nexia device."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/nexia/climate.py b/homeassistant/components/nexia/climate.py
index 81e7800fd01..e9de81cca7c 100644
--- a/homeassistant/components/nexia/climate.py
+++ b/homeassistant/components/nexia/climate.py
@@ -33,7 +33,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import VolDictType
@@ -116,7 +116,7 @@ NEXIA_SUPPORTED = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NexiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate for a Nexia device."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/nexia/coordinator.py b/homeassistant/components/nexia/coordinator.py
index 894c491c45b..85e784218f4 100644
--- a/homeassistant/components/nexia/coordinator.py
+++ b/homeassistant/components/nexia/coordinator.py
@@ -8,6 +8,7 @@ from typing import Any
from nexia.home import NexiaHome
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -19,9 +20,12 @@ DEFAULT_UPDATE_RATE = 120
class NexiaDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""DataUpdateCoordinator for nexia homes."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
nexia_home: NexiaHome,
) -> None:
"""Initialize DataUpdateCoordinator for the nexia home."""
@@ -29,6 +33,7 @@ class NexiaDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Nexia update",
update_interval=timedelta(seconds=DEFAULT_UPDATE_RATE),
always_update=False,
diff --git a/homeassistant/components/nexia/manifest.json b/homeassistant/components/nexia/manifest.json
index 6a439f869c9..e7ab63d4712 100644
--- a/homeassistant/components/nexia/manifest.json
+++ b/homeassistant/components/nexia/manifest.json
@@ -12,5 +12,5 @@
"documentation": "https://www.home-assistant.io/integrations/nexia",
"iot_class": "cloud_polling",
"loggers": ["nexia"],
- "requirements": ["nexia==2.0.9"]
+ "requirements": ["nexia==2.4.0"]
}
diff --git a/homeassistant/components/nexia/number.py b/homeassistant/components/nexia/number.py
index 46cc4d094a3..05d9e5b4614 100644
--- a/homeassistant/components/nexia/number.py
+++ b/homeassistant/components/nexia/number.py
@@ -7,7 +7,7 @@ from nexia.thermostat import NexiaThermostat
from homeassistant.components.number import NumberEntity
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import NexiaDataUpdateCoordinator
from .entity import NexiaThermostatEntity
@@ -18,7 +18,7 @@ from .util import percent_conv
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NexiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for a Nexia device."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/nexia/scene.py b/homeassistant/components/nexia/scene.py
index 60078fab822..fe75eb07e02 100644
--- a/homeassistant/components/nexia/scene.py
+++ b/homeassistant/components/nexia/scene.py
@@ -6,7 +6,7 @@ from nexia.automation import NexiaAutomation
from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from .const import ATTR_DESCRIPTION
@@ -20,7 +20,7 @@ SCENE_ACTIVATION_TIME = 5
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NexiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up automations for a Nexia device."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/nexia/sensor.py b/homeassistant/components/nexia/sensor.py
index e50bd750c2f..293a9308cb4 100644
--- a/homeassistant/components/nexia/sensor.py
+++ b/homeassistant/components/nexia/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import NexiaThermostatEntity, NexiaThermostatZoneEntity
from .types import NexiaConfigEntry
@@ -22,7 +22,7 @@ from .util import percent_conv
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NexiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for a Nexia device."""
diff --git a/homeassistant/components/nexia/strings.json b/homeassistant/components/nexia/strings.json
index d88ce0b898d..43da2cf05c7 100644
--- a/homeassistant/components/nexia/strings.json
+++ b/homeassistant/components/nexia/strings.json
@@ -58,6 +58,9 @@
"switch": {
"hold": {
"name": "Hold"
+ },
+ "emergency_heat": {
+ "name": "Emergency heat"
}
}
},
@@ -83,8 +86,8 @@
}
},
"set_hvac_run_mode": {
- "name": "Set hvac run mode",
- "description": "Sets the HVAC operation mode.",
+ "name": "Set HVAC run mode",
+ "description": "Sets the run and/or operation mode of the HVAC system.",
"fields": {
"run_mode": {
"name": "Run mode",
diff --git a/homeassistant/components/nexia/switch.py b/homeassistant/components/nexia/switch.py
index 9505538e86a..1897ad67414 100644
--- a/homeassistant/components/nexia/switch.py
+++ b/homeassistant/components/nexia/switch.py
@@ -10,7 +10,7 @@ from nexia.zone import NexiaThermostatZone
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import NexiaDataUpdateCoordinator
from .entity import NexiaThermostatEntity, NexiaThermostatZoneEntity
@@ -20,7 +20,7 @@ from .types import NexiaConfigEntry
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NexiaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches for a Nexia device."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/nextbus/sensor.py b/homeassistant/components/nextbus/sensor.py
index 554814fe2db..2e184e13fc7 100644
--- a/homeassistant/components/nextbus/sensor.py
+++ b/homeassistant/components/nextbus/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, CONF_STOP
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.dt import utc_from_timestamp
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load values from configuration and initialize the platform."""
_LOGGER.debug(config.data)
diff --git a/homeassistant/components/nextcloud/binary_sensor.py b/homeassistant/components/nextcloud/binary_sensor.py
index 10e1a000a68..f51796e6c7f 100644
--- a/homeassistant/components/nextcloud/binary_sensor.py
+++ b/homeassistant/components/nextcloud/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import NextcloudConfigEntry
from .entity import NextcloudEntity
@@ -54,7 +54,7 @@ BINARY_SENSORS: Final[list[BinarySensorEntityDescription]] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: NextcloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nextcloud binary sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/nextcloud/sensor.py b/homeassistant/components/nextcloud/sensor.py
index a6722821012..63b31f0edde 100644
--- a/homeassistant/components/nextcloud/sensor.py
+++ b/homeassistant/components/nextcloud/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utc_from_timestamp
from .coordinator import NextcloudConfigEntry
@@ -602,7 +602,7 @@ SENSORS: Final[list[NextcloudSensorEntityDescription]] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: NextcloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nextcloud sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/nextcloud/strings.json b/homeassistant/components/nextcloud/strings.json
index f9f7e4c2294..ef4e3de0f62 100644
--- a/homeassistant/components/nextcloud/strings.json
+++ b/homeassistant/components/nextcloud/strings.json
@@ -21,7 +21,7 @@
},
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]",
- "connection_error_during_import": "Connection error occured during yaml configuration import",
+ "connection_error_during_import": "Connection error occurred during yaml configuration import",
"reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]"
},
"error": {
@@ -70,7 +70,7 @@
"name": "Cache memory"
},
"nextcloud_cache_num_entries": {
- "name": "Cache number of entires"
+ "name": "Cache number of entries"
},
"nextcloud_cache_num_hits": {
"name": "Cache number of hits"
@@ -88,7 +88,7 @@
"name": "Cache start time"
},
"nextcloud_cache_ttl": {
- "name": "Cache ttl"
+ "name": "Cache TTL"
},
"nextcloud_database_size": {
"name": "Database size"
@@ -268,13 +268,13 @@
"name": "Updates available"
},
"nextcloud_system_cpuload_1": {
- "name": "CPU Load last 1 minute"
+ "name": "CPU load last 1 minute"
},
"nextcloud_system_cpuload_15": {
- "name": "CPU Load last 15 minutes"
+ "name": "CPU load last 15 minutes"
},
"nextcloud_system_cpuload_5": {
- "name": "CPU Load last 5 minutes"
+ "name": "CPU load last 5 minutes"
},
"nextcloud_system_freespace": {
"name": "Free space"
diff --git a/homeassistant/components/nextcloud/update.py b/homeassistant/components/nextcloud/update.py
index aad6412b7b3..b991b001117 100644
--- a/homeassistant/components/nextcloud/update.py
+++ b/homeassistant/components/nextcloud/update.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.components.update import UpdateEntity, UpdateEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import NextcloudConfigEntry
from .entity import NextcloudEntity
@@ -13,7 +13,7 @@ from .entity import NextcloudEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: NextcloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nextcloud update entity."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/nextdns/__init__.py b/homeassistant/components/nextdns/__init__.py
index 7f0729bca1e..eb8bd26cb9b 100644
--- a/homeassistant/components/nextdns/__init__.py
+++ b/homeassistant/components/nextdns/__init__.py
@@ -36,6 +36,7 @@ from .const import (
ATTR_SETTINGS,
ATTR_STATUS,
CONF_PROFILE_ID,
+ DOMAIN,
UPDATE_INTERVAL_ANALYTICS,
UPDATE_INTERVAL_CONNECTION,
UPDATE_INTERVAL_SETTINGS,
@@ -88,9 +89,20 @@ async def async_setup_entry(hass: HomeAssistant, entry: NextDnsConfigEntry) -> b
try:
nextdns = await NextDns.create(websession, api_key)
except (ApiError, ClientConnectorError, RetryError, TimeoutError) as err:
- raise ConfigEntryNotReady from err
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="cannot_connect",
+ translation_placeholders={
+ "entry": entry.title,
+ "error": repr(err),
+ },
+ ) from err
except InvalidApiKeyError as err:
- raise ConfigEntryAuthFailed from err
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN,
+ translation_key="auth_error",
+ translation_placeholders={"entry": entry.title},
+ ) from err
tasks = []
coordinators = {}
@@ -98,7 +110,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: NextDnsConfigEntry) -> b
# Independent DataUpdateCoordinator is used for each API endpoint to avoid
# unnecessary requests when entities using this endpoint are disabled.
for coordinator_name, coordinator_class, update_interval in COORDINATORS:
- coordinator = coordinator_class(hass, nextdns, profile_id, update_interval)
+ coordinator = coordinator_class(
+ hass, entry, nextdns, profile_id, update_interval
+ )
tasks.append(coordinator.async_config_entry_first_refresh())
coordinators[coordinator_name] = coordinator
diff --git a/homeassistant/components/nextdns/binary_sensor.py b/homeassistant/components/nextdns/binary_sensor.py
index 08a1f89418f..ed244146efc 100644
--- a/homeassistant/components/nextdns/binary_sensor.py
+++ b/homeassistant/components/nextdns/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import NextDnsConfigEntry
@@ -51,7 +51,7 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: NextDnsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add NextDNS entities from a config_entry."""
coordinator = entry.runtime_data.connection
diff --git a/homeassistant/components/nextdns/button.py b/homeassistant/components/nextdns/button.py
index 164d725b393..2adccaa304f 100644
--- a/homeassistant/components/nextdns/button.py
+++ b/homeassistant/components/nextdns/button.py
@@ -2,15 +2,19 @@
from __future__ import annotations
-from nextdns import AnalyticsStatus
+from aiohttp import ClientError
+from aiohttp.client_exceptions import ClientConnectorError
+from nextdns import AnalyticsStatus, ApiError, InvalidApiKeyError
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import NextDnsConfigEntry
+from .const import DOMAIN
from .coordinator import NextDnsUpdateCoordinator
PARALLEL_UPDATES = 1
@@ -25,7 +29,7 @@ CLEAR_LOGS_BUTTON = ButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: NextDnsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add aNextDNS entities from a config_entry."""
coordinator = entry.runtime_data.status
@@ -53,4 +57,21 @@ class NextDnsButton(
async def async_press(self) -> None:
"""Trigger cleaning logs."""
- await self.coordinator.nextdns.clear_logs(self.coordinator.profile_id)
+ try:
+ await self.coordinator.nextdns.clear_logs(self.coordinator.profile_id)
+ except (
+ ApiError,
+ ClientConnectorError,
+ TimeoutError,
+ ClientError,
+ ) as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="method_error",
+ translation_placeholders={
+ "entity": self.entity_id,
+ "error": repr(err),
+ },
+ ) from err
+ except InvalidApiKeyError:
+ self.coordinator.config_entry.async_start_reauth(self.hass)
diff --git a/homeassistant/components/nextdns/coordinator.py b/homeassistant/components/nextdns/coordinator.py
index 6b35e35a027..41f6ff43a2a 100644
--- a/homeassistant/components/nextdns/coordinator.py
+++ b/homeassistant/components/nextdns/coordinator.py
@@ -1,8 +1,10 @@
"""NextDns coordinator."""
+from __future__ import annotations
+
from datetime import timedelta
import logging
-from typing import TypeVar
+from typing import TYPE_CHECKING, TypeVar
from aiohttp.client_exceptions import ClientConnectorError
from nextdns import (
@@ -25,6 +27,9 @@ from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import NextDnsConfigEntry
+
from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
@@ -35,9 +40,12 @@ CoordinatorDataT = TypeVar("CoordinatorDataT", bound=NextDnsData)
class NextDnsUpdateCoordinator(DataUpdateCoordinator[CoordinatorDataT]):
"""Class to manage fetching NextDNS data API."""
+ config_entry: NextDnsConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: NextDnsConfigEntry,
nextdns: NextDns,
profile_id: str,
update_interval: timedelta,
@@ -54,7 +62,13 @@ class NextDnsUpdateCoordinator(DataUpdateCoordinator[CoordinatorDataT]):
name=self.profile_name,
)
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=update_interval)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=update_interval,
+ )
async def _async_update_data(self) -> CoordinatorDataT:
"""Update data via internal method."""
@@ -65,9 +79,20 @@ class NextDnsUpdateCoordinator(DataUpdateCoordinator[CoordinatorDataT]):
ClientConnectorError,
RetryError,
) as err:
- raise UpdateFailed(err) from err
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="update_error",
+ translation_placeholders={
+ "entry": self.config_entry.title,
+ "error": repr(err),
+ },
+ ) from err
except InvalidApiKeyError as err:
- raise ConfigEntryAuthFailed from err
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN,
+ translation_key="auth_error",
+ translation_placeholders={"entry": self.config_entry.title},
+ ) from err
async def _async_update_data_internal(self) -> CoordinatorDataT:
"""Update data via library."""
diff --git a/homeassistant/components/nextdns/sensor.py b/homeassistant/components/nextdns/sensor.py
index ef2b5140fa1..0a4a8eaad8f 100644
--- a/homeassistant/components/nextdns/sensor.py
+++ b/homeassistant/components/nextdns/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -286,7 +286,7 @@ SENSORS: tuple[NextDnsSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: NextDnsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a NextDNS entities from a config_entry."""
async_add_entities(
diff --git a/homeassistant/components/nextdns/strings.json b/homeassistant/components/nextdns/strings.json
index f2a5fa2816d..38944a0711e 100644
--- a/homeassistant/components/nextdns/strings.json
+++ b/homeassistant/components/nextdns/strings.json
@@ -359,5 +359,19 @@
"name": "Force YouTube restricted mode"
}
}
+ },
+ "exceptions": {
+ "auth_error": {
+ "message": "Authentication failed for {entry}, please update your API key"
+ },
+ "cannot_connect": {
+ "message": "An error occurred while connecting to the NextDNS API for {entry}: {error}"
+ },
+ "method_error": {
+ "message": "An error occurred while calling the NextDNS API method for {entity}: {error}"
+ },
+ "update_error": {
+ "message": "An error occurred while retrieving data from the NextDNS API for {entry}: {error}"
+ }
}
}
diff --git a/homeassistant/components/nextdns/switch.py b/homeassistant/components/nextdns/switch.py
index 37ff22c7521..8bdca76b955 100644
--- a/homeassistant/components/nextdns/switch.py
+++ b/homeassistant/components/nextdns/switch.py
@@ -8,16 +8,17 @@ from typing import Any
from aiohttp import ClientError
from aiohttp.client_exceptions import ClientConnectorError
-from nextdns import ApiError, Settings
+from nextdns import ApiError, InvalidApiKeyError, Settings
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import NextDnsConfigEntry
+from .const import DOMAIN
from .coordinator import NextDnsUpdateCoordinator
PARALLEL_UPDATES = 1
@@ -525,7 +526,7 @@ SWITCHES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: NextDnsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add NextDNS entities from a config_entry."""
coordinator = entry.runtime_data.settings
@@ -582,9 +583,16 @@ class NextDnsSwitch(
ClientError,
) as err:
raise HomeAssistantError(
- "NextDNS API returned an error calling set_setting for"
- f" {self.entity_id}: {err}"
+ translation_domain=DOMAIN,
+ translation_key="method_error",
+ translation_placeholders={
+ "entity": self.entity_id,
+ "error": repr(err),
+ },
) from err
+ except InvalidApiKeyError:
+ self.coordinator.config_entry.async_start_reauth(self.hass)
+ return
if result:
self._attr_is_on = new_state
diff --git a/homeassistant/components/nibe_heatpump/__init__.py b/homeassistant/components/nibe_heatpump/__init__.py
index b3ceb00a834..ac201ed2322 100644
--- a/homeassistant/components/nibe_heatpump/__init__.py
+++ b/homeassistant/components/nibe_heatpump/__init__.py
@@ -81,7 +81,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _async_stop)
)
- coordinator = CoilCoordinator(hass, heatpump, connection)
+ coordinator = CoilCoordinator(hass, entry, heatpump, connection)
data = hass.data.setdefault(DOMAIN, {})
data[entry.entry_id] = coordinator
diff --git a/homeassistant/components/nibe_heatpump/binary_sensor.py b/homeassistant/components/nibe_heatpump/binary_sensor.py
index 0cb16bf4485..284e4d83569 100644
--- a/homeassistant/components/nibe_heatpump/binary_sensor.py
+++ b/homeassistant/components/nibe_heatpump/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import ENTITY_ID_FORMAT, BinarySenso
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import CoilCoordinator
@@ -18,7 +18,7 @@ from .entity import CoilEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nibe_heatpump/button.py b/homeassistant/components/nibe_heatpump/button.py
index df8ceef6479..849912af656 100644
--- a/homeassistant/components/nibe_heatpump/button.py
+++ b/homeassistant/components/nibe_heatpump/button.py
@@ -9,7 +9,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, LOGGER
@@ -19,7 +19,7 @@ from .coordinator import CoilCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nibe_heatpump/climate.py b/homeassistant/components/nibe_heatpump/climate.py
index 94db90e7f58..1b8a0ecc0df 100644
--- a/homeassistant/components/nibe_heatpump/climate.py
+++ b/homeassistant/components/nibe_heatpump/climate.py
@@ -27,7 +27,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -44,7 +44,7 @@ from .coordinator import CoilCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nibe_heatpump/coordinator.py b/homeassistant/components/nibe_heatpump/coordinator.py
index faaac5f165a..2451e2fbda9 100644
--- a/homeassistant/components/nibe_heatpump/coordinator.py
+++ b/homeassistant/components/nibe_heatpump/coordinator.py
@@ -71,12 +71,17 @@ class CoilCoordinator(ContextCoordinator[dict[int, CoilData], int]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
heatpump: HeatPump,
connection: Connection,
) -> None:
"""Initialize coordinator."""
super().__init__(
- hass, LOGGER, name="Nibe Heat Pump", update_interval=timedelta(seconds=60)
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name="Nibe Heat Pump",
+ update_interval=timedelta(seconds=60),
)
self.data = {}
diff --git a/homeassistant/components/nibe_heatpump/manifest.json b/homeassistant/components/nibe_heatpump/manifest.json
index 049ba905f04..a8441fb90d8 100644
--- a/homeassistant/components/nibe_heatpump/manifest.json
+++ b/homeassistant/components/nibe_heatpump/manifest.json
@@ -5,5 +5,5 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/nibe_heatpump",
"iot_class": "local_polling",
- "requirements": ["nibe==2.14.0"]
+ "requirements": ["nibe==2.17.0"]
}
diff --git a/homeassistant/components/nibe_heatpump/number.py b/homeassistant/components/nibe_heatpump/number.py
index cb379139eed..d85e5e9b765 100644
--- a/homeassistant/components/nibe_heatpump/number.py
+++ b/homeassistant/components/nibe_heatpump/number.py
@@ -8,7 +8,7 @@ from homeassistant.components.number import ENTITY_ID_FORMAT, NumberEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import CoilCoordinator
@@ -18,7 +18,7 @@ from .entity import CoilEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nibe_heatpump/select.py b/homeassistant/components/nibe_heatpump/select.py
index 3aecff94649..c92c12a882a 100644
--- a/homeassistant/components/nibe_heatpump/select.py
+++ b/homeassistant/components/nibe_heatpump/select.py
@@ -8,7 +8,7 @@ from homeassistant.components.select import ENTITY_ID_FORMAT, SelectEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import CoilCoordinator
@@ -18,7 +18,7 @@ from .entity import CoilEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nibe_heatpump/sensor.py b/homeassistant/components/nibe_heatpump/sensor.py
index d34fed50977..54cd0f7ea34 100644
--- a/homeassistant/components/nibe_heatpump/sensor.py
+++ b/homeassistant/components/nibe_heatpump/sensor.py
@@ -13,17 +13,20 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
+ PERCENTAGE,
EntityCategory,
UnitOfElectricCurrent,
UnitOfElectricPotential,
UnitOfEnergy,
UnitOfFrequency,
UnitOfPower,
+ UnitOfPressure,
UnitOfTemperature,
UnitOfTime,
+ UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import CoilCoordinator
@@ -114,6 +117,20 @@ UNIT_DESCRIPTIONS = {
state_class=SensorStateClass.TOTAL_INCREASING,
native_unit_of_measurement=UnitOfTime.HOURS,
),
+ "min": SensorEntityDescription(
+ key="min",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.DURATION,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ native_unit_of_measurement=UnitOfTime.MINUTES,
+ ),
+ "s": SensorEntityDescription(
+ key="s",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.DURATION,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ native_unit_of_measurement=UnitOfTime.SECONDS,
+ ),
"Hz": SensorEntityDescription(
key="Hz",
entity_category=EntityCategory.DIAGNOSTIC,
@@ -121,13 +138,55 @@ UNIT_DESCRIPTIONS = {
state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=UnitOfFrequency.HERTZ,
),
+ "Pa": SensorEntityDescription(
+ key="Pa",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.PRESSURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ native_unit_of_measurement=UnitOfPressure.PA,
+ ),
+ "kPa": SensorEntityDescription(
+ key="kPa",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.PRESSURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ native_unit_of_measurement=UnitOfPressure.KPA,
+ ),
+ "bar": SensorEntityDescription(
+ key="bar",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.PRESSURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ native_unit_of_measurement=UnitOfPressure.BAR,
+ ),
+ "l/m": SensorEntityDescription(
+ key="l/m",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.VOLUME_FLOW_RATE,
+ state_class=SensorStateClass.MEASUREMENT,
+ native_unit_of_measurement=UnitOfVolumeFlowRate.LITERS_PER_MINUTE,
+ ),
+ "m³/h": SensorEntityDescription(
+ key="m³/h",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.VOLUME_FLOW_RATE,
+ state_class=SensorStateClass.MEASUREMENT,
+ native_unit_of_measurement=UnitOfVolumeFlowRate.CUBIC_METERS_PER_HOUR,
+ ),
+ "%RH": SensorEntityDescription(
+ key="%RH",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ device_class=SensorDeviceClass.HUMIDITY,
+ state_class=SensorStateClass.MEASUREMENT,
+ native_unit_of_measurement=PERCENTAGE,
+ ),
}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nibe_heatpump/strings.json b/homeassistant/components/nibe_heatpump/strings.json
index 6fa421e0855..c65a76d3364 100644
--- a/homeassistant/components/nibe_heatpump/strings.json
+++ b/homeassistant/components/nibe_heatpump/strings.json
@@ -10,13 +10,13 @@
},
"modbus": {
"data": {
- "model": "Model of Heat Pump",
+ "model": "Model of heat pump",
"modbus_url": "Modbus URL",
- "modbus_unit": "Modbus Unit Identifier"
+ "modbus_unit": "Modbus unit identifier"
},
"data_description": {
- "modbus_url": "Modbus URL that describes the connection to your Heat Pump or MODBUS40 unit. It should be on the form:\n - `tcp://[HOST]:[PORT]` for Modbus TCP connection\n - `serial://[LOCAL DEVICE]` for a local Modbus RTU connection\n - `rfc2217://[HOST]:[PORT]` for a remote telnet based Modbus RTU connection.",
- "modbus_unit": "Unit identification for your Heat Pump. Can usually be left at 0."
+ "modbus_url": "Modbus URL that describes the connection to your heat pump or MODBUS40 unit. It should be in the form:\n - `tcp://[HOST]:[PORT]` for Modbus TCP connection\n - `serial://[LOCAL DEVICE]` for a local Modbus RTU connection\n - `rfc2217://[HOST]:[PORT]` for a remote Telnet-based Modbus RTU connection.",
+ "modbus_unit": "Unit identification for your heat pump. Can usually be left at 0."
}
},
"nibegw": {
diff --git a/homeassistant/components/nibe_heatpump/switch.py b/homeassistant/components/nibe_heatpump/switch.py
index 72b7c20c7b3..2daf3fc48ff 100644
--- a/homeassistant/components/nibe_heatpump/switch.py
+++ b/homeassistant/components/nibe_heatpump/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import ENTITY_ID_FORMAT, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import CoilCoordinator
@@ -20,7 +20,7 @@ from .entity import CoilEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nibe_heatpump/water_heater.py b/homeassistant/components/nibe_heatpump/water_heater.py
index f53df596d27..a72851e7eab 100644
--- a/homeassistant/components/nibe_heatpump/water_heater.py
+++ b/homeassistant/components/nibe_heatpump/water_heater.py
@@ -17,7 +17,7 @@ from homeassistant.components.water_heater import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -32,7 +32,7 @@ from .coordinator import CoilCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform."""
diff --git a/homeassistant/components/nice_go/__init__.py b/homeassistant/components/nice_go/__init__.py
index b217112c192..a8d2bd71ac4 100644
--- a/homeassistant/components/nice_go/__init__.py
+++ b/homeassistant/components/nice_go/__init__.py
@@ -4,11 +4,10 @@ from __future__ import annotations
import logging
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EVENT_HOMEASSISTANT_STOP, Platform
from homeassistant.core import HomeAssistant
-from .coordinator import NiceGOUpdateCoordinator
+from .coordinator import NiceGOConfigEntry, NiceGOUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
PLATFORMS: list[Platform] = [
@@ -18,13 +17,11 @@ PLATFORMS: list[Platform] = [
Platform.SWITCH,
]
-type NiceGOConfigEntry = ConfigEntry[NiceGOUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: NiceGOConfigEntry) -> bool:
"""Set up Nice G.O. from a config entry."""
- coordinator = NiceGOUpdateCoordinator(hass)
+ coordinator = NiceGOUpdateCoordinator(hass, entry)
entry.async_on_unload(
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, coordinator.async_ha_stop)
)
diff --git a/homeassistant/components/nice_go/coordinator.py b/homeassistant/components/nice_go/coordinator.py
index 07b20bbbf10..ffdd9dbd518 100644
--- a/homeassistant/components/nice_go/coordinator.py
+++ b/homeassistant/components/nice_go/coordinator.py
@@ -54,19 +54,18 @@ class NiceGODevice:
vacation_mode: bool | None
+type NiceGOConfigEntry = ConfigEntry[NiceGOUpdateCoordinator]
+
+
class NiceGOUpdateCoordinator(DataUpdateCoordinator[dict[str, NiceGODevice]]):
"""DataUpdateCoordinator for Nice G.O."""
- config_entry: ConfigEntry
+ config_entry: NiceGOConfigEntry
organization_id: str
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: NiceGOConfigEntry) -> None:
"""Initialize DataUpdateCoordinator for Nice G.O."""
- super().__init__(
- hass,
- _LOGGER,
- name="Nice G.O.",
- )
+ super().__init__(hass, _LOGGER, config_entry=config_entry, name="Nice G.O.")
self.refresh_token = self.config_entry.data[CONF_REFRESH_TOKEN]
self.refresh_token_creation_time = self.config_entry.data[
@@ -154,7 +153,7 @@ class NiceGOUpdateCoordinator(DataUpdateCoordinator[dict[str, NiceGODevice]]):
)
try:
if datetime.now().timestamp() >= expiry_time:
- await self._update_refresh_token()
+ await self.update_refresh_token()
else:
await self.api.authenticate_refresh(
self.refresh_token, async_get_clientsession(self.hass)
@@ -179,7 +178,7 @@ class NiceGOUpdateCoordinator(DataUpdateCoordinator[dict[str, NiceGODevice]]):
else:
self.async_set_updated_data(devices)
- async def _update_refresh_token(self) -> None:
+ async def update_refresh_token(self) -> None:
"""Update the refresh token with Nice G.O. API."""
_LOGGER.debug("Updating the refresh token with Nice G.O. API")
try:
diff --git a/homeassistant/components/nice_go/cover.py b/homeassistant/components/nice_go/cover.py
index 6360e398b96..b9b39711a01 100644
--- a/homeassistant/components/nice_go/cover.py
+++ b/homeassistant/components/nice_go/cover.py
@@ -2,21 +2,17 @@
from typing import Any
-from aiohttp import ClientError
-from nice_go import ApiError
-
from homeassistant.components.cover import (
CoverDeviceClass,
CoverEntity,
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NiceGOConfigEntry
-from .const import DOMAIN
+from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity
+from .util import retry
DEVICE_CLASSES = {
"WallStation": CoverDeviceClass.GARAGE,
@@ -29,7 +25,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NiceGOConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Nice G.O. cover."""
coordinator = config_entry.runtime_data
@@ -71,30 +67,18 @@ class NiceGOCoverEntity(NiceGOEntity, CoverEntity):
"""Return if cover is closing."""
return self.data.barrier_status == "closing"
+ @retry("close_cover_error")
async def async_close_cover(self, **kwargs: Any) -> None:
"""Close the garage door."""
if self.is_closed:
return
- try:
- await self.coordinator.api.close_barrier(self._device_id)
- except (ApiError, ClientError) as err:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="close_cover_error",
- translation_placeholders={"exception": str(err)},
- ) from err
+ await self.coordinator.api.close_barrier(self._device_id)
+ @retry("open_cover_error")
async def async_open_cover(self, **kwargs: Any) -> None:
"""Open the garage door."""
if self.is_opened:
return
- try:
- await self.coordinator.api.open_barrier(self._device_id)
- except (ApiError, ClientError) as err:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="open_cover_error",
- translation_placeholders={"exception": str(err)},
- ) from err
+ await self.coordinator.api.open_barrier(self._device_id)
diff --git a/homeassistant/components/nice_go/diagnostics.py b/homeassistant/components/nice_go/diagnostics.py
index 2c9a695d4b5..2a663d8925a 100644
--- a/homeassistant/components/nice_go/diagnostics.py
+++ b/homeassistant/components/nice_go/diagnostics.py
@@ -9,8 +9,8 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
from homeassistant.core import HomeAssistant
-from . import NiceGOConfigEntry
from .const import CONF_REFRESH_TOKEN
+from .coordinator import NiceGOConfigEntry
TO_REDACT = {CONF_PASSWORD, CONF_EMAIL, CONF_REFRESH_TOKEN, "title", "unique_id"}
diff --git a/homeassistant/components/nice_go/event.py b/homeassistant/components/nice_go/event.py
index cd9198bcd26..400cc3d2144 100644
--- a/homeassistant/components/nice_go/event.py
+++ b/homeassistant/components/nice_go/event.py
@@ -5,9 +5,9 @@ from typing import Any
from homeassistant.components.event import EventEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NiceGOConfigEntry
+from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity
_LOGGER = logging.getLogger(__name__)
@@ -16,7 +16,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NiceGOConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Nice G.O. event."""
diff --git a/homeassistant/components/nice_go/light.py b/homeassistant/components/nice_go/light.py
index abb192adde1..bf283ed6eff 100644
--- a/homeassistant/components/nice_go/light.py
+++ b/homeassistant/components/nice_go/light.py
@@ -3,23 +3,19 @@
import logging
from typing import TYPE_CHECKING, Any
-from aiohttp import ClientError
-from nice_go import ApiError
-
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NiceGOConfigEntry
from .const import (
- DOMAIN,
KNOWN_UNSUPPORTED_DEVICE_TYPES,
SUPPORTED_DEVICE_TYPES,
UNSUPPORTED_DEVICE_WARNING,
)
+from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity
+from .util import retry
_LOGGER = logging.getLogger(__name__)
@@ -27,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NiceGOConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Nice G.O. light."""
@@ -63,26 +59,14 @@ class NiceGOLightEntity(NiceGOEntity, LightEntity):
assert self.data.light_status is not None
return self.data.light_status
+ @retry("light_on_error")
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn on the light."""
- try:
- await self.coordinator.api.light_on(self._device_id)
- except (ApiError, ClientError) as error:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="light_on_error",
- translation_placeholders={"exception": str(error)},
- ) from error
+ await self.coordinator.api.light_on(self._device_id)
+ @retry("light_off_error")
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn off the light."""
- try:
- await self.coordinator.api.light_off(self._device_id)
- except (ApiError, ClientError) as error:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="light_off_error",
- translation_placeholders={"exception": str(error)},
- ) from error
+ await self.coordinator.api.light_off(self._device_id)
diff --git a/homeassistant/components/nice_go/switch.py b/homeassistant/components/nice_go/switch.py
index e3b85528f3b..f043a23eab5 100644
--- a/homeassistant/components/nice_go/switch.py
+++ b/homeassistant/components/nice_go/switch.py
@@ -5,23 +5,19 @@ from __future__ import annotations
import logging
from typing import TYPE_CHECKING, Any
-from aiohttp import ClientError
-from nice_go import ApiError
-
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NiceGOConfigEntry
from .const import (
- DOMAIN,
KNOWN_UNSUPPORTED_DEVICE_TYPES,
SUPPORTED_DEVICE_TYPES,
UNSUPPORTED_DEVICE_WARNING,
)
+from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity
+from .util import retry
_LOGGER = logging.getLogger(__name__)
@@ -29,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NiceGOConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Nice G.O. switch."""
coordinator = config_entry.runtime_data
@@ -65,26 +61,14 @@ class NiceGOSwitchEntity(NiceGOEntity, SwitchEntity):
assert self.data.vacation_mode is not None
return self.data.vacation_mode
+ @retry("switch_on_error")
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn the switch on."""
- try:
- await self.coordinator.api.vacation_mode_on(self.data.id)
- except (ApiError, ClientError) as error:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="switch_on_error",
- translation_placeholders={"exception": str(error)},
- ) from error
+ await self.coordinator.api.vacation_mode_on(self.data.id)
+ @retry("switch_off_error")
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the switch off."""
- try:
- await self.coordinator.api.vacation_mode_off(self.data.id)
- except (ApiError, ClientError) as error:
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="switch_off_error",
- translation_placeholders={"exception": str(error)},
- ) from error
+ await self.coordinator.api.vacation_mode_off(self.data.id)
diff --git a/homeassistant/components/nice_go/util.py b/homeassistant/components/nice_go/util.py
new file mode 100644
index 00000000000..02dee6b0ac1
--- /dev/null
+++ b/homeassistant/components/nice_go/util.py
@@ -0,0 +1,66 @@
+"""Utilities for Nice G.O."""
+
+from collections.abc import Callable, Coroutine
+from functools import wraps
+from typing import Any, Protocol, runtime_checkable
+
+from aiohttp import ClientError
+from nice_go import ApiError, AuthFailedError
+
+from homeassistant.exceptions import ConfigEntryAuthFailed, HomeAssistantError
+from homeassistant.helpers.update_coordinator import UpdateFailed
+
+from .const import DOMAIN
+
+
+@runtime_checkable
+class _ArgsProtocol(Protocol):
+ coordinator: Any
+ hass: Any
+
+
+def retry[_R, **P](
+ translation_key: str,
+) -> Callable[
+ [Callable[P, Coroutine[Any, Any, _R]]], Callable[P, Coroutine[Any, Any, _R]]
+]:
+ """Retry decorator to handle API errors."""
+
+ def decorator(
+ func: Callable[P, Coroutine[Any, Any, _R]],
+ ) -> Callable[P, Coroutine[Any, Any, _R]]:
+ @wraps(func)
+ async def wrapper(*args: P.args, **kwargs: P.kwargs):
+ instance = args[0]
+ if not isinstance(instance, _ArgsProtocol):
+ raise TypeError("First argument must have correct attributes")
+ try:
+ return await func(*args, **kwargs)
+ except (ApiError, ClientError) as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key=translation_key,
+ translation_placeholders={"exception": str(err)},
+ ) from err
+ except AuthFailedError:
+ # Try refreshing token and retry
+ try:
+ await instance.coordinator.update_refresh_token()
+ return await func(*args, **kwargs)
+ except (ApiError, ClientError, UpdateFailed) as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key=translation_key,
+ translation_placeholders={"exception": str(err)},
+ ) from err
+ except (AuthFailedError, ConfigEntryAuthFailed) as err:
+ instance.coordinator.config_entry.async_start_reauth(instance.hass)
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key=translation_key,
+ translation_placeholders={"exception": str(err)},
+ ) from err
+
+ return wrapper
+
+ return decorator
diff --git a/homeassistant/components/nightscout/sensor.py b/homeassistant/components/nightscout/sensor.py
index 620349ec3c3..de1dadf1143 100644
--- a/homeassistant/components/nightscout/sensor.py
+++ b/homeassistant/components/nightscout/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_DATE, UnitOfBloodGlucoseConcentration
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_DELTA, ATTR_DEVICE, ATTR_DIRECTION, DOMAIN
@@ -27,7 +27,7 @@ DEFAULT_NAME = "Blood Glucose"
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Glucose Sensor."""
api = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/niko_home_control/cover.py b/homeassistant/components/niko_home_control/cover.py
index b3546b517d5..2ab3438c4d9 100644
--- a/homeassistant/components/niko_home_control/cover.py
+++ b/homeassistant/components/niko_home_control/cover.py
@@ -8,7 +8,7 @@ from nhc.cover import NHCCover
from homeassistant.components.cover import CoverEntity, CoverEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import NikoHomeControlConfigEntry
from .entity import NikoHomeControlEntity
@@ -17,7 +17,7 @@ from .entity import NikoHomeControlEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: NikoHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Niko Home Control cover entry."""
controller = entry.runtime_data
diff --git a/homeassistant/components/niko_home_control/light.py b/homeassistant/components/niko_home_control/light.py
index 7c0d11b3388..b0a2d12b004 100644
--- a/homeassistant/components/niko_home_control/light.py
+++ b/homeassistant/components/niko_home_control/light.py
@@ -19,7 +19,10 @@ from homeassistant.const import CONF_HOST
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant
from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers import config_validation as cv, issue_registry as ir
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import NHCController, NikoHomeControlConfigEntry
@@ -80,7 +83,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: NikoHomeControlConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Niko Home Control light entry."""
controller = entry.runtime_data
diff --git a/homeassistant/components/niko_home_control/manifest.json b/homeassistant/components/niko_home_control/manifest.json
index b50410cd7de..83fca0ca2d6 100644
--- a/homeassistant/components/niko_home_control/manifest.json
+++ b/homeassistant/components/niko_home_control/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/niko_home_control",
"iot_class": "local_push",
"loggers": ["nikohomecontrol"],
- "requirements": ["nhc==0.4.4"]
+ "requirements": ["nhc==0.4.10"]
}
diff --git a/homeassistant/components/nina/__init__.py b/homeassistant/components/nina/__init__.py
index d5b1c5ccb35..b02d6711e74 100644
--- a/homeassistant/components/nina/__init__.py
+++ b/homeassistant/components/nina/__init__.py
@@ -11,7 +11,6 @@ from .const import (
CONF_AREA_FILTER,
CONF_FILTER_CORONA,
CONF_HEADLINE_FILTER,
- CONF_REGIONS,
DOMAIN,
NO_MATCH_REGEX,
)
@@ -22,9 +21,6 @@ PLATFORMS: list[str] = [Platform.BINARY_SENSOR]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up platform from a ConfigEntry."""
-
- regions: dict[str, str] = entry.data[CONF_REGIONS]
-
if CONF_HEADLINE_FILTER not in entry.data:
filter_regex = NO_MATCH_REGEX
@@ -39,12 +35,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
new_data = {**entry.data, CONF_AREA_FILTER: ALL_MATCH_REGEX}
hass.config_entries.async_update_entry(entry, data=new_data)
- coordinator = NINADataUpdateCoordinator(
- hass,
- regions,
- entry.data[CONF_HEADLINE_FILTER],
- entry.data[CONF_AREA_FILTER],
- )
+ coordinator = NINADataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/nina/binary_sensor.py b/homeassistant/components/nina/binary_sensor.py
index 10d3008fd82..3f7d496aca9 100644
--- a/homeassistant/components/nina/binary_sensor.py
+++ b/homeassistant/components/nina/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -36,7 +36,7 @@ from .coordinator import NINADataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entries."""
diff --git a/homeassistant/components/nina/coordinator.py b/homeassistant/components/nina/coordinator.py
index 2d9548f3d12..3c27729ef09 100644
--- a/homeassistant/components/nina/coordinator.py
+++ b/homeassistant/components/nina/coordinator.py
@@ -9,11 +9,19 @@ from typing import Any
from pynina import ApiError, Nina
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import _LOGGER, DOMAIN, SCAN_INTERVAL
+from .const import (
+ _LOGGER,
+ CONF_AREA_FILTER,
+ CONF_HEADLINE_FILTER,
+ CONF_REGIONS,
+ DOMAIN,
+ SCAN_INTERVAL,
+)
@dataclass
@@ -39,23 +47,29 @@ class NINADataUpdateCoordinator(
):
"""Class to manage fetching NINA data API."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- regions: dict[str, str],
- headline_filter: str,
- area_filter: str,
+ config_entry: ConfigEntry,
) -> None:
"""Initialize."""
- self._regions: dict[str, str] = regions
self._nina: Nina = Nina(async_get_clientsession(hass))
- self.headline_filter: str = headline_filter
- self.area_filter: str = area_filter
+ self.headline_filter: str = config_entry.data[CONF_HEADLINE_FILTER]
+ self.area_filter: str = config_entry.data[CONF_AREA_FILTER]
+ regions: dict[str, str] = config_entry.data[CONF_REGIONS]
for region in regions:
self._nina.addRegion(region)
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
+ )
async def _async_update_data(self) -> dict[str, list[NinaWarningData]]:
"""Update data."""
diff --git a/homeassistant/components/nmap_tracker/device_tracker.py b/homeassistant/components/nmap_tracker/device_tracker.py
index c8e7e7c25ea..afac3f06435 100644
--- a/homeassistant/components/nmap_tracker/device_tracker.py
+++ b/homeassistant/components/nmap_tracker/device_tracker.py
@@ -9,7 +9,7 @@ from homeassistant.components.device_tracker import ScannerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import NmapDevice, NmapDeviceScanner, short_hostname, signal_device_update
from .const import DOMAIN
@@ -18,7 +18,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Nmap Tracker component."""
nmap_tracker = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/nmbs/__init__.py b/homeassistant/components/nmbs/__init__.py
index 7d06baf37b6..4a2783143ca 100644
--- a/homeassistant/components/nmbs/__init__.py
+++ b/homeassistant/components/nmbs/__init__.py
@@ -8,6 +8,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.typing import ConfigType
from .const import DOMAIN
@@ -22,13 +23,13 @@ CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the NMBS component."""
- api_client = iRail()
+ api_client = iRail(session=async_get_clientsession(hass))
hass.data.setdefault(DOMAIN, {})
- station_response = await hass.async_add_executor_job(api_client.get_stations)
- if station_response == -1:
+ station_response = await api_client.get_stations()
+ if station_response is None:
return False
- hass.data[DOMAIN] = station_response["station"]
+ hass.data[DOMAIN] = station_response.stations
return True
diff --git a/homeassistant/components/nmbs/config_flow.py b/homeassistant/components/nmbs/config_flow.py
index e45b2d9adeb..60ab015e22b 100644
--- a/homeassistant/components/nmbs/config_flow.py
+++ b/homeassistant/components/nmbs/config_flow.py
@@ -3,11 +3,13 @@
from typing import Any
from pyrail import iRail
+from pyrail.models import StationDetails
import voluptuous as vol
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import Platform
from homeassistant.helpers import entity_registry as er
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.selector import (
BooleanSelector,
SelectOptionDict,
@@ -31,17 +33,15 @@ class NMBSConfigFlow(ConfigFlow, domain=DOMAIN):
def __init__(self) -> None:
"""Initialize."""
- self.api_client = iRail()
- self.stations: list[dict[str, Any]] = []
+ self.stations: list[StationDetails] = []
- async def _fetch_stations(self) -> list[dict[str, Any]]:
+ async def _fetch_stations(self) -> list[StationDetails]:
"""Fetch the stations."""
- stations_response = await self.hass.async_add_executor_job(
- self.api_client.get_stations
- )
- if stations_response == -1:
+ api_client = iRail(session=async_get_clientsession(self.hass))
+ stations_response = await api_client.get_stations()
+ if stations_response is None:
raise CannotConnect("The API is currently unavailable.")
- return stations_response["station"]
+ return stations_response.stations
async def _fetch_stations_choices(self) -> list[SelectOptionDict]:
"""Fetch the stations options."""
@@ -50,7 +50,7 @@ class NMBSConfigFlow(ConfigFlow, domain=DOMAIN):
self.stations = await self._fetch_stations()
return [
- SelectOptionDict(value=station["id"], label=station["standardname"])
+ SelectOptionDict(value=station.id, label=station.standard_name)
for station in self.stations
]
@@ -72,12 +72,12 @@ class NMBSConfigFlow(ConfigFlow, domain=DOMAIN):
[station_from] = [
station
for station in self.stations
- if station["id"] == user_input[CONF_STATION_FROM]
+ if station.id == user_input[CONF_STATION_FROM]
]
[station_to] = [
station
for station in self.stations
- if station["id"] == user_input[CONF_STATION_TO]
+ if station.id == user_input[CONF_STATION_TO]
]
vias = "_excl_vias" if user_input.get(CONF_EXCLUDE_VIAS) else ""
await self.async_set_unique_id(
@@ -85,7 +85,7 @@ class NMBSConfigFlow(ConfigFlow, domain=DOMAIN):
)
self._abort_if_unique_id_configured()
- config_entry_name = f"Train from {station_from['standardname']} to {station_to['standardname']}"
+ config_entry_name = f"Train from {station_from.standard_name} to {station_to.standard_name}"
return self.async_create_entry(
title=config_entry_name,
data=user_input,
@@ -127,18 +127,18 @@ class NMBSConfigFlow(ConfigFlow, domain=DOMAIN):
station_live = None
for station in self.stations:
if user_input[CONF_STATION_FROM] in (
- station["standardname"],
- station["name"],
+ station.standard_name,
+ station.name,
):
station_from = station
if user_input[CONF_STATION_TO] in (
- station["standardname"],
- station["name"],
+ station.standard_name,
+ station.name,
):
station_to = station
if CONF_STATION_LIVE in user_input and user_input[CONF_STATION_LIVE] in (
- station["standardname"],
- station["name"],
+ station.standard_name,
+ station.name,
):
station_live = station
@@ -148,29 +148,29 @@ class NMBSConfigFlow(ConfigFlow, domain=DOMAIN):
return self.async_abort(reason="same_station")
# config flow uses id and not the standard name
- user_input[CONF_STATION_FROM] = station_from["id"]
- user_input[CONF_STATION_TO] = station_to["id"]
+ user_input[CONF_STATION_FROM] = station_from.id
+ user_input[CONF_STATION_TO] = station_to.id
if station_live:
- user_input[CONF_STATION_LIVE] = station_live["id"]
+ user_input[CONF_STATION_LIVE] = station_live.id
entity_registry = er.async_get(self.hass)
prefix = "live"
vias = "_excl_vias" if user_input.get(CONF_EXCLUDE_VIAS, False) else ""
if entity_id := entity_registry.async_get_entity_id(
Platform.SENSOR,
DOMAIN,
- f"{prefix}_{station_live['standardname']}_{station_from['standardname']}_{station_to['standardname']}",
+ f"{prefix}_{station_live.standard_name}_{station_from.standard_name}_{station_to.standard_name}",
):
- new_unique_id = f"{DOMAIN}_{prefix}_{station_live['id']}_{station_from['id']}_{station_to['id']}{vias}"
+ new_unique_id = f"{DOMAIN}_{prefix}_{station_live.id}_{station_from.id}_{station_to.id}{vias}"
entity_registry.async_update_entity(
entity_id, new_unique_id=new_unique_id
)
if entity_id := entity_registry.async_get_entity_id(
Platform.SENSOR,
DOMAIN,
- f"{prefix}_{station_live['name']}_{station_from['name']}_{station_to['name']}",
+ f"{prefix}_{station_live.name}_{station_from.name}_{station_to.name}",
):
- new_unique_id = f"{DOMAIN}_{prefix}_{station_live['id']}_{station_from['id']}_{station_to['id']}{vias}"
+ new_unique_id = f"{DOMAIN}_{prefix}_{station_live.id}_{station_from.id}_{station_to.id}{vias}"
entity_registry.async_update_entity(
entity_id, new_unique_id=new_unique_id
)
diff --git a/homeassistant/components/nmbs/const.py b/homeassistant/components/nmbs/const.py
index fddb7365501..04c8beb327d 100644
--- a/homeassistant/components/nmbs/const.py
+++ b/homeassistant/components/nmbs/const.py
@@ -19,11 +19,7 @@ CONF_SHOW_ON_MAP = "show_on_map"
def find_station_by_name(hass: HomeAssistant, station_name: str):
"""Find given station_name in the station list."""
return next(
- (
- s
- for s in hass.data[DOMAIN]
- if station_name in (s["standardname"], s["name"])
- ),
+ (s for s in hass.data[DOMAIN] if station_name in (s.standard_name, s.name)),
None,
)
@@ -31,6 +27,6 @@ def find_station_by_name(hass: HomeAssistant, station_name: str):
def find_station(hass: HomeAssistant, station_name: str):
"""Find given station_id in the station list."""
return next(
- (s for s in hass.data[DOMAIN] if station_name in s["id"]),
+ (s for s in hass.data[DOMAIN] if station_name in s.id),
None,
)
diff --git a/homeassistant/components/nmbs/manifest.json b/homeassistant/components/nmbs/manifest.json
index 9016eff11f8..37ff9429a54 100644
--- a/homeassistant/components/nmbs/manifest.json
+++ b/homeassistant/components/nmbs/manifest.json
@@ -7,5 +7,5 @@
"iot_class": "cloud_polling",
"loggers": ["pyrail"],
"quality_scale": "legacy",
- "requirements": ["pyrail==0.0.3"]
+ "requirements": ["pyrail==0.4.1"]
}
diff --git a/homeassistant/components/nmbs/sensor.py b/homeassistant/components/nmbs/sensor.py
index 6d13777e10a..822b0236dd0 100644
--- a/homeassistant/components/nmbs/sensor.py
+++ b/homeassistant/components/nmbs/sensor.py
@@ -2,10 +2,12 @@
from __future__ import annotations
+from datetime import datetime
import logging
from typing import Any
from pyrail import iRail
+from pyrail.models import ConnectionDetails, LiveboardDeparture, StationDetails
import voluptuous as vol
from homeassistant.components.sensor import (
@@ -23,7 +25,11 @@ from homeassistant.const import (
)
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
@@ -41,8 +47,6 @@ from .const import ( # noqa: F401
_LOGGER = logging.getLogger(__name__)
-API_FAILURE = -1
-
DEFAULT_NAME = "NMBS"
DEFAULT_ICON = "mdi:train"
@@ -60,12 +64,12 @@ PLATFORM_SCHEMA = SENSOR_PLATFORM_SCHEMA.extend(
)
-def get_time_until(departure_time=None):
+def get_time_until(departure_time: datetime | None = None):
"""Calculate the time between now and a train's departure time."""
if departure_time is None:
return 0
- delta = dt_util.utc_from_timestamp(int(departure_time)) - dt_util.now()
+ delta = dt_util.as_utc(departure_time) - dt_util.utcnow()
return round(delta.total_seconds() / 60)
@@ -74,11 +78,9 @@ def get_delay_in_minutes(delay=0):
return round(int(delay) / 60)
-def get_ride_duration(departure_time, arrival_time, delay=0):
+def get_ride_duration(departure_time: datetime, arrival_time: datetime, delay=0):
"""Calculate the total travel time in minutes."""
- duration = dt_util.utc_from_timestamp(
- int(arrival_time)
- ) - dt_util.utc_from_timestamp(int(departure_time))
+ duration = arrival_time - departure_time
duration_time = int(round(duration.total_seconds() / 60))
return duration_time + get_delay_in_minutes(delay)
@@ -151,10 +153,10 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up NMBS sensor entities based on a config entry."""
- api_client = iRail()
+ api_client = iRail(session=async_get_clientsession(hass))
name = config_entry.data.get(CONF_NAME, None)
show_on_map = config_entry.data.get(CONF_SHOW_ON_MAP, False)
@@ -186,9 +188,9 @@ class NMBSLiveBoard(SensorEntity):
def __init__(
self,
api_client: iRail,
- live_station: dict[str, Any],
- station_from: dict[str, Any],
- station_to: dict[str, Any],
+ live_station: StationDetails,
+ station_from: StationDetails,
+ station_to: StationDetails,
excl_vias: bool,
) -> None:
"""Initialize the sensor for getting liveboard data."""
@@ -198,7 +200,8 @@ class NMBSLiveBoard(SensorEntity):
self._station_to = station_to
self._excl_vias = excl_vias
- self._attrs: dict[str, Any] | None = {}
+ self._attrs: LiveboardDeparture | None = None
+
self._state: str | None = None
self.entity_registry_enabled_default = False
@@ -206,22 +209,20 @@ class NMBSLiveBoard(SensorEntity):
@property
def name(self) -> str:
"""Return the sensor default name."""
- return f"Trains in {self._station['standardname']}"
+ return f"Trains in {self._station.standard_name}"
@property
def unique_id(self) -> str:
"""Return the unique ID."""
- unique_id = (
- f"{self._station['id']}_{self._station_from['id']}_{self._station_to['id']}"
- )
+ unique_id = f"{self._station.id}_{self._station_from.id}_{self._station_to.id}"
vias = "_excl_vias" if self._excl_vias else ""
return f"nmbs_live_{unique_id}{vias}"
@property
def icon(self) -> str:
"""Return the default icon or an alert icon if delays."""
- if self._attrs and int(self._attrs["delay"]) > 0:
+ if self._attrs and int(self._attrs.delay) > 0:
return DEFAULT_ICON_ALERT
return DEFAULT_ICON
@@ -237,15 +238,15 @@ class NMBSLiveBoard(SensorEntity):
if self._state is None or not self._attrs:
return None
- delay = get_delay_in_minutes(self._attrs["delay"])
- departure = get_time_until(self._attrs["time"])
+ delay = get_delay_in_minutes(self._attrs.delay)
+ departure = get_time_until(self._attrs.time)
attrs = {
"departure": f"In {departure} minutes",
"departure_minutes": departure,
- "extra_train": int(self._attrs["isExtra"]) > 0,
- "vehicle_id": self._attrs["vehicle"],
- "monitored_station": self._station["standardname"],
+ "extra_train": self._attrs.is_extra,
+ "vehicle_id": self._attrs.vehicle,
+ "monitored_station": self._station.standard_name,
}
if delay > 0:
@@ -254,28 +255,26 @@ class NMBSLiveBoard(SensorEntity):
return attrs
- def update(self) -> None:
+ async def async_update(self, **kwargs: Any) -> None:
"""Set the state equal to the next departure."""
- liveboard = self._api_client.get_liveboard(self._station["id"])
+ liveboard = await self._api_client.get_liveboard(self._station.id)
- if liveboard == API_FAILURE:
+ if liveboard is None:
_LOGGER.warning("API failed in NMBSLiveBoard")
return
- if not (departures := liveboard.get("departures")):
+ if not (departures := liveboard.departures):
_LOGGER.warning("API returned invalid departures: %r", liveboard)
return
_LOGGER.debug("API returned departures: %r", departures)
- if departures["number"] == "0":
+ if len(departures) == 0:
# No trains are scheduled
return
- next_departure = departures["departure"][0]
+ next_departure = departures[0]
self._attrs = next_departure
- self._state = (
- f"Track {next_departure['platform']} - {next_departure['station']}"
- )
+ self._state = f"Track {next_departure.platform} - {next_departure.station}"
class NMBSSensor(SensorEntity):
@@ -289,8 +288,8 @@ class NMBSSensor(SensorEntity):
api_client: iRail,
name: str,
show_on_map: bool,
- station_from: dict[str, Any],
- station_to: dict[str, Any],
+ station_from: StationDetails,
+ station_to: StationDetails,
excl_vias: bool,
) -> None:
"""Initialize the NMBS connection sensor."""
@@ -301,13 +300,13 @@ class NMBSSensor(SensorEntity):
self._station_to = station_to
self._excl_vias = excl_vias
- self._attrs: dict[str, Any] | None = {}
+ self._attrs: ConnectionDetails | None = None
self._state = None
@property
def unique_id(self) -> str:
"""Return the unique ID."""
- unique_id = f"{self._station_from['id']}_{self._station_to['id']}"
+ unique_id = f"{self._station_from.id}_{self._station_to.id}"
vias = "_excl_vias" if self._excl_vias else ""
return f"nmbs_connection_{unique_id}{vias}"
@@ -316,14 +315,14 @@ class NMBSSensor(SensorEntity):
def name(self) -> str:
"""Return the name of the sensor."""
if self._name is None:
- return f"Train from {self._station_from['standardname']} to {self._station_to['standardname']}"
+ return f"Train from {self._station_from.standard_name} to {self._station_to.standard_name}"
return self._name
@property
def icon(self) -> str:
"""Return the sensor default icon or an alert icon if any delay."""
if self._attrs:
- delay = get_delay_in_minutes(self._attrs["departure"]["delay"])
+ delay = get_delay_in_minutes(self._attrs.departure.delay)
if delay > 0:
return "mdi:alert-octagon"
@@ -335,19 +334,19 @@ class NMBSSensor(SensorEntity):
if self._state is None or not self._attrs:
return None
- delay = get_delay_in_minutes(self._attrs["departure"]["delay"])
- departure = get_time_until(self._attrs["departure"]["time"])
- canceled = int(self._attrs["departure"]["canceled"])
+ delay = get_delay_in_minutes(self._attrs.departure.delay)
+ departure = get_time_until(self._attrs.departure.time)
+ canceled = self._attrs.departure.canceled
attrs = {
- "destination": self._attrs["departure"]["station"],
- "direction": self._attrs["departure"]["direction"]["name"],
- "platform_arriving": self._attrs["arrival"]["platform"],
- "platform_departing": self._attrs["departure"]["platform"],
- "vehicle_id": self._attrs["departure"]["vehicle"],
+ "destination": self._attrs.departure.station,
+ "direction": self._attrs.departure.direction.name,
+ "platform_arriving": self._attrs.arrival.platform,
+ "platform_departing": self._attrs.departure.platform,
+ "vehicle_id": self._attrs.departure.vehicle,
}
- if canceled != 1:
+ if not canceled:
attrs["departure"] = f"In {departure} minutes"
attrs["departure_minutes"] = departure
attrs["canceled"] = False
@@ -361,14 +360,14 @@ class NMBSSensor(SensorEntity):
attrs[ATTR_LONGITUDE] = self.station_coordinates[1]
if self.is_via_connection and not self._excl_vias:
- via = self._attrs["vias"]["via"][0]
+ via = self._attrs.vias.via[0]
- attrs["via"] = via["station"]
- attrs["via_arrival_platform"] = via["arrival"]["platform"]
- attrs["via_transfer_platform"] = via["departure"]["platform"]
+ attrs["via"] = via.station
+ attrs["via_arrival_platform"] = via.arrival.platform
+ attrs["via_transfer_platform"] = via.departure.platform
attrs["via_transfer_time"] = get_delay_in_minutes(
- via["timebetween"]
- ) + get_delay_in_minutes(via["departure"]["delay"])
+ via.timebetween
+ ) + get_delay_in_minutes(via.departure.delay)
if delay > 0:
attrs["delay"] = f"{delay} minutes"
@@ -387,8 +386,8 @@ class NMBSSensor(SensorEntity):
if self._state is None or not self._attrs:
return []
- latitude = float(self._attrs["departure"]["stationinfo"]["locationY"])
- longitude = float(self._attrs["departure"]["stationinfo"]["locationX"])
+ latitude = float(self._attrs.departure.station_info.latitude)
+ longitude = float(self._attrs.departure.station_info.longitude)
return [latitude, longitude]
@property
@@ -397,24 +396,24 @@ class NMBSSensor(SensorEntity):
if not self._attrs:
return False
- return "vias" in self._attrs and int(self._attrs["vias"]["number"]) > 0
+ return self._attrs.vias is not None and len(self._attrs.vias) > 0
- def update(self) -> None:
+ async def async_update(self, **kwargs: Any) -> None:
"""Set the state to the duration of a connection."""
- connections = self._api_client.get_connections(
- self._station_from["id"], self._station_to["id"]
+ connections = await self._api_client.get_connections(
+ self._station_from.id, self._station_to.id
)
- if connections == API_FAILURE:
+ if connections is None:
_LOGGER.warning("API failed in NMBSSensor")
return
- if not (connection := connections.get("connection")):
+ if not (connection := connections.connections):
_LOGGER.warning("API returned invalid connection: %r", connections)
return
_LOGGER.debug("API returned connection: %r", connection)
- if int(connection[0]["departure"]["left"]) > 0:
+ if connection[0].departure.left:
next_connection = connection[1]
else:
next_connection = connection[0]
@@ -428,9 +427,9 @@ class NMBSSensor(SensorEntity):
return
duration = get_ride_duration(
- next_connection["departure"]["time"],
- next_connection["arrival"]["time"],
- next_connection["departure"]["delay"],
+ next_connection.departure.time,
+ next_connection.arrival.time,
+ next_connection.departure.delay,
)
self._state = duration
diff --git a/homeassistant/components/nmbs/strings.json b/homeassistant/components/nmbs/strings.json
index 3e7aa8d05bd..ac11026577a 100644
--- a/homeassistant/components/nmbs/strings.json
+++ b/homeassistant/components/nmbs/strings.json
@@ -29,7 +29,7 @@
"issues": {
"deprecated_yaml_import_issue_station_not_found": {
"title": "The {integration_title} YAML configuration import failed",
- "description": "Configuring {integration_title} using YAML is being removed but there was an problem importing your YAML configuration.\n\nThe used station \"{station_name}\" could not be found. Fix it or remove the {integration_title} YAML configuration from your configuration.yaml file and continue to [set up the integration]({url}) manually."
+ "description": "Configuring {integration_title} using YAML is being removed but there was a problem importing your YAML configuration.\n\nThe used station \"{station_name}\" could not be found. Fix it or remove the {integration_title} YAML configuration from your configuration.yaml file and continue to [set up the integration]({url}) manually."
}
}
}
diff --git a/homeassistant/components/noaa_tides/helpers.py b/homeassistant/components/noaa_tides/helpers.py
new file mode 100644
index 00000000000..734cca68f44
--- /dev/null
+++ b/homeassistant/components/noaa_tides/helpers.py
@@ -0,0 +1,6 @@
+"""Helpers for NOAA Tides integration."""
+
+
+def get_station_unique_id(station_id: str) -> str:
+ """Convert a station ID to a unique ID."""
+ return f"{station_id.lower()}"
diff --git a/homeassistant/components/noaa_tides/manifest.json b/homeassistant/components/noaa_tides/manifest.json
index 8cc81857770..02a189883bc 100644
--- a/homeassistant/components/noaa_tides/manifest.json
+++ b/homeassistant/components/noaa_tides/manifest.json
@@ -6,5 +6,5 @@
"iot_class": "cloud_polling",
"loggers": ["noaa_coops"],
"quality_scale": "legacy",
- "requirements": ["noaa-coops==0.1.9"]
+ "requirements": ["noaa-coops==0.4.0"]
}
diff --git a/homeassistant/components/noaa_tides/sensor.py b/homeassistant/components/noaa_tides/sensor.py
index f6ec9dc4bf2..3b5a13b0f15 100644
--- a/homeassistant/components/noaa_tides/sensor.py
+++ b/homeassistant/components/noaa_tides/sensor.py
@@ -22,6 +22,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util.unit_system import METRIC_SYSTEM
+from .helpers import get_station_unique_id
+
if TYPE_CHECKING:
from pandas import Timestamp
@@ -105,6 +107,7 @@ class NOAATidesAndCurrentsSensor(SensorEntity):
self._unit_system = unit_system
self._station = station
self.data: NOAATidesData | None = None
+ self._attr_unique_id = f"{get_station_unique_id(station_id)}_summary"
@property
def name(self) -> str:
@@ -169,8 +172,8 @@ class NOAATidesAndCurrentsSensor(SensorEntity):
api_data = df_predictions.head()
self.data = NOAATidesData(
time_stamp=list(api_data.index),
- hi_lo=list(api_data["hi_lo"].values),
- predicted_wl=list(api_data["predicted_wl"].values),
+ hi_lo=list(api_data["type"].values),
+ predicted_wl=list(api_data["v"].values),
)
_LOGGER.debug("Data = %s", api_data)
_LOGGER.debug(
diff --git a/homeassistant/components/nobo_hub/climate.py b/homeassistant/components/nobo_hub/climate.py
index a089209cde5..771da420213 100644
--- a/homeassistant/components/nobo_hub/climate.py
+++ b/homeassistant/components/nobo_hub/climate.py
@@ -21,7 +21,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_NAME, PRECISION_TENTHS, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import (
@@ -46,7 +46,7 @@ MAX_TEMPERATURE = 40
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nobø Ecohub platform from UI configuration."""
diff --git a/homeassistant/components/nobo_hub/select.py b/homeassistant/components/nobo_hub/select.py
index 43f177dd7a0..c24dbe3d21d 100644
--- a/homeassistant/components/nobo_hub/select.py
+++ b/homeassistant/components/nobo_hub/select.py
@@ -10,7 +10,7 @@ from homeassistant.const import ATTR_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_HARDWARE_VERSION,
@@ -26,7 +26,7 @@ from .const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up any temperature sensors connected to the Nobø Ecohub."""
diff --git a/homeassistant/components/nobo_hub/sensor.py b/homeassistant/components/nobo_hub/sensor.py
index 1632b6ba5e7..382fd1b0bf4 100644
--- a/homeassistant/components/nobo_hub/sensor.py
+++ b/homeassistant/components/nobo_hub/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_MODEL, ATTR_NAME, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import ATTR_SERIAL, ATTR_ZONE_ID, DOMAIN, NOBO_MANUFACTURER
@@ -22,7 +22,7 @@ from .const import ATTR_SERIAL, ATTR_ZONE_ID, DOMAIN, NOBO_MANUFACTURER
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up any temperature sensors connected to the Nobø Ecohub."""
diff --git a/homeassistant/components/nordpool/sensor.py b/homeassistant/components/nordpool/sensor.py
index 30910f8e5f6..c6993826239 100644
--- a/homeassistant/components/nordpool/sensor.py
+++ b/homeassistant/components/nordpool/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util, slugify
from . import NordPoolConfigEntry
@@ -271,7 +271,7 @@ DAILY_AVERAGE_PRICES_SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: NordPoolConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Nord Pool sensor platform."""
diff --git a/homeassistant/components/nordpool/services.py b/homeassistant/components/nordpool/services.py
index 872bd5b1e6b..6607edfdbcb 100644
--- a/homeassistant/components/nordpool/services.py
+++ b/homeassistant/components/nordpool/services.py
@@ -41,7 +41,7 @@ ATTR_CURRENCY = "currency"
SERVICE_GET_PRICES_FOR_DATE = "get_prices_for_date"
SERVICE_GET_PRICES_SCHEMA = vol.Schema(
{
- vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector(),
+ vol.Required(ATTR_CONFIG_ENTRY): ConfigEntrySelector({"integration": DOMAIN}),
vol.Required(ATTR_DATE): cv.date,
vol.Optional(ATTR_AREAS): vol.All(vol.In(list(AREAS)), cv.ensure_list, [str]),
vol.Optional(ATTR_CURRENCY): vol.All(
diff --git a/homeassistant/components/nordpool/strings.json b/homeassistant/components/nordpool/strings.json
index cc10a1a0640..7b33f032de1 100644
--- a/homeassistant/components/nordpool/strings.json
+++ b/homeassistant/components/nordpool/strings.json
@@ -15,7 +15,7 @@
},
"data_description": {
"currency": "Select currency to display prices in, EUR is the base currency.",
- "areas": "Areas to display prices for according to Nordpool market areas."
+ "areas": "Areas to display prices for according to Nord Pool market areas."
}
},
"reconfigure": {
@@ -95,11 +95,11 @@
"services": {
"get_prices_for_date": {
"name": "Get prices for date",
- "description": "Retrieve the prices for a specific date.",
+ "description": "Retrieves the prices for a specific date.",
"fields": {
"config_entry": {
- "name": "Select Nord Pool configuration entry",
- "description": "Choose the configuration entry."
+ "name": "Config entry",
+ "description": "The Nord Pool configuration entry for this action."
},
"date": {
"name": "Date",
diff --git a/homeassistant/components/notify/strings.json b/homeassistant/components/notify/strings.json
index e832bfc248a..b33af360448 100644
--- a/homeassistant/components/notify/strings.json
+++ b/homeassistant/components/notify/strings.json
@@ -24,7 +24,7 @@
},
"data": {
"name": "Data",
- "description": "Some integrations provide extended functionality. For information on how to use _data_, refer to the integration documentation."
+ "description": "Some integrations provide extended functionality via this field. For more information, refer to the integration documentation."
}
}
},
@@ -56,7 +56,7 @@
},
"data": {
"name": "Data",
- "description": "Some integrations provide extended functionality. For information on how to use _data_, refer to the integration documentation.."
+ "description": "Some integrations provide extended functionality via this field. For more information, refer to the integration documentation."
}
}
}
diff --git a/homeassistant/components/notion/binary_sensor.py b/homeassistant/components/notion/binary_sensor.py
index 8c57310752a..5552305e867 100644
--- a/homeassistant/components/notion/binary_sensor.py
+++ b/homeassistant/components/notion/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
@@ -107,7 +107,9 @@ BINARY_SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Notion sensors based on a config entry."""
coordinator: NotionDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/notion/coordinator.py b/homeassistant/components/notion/coordinator.py
index c3fd23abc84..d77bfa95f47 100644
--- a/homeassistant/components/notion/coordinator.py
+++ b/homeassistant/components/notion/coordinator.py
@@ -117,16 +117,16 @@ class NotionDataUpdateCoordinator(DataUpdateCoordinator[NotionData]):
super().__init__(
hass,
LOGGER,
+ config_entry=entry,
name=entry.data[CONF_USERNAME],
update_interval=DEFAULT_SCAN_INTERVAL,
)
self._client = client
- self._entry = entry
async def _async_update_data(self) -> NotionData:
"""Fetch data from Notion."""
- data = NotionData(hass=self.hass, entry=self._entry)
+ data = NotionData(hass=self.hass, entry=self.config_entry)
try:
async with asyncio.TaskGroup() as tg:
diff --git a/homeassistant/components/notion/sensor.py b/homeassistant/components/notion/sensor.py
index fb853e65d7d..24496c8391a 100644
--- a/homeassistant/components/notion/sensor.py
+++ b/homeassistant/components/notion/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, SENSOR_MOLD, SENSOR_TEMPERATURE
from .coordinator import NotionDataUpdateCoordinator
@@ -42,7 +42,9 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Notion sensors based on a config entry."""
coordinator: NotionDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/nuheat/climate.py b/homeassistant/components/nuheat/climate.py
index 8248c1b9b82..376a07ddb7b 100644
--- a/homeassistant/components/nuheat/climate.py
+++ b/homeassistant/components/nuheat/climate.py
@@ -23,7 +23,7 @@ from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import event as event_helper
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER, NUHEAT_API_STATE_SHIFT_DELAY
@@ -55,7 +55,7 @@ SCHEDULE_MODE_TO_PRESET_MODE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the NuHeat thermostat(s)."""
thermostat, coordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/nuki/__init__.py b/homeassistant/components/nuki/__init__.py
index 4f3f56f7f03..5c02b6e972e 100644
--- a/homeassistant/components/nuki/__init__.py
+++ b/homeassistant/components/nuki/__init__.py
@@ -222,7 +222,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _stop_nuki)
)
- coordinator = NukiCoordinator(hass, bridge, locks, openers)
+ coordinator = NukiCoordinator(hass, entry, bridge, locks, openers)
hass.data[DOMAIN][entry.entry_id] = NukiEntryData(
coordinator=coordinator,
bridge=bridge,
diff --git a/homeassistant/components/nuki/binary_sensor.py b/homeassistant/components/nuki/binary_sensor.py
index 8269c43813e..2785c46ca17 100644
--- a/homeassistant/components/nuki/binary_sensor.py
+++ b/homeassistant/components/nuki/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import NukiEntryData
from .const import DOMAIN as NUKI_DOMAIN
@@ -20,7 +20,9 @@ from .entity import NukiEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nuki binary sensors."""
entry_data: NukiEntryData = hass.data[NUKI_DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/nuki/coordinator.py b/homeassistant/components/nuki/coordinator.py
index 114b4aee4c9..cccff99e397 100644
--- a/homeassistant/components/nuki/coordinator.py
+++ b/homeassistant/components/nuki/coordinator.py
@@ -12,6 +12,7 @@ from pynuki.bridge import InvalidCredentialsException
from pynuki.device import NukiDevice
from requests.exceptions import RequestException
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
@@ -28,9 +29,12 @@ UPDATE_INTERVAL = timedelta(seconds=30)
class NukiCoordinator(DataUpdateCoordinator[None]):
"""Data Update Coordinator for the Nuki integration."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
bridge: NukiBridge,
locks: list[NukiLock],
openers: list[NukiOpener],
@@ -39,9 +43,8 @@ class NukiCoordinator(DataUpdateCoordinator[None]):
super().__init__(
hass,
_LOGGER,
- # Name of the data. For logging purposes.
+ config_entry=config_entry,
name="nuki devices",
- # Polling interval. Will only be polled if there are subscribers.
update_interval=UPDATE_INTERVAL,
)
self.bridge = bridge
diff --git a/homeassistant/components/nuki/lock.py b/homeassistant/components/nuki/lock.py
index a2bf7559fc4..3cc972d3555 100644
--- a/homeassistant/components/nuki/lock.py
+++ b/homeassistant/components/nuki/lock.py
@@ -15,7 +15,7 @@ from homeassistant.components.lock import LockEntity, LockEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import NukiEntryData
from .const import ATTR_ENABLE, ATTR_UNLATCH, DOMAIN as NUKI_DOMAIN, ERROR_STATES
@@ -24,7 +24,9 @@ from .helpers import CannotConnect
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nuki lock platform."""
entry_data: NukiEntryData = hass.data[NUKI_DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/nuki/sensor.py b/homeassistant/components/nuki/sensor.py
index d89202ac7d7..4f3890a10cf 100644
--- a/homeassistant/components/nuki/sensor.py
+++ b/homeassistant/components/nuki/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import NukiEntryData
from .const import DOMAIN as NUKI_DOMAIN
@@ -16,7 +16,9 @@ from .entity import NukiEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Nuki lock sensor."""
entry_data: NukiEntryData = hass.data[NUKI_DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/nuki/strings.json b/homeassistant/components/nuki/strings.json
index beac3cb7f74..daf47bc7de1 100644
--- a/homeassistant/components/nuki/strings.json
+++ b/homeassistant/components/nuki/strings.json
@@ -58,12 +58,12 @@
},
"services": {
"lock_n_go": {
- "name": "Lock 'n' go",
- "description": "Nuki Lock 'n' Go.",
+ "name": "Lock 'n' Go",
+ "description": "Unlocks the door, waits a few seconds then re-locks. The wait period can be customized through the app.",
"fields": {
"unlatch": {
"name": "Unlatch",
- "description": "Whether to unlatch the lock."
+ "description": "Whether to also unlatch the door when unlocking it."
}
}
},
diff --git a/homeassistant/components/number/const.py b/homeassistant/components/number/const.py
index bdde3a4567e..f44a510b1c0 100644
--- a/homeassistant/components/number/const.py
+++ b/homeassistant/components/number/const.py
@@ -11,6 +11,7 @@ from homeassistant.const import (
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
CONCENTRATION_PARTS_PER_BILLION,
CONCENTRATION_PARTS_PER_MILLION,
+ DEGREE,
LIGHT_LUX,
PERCENTAGE,
SIGNAL_STRENGTH_DECIBELS,
@@ -158,7 +159,7 @@ class NumberDeviceClass(StrEnum):
DURATION = "duration"
"""Fixed duration.
- Unit of measurement: `d`, `h`, `min`, `s`, `ms`
+ Unit of measurement: `d`, `h`, `min`, `s`, `ms`, `µs`
"""
ENERGY = "energy"
@@ -424,6 +425,12 @@ class NumberDeviceClass(StrEnum):
- USCS / imperial: `oz`, `lb`
"""
+ WIND_DIRECTION = "wind_direction"
+ """Wind direction.
+
+ Unit of measurement: `°`
+ """
+
WIND_SPEED = "wind_speed"
"""Wind speed.
@@ -455,6 +462,7 @@ DEVICE_CLASS_UNITS: dict[NumberDeviceClass, set[type[StrEnum] | str | None]] = {
UnitOfTime.MINUTES,
UnitOfTime.SECONDS,
UnitOfTime.MILLISECONDS,
+ UnitOfTime.MICROSECONDS,
},
NumberDeviceClass.ENERGY: set(UnitOfEnergy),
NumberDeviceClass.ENERGY_DISTANCE: set(UnitOfEnergyDistance),
@@ -479,6 +487,7 @@ DEVICE_CLASS_UNITS: dict[NumberDeviceClass, set[type[StrEnum] | str | None]] = {
NumberDeviceClass.PM25: {CONCENTRATION_MICROGRAMS_PER_CUBIC_METER},
NumberDeviceClass.POWER_FACTOR: {PERCENTAGE, None},
NumberDeviceClass.POWER: {
+ UnitOfPower.MILLIWATT,
UnitOfPower.WATT,
UnitOfPower.KILO_WATT,
UnitOfPower.MEGA_WATT,
@@ -516,6 +525,7 @@ DEVICE_CLASS_UNITS: dict[NumberDeviceClass, set[type[StrEnum] | str | None]] = {
UnitOfVolume.LITERS,
},
NumberDeviceClass.WEIGHT: set(UnitOfMass),
+ NumberDeviceClass.WIND_DIRECTION: {DEGREE},
NumberDeviceClass.WIND_SPEED: set(UnitOfSpeed),
}
diff --git a/homeassistant/components/number/icons.json b/homeassistant/components/number/icons.json
index 636fa0a7751..49103f5cd41 100644
--- a/homeassistant/components/number/icons.json
+++ b/homeassistant/components/number/icons.json
@@ -150,6 +150,9 @@
"weight": {
"default": "mdi:weight"
},
+ "wind_direction": {
+ "default": "mdi:compass-rose"
+ },
"wind_speed": {
"default": "mdi:weather-windy"
}
diff --git a/homeassistant/components/number/services.yaml b/homeassistant/components/number/services.yaml
index dcbb955d739..6a7083a7613 100644
--- a/homeassistant/components/number/services.yaml
+++ b/homeassistant/components/number/services.yaml
@@ -7,5 +7,6 @@ set_value:
fields:
value:
example: 42
+ required: true
selector:
text:
diff --git a/homeassistant/components/number/strings.json b/homeassistant/components/number/strings.json
index cc77d224d72..993120ef3ad 100644
--- a/homeassistant/components/number/strings.json
+++ b/homeassistant/components/number/strings.json
@@ -169,6 +169,9 @@
"weight": {
"name": "[%key:component::sensor::entity_component::weight::name%]"
},
+ "wind_direction": {
+ "name": "[%key:component::sensor::entity_component::wind_direction::name%]"
+ },
"wind_speed": {
"name": "[%key:component::sensor::entity_component::wind_speed::name%]"
}
diff --git a/homeassistant/components/nut/__init__.py b/homeassistant/components/nut/__init__.py
index 169dbbbff5d..5b188868819 100644
--- a/homeassistant/components/nut/__init__.py
+++ b/homeassistant/components/nut/__init__.py
@@ -103,7 +103,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NutConfigEntry) -> bool:
)
status = coordinator.data
- _LOGGER.debug("NUT Sensors Available: %s", status)
+ _LOGGER.debug("NUT Sensors Available: %s", status if status else None)
entry.async_on_unload(entry.add_update_listener(_async_update_listener))
unique_id = _unique_id_from_status(status)
@@ -111,14 +111,34 @@ async def async_setup_entry(hass: HomeAssistant, entry: NutConfigEntry) -> bool:
unique_id = entry.entry_id
if username is not None and password is not None:
+ # Dynamically add outlet integration commands
+ additional_integration_commands = set()
+ if (num_outlets := status.get("outlet.count")) is not None:
+ for outlet_num in range(1, int(num_outlets) + 1):
+ outlet_num_str: str = str(outlet_num)
+ additional_integration_commands |= {
+ f"outlet.{outlet_num_str}.load.cycle",
+ f"outlet.{outlet_num_str}.load.on",
+ f"outlet.{outlet_num_str}.load.off",
+ }
+
+ valid_integration_commands = (
+ INTEGRATION_SUPPORTED_COMMANDS | additional_integration_commands
+ )
+
user_available_commands = {
- device_supported_command
- for device_supported_command in await data.async_list_commands() or {}
- if device_supported_command in INTEGRATION_SUPPORTED_COMMANDS
+ device_command
+ for device_command in await data.async_list_commands() or {}
+ if device_command in valid_integration_commands
}
else:
user_available_commands = set()
+ _LOGGER.debug(
+ "NUT Commands Available: %s",
+ user_available_commands if user_available_commands else None,
+ )
+
entry.runtime_data = NutRuntimeData(
coordinator, data, unique_id, user_available_commands
)
@@ -240,6 +260,7 @@ class PyNUTData:
self._client = AIONUTClient(self._host, port, username, password, 5, persistent)
self.ups_list: dict[str, str] | None = None
+ self.device_name: str | None = None
self._status: dict[str, str] | None = None
self._device_info: NUTDeviceInfo | None = None
@@ -250,7 +271,7 @@ class PyNUTData:
@property
def name(self) -> str:
- """Return the name of the ups."""
+ """Return the name of the NUT device."""
return self._alias or f"Nut-{self._host}"
@property
@@ -294,6 +315,8 @@ class PyNUTData:
self._status = await self._async_get_status()
if self._device_info is None:
self._device_info = self._get_device_info()
+ if self.device_name is None:
+ self.device_name = self.name.title()
return self._status
async def async_run_command(self, command_name: str) -> None:
diff --git a/homeassistant/components/nut/button.py b/homeassistant/components/nut/button.py
new file mode 100644
index 00000000000..436f06b44d7
--- /dev/null
+++ b/homeassistant/components/nut/button.py
@@ -0,0 +1,65 @@
+"""Provides a switch for switchable NUT outlets."""
+
+from __future__ import annotations
+
+import logging
+
+from homeassistant.components.button import (
+ ButtonDeviceClass,
+ ButtonEntity,
+ ButtonEntityDescription,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import NutConfigEntry
+from .entity import NUTBaseEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: NutConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the NUT buttons."""
+ pynut_data = config_entry.runtime_data
+ coordinator = pynut_data.coordinator
+ status = coordinator.data
+
+ # Dynamically add outlet button types
+ if (num_outlets := status.get("outlet.count")) is None:
+ return
+
+ data = pynut_data.data
+ unique_id = pynut_data.unique_id
+ valid_button_types: dict[str, ButtonEntityDescription] = {}
+ for outlet_num in range(1, int(num_outlets) + 1):
+ outlet_num_str = str(outlet_num)
+ outlet_name: str = status.get(f"outlet.{outlet_num_str}.name") or outlet_num_str
+ valid_button_types |= {
+ f"outlet.{outlet_num_str}.load.cycle": ButtonEntityDescription(
+ key=f"outlet.{outlet_num_str}.load.cycle",
+ translation_key="outlet_number_load_cycle",
+ translation_placeholders={"outlet_name": outlet_name},
+ device_class=ButtonDeviceClass.RESTART,
+ entity_registry_enabled_default=True,
+ ),
+ }
+
+ async_add_entities(
+ NUTButton(coordinator, description, data, unique_id)
+ for button_id, description in valid_button_types.items()
+ if button_id in pynut_data.user_available_commands
+ )
+
+
+class NUTButton(NUTBaseEntity, ButtonEntity):
+ """Representation of a button entity for NUT."""
+
+ async def async_press(self) -> None:
+ """Press the button."""
+ name_list = self.entity_description.key.split(".")
+ command_name = f"{name_list[0]}.{name_list[1]}.load.cycle"
+ await self.pynut_data.async_run_command(command_name)
diff --git a/homeassistant/components/nut/const.py b/homeassistant/components/nut/const.py
index 6db40a910a0..d741d8e95f9 100644
--- a/homeassistant/components/nut/const.py
+++ b/homeassistant/components/nut/const.py
@@ -6,7 +6,11 @@ from homeassistant.const import Platform
DOMAIN = "nut"
-PLATFORMS = [Platform.SENSOR]
+PLATFORMS = [
+ Platform.BUTTON,
+ Platform.SENSOR,
+ Platform.SWITCH,
+]
DEFAULT_NAME = "NUT UPS"
DEFAULT_HOST = "localhost"
diff --git a/homeassistant/components/nut/diagnostics.py b/homeassistant/components/nut/diagnostics.py
index 532e4ece76b..ec59fa65c22 100644
--- a/homeassistant/components/nut/diagnostics.py
+++ b/homeassistant/components/nut/diagnostics.py
@@ -39,8 +39,8 @@ async def async_get_config_entry_diagnostics(
hass_device = device_registry.async_get_device(
identifiers={(DOMAIN, hass_data.unique_id)}
)
- if not hass_device:
- return data
+ # Device is always created
+ assert hass_device is not None
data["device"] = {
**attr.asdict(hass_device),
diff --git a/homeassistant/components/nut/entity.py b/homeassistant/components/nut/entity.py
new file mode 100644
index 00000000000..e6536d8aad6
--- /dev/null
+++ b/homeassistant/components/nut/entity.py
@@ -0,0 +1,67 @@
+"""Base entity for the NUT integration."""
+
+from __future__ import annotations
+
+from dataclasses import asdict
+from typing import cast
+
+from homeassistant.const import (
+ ATTR_MANUFACTURER,
+ ATTR_MODEL,
+ ATTR_SERIAL_NUMBER,
+ ATTR_SW_VERSION,
+)
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity import EntityDescription
+from homeassistant.helpers.update_coordinator import (
+ CoordinatorEntity,
+ DataUpdateCoordinator,
+)
+
+from . import PyNUTData
+from .const import DOMAIN
+
+NUT_DEV_INFO_TO_DEV_INFO: dict[str, str] = {
+ "manufacturer": ATTR_MANUFACTURER,
+ "model": ATTR_MODEL,
+ "firmware": ATTR_SW_VERSION,
+ "serial": ATTR_SERIAL_NUMBER,
+}
+
+
+class NUTBaseEntity(CoordinatorEntity[DataUpdateCoordinator]):
+ """NUT base entity."""
+
+ _attr_has_entity_name = True
+
+ def __init__(
+ self,
+ coordinator: DataUpdateCoordinator,
+ entity_description: EntityDescription,
+ data: PyNUTData,
+ unique_id: str,
+ ) -> None:
+ """Initialize the entity."""
+ super().__init__(coordinator)
+
+ self.entity_description = entity_description
+ self._attr_unique_id = f"{unique_id}_{entity_description.key}"
+
+ self.pynut_data = data
+ self._attr_device_info = DeviceInfo(
+ identifiers={(DOMAIN, unique_id)},
+ name=self.pynut_data.device_name,
+ )
+ self._attr_device_info.update(_get_nut_device_info(data))
+
+
+def _get_nut_device_info(data: PyNUTData) -> DeviceInfo:
+ """Return a DeviceInfo object filled with NUT device info."""
+ nut_dev_infos = asdict(data.device_info)
+ nut_infos = {
+ info_key: nut_dev_infos[nut_key]
+ for nut_key, info_key in NUT_DEV_INFO_TO_DEV_INFO.items()
+ if nut_dev_infos[nut_key] is not None
+ }
+
+ return cast(DeviceInfo, nut_infos)
diff --git a/homeassistant/components/nut/icons.json b/homeassistant/components/nut/icons.json
index e0f78d6400b..bfa4703d65e 100644
--- a/homeassistant/components/nut/icons.json
+++ b/homeassistant/components/nut/icons.json
@@ -1,6 +1,12 @@
{
"entity": {
"sensor": {
+ "ambient_humidity_status": {
+ "default": "mdi:information-outline"
+ },
+ "ambient_temperature_status": {
+ "default": "mdi:information-outline"
+ },
"battery_alarm_threshold": {
"default": "mdi:information-outline"
},
@@ -37,18 +43,48 @@
"input_bypass_phases": {
"default": "mdi:information-outline"
},
+ "input_current_status": {
+ "default": "mdi:information-outline"
+ },
"input_frequency_status": {
"default": "mdi:information-outline"
},
+ "input_load": {
+ "default": "mdi:gauge"
+ },
"input_phases": {
"default": "mdi:information-outline"
},
+ "input_power": {
+ "default": "mdi:gauge"
+ },
"input_sensitivity": {
"default": "mdi:information-outline"
},
"input_transfer_reason": {
"default": "mdi:information-outline"
},
+ "input_voltage_status": {
+ "default": "mdi:information-outline"
+ },
+ "outlet_number_current": {
+ "default": "mdi:gauge"
+ },
+ "outlet_number_current_status": {
+ "default": "mdi:information-outline"
+ },
+ "outlet_number_desc": {
+ "default": "mdi:information-outline"
+ },
+ "outlet_number_power": {
+ "default": "mdi:gauge"
+ },
+ "outlet_number_realpower": {
+ "default": "mdi:gauge"
+ },
+ "outlet_voltage": {
+ "default": "mdi:gauge"
+ },
"output_l1_power_percent": {
"default": "mdi:gauge"
},
@@ -115,6 +151,16 @@
"ups_watchdog_status": {
"default": "mdi:information-outline"
}
+ },
+ "button": {
+ "outlet_number_load_cycle": {
+ "default": "mdi:restart"
+ }
+ },
+ "switch": {
+ "outlet_number_load_poweronoff": {
+ "default": "mdi:power"
+ }
}
}
}
diff --git a/homeassistant/components/nut/manifest.json b/homeassistant/components/nut/manifest.json
index fb6c8561b25..1ee85a84caf 100644
--- a/homeassistant/components/nut/manifest.json
+++ b/homeassistant/components/nut/manifest.json
@@ -1,7 +1,7 @@
{
"domain": "nut",
"name": "Network UPS Tools (NUT)",
- "codeowners": ["@bdraco", "@ollo69", "@pestevez"],
+ "codeowners": ["@bdraco", "@ollo69", "@pestevez", "@tdfountain"],
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/nut",
"integration_type": "device",
diff --git a/homeassistant/components/nut/sensor.py b/homeassistant/components/nut/sensor.py
index bb702873052..80046c6ac22 100644
--- a/homeassistant/components/nut/sensor.py
+++ b/homeassistant/components/nut/sensor.py
@@ -1,10 +1,9 @@
-"""Provides a sensor to track various status aspects of a UPS."""
+"""Provides a sensor to track various status aspects of a NUT device."""
from __future__ import annotations
-from dataclasses import asdict
import logging
-from typing import Final, cast
+from typing import Final
from homeassistant.components.sensor import (
SensorDeviceClass,
@@ -13,10 +12,6 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.const import (
- ATTR_MANUFACTURER,
- ATTR_MODEL,
- ATTR_SERIAL_NUMBER,
- ATTR_SW_VERSION,
PERCENTAGE,
STATE_UNKNOWN,
EntityCategory,
@@ -29,25 +24,30 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.update_coordinator import (
- CoordinatorEntity,
- DataUpdateCoordinator,
-)
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import NutConfigEntry, PyNUTData
-from .const import DOMAIN, KEY_STATUS, KEY_STATUS_DISPLAY, STATE_TYPES
+from . import NutConfigEntry
+from .const import KEY_STATUS, KEY_STATUS_DISPLAY, STATE_TYPES
+from .entity import NUTBaseEntity
-NUT_DEV_INFO_TO_DEV_INFO: dict[str, str] = {
- "manufacturer": ATTR_MANUFACTURER,
- "model": ATTR_MODEL,
- "firmware": ATTR_SW_VERSION,
- "serial": ATTR_SERIAL_NUMBER,
+AMBIENT_PRESENT = "ambient.present"
+AMBIENT_SENSORS = {
+ "ambient.humidity",
+ "ambient.humidity.status",
+ "ambient.temperature",
+ "ambient.temperature.status",
}
+AMBIENT_THRESHOLD_STATUS_OPTIONS = [
+ "good",
+ "warning-low",
+ "critical-low",
+ "warning-high",
+ "critical-high",
+]
_LOGGER = logging.getLogger(__name__)
+
SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
"ups.status.display": SensorEntityDescription(
key="ups.status.display",
@@ -454,6 +454,12 @@ SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
),
+ "input.voltage.status": SensorEntityDescription(
+ key="input.voltage.status",
+ translation_key="input_voltage_status",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ entity_registry_enabled_default=False,
+ ),
"input.L1-N.voltage": SensorEntityDescription(
key="input.L1-N.voltage",
translation_key="input_l1_n_voltage",
@@ -662,6 +668,12 @@ SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
),
+ "input.current.status": SensorEntityDescription(
+ key="input.current.status",
+ translation_key="input_current_status",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ entity_registry_enabled_default=False,
+ ),
"input.L1.current": SensorEntityDescription(
key="input.L1.current",
translation_key="input_l1_current",
@@ -689,12 +701,26 @@ SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
),
+ "input.load": SensorEntityDescription(
+ key="input.load",
+ translation_key="input_load",
+ native_unit_of_measurement=PERCENTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
"input.phases": SensorEntityDescription(
key="input.phases",
translation_key="input_phases",
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
),
+ "input.power": SensorEntityDescription(
+ key="input.power",
+ translation_key="input_power",
+ device_class=SensorDeviceClass.APPARENT_POWER,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ entity_registry_enabled_default=False,
+ ),
"input.realpower": SensorEntityDescription(
key="input.realpower",
translation_key="input_realpower",
@@ -731,6 +757,13 @@ SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
),
+ "outlet.voltage": SensorEntityDescription(
+ key="outlet.voltage",
+ translation_key="outlet_voltage",
+ native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+ device_class=SensorDeviceClass.VOLTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
"output.power.nominal": SensorEntityDescription(
key="output.power.nominal",
translation_key="output_power_nominal",
@@ -930,6 +963,13 @@ SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
state_class=SensorStateClass.MEASUREMENT,
entity_category=EntityCategory.DIAGNOSTIC,
),
+ "ambient.humidity.status": SensorEntityDescription(
+ key="ambient.humidity.status",
+ translation_key="ambient_humidity_status",
+ device_class=SensorDeviceClass.ENUM,
+ options=AMBIENT_THRESHOLD_STATUS_OPTIONS,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
"ambient.temperature": SensorEntityDescription(
key="ambient.temperature",
translation_key="ambient_temperature",
@@ -938,6 +978,13 @@ SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
state_class=SensorStateClass.MEASUREMENT,
entity_category=EntityCategory.DIAGNOSTIC,
),
+ "ambient.temperature.status": SensorEntityDescription(
+ key="ambient.temperature.status",
+ translation_key="ambient_temperature_status",
+ device_class=SensorDeviceClass.ENUM,
+ options=AMBIENT_THRESHOLD_STATUS_OPTIONS,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
"watts": SensorEntityDescription(
key="watts",
translation_key="watts",
@@ -948,24 +995,13 @@ SENSOR_TYPES: Final[dict[str, SensorEntityDescription]] = {
}
-def _get_nut_device_info(data: PyNUTData) -> DeviceInfo:
- """Return a DeviceInfo object filled with NUT device info."""
- nut_dev_infos = asdict(data.device_info)
- nut_infos = {
- info_key: nut_dev_infos[nut_key]
- for nut_key, info_key in NUT_DEV_INFO_TO_DEV_INFO.items()
- if nut_dev_infos[nut_key] is not None
- }
-
- return cast(DeviceInfo, nut_infos)
-
-
async def async_setup_entry(
hass: HomeAssistant,
config_entry: NutConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the NUT sensors."""
+ valid_sensor_types: dict[str, SensorEntityDescription]
pynut_data = config_entry.runtime_data
coordinator = pynut_data.coordinator
@@ -973,16 +1009,75 @@ async def async_setup_entry(
unique_id = pynut_data.unique_id
status = coordinator.data
- resources = [sensor_id for sensor_id in SENSOR_TYPES if sensor_id in status]
+ # Dynamically add outlet sensors to valid sensors dictionary
+ if (num_outlets := status.get("outlet.count")) is not None:
+ additional_sensor_types: dict[str, SensorEntityDescription] = {}
+ for outlet_num in range(1, int(num_outlets) + 1):
+ outlet_num_str: str = str(outlet_num)
+ outlet_name: str = (
+ status.get(f"outlet.{outlet_num_str}.name") or outlet_num_str
+ )
+ additional_sensor_types |= {
+ f"outlet.{outlet_num_str}.current": SensorEntityDescription(
+ key=f"outlet.{outlet_num_str}.current",
+ translation_key="outlet_number_current",
+ translation_placeholders={"outlet_name": outlet_name},
+ native_unit_of_measurement=UnitOfElectricCurrent.AMPERE,
+ device_class=SensorDeviceClass.CURRENT,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ f"outlet.{outlet_num_str}.current_status": SensorEntityDescription(
+ key=f"outlet.{outlet_num_str}.current_status",
+ translation_key="outlet_number_current_status",
+ translation_placeholders={"outlet_name": outlet_name},
+ entity_category=EntityCategory.DIAGNOSTIC,
+ entity_registry_enabled_default=False,
+ ),
+ f"outlet.{outlet_num_str}.desc": SensorEntityDescription(
+ key=f"outlet.{outlet_num_str}.desc",
+ translation_key="outlet_number_desc",
+ translation_placeholders={"outlet_name": outlet_name},
+ ),
+ f"outlet.{outlet_num_str}.power": SensorEntityDescription(
+ key=f"outlet.{outlet_num_str}.power",
+ translation_key="outlet_number_power",
+ translation_placeholders={"outlet_name": outlet_name},
+ native_unit_of_measurement=UnitOfApparentPower.VOLT_AMPERE,
+ device_class=SensorDeviceClass.APPARENT_POWER,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ f"outlet.{outlet_num_str}.realpower": SensorEntityDescription(
+ key=f"outlet.{outlet_num_str}.realpower",
+ translation_key="outlet_number_realpower",
+ translation_placeholders={"outlet_name": outlet_name},
+ native_unit_of_measurement=UnitOfPower.WATT,
+ device_class=SensorDeviceClass.POWER,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ }
+
+ valid_sensor_types = {**SENSOR_TYPES, **additional_sensor_types}
+ else:
+ valid_sensor_types = SENSOR_TYPES
+
+ # If device reports ambient sensors are not present, then remove
+ has_ambient_sensors: bool = status.get(AMBIENT_PRESENT) != "no"
+ resources = [
+ sensor_id
+ for sensor_id in valid_sensor_types
+ if sensor_id in status
+ and (has_ambient_sensors or sensor_id not in AMBIENT_SENSORS)
+ ]
+
# Display status is a special case that falls back to the status value
# of the UPS instead.
- if KEY_STATUS in resources:
+ if KEY_STATUS in status:
resources.append(KEY_STATUS_DISPLAY)
async_add_entities(
NUTSensor(
coordinator,
- SENSOR_TYPES[sensor_type],
+ valid_sensor_types[sensor_type],
data,
unique_id,
)
@@ -990,33 +1085,12 @@ async def async_setup_entry(
)
-class NUTSensor(CoordinatorEntity[DataUpdateCoordinator[dict[str, str]]], SensorEntity):
+class NUTSensor(NUTBaseEntity, SensorEntity):
"""Representation of a sensor entity for NUT status values."""
- _attr_has_entity_name = True
-
- def __init__(
- self,
- coordinator: DataUpdateCoordinator[dict[str, str]],
- sensor_description: SensorEntityDescription,
- data: PyNUTData,
- unique_id: str,
- ) -> None:
- """Initialize the sensor."""
- super().__init__(coordinator)
- self.entity_description = sensor_description
-
- device_name = data.name.title()
- self._attr_unique_id = f"{unique_id}_{sensor_description.key}"
- self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, unique_id)},
- name=device_name,
- )
- self._attr_device_info.update(_get_nut_device_info(data))
-
@property
def native_value(self) -> str | None:
- """Return entity state from ups."""
+ """Return entity state from NUT device."""
status = self.coordinator.data
if self.entity_description.key == KEY_STATUS_DISPLAY:
return _format_display_state(status)
diff --git a/homeassistant/components/nut/strings.json b/homeassistant/components/nut/strings.json
index 83b8d340dc1..3ac5f23a0c1 100644
--- a/homeassistant/components/nut/strings.json
+++ b/homeassistant/components/nut/strings.json
@@ -80,7 +80,9 @@
"entity": {
"sensor": {
"ambient_humidity": { "name": "Ambient humidity" },
+ "ambient_humidity_status": { "name": "Ambient humidity status" },
"ambient_temperature": { "name": "Ambient temperature" },
+ "ambient_temperature_status": { "name": "Ambient temperature status" },
"battery_alarm_threshold": { "name": "Battery alarm threshold" },
"battery_capacity": { "name": "Battery capacity" },
"battery_charge": { "name": "Battery charge" },
@@ -124,6 +126,7 @@
"name": "Input bypass L3 real power"
},
"input_current": { "name": "Input current" },
+ "input_current_status": { "name": "Input current status" },
"input_l1_current": { "name": "Input L1 current" },
"input_l2_current": { "name": "Input L2 current" },
"input_l3_current": { "name": "Input L3 current" },
@@ -134,19 +137,30 @@
"input_l2_frequency": { "name": "Input L2 line frequency" },
"input_l3_frequency": { "name": "Input L3 line frequency" },
"input_phases": { "name": "Input phases" },
+ "input_power": { "name": "Input power" },
"input_realpower": { "name": "Input real power" },
"input_l1_realpower": { "name": "Input L1 real power" },
"input_l2_realpower": { "name": "Input L2 real power" },
"input_l3_realpower": { "name": "Input L3 real power" },
+ "input_load": { "name": "Input load" },
"input_sensitivity": { "name": "Input power sensitivity" },
"input_transfer_high": { "name": "High voltage transfer" },
"input_transfer_low": { "name": "Low voltage transfer" },
"input_transfer_reason": { "name": "Voltage transfer reason" },
"input_voltage": { "name": "Input voltage" },
"input_voltage_nominal": { "name": "Nominal input voltage" },
+ "input_voltage_status": { "name": "Input voltage status" },
"input_l1_n_voltage": { "name": "Input L1 voltage" },
"input_l2_n_voltage": { "name": "Input L2 voltage" },
"input_l3_n_voltage": { "name": "Input L3 voltage" },
+ "outlet_number_current": { "name": "Outlet {outlet_name} current" },
+ "outlet_number_current_status": {
+ "name": "Outlet {outlet_name} current status"
+ },
+ "outlet_number_desc": { "name": "Outlet {outlet_name} description" },
+ "outlet_number_power": { "name": "Outlet {outlet_name} power" },
+ "outlet_number_realpower": { "name": "Outlet {outlet_name} real power" },
+ "outlet_voltage": { "name": "Outlet voltage" },
"output_current": { "name": "Output current" },
"output_current_nominal": { "name": "Nominal output current" },
"output_l1_current": { "name": "Output L1 current" },
@@ -203,6 +217,12 @@
"ups_type": { "name": "UPS type" },
"ups_watchdog_status": { "name": "Watchdog status" },
"watts": { "name": "Watts" }
+ },
+ "button": {
+ "outlet_number_load_cycle": { "name": "Power cycle outlet {outlet_name}" }
+ },
+ "switch": {
+ "outlet_number_load_poweronoff": { "name": "Power outlet {outlet_name}" }
}
}
}
diff --git a/homeassistant/components/nut/switch.py b/homeassistant/components/nut/switch.py
new file mode 100644
index 00000000000..3ab8d0ec60a
--- /dev/null
+++ b/homeassistant/components/nut/switch.py
@@ -0,0 +1,88 @@
+"""Provides a switch for switchable NUT outlets."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from homeassistant.components.switch import (
+ SwitchDeviceClass,
+ SwitchEntity,
+ SwitchEntityDescription,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import NutConfigEntry
+from .entity import NUTBaseEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: NutConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the NUT switches."""
+ pynut_data = config_entry.runtime_data
+ coordinator = pynut_data.coordinator
+ status = coordinator.data
+
+ # Dynamically add outlet switch types
+ if (num_outlets := status.get("outlet.count")) is None:
+ return
+
+ data = pynut_data.data
+ unique_id = pynut_data.unique_id
+ user_available_commands = pynut_data.user_available_commands
+ switch_descriptions = [
+ SwitchEntityDescription(
+ key=f"outlet.{outlet_num!s}.load.poweronoff",
+ translation_key="outlet_number_load_poweronoff",
+ translation_placeholders={
+ "outlet_name": status.get(f"outlet.{outlet_num!s}.name")
+ or str(outlet_num)
+ },
+ device_class=SwitchDeviceClass.OUTLET,
+ entity_registry_enabled_default=True,
+ )
+ for outlet_num in range(1, int(num_outlets) + 1)
+ if (
+ status.get(f"outlet.{outlet_num!s}.switchable") == "yes"
+ and f"outlet.{outlet_num!s}.load.on" in user_available_commands
+ and f"outlet.{outlet_num!s}.load.off" in user_available_commands
+ )
+ ]
+
+ async_add_entities(
+ NUTSwitch(coordinator, description, data, unique_id)
+ for description in switch_descriptions
+ )
+
+
+class NUTSwitch(NUTBaseEntity, SwitchEntity):
+ """Representation of a switch entity for NUT status values."""
+
+ @property
+ def is_on(self) -> bool | None:
+ """Return the state of the switch."""
+ status = self.coordinator.data
+ outlet, outlet_num_str = self.entity_description.key.split(".", 2)[:2]
+ if (state := status.get(f"{outlet}.{outlet_num_str}.status")) is None:
+ return None
+ return bool(state == "on")
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn on the device."""
+
+ outlet, outlet_num_str = self.entity_description.key.split(".", 2)[:2]
+ command_name = f"{outlet}.{outlet_num_str}.load.on"
+ await self.pynut_data.async_run_command(command_name)
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn off the device."""
+
+ outlet, outlet_num_str = self.entity_description.key.split(".", 2)[:2]
+ command_name = f"{outlet}.{outlet_num_str}.load.off"
+ await self.pynut_data.async_run_command(command_name)
diff --git a/homeassistant/components/nws/__init__.py b/homeassistant/components/nws/__init__.py
index c700476ed3d..633619bcf05 100644
--- a/homeassistant/components/nws/__init__.py
+++ b/homeassistant/components/nws/__init__.py
@@ -101,10 +101,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NWSConfigEntry) -> bool:
return update_forecast_hourly
- coordinator_observation = NWSObservationDataUpdateCoordinator(
- hass,
- nws_data,
- )
+ coordinator_observation = NWSObservationDataUpdateCoordinator(hass, entry, nws_data)
# Don't use retries in setup
coordinator_forecast = TimestampDataUpdateCoordinator(
diff --git a/homeassistant/components/nws/coordinator.py b/homeassistant/components/nws/coordinator.py
index 104b1812c67..4e6560947e8 100644
--- a/homeassistant/components/nws/coordinator.py
+++ b/homeassistant/components/nws/coordinator.py
@@ -1,7 +1,10 @@
"""The NWS coordinator."""
+from __future__ import annotations
+
from datetime import datetime
import logging
+from typing import TYPE_CHECKING
from aiohttp import ClientResponseError
from pynws import NwsNoDataError, SimpleNWS, call_with_retry
@@ -14,6 +17,9 @@ from homeassistant.helpers.update_coordinator import (
)
from homeassistant.util.dt import utcnow
+if TYPE_CHECKING:
+ from . import NWSConfigEntry
+
from .const import (
DEBOUNCE_TIME,
DEFAULT_SCAN_INTERVAL,
@@ -29,9 +35,12 @@ _LOGGER = logging.getLogger(__name__)
class NWSObservationDataUpdateCoordinator(TimestampDataUpdateCoordinator[None]):
"""Class to manage fetching NWS observation data."""
+ config_entry: NWSConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: NWSConfigEntry,
nws: SimpleNWS,
) -> None:
"""Initialize."""
@@ -42,6 +51,7 @@ class NWSObservationDataUpdateCoordinator(TimestampDataUpdateCoordinator[None]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"NWS observation station {nws.station}",
update_interval=DEFAULT_SCAN_INTERVAL,
request_refresh_debouncer=debounce.Debouncer(
diff --git a/homeassistant/components/nws/sensor.py b/homeassistant/components/nws/sensor.py
index d1992056d47..4cfb3b85e0f 100644
--- a/homeassistant/components/nws/sensor.py
+++ b/homeassistant/components/nws/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
TimestampDataUpdateCoordinator,
@@ -114,6 +114,7 @@ SENSOR_TYPES: tuple[NWSSensorEntityDescription, ...] = (
icon="mdi:compass-rose",
native_unit_of_measurement=DEGREE,
unit_convert=DEGREE,
+ device_class=SensorDeviceClass.WIND_DIRECTION,
),
NWSSensorEntityDescription(
key="barometricPressure",
@@ -148,7 +149,9 @@ SENSOR_TYPES: tuple[NWSSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: NWSConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NWSConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the NWS weather platform."""
nws_data = entry.runtime_data
diff --git a/homeassistant/components/nws/strings.json b/homeassistant/components/nws/strings.json
index c9ee8349631..72b6a2c86b6 100644
--- a/homeassistant/components/nws/strings.json
+++ b/homeassistant/components/nws/strings.json
@@ -2,7 +2,7 @@
"config": {
"step": {
"user": {
- "description": "If a METAR station code is not specified, the latitude and longitude will be used to find the closest station. For now, an API Key can be anything. It is recommended to use a valid email address.",
+ "description": "If a METAR station code is not specified, the latitude and longitude will be used to find the closest station. For now, the API key can be anything. It is recommended to use a valid email address.",
"title": "Connect to the National Weather Service",
"data": {
"api_key": "[%key:common::config_flow::data::api_key%]",
@@ -30,12 +30,12 @@
},
"services": {
"get_forecasts_extra": {
- "name": "Get extra forecasts data.",
- "description": "Get extra data for weather forecasts.",
+ "name": "Get extra forecasts data",
+ "description": "Retrieves extra data for weather forecasts.",
"fields": {
"type": {
"name": "Forecast type",
- "description": "Forecast type: hourly or twice_daily."
+ "description": "The scope of the weather forecast."
}
}
}
diff --git a/homeassistant/components/nws/weather.py b/homeassistant/components/nws/weather.py
index d34a5abe8af..c90c67edcb7 100644
--- a/homeassistant/components/nws/weather.py
+++ b/homeassistant/components/nws/weather.py
@@ -40,7 +40,7 @@ from homeassistant.core import (
callback,
)
from homeassistant.helpers import entity_platform, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import TimestampDataUpdateCoordinator
from homeassistant.util.json import JsonValueType
from homeassistant.util.unit_conversion import SpeedConverter, TemperatureConverter
@@ -87,7 +87,9 @@ def convert_condition(time: str, weather: tuple[tuple[str, int | None], ...]) ->
async def async_setup_entry(
- hass: HomeAssistant, entry: NWSConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: NWSConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the NWS weather platform."""
entity_registry = er.async_get(hass)
diff --git a/homeassistant/components/nyt_games/__init__.py b/homeassistant/components/nyt_games/__init__.py
index 94dc22fe89e..d1c6ca5c2a4 100644
--- a/homeassistant/components/nyt_games/__init__.py
+++ b/homeassistant/components/nyt_games/__init__.py
@@ -4,21 +4,17 @@ from __future__ import annotations
from nyt_games import NYTGamesClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TOKEN, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_create_clientsession
-from .coordinator import NYTGamesCoordinator
+from .coordinator import NYTGamesConfigEntry, NYTGamesCoordinator
PLATFORMS: list[Platform] = [
Platform.SENSOR,
]
-type NYTGamesConfigEntry = ConfigEntry[NYTGamesCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: NYTGamesConfigEntry) -> bool:
"""Set up NYTGames from a config entry."""
@@ -26,7 +22,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NYTGamesConfigEntry) ->
entry.data[CONF_TOKEN], session=async_create_clientsession(hass)
)
- coordinator = NYTGamesCoordinator(hass, client)
+ coordinator = NYTGamesCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/nyt_games/coordinator.py b/homeassistant/components/nyt_games/coordinator.py
index 5e88a5dd92a..ae9ea4f03a0 100644
--- a/homeassistant/components/nyt_games/coordinator.py
+++ b/homeassistant/components/nyt_games/coordinator.py
@@ -4,18 +4,15 @@ from __future__ import annotations
from dataclasses import dataclass
from datetime import timedelta
-from typing import TYPE_CHECKING
from nyt_games import Connections, NYTGamesClient, NYTGamesError, SpellingBee, Wordle
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import LOGGER
-if TYPE_CHECKING:
- from . import NYTGamesConfigEntry
-
@dataclass
class NYTGamesData:
@@ -26,16 +23,25 @@ class NYTGamesData:
connections: Connections | None
+type NYTGamesConfigEntry = ConfigEntry[NYTGamesCoordinator]
+
+
class NYTGamesCoordinator(DataUpdateCoordinator[NYTGamesData]):
"""Class to manage fetching NYT Games data."""
config_entry: NYTGamesConfigEntry
- def __init__(self, hass: HomeAssistant, client: NYTGamesClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: NYTGamesConfigEntry,
+ client: NYTGamesClient,
+ ) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
logger=LOGGER,
+ config_entry=config_entry,
name="NYT Games",
update_interval=timedelta(minutes=15),
)
diff --git a/homeassistant/components/nyt_games/sensor.py b/homeassistant/components/nyt_games/sensor.py
index 01b2db4620b..5009eafd85a 100644
--- a/homeassistant/components/nyt_games/sensor.py
+++ b/homeassistant/components/nyt_games/sensor.py
@@ -14,11 +14,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import NYTGamesConfigEntry
-from .coordinator import NYTGamesCoordinator
+from .coordinator import NYTGamesConfigEntry, NYTGamesCoordinator
from .entity import ConnectionsEntity, SpellingBeeEntity, WordleEntity
@@ -147,7 +146,7 @@ CONNECTIONS_SENSORS: tuple[NYTGamesConnectionsSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: NYTGamesConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up NYT Games sensor entities based on a config entry."""
diff --git a/homeassistant/components/nzbget/__init__.py b/homeassistant/components/nzbget/__init__.py
index 84456c4c006..e9e5856d524 100644
--- a/homeassistant/components/nzbget/__init__.py
+++ b/homeassistant/components/nzbget/__init__.py
@@ -31,10 +31,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up NZBGet from a config entry."""
hass.data.setdefault(DOMAIN, {})
- coordinator = NZBGetDataUpdateCoordinator(
- hass,
- config=entry.data,
- )
+ coordinator = NZBGetDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/nzbget/coordinator.py b/homeassistant/components/nzbget/coordinator.py
index cf9625ce4ec..9e6b06da760 100644
--- a/homeassistant/components/nzbget/coordinator.py
+++ b/homeassistant/components/nzbget/coordinator.py
@@ -1,13 +1,12 @@
"""Provides the NZBGet DataUpdateCoordinator."""
import asyncio
-from collections.abc import Mapping
from datetime import timedelta
import logging
-from typing import Any
from pynzbgetapi import NZBGetAPI, NZBGetAPIException
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_HOST,
CONF_PASSWORD,
@@ -27,27 +26,32 @@ _LOGGER = logging.getLogger(__name__)
class NZBGetDataUpdateCoordinator(DataUpdateCoordinator):
"""Class to manage fetching NZBGet data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- *,
- config: Mapping[str, Any],
+ config_entry: ConfigEntry,
) -> None:
"""Initialize global NZBGet data updater."""
self.nzbget = NZBGetAPI(
- config[CONF_HOST],
- config.get(CONF_USERNAME),
- config.get(CONF_PASSWORD),
- config[CONF_SSL],
- config[CONF_VERIFY_SSL],
- config[CONF_PORT],
+ config_entry.data[CONF_HOST],
+ config_entry.data.get(CONF_USERNAME),
+ config_entry.data.get(CONF_PASSWORD),
+ config_entry.data[CONF_SSL],
+ config_entry.data[CONF_VERIFY_SSL],
+ config_entry.data[CONF_PORT],
)
self._completed_downloads_init = False
self._completed_downloads = set[tuple]()
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=timedelta(seconds=5)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=timedelta(seconds=5),
)
def _check_completed_downloads(self, history):
diff --git a/homeassistant/components/nzbget/sensor.py b/homeassistant/components/nzbget/sensor.py
index f6a4e4cc973..2328bf453f0 100644
--- a/homeassistant/components/nzbget/sensor.py
+++ b/homeassistant/components/nzbget/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, UnitOfDataRate, UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
@@ -93,7 +93,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up NZBGet sensor based on a config entry."""
coordinator: NZBGetDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
diff --git a/homeassistant/components/nzbget/switch.py b/homeassistant/components/nzbget/switch.py
index 552a1854902..0796f628507 100644
--- a/homeassistant/components/nzbget/switch.py
+++ b/homeassistant/components/nzbget/switch.py
@@ -8,7 +8,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_COORDINATOR, DOMAIN
from .coordinator import NZBGetDataUpdateCoordinator
@@ -18,7 +18,7 @@ from .entity import NZBGetEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up NZBGet sensor based on a config entry."""
coordinator: NZBGetDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
diff --git a/homeassistant/components/obihai/button.py b/homeassistant/components/obihai/button.py
index d1b924b4693..9cef92d3fce 100644
--- a/homeassistant/components/obihai/button.py
+++ b/homeassistant/components/obihai/button.py
@@ -27,7 +27,7 @@ BUTTON_DESCRIPTION = ButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: entity_platform.AddEntitiesCallback,
+ async_add_entities: entity_platform.AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Obihai sensor entries."""
diff --git a/homeassistant/components/obihai/sensor.py b/homeassistant/components/obihai/sensor.py
index c162bd6c559..ec29238201a 100644
--- a/homeassistant/components/obihai/sensor.py
+++ b/homeassistant/components/obihai/sensor.py
@@ -9,7 +9,7 @@ from requests.exceptions import RequestException
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .connectivity import ObihaiConnection
from .const import DOMAIN, LOGGER, OBIHAI
@@ -18,7 +18,9 @@ SCAN_INTERVAL = datetime.timedelta(seconds=5)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Obihai sensor entries."""
diff --git a/homeassistant/components/octoprint/binary_sensor.py b/homeassistant/components/octoprint/binary_sensor.py
index 10a637e5a3b..a20738de150 100644
--- a/homeassistant/components/octoprint/binary_sensor.py
+++ b/homeassistant/components/octoprint/binary_sensor.py
@@ -9,7 +9,7 @@ from pyoctoprintapi import OctoprintPrinterInfo
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import OctoprintDataUpdateCoordinator
@@ -19,7 +19,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the available OctoPrint binary sensors."""
coordinator: OctoprintDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/octoprint/button.py b/homeassistant/components/octoprint/button.py
index 2a2e5015303..3a128fcd7aa 100644
--- a/homeassistant/components/octoprint/button.py
+++ b/homeassistant/components/octoprint/button.py
@@ -6,7 +6,7 @@ from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import OctoprintDataUpdateCoordinator
@@ -16,7 +16,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Octoprint control buttons."""
coordinator: OctoprintDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/octoprint/camera.py b/homeassistant/components/octoprint/camera.py
index e6430c55fa2..37347539d5b 100644
--- a/homeassistant/components/octoprint/camera.py
+++ b/homeassistant/components/octoprint/camera.py
@@ -8,7 +8,7 @@ from homeassistant.components.mjpeg import MjpegCamera
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_VERIFY_SSL
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import OctoprintDataUpdateCoordinator
@@ -18,7 +18,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the available OctoPrint camera."""
coordinator: OctoprintDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/octoprint/coordinator.py b/homeassistant/components/octoprint/coordinator.py
index d4f8f652b80..bb006329ff1 100644
--- a/homeassistant/components/octoprint/coordinator.py
+++ b/homeassistant/components/octoprint/coordinator.py
@@ -39,10 +39,10 @@ class OctoprintDataUpdateCoordinator(DataUpdateCoordinator):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"octoprint-{config_entry.entry_id}",
update_interval=timedelta(seconds=interval),
)
- self.config_entry = config_entry
self._octoprint = octoprint
self._printer_offline = False
self.data = {"printer": None, "job": None, "last_read_time": None}
diff --git a/homeassistant/components/octoprint/sensor.py b/homeassistant/components/octoprint/sensor.py
index fb5f292d669..71db1d804c5 100644
--- a/homeassistant/components/octoprint/sensor.py
+++ b/homeassistant/components/octoprint/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import OctoprintDataUpdateCoordinator
@@ -38,7 +38,7 @@ def _is_printer_printing(printer: OctoprintPrinterInfo) -> bool:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the available OctoPrint binary sensors."""
coordinator: OctoprintDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/ogemray/__init__.py b/homeassistant/components/ogemray/__init__.py
new file mode 100644
index 00000000000..94e19234a6b
--- /dev/null
+++ b/homeassistant/components/ogemray/__init__.py
@@ -0,0 +1 @@
+"""Ogemray virtual integration."""
diff --git a/homeassistant/components/ogemray/manifest.json b/homeassistant/components/ogemray/manifest.json
new file mode 100644
index 00000000000..6a8eb315c7a
--- /dev/null
+++ b/homeassistant/components/ogemray/manifest.json
@@ -0,0 +1,6 @@
+{
+ "domain": "ogemray",
+ "name": "Ogemray",
+ "integration_type": "virtual",
+ "supported_by": "shelly"
+}
diff --git a/homeassistant/components/ohme/__init__.py b/homeassistant/components/ohme/__init__.py
index 8518e55c0a3..e3e252cbf8b 100644
--- a/homeassistant/components/ohme/__init__.py
+++ b/homeassistant/components/ohme/__init__.py
@@ -1,10 +1,7 @@
"""Set up ohme integration."""
-from dataclasses import dataclass
-
from ohme import ApiException, AuthException, OhmeApiClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -15,23 +12,14 @@ from .const import DOMAIN, PLATFORMS
from .coordinator import (
OhmeAdvancedSettingsCoordinator,
OhmeChargeSessionCoordinator,
+ OhmeConfigEntry,
OhmeDeviceInfoCoordinator,
+ OhmeRuntimeData,
)
from .services import async_setup_services
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
-type OhmeConfigEntry = ConfigEntry[OhmeRuntimeData]
-
-
-@dataclass()
-class OhmeRuntimeData:
- """Dataclass to hold ohme coordinators."""
-
- charge_session_coordinator: OhmeChargeSessionCoordinator
- advanced_settings_coordinator: OhmeAdvancedSettingsCoordinator
- device_info_coordinator: OhmeDeviceInfoCoordinator
-
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up Ohme integration."""
@@ -62,9 +50,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: OhmeConfigEntry) -> bool
) from e
coordinators = (
- OhmeChargeSessionCoordinator(hass, client),
- OhmeAdvancedSettingsCoordinator(hass, client),
- OhmeDeviceInfoCoordinator(hass, client),
+ OhmeChargeSessionCoordinator(hass, entry, client),
+ OhmeAdvancedSettingsCoordinator(hass, entry, client),
+ OhmeDeviceInfoCoordinator(hass, entry, client),
)
for coordinator in coordinators:
diff --git a/homeassistant/components/ohme/button.py b/homeassistant/components/ohme/button.py
index 0b0590428ce..6e942215c0f 100644
--- a/homeassistant/components/ohme/button.py
+++ b/homeassistant/components/ohme/button.py
@@ -10,10 +10,10 @@ from ohme import ApiException, ChargerStatus, OhmeApiClient
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import OhmeConfigEntry
from .const import DOMAIN
+from .coordinator import OhmeConfigEntry
from .entity import OhmeEntity, OhmeEntityDescription
PARALLEL_UPDATES = 1
@@ -40,7 +40,7 @@ BUTTON_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OhmeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up buttons."""
coordinator = config_entry.runtime_data.charge_session_coordinator
diff --git a/homeassistant/components/ohme/config_flow.py b/homeassistant/components/ohme/config_flow.py
index 748ea558983..1037c3a7c8b 100644
--- a/homeassistant/components/ohme/config_flow.py
+++ b/homeassistant/components/ohme/config_flow.py
@@ -99,6 +99,29 @@ class OhmeConfigFlow(ConfigFlow, domain=DOMAIN):
errors=errors,
)
+ async def async_step_reconfigure(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle re-configuration."""
+ errors: dict[str, str] = {}
+ reconfigure_entry = self._get_reconfigure_entry()
+ if user_input:
+ errors = await self._validate_account(
+ reconfigure_entry.data[CONF_EMAIL],
+ user_input[CONF_PASSWORD],
+ )
+ if not errors:
+ return self.async_update_reload_and_abort(
+ reconfigure_entry,
+ data_updates=user_input,
+ )
+ return self.async_show_form(
+ step_id="reconfigure",
+ data_schema=REAUTH_SCHEMA,
+ description_placeholders={"email": reconfigure_entry.data[CONF_EMAIL]},
+ errors=errors,
+ )
+
async def _validate_account(self, email: str, password: str) -> dict[str, str]:
"""Validate Ohme account and return dict of errors."""
errors: dict[str, str] = {}
diff --git a/homeassistant/components/ohme/coordinator.py b/homeassistant/components/ohme/coordinator.py
index 199eb7380a7..864b03e9a7c 100644
--- a/homeassistant/components/ohme/coordinator.py
+++ b/homeassistant/components/ohme/coordinator.py
@@ -1,11 +1,15 @@
"""Ohme coordinators."""
+from __future__ import annotations
+
from abc import abstractmethod
+from dataclasses import dataclass
from datetime import timedelta
import logging
from ohme import ApiException, OhmeApiClient
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -14,18 +18,34 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
+@dataclass()
+class OhmeRuntimeData:
+ """Dataclass to hold ohme coordinators."""
+
+ charge_session_coordinator: OhmeChargeSessionCoordinator
+ advanced_settings_coordinator: OhmeAdvancedSettingsCoordinator
+ device_info_coordinator: OhmeDeviceInfoCoordinator
+
+
+type OhmeConfigEntry = ConfigEntry[OhmeRuntimeData]
+
+
class OhmeBaseCoordinator(DataUpdateCoordinator[None]):
"""Base for all Ohme coordinators."""
+ config_entry: OhmeConfigEntry
client: OhmeApiClient
_default_update_interval: timedelta | None = timedelta(minutes=1)
coordinator_name: str = ""
- def __init__(self, hass: HomeAssistant, client: OhmeApiClient) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: OhmeConfigEntry, client: OhmeApiClient
+ ) -> None:
"""Initialise coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="",
update_interval=self._default_update_interval,
)
diff --git a/homeassistant/components/ohme/diagnostics.py b/homeassistant/components/ohme/diagnostics.py
new file mode 100644
index 00000000000..a955b3b76e2
--- /dev/null
+++ b/homeassistant/components/ohme/diagnostics.py
@@ -0,0 +1,24 @@
+"""Provides diagnostics for Ohme."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.core import HomeAssistant
+
+from .coordinator import OhmeConfigEntry
+
+
+async def async_get_config_entry_diagnostics(
+ hass: HomeAssistant, config_entry: OhmeConfigEntry
+) -> dict[str, Any]:
+ """Return diagnostics for Ohme."""
+ coordinators = config_entry.runtime_data
+ client = coordinators.charge_session_coordinator.client
+
+ return {
+ "device_info": client.device_info,
+ "vehicles": client.vehicles,
+ "ct_connected": client.ct_connected,
+ "cap_available": client.cap_available,
+ }
diff --git a/homeassistant/components/ohme/icons.json b/homeassistant/components/ohme/icons.json
index 7a27156b2fe..8613f2542c4 100644
--- a/homeassistant/components/ohme/icons.json
+++ b/homeassistant/components/ohme/icons.json
@@ -6,6 +6,9 @@
}
},
"number": {
+ "preconditioning_duration": {
+ "default": "mdi:fan-clock"
+ },
"target_percentage": {
"default": "mdi:battery-heart"
}
@@ -13,6 +16,9 @@
"select": {
"charge_mode": {
"default": "mdi:play-box"
+ },
+ "vehicle": {
+ "default": "mdi:car"
}
},
"sensor": {
@@ -28,6 +34,9 @@
},
"ct_current": {
"default": "mdi:gauge"
+ },
+ "slot_list": {
+ "default": "mdi:calendar-clock"
}
},
"switch": {
@@ -45,6 +54,9 @@
"state": {
"off": "mdi:sleep-off"
}
+ },
+ "price_cap": {
+ "default": "mdi:car-speed-limiter"
}
},
"time": {
@@ -56,6 +68,9 @@
"services": {
"list_charge_slots": {
"service": "mdi:clock-start"
+ },
+ "set_price_cap": {
+ "service": "mdi:car-speed-limiter"
}
}
}
diff --git a/homeassistant/components/ohme/manifest.json b/homeassistant/components/ohme/manifest.json
index 602c53ced7b..f0021808d92 100644
--- a/homeassistant/components/ohme/manifest.json
+++ b/homeassistant/components/ohme/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "device",
"iot_class": "cloud_polling",
"quality_scale": "silver",
- "requirements": ["ohme==1.2.8"]
+ "requirements": ["ohme==1.4.1"]
}
diff --git a/homeassistant/components/ohme/number.py b/homeassistant/components/ohme/number.py
index d618d4a873b..0c71bab009f 100644
--- a/homeassistant/components/ohme/number.py
+++ b/homeassistant/components/ohme/number.py
@@ -6,13 +6,13 @@ from dataclasses import dataclass
from ohme import ApiException, OhmeApiClient
from homeassistant.components.number import NumberEntity, NumberEntityDescription
-from homeassistant.const import PERCENTAGE
+from homeassistant.const import PERCENTAGE, UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import OhmeConfigEntry
from .const import DOMAIN
+from .coordinator import OhmeConfigEntry
from .entity import OhmeEntity, OhmeEntityDescription
PARALLEL_UPDATES = 1
@@ -37,13 +37,25 @@ NUMBER_DESCRIPTION = [
native_step=1,
native_unit_of_measurement=PERCENTAGE,
),
+ OhmeNumberDescription(
+ key="preconditioning_duration",
+ translation_key="preconditioning_duration",
+ value_fn=lambda client: client.preconditioning,
+ set_fn=lambda client, value: client.async_set_target(
+ pre_condition_length=value
+ ),
+ native_min_value=0,
+ native_max_value=60,
+ native_step=5,
+ native_unit_of_measurement=UnitOfTime.MINUTES,
+ ),
]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OhmeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up numbers."""
coordinators = config_entry.runtime_data
diff --git a/homeassistant/components/ohme/quality_scale.yaml b/homeassistant/components/ohme/quality_scale.yaml
index 497d5ad32e5..f748cf339b4 100644
--- a/homeassistant/components/ohme/quality_scale.yaml
+++ b/homeassistant/components/ohme/quality_scale.yaml
@@ -39,7 +39,7 @@ rules:
# Gold
devices: done
- diagnostics: todo
+ diagnostics: done
discovery:
status: exempt
comment: |
@@ -62,7 +62,7 @@ rules:
entity-translations: done
exception-translations: done
icon-translations: done
- reconfiguration-flow: todo
+ reconfiguration-flow: done
repair-issues:
status: exempt
comment: |
diff --git a/homeassistant/components/ohme/select.py b/homeassistant/components/ohme/select.py
index a357e98f0a6..f065afeb176 100644
--- a/homeassistant/components/ohme/select.py
+++ b/homeassistant/components/ohme/select.py
@@ -11,10 +11,10 @@ from ohme import ApiException, ChargerMode, OhmeApiClient
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import OhmeConfigEntry
from .const import DOMAIN
+from .coordinator import OhmeConfigEntry
from .entity import OhmeEntity, OhmeEntityDescription
PARALLEL_UPDATES = 1
@@ -25,10 +25,12 @@ class OhmeSelectDescription(OhmeEntityDescription, SelectEntityDescription):
"""Class to describe an Ohme select entity."""
select_fn: Callable[[OhmeApiClient, Any], Awaitable[None]]
+ options: list[str] | None = None
+ options_fn: Callable[[OhmeApiClient], list[str]] | None = None
current_option_fn: Callable[[OhmeApiClient], str | None]
-SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
+MODE_SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
key="charge_mode",
translation_key="charge_mode",
select_fn=lambda client, mode: client.async_set_mode(mode),
@@ -37,16 +39,30 @@ SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
available_fn=lambda client: client.mode is not None,
)
+VEHICLE_SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
+ key="vehicle",
+ translation_key="vehicle",
+ select_fn=lambda client, selection: client.async_set_vehicle(selection),
+ options_fn=lambda client: client.vehicles,
+ current_option_fn=lambda client: client.current_vehicle or None,
+)
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OhmeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ohme selects."""
- coordinator = config_entry.runtime_data.charge_session_coordinator
+ charge_sessions_coordinator = config_entry.runtime_data.charge_session_coordinator
+ device_info_coordinator = config_entry.runtime_data.device_info_coordinator
- async_add_entities([OhmeSelect(coordinator, SELECT_DESCRIPTION)])
+ async_add_entities(
+ [
+ OhmeSelect(charge_sessions_coordinator, MODE_SELECT_DESCRIPTION),
+ OhmeSelect(device_info_coordinator, VEHICLE_SELECT_DESCRIPTION),
+ ]
+ )
class OhmeSelect(OhmeEntity, SelectEntity):
@@ -64,6 +80,14 @@ class OhmeSelect(OhmeEntity, SelectEntity):
) from e
await self.coordinator.async_request_refresh()
+ @property
+ def options(self) -> list[str]:
+ """Return a set of selectable options."""
+ if self.entity_description.options_fn:
+ return self.entity_description.options_fn(self.coordinator.client)
+ assert self.entity_description.options
+ return self.entity_description.options
+
@property
def current_option(self) -> str | None:
"""Return the current selected option."""
diff --git a/homeassistant/components/ohme/sensor.py b/homeassistant/components/ohme/sensor.py
index 230314cba83..d0425040b53 100644
--- a/homeassistant/components/ohme/sensor.py
+++ b/homeassistant/components/ohme/sensor.py
@@ -15,14 +15,16 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import (
PERCENTAGE,
+ STATE_UNKNOWN,
UnitOfElectricCurrent,
+ UnitOfElectricPotential,
UnitOfEnergy,
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import OhmeConfigEntry
+from .coordinator import OhmeConfigEntry
from .entity import OhmeEntity, OhmeEntityDescription
PARALLEL_UPDATES = 0
@@ -66,6 +68,13 @@ SENSOR_CHARGE_SESSION = [
state_class=SensorStateClass.TOTAL_INCREASING,
value_fn=lambda client: client.energy,
),
+ OhmeSensorDescription(
+ key="voltage",
+ device_class=SensorDeviceClass.VOLTAGE,
+ native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+ state_class=SensorStateClass.MEASUREMENT,
+ value_fn=lambda client: client.power.volts,
+ ),
OhmeSensorDescription(
key="battery",
translation_key="vehicle_battery",
@@ -74,6 +83,12 @@ SENSOR_CHARGE_SESSION = [
suggested_display_precision=0,
value_fn=lambda client: client.battery,
),
+ OhmeSensorDescription(
+ key="slot_list",
+ translation_key="slot_list",
+ value_fn=lambda client: ", ".join(str(x) for x in client.slots)
+ or STATE_UNKNOWN,
+ ),
]
SENSOR_ADVANCED_SETTINGS = [
@@ -91,7 +106,7 @@ SENSOR_ADVANCED_SETTINGS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OhmeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
coordinators = config_entry.runtime_data
diff --git a/homeassistant/components/ohme/services.py b/homeassistant/components/ohme/services.py
index 7d06b909d88..be044f01740 100644
--- a/homeassistant/components/ohme/services.py
+++ b/homeassistant/components/ohme/services.py
@@ -17,9 +17,11 @@ from homeassistant.helpers import selector
from .const import DOMAIN
-SERVICE_LIST_CHARGE_SLOTS = "list_charge_slots"
ATTR_CONFIG_ENTRY: Final = "config_entry"
-SERVICE_SCHEMA: Final = vol.Schema(
+ATTR_PRICE_CAP: Final = "price_cap"
+
+SERVICE_LIST_CHARGE_SLOTS = "list_charge_slots"
+SERVICE_LIST_CHARGE_SLOTS_SCHEMA: Final = vol.Schema(
{
vol.Required(ATTR_CONFIG_ENTRY): selector.ConfigEntrySelector(
{
@@ -29,6 +31,18 @@ SERVICE_SCHEMA: Final = vol.Schema(
}
)
+SERVICE_SET_PRICE_CAP = "set_price_cap"
+SERVICE_SET_PRICE_CAP_SCHEMA: Final = vol.Schema(
+ {
+ vol.Required(ATTR_CONFIG_ENTRY): selector.ConfigEntrySelector(
+ {
+ "integration": DOMAIN,
+ }
+ ),
+ vol.Required(ATTR_PRICE_CAP): vol.Coerce(float),
+ }
+)
+
def __get_client(call: ServiceCall) -> OhmeApiClient:
"""Get the client from the config entry."""
@@ -66,10 +80,26 @@ def async_setup_services(hass: HomeAssistant) -> None:
return {"slots": client.slots}
+ async def set_price_cap(
+ service_call: ServiceCall,
+ ) -> None:
+ """List of charge slots."""
+ client = __get_client(service_call)
+ price_cap = service_call.data[ATTR_PRICE_CAP]
+ await client.async_change_price_cap(cap=price_cap)
+
hass.services.async_register(
DOMAIN,
SERVICE_LIST_CHARGE_SLOTS,
list_charge_slots,
- schema=SERVICE_SCHEMA,
+ schema=SERVICE_LIST_CHARGE_SLOTS_SCHEMA,
supports_response=SupportsResponse.ONLY,
)
+
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_SET_PRICE_CAP,
+ set_price_cap,
+ schema=SERVICE_SET_PRICE_CAP_SCHEMA,
+ supports_response=SupportsResponse.NONE,
+ )
diff --git a/homeassistant/components/ohme/services.yaml b/homeassistant/components/ohme/services.yaml
index c5c8ee18138..a45bc131511 100644
--- a/homeassistant/components/ohme/services.yaml
+++ b/homeassistant/components/ohme/services.yaml
@@ -5,3 +5,16 @@ list_charge_slots:
selector:
config_entry:
integration: ohme
+set_price_cap:
+ fields:
+ config_entry:
+ required: true
+ selector:
+ config_entry:
+ integration: ohme
+ price_cap:
+ required: true
+ selector:
+ number:
+ min: 0
+ mode: box
diff --git a/homeassistant/components/ohme/strings.json b/homeassistant/components/ohme/strings.json
index eb5bbffda52..4a2170babeb 100644
--- a/homeassistant/components/ohme/strings.json
+++ b/homeassistant/components/ohme/strings.json
@@ -21,6 +21,16 @@
"data_description": {
"password": "Enter the password for your Ohme account"
}
+ },
+ "reconfigure": {
+ "description": "Update your password for {email}",
+ "title": "Reconfigure Ohme Account",
+ "data": {
+ "password": "[%key:common::config_flow::data::password%]"
+ },
+ "data_description": {
+ "password": "Enter the password for your Ohme account"
+ }
}
},
"error": {
@@ -29,7 +39,8 @@
},
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]",
- "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]"
+ "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",
+ "reconfigure_successful": "[%key:common::config_flow::abort::reconfigure_successful%]"
}
},
"services": {
@@ -42,6 +53,20 @@
"description": "The Ohme config entry for which to return charge slots."
}
}
+ },
+ "set_price_cap": {
+ "name": "Set price cap",
+ "description": "Prevents charging when the electricity price exceeds a defined threshold.",
+ "fields": {
+ "config_entry": {
+ "name": "Ohme account",
+ "description": "The Ohme config entry for which to return charge slots."
+ },
+ "price_cap": {
+ "name": "Price cap",
+ "description": "Threshold in 1/100ths of your local currency."
+ }
+ }
}
},
"entity": {
@@ -51,6 +76,9 @@
}
},
"number": {
+ "preconditioning_duration": {
+ "name": "Preconditioning duration"
+ },
"target_percentage": {
"name": "Target percentage"
}
@@ -63,6 +91,9 @@
"max_charge": "Max charge",
"paused": "Paused"
}
+ },
+ "vehicle": {
+ "name": "Vehicle"
}
},
"sensor": {
@@ -71,9 +102,10 @@
"state": {
"unplugged": "Unplugged",
"plugged_in": "Plugged in",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"paused": "[%key:common::state::paused%]",
- "pending_approval": "Pending approval"
+ "pending_approval": "Pending approval",
+ "finished": "Finished charging"
}
},
"ct_current": {
@@ -81,6 +113,9 @@
},
"vehicle_battery": {
"name": "Vehicle battery"
+ },
+ "slot_list": {
+ "name": "Charge slots"
}
},
"switch": {
@@ -92,6 +127,9 @@
},
"sleep_when_inactive": {
"name": "Sleep when inactive"
+ },
+ "price_cap": {
+ "name": "Price cap"
}
},
"time": {
diff --git a/homeassistant/components/ohme/switch.py b/homeassistant/components/ohme/switch.py
index d1eb1a80b56..47e3bf8a99d 100644
--- a/homeassistant/components/ohme/switch.py
+++ b/homeassistant/components/ohme/switch.py
@@ -1,46 +1,56 @@
"""Platform for switch."""
+from collections.abc import Awaitable, Callable
from dataclasses import dataclass
from typing import Any
-from ohme import ApiException
+from ohme import ApiException, OhmeApiClient
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import OhmeConfigEntry
from .const import DOMAIN
+from .coordinator import OhmeConfigEntry
from .entity import OhmeEntity, OhmeEntityDescription
PARALLEL_UPDATES = 1
@dataclass(frozen=True, kw_only=True)
-class OhmeSwitchDescription(OhmeEntityDescription, SwitchEntityDescription):
- """Class describing Ohme switch entities."""
+class OhmeConfigSwitchDescription(OhmeEntityDescription, SwitchEntityDescription):
+ """Class describing Ohme configuration switch entities."""
configuration_key: str
-SWITCH_DEVICE_INFO = [
- OhmeSwitchDescription(
+@dataclass(frozen=True, kw_only=True)
+class OhmeSwitchDescription(OhmeEntityDescription, SwitchEntityDescription):
+ """Class describing basic Ohme switch entities."""
+
+ is_on_fn: Callable[[OhmeApiClient], bool]
+ off_fn: Callable[[OhmeApiClient], Awaitable]
+ on_fn: Callable[[OhmeApiClient], Awaitable]
+
+
+SWITCH_CONFIG = [
+ OhmeConfigSwitchDescription(
key="lock_buttons",
translation_key="lock_buttons",
entity_category=EntityCategory.CONFIG,
is_supported_fn=lambda client: client.is_capable("buttonsLockable"),
configuration_key="buttonsLocked",
),
- OhmeSwitchDescription(
+ OhmeConfigSwitchDescription(
key="require_approval",
translation_key="require_approval",
entity_category=EntityCategory.CONFIG,
is_supported_fn=lambda client: client.is_capable("pluginsRequireApprovalMode"),
configuration_key="pluginsRequireApproval",
),
- OhmeSwitchDescription(
+ OhmeConfigSwitchDescription(
key="sleep_when_inactive",
translation_key="sleep_when_inactive",
entity_category=EntityCategory.CONFIG,
@@ -49,22 +59,35 @@ SWITCH_DEVICE_INFO = [
),
]
+SWITCH_DESCRIPTION = [
+ OhmeSwitchDescription(
+ key="price_cap",
+ translation_key="price_cap",
+ is_supported_fn=lambda client: client.cap_available,
+ is_on_fn=lambda client: client.cap_enabled,
+ on_fn=lambda client: client.async_change_price_cap(True),
+ off_fn=lambda client: client.async_change_price_cap(False),
+ ),
+]
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OhmeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches."""
- coordinators = config_entry.runtime_data
- coordinator_map = [
- (SWITCH_DEVICE_INFO, coordinators.device_info_coordinator),
- ]
+ coordinator = config_entry.runtime_data.device_info_coordinator
+
+ async_add_entities(
+ OhmeConfigSwitch(coordinator, description)
+ for description in SWITCH_CONFIG
+ if description.is_supported_fn(coordinator.client)
+ )
async_add_entities(
OhmeSwitch(coordinator, description)
- for entities, coordinator in coordinator_map
- for description in entities
+ for description in SWITCH_DESCRIPTION
if description.is_supported_fn(coordinator.client)
)
@@ -74,6 +97,27 @@ class OhmeSwitch(OhmeEntity, SwitchEntity):
entity_description: OhmeSwitchDescription
+ @property
+ def is_on(self) -> bool:
+ """Return True if entity is on."""
+ return self.entity_description.is_on_fn(self.coordinator.client)
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the switch off."""
+ await self.entity_description.off_fn(self.coordinator.client)
+ await self.coordinator.async_request_refresh()
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the switch on."""
+ await self.entity_description.on_fn(self.coordinator.client)
+ await self.coordinator.async_request_refresh()
+
+
+class OhmeConfigSwitch(OhmeEntity, SwitchEntity):
+ """Configuration switch for Ohme."""
+
+ entity_description: OhmeConfigSwitchDescription
+
@property
def is_on(self) -> bool:
"""Return the entity value to represent the entity state."""
diff --git a/homeassistant/components/ohme/time.py b/homeassistant/components/ohme/time.py
index a7de913ef8e..264b2afd41a 100644
--- a/homeassistant/components/ohme/time.py
+++ b/homeassistant/components/ohme/time.py
@@ -9,10 +9,10 @@ from ohme import ApiException, OhmeApiClient
from homeassistant.components.time import TimeEntity, TimeEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import OhmeConfigEntry
from .const import DOMAIN
+from .coordinator import OhmeConfigEntry
from .entity import OhmeEntity, OhmeEntityDescription
PARALLEL_UPDATES = 1
@@ -43,7 +43,7 @@ TIME_DESCRIPTION = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OhmeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up time entities."""
coordinators = config_entry.runtime_data
diff --git a/homeassistant/components/ollama/conversation.py b/homeassistant/components/ollama/conversation.py
index c0fbfae6444..ab9e05b5fbe 100644
--- a/homeassistant/components/ollama/conversation.py
+++ b/homeassistant/components/ollama/conversation.py
@@ -2,25 +2,21 @@
from __future__ import annotations
-from collections.abc import Callable
+from collections.abc import AsyncGenerator, Callable
import json
import logging
-import time
from typing import Any, Literal
import ollama
-import voluptuous as vol
from voluptuous_openapi import convert
from homeassistant.components import assist_pipeline, conversation
-from homeassistant.components.conversation import trace
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_LLM_HASS_API, MATCH_ALL
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import HomeAssistantError, TemplateError
-from homeassistant.helpers import intent, llm, template
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.util import ulid as ulid_util
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers import intent, llm
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_KEEP_ALIVE,
@@ -32,7 +28,6 @@ from .const import (
DEFAULT_MAX_HISTORY,
DEFAULT_NUM_CTX,
DOMAIN,
- MAX_HISTORY_SECONDS,
)
from .models import MessageHistory, MessageRole
@@ -45,7 +40,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up conversation entities."""
agent = OllamaConversationEntity(config_entry)
@@ -93,6 +88,84 @@ def _parse_tool_args(arguments: dict[str, Any]) -> dict[str, Any]:
return {k: _fix_invalid_arguments(v) for k, v in arguments.items() if v}
+def _convert_content(
+ chat_content: conversation.Content
+ | conversation.ToolResultContent
+ | conversation.AssistantContent,
+) -> ollama.Message:
+ """Create tool response content."""
+ if isinstance(chat_content, conversation.ToolResultContent):
+ return ollama.Message(
+ role=MessageRole.TOOL.value,
+ content=json.dumps(chat_content.tool_result),
+ )
+ if isinstance(chat_content, conversation.AssistantContent):
+ return ollama.Message(
+ role=MessageRole.ASSISTANT.value,
+ content=chat_content.content,
+ tool_calls=[
+ ollama.Message.ToolCall(
+ function=ollama.Message.ToolCall.Function(
+ name=tool_call.tool_name,
+ arguments=tool_call.tool_args,
+ )
+ )
+ for tool_call in chat_content.tool_calls or ()
+ ],
+ )
+ if isinstance(chat_content, conversation.UserContent):
+ return ollama.Message(
+ role=MessageRole.USER.value,
+ content=chat_content.content,
+ )
+ if isinstance(chat_content, conversation.SystemContent):
+ return ollama.Message(
+ role=MessageRole.SYSTEM.value,
+ content=chat_content.content,
+ )
+ raise TypeError(f"Unexpected content type: {type(chat_content)}")
+
+
+async def _transform_stream(
+ result: AsyncGenerator[ollama.Message],
+) -> AsyncGenerator[conversation.AssistantContentDeltaDict]:
+ """Transform the response stream into HA format.
+
+ An Ollama streaming response may come in chunks like this:
+
+ response: message=Message(role="assistant", content="Paris")
+ response: message=Message(role="assistant", content=".")
+ response: message=Message(role="assistant", content=""), done: True, done_reason: "stop"
+ response: message=Message(role="assistant", tool_calls=[...])
+ response: message=Message(role="assistant", content=""), done: True, done_reason: "stop"
+
+ This generator conforms to the chatlog delta stream expectations in that it
+ yields deltas, then the role only once the response is done.
+ """
+
+ new_msg = True
+ async for response in result:
+ _LOGGER.debug("Received response: %s", response)
+ response_message = response["message"]
+ chunk: conversation.AssistantContentDeltaDict = {}
+ if new_msg:
+ new_msg = False
+ chunk["role"] = "assistant"
+ if (tool_calls := response_message.get("tool_calls")) is not None:
+ chunk["tool_calls"] = [
+ llm.ToolInput(
+ tool_name=tool_call["function"]["name"],
+ tool_args=_parse_tool_args(tool_call["function"]["arguments"]),
+ )
+ for tool_call in tool_calls
+ ]
+ if (content := response_message.get("content")) is not None:
+ chunk["content"] = content
+ if response_message.get("done"):
+ new_msg = True
+ yield chunk
+
+
class OllamaConversationEntity(
conversation.ConversationEntity, conversation.AbstractConversationAgent
):
@@ -105,7 +178,6 @@ class OllamaConversationEntity(
self.entry = entry
# conversation id -> message history
- self._history: dict[str, MessageHistory] = {}
self._attr_name = entry.title
self._attr_unique_id = entry.entry_id
if self.entry.options.get(CONF_LLM_HASS_API):
@@ -134,212 +206,106 @@ class OllamaConversationEntity(
"""Return a list of supported languages."""
return MATCH_ALL
- async def async_process(
- self, user_input: conversation.ConversationInput
+ async def _async_handle_message(
+ self,
+ user_input: conversation.ConversationInput,
+ chat_log: conversation.ChatLog,
) -> conversation.ConversationResult:
- """Process a sentence."""
+ """Call the API."""
settings = {**self.entry.data, **self.entry.options}
client = self.hass.data[DOMAIN][self.entry.entry_id]
- conversation_id = user_input.conversation_id or ulid_util.ulid_now()
model = settings[CONF_MODEL]
- intent_response = intent.IntentResponse(language=user_input.language)
- llm_api: llm.APIInstance | None = None
- tools: list[dict[str, Any]] | None = None
- user_name: str | None = None
- llm_context = llm.LLMContext(
- platform=DOMAIN,
- context=user_input.context,
- user_prompt=user_input.text,
- language=user_input.language,
- assistant=conversation.DOMAIN,
- device_id=user_input.device_id,
- )
- if settings.get(CONF_LLM_HASS_API):
- try:
- llm_api = await llm.async_get_api(
- self.hass,
- settings[CONF_LLM_HASS_API],
- llm_context,
- )
- except HomeAssistantError as err:
- _LOGGER.error("Error getting LLM API: %s", err)
- intent_response.async_set_error(
- intent.IntentResponseErrorCode.UNKNOWN,
- f"Error preparing LLM API: {err}",
- )
- return conversation.ConversationResult(
- response=intent_response, conversation_id=user_input.conversation_id
- )
+ try:
+ await chat_log.async_update_llm_data(
+ DOMAIN,
+ user_input,
+ settings.get(CONF_LLM_HASS_API),
+ settings.get(CONF_PROMPT),
+ )
+ except conversation.ConverseError as err:
+ return err.as_conversation_result()
+
+ tools: list[dict[str, Any]] | None = None
+ if chat_log.llm_api:
tools = [
- _format_tool(tool, llm_api.custom_serializer) for tool in llm_api.tools
+ _format_tool(tool, chat_log.llm_api.custom_serializer)
+ for tool in chat_log.llm_api.tools
]
- if (
- user_input.context
- and user_input.context.user_id
- and (
- user := await self.hass.auth.async_get_user(user_input.context.user_id)
- )
- ):
- user_name = user.name
-
- # Look up message history
- message_history: MessageHistory | None = None
- message_history = self._history.get(conversation_id)
- if message_history is None:
- # New history
- #
- # Render prompt and error out early if there's a problem
- try:
- prompt_parts = [
- template.Template(
- llm.BASE_PROMPT
- + settings.get(CONF_PROMPT, llm.DEFAULT_INSTRUCTIONS_PROMPT),
- self.hass,
- ).async_render(
- {
- "ha_name": self.hass.config.location_name,
- "user_name": user_name,
- "llm_context": llm_context,
- },
- parse_result=False,
- )
- ]
-
- except TemplateError as err:
- _LOGGER.error("Error rendering prompt: %s", err)
- intent_response.async_set_error(
- intent.IntentResponseErrorCode.UNKNOWN,
- f"Sorry, I had a problem generating my prompt: {err}",
- )
- return conversation.ConversationResult(
- response=intent_response, conversation_id=conversation_id
- )
-
- if llm_api:
- prompt_parts.append(llm_api.api_prompt)
-
- prompt = "\n".join(prompt_parts)
- _LOGGER.debug("Prompt: %s", prompt)
- _LOGGER.debug("Tools: %s", tools)
-
- message_history = MessageHistory(
- timestamp=time.monotonic(),
- messages=[
- ollama.Message(role=MessageRole.SYSTEM.value, content=prompt)
- ],
- )
- self._history[conversation_id] = message_history
- else:
- # Bump timestamp so this conversation won't get cleaned up
- message_history.timestamp = time.monotonic()
-
- # Clean up old histories
- self._prune_old_histories()
-
- # Trim this message history to keep a maximum number of *user* messages
+ message_history: MessageHistory = MessageHistory(
+ [_convert_content(content) for content in chat_log.content]
+ )
max_messages = int(settings.get(CONF_MAX_HISTORY, DEFAULT_MAX_HISTORY))
self._trim_history(message_history, max_messages)
- # Add new user message
- message_history.messages.append(
- ollama.Message(role=MessageRole.USER.value, content=user_input.text)
- )
-
- trace.async_conversation_trace_append(
- trace.ConversationTraceEventType.AGENT_DETAIL,
- {"messages": message_history.messages},
- )
-
# Get response
# To prevent infinite loops, we limit the number of iterations
for _iteration in range(MAX_TOOL_ITERATIONS):
try:
- response = await client.chat(
+ response_generator = await client.chat(
model=model,
# Make a copy of the messages because we mutate the list later
messages=list(message_history.messages),
tools=tools,
- stream=False,
+ stream=True,
# keep_alive requires specifying unit. In this case, seconds
keep_alive=f"{settings.get(CONF_KEEP_ALIVE, DEFAULT_KEEP_ALIVE)}s",
options={CONF_NUM_CTX: settings.get(CONF_NUM_CTX, DEFAULT_NUM_CTX)},
)
except (ollama.RequestError, ollama.ResponseError) as err:
_LOGGER.error("Unexpected error talking to Ollama server: %s", err)
- intent_response.async_set_error(
- intent.IntentResponseErrorCode.UNKNOWN,
- f"Sorry, I had a problem talking to the Ollama server: {err}",
- )
- return conversation.ConversationResult(
- response=intent_response, conversation_id=conversation_id
- )
+ raise HomeAssistantError(
+ f"Sorry, I had a problem talking to the Ollama server: {err}"
+ ) from err
- response_message = response["message"]
- message_history.messages.append(
- ollama.Message(
- role=response_message["role"],
- content=response_message.get("content"),
- tool_calls=response_message.get("tool_calls"),
- )
+ message_history.messages.extend(
+ [
+ _convert_content(content)
+ async for content in chat_log.async_add_delta_content_stream(
+ user_input.agent_id, _transform_stream(response_generator)
+ )
+ ]
)
- tool_calls = response_message.get("tool_calls")
- if not tool_calls or not llm_api:
+ if not chat_log.unresponded_tool_results:
break
- for tool_call in tool_calls:
- tool_input = llm.ToolInput(
- tool_name=tool_call["function"]["name"],
- tool_args=_parse_tool_args(tool_call["function"]["arguments"]),
- )
- _LOGGER.debug(
- "Tool call: %s(%s)", tool_input.tool_name, tool_input.tool_args
- )
-
- try:
- tool_response = await llm_api.async_call_tool(tool_input)
- except (HomeAssistantError, vol.Invalid) as e:
- tool_response = {"error": type(e).__name__}
- if str(e):
- tool_response["error_text"] = str(e)
-
- _LOGGER.debug("Tool response: %s", tool_response)
- message_history.messages.append(
- ollama.Message(
- role=MessageRole.TOOL.value,
- content=json.dumps(tool_response),
- )
- )
-
# Create intent response
- intent_response.async_set_speech(response_message["content"])
+ intent_response = intent.IntentResponse(language=user_input.language)
+ if not isinstance(chat_log.content[-1], conversation.AssistantContent):
+ raise TypeError(
+ f"Unexpected last message type: {type(chat_log.content[-1])}"
+ )
+ intent_response.async_set_speech(chat_log.content[-1].content or "")
return conversation.ConversationResult(
- response=intent_response, conversation_id=conversation_id
+ response=intent_response,
+ conversation_id=chat_log.conversation_id,
+ continue_conversation=chat_log.continue_conversation,
)
- def _prune_old_histories(self) -> None:
- """Remove old message histories."""
- now = time.monotonic()
- self._history = {
- conversation_id: message_history
- for conversation_id, message_history in self._history.items()
- if (now - message_history.timestamp) <= MAX_HISTORY_SECONDS
- }
-
def _trim_history(self, message_history: MessageHistory, max_messages: int) -> None:
- """Trims excess messages from a single history."""
+ """Trims excess messages from a single history.
+
+ This sets the max history to allow a configurable size history may take
+ up in the context window.
+
+ Note that some messages in the history may not be from ollama only, and
+ may come from other anents, so the assumptions here may not strictly hold,
+ but generally should be effective.
+ """
if max_messages < 1:
# Keep all messages
return
- if message_history.num_user_messages >= max_messages:
+ # Ignore the in progress user message
+ num_previous_rounds = message_history.num_user_messages - 1
+ if num_previous_rounds >= max_messages:
# Trim history but keep system prompt (first message).
# Every other message should be an assistant message, so keep 2x
- # message objects.
- num_keep = 2 * max_messages
+ # message objects. Also keep the last in progress user message
+ num_keep = 2 * max_messages + 1
drop_index = len(message_history.messages) - num_keep
message_history.messages = [
message_history.messages[0]
diff --git a/homeassistant/components/ollama/models.py b/homeassistant/components/ollama/models.py
index 3b6fc958587..fd268664919 100644
--- a/homeassistant/components/ollama/models.py
+++ b/homeassistant/components/ollama/models.py
@@ -19,9 +19,6 @@ class MessageRole(StrEnum):
class MessageHistory:
"""Chat message history."""
- timestamp: float
- """Timestamp of last use in seconds."""
-
messages: list[ollama.Message]
"""List of message history, including system prompt and assistant responses."""
diff --git a/homeassistant/components/omnilogic/sensor.py b/homeassistant/components/omnilogic/sensor.py
index c87b589e1f6..d941eb3ae4d 100644
--- a/homeassistant/components/omnilogic/sensor.py
+++ b/homeassistant/components/omnilogic/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import check_guard
from .const import COORDINATOR, DEFAULT_PH_OFFSET, DOMAIN, PUMP_TYPES
@@ -22,7 +22,9 @@ from .entity import OmniLogicEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
diff --git a/homeassistant/components/omnilogic/strings.json b/homeassistant/components/omnilogic/strings.json
index 5b193b7f5ba..6f207337789 100644
--- a/homeassistant/components/omnilogic/strings.json
+++ b/homeassistant/components/omnilogic/strings.json
@@ -34,7 +34,7 @@
"fields": {
"speed": {
"name": "Speed",
- "description": "Speed for the VSP between min and max speed."
+ "description": "Speed for the pump between min and max speed."
}
}
}
diff --git a/homeassistant/components/omnilogic/switch.py b/homeassistant/components/omnilogic/switch.py
index eb57d03bc34..a9f8bc77d8a 100644
--- a/homeassistant/components/omnilogic/switch.py
+++ b/homeassistant/components/omnilogic/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import check_guard
from .const import COORDINATOR, DOMAIN, PUMP_TYPES
@@ -22,7 +22,9 @@ OMNILOGIC_SWITCH_OFF = 7
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the light platform."""
diff --git a/homeassistant/components/onboarding/manifest.json b/homeassistant/components/onboarding/manifest.json
index 8e253d4bff9..a4cf814eb2a 100644
--- a/homeassistant/components/onboarding/manifest.json
+++ b/homeassistant/components/onboarding/manifest.json
@@ -1,7 +1,6 @@
{
"domain": "onboarding",
"name": "Home Assistant Onboarding",
- "after_dependencies": ["backup", "hassio"],
"codeowners": ["@home-assistant/core"],
"dependencies": ["auth", "http", "person"],
"documentation": "https://www.home-assistant.io/integrations/onboarding",
diff --git a/homeassistant/components/onboarding/views.py b/homeassistant/components/onboarding/views.py
index 1e29860e3c5..5f1d908f7f8 100644
--- a/homeassistant/components/onboarding/views.py
+++ b/homeassistant/components/onboarding/views.py
@@ -20,7 +20,6 @@ from homeassistant.components.backup import (
BackupManager,
Folder,
IncorrectPasswordError,
- async_get_manager as async_get_backup_manager,
http as backup_http,
)
from homeassistant.components.http import KEY_HASS, KEY_HASS_REFRESH_TOKEN_ID
@@ -29,11 +28,10 @@ from homeassistant.components.http.view import HomeAssistantView
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import area_registry as ar
-from homeassistant.helpers.hassio import is_hassio
+from homeassistant.helpers.backup import async_get_manager as async_get_backup_manager
from homeassistant.helpers.system_info import async_get_system_info
from homeassistant.helpers.translation import async_get_translations
from homeassistant.setup import async_setup_component
-from homeassistant.util.async_ import create_eager_task
if TYPE_CHECKING:
from . import OnboardingData, OnboardingStorage, OnboardingStoreData
@@ -225,32 +223,21 @@ class CoreConfigOnboardingView(_BaseOnboardingView):
"shopping_list",
]
- # pylint: disable-next=import-outside-toplevel
- from homeassistant.components import hassio
-
- if (
- is_hassio(hass)
- and (core_info := hassio.get_core_info(hass))
- and "raspberrypi" in core_info["machine"]
- ):
- onboard_integrations.append("rpi_power")
-
- coros: list[Coroutine[Any, Any, Any]] = [
- hass.config_entries.flow.async_init(
- domain, context={"source": "onboarding"}
+ for domain in onboard_integrations:
+ # Create tasks so onboarding isn't affected
+ # by errors in these integrations.
+ hass.async_create_task(
+ hass.config_entries.flow.async_init(
+ domain, context={"source": "onboarding"}
+ ),
+ f"onboarding_setup_{domain}",
)
- for domain in onboard_integrations
- ]
if "analytics" not in hass.config.components:
# If by some chance that analytics has not finished
# setting up, wait for it here so its ready for the
# next step.
- coros.append(async_setup_component(hass, "analytics", {}))
-
- # Set up integrations after onboarding and ensure
- # analytics is ready for the next step.
- await asyncio.gather(*(create_eager_task(coro) for coro in coros))
+ await async_setup_component(hass, "analytics", {})
return self.json({})
@@ -354,10 +341,10 @@ def with_backup_manager[_ViewT: BackupOnboardingView, **_P](
raise HTTPUnauthorized
try:
- manager = async_get_backup_manager(request.app[KEY_HASS])
+ manager = await async_get_backup_manager(request.app[KEY_HASS])
except HomeAssistantError:
return self.json(
- {"error": "backup_disabled"},
+ {"code": "backup_disabled"},
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
)
@@ -380,7 +367,7 @@ class BackupInfoView(BackupOnboardingView):
{
"backups": list(backups.values()),
"state": manager.state,
- "last_non_idle_event": manager.last_non_idle_event,
+ "last_action_event": manager.last_action_event,
}
)
@@ -420,7 +407,12 @@ class RestoreBackupView(BackupOnboardingView):
)
except IncorrectPasswordError:
return self.json(
- {"message": "incorrect_password"}, status_code=HTTPStatus.BAD_REQUEST
+ {"code": "incorrect_password"}, status_code=HTTPStatus.BAD_REQUEST
+ )
+ except HomeAssistantError as err:
+ return self.json(
+ {"code": "restore_failed", "message": str(err)},
+ status_code=HTTPStatus.BAD_REQUEST,
)
return web.Response(status=HTTPStatus.OK)
diff --git a/homeassistant/components/oncue/binary_sensor.py b/homeassistant/components/oncue/binary_sensor.py
index 961b082a5c5..8dc9ba1be6f 100644
--- a/homeassistant/components/oncue/binary_sensor.py
+++ b/homeassistant/components/oncue/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import OncueEntity
from .types import OncueConfigEntry
@@ -28,7 +28,7 @@ SENSOR_MAP = {description.key: description for description in SENSOR_TYPES}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OncueConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/oncue/sensor.py b/homeassistant/components/oncue/sensor.py
index a0f275ef692..669c34157d4 100644
--- a/homeassistant/components/oncue/sensor.py
+++ b/homeassistant/components/oncue/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .entity import OncueEntity
@@ -180,7 +180,7 @@ UNIT_MAPPINGS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OncueConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/ondilo_ico/__init__.py b/homeassistant/components/ondilo_ico/__init__.py
index fb78035c630..93aadb5b6ea 100644
--- a/homeassistant/components/ondilo_ico/__init__.py
+++ b/homeassistant/components/ondilo_ico/__init__.py
@@ -1,34 +1,46 @@
"""The Ondilo ICO integration."""
+from homeassistant.components.application_credentials import (
+ ClientCredential,
+ async_import_client_credential,
+)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers import config_entry_oauth2_flow
+from homeassistant.helpers import config_entry_oauth2_flow, config_validation as cv
+from homeassistant.helpers.typing import ConfigType
from .api import OndiloClient
-from .config_flow import OndiloIcoOAuth2FlowHandler
-from .const import DOMAIN
-from .coordinator import OndiloIcoCoordinator
-from .oauth_impl import OndiloOauth2Implementation
+from .const import DOMAIN, OAUTH2_CLIENT_ID, OAUTH2_CLIENT_SECRET
+from .coordinator import OndiloIcoPoolsCoordinator
+CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
PLATFORMS = [Platform.SENSOR]
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
+ """Set up the Ondilo ICO integration."""
+ # Import the default client credential.
+ await async_import_client_credential(
+ hass,
+ DOMAIN,
+ ClientCredential(OAUTH2_CLIENT_ID, OAUTH2_CLIENT_SECRET, name="Ondilo ICO"),
+ )
+
+ return True
+
+
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Ondilo ICO from a config entry."""
-
- OndiloIcoOAuth2FlowHandler.async_register_implementation(
- hass,
- OndiloOauth2Implementation(hass),
- )
-
implementation = (
await config_entry_oauth2_flow.async_get_config_entry_implementation(
hass, entry
)
)
- coordinator = OndiloIcoCoordinator(hass, OndiloClient(hass, entry, implementation))
+ coordinator = OndiloIcoPoolsCoordinator(
+ hass, entry, OndiloClient(hass, entry, implementation)
+ )
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/ondilo_ico/api.py b/homeassistant/components/ondilo_ico/api.py
index f6ab0baa576..696acf1b2d6 100644
--- a/homeassistant/components/ondilo_ico/api.py
+++ b/homeassistant/components/ondilo_ico/api.py
@@ -1,15 +1,12 @@
"""API for Ondilo ICO bound to Home Assistant OAuth."""
from asyncio import run_coroutine_threadsafe
-import logging
from ondilo import Ondilo
from homeassistant import config_entries, core
from homeassistant.helpers import config_entry_oauth2_flow
-_LOGGER = logging.getLogger(__name__)
-
class OndiloClient(Ondilo):
"""Provide Ondilo ICO authentication tied to an OAuth2 based config entry."""
diff --git a/homeassistant/components/ondilo_ico/application_credentials.py b/homeassistant/components/ondilo_ico/application_credentials.py
new file mode 100644
index 00000000000..5481a88bc1b
--- /dev/null
+++ b/homeassistant/components/ondilo_ico/application_credentials.py
@@ -0,0 +1,14 @@
+"""Application credentials platform for Ondilo ICO."""
+
+from homeassistant.components.application_credentials import AuthorizationServer
+from homeassistant.core import HomeAssistant
+
+from .const import OAUTH2_AUTHORIZE, OAUTH2_TOKEN
+
+
+async def async_get_authorization_server(hass: HomeAssistant) -> AuthorizationServer:
+ """Return authorization server."""
+ return AuthorizationServer(
+ authorize_url=OAUTH2_AUTHORIZE,
+ token_url=OAUTH2_TOKEN,
+ )
diff --git a/homeassistant/components/ondilo_ico/config_flow.py b/homeassistant/components/ondilo_ico/config_flow.py
index fe0b89e7258..6839d2089bf 100644
--- a/homeassistant/components/ondilo_ico/config_flow.py
+++ b/homeassistant/components/ondilo_ico/config_flow.py
@@ -3,11 +3,14 @@
import logging
from typing import Any
+from homeassistant.components.application_credentials import (
+ ClientCredential,
+ async_import_client_credential,
+)
from homeassistant.config_entries import ConfigFlowResult
from homeassistant.helpers.config_entry_oauth2_flow import AbstractOAuth2FlowHandler
-from .const import DOMAIN
-from .oauth_impl import OndiloOauth2Implementation
+from .const import DOMAIN, OAUTH2_CLIENT_ID, OAUTH2_CLIENT_SECRET
class OndiloIcoOAuth2FlowHandler(AbstractOAuth2FlowHandler, domain=DOMAIN):
@@ -18,14 +21,13 @@ class OndiloIcoOAuth2FlowHandler(AbstractOAuth2FlowHandler, domain=DOMAIN):
async def async_step_user(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
- """Handle a flow initialized by the user."""
- await self.async_set_unique_id(DOMAIN)
-
- self.async_register_implementation(
+ """Handle a flow start."""
+ # Import the default client credential.
+ await async_import_client_credential(
self.hass,
- OndiloOauth2Implementation(self.hass),
+ DOMAIN,
+ ClientCredential(OAUTH2_CLIENT_ID, OAUTH2_CLIENT_SECRET, name="Ondilo ICO"),
)
-
return await super().async_step_user(user_input)
@property
diff --git a/homeassistant/components/ondilo_ico/const.py b/homeassistant/components/ondilo_ico/const.py
index 3c947776857..8dec6072556 100644
--- a/homeassistant/components/ondilo_ico/const.py
+++ b/homeassistant/components/ondilo_ico/const.py
@@ -4,5 +4,5 @@ DOMAIN = "ondilo_ico"
OAUTH2_AUTHORIZE = "https://interop.ondilo.com/oauth2/authorize"
OAUTH2_TOKEN = "https://interop.ondilo.com/oauth2/token"
-OAUTH2_CLIENTID = "customer_api"
-OAUTH2_CLIENTSECRET = ""
+OAUTH2_CLIENT_ID = "customer_api"
+OAUTH2_CLIENT_SECRET = ""
diff --git a/homeassistant/components/ondilo_ico/coordinator.py b/homeassistant/components/ondilo_ico/coordinator.py
index bc092ad0b9a..7545f6d61e0 100644
--- a/homeassistant/components/ondilo_ico/coordinator.py
+++ b/homeassistant/components/ondilo_ico/coordinator.py
@@ -1,77 +1,191 @@
"""Define an object to coordinate fetching Ondilo ICO data."""
-from dataclasses import dataclass
-from datetime import timedelta
+from __future__ import annotations
+
+import asyncio
+from dataclasses import dataclass, field
+from datetime import datetime, timedelta
import logging
from typing import Any
from ondilo import OndiloError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
+from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+from homeassistant.util import dt as dt_util
from . import DOMAIN
from .api import OndiloClient
_LOGGER = logging.getLogger(__name__)
+TIME_TO_NEXT_UPDATE = timedelta(hours=1, minutes=5)
+UPDATE_LOCK = asyncio.Lock()
+
@dataclass
-class OndiloIcoData:
- """Class for storing the data."""
+class OndiloIcoPoolData:
+ """Store the pools the data."""
ico: dict[str, Any]
pool: dict[str, Any]
+ measures_coordinator: OndiloIcoMeasuresCoordinator = field(init=False)
+
+
+@dataclass
+class OndiloIcoMeasurementData:
+ """Store the measurement data for one pool."""
+
sensors: dict[str, Any]
-class OndiloIcoCoordinator(DataUpdateCoordinator[dict[str, OndiloIcoData]]):
- """Class to manage fetching Ondilo ICO data from API."""
+class OndiloIcoPoolsCoordinator(DataUpdateCoordinator[dict[str, OndiloIcoPoolData]]):
+ """Fetch Ondilo ICO pools data from API."""
- def __init__(self, hass: HomeAssistant, api: OndiloClient) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: OndiloClient
+ ) -> None:
"""Initialize."""
super().__init__(
hass,
logger=_LOGGER,
- name=DOMAIN,
+ config_entry=config_entry,
+ name=f"{DOMAIN}_pools",
update_interval=timedelta(minutes=20),
)
self.api = api
+ self.config_entry = config_entry
+ self._device_registry = dr.async_get(self.hass)
- async def _async_update_data(self) -> dict[str, OndiloIcoData]:
- """Fetch data from API endpoint."""
+ async def _async_update_data(self) -> dict[str, OndiloIcoPoolData]:
+ """Fetch pools data from API endpoint and update devices."""
+ known_pools: set[str] = set(self.data) if self.data else set()
try:
- return await self.hass.async_add_executor_job(self._update_data)
+ async with UPDATE_LOCK:
+ data = await self.hass.async_add_executor_job(self._update_data)
except OndiloError as err:
raise UpdateFailed(f"Error communicating with API: {err}") from err
- def _update_data(self) -> dict[str, OndiloIcoData]:
- """Fetch data from API endpoint."""
+ current_pools = set(data)
+
+ new_pools = current_pools - known_pools
+ for pool_id in new_pools:
+ pool_data = data[pool_id]
+ pool_data.measures_coordinator = OndiloIcoMeasuresCoordinator(
+ self.hass, self.config_entry, self.api, pool_id
+ )
+ self._device_registry.async_get_or_create(
+ config_entry_id=self.config_entry.entry_id,
+ identifiers={(DOMAIN, pool_data.ico["serial_number"])},
+ manufacturer="Ondilo",
+ model="ICO",
+ name=pool_data.pool["name"],
+ sw_version=pool_data.ico["sw_version"],
+ )
+
+ removed_pools = known_pools - current_pools
+ for pool_id in removed_pools:
+ pool_data = self.data.pop(pool_id)
+ await pool_data.measures_coordinator.async_shutdown()
+ device_entry = self._device_registry.async_get_device(
+ identifiers={(DOMAIN, pool_data.ico["serial_number"])}
+ )
+ if device_entry:
+ self._device_registry.async_update_device(
+ device_id=device_entry.id,
+ remove_config_entry_id=self.config_entry.entry_id,
+ )
+
+ for pool_id in current_pools:
+ pool_data = data[pool_id]
+ measures_coordinator = pool_data.measures_coordinator
+ measures_coordinator.set_next_refresh(pool_data)
+ if not measures_coordinator.data:
+ await measures_coordinator.async_refresh()
+
+ return data
+
+ def _update_data(self) -> dict[str, OndiloIcoPoolData]:
+ """Fetch pools data from API endpoint."""
res = {}
pools = self.api.get_pools()
_LOGGER.debug("Pools: %s", pools)
error: OndiloError | None = None
for pool in pools:
pool_id = pool["id"]
+ if (data := self.data) and pool_id in data:
+ pool_data = res[pool_id] = data[pool_id]
+ pool_data.pool = pool
+ # Skip requesting new ICO data for known pools
+ # to avoid unnecessary API calls.
+ continue
try:
ico = self.api.get_ICO_details(pool_id)
- if not ico:
- _LOGGER.debug(
- "The pool id %s does not have any ICO attached", pool_id
- )
- continue
- sensors = self.api.get_last_pool_measures(pool_id)
except OndiloError as err:
error = err
_LOGGER.debug("Error communicating with API for %s: %s", pool_id, err)
continue
- res[pool_id] = OndiloIcoData(
- ico=ico,
- pool=pool,
- sensors={sensor["data_type"]: sensor["value"] for sensor in sensors},
- )
+
+ if not ico:
+ _LOGGER.debug("The pool id %s does not have any ICO attached", pool_id)
+ continue
+
+ res[pool_id] = OndiloIcoPoolData(ico=ico, pool=pool)
if not res:
if error:
raise UpdateFailed(f"Error communicating with API: {error}") from error
- raise UpdateFailed("No data available")
return res
+
+
+class OndiloIcoMeasuresCoordinator(DataUpdateCoordinator[OndiloIcoMeasurementData]):
+ """Fetch Ondilo ICO measurement data for one pool from API."""
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ api: OndiloClient,
+ pool_id: str,
+ ) -> None:
+ """Initialize."""
+ super().__init__(
+ hass,
+ config_entry=config_entry,
+ logger=_LOGGER,
+ name=f"{DOMAIN}_measures_{pool_id}",
+ )
+ self.api = api
+ self._next_refresh: datetime | None = None
+ self._pool_id = pool_id
+
+ async def _async_update_data(self) -> OndiloIcoMeasurementData:
+ """Fetch measurement data from API endpoint."""
+ async with UPDATE_LOCK:
+ data = await self.hass.async_add_executor_job(self._update_data)
+ if next_refresh := self._next_refresh:
+ now = dt_util.utcnow()
+ # If we've missed the next refresh, schedule a refresh in one hour.
+ if next_refresh <= now:
+ next_refresh = now + timedelta(hours=1)
+ self.update_interval = next_refresh - now
+
+ return data
+
+ def _update_data(self) -> OndiloIcoMeasurementData:
+ """Fetch measurement data from API endpoint."""
+ try:
+ sensors = self.api.get_last_pool_measures(self._pool_id)
+ except OndiloError as err:
+ raise UpdateFailed(f"Error communicating with API: {err}") from err
+ return OndiloIcoMeasurementData(
+ sensors={sensor["data_type"]: sensor["value"] for sensor in sensors},
+ )
+
+ def set_next_refresh(self, pool_data: OndiloIcoPoolData) -> None:
+ """Set next refresh of this coordinator."""
+ last_update = datetime.fromisoformat(pool_data.pool["updated_at"])
+ self._next_refresh = last_update + TIME_TO_NEXT_UPDATE
diff --git a/homeassistant/components/ondilo_ico/manifest.json b/homeassistant/components/ondilo_ico/manifest.json
index 84862a89fbb..3553797b9cd 100644
--- a/homeassistant/components/ondilo_ico/manifest.json
+++ b/homeassistant/components/ondilo_ico/manifest.json
@@ -3,7 +3,7 @@
"name": "Ondilo ICO",
"codeowners": ["@JeromeHXP"],
"config_flow": true,
- "dependencies": ["auth"],
+ "dependencies": ["application_credentials"],
"documentation": "https://www.home-assistant.io/integrations/ondilo_ico",
"integration_type": "hub",
"iot_class": "cloud_polling",
diff --git a/homeassistant/components/ondilo_ico/oauth_impl.py b/homeassistant/components/ondilo_ico/oauth_impl.py
deleted file mode 100644
index e1c6e6fdb90..00000000000
--- a/homeassistant/components/ondilo_ico/oauth_impl.py
+++ /dev/null
@@ -1,32 +0,0 @@
-"""Local implementation of OAuth2 specific to Ondilo to hard code client id and secret and return a proper name."""
-
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.config_entry_oauth2_flow import LocalOAuth2Implementation
-
-from .const import (
- DOMAIN,
- OAUTH2_AUTHORIZE,
- OAUTH2_CLIENTID,
- OAUTH2_CLIENTSECRET,
- OAUTH2_TOKEN,
-)
-
-
-class OndiloOauth2Implementation(LocalOAuth2Implementation):
- """Local implementation of OAuth2 specific to Ondilo to hard code client id and secret and return a proper name."""
-
- def __init__(self, hass: HomeAssistant) -> None:
- """Just init default class with default values."""
- super().__init__(
- hass,
- DOMAIN,
- OAUTH2_CLIENTID,
- OAUTH2_CLIENTSECRET,
- OAUTH2_AUTHORIZE,
- OAUTH2_TOKEN,
- )
-
- @property
- def name(self) -> str:
- """Name of the implementation."""
- return "Ondilo"
diff --git a/homeassistant/components/ondilo_ico/sensor.py b/homeassistant/components/ondilo_ico/sensor.py
index 66b07335663..da5ccae11a5 100644
--- a/homeassistant/components/ondilo_ico/sensor.py
+++ b/homeassistant/components/ondilo_ico/sensor.py
@@ -12,17 +12,22 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONCENTRATION_PARTS_PER_MILLION,
PERCENTAGE,
+ EntityCategory,
UnitOfElectricPotential,
UnitOfTemperature,
)
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
-from .coordinator import OndiloIcoCoordinator, OndiloIcoData
+from .coordinator import (
+ OndiloIcoMeasuresCoordinator,
+ OndiloIcoPoolData,
+ OndiloIcoPoolsCoordinator,
+)
SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
SensorEntityDescription(
@@ -52,12 +57,14 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
key="battery",
native_unit_of_measurement=PERCENTAGE,
device_class=SensorDeviceClass.BATTERY,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.MEASUREMENT,
),
SensorEntityDescription(
key="rssi",
translation_key="rssi",
native_unit_of_measurement=PERCENTAGE,
+ entity_category=EntityCategory.DIAGNOSTIC,
state_class=SensorStateClass.MEASUREMENT,
),
SensorEntityDescription(
@@ -70,53 +77,72 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ondilo ICO sensors."""
+ pools_coordinator: OndiloIcoPoolsCoordinator = hass.data[DOMAIN][entry.entry_id]
+ known_entities: set[str] = set()
- coordinator: OndiloIcoCoordinator = hass.data[DOMAIN][entry.entry_id]
+ async_add_entities(get_new_entities(pools_coordinator, known_entities))
- async_add_entities(
- OndiloICO(coordinator, pool_id, description)
- for pool_id, pool in coordinator.data.items()
- for description in SENSOR_TYPES
- if description.key in pool.sensors
- )
+ @callback
+ def add_new_entities():
+ """Add any new entities after update of the pools coordinator."""
+ async_add_entities(get_new_entities(pools_coordinator, known_entities))
+
+ entry.async_on_unload(pools_coordinator.async_add_listener(add_new_entities))
-class OndiloICO(CoordinatorEntity[OndiloIcoCoordinator], SensorEntity):
+@callback
+def get_new_entities(
+ pools_coordinator: OndiloIcoPoolsCoordinator,
+ known_entities: set[str],
+) -> list[OndiloICO]:
+ """Return new Ondilo ICO sensor entities."""
+ entities = []
+ for pool_id, pool_data in pools_coordinator.data.items():
+ for description in SENSOR_TYPES:
+ measurement_id = f"{pool_id}-{description.key}"
+ if (
+ measurement_id in known_entities
+ or (data := pool_data.measures_coordinator.data) is None
+ or description.key not in data.sensors
+ ):
+ continue
+ known_entities.add(measurement_id)
+ entities.append(
+ OndiloICO(
+ pool_data.measures_coordinator, description, pool_id, pool_data
+ )
+ )
+
+ return entities
+
+
+class OndiloICO(CoordinatorEntity[OndiloIcoMeasuresCoordinator], SensorEntity):
"""Representation of a Sensor."""
_attr_has_entity_name = True
def __init__(
self,
- coordinator: OndiloIcoCoordinator,
- pool_id: str,
+ coordinator: OndiloIcoMeasuresCoordinator,
description: SensorEntityDescription,
+ pool_id: str,
+ pool_data: OndiloIcoPoolData,
) -> None:
"""Initialize sensor entity with data from coordinator."""
super().__init__(coordinator)
self.entity_description = description
-
self._pool_id = pool_id
-
- data = self.pool_data
- self._attr_unique_id = f"{data.ico['serial_number']}-{description.key}"
+ self._attr_unique_id = f"{pool_data.ico['serial_number']}-{description.key}"
self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, data.ico["serial_number"])},
- manufacturer="Ondilo",
- model="ICO",
- name=data.pool["name"],
- sw_version=data.ico["sw_version"],
+ identifiers={(DOMAIN, pool_data.ico["serial_number"])},
)
- @property
- def pool_data(self) -> OndiloIcoData:
- """Get pool data."""
- return self.coordinator.data[self._pool_id]
-
@property
def native_value(self) -> StateType:
"""Last value of the sensor."""
- return self.pool_data.sensors[self.entity_description.key]
+ return self.coordinator.data.sensors[self.entity_description.key]
diff --git a/homeassistant/components/onedrive/__init__.py b/homeassistant/components/onedrive/__init__.py
index 9716f692ec8..f5d841683d5 100644
--- a/homeassistant/components/onedrive/__init__.py
+++ b/homeassistant/components/onedrive/__init__.py
@@ -3,7 +3,6 @@
from __future__ import annotations
from collections.abc import Awaitable, Callable
-from dataclasses import dataclass
from html import unescape
from json import dumps, loads
import logging
@@ -12,83 +11,89 @@ from typing import cast
from onedrive_personal_sdk import OneDriveClient
from onedrive_personal_sdk.exceptions import (
AuthenticationError,
- HttpRequestException,
+ NotFoundError,
OneDriveException,
)
-from onedrive_personal_sdk.models.items import ItemUpdate
+from onedrive_personal_sdk.models.items import Item, ItemUpdate
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_ACCESS_TOKEN
-from homeassistant.core import HomeAssistant, callback
+from homeassistant.const import CONF_ACCESS_TOKEN, Platform
+from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
+from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.config_entry_oauth2_flow import (
OAuth2Session,
async_get_config_entry_implementation,
)
from homeassistant.helpers.instance_id import async_get as async_get_instance_id
+from homeassistant.helpers.typing import ConfigType
-from .const import DATA_BACKUP_AGENT_LISTENERS, DOMAIN
+from .const import CONF_FOLDER_ID, CONF_FOLDER_NAME, DATA_BACKUP_AGENT_LISTENERS, DOMAIN
+from .coordinator import (
+ OneDriveConfigEntry,
+ OneDriveRuntimeData,
+ OneDriveUpdateCoordinator,
+)
+from .services import async_register_services
-
-@dataclass
-class OneDriveRuntimeData:
- """Runtime data for the OneDrive integration."""
-
- client: OneDriveClient
- token_function: Callable[[], Awaitable[str]]
- backup_folder_id: str
-
-
-type OneDriveConfigEntry = ConfigEntry[OneDriveRuntimeData]
+CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
+PLATFORMS = [Platform.SENSOR]
_LOGGER = logging.getLogger(__name__)
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
+ """Set up the OneDrive integration."""
+ async_register_services(hass)
+ return True
+
+
async def async_setup_entry(hass: HomeAssistant, entry: OneDriveConfigEntry) -> bool:
"""Set up OneDrive from a config entry."""
- implementation = await async_get_config_entry_implementation(hass, entry)
- session = OAuth2Session(hass, entry, implementation)
-
- async def get_access_token() -> str:
- await session.async_ensure_token_valid()
- return cast(str, session.token[CONF_ACCESS_TOKEN])
-
- client = OneDriveClient(get_access_token, async_get_clientsession(hass))
+ client, get_access_token = await _get_onedrive_client(hass, entry)
# get approot, will be created automatically if it does not exist
- try:
- approot = await client.get_approot()
- except AuthenticationError as err:
- raise ConfigEntryAuthFailed(
- translation_domain=DOMAIN, translation_key="authentication_failed"
- ) from err
- except (HttpRequestException, OneDriveException, TimeoutError) as err:
- _LOGGER.debug("Failed to get approot", exc_info=True)
- raise ConfigEntryNotReady(
- translation_domain=DOMAIN,
- translation_key="failed_to_get_folder",
- translation_placeholders={"folder": "approot"},
- ) from err
+ approot = await _handle_item_operation(client.get_approot, "approot")
+ folder_name = entry.data[CONF_FOLDER_NAME]
- instance_id = await async_get_instance_id(hass)
- backup_folder_name = f"backups_{instance_id[:8]}"
try:
- backup_folder = await client.create_folder(
- parent_id=approot.id, name=backup_folder_name
+ backup_folder = await _handle_item_operation(
+ lambda: client.get_drive_item(path_or_id=entry.data[CONF_FOLDER_ID]),
+ folder_name,
)
- except (HttpRequestException, OneDriveException, TimeoutError) as err:
- _LOGGER.debug("Failed to create backup folder", exc_info=True)
- raise ConfigEntryNotReady(
- translation_domain=DOMAIN,
- translation_key="failed_to_get_folder",
- translation_placeholders={"folder": backup_folder_name},
- ) from err
+ except NotFoundError:
+ _LOGGER.debug("Creating backup folder %s", folder_name)
+ backup_folder = await _handle_item_operation(
+ lambda: client.create_folder(parent_id=approot.id, name=folder_name),
+ folder_name,
+ )
+ hass.config_entries.async_update_entry(
+ entry, data={**entry.data, CONF_FOLDER_ID: backup_folder.id}
+ )
+
+ # write instance id to description
+ if backup_folder.description != (instance_id := await async_get_instance_id(hass)):
+ await _handle_item_operation(
+ lambda: client.update_drive_item(
+ backup_folder.id, ItemUpdate(description=instance_id)
+ ),
+ folder_name,
+ )
+
+ # update in case folder was renamed manually inside OneDrive
+ if backup_folder.name != entry.data[CONF_FOLDER_NAME]:
+ hass.config_entries.async_update_entry(
+ entry, data={**entry.data, CONF_FOLDER_NAME: backup_folder.name}
+ )
+
+ coordinator = OneDriveUpdateCoordinator(hass, entry, client)
+ await coordinator.async_config_entry_first_refresh()
entry.runtime_data = OneDriveRuntimeData(
client=client,
token_function=get_access_token,
backup_folder_id=backup_folder.id,
+ coordinator=coordinator,
)
try:
@@ -99,25 +104,20 @@ async def async_setup_entry(hass: HomeAssistant, entry: OneDriveConfigEntry) ->
translation_key="failed_to_migrate_files",
) from err
- _async_notify_backup_listeners_soon(hass)
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+
+ def async_notify_backup_listeners() -> None:
+ for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
+ listener()
+
+ entry.async_on_unload(entry.async_on_state_change(async_notify_backup_listeners))
return True
async def async_unload_entry(hass: HomeAssistant, entry: OneDriveConfigEntry) -> bool:
"""Unload a OneDrive config entry."""
- _async_notify_backup_listeners_soon(hass)
- return True
-
-
-def _async_notify_backup_listeners(hass: HomeAssistant) -> None:
- for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
- listener()
-
-
-@callback
-def _async_notify_backup_listeners_soon(hass: HomeAssistant) -> None:
- hass.loop.call_soon(_async_notify_backup_listeners, hass)
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
async def _migrate_backup_files(client: OneDriveClient, backup_folder_id: str) -> None:
@@ -133,7 +133,7 @@ async def _migrate_backup_files(client: OneDriveClient, backup_folder_id: str) -
metadata_file = await client.upload_file(
backup_folder_id,
metadata_filename,
- dumps(metadata), # type: ignore[arg-type]
+ dumps(metadata),
)
metadata_description = {
"metadata_version": 2,
@@ -149,3 +149,74 @@ async def _migrate_backup_files(client: OneDriveClient, backup_folder_id: str) -
data=ItemUpdate(description=""),
)
_LOGGER.debug("Migrated backup file %s", file.name)
+
+
+async def async_migrate_entry(hass: HomeAssistant, entry: OneDriveConfigEntry) -> bool:
+ """Migrate old entry."""
+ if entry.version > 1:
+ # This means the user has downgraded from a future version
+ return False
+
+ if (version := entry.version) == 1 and (minor_version := entry.minor_version) == 1:
+ _LOGGER.debug(
+ "Migrating OneDrive config entry from version %s.%s", version, minor_version
+ )
+ client, _ = await _get_onedrive_client(hass, entry)
+ instance_id = await async_get_instance_id(hass)
+ try:
+ approot = await client.get_approot()
+ folder = await client.get_drive_item(
+ f"{approot.id}:/backups_{instance_id[:8]}:"
+ )
+ except OneDriveException:
+ _LOGGER.exception("Migration to version 1.2 failed")
+ return False
+
+ hass.config_entries.async_update_entry(
+ entry,
+ data={
+ **entry.data,
+ CONF_FOLDER_ID: folder.id,
+ CONF_FOLDER_NAME: f"backups_{instance_id[:8]}",
+ },
+ minor_version=2,
+ )
+ _LOGGER.debug("Migration to version 1.2 successful")
+ return True
+
+
+async def _get_onedrive_client(
+ hass: HomeAssistant, entry: OneDriveConfigEntry
+) -> tuple[OneDriveClient, Callable[[], Awaitable[str]]]:
+ """Get OneDrive client."""
+ implementation = await async_get_config_entry_implementation(hass, entry)
+ session = OAuth2Session(hass, entry, implementation)
+
+ async def get_access_token() -> str:
+ await session.async_ensure_token_valid()
+ return cast(str, session.token[CONF_ACCESS_TOKEN])
+
+ return (
+ OneDriveClient(get_access_token, async_get_clientsession(hass)),
+ get_access_token,
+ )
+
+
+async def _handle_item_operation(
+ func: Callable[[], Awaitable[Item]], folder: str
+) -> Item:
+ try:
+ return await func()
+ except NotFoundError:
+ raise
+ except AuthenticationError as err:
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN, translation_key="authentication_failed"
+ ) from err
+ except (OneDriveException, TimeoutError) as err:
+ _LOGGER.debug("Failed to get approot", exc_info=True)
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="failed_to_get_folder",
+ translation_placeholders={"folder": folder},
+ ) from err
diff --git a/homeassistant/components/onedrive/backup.py b/homeassistant/components/onedrive/backup.py
index 182e29aa63f..41a244506ea 100644
--- a/homeassistant/components/onedrive/backup.py
+++ b/homeassistant/components/onedrive/backup.py
@@ -3,10 +3,12 @@
from __future__ import annotations
from collections.abc import AsyncIterator, Callable, Coroutine
+from dataclasses import dataclass
from functools import wraps
from html import unescape
from json import dumps, loads
import logging
+from time import time
from typing import Any, Concatenate
from aiohttp import ClientTimeout
@@ -16,25 +18,27 @@ from onedrive_personal_sdk.exceptions import (
HashMismatchError,
OneDriveException,
)
-from onedrive_personal_sdk.models.items import File, Folder, ItemUpdate
+from onedrive_personal_sdk.models.items import ItemUpdate
from onedrive_personal_sdk.models.upload import FileInfo
from homeassistant.components.backup import (
AgentBackup,
BackupAgent,
BackupAgentError,
+ BackupNotFound,
suggested_filename,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from . import OneDriveConfigEntry
-from .const import DATA_BACKUP_AGENT_LISTENERS, DOMAIN
+from .const import CONF_DELETE_PERMANENTLY, DATA_BACKUP_AGENT_LISTENERS, DOMAIN
+from .coordinator import OneDriveConfigEntry
_LOGGER = logging.getLogger(__name__)
UPLOAD_CHUNK_SIZE = 16 * 320 * 1024 # 5.2MB
TIMEOUT = ClientTimeout(connect=10, total=43200) # 12 hours
METADATA_VERSION = 2
+CACHE_TTL = 300
async def async_get_backup_agents(
@@ -70,7 +74,7 @@ def async_register_backup_agents_listener(
def handle_backup_errors[_R, **P](
func: Callable[Concatenate[OneDriveBackupAgent, P], Coroutine[Any, Any, _R]],
) -> Callable[Concatenate[OneDriveBackupAgent, P], Coroutine[Any, Any, _R]]:
- """Handle backup errors with a specific translation key."""
+ """Handle backup errors."""
@wraps(func)
async def wrapper(
@@ -99,6 +103,15 @@ def handle_backup_errors[_R, **P](
return wrapper
+@dataclass(kw_only=True)
+class OneDriveBackup:
+ """Define a OneDrive backup."""
+
+ backup: AgentBackup
+ backup_file_id: str
+ metadata_file_id: str
+
+
class OneDriveBackupAgent(BackupAgent):
"""OneDrive backup agent."""
@@ -115,24 +128,20 @@ class OneDriveBackupAgent(BackupAgent):
self.name = entry.title
assert entry.unique_id
self.unique_id = entry.unique_id
+ self._backup_cache: dict[str, OneDriveBackup] = {}
+ self._cache_expiration = time()
@handle_backup_errors
async def async_download_backup(
self, backup_id: str, **kwargs: Any
) -> AsyncIterator[bytes]:
"""Download a backup file."""
- metadata_item = await self._find_item_by_backup_id(backup_id)
- if (
- metadata_item is None
- or metadata_item.description is None
- or "backup_file_id" not in metadata_item.description
- ):
- raise BackupAgentError("Backup not found")
-
- metadata_info = loads(unescape(metadata_item.description))
+ backups = await self._list_cached_backups()
+ if backup_id not in backups:
+ raise BackupNotFound(f"Backup {backup_id} not found")
stream = await self._client.download_drive_item(
- metadata_info["backup_file_id"], timeout=TIMEOUT
+ backups[backup_id].backup_file_id, timeout=TIMEOUT
)
return stream.iter_chunked(1024)
@@ -168,7 +177,7 @@ class OneDriveBackupAgent(BackupAgent):
metadata_file = await self._client.upload_file(
self._folder_id,
metadata_filename,
- description, # type: ignore[arg-type]
+ description,
)
# add metadata to the metadata file
@@ -181,6 +190,7 @@ class OneDriveBackupAgent(BackupAgent):
path_or_id=metadata_file.id,
data=ItemUpdate(description=dumps(metadata_description)),
)
+ self._cache_expiration = time()
@handle_backup_errors
async def async_delete_backup(
@@ -189,56 +199,60 @@ class OneDriveBackupAgent(BackupAgent):
**kwargs: Any,
) -> None:
"""Delete a backup file."""
- metadata_item = await self._find_item_by_backup_id(backup_id)
- if (
- metadata_item is None
- or metadata_item.description is None
- or "backup_file_id" not in metadata_item.description
- ):
- return
- metadata_info = loads(unescape(metadata_item.description))
+ backups = await self._list_cached_backups()
+ if backup_id not in backups:
+ raise BackupNotFound(f"Backup {backup_id} not found")
- await self._client.delete_drive_item(metadata_info["backup_file_id"])
- await self._client.delete_drive_item(metadata_item.id)
+ backup = backups[backup_id]
+
+ delete_permanently = self._entry.options.get(CONF_DELETE_PERMANENTLY, False)
+
+ await self._client.delete_drive_item(backup.backup_file_id, delete_permanently)
+ await self._client.delete_drive_item(
+ backup.metadata_file_id, delete_permanently
+ )
+ self._cache_expiration = time()
@handle_backup_errors
async def async_list_backups(self, **kwargs: Any) -> list[AgentBackup]:
"""List backups."""
- items = await self._client.list_drive_items(self._folder_id)
return [
- await self._download_backup_metadata(item.id)
- for item in items
- if item.description
- and "backup_id" in item.description
- and f'"metadata_version": {METADATA_VERSION}' in unescape(item.description)
+ backup.backup for backup in (await self._list_cached_backups()).values()
]
@handle_backup_errors
- async def async_get_backup(
- self, backup_id: str, **kwargs: Any
- ) -> AgentBackup | None:
+ async def async_get_backup(self, backup_id: str, **kwargs: Any) -> AgentBackup:
"""Return a backup."""
- metadata_file = await self._find_item_by_backup_id(backup_id)
- if metadata_file is None or metadata_file.description is None:
- return None
+ backups = await self._list_cached_backups()
+ if backup_id not in backups:
+ raise BackupNotFound(f"Backup {backup_id} not found")
+ return backups[backup_id].backup
- return await self._download_backup_metadata(metadata_file.id)
+ async def _list_cached_backups(self) -> dict[str, OneDriveBackup]:
+ """List backups with a cache."""
+ if time() <= self._cache_expiration:
+ return self._backup_cache
- async def _find_item_by_backup_id(self, backup_id: str) -> File | Folder | None:
- """Find an item by backup ID."""
- return next(
- (
- item
- for item in await self._client.list_drive_items(self._folder_id)
- if item.description
- and backup_id in item.description
- and f'"metadata_version": {METADATA_VERSION}'
- in unescape(item.description)
- ),
- None,
- )
+ items = await self._client.list_drive_items(self._folder_id)
- async def _download_backup_metadata(self, item_id: str) -> AgentBackup:
- metadata_stream = await self._client.download_drive_item(item_id)
- metadata_json = loads(await metadata_stream.read())
- return AgentBackup.from_dict(metadata_json)
+ async def download_backup_metadata(item_id: str) -> AgentBackup:
+ metadata_stream = await self._client.download_drive_item(item_id)
+ metadata_json = loads(await metadata_stream.read())
+ return AgentBackup.from_dict(metadata_json)
+
+ backups: dict[str, OneDriveBackup] = {}
+ for item in items:
+ if item.description and f'"metadata_version": {METADATA_VERSION}' in (
+ metadata_description_json := unescape(item.description)
+ ):
+ backup = await download_backup_metadata(item.id)
+ metadata_description = loads(metadata_description_json)
+ backups[backup.backup_id] = OneDriveBackup(
+ backup=backup,
+ backup_file_id=metadata_description["backup_file_id"],
+ metadata_file_id=item.id,
+ )
+
+ self._cache_expiration = time() + CACHE_TTL
+ self._backup_cache = backups
+ return backups
diff --git a/homeassistant/components/onedrive/config_flow.py b/homeassistant/components/onedrive/config_flow.py
index 900db0177d9..3374c0369ee 100644
--- a/homeassistant/components/onedrive/config_flow.py
+++ b/homeassistant/components/onedrive/config_flow.py
@@ -1,24 +1,54 @@
"""Config flow for OneDrive."""
+from __future__ import annotations
+
from collections.abc import Mapping
import logging
from typing import Any, cast
from onedrive_personal_sdk.clients.client import OneDriveClient
from onedrive_personal_sdk.exceptions import OneDriveException
+from onedrive_personal_sdk.models.items import AppRoot, ItemUpdate
+import voluptuous as vol
-from homeassistant.config_entries import SOURCE_REAUTH, ConfigFlowResult
+from homeassistant.config_entries import (
+ SOURCE_REAUTH,
+ SOURCE_RECONFIGURE,
+ SOURCE_USER,
+ ConfigFlowResult,
+ OptionsFlow,
+)
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_TOKEN
+from homeassistant.core import callback
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.config_entry_oauth2_flow import AbstractOAuth2FlowHandler
+from homeassistant.helpers.instance_id import async_get as async_get_instance_id
-from .const import DOMAIN, OAUTH_SCOPES
+from .const import (
+ CONF_DELETE_PERMANENTLY,
+ CONF_FOLDER_ID,
+ CONF_FOLDER_NAME,
+ DOMAIN,
+ OAUTH_SCOPES,
+)
+from .coordinator import OneDriveConfigEntry
+
+FOLDER_NAME_SCHEMA = vol.Schema({vol.Required(CONF_FOLDER_NAME): str})
class OneDriveConfigFlow(AbstractOAuth2FlowHandler, domain=DOMAIN):
"""Config flow to handle OneDrive OAuth2 authentication."""
DOMAIN = DOMAIN
+ MINOR_VERSION = 2
+
+ client: OneDriveClient
+ approot: AppRoot
+
+ def __init__(self) -> None:
+ """Initialize the OneDrive config flow."""
+ super().__init__()
+ self.step_data: dict[str, Any] = {}
@property
def logger(self) -> logging.Logger:
@@ -30,6 +60,15 @@ class OneDriveConfigFlow(AbstractOAuth2FlowHandler, domain=DOMAIN):
"""Extra data that needs to be appended to the authorize url."""
return {"scope": " ".join(OAUTH_SCOPES)}
+ @property
+ def apps_folder(self) -> str:
+ """Return the name of the Apps folder (translated)."""
+ return (
+ path.split("/")[-1]
+ if (path := self.approot.parent_reference.path)
+ else "Apps"
+ )
+
async def async_oauth_create_entry(
self,
data: dict[str, Any],
@@ -39,12 +78,12 @@ class OneDriveConfigFlow(AbstractOAuth2FlowHandler, domain=DOMAIN):
async def get_access_token() -> str:
return cast(str, data[CONF_TOKEN][CONF_ACCESS_TOKEN])
- graph_client = OneDriveClient(
+ self.client = OneDriveClient(
get_access_token, async_get_clientsession(self.hass)
)
try:
- approot = await graph_client.get_approot()
+ self.approot = await self.client.get_approot()
except OneDriveException:
self.logger.exception("Failed to connect to OneDrive")
return self.async_abort(reason="connection_error")
@@ -52,26 +91,118 @@ class OneDriveConfigFlow(AbstractOAuth2FlowHandler, domain=DOMAIN):
self.logger.exception("Unknown error")
return self.async_abort(reason="unknown")
- await self.async_set_unique_id(approot.parent_reference.drive_id)
+ await self.async_set_unique_id(self.approot.parent_reference.drive_id)
- if self.source == SOURCE_REAUTH:
- reauth_entry = self._get_reauth_entry()
+ if self.source != SOURCE_USER:
self._abort_if_unique_id_mismatch(
reason="wrong_drive",
)
+
+ if self.source == SOURCE_REAUTH:
+ reauth_entry = self._get_reauth_entry()
return self.async_update_reload_and_abort(
entry=reauth_entry,
data=data,
)
- self._abort_if_unique_id_configured()
+ if self.source != SOURCE_RECONFIGURE:
+ self._abort_if_unique_id_configured()
- title = (
- f"{approot.created_by.user.display_name}'s OneDrive"
- if approot.created_by.user and approot.created_by.user.display_name
- else "OneDrive"
+ self.step_data = data
+
+ if self.source == SOURCE_RECONFIGURE:
+ return await self.async_step_reconfigure_folder()
+
+ return await self.async_step_folder_name()
+
+ async def async_step_folder_name(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Step to ask for the folder name."""
+ errors: dict[str, str] = {}
+ instance_id = await async_get_instance_id(self.hass)
+ if user_input is not None:
+ try:
+ folder = await self.client.create_folder(
+ self.approot.id, user_input[CONF_FOLDER_NAME]
+ )
+ except OneDriveException:
+ self.logger.debug("Failed to create folder", exc_info=True)
+ errors["base"] = "folder_creation_error"
+ else:
+ if folder.description and folder.description != instance_id:
+ errors[CONF_FOLDER_NAME] = "folder_already_in_use"
+ if not errors:
+ title = (
+ f"{self.approot.created_by.user.display_name}'s OneDrive"
+ if self.approot.created_by.user
+ and self.approot.created_by.user.display_name
+ else "OneDrive"
+ )
+ return self.async_create_entry(
+ title=title,
+ data={
+ **self.step_data,
+ CONF_FOLDER_ID: folder.id,
+ CONF_FOLDER_NAME: user_input[CONF_FOLDER_NAME],
+ },
+ )
+
+ default_folder_name = (
+ f"backups_{instance_id[:8]}"
+ if user_input is None
+ else user_input[CONF_FOLDER_NAME]
+ )
+
+ return self.async_show_form(
+ step_id="folder_name",
+ data_schema=self.add_suggested_values_to_schema(
+ FOLDER_NAME_SCHEMA, {CONF_FOLDER_NAME: default_folder_name}
+ ),
+ description_placeholders={
+ "apps_folder": self.apps_folder,
+ "approot": self.approot.name,
+ },
+ errors=errors,
+ )
+
+ async def async_step_reconfigure_folder(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Reconfigure the folder name."""
+ errors: dict[str, str] = {}
+ reconfigure_entry = self._get_reconfigure_entry()
+
+ if user_input is not None:
+ if (
+ new_folder_name := user_input[CONF_FOLDER_NAME]
+ ) != reconfigure_entry.data[CONF_FOLDER_NAME]:
+ try:
+ await self.client.update_drive_item(
+ reconfigure_entry.data[CONF_FOLDER_ID],
+ ItemUpdate(name=new_folder_name),
+ )
+ except OneDriveException:
+ self.logger.debug("Failed to update folder", exc_info=True)
+ errors["base"] = "folder_rename_error"
+ if not errors:
+ return self.async_update_reload_and_abort(
+ reconfigure_entry,
+ data={**reconfigure_entry.data, CONF_FOLDER_NAME: new_folder_name},
+ )
+
+ return self.async_show_form(
+ step_id="reconfigure_folder",
+ data_schema=self.add_suggested_values_to_schema(
+ FOLDER_NAME_SCHEMA,
+ {CONF_FOLDER_NAME: reconfigure_entry.data[CONF_FOLDER_NAME]},
+ ),
+ description_placeholders={
+ "apps_folder": self.apps_folder,
+ "approot": self.approot.name,
+ },
+ errors=errors,
)
- return self.async_create_entry(title=title, data=data)
async def async_step_reauth(
self, entry_data: Mapping[str, Any]
@@ -86,3 +217,44 @@ class OneDriveConfigFlow(AbstractOAuth2FlowHandler, domain=DOMAIN):
if user_input is None:
return self.async_show_form(step_id="reauth_confirm")
return await self.async_step_user()
+
+ async def async_step_reconfigure(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Reconfigure the entry."""
+ return await self.async_step_user()
+
+ @staticmethod
+ @callback
+ def async_get_options_flow(
+ config_entry: OneDriveConfigEntry,
+ ) -> OneDriveOptionsFlowHandler:
+ """Create the options flow."""
+ return OneDriveOptionsFlowHandler()
+
+
+class OneDriveOptionsFlowHandler(OptionsFlow):
+ """Handles options flow for the component."""
+
+ async def async_step_init(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Manage the options for OneDrive."""
+ if user_input:
+ return self.async_create_entry(title="", data=user_input)
+
+ options_schema = vol.Schema(
+ {
+ vol.Optional(
+ CONF_DELETE_PERMANENTLY,
+ default=self.config_entry.options.get(
+ CONF_DELETE_PERMANENTLY, False
+ ),
+ ): bool,
+ }
+ )
+
+ return self.async_show_form(
+ step_id="init",
+ data_schema=options_schema,
+ )
diff --git a/homeassistant/components/onedrive/const.py b/homeassistant/components/onedrive/const.py
index f9d49b141e5..fd21d84369c 100644
--- a/homeassistant/components/onedrive/const.py
+++ b/homeassistant/components/onedrive/const.py
@@ -6,6 +6,10 @@ from typing import Final
from homeassistant.util.hass_dict import HassKey
DOMAIN: Final = "onedrive"
+CONF_FOLDER_NAME: Final = "folder_name"
+CONF_FOLDER_ID: Final = "folder_id"
+
+CONF_DELETE_PERMANENTLY: Final = "delete_permanently"
# replace "consumers" with "common", when adding SharePoint or OneDrive for Business support
OAUTH2_AUTHORIZE: Final = (
diff --git a/homeassistant/components/onedrive/coordinator.py b/homeassistant/components/onedrive/coordinator.py
new file mode 100644
index 00000000000..7b2dbaab87a
--- /dev/null
+++ b/homeassistant/components/onedrive/coordinator.py
@@ -0,0 +1,95 @@
+"""Coordinator for OneDrive."""
+
+from __future__ import annotations
+
+from collections.abc import Awaitable, Callable
+from dataclasses import dataclass
+from datetime import timedelta
+import logging
+
+from onedrive_personal_sdk import OneDriveClient
+from onedrive_personal_sdk.const import DriveState
+from onedrive_personal_sdk.exceptions import AuthenticationError, OneDriveException
+from onedrive_personal_sdk.models.items import Drive
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed
+from homeassistant.helpers import issue_registry as ir
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+from .const import DOMAIN
+
+SCAN_INTERVAL = timedelta(minutes=5)
+
+_LOGGER = logging.getLogger(__name__)
+
+
+@dataclass
+class OneDriveRuntimeData:
+ """Runtime data for the OneDrive integration."""
+
+ client: OneDriveClient
+ token_function: Callable[[], Awaitable[str]]
+ backup_folder_id: str
+ coordinator: OneDriveUpdateCoordinator
+
+
+type OneDriveConfigEntry = ConfigEntry[OneDriveRuntimeData]
+
+
+class OneDriveUpdateCoordinator(DataUpdateCoordinator[Drive]):
+ """Class to handle fetching data from the Graph API centrally."""
+
+ config_entry: OneDriveConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, entry: OneDriveConfigEntry, client: OneDriveClient
+ ) -> None:
+ """Initialize coordinator."""
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=entry,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
+ )
+ self._client = client
+
+ async def _async_update_data(self) -> Drive:
+ """Fetch data from API endpoint."""
+
+ try:
+ drive = await self._client.get_drive()
+ except AuthenticationError as err:
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN, translation_key="authentication_failed"
+ ) from err
+ except OneDriveException as err:
+ raise UpdateFailed(
+ translation_domain=DOMAIN, translation_key="update_failed"
+ ) from err
+
+ # create an issue if the drive is almost full
+ if drive.quota and (state := drive.quota.state) in (
+ DriveState.CRITICAL,
+ DriveState.EXCEEDED,
+ ):
+ key = "drive_full" if state is DriveState.EXCEEDED else "drive_almost_full"
+ ir.async_create_issue(
+ self.hass,
+ DOMAIN,
+ key,
+ is_fixable=False,
+ severity=(
+ ir.IssueSeverity.ERROR
+ if state is DriveState.EXCEEDED
+ else ir.IssueSeverity.WARNING
+ ),
+ translation_key=key,
+ translation_placeholders={
+ "total": str(drive.quota.total),
+ "used": str(drive.quota.used),
+ },
+ )
+ return drive
diff --git a/homeassistant/components/onedrive/diagnostics.py b/homeassistant/components/onedrive/diagnostics.py
new file mode 100644
index 00000000000..0e1ed94e155
--- /dev/null
+++ b/homeassistant/components/onedrive/diagnostics.py
@@ -0,0 +1,33 @@
+"""Diagnostics support for OneDrive."""
+
+from __future__ import annotations
+
+from dataclasses import asdict
+from typing import Any
+
+from homeassistant.components.diagnostics import async_redact_data
+from homeassistant.const import CONF_ACCESS_TOKEN, CONF_TOKEN
+from homeassistant.core import HomeAssistant
+
+from .coordinator import OneDriveConfigEntry
+
+TO_REDACT = {"display_name", "email", CONF_ACCESS_TOKEN, CONF_TOKEN}
+
+
+async def async_get_config_entry_diagnostics(
+ hass: HomeAssistant,
+ entry: OneDriveConfigEntry,
+) -> dict[str, Any]:
+ """Return diagnostics for a config entry."""
+
+ coordinator = entry.runtime_data.coordinator
+
+ data = {
+ "drive": asdict(coordinator.data),
+ "config": {
+ **entry.data,
+ **entry.options,
+ },
+ }
+
+ return async_redact_data(data, TO_REDACT)
diff --git a/homeassistant/components/onedrive/icons.json b/homeassistant/components/onedrive/icons.json
new file mode 100644
index 00000000000..2ac4921439c
--- /dev/null
+++ b/homeassistant/components/onedrive/icons.json
@@ -0,0 +1,29 @@
+{
+ "entity": {
+ "sensor": {
+ "total_size": {
+ "default": "mdi:database"
+ },
+ "used_size": {
+ "default": "mdi:database"
+ },
+ "remaining_size": {
+ "default": "mdi:database"
+ },
+ "drive_state": {
+ "default": "mdi:harddisk",
+ "state": {
+ "normal": "mdi:harddisk",
+ "nearing": "mdi:alert-circle-outline",
+ "critical": "mdi:alert",
+ "exceeded": "mdi:alert-octagon"
+ }
+ }
+ }
+ },
+ "services": {
+ "upload": {
+ "service": "mdi:cloud-upload"
+ }
+ }
+}
diff --git a/homeassistant/components/onedrive/manifest.json b/homeassistant/components/onedrive/manifest.json
index 88d51e6d73a..c3d98200b03 100644
--- a/homeassistant/components/onedrive/manifest.json
+++ b/homeassistant/components/onedrive/manifest.json
@@ -8,6 +8,6 @@
"integration_type": "service",
"iot_class": "cloud_polling",
"loggers": ["onedrive_personal_sdk"],
- "quality_scale": "bronze",
- "requirements": ["onedrive-personal-sdk==0.0.8"]
+ "quality_scale": "platinum",
+ "requirements": ["onedrive-personal-sdk==0.0.13"]
}
diff --git a/homeassistant/components/onedrive/quality_scale.yaml b/homeassistant/components/onedrive/quality_scale.yaml
index f0d58d89c9a..1632c2670e0 100644
--- a/homeassistant/components/onedrive/quality_scale.yaml
+++ b/homeassistant/components/onedrive/quality_scale.yaml
@@ -1,21 +1,13 @@
rules:
# Bronze
- action-setup:
- status: exempt
- comment: Integration does not register custom actions.
- appropriate-polling:
- status: exempt
- comment: |
- This integration does not poll.
+ action-setup: done
+ appropriate-polling: done
brands: done
common-modules: done
config-flow-test-coverage: done
config-flow: done
dependency-transparency: done
- docs-actions:
- status: exempt
- comment: |
- This integration does not have any custom actions.
+ docs-actions: done
docs-high-level-description: done
docs-installation-instructions: done
docs-removal-instructions: done
@@ -23,14 +15,8 @@ rules:
status: exempt
comment: |
Entities of this integration does not explicitly subscribe to events.
- entity-unique-id:
- status: exempt
- comment: |
- This integration does not have entities.
- has-entity-name:
- status: exempt
- comment: |
- This integration does not have entities.
+ entity-unique-id: done
+ has-entity-name: done
runtime-data: done
test-before-configure: done
test-before-setup: done
@@ -39,36 +25,18 @@ rules:
# Silver
action-exceptions: done
config-entry-unloading: done
- docs-configuration-parameters:
- status: exempt
- comment: |
- No Options flow.
+ docs-configuration-parameters: done
docs-installation-parameters: done
- entity-unavailable:
- status: exempt
- comment: |
- This integration does not have entities.
+ entity-unavailable: done
integration-owner: done
- log-when-unavailable:
- status: exempt
- comment: |
- This integration does not have entities.
- parallel-updates:
- status: exempt
- comment: |
- This integration does not have platforms.
+ log-when-unavailable: done
+ parallel-updates: done
reauthentication-flow: done
- test-coverage: todo
+ test-coverage: done
# Gold
- devices:
- status: exempt
- comment: |
- This integration connects to a single service.
- diagnostics:
- status: exempt
- comment: |
- There is no data to diagnose.
+ devices: done
+ diagnostics: done
discovery-update-info:
status: exempt
comment: |
@@ -77,57 +45,27 @@ rules:
status: exempt
comment: |
This integration is a cloud service and does not support discovery.
- docs-data-update:
- status: exempt
- comment: |
- This integration does not poll or push.
- docs-examples:
- status: exempt
- comment: |
- This integration only serves backup.
+ docs-data-update: done
+ docs-examples: done
docs-known-limitations: done
docs-supported-devices:
status: exempt
comment: |
This integration is a cloud service.
- docs-supported-functions:
- status: exempt
- comment: |
- This integration does not have entities.
- docs-troubleshooting:
- status: exempt
- comment: |
- No issues known to troubleshoot.
+ docs-supported-functions: done
+ docs-troubleshooting: done
docs-use-cases: done
dynamic-devices:
status: exempt
comment: |
This integration connects to a single service.
- entity-category:
- status: exempt
- comment: |
- This integration does not have entities.
- entity-device-class:
- status: exempt
- comment: |
- This integration does not have entities.
- entity-disabled-by-default:
- status: exempt
- comment: |
- This integration does not have entities.
- entity-translations:
- status: exempt
- comment: |
- This integration does not have entities.
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default: done
+ entity-translations: done
exception-translations: done
- icon-translations:
- status: exempt
- comment: |
- This integration does not have entities.
- reconfiguration-flow:
- status: exempt
- comment: |
- Nothing to reconfigure.
+ icon-translations: done
+ reconfiguration-flow: done
repair-issues: done
stale-devices:
status: exempt
diff --git a/homeassistant/components/onedrive/sensor.py b/homeassistant/components/onedrive/sensor.py
new file mode 100644
index 00000000000..fa7c0b125fe
--- /dev/null
+++ b/homeassistant/components/onedrive/sensor.py
@@ -0,0 +1,122 @@
+"""Sensors for OneDrive."""
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from onedrive_personal_sdk.const import DriveState
+from onedrive_personal_sdk.models.items import DriveQuota
+
+from homeassistant.components.sensor import (
+ SensorDeviceClass,
+ SensorEntity,
+ SensorEntityDescription,
+)
+from homeassistant.const import EntityCategory, UnitOfInformation
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.typing import StateType
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+from .coordinator import OneDriveConfigEntry, OneDriveUpdateCoordinator
+
+PARALLEL_UPDATES = 0
+
+
+@dataclass(kw_only=True, frozen=True)
+class OneDriveSensorEntityDescription(SensorEntityDescription):
+ """Describes OneDrive sensor entity."""
+
+ value_fn: Callable[[DriveQuota], StateType]
+
+
+DRIVE_STATE_ENTITIES: tuple[OneDriveSensorEntityDescription, ...] = (
+ OneDriveSensorEntityDescription(
+ key="total_size",
+ value_fn=lambda quota: quota.total,
+ native_unit_of_measurement=UnitOfInformation.BYTES,
+ suggested_unit_of_measurement=UnitOfInformation.GIBIBYTES,
+ suggested_display_precision=0,
+ device_class=SensorDeviceClass.DATA_SIZE,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ entity_registry_enabled_default=False,
+ ),
+ OneDriveSensorEntityDescription(
+ key="used_size",
+ value_fn=lambda quota: quota.used,
+ native_unit_of_measurement=UnitOfInformation.BYTES,
+ suggested_unit_of_measurement=UnitOfInformation.GIBIBYTES,
+ suggested_display_precision=2,
+ device_class=SensorDeviceClass.DATA_SIZE,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ OneDriveSensorEntityDescription(
+ key="remaining_size",
+ value_fn=lambda quota: quota.remaining,
+ native_unit_of_measurement=UnitOfInformation.BYTES,
+ suggested_unit_of_measurement=UnitOfInformation.GIBIBYTES,
+ suggested_display_precision=2,
+ device_class=SensorDeviceClass.DATA_SIZE,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ OneDriveSensorEntityDescription(
+ key="drive_state",
+ value_fn=lambda quota: quota.state.value,
+ options=[state.value for state in DriveState],
+ device_class=SensorDeviceClass.ENUM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: OneDriveConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up OneDrive sensors based on a config entry."""
+ coordinator = entry.runtime_data.coordinator
+ async_add_entities(
+ OneDriveDriveStateSensor(coordinator, description)
+ for description in DRIVE_STATE_ENTITIES
+ )
+
+
+class OneDriveDriveStateSensor(
+ CoordinatorEntity[OneDriveUpdateCoordinator], SensorEntity
+):
+ """Define a OneDrive sensor."""
+
+ entity_description: OneDriveSensorEntityDescription
+ _attr_has_entity_name = True
+
+ def __init__(
+ self,
+ coordinator: OneDriveUpdateCoordinator,
+ description: OneDriveSensorEntityDescription,
+ ) -> None:
+ """Initialize the sensor."""
+ super().__init__(coordinator)
+ self.entity_description = description
+ self._attr_translation_key = description.key
+ self._attr_unique_id = f"{coordinator.data.id}_{description.key}"
+ self._attr_device_info = DeviceInfo(
+ entry_type=DeviceEntryType.SERVICE,
+ name=coordinator.data.name or coordinator.config_entry.title,
+ identifiers={(DOMAIN, coordinator.data.id)},
+ manufacturer="Microsoft",
+ model=f"OneDrive {coordinator.data.drive_type.value.capitalize()}",
+ configuration_url=f"https://onedrive.live.com/?id=root&cid={coordinator.data.id}",
+ )
+
+ @property
+ def native_value(self) -> StateType:
+ """Return the state of the sensor."""
+ assert self.coordinator.data.quota
+ return self.entity_description.value_fn(self.coordinator.data.quota)
+
+ @property
+ def available(self) -> bool:
+ """Availability of the sensor."""
+ return super().available and self.coordinator.data.quota is not None
diff --git a/homeassistant/components/onedrive/services.py b/homeassistant/components/onedrive/services.py
new file mode 100644
index 00000000000..1f1afe1507c
--- /dev/null
+++ b/homeassistant/components/onedrive/services.py
@@ -0,0 +1,131 @@
+"""OneDrive services."""
+
+from __future__ import annotations
+
+import asyncio
+from dataclasses import asdict
+from pathlib import Path
+from typing import cast
+
+from onedrive_personal_sdk.exceptions import OneDriveException
+import voluptuous as vol
+
+from homeassistant.const import CONF_FILENAME
+from homeassistant.core import (
+ HomeAssistant,
+ ServiceCall,
+ ServiceResponse,
+ SupportsResponse,
+)
+from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
+from homeassistant.helpers import config_validation as cv
+
+from .const import DOMAIN
+from .coordinator import OneDriveConfigEntry
+
+CONF_CONFIG_ENTRY_ID = "config_entry_id"
+CONF_DESTINATION_FOLDER = "destination_folder"
+
+UPLOAD_SERVICE = "upload"
+UPLOAD_SERVICE_SCHEMA = vol.Schema(
+ {
+ vol.Required(CONF_CONFIG_ENTRY_ID): cv.string,
+ vol.Required(CONF_FILENAME): vol.All(cv.ensure_list, [cv.string]),
+ vol.Required(CONF_DESTINATION_FOLDER): cv.string,
+ }
+)
+CONTENT_SIZE_LIMIT = 250 * 1024 * 1024
+
+
+def _read_file_contents(
+ hass: HomeAssistant, filenames: list[str]
+) -> list[tuple[str, bytes]]:
+ """Return the mime types and file contents for each file."""
+ results = []
+ for filename in filenames:
+ if not hass.config.is_allowed_path(filename):
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="no_access_to_path",
+ translation_placeholders={"filename": filename},
+ )
+ filename_path = Path(filename)
+ if not filename_path.exists():
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="filename_does_not_exist",
+ translation_placeholders={"filename": filename},
+ )
+ if filename_path.stat().st_size > CONTENT_SIZE_LIMIT:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="file_too_large",
+ translation_placeholders={
+ "filename": filename,
+ "size": str(filename_path.stat().st_size),
+ "limit": str(CONTENT_SIZE_LIMIT),
+ },
+ )
+ results.append((filename_path.name, filename_path.read_bytes()))
+ return results
+
+
+def async_register_services(hass: HomeAssistant) -> None:
+ """Register OneDrive services."""
+
+ async def async_handle_upload(call: ServiceCall) -> ServiceResponse:
+ """Generate content from text and optionally images."""
+ config_entry: OneDriveConfigEntry | None = hass.config_entries.async_get_entry(
+ call.data[CONF_CONFIG_ENTRY_ID]
+ )
+ if not config_entry:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="integration_not_found",
+ translation_placeholders={"target": DOMAIN},
+ )
+ client = config_entry.runtime_data.client
+ upload_tasks = []
+ file_results = await hass.async_add_executor_job(
+ _read_file_contents, hass, call.data[CONF_FILENAME]
+ )
+
+ # make sure the destination folder exists
+ try:
+ folder_id = (await client.get_approot()).id
+ for folder in (
+ cast(str, call.data[CONF_DESTINATION_FOLDER]).strip("/").split("/")
+ ):
+ folder_id = (await client.create_folder(folder_id, folder)).id
+ except OneDriveException as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="create_folder_error",
+ translation_placeholders={"message": str(err)},
+ ) from err
+
+ upload_tasks = [
+ client.upload_file(folder_id, file_name, content)
+ for file_name, content in file_results
+ ]
+ try:
+ upload_results = await asyncio.gather(*upload_tasks)
+ except OneDriveException as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="upload_error",
+ translation_placeholders={"message": str(err)},
+ ) from err
+
+ if call.return_response:
+ return {"files": [asdict(item_result) for item_result in upload_results]}
+ return None
+
+ if not hass.services.has_service(DOMAIN, UPLOAD_SERVICE):
+ hass.services.async_register(
+ DOMAIN,
+ UPLOAD_SERVICE,
+ async_handle_upload,
+ schema=UPLOAD_SERVICE_SCHEMA,
+ supports_response=SupportsResponse.OPTIONAL,
+ )
diff --git a/homeassistant/components/onedrive/services.yaml b/homeassistant/components/onedrive/services.yaml
new file mode 100644
index 00000000000..0cf0faf6b60
--- /dev/null
+++ b/homeassistant/components/onedrive/services.yaml
@@ -0,0 +1,15 @@
+upload:
+ fields:
+ config_entry_id:
+ required: true
+ selector:
+ config_entry:
+ integration: onedrive
+ filename:
+ required: false
+ selector:
+ object:
+ destination_folder:
+ required: true
+ selector:
+ text:
diff --git a/homeassistant/components/onedrive/strings.json b/homeassistant/components/onedrive/strings.json
index ebc46d3eb12..90fa4efc3ec 100644
--- a/homeassistant/components/onedrive/strings.json
+++ b/homeassistant/components/onedrive/strings.json
@@ -7,6 +7,26 @@
"reauth_confirm": {
"title": "[%key:common::config_flow::title::reauth%]",
"description": "The OneDrive integration needs to re-authenticate your account"
+ },
+ "folder_name": {
+ "title": "Pick a folder name",
+ "description": "This name will be used to create a folder that is specific for this Home Assistant instance. This folder will be created inside `{apps_folder}/{approot}`",
+ "data": {
+ "folder_name": "Folder name"
+ },
+ "data_description": {
+ "folder_name": "Name of the folder"
+ }
+ },
+ "reconfigure_folder": {
+ "title": "Change the folder name",
+ "description": "Rename the instance specific folder inside `{apps_folder}/{approot}`. This will only rename the folder (and does not select another folder), so make sure the new name is not already in use.",
+ "data": {
+ "folder_name": "[%key:component::onedrive::config::step::folder_name::data::folder_name%]"
+ },
+ "data_description": {
+ "folder_name": "[%key:component::onedrive::config::step::folder_name::data_description::folder_name%]"
+ }
}
},
"abort": {
@@ -23,10 +43,39 @@
"connection_error": "Failed to connect to OneDrive.",
"wrong_drive": "New account does not contain previously configured OneDrive.",
"unknown": "[%key:common::config_flow::error::unknown%]",
- "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]"
+ "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",
+ "reconfigure_successful": "[%key:common::config_flow::abort::reconfigure_successful%]"
},
"create_entry": {
"default": "[%key:common::config_flow::create_entry::authenticated%]"
+ },
+ "error": {
+ "folder_rename_error": "Failed to rename folder",
+ "folder_creation_error": "Failed to create folder",
+ "folder_already_in_use": "Folder already used for backups from another Home Assistant instance"
+ }
+ },
+ "options": {
+ "step": {
+ "init": {
+ "description": "By default, files are put into the Recycle Bin when deleted, where they remain available for another 30 days. If you enable this option, files will be deleted immediately when they are cleaned up by the backup system.",
+ "data": {
+ "delete_permanently": "Delete files permanently"
+ },
+ "data_description": {
+ "delete_permanently": "Delete files without moving them to the Recycle Bin"
+ }
+ }
+ }
+ },
+ "issues": {
+ "drive_full": {
+ "title": "OneDrive data cap exceeded",
+ "description": "Your OneDrive has exceeded your quota limit. This means your next backup will fail. Please free up some space or upgrade your OneDrive plan. Currently using {used} GiB of {total} GiB."
+ },
+ "drive_almost_full": {
+ "title": "OneDrive near data cap",
+ "description": "Your OneDrive is near your quota limit. If you go over this limit your drive will be temporarily frozen and your backups will start failing. Please free up some space or upgrade your OneDrive plan. Currently using {used} GiB of {total} GiB."
}
},
"exceptions": {
@@ -38,6 +87,71 @@
},
"failed_to_migrate_files": {
"message": "Failed to migrate metadata to separate files"
+ },
+ "update_failed": {
+ "message": "Failed to update drive state"
+ },
+ "integration_not_found": {
+ "message": "Integration \"{target}\" not found in registry."
+ },
+ "no_access_to_path": {
+ "message": "Cannot read {filename}, no access to path; `allowlist_external_dirs` may need to be adjusted in `configuration.yaml`"
+ },
+ "filename_does_not_exist": {
+ "message": "`{filename}` does not exist"
+ },
+ "file_too_large": {
+ "message": "`{filename}` is too large ({size} > {limit})"
+ },
+ "upload_error": {
+ "message": "Failed to upload content: {message}"
+ },
+ "create_folder_error": {
+ "message": "Failed to create folder: {message}"
+ }
+ },
+ "entity": {
+ "sensor": {
+ "total_size": {
+ "name": "Total available storage"
+ },
+ "used_size": {
+ "name": "Used storage"
+ },
+ "remaining_size": {
+ "name": "Remaining storage"
+ },
+ "drive_state": {
+ "name": "Drive state",
+ "state": {
+ "normal": "Normal",
+ "nearing": "Nearing limit",
+ "critical": "Critical",
+ "exceeded": "Exceeded"
+ }
+ }
+ }
+ },
+ "services": {
+ "upload": {
+ "name": "Upload file",
+ "description": "Uploads files to OneDrive.",
+ "fields": {
+ "config_entry_id": {
+ "name": "Config entry ID",
+ "description": "The config entry representing the OneDrive you want to upload to."
+ },
+ "filename": {
+ "name": "Filename",
+ "description": "Path to the file to upload.",
+ "example": "/config/www/image.jpg"
+ },
+ "destination_folder": {
+ "name": "Destination folder",
+ "description": "Folder inside the Home Assistant app folder (Apps/Home Assistant) you want to upload the file to. Will be created if it does not exist.",
+ "example": "photos/snapshots"
+ }
+ }
}
}
}
diff --git a/homeassistant/components/onewire/binary_sensor.py b/homeassistant/components/onewire/binary_sensor.py
index 60a1d165b15..2bb393e48a8 100644
--- a/homeassistant/components/onewire/binary_sensor.py
+++ b/homeassistant/components/onewire/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DEVICE_KEYS_0_3, DEVICE_KEYS_0_7, DEVICE_KEYS_A_B, READ_MODE_BOOL
from .entity import OneWireEntity, OneWireEntityDescription
@@ -101,7 +101,7 @@ def get_sensor_types(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OneWireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up 1-Wire platform."""
diff --git a/homeassistant/components/onewire/config_flow.py b/homeassistant/components/onewire/config_flow.py
index 8a5623772f7..2099d9aabb5 100644
--- a/homeassistant/components/onewire/config_flow.py
+++ b/homeassistant/components/onewire/config_flow.py
@@ -234,12 +234,7 @@ class OnewireOptionsFlowHandler(OptionsFlow):
INPUT_ENTRY_DEVICE_SELECTION,
default=self._get_current_configured_sensors(),
description="Multiselect with list of devices to choose from",
- ): cv.multi_select(
- {
- friendly_name: False
- for friendly_name in self.configurable_devices
- }
- ),
+ ): cv.multi_select(dict.fromkeys(self.configurable_devices, False)),
}
),
errors=errors,
diff --git a/homeassistant/components/onewire/onewirehub.py b/homeassistant/components/onewire/onewirehub.py
index d65d7a90950..dc894a4242e 100644
--- a/homeassistant/components/onewire/onewirehub.py
+++ b/homeassistant/components/onewire/onewirehub.py
@@ -2,6 +2,7 @@
from __future__ import annotations
+import contextlib
from datetime import datetime, timedelta
import logging
import os
@@ -58,7 +59,7 @@ class OneWireHub:
owproxy: protocol._Proxy
devices: list[OWDeviceDescription]
- _version: str
+ _version: str | None = None
def __init__(self, hass: HomeAssistant, config_entry: OneWireConfigEntry) -> None:
"""Initialize."""
@@ -74,7 +75,9 @@ class OneWireHub:
port = self._config_entry.data[CONF_PORT]
_LOGGER.debug("Initializing connection to %s:%s", host, port)
self.owproxy = protocol.proxy(host, port)
- self._version = self.owproxy.read(protocol.PTH_VERSION).decode()
+ with contextlib.suppress(protocol.OwnetError):
+ # Version is not available on all servers
+ self._version = self.owproxy.read(protocol.PTH_VERSION).decode()
self.devices = _discover_devices(self.owproxy)
async def initialize(self) -> None:
diff --git a/homeassistant/components/onewire/select.py b/homeassistant/components/onewire/select.py
index 7a26ecdbb31..7f4111243aa 100644
--- a/homeassistant/components/onewire/select.py
+++ b/homeassistant/components/onewire/select.py
@@ -10,7 +10,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import READ_MODE_INT
from .entity import OneWireEntity, OneWireEntityDescription
@@ -48,7 +48,7 @@ ENTITY_DESCRIPTIONS: dict[str, tuple[OneWireEntityDescription, ...]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OneWireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up 1-Wire platform."""
diff --git a/homeassistant/components/onewire/sensor.py b/homeassistant/components/onewire/sensor.py
index 04141f87847..5e1c7d35bd6 100644
--- a/homeassistant/components/onewire/sensor.py
+++ b/homeassistant/components/onewire/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -388,7 +388,7 @@ def get_sensor_types(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OneWireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up 1-Wire platform."""
diff --git a/homeassistant/components/onewire/switch.py b/homeassistant/components/onewire/switch.py
index 7215b1ec020..d2cc3b80185 100644
--- a/homeassistant/components/onewire/switch.py
+++ b/homeassistant/components/onewire/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DEVICE_KEYS_0_3, DEVICE_KEYS_0_7, DEVICE_KEYS_A_B, READ_MODE_BOOL
from .entity import OneWireEntity, OneWireEntityDescription
@@ -161,7 +161,7 @@ def get_sensor_types(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OneWireConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up 1-Wire platform."""
diff --git a/homeassistant/components/onkyo/__init__.py b/homeassistant/components/onkyo/__init__.py
index fd5c0ba634a..2ebe86da561 100644
--- a/homeassistant/components/onkyo/__init__.py
+++ b/homeassistant/components/onkyo/__init__.py
@@ -1,6 +1,7 @@
"""The onkyo component."""
from dataclasses import dataclass
+import logging
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
@@ -9,10 +10,18 @@ from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.typing import ConfigType
-from .const import DOMAIN, OPTION_INPUT_SOURCES, InputSource
+from .const import (
+ DOMAIN,
+ OPTION_INPUT_SOURCES,
+ OPTION_LISTENING_MODES,
+ InputSource,
+ ListeningMode,
+)
from .receiver import Receiver, async_interview
from .services import DATA_MP_ENTITIES, async_register_services
+_LOGGER = logging.getLogger(__name__)
+
PLATFORMS = [Platform.MEDIA_PLAYER]
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
@@ -24,6 +33,7 @@ class OnkyoData:
receiver: Receiver
sources: dict[InputSource, str]
+ sound_modes: dict[ListeningMode, str]
type OnkyoConfigEntry = ConfigEntry[OnkyoData]
@@ -50,7 +60,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: OnkyoConfigEntry) -> boo
sources_store: dict[str, str] = entry.options[OPTION_INPUT_SOURCES]
sources = {InputSource(k): v for k, v in sources_store.items()}
- entry.runtime_data = OnkyoData(receiver, sources)
+ sound_modes_store: dict[str, str] = entry.options.get(OPTION_LISTENING_MODES, {})
+ sound_modes = {ListeningMode(k): v for k, v in sound_modes_store.items()}
+
+ entry.runtime_data = OnkyoData(receiver, sources, sound_modes)
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/onkyo/config_flow.py b/homeassistant/components/onkyo/config_flow.py
index 228748d5257..5d941be959a 100644
--- a/homeassistant/components/onkyo/config_flow.py
+++ b/homeassistant/components/onkyo/config_flow.py
@@ -1,5 +1,6 @@
"""Config flow for Onkyo."""
+from collections.abc import Mapping
import logging
from typing import Any
@@ -33,12 +34,14 @@ from .const import (
CONF_SOURCES,
DOMAIN,
OPTION_INPUT_SOURCES,
+ OPTION_LISTENING_MODES,
OPTION_MAX_VOLUME,
OPTION_MAX_VOLUME_DEFAULT,
OPTION_VOLUME_RESOLUTION,
OPTION_VOLUME_RESOLUTION_DEFAULT,
VOLUME_RESOLUTION_ALLOWED,
InputSource,
+ ListeningMode,
)
from .receiver import ReceiverInfo, async_discover, async_interview
@@ -46,9 +49,14 @@ _LOGGER = logging.getLogger(__name__)
CONF_DEVICE = "device"
-INPUT_SOURCES_ALL_MEANINGS = [
- input_source.value_meaning for input_source in InputSource
-]
+INPUT_SOURCES_DEFAULT: dict[str, str] = {}
+LISTENING_MODES_DEFAULT: dict[str, str] = {}
+INPUT_SOURCES_ALL_MEANINGS = {
+ input_source.value_meaning: input_source for input_source in InputSource
+}
+LISTENING_MODES_ALL_MEANINGS = {
+ listening_mode.value_meaning: listening_mode for listening_mode in ListeningMode
+}
STEP_MANUAL_SCHEMA = vol.Schema({vol.Required(CONF_HOST): str})
STEP_RECONFIGURE_SCHEMA = vol.Schema(
{
@@ -59,7 +67,14 @@ STEP_CONFIGURE_SCHEMA = STEP_RECONFIGURE_SCHEMA.extend(
{
vol.Required(OPTION_INPUT_SOURCES): SelectSelector(
SelectSelectorConfig(
- options=INPUT_SOURCES_ALL_MEANINGS,
+ options=list(INPUT_SOURCES_ALL_MEANINGS),
+ multiple=True,
+ mode=SelectSelectorMode.DROPDOWN,
+ )
+ ),
+ vol.Required(OPTION_LISTENING_MODES): SelectSelector(
+ SelectSelectorConfig(
+ options=list(LISTENING_MODES_ALL_MEANINGS),
multiple=True,
mode=SelectSelectorMode.DROPDOWN,
)
@@ -238,9 +253,8 @@ class OnkyoConfigFlow(ConfigFlow, domain=DOMAIN):
CONF_HOST: self._receiver_info.host,
},
options={
+ **entry_options,
OPTION_VOLUME_RESOLUTION: volume_resolution,
- OPTION_MAX_VOLUME: entry_options[OPTION_MAX_VOLUME],
- OPTION_INPUT_SOURCES: entry_options[OPTION_INPUT_SOURCES],
},
)
@@ -250,12 +264,24 @@ class OnkyoConfigFlow(ConfigFlow, domain=DOMAIN):
input_source_meanings: list[str] = user_input[OPTION_INPUT_SOURCES]
if not input_source_meanings:
errors[OPTION_INPUT_SOURCES] = "empty_input_source_list"
- else:
+
+ listening_modes: list[str] = user_input[OPTION_LISTENING_MODES]
+ if not listening_modes:
+ errors[OPTION_LISTENING_MODES] = "empty_listening_mode_list"
+
+ if not errors:
input_sources_store: dict[str, str] = {}
for input_source_meaning in input_source_meanings:
- input_source = InputSource.from_meaning(input_source_meaning)
+ input_source = INPUT_SOURCES_ALL_MEANINGS[input_source_meaning]
input_sources_store[input_source.value] = input_source_meaning
+ listening_modes_store: dict[str, str] = {}
+ for listening_mode_meaning in listening_modes:
+ listening_mode = LISTENING_MODES_ALL_MEANINGS[
+ listening_mode_meaning
+ ]
+ listening_modes_store[listening_mode.value] = listening_mode_meaning
+
result = self.async_create_entry(
title=self._receiver_info.model_name,
data={
@@ -265,6 +291,7 @@ class OnkyoConfigFlow(ConfigFlow, domain=DOMAIN):
OPTION_VOLUME_RESOLUTION: volume_resolution,
OPTION_MAX_VOLUME: OPTION_MAX_VOLUME_DEFAULT,
OPTION_INPUT_SOURCES: input_sources_store,
+ OPTION_LISTENING_MODES: listening_modes_store,
},
)
@@ -278,16 +305,13 @@ class OnkyoConfigFlow(ConfigFlow, domain=DOMAIN):
if reconfigure_entry is None:
suggested_values = {
OPTION_VOLUME_RESOLUTION: OPTION_VOLUME_RESOLUTION_DEFAULT,
- OPTION_INPUT_SOURCES: [],
+ OPTION_INPUT_SOURCES: INPUT_SOURCES_DEFAULT,
+ OPTION_LISTENING_MODES: LISTENING_MODES_DEFAULT,
}
else:
entry_options = reconfigure_entry.options
suggested_values = {
OPTION_VOLUME_RESOLUTION: entry_options[OPTION_VOLUME_RESOLUTION],
- OPTION_INPUT_SOURCES: [
- InputSource(input_source).value_meaning
- for input_source in entry_options[OPTION_INPUT_SOURCES]
- ],
}
return self.async_show_form(
@@ -356,6 +380,7 @@ class OnkyoConfigFlow(ConfigFlow, domain=DOMAIN):
OPTION_VOLUME_RESOLUTION: volume_resolution,
OPTION_MAX_VOLUME: max_volume,
OPTION_INPUT_SOURCES: sources_store,
+ OPTION_LISTENING_MODES: LISTENING_MODES_DEFAULT,
},
)
@@ -373,7 +398,14 @@ OPTIONS_STEP_INIT_SCHEMA = vol.Schema(
),
vol.Required(OPTION_INPUT_SOURCES): SelectSelector(
SelectSelectorConfig(
- options=INPUT_SOURCES_ALL_MEANINGS,
+ options=list(INPUT_SOURCES_ALL_MEANINGS),
+ multiple=True,
+ mode=SelectSelectorMode.DROPDOWN,
+ )
+ ),
+ vol.Required(OPTION_LISTENING_MODES): SelectSelector(
+ SelectSelectorConfig(
+ options=list(LISTENING_MODES_ALL_MEANINGS),
multiple=True,
mode=SelectSelectorMode.DROPDOWN,
)
@@ -387,6 +419,7 @@ class OnkyoOptionsFlowHandler(OptionsFlow):
_data: dict[str, Any]
_input_sources: dict[InputSource, str]
+ _listening_modes: dict[ListeningMode, str]
async def async_step_init(
self, user_input: dict[str, Any] | None = None
@@ -394,20 +427,40 @@ class OnkyoOptionsFlowHandler(OptionsFlow):
"""Manage the options."""
errors = {}
- entry_options = self.config_entry.options
+ entry_options: Mapping[str, Any] = self.config_entry.options
+ entry_options = {
+ OPTION_LISTENING_MODES: LISTENING_MODES_DEFAULT,
+ **entry_options,
+ }
if user_input is not None:
- self._input_sources = {}
- for input_source_meaning in user_input[OPTION_INPUT_SOURCES]:
- input_source = InputSource.from_meaning(input_source_meaning)
- input_source_name = entry_options[OPTION_INPUT_SOURCES].get(
- input_source.value, input_source_meaning
- )
- self._input_sources[input_source] = input_source_name
-
- if not self._input_sources:
+ input_source_meanings: list[str] = user_input[OPTION_INPUT_SOURCES]
+ if not input_source_meanings:
errors[OPTION_INPUT_SOURCES] = "empty_input_source_list"
- else:
+
+ listening_mode_meanings: list[str] = user_input[OPTION_LISTENING_MODES]
+ if not listening_mode_meanings:
+ errors[OPTION_LISTENING_MODES] = "empty_listening_mode_list"
+
+ if not errors:
+ self._input_sources = {}
+ for input_source_meaning in input_source_meanings:
+ input_source = INPUT_SOURCES_ALL_MEANINGS[input_source_meaning]
+ input_source_name = entry_options[OPTION_INPUT_SOURCES].get(
+ input_source.value, input_source_meaning
+ )
+ self._input_sources[input_source] = input_source_name
+
+ self._listening_modes = {}
+ for listening_mode_meaning in listening_mode_meanings:
+ listening_mode = LISTENING_MODES_ALL_MEANINGS[
+ listening_mode_meaning
+ ]
+ listening_mode_name = entry_options[OPTION_LISTENING_MODES].get(
+ listening_mode.value, listening_mode_meaning
+ )
+ self._listening_modes[listening_mode] = listening_mode_name
+
self._data = {
OPTION_VOLUME_RESOLUTION: entry_options[OPTION_VOLUME_RESOLUTION],
OPTION_MAX_VOLUME: user_input[OPTION_MAX_VOLUME],
@@ -423,6 +476,10 @@ class OnkyoOptionsFlowHandler(OptionsFlow):
InputSource(input_source).value_meaning
for input_source in entry_options[OPTION_INPUT_SOURCES]
],
+ OPTION_LISTENING_MODES: [
+ ListeningMode(listening_mode).value_meaning
+ for listening_mode in entry_options[OPTION_LISTENING_MODES]
+ ],
}
return self.async_show_form(
@@ -440,28 +497,48 @@ class OnkyoOptionsFlowHandler(OptionsFlow):
if user_input is not None:
input_sources_store: dict[str, str] = {}
for input_source_meaning, input_source_name in user_input[
- "input_sources"
+ OPTION_INPUT_SOURCES
].items():
- input_source = InputSource.from_meaning(input_source_meaning)
+ input_source = INPUT_SOURCES_ALL_MEANINGS[input_source_meaning]
input_sources_store[input_source.value] = input_source_name
+ listening_modes_store: dict[str, str] = {}
+ for listening_mode_meaning, listening_mode_name in user_input[
+ OPTION_LISTENING_MODES
+ ].items():
+ listening_mode = LISTENING_MODES_ALL_MEANINGS[listening_mode_meaning]
+ listening_modes_store[listening_mode.value] = listening_mode_name
+
return self.async_create_entry(
data={
**self._data,
OPTION_INPUT_SOURCES: input_sources_store,
+ OPTION_LISTENING_MODES: listening_modes_store,
}
)
- schema_dict: dict[Any, Selector] = {}
-
+ input_sources_schema_dict: dict[Any, Selector] = {}
for input_source, input_source_name in self._input_sources.items():
- schema_dict[
+ input_sources_schema_dict[
vol.Required(input_source.value_meaning, default=input_source_name)
] = TextSelector()
+ listening_modes_schema_dict: dict[Any, Selector] = {}
+ for listening_mode, listening_mode_name in self._listening_modes.items():
+ listening_modes_schema_dict[
+ vol.Required(listening_mode.value_meaning, default=listening_mode_name)
+ ] = TextSelector()
+
return self.async_show_form(
step_id="names",
data_schema=vol.Schema(
- {vol.Required("input_sources"): section(vol.Schema(schema_dict))}
+ {
+ vol.Required(OPTION_INPUT_SOURCES): section(
+ vol.Schema(input_sources_schema_dict)
+ ),
+ vol.Required(OPTION_LISTENING_MODES): section(
+ vol.Schema(listening_modes_schema_dict)
+ ),
+ }
),
)
diff --git a/homeassistant/components/onkyo/const.py b/homeassistant/components/onkyo/const.py
index bd4fe98ae7d..fcb1a8a0a9e 100644
--- a/homeassistant/components/onkyo/const.py
+++ b/homeassistant/components/onkyo/const.py
@@ -2,7 +2,7 @@
from enum import Enum
import typing
-from typing import ClassVar, Literal, Self
+from typing import Literal, Self
import pyeiscp
@@ -24,7 +24,27 @@ VOLUME_RESOLUTION_ALLOWED: tuple[VolumeResolution, ...] = typing.get_args(
OPTION_MAX_VOLUME = "max_volume"
OPTION_MAX_VOLUME_DEFAULT = 100.0
+
+class EnumWithMeaning(Enum):
+ """Enum with meaning."""
+
+ value_meaning: str
+
+ def __new__(cls, value: str) -> Self:
+ """Create enum."""
+ obj = object.__new__(cls)
+ obj._value_ = value
+ obj.value_meaning = cls._get_meanings()[value]
+
+ return obj
+
+ @staticmethod
+ def _get_meanings() -> dict[str, str]:
+ raise NotImplementedError
+
+
OPTION_INPUT_SOURCES = "input_sources"
+OPTION_LISTENING_MODES = "listening_modes"
_INPUT_SOURCE_MEANINGS = {
"00": "VIDEO1 ··· VCR/DVR ··· STB/DVR",
@@ -71,7 +91,7 @@ _INPUT_SOURCE_MEANINGS = {
}
-class InputSource(Enum):
+class InputSource(EnumWithMeaning):
"""Receiver input source."""
DVR = "00"
@@ -116,24 +136,100 @@ class InputSource(Enum):
HDMI_7 = "57"
MAIN_SOURCE = "80"
- __meaning_mapping: ClassVar[dict[str, Self]] = {} # type: ignore[misc]
+ @staticmethod
+ def _get_meanings() -> dict[str, str]:
+ return _INPUT_SOURCE_MEANINGS
- value_meaning: str
- def __new__(cls, value: str) -> Self:
- """Create InputSource enum."""
- obj = object.__new__(cls)
- obj._value_ = value
- obj.value_meaning = _INPUT_SOURCE_MEANINGS[value]
+_LISTENING_MODE_MEANINGS = {
+ "00": "STEREO",
+ "01": "DIRECT",
+ "02": "SURROUND",
+ "03": "FILM ··· GAME RPG ··· ADVANCED GAME",
+ "04": "THX",
+ "05": "ACTION ··· GAME ACTION",
+ "06": "MUSICAL ··· GAME ROCK ··· ROCK/POP",
+ "07": "MONO MOVIE",
+ "08": "ORCHESTRA ··· CLASSICAL",
+ "09": "UNPLUGGED",
+ "0A": "STUDIO MIX ··· ENTERTAINMENT SHOW",
+ "0B": "TV LOGIC ··· DRAMA",
+ "0C": "ALL CH STEREO ··· EXTENDED STEREO",
+ "0D": "THEATER DIMENSIONAL ··· FRONT STAGE SURROUND",
+ "0E": "ENHANCED 7/ENHANCE ··· GAME SPORTS ··· SPORTS",
+ "0F": "MONO",
+ "11": "PURE AUDIO ··· PURE DIRECT",
+ "12": "MULTIPLEX",
+ "13": "FULL MONO ··· MONO MUSIC",
+ "14": "DOLBY VIRTUAL/SURROUND ENHANCER",
+ "15": "DTS SURROUND SENSATION",
+ "16": "AUDYSSEY DSX",
+ "17": "DTS VIRTUAL:X",
+ "1F": "WHOLE HOUSE MODE ··· MULTI ZONE MUSIC",
+ "23": "STAGE (JAPAN GENRE CONTROL)",
+ "25": "ACTION (JAPAN GENRE CONTROL)",
+ "26": "MUSIC (JAPAN GENRE CONTROL)",
+ "2E": "SPORTS (JAPAN GENRE CONTROL)",
+ "40": "STRAIGHT DECODE ··· 5.1 CH SURROUND",
+ "41": "DOLBY EX/DTS ES",
+ "42": "THX CINEMA",
+ "43": "THX SURROUND EX",
+ "44": "THX MUSIC",
+ "45": "THX GAMES",
+ "50": "THX U(2)/S(2)/I/S CINEMA",
+ "51": "THX U(2)/S(2)/I/S MUSIC",
+ "52": "THX U(2)/S(2)/I/S GAMES",
+ "80": "DOLBY ATMOS/DOLBY SURROUND ··· PLII/PLIIx MOVIE",
+ "81": "PLII/PLIIx MUSIC",
+ "82": "DTS:X/NEURAL:X ··· NEO:6/NEO:X CINEMA",
+ "83": "NEO:6/NEO:X MUSIC",
+ "84": "DOLBY SURROUND THX CINEMA ··· PLII/PLIIx THX CINEMA",
+ "85": "DTS NEURAL:X THX CINEMA ··· NEO:6/NEO:X THX CINEMA",
+ "86": "PLII/PLIIx GAME",
+ "87": "NEURAL SURR",
+ "88": "NEURAL THX/NEURAL SURROUND",
+ "89": "DOLBY SURROUND THX GAMES ··· PLII/PLIIx THX GAMES",
+ "8A": "DTS NEURAL:X THX GAMES ··· NEO:6/NEO:X THX GAMES",
+ "8B": "DOLBY SURROUND THX MUSIC ··· PLII/PLIIx THX MUSIC",
+ "8C": "DTS NEURAL:X THX MUSIC ··· NEO:6/NEO:X THX MUSIC",
+ "8D": "NEURAL THX CINEMA",
+ "8E": "NEURAL THX MUSIC",
+ "8F": "NEURAL THX GAMES",
+ "90": "PLIIz HEIGHT",
+ "91": "NEO:6 CINEMA DTS SURROUND SENSATION",
+ "92": "NEO:6 MUSIC DTS SURROUND SENSATION",
+ "93": "NEURAL DIGITAL MUSIC",
+ "94": "PLIIz HEIGHT + THX CINEMA",
+ "95": "PLIIz HEIGHT + THX MUSIC",
+ "96": "PLIIz HEIGHT + THX GAMES",
+ "97": "PLIIz HEIGHT + THX U2/S2 CINEMA",
+ "98": "PLIIz HEIGHT + THX U2/S2 MUSIC",
+ "99": "PLIIz HEIGHT + THX U2/S2 GAMES",
+ "9A": "NEO:X GAME",
+ "A0": "PLIIx/PLII Movie + AUDYSSEY DSX",
+ "A1": "PLIIx/PLII MUSIC + AUDYSSEY DSX",
+ "A2": "PLIIx/PLII GAME + AUDYSSEY DSX",
+ "A3": "NEO:6 CINEMA + AUDYSSEY DSX",
+ "A4": "NEO:6 MUSIC + AUDYSSEY DSX",
+ "A5": "NEURAL SURROUND + AUDYSSEY DSX",
+ "A6": "NEURAL DIGITAL MUSIC + AUDYSSEY DSX",
+ "A7": "DOLBY EX + AUDYSSEY DSX",
+ "FF": "AUTO SURROUND",
+}
- cls.__meaning_mapping[obj.value_meaning] = obj
- return obj
+class ListeningMode(EnumWithMeaning):
+ """Receiver listening mode."""
- @classmethod
- def from_meaning(cls, meaning: str) -> Self:
- """Get InputSource enum from its meaning."""
- return cls.__meaning_mapping[meaning]
+ _ignore_ = "ListeningMode _k _v _meaning"
+
+ ListeningMode = vars()
+ for _k in _LISTENING_MODE_MEANINGS:
+ ListeningMode["I" + _k] = _k
+
+ @staticmethod
+ def _get_meanings() -> dict[str, str]:
+ return _LISTENING_MODE_MEANINGS
ZONES = {"main": "Main", "zone2": "Zone 2", "zone3": "Zone 3", "zone4": "Zone 4"}
diff --git a/homeassistant/components/onkyo/media_player.py b/homeassistant/components/onkyo/media_player.py
index 97a82fc8a1a..f7fe83c57a3 100644
--- a/homeassistant/components/onkyo/media_player.py
+++ b/homeassistant/components/onkyo/media_player.py
@@ -3,6 +3,7 @@
from __future__ import annotations
import asyncio
+from enum import Enum
from functools import cache
import logging
from typing import Any, Literal
@@ -22,7 +23,10 @@ from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant, ca
from homeassistant.data_entry_flow import FlowResultType
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -36,6 +40,7 @@ from .const import (
PYEISCP_COMMANDS,
ZONES,
InputSource,
+ ListeningMode,
VolumeResolution,
)
from .receiver import Receiver, async_discover
@@ -60,6 +65,8 @@ CONF_SOURCES_DEFAULT = {
"fm": "Radio",
}
+ISSUE_URL_PLACEHOLDER = "/config/integrations/dashboard/add?domain=onkyo"
+
PLATFORM_SCHEMA = MEDIA_PLAYER_PLATFORM_SCHEMA.extend(
{
vol.Optional(CONF_HOST): cv.string,
@@ -76,23 +83,23 @@ PLATFORM_SCHEMA = MEDIA_PLAYER_PLATFORM_SCHEMA.extend(
}
)
-SUPPORT_ONKYO_WO_VOLUME = (
+
+SUPPORTED_FEATURES_BASE = (
MediaPlayerEntityFeature.TURN_ON
| MediaPlayerEntityFeature.TURN_OFF
| MediaPlayerEntityFeature.SELECT_SOURCE
| MediaPlayerEntityFeature.PLAY_MEDIA
)
-SUPPORT_ONKYO = (
- SUPPORT_ONKYO_WO_VOLUME
- | MediaPlayerEntityFeature.VOLUME_SET
+SUPPORTED_FEATURES_VOLUME = (
+ MediaPlayerEntityFeature.VOLUME_SET
| MediaPlayerEntityFeature.VOLUME_MUTE
| MediaPlayerEntityFeature.VOLUME_STEP
)
-DEFAULT_PLAYABLE_SOURCES = (
- InputSource.from_meaning("FM"),
- InputSource.from_meaning("AM"),
- InputSource.from_meaning("TUNER"),
+PLAYABLE_SOURCES = (
+ InputSource.FM,
+ InputSource.AM,
+ InputSource.DAB,
)
ATTR_PRESET = "preset"
@@ -115,7 +122,6 @@ AUDIO_INFORMATION_MAPPING = [
"auto_phase_control_phase",
"upmix_mode",
]
-
VIDEO_INFORMATION_MAPPING = [
"video_input_port",
"input_resolution",
@@ -128,7 +134,6 @@ VIDEO_INFORMATION_MAPPING = [
"picture_mode",
"input_hdr",
]
-ISSUE_URL_PLACEHOLDER = "/config/integrations/dashboard/add?domain=onkyo"
type LibValue = str | tuple[str, ...]
@@ -136,7 +141,19 @@ type LibValue = str | tuple[str, ...]
def _get_single_lib_value(value: LibValue) -> str:
if isinstance(value, str):
return value
- return value[0]
+ return value[-1]
+
+
+def _get_lib_mapping[T: Enum](cmds: Any, cls: type[T]) -> dict[T, LibValue]:
+ result: dict[T, LibValue] = {}
+ for k, v in cmds["values"].items():
+ try:
+ key = cls(k)
+ except ValueError:
+ continue
+ result[key] = v["name"]
+
+ return result
@cache
@@ -151,15 +168,7 @@ def _input_source_lib_mappings(zone: str) -> dict[InputSource, LibValue]:
case "zone4":
cmds = PYEISCP_COMMANDS["zone4"]["SL4"]
- result: dict[InputSource, LibValue] = {}
- for k, v in cmds["values"].items():
- try:
- source = InputSource(k)
- except ValueError:
- continue
- result[source] = v["name"]
-
- return result
+ return _get_lib_mapping(cmds, InputSource)
@cache
@@ -167,6 +176,24 @@ def _rev_input_source_lib_mappings(zone: str) -> dict[LibValue, InputSource]:
return {value: key for key, value in _input_source_lib_mappings(zone).items()}
+@cache
+def _listening_mode_lib_mappings(zone: str) -> dict[ListeningMode, LibValue]:
+ match zone:
+ case "main":
+ cmds = PYEISCP_COMMANDS["main"]["LMD"]
+ case "zone2":
+ cmds = PYEISCP_COMMANDS["zone2"]["LMZ"]
+ case _:
+ return {}
+
+ return _get_lib_mapping(cmds, ListeningMode)
+
+
+@cache
+def _rev_listening_mode_lib_mappings(zone: str) -> dict[LibValue, ListeningMode]:
+ return {value: key for key, value in _listening_mode_lib_mappings(zone).items()}
+
+
async def async_setup_platform(
hass: HomeAssistant,
config: ConfigType,
@@ -286,7 +313,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: OnkyoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MediaPlayer for config entry."""
data = entry.runtime_data
@@ -300,6 +327,7 @@ async def async_setup_entry(
volume_resolution: VolumeResolution = entry.options[OPTION_VOLUME_RESOLUTION]
max_volume: float = entry.options[OPTION_MAX_VOLUME]
sources = data.sources
+ sound_modes = data.sound_modes
def connect_callback(receiver: Receiver) -> None:
if not receiver.first_connect:
@@ -328,6 +356,7 @@ async def async_setup_entry(
volume_resolution=volume_resolution,
max_volume=max_volume,
sources=sources,
+ sound_modes=sound_modes,
)
entities[zone] = zone_entity
async_add_entities([zone_entity])
@@ -342,6 +371,7 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
_attr_should_poll = False
_supports_volume: bool = False
+ _supports_sound_mode: bool = False
_supports_audio_info: bool = False
_supports_video_info: bool = False
_query_timer: asyncio.TimerHandle | None = None
@@ -354,6 +384,7 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
volume_resolution: VolumeResolution,
max_volume: float,
sources: dict[InputSource, str],
+ sound_modes: dict[ListeningMode, str],
) -> None:
"""Initialize the Onkyo Receiver."""
self._receiver = receiver
@@ -367,6 +398,7 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
self._volume_resolution = volume_resolution
self._max_volume = max_volume
+ self._options_sources = sources
self._source_lib_mapping = _input_source_lib_mappings(zone)
self._rev_source_lib_mapping = _rev_input_source_lib_mappings(zone)
self._source_mapping = {
@@ -378,7 +410,28 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
value: key for key, value in self._source_mapping.items()
}
+ self._options_sound_modes = sound_modes
+ self._sound_mode_lib_mapping = _listening_mode_lib_mappings(zone)
+ self._rev_sound_mode_lib_mapping = _rev_listening_mode_lib_mappings(zone)
+ self._sound_mode_mapping = {
+ key: value
+ for key, value in sound_modes.items()
+ if key in self._sound_mode_lib_mapping
+ }
+ self._rev_sound_mode_mapping = {
+ value: key for key, value in self._sound_mode_mapping.items()
+ }
+
self._attr_source_list = list(self._rev_source_mapping)
+ self._attr_sound_mode_list = list(self._rev_sound_mode_mapping)
+
+ self._attr_supported_features = SUPPORTED_FEATURES_BASE
+ if zone == "main":
+ self._attr_supported_features |= SUPPORTED_FEATURES_VOLUME
+ self._supports_volume = True
+ self._attr_supported_features |= MediaPlayerEntityFeature.SELECT_SOUND_MODE
+ self._supports_sound_mode = True
+
self._attr_extra_state_attributes = {}
async def async_added_to_hass(self) -> None:
@@ -391,13 +444,6 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
self._query_timer.cancel()
self._query_timer = None
- @property
- def supported_features(self) -> MediaPlayerEntityFeature:
- """Return media player features that are supported."""
- if self._supports_volume:
- return SUPPORT_ONKYO
- return SUPPORT_ONKYO_WO_VOLUME
-
@callback
def _update_receiver(self, propname: str, value: Any) -> None:
"""Update a property in the receiver."""
@@ -463,6 +509,24 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
"input-selector" if self._zone == "main" else "selector", source_lib_single
)
+ async def async_select_sound_mode(self, sound_mode: str) -> None:
+ """Select listening sound mode."""
+ if not self.sound_mode_list or sound_mode not in self.sound_mode_list:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="invalid_sound_mode",
+ translation_placeholders={
+ "invalid_sound_mode": sound_mode,
+ "entity_id": self.entity_id,
+ },
+ )
+
+ sound_mode_lib = self._sound_mode_lib_mapping[
+ self._rev_sound_mode_mapping[sound_mode]
+ ]
+ sound_mode_lib_single = _get_single_lib_value(sound_mode_lib)
+ self._update_receiver("listening-mode", sound_mode_lib_single)
+
async def async_select_output(self, hdmi_output: str) -> None:
"""Set hdmi-out."""
self._update_receiver("hdmi-output-selector", hdmi_output)
@@ -473,7 +537,7 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
"""Play radio station by preset number."""
if self.source is not None:
source = self._rev_source_mapping[self.source]
- if media_type.lower() == "radio" and source in DEFAULT_PLAYABLE_SOURCES:
+ if media_type.lower() == "radio" and source in PLAYABLE_SOURCES:
self._update_receiver("preset", media_id)
@callback
@@ -514,7 +578,9 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
self._attr_extra_state_attributes.pop(ATTR_PRESET, None)
self._attr_extra_state_attributes.pop(ATTR_VIDEO_OUT, None)
elif command in ["volume", "master-volume"] and value != "N/A":
- self._supports_volume = True
+ if not self._supports_volume:
+ self._attr_supported_features |= SUPPORTED_FEATURES_VOLUME
+ self._supports_volume = True
# AMP_VOL / (VOL_RESOLUTION * (MAX_VOL / 100))
volume_level: float = value / (
self._volume_resolution * self._max_volume / 100
@@ -522,7 +588,7 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
self._attr_volume_level = min(1, volume_level)
elif command in ["muting", "audio-muting"]:
self._attr_is_volume_muted = bool(value == "on")
- elif command in ["selector", "input-selector"]:
+ elif command in ["selector", "input-selector"] and value != "N/A":
self._parse_source(value)
self._query_av_info_delayed()
elif command == "hdmi-output-selector":
@@ -532,6 +598,14 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
self._attr_extra_state_attributes[ATTR_PRESET] = value
elif ATTR_PRESET in self._attr_extra_state_attributes:
del self._attr_extra_state_attributes[ATTR_PRESET]
+ elif command == "listening-mode" and value != "N/A":
+ if not self._supports_sound_mode:
+ self._attr_supported_features |= (
+ MediaPlayerEntityFeature.SELECT_SOUND_MODE
+ )
+ self._supports_sound_mode = True
+ self._parse_sound_mode(value)
+ self._query_av_info_delayed()
elif command == "audio-information":
self._supports_audio_info = True
self._parse_audio_information(value)
@@ -551,13 +625,46 @@ class OnkyoMediaPlayer(MediaPlayerEntity):
return
source_meaning = source.value_meaning
- _LOGGER.error(
- 'Input source "%s" is invalid for entity: %s',
- source_meaning,
- self.entity_id,
- )
+
+ if source not in self._options_sources:
+ _LOGGER.warning(
+ 'Input source "%s" for entity: %s is not in the list. Check integration options',
+ source_meaning,
+ self.entity_id,
+ )
+ else:
+ _LOGGER.error(
+ 'Input source "%s" is invalid for entity: %s',
+ source_meaning,
+ self.entity_id,
+ )
+
self._attr_source = source_meaning
+ @callback
+ def _parse_sound_mode(self, mode_lib: LibValue) -> None:
+ sound_mode = self._rev_sound_mode_lib_mapping[mode_lib]
+ if sound_mode in self._sound_mode_mapping:
+ self._attr_sound_mode = self._sound_mode_mapping[sound_mode]
+ return
+
+ sound_mode_meaning = sound_mode.value_meaning
+
+ if sound_mode not in self._options_sound_modes:
+ _LOGGER.warning(
+ 'Listening mode "%s" for entity: %s is not in the list. Check integration options',
+ sound_mode_meaning,
+ self.entity_id,
+ )
+ else:
+ _LOGGER.error(
+ 'Listening mode "%s" is invalid for entity: %s',
+ sound_mode_meaning,
+ self.entity_id,
+ )
+
+ self._attr_sound_mode = sound_mode_meaning
+
@callback
def _parse_audio_information(
self, audio_information: tuple[str] | Literal["N/A"]
diff --git a/homeassistant/components/onkyo/strings.json b/homeassistant/components/onkyo/strings.json
index b3b14efec44..d8131dd1149 100644
--- a/homeassistant/components/onkyo/strings.json
+++ b/homeassistant/components/onkyo/strings.json
@@ -27,17 +27,20 @@
"description": "Configure {name}",
"data": {
"volume_resolution": "Volume resolution",
- "input_sources": "[%key:component::onkyo::options::step::init::data::input_sources%]"
+ "input_sources": "[%key:component::onkyo::options::step::init::data::input_sources%]",
+ "listening_modes": "[%key:component::onkyo::options::step::init::data::listening_modes%]"
},
"data_description": {
"volume_resolution": "Number of steps it takes for the receiver to go from the lowest to the highest possible volume.",
- "input_sources": "[%key:component::onkyo::options::step::init::data_description::input_sources%]"
+ "input_sources": "[%key:component::onkyo::options::step::init::data_description::input_sources%]",
+ "listening_modes": "[%key:component::onkyo::options::step::init::data_description::listening_modes%]"
}
}
},
"error": {
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
"empty_input_source_list": "[%key:component::onkyo::options::error::empty_input_source_list%]",
+ "empty_listening_mode_list": "[%key:component::onkyo::options::error::empty_listening_mode_list%]",
"unknown": "[%key:common::config_flow::error::unknown%]"
},
"abort": {
@@ -53,11 +56,13 @@
"init": {
"data": {
"max_volume": "Maximum volume limit (%)",
- "input_sources": "Input sources"
+ "input_sources": "Input sources",
+ "listening_modes": "Listening modes"
},
"data_description": {
"max_volume": "Maximum volume limit as a percentage. This will associate Home Assistant's maximum volume to this value on the receiver, i.e., if you set this to 50%, then setting the volume to 100% in Home Assistant will cause the volume on the receiver to be set to 50% of its maximum value.",
- "input_sources": "List of input sources supported by the receiver."
+ "input_sources": "List of input sources supported by the receiver.",
+ "listening_modes": "List of listening modes supported by the receiver."
}
},
"names": {
@@ -65,12 +70,17 @@
"input_sources": {
"name": "Input source names",
"description": "Mappings of receiver's input sources to their names."
+ },
+ "listening_modes": {
+ "name": "Listening mode names",
+ "description": "Mappings of receiver's listening modes to their names."
}
}
}
},
"error": {
- "empty_input_source_list": "Input source list cannot be empty"
+ "empty_input_source_list": "Input source list cannot be empty",
+ "empty_listening_mode_list": "Listening mode list cannot be empty"
}
},
"issues": {
@@ -84,6 +94,9 @@
}
},
"exceptions": {
+ "invalid_sound_mode": {
+ "message": "Cannot select sound mode \"{invalid_sound_mode}\" for entity: {entity_id}."
+ },
"invalid_source": {
"message": "Cannot select input source \"{invalid_source}\" for entity: {entity_id}."
}
diff --git a/homeassistant/components/onvif/__init__.py b/homeassistant/components/onvif/__init__.py
index 02e7e28ea18..09a4aba52bf 100644
--- a/homeassistant/components/onvif/__init__.py
+++ b/homeassistant/components/onvif/__init__.py
@@ -19,8 +19,9 @@ from homeassistant.const import (
HTTP_DIGEST_AUTHENTICATION,
Platform,
)
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
+from homeassistant.helpers import entity_registry as er
from .const import (
CONF_ENABLE_WEBHOOKS,
@@ -99,6 +100,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
if device.capabilities.imaging:
device.platforms += [Platform.SWITCH]
+ _async_migrate_camera_entities_unique_ids(hass, entry, device)
+
await hass.config_entries.async_forward_entry_setups(entry, device.platforms)
entry.async_on_unload(
@@ -155,3 +158,58 @@ async def async_populate_options(hass: HomeAssistant, entry: ConfigEntry) -> Non
}
hass.config_entries.async_update_entry(entry, options=options)
+
+
+@callback
+def _async_migrate_camera_entities_unique_ids(
+ hass: HomeAssistant, config_entry: ConfigEntry, device: ONVIFDevice
+) -> None:
+ """Migrate unique ids of camera entities from profile index to profile token."""
+ entity_reg = er.async_get(hass)
+ entities: list[er.RegistryEntry] = er.async_entries_for_config_entry(
+ entity_reg, config_entry.entry_id
+ )
+
+ mac_or_serial = device.info.mac or device.info.serial_number
+ old_uid_start = f"{mac_or_serial}_"
+ new_uid_start = f"{mac_or_serial}#"
+
+ for entity in entities:
+ if entity.domain != Platform.CAMERA:
+ continue
+
+ if (
+ not entity.unique_id.startswith(old_uid_start)
+ and entity.unique_id != mac_or_serial
+ ):
+ continue
+
+ index = 0
+ if entity.unique_id.startswith(old_uid_start):
+ try:
+ index = int(entity.unique_id[len(old_uid_start) :])
+ except ValueError:
+ LOGGER.error(
+ "Failed to migrate unique id for '%s' as the ONVIF profile index could not be parsed from unique id '%s'",
+ entity.entity_id,
+ entity.unique_id,
+ )
+ continue
+ try:
+ token = device.profiles[index].token
+ except IndexError:
+ LOGGER.error(
+ "Failed to migrate unique id for '%s' as the ONVIF profile index '%d' parsed from unique id '%s' could not be found",
+ entity.entity_id,
+ index,
+ entity.unique_id,
+ )
+ continue
+ new_uid = f"{new_uid_start}{token}"
+ LOGGER.debug(
+ "Migrating unique id for '%s' from '%s' to '%s'",
+ entity.entity_id,
+ entity.unique_id,
+ new_uid,
+ )
+ entity_reg.async_update_entity(entity.entity_id, new_unique_id=new_uid)
diff --git a/homeassistant/components/onvif/binary_sensor.py b/homeassistant/components/onvif/binary_sensor.py
index 92c5ab45129..d29f732ef67 100644
--- a/homeassistant/components/onvif/binary_sensor.py
+++ b/homeassistant/components/onvif/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_ON
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.util.enum import try_parse_enum
@@ -22,7 +22,7 @@ from .entity import ONVIFBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a ONVIF binary sensor."""
device: ONVIFDevice = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/onvif/button.py b/homeassistant/components/onvif/button.py
index 644a7c942f7..8e92cb07a8c 100644
--- a/homeassistant/components/onvif/button.py
+++ b/homeassistant/components/onvif/button.py
@@ -4,7 +4,7 @@ from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .device import ONVIFDevice
@@ -14,7 +14,7 @@ from .entity import ONVIFBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ONVIF button based on a config entry."""
device = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/onvif/camera.py b/homeassistant/components/onvif/camera.py
index 8c0fd027b95..fc17e912fcc 100644
--- a/homeassistant/components/onvif/camera.py
+++ b/homeassistant/components/onvif/camera.py
@@ -22,7 +22,7 @@ from homeassistant.const import HTTP_BASIC_AUTHENTICATION
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ABSOLUTE_MOVE,
@@ -57,7 +57,7 @@ from .models import Profile
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ONVIF camera video stream."""
platform = entity_platform.async_get_current_platform()
@@ -117,10 +117,7 @@ class ONVIFCameraEntity(ONVIFBaseEntity, Camera):
self._attr_entity_registry_enabled_default = (
device.max_resolution == profile.video.resolution.width
)
- if profile.index:
- self._attr_unique_id = f"{self.mac_or_serial}_{profile.index}"
- else:
- self._attr_unique_id = self.mac_or_serial
+ self._attr_unique_id = f"{self.mac_or_serial}#{profile.token}"
self._attr_name = f"{device.name} {profile.name}"
@property
diff --git a/homeassistant/components/onvif/device.py b/homeassistant/components/onvif/device.py
index 6d1a340fc7b..3f37ba42397 100644
--- a/homeassistant/components/onvif/device.py
+++ b/homeassistant/components/onvif/device.py
@@ -235,7 +235,7 @@ class ONVIFDevice:
LOGGER.debug("%s: Retrieving current device date/time", self.name)
try:
device_time = await device_mgmt.GetSystemDateAndTime()
- except RequestError as err:
+ except (RequestError, Fault) as err:
LOGGER.warning(
"Couldn't get device '%s' date/time. Error: %s", self.name, err
)
diff --git a/homeassistant/components/onvif/entity.py b/homeassistant/components/onvif/entity.py
index c9900106256..783df743e86 100644
--- a/homeassistant/components/onvif/entity.py
+++ b/homeassistant/components/onvif/entity.py
@@ -17,7 +17,7 @@ class ONVIFBaseEntity(Entity):
self.device: ONVIFDevice = device
@property
- def available(self):
+ def available(self) -> bool:
"""Return True if device is available."""
return self.device.available
diff --git a/homeassistant/components/onvif/event.py b/homeassistant/components/onvif/event.py
index b7b34f7be9f..d1b93304ccc 100644
--- a/homeassistant/components/onvif/event.py
+++ b/homeassistant/components/onvif/event.py
@@ -174,11 +174,20 @@ class EventManager:
UNHANDLED_TOPICS.add(topic)
continue
- event = await parser(unique_id, msg)
+ try:
+ event = await parser(unique_id, msg)
+ error = None
+ except (AttributeError, KeyError) as e:
+ event = None
+ error = e
if not event:
LOGGER.warning(
- "%s: Unable to parse event from %s: %s", self.name, unique_id, msg
+ "%s: Unable to parse event from %s: %s: %s",
+ self.name,
+ unique_id,
+ error,
+ msg,
)
return
diff --git a/homeassistant/components/onvif/parsers.py b/homeassistant/components/onvif/parsers.py
index 6eb1d001796..e5a731c73f6 100644
--- a/homeassistant/components/onvif/parsers.py
+++ b/homeassistant/components/onvif/parsers.py
@@ -49,24 +49,22 @@ def local_datetime_or_none(value: str) -> datetime.datetime | None:
@PARSERS.register("tns1:VideoSource/MotionAlarm")
+@PARSERS.register("tns1:Device/Trigger/tnshik:AlarmIn")
async def async_parse_motion_alarm(uid: str, msg) -> Event | None:
"""Handle parsing event message.
Topic: tns1:VideoSource/MotionAlarm
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Motion Alarm",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Motion Alarm",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:VideoSource/ImageTooBlurry/AnalyticsService")
@@ -77,20 +75,17 @@ async def async_parse_image_too_blurry(uid: str, msg) -> Event | None:
Topic: tns1:VideoSource/ImageTooBlurry/*
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Image Too Blurry",
- "binary_sensor",
- "problem",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Image Too Blurry",
+ "binary_sensor",
+ "problem",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:VideoSource/ImageTooDark/AnalyticsService")
@@ -101,20 +96,17 @@ async def async_parse_image_too_dark(uid: str, msg) -> Event | None:
Topic: tns1:VideoSource/ImageTooDark/*
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Image Too Dark",
- "binary_sensor",
- "problem",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Image Too Dark",
+ "binary_sensor",
+ "problem",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:VideoSource/ImageTooBright/AnalyticsService")
@@ -125,20 +117,17 @@ async def async_parse_image_too_bright(uid: str, msg) -> Event | None:
Topic: tns1:VideoSource/ImageTooBright/*
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Image Too Bright",
- "binary_sensor",
- "problem",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Image Too Bright",
+ "binary_sensor",
+ "problem",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:VideoSource/GlobalSceneChange/AnalyticsService")
@@ -149,19 +138,16 @@ async def async_parse_scene_change(uid: str, msg) -> Event | None:
Topic: tns1:VideoSource/GlobalSceneChange/*
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Global Scene Change",
- "binary_sensor",
- "problem",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Global Scene Change",
+ "binary_sensor",
+ "problem",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:AudioAnalytics/Audio/DetectedSound")
@@ -170,29 +156,26 @@ async def async_parse_detected_sound(uid: str, msg) -> Event | None:
Topic: tns1:AudioAnalytics/Audio/DetectedSound
"""
- try:
- audio_source = ""
- audio_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "AudioSourceConfigurationToken":
- audio_source = source.Value
- if source.Name == "AudioAnalyticsConfigurationToken":
- audio_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ audio_source = ""
+ audio_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "AudioSourceConfigurationToken":
+ audio_source = source.Value
+ if source.Name == "AudioAnalyticsConfigurationToken":
+ audio_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- return Event(
- f"{uid}_{topic}_{audio_source}_{audio_analytics}_{rule}",
- "Detected Sound",
- "binary_sensor",
- "sound",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{audio_source}_{audio_analytics}_{rule}",
+ "Detected Sound",
+ "binary_sensor",
+ "sound",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:RuleEngine/FieldDetector/ObjectsInside")
@@ -201,30 +184,26 @@ async def async_parse_field_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/FieldDetector/ObjectsInside
"""
- try:
- video_source = ""
- video_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = _normalize_video_source(source.Value)
- if source.Name == "VideoAnalyticsConfigurationToken":
- video_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ video_source = ""
+ video_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = _normalize_video_source(source.Value)
+ if source.Name == "VideoAnalyticsConfigurationToken":
+ video_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- evt = Event(
- f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
- "Field Detection",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
- return evt
+ return Event(
+ f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
+ "Field Detection",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:RuleEngine/CellMotionDetector/Motion")
@@ -233,29 +212,26 @@ async def async_parse_cell_motion_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/CellMotionDetector/Motion
"""
- try:
- video_source = ""
- video_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = _normalize_video_source(source.Value)
- if source.Name == "VideoAnalyticsConfigurationToken":
- video_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ video_source = ""
+ video_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = _normalize_video_source(source.Value)
+ if source.Name == "VideoAnalyticsConfigurationToken":
+ video_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- return Event(
- f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
- "Cell Motion Detection",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
+ "Cell Motion Detection",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:RuleEngine/MotionRegionDetector/Motion")
@@ -264,29 +240,26 @@ async def async_parse_motion_region_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/MotionRegionDetector/Motion
"""
- try:
- video_source = ""
- video_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = _normalize_video_source(source.Value)
- if source.Name == "VideoAnalyticsConfigurationToken":
- video_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ video_source = ""
+ video_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = _normalize_video_source(source.Value)
+ if source.Name == "VideoAnalyticsConfigurationToken":
+ video_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- return Event(
- f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
- "Motion Region Detection",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value in ["1", "true"],
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
+ "Motion Region Detection",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value in ["1", "true"],
+ )
@PARSERS.register("tns1:RuleEngine/TamperDetector/Tamper")
@@ -295,30 +268,27 @@ async def async_parse_tamper_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/TamperDetector/Tamper
"""
- try:
- video_source = ""
- video_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = _normalize_video_source(source.Value)
- if source.Name == "VideoAnalyticsConfigurationToken":
- video_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ video_source = ""
+ video_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = _normalize_video_source(source.Value)
+ if source.Name == "VideoAnalyticsConfigurationToken":
+ video_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- return Event(
- f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
- "Tamper Detection",
- "binary_sensor",
- "problem",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
+ "Tamper Detection",
+ "binary_sensor",
+ "problem",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:RuleEngine/MyRuleDetector/DogCatDetect")
@@ -327,23 +297,20 @@ async def async_parse_dog_cat_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/MyRuleDetector/DogCatDetect
"""
- try:
- video_source = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "Source":
- video_source = _normalize_video_source(source.Value)
+ video_source = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "Source":
+ video_source = _normalize_video_source(source.Value)
- return Event(
- f"{uid}_{topic}_{video_source}",
- "Pet Detection",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}",
+ "Pet Detection",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:RuleEngine/MyRuleDetector/VehicleDetect")
@@ -352,23 +319,20 @@ async def async_parse_vehicle_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/MyRuleDetector/VehicleDetect
"""
- try:
- video_source = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "Source":
- video_source = _normalize_video_source(source.Value)
+ video_source = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "Source":
+ video_source = _normalize_video_source(source.Value)
- return Event(
- f"{uid}_{topic}_{video_source}",
- "Vehicle Detection",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}",
+ "Vehicle Detection",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
_TAPO_EVENT_TEMPLATES: dict[str, Event] = {
@@ -420,32 +384,28 @@ async def async_parse_tplink_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/PeopleDetector/People
Topic: tns1:RuleEngine/TPSmartEventDetector/TPSmartEvent
"""
- try:
- video_source = ""
- video_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = _normalize_video_source(source.Value)
- if source.Name == "VideoAnalyticsConfigurationToken":
- video_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ video_source = ""
+ video_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = _normalize_video_source(source.Value)
+ if source.Name == "VideoAnalyticsConfigurationToken":
+ video_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- for item in payload.Data.SimpleItem:
- event_template = _TAPO_EVENT_TEMPLATES.get(item.Name, None)
- if event_template is None:
- continue
+ for item in payload.Data.SimpleItem:
+ event_template = _TAPO_EVENT_TEMPLATES.get(item.Name, None)
+ if event_template is None:
+ continue
- return dataclasses.replace(
- event_template,
- uid=f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
- value=item.Value == "true",
- )
-
- except (AttributeError, KeyError):
- return None
+ return dataclasses.replace(
+ event_template,
+ uid=f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
+ value=item.Value == "true",
+ )
return None
@@ -456,23 +416,20 @@ async def async_parse_person_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/MyRuleDetector/PeopleDetect
"""
- try:
- video_source = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "Source":
- video_source = _normalize_video_source(source.Value)
+ video_source = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "Source":
+ video_source = _normalize_video_source(source.Value)
- return Event(
- f"{uid}_{topic}_{video_source}",
- "Person Detection",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}",
+ "Person Detection",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:RuleEngine/MyRuleDetector/FaceDetect")
@@ -481,23 +438,20 @@ async def async_parse_face_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/MyRuleDetector/FaceDetect
"""
- try:
- video_source = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "Source":
- video_source = _normalize_video_source(source.Value)
+ video_source = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "Source":
+ video_source = _normalize_video_source(source.Value)
- return Event(
- f"{uid}_{topic}_{video_source}",
- "Face Detection",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}",
+ "Face Detection",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:RuleEngine/MyRuleDetector/Visitor")
@@ -506,23 +460,42 @@ async def async_parse_visitor_detector(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/MyRuleDetector/Visitor
"""
- try:
- video_source = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "Source":
- video_source = _normalize_video_source(source.Value)
+ video_source = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "Source":
+ video_source = _normalize_video_source(source.Value)
- return Event(
- f"{uid}_{topic}_{video_source}",
- "Visitor Detection",
- "binary_sensor",
- "occupancy",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}",
+ "Visitor Detection",
+ "binary_sensor",
+ "occupancy",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
+
+
+@PARSERS.register("tns1:RuleEngine/MyRuleDetector/Package")
+async def async_parse_package_detector(uid: str, msg) -> Event | None:
+ """Handle parsing event message.
+
+ Topic: tns1:RuleEngine/MyRuleDetector/Package
+ """
+ video_source = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "Source":
+ video_source = _normalize_video_source(source.Value)
+
+ return Event(
+ f"{uid}_{topic}_{video_source}",
+ "Package Detection",
+ "binary_sensor",
+ "occupancy",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:Device/Trigger/DigitalInput")
@@ -531,19 +504,16 @@ async def async_parse_digital_input(uid: str, msg) -> Event | None:
Topic: tns1:Device/Trigger/DigitalInput
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Digital Input",
- "binary_sensor",
- None,
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Digital Input",
+ "binary_sensor",
+ None,
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
@PARSERS.register("tns1:Device/Trigger/Relay")
@@ -552,19 +522,16 @@ async def async_parse_relay(uid: str, msg) -> Event | None:
Topic: tns1:Device/Trigger/Relay
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Relay Triggered",
- "binary_sensor",
- None,
- None,
- payload.Data.SimpleItem[0].Value == "active",
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Relay Triggered",
+ "binary_sensor",
+ None,
+ None,
+ payload.Data.SimpleItem[0].Value == "active",
+ )
@PARSERS.register("tns1:Device/HardwareFailure/StorageFailure")
@@ -573,20 +540,17 @@ async def async_parse_storage_failure(uid: str, msg) -> Event | None:
Topic: tns1:Device/HardwareFailure/StorageFailure
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Storage Failure",
- "binary_sensor",
- "problem",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Storage Failure",
+ "binary_sensor",
+ "problem",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:Monitoring/ProcessorUsage")
@@ -595,23 +559,20 @@ async def async_parse_processor_usage(uid: str, msg) -> Event | None:
Topic: tns1:Monitoring/ProcessorUsage
"""
- try:
- topic, payload = extract_message(msg)
- usage = float(payload.Data.SimpleItem[0].Value)
- if usage <= 1:
- usage *= 100
+ topic, payload = extract_message(msg)
+ usage = float(payload.Data.SimpleItem[0].Value)
+ if usage <= 1:
+ usage *= 100
- return Event(
- f"{uid}_{topic}",
- "Processor Usage",
- "sensor",
- None,
- "percent",
- int(usage),
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}",
+ "Processor Usage",
+ "sensor",
+ None,
+ "percent",
+ int(usage),
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:Monitoring/OperatingTime/LastReboot")
@@ -620,20 +581,17 @@ async def async_parse_last_reboot(uid: str, msg) -> Event | None:
Topic: tns1:Monitoring/OperatingTime/LastReboot
"""
- try:
- topic, payload = extract_message(msg)
- date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
- return Event(
- f"{uid}_{topic}",
- "Last Reboot",
- "sensor",
- "timestamp",
- None,
- date_time,
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
+ return Event(
+ f"{uid}_{topic}",
+ "Last Reboot",
+ "sensor",
+ "timestamp",
+ None,
+ date_time,
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:Monitoring/OperatingTime/LastReset")
@@ -642,21 +600,18 @@ async def async_parse_last_reset(uid: str, msg) -> Event | None:
Topic: tns1:Monitoring/OperatingTime/LastReset
"""
- try:
- topic, payload = extract_message(msg)
- date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
- return Event(
- f"{uid}_{topic}",
- "Last Reset",
- "sensor",
- "timestamp",
- None,
- date_time,
- EntityCategory.DIAGNOSTIC,
- entity_enabled=False,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
+ return Event(
+ f"{uid}_{topic}",
+ "Last Reset",
+ "sensor",
+ "timestamp",
+ None,
+ date_time,
+ EntityCategory.DIAGNOSTIC,
+ entity_enabled=False,
+ )
@PARSERS.register("tns1:Monitoring/Backup/Last")
@@ -665,22 +620,18 @@ async def async_parse_backup_last(uid: str, msg) -> Event | None:
Topic: tns1:Monitoring/Backup/Last
"""
-
- try:
- topic, payload = extract_message(msg)
- date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
- return Event(
- f"{uid}_{topic}",
- "Last Backup",
- "sensor",
- "timestamp",
- None,
- date_time,
- EntityCategory.DIAGNOSTIC,
- entity_enabled=False,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
+ return Event(
+ f"{uid}_{topic}",
+ "Last Backup",
+ "sensor",
+ "timestamp",
+ None,
+ date_time,
+ EntityCategory.DIAGNOSTIC,
+ entity_enabled=False,
+ )
@PARSERS.register("tns1:Monitoring/OperatingTime/LastClockSynchronization")
@@ -689,21 +640,18 @@ async def async_parse_last_clock_sync(uid: str, msg) -> Event | None:
Topic: tns1:Monitoring/OperatingTime/LastClockSynchronization
"""
- try:
- topic, payload = extract_message(msg)
- date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
- return Event(
- f"{uid}_{topic}",
- "Last Clock Synchronization",
- "sensor",
- "timestamp",
- None,
- date_time,
- EntityCategory.DIAGNOSTIC,
- entity_enabled=False,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ date_time = local_datetime_or_none(payload.Data.SimpleItem[0].Value)
+ return Event(
+ f"{uid}_{topic}",
+ "Last Clock Synchronization",
+ "sensor",
+ "timestamp",
+ None,
+ date_time,
+ EntityCategory.DIAGNOSTIC,
+ entity_enabled=False,
+ )
@PARSERS.register("tns1:RecordingConfig/JobState")
@@ -713,20 +661,17 @@ async def async_parse_jobstate(uid: str, msg) -> Event | None:
Topic: tns1:RecordingConfig/JobState
"""
- try:
- topic, payload = extract_message(msg)
- source = payload.Source.SimpleItem[0].Value
- return Event(
- f"{uid}_{topic}_{source}",
- "Recording Job State",
- "binary_sensor",
- None,
- None,
- payload.Data.SimpleItem[0].Value == "Active",
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ topic, payload = extract_message(msg)
+ source = payload.Source.SimpleItem[0].Value
+ return Event(
+ f"{uid}_{topic}_{source}",
+ "Recording Job State",
+ "binary_sensor",
+ None,
+ None,
+ payload.Data.SimpleItem[0].Value == "Active",
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:RuleEngine/LineDetector/Crossed")
@@ -735,30 +680,27 @@ async def async_parse_linedetector_crossed(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/LineDetector/Crossed
"""
- try:
- video_source = ""
- video_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = source.Value
- if source.Name == "VideoAnalyticsConfigurationToken":
- video_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ video_source = ""
+ video_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = source.Value
+ if source.Name == "VideoAnalyticsConfigurationToken":
+ video_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- return Event(
- f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
- "Line Detector Crossed",
- "sensor",
- None,
- None,
- payload.Data.SimpleItem[0].Value,
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
+ "Line Detector Crossed",
+ "sensor",
+ None,
+ None,
+ payload.Data.SimpleItem[0].Value,
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:RuleEngine/CountAggregation/Counter")
@@ -767,30 +709,27 @@ async def async_parse_count_aggregation_counter(uid: str, msg) -> Event | None:
Topic: tns1:RuleEngine/CountAggregation/Counter
"""
- try:
- video_source = ""
- video_analytics = ""
- rule = ""
- topic, payload = extract_message(msg)
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = _normalize_video_source(source.Value)
- if source.Name == "VideoAnalyticsConfigurationToken":
- video_analytics = source.Value
- if source.Name == "Rule":
- rule = source.Value
+ video_source = ""
+ video_analytics = ""
+ rule = ""
+ topic, payload = extract_message(msg)
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = _normalize_video_source(source.Value)
+ if source.Name == "VideoAnalyticsConfigurationToken":
+ video_analytics = source.Value
+ if source.Name == "Rule":
+ rule = source.Value
- return Event(
- f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
- "Count Aggregation Counter",
- "sensor",
- None,
- None,
- payload.Data.SimpleItem[0].Value,
- EntityCategory.DIAGNOSTIC,
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}_{video_analytics}_{rule}",
+ "Count Aggregation Counter",
+ "sensor",
+ None,
+ None,
+ payload.Data.SimpleItem[0].Value,
+ EntityCategory.DIAGNOSTIC,
+ )
@PARSERS.register("tns1:UserAlarm/IVA/HumanShapeDetect")
@@ -799,21 +738,18 @@ async def async_parse_human_shape_detect(uid: str, msg) -> Event | None:
Topic: tns1:UserAlarm/IVA/HumanShapeDetect
"""
- try:
- topic, payload = extract_message(msg)
- video_source = ""
- for source in payload.Source.SimpleItem:
- if source.Name == "VideoSourceConfigurationToken":
- video_source = _normalize_video_source(source.Value)
- break
+ topic, payload = extract_message(msg)
+ video_source = ""
+ for source in payload.Source.SimpleItem:
+ if source.Name == "VideoSourceConfigurationToken":
+ video_source = _normalize_video_source(source.Value)
+ break
- return Event(
- f"{uid}_{topic}_{video_source}",
- "Human Shape Detect",
- "binary_sensor",
- "motion",
- None,
- payload.Data.SimpleItem[0].Value == "true",
- )
- except (AttributeError, KeyError):
- return None
+ return Event(
+ f"{uid}_{topic}_{video_source}",
+ "Human Shape Detect",
+ "binary_sensor",
+ "motion",
+ None,
+ payload.Data.SimpleItem[0].Value == "true",
+ )
diff --git a/homeassistant/components/onvif/sensor.py b/homeassistant/components/onvif/sensor.py
index 46db26361bc..a0162a05f76 100644
--- a/homeassistant/components/onvif/sensor.py
+++ b/homeassistant/components/onvif/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import RestoreSensor, SensorDeviceClass
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.enum import try_parse_enum
@@ -21,7 +21,7 @@ from .entity import ONVIFBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a ONVIF binary sensor."""
device: ONVIFDevice = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/onvif/strings.json b/homeassistant/components/onvif/strings.json
index 0afb5e59e8e..7988c50b1ac 100644
--- a/homeassistant/components/onvif/strings.json
+++ b/homeassistant/components/onvif/strings.json
@@ -62,12 +62,12 @@
"step": {
"onvif_devices": {
"data": {
- "extra_arguments": "Extra FFMPEG arguments",
+ "extra_arguments": "Extra FFmpeg arguments",
"rtsp_transport": "RTSP transport mechanism",
"use_wallclock_as_timestamps": "Use wall clock as timestamps",
- "enable_webhooks": "Enable Webhooks"
+ "enable_webhooks": "Enable webhooks"
},
- "title": "ONVIF Device Options"
+ "title": "ONVIF device options"
}
}
},
diff --git a/homeassistant/components/onvif/switch.py b/homeassistant/components/onvif/switch.py
index ff62e469af0..d8e1020c6a3 100644
--- a/homeassistant/components/onvif/switch.py
+++ b/homeassistant/components/onvif/switch.py
@@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .device import ONVIFDevice
@@ -66,7 +66,7 @@ SWITCHES: tuple[ONVIFSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a ONVIF switch platform."""
device = hass.data[DOMAIN][config_entry.unique_id]
diff --git a/homeassistant/components/open_meteo/weather.py b/homeassistant/components/open_meteo/weather.py
index 51ee91de083..9782051ab22 100644
--- a/homeassistant/components/open_meteo/weather.py
+++ b/homeassistant/components/open_meteo/weather.py
@@ -20,7 +20,7 @@ from homeassistant.components.weather import (
from homeassistant.const import UnitOfPrecipitationDepth, UnitOfSpeed, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from homeassistant.util import dt as dt_util
@@ -31,7 +31,7 @@ from .coordinator import OpenMeteoConfigEntry
async def async_setup_entry(
hass: HomeAssistant,
entry: OpenMeteoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Open-Meteo weather entity based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/openai_conversation/__init__.py b/homeassistant/components/openai_conversation/__init__.py
index 0fbda9b7f4a..fcf6ab298dc 100644
--- a/homeassistant/components/openai_conversation/__init__.py
+++ b/homeassistant/components/openai_conversation/__init__.py
@@ -2,7 +2,20 @@
from __future__ import annotations
+import base64
+from mimetypes import guess_file_type
+from pathlib import Path
+
import openai
+from openai.types.images_response import ImagesResponse
+from openai.types.responses import (
+ EasyInputMessageParam,
+ Response,
+ ResponseInputImageParam,
+ ResponseInputMessageContentListParam,
+ ResponseInputParam,
+ ResponseInputTextParam,
+)
import voluptuous as vol
from homeassistant.config_entries import ConfigEntry
@@ -22,15 +35,41 @@ from homeassistant.helpers import config_validation as cv, selector
from homeassistant.helpers.httpx_client import get_async_client
from homeassistant.helpers.typing import ConfigType
-from .const import DOMAIN, LOGGER
+from .const import (
+ CONF_CHAT_MODEL,
+ CONF_FILENAMES,
+ CONF_MAX_TOKENS,
+ CONF_PROMPT,
+ CONF_REASONING_EFFORT,
+ CONF_TEMPERATURE,
+ CONF_TOP_P,
+ DOMAIN,
+ LOGGER,
+ RECOMMENDED_CHAT_MODEL,
+ RECOMMENDED_MAX_TOKENS,
+ RECOMMENDED_REASONING_EFFORT,
+ RECOMMENDED_TEMPERATURE,
+ RECOMMENDED_TOP_P,
+)
SERVICE_GENERATE_IMAGE = "generate_image"
+SERVICE_GENERATE_CONTENT = "generate_content"
+
PLATFORMS = (Platform.CONVERSATION,)
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
type OpenAIConfigEntry = ConfigEntry[openai.AsyncClient]
+def encode_file(file_path: str) -> tuple[str, str]:
+ """Return base64 version of file contents."""
+ mime_type, _ = guess_file_type(file_path)
+ if mime_type is None:
+ mime_type = "application/octet-stream"
+ with open(file_path, "rb") as image_file:
+ return (mime_type, base64.b64encode(image_file.read()).decode("utf-8"))
+
+
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up OpenAI Conversation."""
@@ -49,9 +88,9 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
client: openai.AsyncClient = entry.runtime_data
try:
- response = await client.images.generate(
+ response: ImagesResponse = await client.images.generate(
model="dall-e-3",
- prompt=call.data["prompt"],
+ prompt=call.data[CONF_PROMPT],
size=call.data["size"],
quality=call.data["quality"],
style=call.data["style"],
@@ -63,6 +102,108 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
return response.data[0].model_dump(exclude={"b64_json"})
+ async def send_prompt(call: ServiceCall) -> ServiceResponse:
+ """Send a prompt to ChatGPT and return the response."""
+ entry_id = call.data["config_entry"]
+ entry = hass.config_entries.async_get_entry(entry_id)
+
+ if entry is None or entry.domain != DOMAIN:
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="invalid_config_entry",
+ translation_placeholders={"config_entry": entry_id},
+ )
+
+ model: str = entry.options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL)
+ client: openai.AsyncClient = entry.runtime_data
+
+ content: ResponseInputMessageContentListParam = [
+ ResponseInputTextParam(type="input_text", text=call.data[CONF_PROMPT])
+ ]
+
+ def append_files_to_content() -> None:
+ for filename in call.data[CONF_FILENAMES]:
+ if not hass.config.is_allowed_path(filename):
+ raise HomeAssistantError(
+ f"Cannot read `{filename}`, no access to path; "
+ "`allowlist_external_dirs` may need to be adjusted in "
+ "`configuration.yaml`"
+ )
+ if not Path(filename).exists():
+ raise HomeAssistantError(f"`{filename}` does not exist")
+ mime_type, base64_file = encode_file(filename)
+ if "image/" not in mime_type:
+ raise HomeAssistantError(
+ "Only images are supported by the OpenAI API,"
+ f"`{filename}` is not an image file"
+ )
+ content.append(
+ ResponseInputImageParam(
+ type="input_image",
+ file_id=filename,
+ image_url=f"data:{mime_type};base64,{base64_file}",
+ detail="auto",
+ )
+ )
+
+ if CONF_FILENAMES in call.data:
+ await hass.async_add_executor_job(append_files_to_content)
+
+ messages: ResponseInputParam = [
+ EasyInputMessageParam(type="message", role="user", content=content)
+ ]
+
+ try:
+ model_args = {
+ "model": model,
+ "input": messages,
+ "max_output_tokens": entry.options.get(
+ CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS
+ ),
+ "top_p": entry.options.get(CONF_TOP_P, RECOMMENDED_TOP_P),
+ "temperature": entry.options.get(
+ CONF_TEMPERATURE, RECOMMENDED_TEMPERATURE
+ ),
+ "user": call.context.user_id,
+ "store": False,
+ }
+
+ if model.startswith("o"):
+ model_args["reasoning"] = {
+ "effort": entry.options.get(
+ CONF_REASONING_EFFORT, RECOMMENDED_REASONING_EFFORT
+ )
+ }
+
+ response: Response = await client.responses.create(**model_args)
+
+ except openai.OpenAIError as err:
+ raise HomeAssistantError(f"Error generating content: {err}") from err
+ except FileNotFoundError as err:
+ raise HomeAssistantError(f"Error generating content: {err}") from err
+
+ return {"text": response.output_text}
+
+ hass.services.async_register(
+ DOMAIN,
+ SERVICE_GENERATE_CONTENT,
+ send_prompt,
+ schema=vol.Schema(
+ {
+ vol.Required("config_entry"): selector.ConfigEntrySelector(
+ {
+ "integration": DOMAIN,
+ }
+ ),
+ vol.Required(CONF_PROMPT): cv.string,
+ vol.Optional(CONF_FILENAMES, default=[]): vol.All(
+ cv.ensure_list, [cv.string]
+ ),
+ }
+ ),
+ supports_response=SupportsResponse.ONLY,
+ )
+
hass.services.async_register(
DOMAIN,
SERVICE_GENERATE_IMAGE,
@@ -74,7 +215,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"integration": DOMAIN,
}
),
- vol.Required("prompt"): cv.string,
+ vol.Required(CONF_PROMPT): cv.string,
vol.Optional("size", default="1024x1024"): vol.In(
("1024x1024", "1024x1792", "1792x1024")
),
@@ -84,6 +225,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
),
supports_response=SupportsResponse.ONLY,
)
+
return True
diff --git a/homeassistant/components/openai_conversation/const.py b/homeassistant/components/openai_conversation/const.py
index 793e021e332..c9987cb81b9 100644
--- a/homeassistant/components/openai_conversation/const.py
+++ b/homeassistant/components/openai_conversation/const.py
@@ -3,22 +3,24 @@
import logging
DOMAIN = "openai_conversation"
-LOGGER = logging.getLogger(__package__)
+LOGGER: logging.Logger = logging.getLogger(__package__)
-CONF_RECOMMENDED = "recommended"
-CONF_PROMPT = "prompt"
CONF_CHAT_MODEL = "chat_model"
-RECOMMENDED_CHAT_MODEL = "gpt-4o-mini"
+CONF_FILENAMES = "filenames"
CONF_MAX_TOKENS = "max_tokens"
-RECOMMENDED_MAX_TOKENS = 150
-CONF_TOP_P = "top_p"
-RECOMMENDED_TOP_P = 1.0
-CONF_TEMPERATURE = "temperature"
-RECOMMENDED_TEMPERATURE = 1.0
+CONF_PROMPT = "prompt"
+CONF_PROMPT = "prompt"
CONF_REASONING_EFFORT = "reasoning_effort"
+CONF_RECOMMENDED = "recommended"
+CONF_TEMPERATURE = "temperature"
+CONF_TOP_P = "top_p"
+RECOMMENDED_CHAT_MODEL = "gpt-4o-mini"
+RECOMMENDED_MAX_TOKENS = 150
RECOMMENDED_REASONING_EFFORT = "low"
+RECOMMENDED_TEMPERATURE = 1.0
+RECOMMENDED_TOP_P = 1.0
-UNSUPPORTED_MODELS = [
+UNSUPPORTED_MODELS: list[str] = [
"o1-mini",
"o1-mini-2024-09-12",
"o1-preview",
diff --git a/homeassistant/components/openai_conversation/conversation.py b/homeassistant/components/openai_conversation/conversation.py
index 73dafa1c48d..32ac20b2680 100644
--- a/homeassistant/components/openai_conversation/conversation.py
+++ b/homeassistant/components/openai_conversation/conversation.py
@@ -1,21 +1,27 @@
"""Conversation support for OpenAI."""
-from collections.abc import Callable
+from collections.abc import AsyncGenerator, Callable
import json
-from typing import Any, Literal, cast
+from typing import Any, Literal
import openai
-from openai._types import NOT_GIVEN
-from openai.types.chat import (
- ChatCompletionAssistantMessageParam,
- ChatCompletionMessage,
- ChatCompletionMessageParam,
- ChatCompletionMessageToolCallParam,
- ChatCompletionToolMessageParam,
- ChatCompletionToolParam,
+from openai._streaming import AsyncStream
+from openai.types.responses import (
+ EasyInputMessageParam,
+ FunctionToolParam,
+ ResponseCompletedEvent,
+ ResponseFunctionCallArgumentsDeltaEvent,
+ ResponseFunctionCallArgumentsDoneEvent,
+ ResponseFunctionToolCall,
+ ResponseFunctionToolCallParam,
+ ResponseInputParam,
+ ResponseOutputItemAddedEvent,
+ ResponseOutputMessage,
+ ResponseStreamEvent,
+ ResponseTextDeltaEvent,
+ ToolParam,
)
-from openai.types.chat.chat_completion_message_tool_call_param import Function
-from openai.types.shared_params import FunctionDefinition
+from openai.types.responses.response_input_param import FunctionCallOutput
from voluptuous_openapi import convert
from homeassistant.components import assist_pipeline, conversation
@@ -23,8 +29,8 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_LLM_HASS_API, MATCH_ALL
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers import chat_session, device_registry as dr, intent, llm
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers import device_registry as dr, intent, llm
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OpenAIConfigEntry
from .const import (
@@ -50,7 +56,7 @@ MAX_TOOL_ITERATIONS = 10
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OpenAIConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up conversation entities."""
agent = OpenAIConversationEntity(config_entry)
@@ -59,80 +65,92 @@ async def async_setup_entry(
def _format_tool(
tool: llm.Tool, custom_serializer: Callable[[Any], Any] | None
-) -> ChatCompletionToolParam:
+) -> FunctionToolParam:
"""Format tool specification."""
- tool_spec = FunctionDefinition(
+ return FunctionToolParam(
+ type="function",
name=tool.name,
parameters=convert(tool.parameters, custom_serializer=custom_serializer),
+ description=tool.description,
+ strict=False,
)
- if tool.description:
- tool_spec["description"] = tool.description
- return ChatCompletionToolParam(type="function", function=tool_spec)
-
-
-def _convert_message_to_param(
- message: ChatCompletionMessage,
-) -> ChatCompletionMessageParam:
- """Convert from class to TypedDict."""
- tool_calls: list[ChatCompletionMessageToolCallParam] = []
- if message.tool_calls:
- tool_calls = [
- ChatCompletionMessageToolCallParam(
- id=tool_call.id,
- function=Function(
- arguments=tool_call.function.arguments,
- name=tool_call.function.name,
- ),
- type=tool_call.type,
- )
- for tool_call in message.tool_calls
- ]
- param = ChatCompletionAssistantMessageParam(
- role=message.role,
- content=message.content,
- )
- if tool_calls:
- param["tool_calls"] = tool_calls
- return param
def _convert_content_to_param(
content: conversation.Content,
-) -> ChatCompletionMessageParam:
+) -> ResponseInputParam:
"""Convert any native chat message for this agent to the native format."""
- if content.role == "tool_result":
- assert type(content) is conversation.ToolResultContent
- return ChatCompletionToolMessageParam(
- role="tool",
- tool_call_id=content.tool_call_id,
- content=json.dumps(content.tool_result),
- )
- if content.role != "assistant" or not content.tool_calls: # type: ignore[union-attr]
- role = content.role
+ messages: ResponseInputParam = []
+ if isinstance(content, conversation.ToolResultContent):
+ return [
+ FunctionCallOutput(
+ type="function_call_output",
+ call_id=content.tool_call_id,
+ output=json.dumps(content.tool_result),
+ )
+ ]
+
+ if content.content:
+ role: Literal["user", "assistant", "system", "developer"] = content.role
if role == "system":
role = "developer"
- return cast(
- ChatCompletionMessageParam,
- {"role": content.role, "content": content.content}, # type: ignore[union-attr]
+ messages.append(
+ EasyInputMessageParam(type="message", role=role, content=content.content)
)
- # Handle the Assistant content including tool calls.
- assert type(content) is conversation.AssistantContent
- return ChatCompletionAssistantMessageParam(
- role="assistant",
- content=content.content,
- tool_calls=[
- ChatCompletionMessageToolCallParam(
- id=tool_call.id,
- function=Function(
- arguments=json.dumps(tool_call.tool_args),
- name=tool_call.tool_name,
- ),
- type="function",
+ if isinstance(content, conversation.AssistantContent) and content.tool_calls:
+ messages.extend(
+ ResponseFunctionToolCallParam(
+ type="function_call",
+ name=tool_call.tool_name,
+ arguments=json.dumps(tool_call.tool_args),
+ call_id=tool_call.id,
)
for tool_call in content.tool_calls
- ],
- )
+ )
+ return messages
+
+
+async def _transform_stream(
+ chat_log: conversation.ChatLog,
+ result: AsyncStream[ResponseStreamEvent],
+) -> AsyncGenerator[conversation.AssistantContentDeltaDict]:
+ """Transform an OpenAI delta stream into HA format."""
+ async for event in result:
+ LOGGER.debug("Received event: %s", event)
+
+ if isinstance(event, ResponseOutputItemAddedEvent):
+ if isinstance(event.item, ResponseOutputMessage):
+ yield {"role": event.item.role}
+ elif isinstance(event.item, ResponseFunctionToolCall):
+ current_tool_call = event.item
+ elif isinstance(event, ResponseTextDeltaEvent):
+ yield {"content": event.delta}
+ elif isinstance(event, ResponseFunctionCallArgumentsDeltaEvent):
+ current_tool_call.arguments += event.delta
+ elif isinstance(event, ResponseFunctionCallArgumentsDoneEvent):
+ current_tool_call.status = "completed"
+ yield {
+ "tool_calls": [
+ llm.ToolInput(
+ id=current_tool_call.call_id,
+ tool_name=current_tool_call.name,
+ tool_args=json.loads(current_tool_call.arguments),
+ )
+ ]
+ }
+ elif (
+ isinstance(event, ResponseCompletedEvent)
+ and (usage := event.response.usage) is not None
+ ):
+ chat_log.async_trace(
+ {
+ "stats": {
+ "input_tokens": usage.input_tokens,
+ "output_tokens": usage.output_tokens,
+ }
+ }
+ )
class OpenAIConversationEntity(
@@ -180,25 +198,12 @@ class OpenAIConversationEntity(
conversation.async_unset_agent(self.hass, self.entry)
await super().async_will_remove_from_hass()
- async def async_process(
- self, user_input: conversation.ConversationInput
- ) -> conversation.ConversationResult:
- """Process a sentence."""
- with (
- chat_session.async_get_chat_session(
- self.hass, user_input.conversation_id
- ) as session,
- conversation.async_get_chat_log(self.hass, session, user_input) as chat_log,
- ):
- return await self._async_handle_message(user_input, chat_log)
-
async def _async_handle_message(
self,
user_input: conversation.ConversationInput,
chat_log: conversation.ChatLog,
) -> conversation.ConversationResult:
"""Call the API."""
- assert user_input.agent_id
options = self.entry.options
try:
@@ -211,7 +216,7 @@ class OpenAIConversationEntity(
except conversation.ConverseError as err:
return err.as_conversation_result()
- tools: list[ChatCompletionToolParam] | None = None
+ tools: list[ToolParam] | None = None
if chat_log.llm_api:
tools = [
_format_tool(tool, chat_log.llm_api.custom_serializer)
@@ -219,7 +224,11 @@ class OpenAIConversationEntity(
]
model = options.get(CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL)
- messages = [_convert_content_to_param(content) for content in chat_log.content]
+ messages = [
+ m
+ for content in chat_log.content
+ for m in _convert_content_to_param(content)
+ ]
client = self.entry.runtime_data
@@ -227,62 +236,50 @@ class OpenAIConversationEntity(
for _iteration in range(MAX_TOOL_ITERATIONS):
model_args = {
"model": model,
- "messages": messages,
- "tools": tools or NOT_GIVEN,
- "max_completion_tokens": options.get(
+ "input": messages,
+ "max_output_tokens": options.get(
CONF_MAX_TOKENS, RECOMMENDED_MAX_TOKENS
),
"top_p": options.get(CONF_TOP_P, RECOMMENDED_TOP_P),
"temperature": options.get(CONF_TEMPERATURE, RECOMMENDED_TEMPERATURE),
"user": chat_log.conversation_id,
+ "store": False,
+ "stream": True,
}
+ if tools:
+ model_args["tools"] = tools
if model.startswith("o"):
- model_args["reasoning_effort"] = options.get(
- CONF_REASONING_EFFORT, RECOMMENDED_REASONING_EFFORT
- )
+ model_args["reasoning"] = {
+ "effort": options.get(
+ CONF_REASONING_EFFORT, RECOMMENDED_REASONING_EFFORT
+ )
+ }
try:
- result = await client.chat.completions.create(**model_args)
+ result = await client.responses.create(**model_args)
+ except openai.RateLimitError as err:
+ LOGGER.error("Rate limited by OpenAI: %s", err)
+ raise HomeAssistantError("Rate limited or insufficient funds") from err
except openai.OpenAIError as err:
LOGGER.error("Error talking to OpenAI: %s", err)
raise HomeAssistantError("Error talking to OpenAI") from err
- LOGGER.debug("Response %s", result)
- response = result.choices[0].message
- messages.append(_convert_message_to_param(response))
+ async for content in chat_log.async_add_delta_content_stream(
+ user_input.agent_id, _transform_stream(chat_log, result)
+ ):
+ messages.extend(_convert_content_to_param(content))
- tool_calls: list[llm.ToolInput] | None = None
- if response.tool_calls:
- tool_calls = [
- llm.ToolInput(
- id=tool_call.id,
- tool_name=tool_call.function.name,
- tool_args=json.loads(tool_call.function.arguments),
- )
- for tool_call in response.tool_calls
- ]
-
- messages.extend(
- [
- _convert_content_to_param(tool_response)
- async for tool_response in chat_log.async_add_assistant_content(
- conversation.AssistantContent(
- agent_id=user_input.agent_id,
- content=response.content or "",
- tool_calls=tool_calls,
- )
- )
- ]
- )
-
- if not tool_calls:
+ if not chat_log.unresponded_tool_results:
break
intent_response = intent.IntentResponse(language=user_input.language)
- intent_response.async_set_speech(response.content or "")
+ assert type(chat_log.content[-1]) is conversation.AssistantContent
+ intent_response.async_set_speech(chat_log.content[-1].content or "")
return conversation.ConversationResult(
- response=intent_response, conversation_id=chat_log.conversation_id
+ response=intent_response,
+ conversation_id=chat_log.conversation_id,
+ continue_conversation=chat_log.continue_conversation,
)
async def _async_entry_update_listener(
diff --git a/homeassistant/components/openai_conversation/icons.json b/homeassistant/components/openai_conversation/icons.json
index 3abecd640d1..f0ece31c304 100644
--- a/homeassistant/components/openai_conversation/icons.json
+++ b/homeassistant/components/openai_conversation/icons.json
@@ -2,6 +2,9 @@
"services": {
"generate_image": {
"service": "mdi:image-sync"
+ },
+ "generate_content": {
+ "service": "mdi:receipt-text"
}
}
}
diff --git a/homeassistant/components/openai_conversation/manifest.json b/homeassistant/components/openai_conversation/manifest.json
index a7aa7884dc4..988dd2321d5 100644
--- a/homeassistant/components/openai_conversation/manifest.json
+++ b/homeassistant/components/openai_conversation/manifest.json
@@ -8,5 +8,5 @@
"documentation": "https://www.home-assistant.io/integrations/openai_conversation",
"integration_type": "service",
"iot_class": "cloud_polling",
- "requirements": ["openai==1.61.0"]
+ "requirements": ["openai==1.68.2"]
}
diff --git a/homeassistant/components/openai_conversation/services.yaml b/homeassistant/components/openai_conversation/services.yaml
index 3db71cae383..75fa097f25d 100644
--- a/homeassistant/components/openai_conversation/services.yaml
+++ b/homeassistant/components/openai_conversation/services.yaml
@@ -38,3 +38,23 @@ generate_image:
options:
- "vivid"
- "natural"
+generate_content:
+ fields:
+ config_entry:
+ required: true
+ selector:
+ config_entry:
+ integration: openai_conversation
+ prompt:
+ required: true
+ selector:
+ text:
+ multiline: true
+ example: "Hello, how can I help you?"
+ filenames:
+ selector:
+ text:
+ multiline: true
+ example: |
+ - /path/to/file1.txt
+ - /path/to/file2.txt
diff --git a/homeassistant/components/openai_conversation/strings.json b/homeassistant/components/openai_conversation/strings.json
index b8768f8abbe..c9d7ee112bd 100644
--- a/homeassistant/components/openai_conversation/strings.json
+++ b/homeassistant/components/openai_conversation/strings.json
@@ -48,10 +48,10 @@
"services": {
"generate_image": {
"name": "Generate image",
- "description": "Turn a prompt into an image",
+ "description": "Turns a prompt into an image",
"fields": {
"config_entry": {
- "name": "Config Entry",
+ "name": "Config entry",
"description": "The config entry to use for this action"
},
"prompt": {
@@ -72,6 +72,24 @@
"description": "The style of the generated image"
}
}
+ },
+ "generate_content": {
+ "name": "Generate content",
+ "description": "Sends a conversational query to ChatGPT including any attached image files",
+ "fields": {
+ "config_entry": {
+ "name": "Config entry",
+ "description": "The config entry to use for this action"
+ },
+ "prompt": {
+ "name": "Prompt",
+ "description": "The prompt to send"
+ },
+ "filenames": {
+ "name": "Files",
+ "description": "List of files to upload"
+ }
+ }
}
},
"exceptions": {
diff --git a/homeassistant/components/openexchangerates/__init__.py b/homeassistant/components/openexchangerates/__init__.py
index 65005235c6b..ed704a61fed 100644
--- a/homeassistant/components/openexchangerates/__init__.py
+++ b/homeassistant/components/openexchangerates/__init__.py
@@ -33,6 +33,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
update_interval = BASE_UPDATE_INTERVAL * (len(existing_coordinator_for_api_key) + 1)
coordinator = OpenexchangeratesCoordinator(
hass,
+ entry,
async_get_clientsession(hass),
api_key,
base,
diff --git a/homeassistant/components/openexchangerates/coordinator.py b/homeassistant/components/openexchangerates/coordinator.py
index 627e0d92e32..6245877ddbd 100644
--- a/homeassistant/components/openexchangerates/coordinator.py
+++ b/homeassistant/components/openexchangerates/coordinator.py
@@ -13,6 +13,7 @@ from aioopenexchangerates import (
OpenExchangeRatesClientError,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -23,9 +24,12 @@ from .const import CLIENT_TIMEOUT, DOMAIN, LOGGER
class OpenexchangeratesCoordinator(DataUpdateCoordinator[Latest]):
"""Represent a coordinator for Open Exchange Rates API."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
session: ClientSession,
api_key: str,
base: str,
@@ -33,7 +37,11 @@ class OpenexchangeratesCoordinator(DataUpdateCoordinator[Latest]):
) -> None:
"""Initialize the coordinator."""
super().__init__(
- hass, LOGGER, name=f"{DOMAIN} base {base}", update_interval=update_interval
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=f"{DOMAIN} base {base}",
+ update_interval=update_interval,
)
self.base = base
self.client = Client(api_key, session)
diff --git a/homeassistant/components/openexchangerates/sensor.py b/homeassistant/components/openexchangerates/sensor.py
index 55ca7bd2fb9..756823ff0ec 100644
--- a/homeassistant/components/openexchangerates/sensor.py
+++ b/homeassistant/components/openexchangerates/sensor.py
@@ -7,7 +7,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_QUOTE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -19,7 +19,7 @@ ATTRIBUTION = "Data provided by openexchangerates.org"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Open Exchange Rates sensor."""
quote: str = config_entry.data.get(CONF_QUOTE, "EUR")
diff --git a/homeassistant/components/opengarage/__init__.py b/homeassistant/components/opengarage/__init__.py
index 12c2f96d7e4..f1f080b30f8 100644
--- a/homeassistant/components/opengarage/__init__.py
+++ b/homeassistant/components/opengarage/__init__.py
@@ -24,8 +24,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async_get_clientsession(hass),
)
open_garage_data_coordinator = OpenGarageDataUpdateCoordinator(
- hass,
- open_garage_connection=open_garage_connection,
+ hass, entry, open_garage_connection
)
await open_garage_data_coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = open_garage_data_coordinator
diff --git a/homeassistant/components/opengarage/binary_sensor.py b/homeassistant/components/opengarage/binary_sensor.py
index 55cacfb5f90..33420ab3fd5 100644
--- a/homeassistant/components/opengarage/binary_sensor.py
+++ b/homeassistant/components/opengarage/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import OpenGarageDataUpdateCoordinator
@@ -29,7 +29,9 @@ SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenGarage binary sensors."""
open_garage_data_coordinator: OpenGarageDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/opengarage/button.py b/homeassistant/components/opengarage/button.py
index 9f93e0fa716..64a4f2f20e7 100644
--- a/homeassistant/components/opengarage/button.py
+++ b/homeassistant/components/opengarage/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import OpenGarageDataUpdateCoordinator
@@ -43,7 +43,7 @@ BUTTONS: tuple[OpenGarageButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenGarage button entities."""
coordinator: OpenGarageDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/opengarage/coordinator.py b/homeassistant/components/opengarage/coordinator.py
index d35dc22d288..5d5440d6b1b 100644
--- a/homeassistant/components/opengarage/coordinator.py
+++ b/homeassistant/components/opengarage/coordinator.py
@@ -8,6 +8,7 @@ from typing import Any
import opengarage
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import update_coordinator
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -20,10 +21,12 @@ _LOGGER = logging.getLogger(__name__)
class OpenGarageDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching Opengarage data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- *,
+ config_entry: ConfigEntry,
open_garage_connection: opengarage.OpenGarage,
) -> None:
"""Initialize global Opengarage data updater."""
@@ -32,6 +35,7 @@ class OpenGarageDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=5),
)
diff --git a/homeassistant/components/opengarage/cover.py b/homeassistant/components/opengarage/cover.py
index 9623050c090..859e3382772 100644
--- a/homeassistant/components/opengarage/cover.py
+++ b/homeassistant/components/opengarage/cover.py
@@ -13,7 +13,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import OpenGarageDataUpdateCoordinator
@@ -25,7 +25,9 @@ STATES_MAP = {0: CoverState.CLOSED, 1: CoverState.OPEN}
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenGarage covers."""
async_add_entities(
diff --git a/homeassistant/components/opengarage/sensor.py b/homeassistant/components/opengarage/sensor.py
index 003e0e0fa5a..14d14dd5d23 100644
--- a/homeassistant/components/opengarage/sensor.py
+++ b/homeassistant/components/opengarage/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import OpenGarageDataUpdateCoordinator
@@ -59,7 +59,9 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenGarage sensors."""
open_garage_data_coordinator: OpenGarageDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/openhome/media_player.py b/homeassistant/components/openhome/media_player.py
index 8c903c90bbb..9f8840b8487 100644
--- a/homeassistant/components/openhome/media_player.py
+++ b/homeassistant/components/openhome/media_player.py
@@ -24,7 +24,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_PIN_INDEX, DOMAIN, SERVICE_INVOKE_PIN
@@ -40,7 +40,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Openhome config entry."""
diff --git a/homeassistant/components/openhome/update.py b/homeassistant/components/openhome/update.py
index bbe4fdac3b3..cc210866e64 100644
--- a/homeassistant/components/openhome/update.py
+++ b/homeassistant/components/openhome/update.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up update entities for Reolink component."""
diff --git a/homeassistant/components/opensky/__init__.py b/homeassistant/components/opensky/__init__.py
index c95dc1283a4..c69cade5842 100644
--- a/homeassistant/components/opensky/__init__.py
+++ b/homeassistant/components/opensky/__init__.py
@@ -32,7 +32,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except OpenSkyError as exc:
raise ConfigEntryNotReady from exc
- coordinator = OpenSkyDataUpdateCoordinator(hass, client)
+ coordinator = OpenSkyDataUpdateCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/opensky/coordinator.py b/homeassistant/components/opensky/coordinator.py
index f54e01b0006..f9aab88c904 100644
--- a/homeassistant/components/opensky/coordinator.py
+++ b/homeassistant/components/opensky/coordinator.py
@@ -36,11 +36,14 @@ class OpenSkyDataUpdateCoordinator(DataUpdateCoordinator[int]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, opensky: OpenSky) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, opensky: OpenSky
+ ) -> None:
"""Initialize the OpenSky data coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval={
True: timedelta(seconds=90),
@@ -50,11 +53,11 @@ class OpenSkyDataUpdateCoordinator(DataUpdateCoordinator[int]):
self._opensky = opensky
self._previously_tracked: set[str] | None = None
self._bounding_box = OpenSky.get_bounding_box(
- self.config_entry.data[CONF_LATITUDE],
- self.config_entry.data[CONF_LONGITUDE],
- self.config_entry.options[CONF_RADIUS],
+ config_entry.data[CONF_LATITUDE],
+ config_entry.data[CONF_LONGITUDE],
+ config_entry.options[CONF_RADIUS],
)
- self._altitude = self.config_entry.options.get(CONF_ALTITUDE, DEFAULT_ALTITUDE)
+ self._altitude = config_entry.options.get(CONF_ALTITUDE, DEFAULT_ALTITUDE)
async def _async_update_data(self) -> int:
try:
diff --git a/homeassistant/components/opensky/sensor.py b/homeassistant/components/opensky/sensor.py
index 9d317ae3e0d..0ab5b49f086 100644
--- a/homeassistant/components/opensky/sensor.py
+++ b/homeassistant/components/opensky/sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.sensor import SensorEntity, SensorStateClass
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER
@@ -16,7 +16,7 @@ from .coordinator import OpenSkyDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize the entries."""
diff --git a/homeassistant/components/opentherm_gw/__init__.py b/homeassistant/components/opentherm_gw/__init__.py
index 8c92c70ab49..87da159872d 100644
--- a/homeassistant/components/opentherm_gw/__init__.py
+++ b/homeassistant/components/opentherm_gw/__init__.py
@@ -9,8 +9,7 @@ import pyotgw.vars as gw_vars
from serial import SerialException
import voluptuous as vol
-from homeassistant.components.climate import DOMAIN as CLIMATE_DOMAIN
-from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ATTR_DATE,
ATTR_ID,
@@ -21,21 +20,12 @@ from homeassistant.const import (
CONF_ID,
CONF_NAME,
EVENT_HOMEASSISTANT_STOP,
- PRECISION_HALVES,
- PRECISION_TENTHS,
- PRECISION_WHOLE,
Platform,
)
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.exceptions import ConfigEntryNotReady
-from homeassistant.helpers import (
- config_validation as cv,
- device_registry as dr,
- entity_registry as er,
- issue_registry as ir,
-)
+from homeassistant.helpers import config_validation as cv, device_registry as dr
from homeassistant.helpers.dispatcher import async_dispatcher_send
-from homeassistant.helpers.typing import ConfigType
from .const import (
ATTR_CH_OVRD,
@@ -44,9 +34,6 @@ from .const import (
ATTR_LEVEL,
ATTR_TRANSP_ARG,
ATTR_TRANSP_CMD,
- CONF_CLIMATE,
- CONF_FLOOR_TEMP,
- CONF_PRECISION,
CONF_TEMPORARY_OVRD_MODE,
CONNECTION_TIMEOUT,
DATA_GATEWAYS,
@@ -70,29 +57,6 @@ from .const import (
_LOGGER = logging.getLogger(__name__)
-# *_SCHEMA required for deprecated import from configuration.yaml, can be removed in 2025.4.0
-CLIMATE_SCHEMA = vol.Schema(
- {
- vol.Optional(CONF_PRECISION): vol.In(
- [PRECISION_TENTHS, PRECISION_HALVES, PRECISION_WHOLE]
- ),
- vol.Optional(CONF_FLOOR_TEMP, default=False): cv.boolean,
- }
-)
-
-CONFIG_SCHEMA = vol.Schema(
- {
- DOMAIN: cv.schema_with_slug_keys(
- {
- vol.Required(CONF_DEVICE): cv.string,
- vol.Optional(CONF_CLIMATE, default={}): CLIMATE_SCHEMA,
- vol.Optional(CONF_NAME): cv.string,
- }
- )
- },
- extra=vol.ALLOW_EXTRA,
-)
-
PLATFORMS = [
Platform.BINARY_SENSOR,
Platform.BUTTON,
@@ -118,35 +82,6 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
gateway = OpenThermGatewayHub(hass, config_entry)
hass.data[DATA_OPENTHERM_GW][DATA_GATEWAYS][config_entry.data[CONF_ID]] = gateway
- # Migration can be removed in 2025.4.0
- dev_reg = dr.async_get(hass)
- if (
- migrate_device := dev_reg.async_get_device(
- {(DOMAIN, config_entry.data[CONF_ID])}
- )
- ) is not None:
- dev_reg.async_update_device(
- migrate_device.id,
- new_identifiers={
- (
- DOMAIN,
- f"{config_entry.data[CONF_ID]}-{OpenThermDeviceIdentifier.GATEWAY}",
- )
- },
- )
-
- # Migration can be removed in 2025.4.0
- ent_reg = er.async_get(hass)
- if (
- entity_id := ent_reg.async_get_entity_id(
- CLIMATE_DOMAIN, DOMAIN, config_entry.data[CONF_ID]
- )
- ) is not None:
- ent_reg.async_update_entity(
- entity_id,
- new_unique_id=f"{config_entry.data[CONF_ID]}-{OpenThermDeviceIdentifier.THERMOSTAT}-thermostat_entity",
- )
-
config_entry.add_update_listener(options_updated)
try:
@@ -164,33 +99,6 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
return True
-# Deprecated import from configuration.yaml, can be removed in 2025.4.0
-async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
- """Set up the OpenTherm Gateway component."""
- if DOMAIN in config:
- ir.async_create_issue(
- hass,
- DOMAIN,
- "deprecated_import_from_configuration_yaml",
- breaks_in_ha_version="2025.4.0",
- is_fixable=False,
- is_persistent=False,
- severity=ir.IssueSeverity.WARNING,
- translation_key="deprecated_import_from_configuration_yaml",
- )
- if not hass.config_entries.async_entries(DOMAIN) and DOMAIN in config:
- conf = config[DOMAIN]
- for device_id, device_config in conf.items():
- device_config[CONF_ID] = device_id
-
- hass.async_create_task(
- hass.config_entries.flow.async_init(
- DOMAIN, context={"source": SOURCE_IMPORT}, data=device_config
- )
- )
- return True
-
-
def register_services(hass: HomeAssistant) -> None:
"""Register services for the component."""
service_reset_schema = vol.Schema(
diff --git a/homeassistant/components/opentherm_gw/binary_sensor.py b/homeassistant/components/opentherm_gw/binary_sensor.py
index 5d542bedc07..8e73392da05 100644
--- a/homeassistant/components/opentherm_gw/binary_sensor.py
+++ b/homeassistant/components/opentherm_gw/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
BOILER_DEVICE_DESCRIPTION,
@@ -393,7 +393,7 @@ BINARY_SENSOR_DESCRIPTIONS: tuple[OpenThermBinarySensorEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenTherm Gateway binary sensors."""
gw_hub = hass.data[DATA_OPENTHERM_GW][DATA_GATEWAYS][config_entry.data[CONF_ID]]
diff --git a/homeassistant/components/opentherm_gw/button.py b/homeassistant/components/opentherm_gw/button.py
index 00b91ad33e0..046b44bfa8c 100644
--- a/homeassistant/components/opentherm_gw/button.py
+++ b/homeassistant/components/opentherm_gw/button.py
@@ -13,7 +13,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OpenThermGatewayHub
from .const import (
@@ -53,7 +53,7 @@ BUTTON_DESCRIPTIONS: tuple[OpenThermButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenTherm Gateway buttons."""
gw_hub = hass.data[DATA_OPENTHERM_GW][DATA_GATEWAYS][config_entry.data[CONF_ID]]
diff --git a/homeassistant/components/opentherm_gw/climate.py b/homeassistant/components/opentherm_gw/climate.py
index e8aa99f7325..c69151c293a 100644
--- a/homeassistant/components/opentherm_gw/climate.py
+++ b/homeassistant/components/opentherm_gw/climate.py
@@ -22,7 +22,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, CONF_ID, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OpenThermGatewayHub
from .const import (
@@ -50,7 +50,7 @@ class OpenThermClimateEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an OpenTherm Gateway climate entity."""
ents = []
diff --git a/homeassistant/components/opentherm_gw/config_flow.py b/homeassistant/components/opentherm_gw/config_flow.py
index bcbf279f3f7..a100dcb730f 100644
--- a/homeassistant/components/opentherm_gw/config_flow.py
+++ b/homeassistant/components/opentherm_gw/config_flow.py
@@ -95,19 +95,6 @@ class OpenThermGwConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle manual initiation of the config flow."""
return await self.async_step_init(user_input)
- # Deprecated import from configuration.yaml, can be removed in 2025.4.0
- async def async_step_import(self, import_data: dict[str, Any]) -> ConfigFlowResult:
- """Import an OpenTherm Gateway device as a config entry.
-
- This flow is triggered by `async_setup` for configured devices.
- """
- formatted_config = {
- CONF_NAME: import_data.get(CONF_NAME, import_data[CONF_ID]),
- CONF_DEVICE: import_data[CONF_DEVICE],
- CONF_ID: import_data[CONF_ID],
- }
- return await self.async_step_init(info=formatted_config)
-
def _show_form(self, errors: dict[str, str] | None = None) -> ConfigFlowResult:
"""Show the config flow form with possible errors."""
return self.async_show_form(
diff --git a/homeassistant/components/opentherm_gw/select.py b/homeassistant/components/opentherm_gw/select.py
index cee1632dc48..da3fa1e80ec 100644
--- a/homeassistant/components/opentherm_gw/select.py
+++ b/homeassistant/components/opentherm_gw/select.py
@@ -20,7 +20,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OpenThermGatewayHub
from .const import (
@@ -234,7 +234,7 @@ SELECT_DESCRIPTIONS: tuple[OpenThermSelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenTherm Gateway select entities."""
gw_hub = hass.data[DATA_OPENTHERM_GW][DATA_GATEWAYS][config_entry.data[CONF_ID]]
diff --git a/homeassistant/components/opentherm_gw/sensor.py b/homeassistant/components/opentherm_gw/sensor.py
index 5ccb4166665..f9ac1b272be 100644
--- a/homeassistant/components/opentherm_gw/sensor.py
+++ b/homeassistant/components/opentherm_gw/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
BOILER_DEVICE_DESCRIPTION,
@@ -875,7 +875,7 @@ SENSOR_DESCRIPTIONS: tuple[OpenThermSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenTherm Gateway sensors."""
gw_hub = hass.data[DATA_OPENTHERM_GW][DATA_GATEWAYS][config_entry.data[CONF_ID]]
diff --git a/homeassistant/components/opentherm_gw/strings.json b/homeassistant/components/opentherm_gw/strings.json
index 405af126c03..cc57a7d9e0c 100644
--- a/homeassistant/components/opentherm_gw/strings.json
+++ b/homeassistant/components/opentherm_gw/strings.json
@@ -354,12 +354,6 @@
}
}
},
- "issues": {
- "deprecated_import_from_configuration_yaml": {
- "title": "Deprecated configuration",
- "description": "Configuration of the OpenTherm Gateway integration through configuration.yaml is deprecated. Your configuration has been migrated to config entries. Please remove any OpenTherm Gateway configuration from your configuration.yaml."
- }
- },
"options": {
"step": {
"init": {
@@ -385,7 +379,7 @@
},
"set_central_heating_ovrd": {
"name": "Set central heating override",
- "description": "Sets the central heating override option on the gateway. When overriding the control setpoint (via a set_control_setpoint action with a value other than 0), the gateway automatically enables the central heating override to start heating. This action can then be used to control the central heating override status. To return control of the central heating to the thermostat, use the set_control_setpoint action with temperature value 0. You will only need this if you are writing your own software thermostat.",
+ "description": "Sets the central heating override option on the gateway. When overriding the control setpoint (via a 'Set control set point' action with a value other than 0), the gateway automatically enables the central heating override to start heating. This action can then be used to control the central heating override status. To return control of the central heating to the thermostat, use the 'Set control set point' action with temperature value 0. You will only need this if you are writing your own software thermostat.",
"fields": {
"gateway_id": {
"name": "[%key:component::opentherm_gw::services::reset_gateway::fields::gateway_id::name%]",
@@ -393,7 +387,7 @@
},
"ch_override": {
"name": "Central heating override",
- "description": "The desired boolean value for the central heating override."
+ "description": "Whether to enable or disable the override."
}
}
},
diff --git a/homeassistant/components/opentherm_gw/switch.py b/homeassistant/components/opentherm_gw/switch.py
index 41ffa03a932..873675f0211 100644
--- a/homeassistant/components/opentherm_gw/switch.py
+++ b/homeassistant/components/opentherm_gw/switch.py
@@ -8,7 +8,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OpenThermGatewayHub
from .const import DATA_GATEWAYS, DATA_OPENTHERM_GW, GATEWAY_DEVICE_DESCRIPTION
@@ -48,7 +48,7 @@ SWITCH_DESCRIPTIONS: tuple[OpenThermSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OpenTherm Gateway switches."""
gw_hub = hass.data[DATA_OPENTHERM_GW][DATA_GATEWAYS][config_entry.data[CONF_ID]]
diff --git a/homeassistant/components/openuv/binary_sensor.py b/homeassistant/components/openuv/binary_sensor.py
index 018d91710df..f45404ce38e 100644
--- a/homeassistant/components/openuv/binary_sensor.py
+++ b/homeassistant/components/openuv/binary_sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import as_local, parse_datetime, utcnow
from .const import DATA_PROTECTION_WINDOW, DOMAIN, LOGGER, TYPE_PROTECTION_WINDOW
@@ -25,7 +25,9 @@ BINARY_SENSOR_DESCRIPTION_PROTECTION_WINDOW = BinarySensorEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
# Once we've successfully authenticated, we re-enable client request retries:
"""Set up an OpenUV sensor based on a config entry."""
diff --git a/homeassistant/components/openuv/coordinator.py b/homeassistant/components/openuv/coordinator.py
index 32d502cb8ce..cc09161b3e9 100644
--- a/homeassistant/components/openuv/coordinator.py
+++ b/homeassistant/components/openuv/coordinator.py
@@ -38,6 +38,7 @@ class OpenUvCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
LOGGER,
+ config_entry=entry,
name=name,
update_method=update_method,
request_refresh_debouncer=Debouncer(
@@ -48,7 +49,6 @@ class OpenUvCoordinator(DataUpdateCoordinator[dict[str, Any]]):
),
)
- self._entry = entry
self.latitude = latitude
self.longitude = longitude
diff --git a/homeassistant/components/openuv/sensor.py b/homeassistant/components/openuv/sensor.py
index 742017be639..5b681655e2b 100644
--- a/homeassistant/components/openuv/sensor.py
+++ b/homeassistant/components/openuv/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UV_INDEX, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import as_local, parse_datetime
from .const import (
@@ -166,7 +166,9 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a OpenUV sensor based on a config entry."""
coordinators: dict[str, OpenUvCoordinator] = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/openweathermap/__init__.py b/homeassistant/components/openweathermap/__init__.py
index 33cd23c4f6c..40ddf0ff37e 100644
--- a/homeassistant/components/openweathermap/__init__.py
+++ b/homeassistant/components/openweathermap/__init__.py
@@ -8,17 +8,10 @@ import logging
from pyopenweathermap import create_owm_client
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import (
- CONF_API_KEY,
- CONF_LANGUAGE,
- CONF_LATITUDE,
- CONF_LONGITUDE,
- CONF_MODE,
- CONF_NAME,
-)
+from homeassistant.const import CONF_API_KEY, CONF_LANGUAGE, CONF_MODE, CONF_NAME
from homeassistant.core import HomeAssistant
-from .const import CONFIG_FLOW_VERSION, OWM_MODE_V25, PLATFORMS
+from .const import CONFIG_FLOW_VERSION, DEFAULT_OWM_MODE, OWM_MODES, PLATFORMS
from .coordinator import WeatherUpdateCoordinator
from .repairs import async_create_issue, async_delete_issue
from .utils import build_data_and_options
@@ -43,20 +36,16 @@ async def async_setup_entry(
"""Set up OpenWeatherMap as config entry."""
name = entry.data[CONF_NAME]
api_key = entry.data[CONF_API_KEY]
- latitude = entry.data.get(CONF_LATITUDE, hass.config.latitude)
- longitude = entry.data.get(CONF_LONGITUDE, hass.config.longitude)
language = entry.options[CONF_LANGUAGE]
mode = entry.options[CONF_MODE]
- if mode == OWM_MODE_V25:
+ if mode not in OWM_MODES:
async_create_issue(hass, entry.entry_id)
else:
async_delete_issue(hass, entry.entry_id)
owm_client = create_owm_client(api_key, mode, lang=language)
- weather_coordinator = WeatherUpdateCoordinator(
- owm_client, latitude, longitude, hass
- )
+ weather_coordinator = WeatherUpdateCoordinator(hass, entry, owm_client)
await weather_coordinator.async_config_entry_first_refresh()
@@ -69,7 +58,9 @@ async def async_setup_entry(
return True
-async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, entry: OpenweathermapConfigEntry
+) -> bool:
"""Migrate old entry."""
config_entries = hass.config_entries
data = entry.data
@@ -79,7 +70,7 @@ async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
_LOGGER.debug("Migrating OpenWeatherMap entry from version %s", version)
if version < 5:
- combined_data = {**data, **options, CONF_MODE: OWM_MODE_V25}
+ combined_data = {**data, **options, CONF_MODE: DEFAULT_OWM_MODE}
new_data, new_options = build_data_and_options(combined_data)
config_entries.async_update_entry(
entry,
@@ -93,7 +84,9 @@ async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True
-async def async_update_options(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def async_update_options(
+ hass: HomeAssistant, entry: OpenweathermapConfigEntry
+) -> None:
"""Update options."""
await hass.config_entries.async_reload(entry.entry_id)
diff --git a/homeassistant/components/openweathermap/const.py b/homeassistant/components/openweathermap/const.py
index 81a6544c7ce..fbd2cb1aee2 100644
--- a/homeassistant/components/openweathermap/const.py
+++ b/homeassistant/components/openweathermap/const.py
@@ -48,6 +48,7 @@ ATTR_API_WEATHER_CODE = "weather_code"
ATTR_API_CLOUD_COVERAGE = "cloud_coverage"
ATTR_API_FORECAST = "forecast"
ATTR_API_CURRENT = "current"
+ATTR_API_MINUTE_FORECAST = "minute_forecast"
ATTR_API_HOURLY_FORECAST = "hourly_forecast"
ATTR_API_DAILY_FORECAST = "daily_forecast"
UPDATE_LISTENER = "update_listener"
@@ -61,10 +62,8 @@ FORECAST_MODE_ONECALL_DAILY = "onecall_daily"
OWM_MODE_FREE_CURRENT = "current"
OWM_MODE_FREE_FORECAST = "forecast"
OWM_MODE_V30 = "v3.0"
-OWM_MODE_V25 = "v2.5"
OWM_MODES = [
OWM_MODE_V30,
- OWM_MODE_V25,
OWM_MODE_FREE_CURRENT,
OWM_MODE_FREE_FORECAST,
]
diff --git a/homeassistant/components/openweathermap/coordinator.py b/homeassistant/components/openweathermap/coordinator.py
index 3ef0eda0c8f..994949b5e03 100644
--- a/homeassistant/components/openweathermap/coordinator.py
+++ b/homeassistant/components/openweathermap/coordinator.py
@@ -1,12 +1,16 @@
"""Weather data coordinator for the OpenWeatherMap (OWM) service."""
+from __future__ import annotations
+
from datetime import timedelta
import logging
+from typing import TYPE_CHECKING
from pyopenweathermap import (
CurrentWeather,
DailyWeatherForecast,
HourlyWeatherForecast,
+ MinutelyWeatherForecast,
OWMClient,
RequestError,
WeatherReport,
@@ -17,20 +21,28 @@ from homeassistant.components.weather import (
ATTR_CONDITION_SUNNY,
Forecast,
)
+from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import sun
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt as dt_util
+if TYPE_CHECKING:
+ from . import OpenweathermapConfigEntry
+
from .const import (
ATTR_API_CLOUDS,
ATTR_API_CONDITION,
ATTR_API_CURRENT,
ATTR_API_DAILY_FORECAST,
+ ATTR_API_DATETIME,
ATTR_API_DEW_POINT,
ATTR_API_FEELS_LIKE_TEMPERATURE,
+ ATTR_API_FORECAST,
ATTR_API_HOURLY_FORECAST,
ATTR_API_HUMIDITY,
+ ATTR_API_MINUTE_FORECAST,
+ ATTR_API_PRECIPITATION,
ATTR_API_PRECIPITATION_KIND,
ATTR_API_PRESSURE,
ATTR_API_RAIN,
@@ -56,20 +68,25 @@ WEATHER_UPDATE_INTERVAL = timedelta(minutes=10)
class WeatherUpdateCoordinator(DataUpdateCoordinator):
"""Weather data update coordinator."""
+ config_entry: OpenweathermapConfigEntry
+
def __init__(
self,
- owm_client: OWMClient,
- latitude,
- longitude,
hass: HomeAssistant,
+ config_entry: OpenweathermapConfigEntry,
+ owm_client: OWMClient,
) -> None:
"""Initialize coordinator."""
self._owm_client = owm_client
- self._latitude = latitude
- self._longitude = longitude
+ self._latitude = config_entry.data.get(CONF_LATITUDE, hass.config.latitude)
+ self._longitude = config_entry.data.get(CONF_LONGITUDE, hass.config.longitude)
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=WEATHER_UPDATE_INTERVAL
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=WEATHER_UPDATE_INTERVAL,
)
async def _async_update_data(self):
@@ -94,6 +111,11 @@ class WeatherUpdateCoordinator(DataUpdateCoordinator):
return {
ATTR_API_CURRENT: current_weather,
+ ATTR_API_MINUTE_FORECAST: (
+ self._get_minute_weather_data(weather_report.minutely_forecast)
+ if weather_report.minutely_forecast is not None
+ else {}
+ ),
ATTR_API_HOURLY_FORECAST: [
self._get_hourly_forecast_weather_data(item)
for item in weather_report.hourly_forecast
@@ -104,6 +126,20 @@ class WeatherUpdateCoordinator(DataUpdateCoordinator):
],
}
+ def _get_minute_weather_data(
+ self, minute_forecast: list[MinutelyWeatherForecast]
+ ) -> dict:
+ """Get minute weather data from the forecast."""
+ return {
+ ATTR_API_FORECAST: [
+ {
+ ATTR_API_DATETIME: item.date_time,
+ ATTR_API_PRECIPITATION: round(item.precipitation, 2),
+ }
+ for item in minute_forecast
+ ]
+ }
+
def _get_current_weather_data(self, current_weather: CurrentWeather):
return {
ATTR_API_CONDITION: self._get_condition(current_weather.condition.id),
diff --git a/homeassistant/components/openweathermap/icons.json b/homeassistant/components/openweathermap/icons.json
new file mode 100644
index 00000000000..d493b1538ba
--- /dev/null
+++ b/homeassistant/components/openweathermap/icons.json
@@ -0,0 +1,7 @@
+{
+ "services": {
+ "get_minute_forecast": {
+ "service": "mdi:weather-snowy-rainy"
+ }
+ }
+}
diff --git a/homeassistant/components/openweathermap/manifest.json b/homeassistant/components/openweathermap/manifest.json
index 14313a5a77e..88510aaae8c 100644
--- a/homeassistant/components/openweathermap/manifest.json
+++ b/homeassistant/components/openweathermap/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/openweathermap",
"iot_class": "cloud_polling",
"loggers": ["pyopenweathermap"],
- "requirements": ["pyopenweathermap==0.2.1"]
+ "requirements": ["pyopenweathermap==0.2.2"]
}
diff --git a/homeassistant/components/openweathermap/repairs.py b/homeassistant/components/openweathermap/repairs.py
index c54484e1e1e..2bde5750ca4 100644
--- a/homeassistant/components/openweathermap/repairs.py
+++ b/homeassistant/components/openweathermap/repairs.py
@@ -1,14 +1,18 @@
"""Issues for OpenWeatherMap."""
-from typing import cast
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
from homeassistant import data_entry_flow
from homeassistant.components.repairs import RepairsFlow
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY, CONF_MODE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import issue_registry as ir
+if TYPE_CHECKING:
+ from . import OpenweathermapConfigEntry
+
from .const import DOMAIN, OWM_MODE_V30
from .utils import validate_api_key
@@ -16,7 +20,7 @@ from .utils import validate_api_key
class DeprecatedV25RepairFlow(RepairsFlow):
"""Handler for an issue fixing flow."""
- def __init__(self, entry: ConfigEntry) -> None:
+ def __init__(self, entry: OpenweathermapConfigEntry) -> None:
"""Create flow."""
super().__init__()
self.entry = entry
diff --git a/homeassistant/components/openweathermap/sensor.py b/homeassistant/components/openweathermap/sensor.py
index 46789f4b3d2..0afab69b638 100644
--- a/homeassistant/components/openweathermap/sensor.py
+++ b/homeassistant/components/openweathermap/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -156,7 +156,7 @@ WEATHER_SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OpenweathermapConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up OpenWeatherMap sensor entities based on a config entry."""
domain_data = config_entry.runtime_data
diff --git a/homeassistant/components/openweathermap/services.yaml b/homeassistant/components/openweathermap/services.yaml
new file mode 100644
index 00000000000..6bbcf1b23e4
--- /dev/null
+++ b/homeassistant/components/openweathermap/services.yaml
@@ -0,0 +1,5 @@
+get_minute_forecast:
+ target:
+ entity:
+ domain: weather
+ integration: openweathermap
diff --git a/homeassistant/components/openweathermap/strings.json b/homeassistant/components/openweathermap/strings.json
index 46b5feab75c..1aa161c87dc 100644
--- a/homeassistant/components/openweathermap/strings.json
+++ b/homeassistant/components/openweathermap/strings.json
@@ -47,5 +47,16 @@
}
}
}
+ },
+ "services": {
+ "get_minute_forecast": {
+ "name": "Get minute forecast",
+ "description": "Retrieves a minute-by-minute weather forecast for one hour."
+ }
+ },
+ "exceptions": {
+ "service_minute_forecast_mode": {
+ "message": "Minute forecast is available only when {name} mode is set to v3.0"
+ }
}
}
diff --git a/homeassistant/components/openweathermap/weather.py b/homeassistant/components/openweathermap/weather.py
index 3a134a0ee26..12d883c871a 100644
--- a/homeassistant/components/openweathermap/weather.py
+++ b/homeassistant/components/openweathermap/weather.py
@@ -14,9 +14,11 @@ from homeassistant.const import (
UnitOfSpeed,
UnitOfTemperature,
)
-from homeassistant.core import HomeAssistant, callback
+from homeassistant.core import HomeAssistant, SupportsResponse, callback
+from homeassistant.exceptions import ServiceValidationError
+from homeassistant.helpers import entity_platform
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OpenweathermapConfigEntry
from .const import (
@@ -28,6 +30,7 @@ from .const import (
ATTR_API_FEELS_LIKE_TEMPERATURE,
ATTR_API_HOURLY_FORECAST,
ATTR_API_HUMIDITY,
+ ATTR_API_MINUTE_FORECAST,
ATTR_API_PRESSURE,
ATTR_API_TEMPERATURE,
ATTR_API_VISIBILITY_DISTANCE,
@@ -39,16 +42,17 @@ from .const import (
DOMAIN,
MANUFACTURER,
OWM_MODE_FREE_FORECAST,
- OWM_MODE_V25,
OWM_MODE_V30,
)
from .coordinator import WeatherUpdateCoordinator
+SERVICE_GET_MINUTE_FORECAST = "get_minute_forecast"
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OpenweathermapConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up OpenWeatherMap weather entity based on a config entry."""
domain_data = config_entry.runtime_data
@@ -61,6 +65,14 @@ async def async_setup_entry(
async_add_entities([owm_weather], False)
+ platform = entity_platform.async_get_current_platform()
+ platform.async_register_entity_service(
+ name=SERVICE_GET_MINUTE_FORECAST,
+ schema=None,
+ func="async_get_minute_forecast",
+ supports_response=SupportsResponse.ONLY,
+ )
+
class OpenWeatherMapWeather(SingleCoordinatorWeatherEntity[WeatherUpdateCoordinator]):
"""Implementation of an OpenWeatherMap sensor."""
@@ -91,8 +103,9 @@ class OpenWeatherMapWeather(SingleCoordinatorWeatherEntity[WeatherUpdateCoordina
manufacturer=MANUFACTURER,
name=DEFAULT_NAME,
)
+ self.mode = mode
- if mode in (OWM_MODE_V30, OWM_MODE_V25):
+ if mode == OWM_MODE_V30:
self._attr_supported_features = (
WeatherEntityFeature.FORECAST_DAILY
| WeatherEntityFeature.FORECAST_HOURLY
@@ -100,6 +113,17 @@ class OpenWeatherMapWeather(SingleCoordinatorWeatherEntity[WeatherUpdateCoordina
elif mode == OWM_MODE_FREE_FORECAST:
self._attr_supported_features = WeatherEntityFeature.FORECAST_HOURLY
+ async def async_get_minute_forecast(self) -> dict[str, list[dict]] | dict:
+ """Return Minute forecast."""
+
+ if self.mode == OWM_MODE_V30:
+ return self.coordinator.data[ATTR_API_MINUTE_FORECAST]
+ raise ServiceValidationError(
+ translation_domain=DOMAIN,
+ translation_key="service_minute_forecast_mode",
+ translation_placeholders={"name": DEFAULT_NAME},
+ )
+
@property
def condition(self) -> str | None:
"""Return the current condition."""
diff --git a/homeassistant/components/opower/__init__.py b/homeassistant/components/opower/__init__.py
index 136a1a4e57a..23c8e7a8136 100644
--- a/homeassistant/components/opower/__init__.py
+++ b/homeassistant/components/opower/__init__.py
@@ -2,22 +2,18 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .coordinator import OpowerCoordinator
+from .coordinator import OpowerConfigEntry, OpowerCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type OpowerConfigEntry = ConfigEntry[OpowerCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: OpowerConfigEntry) -> bool:
"""Set up Opower from a config entry."""
- coordinator = OpowerCoordinator(hass, entry.data)
+ coordinator = OpowerCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/opower/coordinator.py b/homeassistant/components/opower/coordinator.py
index 6957ae4984c..aed89ccf46e 100644
--- a/homeassistant/components/opower/coordinator.py
+++ b/homeassistant/components/opower/coordinator.py
@@ -2,8 +2,7 @@
from datetime import datetime, timedelta
import logging
-from types import MappingProxyType
-from typing import Any, cast
+from typing import cast
from opower import (
Account,
@@ -23,6 +22,7 @@ from homeassistant.components.recorder.statistics import (
get_last_statistics,
statistics_during_period,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, UnitOfEnergy, UnitOfVolume
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryAuthFailed
@@ -34,19 +34,24 @@ from .const import CONF_TOTP_SECRET, CONF_UTILITY, DOMAIN
_LOGGER = logging.getLogger(__name__)
+type OpowerConfigEntry = ConfigEntry[OpowerCoordinator]
+
class OpowerCoordinator(DataUpdateCoordinator[dict[str, Forecast]]):
"""Handle fetching Opower data, updating sensors and inserting statistics."""
+ config_entry: OpowerConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- entry_data: MappingProxyType[str, Any],
+ config_entry: OpowerConfigEntry,
) -> None:
"""Initialize the data handler."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Opower",
# Data is updated daily on Opower.
# Refresh every 12h to be at most 12h behind.
@@ -54,10 +59,10 @@ class OpowerCoordinator(DataUpdateCoordinator[dict[str, Forecast]]):
)
self.api = Opower(
aiohttp_client.async_get_clientsession(hass),
- entry_data[CONF_UTILITY],
- entry_data[CONF_USERNAME],
- entry_data[CONF_PASSWORD],
- entry_data.get(CONF_TOTP_SECRET),
+ config_entry.data[CONF_UTILITY],
+ config_entry.data[CONF_USERNAME],
+ config_entry.data[CONF_PASSWORD],
+ config_entry.data.get(CONF_TOTP_SECRET),
)
@callback
diff --git a/homeassistant/components/opower/manifest.json b/homeassistant/components/opower/manifest.json
index d168cba5752..2da4511c0aa 100644
--- a/homeassistant/components/opower/manifest.json
+++ b/homeassistant/components/opower/manifest.json
@@ -7,5 +7,5 @@
"documentation": "https://www.home-assistant.io/integrations/opower",
"iot_class": "cloud_polling",
"loggers": ["opower"],
- "requirements": ["opower==0.8.9"]
+ "requirements": ["opower==0.9.0"]
}
diff --git a/homeassistant/components/opower/sensor.py b/homeassistant/components/opower/sensor.py
index f9d0fe62332..46aa9e9b318 100644
--- a/homeassistant/components/opower/sensor.py
+++ b/homeassistant/components/opower/sensor.py
@@ -4,6 +4,7 @@ from __future__ import annotations
from collections.abc import Callable
from dataclasses import dataclass
+from datetime import date
from opower import Forecast, MeterType, UnitOfMeasure
@@ -16,20 +17,19 @@ from homeassistant.components.sensor import (
from homeassistant.const import EntityCategory, UnitOfEnergy, UnitOfVolume
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import OpowerConfigEntry
from .const import DOMAIN
-from .coordinator import OpowerCoordinator
+from .coordinator import OpowerConfigEntry, OpowerCoordinator
@dataclass(frozen=True, kw_only=True)
class OpowerEntityDescription(SensorEntityDescription):
"""Class describing Opower sensors entities."""
- value_fn: Callable[[Forecast], str | float]
+ value_fn: Callable[[Forecast], str | float | date]
# suggested_display_precision=0 for all sensors since
@@ -97,7 +97,7 @@ ELEC_SENSORS: tuple[OpowerEntityDescription, ...] = (
device_class=SensorDeviceClass.DATE,
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
- value_fn=lambda data: str(data.start_date),
+ value_fn=lambda data: data.start_date,
),
OpowerEntityDescription(
key="elec_end_date",
@@ -105,7 +105,7 @@ ELEC_SENSORS: tuple[OpowerEntityDescription, ...] = (
device_class=SensorDeviceClass.DATE,
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
- value_fn=lambda data: str(data.end_date),
+ value_fn=lambda data: data.end_date,
),
)
GAS_SENSORS: tuple[OpowerEntityDescription, ...] = (
@@ -169,7 +169,7 @@ GAS_SENSORS: tuple[OpowerEntityDescription, ...] = (
device_class=SensorDeviceClass.DATE,
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
- value_fn=lambda data: str(data.start_date),
+ value_fn=lambda data: data.start_date,
),
OpowerEntityDescription(
key="gas_end_date",
@@ -177,7 +177,7 @@ GAS_SENSORS: tuple[OpowerEntityDescription, ...] = (
device_class=SensorDeviceClass.DATE,
entity_category=EntityCategory.DIAGNOSTIC,
entity_registry_enabled_default=False,
- value_fn=lambda data: str(data.end_date),
+ value_fn=lambda data: data.end_date,
),
)
@@ -185,7 +185,7 @@ GAS_SENSORS: tuple[OpowerEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: OpowerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Opower sensor."""
@@ -247,7 +247,7 @@ class OpowerSensor(CoordinatorEntity[OpowerCoordinator], SensorEntity):
self.utility_account_id = utility_account_id
@property
- def native_value(self) -> StateType:
+ def native_value(self) -> StateType | date:
"""Return the state."""
if self.coordinator.data is not None:
return self.entity_description.value_fn(
diff --git a/homeassistant/components/oralb/sensor.py b/homeassistant/components/oralb/sensor.py
index 9994bfc6443..3b345f4b36a 100644
--- a/homeassistant/components/oralb/sensor.py
+++ b/homeassistant/components/oralb/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from . import OralBConfigEntry
@@ -108,7 +108,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: OralBConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OralB BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/osoenergy/binary_sensor.py b/homeassistant/components/osoenergy/binary_sensor.py
index 0cf0ac74d36..a2ba61ccbe4 100644
--- a/homeassistant/components/osoenergy/binary_sensor.py
+++ b/homeassistant/components/osoenergy/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import OSOEnergyEntity
@@ -45,7 +45,9 @@ SENSOR_TYPES: dict[str, OSOEnergyBinarySensorEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up OSO Energy binary sensor."""
osoenergy: OSOEnergy = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/osoenergy/sensor.py b/homeassistant/components/osoenergy/sensor.py
index 40ec33e3e02..18859627952 100644
--- a/homeassistant/components/osoenergy/sensor.py
+++ b/homeassistant/components/osoenergy/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN
@@ -138,7 +138,9 @@ SENSOR_TYPES: dict[str, OSOEnergySensorEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up OSO Energy sensor."""
osoenergy = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/osoenergy/strings.json b/homeassistant/components/osoenergy/strings.json
index ca23265048f..7e10168d941 100644
--- a/homeassistant/components/osoenergy/strings.json
+++ b/homeassistant/components/osoenergy/strings.json
@@ -215,7 +215,7 @@
"fields": {
"until_temp_limit": {
"name": "Until temperature limit",
- "description": "Choose if heating should be off until min temperature (True) is reached or for one hour (False)"
+ "description": "Whether heating should be off until the minimum temperature is reached instead of for one hour."
}
}
},
@@ -225,7 +225,7 @@
"fields": {
"until_temp_limit": {
"name": "Until temperature limit",
- "description": "Choose if heating should be on until max temperature (True) is reached or for one hour (False)"
+ "description": "Whether heating should be on until the maximum temperature is reached instead of for one hour."
}
}
}
diff --git a/homeassistant/components/osoenergy/water_heater.py b/homeassistant/components/osoenergy/water_heater.py
index b3281193da3..07820ee97d5 100644
--- a/homeassistant/components/osoenergy/water_heater.py
+++ b/homeassistant/components/osoenergy/water_heater.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant, ServiceResponse, SupportsResponse
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from homeassistant.util.json import JsonValueType
@@ -49,7 +49,9 @@ SERVICE_TURN_ON = "turn_on"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up OSO Energy heater based on a config entry."""
osoenergy = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/otbr/__init__.py b/homeassistant/components/otbr/__init__.py
index 4b95be1d40d..0756f32ab18 100644
--- a/homeassistant/components/otbr/__init__.py
+++ b/homeassistant/components/otbr/__init__.py
@@ -7,16 +7,20 @@ import logging
import aiohttp
import python_otbr_api
+from homeassistant.components.homeassistant_hardware.helpers import (
+ async_notify_firmware_info,
+ async_register_firmware_info_provider,
+)
from homeassistant.components.thread import async_add_dataset
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
from homeassistant.helpers import config_validation as cv, issue_registry as ir
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.typing import ConfigType
-from . import websocket_api
+from . import homeassistant_hardware, websocket_api
from .const import DOMAIN
+from .types import OTBRConfigEntry
from .util import (
GetBorderAgentIdNotSupported,
OTBRData,
@@ -28,12 +32,13 @@ _LOGGER = logging.getLogger(__name__)
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
-type OTBRConfigEntry = ConfigEntry[OTBRData]
-
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the Open Thread Border Router component."""
websocket_api.async_setup(hass)
+
+ async_register_firmware_info_provider(hass, DOMAIN, homeassistant_hardware)
+
return True
@@ -77,6 +82,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: OTBRConfigEntry) -> bool
entry.async_on_unload(entry.add_update_listener(async_reload_entry))
entry.runtime_data = otbrdata
+ if fw_info := await homeassistant_hardware.async_get_firmware_info(hass, entry):
+ await async_notify_firmware_info(hass, DOMAIN, fw_info)
+
return True
diff --git a/homeassistant/components/otbr/config_flow.py b/homeassistant/components/otbr/config_flow.py
index aff79ca4651..514f6c7617c 100644
--- a/homeassistant/components/otbr/config_flow.py
+++ b/homeassistant/components/otbr/config_flow.py
@@ -16,7 +16,12 @@ import yarl
from homeassistant.components.hassio import AddonError, AddonManager
from homeassistant.components.homeassistant_yellow import hardware as yellow_hardware
from homeassistant.components.thread import async_get_preferred_dataset
-from homeassistant.config_entries import SOURCE_HASSIO, ConfigFlow, ConfigFlowResult
+from homeassistant.config_entries import (
+ SOURCE_HASSIO,
+ ConfigEntryState,
+ ConfigFlow,
+ ConfigFlowResult,
+)
from homeassistant.const import CONF_URL
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
@@ -201,12 +206,23 @@ class OTBRConfigFlow(ConfigFlow, domain=DOMAIN):
# we have to assume it's the first version
# This check can be removed in HA Core 2025.9
unique_id = discovery_info.uuid
+
+ if unique_id != discovery_info.uuid:
+ continue
+
if (
- unique_id != discovery_info.uuid
- or current_url.host != config["host"]
+ current_url.host != config["host"]
or current_url.port == config["port"]
):
+ # Reload the entry since OTBR has restarted
+ if current_entry.state == ConfigEntryState.LOADED:
+ assert current_entry.unique_id is not None
+ await self.hass.config_entries.async_reload(
+ current_entry.entry_id
+ )
+
continue
+
# Update URL with the new port
self.hass.config_entries.async_update_entry(
current_entry,
diff --git a/homeassistant/components/otbr/homeassistant_hardware.py b/homeassistant/components/otbr/homeassistant_hardware.py
new file mode 100644
index 00000000000..94193be1359
--- /dev/null
+++ b/homeassistant/components/otbr/homeassistant_hardware.py
@@ -0,0 +1,76 @@
+"""Home Assistant Hardware firmware utilities."""
+
+from __future__ import annotations
+
+import logging
+
+from yarl import URL
+
+from homeassistant.components.hassio import AddonManager
+from homeassistant.components.homeassistant_hardware.util import (
+ ApplicationType,
+ FirmwareInfo,
+ OwningAddon,
+ OwningIntegration,
+ get_otbr_addon_firmware_info,
+)
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.hassio import is_hassio
+
+from .const import DOMAIN
+from .types import OTBRConfigEntry
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_get_firmware_info(
+ hass: HomeAssistant, config_entry: OTBRConfigEntry
+) -> FirmwareInfo | None:
+ """Return firmware information for the OpenThread Border Router."""
+ owners: list[OwningIntegration | OwningAddon] = [
+ OwningIntegration(config_entry_id=config_entry.entry_id)
+ ]
+
+ device = None
+
+ if is_hassio(hass) and (host := URL(config_entry.data["url"]).host) is not None:
+ otbr_addon_manager = AddonManager(
+ hass=hass,
+ logger=_LOGGER,
+ addon_name="OpenThread Border Router",
+ addon_slug=host.replace("-", "_"),
+ )
+
+ if (
+ addon_fw_info := await get_otbr_addon_firmware_info(
+ hass, otbr_addon_manager
+ )
+ ) is not None:
+ device = addon_fw_info.device
+ owners.extend(addon_fw_info.owners)
+
+ firmware_version = None
+
+ if config_entry.state in (
+ # This function is called during OTBR config entry setup so we need to account
+ # for both config entry states
+ ConfigEntryState.LOADED,
+ ConfigEntryState.SETUP_IN_PROGRESS,
+ ):
+ try:
+ firmware_version = await config_entry.runtime_data.get_coprocessor_version()
+ except HomeAssistantError:
+ firmware_version = None
+
+ if device is None:
+ return None
+
+ return FirmwareInfo(
+ device=device,
+ firmware_type=ApplicationType.SPINEL,
+ firmware_version=firmware_version,
+ source=DOMAIN,
+ owners=owners,
+ )
diff --git a/homeassistant/components/otbr/strings.json b/homeassistant/components/otbr/strings.json
index e1afa5b8909..3a9661c454d 100644
--- a/homeassistant/components/otbr/strings.json
+++ b/homeassistant/components/otbr/strings.json
@@ -5,7 +5,7 @@
"data": {
"url": "[%key:common::config_flow::data::url%]"
},
- "description": "Provide URL for the Open Thread Border Router's REST API"
+ "description": "Provide URL for the OpenThread Border Router's REST API"
}
},
"error": {
@@ -20,8 +20,8 @@
},
"issues": {
"get_get_border_agent_id_unsupported": {
- "title": "The OTBR does not support border agent ID",
- "description": "Your OTBR does not support border agent ID.\n\nTo fix this issue, update the OTBR to the latest version and restart Home Assistant.\nTo update the OTBR, update the Open Thread Border Router or Silicon Labs Multiprotocol add-on if you use the OTBR from the add-on, otherwise update your self managed OTBR."
+ "title": "The OTBR does not support Border Agent ID",
+ "description": "Your OTBR does not support Border Agent ID.\n\nTo fix this issue, update the OTBR to the latest version and restart Home Assistant.\nIf you are using an OTBR integrated in Home Assistant, update either the OpenThread Border Router add-on or the Silicon Labs Multiprotocol add-on. Otherwise update your self-managed OTBR."
},
"insecure_thread_network": {
"title": "Insecure Thread network settings detected",
diff --git a/homeassistant/components/otbr/types.py b/homeassistant/components/otbr/types.py
new file mode 100644
index 00000000000..eff6aa980d6
--- /dev/null
+++ b/homeassistant/components/otbr/types.py
@@ -0,0 +1,7 @@
+"""The Open Thread Border Router integration types."""
+
+from homeassistant.config_entries import ConfigEntry
+
+from .util import OTBRData
+
+type OTBRConfigEntry = ConfigEntry[OTBRData]
diff --git a/homeassistant/components/otbr/util.py b/homeassistant/components/otbr/util.py
index 351e23c7736..30e456e11a8 100644
--- a/homeassistant/components/otbr/util.py
+++ b/homeassistant/components/otbr/util.py
@@ -163,6 +163,11 @@ class OTBRData:
"""Get extended address (EUI-64)."""
return await self.api.get_extended_address()
+ @_handle_otbr_error
+ async def get_coprocessor_version(self) -> str:
+ """Get coprocessor firmware version."""
+ return await self.api.get_coprocessor_version()
+
async def get_allowed_channel(hass: HomeAssistant, otbr_url: str) -> int | None:
"""Return the allowed channel, or None if there's no restriction."""
diff --git a/homeassistant/components/otp/sensor.py b/homeassistant/components/otp/sensor.py
index 255bc0ded34..af508d2e915 100644
--- a/homeassistant/components/otp/sensor.py
+++ b/homeassistant/components/otp/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, CONF_TOKEN
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN
@@ -20,7 +20,9 @@ TIME_STEP = 30 # Default time step assumed by Google Authenticator
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OTP sensor."""
diff --git a/homeassistant/components/ourgroceries/__init__.py b/homeassistant/components/ourgroceries/__init__.py
index 5086a5cfc9b..a83430b3531 100644
--- a/homeassistant/components/ourgroceries/__init__.py
+++ b/homeassistant/components/ourgroceries/__init__.py
@@ -30,7 +30,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except InvalidLoginException:
return False
- coordinator = OurGroceriesDataUpdateCoordinator(hass, og)
+ coordinator = OurGroceriesDataUpdateCoordinator(hass, entry, og)
await coordinator.async_config_entry_first_refresh()
hass.data[DOMAIN][entry.entry_id] = coordinator
diff --git a/homeassistant/components/ourgroceries/coordinator.py b/homeassistant/components/ourgroceries/coordinator.py
index bc645b2bdb3..a822931e88c 100644
--- a/homeassistant/components/ourgroceries/coordinator.py
+++ b/homeassistant/components/ourgroceries/coordinator.py
@@ -8,6 +8,7 @@ import logging
from ourgroceries import OurGroceries
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -21,7 +22,11 @@ _LOGGER = logging.getLogger(__name__)
class OurGroceriesDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
"""Class to manage fetching OurGroceries data."""
- def __init__(self, hass: HomeAssistant, og: OurGroceries) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, og: OurGroceries
+ ) -> None:
"""Initialize global OurGroceries data updater."""
self.og = og
self.lists: list[dict] = []
@@ -30,6 +35,7 @@ class OurGroceriesDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=interval,
)
diff --git a/homeassistant/components/ourgroceries/todo.py b/homeassistant/components/ourgroceries/todo.py
index 5b8d19e5aa1..f257ef481c7 100644
--- a/homeassistant/components/ourgroceries/todo.py
+++ b/homeassistant/components/ourgroceries/todo.py
@@ -11,7 +11,7 @@ from homeassistant.components.todo import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -19,7 +19,9 @@ from .coordinator import OurGroceriesDataUpdateCoordinator
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the OurGroceries todo platform config entry."""
coordinator: OurGroceriesDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/overkiz/__init__.py b/homeassistant/components/overkiz/__init__.py
index 51efb52e55d..8aa1ed0e4fe 100644
--- a/homeassistant/components/overkiz/__init__.py
+++ b/homeassistant/components/overkiz/__init__.py
@@ -39,7 +39,6 @@ from .const import (
LOGGER,
OVERKIZ_DEVICE_TO_PLATFORM,
PLATFORMS,
- UPDATE_INTERVAL,
UPDATE_INTERVAL_ALL_ASSUMED_STATE,
UPDATE_INTERVAL_LOCAL,
)
@@ -104,13 +103,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: OverkizDataConfigEntry)
coordinator = OverkizDataUpdateCoordinator(
hass,
+ entry,
LOGGER,
- name="device events",
client=client,
devices=setup.devices,
places=setup.root_place,
- update_interval=UPDATE_INTERVAL,
- config_entry_id=entry.entry_id,
)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/overkiz/alarm_control_panel.py b/homeassistant/components/overkiz/alarm_control_panel.py
index 90c135291c3..1a5490dd329 100644
--- a/homeassistant/components/overkiz/alarm_control_panel.py
+++ b/homeassistant/components/overkiz/alarm_control_panel.py
@@ -19,7 +19,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .coordinator import OverkizDataUpdateCoordinator
@@ -209,7 +209,7 @@ SUPPORTED_DEVICES = {description.key: description for description in ALARM_DESCR
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz alarm control panel from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/binary_sensor.py b/homeassistant/components/overkiz/binary_sensor.py
index 3a75cd77c2f..09319d59932 100644
--- a/homeassistant/components/overkiz/binary_sensor.py
+++ b/homeassistant/components/overkiz/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .const import IGNORED_OVERKIZ_DEVICES
@@ -143,7 +143,7 @@ SUPPORTED_STATES = {
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz binary sensors from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/button.py b/homeassistant/components/overkiz/button.py
index 92711ac8ca8..f4e051ef9ca 100644
--- a/homeassistant/components/overkiz/button.py
+++ b/homeassistant/components/overkiz/button.py
@@ -14,7 +14,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .const import IGNORED_OVERKIZ_DEVICES
@@ -100,7 +100,7 @@ SUPPORTED_COMMANDS = {
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz button from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/climate/__init__.py b/homeassistant/components/overkiz/climate/__init__.py
index 3276a1979cc..058c3aefdb7 100644
--- a/homeassistant/components/overkiz/climate/__init__.py
+++ b/homeassistant/components/overkiz/climate/__init__.py
@@ -10,7 +10,7 @@ from pyoverkiz.enums.ui import UIWidget
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .. import OverkizDataConfigEntry
from .atlantic_electrical_heater import AtlanticElectricalHeater
@@ -82,7 +82,7 @@ WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY = {
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz climate from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/coordinator.py b/homeassistant/components/overkiz/coordinator.py
index 484ef138cf7..4b79cfc9c06 100644
--- a/homeassistant/components/overkiz/coordinator.py
+++ b/homeassistant/components/overkiz/coordinator.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from collections.abc import Callable, Coroutine
from datetime import timedelta
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
from aiohttp import ClientConnectorError, ServerDisconnectedError
from pyoverkiz.client import OverkizClient
@@ -26,7 +26,10 @@ from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util.decorator import Registry
-from .const import DOMAIN, IGNORED_OVERKIZ_DEVICES, LOGGER
+if TYPE_CHECKING:
+ from . import OverkizDataConfigEntry
+
+from .const import DOMAIN, IGNORED_OVERKIZ_DEVICES, LOGGER, UPDATE_INTERVAL
EVENT_HANDLERS: Registry[
str, Callable[[OverkizDataUpdateCoordinator, Event], Coroutine[Any, Any, None]]
@@ -36,26 +39,26 @@ EVENT_HANDLERS: Registry[
class OverkizDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Device]]):
"""Class to manage fetching data from Overkiz platform."""
+ config_entry: OverkizDataConfigEntry
_default_update_interval: timedelta
def __init__(
self,
hass: HomeAssistant,
+ config_entry: OverkizDataConfigEntry,
logger: logging.Logger,
*,
- name: str,
client: OverkizClient,
devices: list[Device],
places: Place | None,
- update_interval: timedelta,
- config_entry_id: str,
) -> None:
"""Initialize global data updater."""
super().__init__(
hass,
logger,
- name=name,
- update_interval=update_interval,
+ config_entry=config_entry,
+ name="device events",
+ update_interval=UPDATE_INTERVAL,
)
self.data = {}
@@ -63,8 +66,7 @@ class OverkizDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Device]]):
self.devices: dict[str, Device] = {d.device_url: d for d in devices}
self.executions: dict[str, dict[str, str]] = {}
self.areas = self._places_to_area(places) if places else None
- self.config_entry_id = config_entry_id
- self._default_update_interval = update_interval
+ self._default_update_interval = UPDATE_INTERVAL
self.is_stateless = all(
device.protocol in (Protocol.RTS, Protocol.INTERNAL)
@@ -164,7 +166,7 @@ async def on_device_created_updated(
) -> None:
"""Handle device unavailable / disabled event."""
coordinator.hass.async_create_task(
- coordinator.hass.config_entries.async_reload(coordinator.config_entry_id)
+ coordinator.hass.config_entries.async_reload(coordinator.config_entry.entry_id)
)
diff --git a/homeassistant/components/overkiz/cover/__init__.py b/homeassistant/components/overkiz/cover/__init__.py
index 38c02eba1bb..dd3216f9c10 100644
--- a/homeassistant/components/overkiz/cover/__init__.py
+++ b/homeassistant/components/overkiz/cover/__init__.py
@@ -4,7 +4,7 @@ from pyoverkiz.enums import OverkizCommand, UIClass
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .. import OverkizDataConfigEntry
from .awning import Awning
@@ -15,7 +15,7 @@ from .vertical_cover import LowSpeedCover, VerticalCover
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz covers from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/light.py b/homeassistant/components/overkiz/light.py
index 933d4cf695b..acd63140196 100644
--- a/homeassistant/components/overkiz/light.py
+++ b/homeassistant/components/overkiz/light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .coordinator import OverkizDataUpdateCoordinator
@@ -24,7 +24,7 @@ from .entity import OverkizEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz lights from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/lock.py b/homeassistant/components/overkiz/lock.py
index 1c073d2f9aa..16ec32b0667 100644
--- a/homeassistant/components/overkiz/lock.py
+++ b/homeassistant/components/overkiz/lock.py
@@ -9,7 +9,7 @@ from pyoverkiz.enums import OverkizCommand, OverkizCommandParam, OverkizState
from homeassistant.components.lock import LockEntity
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .entity import OverkizEntity
@@ -18,7 +18,7 @@ from .entity import OverkizEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz locks from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/manifest.json b/homeassistant/components/overkiz/manifest.json
index c25accd87f3..cfaed4ceb8b 100644
--- a/homeassistant/components/overkiz/manifest.json
+++ b/homeassistant/components/overkiz/manifest.json
@@ -13,7 +13,7 @@
"integration_type": "hub",
"iot_class": "local_polling",
"loggers": ["boto3", "botocore", "pyhumps", "pyoverkiz", "s3transfer"],
- "requirements": ["pyoverkiz==1.16.0"],
+ "requirements": ["pyoverkiz==1.16.4"],
"zeroconf": [
{
"type": "_kizbox._tcp.local.",
diff --git a/homeassistant/components/overkiz/number.py b/homeassistant/components/overkiz/number.py
index 0e03e822424..83c0e7cf7a8 100644
--- a/homeassistant/components/overkiz/number.py
+++ b/homeassistant/components/overkiz/number.py
@@ -16,7 +16,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .const import IGNORED_OVERKIZ_DEVICES
@@ -191,7 +191,7 @@ SUPPORTED_STATES = {description.key: description for description in NUMBER_DESCR
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz number from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/scene.py b/homeassistant/components/overkiz/scene.py
index 4533ed3245c..bd362b4b372 100644
--- a/homeassistant/components/overkiz/scene.py
+++ b/homeassistant/components/overkiz/scene.py
@@ -9,7 +9,7 @@ from pyoverkiz.models import Scenario
from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
@@ -17,7 +17,7 @@ from . import OverkizDataConfigEntry
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz scenes from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/select.py b/homeassistant/components/overkiz/select.py
index ac467eaaa7a..e23dafdaab8 100644
--- a/homeassistant/components/overkiz/select.py
+++ b/homeassistant/components/overkiz/select.py
@@ -10,7 +10,7 @@ from pyoverkiz.enums import OverkizCommand, OverkizCommandParam, OverkizState
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .const import IGNORED_OVERKIZ_DEVICES
@@ -129,7 +129,7 @@ SUPPORTED_STATES = {description.key: description for description in SELECT_DESCR
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz select from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/sensor.py b/homeassistant/components/overkiz/sensor.py
index 81a9ab41d2d..9214398a37b 100644
--- a/homeassistant/components/overkiz/sensor.py
+++ b/homeassistant/components/overkiz/sensor.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import OverkizDataConfigEntry
@@ -483,7 +483,7 @@ SUPPORTED_STATES = {description.key: description for description in SENSOR_DESCR
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz sensors from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/siren.py b/homeassistant/components/overkiz/siren.py
index f7246e50ec0..af761611444 100644
--- a/homeassistant/components/overkiz/siren.py
+++ b/homeassistant/components/overkiz/siren.py
@@ -12,7 +12,7 @@ from homeassistant.components.siren import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .entity import OverkizEntity
@@ -21,7 +21,7 @@ from .entity import OverkizEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz sirens from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/switch.py b/homeassistant/components/overkiz/switch.py
index c921dbab776..d14b2792947 100644
--- a/homeassistant/components/overkiz/switch.py
+++ b/homeassistant/components/overkiz/switch.py
@@ -17,7 +17,7 @@ from homeassistant.components.switch import (
)
from homeassistant.const import EntityCategory, Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OverkizDataConfigEntry
from .entity import OverkizDescriptiveEntity
@@ -110,7 +110,7 @@ SUPPORTED_DEVICES = {
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz switch from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overkiz/water_heater/__init__.py b/homeassistant/components/overkiz/water_heater/__init__.py
index 1dd1d596a33..9895ea84c2c 100644
--- a/homeassistant/components/overkiz/water_heater/__init__.py
+++ b/homeassistant/components/overkiz/water_heater/__init__.py
@@ -6,7 +6,7 @@ from pyoverkiz.enums.ui import UIWidget
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .. import OverkizDataConfigEntry
from ..entity import OverkizEntity
@@ -21,7 +21,7 @@ from .hitachi_dhw import HitachiDHW
async def async_setup_entry(
hass: HomeAssistant,
entry: OverkizDataConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Overkiz DHW from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/overseerr/event.py b/homeassistant/components/overseerr/event.py
index 589a80c5404..1ffb1e71771 100644
--- a/homeassistant/components/overseerr/event.py
+++ b/homeassistant/components/overseerr/event.py
@@ -8,7 +8,7 @@ from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, EVENT_KEY
from .coordinator import OverseerrConfigEntry, OverseerrCoordinator
@@ -44,7 +44,7 @@ EVENTS: tuple[OverseerrEventEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: OverseerrConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Overseerr sensor entities based on a config entry."""
diff --git a/homeassistant/components/overseerr/manifest.json b/homeassistant/components/overseerr/manifest.json
index 6258481adcf..3c4321ebb37 100644
--- a/homeassistant/components/overseerr/manifest.json
+++ b/homeassistant/components/overseerr/manifest.json
@@ -9,5 +9,5 @@
"integration_type": "service",
"iot_class": "local_push",
"quality_scale": "platinum",
- "requirements": ["python-overseerr==0.7.0"]
+ "requirements": ["python-overseerr==0.7.1"]
}
diff --git a/homeassistant/components/overseerr/sensor.py b/homeassistant/components/overseerr/sensor.py
index 2daaa3de0cb..510e6f52c59 100644
--- a/homeassistant/components/overseerr/sensor.py
+++ b/homeassistant/components/overseerr/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import REQUESTS
from .coordinator import OverseerrConfigEntry, OverseerrCoordinator
@@ -76,7 +76,7 @@ SENSORS: tuple[OverseerrSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: OverseerrConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Overseerr sensor entities based on a config entry."""
diff --git a/homeassistant/components/overseerr/strings.json b/homeassistant/components/overseerr/strings.json
index 14650fd5c25..ce8b9fe9fec 100644
--- a/homeassistant/components/overseerr/strings.json
+++ b/homeassistant/components/overseerr/strings.json
@@ -90,7 +90,7 @@
"services": {
"get_requests": {
"name": "Get requests",
- "description": "Get media requests from Overseerr.",
+ "description": "Retrieves a list of media requests from Overseerr.",
"fields": {
"config_entry_id": {
"name": "Overseerr instance",
@@ -106,7 +106,7 @@
},
"requested_by": {
"name": "Requested by",
- "description": "Filter the requests by the user id that requested them."
+ "description": "Filter the requests by the user ID that requested them."
}
}
}
diff --git a/homeassistant/components/ovo_energy/sensor.py b/homeassistant/components/ovo_energy/sensor.py
index 8cada86da34..1dc12c7f008 100644
--- a/homeassistant/components/ovo_energy/sensor.py
+++ b/homeassistant/components/ovo_energy/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfEnergy
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from homeassistant.util import dt as dt_util
@@ -111,7 +111,9 @@ SENSOR_TYPES_GAS: tuple[OVOEnergySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up OVO Energy sensor based on a config entry."""
coordinator: DataUpdateCoordinator[OVODailyUsage] = hass.data[DOMAIN][
diff --git a/homeassistant/components/owntracks/device_tracker.py b/homeassistant/components/owntracks/device_tracker.py
index 6a6f0f078b1..7ccbbb69aa1 100644
--- a/homeassistant/components/owntracks/device_tracker.py
+++ b/homeassistant/components/owntracks/device_tracker.py
@@ -16,14 +16,16 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import DOMAIN as OT_DOMAIN
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up OwnTracks based off an entry."""
# Restore previously loaded devices
diff --git a/homeassistant/components/p1_monitor/__init__.py b/homeassistant/components/p1_monitor/__init__.py
index d2ccc83972a..e12c092453c 100644
--- a/homeassistant/components/p1_monitor/__init__.py
+++ b/homeassistant/components/p1_monitor/__init__.py
@@ -2,23 +2,20 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from .const import LOGGER
-from .coordinator import P1MonitorDataUpdateCoordinator
+from .coordinator import P1MonitorConfigEntry, P1MonitorDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type P1MonitorConfigEntry = ConfigEntry[P1MonitorDataUpdateCoordinator]
-
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: P1MonitorConfigEntry) -> bool:
"""Set up P1 Monitor from a config entry."""
- coordinator = P1MonitorDataUpdateCoordinator(hass)
+ coordinator = P1MonitorDataUpdateCoordinator(hass, entry)
try:
await coordinator.async_config_entry_first_refresh()
except ConfigEntryNotReady:
@@ -31,7 +28,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True
-async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, config_entry: P1MonitorConfigEntry
+) -> bool:
"""Migrate old entry."""
LOGGER.debug("Migrating from version %s", config_entry.version)
@@ -54,6 +53,6 @@ async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: P1MonitorConfigEntry) -> bool:
"""Unload P1 Monitor config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/p1_monitor/coordinator.py b/homeassistant/components/p1_monitor/coordinator.py
index 5459f88c388..3be78f8efd5 100644
--- a/homeassistant/components/p1_monitor/coordinator.py
+++ b/homeassistant/components/p1_monitor/coordinator.py
@@ -30,6 +30,8 @@ from .const import (
SERVICE_WATERMETER,
)
+type P1MonitorConfigEntry = ConfigEntry[P1MonitorDataUpdateCoordinator]
+
class P1MonitorData(TypedDict):
"""Class for defining data in dict."""
@@ -43,17 +45,19 @@ class P1MonitorData(TypedDict):
class P1MonitorDataUpdateCoordinator(DataUpdateCoordinator[P1MonitorData]):
"""Class to manage fetching P1 Monitor data from single endpoint."""
- config_entry: ConfigEntry
+ config_entry: P1MonitorConfigEntry
has_water_meter: bool | None = None
def __init__(
self,
hass: HomeAssistant,
+ config_entry: P1MonitorConfigEntry,
) -> None:
"""Initialize global P1 Monitor data updater."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/p1_monitor/diagnostics.py b/homeassistant/components/p1_monitor/diagnostics.py
index d2e2ec5c24e..ac670486e79 100644
--- a/homeassistant/components/p1_monitor/diagnostics.py
+++ b/homeassistant/components/p1_monitor/diagnostics.py
@@ -6,7 +6,6 @@ from dataclasses import asdict
from typing import TYPE_CHECKING, Any, cast
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT
from homeassistant.core import HomeAssistant
@@ -16,6 +15,7 @@ from .const import (
SERVICE_SMARTMETER,
SERVICE_WATERMETER,
)
+from .coordinator import P1MonitorConfigEntry
if TYPE_CHECKING:
from _typeshed import DataclassInstance
@@ -24,7 +24,7 @@ TO_REDACT = {CONF_HOST, CONF_PORT}
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, entry: ConfigEntry
+ hass: HomeAssistant, entry: P1MonitorConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
data = {
diff --git a/homeassistant/components/p1_monitor/sensor.py b/homeassistant/components/p1_monitor/sensor.py
index 771ef0e19af..15a8f510fd7 100644
--- a/homeassistant/components/p1_monitor/sensor.py
+++ b/homeassistant/components/p1_monitor/sensor.py
@@ -10,7 +10,6 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_HOST,
CURRENCY_EURO,
@@ -22,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -33,7 +32,7 @@ from .const import (
SERVICE_SMARTMETER,
SERVICE_WATERMETER,
)
-from .coordinator import P1MonitorDataUpdateCoordinator
+from .coordinator import P1MonitorConfigEntry, P1MonitorDataUpdateCoordinator
SENSORS_SMARTMETER: tuple[SensorEntityDescription, ...] = (
SensorEntityDescription(
@@ -236,7 +235,9 @@ SENSORS_WATERMETER: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: P1MonitorConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up P1 Monitor Sensors based on a config entry."""
entities: list[P1MonitorSensorEntity] = []
@@ -290,7 +291,7 @@ class P1MonitorSensorEntity(
def __init__(
self,
*,
- entry: ConfigEntry,
+ entry: P1MonitorConfigEntry,
description: SensorEntityDescription,
name: str,
service: Literal["smartmeter", "watermeter", "phases", "settings"],
diff --git a/homeassistant/components/palazzetti/__init__.py b/homeassistant/components/palazzetti/__init__.py
index dbf1baa0c28..a698cdcd8b7 100644
--- a/homeassistant/components/palazzetti/__init__.py
+++ b/homeassistant/components/palazzetti/__init__.py
@@ -18,7 +18,7 @@ PLATFORMS: list[Platform] = [
async def async_setup_entry(hass: HomeAssistant, entry: PalazzettiConfigEntry) -> bool:
"""Set up Palazzetti from a config entry."""
- coordinator = PalazzettiDataUpdateCoordinator(hass)
+ coordinator = PalazzettiDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/palazzetti/button.py b/homeassistant/components/palazzetti/button.py
index cd4765576ed..319a1174542 100644
--- a/homeassistant/components/palazzetti/button.py
+++ b/homeassistant/components/palazzetti/button.py
@@ -7,18 +7,17 @@ from pypalazzetti.exceptions import CommunicationError
from homeassistant.components.button import ButtonEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PalazzettiConfigEntry
from .const import DOMAIN
-from .coordinator import PalazzettiDataUpdateCoordinator
+from .coordinator import PalazzettiConfigEntry, PalazzettiDataUpdateCoordinator
from .entity import PalazzettiEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PalazzettiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Palazzetti button platform."""
diff --git a/homeassistant/components/palazzetti/climate.py b/homeassistant/components/palazzetti/climate.py
index 0722b97e4b7..5a4097e083a 100644
--- a/homeassistant/components/palazzetti/climate.py
+++ b/homeassistant/components/palazzetti/climate.py
@@ -13,18 +13,17 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PalazzettiConfigEntry
from .const import DOMAIN, FAN_AUTO, FAN_HIGH, FAN_MODES
-from .coordinator import PalazzettiDataUpdateCoordinator
+from .coordinator import PalazzettiConfigEntry, PalazzettiDataUpdateCoordinator
from .entity import PalazzettiEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: PalazzettiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Palazzetti climates based on a config entry."""
async_add_entities([PalazzettiClimateEntity(entry.runtime_data)])
diff --git a/homeassistant/components/palazzetti/coordinator.py b/homeassistant/components/palazzetti/coordinator.py
index d992bd3fb62..1e4069e58ea 100644
--- a/homeassistant/components/palazzetti/coordinator.py
+++ b/homeassistant/components/palazzetti/coordinator.py
@@ -22,11 +22,13 @@ class PalazzettiDataUpdateCoordinator(DataUpdateCoordinator[None]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: PalazzettiConfigEntry,
) -> None:
"""Initialize global Palazzetti data updater."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/palazzetti/diagnostics.py b/homeassistant/components/palazzetti/diagnostics.py
index 3843f0ec111..e386ffc7833 100644
--- a/homeassistant/components/palazzetti/diagnostics.py
+++ b/homeassistant/components/palazzetti/diagnostics.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.core import HomeAssistant
-from . import PalazzettiConfigEntry
+from .coordinator import PalazzettiConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/palazzetti/number.py b/homeassistant/components/palazzetti/number.py
index 2b303f71fd6..63c1ed16f0c 100644
--- a/homeassistant/components/palazzetti/number.py
+++ b/homeassistant/components/palazzetti/number.py
@@ -8,18 +8,17 @@ from pypalazzetti.fan import FanType
from homeassistant.components.number import NumberDeviceClass, NumberEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PalazzettiConfigEntry
from .const import DOMAIN
-from .coordinator import PalazzettiDataUpdateCoordinator
+from .coordinator import PalazzettiConfigEntry, PalazzettiDataUpdateCoordinator
from .entity import PalazzettiEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PalazzettiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Palazzetti number platform."""
diff --git a/homeassistant/components/palazzetti/sensor.py b/homeassistant/components/palazzetti/sensor.py
index 11462201f4e..57d5ca861a2 100644
--- a/homeassistant/components/palazzetti/sensor.py
+++ b/homeassistant/components/palazzetti/sensor.py
@@ -10,12 +10,11 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfLength, UnitOfMass, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import PalazzettiConfigEntry
from .const import STATUS_TO_HA
-from .coordinator import PalazzettiDataUpdateCoordinator
+from .coordinator import PalazzettiConfigEntry, PalazzettiDataUpdateCoordinator
from .entity import PalazzettiEntity
@@ -60,7 +59,7 @@ PROPERTY_SENSOR_DESCRIPTIONS: list[PropertySensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PalazzettiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Palazzetti sensor entities based on a config entry."""
diff --git a/homeassistant/components/panasonic_viera/media_player.py b/homeassistant/components/panasonic_viera/media_player.py
index 8738b897d29..a78920f33a5 100644
--- a/homeassistant/components/panasonic_viera/media_player.py
+++ b/homeassistant/components/panasonic_viera/media_player.py
@@ -21,7 +21,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_DEVICE_INFO,
@@ -40,7 +40,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Panasonic Viera TV from a config entry."""
diff --git a/homeassistant/components/panasonic_viera/remote.py b/homeassistant/components/panasonic_viera/remote.py
index ad40a97f700..5fa4be9ca2b 100644
--- a/homeassistant/components/panasonic_viera/remote.py
+++ b/homeassistant/components/panasonic_viera/remote.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, STATE_ON
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Remote
from .const import (
@@ -28,7 +28,7 @@ from .const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Panasonic Viera TV Remote from a config entry."""
diff --git a/homeassistant/components/peblar/binary_sensor.py b/homeassistant/components/peblar/binary_sensor.py
index e8e5095f050..8834a2ba2a0 100644
--- a/homeassistant/components/peblar/binary_sensor.py
+++ b/homeassistant/components/peblar/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PeblarConfigEntry, PeblarData, PeblarDataUpdateCoordinator
from .entity import PeblarEntity
@@ -50,7 +50,7 @@ DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PeblarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Peblar binary sensor based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/peblar/button.py b/homeassistant/components/peblar/button.py
index 22150c82649..8c60c8d84d3 100644
--- a/homeassistant/components/peblar/button.py
+++ b/homeassistant/components/peblar/button.py
@@ -15,7 +15,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PeblarConfigEntry, PeblarUserConfigurationDataUpdateCoordinator
from .entity import PeblarEntity
@@ -52,7 +52,7 @@ DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PeblarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Peblar buttons based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/peblar/coordinator.py b/homeassistant/components/peblar/coordinator.py
index 058f2aefb3b..36708b207c5 100644
--- a/homeassistant/components/peblar/coordinator.py
+++ b/homeassistant/components/peblar/coordinator.py
@@ -34,6 +34,7 @@ class PeblarRuntimeData:
"""Class to hold runtime data."""
data_coordinator: PeblarDataUpdateCoordinator
+ last_known_charging_limit = 6
system_information: PeblarSystemInformation
user_configuration_coordinator: PeblarUserConfigurationDataUpdateCoordinator
version_coordinator: PeblarVersionDataUpdateCoordinator
@@ -137,6 +138,8 @@ class PeblarVersionDataUpdateCoordinator(
class PeblarDataUpdateCoordinator(DataUpdateCoordinator[PeblarData]):
"""Class to manage fetching Peblar active data."""
+ config_entry: PeblarConfigEntry
+
def __init__(
self, hass: HomeAssistant, entry: PeblarConfigEntry, api: PeblarApi
) -> None:
diff --git a/homeassistant/components/peblar/icons.json b/homeassistant/components/peblar/icons.json
index 6244945077b..a954d112c4a 100644
--- a/homeassistant/components/peblar/icons.json
+++ b/homeassistant/components/peblar/icons.json
@@ -36,6 +36,9 @@
}
},
"switch": {
+ "charge": {
+ "default": "mdi:ev-plug-type2"
+ },
"force_single_phase": {
"default": "mdi:power-cycle"
}
diff --git a/homeassistant/components/peblar/number.py b/homeassistant/components/peblar/number.py
index 1a7cec43295..bff1bb26db4 100644
--- a/homeassistant/components/peblar/number.py
+++ b/homeassistant/components/peblar/number.py
@@ -2,101 +2,129 @@
from __future__ import annotations
-from collections.abc import Awaitable, Callable
-from dataclasses import dataclass
-from typing import Any
-
-from peblar import PeblarApi
-
from homeassistant.components.number import (
NumberDeviceClass,
- NumberEntity,
NumberEntityDescription,
+ RestoreNumber,
)
-from homeassistant.const import EntityCategory, UnitOfElectricCurrent
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.const import (
+ STATE_UNAVAILABLE,
+ STATE_UNKNOWN,
+ EntityCategory,
+ UnitOfElectricCurrent,
+)
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .coordinator import (
- PeblarConfigEntry,
- PeblarData,
- PeblarDataUpdateCoordinator,
- PeblarRuntimeData,
-)
+from .coordinator import PeblarConfigEntry, PeblarDataUpdateCoordinator
from .entity import PeblarEntity
from .helpers import peblar_exception_handler
PARALLEL_UPDATES = 1
-@dataclass(frozen=True, kw_only=True)
-class PeblarNumberEntityDescription(NumberEntityDescription):
- """Describe a Peblar number."""
-
- native_max_value_fn: Callable[[PeblarRuntimeData], int]
- set_value_fn: Callable[[PeblarApi, float], Awaitable[Any]]
- value_fn: Callable[[PeblarData], int | None]
-
-
-DESCRIPTIONS = [
- PeblarNumberEntityDescription(
- key="charge_current_limit",
- translation_key="charge_current_limit",
- device_class=NumberDeviceClass.CURRENT,
- entity_category=EntityCategory.CONFIG,
- native_step=1,
- native_min_value=6,
- native_max_value_fn=lambda x: x.user_configuration_coordinator.data.user_defined_charge_limit_current,
- native_unit_of_measurement=UnitOfElectricCurrent.AMPERE,
- set_value_fn=lambda x, v: x.ev_interface(charge_current_limit=int(v) * 1000),
- value_fn=lambda x: round(x.ev.charge_current_limit / 1000),
- ),
-]
-
-
async def async_setup_entry(
hass: HomeAssistant,
entry: PeblarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Peblar number based on a config entry."""
async_add_entities(
- PeblarNumberEntity(
- entry=entry,
- coordinator=entry.runtime_data.data_coordinator,
- description=description,
- )
- for description in DESCRIPTIONS
+ [
+ PeblarChargeCurrentLimitNumberEntity(
+ entry=entry,
+ coordinator=entry.runtime_data.data_coordinator,
+ )
+ ]
)
-class PeblarNumberEntity(
+class PeblarChargeCurrentLimitNumberEntity(
PeblarEntity[PeblarDataUpdateCoordinator],
- NumberEntity,
+ RestoreNumber,
):
- """Defines a Peblar number."""
+ """Defines a Peblar charge current limit number.
- entity_description: PeblarNumberEntityDescription
+ This entity is a little bit different from the other entities, any value
+ below 6 amps is ignored. It means the Peblar is not charging.
+ Peblar has assigned a dual functionality to the charge current limit
+ number, it is used to set the current charging value and to start/stop/pauze
+ the charging process.
+ """
+
+ _attr_device_class = NumberDeviceClass.CURRENT
+ _attr_entity_category = EntityCategory.CONFIG
+ _attr_native_min_value = 6
+ _attr_native_step = 1
+ _attr_native_unit_of_measurement = UnitOfElectricCurrent.AMPERE
+ _attr_translation_key = "charge_current_limit"
def __init__(
self,
entry: PeblarConfigEntry,
coordinator: PeblarDataUpdateCoordinator,
- description: PeblarNumberEntityDescription,
) -> None:
- """Initialize the Peblar entity."""
- super().__init__(entry=entry, coordinator=coordinator, description=description)
- self._attr_native_max_value = description.native_max_value_fn(
- entry.runtime_data
+ """Initialize the Peblar charge current limit entity."""
+ super().__init__(
+ entry=entry,
+ coordinator=coordinator,
+ description=NumberEntityDescription(key="charge_current_limit"),
)
+ configuration = entry.runtime_data.user_configuration_coordinator.data
+ self._attr_native_max_value = configuration.user_defined_charge_limit_current
- @property
- def native_value(self) -> int | None:
- """Return the number value."""
- return self.entity_description.value_fn(self.coordinator.data)
+ async def async_added_to_hass(self) -> None:
+ """Load the last known state when adding this entity."""
+ if (
+ (last_state := await self.async_get_last_state())
+ and (last_number_data := await self.async_get_last_number_data())
+ and last_state.state not in (STATE_UNKNOWN, STATE_UNAVAILABLE)
+ and last_number_data.native_value
+ ):
+ self._attr_native_value = last_number_data.native_value
+ # Set the last known charging limit in the runtime data the
+ # start/stop/pauze functionality needs it in order to restore
+ # the last known charging limits when charging is resumed.
+ self.coordinator.config_entry.runtime_data.last_known_charging_limit = int(
+ last_number_data.native_value
+ )
+ await super().async_added_to_hass()
+ self._handle_coordinator_update()
+
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle coordinator update.
+
+ Ignore any update that provides a ampere value that is below the
+ minimum value (6 amps). It means the Peblar is currently not charging.
+ """
+ if (
+ current_charge_limit := round(
+ self.coordinator.data.ev.charge_current_limit / 1000
+ )
+ ) < 6:
+ return
+ self._attr_native_value = current_charge_limit
+ # Update the last known charging limit in the runtime data the
+ # start/stop/pauze functionality needs it in order to restore
+ # the last known charging limits when charging is resumed.
+ self.coordinator.config_entry.runtime_data.last_known_charging_limit = (
+ current_charge_limit
+ )
+ super()._handle_coordinator_update()
@peblar_exception_handler
async def async_set_native_value(self, value: float) -> None:
- """Change to new number value."""
- await self.entity_description.set_value_fn(self.coordinator.api, value)
+ """Change the current charging value."""
+ # If charging is currently disabled (below 6 amps), just set the value
+ # as the native value and the last known charging limit in the runtime
+ # data. So we can pick it up once charging gets enabled again.
+ if self.coordinator.data.ev.charge_current_limit < 6000:
+ self._attr_native_value = int(value)
+ self.coordinator.config_entry.runtime_data.last_known_charging_limit = int(
+ value
+ )
+ self.async_write_ha_state()
+ return
+ await self.coordinator.api.ev_interface(charge_current_limit=int(value) * 1000)
await self.coordinator.async_request_refresh()
diff --git a/homeassistant/components/peblar/select.py b/homeassistant/components/peblar/select.py
index a2a0997a797..17503951ccd 100644
--- a/homeassistant/components/peblar/select.py
+++ b/homeassistant/components/peblar/select.py
@@ -11,7 +11,7 @@ from peblar import Peblar, PeblarUserConfiguration, SmartChargingMode
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PeblarConfigEntry, PeblarUserConfigurationDataUpdateCoordinator
from .entity import PeblarEntity
@@ -49,7 +49,7 @@ DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PeblarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Peblar select based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/peblar/sensor.py b/homeassistant/components/peblar/sensor.py
index e655253d75c..81476eef9aa 100644
--- a/homeassistant/components/peblar/sensor.py
+++ b/homeassistant/components/peblar/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from .const import (
@@ -231,7 +231,7 @@ DESCRIPTIONS: tuple[PeblarSensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PeblarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Peblar sensors based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/peblar/strings.json b/homeassistant/components/peblar/strings.json
index a33667fa533..416f1a2c062 100644
--- a/homeassistant/components/peblar/strings.json
+++ b/homeassistant/components/peblar/strings.json
@@ -107,7 +107,7 @@
"cp_state": {
"name": "State",
"state": {
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"error": "Error",
"fault": "Fault",
"invalid": "Invalid",
@@ -153,6 +153,9 @@
}
},
"switch": {
+ "charge": {
+ "name": "Charge"
+ },
"force_single_phase": {
"name": "Force single phase"
}
diff --git a/homeassistant/components/peblar/switch.py b/homeassistant/components/peblar/switch.py
index e56c2fcdaec..f2e1ae13ae2 100644
--- a/homeassistant/components/peblar/switch.py
+++ b/homeassistant/components/peblar/switch.py
@@ -6,12 +6,12 @@ from collections.abc import Awaitable, Callable
from dataclasses import dataclass
from typing import Any
-from peblar import PeblarApi
+from peblar import PeblarEVInterface
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import (
PeblarConfigEntry,
@@ -31,7 +31,19 @@ class PeblarSwitchEntityDescription(SwitchEntityDescription):
has_fn: Callable[[PeblarRuntimeData], bool] = lambda x: True
is_on_fn: Callable[[PeblarData], bool]
- set_fn: Callable[[PeblarApi, bool], Awaitable[Any]]
+ set_fn: Callable[[PeblarDataUpdateCoordinator, bool], Awaitable[Any]]
+
+
+def _async_peblar_charge(
+ coordinator: PeblarDataUpdateCoordinator, on: bool
+) -> Awaitable[PeblarEVInterface]:
+ """Set the charge state."""
+ charge_current_limit = 0
+ if on:
+ charge_current_limit = (
+ coordinator.config_entry.runtime_data.last_known_charging_limit * 1000
+ )
+ return coordinator.api.ev_interface(charge_current_limit=charge_current_limit)
DESCRIPTIONS = [
@@ -44,7 +56,14 @@ DESCRIPTIONS = [
and x.user_configuration_coordinator.data.connected_phases > 1
),
is_on_fn=lambda x: x.ev.force_single_phase,
- set_fn=lambda x, on: x.ev_interface(force_single_phase=on),
+ set_fn=lambda x, on: x.api.ev_interface(force_single_phase=on),
+ ),
+ PeblarSwitchEntityDescription(
+ key="charge",
+ translation_key="charge",
+ entity_category=EntityCategory.CONFIG,
+ is_on_fn=lambda x: (x.ev.charge_current_limit >= 6000),
+ set_fn=_async_peblar_charge,
),
]
@@ -52,7 +71,7 @@ DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PeblarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Peblar switch based on a config entry."""
async_add_entities(
@@ -82,11 +101,11 @@ class PeblarSwitchEntity(
@peblar_exception_handler
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn the entity on."""
- await self.entity_description.set_fn(self.coordinator.api, True)
+ await self.entity_description.set_fn(self.coordinator, True)
await self.coordinator.async_request_refresh()
@peblar_exception_handler
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the entity off."""
- await self.entity_description.set_fn(self.coordinator.api, False)
+ await self.entity_description.set_fn(self.coordinator, False)
await self.coordinator.async_request_refresh()
diff --git a/homeassistant/components/peblar/update.py b/homeassistant/components/peblar/update.py
index 58c2fbdc899..88966916069 100644
--- a/homeassistant/components/peblar/update.py
+++ b/homeassistant/components/peblar/update.py
@@ -11,7 +11,7 @@ from homeassistant.components.update import (
UpdateEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import (
PeblarConfigEntry,
@@ -53,7 +53,7 @@ DESCRIPTIONS: tuple[PeblarUpdateEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PeblarConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Peblar update based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/peco/binary_sensor.py b/homeassistant/components/peco/binary_sensor.py
index a55f0fcc731..a4d59a8c9a2 100644
--- a/homeassistant/components/peco/binary_sensor.py
+++ b/homeassistant/components/peco/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -24,7 +24,7 @@ PARALLEL_UPDATES: Final = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor for PECO."""
if "smart_meter" not in hass.data[DOMAIN][config_entry.entry_id]:
diff --git a/homeassistant/components/peco/sensor.py b/homeassistant/components/peco/sensor.py
index d08947eb0ec..eafa36c98e9 100644
--- a/homeassistant/components/peco/sensor.py
+++ b/homeassistant/components/peco/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -76,7 +76,7 @@ SENSOR_LIST: tuple[PECOSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
county: str = config_entry.data[CONF_COUNTY]
diff --git a/homeassistant/components/pegel_online/sensor.py b/homeassistant/components/pegel_online/sensor.py
index 181c0f5dc6d..fd90683a9b2 100644
--- a/homeassistant/components/pegel_online/sensor.py
+++ b/homeassistant/components/pegel_online/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import PegelOnlineConfigEntry, PegelOnlineDataUpdateCoordinator
from .entity import PegelOnlineEntity
@@ -92,7 +92,7 @@ SENSORS: tuple[PegelOnlineSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PegelOnlineConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the PEGELONLINE sensor."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/permobil/__init__.py b/homeassistant/components/permobil/__init__.py
index 675a803ce91..441c6a2646e 100644
--- a/homeassistant/components/permobil/__init__.py
+++ b/homeassistant/components/permobil/__init__.py
@@ -48,7 +48,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
raise ConfigEntryAuthFailed(f"Config error for {p_api.email}") from err
# create the coordinator with the API object
- coordinator = MyPermobilCoordinator(hass, p_api)
+ coordinator = MyPermobilCoordinator(hass, entry, p_api)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/permobil/binary_sensor.py b/homeassistant/components/permobil/binary_sensor.py
index 4b768cf5af5..c2d51067e19 100644
--- a/homeassistant/components/permobil/binary_sensor.py
+++ b/homeassistant/components/permobil/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MyPermobilCoordinator
@@ -42,7 +42,7 @@ BINARY_SENSOR_DESCRIPTIONS: tuple[PermobilBinarySensorEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create and setup the binary sensor."""
diff --git a/homeassistant/components/permobil/coordinator.py b/homeassistant/components/permobil/coordinator.py
index 6efde26d341..ea7ddadff9f 100644
--- a/homeassistant/components/permobil/coordinator.py
+++ b/homeassistant/components/permobil/coordinator.py
@@ -7,6 +7,7 @@ import logging
from mypermobil import MyPermobil, MyPermobilAPIException
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -25,11 +26,16 @@ class MyPermobilData:
class MyPermobilCoordinator(DataUpdateCoordinator[MyPermobilData]):
"""MyPermobil coordinator."""
- def __init__(self, hass: HomeAssistant, p_api: MyPermobil) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, p_api: MyPermobil
+ ) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="permobil",
update_interval=timedelta(minutes=5),
)
diff --git a/homeassistant/components/permobil/sensor.py b/homeassistant/components/permobil/sensor.py
index 54d3a61c519..5f8cb88290a 100644
--- a/homeassistant/components/permobil/sensor.py
+++ b/homeassistant/components/permobil/sensor.py
@@ -32,7 +32,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfEnergy, UnitOfLength, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import BATTERY_ASSUMED_VOLTAGE, DOMAIN, KM, MILES
from .coordinator import MyPermobilCoordinator
@@ -175,7 +175,7 @@ DISTANCE_UNITS: dict[Any, UnitOfLength] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create sensors from a config entry created in the integrations UI."""
diff --git a/homeassistant/components/person/manifest.json b/homeassistant/components/person/manifest.json
index 7f370be6fbe..0c1792e9277 100644
--- a/homeassistant/components/person/manifest.json
+++ b/homeassistant/components/person/manifest.json
@@ -1,7 +1,6 @@
{
"domain": "person",
"name": "Person",
- "after_dependencies": ["device_tracker"],
"codeowners": [],
"dependencies": ["image_upload", "http"],
"documentation": "https://www.home-assistant.io/integrations/person",
diff --git a/homeassistant/components/pglab/__init__.py b/homeassistant/components/pglab/__init__.py
new file mode 100644
index 00000000000..8bce7be26e8
--- /dev/null
+++ b/homeassistant/components/pglab/__init__.py
@@ -0,0 +1,89 @@
+"""PG LAB Electronics integration."""
+
+from __future__ import annotations
+
+from pypglab.mqtt import (
+ Client as PyPGLabMqttClient,
+ Sub_State as PyPGLabSubState,
+ Subcribe_CallBack as PyPGLabSubscribeCallBack,
+)
+
+from homeassistant.components import mqtt
+from homeassistant.components.mqtt import (
+ ReceiveMessage,
+ async_prepare_subscribe_topics,
+ async_subscribe_topics,
+ async_unsubscribe_topics,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import ConfigEntryNotReady
+from homeassistant.helpers import config_validation as cv
+
+from .const import DOMAIN, LOGGER
+from .discovery import PGLabDiscovery
+
+type PGLabConfigEntry = ConfigEntry[PGLabDiscovery]
+
+CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant, config_entry: PGLabConfigEntry
+) -> bool:
+ """Set up PG LAB Electronics integration from a config entry."""
+
+ async def mqtt_publish(topic: str, payload: str, qos: int, retain: bool) -> None:
+ """Publish an MQTT message using the Home Assistant MQTT client."""
+ await mqtt.async_publish(hass, topic, payload, qos, retain)
+
+ async def mqtt_subscribe(
+ sub_state: PyPGLabSubState, topic: str, callback_func: PyPGLabSubscribeCallBack
+ ) -> PyPGLabSubState:
+ """Subscribe to MQTT topics using the Home Assistant MQTT client."""
+
+ @callback
+ def mqtt_message_received(msg: ReceiveMessage) -> None:
+ """Handle PGLab mqtt messages."""
+ callback_func(msg.topic, msg.payload)
+
+ topics = {
+ "pglab_subscribe_topic": {
+ "topic": topic,
+ "msg_callback": mqtt_message_received,
+ }
+ }
+
+ sub_state = async_prepare_subscribe_topics(hass, sub_state, topics)
+ await async_subscribe_topics(hass, sub_state)
+ return sub_state
+
+ async def mqtt_unsubscribe(sub_state: PyPGLabSubState) -> None:
+ async_unsubscribe_topics(hass, sub_state)
+
+ if not await mqtt.async_wait_for_mqtt_client(hass):
+ LOGGER.error("MQTT integration not available")
+ raise ConfigEntryNotReady("MQTT integration not available")
+
+ # Create an MQTT client for PGLab used for PGLab python module.
+ pglab_mqtt = PyPGLabMqttClient(mqtt_publish, mqtt_subscribe, mqtt_unsubscribe)
+
+ # Setup PGLab device discovery.
+ config_entry.runtime_data = PGLabDiscovery()
+
+ # Start to discovery PG Lab devices.
+ await config_entry.runtime_data.start(hass, pglab_mqtt, config_entry)
+
+ return True
+
+
+async def async_unload_entry(
+ hass: HomeAssistant, config_entry: PGLabConfigEntry
+) -> bool:
+ """Unload a config entry."""
+
+ # Stop PGLab device discovery.
+ pglab_discovery = config_entry.runtime_data
+ await pglab_discovery.stop(hass, config_entry)
+
+ return True
diff --git a/homeassistant/components/pglab/config_flow.py b/homeassistant/components/pglab/config_flow.py
new file mode 100644
index 00000000000..606de757622
--- /dev/null
+++ b/homeassistant/components/pglab/config_flow.py
@@ -0,0 +1,73 @@
+"""Config flow for PG LAB Electronics integration."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.components import mqtt
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.helpers.service_info.mqtt import MqttServiceInfo
+
+from .const import DISCOVERY_TOPIC, DOMAIN
+
+
+class PGLabFlowHandler(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow."""
+
+ VERSION = 1
+
+ async def async_step_mqtt(
+ self, discovery_info: MqttServiceInfo
+ ) -> ConfigFlowResult:
+ """Handle a flow initialized by MQTT discovery."""
+
+ await self.async_set_unique_id(DOMAIN)
+
+ # Validate the message, abort if it fails.
+ if not discovery_info.topic.endswith("/config"):
+ # Not a PGLab Electronics discovery message.
+ return self.async_abort(reason="invalid_discovery_info")
+ if not discovery_info.payload:
+ # Empty payload, unexpected payload.
+ return self.async_abort(reason="invalid_discovery_info")
+
+ return await self.async_step_confirm_from_mqtt()
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle a flow initialized by the user."""
+ try:
+ if not mqtt.is_connected(self.hass):
+ return self.async_abort(reason="mqtt_not_connected")
+ except KeyError:
+ return self.async_abort(reason="mqtt_not_configured")
+
+ return await self.async_step_confirm_from_user()
+
+ def step_confirm(
+ self, step_id: str, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Confirm the setup."""
+
+ if user_input is not None:
+ return self.async_create_entry(
+ title="PG LAB Electronics",
+ data={
+ "discovery_prefix": DISCOVERY_TOPIC,
+ },
+ )
+
+ return self.async_show_form(step_id=step_id)
+
+ async def async_step_confirm_from_mqtt(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Confirm the setup from MQTT discovered."""
+ return self.step_confirm(step_id="confirm_from_mqtt", user_input=user_input)
+
+ async def async_step_confirm_from_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Confirm the setup from user add integration."""
+ return self.step_confirm(step_id="confirm_from_user", user_input=user_input)
diff --git a/homeassistant/components/pglab/const.py b/homeassistant/components/pglab/const.py
new file mode 100644
index 00000000000..de076ac37f0
--- /dev/null
+++ b/homeassistant/components/pglab/const.py
@@ -0,0 +1,12 @@
+"""Constants used by PG LAB Electronics integration."""
+
+import logging
+
+# The domain of the integration.
+DOMAIN = "pglab"
+
+# The message logger.
+LOGGER = logging.getLogger(__package__)
+
+# The MQTT message used to subscribe to get a new PG LAB device.
+DISCOVERY_TOPIC = "pglab/discovery"
diff --git a/homeassistant/components/pglab/coordinator.py b/homeassistant/components/pglab/coordinator.py
new file mode 100644
index 00000000000..53c5dbc3b58
--- /dev/null
+++ b/homeassistant/components/pglab/coordinator.py
@@ -0,0 +1,78 @@
+"""Coordinator for PG LAB Electronics."""
+
+from __future__ import annotations
+
+from datetime import datetime, timedelta
+from typing import TYPE_CHECKING, Any
+
+from pypglab.const import SENSOR_REBOOT_TIME, SENSOR_TEMPERATURE, SENSOR_VOLTAGE
+from pypglab.device import Device as PyPGLabDevice
+from pypglab.sensor import Sensor as PyPGLabSensors
+
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
+from homeassistant.util.dt import utcnow
+
+from .const import DOMAIN, LOGGER
+
+if TYPE_CHECKING:
+ from . import PGLabConfigEntry
+
+
+class PGLabSensorsCoordinator(DataUpdateCoordinator[dict[str, Any]]):
+ """Class to update Sensor Entities when receiving new data."""
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: PGLabConfigEntry,
+ pglab_device: PyPGLabDevice,
+ ) -> None:
+ """Initialize."""
+
+ # get a reference of PG Lab device internal sensors state
+ self._sensors: PyPGLabSensors = pglab_device.sensors
+
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ )
+
+ @callback
+ def _new_sensors_data(self, payload: str) -> None:
+ """Handle new sensor data."""
+
+ # notify all listeners that new sensor values are available
+ self.async_set_updated_data(self._sensors.state)
+
+ async def subscribe_topics(self) -> None:
+ """Subscribe the sensors state to be notifty from MQTT update messages."""
+
+ # subscribe to the pypglab sensors to receive updates from the mqtt broker
+ # when a new sensor values are available
+ await self._sensors.subscribe_topics()
+
+ # set the callback to be called when a new sensor values are available
+ self._sensors.set_on_state_callback(self._new_sensors_data)
+
+ def get_sensor_value(self, sensor_key: str) -> float | datetime | None:
+ """Return the value of a sensor."""
+
+ if self.data:
+ value = self.data[sensor_key]
+
+ if (sensor_key == SENSOR_REBOOT_TIME) and value:
+ # convert the reboot time to a datetime object
+ return utcnow() - timedelta(seconds=value)
+
+ if (sensor_key == SENSOR_TEMPERATURE) and value:
+ # convert the temperature value to a float
+ return float(value)
+
+ if (sensor_key == SENSOR_VOLTAGE) and value:
+ # convert the voltage value to a float
+ return float(value)
+
+ return None
diff --git a/homeassistant/components/pglab/discovery.py b/homeassistant/components/pglab/discovery.py
new file mode 100644
index 00000000000..e34f80a2e2d
--- /dev/null
+++ b/homeassistant/components/pglab/discovery.py
@@ -0,0 +1,311 @@
+"""Discovery PG LAB Electronics devices."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass
+import json
+from typing import TYPE_CHECKING, Any
+
+from pypglab.device import Device as PyPGLabDevice
+from pypglab.mqtt import Client as PyPGLabMqttClient
+
+from homeassistant.components.mqtt import (
+ EntitySubscription,
+ ReceiveMessage,
+ async_prepare_subscribe_topics,
+ async_subscribe_topics,
+ async_unsubscribe_topics,
+)
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import device_registry as dr, entity_registry as er
+from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
+from homeassistant.helpers.dispatcher import (
+ async_dispatcher_connect,
+ async_dispatcher_send,
+)
+
+from .const import DISCOVERY_TOPIC, DOMAIN, LOGGER
+from .coordinator import PGLabSensorsCoordinator
+
+if TYPE_CHECKING:
+ from . import PGLabConfigEntry
+
+# Supported platforms.
+PLATFORMS = [
+ Platform.SENSOR,
+ Platform.SWITCH,
+]
+
+# Used to create a new component entity.
+CREATE_NEW_ENTITY = {
+ Platform.SENSOR: "pglab_create_new_entity_sensor",
+ Platform.SWITCH: "pglab_create_new_entity_switch",
+}
+
+
+class PGLabDiscoveryError(Exception):
+ """Raised when a discovery has failed."""
+
+
+def get_device_id_from_discovery_topic(topic: str) -> str | None:
+ """From the discovery topic get the PG LAB Electronics device id."""
+
+ # The discovery topic has the following format "pglab/discovery/[Device ID]/config"
+ split_topic = topic.split("/", 5)
+
+ # Do a sanity check on the string.
+ if len(split_topic) != 4:
+ return None
+
+ if split_topic[3] != "config":
+ return None
+
+ return split_topic[2]
+
+
+class DiscoverDeviceInfo:
+ """Keeps information of the PGLab discovered device."""
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: PGLabConfigEntry,
+ pglab_device: PyPGLabDevice,
+ ) -> None:
+ """Initialize the device discovery info."""
+
+ # Hash string represents the devices actual configuration,
+ # it depends on the number of available relays and shutters.
+ # When the hash string changes the devices entities must be rebuilt.
+ self._hash = pglab_device.hash
+ self._entities: list[tuple[str, str]] = []
+ self.coordinator = PGLabSensorsCoordinator(hass, config_entry, pglab_device)
+
+ def add_entity(self, platform_domain: str, entity_unique_id: str | None) -> None:
+ """Add an entity."""
+
+ # PGLabEntity always have unique IDs
+ if TYPE_CHECKING:
+ assert entity_unique_id is not None
+ self._entities.append((platform_domain, entity_unique_id))
+
+ @property
+ def hash(self) -> int:
+ """Return the hash for this configuration."""
+ return self._hash
+
+ @property
+ def entities(self) -> list[tuple[str, str]]:
+ """Return array of entities available."""
+ return self._entities
+
+
+async def create_discover_device_info(
+ hass: HomeAssistant, config_entry: PGLabConfigEntry, pglab_device: PyPGLabDevice
+) -> DiscoverDeviceInfo:
+ """Create a new DiscoverDeviceInfo instance."""
+ discovery_info = DiscoverDeviceInfo(hass, config_entry, pglab_device)
+
+ # Subscribe to sensor state changes.
+ await discovery_info.coordinator.subscribe_topics()
+ return discovery_info
+
+
+@dataclass
+class PGLabDiscovery:
+ """Discovery a PGLab device with the following MQTT topic format pglab/discovery/[device]/config."""
+
+ def __init__(self) -> None:
+ """Initialize the discovery class."""
+ self._substate: dict[str, EntitySubscription] = {}
+ self._discovery_topic = DISCOVERY_TOPIC
+ self._mqtt_client = None
+ self._discovered: dict[str, DiscoverDeviceInfo] = {}
+ self._disconnect_platform: list = []
+
+ async def __build_device(
+ self, mqtt: PyPGLabMqttClient, msg: ReceiveMessage
+ ) -> PyPGLabDevice:
+ """Build a PGLab device."""
+
+ # Check if the discovery message is in valid json format.
+ try:
+ payload = json.loads(msg.payload)
+ except ValueError as err:
+ raise PGLabDiscoveryError(
+ f"Can't decode discovery payload: {msg.payload!r}"
+ ) from err
+
+ device_id = "id"
+
+ # Check if the key id is present in the payload. It must always be present.
+ if device_id not in payload:
+ raise PGLabDiscoveryError(
+ "Unexpected discovery payload format, id key not present"
+ )
+
+ # Do a sanity check: the id must match the discovery topic /pglab/discovery/[id]/config
+ topic = msg.topic
+ if not topic.endswith(f"{payload[device_id]}/config"):
+ raise PGLabDiscoveryError("Unexpected discovery topic format")
+
+ # Build and configure the PGLab device.
+ pglab_device = PyPGLabDevice()
+ if not await pglab_device.config(mqtt, payload):
+ raise PGLabDiscoveryError("Error during setup of a new discovered device")
+
+ return pglab_device
+
+ def __clean_discovered_device(self, hass: HomeAssistant, device_id: str) -> None:
+ """Destroy the device and any entities connected to the device."""
+
+ if device_id not in self._discovered:
+ return
+
+ discovery_info = self._discovered[device_id]
+
+ # Destroy all entities connected to the device.
+ entity_registry = er.async_get(hass)
+ for platform, unique_id in discovery_info.entities:
+ if entity_id := entity_registry.async_get_entity_id(
+ platform, DOMAIN, unique_id
+ ):
+ entity_registry.async_remove(entity_id)
+
+ # Destroy the device.
+ device_registry = dr.async_get(hass)
+ if device_entry := device_registry.async_get_device(
+ identifiers={(DOMAIN, device_id)}
+ ):
+ device_registry.async_remove_device(device_entry.id)
+
+ # Clean the discovery info.
+ del self._discovered[device_id]
+
+ async def start(
+ self,
+ hass: HomeAssistant,
+ mqtt: PyPGLabMqttClient,
+ config_entry: PGLabConfigEntry,
+ ) -> None:
+ """Start discovering a PGLab devices."""
+
+ async def discovery_message_received(msg: ReceiveMessage) -> None:
+ """Received a new discovery message."""
+
+ # Create a PGLab device and add entities.
+ try:
+ pglab_device = await self.__build_device(mqtt, msg)
+ except PGLabDiscoveryError as err:
+ LOGGER.warning("Can't create PGLabDiscovery instance(%s) ", str(err))
+
+ # For some reason it's not possible to create the device with the discovery message,
+ # be sure that any previous device with the same topic is now destroyed.
+ device_id = get_device_id_from_discovery_topic(msg.topic)
+
+ # If there is a valid topic device_id clean everything relative to the device.
+ if device_id:
+ self.__clean_discovered_device(hass, device_id)
+
+ return
+
+ # Create a new device.
+ device_registry = dr.async_get(hass)
+ device_registry.async_get_or_create(
+ config_entry_id=config_entry.entry_id,
+ configuration_url=f"http://{pglab_device.ip}/",
+ connections={(CONNECTION_NETWORK_MAC, pglab_device.mac)},
+ identifiers={(DOMAIN, pglab_device.id)},
+ manufacturer=pglab_device.manufactor,
+ model=pglab_device.type,
+ name=pglab_device.name,
+ sw_version=pglab_device.firmware_version,
+ hw_version=pglab_device.hardware_version,
+ )
+
+ # Do some checking if previous entities must be updated.
+ if pglab_device.id in self._discovered:
+ # The device is already been discovered,
+ # get the old discovery info data.
+ discovery_info = self._discovered[pglab_device.id]
+
+ if discovery_info.hash == pglab_device.hash:
+ # Best case, there is nothing to do.
+ # The device is still in the same configuration. Same name, same shutters, same relay etc.
+ return
+
+ LOGGER.warning(
+ "Changed internal configuration of device(%s). Rebuilding all entities",
+ pglab_device.id,
+ )
+
+ # Something has changed, all previous entities must be destroyed and re-created.
+ self.__clean_discovered_device(hass, pglab_device.id)
+
+ # Add a new device.
+ discovery_info = await create_discover_device_info(
+ hass, config_entry, pglab_device
+ )
+ self._discovered[pglab_device.id] = discovery_info
+
+ # Create all new relay entities.
+ for r in pglab_device.relays:
+ # The HA entity is not yet created, send a message to create it.
+ async_dispatcher_send(
+ hass, CREATE_NEW_ENTITY[Platform.SWITCH], pglab_device, r
+ )
+
+ # Create all new sensor entities.
+ async_dispatcher_send(
+ hass,
+ CREATE_NEW_ENTITY[Platform.SENSOR],
+ pglab_device,
+ discovery_info.coordinator,
+ )
+
+ topics = {
+ "discovery_topic": {
+ "topic": f"{self._discovery_topic}/#",
+ "msg_callback": discovery_message_received,
+ }
+ }
+
+ # Forward setup all HA supported platforms.
+ await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
+
+ self._mqtt_client = mqtt
+ self._substate = async_prepare_subscribe_topics(hass, self._substate, topics)
+ await async_subscribe_topics(hass, self._substate)
+
+ async def register_platform(
+ self, hass: HomeAssistant, platform: Platform, target: Callable[..., Any]
+ ):
+ """Register a callback to create entity of a specific HA platform."""
+ disconnect_callback = async_dispatcher_connect(
+ hass, CREATE_NEW_ENTITY[platform], target
+ )
+ self._disconnect_platform.append(disconnect_callback)
+
+ async def stop(self, hass: HomeAssistant, config_entry: PGLabConfigEntry) -> None:
+ """Stop to discovery PG LAB devices."""
+ await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
+
+ # Disconnect all registered platforms.
+ for disconnect_callback in self._disconnect_platform:
+ disconnect_callback()
+
+ async_unsubscribe_topics(hass, self._substate)
+
+ async def add_entity(
+ self, platform_domain: str, entity_unique_id: str | None, device_id: str
+ ):
+ """Save a new PG LAB device entity."""
+
+ # Be sure that the device is been discovered.
+ if device_id not in self._discovered:
+ raise PGLabDiscoveryError("Unknown device, device_id not discovered")
+
+ discovery_info = self._discovered[device_id]
+ discovery_info.add_entity(platform_domain, entity_unique_id)
diff --git a/homeassistant/components/pglab/entity.py b/homeassistant/components/pglab/entity.py
new file mode 100644
index 00000000000..59a4e28de89
--- /dev/null
+++ b/homeassistant/components/pglab/entity.py
@@ -0,0 +1,112 @@
+"""Entity for PG LAB Electronics."""
+
+from __future__ import annotations
+
+from pypglab.device import Device as PyPGLabDevice
+from pypglab.entity import Entity as PyPGLabEntity
+
+from homeassistant.core import callback
+from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
+from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+from .coordinator import PGLabSensorsCoordinator
+from .discovery import PGLabDiscovery
+
+
+class PGLabBaseEntity(Entity):
+ """Base class of a PGLab entity in Home Assistant."""
+
+ _attr_has_entity_name = True
+
+ def __init__(
+ self,
+ pglab_discovery: PGLabDiscovery,
+ pglab_device: PyPGLabDevice,
+ ) -> None:
+ """Initialize the class."""
+
+ self._device_id = pglab_device.id
+ self._discovery = pglab_discovery
+
+ # Information about the device that is partially visible in the UI.
+ self._attr_device_info = DeviceInfo(
+ identifiers={(DOMAIN, pglab_device.id)},
+ name=pglab_device.name,
+ sw_version=pglab_device.firmware_version,
+ hw_version=pglab_device.hardware_version,
+ model=pglab_device.type,
+ manufacturer=pglab_device.manufactor,
+ configuration_url=f"http://{pglab_device.ip}/",
+ connections={(CONNECTION_NETWORK_MAC, pglab_device.mac)},
+ )
+
+ async def async_added_to_hass(self) -> None:
+ """Update the device discovery info."""
+
+ # Inform PGLab discovery instance that a new entity is available.
+ # This is important to know in case the device needs to be reconfigured
+ # and the entity can be potentially destroyed.
+ await self._discovery.add_entity(
+ self.platform.domain,
+ self.unique_id,
+ self._device_id,
+ )
+
+ # propagate the async_added_to_hass to the super class
+ await super().async_added_to_hass()
+
+
+class PGLabEntity(PGLabBaseEntity):
+ """Representation of a PGLab entity in Home Assistant."""
+
+ def __init__(
+ self,
+ pglab_discovery: PGLabDiscovery,
+ pglab_device: PyPGLabDevice,
+ pglab_entity: PyPGLabEntity,
+ ) -> None:
+ """Initialize the class."""
+
+ super().__init__(pglab_discovery, pglab_device)
+
+ self._id = pglab_entity.id
+ self._entity: PyPGLabEntity = pglab_entity
+
+ async def async_added_to_hass(self) -> None:
+ """Subscribe pypglab entity to be updated from mqtt when pypglab entity internal state change."""
+
+ # set the callback to be called when pypglab entity state is changed
+ self._entity.set_on_state_callback(self.state_updated)
+
+ # subscribe to the pypglab entity to receive updates from the mqtt broker
+ await self._entity.subscribe_topics()
+ await super().async_added_to_hass()
+
+ async def async_will_remove_from_hass(self) -> None:
+ """Unsubscribe when removed."""
+
+ await super().async_will_remove_from_hass()
+ await self._entity.unsubscribe_topics()
+ self._entity.set_on_state_callback(None)
+
+ @callback
+ def state_updated(self, payload: str) -> None:
+ """Handle state updates."""
+ self.async_write_ha_state()
+
+
+class PGLabSensorEntity(PGLabBaseEntity, CoordinatorEntity[PGLabSensorsCoordinator]):
+ """Representation of a PGLab sensor entity in Home Assistant."""
+
+ def __init__(
+ self,
+ pglab_discovery: PGLabDiscovery,
+ pglab_device: PyPGLabDevice,
+ pglab_coordinator: PGLabSensorsCoordinator,
+ ) -> None:
+ """Initialize the class."""
+
+ PGLabBaseEntity.__init__(self, pglab_discovery, pglab_device)
+ CoordinatorEntity.__init__(self, pglab_coordinator)
diff --git a/homeassistant/components/pglab/manifest.json b/homeassistant/components/pglab/manifest.json
new file mode 100644
index 00000000000..7f7d596be77
--- /dev/null
+++ b/homeassistant/components/pglab/manifest.json
@@ -0,0 +1,14 @@
+{
+ "domain": "pglab",
+ "name": "PG LAB Electronics",
+ "codeowners": ["@pglab-electronics"],
+ "config_flow": true,
+ "dependencies": ["mqtt"],
+ "documentation": "https://www.home-assistant.io/integrations/pglab",
+ "iot_class": "local_push",
+ "loggers": ["pglab"],
+ "mqtt": ["pglab/discovery/#"],
+ "quality_scale": "bronze",
+ "requirements": ["pypglab==0.0.3"],
+ "single_config_entry": true
+}
diff --git a/homeassistant/components/pglab/quality_scale.yaml b/homeassistant/components/pglab/quality_scale.yaml
new file mode 100644
index 00000000000..dda637e5833
--- /dev/null
+++ b/homeassistant/components/pglab/quality_scale.yaml
@@ -0,0 +1,80 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: The integration does not provide any additional actions.
+ appropriate-polling:
+ status: exempt
+ comment: The integration does not poll.
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: The integration does not provide any additional actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup: done
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure:
+ status: exempt
+ comment: The integration relies solely on auto-discovery.
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions:
+ status: exempt
+ comment: The integration does not provide any additional actions.
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: No options flow.
+ docs-installation-parameters:
+ status: exempt
+ comment: There are no parameters.
+ entity-unavailable: todo
+ integration-owner: done
+ log-when-unavailable: todo
+ parallel-updates: done
+ reauthentication-flow:
+ status: exempt
+ comment: The integration does not require authentication.
+ test-coverage: todo
+
+ # Gold
+ devices: done
+ diagnostics: todo
+ discovery-update-info: done
+ discovery: done
+ docs-data-update: todo
+ docs-examples: todo
+ docs-known-limitations: todo
+ docs-supported-devices: todo
+ docs-supported-functions: done
+ docs-troubleshooting: todo
+ docs-use-cases: todo
+ dynamic-devices: done
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default: done
+ entity-translations: todo
+ exception-translations: todo
+ icon-translations: todo
+ reconfiguration-flow:
+ status: exempt
+ comment: The integration has no settings.
+ repair-issues: todo
+ stale-devices: todo
+
+ # Platinum
+ async-dependency: done
+ inject-websession:
+ status: exempt
+ comment: The integration does not make HTTP requests.
+ strict-typing: todo
diff --git a/homeassistant/components/pglab/sensor.py b/homeassistant/components/pglab/sensor.py
new file mode 100644
index 00000000000..ce19ec3a21a
--- /dev/null
+++ b/homeassistant/components/pglab/sensor.py
@@ -0,0 +1,109 @@
+"""Sensor for PG LAB Electronics."""
+
+from __future__ import annotations
+
+from pypglab.const import SENSOR_REBOOT_TIME, SENSOR_TEMPERATURE, SENSOR_VOLTAGE
+from pypglab.device import Device as PyPGLabDevice
+
+from homeassistant.components.sensor import (
+ SensorDeviceClass,
+ SensorEntity,
+ SensorEntityDescription,
+ SensorStateClass,
+)
+from homeassistant.const import Platform, UnitOfElectricPotential, UnitOfTemperature
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import PGLabConfigEntry
+from .coordinator import PGLabSensorsCoordinator
+from .discovery import PGLabDiscovery
+from .entity import PGLabSensorEntity
+
+PARALLEL_UPDATES = 0
+
+SENSOR_INFO: list[SensorEntityDescription] = [
+ SensorEntityDescription(
+ key=SENSOR_TEMPERATURE,
+ native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=SENSOR_VOLTAGE,
+ translation_key="mpu_voltage",
+ native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+ device_class=SensorDeviceClass.VOLTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=SENSOR_REBOOT_TIME,
+ translation_key="runtime",
+ device_class=SensorDeviceClass.TIMESTAMP,
+ icon="mdi:progress-clock",
+ ),
+]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: PGLabConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up sensor for device."""
+
+ @callback
+ def async_discover(
+ pglab_device: PyPGLabDevice,
+ pglab_coordinator: PGLabSensorsCoordinator,
+ ) -> None:
+ """Discover and add a PG LAB Sensor."""
+ pglab_discovery = config_entry.runtime_data
+
+ sensors: list[PGLabSensor] = [
+ PGLabSensor(
+ description,
+ pglab_discovery,
+ pglab_device,
+ pglab_coordinator,
+ )
+ for description in SENSOR_INFO
+ ]
+
+ async_add_entities(sensors)
+
+ # Register the callback to create the sensor entity when discovered.
+ pglab_discovery = config_entry.runtime_data
+ await pglab_discovery.register_platform(hass, Platform.SENSOR, async_discover)
+
+
+class PGLabSensor(PGLabSensorEntity, SensorEntity):
+ """A PGLab sensor."""
+
+ def __init__(
+ self,
+ description: SensorEntityDescription,
+ pglab_discovery: PGLabDiscovery,
+ pglab_device: PyPGLabDevice,
+ pglab_coordinator: PGLabSensorsCoordinator,
+ ) -> None:
+ """Initialize the Sensor class."""
+
+ super().__init__(pglab_discovery, pglab_device, pglab_coordinator)
+
+ self._attr_unique_id = f"{pglab_device.id}_{description.key}"
+ self.entity_description = description
+
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Update attributes when the coordinator updates."""
+
+ self._attr_native_value = self.coordinator.get_sensor_value(
+ self.entity_description.key
+ )
+ super()._handle_coordinator_update()
+
+ @property
+ def available(self) -> bool:
+ """Return PG LAB sensor availability."""
+ return super().available and self.native_value is not None
diff --git a/homeassistant/components/pglab/strings.json b/homeassistant/components/pglab/strings.json
new file mode 100644
index 00000000000..4fad408ad98
--- /dev/null
+++ b/homeassistant/components/pglab/strings.json
@@ -0,0 +1,35 @@
+{
+ "config": {
+ "step": {
+ "confirm_from_user": {
+ "description": "In order to be found PG LAB Electronics devices need to be connected to the same broker as the Home Assistant MQTT integration client. Do you want to continue?"
+ },
+ "confirm_from_mqtt": {
+ "description": "Do you want to set up PG LAB Electronics?"
+ }
+ },
+ "abort": {
+ "single_instance_allowed": "[%key:common::config_flow::abort::single_instance_allowed%]",
+ "mqtt_not_connected": "Home Assistant MQTT integration not connected to MQTT broker.",
+ "mqtt_not_configured": "Home Assistant MQTT integration not configured."
+ }
+ },
+ "entity": {
+ "switch": {
+ "relay": {
+ "name": "Relay {relay_id}"
+ }
+ },
+ "sensor": {
+ "temperature": {
+ "name": "Temperature"
+ },
+ "runtime": {
+ "name": "Run time"
+ },
+ "mpu_voltage": {
+ "name": "MPU voltage"
+ }
+ }
+ }
+}
diff --git a/homeassistant/components/pglab/switch.py b/homeassistant/components/pglab/switch.py
new file mode 100644
index 00000000000..76b177e84c4
--- /dev/null
+++ b/homeassistant/components/pglab/switch.py
@@ -0,0 +1,76 @@
+"""Switch for PG LAB Electronics."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from pypglab.device import Device as PyPGLabDevice
+from pypglab.relay import Relay as PyPGLabRelay
+
+from homeassistant.components.switch import SwitchEntity
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import PGLabConfigEntry
+from .discovery import PGLabDiscovery
+from .entity import PGLabEntity
+
+PARALLEL_UPDATES = 0
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: PGLabConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up switches for device."""
+
+ @callback
+ def async_discover(pglab_device: PyPGLabDevice, pglab_relay: PyPGLabRelay) -> None:
+ """Discover and add a PGLab Relay."""
+ pglab_discovery = config_entry.runtime_data
+ pglab_switch = PGLabSwitch(pglab_discovery, pglab_device, pglab_relay)
+ async_add_entities([pglab_switch])
+
+ # Register the callback to create the switch entity when discovered.
+ pglab_discovery = config_entry.runtime_data
+ await pglab_discovery.register_platform(hass, Platform.SWITCH, async_discover)
+
+
+class PGLabSwitch(PGLabEntity, SwitchEntity):
+ """A PGLab switch."""
+
+ _attr_translation_key = "relay"
+
+ def __init__(
+ self,
+ pglab_discovery: PGLabDiscovery,
+ pglab_device: PyPGLabDevice,
+ pglab_relay: PyPGLabRelay,
+ ) -> None:
+ """Initialize the Switch class."""
+
+ super().__init__(
+ pglab_discovery,
+ pglab_device,
+ pglab_relay,
+ )
+
+ self._attr_unique_id = f"{pglab_device.id}_relay{pglab_relay.id}"
+ self._attr_translation_placeholders = {"relay_id": pglab_relay.id}
+
+ self._relay = pglab_relay
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the device on."""
+ await self._relay.turn_on()
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the device off."""
+ await self._relay.turn_off()
+
+ @property
+ def is_on(self) -> bool:
+ """Return true if device is on."""
+ return self._relay.state
diff --git a/homeassistant/components/philips_js/__init__.py b/homeassistant/components/philips_js/__init__.py
index 93f869e849d..9ff101915b8 100644
--- a/homeassistant/components/philips_js/__init__.py
+++ b/homeassistant/components/philips_js/__init__.py
@@ -7,7 +7,6 @@ import logging
from haphilipsjs import PhilipsTV
from haphilipsjs.typing import SystemType
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_API_VERSION,
CONF_HOST,
@@ -18,7 +17,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from .const import CONF_SYSTEM
-from .coordinator import PhilipsTVDataUpdateCoordinator
+from .coordinator import PhilipsTVConfigEntry, PhilipsTVDataUpdateCoordinator
PLATFORMS = [
Platform.BINARY_SENSOR,
@@ -30,8 +29,6 @@ PLATFORMS = [
LOGGER = logging.getLogger(__name__)
-PhilipsTVConfigEntry = ConfigEntry[PhilipsTVDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: PhilipsTVConfigEntry) -> bool:
"""Set up Philips TV from a config entry."""
@@ -44,7 +41,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: PhilipsTVConfigEntry) ->
password=entry.data.get(CONF_PASSWORD),
system=system,
)
- coordinator = PhilipsTVDataUpdateCoordinator(hass, tvapi, entry.options)
+ coordinator = PhilipsTVDataUpdateCoordinator(hass, entry, tvapi)
await coordinator.async_refresh()
diff --git a/homeassistant/components/philips_js/binary_sensor.py b/homeassistant/components/philips_js/binary_sensor.py
index 6de814efd97..3667d37dc48 100644
--- a/homeassistant/components/philips_js/binary_sensor.py
+++ b/homeassistant/components/philips_js/binary_sensor.py
@@ -11,10 +11,9 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PhilipsTVConfigEntry
-from .coordinator import PhilipsTVDataUpdateCoordinator
+from .coordinator import PhilipsTVConfigEntry, PhilipsTVDataUpdateCoordinator
from .entity import PhilipsJsEntity
@@ -42,7 +41,7 @@ DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PhilipsTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the configuration entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/philips_js/coordinator.py b/homeassistant/components/philips_js/coordinator.py
index cae59fa5123..f450e971093 100644
--- a/homeassistant/components/philips_js/coordinator.py
+++ b/homeassistant/components/philips_js/coordinator.py
@@ -3,10 +3,8 @@
from __future__ import annotations
import asyncio
-from collections.abc import Mapping
from datetime import timedelta
import logging
-from typing import Any
from haphilipsjs import (
AutenticationFailure,
@@ -27,23 +25,28 @@ from .const import CONF_ALLOW_NOTIFY, CONF_SYSTEM, DOMAIN
_LOGGER = logging.getLogger(__name__)
+type PhilipsTVConfigEntry = ConfigEntry[PhilipsTVDataUpdateCoordinator]
+
class PhilipsTVDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Coordinator to update data."""
- config_entry: ConfigEntry
+ config_entry: PhilipsTVConfigEntry
def __init__(
- self, hass: HomeAssistant, api: PhilipsTV, options: Mapping[str, Any]
+ self,
+ hass: HomeAssistant,
+ config_entry: PhilipsTVConfigEntry,
+ api: PhilipsTV,
) -> None:
"""Set up the coordinator."""
self.api = api
- self.options = options
self._notify_future: asyncio.Task | None = None
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
request_refresh_debouncer=Debouncer(
@@ -91,7 +94,7 @@ class PhilipsTVDataUpdateCoordinator(DataUpdateCoordinator[None]):
self.api.on
and self.api.powerstate == "On"
and self.api.notify_change_supported
- and self.options.get(CONF_ALLOW_NOTIFY, False)
+ and self.config_entry.options.get(CONF_ALLOW_NOTIFY, False)
)
async def _notify_task(self):
diff --git a/homeassistant/components/philips_js/diagnostics.py b/homeassistant/components/philips_js/diagnostics.py
index 625b77f6c25..99b27b2c85a 100644
--- a/homeassistant/components/philips_js/diagnostics.py
+++ b/homeassistant/components/philips_js/diagnostics.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant
-from . import PhilipsTVConfigEntry
+from .coordinator import PhilipsTVConfigEntry
TO_REDACT = {
"serialnumber_encrypted",
diff --git a/homeassistant/components/philips_js/light.py b/homeassistant/components/philips_js/light.py
index 1d63b2062e6..bf15292335e 100644
--- a/homeassistant/components/philips_js/light.py
+++ b/homeassistant/components/philips_js/light.py
@@ -18,11 +18,10 @@ from homeassistant.components.light import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.color import color_hsv_to_RGB, color_RGB_to_hsv
-from . import PhilipsTVConfigEntry
-from .coordinator import PhilipsTVDataUpdateCoordinator
+from .coordinator import PhilipsTVConfigEntry, PhilipsTVDataUpdateCoordinator
from .entity import PhilipsJsEntity
EFFECT_PARTITION = ": "
@@ -35,7 +34,7 @@ EFFECT_EXPERT_STYLES = {"FOLLOW_AUDIO", "FOLLOW_COLOR", "Lounge light"}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PhilipsTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the configuration entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/philips_js/media_player.py b/homeassistant/components/philips_js/media_player.py
index bd8727ae9c1..a433a63f31f 100644
--- a/homeassistant/components/philips_js/media_player.py
+++ b/homeassistant/components/philips_js/media_player.py
@@ -18,11 +18,11 @@ from homeassistant.components.media_player import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.trigger import PluggableAction
-from . import LOGGER as _LOGGER, PhilipsTVConfigEntry
-from .coordinator import PhilipsTVDataUpdateCoordinator
+from . import LOGGER as _LOGGER
+from .coordinator import PhilipsTVConfigEntry, PhilipsTVDataUpdateCoordinator
from .entity import PhilipsJsEntity
from .helpers import async_get_turn_on_trigger
@@ -49,7 +49,7 @@ def _inverted(data):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PhilipsTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the configuration entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/philips_js/remote.py b/homeassistant/components/philips_js/remote.py
index f8d9cb0885d..b026b33a857 100644
--- a/homeassistant/components/philips_js/remote.py
+++ b/homeassistant/components/philips_js/remote.py
@@ -13,11 +13,11 @@ from homeassistant.components.remote import (
RemoteEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.trigger import PluggableAction
-from . import LOGGER, PhilipsTVConfigEntry
-from .coordinator import PhilipsTVDataUpdateCoordinator
+from . import LOGGER
+from .coordinator import PhilipsTVConfigEntry, PhilipsTVDataUpdateCoordinator
from .entity import PhilipsJsEntity
from .helpers import async_get_turn_on_trigger
@@ -25,7 +25,7 @@ from .helpers import async_get_turn_on_trigger
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PhilipsTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the configuration entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/philips_js/switch.py b/homeassistant/components/philips_js/switch.py
index b35b2ad4ff1..45963432665 100644
--- a/homeassistant/components/philips_js/switch.py
+++ b/homeassistant/components/philips_js/switch.py
@@ -6,10 +6,9 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PhilipsTVConfigEntry
-from .coordinator import PhilipsTVDataUpdateCoordinator
+from .coordinator import PhilipsTVConfigEntry, PhilipsTVDataUpdateCoordinator
from .entity import PhilipsJsEntity
HUE_POWER_OFF = "Off"
@@ -19,7 +18,7 @@ HUE_POWER_ON = "On"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PhilipsTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the configuration entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/pi_hole/binary_sensor.py b/homeassistant/components/pi_hole/binary_sensor.py
index 5e3ce560ab4..1d12307b6e5 100644
--- a/homeassistant/components/pi_hole/binary_sensor.py
+++ b/homeassistant/components/pi_hole/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from . import PiHoleConfigEntry
@@ -41,7 +41,7 @@ BINARY_SENSOR_TYPES: tuple[PiHoleBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PiHoleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Pi-hole binary sensor."""
name = entry.data[CONF_NAME]
diff --git a/homeassistant/components/pi_hole/sensor.py b/homeassistant/components/pi_hole/sensor.py
index 4cf5133e700..54a9cb23d02 100644
--- a/homeassistant/components/pi_hole/sensor.py
+++ b/homeassistant/components/pi_hole/sensor.py
@@ -7,7 +7,7 @@ from hole import Hole
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.const import CONF_NAME, PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -47,7 +47,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PiHoleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Pi-hole sensor."""
name = entry.data[CONF_NAME]
diff --git a/homeassistant/components/pi_hole/switch.py b/homeassistant/components/pi_hole/switch.py
index 805ba479a9e..84ffe7e51a4 100644
--- a/homeassistant/components/pi_hole/switch.py
+++ b/homeassistant/components/pi_hole/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import PiHoleConfigEntry
from .const import SERVICE_DISABLE, SERVICE_DISABLE_ATTR_DURATION
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: PiHoleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Pi-hole switch."""
name = entry.data[CONF_NAME]
diff --git a/homeassistant/components/pi_hole/update.py b/homeassistant/components/pi_hole/update.py
index 510f5d1dc19..56e92b47289 100644
--- a/homeassistant/components/pi_hole/update.py
+++ b/homeassistant/components/pi_hole/update.py
@@ -10,7 +10,7 @@ from hole import Hole
from homeassistant.components.update import UpdateEntity, UpdateEntityDescription
from homeassistant.const import CONF_NAME, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from . import PiHoleConfigEntry
@@ -65,7 +65,7 @@ UPDATE_ENTITY_TYPES: tuple[PiHoleUpdateEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PiHoleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Pi-hole update entities."""
name = entry.data[CONF_NAME]
diff --git a/homeassistant/components/picnic/__init__.py b/homeassistant/components/picnic/__init__.py
index d2f023af79f..8de407133cd 100644
--- a/homeassistant/components/picnic/__init__.py
+++ b/homeassistant/components/picnic/__init__.py
@@ -1,6 +1,6 @@
"""The Picnic integration."""
-from python_picnic_api import PicnicAPI
+from python_picnic_api2 import PicnicAPI
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_COUNTRY_CODE, Platform
diff --git a/homeassistant/components/picnic/config_flow.py b/homeassistant/components/picnic/config_flow.py
index 4c8281f21de..a60086173a8 100644
--- a/homeassistant/components/picnic/config_flow.py
+++ b/homeassistant/components/picnic/config_flow.py
@@ -6,8 +6,8 @@ from collections.abc import Mapping
import logging
from typing import Any
-from python_picnic_api import PicnicAPI
-from python_picnic_api.session import PicnicAuthError
+from python_picnic_api2 import PicnicAPI
+from python_picnic_api2.session import PicnicAuthError
import requests
import voluptuous as vol
diff --git a/homeassistant/components/picnic/coordinator.py b/homeassistant/components/picnic/coordinator.py
index de686cad37d..9b23157dbf3 100644
--- a/homeassistant/components/picnic/coordinator.py
+++ b/homeassistant/components/picnic/coordinator.py
@@ -6,8 +6,8 @@ import copy
from datetime import timedelta
import logging
-from python_picnic_api import PicnicAPI
-from python_picnic_api.session import PicnicAuthError
+from python_picnic_api2 import PicnicAPI
+from python_picnic_api2.session import PicnicAuthError
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN
diff --git a/homeassistant/components/picnic/manifest.json b/homeassistant/components/picnic/manifest.json
index 947dd0241d2..09f28da39a4 100644
--- a/homeassistant/components/picnic/manifest.json
+++ b/homeassistant/components/picnic/manifest.json
@@ -1,10 +1,10 @@
{
"domain": "picnic",
"name": "Picnic",
- "codeowners": ["@corneyl"],
+ "codeowners": ["@corneyl", "@codesalatdev"],
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/picnic",
"iot_class": "cloud_polling",
- "loggers": ["python_picnic_api"],
- "requirements": ["python-picnic-api==1.1.0"]
+ "loggers": ["python_picnic_api2"],
+ "requirements": ["python-picnic-api2==1.2.2"]
}
diff --git a/homeassistant/components/picnic/sensor.py b/homeassistant/components/picnic/sensor.py
index 866bd6b56c1..dcfd9086491 100644
--- a/homeassistant/components/picnic/sensor.py
+++ b/homeassistant/components/picnic/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CURRENCY_EURO
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -203,7 +203,7 @@ SENSOR_TYPES: tuple[PicnicSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Picnic sensor entries."""
picnic_coordinator = hass.data[DOMAIN][config_entry.entry_id][CONF_COORDINATOR]
diff --git a/homeassistant/components/picnic/services.py b/homeassistant/components/picnic/services.py
index bbc775891b7..76d7b8a6c44 100644
--- a/homeassistant/components/picnic/services.py
+++ b/homeassistant/components/picnic/services.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from typing import cast
-from python_picnic_api import PicnicAPI
+from python_picnic_api2 import PicnicAPI
import voluptuous as vol
from homeassistant.core import HomeAssistant, ServiceCall
diff --git a/homeassistant/components/picnic/todo.py b/homeassistant/components/picnic/todo.py
index 7fa2bbccd3e..383c236de3c 100644
--- a/homeassistant/components/picnic/todo.py
+++ b/homeassistant/components/picnic/todo.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_COORDINATOR, DOMAIN
@@ -28,7 +28,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Picnic shopping cart todo platform config entry."""
picnic_coordinator = hass.data[DOMAIN][config_entry.entry_id][CONF_COORDINATOR]
diff --git a/homeassistant/components/pilight/entity.py b/homeassistant/components/pilight/entity.py
index fbb924d7f8f..fbfa5cfb5e1 100644
--- a/homeassistant/components/pilight/entity.py
+++ b/homeassistant/components/pilight/entity.py
@@ -86,7 +86,7 @@ class PilightBaseDevice(RestoreEntity):
self._brightness = 255
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Call when entity about to be added to hass."""
await super().async_added_to_hass()
if state := await self.async_get_last_state():
@@ -99,7 +99,7 @@ class PilightBaseDevice(RestoreEntity):
return self._name
@property
- def assumed_state(self):
+ def assumed_state(self) -> bool:
"""Return True if unable to access real state of the entity."""
return True
diff --git a/homeassistant/components/ping/__init__.py b/homeassistant/components/ping/__init__.py
index 4b03e5e4407..14203541359 100644
--- a/homeassistant/components/ping/__init__.py
+++ b/homeassistant/components/ping/__init__.py
@@ -6,7 +6,6 @@ import logging
from icmplib import SocketPermissionError, async_ping
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
@@ -14,7 +13,7 @@ from homeassistant.helpers.typing import ConfigType
from homeassistant.util.hass_dict import HassKey
from .const import CONF_PING_COUNT, DOMAIN
-from .coordinator import PingUpdateCoordinator
+from .coordinator import PingConfigEntry, PingUpdateCoordinator
from .helpers import PingDataICMPLib, PingDataSubProcess
_LOGGER = logging.getLogger(__name__)
@@ -24,9 +23,6 @@ PLATFORMS = [Platform.BINARY_SENSOR, Platform.DEVICE_TRACKER, Platform.SENSOR]
DATA_PRIVILEGED_KEY: HassKey[bool | None] = HassKey(DOMAIN)
-type PingConfigEntry = ConfigEntry[PingUpdateCoordinator]
-
-
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the ping integration."""
hass.data[DATA_PRIVILEGED_KEY] = await _can_use_icmp_lib_with_privilege()
@@ -47,7 +43,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: PingConfigEntry) -> bool
ping_cls = PingDataICMPLib
coordinator = PingUpdateCoordinator(
- hass=hass, ping=ping_cls(hass, host, count, privileged)
+ hass=hass, config_entry=entry, ping=ping_cls(hass, host, count, privileged)
)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/ping/binary_sensor.py b/homeassistant/components/ping/binary_sensor.py
index 5c50e4335f9..35bf2707694 100644
--- a/homeassistant/components/ping/binary_sensor.py
+++ b/homeassistant/components/ping/binary_sensor.py
@@ -6,18 +6,18 @@ from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
BinarySensorEntity,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PingConfigEntry
from .const import CONF_IMPORTED_BY
-from .coordinator import PingUpdateCoordinator
+from .coordinator import PingConfigEntry, PingUpdateCoordinator
from .entity import PingEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: PingConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: PingConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Ping config entry."""
async_add_entities([PingBinarySensor(entry, entry.runtime_data)])
@@ -31,7 +31,7 @@ class PingBinarySensor(PingEntity, BinarySensorEntity):
_attr_name = None
def __init__(
- self, config_entry: ConfigEntry, coordinator: PingUpdateCoordinator
+ self, config_entry: PingConfigEntry, coordinator: PingUpdateCoordinator
) -> None:
"""Initialize the Ping Binary sensor."""
super().__init__(config_entry, coordinator, config_entry.entry_id)
diff --git a/homeassistant/components/ping/coordinator.py b/homeassistant/components/ping/coordinator.py
index 38ab2e79ffc..afb7de4dce3 100644
--- a/homeassistant/components/ping/coordinator.py
+++ b/homeassistant/components/ping/coordinator.py
@@ -7,6 +7,7 @@ from datetime import timedelta
import logging
from typing import Any
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -14,6 +15,8 @@ from .helpers import PingDataICMPLib, PingDataSubProcess
_LOGGER = logging.getLogger(__name__)
+type PingConfigEntry = ConfigEntry[PingUpdateCoordinator]
+
@dataclass(slots=True, frozen=True)
class PingResult:
@@ -27,11 +30,13 @@ class PingResult:
class PingUpdateCoordinator(DataUpdateCoordinator[PingResult]):
"""The Ping update coordinator."""
+ config_entry: PingConfigEntry
ping: PingDataSubProcess | PingDataICMPLib
def __init__(
self,
hass: HomeAssistant,
+ config_entry: PingConfigEntry,
ping: PingDataSubProcess | PingDataICMPLib,
) -> None:
"""Initialize the Ping coordinator."""
@@ -40,6 +45,7 @@ class PingUpdateCoordinator(DataUpdateCoordinator[PingResult]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"Ping {ping.ip_address}",
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/ping/device_tracker.py b/homeassistant/components/ping/device_tracker.py
index 29a4e922234..9d093da262d 100644
--- a/homeassistant/components/ping/device_tracker.py
+++ b/homeassistant/components/ping/device_tracker.py
@@ -9,19 +9,19 @@ from homeassistant.components.device_tracker import (
DEFAULT_CONSIDER_HOME,
ScannerEntity,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
-from . import PingConfigEntry
from .const import CONF_IMPORTED_BY
-from .coordinator import PingUpdateCoordinator
+from .coordinator import PingConfigEntry, PingUpdateCoordinator
async def async_setup_entry(
- hass: HomeAssistant, entry: PingConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: PingConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Ping config entry."""
async_add_entities([PingDeviceTracker(entry, entry.runtime_data)])
@@ -33,7 +33,7 @@ class PingDeviceTracker(CoordinatorEntity[PingUpdateCoordinator], ScannerEntity)
_last_seen: datetime | None = None
def __init__(
- self, config_entry: ConfigEntry, coordinator: PingUpdateCoordinator
+ self, config_entry: PingConfigEntry, coordinator: PingUpdateCoordinator
) -> None:
"""Initialize the Ping device tracker."""
super().__init__(coordinator)
diff --git a/homeassistant/components/ping/entity.py b/homeassistant/components/ping/entity.py
index a1f84f6ef32..d592ef6b549 100644
--- a/homeassistant/components/ping/entity.py
+++ b/homeassistant/components/ping/entity.py
@@ -1,11 +1,10 @@
"""Base entity for the Ping component."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .coordinator import PingUpdateCoordinator
+from .coordinator import PingConfigEntry, PingUpdateCoordinator
class PingEntity(CoordinatorEntity[PingUpdateCoordinator]):
@@ -15,7 +14,7 @@ class PingEntity(CoordinatorEntity[PingUpdateCoordinator]):
def __init__(
self,
- config_entry: ConfigEntry,
+ config_entry: PingConfigEntry,
coordinator: PingUpdateCoordinator,
unique_id: str,
) -> None:
diff --git a/homeassistant/components/ping/sensor.py b/homeassistant/components/ping/sensor.py
index 6e6c4cf2cde..82d88064e02 100644
--- a/homeassistant/components/ping/sensor.py
+++ b/homeassistant/components/ping/sensor.py
@@ -12,10 +12,9 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PingConfigEntry
-from .coordinator import PingResult, PingUpdateCoordinator
+from .coordinator import PingConfigEntry, PingResult, PingUpdateCoordinator
from .entity import PingEntity
@@ -76,7 +75,9 @@ SENSORS: tuple[PingSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: PingConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: PingConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ping sensors from config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/ping/strings.json b/homeassistant/components/ping/strings.json
index ef9f74b4207..c301a1b277d 100644
--- a/homeassistant/components/ping/strings.json
+++ b/homeassistant/components/ping/strings.json
@@ -2,16 +2,16 @@
"entity": {
"sensor": {
"round_trip_time_avg": {
- "name": "Round Trip Time Average"
+ "name": "Round-trip time average"
},
"round_trip_time_max": {
- "name": "Round Trip Time Maximum"
+ "name": "Round-trip time maximum"
},
"round_trip_time_mdev": {
- "name": "Round Trip Time Mean Deviation"
+ "name": "Round-trip time mean deviation"
},
"round_trip_time_min": {
- "name": "Round Trip Time Minimum"
+ "name": "Round-trip time minimum"
}
}
},
diff --git a/homeassistant/components/plaato/__init__.py b/homeassistant/components/plaato/__init__.py
index 6001a243a2d..14e757d4623 100644
--- a/homeassistant/components/plaato/__init__.py
+++ b/homeassistant/components/plaato/__init__.py
@@ -121,7 +121,9 @@ async def async_setup_coordinator(hass: HomeAssistant, entry: ConfigEntry):
else:
update_interval = timedelta(minutes=DEFAULT_SCAN_INTERVAL)
- coordinator = PlaatoCoordinator(hass, auth_token, device_type, update_interval)
+ coordinator = PlaatoCoordinator(
+ hass, entry, auth_token, device_type, update_interval
+ )
await coordinator.async_config_entry_first_refresh()
_set_entry_data(entry, hass, coordinator, auth_token)
diff --git a/homeassistant/components/plaato/binary_sensor.py b/homeassistant/components/plaato/binary_sensor.py
index 42019bbec9b..b71673aa1fd 100644
--- a/homeassistant/components/plaato/binary_sensor.py
+++ b/homeassistant/components/plaato/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_USE_WEBHOOK, COORDINATOR, DOMAIN
from .entity import PlaatoEntity
@@ -19,7 +19,7 @@ from .entity import PlaatoEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plaato from a config entry."""
diff --git a/homeassistant/components/plaato/coordinator.py b/homeassistant/components/plaato/coordinator.py
index 8d21f17880a..df360d50068 100644
--- a/homeassistant/components/plaato/coordinator.py
+++ b/homeassistant/components/plaato/coordinator.py
@@ -5,6 +5,7 @@ import logging
from pyplaato.plaato import Plaato, PlaatoDeviceType
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import aiohttp_client
@@ -18,9 +19,12 @@ _LOGGER = logging.getLogger(__name__)
class PlaatoCoordinator(DataUpdateCoordinator):
"""Class to manage fetching data from the API."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
auth_token: str,
device_type: PlaatoDeviceType,
update_interval: timedelta,
@@ -34,6 +38,7 @@ class PlaatoCoordinator(DataUpdateCoordinator):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=update_interval,
)
diff --git a/homeassistant/components/plaato/entity.py b/homeassistant/components/plaato/entity.py
index 7ab8367bd1d..9cc63a38a64 100644
--- a/homeassistant/components/plaato/entity.py
+++ b/homeassistant/components/plaato/entity.py
@@ -73,13 +73,13 @@ class PlaatoEntity(entity.Entity):
return None
@property
- def available(self):
+ def available(self) -> bool:
"""Return if sensor is available."""
if self._coordinator is not None:
return self._coordinator.last_update_success
return True
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""When entity is added to hass."""
if self._coordinator is not None:
self.async_on_remove(
diff --git a/homeassistant/components/plaato/sensor.py b/homeassistant/components/plaato/sensor.py
index b11bac40144..7a98c8a1ced 100644
--- a/homeassistant/components/plaato/sensor.py
+++ b/homeassistant/components/plaato/sensor.py
@@ -12,7 +12,10 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import ATTR_TEMP, SENSOR_UPDATE
@@ -38,7 +41,9 @@ async def async_setup_platform(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plaato from a config entry."""
entry_data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/plex/button.py b/homeassistant/components/plex/button.py
index 8bb34be38ce..5ed34eac6b2 100644
--- a/homeassistant/components/plex/button.py
+++ b/homeassistant/components/plex/button.py
@@ -8,7 +8,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_send
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import PlexServer
from .const import CONF_SERVER_IDENTIFIER, DOMAIN, PLEX_UPDATE_PLATFORMS_SIGNAL
@@ -18,7 +18,7 @@ from .helpers import get_plex_server
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plex button from config entry."""
server_id: str = config_entry.data[CONF_SERVER_IDENTIFIER]
diff --git a/homeassistant/components/plex/media_player.py b/homeassistant/components/plex/media_player.py
index 1dd79ad27a5..4a1654959f6 100644
--- a/homeassistant/components/plex/media_player.py
+++ b/homeassistant/components/plex/media_player.py
@@ -27,7 +27,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.network import is_internal_request
from .const import (
@@ -68,7 +68,7 @@ def needs_session[_PlexMediaPlayerT: PlexMediaPlayer, **_P, _R](
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plex media_player from a config entry."""
server_id = config_entry.data[CONF_SERVER_IDENTIFIER]
diff --git a/homeassistant/components/plex/sensor.py b/homeassistant/components/plex/sensor.py
index eb27f465a7e..66e513dd83a 100644
--- a/homeassistant/components/plex/sensor.py
+++ b/homeassistant/components/plex/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_SERVER_IDENTIFIER,
@@ -52,7 +52,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plex sensor from a config entry."""
server_id = config_entry.data[CONF_SERVER_IDENTIFIER]
diff --git a/homeassistant/components/plex/update.py b/homeassistant/components/plex/update.py
index 7acf4551f33..9b7645cd078 100644
--- a/homeassistant/components/plex/update.py
+++ b/homeassistant/components/plex/update.py
@@ -11,7 +11,7 @@ from homeassistant.components.update import UpdateEntity, UpdateEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_SERVER_IDENTIFIER
from .helpers import get_plex_server
@@ -22,7 +22,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plex update entities from a config entry."""
server_id = config_entry.data[CONF_SERVER_IDENTIFIER]
diff --git a/homeassistant/components/plugwise/__init__.py b/homeassistant/components/plugwise/__init__.py
index a100103b029..e97493a78a7 100644
--- a/homeassistant/components/plugwise/__init__.py
+++ b/homeassistant/components/plugwise/__init__.py
@@ -4,22 +4,19 @@ from __future__ import annotations
from typing import Any
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
from .const import DOMAIN, LOGGER, PLATFORMS
-from .coordinator import PlugwiseDataUpdateCoordinator
-
-type PlugwiseConfigEntry = ConfigEntry[PlugwiseDataUpdateCoordinator]
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: PlugwiseConfigEntry) -> bool:
"""Set up Plugwise components from a config entry."""
await er.async_migrate_entries(hass, entry.entry_id, async_migrate_entity_entry)
- coordinator = PlugwiseDataUpdateCoordinator(hass)
+ coordinator = PlugwiseDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
migrate_sensor_entities(hass, coordinator)
@@ -82,7 +79,7 @@ def migrate_sensor_entities(
# Migrating opentherm_outdoor_temperature
# to opentherm_outdoor_air_temperature sensor
- for device_id, device in coordinator.data.devices.items():
+ for device_id, device in coordinator.data.items():
if device["dev_class"] != "heater_central":
continue
diff --git a/homeassistant/components/plugwise/binary_sensor.py b/homeassistant/components/plugwise/binary_sensor.py
index 539fa243d6c..f2c2fd6ed68 100644
--- a/homeassistant/components/plugwise/binary_sensor.py
+++ b/homeassistant/components/plugwise/binary_sensor.py
@@ -15,10 +15,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PlugwiseConfigEntry
-from .coordinator import PlugwiseDataUpdateCoordinator
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
from .entity import PlugwiseEntity
SEVERITIES = ["other", "info", "warning", "error"]
@@ -86,7 +85,7 @@ BINARY_SENSORS: tuple[PlugwiseBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PlugwiseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smile binary_sensors from a config entry."""
coordinator = entry.runtime_data
@@ -100,11 +99,7 @@ async def async_setup_entry(
async_add_entities(
PlugwiseBinarySensorEntity(coordinator, device_id, description)
for device_id in coordinator.new_devices
- if (
- binary_sensors := coordinator.data.devices[device_id].get(
- "binary_sensors"
- )
- )
+ if (binary_sensors := coordinator.data[device_id].get("binary_sensors"))
for description in BINARY_SENSORS
if description.key in binary_sensors
)
@@ -141,7 +136,8 @@ class PlugwiseBinarySensorEntity(PlugwiseEntity, BinarySensorEntity):
return None
attrs: dict[str, list[str]] = {f"{severity}_msg": [] for severity in SEVERITIES}
- if notify := self.coordinator.data.gateway["notifications"]:
+ gateway_id = self.coordinator.api.gateway_id
+ if notify := self.coordinator.data[gateway_id]["notifications"]:
for details in notify.values():
for msg_type, msg in details.items():
msg_type = msg_type.lower()
diff --git a/homeassistant/components/plugwise/button.py b/homeassistant/components/plugwise/button.py
index 8a05ede3496..c0896b602f0 100644
--- a/homeassistant/components/plugwise/button.py
+++ b/homeassistant/components/plugwise/button.py
@@ -5,11 +5,10 @@ from __future__ import annotations
from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PlugwiseConfigEntry
-from .const import GATEWAY_ID, REBOOT
-from .coordinator import PlugwiseDataUpdateCoordinator
+from .const import REBOOT
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
from .entity import PlugwiseEntity
from .util import plugwise_command
@@ -19,16 +18,15 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: PlugwiseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Plugwise buttons from a ConfigEntry."""
coordinator = entry.runtime_data
- gateway = coordinator.data.gateway
async_add_entities(
PlugwiseButtonEntity(coordinator, device_id)
- for device_id in coordinator.data.devices
- if device_id == gateway[GATEWAY_ID] and REBOOT in gateway
+ for device_id in coordinator.data
+ if device_id == coordinator.api.gateway_id and coordinator.api.reboot
)
diff --git a/homeassistant/components/plugwise/climate.py b/homeassistant/components/plugwise/climate.py
index 3caed1e7bc2..c7fac07f1cb 100644
--- a/homeassistant/components/plugwise/climate.py
+++ b/homeassistant/components/plugwise/climate.py
@@ -16,11 +16,10 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PlugwiseConfigEntry
from .const import DOMAIN, MASTER_THERMOSTATS
-from .coordinator import PlugwiseDataUpdateCoordinator
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
from .entity import PlugwiseEntity
from .util import plugwise_command
@@ -30,7 +29,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: PlugwiseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smile Thermostats from a config entry."""
coordinator = entry.runtime_data
@@ -41,18 +40,17 @@ async def async_setup_entry(
if not coordinator.new_devices:
return
- if coordinator.data.gateway["smile_name"] == "Adam":
+ if coordinator.api.smile_name == "Adam":
async_add_entities(
PlugwiseClimateEntity(coordinator, device_id)
for device_id in coordinator.new_devices
- if coordinator.data.devices[device_id]["dev_class"] == "climate"
+ if coordinator.data[device_id]["dev_class"] == "climate"
)
else:
async_add_entities(
PlugwiseClimateEntity(coordinator, device_id)
for device_id in coordinator.new_devices
- if coordinator.data.devices[device_id]["dev_class"]
- in MASTER_THERMOSTATS
+ if coordinator.data[device_id]["dev_class"] in MASTER_THERMOSTATS
)
_add_entities()
@@ -77,10 +75,8 @@ class PlugwiseClimateEntity(PlugwiseEntity, ClimateEntity):
super().__init__(coordinator, device_id)
self._attr_unique_id = f"{device_id}-climate"
- self._devices = coordinator.data.devices
- self._gateway = coordinator.data.gateway
- gateway_id: str = self._gateway["gateway_id"]
- self._gateway_data = self._devices[gateway_id]
+ gateway_id: str = coordinator.api.gateway_id
+ self._gateway_data = coordinator.data[gateway_id]
self._location = device_id
if (location := self.device.get("location")) is not None:
@@ -88,7 +84,10 @@ class PlugwiseClimateEntity(PlugwiseEntity, ClimateEntity):
# Determine supported features
self._attr_supported_features = ClimateEntityFeature.TARGET_TEMPERATURE
- if self._gateway["cooling_present"] and self._gateway["smile_name"] != "Adam":
+ if (
+ self.coordinator.api.cooling_present
+ and coordinator.api.smile_name != "Adam"
+ ):
self._attr_supported_features = (
ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
)
@@ -170,7 +169,7 @@ class PlugwiseClimateEntity(PlugwiseEntity, ClimateEntity):
if "available_schedules" in self.device:
hvac_modes.append(HVACMode.AUTO)
- if self._gateway["cooling_present"]:
+ if self.coordinator.api.cooling_present:
if "regulation_modes" in self._gateway_data:
if self._gateway_data["select_regulation_mode"] == "cooling":
hvac_modes.append(HVACMode.COOL)
diff --git a/homeassistant/components/plugwise/const.py b/homeassistant/components/plugwise/const.py
index 5e4dea5586b..176ae39b1ba 100644
--- a/homeassistant/components/plugwise/const.py
+++ b/homeassistant/components/plugwise/const.py
@@ -17,7 +17,6 @@ FLOW_SMILE: Final = "smile (Adam/Anna/P1)"
FLOW_STRETCH: Final = "stretch (Stretch)"
FLOW_TYPE: Final = "flow_type"
GATEWAY: Final = "gateway"
-GATEWAY_ID: Final = "gateway_id"
LOCATION: Final = "location"
PW_TYPE: Final = "plugwise_type"
REBOOT: Final = "reboot"
diff --git a/homeassistant/components/plugwise/coordinator.py b/homeassistant/components/plugwise/coordinator.py
index 7ac0cc21c51..b346f26492c 100644
--- a/homeassistant/components/plugwise/coordinator.py
+++ b/homeassistant/components/plugwise/coordinator.py
@@ -3,7 +3,7 @@
from datetime import timedelta
from packaging.version import Version
-from plugwise import PlugwiseData, Smile
+from plugwise import GwEntityData, Smile
from plugwise.exceptions import (
ConnectionFailedError,
InvalidAuthentication,
@@ -22,21 +22,24 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import DEFAULT_PORT, DEFAULT_USERNAME, DOMAIN, GATEWAY_ID, LOGGER
+from .const import DEFAULT_PORT, DEFAULT_USERNAME, DOMAIN, LOGGER
+
+type PlugwiseConfigEntry = ConfigEntry[PlugwiseDataUpdateCoordinator]
-class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[PlugwiseData]):
+class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[dict[str, GwEntityData]]):
"""Class to manage fetching Plugwise data from single endpoint."""
_connected: bool = False
- config_entry: ConfigEntry
+ config_entry: PlugwiseConfigEntry
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: PlugwiseConfigEntry) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=60),
# Don't refresh immediately, give the device time to process
@@ -63,10 +66,8 @@ class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[PlugwiseData]):
"""Connect to the Plugwise Smile."""
version = await self.api.connect()
self._connected = isinstance(version, Version)
- if self._connected:
- self.api.get_all_gateway_entities()
- async def _async_update_data(self) -> PlugwiseData:
+ async def _async_update_data(self) -> dict[str, GwEntityData]:
"""Fetch data from Plugwise."""
try:
if not self._connected:
@@ -101,26 +102,28 @@ class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[PlugwiseData]):
self._async_add_remove_devices(data, self.config_entry)
return data
- def _async_add_remove_devices(self, data: PlugwiseData, entry: ConfigEntry) -> None:
+ def _async_add_remove_devices(
+ self, data: dict[str, GwEntityData], entry: ConfigEntry
+ ) -> None:
"""Add new Plugwise devices, remove non-existing devices."""
# Check for new or removed devices
- self.new_devices = set(data.devices) - self._current_devices
- removed_devices = self._current_devices - set(data.devices)
- self._current_devices = set(data.devices)
+ self.new_devices = set(data) - self._current_devices
+ removed_devices = self._current_devices - set(data)
+ self._current_devices = set(data)
if removed_devices:
self._async_remove_devices(data, entry)
- def _async_remove_devices(self, data: PlugwiseData, entry: ConfigEntry) -> None:
+ def _async_remove_devices(
+ self, data: dict[str, GwEntityData], entry: ConfigEntry
+ ) -> None:
"""Clean registries when removed devices found."""
device_reg = dr.async_get(self.hass)
device_list = dr.async_entries_for_config_entry(
device_reg, self.config_entry.entry_id
)
# First find the Plugwise via_device
- gateway_device = device_reg.async_get_device(
- {(DOMAIN, data.gateway[GATEWAY_ID])}
- )
+ gateway_device = device_reg.async_get_device({(DOMAIN, self.api.gateway_id)})
assert gateway_device is not None
via_device_id = gateway_device.id
@@ -130,7 +133,7 @@ class PlugwiseDataUpdateCoordinator(DataUpdateCoordinator[PlugwiseData]):
if identifier[0] == DOMAIN:
if (
device_entry.via_device_id == via_device_id
- and identifier[1] not in data.devices
+ and identifier[1] not in data
):
device_reg.async_update_device(
device_entry.id, remove_config_entry_id=entry.entry_id
diff --git a/homeassistant/components/plugwise/diagnostics.py b/homeassistant/components/plugwise/diagnostics.py
index 47ff7d1a9fb..e97405f6279 100644
--- a/homeassistant/components/plugwise/diagnostics.py
+++ b/homeassistant/components/plugwise/diagnostics.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.core import HomeAssistant
-from . import PlugwiseConfigEntry
+from .coordinator import PlugwiseConfigEntry
async def async_get_config_entry_diagnostics(
@@ -14,7 +14,4 @@ async def async_get_config_entry_diagnostics(
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
coordinator = entry.runtime_data
- return {
- "devices": coordinator.data.devices,
- "gateway": coordinator.data.gateway,
- }
+ return coordinator.data
diff --git a/homeassistant/components/plugwise/entity.py b/homeassistant/components/plugwise/entity.py
index 3f63abaff43..39838c38fde 100644
--- a/homeassistant/components/plugwise/entity.py
+++ b/homeassistant/components/plugwise/entity.py
@@ -34,7 +34,7 @@ class PlugwiseEntity(CoordinatorEntity[PlugwiseDataUpdateCoordinator]):
if entry := self.coordinator.config_entry:
configuration_url = f"http://{entry.data[CONF_HOST]}"
- data = coordinator.data.devices[device_id]
+ data = coordinator.data[device_id]
connections = set()
if mac := data.get("mac_address"):
connections.add((CONNECTION_NETWORK_MAC, mac))
@@ -48,18 +48,18 @@ class PlugwiseEntity(CoordinatorEntity[PlugwiseDataUpdateCoordinator]):
manufacturer=data.get("vendor"),
model=data.get("model"),
model_id=data.get("model_id"),
- name=coordinator.data.gateway["smile_name"],
+ name=coordinator.api.smile_name,
sw_version=data.get("firmware"),
hw_version=data.get("hardware"),
)
- if device_id != coordinator.data.gateway["gateway_id"]:
+ if device_id != coordinator.api.gateway_id:
self._attr_device_info.update(
{
ATTR_NAME: data.get("name"),
ATTR_VIA_DEVICE: (
DOMAIN,
- str(self.coordinator.data.gateway["gateway_id"]),
+ str(self.coordinator.api.gateway_id),
),
}
)
@@ -68,7 +68,7 @@ class PlugwiseEntity(CoordinatorEntity[PlugwiseDataUpdateCoordinator]):
def available(self) -> bool:
"""Return if entity is available."""
return (
- self._dev_id in self.coordinator.data.devices
+ self._dev_id in self.coordinator.data
and ("available" not in self.device or self.device["available"] is True)
and super().available
)
@@ -76,4 +76,4 @@ class PlugwiseEntity(CoordinatorEntity[PlugwiseDataUpdateCoordinator]):
@property
def device(self) -> GwEntityData:
"""Return data for this device."""
- return self.coordinator.data.devices[self._dev_id]
+ return self.coordinator.data[self._dev_id]
diff --git a/homeassistant/components/plugwise/manifest.json b/homeassistant/components/plugwise/manifest.json
index f7bd646f801..87878980f2d 100644
--- a/homeassistant/components/plugwise/manifest.json
+++ b/homeassistant/components/plugwise/manifest.json
@@ -8,6 +8,6 @@
"iot_class": "local_polling",
"loggers": ["plugwise"],
"quality_scale": "platinum",
- "requirements": ["plugwise==1.6.4"],
+ "requirements": ["plugwise==1.7.2"],
"zeroconf": ["_plugwise._tcp.local."]
}
diff --git a/homeassistant/components/plugwise/number.py b/homeassistant/components/plugwise/number.py
index 1d0b1382c24..1dbb0506748 100644
--- a/homeassistant/components/plugwise/number.py
+++ b/homeassistant/components/plugwise/number.py
@@ -12,11 +12,10 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PlugwiseConfigEntry
from .const import NumberType
-from .coordinator import PlugwiseDataUpdateCoordinator
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
from .entity import PlugwiseEntity
from .util import plugwise_command
@@ -58,7 +57,7 @@ NUMBER_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PlugwiseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plugwise number platform."""
coordinator = entry.runtime_data
@@ -73,7 +72,7 @@ async def async_setup_entry(
PlugwiseNumberEntity(coordinator, device_id, description)
for device_id in coordinator.new_devices
for description in NUMBER_TYPES
- if description.key in coordinator.data.devices[device_id]
+ if description.key in coordinator.data[device_id]
)
_add_entities()
diff --git a/homeassistant/components/plugwise/select.py b/homeassistant/components/plugwise/select.py
index ff268d8eded..6ca1d4ce7a2 100644
--- a/homeassistant/components/plugwise/select.py
+++ b/homeassistant/components/plugwise/select.py
@@ -7,11 +7,10 @@ from dataclasses import dataclass
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PlugwiseConfigEntry
from .const import SelectOptionsType, SelectType
-from .coordinator import PlugwiseDataUpdateCoordinator
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
from .entity import PlugwiseEntity
from .util import plugwise_command
@@ -56,7 +55,7 @@ SELECT_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PlugwiseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smile selector from a config entry."""
coordinator = entry.runtime_data
@@ -71,7 +70,7 @@ async def async_setup_entry(
PlugwiseSelectEntity(coordinator, device_id, description)
for device_id in coordinator.new_devices
for description in SELECT_TYPES
- if description.options_key in coordinator.data.devices[device_id]
+ if description.options_key in coordinator.data[device_id]
)
_add_entities()
diff --git a/homeassistant/components/plugwise/sensor.py b/homeassistant/components/plugwise/sensor.py
index 14b42682376..7bd93e2ff84 100644
--- a/homeassistant/components/plugwise/sensor.py
+++ b/homeassistant/components/plugwise/sensor.py
@@ -25,10 +25,9 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PlugwiseConfigEntry
-from .coordinator import PlugwiseDataUpdateCoordinator
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
from .entity import PlugwiseEntity
# Coordinator is used to centralize the data updates
@@ -406,7 +405,7 @@ SENSORS: tuple[PlugwiseSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PlugwiseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smile sensors from a config entry."""
coordinator = entry.runtime_data
@@ -420,7 +419,7 @@ async def async_setup_entry(
async_add_entities(
PlugwiseSensorEntity(coordinator, device_id, description)
for device_id in coordinator.new_devices
- if (sensors := coordinator.data.devices[device_id].get("sensors"))
+ if (sensors := coordinator.data[device_id].get("sensors"))
for description in SENSORS
if description.key in sensors
)
diff --git a/homeassistant/components/plugwise/switch.py b/homeassistant/components/plugwise/switch.py
index ea6d6f18b7f..8179fb546b4 100644
--- a/homeassistant/components/plugwise/switch.py
+++ b/homeassistant/components/plugwise/switch.py
@@ -14,10 +14,9 @@ from homeassistant.components.switch import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PlugwiseConfigEntry
-from .coordinator import PlugwiseDataUpdateCoordinator
+from .coordinator import PlugwiseConfigEntry, PlugwiseDataUpdateCoordinator
from .entity import PlugwiseEntity
from .util import plugwise_command
@@ -58,7 +57,7 @@ SWITCHES: tuple[PlugwiseSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PlugwiseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smile switches from a config entry."""
coordinator = entry.runtime_data
@@ -72,7 +71,7 @@ async def async_setup_entry(
async_add_entities(
PlugwiseSwitchEntity(coordinator, device_id, description)
for device_id in coordinator.new_devices
- if (switches := coordinator.data.devices[device_id].get("switches"))
+ if (switches := coordinator.data[device_id].get("switches"))
for description in SWITCHES
if description.key in switches
)
diff --git a/homeassistant/components/plum_lightpad/light.py b/homeassistant/components/plum_lightpad/light.py
index 08a3d0ab0b9..78743c12808 100644
--- a/homeassistant/components/plum_lightpad/light.py
+++ b/homeassistant/components/plum_lightpad/light.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .const import DOMAIN
@@ -25,7 +25,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Plum Lightpad dimmer lights and glow rings."""
diff --git a/homeassistant/components/point/alarm_control_panel.py b/homeassistant/components/point/alarm_control_panel.py
index 4e4e4238176..0f501d2ee09 100644
--- a/homeassistant/components/point/alarm_control_panel.py
+++ b/homeassistant/components/point/alarm_control_panel.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MinutPointClient
from .const import DOMAIN as POINT_DOMAIN, POINT_DISCOVERY_NEW, SIGNAL_WEBHOOK
@@ -33,7 +33,7 @@ EVENT_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Point's alarm_control_panel based on a config entry."""
diff --git a/homeassistant/components/point/binary_sensor.py b/homeassistant/components/point/binary_sensor.py
index 546c7d9cb0f..c9338cb63f2 100644
--- a/homeassistant/components/point/binary_sensor.py
+++ b/homeassistant/components/point/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN as POINT_DOMAIN, POINT_DISCOVERY_NEW, SIGNAL_WEBHOOK
from .entity import MinutPointEntity
@@ -43,7 +43,7 @@ DEVICES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Point's binary sensors based on a config entry."""
diff --git a/homeassistant/components/point/config_flow.py b/homeassistant/components/point/config_flow.py
index a0a51c7b9e6..b26ade8b725 100644
--- a/homeassistant/components/point/config_flow.py
+++ b/homeassistant/components/point/config_flow.py
@@ -11,6 +11,8 @@ from homeassistant.helpers.config_entry_oauth2_flow import AbstractOAuth2FlowHan
from .const import DOMAIN
+_LOGGER = logging.getLogger(__name__)
+
class OAuth2FlowHandler(AbstractOAuth2FlowHandler, domain=DOMAIN):
"""Config flow to handle Minut Point OAuth2 authentication."""
@@ -56,7 +58,7 @@ class OAuth2FlowHandler(AbstractOAuth2FlowHandler, domain=DOMAIN):
if reauth_entry.unique_id is not None:
self._abort_if_unique_id_mismatch(reason="wrong_account")
- logging.debug("user_id: %s", user_id)
+ _LOGGER.debug("user_id: %s", user_id)
return self.async_update_reload_and_abort(
reauth_entry, data_updates=data, unique_id=user_id
)
diff --git a/homeassistant/components/point/entity.py b/homeassistant/components/point/entity.py
index 4784dd43180..5c52e81e6f7 100644
--- a/homeassistant/components/point/entity.py
+++ b/homeassistant/components/point/entity.py
@@ -52,7 +52,7 @@ class MinutPointEntity(Entity):
)
await self._update_callback()
- async def async_will_remove_from_hass(self):
+ async def async_will_remove_from_hass(self) -> None:
"""Disconnect dispatcher listener when removed."""
if self._async_unsub_dispatcher_connect:
self._async_unsub_dispatcher_connect()
@@ -61,7 +61,7 @@ class MinutPointEntity(Entity):
"""Update the value of the sensor."""
@property
- def available(self):
+ def available(self) -> bool:
"""Return true if device is not offline."""
return self._client.is_available(self.device_id)
diff --git a/homeassistant/components/point/sensor.py b/homeassistant/components/point/sensor.py
index d864c8bb18c..c959d09d606 100644
--- a/homeassistant/components/point/sensor.py
+++ b/homeassistant/components/point/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfSoundPressure, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import parse_datetime
from .const import DOMAIN as POINT_DOMAIN, POINT_DISCOVERY_NEW
@@ -48,7 +48,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Point's sensors based on a config entry."""
diff --git a/homeassistant/components/poolsense/__init__.py b/homeassistant/components/poolsense/__init__.py
index a4b6f7b60d8..a2e54712566 100644
--- a/homeassistant/components/poolsense/__init__.py
+++ b/homeassistant/components/poolsense/__init__.py
@@ -4,14 +4,11 @@ import logging
from poolsense import PoolSense
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import aiohttp_client
-from .coordinator import PoolSenseDataUpdateCoordinator
-
-type PoolSenseConfigEntry = ConfigEntry[PoolSenseDataUpdateCoordinator]
+from .coordinator import PoolSenseConfigEntry, PoolSenseDataUpdateCoordinator
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR]
@@ -33,7 +30,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: PoolSenseConfigEntry) ->
_LOGGER.error("Invalid authentication")
return False
- coordinator = PoolSenseDataUpdateCoordinator(hass, poolsense)
+ coordinator = PoolSenseDataUpdateCoordinator(hass, entry, poolsense)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/poolsense/binary_sensor.py b/homeassistant/components/poolsense/binary_sensor.py
index 7668845f318..b93f017501d 100644
--- a/homeassistant/components/poolsense/binary_sensor.py
+++ b/homeassistant/components/poolsense/binary_sensor.py
@@ -8,9 +8,9 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PoolSenseConfigEntry
+from .coordinator import PoolSenseConfigEntry
from .entity import PoolSenseEntity
BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
@@ -30,7 +30,7 @@ BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PoolSenseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/poolsense/coordinator.py b/homeassistant/components/poolsense/coordinator.py
index d9e7e8468ff..557686f9145 100644
--- a/homeassistant/components/poolsense/coordinator.py
+++ b/homeassistant/components/poolsense/coordinator.py
@@ -5,11 +5,11 @@ from __future__ import annotations
import asyncio
from datetime import timedelta
import logging
-from typing import TYPE_CHECKING
from poolsense import PoolSense
from poolsense.exceptions import PoolSenseError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL
from homeassistant.core import HomeAssistant
from homeassistant.helpers.typing import StateType
@@ -17,20 +17,30 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN
-if TYPE_CHECKING:
- from . import PoolSenseConfigEntry
-
_LOGGER = logging.getLogger(__name__)
+type PoolSenseConfigEntry = ConfigEntry[PoolSenseDataUpdateCoordinator]
+
class PoolSenseDataUpdateCoordinator(DataUpdateCoordinator[dict[str, StateType]]):
"""Define an object to hold PoolSense data."""
config_entry: PoolSenseConfigEntry
- def __init__(self, hass: HomeAssistant, poolsense: PoolSense) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: PoolSenseConfigEntry,
+ poolsense: PoolSense,
+ ) -> None:
"""Initialize."""
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=timedelta(hours=1))
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=timedelta(hours=1),
+ )
self.poolsense = poolsense
self.email = self.config_entry.data[CONF_EMAIL]
diff --git a/homeassistant/components/poolsense/sensor.py b/homeassistant/components/poolsense/sensor.py
index 8cfb982d33b..b0ac4404237 100644
--- a/homeassistant/components/poolsense/sensor.py
+++ b/homeassistant/components/poolsense/sensor.py
@@ -9,10 +9,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfElectricPotential, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import PoolSenseConfigEntry
+from .coordinator import PoolSenseConfigEntry
from .entity import PoolSenseEntity
SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
@@ -65,7 +65,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: PoolSenseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Defer sensor setup to the shared sensor module."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/powerfox/__init__.py b/homeassistant/components/powerfox/__init__.py
index 243f3aacc4f..8e51985211d 100644
--- a/homeassistant/components/powerfox/__init__.py
+++ b/homeassistant/components/powerfox/__init__.py
@@ -6,18 +6,15 @@ import asyncio
from powerfox import Powerfox, PowerfoxConnectionError
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .coordinator import PowerfoxDataUpdateCoordinator
+from .coordinator import PowerfoxConfigEntry, PowerfoxDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type PowerfoxConfigEntry = ConfigEntry[list[PowerfoxDataUpdateCoordinator]]
-
async def async_setup_entry(hass: HomeAssistant, entry: PowerfoxConfigEntry) -> bool:
"""Set up Powerfox from a config entry."""
@@ -34,7 +31,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: PowerfoxConfigEntry) ->
raise ConfigEntryNotReady from err
coordinators: list[PowerfoxDataUpdateCoordinator] = [
- PowerfoxDataUpdateCoordinator(hass, client, device) for device in devices
+ PowerfoxDataUpdateCoordinator(hass, entry, client, device) for device in devices
]
await asyncio.gather(
diff --git a/homeassistant/components/powerfox/coordinator.py b/homeassistant/components/powerfox/coordinator.py
index a4a26759b69..bd76b7cc166 100644
--- a/homeassistant/components/powerfox/coordinator.py
+++ b/homeassistant/components/powerfox/coordinator.py
@@ -18,15 +18,18 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN, LOGGER, SCAN_INTERVAL
+type PowerfoxConfigEntry = ConfigEntry[list[PowerfoxDataUpdateCoordinator]]
+
class PowerfoxDataUpdateCoordinator(DataUpdateCoordinator[Poweropti]):
"""Class to manage fetching Powerfox data from the API."""
- config_entry: ConfigEntry
+ config_entry: PowerfoxConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: PowerfoxConfigEntry,
client: Powerfox,
device: Device,
) -> None:
@@ -34,6 +37,7 @@ class PowerfoxDataUpdateCoordinator(DataUpdateCoordinator[Poweropti]):
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/powerfox/diagnostics.py b/homeassistant/components/powerfox/diagnostics.py
index 4c6b0f8c6eb..8514e42537e 100644
--- a/homeassistant/components/powerfox/diagnostics.py
+++ b/homeassistant/components/powerfox/diagnostics.py
@@ -9,7 +9,7 @@ from powerfox import HeatMeter, PowerMeter, WaterMeter
from homeassistant.core import HomeAssistant
-from . import PowerfoxConfigEntry, PowerfoxDataUpdateCoordinator
+from .coordinator import PowerfoxConfigEntry, PowerfoxDataUpdateCoordinator
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/powerfox/sensor.py b/homeassistant/components/powerfox/sensor.py
index 6505139fcd9..ab60c99a58b 100644
--- a/homeassistant/components/powerfox/sensor.py
+++ b/homeassistant/components/powerfox/sensor.py
@@ -15,10 +15,9 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfEnergy, UnitOfPower, UnitOfVolume
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PowerfoxConfigEntry
-from .coordinator import PowerfoxDataUpdateCoordinator
+from .coordinator import PowerfoxConfigEntry, PowerfoxDataUpdateCoordinator
from .entity import PowerfoxEntity
@@ -131,7 +130,7 @@ SENSORS_HEAT: tuple[PowerfoxSensorEntityDescription[HeatMeter], ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerfoxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Powerfox sensors based on a config entry."""
entities: list[SensorEntity] = []
diff --git a/homeassistant/components/powerwall/binary_sensor.py b/homeassistant/components/powerwall/binary_sensor.py
index c50876e22fb..100e31b1c21 100644
--- a/homeassistant/components/powerwall/binary_sensor.py
+++ b/homeassistant/components/powerwall/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import PowerWallEntity
from .models import PowerwallConfigEntry
@@ -23,7 +23,7 @@ CONNECTED_GRID_STATUSES = {
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerwallConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the powerwall sensors."""
powerwall_data = entry.runtime_data
diff --git a/homeassistant/components/powerwall/sensor.py b/homeassistant/components/powerwall/sensor.py
index 28506e2a60c..b4988133727 100644
--- a/homeassistant/components/powerwall/sensor.py
+++ b/homeassistant/components/powerwall/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import POWERWALL_COORDINATOR
from .entity import BatteryEntity, PowerWallEntity
@@ -213,7 +213,7 @@ BATTERY_INSTANT_SENSORS: list[PowerwallSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerwallConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the powerwall sensors."""
powerwall_data = entry.runtime_data
diff --git a/homeassistant/components/powerwall/switch.py b/homeassistant/components/powerwall/switch.py
index 214ca01fb63..a874161de5b 100644
--- a/homeassistant/components/powerwall/switch.py
+++ b/homeassistant/components/powerwall/switch.py
@@ -8,7 +8,7 @@ from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import PowerWallEntity
from .models import PowerwallConfigEntry, PowerwallRuntimeData
@@ -22,7 +22,7 @@ OFF_GRID_STATUSES = {
async def async_setup_entry(
hass: HomeAssistant,
entry: PowerwallConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Powerwall switch platform from Powerwall resources."""
async_add_entities([PowerwallOffGridEnabledEntity(entry.runtime_data)])
diff --git a/homeassistant/components/private_ble_device/device_tracker.py b/homeassistant/components/private_ble_device/device_tracker.py
index fbaf0d44751..eaccbd6c785 100644
--- a/homeassistant/components/private_ble_device/device_tracker.py
+++ b/homeassistant/components/private_ble_device/device_tracker.py
@@ -11,7 +11,7 @@ from homeassistant.components.device_tracker.config_entry import BaseTrackerEnti
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_HOME, STATE_NOT_HOME
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import BasePrivateDeviceEntity
@@ -21,7 +21,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Device Tracker entities for a config entry."""
async_add_entities([BasePrivateDeviceTracker(config_entry)])
diff --git a/homeassistant/components/private_ble_device/manifest.json b/homeassistant/components/private_ble_device/manifest.json
index 445affbcd57..810fce41e05 100644
--- a/homeassistant/components/private_ble_device/manifest.json
+++ b/homeassistant/components/private_ble_device/manifest.json
@@ -6,5 +6,5 @@
"dependencies": ["bluetooth_adapters"],
"documentation": "https://www.home-assistant.io/integrations/private_ble_device",
"iot_class": "local_push",
- "requirements": ["bluetooth-data-tools==1.23.4"]
+ "requirements": ["bluetooth-data-tools==1.26.1"]
}
diff --git a/homeassistant/components/private_ble_device/sensor.py b/homeassistant/components/private_ble_device/sensor.py
index e2c4fb0c7da..d8c09500332 100644
--- a/homeassistant/components/private_ble_device/sensor.py
+++ b/homeassistant/components/private_ble_device/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import BasePrivateDeviceEntity
@@ -92,7 +92,9 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for Private BLE component."""
async_add_entities(
diff --git a/homeassistant/components/progettihwsw/binary_sensor.py b/homeassistant/components/progettihwsw/binary_sensor.py
index a89b8b3c3f1..40296dcac90 100644
--- a/homeassistant/components/progettihwsw/binary_sensor.py
+++ b/homeassistant/components/progettihwsw/binary_sensor.py
@@ -9,7 +9,7 @@ from ProgettiHWSW.input import Input
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(DOMAIN)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary sensors from a config entry."""
board_api = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/progettihwsw/switch.py b/homeassistant/components/progettihwsw/switch.py
index 983a2383e99..256d90ae5b7 100644
--- a/homeassistant/components/progettihwsw/switch.py
+++ b/homeassistant/components/progettihwsw/switch.py
@@ -10,7 +10,7 @@ from ProgettiHWSW.relay import Relay
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -25,7 +25,7 @@ _LOGGER = logging.getLogger(DOMAIN)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switches from a config entry."""
board_api = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/prosegur/alarm_control_panel.py b/homeassistant/components/prosegur/alarm_control_panel.py
index 1c58b64cf55..1f0f89c5f04 100644
--- a/homeassistant/components/prosegur/alarm_control_panel.py
+++ b/homeassistant/components/prosegur/alarm_control_panel.py
@@ -15,7 +15,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
@@ -30,7 +30,9 @@ STATE_MAPPING = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Prosegur alarm control panel platform."""
async_add_entities(
diff --git a/homeassistant/components/prosegur/camera.py b/homeassistant/components/prosegur/camera.py
index 2df6ff62038..3e1c91713e1 100644
--- a/homeassistant/components/prosegur/camera.py
+++ b/homeassistant/components/prosegur/camera.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
@@ -24,7 +24,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Prosegur camera platform."""
diff --git a/homeassistant/components/prosegur/manifest.json b/homeassistant/components/prosegur/manifest.json
index adf5e985fe9..2e649ebd5bd 100644
--- a/homeassistant/components/prosegur/manifest.json
+++ b/homeassistant/components/prosegur/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/prosegur",
"iot_class": "cloud_polling",
"loggers": ["pyprosegur"],
- "requirements": ["pyprosegur==0.0.9"]
+ "requirements": ["pyprosegur==0.0.14"]
}
diff --git a/homeassistant/components/proximity/coordinator.py b/homeassistant/components/proximity/coordinator.py
index 055c15125f1..856138c9051 100644
--- a/homeassistant/components/proximity/coordinator.py
+++ b/homeassistant/components/proximity/coordinator.py
@@ -164,7 +164,7 @@ class ProximityDataUpdateCoordinator(DataUpdateCoordinator[ProximityData]):
)
return None
- distance_to_zone = distance(
+ distance_to_centre = distance(
zone.attributes[ATTR_LATITUDE],
zone.attributes[ATTR_LONGITUDE],
latitude,
@@ -172,8 +172,13 @@ class ProximityDataUpdateCoordinator(DataUpdateCoordinator[ProximityData]):
)
# it is ensured, that distance can't be None, since zones must have lat/lon coordinates
- assert distance_to_zone is not None
- return round(distance_to_zone)
+ assert distance_to_centre is not None
+
+ zone_radius: float = zone.attributes["radius"]
+ if zone_radius > distance_to_centre:
+ # we've arrived the zone
+ return 0
+ return round(distance_to_centre - zone_radius)
def _calc_direction_of_travel(
self,
diff --git a/homeassistant/components/proximity/sensor.py b/homeassistant/components/proximity/sensor.py
index 55d4ca02b9b..72203a2dff4 100644
--- a/homeassistant/components/proximity/sensor.py
+++ b/homeassistant/components/proximity/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.const import UnitOfLength
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -82,7 +82,7 @@ def _device_info(coordinator: ProximityDataUpdateCoordinator) -> DeviceInfo:
async def async_setup_entry(
hass: HomeAssistant,
entry: ProximityConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the proximity sensors."""
diff --git a/homeassistant/components/proximity/strings.json b/homeassistant/components/proximity/strings.json
index 118004e908e..5f713174f50 100644
--- a/homeassistant/components/proximity/strings.json
+++ b/homeassistant/components/proximity/strings.json
@@ -61,7 +61,7 @@
"step": {
"confirm": {
"title": "[%key:component::proximity::issues::tracked_entity_removed::title%]",
- "description": "The entity `{entity_id}` has been removed from HA, but is used in proximity {name}. Please remove `{entity_id}` from the list of tracked entities. Related proximity sensor entites were set to unavailable and can be removed."
+ "description": "The entity `{entity_id}` has been removed from HA, but is used in proximity {name}. Please remove `{entity_id}` from the list of tracked entities. Related proximity sensor entities were set to unavailable and can be removed."
}
}
}
diff --git a/homeassistant/components/prusalink/__init__.py b/homeassistant/components/prusalink/__init__.py
index 1415e3dd0a6..4bb7dee411d 100644
--- a/homeassistant/components/prusalink/__init__.py
+++ b/homeassistant/components/prusalink/__init__.py
@@ -24,6 +24,7 @@ from .coordinator import (
InfoUpdateCoordinator,
JobUpdateCoordinator,
LegacyStatusCoordinator,
+ PrusaLinkUpdateCoordinator,
StatusCoordinator,
)
@@ -47,11 +48,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
entry.data[CONF_PASSWORD],
)
- coordinators = {
- "legacy_status": LegacyStatusCoordinator(hass, api),
- "status": StatusCoordinator(hass, api),
- "job": JobUpdateCoordinator(hass, api),
- "info": InfoUpdateCoordinator(hass, api),
+ coordinators: dict[str, PrusaLinkUpdateCoordinator] = {
+ "legacy_status": LegacyStatusCoordinator(hass, entry, api),
+ "status": StatusCoordinator(hass, entry, api),
+ "job": JobUpdateCoordinator(hass, entry, api),
+ "info": InfoUpdateCoordinator(hass, entry, api),
}
for coordinator in coordinators.values():
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/prusalink/binary_sensor.py b/homeassistant/components/prusalink/binary_sensor.py
index d40ac8a4cfa..56be36c3e9d 100644
--- a/homeassistant/components/prusalink/binary_sensor.py
+++ b/homeassistant/components/prusalink/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import PrusaLinkUpdateCoordinator
@@ -57,7 +57,7 @@ BINARY_SENSORS: dict[str, tuple[PrusaLinkBinarySensorEntityDescription, ...]] =
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up PrusaLink sensor based on a config entry."""
coordinators: dict[str, PrusaLinkUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/prusalink/button.py b/homeassistant/components/prusalink/button.py
index 06d356b2ca6..59a63d874ee 100644
--- a/homeassistant/components/prusalink/button.py
+++ b/homeassistant/components/prusalink/button.py
@@ -13,7 +13,7 @@ from homeassistant.components.button import ButtonEntity, ButtonEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import PrusaLinkUpdateCoordinator
@@ -72,7 +72,7 @@ BUTTONS: dict[str, tuple[PrusaLinkButtonEntityDescription, ...]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up PrusaLink buttons based on a config entry."""
coordinators: dict[str, PrusaLinkUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/prusalink/camera.py b/homeassistant/components/prusalink/camera.py
index eee655447cc..6aac03ca179 100644
--- a/homeassistant/components/prusalink/camera.py
+++ b/homeassistant/components/prusalink/camera.py
@@ -7,7 +7,7 @@ from pyprusalink.types import PrinterState
from homeassistant.components.camera import Camera
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import JobUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import PrusaLinkEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up PrusaLink camera."""
coordinator: JobUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]["job"]
diff --git a/homeassistant/components/prusalink/coordinator.py b/homeassistant/components/prusalink/coordinator.py
index 1d887983931..e6f54bc6fa5 100644
--- a/homeassistant/components/prusalink/coordinator.py
+++ b/homeassistant/components/prusalink/coordinator.py
@@ -37,12 +37,18 @@ class PrusaLinkUpdateCoordinator(DataUpdateCoordinator[T], ABC):
config_entry: ConfigEntry
expect_change_until = 0.0
- def __init__(self, hass: HomeAssistant, api: PrusaLink) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: PrusaLink
+ ) -> None:
"""Initialize the update coordinator."""
self.api = api
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=self._get_update_interval(None)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=self._get_update_interval(None),
)
async def _async_update_data(self) -> T:
diff --git a/homeassistant/components/prusalink/sensor.py b/homeassistant/components/prusalink/sensor.py
index 0c746adbe2e..b9588f72a3c 100644
--- a/homeassistant/components/prusalink/sensor.py
+++ b/homeassistant/components/prusalink/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
from homeassistant.util.variance import ignore_variance
@@ -205,7 +205,7 @@ SENSORS: dict[str, tuple[PrusaLinkSensorEntityDescription, ...]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up PrusaLink sensor based on a config entry."""
coordinators: dict[str, PrusaLinkUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/ps4/media_player.py b/homeassistant/components/ps4/media_player.py
index 8db24beae20..4de7cbeb463 100644
--- a/homeassistant/components/ps4/media_player.py
+++ b/homeassistant/components/ps4/media_player.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.json import JsonObjectType
from . import format_unique_id, load_games, save_games
@@ -48,7 +48,7 @@ DEFAULT_RETRIES = 2
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up PS4 from a config entry."""
config = config_entry
diff --git a/homeassistant/components/pure_energie/__init__.py b/homeassistant/components/pure_energie/__init__.py
index 4de1ce02810..4ece35a3f1c 100644
--- a/homeassistant/components/pure_energie/__init__.py
+++ b/homeassistant/components/pure_energie/__init__.py
@@ -2,22 +2,19 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
-from .coordinator import PureEnergieDataUpdateCoordinator
+from .coordinator import PureEnergieConfigEntry, PureEnergieDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type PureEnergieConfigEntry = ConfigEntry[PureEnergieDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: PureEnergieConfigEntry) -> bool:
"""Set up Pure Energie from a config entry."""
- coordinator = PureEnergieDataUpdateCoordinator(hass)
+ coordinator = PureEnergieDataUpdateCoordinator(hass, entry)
try:
await coordinator.async_config_entry_first_refresh()
except ConfigEntryNotReady:
diff --git a/homeassistant/components/pure_energie/coordinator.py b/homeassistant/components/pure_energie/coordinator.py
index fdd848eb4c6..cd66ab060eb 100644
--- a/homeassistant/components/pure_energie/coordinator.py
+++ b/homeassistant/components/pure_energie/coordinator.py
@@ -14,6 +14,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN, LOGGER, SCAN_INTERVAL
+type PureEnergieConfigEntry = ConfigEntry[PureEnergieDataUpdateCoordinator]
+
class PureEnergieData(NamedTuple):
"""Class for defining data in dict."""
@@ -25,16 +27,18 @@ class PureEnergieData(NamedTuple):
class PureEnergieDataUpdateCoordinator(DataUpdateCoordinator[PureEnergieData]):
"""Class to manage fetching Pure Energie data from single eindpoint."""
- config_entry: ConfigEntry
+ config_entry: PureEnergieConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: PureEnergieConfigEntry,
) -> None:
"""Initialize global Pure Energie data updater."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/pure_energie/diagnostics.py b/homeassistant/components/pure_energie/diagnostics.py
index de9134129ed..5098a298e85 100644
--- a/homeassistant/components/pure_energie/diagnostics.py
+++ b/homeassistant/components/pure_energie/diagnostics.py
@@ -9,7 +9,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
-from . import PureEnergieConfigEntry
+from .coordinator import PureEnergieConfigEntry
TO_REDACT = {
CONF_HOST,
diff --git a/homeassistant/components/pure_energie/sensor.py b/homeassistant/components/pure_energie/sensor.py
index 468858f117f..ad57206adeb 100644
--- a/homeassistant/components/pure_energie/sensor.py
+++ b/homeassistant/components/pure_energie/sensor.py
@@ -15,12 +15,15 @@ from homeassistant.components.sensor import (
from homeassistant.const import CONF_HOST, UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import PureEnergieConfigEntry
from .const import DOMAIN
-from .coordinator import PureEnergieData, PureEnergieDataUpdateCoordinator
+from .coordinator import (
+ PureEnergieConfigEntry,
+ PureEnergieData,
+ PureEnergieDataUpdateCoordinator,
+)
@dataclass(frozen=True, kw_only=True)
@@ -61,7 +64,7 @@ SENSORS: tuple[PureEnergieSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PureEnergieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Pure Energie Sensors based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/purpleair/__init__.py b/homeassistant/components/purpleair/__init__.py
index 2d4022946b2..78986b34351 100644
--- a/homeassistant/components/purpleair/__init__.py
+++ b/homeassistant/components/purpleair/__init__.py
@@ -2,37 +2,34 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator
-
-PLATFORMS = [Platform.SENSOR]
+from .const import PLATFORMS
+from .coordinator import PurpleAirConfigEntry, PurpleAirDataUpdateCoordinator
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
- """Set up PurpleAir from a config entry."""
- coordinator = PurpleAirDataUpdateCoordinator(hass, entry)
+async def async_setup_entry(hass: HomeAssistant, entry: PurpleAirConfigEntry) -> bool:
+ """Set up PurpleAir config entry."""
+ coordinator = PurpleAirDataUpdateCoordinator(
+ hass,
+ entry,
+ )
+ entry.runtime_data = coordinator
+
await coordinator.async_config_entry_first_refresh()
- hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
- entry.async_on_unload(entry.add_update_listener(async_handle_entry_update))
+ entry.async_on_unload(entry.add_update_listener(async_reload_entry))
return True
-async def async_handle_entry_update(hass: HomeAssistant, entry: ConfigEntry) -> None:
- """Handle an options update."""
+async def async_reload_entry(hass: HomeAssistant, entry: PurpleAirConfigEntry) -> None:
+ """Reload config entry."""
await hass.config_entries.async_reload(entry.entry_id)
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
- """Unload a config entry."""
- if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
- hass.data[DOMAIN].pop(entry.entry_id)
-
- return unload_ok
+async def async_unload_entry(hass: HomeAssistant, entry: PurpleAirConfigEntry) -> bool:
+ """Unload config entry."""
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/purpleair/const.py b/homeassistant/components/purpleair/const.py
index 5f1ec84d469..fcb928bd4f3 100644
--- a/homeassistant/components/purpleair/const.py
+++ b/homeassistant/components/purpleair/const.py
@@ -1,10 +1,13 @@
"""Constants for the PurpleAir integration."""
import logging
+from typing import Final
-DOMAIN = "purpleair"
+from homeassistant.const import Platform
-LOGGER = logging.getLogger(__package__)
+LOGGER: Final = logging.getLogger(__package__)
+PLATFORMS: Final = [Platform.SENSOR]
-CONF_READ_KEY = "read_key"
-CONF_SENSOR_INDICES = "sensor_indices"
+DOMAIN: Final[str] = "purpleair"
+
+CONF_SENSOR_INDICES: Final[str] = "sensor_indices"
diff --git a/homeassistant/components/purpleair/coordinator.py b/homeassistant/components/purpleair/coordinator.py
index 7bf0770c6fc..4ed0c0340c6 100644
--- a/homeassistant/components/purpleair/coordinator.py
+++ b/homeassistant/components/purpleair/coordinator.py
@@ -46,19 +46,27 @@ SENSOR_FIELDS_TO_RETRIEVE = [
UPDATE_INTERVAL = timedelta(minutes=2)
+type PurpleAirConfigEntry = ConfigEntry[PurpleAirDataUpdateCoordinator]
+
+
class PurpleAirDataUpdateCoordinator(DataUpdateCoordinator[GetSensorsResponse]):
"""Define a PurpleAir-specific coordinator."""
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
+ config_entry: PurpleAirConfigEntry
+
+ def __init__(self, hass: HomeAssistant, entry: PurpleAirConfigEntry) -> None:
"""Initialize."""
- self._entry = entry
self._api = API(
entry.data[CONF_API_KEY],
session=aiohttp_client.async_get_clientsession(hass),
)
super().__init__(
- hass, LOGGER, name=entry.title, update_interval=UPDATE_INTERVAL
+ hass,
+ LOGGER,
+ config_entry=entry,
+ name=entry.title,
+ update_interval=UPDATE_INTERVAL,
)
async def _async_update_data(self) -> GetSensorsResponse:
@@ -66,7 +74,7 @@ class PurpleAirDataUpdateCoordinator(DataUpdateCoordinator[GetSensorsResponse]):
try:
return await self._api.sensors.async_get_sensors(
SENSOR_FIELDS_TO_RETRIEVE,
- sensor_indices=self._entry.options[CONF_SENSOR_INDICES],
+ sensor_indices=self.config_entry.options[CONF_SENSOR_INDICES],
)
except InvalidApiKeyError as err:
raise ConfigEntryAuthFailed("Invalid API key") from err
diff --git a/homeassistant/components/purpleair/diagnostics.py b/homeassistant/components/purpleair/diagnostics.py
index f7c44b7e9b2..71b83e277d3 100644
--- a/homeassistant/components/purpleair/diagnostics.py
+++ b/homeassistant/components/purpleair/diagnostics.py
@@ -5,7 +5,6 @@ from __future__ import annotations
from typing import Any
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_API_KEY,
CONF_LATITUDE,
@@ -14,8 +13,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator
+from .coordinator import PurpleAirConfigEntry
CONF_TITLE = "title"
@@ -30,14 +28,13 @@ TO_REDACT = {
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, entry: ConfigEntry
+ hass: HomeAssistant, entry: PurpleAirConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
- coordinator: PurpleAirDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
return async_redact_data(
{
"entry": entry.as_dict(),
- "data": coordinator.data.model_dump(),
+ "data": entry.runtime_data.data.model_dump(),
},
TO_REDACT,
)
diff --git a/homeassistant/components/purpleair/entity.py b/homeassistant/components/purpleair/entity.py
index 4f7be1874ed..410fdd9b942 100644
--- a/homeassistant/components/purpleair/entity.py
+++ b/homeassistant/components/purpleair/entity.py
@@ -7,13 +7,12 @@ from typing import Any
from aiopurpleair.models.sensors import SensorModel
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE, CONF_SHOW_ON_MAP
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator
+from .coordinator import PurpleAirConfigEntry, PurpleAirDataUpdateCoordinator
class PurpleAirEntity(CoordinatorEntity[PurpleAirDataUpdateCoordinator]):
@@ -23,12 +22,11 @@ class PurpleAirEntity(CoordinatorEntity[PurpleAirDataUpdateCoordinator]):
def __init__(
self,
- coordinator: PurpleAirDataUpdateCoordinator,
- entry: ConfigEntry,
+ entry: PurpleAirConfigEntry,
sensor_index: int,
) -> None:
"""Initialize."""
- super().__init__(coordinator)
+ super().__init__(entry.runtime_data)
self._sensor_index = sensor_index
diff --git a/homeassistant/components/purpleair/sensor.py b/homeassistant/components/purpleair/sensor.py
index 9fb0249a360..a85a23b6144 100644
--- a/homeassistant/components/purpleair/sensor.py
+++ b/homeassistant/components/purpleair/sensor.py
@@ -13,7 +13,6 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
PERCENTAGE,
@@ -25,10 +24,10 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import CONF_SENSOR_INDICES, DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator
+from .const import CONF_SENSOR_INDICES
+from .coordinator import PurpleAirConfigEntry
from .entity import PurpleAirEntity
CONCENTRATION_PARTICLES_PER_100_MILLILITERS = f"particles/100{UnitOfVolume.MILLILITERS}"
@@ -165,13 +164,12 @@ SENSOR_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
- entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: PurpleAirConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up PurpleAir sensors based on a config entry."""
- coordinator: PurpleAirDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
async_add_entities(
- PurpleAirSensorEntity(coordinator, entry, sensor_index, description)
+ PurpleAirSensorEntity(entry, sensor_index, description)
for sensor_index in entry.options[CONF_SENSOR_INDICES]
for description in SENSOR_DESCRIPTIONS
)
@@ -184,13 +182,12 @@ class PurpleAirSensorEntity(PurpleAirEntity, SensorEntity):
def __init__(
self,
- coordinator: PurpleAirDataUpdateCoordinator,
- entry: ConfigEntry,
+ entry: PurpleAirConfigEntry,
sensor_index: int,
description: PurpleAirSensorEntityDescription,
) -> None:
"""Initialize."""
- super().__init__(coordinator, entry, sensor_index)
+ super().__init__(entry, sensor_index)
self._attr_unique_id = f"{self._sensor_index}-{description.key}"
self.entity_description = description
diff --git a/homeassistant/components/pushbullet/sensor.py b/homeassistant/components/pushbullet/sensor.py
index 4989fc91d5e..2dbaa8fc713 100644
--- a/homeassistant/components/pushbullet/sensor.py
+++ b/homeassistant/components/pushbullet/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .api import PushBulletNotificationProvider
from .const import DATA_UPDATED, DOMAIN
@@ -68,7 +68,9 @@ SENSOR_KEYS: list[str] = [desc.key for desc in SENSOR_TYPES]
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Pushbullet sensors from config entry."""
diff --git a/homeassistant/components/pvoutput/coordinator.py b/homeassistant/components/pvoutput/coordinator.py
index 5c38792c553..ce3642421bf 100644
--- a/homeassistant/components/pvoutput/coordinator.py
+++ b/homeassistant/components/pvoutput/coordinator.py
@@ -21,14 +21,15 @@ class PVOutputDataUpdateCoordinator(DataUpdateCoordinator[Status]):
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize the PVOutput coordinator."""
- self.config_entry = entry
self.pvoutput = PVOutput(
api_key=entry.data[CONF_API_KEY],
system_id=entry.data[CONF_SYSTEM_ID],
session=async_get_clientsession(hass),
)
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass, LOGGER, config_entry=entry, name=DOMAIN, update_interval=SCAN_INTERVAL
+ )
async def _async_update_data(self) -> Status:
"""Fetch system status from PVOutput."""
diff --git a/homeassistant/components/pvoutput/sensor.py b/homeassistant/components/pvoutput/sensor.py
index ef2bb3eb660..b4ed3f93945 100644
--- a/homeassistant/components/pvoutput/sensor.py
+++ b/homeassistant/components/pvoutput/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_SYSTEM_ID, DOMAIN
@@ -98,7 +98,7 @@ SENSORS: tuple[PVOutputSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a PVOutput sensors based on a config entry."""
coordinator: PVOutputDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/pvpc_hourly_pricing/coordinator.py b/homeassistant/components/pvpc_hourly_pricing/coordinator.py
index 171e516abdc..28e676d37ed 100644
--- a/homeassistant/components/pvpc_hourly_pricing/coordinator.py
+++ b/homeassistant/components/pvpc_hourly_pricing/coordinator.py
@@ -21,6 +21,8 @@ _LOGGER = logging.getLogger(__name__)
class ElecPricesDataUpdateCoordinator(DataUpdateCoordinator[EsiosApiData]):
"""Class to manage fetching Electricity prices data from API."""
+ config_entry: ConfigEntry
+
def __init__(
self, hass: HomeAssistant, entry: ConfigEntry, sensor_keys: set[str]
) -> None:
@@ -35,14 +37,17 @@ class ElecPricesDataUpdateCoordinator(DataUpdateCoordinator[EsiosApiData]):
sensor_keys=tuple(sensor_keys),
)
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=timedelta(minutes=30)
+ hass,
+ _LOGGER,
+ config_entry=entry,
+ name=DOMAIN,
+ update_interval=timedelta(minutes=30),
)
- self._entry = entry
@property
def entry_id(self) -> str:
"""Return entry ID."""
- return self._entry.entry_id
+ return self.config_entry.entry_id
async def _async_update_data(self) -> EsiosApiData:
"""Update electricity prices from the ESIOS API."""
diff --git a/homeassistant/components/pvpc_hourly_pricing/sensor.py b/homeassistant/components/pvpc_hourly_pricing/sensor.py
index 9d9fe5b9661..1b92cfc533d 100644
--- a/homeassistant/components/pvpc_hourly_pricing/sensor.py
+++ b/homeassistant/components/pvpc_hourly_pricing/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CURRENCY_EURO, UnitOfEnergy
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_change
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -148,7 +148,9 @@ _PRICE_SENSOR_ATTRIBUTES_MAP = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the electricity price sensor from config_entry."""
coordinator: ElecPricesDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/pyload/__init__.py b/homeassistant/components/pyload/__init__.py
index f07db509630..ca7bbb0c1dc 100644
--- a/homeassistant/components/pyload/__init__.py
+++ b/homeassistant/components/pyload/__init__.py
@@ -2,40 +2,35 @@
from __future__ import annotations
-from aiohttp import CookieJar
-from pyloadapi.api import PyLoadAPI
-from pyloadapi.exceptions import CannotConnect, InvalidAuth, ParserError
+import logging
+
+from aiohttp import CookieJar
+from pyloadapi import PyLoadAPI
+from yarl import URL
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_HOST,
CONF_PASSWORD,
CONF_PORT,
CONF_SSL,
+ CONF_URL,
CONF_USERNAME,
CONF_VERIFY_SSL,
Platform,
)
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_create_clientsession
-from .const import DOMAIN
-from .coordinator import PyLoadCoordinator
+from .coordinator import PyLoadConfigEntry, PyLoadCoordinator
+
+_LOGGER = logging.getLogger(__name__)
PLATFORMS: list[Platform] = [Platform.BUTTON, Platform.SENSOR, Platform.SWITCH]
-type PyLoadConfigEntry = ConfigEntry[PyLoadCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bool:
"""Set up pyLoad from a config entry."""
- url = (
- f"{'https' if entry.data[CONF_SSL] else 'http'}://"
- f"{entry.data[CONF_HOST]}:{entry.data[CONF_PORT]}/"
- )
-
session = async_create_clientsession(
hass,
verify_ssl=entry.data[CONF_VERIFY_SSL],
@@ -43,30 +38,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bo
)
pyloadapi = PyLoadAPI(
session,
- api_url=url,
+ api_url=URL(entry.data[CONF_URL]),
username=entry.data[CONF_USERNAME],
password=entry.data[CONF_PASSWORD],
)
- try:
- await pyloadapi.login()
- except CannotConnect as e:
- raise ConfigEntryNotReady(
- translation_domain=DOMAIN,
- translation_key="setup_request_exception",
- ) from e
- except ParserError as e:
- raise ConfigEntryNotReady(
- translation_domain=DOMAIN,
- translation_key="setup_parse_exception",
- ) from e
- except InvalidAuth as e:
- raise ConfigEntryAuthFailed(
- translation_domain=DOMAIN,
- translation_key="setup_authentication_exception",
- translation_placeholders={CONF_USERNAME: entry.data[CONF_USERNAME]},
- ) from e
- coordinator = PyLoadCoordinator(hass, pyloadapi)
+ coordinator = PyLoadCoordinator(hass, entry, pyloadapi)
await coordinator.async_config_entry_first_refresh()
@@ -79,3 +56,27 @@ async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bo
async def async_unload_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+
+
+async def async_migrate_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bool:
+ """Migrate config entry."""
+ _LOGGER.debug(
+ "Migrating configuration from version %s.%s", entry.version, entry.minor_version
+ )
+
+ if entry.version == 1 and entry.minor_version == 0:
+ url = URL.build(
+ scheme="https" if entry.data[CONF_SSL] else "http",
+ host=entry.data[CONF_HOST],
+ port=entry.data[CONF_PORT],
+ ).human_repr()
+ hass.config_entries.async_update_entry(
+ entry, data={**entry.data, CONF_URL: url}, minor_version=1, version=1
+ )
+
+ _LOGGER.debug(
+ "Migration to configuration version %s.%s successful",
+ entry.version,
+ entry.minor_version,
+ )
+ return True
diff --git a/homeassistant/components/pyload/button.py b/homeassistant/components/pyload/button.py
index 386fe6968de..5ee10a327d1 100644
--- a/homeassistant/components/pyload/button.py
+++ b/homeassistant/components/pyload/button.py
@@ -7,17 +7,19 @@ from dataclasses import dataclass
from enum import StrEnum
from typing import Any
-from pyloadapi.api import CannotConnect, InvalidAuth, PyLoadAPI
+from pyloadapi import CannotConnect, InvalidAuth, PyLoadAPI
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PyLoadConfigEntry
from .const import DOMAIN
+from .coordinator import PyLoadConfigEntry
from .entity import BasePyLoadEntity
+PARALLEL_UPDATES = 1
+
@dataclass(kw_only=True, frozen=True)
class PyLoadButtonEntityDescription(ButtonEntityDescription):
@@ -63,7 +65,7 @@ SENSOR_DESCRIPTIONS: tuple[PyLoadButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PyLoadConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up buttons from a config entry."""
diff --git a/homeassistant/components/pyload/config_flow.py b/homeassistant/components/pyload/config_flow.py
index b9bfc579cfc..50d354d345d 100644
--- a/homeassistant/components/pyload/config_flow.py
+++ b/homeassistant/components/pyload/config_flow.py
@@ -7,22 +7,19 @@ import logging
from typing import Any
from aiohttp import CookieJar
-from pyloadapi.api import PyLoadAPI
-from pyloadapi.exceptions import CannotConnect, InvalidAuth, ParserError
+from pyloadapi import CannotConnect, InvalidAuth, ParserError, PyLoadAPI
import voluptuous as vol
+from yarl import URL
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import (
- CONF_HOST,
CONF_NAME,
CONF_PASSWORD,
- CONF_PORT,
- CONF_SSL,
+ CONF_URL,
CONF_USERNAME,
CONF_VERIFY_SSL,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_create_clientsession
from homeassistant.helpers.selector import (
TextSelector,
@@ -30,15 +27,18 @@ from homeassistant.helpers.selector import (
TextSelectorType,
)
-from .const import DEFAULT_NAME, DEFAULT_PORT, DOMAIN
+from .const import DEFAULT_NAME, DOMAIN
_LOGGER = logging.getLogger(__name__)
STEP_USER_DATA_SCHEMA = vol.Schema(
{
- vol.Required(CONF_HOST): str,
- vol.Required(CONF_PORT, default=DEFAULT_PORT): cv.port,
- vol.Required(CONF_SSL, default=False): cv.boolean,
+ vol.Required(CONF_URL): TextSelector(
+ TextSelectorConfig(
+ type=TextSelectorType.URL,
+ autocomplete="url",
+ ),
+ ),
vol.Required(CONF_VERIFY_SSL, default=True): bool,
vol.Required(CONF_USERNAME): TextSelector(
TextSelectorConfig(
@@ -81,14 +81,9 @@ async def validate_input(hass: HomeAssistant, user_input: dict[str, Any]) -> Non
user_input[CONF_VERIFY_SSL],
cookie_jar=CookieJar(unsafe=True),
)
-
- url = (
- f"{'https' if user_input[CONF_SSL] else 'http'}://"
- f"{user_input[CONF_HOST]}:{user_input[CONF_PORT]}/"
- )
pyload = PyLoadAPI(
session,
- api_url=url,
+ api_url=URL(user_input[CONF_URL]),
username=user_input[CONF_USERNAME],
password=user_input[CONF_PASSWORD],
)
@@ -100,6 +95,7 @@ class PyLoadConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a config flow for pyLoad."""
VERSION = 1
+ MINOR_VERSION = 1
async def async_step_user(
self, user_input: dict[str, Any] | None = None
@@ -107,9 +103,8 @@ class PyLoadConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle the initial step."""
errors: dict[str, str] = {}
if user_input is not None:
- self._async_abort_entries_match(
- {CONF_HOST: user_input[CONF_HOST], CONF_PORT: user_input[CONF_PORT]}
- )
+ url = URL(user_input[CONF_URL]).human_repr()
+ self._async_abort_entries_match({CONF_URL: url})
try:
await validate_input(self.hass, user_input)
except (CannotConnect, ParserError):
@@ -121,7 +116,14 @@ class PyLoadConfigFlow(ConfigFlow, domain=DOMAIN):
errors["base"] = "unknown"
else:
title = DEFAULT_NAME
- return self.async_create_entry(title=title, data=user_input)
+
+ return self.async_create_entry(
+ title=title,
+ data={
+ **user_input,
+ CONF_URL: url,
+ },
+ )
return self.async_show_form(
step_id="user",
@@ -145,9 +147,8 @@ class PyLoadConfigFlow(ConfigFlow, domain=DOMAIN):
reauth_entry = self._get_reauth_entry()
if user_input is not None:
- new_input = reauth_entry.data | user_input
try:
- await validate_input(self.hass, new_input)
+ await validate_input(self.hass, {**reauth_entry.data, **user_input})
except (CannotConnect, ParserError):
errors["base"] = "cannot_connect"
except InvalidAuth:
@@ -156,7 +157,9 @@ class PyLoadConfigFlow(ConfigFlow, domain=DOMAIN):
_LOGGER.exception("Unexpected exception")
errors["base"] = "unknown"
else:
- return self.async_update_reload_and_abort(reauth_entry, data=new_input)
+ return self.async_update_reload_and_abort(
+ reauth_entry, data_updates=user_input
+ )
return self.async_show_form(
step_id="reauth_confirm",
@@ -192,15 +195,18 @@ class PyLoadConfigFlow(ConfigFlow, domain=DOMAIN):
else:
return self.async_update_reload_and_abort(
reconfig_entry,
- data=user_input,
+ data={
+ **user_input,
+ CONF_URL: URL(user_input[CONF_URL]).human_repr(),
+ },
reload_even_if_entry_is_unchanged=False,
)
-
+ suggested_values = user_input if user_input else reconfig_entry.data
return self.async_show_form(
step_id="reconfigure",
data_schema=self.add_suggested_values_to_schema(
STEP_USER_DATA_SCHEMA,
- user_input or reconfig_entry.data,
+ suggested_values,
),
description_placeholders={CONF_NAME: reconfig_entry.data[CONF_USERNAME]},
errors=errors,
diff --git a/homeassistant/components/pyload/coordinator.py b/homeassistant/components/pyload/coordinator.py
index 7eadefcd260..c57dfa7720d 100644
--- a/homeassistant/components/pyload/coordinator.py
+++ b/homeassistant/components/pyload/coordinator.py
@@ -9,7 +9,7 @@ from pyloadapi import CannotConnect, InvalidAuth, ParserError, PyLoadAPI
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_USERNAME
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import ConfigEntryAuthFailed
+from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN
@@ -34,16 +34,22 @@ class PyLoadData:
free_space: int
+type PyLoadConfigEntry = ConfigEntry[PyLoadCoordinator]
+
+
class PyLoadCoordinator(DataUpdateCoordinator[PyLoadData]):
"""pyLoad coordinator."""
- config_entry: ConfigEntry
+ config_entry: PyLoadConfigEntry
- def __init__(self, hass: HomeAssistant, pyload: PyLoadAPI) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: PyLoadConfigEntry, pyload: PyLoadAPI
+ ) -> None:
"""Initialize pyLoad coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
@@ -53,14 +59,11 @@ class PyLoadCoordinator(DataUpdateCoordinator[PyLoadData]):
async def _async_update_data(self) -> PyLoadData:
"""Fetch data from API endpoint."""
try:
- if not self.version:
- self.version = await self.pyload.version()
return PyLoadData(
**await self.pyload.get_status(),
free_space=await self.pyload.free_space(),
)
-
- except InvalidAuth as e:
+ except InvalidAuth:
try:
await self.pyload.login()
except InvalidAuth as exc:
@@ -69,13 +72,42 @@ class PyLoadCoordinator(DataUpdateCoordinator[PyLoadData]):
translation_key="setup_authentication_exception",
translation_placeholders={CONF_USERNAME: self.pyload.username},
) from exc
-
- raise UpdateFailed(
- "Unable to retrieve data due to cookie expiration"
- ) from e
+ _LOGGER.debug(
+ "Unable to retrieve data due to cookie expiration, retrying after 20 seconds"
+ )
+ return self.data
except CannotConnect as e:
raise UpdateFailed(
- "Unable to connect and retrieve data from pyLoad API"
+ translation_domain=DOMAIN,
+ translation_key="setup_request_exception",
) from e
except ParserError as e:
- raise UpdateFailed("Unable to parse data from pyLoad API") from e
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="setup_parse_exception",
+ ) from e
+
+ async def _async_setup(self) -> None:
+ """Set up the coordinator."""
+
+ try:
+ await self.pyload.login()
+ self.version = await self.pyload.version()
+ except CannotConnect as e:
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="setup_request_exception",
+ ) from e
+ except ParserError as e:
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="setup_parse_exception",
+ ) from e
+ except InvalidAuth as e:
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN,
+ translation_key="setup_authentication_exception",
+ translation_placeholders={
+ CONF_USERNAME: self.config_entry.data[CONF_USERNAME]
+ },
+ ) from e
diff --git a/homeassistant/components/pyload/diagnostics.py b/homeassistant/components/pyload/diagnostics.py
index e9688a3369b..98fab38da1d 100644
--- a/homeassistant/components/pyload/diagnostics.py
+++ b/homeassistant/components/pyload/diagnostics.py
@@ -5,14 +5,15 @@ from __future__ import annotations
from dataclasses import asdict
from typing import Any
-from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
+from yarl import URL
+
+from homeassistant.components.diagnostics import REDACTED, async_redact_data
+from homeassistant.const import CONF_PASSWORD, CONF_URL, CONF_USERNAME
from homeassistant.core import HomeAssistant
-from . import PyLoadConfigEntry
-from .coordinator import PyLoadData
+from .coordinator import PyLoadConfigEntry, PyLoadData
-TO_REDACT = {CONF_USERNAME, CONF_PASSWORD, CONF_HOST}
+TO_REDACT = {CONF_USERNAME, CONF_PASSWORD, CONF_URL}
async def async_get_config_entry_diagnostics(
@@ -22,6 +23,9 @@ async def async_get_config_entry_diagnostics(
pyload_data: PyLoadData = config_entry.runtime_data.data
return {
- "config_entry_data": async_redact_data(dict(config_entry.data), TO_REDACT),
+ "config_entry_data": {
+ **async_redact_data(dict(config_entry.data), TO_REDACT),
+ CONF_URL: URL(config_entry.data[CONF_URL]).with_host(REDACTED).human_repr(),
+ },
"pyload_data": asdict(pyload_data),
}
diff --git a/homeassistant/components/pyload/manifest.json b/homeassistant/components/pyload/manifest.json
index e21167cf10b..feaa23af7de 100644
--- a/homeassistant/components/pyload/manifest.json
+++ b/homeassistant/components/pyload/manifest.json
@@ -7,5 +7,6 @@
"integration_type": "service",
"iot_class": "local_polling",
"loggers": ["pyloadapi"],
- "requirements": ["PyLoadAPI==1.3.2"]
+ "quality_scale": "platinum",
+ "requirements": ["PyLoadAPI==1.4.2"]
}
diff --git a/homeassistant/components/pyload/quality_scale.yaml b/homeassistant/components/pyload/quality_scale.yaml
new file mode 100644
index 00000000000..a9ce552961b
--- /dev/null
+++ b/homeassistant/components/pyload/quality_scale.yaml
@@ -0,0 +1,82 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: The integration registers no actions.
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: The integration registers no actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: exempt
+ comment: The integration registers no events.
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions:
+ status: exempt
+ comment: The integration registers no actions.
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: Integration has no configuration parameters
+ docs-installation-parameters: done
+ entity-unavailable: done
+ integration-owner: done
+ log-when-unavailable: done
+ parallel-updates: done
+ reauthentication-flow: done
+ test-coverage: done
+
+ # Gold
+ devices: done
+ diagnostics: done
+ discovery-update-info:
+ status: exempt
+ comment: The integration is a web service, there are no discoverable devices.
+ discovery:
+ status: exempt
+ comment: The integration is a web service, there are no discoverable devices.
+ docs-data-update: done
+ docs-examples: done
+ docs-known-limitations: done
+ docs-supported-devices:
+ status: exempt
+ comment: The integration is a web service, there are no devices.
+ docs-supported-functions: done
+ docs-troubleshooting: done
+ docs-use-cases: done
+ dynamic-devices:
+ status: exempt
+ comment: The integration is a web service, there are no devices.
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default: done
+ entity-translations: done
+ exception-translations: done
+ icon-translations: done
+ reconfiguration-flow: done
+ repair-issues:
+ status: exempt
+ comment: The integration has no repairs.
+ stale-devices:
+ status: exempt
+ comment: The integration is a web service, there are no devices.
+
+ # Platinum
+ async-dependency: done
+ inject-websession: done
+ strict-typing: done
diff --git a/homeassistant/components/pyload/sensor.py b/homeassistant/components/pyload/sensor.py
index 38f681d30d5..7425c543fe1 100644
--- a/homeassistant/components/pyload/sensor.py
+++ b/homeassistant/components/pyload/sensor.py
@@ -14,14 +14,15 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfDataRate, UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import PyLoadConfigEntry
from .const import UNIT_DOWNLOADS
-from .coordinator import PyLoadData
+from .coordinator import PyLoadConfigEntry, PyLoadData
from .entity import BasePyLoadEntity
+PARALLEL_UPDATES = 0
+
class PyLoadSensorEntity(StrEnum):
"""pyLoad Sensor Entities."""
@@ -86,7 +87,7 @@ SENSOR_DESCRIPTIONS: tuple[PyLoadSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PyLoadConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the pyLoad sensors."""
diff --git a/homeassistant/components/pyload/strings.json b/homeassistant/components/pyload/strings.json
index 0fd9b4befcf..9414f7f7bb8 100644
--- a/homeassistant/components/pyload/strings.json
+++ b/homeassistant/components/pyload/strings.json
@@ -3,30 +3,30 @@
"step": {
"user": {
"data": {
- "host": "[%key:common::config_flow::data::host%]",
+ "url": "[%key:common::config_flow::data::url%]",
"username": "[%key:common::config_flow::data::username%]",
"password": "[%key:common::config_flow::data::password%]",
- "ssl": "[%key:common::config_flow::data::ssl%]",
- "verify_ssl": "[%key:common::config_flow::data::verify_ssl%]",
- "port": "[%key:common::config_flow::data::port%]"
+ "verify_ssl": "[%key:common::config_flow::data::verify_ssl%]"
},
"data_description": {
- "host": "The hostname or IP address of the device running your pyLoad instance.",
- "port": "pyLoad uses port 8000 by default."
+ "url": "Specify the full URL of your pyLoad web interface, including the protocol (HTTP or HTTPS), hostname or IP address, port (pyLoad uses 8000 by default), and any path prefix if applicable.\nExample: `https://example.com:8000/path`",
+ "username": "The username used to access the pyLoad instance.",
+ "password": "The password associated with the pyLoad account.",
+ "verify_ssl": "If checked, the SSL certificate will be validated to ensure a secure connection."
}
},
"reconfigure": {
"data": {
- "host": "[%key:common::config_flow::data::host%]",
+ "url": "[%key:common::config_flow::data::url%]",
"username": "[%key:common::config_flow::data::username%]",
"password": "[%key:common::config_flow::data::password%]",
- "ssl": "[%key:common::config_flow::data::ssl%]",
- "verify_ssl": "[%key:common::config_flow::data::verify_ssl%]",
- "port": "[%key:common::config_flow::data::port%]"
+ "verify_ssl": "[%key:common::config_flow::data::verify_ssl%]"
},
"data_description": {
- "host": "The hostname or IP address of the device running your pyLoad instance.",
- "port": "pyLoad uses port 8000 by default."
+ "url": "[%key:component::pyload::config::step::user::data_description::url%]",
+ "username": "[%key:component::pyload::config::step::user::data_description::username%]",
+ "password": "[%key:component::pyload::config::step::user::data_description::password%]",
+ "verify_ssl": "[%key:component::pyload::config::step::user::data_description::verify_ssl%]"
}
},
"reauth_confirm": {
@@ -34,6 +34,10 @@
"data": {
"username": "[%key:common::config_flow::data::username%]",
"password": "[%key:common::config_flow::data::password%]"
+ },
+ "data_description": {
+ "username": "[%key:component::pyload::config::step::user::data_description::username%]",
+ "password": "[%key:component::pyload::config::step::user::data_description::password%]"
}
}
},
@@ -91,10 +95,10 @@
},
"exceptions": {
"setup_request_exception": {
- "message": "Unable to connect and retrieve data from pyLoad API, try again later"
+ "message": "Unable to connect and retrieve data from pyLoad API"
},
"setup_parse_exception": {
- "message": "Unable to parse data from pyLoad API, try again later"
+ "message": "Unable to parse data from pyLoad API"
},
"setup_authentication_exception": {
"message": "Authentication failed for {username}, verify your login credentials"
diff --git a/homeassistant/components/pyload/switch.py b/homeassistant/components/pyload/switch.py
index ea189ed9a8f..46a54451b9a 100644
--- a/homeassistant/components/pyload/switch.py
+++ b/homeassistant/components/pyload/switch.py
@@ -7,7 +7,7 @@ from dataclasses import dataclass
from enum import StrEnum
from typing import Any
-from pyloadapi.api import CannotConnect, InvalidAuth, PyLoadAPI
+from pyloadapi import CannotConnect, InvalidAuth, PyLoadAPI
from homeassistant.components.switch import (
SwitchDeviceClass,
@@ -16,13 +16,14 @@ from homeassistant.components.switch import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import PyLoadConfigEntry
from .const import DOMAIN
-from .coordinator import PyLoadData
+from .coordinator import PyLoadConfigEntry, PyLoadData
from .entity import BasePyLoadEntity
+PARALLEL_UPDATES = 1
+
class PyLoadSwitch(StrEnum):
"""PyLoad Switch Entities."""
@@ -66,7 +67,7 @@ SENSOR_DESCRIPTIONS: tuple[PyLoadSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: PyLoadConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the pyLoad sensors."""
diff --git a/homeassistant/components/qbittorrent/__init__.py b/homeassistant/components/qbittorrent/__init__.py
index d95136965f8..513b49d3561 100644
--- a/homeassistant/components/qbittorrent/__init__.py
+++ b/homeassistant/components/qbittorrent/__init__.py
@@ -124,7 +124,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
except APIConnectionError as exc:
raise ConfigEntryNotReady("Fail to connect to qBittorrent") from exc
- coordinator = QBittorrentDataCoordinator(hass, client)
+ coordinator = QBittorrentDataCoordinator(hass, config_entry, client)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = coordinator
diff --git a/homeassistant/components/qbittorrent/coordinator.py b/homeassistant/components/qbittorrent/coordinator.py
index c590bb9d81a..8fd23fb3b5b 100644
--- a/homeassistant/components/qbittorrent/coordinator.py
+++ b/homeassistant/components/qbittorrent/coordinator.py
@@ -15,6 +15,7 @@ from qbittorrentapi import (
)
from qbittorrentapi.torrents import TorrentStatusesT
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -27,7 +28,11 @@ _LOGGER = logging.getLogger(__name__)
class QBittorrentDataCoordinator(DataUpdateCoordinator[SyncMainDataDictionary]):
"""Coordinator for updating QBittorrent data."""
- def __init__(self, hass: HomeAssistant, client: Client) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, client: Client
+ ) -> None:
"""Initialize coordinator."""
self.client = client
self._is_alternative_mode_enabled = False
@@ -42,6 +47,7 @@ class QBittorrentDataCoordinator(DataUpdateCoordinator[SyncMainDataDictionary]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/qbittorrent/sensor.py b/homeassistant/components/qbittorrent/sensor.py
index 67eb856bb83..23ec485fcd4 100644
--- a/homeassistant/components/qbittorrent/sensor.py
+++ b/homeassistant/components/qbittorrent/sensor.py
@@ -14,10 +14,10 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import STATE_IDLE, UnitOfDataRate
+from homeassistant.const import STATE_IDLE, UnitOfDataRate, UnitOfInformation
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -27,8 +27,14 @@ from .coordinator import QBittorrentDataCoordinator
_LOGGER = logging.getLogger(__name__)
SENSOR_TYPE_CURRENT_STATUS = "current_status"
+SENSOR_TYPE_CONNECTION_STATUS = "connection_status"
SENSOR_TYPE_DOWNLOAD_SPEED = "download_speed"
SENSOR_TYPE_UPLOAD_SPEED = "upload_speed"
+SENSOR_TYPE_DOWNLOAD_SPEED_LIMIT = "download_speed_limit"
+SENSOR_TYPE_UPLOAD_SPEED_LIMIT = "upload_speed_limit"
+SENSOR_TYPE_ALLTIME_DOWNLOAD = "alltime_download"
+SENSOR_TYPE_ALLTIME_UPLOAD = "alltime_upload"
+SENSOR_TYPE_GLOBAL_RATIO = "global_ratio"
SENSOR_TYPE_ALL_TORRENTS = "all_torrents"
SENSOR_TYPE_PAUSED_TORRENTS = "paused_torrents"
SENSOR_TYPE_ACTIVE_TORRENTS = "active_torrents"
@@ -50,18 +56,54 @@ def get_state(coordinator: QBittorrentDataCoordinator) -> str:
return STATE_IDLE
-def get_dl(coordinator: QBittorrentDataCoordinator) -> int:
+def get_connection_status(coordinator: QBittorrentDataCoordinator) -> str:
+ """Get current download/upload state."""
+ server_state = cast(Mapping, coordinator.data.get("server_state"))
+ return cast(str, server_state.get("connection_status"))
+
+
+def get_download_speed(coordinator: QBittorrentDataCoordinator) -> int:
"""Get current download speed."""
server_state = cast(Mapping, coordinator.data.get("server_state"))
return cast(int, server_state.get("dl_info_speed"))
-def get_up(coordinator: QBittorrentDataCoordinator) -> int:
+def get_upload_speed(coordinator: QBittorrentDataCoordinator) -> int:
"""Get current upload speed."""
server_state = cast(Mapping[str, Any], coordinator.data.get("server_state"))
return cast(int, server_state.get("up_info_speed"))
+def get_download_speed_limit(coordinator: QBittorrentDataCoordinator) -> int:
+ """Get current download speed."""
+ server_state = cast(Mapping, coordinator.data.get("server_state"))
+ return cast(int, server_state.get("dl_rate_limit"))
+
+
+def get_upload_speed_limit(coordinator: QBittorrentDataCoordinator) -> int:
+ """Get current upload speed."""
+ server_state = cast(Mapping[str, Any], coordinator.data.get("server_state"))
+ return cast(int, server_state.get("up_rate_limit"))
+
+
+def get_alltime_download(coordinator: QBittorrentDataCoordinator) -> int:
+ """Get current download speed."""
+ server_state = cast(Mapping, coordinator.data.get("server_state"))
+ return cast(int, server_state.get("alltime_dl"))
+
+
+def get_alltime_upload(coordinator: QBittorrentDataCoordinator) -> int:
+ """Get current download speed."""
+ server_state = cast(Mapping, coordinator.data.get("server_state"))
+ return cast(int, server_state.get("alltime_ul"))
+
+
+def get_global_ratio(coordinator: QBittorrentDataCoordinator) -> float:
+ """Get current download speed."""
+ server_state = cast(Mapping, coordinator.data.get("server_state"))
+ return cast(float, server_state.get("global_ratio"))
+
+
@dataclass(frozen=True, kw_only=True)
class QBittorrentSensorEntityDescription(SensorEntityDescription):
"""Entity description class for qBittorent sensors."""
@@ -77,6 +119,13 @@ SENSOR_TYPES: tuple[QBittorrentSensorEntityDescription, ...] = (
options=[STATE_IDLE, STATE_UP_DOWN, STATE_SEEDING, STATE_DOWNLOADING],
value_fn=get_state,
),
+ QBittorrentSensorEntityDescription(
+ key=SENSOR_TYPE_CONNECTION_STATUS,
+ translation_key="connection_status",
+ device_class=SensorDeviceClass.ENUM,
+ options=["connected", "firewalled", "disconnected"],
+ value_fn=get_connection_status,
+ ),
QBittorrentSensorEntityDescription(
key=SENSOR_TYPE_DOWNLOAD_SPEED,
translation_key="download_speed",
@@ -85,7 +134,7 @@ SENSOR_TYPES: tuple[QBittorrentSensorEntityDescription, ...] = (
native_unit_of_measurement=UnitOfDataRate.BYTES_PER_SECOND,
suggested_display_precision=2,
suggested_unit_of_measurement=UnitOfDataRate.MEGABYTES_PER_SECOND,
- value_fn=get_dl,
+ value_fn=get_download_speed,
),
QBittorrentSensorEntityDescription(
key=SENSOR_TYPE_UPLOAD_SPEED,
@@ -95,7 +144,56 @@ SENSOR_TYPES: tuple[QBittorrentSensorEntityDescription, ...] = (
native_unit_of_measurement=UnitOfDataRate.BYTES_PER_SECOND,
suggested_display_precision=2,
suggested_unit_of_measurement=UnitOfDataRate.MEGABYTES_PER_SECOND,
- value_fn=get_up,
+ value_fn=get_upload_speed,
+ ),
+ QBittorrentSensorEntityDescription(
+ key=SENSOR_TYPE_DOWNLOAD_SPEED_LIMIT,
+ translation_key="download_speed_limit",
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.DATA_RATE,
+ native_unit_of_measurement=UnitOfDataRate.BYTES_PER_SECOND,
+ suggested_display_precision=2,
+ suggested_unit_of_measurement=UnitOfDataRate.MEGABYTES_PER_SECOND,
+ value_fn=get_download_speed_limit,
+ entity_registry_enabled_default=False,
+ ),
+ QBittorrentSensorEntityDescription(
+ key=SENSOR_TYPE_UPLOAD_SPEED_LIMIT,
+ translation_key="upload_speed_limit",
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.DATA_RATE,
+ native_unit_of_measurement=UnitOfDataRate.BYTES_PER_SECOND,
+ suggested_display_precision=2,
+ suggested_unit_of_measurement=UnitOfDataRate.MEGABYTES_PER_SECOND,
+ value_fn=get_upload_speed_limit,
+ entity_registry_enabled_default=False,
+ ),
+ QBittorrentSensorEntityDescription(
+ key=SENSOR_TYPE_ALLTIME_DOWNLOAD,
+ translation_key="alltime_download",
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ device_class=SensorDeviceClass.DATA_SIZE,
+ native_unit_of_measurement=UnitOfInformation.BYTES,
+ suggested_display_precision=2,
+ suggested_unit_of_measurement=UnitOfInformation.TEBIBYTES,
+ value_fn=get_alltime_download,
+ ),
+ QBittorrentSensorEntityDescription(
+ key=SENSOR_TYPE_ALLTIME_UPLOAD,
+ translation_key="alltime_upload",
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ device_class=SensorDeviceClass.DATA_SIZE,
+ native_unit_of_measurement="B",
+ suggested_display_precision=2,
+ suggested_unit_of_measurement="TiB",
+ value_fn=get_alltime_upload,
+ ),
+ QBittorrentSensorEntityDescription(
+ key=SENSOR_TYPE_GLOBAL_RATIO,
+ translation_key="global_ratio",
+ state_class=SensorStateClass.MEASUREMENT,
+ value_fn=get_global_ratio,
+ entity_registry_enabled_default=False,
),
QBittorrentSensorEntityDescription(
key=SENSOR_TYPE_ALL_TORRENTS,
@@ -129,7 +227,7 @@ SENSOR_TYPES: tuple[QBittorrentSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up qBittorrent sensor entries."""
diff --git a/homeassistant/components/qbittorrent/strings.json b/homeassistant/components/qbittorrent/strings.json
index 9c9ee371737..ee613eb96c2 100644
--- a/homeassistant/components/qbittorrent/strings.json
+++ b/homeassistant/components/qbittorrent/strings.json
@@ -26,6 +26,21 @@
"upload_speed": {
"name": "Upload speed"
},
+ "download_speed_limit": {
+ "name": "Download speed limit"
+ },
+ "upload_speed_limit": {
+ "name": "Upload speed limit"
+ },
+ "alltime_download": {
+ "name": "All-time download"
+ },
+ "alltime_upload": {
+ "name": "All-time upload"
+ },
+ "global_ratio": {
+ "name": "Global ratio"
+ },
"current_status": {
"name": "Status",
"state": {
@@ -35,6 +50,14 @@
"downloading": "Downloading"
}
},
+ "connection_status": {
+ "name": "Connection status",
+ "state": {
+ "connected": "Connected",
+ "firewalled": "Firewalled",
+ "disconnected": "Disconnected"
+ }
+ },
"active_torrents": {
"name": "Active torrents",
"unit_of_measurement": "torrents"
@@ -86,16 +109,16 @@
},
"exceptions": {
"invalid_device": {
- "message": "No device with id {device_id} was found"
+ "message": "No device with ID {device_id} was found"
},
"invalid_entry_id": {
- "message": "No entry with id {device_id} was found"
+ "message": "No entry with ID {device_id} was found"
},
"login_error": {
- "message": "A login error occured. Please check you username and password."
+ "message": "A login error occurred. Please check your username and password."
},
"cannot_connect": {
- "message": "Can't connect to QBittorrent, please check your configuration."
+ "message": "Can't connect to qBittorrent, please check your configuration."
}
}
}
diff --git a/homeassistant/components/qbittorrent/switch.py b/homeassistant/components/qbittorrent/switch.py
index f12118e5233..dd61f130ca1 100644
--- a/homeassistant/components/qbittorrent/switch.py
+++ b/homeassistant/components/qbittorrent/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -43,7 +43,7 @@ SWITCH_TYPES: tuple[QBittorrentSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up qBittorrent switch entries."""
diff --git a/homeassistant/components/qbus/__init__.py b/homeassistant/components/qbus/__init__.py
index da9dcfe69be..f77f439ecc1 100644
--- a/homeassistant/components/qbus/__init__.py
+++ b/homeassistant/components/qbus/__init__.py
@@ -71,17 +71,14 @@ async def async_unload_entry(hass: HomeAssistant, entry: QbusConfigEntry) -> boo
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
entry.runtime_data.shutdown()
- cleanup(hass, entry)
+ _cleanup(hass, entry)
return unload_ok
-def cleanup(hass: HomeAssistant, entry: QbusConfigEntry) -> None:
+def _cleanup(hass: HomeAssistant, entry: QbusConfigEntry) -> None:
"""Shutdown if no more entries are loaded."""
- entries = hass.config_entries.async_loaded_entries(DOMAIN)
- count = len(entries)
-
- # During unloading of the entry, it is not marked as unloaded yet. So
- # count can be 1 if it is the last one.
- if count <= 1 and (config_coordinator := hass.data.get(QBUS_KEY)):
+ if not hass.config_entries.async_loaded_entries(DOMAIN) and (
+ config_coordinator := hass.data.get(QBUS_KEY)
+ ):
config_coordinator.shutdown()
diff --git a/homeassistant/components/qbus/const.py b/homeassistant/components/qbus/const.py
index ddfb8963cb7..b9e42f13766 100644
--- a/homeassistant/components/qbus/const.py
+++ b/homeassistant/components/qbus/const.py
@@ -5,7 +5,10 @@ from typing import Final
from homeassistant.const import Platform
DOMAIN: Final = "qbus"
-PLATFORMS: list[Platform] = [Platform.SWITCH]
+PLATFORMS: list[Platform] = [
+ Platform.LIGHT,
+ Platform.SWITCH,
+]
CONF_SERIAL_NUMBER: Final = "serial"
diff --git a/homeassistant/components/qbus/entity.py b/homeassistant/components/qbus/entity.py
index 39bcddaaf4f..4ab1913c4dc 100644
--- a/homeassistant/components/qbus/entity.py
+++ b/homeassistant/components/qbus/entity.py
@@ -1,6 +1,9 @@
"""Base class for Qbus entities."""
+from __future__ import annotations
+
from abc import ABC, abstractmethod
+from collections.abc import Callable
import re
from qbusmqttapi.discovery import QbusMqttOutput
@@ -10,12 +13,36 @@ from qbusmqttapi.state import QbusMqttState
from homeassistant.components.mqtt import ReceiveMessage, client as mqtt
from homeassistant.helpers.device_registry import DeviceInfo, format_mac
from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, MANUFACTURER
+from .coordinator import QbusControllerCoordinator
_REFID_REGEX = re.compile(r"^\d+\/(\d+(?:\/\d+)?)$")
+def add_new_outputs(
+ coordinator: QbusControllerCoordinator,
+ added_outputs: list[QbusMqttOutput],
+ filter_fn: Callable[[QbusMqttOutput], bool],
+ entity_type: type[QbusEntity],
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Call async_add_entities for new outputs."""
+
+ added_ref_ids = {k.ref_id for k in added_outputs}
+
+ new_outputs = [
+ output
+ for output in coordinator.data
+ if filter_fn(output) and output.ref_id not in added_ref_ids
+ ]
+
+ if new_outputs:
+ added_outputs.extend(new_outputs)
+ async_add_entities([entity_type(output) for output in new_outputs])
+
+
def format_ref_id(ref_id: str) -> str | None:
"""Format the Qbus ref_id."""
matches: list[str] = re.findall(_REFID_REGEX, ref_id)
diff --git a/homeassistant/components/qbus/light.py b/homeassistant/components/qbus/light.py
new file mode 100644
index 00000000000..5ec76f5e807
--- /dev/null
+++ b/homeassistant/components/qbus/light.py
@@ -0,0 +1,110 @@
+"""Support for Qbus light."""
+
+from typing import Any
+
+from qbusmqttapi.discovery import QbusMqttOutput
+from qbusmqttapi.state import QbusMqttAnalogState, StateType
+
+from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
+from homeassistant.components.mqtt import ReceiveMessage
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.util.color import brightness_to_value, value_to_brightness
+
+from .coordinator import QbusConfigEntry
+from .entity import QbusEntity, add_new_outputs
+
+PARALLEL_UPDATES = 0
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: QbusConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up light entities."""
+
+ coordinator = entry.runtime_data
+ added_outputs: list[QbusMqttOutput] = []
+
+ def _check_outputs() -> None:
+ add_new_outputs(
+ coordinator,
+ added_outputs,
+ lambda output: output.type == "analog",
+ QbusLight,
+ async_add_entities,
+ )
+
+ _check_outputs()
+ entry.async_on_unload(coordinator.async_add_listener(_check_outputs))
+
+
+class QbusLight(QbusEntity, LightEntity):
+ """Representation of a Qbus light entity."""
+
+ _attr_supported_color_modes = {ColorMode.BRIGHTNESS}
+ _attr_color_mode = ColorMode.BRIGHTNESS
+
+ def __init__(self, mqtt_output: QbusMqttOutput) -> None:
+ """Initialize light entity."""
+
+ super().__init__(mqtt_output)
+
+ self._set_state()
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the entity on."""
+ brightness = kwargs.get(ATTR_BRIGHTNESS)
+
+ percentage: int | None = None
+ on: bool | None = None
+
+ state = QbusMqttAnalogState(id=self._mqtt_output.id)
+
+ if brightness is None:
+ on = True
+
+ state.type = StateType.ACTION
+ state.write_on_off(on)
+ else:
+ percentage = round(brightness_to_value((1, 100), brightness))
+
+ state.type = StateType.STATE
+ state.write_percentage(percentage)
+
+ await self._async_publish_output_state(state)
+ self._set_state(percentage=percentage, on=on)
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the entity off."""
+ state = QbusMqttAnalogState(id=self._mqtt_output.id, type=StateType.ACTION)
+ state.write_on_off(on=False)
+
+ await self._async_publish_output_state(state)
+ self._set_state(on=False)
+
+ async def _state_received(self, msg: ReceiveMessage) -> None:
+ output = self._message_factory.parse_output_state(
+ QbusMqttAnalogState, msg.payload
+ )
+
+ if output is not None:
+ percentage = round(output.read_percentage())
+ self._set_state(percentage=percentage)
+ self.async_schedule_update_ha_state()
+
+ def _set_state(
+ self, *, percentage: int | None = None, on: bool | None = None
+ ) -> None:
+ if percentage is None:
+ # When turning on without brightness, we don't know the desired
+ # brightness. It will be set during _state_received().
+ if on is True:
+ self._attr_is_on = True
+ else:
+ self._attr_is_on = False
+ self._attr_brightness = 0
+ else:
+ self._attr_is_on = percentage > 0
+ self._attr_brightness = value_to_brightness((1, 100), percentage)
diff --git a/homeassistant/components/qbus/manifest.json b/homeassistant/components/qbus/manifest.json
index b7d277f3953..17101da7c33 100644
--- a/homeassistant/components/qbus/manifest.json
+++ b/homeassistant/components/qbus/manifest.json
@@ -13,5 +13,5 @@
"cloudapp/QBUSMQTTGW/+/state"
],
"quality_scale": "bronze",
- "requirements": ["qbusmqttapi==1.2.4"]
+ "requirements": ["qbusmqttapi==1.3.0"]
}
diff --git a/homeassistant/components/qbus/strings.json b/homeassistant/components/qbus/strings.json
index b8918497c41..e6df18c393c 100644
--- a/homeassistant/components/qbus/strings.json
+++ b/homeassistant/components/qbus/strings.json
@@ -10,7 +10,7 @@
"abort": {
"already_configured": "Controller already configured",
"discovery_in_progress": "Discovery in progress",
- "not_supported": "Configuration for QBUS happens through MQTT discovery. If your controller is not automatically discovered, check the prerequisites and troubleshooting section of the documention."
+ "not_supported": "Configuration for QBUS happens through MQTT discovery. If your controller is not automatically discovered, check the prerequisites and troubleshooting section of the documentation."
},
"error": {
"no_controller": "No controllers were found"
diff --git a/homeassistant/components/qbus/switch.py b/homeassistant/components/qbus/switch.py
index 2413b8f152f..002ad43e904 100644
--- a/homeassistant/components/qbus/switch.py
+++ b/homeassistant/components/qbus/switch.py
@@ -8,35 +8,32 @@ from qbusmqttapi.state import QbusMqttOnOffState, StateType
from homeassistant.components.mqtt import ReceiveMessage
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import QbusConfigEntry
-from .entity import QbusEntity
+from .entity import QbusEntity, add_new_outputs
PARALLEL_UPDATES = 0
async def async_setup_entry(
- hass: HomeAssistant, entry: QbusConfigEntry, add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: QbusConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch entities."""
- coordinator = entry.runtime_data
+ coordinator = entry.runtime_data
added_outputs: list[QbusMqttOutput] = []
- # Local function that calls add_entities for new entities
def _check_outputs() -> None:
- added_output_ids = {k.id for k in added_outputs}
-
- new_outputs = [
- item
- for item in coordinator.data
- if item.type == "onoff" and item.id not in added_output_ids
- ]
-
- if new_outputs:
- added_outputs.extend(new_outputs)
- add_entities([QbusSwitch(output) for output in new_outputs])
+ add_new_outputs(
+ coordinator,
+ added_outputs,
+ lambda output: output.type == "onoff",
+ QbusSwitch,
+ async_add_entities,
+ )
_check_outputs()
entry.async_on_unload(coordinator.async_add_listener(_check_outputs))
@@ -47,10 +44,7 @@ class QbusSwitch(QbusEntity, SwitchEntity):
_attr_device_class = SwitchDeviceClass.SWITCH
- def __init__(
- self,
- mqtt_output: QbusMqttOutput,
- ) -> None:
+ def __init__(self, mqtt_output: QbusMqttOutput) -> None:
"""Initialize switch entity."""
super().__init__(mqtt_output)
diff --git a/homeassistant/components/qingping/binary_sensor.py b/homeassistant/components/qingping/binary_sensor.py
index 5f1367fbce8..3431204595a 100644
--- a/homeassistant/components/qingping/binary_sensor.py
+++ b/homeassistant/components/qingping/binary_sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.bluetooth.passive_update_processor import (
PassiveBluetoothProcessorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from . import QingpingConfigEntry
@@ -74,7 +74,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: QingpingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Qingping BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/qingping/sensor.py b/homeassistant/components/qingping/sensor.py
index 3d5f30c61fc..ee2a63b169a 100644
--- a/homeassistant/components/qingping/sensor.py
+++ b/homeassistant/components/qingping/sensor.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from . import QingpingConfigEntry
@@ -142,7 +142,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: QingpingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Qingping BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/qnap/coordinator.py b/homeassistant/components/qnap/coordinator.py
index 8c2bf81a47f..297f6569d2b 100644
--- a/homeassistant/components/qnap/coordinator.py
+++ b/homeassistant/components/qnap/coordinator.py
@@ -33,7 +33,13 @@ class QnapCoordinator(DataUpdateCoordinator[dict[str, dict[str, Any]]]):
def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize the qnap coordinator."""
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=UPDATE_INTERVAL,
+ )
protocol = "https" if config_entry.data[CONF_SSL] else "http"
self._api = QNAPStats(
diff --git a/homeassistant/components/qnap/sensor.py b/homeassistant/components/qnap/sensor.py
index 383a4e5f572..381455cb7e1 100644
--- a/homeassistant/components/qnap/sensor.py
+++ b/homeassistant/components/qnap/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -248,7 +248,7 @@ SENSOR_KEYS: list[str] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
coordinator = QnapCoordinator(hass, config_entry)
diff --git a/homeassistant/components/qnap_qsw/__init__.py b/homeassistant/components/qnap_qsw/__init__.py
index d7352435b07..f9faca025a5 100644
--- a/homeassistant/components/qnap_qsw/__init__.py
+++ b/homeassistant/components/qnap_qsw/__init__.py
@@ -35,10 +35,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
qsw = QnapQswApi(aiohttp_client.async_get_clientsession(hass), options)
- coord_data = QswDataCoordinator(hass, qsw)
+ coord_data = QswDataCoordinator(hass, entry, qsw)
await coord_data.async_config_entry_first_refresh()
- coord_fw = QswFirmwareCoordinator(hass, qsw)
+ coord_fw = QswFirmwareCoordinator(hass, entry, qsw)
try:
await coord_fw.async_config_entry_first_refresh()
except ConfigEntryNotReady as error:
diff --git a/homeassistant/components/qnap_qsw/binary_sensor.py b/homeassistant/components/qnap_qsw/binary_sensor.py
index a9c025b86ce..c1f77d068df 100644
--- a/homeassistant/components/qnap_qsw/binary_sensor.py
+++ b/homeassistant/components/qnap_qsw/binary_sensor.py
@@ -23,7 +23,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UNDEFINED
from .const import ATTR_MESSAGE, DOMAIN, QSW_COORD_DATA
@@ -78,7 +78,9 @@ PORT_BINARY_SENSOR_TYPES: Final[tuple[QswBinarySensorEntityDescription, ...]] =
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add QNAP QSW binary sensors from a config_entry."""
coordinator: QswDataCoordinator = hass.data[DOMAIN][entry.entry_id][QSW_COORD_DATA]
diff --git a/homeassistant/components/qnap_qsw/button.py b/homeassistant/components/qnap_qsw/button.py
index 091c6786a92..02cf96766f2 100644
--- a/homeassistant/components/qnap_qsw/button.py
+++ b/homeassistant/components/qnap_qsw/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, QSW_COORD_DATA, QSW_REBOOT
from .coordinator import QswDataCoordinator
@@ -41,7 +41,9 @@ BUTTON_TYPES: Final[tuple[QswButtonDescription, ...]] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add QNAP QSW buttons from a config_entry."""
coordinator: QswDataCoordinator = hass.data[DOMAIN][entry.entry_id][QSW_COORD_DATA]
diff --git a/homeassistant/components/qnap_qsw/coordinator.py b/homeassistant/components/qnap_qsw/coordinator.py
index c873f2a773d..b72bed7105c 100644
--- a/homeassistant/components/qnap_qsw/coordinator.py
+++ b/homeassistant/components/qnap_qsw/coordinator.py
@@ -10,6 +10,7 @@ from typing import Any
from aioqsw.exceptions import QswError
from aioqsw.localapi import QnapQswApi
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -24,13 +25,18 @@ _LOGGER = logging.getLogger(__name__)
class QswDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching data from the QNAP QSW device."""
- def __init__(self, hass: HomeAssistant, qsw: QnapQswApi) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, qsw: QnapQswApi
+ ) -> None:
"""Initialize."""
self.qsw = qsw
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=DATA_SCAN_INTERVAL,
)
@@ -48,13 +54,18 @@ class QswDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class QswFirmwareCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching firmware data from the QNAP QSW device."""
- def __init__(self, hass: HomeAssistant, qsw: QnapQswApi) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, qsw: QnapQswApi
+ ) -> None:
"""Initialize."""
self.qsw = qsw
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=FW_SCAN_INTERVAL,
)
diff --git a/homeassistant/components/qnap_qsw/sensor.py b/homeassistant/components/qnap_qsw/sensor.py
index e7f2c18638f..af02c121656 100644
--- a/homeassistant/components/qnap_qsw/sensor.py
+++ b/homeassistant/components/qnap_qsw/sensor.py
@@ -44,7 +44,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UNDEFINED, StateType
from homeassistant.util import dt as dt_util
@@ -286,7 +286,9 @@ PORT_SENSOR_TYPES: Final[tuple[QswSensorEntityDescription, ...]] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add QNAP QSW sensors from a config_entry."""
coordinator: QswDataCoordinator = hass.data[DOMAIN][entry.entry_id][QSW_COORD_DATA]
diff --git a/homeassistant/components/qnap_qsw/update.py b/homeassistant/components/qnap_qsw/update.py
index ac789235271..c5cef729849 100644
--- a/homeassistant/components/qnap_qsw/update.py
+++ b/homeassistant/components/qnap_qsw/update.py
@@ -20,7 +20,7 @@ from homeassistant.components.update import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, QSW_COORD_FW, QSW_UPDATE
from .coordinator import QswFirmwareCoordinator
@@ -36,7 +36,9 @@ UPDATE_TYPES: Final[tuple[UpdateEntityDescription, ...]] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add QNAP QSW updates from a config_entry."""
coordinator: QswFirmwareCoordinator = hass.data[DOMAIN][entry.entry_id][
diff --git a/homeassistant/components/qwikswitch/entity.py b/homeassistant/components/qwikswitch/entity.py
index 3a2ec5a9206..ff7a1d2e98a 100644
--- a/homeassistant/components/qwikswitch/entity.py
+++ b/homeassistant/components/qwikswitch/entity.py
@@ -35,7 +35,7 @@ class QSEntity(Entity):
"""Receive update packet from QSUSB. Match dispather_send signature."""
self.async_write_ha_state()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Listen for updates from QSUSb via dispatcher."""
self.async_on_remove(
async_dispatcher_connect(self.hass, self.qsid, self.update_packet)
diff --git a/homeassistant/components/rabbitair/__init__.py b/homeassistant/components/rabbitair/__init__.py
index e4eb67a67f5..d6530b322b0 100644
--- a/homeassistant/components/rabbitair/__init__.py
+++ b/homeassistant/components/rabbitair/__init__.py
@@ -26,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
zeroconf_instance = await zeroconf.async_get_async_instance(hass)
device: Client = UdpClient(host, token, zeroconf=zeroconf_instance)
- coordinator = RabbitAirDataUpdateCoordinator(hass, device)
+ coordinator = RabbitAirDataUpdateCoordinator(hass, entry, device)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/rabbitair/coordinator.py b/homeassistant/components/rabbitair/coordinator.py
index 3c7db126c7d..75453fe4d24 100644
--- a/homeassistant/components/rabbitair/coordinator.py
+++ b/homeassistant/components/rabbitair/coordinator.py
@@ -7,6 +7,7 @@ from typing import Any, cast
from rabbitair import Client, State
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -42,12 +43,17 @@ class RabbitAirDebouncer(Debouncer[Coroutine[Any, Any, None]]):
class RabbitAirDataUpdateCoordinator(DataUpdateCoordinator[State]):
"""Class to manage fetching data from single endpoint."""
- def __init__(self, hass: HomeAssistant, device: Client) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, device: Client
+ ) -> None:
"""Initialize global data updater."""
self.device = device
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="rabbitair",
update_interval=timedelta(seconds=10),
request_refresh_debouncer=RabbitAirDebouncer(hass),
diff --git a/homeassistant/components/rabbitair/fan.py b/homeassistant/components/rabbitair/fan.py
index cfbee0be67c..4c13f3a8b02 100644
--- a/homeassistant/components/rabbitair/fan.py
+++ b/homeassistant/components/rabbitair/fan.py
@@ -9,7 +9,7 @@ from rabbitair import Mode, Model, Speed
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -39,7 +39,9 @@ PRESET_MODES = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
coordinator: RabbitAirDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/rachio/binary_sensor.py b/homeassistant/components/rachio/binary_sensor.py
index 189a08e998d..3bf0f716c6d 100644
--- a/homeassistant/components/rachio/binary_sensor.py
+++ b/homeassistant/components/rachio/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN as DOMAIN_RACHIO,
@@ -46,7 +46,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Rachio binary sensors."""
entities = await hass.async_add_executor_job(_create_entities, hass, config_entry)
diff --git a/homeassistant/components/rachio/calendar.py b/homeassistant/components/rachio/calendar.py
index 5c7e13c748a..91ad29fac9f 100644
--- a/homeassistant/components/rachio/calendar.py
+++ b/homeassistant/components/rachio/calendar.py
@@ -12,7 +12,7 @@ from homeassistant.components.calendar import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -41,7 +41,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry for Rachio smart hose timer calendar."""
person: RachioPerson = hass.data[DOMAIN_RACHIO][config_entry.entry_id]
diff --git a/homeassistant/components/rachio/strings.json b/homeassistant/components/rachio/strings.json
index 308403d805d..d51a1d5f920 100644
--- a/homeassistant/components/rachio/strings.json
+++ b/homeassistant/components/rachio/strings.json
@@ -3,7 +3,7 @@
"step": {
"user": {
"title": "Connect to your Rachio device",
- "description": "You will need the API Key from https://app.rach.io/. Go to Settings, then select 'GET API KEY'.",
+ "description": "You will need the API key from https://app.rach.io/. Go to Settings, then select 'GET API KEY'.",
"data": {
"api_key": "[%key:common::config_flow::data::api_key%]"
}
@@ -70,7 +70,7 @@
},
"start_watering": {
"name": "Start watering",
- "description": "Start a single zone, a schedule or any number of smart hose timers.",
+ "description": "Starts a single zone, a schedule or any number of smart hose timers.",
"fields": {
"duration": {
"name": "Duration",
@@ -80,7 +80,7 @@
},
"pause_watering": {
"name": "Pause watering",
- "description": "Pause any currently running zones or schedules.",
+ "description": "Pauses any currently running zones or schedules.",
"fields": {
"devices": {
"name": "Devices",
@@ -94,7 +94,7 @@
},
"resume_watering": {
"name": "Resume watering",
- "description": "Resume any paused zone runs or schedules.",
+ "description": "Resumes any paused zone runs or schedules.",
"fields": {
"devices": {
"name": "[%key:component::rachio::services::pause_watering::fields::devices::name%]",
@@ -104,7 +104,7 @@
},
"stop_watering": {
"name": "Stop watering",
- "description": "Stop any currently running zones or schedules.",
+ "description": "Stops any currently running zones or schedules.",
"fields": {
"devices": {
"name": "[%key:component::rachio::services::pause_watering::fields::devices::name%]",
diff --git a/homeassistant/components/rachio/switch.py b/homeassistant/components/rachio/switch.py
index 92e7c0ea2ba..25cdeac62f7 100644
--- a/homeassistant/components/rachio/switch.py
+++ b/homeassistant/components/rachio/switch.py
@@ -16,7 +16,7 @@ from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_point_in_utc_time
from homeassistant.util.dt import as_timestamp, now, parse_datetime, utc_from_timestamp
@@ -102,7 +102,7 @@ START_MULTIPLE_ZONES_SCHEMA = vol.Schema(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Rachio switches."""
zone_entities = []
diff --git a/homeassistant/components/radarr/__init__.py b/homeassistant/components/radarr/__init__.py
index 5c225697f98..11a9b6b4dc0 100644
--- a/homeassistant/components/radarr/__init__.py
+++ b/homeassistant/components/radarr/__init__.py
@@ -2,12 +2,11 @@
from __future__ import annotations
-from dataclasses import dataclass, fields
+from dataclasses import fields
from aiopyarr.models.host_configuration import PyArrHostConfiguration
from aiopyarr.radarr_client import RadarrClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY, CONF_URL, CONF_VERIFY_SSL, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -18,24 +17,13 @@ from .coordinator import (
HealthDataUpdateCoordinator,
MoviesDataUpdateCoordinator,
QueueDataUpdateCoordinator,
+ RadarrConfigEntry,
+ RadarrData,
RadarrDataUpdateCoordinator,
StatusDataUpdateCoordinator,
)
PLATFORMS = [Platform.BINARY_SENSOR, Platform.CALENDAR, Platform.SENSOR]
-type RadarrConfigEntry = ConfigEntry[RadarrData]
-
-
-@dataclass(kw_only=True, slots=True)
-class RadarrData:
- """Radarr data type."""
-
- calendar: CalendarUpdateCoordinator
- disk_space: DiskSpaceDataUpdateCoordinator
- health: HealthDataUpdateCoordinator
- movie: MoviesDataUpdateCoordinator
- queue: QueueDataUpdateCoordinator
- status: StatusDataUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: RadarrConfigEntry) -> bool:
@@ -50,12 +38,14 @@ async def async_setup_entry(hass: HomeAssistant, entry: RadarrConfigEntry) -> bo
session=async_get_clientsession(hass, entry.data[CONF_VERIFY_SSL]),
)
data = RadarrData(
- calendar=CalendarUpdateCoordinator(hass, host_configuration, radarr),
- disk_space=DiskSpaceDataUpdateCoordinator(hass, host_configuration, radarr),
- health=HealthDataUpdateCoordinator(hass, host_configuration, radarr),
- movie=MoviesDataUpdateCoordinator(hass, host_configuration, radarr),
- queue=QueueDataUpdateCoordinator(hass, host_configuration, radarr),
- status=StatusDataUpdateCoordinator(hass, host_configuration, radarr),
+ calendar=CalendarUpdateCoordinator(hass, entry, host_configuration, radarr),
+ disk_space=DiskSpaceDataUpdateCoordinator(
+ hass, entry, host_configuration, radarr
+ ),
+ health=HealthDataUpdateCoordinator(hass, entry, host_configuration, radarr),
+ movie=MoviesDataUpdateCoordinator(hass, entry, host_configuration, radarr),
+ queue=QueueDataUpdateCoordinator(hass, entry, host_configuration, radarr),
+ status=StatusDataUpdateCoordinator(hass, entry, host_configuration, radarr),
)
for field in fields(data):
coordinator: RadarrDataUpdateCoordinator = getattr(data, field.name)
diff --git a/homeassistant/components/radarr/binary_sensor.py b/homeassistant/components/radarr/binary_sensor.py
index 953c7dead18..f09e6015b53 100644
--- a/homeassistant/components/radarr/binary_sensor.py
+++ b/homeassistant/components/radarr/binary_sensor.py
@@ -11,10 +11,10 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RadarrConfigEntry
from .const import HEALTH_ISSUES
+from .coordinator import RadarrConfigEntry
from .entity import RadarrEntity
BINARY_SENSOR_TYPE = BinarySensorEntityDescription(
@@ -28,7 +28,7 @@ BINARY_SENSOR_TYPE = BinarySensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: RadarrConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Radarr sensors based on a config entry."""
coordinator = entry.runtime_data.health
diff --git a/homeassistant/components/radarr/calendar.py b/homeassistant/components/radarr/calendar.py
index c741c178862..00df27f21bd 100644
--- a/homeassistant/components/radarr/calendar.py
+++ b/homeassistant/components/radarr/calendar.py
@@ -7,10 +7,9 @@ from datetime import datetime
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RadarrConfigEntry
-from .coordinator import CalendarUpdateCoordinator, RadarrEvent
+from .coordinator import CalendarUpdateCoordinator, RadarrConfigEntry, RadarrEvent
from .entity import RadarrEntity
CALENDAR_TYPE = EntityDescription(
@@ -22,7 +21,7 @@ CALENDAR_TYPE = EntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: RadarrConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Radarr calendar entity."""
coordinator = entry.runtime_data.calendar
diff --git a/homeassistant/components/radarr/coordinator.py b/homeassistant/components/radarr/coordinator.py
index 6e8a3d55d3e..d343675d7ea 100644
--- a/homeassistant/components/radarr/coordinator.py
+++ b/homeassistant/components/radarr/coordinator.py
@@ -6,7 +6,7 @@ from abc import ABC, abstractmethod
import asyncio
from dataclasses import dataclass
from datetime import date, datetime, timedelta
-from typing import TYPE_CHECKING, Generic, TypeVar, cast
+from typing import Generic, TypeVar, cast
from aiopyarr import (
Health,
@@ -20,14 +20,27 @@ from aiopyarr.models.host_configuration import PyArrHostConfiguration
from aiopyarr.radarr_client import RadarrClient
from homeassistant.components.calendar import CalendarEvent
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DEFAULT_MAX_RECORDS, DOMAIN, LOGGER
-if TYPE_CHECKING:
- from . import RadarrConfigEntry
+
+@dataclass(kw_only=True, slots=True)
+class RadarrData:
+ """Radarr data type."""
+
+ calendar: CalendarUpdateCoordinator
+ disk_space: DiskSpaceDataUpdateCoordinator
+ health: HealthDataUpdateCoordinator
+ movie: MoviesDataUpdateCoordinator
+ queue: QueueDataUpdateCoordinator
+ status: StatusDataUpdateCoordinator
+
+
+type RadarrConfigEntry = ConfigEntry[RadarrData]
T = TypeVar("T", bound=SystemStatus | list[RootFolder] | list[Health] | int | None)
@@ -53,6 +66,7 @@ class RadarrDataUpdateCoordinator(DataUpdateCoordinator[T], Generic[T], ABC):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: RadarrConfigEntry,
host_configuration: PyArrHostConfiguration,
api_client: RadarrClient,
) -> None:
@@ -60,6 +74,7 @@ class RadarrDataUpdateCoordinator(DataUpdateCoordinator[T], Generic[T], ABC):
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=self._update_interval,
)
@@ -140,11 +155,12 @@ class CalendarUpdateCoordinator(RadarrDataUpdateCoordinator[None]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: RadarrConfigEntry,
host_configuration: PyArrHostConfiguration,
api_client: RadarrClient,
) -> None:
"""Initialize."""
- super().__init__(hass, host_configuration, api_client)
+ super().__init__(hass, config_entry, host_configuration, api_client)
self.event: RadarrEvent | None = None
self._events: list[RadarrEvent] = []
diff --git a/homeassistant/components/radarr/sensor.py b/homeassistant/components/radarr/sensor.py
index df1a0686e00..a6d29ee9d1d 100644
--- a/homeassistant/components/radarr/sensor.py
+++ b/homeassistant/components/radarr/sensor.py
@@ -17,10 +17,9 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory, UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RadarrConfigEntry
-from .coordinator import RadarrDataUpdateCoordinator, T
+from .coordinator import RadarrConfigEntry, RadarrDataUpdateCoordinator, T
from .entity import RadarrEntity
@@ -82,14 +81,12 @@ SENSOR_TYPES: dict[str, RadarrSensorEntityDescription[Any]] = {
"movie": RadarrSensorEntityDescription[int](
key="movies",
translation_key="movies",
- native_unit_of_measurement="Movies",
entity_registry_enabled_default=False,
value_fn=lambda data, _: data,
),
"queue": RadarrSensorEntityDescription[int](
key="queue",
translation_key="queue",
- native_unit_of_measurement="Movies",
entity_registry_enabled_default=False,
state_class=SensorStateClass.TOTAL,
value_fn=lambda data, _: data,
@@ -117,7 +114,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: RadarrConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Radarr sensors based on a config entry."""
entities: list[RadarrSensor[Any]] = []
diff --git a/homeassistant/components/radarr/strings.json b/homeassistant/components/radarr/strings.json
index ec1baf6ffd8..268d7955c1b 100644
--- a/homeassistant/components/radarr/strings.json
+++ b/homeassistant/components/radarr/strings.json
@@ -43,10 +43,12 @@
},
"sensor": {
"movies": {
- "name": "Movies"
+ "name": "Movies",
+ "unit_of_measurement": "movies"
},
"queue": {
- "name": "Queue"
+ "name": "Queue",
+ "unit_of_measurement": "[%key:component::radarr::entity::sensor::movies::unit_of_measurement%]"
},
"start_time": {
"name": "Start time"
diff --git a/homeassistant/components/radiotherm/__init__.py b/homeassistant/components/radiotherm/__init__.py
index 7b2eaba52c4..80dbcf44bc9 100644
--- a/homeassistant/components/radiotherm/__init__.py
+++ b/homeassistant/components/radiotherm/__init__.py
@@ -43,7 +43,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
host = entry.data[CONF_HOST]
init_coro = async_get_init_data(hass, host)
init_data = await _async_call_or_raise_not_ready(init_coro, host)
- coordinator = RadioThermUpdateCoordinator(hass, init_data)
+ coordinator = RadioThermUpdateCoordinator(hass, entry, init_data)
await coordinator.async_config_entry_first_refresh()
# Only set the time if the thermostat is
diff --git a/homeassistant/components/radiotherm/climate.py b/homeassistant/components/radiotherm/climate.py
index af52c5fcea3..09ac5b42b60 100644
--- a/homeassistant/components/radiotherm/climate.py
+++ b/homeassistant/components/radiotherm/climate.py
@@ -20,7 +20,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_HALVES, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
from .coordinator import RadioThermUpdateCoordinator
@@ -93,7 +93,7 @@ def round_temp(temperature):
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate for a radiotherm device."""
coordinator: RadioThermUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/radiotherm/coordinator.py b/homeassistant/components/radiotherm/coordinator.py
index 06e3554c8d7..7d483426c83 100644
--- a/homeassistant/components/radiotherm/coordinator.py
+++ b/homeassistant/components/radiotherm/coordinator.py
@@ -8,6 +8,7 @@ from urllib.error import URLError
from radiotherm.validate import RadiothermTstatError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -21,13 +22,21 @@ UPDATE_INTERVAL = timedelta(seconds=15)
class RadioThermUpdateCoordinator(DataUpdateCoordinator[RadioThermUpdate]):
"""DataUpdateCoordinator to gather data for radio thermostats."""
- def __init__(self, hass: HomeAssistant, init_data: RadioThermInitData) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ init_data: RadioThermInitData,
+ ) -> None:
"""Initialize DataUpdateCoordinator."""
self.init_data = init_data
self._description = f"{init_data.name} ({init_data.host})"
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"radiotherm {self.init_data.name}",
update_interval=UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/radiotherm/switch.py b/homeassistant/components/radiotherm/switch.py
index e7b463e3def..2952e1e5817 100644
--- a/homeassistant/components/radiotherm/switch.py
+++ b/homeassistant/components/radiotherm/switch.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RadioThermUpdateCoordinator
@@ -19,7 +19,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches for a radiotherm device."""
coordinator: RadioThermUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/rainbird/__init__.py b/homeassistant/components/rainbird/__init__.py
index d8b71e2df0b..f9cd751a81e 100644
--- a/homeassistant/components/rainbird/__init__.py
+++ b/homeassistant/components/rainbird/__init__.py
@@ -101,18 +101,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: RainbirdConfigEntry) ->
data = RainbirdData(
controller,
model_info,
- coordinator=RainbirdUpdateCoordinator(
- hass,
- name=entry.title,
- controller=controller,
- unique_id=entry.unique_id,
- model_info=model_info,
- ),
- schedule_coordinator=RainbirdScheduleUpdateCoordinator(
- hass,
- name=f"{entry.title} Schedule",
- controller=controller,
- ),
+ coordinator=RainbirdUpdateCoordinator(hass, entry, controller, model_info),
+ schedule_coordinator=RainbirdScheduleUpdateCoordinator(hass, entry, controller),
)
await data.coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/rainbird/binary_sensor.py b/homeassistant/components/rainbird/binary_sensor.py
index 5722b8852dd..0b27c7e33c4 100644
--- a/homeassistant/components/rainbird/binary_sensor.py
+++ b/homeassistant/components/rainbird/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import RainbirdUpdateCoordinator
@@ -27,7 +27,7 @@ RAIN_SENSOR_ENTITY_DESCRIPTION = BinarySensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RainbirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry for a Rain Bird binary_sensor."""
coordinator = config_entry.runtime_data.coordinator
diff --git a/homeassistant/components/rainbird/calendar.py b/homeassistant/components/rainbird/calendar.py
index 160fe70c61e..c48ca438146 100644
--- a/homeassistant/components/rainbird/calendar.py
+++ b/homeassistant/components/rainbird/calendar.py
@@ -9,7 +9,7 @@ from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -22,7 +22,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RainbirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry for a Rain Bird irrigation calendar."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/rainbird/coordinator.py b/homeassistant/components/rainbird/coordinator.py
index 2ccfa0af62a..426df625697 100644
--- a/homeassistant/components/rainbird/coordinator.py
+++ b/homeassistant/components/rainbird/coordinator.py
@@ -58,26 +58,28 @@ def async_create_clientsession() -> aiohttp.ClientSession:
class RainbirdUpdateCoordinator(DataUpdateCoordinator[RainbirdDeviceState]):
"""Coordinator for rainbird API calls."""
+ config_entry: RainbirdConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- name: str,
+ config_entry: RainbirdConfigEntry,
controller: AsyncRainbirdController,
- unique_id: str | None,
model_info: ModelAndVersion,
) -> None:
"""Initialize RainbirdUpdateCoordinator."""
super().__init__(
hass,
_LOGGER,
- name=name,
+ config_entry=config_entry,
+ name=config_entry.title,
update_interval=UPDATE_INTERVAL,
request_refresh_debouncer=Debouncer(
hass, _LOGGER, cooldown=DEBOUNCER_COOLDOWN, immediate=False
),
)
self._controller = controller
- self._unique_id = unique_id
+ self._unique_id = config_entry.unique_id
self._zones: set[int] | None = None
self._model_info = model_info
@@ -145,14 +147,15 @@ class RainbirdScheduleUpdateCoordinator(DataUpdateCoordinator[Schedule]):
def __init__(
self,
hass: HomeAssistant,
- name: str,
+ config_entry: RainbirdConfigEntry,
controller: AsyncRainbirdController,
) -> None:
"""Initialize ZoneStateUpdateCoordinator."""
super().__init__(
hass,
_LOGGER,
- name=name,
+ config_entry=config_entry,
+ name=f"{config_entry.title} Schedule",
update_method=self._async_update_data,
update_interval=CALENDAR_UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/rainbird/number.py b/homeassistant/components/rainbird/number.py
index d8081a796b9..7f1dfe74752 100644
--- a/homeassistant/components/rainbird/number.py
+++ b/homeassistant/components/rainbird/number.py
@@ -10,7 +10,7 @@ from homeassistant.components.number import NumberEntity
from homeassistant.const import UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import RainbirdUpdateCoordinator
@@ -22,7 +22,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RainbirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry for a Rain Bird number platform."""
async_add_entities(
diff --git a/homeassistant/components/rainbird/sensor.py b/homeassistant/components/rainbird/sensor.py
index 4725a33bc9a..9fab1af0a23 100644
--- a/homeassistant/components/rainbird/sensor.py
+++ b/homeassistant/components/rainbird/sensor.py
@@ -6,7 +6,7 @@ import logging
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -25,7 +25,7 @@ RAIN_DELAY_ENTITY_DESCRIPTION = SensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RainbirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry for a Rain Bird sensor."""
async_add_entities(
diff --git a/homeassistant/components/rainbird/switch.py b/homeassistant/components/rainbird/switch.py
index f622a1b9b2c..f188350138e 100644
--- a/homeassistant/components/rainbird/switch.py
+++ b/homeassistant/components/rainbird/switch.py
@@ -12,7 +12,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -32,7 +32,7 @@ SERVICE_SCHEMA_IRRIGATION: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RainbirdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry for a Rain Bird irrigation switches."""
coordinator = config_entry.runtime_data.coordinator
diff --git a/homeassistant/components/raincloud/entity.py b/homeassistant/components/raincloud/entity.py
index 337324d96eb..b45684ac72b 100644
--- a/homeassistant/components/raincloud/entity.py
+++ b/homeassistant/components/raincloud/entity.py
@@ -45,7 +45,7 @@ class RainCloudEntity(Entity):
"""Return the name of the sensor."""
return self._name
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register callbacks."""
self.async_on_remove(
async_dispatcher_connect(
diff --git a/homeassistant/components/rainforest_eagle/coordinator.py b/homeassistant/components/rainforest_eagle/coordinator.py
index 9c714a291ee..11956681638 100644
--- a/homeassistant/components/rainforest_eagle/coordinator.py
+++ b/homeassistant/components/rainforest_eagle/coordinator.py
@@ -29,13 +29,13 @@ _LOGGER = logging.getLogger(__name__)
class EagleDataCoordinator(DataUpdateCoordinator):
"""Get the latest data from the Eagle device."""
+ config_entry: ConfigEntry
eagle100_reader: Eagle100Reader | None = None
eagle200_meter: aioeagle.ElectricMeter | None = None
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize the data object."""
- self.entry = entry
- if self.type == TYPE_EAGLE_100:
+ if config_entry.data[CONF_TYPE] == TYPE_EAGLE_100:
self.model = "EAGLE-100"
update_method = self._async_update_data_100
else:
@@ -45,7 +45,8 @@ class EagleDataCoordinator(DataUpdateCoordinator):
super().__init__(
hass,
_LOGGER,
- name=entry.data[CONF_CLOUD_ID],
+ config_entry=config_entry,
+ name=config_entry.data[CONF_CLOUD_ID],
update_interval=timedelta(seconds=30),
update_method=update_method,
)
@@ -53,17 +54,12 @@ class EagleDataCoordinator(DataUpdateCoordinator):
@property
def cloud_id(self):
"""Return the cloud ID."""
- return self.entry.data[CONF_CLOUD_ID]
-
- @property
- def type(self):
- """Return entry type."""
- return self.entry.data[CONF_TYPE]
+ return self.config_entry.data[CONF_CLOUD_ID]
@property
def hardware_address(self):
"""Return hardware address of meter."""
- return self.entry.data[CONF_HARDWARE_ADDRESS]
+ return self.config_entry.data[CONF_HARDWARE_ADDRESS]
@property
def is_connected(self):
@@ -79,8 +75,8 @@ class EagleDataCoordinator(DataUpdateCoordinator):
hub = aioeagle.EagleHub(
aiohttp_client.async_get_clientsession(self.hass),
self.cloud_id,
- self.entry.data[CONF_INSTALL_CODE],
- host=self.entry.data[CONF_HOST],
+ self.config_entry.data[CONF_INSTALL_CODE],
+ host=self.config_entry.data[CONF_HOST],
)
eagle200_meter = aioeagle.ElectricMeter.create_instance(
hub, self.hardware_address
@@ -115,8 +111,8 @@ class EagleDataCoordinator(DataUpdateCoordinator):
if self.eagle100_reader is None:
self.eagle100_reader = Eagle100Reader(
self.cloud_id,
- self.entry.data[CONF_INSTALL_CODE],
- self.entry.data[CONF_HOST],
+ self.config_entry.data[CONF_INSTALL_CODE],
+ self.config_entry.data[CONF_HOST],
)
out = {}
diff --git a/homeassistant/components/rainforest_eagle/sensor.py b/homeassistant/components/rainforest_eagle/sensor.py
index 8c4c5927998..58427b0e5ba 100644
--- a/homeassistant/components/rainforest_eagle/sensor.py
+++ b/homeassistant/components/rainforest_eagle/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -45,7 +45,9 @@ SENSORS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/rainforest_raven/sensor.py b/homeassistant/components/rainforest_raven/sensor.py
index 1025e92ef86..3d358322b70 100644
--- a/homeassistant/components/rainforest_raven/sensor.py
+++ b/homeassistant/components/rainforest_raven/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -80,7 +80,7 @@ DIAGNOSTICS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RAVEnConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/rainmachine/__init__.py b/homeassistant/components/rainmachine/__init__.py
index 4d486c9c6aa..65648b8d44f 100644
--- a/homeassistant/components/rainmachine/__init__.py
+++ b/homeassistant/components/rainmachine/__init__.py
@@ -13,7 +13,7 @@ from regenmaschine.controller import Controller
from regenmaschine.errors import RainMachineError, UnknownAPICallError
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_DEVICE_ID,
CONF_IP_ADDRESS,
@@ -465,12 +465,7 @@ async def async_unload_entry(
) -> bool:
"""Unload an RainMachine config entry."""
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state is ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
# If this is the last loaded instance of RainMachine, deregister any services
# defined during integration setup:
for service_name in (
diff --git a/homeassistant/components/rainmachine/binary_sensor.py b/homeassistant/components/rainmachine/binary_sensor.py
index 4ba9b58d596..610505e2b7f 100644
--- a/homeassistant/components/rainmachine/binary_sensor.py
+++ b/homeassistant/components/rainmachine/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RainMachineConfigEntry
from .const import DATA_PROVISION_SETTINGS, DATA_RESTRICTIONS_CURRENT
@@ -94,7 +94,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RainMachineConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up RainMachine binary sensors based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/rainmachine/button.py b/homeassistant/components/rainmachine/button.py
index 2f68c6a8a9c..e4ed00930dd 100644
--- a/homeassistant/components/rainmachine/button.py
+++ b/homeassistant/components/rainmachine/button.py
@@ -17,7 +17,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_send
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RainMachineConfigEntry
from .const import DATA_PROVISION_SETTINGS
@@ -53,7 +53,7 @@ BUTTON_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RainMachineConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up RainMachine buttons based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/rainmachine/coordinator.py b/homeassistant/components/rainmachine/coordinator.py
index df7972ef31d..de43e5a073f 100644
--- a/homeassistant/components/rainmachine/coordinator.py
+++ b/homeassistant/components/rainmachine/coordinator.py
@@ -41,6 +41,7 @@ class RainMachineDataUpdateCoordinator(DataUpdateCoordinator[dict]):
super().__init__(
hass,
LOGGER,
+ config_entry=entry,
name=name,
update_interval=update_interval,
update_method=update_method,
@@ -49,7 +50,6 @@ class RainMachineDataUpdateCoordinator(DataUpdateCoordinator[dict]):
self._rebooting = False
self._signal_handler_unsubs: list[Callable[[], None]] = []
- self.config_entry = entry
self.signal_reboot_completed = SIGNAL_REBOOT_COMPLETED.format(
self.config_entry.entry_id
)
diff --git a/homeassistant/components/rainmachine/select.py b/homeassistant/components/rainmachine/select.py
index 1d9225a5bb2..5b23a5d79ef 100644
--- a/homeassistant/components/rainmachine/select.py
+++ b/homeassistant/components/rainmachine/select.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM, UnitSystem
from . import RainMachineConfigEntry, RainMachineData
@@ -83,7 +83,7 @@ SELECT_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RainMachineConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up RainMachine selects based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/rainmachine/sensor.py b/homeassistant/components/rainmachine/sensor.py
index 64f9ecf3990..4677a6d8bca 100644
--- a/homeassistant/components/rainmachine/sensor.py
+++ b/homeassistant/components/rainmachine/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfVolume
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utc_from_timestamp, utcnow
from . import RainMachineConfigEntry, RainMachineData
@@ -153,7 +153,7 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RainMachineConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up RainMachine sensors based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/rainmachine/switch.py b/homeassistant/components/rainmachine/switch.py
index 2a065f18976..9b62b15d196 100644
--- a/homeassistant/components/rainmachine/switch.py
+++ b/homeassistant/components/rainmachine/switch.py
@@ -17,7 +17,7 @@ from homeassistant.const import ATTR_ID, EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from . import RainMachineConfigEntry, RainMachineData, async_update_programs_and_zones
@@ -174,7 +174,7 @@ RESTRICTIONS_SWITCH_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RainMachineConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up RainMachine switches based on a config entry."""
platform = entity_platform.async_get_current_platform()
diff --git a/homeassistant/components/rainmachine/update.py b/homeassistant/components/rainmachine/update.py
index 39156b05cd4..312937184e4 100644
--- a/homeassistant/components/rainmachine/update.py
+++ b/homeassistant/components/rainmachine/update.py
@@ -16,7 +16,7 @@ from homeassistant.components.update import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RainMachineConfigEntry
from .const import DATA_MACHINE_FIRMWARE_UPDATE_STATUS
@@ -60,7 +60,7 @@ UPDATE_DESCRIPTION = RainMachineUpdateEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: RainMachineConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Rainmachine update based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/random/binary_sensor.py b/homeassistant/components/random/binary_sensor.py
index fadc966bc3d..1af85b43486 100644
--- a/homeassistant/components/random/binary_sensor.py
+++ b/homeassistant/components/random/binary_sensor.py
@@ -17,7 +17,10 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE_CLASS, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
DEFAULT_NAME = "Random binary sensor"
@@ -44,7 +47,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
async_add_entities(
diff --git a/homeassistant/components/random/sensor.py b/homeassistant/components/random/sensor.py
index 590b391c3a0..6ea296c791e 100644
--- a/homeassistant/components/random/sensor.py
+++ b/homeassistant/components/random/sensor.py
@@ -22,7 +22,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import DEFAULT_MAX, DEFAULT_MIN
@@ -57,7 +60,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
diff --git a/homeassistant/components/rapt_ble/sensor.py b/homeassistant/components/rapt_ble/sensor.py
index fd88cbcb54c..01aeedbd344 100644
--- a/homeassistant/components/rapt_ble/sensor.py
+++ b/homeassistant/components/rapt_ble/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -99,7 +99,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the RAPT Pill BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/rdw/binary_sensor.py b/homeassistant/components/rdw/binary_sensor.py
index 5360ce4a7fe..58e1c2e8237 100644
--- a/homeassistant/components/rdw/binary_sensor.py
+++ b/homeassistant/components/rdw/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -49,7 +49,7 @@ BINARY_SENSORS: tuple[RDWBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up RDW binary sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/rdw/sensor.py b/homeassistant/components/rdw/sensor.py
index 2c9c9addcfb..4133082bcf4 100644
--- a/homeassistant/components/rdw/sensor.py
+++ b/homeassistant/components/rdw/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -51,7 +51,7 @@ SENSORS: tuple[RDWSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up RDW sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/recollect_waste/calendar.py b/homeassistant/components/recollect_waste/calendar.py
index 3a76451358e..8145a93a2b7 100644
--- a/homeassistant/components/recollect_waste/calendar.py
+++ b/homeassistant/components/recollect_waste/calendar.py
@@ -9,7 +9,7 @@ from aiorecollect.client import PickupEvent
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN
@@ -35,7 +35,9 @@ def async_get_calendar_event_from_pickup_event(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ReCollect Waste sensors based on a config entry."""
coordinator: DataUpdateCoordinator[list[PickupEvent]] = hass.data[DOMAIN][
diff --git a/homeassistant/components/recollect_waste/sensor.py b/homeassistant/components/recollect_waste/sensor.py
index 36658fb5008..69b1772b9fa 100644
--- a/homeassistant/components/recollect_waste/sensor.py
+++ b/homeassistant/components/recollect_waste/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import DOMAIN, LOGGER
@@ -39,7 +39,9 @@ SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ReCollect Waste sensors based on a config entry."""
coordinator: DataUpdateCoordinator[list[PickupEvent]] = hass.data[DOMAIN][
diff --git a/homeassistant/components/recorder/__init__.py b/homeassistant/components/recorder/__init__.py
index 5a95ace92cb..7cb71e70f65 100644
--- a/homeassistant/components/recorder/__init__.py
+++ b/homeassistant/components/recorder/__init__.py
@@ -149,9 +149,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
commit_interval = conf[CONF_COMMIT_INTERVAL]
db_max_retries = conf[CONF_DB_MAX_RETRIES]
db_retry_wait = conf[CONF_DB_RETRY_WAIT]
- db_url = conf.get(CONF_DB_URL) or DEFAULT_URL.format(
- hass_config_path=hass.config.path(DEFAULT_DB_FILE)
- )
+ db_url = conf.get(CONF_DB_URL) or get_default_url(hass)
exclude = conf[CONF_EXCLUDE]
exclude_event_types: set[EventType[Any] | str] = set(
exclude.get(CONF_EVENT_TYPES, [])
@@ -200,3 +198,8 @@ async def _async_setup_integration_platform(
instance.queue_task(AddRecorderPlatformTask(domain, platform))
await async_process_integration_platforms(hass, DOMAIN, _process_recorder_platform)
+
+
+def get_default_url(hass: HomeAssistant) -> str:
+ """Return the default URL."""
+ return DEFAULT_URL.format(hass_config_path=hass.config.path(DEFAULT_DB_FILE))
diff --git a/homeassistant/components/recorder/auto_repairs/schema.py b/homeassistant/components/recorder/auto_repairs/schema.py
index 1373f466bc2..cf3addd4f20 100644
--- a/homeassistant/components/recorder/auto_repairs/schema.py
+++ b/homeassistant/components/recorder/auto_repairs/schema.py
@@ -175,7 +175,7 @@ def _validate_db_schema_precision(
# Mark the session as read_only to ensure that the test data is not committed
# to the database and we always rollback when the scope is exited
with session_scope(session=instance.get_session(), read_only=True) as session:
- db_object = table_object(**{column: PRECISE_NUMBER for column in columns})
+ db_object = table_object(**dict.fromkeys(columns, PRECISE_NUMBER))
table = table_object.__tablename__
try:
session.add(db_object)
@@ -184,7 +184,7 @@ def _validate_db_schema_precision(
_check_columns(
schema_errors=schema_errors,
stored={column: getattr(db_object, column) for column in columns},
- expected={column: PRECISE_NUMBER for column in columns},
+ expected=dict.fromkeys(columns, PRECISE_NUMBER),
columns=columns,
table_name=table,
supports="double precision",
diff --git a/homeassistant/components/recorder/backup.py b/homeassistant/components/recorder/backup.py
index d47cbe92bd4..eeebe328007 100644
--- a/homeassistant/components/recorder/backup.py
+++ b/homeassistant/components/recorder/backup.py
@@ -2,7 +2,7 @@
from logging import getLogger
-from homeassistant.core import HomeAssistant
+from homeassistant.core import CoreState, HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from .util import async_migration_in_progress, get_instance
@@ -14,6 +14,8 @@ async def async_pre_backup(hass: HomeAssistant) -> None:
"""Perform operations before a backup starts."""
_LOGGER.info("Backup start notification, locking database for writes")
instance = get_instance(hass)
+ if hass.state is not CoreState.running:
+ raise HomeAssistantError("Home Assistant is not running")
if async_migration_in_progress(hass):
raise HomeAssistantError("Database migration in progress")
await instance.lock_database()
diff --git a/homeassistant/components/recorder/basic_websocket_api.py b/homeassistant/components/recorder/basic_websocket_api.py
index 9cbc77b30c0..ce9aa452fae 100644
--- a/homeassistant/components/recorder/basic_websocket_api.py
+++ b/homeassistant/components/recorder/basic_websocket_api.py
@@ -8,7 +8,9 @@ import voluptuous as vol
from homeassistant.components import websocket_api
from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers import recorder as recorder_helper
+from . import get_default_url
from .util import get_instance
@@ -23,30 +25,28 @@ def async_setup(hass: HomeAssistant) -> None:
vol.Required("type"): "recorder/info",
}
)
-@callback
-def ws_info(
+@websocket_api.async_response
+async def ws_info(
hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]
) -> None:
"""Return status of the recorder."""
- if instance := get_instance(hass):
- backlog = instance.backlog
- migration_in_progress = instance.migration_in_progress
- migration_is_live = instance.migration_is_live
- recording = instance.recording
- # We avoid calling is_alive() as it can block waiting
- # for the thread state lock which will block the event loop.
- is_running = instance.is_running
- max_backlog = instance.max_backlog
- else:
- backlog = None
- migration_in_progress = False
- migration_is_live = False
- recording = False
- is_running = False
- max_backlog = None
+ # Wait for db_connected to ensure the recorder instance is created and the
+ # migration flags are set.
+ await hass.data[recorder_helper.DATA_RECORDER].db_connected
+ instance = get_instance(hass)
+ backlog = instance.backlog
+ db_in_default_location = instance.db_url == get_default_url(hass)
+ migration_in_progress = instance.migration_in_progress
+ migration_is_live = instance.migration_is_live
+ recording = instance.recording
+ # We avoid calling is_alive() as it can block waiting
+ # for the thread state lock which will block the event loop.
+ is_running = instance.is_running
+ max_backlog = instance.max_backlog
recorder_info = {
"backlog": backlog,
+ "db_in_default_location": db_in_default_location,
"max_backlog": max_backlog,
"migration_in_progress": migration_in_progress,
"migration_is_live": migration_is_live,
diff --git a/homeassistant/components/recorder/const.py b/homeassistant/components/recorder/const.py
index c91845e8436..36ff63a0496 100644
--- a/homeassistant/components/recorder/const.py
+++ b/homeassistant/components/recorder/const.py
@@ -30,6 +30,12 @@ CONF_DB_INTEGRITY_CHECK = "db_integrity_check"
MAX_QUEUE_BACKLOG_MIN_VALUE = 65000
MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG = 256 * 1024**2
+# As soon as we have more than 999 ids, split the query as the
+# MySQL optimizer handles it poorly and will no longer
+# do an index only scan with a group-by
+# https://github.com/home-assistant/core/issues/132865#issuecomment-2543160459
+MAX_IDS_FOR_INDEXED_GROUP_BY = 999
+
# The maximum number of rows (events) we purge in one delete statement
DEFAULT_MAX_BIND_VARS = 4000
@@ -50,6 +56,11 @@ STATES_META_SCHEMA_VERSION = 38
LAST_REPORTED_SCHEMA_VERSION = 43
LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION = 28
+LEGACY_STATES_EVENT_FOREIGN_KEYS_FIXED_SCHEMA_VERSION = 43
+# https://github.com/home-assistant/core/pull/120779
+# fixed the foreign keys in the states table but it did
+# not bump the schema version which means only databases
+# created with schema 44 and later do not need the rebuild.
INTEGRATION_PLATFORM_COMPILE_STATISTICS = "compile_statistics"
INTEGRATION_PLATFORM_LIST_STATISTIC_IDS = "list_statistic_ids"
diff --git a/homeassistant/components/recorder/core.py b/homeassistant/components/recorder/core.py
index 05a5731e791..62afa0e7b04 100644
--- a/homeassistant/components/recorder/core.py
+++ b/homeassistant/components/recorder/core.py
@@ -43,6 +43,7 @@ from homeassistant.helpers.event import (
async_track_time_interval,
async_track_utc_time_change,
)
+from homeassistant.helpers.recorder import DATA_RECORDER
from homeassistant.helpers.start import async_at_started
from homeassistant.helpers.typing import UNDEFINED, UndefinedType
from homeassistant.util import dt as dt_util
@@ -122,8 +123,6 @@ from .util import (
_LOGGER = logging.getLogger(__name__)
-DEFAULT_URL = "sqlite:///{hass_config_path}"
-
# Controls how often we clean up
# States and Events objects
EXPIRE_AFTER_COMMITS = 120
@@ -183,7 +182,7 @@ class Recorder(threading.Thread):
self.db_retry_wait = db_retry_wait
self.database_engine: DatabaseEngine | None = None
# Database connection is ready, but non-live migration may be in progress
- db_connected: asyncio.Future[bool] = hass.data[DOMAIN].db_connected
+ db_connected: asyncio.Future[bool] = hass.data[DATA_RECORDER].db_connected
self.async_db_connected: asyncio.Future[bool] = db_connected
# Database is ready to use but live migration may be in progress
self.async_db_ready: asyncio.Future[bool] = hass.loop.create_future()
diff --git a/homeassistant/components/recorder/db_schema.py b/homeassistant/components/recorder/db_schema.py
index d1a2405406e..bc8fcd1310e 100644
--- a/homeassistant/components/recorder/db_schema.py
+++ b/homeassistant/components/recorder/db_schema.py
@@ -203,11 +203,11 @@ UINT_32_TYPE = BigInteger().with_variant(
"mariadb",
)
JSON_VARIANT_CAST = Text().with_variant(
- postgresql.JSON(none_as_null=True), # type: ignore[no-untyped-call]
+ postgresql.JSON(none_as_null=True),
"postgresql",
)
JSONB_VARIANT_CAST = Text().with_variant(
- postgresql.JSONB(none_as_null=True), # type: ignore[no-untyped-call]
+ postgresql.JSONB(none_as_null=True),
"postgresql",
)
DATETIME_TYPE = (
diff --git a/homeassistant/components/recorder/history/modern.py b/homeassistant/components/recorder/history/modern.py
index 8958913bce6..566e30713f0 100644
--- a/homeassistant/components/recorder/history/modern.py
+++ b/homeassistant/components/recorder/history/modern.py
@@ -6,11 +6,12 @@ from collections.abc import Callable, Iterable, Iterator
from datetime import datetime
from itertools import groupby
from operator import itemgetter
-from typing import Any, cast
+from typing import TYPE_CHECKING, Any, cast
from sqlalchemy import (
CompoundSelect,
Select,
+ StatementLambdaElement,
Subquery,
and_,
func,
@@ -26,8 +27,9 @@ from homeassistant.const import COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_
from homeassistant.core import HomeAssistant, State, split_entity_id
from homeassistant.helpers.recorder import get_instance
from homeassistant.util import dt as dt_util
+from homeassistant.util.collection import chunked_or_all
-from ..const import LAST_REPORTED_SCHEMA_VERSION
+from ..const import LAST_REPORTED_SCHEMA_VERSION, MAX_IDS_FOR_INDEXED_GROUP_BY
from ..db_schema import (
SHARED_ATTR_OR_LEGACY_ATTRIBUTES,
StateAttributes,
@@ -149,6 +151,7 @@ def _significant_states_stmt(
no_attributes: bool,
include_start_time_state: bool,
run_start_ts: float | None,
+ slow_dependent_subquery: bool,
) -> Select | CompoundSelect:
"""Query the database for significant state changes."""
include_last_changed = not significant_changes_only
@@ -187,6 +190,7 @@ def _significant_states_stmt(
metadata_ids,
no_attributes,
include_last_changed,
+ slow_dependent_subquery,
).subquery(),
no_attributes,
include_last_changed,
@@ -257,7 +261,68 @@ def get_significant_states_with_session(
start_time_ts = start_time.timestamp()
end_time_ts = datetime_to_timestamp_or_none(end_time)
single_metadata_id = metadata_ids[0] if len(metadata_ids) == 1 else None
- stmt = lambda_stmt(
+ rows: list[Row] = []
+ if TYPE_CHECKING:
+ assert instance.database_engine is not None
+ slow_dependent_subquery = instance.database_engine.optimizer.slow_dependent_subquery
+ if include_start_time_state and slow_dependent_subquery:
+ # https://github.com/home-assistant/core/issues/137178
+ # If we include the start time state we need to limit the
+ # number of metadata_ids we query for at a time to avoid
+ # hitting limits in the MySQL optimizer that prevent
+ # the start time state query from using an index-only optimization
+ # to find the start time state.
+ iter_metadata_ids = chunked_or_all(metadata_ids, MAX_IDS_FOR_INDEXED_GROUP_BY)
+ else:
+ iter_metadata_ids = (metadata_ids,)
+ for metadata_ids_chunk in iter_metadata_ids:
+ stmt = _generate_significant_states_with_session_stmt(
+ start_time_ts,
+ end_time_ts,
+ single_metadata_id,
+ metadata_ids_chunk,
+ metadata_ids_in_significant_domains,
+ significant_changes_only,
+ no_attributes,
+ include_start_time_state,
+ oldest_ts,
+ slow_dependent_subquery,
+ )
+ row_chunk = cast(
+ list[Row],
+ execute_stmt_lambda_element(session, stmt, None, end_time, orm_rows=False),
+ )
+ if rows:
+ rows += row_chunk
+ else:
+ # If we have no rows yet, we can just assign the chunk
+ # as this is the common case since its rare that
+ # we exceed the MAX_IDS_FOR_INDEXED_GROUP_BY limit
+ rows = row_chunk
+ return _sorted_states_to_dict(
+ rows,
+ start_time_ts if include_start_time_state else None,
+ entity_ids,
+ entity_id_to_metadata_id,
+ minimal_response,
+ compressed_state_format,
+ no_attributes=no_attributes,
+ )
+
+
+def _generate_significant_states_with_session_stmt(
+ start_time_ts: float,
+ end_time_ts: float | None,
+ single_metadata_id: int | None,
+ metadata_ids: list[int],
+ metadata_ids_in_significant_domains: list[int],
+ significant_changes_only: bool,
+ no_attributes: bool,
+ include_start_time_state: bool,
+ oldest_ts: float | None,
+ slow_dependent_subquery: bool,
+) -> StatementLambdaElement:
+ return lambda_stmt(
lambda: _significant_states_stmt(
start_time_ts,
end_time_ts,
@@ -268,6 +333,7 @@ def get_significant_states_with_session(
no_attributes,
include_start_time_state,
oldest_ts,
+ slow_dependent_subquery,
),
track_on=[
bool(single_metadata_id),
@@ -276,17 +342,9 @@ def get_significant_states_with_session(
significant_changes_only,
no_attributes,
include_start_time_state,
+ slow_dependent_subquery,
],
)
- return _sorted_states_to_dict(
- execute_stmt_lambda_element(session, stmt, None, end_time, orm_rows=False),
- start_time_ts if include_start_time_state else None,
- entity_ids,
- entity_id_to_metadata_id,
- minimal_response,
- compressed_state_format,
- no_attributes=no_attributes,
- )
def get_full_significant_states_with_session(
@@ -554,13 +612,14 @@ def get_last_state_changes(
)
-def _get_start_time_state_for_entities_stmt(
+def _get_start_time_state_for_entities_stmt_dependent_sub_query(
epoch_time: float,
metadata_ids: list[int],
no_attributes: bool,
include_last_changed: bool,
) -> Select:
"""Baked query to get states for specific entities."""
+ # Engine has a fast dependent subquery optimizer
# This query is the result of significant research in
# https://github.com/home-assistant/core/issues/132865
# A reverse index scan with a limit 1 is the fastest way to get the
@@ -570,7 +629,9 @@ def _get_start_time_state_for_entities_stmt(
# before a specific point in time for all entities.
stmt = (
_stmt_and_join_attributes_for_start_state(
- no_attributes, include_last_changed, False
+ no_attributes=no_attributes,
+ include_last_changed=include_last_changed,
+ include_last_reported=False,
)
.select_from(StatesMeta)
.join(
@@ -600,6 +661,55 @@ def _get_start_time_state_for_entities_stmt(
)
+def _get_start_time_state_for_entities_stmt_group_by(
+ epoch_time: float,
+ metadata_ids: list[int],
+ no_attributes: bool,
+ include_last_changed: bool,
+) -> Select:
+ """Baked query to get states for specific entities."""
+ # Simple group-by for MySQL, must use less
+ # than 1000 metadata_ids in the IN clause for MySQL
+ # or it will optimize poorly. Callers are responsible
+ # for ensuring that the number of metadata_ids is less
+ # than 1000.
+ most_recent_states_for_entities_by_date = (
+ select(
+ States.metadata_id.label("max_metadata_id"),
+ func.max(States.last_updated_ts).label("max_last_updated"),
+ )
+ .filter(
+ (States.last_updated_ts < epoch_time) & States.metadata_id.in_(metadata_ids)
+ )
+ .group_by(States.metadata_id)
+ .subquery()
+ )
+ stmt = (
+ _stmt_and_join_attributes_for_start_state(
+ no_attributes=no_attributes,
+ include_last_changed=include_last_changed,
+ include_last_reported=False,
+ )
+ .join(
+ most_recent_states_for_entities_by_date,
+ and_(
+ States.metadata_id
+ == most_recent_states_for_entities_by_date.c.max_metadata_id,
+ States.last_updated_ts
+ == most_recent_states_for_entities_by_date.c.max_last_updated,
+ ),
+ )
+ .filter(
+ (States.last_updated_ts < epoch_time) & States.metadata_id.in_(metadata_ids)
+ )
+ )
+ if no_attributes:
+ return stmt
+ return stmt.outerjoin(
+ StateAttributes, (States.attributes_id == StateAttributes.attributes_id)
+ )
+
+
def _get_oldest_possible_ts(
hass: HomeAssistant, utc_point_in_time: datetime
) -> float | None:
@@ -620,6 +730,7 @@ def _get_start_time_state_stmt(
metadata_ids: list[int],
no_attributes: bool,
include_last_changed: bool,
+ slow_dependent_subquery: bool,
) -> Select:
"""Return the states at a specific point in time."""
if single_metadata_id:
@@ -634,7 +745,15 @@ def _get_start_time_state_stmt(
)
# We have more than one entity to look at so we need to do a query on states
# since the last recorder run started.
- return _get_start_time_state_for_entities_stmt(
+ if slow_dependent_subquery:
+ return _get_start_time_state_for_entities_stmt_group_by(
+ epoch_time,
+ metadata_ids,
+ no_attributes,
+ include_last_changed,
+ )
+
+ return _get_start_time_state_for_entities_stmt_dependent_sub_query(
epoch_time,
metadata_ids,
no_attributes,
diff --git a/homeassistant/components/recorder/manifest.json b/homeassistant/components/recorder/manifest.json
index 0b8532bedea..f5336e2a85b 100644
--- a/homeassistant/components/recorder/manifest.json
+++ b/homeassistant/components/recorder/manifest.json
@@ -7,8 +7,8 @@
"iot_class": "local_push",
"quality_scale": "internal",
"requirements": [
- "SQLAlchemy==2.0.38",
- "fnv-hash-fast==1.2.2",
+ "SQLAlchemy==2.0.39",
+ "fnv-hash-fast==1.4.0",
"psutil-home-assistant==0.0.1"
]
}
diff --git a/homeassistant/components/recorder/migration.py b/homeassistant/components/recorder/migration.py
index c6cdd6d317f..c5eea0f7088 100644
--- a/homeassistant/components/recorder/migration.py
+++ b/homeassistant/components/recorder/migration.py
@@ -9,7 +9,7 @@ from dataclasses import dataclass, replace as dataclass_replace
from datetime import timedelta
import logging
from time import time
-from typing import TYPE_CHECKING, Any, cast, final
+from typing import TYPE_CHECKING, Any, TypedDict, cast, final
from uuid import UUID
import sqlalchemy
@@ -52,6 +52,7 @@ from .auto_repairs.statistics.schema import (
from .const import (
CONTEXT_ID_AS_BINARY_SCHEMA_VERSION,
EVENT_TYPE_IDS_SCHEMA_VERSION,
+ LEGACY_STATES_EVENT_FOREIGN_KEYS_FIXED_SCHEMA_VERSION,
LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION,
STATES_META_SCHEMA_VERSION,
SupportedDialect,
@@ -711,6 +712,11 @@ def _modify_columns(
raise
+class _FKAlterDict(TypedDict):
+ old_fk: ForeignKeyConstraint
+ columns: list[str]
+
+
def _update_states_table_with_foreign_key_options(
session_maker: Callable[[], Session], engine: Engine
) -> None:
@@ -728,7 +734,7 @@ def _update_states_table_with_foreign_key_options(
inspector = sqlalchemy.inspect(engine)
tmp_states_table = Table(TABLE_STATES, MetaData())
- alters = [
+ alters: list[_FKAlterDict] = [
{
"old_fk": ForeignKeyConstraint(
(), (), name=foreign_key["name"], table=tmp_states_table
@@ -754,14 +760,14 @@ def _update_states_table_with_foreign_key_options(
with session_scope(session=session_maker()) as session:
try:
connection = session.connection()
- connection.execute(DropConstraint(alter["old_fk"])) # type: ignore[no-untyped-call]
+ connection.execute(DropConstraint(alter["old_fk"]))
for fkc in states_key_constraints:
if fkc.column_keys == alter["columns"]:
# AddConstraint mutates the constraint passed to it, we need to
# undo that to avoid changing the behavior of the table schema.
# https://github.com/sqlalchemy/sqlalchemy/blob/96f1172812f858fead45cdc7874abac76f45b339/lib/sqlalchemy/sql/ddl.py#L746-L748
create_rule = fkc._create_rule # noqa: SLF001
- add_constraint = AddConstraint(fkc) # type: ignore[no-untyped-call]
+ add_constraint = AddConstraint(fkc)
fkc._create_rule = create_rule # noqa: SLF001
connection.execute(add_constraint)
except (InternalError, OperationalError):
@@ -799,7 +805,7 @@ def _drop_foreign_key_constraints(
with session_scope(session=session_maker()) as session:
try:
connection = session.connection()
- connection.execute(DropConstraint(drop)) # type: ignore[no-untyped-call]
+ connection.execute(DropConstraint(drop))
except (InternalError, OperationalError):
_LOGGER.exception(
"Could not drop foreign constraints in %s table on %s",
@@ -844,7 +850,7 @@ def _restore_foreign_key_constraints(
# undo that to avoid changing the behavior of the table schema.
# https://github.com/sqlalchemy/sqlalchemy/blob/96f1172812f858fead45cdc7874abac76f45b339/lib/sqlalchemy/sql/ddl.py#L746-L748
create_rule = constraint._create_rule # noqa: SLF001
- add_constraint = AddConstraint(constraint) # type: ignore[no-untyped-call]
+ add_constraint = AddConstraint(constraint)
constraint._create_rule = create_rule # noqa: SLF001
try:
_add_constraint(session_maker, add_constraint, table, column)
@@ -2490,9 +2496,10 @@ class BaseMigration(ABC):
if self.initial_schema_version > self.max_initial_schema_version:
_LOGGER.debug(
"Data migration '%s' not needed, database created with version %s "
- "after migrator was added",
+ "after migrator was added in version %s",
self.migration_id,
self.initial_schema_version,
+ self.max_initial_schema_version,
)
return False
if self.start_schema_version < self.required_schema_version:
@@ -2868,7 +2875,14 @@ class EventIDPostMigration(BaseRunTimeMigration):
"""Migration to remove old event_id index from states."""
migration_id = "event_id_post_migration"
- max_initial_schema_version = LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION - 1
+ # Note we don't subtract 1 from the max_initial_schema_version
+ # in this case because we need to run this migration on databases
+ # version >= 43 because the schema was not bumped when the table
+ # rebuild was added in
+ # https://github.com/home-assistant/core/pull/120779
+ # which means its only safe to assume version 44 and later
+ # do not need the table rebuild
+ max_initial_schema_version = LEGACY_STATES_EVENT_FOREIGN_KEYS_FIXED_SCHEMA_VERSION
task = MigrationTask
migration_version = 2
diff --git a/homeassistant/components/recorder/models/database.py b/homeassistant/components/recorder/models/database.py
index b86fd299793..2a4924edab3 100644
--- a/homeassistant/components/recorder/models/database.py
+++ b/homeassistant/components/recorder/models/database.py
@@ -37,3 +37,13 @@ class DatabaseOptimizer:
# https://wiki.postgresql.org/wiki/Loose_indexscan
# https://github.com/home-assistant/core/issues/126084
slow_range_in_select: bool
+
+ # MySQL 8.x+ can end up with a file-sort on a dependent subquery
+ # which makes the query painfully slow.
+ # https://github.com/home-assistant/core/issues/137178
+ # The solution is to use multiple indexed group-by queries instead
+ # of the subquery as long as the group by does not exceed
+ # 999 elements since as soon as we hit 1000 elements MySQL
+ # will no longer use the group_index_range optimization.
+ # https://github.com/home-assistant/core/issues/132865#issuecomment-2543160459
+ slow_dependent_subquery: bool
diff --git a/homeassistant/components/recorder/statistics.py b/homeassistant/components/recorder/statistics.py
index 2b6640270ed..e26a69c0db9 100644
--- a/homeassistant/components/recorder/statistics.py
+++ b/homeassistant/components/recorder/statistics.py
@@ -24,9 +24,11 @@ import voluptuous as vol
from homeassistant.const import ATTR_UNIT_OF_MEASUREMENT
from homeassistant.core import HomeAssistant, callback, valid_entity_id
from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.recorder import DATA_RECORDER
from homeassistant.helpers.singleton import singleton
from homeassistant.helpers.typing import UNDEFINED, UndefinedType
from homeassistant.util import dt as dt_util
+from homeassistant.util.collection import chunked_or_all
from homeassistant.util.unit_conversion import (
AreaConverter,
BaseUnitConverter,
@@ -58,6 +60,7 @@ from .const import (
INTEGRATION_PLATFORM_LIST_STATISTIC_IDS,
INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES,
INTEGRATION_PLATFORM_VALIDATE_STATISTICS,
+ MAX_IDS_FOR_INDEXED_GROUP_BY,
SupportedDialect,
)
from .db_schema import (
@@ -133,31 +136,28 @@ QUERY_STATISTICS_SUMMARY_SUM = (
STATISTIC_UNIT_TO_UNIT_CONVERTER: dict[str | None, type[BaseUnitConverter]] = {
- **{unit: AreaConverter for unit in AreaConverter.VALID_UNITS},
- **{
- unit: BloodGlucoseConcentrationConverter
- for unit in BloodGlucoseConcentrationConverter.VALID_UNITS
- },
- **{unit: ConductivityConverter for unit in ConductivityConverter.VALID_UNITS},
- **{unit: DataRateConverter for unit in DataRateConverter.VALID_UNITS},
- **{unit: DistanceConverter for unit in DistanceConverter.VALID_UNITS},
- **{unit: DurationConverter for unit in DurationConverter.VALID_UNITS},
- **{unit: ElectricCurrentConverter for unit in ElectricCurrentConverter.VALID_UNITS},
- **{
- unit: ElectricPotentialConverter
- for unit in ElectricPotentialConverter.VALID_UNITS
- },
- **{unit: EnergyConverter for unit in EnergyConverter.VALID_UNITS},
- **{unit: EnergyDistanceConverter for unit in EnergyDistanceConverter.VALID_UNITS},
- **{unit: InformationConverter for unit in InformationConverter.VALID_UNITS},
- **{unit: MassConverter for unit in MassConverter.VALID_UNITS},
- **{unit: PowerConverter for unit in PowerConverter.VALID_UNITS},
- **{unit: PressureConverter for unit in PressureConverter.VALID_UNITS},
- **{unit: SpeedConverter for unit in SpeedConverter.VALID_UNITS},
- **{unit: TemperatureConverter for unit in TemperatureConverter.VALID_UNITS},
- **{unit: UnitlessRatioConverter for unit in UnitlessRatioConverter.VALID_UNITS},
- **{unit: VolumeConverter for unit in VolumeConverter.VALID_UNITS},
- **{unit: VolumeFlowRateConverter for unit in VolumeFlowRateConverter.VALID_UNITS},
+ **dict.fromkeys(AreaConverter.VALID_UNITS, AreaConverter),
+ **dict.fromkeys(
+ BloodGlucoseConcentrationConverter.VALID_UNITS,
+ BloodGlucoseConcentrationConverter,
+ ),
+ **dict.fromkeys(ConductivityConverter.VALID_UNITS, ConductivityConverter),
+ **dict.fromkeys(DataRateConverter.VALID_UNITS, DataRateConverter),
+ **dict.fromkeys(DistanceConverter.VALID_UNITS, DistanceConverter),
+ **dict.fromkeys(DurationConverter.VALID_UNITS, DurationConverter),
+ **dict.fromkeys(ElectricCurrentConverter.VALID_UNITS, ElectricCurrentConverter),
+ **dict.fromkeys(ElectricPotentialConverter.VALID_UNITS, ElectricPotentialConverter),
+ **dict.fromkeys(EnergyConverter.VALID_UNITS, EnergyConverter),
+ **dict.fromkeys(EnergyDistanceConverter.VALID_UNITS, EnergyDistanceConverter),
+ **dict.fromkeys(InformationConverter.VALID_UNITS, InformationConverter),
+ **dict.fromkeys(MassConverter.VALID_UNITS, MassConverter),
+ **dict.fromkeys(PowerConverter.VALID_UNITS, PowerConverter),
+ **dict.fromkeys(PressureConverter.VALID_UNITS, PressureConverter),
+ **dict.fromkeys(SpeedConverter.VALID_UNITS, SpeedConverter),
+ **dict.fromkeys(TemperatureConverter.VALID_UNITS, TemperatureConverter),
+ **dict.fromkeys(UnitlessRatioConverter.VALID_UNITS, UnitlessRatioConverter),
+ **dict.fromkeys(VolumeConverter.VALID_UNITS, VolumeConverter),
+ **dict.fromkeys(VolumeFlowRateConverter.VALID_UNITS, VolumeFlowRateConverter),
}
@@ -561,7 +561,9 @@ def _compile_statistics(
platform_stats: list[StatisticResult] = []
current_metadata: dict[str, tuple[int, StatisticMetaData]] = {}
# Collect statistics from all platforms implementing support
- for domain, platform in instance.hass.data[DOMAIN].recorder_platforms.items():
+ for domain, platform in instance.hass.data[
+ DATA_RECORDER
+ ].recorder_platforms.items():
if not (
platform_compile_statistics := getattr(
platform, INTEGRATION_PLATFORM_COMPILE_STATISTICS, None
@@ -599,7 +601,7 @@ def _compile_statistics(
if start.minute == 50:
# Once every hour, update issues
- for platform in instance.hass.data[DOMAIN].recorder_platforms.values():
+ for platform in instance.hass.data[DATA_RECORDER].recorder_platforms.values():
if not (
platform_update_issues := getattr(
platform, INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES, None
@@ -882,7 +884,7 @@ def list_statistic_ids(
# the integrations for the missing ones.
#
# Query all integrations with a registered recorder platform
- for platform in hass.data[DOMAIN].recorder_platforms.values():
+ for platform in hass.data[DATA_RECORDER].recorder_platforms.values():
if not (
platform_list_statistic_ids := getattr(
platform, INTEGRATION_PLATFORM_LIST_STATISTIC_IDS, None
@@ -1666,6 +1668,7 @@ def _augment_result_with_change(
drop_sum = "sum" not in _types
prev_sums = {}
if tmp := _statistics_at_time(
+ get_instance(hass),
session,
{metadata[statistic_id][0] for statistic_id in result},
table,
@@ -2024,7 +2027,39 @@ def get_latest_short_term_statistics_with_session(
)
-def _generate_statistics_at_time_stmt(
+def _generate_statistics_at_time_stmt_group_by(
+ table: type[StatisticsBase],
+ metadata_ids: set[int],
+ start_time_ts: float,
+ types: set[Literal["last_reset", "max", "mean", "min", "state", "sum"]],
+) -> StatementLambdaElement:
+ """Create the statement for finding the statistics for a given time."""
+ # Simple group-by for MySQL, must use less
+ # than 1000 metadata_ids in the IN clause for MySQL
+ # or it will optimize poorly. Callers are responsible
+ # for ensuring that the number of metadata_ids is less
+ # than 1000.
+ return _generate_select_columns_for_types_stmt(table, types) + (
+ lambda q: q.join(
+ most_recent_statistic_ids := (
+ select(
+ func.max(table.start_ts).label("max_start_ts"),
+ table.metadata_id.label("max_metadata_id"),
+ )
+ .filter(table.start_ts < start_time_ts)
+ .filter(table.metadata_id.in_(metadata_ids))
+ .group_by(table.metadata_id)
+ .subquery()
+ ),
+ and_(
+ table.start_ts == most_recent_statistic_ids.c.max_start_ts,
+ table.metadata_id == most_recent_statistic_ids.c.max_metadata_id,
+ ),
+ )
+ )
+
+
+def _generate_statistics_at_time_stmt_dependent_sub_query(
table: type[StatisticsBase],
metadata_ids: set[int],
start_time_ts: float,
@@ -2038,8 +2073,7 @@ def _generate_statistics_at_time_stmt(
# databases. Since all databases support this query as a join
# condition we can use it as a subquery to get the last start_time_ts
# before a specific point in time for all entities.
- stmt = _generate_select_columns_for_types_stmt(table, types)
- stmt += (
+ return _generate_select_columns_for_types_stmt(table, types) + (
lambda q: q.select_from(StatisticsMeta)
.join(
table,
@@ -2061,10 +2095,10 @@ def _generate_statistics_at_time_stmt(
)
.where(table.metadata_id.in_(metadata_ids))
)
- return stmt
def _statistics_at_time(
+ instance: Recorder,
session: Session,
metadata_ids: set[int],
table: type[StatisticsBase],
@@ -2073,8 +2107,41 @@ def _statistics_at_time(
) -> Sequence[Row] | None:
"""Return last known statistics, earlier than start_time, for the metadata_ids."""
start_time_ts = start_time.timestamp()
- stmt = _generate_statistics_at_time_stmt(table, metadata_ids, start_time_ts, types)
- return cast(Sequence[Row], execute_stmt_lambda_element(session, stmt))
+ if TYPE_CHECKING:
+ assert instance.database_engine is not None
+ if not instance.database_engine.optimizer.slow_dependent_subquery:
+ stmt = _generate_statistics_at_time_stmt_dependent_sub_query(
+ table=table,
+ metadata_ids=metadata_ids,
+ start_time_ts=start_time_ts,
+ types=types,
+ )
+ return cast(list[Row], execute_stmt_lambda_element(session, stmt))
+ rows: list[Row] = []
+ # https://github.com/home-assistant/core/issues/132865
+ # If we include the start time state we need to limit the
+ # number of metadata_ids we query for at a time to avoid
+ # hitting limits in the MySQL optimizer that prevent
+ # the start time state query from using an index-only optimization
+ # to find the start time state.
+ for metadata_ids_chunk in chunked_or_all(
+ metadata_ids, MAX_IDS_FOR_INDEXED_GROUP_BY
+ ):
+ stmt = _generate_statistics_at_time_stmt_group_by(
+ table=table,
+ metadata_ids=metadata_ids_chunk,
+ start_time_ts=start_time_ts,
+ types=types,
+ )
+ row_chunk = cast(list[Row], execute_stmt_lambda_element(session, stmt))
+ if rows:
+ rows += row_chunk
+ else:
+ # If we have no rows yet, we can just assign the chunk
+ # as this is the common case since its rare that
+ # we exceed the MAX_IDS_FOR_INDEXED_GROUP_BY limit
+ rows = row_chunk
+ return rows
def _build_sum_converted_stats(
@@ -2232,7 +2299,7 @@ def _sorted_statistics_to_dict(
def validate_statistics(hass: HomeAssistant) -> dict[str, list[ValidationIssue]]:
"""Validate statistics."""
platform_validation: dict[str, list[ValidationIssue]] = {}
- for platform in hass.data[DOMAIN].recorder_platforms.values():
+ for platform in hass.data[DATA_RECORDER].recorder_platforms.values():
if platform_validate_statistics := getattr(
platform, INTEGRATION_PLATFORM_VALIDATE_STATISTICS, None
):
@@ -2243,7 +2310,7 @@ def validate_statistics(hass: HomeAssistant) -> dict[str, list[ValidationIssue]]
def update_statistics_issues(hass: HomeAssistant) -> None:
"""Update statistics issues."""
with session_scope(hass=hass, read_only=True) as session:
- for platform in hass.data[DOMAIN].recorder_platforms.values():
+ for platform in hass.data[DATA_RECORDER].recorder_platforms.values():
if platform_update_statistics_issues := getattr(
platform, INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES, None
):
diff --git a/homeassistant/components/recorder/tasks.py b/homeassistant/components/recorder/tasks.py
index fa10c12aa68..4eb9547ee9d 100644
--- a/homeassistant/components/recorder/tasks.py
+++ b/homeassistant/components/recorder/tasks.py
@@ -11,11 +11,11 @@ import logging
import threading
from typing import TYPE_CHECKING, Any
+from homeassistant.helpers.recorder import DATA_RECORDER
from homeassistant.helpers.typing import UndefinedType
from homeassistant.util.event_type import EventType
from . import entity_registry, purge, statistics
-from .const import DOMAIN
from .db_schema import Statistics, StatisticsShortTerm
from .models import StatisticData, StatisticMetaData
from .util import periodic_db_cleanups, session_scope
@@ -308,7 +308,7 @@ class AddRecorderPlatformTask(RecorderTask):
hass = instance.hass
domain = self.domain
platform = self.platform
- platforms: dict[str, Any] = hass.data[DOMAIN].recorder_platforms
+ platforms: dict[str, Any] = hass.data[DATA_RECORDER].recorder_platforms
platforms[domain] = platform
diff --git a/homeassistant/components/recorder/util.py b/homeassistant/components/recorder/util.py
index a686c7c6498..0acaf0aa68f 100644
--- a/homeassistant/components/recorder/util.py
+++ b/homeassistant/components/recorder/util.py
@@ -464,6 +464,7 @@ def setup_connection_for_dialect(
"""Execute statements needed for dialect connection."""
version: AwesomeVersion | None = None
slow_range_in_select = False
+ slow_dependent_subquery = False
if dialect_name == SupportedDialect.SQLITE:
if first_connection:
old_isolation = dbapi_connection.isolation_level # type: ignore[attr-defined]
@@ -505,9 +506,8 @@ def setup_connection_for_dialect(
result = query_on_connection(dbapi_connection, "SELECT VERSION()")
version_string = result[0][0]
version = _extract_version_from_server_response(version_string)
- is_maria_db = "mariadb" in version_string.lower()
- if is_maria_db:
+ if "mariadb" in version_string.lower():
if not version or version < MIN_VERSION_MARIA_DB:
_raise_if_version_unsupported(
version or version_string, "MariaDB", MIN_VERSION_MARIA_DB
@@ -523,19 +523,21 @@ def setup_connection_for_dialect(
instance.hass,
version,
)
-
+ slow_range_in_select = bool(
+ not version
+ or version < MARIADB_WITH_FIXED_IN_QUERIES_105
+ or MARIA_DB_106 <= version < MARIADB_WITH_FIXED_IN_QUERIES_106
+ or MARIA_DB_107 <= version < MARIADB_WITH_FIXED_IN_QUERIES_107
+ or MARIA_DB_108 <= version < MARIADB_WITH_FIXED_IN_QUERIES_108
+ )
elif not version or version < MIN_VERSION_MYSQL:
_raise_if_version_unsupported(
version or version_string, "MySQL", MIN_VERSION_MYSQL
)
-
- slow_range_in_select = bool(
- not version
- or version < MARIADB_WITH_FIXED_IN_QUERIES_105
- or MARIA_DB_106 <= version < MARIADB_WITH_FIXED_IN_QUERIES_106
- or MARIA_DB_107 <= version < MARIADB_WITH_FIXED_IN_QUERIES_107
- or MARIA_DB_108 <= version < MARIADB_WITH_FIXED_IN_QUERIES_108
- )
+ else:
+ # MySQL
+ # https://github.com/home-assistant/core/issues/137178
+ slow_dependent_subquery = True
# Ensure all times are using UTC to avoid issues with daylight savings
execute_on_connection(dbapi_connection, "SET time_zone = '+00:00'")
@@ -565,7 +567,10 @@ def setup_connection_for_dialect(
return DatabaseEngine(
dialect=SupportedDialect(dialect_name),
version=version,
- optimizer=DatabaseOptimizer(slow_range_in_select=slow_range_in_select),
+ optimizer=DatabaseOptimizer(
+ slow_range_in_select=slow_range_in_select,
+ slow_dependent_subquery=slow_dependent_subquery,
+ ),
max_bind_vars=DEFAULT_MAX_BIND_VARS,
)
diff --git a/homeassistant/components/recorder/websocket_api.py b/homeassistant/components/recorder/websocket_api.py
index 03d9e725170..d23ecab3dac 100644
--- a/homeassistant/components/recorder/websocket_api.py
+++ b/homeassistant/components/recorder/websocket_api.py
@@ -297,13 +297,13 @@ async def ws_list_statistic_ids(
async def ws_validate_statistics(
hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]
) -> None:
- """Fetch a list of available statistic_id."""
+ """Validate statistics and return issues found."""
instance = get_instance(hass)
- statistic_ids = await instance.async_add_executor_job(
+ validation_issues = await instance.async_add_executor_job(
validate_statistics,
hass,
)
- connection.send_result(msg["id"], statistic_ids)
+ connection.send_result(msg["id"], validation_issues)
@websocket_api.websocket_command(
diff --git a/homeassistant/components/refoss/__init__.py b/homeassistant/components/refoss/__init__.py
index 0f0c852b043..eb2085efda4 100644
--- a/homeassistant/components/refoss/__init__.py
+++ b/homeassistant/components/refoss/__init__.py
@@ -24,7 +24,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Refoss from a config entry."""
hass.data.setdefault(DOMAIN, {})
discover = await refoss_discovery_server(hass)
- refoss_discovery = DiscoveryService(hass, discover)
+ refoss_discovery = DiscoveryService(hass, entry, discover)
hass.data[DOMAIN][DATA_DISCOVERY_SERVICE] = refoss_discovery
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/refoss/bridge.py b/homeassistant/components/refoss/bridge.py
index 11e92620fbb..a3ba9ea663d 100644
--- a/homeassistant/components/refoss/bridge.py
+++ b/homeassistant/components/refoss/bridge.py
@@ -6,6 +6,7 @@ from refoss_ha.device import DeviceInfo
from refoss_ha.device_manager import async_build_base_device
from refoss_ha.discovery import Discovery, Listener
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_send
@@ -16,9 +17,12 @@ from .coordinator import RefossDataUpdateCoordinator
class DiscoveryService(Listener):
"""Discovery event handler for refoss devices."""
- def __init__(self, hass: HomeAssistant, discovery: Discovery) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, discovery: Discovery
+ ) -> None:
"""Init discovery service."""
self.hass = hass
+ self.config_entry = config_entry
self.discovery = discovery
self.discovery.add_listener(self)
@@ -32,7 +36,7 @@ class DiscoveryService(Listener):
if device is None:
return
- coordo = RefossDataUpdateCoordinator(self.hass, device)
+ coordo = RefossDataUpdateCoordinator(self.hass, self.config_entry, device)
self.hass.data[DOMAIN][COORDINATORS].append(coordo)
await coordo.async_refresh()
diff --git a/homeassistant/components/refoss/coordinator.py b/homeassistant/components/refoss/coordinator.py
index 929d1b3962b..381f64614b5 100644
--- a/homeassistant/components/refoss/coordinator.py
+++ b/homeassistant/components/refoss/coordinator.py
@@ -7,6 +7,7 @@ from datetime import timedelta
from refoss_ha.controller.device import BaseDevice
from refoss_ha.exceptions import DeviceTimeoutError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -16,11 +17,16 @@ from .const import _LOGGER, DOMAIN, MAX_ERRORS
class RefossDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Manages polling for state changes from the device."""
- def __init__(self, hass: HomeAssistant, device: BaseDevice) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, device: BaseDevice
+ ) -> None:
"""Initialize the data update coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"{DOMAIN}-{device.device_info.dev_name}",
update_interval=timedelta(seconds=15),
)
diff --git a/homeassistant/components/refoss/sensor.py b/homeassistant/components/refoss/sensor.py
index 7065470657f..92090a192e8 100644
--- a/homeassistant/components/refoss/sensor.py
+++ b/homeassistant/components/refoss/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .bridge import RefossDataUpdateCoordinator
@@ -94,7 +94,7 @@ SENSORS: dict[str, tuple[RefossSensorEntityDescription, ...]] = {
key="energy",
translation_key="this_month_energy",
device_class=SensorDeviceClass.ENERGY,
- state_class=SensorStateClass.TOTAL,
+ state_class=SensorStateClass.TOTAL_INCREASING,
native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
suggested_display_precision=2,
subkey="mConsume",
@@ -104,7 +104,7 @@ SENSORS: dict[str, tuple[RefossSensorEntityDescription, ...]] = {
key="energy_returned",
translation_key="this_month_energy_returned",
device_class=SensorDeviceClass.ENERGY,
- state_class=SensorStateClass.TOTAL,
+ state_class=SensorStateClass.TOTAL_INCREASING,
native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
suggested_display_precision=2,
subkey="mConsume",
@@ -117,7 +117,7 @@ SENSORS: dict[str, tuple[RefossSensorEntityDescription, ...]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Refoss device from a config entry."""
diff --git a/homeassistant/components/refoss/switch.py b/homeassistant/components/refoss/switch.py
index aed132ecc3a..1d465f7f319 100644
--- a/homeassistant/components/refoss/switch.py
+++ b/homeassistant/components/refoss/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .bridge import RefossDataUpdateCoordinator
from .const import _LOGGER, COORDINATORS, DISPATCH_DEVICE_DISCOVERED, DOMAIN
@@ -20,7 +20,7 @@ from .entity import RefossEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Refoss device from a config entry."""
diff --git a/homeassistant/components/remember_the_milk/__init__.py b/homeassistant/components/remember_the_milk/__init__.py
index 0d1c54efb56..df9eec0622f 100644
--- a/homeassistant/components/remember_the_milk/__init__.py
+++ b/homeassistant/components/remember_the_milk/__init__.py
@@ -1,33 +1,25 @@
"""Support to interact with Remember The Milk."""
-import json
-import logging
-import os
-
from rtmapi import Rtm
import voluptuous as vol
from homeassistant.components import configurator
-from homeassistant.const import CONF_API_KEY, CONF_ID, CONF_NAME, CONF_TOKEN
+from homeassistant.const import CONF_API_KEY, CONF_ID, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_component import EntityComponent
from homeassistant.helpers.typing import ConfigType
+from .const import LOGGER
from .entity import RememberTheMilkEntity
+from .storage import RememberTheMilkConfiguration
# httplib2 is a transitive dependency from RtmAPI. If this dependency is not
# set explicitly, the library does not work.
-_LOGGER = logging.getLogger(__name__)
DOMAIN = "remember_the_milk"
-DEFAULT_NAME = DOMAIN
CONF_SHARED_SECRET = "shared_secret"
-CONF_ID_MAP = "id_map"
-CONF_LIST_ID = "list_id"
-CONF_TIMESERIES_ID = "timeseries_id"
-CONF_TASK_ID = "task_id"
RTM_SCHEMA = vol.Schema(
{
@@ -41,7 +33,6 @@ CONFIG_SCHEMA = vol.Schema(
{DOMAIN: vol.All(cv.ensure_list, [RTM_SCHEMA])}, extra=vol.ALLOW_EXTRA
)
-CONFIG_FILE_NAME = ".remember_the_milk.conf"
SERVICE_CREATE_TASK = "create_task"
SERVICE_COMPLETE_TASK = "complete_task"
@@ -54,17 +45,17 @@ SERVICE_SCHEMA_COMPLETE_TASK = vol.Schema({vol.Required(CONF_ID): cv.string})
def setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the Remember the milk component."""
- component = EntityComponent[RememberTheMilkEntity](_LOGGER, DOMAIN, hass)
+ component = EntityComponent[RememberTheMilkEntity](LOGGER, DOMAIN, hass)
stored_rtm_config = RememberTheMilkConfiguration(hass)
for rtm_config in config[DOMAIN]:
account_name = rtm_config[CONF_NAME]
- _LOGGER.debug("Adding Remember the milk account %s", account_name)
+ LOGGER.debug("Adding Remember the milk account %s", account_name)
api_key = rtm_config[CONF_API_KEY]
shared_secret = rtm_config[CONF_SHARED_SECRET]
token = stored_rtm_config.get_token(account_name)
if token:
- _LOGGER.debug("found token for account %s", account_name)
+ LOGGER.debug("found token for account %s", account_name)
_create_instance(
hass,
account_name,
@@ -79,13 +70,19 @@ def setup(hass: HomeAssistant, config: ConfigType) -> bool:
hass, account_name, api_key, shared_secret, stored_rtm_config, component
)
- _LOGGER.debug("Finished adding all Remember the milk accounts")
+ LOGGER.debug("Finished adding all Remember the milk accounts")
return True
def _create_instance(
- hass, account_name, api_key, shared_secret, token, stored_rtm_config, component
-):
+ hass: HomeAssistant,
+ account_name: str,
+ api_key: str,
+ shared_secret: str,
+ token: str,
+ stored_rtm_config: RememberTheMilkConfiguration,
+ component: EntityComponent[RememberTheMilkEntity],
+) -> None:
entity = RememberTheMilkEntity(
account_name, api_key, shared_secret, token, stored_rtm_config
)
@@ -105,26 +102,30 @@ def _create_instance(
def _register_new_account(
- hass, account_name, api_key, shared_secret, stored_rtm_config, component
-):
- request_id = None
+ hass: HomeAssistant,
+ account_name: str,
+ api_key: str,
+ shared_secret: str,
+ stored_rtm_config: RememberTheMilkConfiguration,
+ component: EntityComponent[RememberTheMilkEntity],
+) -> None:
api = Rtm(api_key, shared_secret, "write", None)
url, frob = api.authenticate_desktop()
- _LOGGER.debug("Sent authentication request to server")
+ LOGGER.debug("Sent authentication request to server")
def register_account_callback(fields: list[dict[str, str]]) -> None:
"""Call for register the configurator."""
api.retrieve_token(frob)
token = api.token
if api.token is None:
- _LOGGER.error("Failed to register, please try again")
+ LOGGER.error("Failed to register, please try again")
configurator.notify_errors(
hass, request_id, "Failed to register, please try again."
)
return
stored_rtm_config.set_token(account_name, token)
- _LOGGER.debug("Retrieved new token from server")
+ LOGGER.debug("Retrieved new token from server")
_create_instance(
hass,
@@ -152,89 +153,3 @@ def _register_new_account(
link_url=url,
submit_caption="login completed",
)
-
-
-class RememberTheMilkConfiguration:
- """Internal configuration data for RememberTheMilk class.
-
- This class stores the authentication token it get from the backend.
- """
-
- def __init__(self, hass):
- """Create new instance of configuration."""
- self._config_file_path = hass.config.path(CONFIG_FILE_NAME)
- if not os.path.isfile(self._config_file_path):
- self._config = {}
- return
- try:
- _LOGGER.debug("Loading configuration from file: %s", self._config_file_path)
- with open(self._config_file_path, encoding="utf8") as config_file:
- self._config = json.load(config_file)
- except ValueError:
- _LOGGER.error(
- "Failed to load configuration file, creating a new one: %s",
- self._config_file_path,
- )
- self._config = {}
-
- def save_config(self):
- """Write the configuration to a file."""
- with open(self._config_file_path, "w", encoding="utf8") as config_file:
- json.dump(self._config, config_file)
-
- def get_token(self, profile_name):
- """Get the server token for a profile."""
- if profile_name in self._config:
- return self._config[profile_name][CONF_TOKEN]
- return None
-
- def set_token(self, profile_name, token):
- """Store a new server token for a profile."""
- self._initialize_profile(profile_name)
- self._config[profile_name][CONF_TOKEN] = token
- self.save_config()
-
- def delete_token(self, profile_name):
- """Delete a token for a profile.
-
- Usually called when the token has expired.
- """
- self._config.pop(profile_name, None)
- self.save_config()
-
- def _initialize_profile(self, profile_name):
- """Initialize the data structures for a profile."""
- if profile_name not in self._config:
- self._config[profile_name] = {}
- if CONF_ID_MAP not in self._config[profile_name]:
- self._config[profile_name][CONF_ID_MAP] = {}
-
- def get_rtm_id(self, profile_name, hass_id):
- """Get the RTM ids for a Home Assistant task ID.
-
- The id of a RTM tasks consists of the tuple:
- list id, timeseries id and the task id.
- """
- self._initialize_profile(profile_name)
- ids = self._config[profile_name][CONF_ID_MAP].get(hass_id)
- if ids is None:
- return None
- return ids[CONF_LIST_ID], ids[CONF_TIMESERIES_ID], ids[CONF_TASK_ID]
-
- def set_rtm_id(self, profile_name, hass_id, list_id, time_series_id, rtm_task_id):
- """Add/Update the RTM task ID for a Home Assistant task IS."""
- self._initialize_profile(profile_name)
- id_tuple = {
- CONF_LIST_ID: list_id,
- CONF_TIMESERIES_ID: time_series_id,
- CONF_TASK_ID: rtm_task_id,
- }
- self._config[profile_name][CONF_ID_MAP][hass_id] = id_tuple
- self.save_config()
-
- def delete_rtm_id(self, profile_name, hass_id):
- """Delete a key mapping."""
- self._initialize_profile(profile_name)
- if hass_id in self._config[profile_name][CONF_ID_MAP]:
- del self._config[profile_name][CONF_ID_MAP][hass_id]
- self.save_config()
diff --git a/homeassistant/components/remember_the_milk/const.py b/homeassistant/components/remember_the_milk/const.py
new file mode 100644
index 00000000000..2fccbf3ee52
--- /dev/null
+++ b/homeassistant/components/remember_the_milk/const.py
@@ -0,0 +1,5 @@
+"""Constants for the Remember The Milk integration."""
+
+import logging
+
+LOGGER = logging.getLogger(__package__)
diff --git a/homeassistant/components/remember_the_milk/entity.py b/homeassistant/components/remember_the_milk/entity.py
index 8fa52b6c06c..be69d16f72f 100644
--- a/homeassistant/components/remember_the_milk/entity.py
+++ b/homeassistant/components/remember_the_milk/entity.py
@@ -1,20 +1,26 @@
"""Support to interact with Remember The Milk."""
-import logging
-
from rtmapi import Rtm, RtmRequestFailedException
from homeassistant.const import CONF_ID, CONF_NAME, STATE_OK
from homeassistant.core import ServiceCall
from homeassistant.helpers.entity import Entity
-_LOGGER = logging.getLogger(__name__)
+from .const import LOGGER
+from .storage import RememberTheMilkConfiguration
class RememberTheMilkEntity(Entity):
"""Representation of an interface to Remember The Milk."""
- def __init__(self, name, api_key, shared_secret, token, rtm_config):
+ def __init__(
+ self,
+ name: str,
+ api_key: str,
+ shared_secret: str,
+ token: str,
+ rtm_config: RememberTheMilkConfiguration,
+ ) -> None:
"""Create new instance of Remember The Milk component."""
self._name = name
self._api_key = api_key
@@ -22,11 +28,11 @@ class RememberTheMilkEntity(Entity):
self._token = token
self._rtm_config = rtm_config
self._rtm_api = Rtm(api_key, shared_secret, "delete", token)
- self._token_valid = None
+ self._token_valid = False
self._check_token()
- _LOGGER.debug("Instance created for account %s", self._name)
+ LOGGER.debug("Instance created for account %s", self._name)
- def _check_token(self):
+ def _check_token(self) -> bool:
"""Check if the API token is still valid.
If it is not valid any more, delete it from the configuration. This
@@ -34,7 +40,7 @@ class RememberTheMilkEntity(Entity):
"""
valid = self._rtm_api.token_valid()
if not valid:
- _LOGGER.error(
+ LOGGER.error(
"Token for account %s is invalid. You need to register again!",
self.name,
)
@@ -60,20 +66,21 @@ class RememberTheMilkEntity(Entity):
result = self._rtm_api.rtm.timelines.create()
timeline = result.timeline.value
- if hass_id is None or rtm_id is None:
+ if rtm_id is None:
result = self._rtm_api.rtm.tasks.add(
timeline=timeline, name=task_name, parse="1"
)
- _LOGGER.debug(
+ LOGGER.debug(
"Created new task '%s' in account %s", task_name, self.name
)
- self._rtm_config.set_rtm_id(
- self._name,
- hass_id,
- result.list.id,
- result.list.taskseries.id,
- result.list.taskseries.task.id,
- )
+ if hass_id is not None:
+ self._rtm_config.set_rtm_id(
+ self._name,
+ hass_id,
+ result.list.id,
+ result.list.taskseries.id,
+ result.list.taskseries.task.id,
+ )
else:
self._rtm_api.rtm.tasks.setName(
name=task_name,
@@ -82,14 +89,14 @@ class RememberTheMilkEntity(Entity):
task_id=rtm_id[2],
timeline=timeline,
)
- _LOGGER.debug(
+ LOGGER.debug(
"Updated task with id '%s' in account %s to name %s",
hass_id,
self.name,
task_name,
)
except RtmRequestFailedException as rtm_exception:
- _LOGGER.error(
+ LOGGER.error(
"Error creating new Remember The Milk task for account %s: %s",
self._name,
rtm_exception,
@@ -100,7 +107,7 @@ class RememberTheMilkEntity(Entity):
hass_id = call.data[CONF_ID]
rtm_id = self._rtm_config.get_rtm_id(self._name, hass_id)
if rtm_id is None:
- _LOGGER.error(
+ LOGGER.error(
(
"Could not find task with ID %s in account %s. "
"So task could not be closed"
@@ -119,23 +126,21 @@ class RememberTheMilkEntity(Entity):
timeline=timeline,
)
self._rtm_config.delete_rtm_id(self._name, hass_id)
- _LOGGER.debug(
- "Completed task with id %s in account %s", hass_id, self._name
- )
+ LOGGER.debug("Completed task with id %s in account %s", hass_id, self._name)
except RtmRequestFailedException as rtm_exception:
- _LOGGER.error(
+ LOGGER.error(
"Error creating new Remember The Milk task for account %s: %s",
self._name,
rtm_exception,
)
@property
- def name(self):
+ def name(self) -> str:
"""Return the name of the device."""
return self._name
@property
- def state(self):
+ def state(self) -> str:
"""Return the state of the device."""
if not self._token_valid:
return "API token invalid"
diff --git a/homeassistant/components/remember_the_milk/storage.py b/homeassistant/components/remember_the_milk/storage.py
new file mode 100644
index 00000000000..593abb7da2c
--- /dev/null
+++ b/homeassistant/components/remember_the_milk/storage.py
@@ -0,0 +1,116 @@
+"""Store RTM configuration in Home Assistant storage."""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+from typing import cast
+
+from homeassistant.const import CONF_TOKEN
+from homeassistant.core import HomeAssistant
+
+from .const import LOGGER
+
+CONFIG_FILE_NAME = ".remember_the_milk.conf"
+CONF_ID_MAP = "id_map"
+CONF_LIST_ID = "list_id"
+CONF_TASK_ID = "task_id"
+CONF_TIMESERIES_ID = "timeseries_id"
+
+
+class RememberTheMilkConfiguration:
+ """Internal configuration data for Remember The Milk."""
+
+ def __init__(self, hass: HomeAssistant) -> None:
+ """Create new instance of configuration."""
+ self._config_file_path = hass.config.path(CONFIG_FILE_NAME)
+ self._config = {}
+ LOGGER.debug("Loading configuration from file: %s", self._config_file_path)
+ try:
+ self._config = json.loads(
+ Path(self._config_file_path).read_text(encoding="utf8")
+ )
+ except FileNotFoundError:
+ LOGGER.debug("Missing configuration file: %s", self._config_file_path)
+ except OSError:
+ LOGGER.debug(
+ "Failed to read from configuration file, %s, using empty configuration",
+ self._config_file_path,
+ )
+ except ValueError:
+ LOGGER.error(
+ "Failed to parse configuration file, %s, using empty configuration",
+ self._config_file_path,
+ )
+
+ def _save_config(self) -> None:
+ """Write the configuration to a file."""
+ Path(self._config_file_path).write_text(
+ json.dumps(self._config), encoding="utf8"
+ )
+
+ def get_token(self, profile_name: str) -> str | None:
+ """Get the server token for a profile."""
+ if profile_name in self._config:
+ return cast(str, self._config[profile_name][CONF_TOKEN])
+ return None
+
+ def set_token(self, profile_name: str, token: str) -> None:
+ """Store a new server token for a profile."""
+ self._initialize_profile(profile_name)
+ self._config[profile_name][CONF_TOKEN] = token
+ self._save_config()
+
+ def delete_token(self, profile_name: str) -> None:
+ """Delete a token for a profile.
+
+ Usually called when the token has expired.
+ """
+ self._config.pop(profile_name, None)
+ self._save_config()
+
+ def _initialize_profile(self, profile_name: str) -> None:
+ """Initialize the data structures for a profile."""
+ if profile_name not in self._config:
+ self._config[profile_name] = {}
+ if CONF_ID_MAP not in self._config[profile_name]:
+ self._config[profile_name][CONF_ID_MAP] = {}
+
+ def get_rtm_id(
+ self, profile_name: str, hass_id: str
+ ) -> tuple[str, str, str] | None:
+ """Get the RTM ids for a Home Assistant task ID.
+
+ The id of a RTM tasks consists of the tuple:
+ list id, timeseries id and the task id.
+ """
+ self._initialize_profile(profile_name)
+ ids = self._config[profile_name][CONF_ID_MAP].get(hass_id)
+ if ids is None:
+ return None
+ return ids[CONF_LIST_ID], ids[CONF_TIMESERIES_ID], ids[CONF_TASK_ID]
+
+ def set_rtm_id(
+ self,
+ profile_name: str,
+ hass_id: str,
+ list_id: str,
+ time_series_id: str,
+ rtm_task_id: str,
+ ) -> None:
+ """Add/Update the RTM task ID for a Home Assistant task IS."""
+ self._initialize_profile(profile_name)
+ id_tuple = {
+ CONF_LIST_ID: list_id,
+ CONF_TIMESERIES_ID: time_series_id,
+ CONF_TASK_ID: rtm_task_id,
+ }
+ self._config[profile_name][CONF_ID_MAP][hass_id] = id_tuple
+ self._save_config()
+
+ def delete_rtm_id(self, profile_name: str, hass_id: str) -> None:
+ """Delete a key mapping."""
+ self._initialize_profile(profile_name)
+ if hass_id in self._config[profile_name][CONF_ID_MAP]:
+ del self._config[profile_name][CONF_ID_MAP][hass_id]
+ self._save_config()
diff --git a/homeassistant/components/remote_calendar/__init__.py b/homeassistant/components/remote_calendar/__init__.py
new file mode 100644
index 00000000000..910eeae8268
--- /dev/null
+++ b/homeassistant/components/remote_calendar/__init__.py
@@ -0,0 +1,33 @@
+"""The Remote Calendar integration."""
+
+import logging
+
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant
+
+from .const import DOMAIN
+from .coordinator import RemoteCalendarConfigEntry, RemoteCalendarDataUpdateCoordinator
+
+_LOGGER = logging.getLogger(__name__)
+
+
+PLATFORMS: list[Platform] = [Platform.CALENDAR]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant, entry: RemoteCalendarConfigEntry
+) -> bool:
+ """Set up Remote Calendar from a config entry."""
+ hass.data.setdefault(DOMAIN, {})
+ coordinator = RemoteCalendarDataUpdateCoordinator(hass, entry)
+ await coordinator.async_config_entry_first_refresh()
+ entry.runtime_data = coordinator
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+ return True
+
+
+async def async_unload_entry(
+ hass: HomeAssistant, entry: RemoteCalendarConfigEntry
+) -> bool:
+ """Handle unload of an entry."""
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/remote_calendar/calendar.py b/homeassistant/components/remote_calendar/calendar.py
new file mode 100644
index 00000000000..bd83a5f18cc
--- /dev/null
+++ b/homeassistant/components/remote_calendar/calendar.py
@@ -0,0 +1,92 @@
+"""Calendar platform for a Remote Calendar."""
+
+from datetime import datetime
+import logging
+
+from ical.event import Event
+
+from homeassistant.components.calendar import CalendarEntity, CalendarEvent
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+from homeassistant.util import dt as dt_util
+
+from . import RemoteCalendarConfigEntry
+from .const import CONF_CALENDAR_NAME
+from .coordinator import RemoteCalendarDataUpdateCoordinator
+
+_LOGGER = logging.getLogger(__name__)
+
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: RemoteCalendarConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the remote calendar platform."""
+ coordinator = entry.runtime_data
+ entity = RemoteCalendarEntity(coordinator, entry)
+ async_add_entities([entity])
+
+
+class RemoteCalendarEntity(
+ CoordinatorEntity[RemoteCalendarDataUpdateCoordinator], CalendarEntity
+):
+ """A calendar entity backed by a remote iCalendar url."""
+
+ _attr_has_entity_name = True
+
+ def __init__(
+ self,
+ coordinator: RemoteCalendarDataUpdateCoordinator,
+ entry: RemoteCalendarConfigEntry,
+ ) -> None:
+ """Initialize RemoteCalendarEntity."""
+ super().__init__(coordinator)
+ self._attr_name = entry.data[CONF_CALENDAR_NAME]
+ self._attr_unique_id = entry.entry_id
+
+ @property
+ def event(self) -> CalendarEvent | None:
+ """Return the next upcoming event."""
+ now = dt_util.now()
+ events = self.coordinator.data.timeline_tz(now.tzinfo).active_after(now)
+ if event := next(events, None):
+ return _get_calendar_event(event)
+ return None
+
+ async def async_get_events(
+ self, hass: HomeAssistant, start_date: datetime, end_date: datetime
+ ) -> list[CalendarEvent]:
+ """Get all events in a specific time frame."""
+ events = self.coordinator.data.timeline_tz(start_date.tzinfo).overlapping(
+ start_date,
+ end_date,
+ )
+ return [_get_calendar_event(event) for event in events]
+
+
+def _get_calendar_event(event: Event) -> CalendarEvent:
+ """Return a CalendarEvent from an API event."""
+
+ return CalendarEvent(
+ summary=event.summary,
+ start=(
+ dt_util.as_local(event.start)
+ if isinstance(event.start, datetime)
+ else event.start
+ ),
+ end=(
+ dt_util.as_local(event.end)
+ if isinstance(event.end, datetime)
+ else event.end
+ ),
+ description=event.description,
+ uid=event.uid,
+ rrule=event.rrule.as_rrule_str() if event.rrule else None,
+ recurrence_id=event.recurrence_id,
+ location=event.location,
+ )
diff --git a/homeassistant/components/remote_calendar/config_flow.py b/homeassistant/components/remote_calendar/config_flow.py
new file mode 100644
index 00000000000..03d0e7ea96a
--- /dev/null
+++ b/homeassistant/components/remote_calendar/config_flow.py
@@ -0,0 +1,70 @@
+"""Config flow for Remote Calendar integration."""
+
+import logging
+from typing import Any
+
+from httpx import HTTPError, InvalidURL
+from ical.calendar_stream import IcsCalendarStream
+from ical.exceptions import CalendarParseError
+import voluptuous as vol
+
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.const import CONF_URL
+from homeassistant.helpers.httpx_client import get_async_client
+
+from .const import CONF_CALENDAR_NAME, DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+STEP_USER_DATA_SCHEMA = vol.Schema(
+ {
+ vol.Required(CONF_CALENDAR_NAME): str,
+ vol.Required(CONF_URL): str,
+ }
+)
+
+
+class RemoteCalendarConfigFlow(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow for Remote Calendar."""
+
+ VERSION = 1
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle the initial step."""
+ if user_input is None:
+ return self.async_show_form(
+ step_id="user", data_schema=STEP_USER_DATA_SCHEMA
+ )
+ errors: dict = {}
+ _LOGGER.debug("User input: %s", user_input)
+ self._async_abort_entries_match(
+ {CONF_CALENDAR_NAME: user_input[CONF_CALENDAR_NAME]}
+ )
+ self._async_abort_entries_match({CONF_URL: user_input[CONF_URL]})
+ client = get_async_client(self.hass)
+ try:
+ res = await client.get(user_input[CONF_URL], follow_redirects=True)
+ res.raise_for_status()
+ except (HTTPError, InvalidURL) as err:
+ errors["base"] = "cannot_connect"
+ _LOGGER.debug("An error occurred: %s", err)
+ else:
+ try:
+ await self.hass.async_add_executor_job(
+ IcsCalendarStream.calendar_from_ics, res.text
+ )
+ except CalendarParseError as err:
+ errors["base"] = "invalid_ics_file"
+ _LOGGER.debug("Invalid .ics file: %s", err)
+ else:
+ return self.async_create_entry(
+ title=user_input[CONF_CALENDAR_NAME], data=user_input
+ )
+
+ return self.async_show_form(
+ step_id="user",
+ data_schema=STEP_USER_DATA_SCHEMA,
+ errors=errors,
+ )
diff --git a/homeassistant/components/remote_calendar/const.py b/homeassistant/components/remote_calendar/const.py
new file mode 100644
index 00000000000..060d7633111
--- /dev/null
+++ b/homeassistant/components/remote_calendar/const.py
@@ -0,0 +1,4 @@
+"""Constants for the Remote Calendar integration."""
+
+DOMAIN = "remote_calendar"
+CONF_CALENDAR_NAME = "calendar_name"
diff --git a/homeassistant/components/remote_calendar/coordinator.py b/homeassistant/components/remote_calendar/coordinator.py
new file mode 100644
index 00000000000..6caec297c1a
--- /dev/null
+++ b/homeassistant/components/remote_calendar/coordinator.py
@@ -0,0 +1,71 @@
+"""Data UpdateCoordinator for the Remote Calendar integration."""
+
+from datetime import timedelta
+import logging
+
+from httpx import HTTPError, InvalidURL
+from ical.calendar import Calendar
+from ical.calendar_stream import IcsCalendarStream
+from ical.exceptions import CalendarParseError
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_URL
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.httpx_client import get_async_client
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+from .const import DOMAIN
+
+type RemoteCalendarConfigEntry = ConfigEntry[RemoteCalendarDataUpdateCoordinator]
+
+_LOGGER = logging.getLogger(__name__)
+SCAN_INTERVAL = timedelta(days=1)
+
+
+class RemoteCalendarDataUpdateCoordinator(DataUpdateCoordinator[Calendar]):
+ """Class to manage fetching calendar data."""
+
+ config_entry: RemoteCalendarConfigEntry
+ ics: str
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: RemoteCalendarConfigEntry,
+ ) -> None:
+ """Initialize data updater."""
+ super().__init__(
+ hass,
+ _LOGGER,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
+ always_update=True,
+ )
+ self._client = get_async_client(hass)
+ self._url = config_entry.data[CONF_URL]
+
+ async def _async_update_data(self) -> Calendar:
+ """Update data from the url."""
+ try:
+ res = await self._client.get(self._url, follow_redirects=True)
+ res.raise_for_status()
+ except (HTTPError, InvalidURL) as err:
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="unable_to_fetch",
+ translation_placeholders={"err": str(err)},
+ ) from err
+ try:
+ # calendar_from_ics will dynamically load packages
+ # the first time it is called, so we need to do it
+ # in a separate thread to avoid blocking the event loop
+ self.ics = res.text
+ return await self.hass.async_add_executor_job(
+ IcsCalendarStream.calendar_from_ics, self.ics
+ )
+ except CalendarParseError as err:
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="unable_to_parse",
+ translation_placeholders={"err": str(err)},
+ ) from err
diff --git a/homeassistant/components/remote_calendar/diagnostics.py b/homeassistant/components/remote_calendar/diagnostics.py
new file mode 100644
index 00000000000..5ebfb3d3812
--- /dev/null
+++ b/homeassistant/components/remote_calendar/diagnostics.py
@@ -0,0 +1,25 @@
+"""Provides diagnostics for the remote calendar."""
+
+import datetime
+from typing import Any
+
+from ical.diagnostics import redact_ics
+
+from homeassistant.core import HomeAssistant
+from homeassistant.util import dt as dt_util
+
+from . import RemoteCalendarConfigEntry
+
+
+async def async_get_config_entry_diagnostics(
+ hass: HomeAssistant, entry: RemoteCalendarConfigEntry
+) -> dict[str, Any]:
+ """Return diagnostics for a config entry."""
+ coordinator = entry.runtime_data
+ payload: dict[str, Any] = {
+ "now": dt_util.now().isoformat(),
+ "timezone": str(dt_util.get_default_time_zone()),
+ "system_timezone": str(datetime.datetime.now().astimezone().tzinfo),
+ }
+ payload["ics"] = "\n".join(redact_ics(coordinator.ics))
+ return payload
diff --git a/homeassistant/components/remote_calendar/manifest.json b/homeassistant/components/remote_calendar/manifest.json
new file mode 100644
index 00000000000..fe17a3d2c34
--- /dev/null
+++ b/homeassistant/components/remote_calendar/manifest.json
@@ -0,0 +1,12 @@
+{
+ "domain": "remote_calendar",
+ "name": "Remote Calendar",
+ "codeowners": ["@Thomas55555"],
+ "config_flow": true,
+ "documentation": "https://www.home-assistant.io/integrations/remote_calendar",
+ "integration_type": "service",
+ "iot_class": "cloud_polling",
+ "loggers": ["ical"],
+ "quality_scale": "silver",
+ "requirements": ["ical==9.0.1"]
+}
diff --git a/homeassistant/components/remote_calendar/quality_scale.yaml b/homeassistant/components/remote_calendar/quality_scale.yaml
new file mode 100644
index 00000000000..964b63d7116
--- /dev/null
+++ b/homeassistant/components/remote_calendar/quality_scale.yaml
@@ -0,0 +1,98 @@
+rules:
+ # Bronze
+ config-flow: done
+ test-before-configure: done
+ unique-config-entry:
+ status: exempt
+ comment: |
+ No unique identifier.
+ config-flow-test-coverage: done
+ runtime-data: done
+ test-before-setup: done
+ appropriate-polling: done
+ entity-unique-id: done
+ has-entity-name: done
+ entity-event-setup:
+ status: exempt
+ comment: |
+ Entities of this integration does not explicitly subscribe to events.
+ dependency-transparency: done
+ action-setup:
+ status: exempt
+ comment: |
+ There are no actions.
+ common-modules: done
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ docs-actions:
+ status: exempt
+ comment: No actions available.
+ brands: done
+ # Silver
+ config-entry-unloading: done
+ log-when-unavailable: done
+ entity-unavailable: done
+ action-exceptions:
+ status: exempt
+ comment: |
+ There are no actions.
+ reauthentication-flow:
+ status: exempt
+ comment: |
+ There is no authentication required.
+ parallel-updates: done
+ test-coverage: done
+ integration-owner: done
+ docs-installation-parameters: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: no configuration options
+
+ # Gold
+ devices:
+ status: exempt
+ comment: No devices. One URL is always assigned to one calendar.
+ diagnostics: done
+ discovery-update-info:
+ status: todo
+ comment: No discovery protocol available.
+ discovery:
+ status: exempt
+ comment: No discovery protocol available.
+ docs-data-update: todo
+ docs-examples: todo
+ docs-known-limitations: todo
+ docs-supported-devices: todo
+ docs-supported-functions: todo
+ docs-troubleshooting: todo
+ docs-use-cases: todo
+ dynamic-devices:
+ status: exempt
+ comment: No devices. One URL is always assigned to one calendar.
+ entity-category: done
+ entity-device-class:
+ status: exempt
+ comment: No devices classes for calendars.
+ entity-disabled-by-default:
+ status: exempt
+ comment: Only one entity per entry.
+ entity-translations:
+ status: exempt
+ comment: Entity name is defined by the user, so no translation possible.
+ exception-translations: done
+ icon-translations:
+ status: exempt
+ comment: Only the default icon is used.
+ reconfiguration-flow:
+ status: exempt
+ comment: no configuration possible
+ repair-issues: todo
+ stale-devices:
+ status: exempt
+ comment: No devices. One URL is always assigned to one calendar.
+
+ # Platinum
+ async-dependency: todo
+ inject-websession: done
+ strict-typing: done
diff --git a/homeassistant/components/remote_calendar/strings.json b/homeassistant/components/remote_calendar/strings.json
new file mode 100644
index 00000000000..1ad62821818
--- /dev/null
+++ b/homeassistant/components/remote_calendar/strings.json
@@ -0,0 +1,33 @@
+{
+ "title": "Remote Calendar",
+ "config": {
+ "step": {
+ "user": {
+ "description": "Please choose a name for the calendar to be imported",
+ "data": {
+ "calendar_name": "Calendar name",
+ "url": "Calendar URL"
+ },
+ "data_description": {
+ "calendar_name": "The name of the calendar shown in the UI.",
+ "url": "The URL of the remote calendar."
+ }
+ }
+ },
+ "abort": {
+ "already_configured": "[%key:common::config_flow::abort::already_configured_service%]"
+ },
+ "error": {
+ "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
+ "invalid_ics_file": "[%key:component::local_calendar::config::error::invalid_ics_file%]"
+ }
+ },
+ "exceptions": {
+ "unable_to_fetch": {
+ "message": "Unable to fetch calendar data: {err}"
+ },
+ "unable_to_parse": {
+ "message": "Unable to parse calendar data: {err}"
+ }
+ }
+}
diff --git a/homeassistant/components/remote_rpi_gpio/binary_sensor.py b/homeassistant/components/remote_rpi_gpio/binary_sensor.py
index 42e8517c1e8..1d970bb3541 100644
--- a/homeassistant/components/remote_rpi_gpio/binary_sensor.py
+++ b/homeassistant/components/remote_rpi_gpio/binary_sensor.py
@@ -2,6 +2,7 @@
from __future__ import annotations
+from gpiozero import DigitalInputDevice
import requests
import voluptuous as vol
@@ -48,10 +49,10 @@ def setup_platform(
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up the Raspberry PI GPIO devices."""
- address = config["host"]
+ address = config[CONF_HOST]
invert_logic = config[CONF_INVERT_LOGIC]
pull_mode = config[CONF_PULL_MODE]
- ports = config["ports"]
+ ports = config[CONF_PORTS]
bouncetime = config[CONF_BOUNCETIME] / 1000
devices = []
@@ -71,9 +72,11 @@ class RemoteRPiGPIOBinarySensor(BinarySensorEntity):
_attr_should_poll = False
- def __init__(self, name, sensor, invert_logic):
+ def __init__(
+ self, name: str | None, sensor: DigitalInputDevice, invert_logic: bool
+ ) -> None:
"""Initialize the RPi binary sensor."""
- self._name = name
+ self._attr_name = name
self._invert_logic = invert_logic
self._state = False
self._sensor = sensor
@@ -90,20 +93,10 @@ class RemoteRPiGPIOBinarySensor(BinarySensorEntity):
self._sensor.when_activated = read_gpio
@property
- def name(self):
- """Return the name of the sensor."""
- return self._name
-
- @property
- def is_on(self):
+ def is_on(self) -> bool:
"""Return the state of the entity."""
return self._state != self._invert_logic
- @property
- def device_class(self):
- """Return the class of this sensor, from DEVICE_CLASSES."""
- return
-
def update(self) -> None:
"""Update the GPIO state."""
try:
diff --git a/homeassistant/components/remote_rpi_gpio/switch.py b/homeassistant/components/remote_rpi_gpio/switch.py
index 91b389c5a1e..25f95045e4b 100644
--- a/homeassistant/components/remote_rpi_gpio/switch.py
+++ b/homeassistant/components/remote_rpi_gpio/switch.py
@@ -4,6 +4,7 @@ from __future__ import annotations
from typing import Any
+from gpiozero import LED
import voluptuous as vol
from homeassistant.components.switch import (
@@ -57,37 +58,23 @@ def setup_platform(
class RemoteRPiGPIOSwitch(SwitchEntity):
"""Representation of a Remote Raspberry Pi GPIO."""
+ _attr_assumed_state = True
_attr_should_poll = False
- def __init__(self, name, led):
+ def __init__(self, name: str | None, led: LED) -> None:
"""Initialize the pin."""
- self._name = name or DEVICE_DEFAULT_NAME
- self._state = False
+ self._attr_name = name or DEVICE_DEFAULT_NAME
+ self._attr_is_on = False
self._switch = led
- @property
- def name(self):
- """Return the name of the switch."""
- return self._name
-
- @property
- def assumed_state(self):
- """If unable to access real state of the entity."""
- return True
-
- @property
- def is_on(self):
- """Return true if device is on."""
- return self._state
-
def turn_on(self, **kwargs: Any) -> None:
"""Turn the device on."""
write_output(self._switch, 1)
- self._state = True
+ self._attr_is_on = True
self.schedule_update_ha_state()
def turn_off(self, **kwargs: Any) -> None:
"""Turn the device off."""
write_output(self._switch, 0)
- self._state = False
+ self._attr_is_on = False
self.schedule_update_ha_state()
diff --git a/homeassistant/components/renault/binary_sensor.py b/homeassistant/components/renault/binary_sensor.py
index a8fdf324f1c..0aebd3bd835 100644
--- a/homeassistant/components/renault/binary_sensor.py
+++ b/homeassistant/components/renault/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import RenaultConfigEntry
@@ -37,7 +37,7 @@ class RenaultBinarySensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RenaultConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renault entities from config entry."""
entities: list[RenaultBinarySensor] = [
diff --git a/homeassistant/components/renault/button.py b/homeassistant/components/renault/button.py
index 6a9f5e05a38..82b811821ea 100644
--- a/homeassistant/components/renault/button.py
+++ b/homeassistant/components/renault/button.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RenaultConfigEntry
from .entity import RenaultEntity
@@ -29,7 +29,7 @@ class RenaultButtonEntityDescription(ButtonEntityDescription):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RenaultConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renault entities from config entry."""
entities: list[RenaultButtonEntity] = [
diff --git a/homeassistant/components/renault/coordinator.py b/homeassistant/components/renault/coordinator.py
index 89e62867130..a90331730bc 100644
--- a/homeassistant/components/renault/coordinator.py
+++ b/homeassistant/components/renault/coordinator.py
@@ -6,7 +6,7 @@ import asyncio
from collections.abc import Awaitable, Callable
from datetime import timedelta
import logging
-from typing import TypeVar
+from typing import TYPE_CHECKING, TypeVar
from renault_api.kamereon.exceptions import (
AccessDeniedException,
@@ -18,6 +18,9 @@ from renault_api.kamereon.models import KamereonVehicleDataAttributes
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import RenaultConfigEntry
+
T = TypeVar("T", bound=KamereonVehicleDataAttributes)
# We have potentially 7 coordinators per vehicle
@@ -27,11 +30,13 @@ _PARALLEL_SEMAPHORE = asyncio.Semaphore(1)
class RenaultDataUpdateCoordinator(DataUpdateCoordinator[T]):
"""Handle vehicle communication with Renault servers."""
+ config_entry: RenaultConfigEntry
update_method: Callable[[], Awaitable[T]]
def __init__(
self,
hass: HomeAssistant,
+ config_entry: RenaultConfigEntry,
logger: logging.Logger,
*,
name: str,
@@ -42,6 +47,7 @@ class RenaultDataUpdateCoordinator(DataUpdateCoordinator[T]):
super().__init__(
hass,
logger,
+ config_entry=config_entry,
name=name,
update_interval=update_interval,
update_method=update_method,
diff --git a/homeassistant/components/renault/device_tracker.py b/homeassistant/components/renault/device_tracker.py
index 08a2a698802..c55ddeb2190 100644
--- a/homeassistant/components/renault/device_tracker.py
+++ b/homeassistant/components/renault/device_tracker.py
@@ -11,7 +11,7 @@ from homeassistant.components.device_tracker import (
TrackerEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RenaultConfigEntry
from .entity import RenaultDataEntity, RenaultDataEntityDescription
@@ -30,7 +30,7 @@ class RenaultTrackerEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RenaultConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renault entities from config entry."""
entities: list[RenaultDeviceTracker] = [
diff --git a/homeassistant/components/renault/renault_hub.py b/homeassistant/components/renault/renault_hub.py
index 76b197b2aaf..b37390526cf 100644
--- a/homeassistant/components/renault/renault_hub.py
+++ b/homeassistant/components/renault/renault_hub.py
@@ -5,13 +5,13 @@ from __future__ import annotations
import asyncio
from datetime import timedelta
import logging
+from typing import TYPE_CHECKING
from renault_api.gigya.exceptions import InvalidCredentialsException
from renault_api.kamereon.models import KamereonVehiclesLink
from renault_api.renault_account import RenaultAccount
from renault_api.renault_client import RenaultClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ATTR_IDENTIFIERS,
ATTR_MANUFACTURER,
@@ -24,6 +24,9 @@ from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.aiohttp_client import async_get_clientsession
+if TYPE_CHECKING:
+ from . import RenaultConfigEntry
+
from .const import CONF_KAMEREON_ACCOUNT_ID, DEFAULT_SCAN_INTERVAL
from .renault_vehicle import RenaultVehicleProxy
@@ -52,7 +55,7 @@ class RenaultHub:
return True
return False
- async def async_initialise(self, config_entry: ConfigEntry) -> None:
+ async def async_initialise(self, config_entry: RenaultConfigEntry) -> None:
"""Set up proxy."""
account_id: str = config_entry.data[CONF_KAMEREON_ACCOUNT_ID]
scan_interval = timedelta(seconds=DEFAULT_SCAN_INTERVAL)
@@ -86,7 +89,7 @@ class RenaultHub:
vehicle_link: KamereonVehiclesLink,
renault_account: RenaultAccount,
scan_interval: timedelta,
- config_entry: ConfigEntry,
+ config_entry: RenaultConfigEntry,
device_registry: dr.DeviceRegistry,
) -> None:
"""Set up proxy."""
@@ -95,6 +98,7 @@ class RenaultHub:
# Generate vehicle proxy
vehicle = RenaultVehicleProxy(
hass=self._hass,
+ config_entry=config_entry,
vehicle=await renault_account.get_api_vehicle(vehicle_link.vin),
details=vehicle_link.vehicleDetails,
scan_interval=scan_interval,
diff --git a/homeassistant/components/renault/renault_vehicle.py b/homeassistant/components/renault/renault_vehicle.py
index d8266d75319..1cce0e4459f 100644
--- a/homeassistant/components/renault/renault_vehicle.py
+++ b/homeassistant/components/renault/renault_vehicle.py
@@ -8,7 +8,7 @@ from dataclasses import dataclass
from datetime import datetime, timedelta
from functools import wraps
import logging
-from typing import Any, Concatenate, cast
+from typing import TYPE_CHECKING, Any, Concatenate, cast
from renault_api.exceptions import RenaultException
from renault_api.kamereon import models
@@ -18,6 +18,9 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
+if TYPE_CHECKING:
+ from . import RenaultConfigEntry
+
from .const import DOMAIN
from .coordinator import RenaultDataUpdateCoordinator
@@ -64,12 +67,14 @@ class RenaultVehicleProxy:
def __init__(
self,
hass: HomeAssistant,
+ config_entry: RenaultConfigEntry,
vehicle: RenaultVehicle,
details: models.KamereonVehicleDetails,
scan_interval: timedelta,
) -> None:
"""Initialise vehicle proxy."""
self.hass = hass
+ self.config_entry = config_entry
self._vehicle = vehicle
self._details = details
self._device_info = DeviceInfo(
@@ -98,11 +103,10 @@ class RenaultVehicleProxy:
self.coordinators = {
coord.key: RenaultDataUpdateCoordinator(
self.hass,
+ self.config_entry,
LOGGER,
- # Name of the data. For logging purposes.
name=f"{self.details.vin} {coord.key}",
update_method=coord.update_method(self._vehicle),
- # Polling interval. Will only be polled if there are subscribers.
update_interval=self._scan_interval,
)
for coord in COORDINATORS
diff --git a/homeassistant/components/renault/select.py b/homeassistant/components/renault/select.py
index cab1d1f4d8a..cddf83bb860 100644
--- a/homeassistant/components/renault/select.py
+++ b/homeassistant/components/renault/select.py
@@ -9,7 +9,7 @@ from renault_api.kamereon.models import KamereonVehicleBatteryStatusData
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import RenaultConfigEntry
@@ -32,7 +32,7 @@ class RenaultSelectEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RenaultConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renault entities from config entry."""
entities: list[RenaultSelectEntity] = [
diff --git a/homeassistant/components/renault/sensor.py b/homeassistant/components/renault/sensor.py
index 7854d70b1c4..7c513c1b9de 100644
--- a/homeassistant/components/renault/sensor.py
+++ b/homeassistant/components/renault/sensor.py
@@ -31,7 +31,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import as_utc, parse_datetime
@@ -60,7 +60,7 @@ class RenaultSensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RenaultConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renault entities from config entry."""
entities: list[RenaultSensor[Any]] = [
diff --git a/homeassistant/components/renault/services.py b/homeassistant/components/renault/services.py
index 80fb2363b1e..df65d16b0b8 100644
--- a/homeassistant/components/renault/services.py
+++ b/homeassistant/components/renault/services.py
@@ -9,7 +9,6 @@ from typing import TYPE_CHECKING, Any
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv, device_registry as dr
@@ -178,9 +177,8 @@ def setup_services(hass: HomeAssistant) -> None:
loaded_entries: list[RenaultConfigEntry] = [
entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- and entry.entry_id in device_entry.config_entries
+ for entry in hass.config_entries.async_loaded_entries(DOMAIN)
+ if entry.entry_id in device_entry.config_entries
]
for entry in loaded_entries:
for vin, vehicle in entry.runtime_data.vehicles.items():
diff --git a/homeassistant/components/renault/strings.json b/homeassistant/components/renault/strings.json
index 7d9cae1bcf1..8649a5c7b47 100644
--- a/homeassistant/components/renault/strings.json
+++ b/homeassistant/components/renault/strings.json
@@ -18,7 +18,7 @@
"data_description": {
"kamereon_account_id": "The Kamereon account ID associated with your vehicle"
},
- "title": "Kamereon Account ID",
+ "title": "Kamereon account ID",
"description": "You have multiple Kamereon accounts associated to this email, please select one"
},
"reauth_confirm": {
@@ -228,10 +228,10 @@
},
"exceptions": {
"invalid_device_id": {
- "message": "No device with id {device_id} was found"
+ "message": "No device with ID {device_id} was found"
},
"no_config_entry_for_device": {
- "message": "No loaded config entry was found for device with id {device_id}"
+ "message": "No loaded config entry was found for device with ID {device_id}"
}
}
}
diff --git a/homeassistant/components/renson/__init__.py b/homeassistant/components/renson/__init__.py
index d1eebdf0a5f..b88f9bb036a 100644
--- a/homeassistant/components/renson/__init__.py
+++ b/homeassistant/components/renson/__init__.py
@@ -37,7 +37,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Renson from a config entry."""
api = RensonVentilation(entry.data[CONF_HOST])
- coordinator = RensonCoordinator("Renson", hass, api)
+ coordinator = RensonCoordinator(hass, entry, api)
if not await hass.async_add_executor_job(api.connect):
raise ConfigEntryNotReady("Cannot connect to Renson device")
diff --git a/homeassistant/components/renson/binary_sensor.py b/homeassistant/components/renson/binary_sensor.py
index 46f832ed15c..60b4f54b85c 100644
--- a/homeassistant/components/renson/binary_sensor.py
+++ b/homeassistant/components/renson/binary_sensor.py
@@ -24,7 +24,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RensonCoordinator
@@ -86,7 +86,7 @@ BINARY_SENSORS: tuple[RensonBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Call the Renson integration to setup."""
diff --git a/homeassistant/components/renson/button.py b/homeassistant/components/renson/button.py
index 02278a0d6f6..830e5a03a4a 100644
--- a/homeassistant/components/renson/button.py
+++ b/homeassistant/components/renson/button.py
@@ -15,7 +15,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RensonCoordinator, RensonData
from .const import DOMAIN
@@ -54,7 +54,7 @@ ENTITY_DESCRIPTIONS: tuple[RensonButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renson button platform."""
diff --git a/homeassistant/components/renson/coordinator.py b/homeassistant/components/renson/coordinator.py
index 8613220eee1..5d0a20e1c29 100644
--- a/homeassistant/components/renson/coordinator.py
+++ b/homeassistant/components/renson/coordinator.py
@@ -9,30 +9,35 @@ from typing import Any
from renson_endura_delta.renson import RensonVentilation
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
+from .const import DOMAIN
+
_LOGGER = logging.getLogger(__name__)
class RensonCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Data update coordinator for Renson."""
+ config_entry: ConfigEntry
+
def __init__(
self,
- name: str,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
api: RensonVentilation,
- update_interval=timedelta(seconds=30),
) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
# Name of the data. For logging purposes.
- name=name,
+ name=DOMAIN,
# Polling interval. Will only be polled if there are subscribers.
- update_interval=update_interval,
+ update_interval=timedelta(seconds=30),
)
self.api = api
diff --git a/homeassistant/components/renson/fan.py b/homeassistant/components/renson/fan.py
index 00edd4547cb..474ab640943 100644
--- a/homeassistant/components/renson/fan.py
+++ b/homeassistant/components/renson/fan.py
@@ -19,7 +19,7 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from homeassistant.util.percentage import (
percentage_to_ranged_value,
@@ -85,7 +85,7 @@ SPEED_RANGE: tuple[float, float] = (1, 4)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renson fan platform."""
diff --git a/homeassistant/components/renson/number.py b/homeassistant/components/renson/number.py
index fb8ab8fc552..67fde1c56dc 100644
--- a/homeassistant/components/renson/number.py
+++ b/homeassistant/components/renson/number.py
@@ -15,7 +15,7 @@ from homeassistant.components.number import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RensonCoordinator
@@ -40,7 +40,7 @@ RENSON_NUMBER_DESCRIPTION = NumberEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renson number platform."""
diff --git a/homeassistant/components/renson/sensor.py b/homeassistant/components/renson/sensor.py
index 1df62e12312..ce7e71b1c0b 100644
--- a/homeassistant/components/renson/sensor.py
+++ b/homeassistant/components/renson/sensor.py
@@ -43,7 +43,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RensonData
from .const import DOMAIN
@@ -272,7 +272,7 @@ class RensonSensor(RensonEntity, SensorEntity):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renson sensor platform."""
diff --git a/homeassistant/components/renson/switch.py b/homeassistant/components/renson/switch.py
index 2cd44d20a6a..3b73bb3dffe 100644
--- a/homeassistant/components/renson/switch.py
+++ b/homeassistant/components/renson/switch.py
@@ -11,7 +11,7 @@ from renson_endura_delta.renson import Level, RensonVentilation
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RensonCoordinator
from .const import DOMAIN
@@ -68,7 +68,7 @@ class RensonBreezeSwitch(RensonEntity, SwitchEntity):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Call the Renson integration to setup."""
diff --git a/homeassistant/components/renson/time.py b/homeassistant/components/renson/time.py
index feb47fadf99..0a07fd2ec4f 100644
--- a/homeassistant/components/renson/time.py
+++ b/homeassistant/components/renson/time.py
@@ -13,7 +13,7 @@ from homeassistant.components.time import TimeEntity, TimeEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RensonData
from .const import DOMAIN
@@ -50,7 +50,7 @@ ENTITY_DESCRIPTIONS: tuple[RensonTimeEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Renson time platform."""
diff --git a/homeassistant/components/reolink/__init__.py b/homeassistant/components/reolink/__init__.py
index 71ca5428740..99ca91c5bdf 100644
--- a/homeassistant/components/reolink/__init__.py
+++ b/homeassistant/components/reolink/__init__.py
@@ -28,7 +28,7 @@ from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.typing import ConfigType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS, DOMAIN
+from .const import CONF_BC_PORT, CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS, DOMAIN
from .exceptions import PasswordIncompatible, ReolinkException, UserNotAdmin
from .host import ReolinkHost
from .services import async_setup_services
@@ -67,9 +67,7 @@ async def async_setup_entry(
hass: HomeAssistant, config_entry: ReolinkConfigEntry
) -> bool:
"""Set up Reolink from a config entry."""
- host = ReolinkHost(
- hass, config_entry.data, config_entry.options, config_entry.entry_id
- )
+ host = ReolinkHost(hass, config_entry.data, config_entry.options, config_entry)
try:
await host.async_init()
@@ -100,6 +98,7 @@ async def async_setup_entry(
or host.api.use_https != config_entry.data[CONF_USE_HTTPS]
or host.api.supported(None, "privacy_mode")
!= config_entry.data.get(CONF_SUPPORTS_PRIVACY_MODE)
+ or host.api.baichuan.port != config_entry.data.get(CONF_BC_PORT)
):
if host.api.port != config_entry.data[CONF_PORT]:
_LOGGER.warning(
@@ -108,10 +107,21 @@ async def async_setup_entry(
config_entry.data[CONF_PORT],
host.api.port,
)
+ if (
+ config_entry.data.get(CONF_BC_PORT, host.api.baichuan.port)
+ != host.api.baichuan.port
+ ):
+ _LOGGER.warning(
+ "Baichuan port of Reolink %s, changed from %s to %s",
+ host.api.nvr_name,
+ config_entry.data.get(CONF_BC_PORT),
+ host.api.baichuan.port,
+ )
data = {
**config_entry.data,
CONF_PORT: host.api.port,
CONF_USE_HTTPS: host.api.use_https,
+ CONF_BC_PORT: host.api.baichuan.port,
CONF_SUPPORTS_PRIVACY_MODE: host.api.supported(None, "privacy_mode"),
}
hass.config_entries.async_update_entry(config_entry, data=data)
diff --git a/homeassistant/components/reolink/binary_sensor.py b/homeassistant/components/reolink/binary_sensor.py
index 2191dedc9cf..39910bbc52a 100644
--- a/homeassistant/components/reolink/binary_sensor.py
+++ b/homeassistant/components/reolink/binary_sensor.py
@@ -23,9 +23,13 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .entity import ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription
+from .entity import (
+ ReolinkChannelCoordinatorEntity,
+ ReolinkChannelEntityDescription,
+ ReolinkEntityDescription,
+)
from .util import ReolinkConfigEntry, ReolinkData
PARALLEL_UPDATES = 0
@@ -41,6 +45,18 @@ class ReolinkBinarySensorEntityDescription(
value: Callable[[Host, int], bool]
+@dataclass(frozen=True, kw_only=True)
+class ReolinkSmartAIBinarySensorEntityDescription(
+ BinarySensorEntityDescription,
+ ReolinkEntityDescription,
+):
+ """A class that describes Smart AI binary sensor entities."""
+
+ smart_type: str
+ value: Callable[[Host, int, int], bool]
+ supported: Callable[[Host, int, int], bool] = lambda api, ch, loc: True
+
+
BINARY_PUSH_SENSORS = (
ReolinkBinarySensorEntityDescription(
key="motion",
@@ -121,26 +137,173 @@ BINARY_SENSORS = (
),
)
+BINARY_SMART_AI_SENSORS = (
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="crossline_person",
+ smart_type="crossline",
+ cmd_id=33,
+ translation_key="crossline_person",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "crossline", loc, "people")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_crossline")
+ and "people" in api.baichuan.smart_ai_type_list(ch, "crossline", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="crossline_vehicle",
+ smart_type="crossline",
+ cmd_id=33,
+ translation_key="crossline_vehicle",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "crossline", loc, "vehicle")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_crossline")
+ and "vehicle" in api.baichuan.smart_ai_type_list(ch, "crossline", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="crossline_dog_cat",
+ smart_type="crossline",
+ cmd_id=33,
+ translation_key="crossline_dog_cat",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "crossline", loc, "dog_cat")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_crossline")
+ and "dog_cat" in api.baichuan.smart_ai_type_list(ch, "crossline", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="intrusion_person",
+ smart_type="intrusion",
+ cmd_id=33,
+ translation_key="intrusion_person",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "intrusion", loc, "people")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_intrusion")
+ and "people" in api.baichuan.smart_ai_type_list(ch, "intrusion", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="intrusion_vehicle",
+ smart_type="intrusion",
+ cmd_id=33,
+ translation_key="intrusion_vehicle",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "intrusion", loc, "vehicle")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_intrusion")
+ and "vehicle" in api.baichuan.smart_ai_type_list(ch, "intrusion", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="intrusion_dog_cat",
+ smart_type="intrusion",
+ cmd_id=33,
+ translation_key="intrusion_dog_cat",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "intrusion", loc, "dog_cat")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_intrusion")
+ and "dog_cat" in api.baichuan.smart_ai_type_list(ch, "intrusion", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="linger_person",
+ smart_type="loitering",
+ cmd_id=33,
+ translation_key="linger_person",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "loitering", loc, "people")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_linger")
+ and "people" in api.baichuan.smart_ai_type_list(ch, "loitering", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="linger_vehicle",
+ smart_type="loitering",
+ cmd_id=33,
+ translation_key="linger_vehicle",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "loitering", loc, "vehicle")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_linger")
+ and "vehicle" in api.baichuan.smart_ai_type_list(ch, "loitering", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="linger_dog_cat",
+ smart_type="loitering",
+ cmd_id=33,
+ translation_key="linger_dog_cat",
+ value=lambda api, ch, loc: (
+ api.baichuan.smart_ai_state(ch, "loitering", loc, "dog_cat")
+ ),
+ supported=lambda api, ch, loc: (
+ api.supported(ch, "ai_linger")
+ and "dog_cat" in api.baichuan.smart_ai_type_list(ch, "loitering", loc)
+ ),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="forgotten_item",
+ smart_type="legacy",
+ cmd_id=33,
+ translation_key="forgotten_item",
+ value=lambda api, ch, loc: (api.baichuan.smart_ai_state(ch, "legacy", loc)),
+ supported=lambda api, ch, loc: api.supported(ch, "ai_forgotten_item"),
+ ),
+ ReolinkSmartAIBinarySensorEntityDescription(
+ key="taken_item",
+ smart_type="loss",
+ cmd_id=33,
+ translation_key="taken_item",
+ value=lambda api, ch, loc: (api.baichuan.smart_ai_state(ch, "loss", loc)),
+ supported=lambda api, ch, loc: api.supported(ch, "ai_taken_item"),
+ ),
+)
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink IP Camera."""
reolink_data: ReolinkData = config_entry.runtime_data
+ api = reolink_data.host.api
- entities: list[ReolinkBinarySensorEntity] = []
- for channel in reolink_data.host.api.channels:
+ entities: list[ReolinkBinarySensorEntity | ReolinkSmartAIBinarySensorEntity] = []
+ for channel in api.channels:
entities.extend(
ReolinkPushBinarySensorEntity(reolink_data, channel, entity_description)
for entity_description in BINARY_PUSH_SENSORS
- if entity_description.supported(reolink_data.host.api, channel)
+ if entity_description.supported(api, channel)
)
entities.extend(
ReolinkBinarySensorEntity(reolink_data, channel, entity_description)
for entity_description in BINARY_SENSORS
- if entity_description.supported(reolink_data.host.api, channel)
+ if entity_description.supported(api, channel)
+ )
+ entities.extend(
+ ReolinkSmartAIBinarySensorEntity(
+ reolink_data, channel, location, entity_description
+ )
+ for entity_description in BINARY_SMART_AI_SENSORS
+ for location in api.baichuan.smart_location_list(
+ channel, entity_description.key
+ )
+ if entity_description.supported(api, channel, location)
)
async_add_entities(entities)
@@ -198,3 +361,40 @@ class ReolinkPushBinarySensorEntity(ReolinkBinarySensorEntity):
async def _async_handle_event(self, event: str) -> None:
"""Handle incoming event for motion detection."""
self.async_write_ha_state()
+
+
+class ReolinkSmartAIBinarySensorEntity(
+ ReolinkChannelCoordinatorEntity, BinarySensorEntity
+):
+ """Binary-sensor class for Reolink IP camera Smart AI sensors."""
+
+ entity_description: ReolinkSmartAIBinarySensorEntityDescription
+
+ def __init__(
+ self,
+ reolink_data: ReolinkData,
+ channel: int,
+ location: int,
+ entity_description: ReolinkSmartAIBinarySensorEntityDescription,
+ ) -> None:
+ """Initialize Reolink binary sensor."""
+ self.entity_description = entity_description
+ super().__init__(reolink_data, channel)
+ unique_index = self._host.api.baichuan.smart_ai_index(
+ channel, entity_description.smart_type, location
+ )
+ self._attr_unique_id = f"{self._attr_unique_id}_{unique_index}"
+
+ self._location = location
+ self._attr_translation_placeholders = {
+ "zone_name": self._host.api.baichuan.smart_ai_name(
+ channel, entity_description.smart_type, location
+ )
+ }
+
+ @property
+ def is_on(self) -> bool:
+ """State of the sensor."""
+ return self.entity_description.value(
+ self._host.api, self._channel, self._location
+ )
diff --git a/homeassistant/components/reolink/button.py b/homeassistant/components/reolink/button.py
index c1a2aed4119..a67b30a394c 100644
--- a/homeassistant/components/reolink/button.py
+++ b/homeassistant/components/reolink/button.py
@@ -19,7 +19,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
@@ -151,7 +151,7 @@ HOST_BUTTON_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink button entities."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/camera.py b/homeassistant/components/reolink/camera.py
index a597be3ec7a..329ef9028de 100644
--- a/homeassistant/components/reolink/camera.py
+++ b/homeassistant/components/reolink/camera.py
@@ -13,7 +13,7 @@ from homeassistant.components.camera import (
CameraEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription
from .util import ReolinkConfigEntry, ReolinkData, raise_translated_error
@@ -89,7 +89,7 @@ CAMERA_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink IP Camera."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/config_flow.py b/homeassistant/components/reolink/config_flow.py
index 7943cadef21..12ccd455be3 100644
--- a/homeassistant/components/reolink/config_flow.py
+++ b/homeassistant/components/reolink/config_flow.py
@@ -8,6 +8,7 @@ import logging
from typing import Any
from reolink_aio.api import ALLOWED_SPECIAL_CHARS
+from reolink_aio.baichuan import DEFAULT_BC_PORT
from reolink_aio.exceptions import (
ApiError,
CredentialsInvalidError,
@@ -37,7 +38,7 @@ from homeassistant.helpers import config_validation as cv, selector
from homeassistant.helpers.device_registry import format_mac
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
-from .const import CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS, DOMAIN
+from .const import CONF_BC_PORT, CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS, DOMAIN
from .exceptions import (
PasswordIncompatible,
ReolinkException,
@@ -287,6 +288,7 @@ class ReolinkFlowHandler(ConfigFlow, domain=DOMAIN):
if not errors:
user_input[CONF_PORT] = host.api.port
user_input[CONF_USE_HTTPS] = host.api.use_https
+ user_input[CONF_BC_PORT] = host.api.baichuan.port
user_input[CONF_SUPPORTS_PRIVACY_MODE] = host.api.supported(
None, "privacy_mode"
)
@@ -326,8 +328,9 @@ class ReolinkFlowHandler(ConfigFlow, domain=DOMAIN):
if errors:
data_schema = data_schema.extend(
{
- vol.Optional(CONF_PORT): cv.positive_int,
+ vol.Optional(CONF_PORT): cv.port,
vol.Required(CONF_USE_HTTPS, default=False): bool,
+ vol.Required(CONF_BC_PORT, default=DEFAULT_BC_PORT): cv.port,
}
)
diff --git a/homeassistant/components/reolink/const.py b/homeassistant/components/reolink/const.py
index 7bd93337c46..026d1219881 100644
--- a/homeassistant/components/reolink/const.py
+++ b/homeassistant/components/reolink/const.py
@@ -3,4 +3,5 @@
DOMAIN = "reolink"
CONF_USE_HTTPS = "use_https"
+CONF_BC_PORT = "baichuan_port"
CONF_SUPPORTS_PRIVACY_MODE = "privacy_mode_supported"
diff --git a/homeassistant/components/reolink/diagnostics.py b/homeassistant/components/reolink/diagnostics.py
index 693f2ba59a4..1d0e5d919e7 100644
--- a/homeassistant/components/reolink/diagnostics.py
+++ b/homeassistant/components/reolink/diagnostics.py
@@ -25,6 +25,14 @@ async def async_get_config_entry_diagnostics(
IPC_cam[ch]["firmware version"] = api.camera_sw_version(ch)
IPC_cam[ch]["encoding main"] = await api.get_encoding(ch)
+ chimes: dict[int, dict[str, Any]] = {}
+ for chime in api.chime_list:
+ chimes[chime.dev_id] = {}
+ chimes[chime.dev_id]["channel"] = chime.channel
+ chimes[chime.dev_id]["name"] = chime.name
+ chimes[chime.dev_id]["online"] = chime.online
+ chimes[chime.dev_id]["event_types"] = chime.chime_event_types
+
return {
"model": api.model,
"hardware version": api.hardware_version,
@@ -41,9 +49,11 @@ async def async_get_config_entry_diagnostics(
"channels": api.channels,
"stream channels": api.stream_channels,
"IPC cams": IPC_cam,
+ "Chimes": chimes,
"capabilities": api.capabilities,
"cmd list": host.update_cmd,
"firmware ch list": host.firmware_ch_list,
"api versions": api.checked_api_versions,
"abilities": api.abilities,
+ "BC_abilities": api.baichuan.abilities,
}
diff --git a/homeassistant/components/reolink/entity.py b/homeassistant/components/reolink/entity.py
index e3a84579865..55ce4ce891e 100644
--- a/homeassistant/components/reolink/entity.py
+++ b/homeassistant/components/reolink/entity.py
@@ -107,10 +107,10 @@ class ReolinkHostCoordinatorEntity(CoordinatorEntity[DataUpdateCoordinator[None]
"""Handle incoming TCP push event."""
self.async_write_ha_state()
- def register_callback(self, unique_id: str, cmd_id: int) -> None:
+ def register_callback(self, callback_id: str, cmd_id: int) -> None:
"""Register callback for TCP push events."""
self._host.api.baichuan.register_callback( # pragma: no cover
- unique_id, self._push_callback, cmd_id
+ callback_id, self._push_callback, cmd_id
)
async def async_added_to_hass(self) -> None:
@@ -118,23 +118,25 @@ class ReolinkHostCoordinatorEntity(CoordinatorEntity[DataUpdateCoordinator[None]
await super().async_added_to_hass()
cmd_key = self.entity_description.cmd_key
cmd_id = self.entity_description.cmd_id
+ callback_id = f"{self.platform.domain}_{self._attr_unique_id}"
if cmd_key is not None:
self._host.async_register_update_cmd(cmd_key)
if cmd_id is not None:
- self.register_callback(self._attr_unique_id, cmd_id)
+ self.register_callback(callback_id, cmd_id)
# Privacy mode
- self.register_callback(f"{self._attr_unique_id}_623", 623)
+ self.register_callback(f"{callback_id}_623", 623)
async def async_will_remove_from_hass(self) -> None:
"""Entity removed."""
cmd_key = self.entity_description.cmd_key
cmd_id = self.entity_description.cmd_id
+ callback_id = f"{self.platform.domain}_{self._attr_unique_id}"
if cmd_key is not None:
self._host.async_unregister_update_cmd(cmd_key)
if cmd_id is not None:
- self._host.api.baichuan.unregister_callback(self._attr_unique_id)
+ self._host.api.baichuan.unregister_callback(callback_id)
# Privacy mode
- self._host.api.baichuan.unregister_callback(f"{self._attr_unique_id}_623")
+ self._host.api.baichuan.unregister_callback(f"{callback_id}_623")
await super().async_will_remove_from_hass()
@@ -193,10 +195,10 @@ class ReolinkChannelCoordinatorEntity(ReolinkHostCoordinatorEntity):
"""Return True if entity is available."""
return super().available and self._host.api.camera_online(self._channel)
- def register_callback(self, unique_id: str, cmd_id: int) -> None:
+ def register_callback(self, callback_id: str, cmd_id: int) -> None:
"""Register callback for TCP push events."""
self._host.api.baichuan.register_callback(
- unique_id, self._push_callback, cmd_id, self._channel
+ callback_id, self._push_callback, cmd_id, self._channel
)
async def async_added_to_hass(self) -> None:
diff --git a/homeassistant/components/reolink/host.py b/homeassistant/components/reolink/host.py
index 2f646ba9090..a027177f1fc 100644
--- a/homeassistant/components/reolink/host.py
+++ b/homeassistant/components/reolink/host.py
@@ -12,6 +12,7 @@ from typing import Any, Literal
import aiohttp
from aiohttp.web import Request
from reolink_aio.api import ALLOWED_SPECIAL_CHARS, Host
+from reolink_aio.baichuan import DEFAULT_BC_PORT
from reolink_aio.enums import SubType
from reolink_aio.exceptions import NotSupportedError, ReolinkError, SubscriptionError
@@ -33,14 +34,14 @@ from homeassistant.helpers.network import NoURLAvailableError, get_url
from homeassistant.helpers.storage import Store
from homeassistant.util.ssl import SSLCipherList
-from .const import CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS, DOMAIN
+from .const import CONF_BC_PORT, CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS, DOMAIN
from .exceptions import (
PasswordIncompatible,
ReolinkSetupException,
ReolinkWebhookException,
UserNotAdmin,
)
-from .util import get_store
+from .util import ReolinkConfigEntry, get_store
DEFAULT_TIMEOUT = 30
FIRST_TCP_PUSH_TIMEOUT = 10
@@ -66,11 +67,11 @@ class ReolinkHost:
hass: HomeAssistant,
config: Mapping[str, Any],
options: Mapping[str, Any],
- config_entry_id: str | None = None,
+ config_entry: ReolinkConfigEntry | None = None,
) -> None:
"""Initialize Reolink Host. Could be either NVR, or Camera."""
self._hass: HomeAssistant = hass
- self._config_entry_id = config_entry_id
+ self._config_entry = config_entry
self._config = config
self._unique_id: str = ""
@@ -91,6 +92,7 @@ class ReolinkHost:
protocol=options[CONF_PROTOCOL],
timeout=DEFAULT_TIMEOUT,
aiohttp_get_session_callback=get_aiohttp_session,
+ bc_port=config.get(CONF_BC_PORT, DEFAULT_BC_PORT),
)
self.last_wake: float = 0
@@ -149,15 +151,33 @@ class ReolinkHost:
async def async_init(self) -> None:
"""Connect to Reolink host."""
if not self._api.valid_password():
+ if (
+ len(self._config[CONF_PASSWORD]) >= 32
+ and self._config_entry is not None
+ ):
+ ir.async_create_issue(
+ self._hass,
+ DOMAIN,
+ f"password_too_long_{self._config_entry.entry_id}",
+ is_fixable=True,
+ severity=ir.IssueSeverity.ERROR,
+ translation_key="password_too_long",
+ translation_placeholders={"name": self._config_entry.title},
+ )
+
raise PasswordIncompatible(
- "Reolink password contains incompatible special character, "
- "please change the password to only contain characters: "
- f"a-z, A-Z, 0-9 or {ALLOWED_SPECIAL_CHARS}"
+ "Reolink password contains incompatible special character or "
+ "is too long, please change the password to only contain characters: "
+ f"a-z, A-Z, 0-9 or {ALLOWED_SPECIAL_CHARS} "
+ "and not be longer than 31 characters"
)
store: Store[str] | None = None
- if self._config_entry_id is not None:
- store = get_store(self._hass, self._config_entry_id)
+ if self._config_entry is not None:
+ ir.async_delete_issue(
+ self._hass, DOMAIN, f"password_too_long_{self._config_entry.entry_id}"
+ )
+ store = get_store(self._hass, self._config_entry.entry_id)
if self._config.get(CONF_SUPPORTS_PRIVACY_MODE) and (
data := await store.async_load()
):
diff --git a/homeassistant/components/reolink/icons.json b/homeassistant/components/reolink/icons.json
index 26198a11594..00045c4cda2 100644
--- a/homeassistant/components/reolink/icons.json
+++ b/homeassistant/components/reolink/icons.json
@@ -54,6 +54,72 @@
"state": {
"on": "mdi:sleep"
}
+ },
+ "crossline_person": {
+ "default": "mdi:fence",
+ "state": {
+ "on": "mdi:fence-electric"
+ }
+ },
+ "crossline_vehicle": {
+ "default": "mdi:fence",
+ "state": {
+ "on": "mdi:fence-electric"
+ }
+ },
+ "crossline_dog_cat": {
+ "default": "mdi:fence",
+ "state": {
+ "on": "mdi:fence-electric"
+ }
+ },
+ "intrusion_person": {
+ "default": "mdi:location-enter",
+ "state": {
+ "on": "mdi:alert-circle-outline"
+ }
+ },
+ "intrusion_vehicle": {
+ "default": "mdi:location-enter",
+ "state": {
+ "on": "mdi:alert-circle-outline"
+ }
+ },
+ "intrusion_dog_cat": {
+ "default": "mdi:location-enter",
+ "state": {
+ "on": "mdi:alert-circle-outline"
+ }
+ },
+ "linger_person": {
+ "default": "mdi:account-switch",
+ "state": {
+ "on": "mdi:account-alert"
+ }
+ },
+ "linger_vehicle": {
+ "default": "mdi:account-switch",
+ "state": {
+ "on": "mdi:account-alert"
+ }
+ },
+ "linger_dog_cat": {
+ "default": "mdi:account-switch",
+ "state": {
+ "on": "mdi:account-alert"
+ }
+ },
+ "forgotten_item": {
+ "default": "mdi:package-variant-closed-plus",
+ "state": {
+ "on": "mdi:package-variant-closed-check"
+ }
+ },
+ "taken_item": {
+ "default": "mdi:package-variant-closed-minus",
+ "state": {
+ "on": "mdi:package-variant-closed-check"
+ }
}
},
"button": {
@@ -284,6 +350,9 @@
},
"sub_bit_rate": {
"default": "mdi:play-speed"
+ },
+ "scene_mode": {
+ "default": "mdi:view-list"
}
},
"sensor": {
@@ -299,6 +368,9 @@
"battery_state": {
"default": "mdi:battery-charging"
},
+ "day_night_state": {
+ "default": "mdi:theme-light-dark"
+ },
"wifi_signal": {
"default": "mdi:wifi"
},
diff --git a/homeassistant/components/reolink/light.py b/homeassistant/components/reolink/light.py
index bbb9592dd76..d48790264d1 100644
--- a/homeassistant/components/reolink/light.py
+++ b/homeassistant/components/reolink/light.py
@@ -16,7 +16,7 @@ from homeassistant.components.light import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import (
ReolinkChannelCoordinatorEntity,
@@ -92,7 +92,7 @@ HOST_LIGHT_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink light entities."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/manifest.json b/homeassistant/components/reolink/manifest.json
index 505358a07f7..41cfe1f9ae3 100644
--- a/homeassistant/components/reolink/manifest.json
+++ b/homeassistant/components/reolink/manifest.json
@@ -19,5 +19,5 @@
"iot_class": "local_push",
"loggers": ["reolink_aio"],
"quality_scale": "platinum",
- "requirements": ["reolink-aio==0.11.10"]
+ "requirements": ["reolink-aio==0.12.3"]
}
diff --git a/homeassistant/components/reolink/media_source.py b/homeassistant/components/reolink/media_source.py
index e912bfb5100..39514d58cb7 100644
--- a/homeassistant/components/reolink/media_source.py
+++ b/homeassistant/components/reolink/media_source.py
@@ -18,7 +18,6 @@ from homeassistant.components.media_source import (
Unresolvable,
)
from homeassistant.components.stream import create_stream
-from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
@@ -71,7 +70,7 @@ class ReolinkVODMediaSource(MediaSource):
host = get_host(self.hass, config_entry_id)
def get_vod_type() -> VodRequestType:
- if filename.endswith(".mp4"):
+ if filename.endswith((".mp4", ".vref")):
if host.api.is_nvr:
return VodRequestType.DOWNLOAD
return VodRequestType.PLAYBACK
@@ -151,9 +150,7 @@ class ReolinkVODMediaSource(MediaSource):
entity_reg = er.async_get(self.hass)
device_reg = dr.async_get(self.hass)
- for config_entry in self.hass.config_entries.async_entries(DOMAIN):
- if config_entry.state != ConfigEntryState.LOADED:
- continue
+ for config_entry in self.hass.config_entries.async_loaded_entries(DOMAIN):
channels: list[str] = []
host = config_entry.runtime_data.host
entities = er.async_entries_for_config_entry(
@@ -222,7 +219,7 @@ class ReolinkVODMediaSource(MediaSource):
if main_enc == "h265":
_LOGGER.debug(
"Reolink camera %s uses h265 encoding for main stream,"
- "playback only possible using sub stream",
+ "playback at high resolution may not work in all browsers/apps",
host.api.camera_name(channel),
)
@@ -236,34 +233,29 @@ class ReolinkVODMediaSource(MediaSource):
can_play=False,
can_expand=True,
),
+ BrowseMediaSource(
+ domain=DOMAIN,
+ identifier=f"RES|{config_entry_id}|{channel}|main",
+ media_class=MediaClass.CHANNEL,
+ media_content_type=MediaType.PLAYLIST,
+ title="High resolution",
+ can_play=False,
+ can_expand=True,
+ ),
]
- if main_enc != "h265":
- children.append(
- BrowseMediaSource(
- domain=DOMAIN,
- identifier=f"RES|{config_entry_id}|{channel}|main",
- media_class=MediaClass.CHANNEL,
- media_content_type=MediaType.PLAYLIST,
- title="High resolution",
- can_play=False,
- can_expand=True,
- ),
- )
if host.api.supported(channel, "autotrack_stream"):
- children.append(
- BrowseMediaSource(
- domain=DOMAIN,
- identifier=f"RES|{config_entry_id}|{channel}|autotrack_sub",
- media_class=MediaClass.CHANNEL,
- media_content_type=MediaType.PLAYLIST,
- title="Autotrack low resolution",
- can_play=False,
- can_expand=True,
- ),
- )
- if main_enc != "h265":
- children.append(
+ children.extend(
+ [
+ BrowseMediaSource(
+ domain=DOMAIN,
+ identifier=f"RES|{config_entry_id}|{channel}|autotrack_sub",
+ media_class=MediaClass.CHANNEL,
+ media_content_type=MediaType.PLAYLIST,
+ title="Autotrack low resolution",
+ can_play=False,
+ can_expand=True,
+ ),
BrowseMediaSource(
domain=DOMAIN,
identifier=f"RES|{config_entry_id}|{channel}|autotrack_main",
@@ -273,11 +265,7 @@ class ReolinkVODMediaSource(MediaSource):
can_play=False,
can_expand=True,
),
- )
-
- if len(children) == 1:
- return await self._async_generate_camera_days(
- config_entry_id, channel, "sub"
+ ]
)
title = host.api.camera_name(channel)
diff --git a/homeassistant/components/reolink/number.py b/homeassistant/components/reolink/number.py
index d8fabfaa3b8..48382df4cbc 100644
--- a/homeassistant/components/reolink/number.py
+++ b/homeassistant/components/reolink/number.py
@@ -15,7 +15,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import (
ReolinkChannelCoordinatorEntity,
@@ -538,7 +538,7 @@ CHIME_NUMBER_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink number entities."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/select.py b/homeassistant/components/reolink/select.py
index df8c0269957..e5d66ed3901 100644
--- a/homeassistant/components/reolink/select.py
+++ b/homeassistant/components/reolink/select.py
@@ -23,13 +23,15 @@ from reolink_aio.api import (
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory, UnitOfDataRate, UnitOfFrequency
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import (
ReolinkChannelCoordinatorEntity,
ReolinkChannelEntityDescription,
ReolinkChimeCoordinatorEntity,
ReolinkChimeEntityDescription,
+ ReolinkHostCoordinatorEntity,
+ ReolinkHostEntityDescription,
)
from .util import ReolinkConfigEntry, ReolinkData, raise_translated_error
@@ -49,6 +51,18 @@ class ReolinkSelectEntityDescription(
value: Callable[[Host, int], str] | None = None
+@dataclass(frozen=True, kw_only=True)
+class ReolinkHostSelectEntityDescription(
+ SelectEntityDescription,
+ ReolinkHostEntityDescription,
+):
+ """A class that describes host select entities."""
+
+ get_options: Callable[[Host], list[str]]
+ method: Callable[[Host, str], Any]
+ value: Callable[[Host], str]
+
+
@dataclass(frozen=True, kw_only=True)
class ReolinkChimeSelectEntityDescription(
SelectEntityDescription,
@@ -238,6 +252,19 @@ SELECT_ENTITIES = (
),
)
+HOST_SELECT_ENTITIES = (
+ ReolinkHostSelectEntityDescription(
+ key="scene_mode",
+ cmd_key="GetScene",
+ translation_key="scene_mode",
+ entity_category=EntityCategory.CONFIG,
+ get_options=lambda api: api.baichuan.scene_names,
+ supported=lambda api: api.supported(None, "scenes"),
+ value=lambda api: api.baichuan.active_scene,
+ method=lambda api, name: api.baichuan.set_scene(scene_name=name),
+ ),
+)
+
CHIME_SELECT_ENTITIES = (
ReolinkChimeSelectEntityDescription(
key="motion_tone",
@@ -295,17 +322,24 @@ CHIME_SELECT_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink select entities."""
reolink_data: ReolinkData = config_entry.runtime_data
- entities: list[ReolinkSelectEntity | ReolinkChimeSelectEntity] = [
+ entities: list[
+ ReolinkSelectEntity | ReolinkHostSelectEntity | ReolinkChimeSelectEntity
+ ] = [
ReolinkSelectEntity(reolink_data, channel, entity_description)
for entity_description in SELECT_ENTITIES
for channel in reolink_data.host.api.channels
if entity_description.supported(reolink_data.host.api, channel)
]
+ entities.extend(
+ ReolinkHostSelectEntity(reolink_data, entity_description)
+ for entity_description in HOST_SELECT_ENTITIES
+ if entity_description.supported(reolink_data.host.api)
+ )
entities.extend(
ReolinkChimeSelectEntity(reolink_data, chime, entity_description)
for entity_description in CHIME_SELECT_ENTITIES
@@ -360,6 +394,33 @@ class ReolinkSelectEntity(ReolinkChannelCoordinatorEntity, SelectEntity):
self.async_write_ha_state()
+class ReolinkHostSelectEntity(ReolinkHostCoordinatorEntity, SelectEntity):
+ """Base select entity class for Reolink Host."""
+
+ entity_description: ReolinkHostSelectEntityDescription
+
+ def __init__(
+ self,
+ reolink_data: ReolinkData,
+ entity_description: ReolinkHostSelectEntityDescription,
+ ) -> None:
+ """Initialize Reolink select entity."""
+ self.entity_description = entity_description
+ super().__init__(reolink_data)
+ self._attr_options = entity_description.get_options(self._host.api)
+
+ @property
+ def current_option(self) -> str | None:
+ """Return the current option."""
+ return self.entity_description.value(self._host.api)
+
+ @raise_translated_error
+ async def async_select_option(self, option: str) -> None:
+ """Change the selected option."""
+ await self.entity_description.method(self._host.api, option)
+ self.async_write_ha_state()
+
+
class ReolinkChimeSelectEntity(ReolinkChimeCoordinatorEntity, SelectEntity):
"""Base select entity class for Reolink IP cameras."""
diff --git a/homeassistant/components/reolink/sensor.py b/homeassistant/components/reolink/sensor.py
index 36900da99ca..85de03dd1a3 100644
--- a/homeassistant/components/reolink/sensor.py
+++ b/homeassistant/components/reolink/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .entity import (
@@ -107,6 +107,17 @@ SENSORS = (
value=lambda api, ch: BatteryEnum(api.battery_status(ch)).name,
supported=lambda api, ch: api.supported(ch, "battery"),
),
+ ReolinkSensorEntityDescription(
+ key="day_night_state",
+ cmd_id=33,
+ cmd_key="296",
+ translation_key="day_night_state",
+ device_class=SensorDeviceClass.ENUM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ options=["day", "night", "led_day"],
+ value=lambda api, ch: api.baichuan.day_night_state(ch),
+ supported=lambda api, ch: api.supported(ch, "day_night_state"),
+ ),
)
HOST_SENSORS = (
@@ -150,7 +161,7 @@ HDD_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink IP Camera."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/services.py b/homeassistant/components/reolink/services.py
index acd31fe0d7d..d170aa32379 100644
--- a/homeassistant/components/reolink/services.py
+++ b/homeassistant/components/reolink/services.py
@@ -40,7 +40,7 @@ def async_setup_services(hass: HomeAssistant) -> None:
if (
config_entry is None
or device is None
- or config_entry.state == ConfigEntryState.NOT_LOADED
+ or config_entry.state != ConfigEntryState.LOADED
):
raise ServiceValidationError(
translation_domain=DOMAIN,
diff --git a/homeassistant/components/reolink/siren.py b/homeassistant/components/reolink/siren.py
index 74bb227d078..f5d2de977ae 100644
--- a/homeassistant/components/reolink/siren.py
+++ b/homeassistant/components/reolink/siren.py
@@ -13,7 +13,7 @@ from homeassistant.components.siren import (
SirenEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription
from .util import ReolinkConfigEntry, ReolinkData, raise_translated_error
@@ -40,7 +40,7 @@ SIREN_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink siren entities."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/strings.json b/homeassistant/components/reolink/strings.json
index b72e7bbd00d..7ad2e1ea217 100644
--- a/homeassistant/components/reolink/strings.json
+++ b/homeassistant/components/reolink/strings.json
@@ -8,13 +8,15 @@
"host": "[%key:common::config_flow::data::host%]",
"port": "[%key:common::config_flow::data::port%]",
"use_https": "Enable HTTPS",
+ "baichuan_port": "Basic service port",
"username": "[%key:common::config_flow::data::username%]",
"password": "[%key:common::config_flow::data::password%]"
},
"data_description": {
"host": "The hostname or IP address of your Reolink device. For example: '192.168.1.25'.",
- "port": "The port to connect to the Reolink device. For HTTP normally: '80', for HTTPS normally '443'.",
+ "port": "The HTTP(s) port to connect to the Reolink device API. For HTTP normally: '80', for HTTPS normally '443'.",
"use_https": "Use a HTTPS (SSL) connection to the Reolink device.",
+ "baichuan_port": "The 'Basic Service Port' to connect to the Reolink device over TCP. Normally '9000' unless manually changed in the Reolink desktop client.",
"username": "Username to login to the Reolink device itself. Not the Reolink cloud account.",
"password": "Password to login to the Reolink device itself. Not the Reolink cloud account."
}
@@ -29,7 +31,7 @@
"cannot_connect": "Failed to connect, check the IP address of the camera",
"invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
"not_admin": "User needs to be admin, user \"{username}\" has authorisation level \"{userlevel}\"",
- "password_incompatible": "Password contains incompatible special character, only these characters are allowed: a-z, A-Z, 0-9 or {special_chars}",
+ "password_incompatible": "Password contains incompatible special character or is too long, maximum 31 characters and only these characters are allowed: a-z, A-Z, 0-9 or {special_chars}",
"unknown": "[%key:common::config_flow::error::unknown%]",
"update_needed": "Failed to login because of outdated firmware, please update the firmware to version {needed_firmware} using the Reolink Download Center: {download_center_url}, currently version {current_firmware} is installed",
"webhook_exception": "Home Assistant URL is not available, go to Settings > System > Network > Home Assistant URL and correct the URLs, see {more_info}"
@@ -100,7 +102,7 @@
"message": "Error trying to update Reolink firmware: {err}"
},
"config_entry_not_ready": {
- "message": "Error while trying to setup {host}: {err}"
+ "message": "Error while trying to set up {host}: {err}"
}
},
"issues": {
@@ -126,7 +128,11 @@
},
"hub_switch_deprecated": {
"title": "Reolink Home Hub switches deprecated",
- "description": "The redundant 'Record', 'Email on event', 'FTP upload', 'Push notifications', and 'Buzzer on event' switches on the Reolink Home Hub are depricated since the new firmware no longer supports these. Please use the equally named switches under each of the camera devices connected to the Home Hub instead. To remove this issue, please adjust automations accordingly and disable the switch entities mentioned."
+ "description": "The redundant 'Record', 'Email on event', 'FTP upload', 'Push notifications', and 'Buzzer on event' switches on the Reolink Home Hub are deprecated since the new firmware no longer supports these. Please use the equally named switches under each of the camera devices connected to the Home Hub instead. To remove this issue, please adjust automations accordingly and disable the switch entities mentioned."
+ },
+ "password_too_long": {
+ "title": "Reolink password too long",
+ "description": "The password for \"{name}\" is more than 31 characters long, this is no longer compatible with the Reolink API. Please change the password using the Reolink app/client to a password with is shorter than 32 characters. After changing the password, fill in the new password in the Reolink Re-authentication flow to continue using this integration. The latest version of the Reolink app/client also has a password limit of 31 characters."
}
},
"services": {
@@ -335,6 +341,83 @@
"off": "Awake",
"on": "Sleeping"
}
+ },
+ "crossline_person": {
+ "name": "Crossline {zone_name} person",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "crossline_vehicle": {
+ "name": "Crossline {zone_name} vehicle",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "crossline_dog_cat": {
+ "name": "Crossline {zone_name} animal",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "intrusion_person": {
+ "name": "Intrusion {zone_name} person",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "intrusion_vehicle": {
+ "name": "Intrusion {zone_name} vehicle",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "intrusion_dog_cat": {
+ "name": "Intrusion {zone_name} animal",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "linger_person": {
+ "name": "Linger {zone_name} person",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "linger_vehicle": {
+ "name": "Linger {zone_name} vehicle",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "linger_dog_cat": {
+ "name": "Linger {zone_name} animal",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "forgotten_item": {
+ "name": "Item forgotten {zone_name}",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
+ },
+ "taken_item": {
+ "name": "Item taken {zone_name}",
+ "state": {
+ "off": "[%key:component::binary_sensor::entity_component::gas::state::off%]",
+ "on": "[%key:component::binary_sensor::entity_component::gas::state::on%]"
+ }
}
},
"button": {
@@ -720,6 +803,15 @@
},
"sub_bit_rate": {
"name": "Fluent bit rate"
+ },
+ "scene_mode": {
+ "name": "Scene mode",
+ "state": {
+ "off": "[%key:common::state::off%]",
+ "disarm": "Disarmed",
+ "home": "Home",
+ "away": "Away"
+ }
}
},
"sensor": {
@@ -741,11 +833,19 @@
"battery_state": {
"name": "Battery state",
"state": {
- "discharging": "Discharging",
- "charging": "Charging",
+ "discharging": "[%key:common::state::discharging%]",
+ "charging": "[%key:common::state::charging%]",
"chargecomplete": "Charge complete"
}
},
+ "day_night_state": {
+ "name": "Day night state",
+ "state": {
+ "day": "Color",
+ "night": "Black & white",
+ "led_day": "Color with floodlight"
+ }
+ },
"hdd_storage": {
"name": "HDD {hdd_index} storage"
},
diff --git a/homeassistant/components/reolink/switch.py b/homeassistant/components/reolink/switch.py
index a0b8824782a..0f106c0f2cc 100644
--- a/homeassistant/components/reolink/switch.py
+++ b/homeassistant/components/reolink/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er, issue_registry as ir
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import (
@@ -330,7 +330,7 @@ DEPRECATED_NVR_SWITCHES = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Reolink switch entities."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/update.py b/homeassistant/components/reolink/update.py
index 5a8c7d7dc08..0744d66fb5b 100644
--- a/homeassistant/components/reolink/update.py
+++ b/homeassistant/components/reolink/update.py
@@ -16,7 +16,7 @@ from homeassistant.components.update import (
)
from homeassistant.core import CALLBACK_TYPE, HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
@@ -75,7 +75,7 @@ HOST_UPDATE_ENTITIES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ReolinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up update entities for Reolink component."""
reolink_data: ReolinkData = config_entry.runtime_data
diff --git a/homeassistant/components/reolink/views.py b/homeassistant/components/reolink/views.py
index 1a4585bc997..44265244b18 100644
--- a/homeassistant/components/reolink/views.py
+++ b/homeassistant/components/reolink/views.py
@@ -83,7 +83,16 @@ class PlaybackProxyView(HomeAssistantView):
_LOGGER.warning("Reolink playback proxy error: %s", str(err))
return web.Response(body=str(err), status=HTTPStatus.BAD_REQUEST)
+ headers = dict(request.headers)
+ headers.pop("Host", None)
+ headers.pop("Referer", None)
+
if _LOGGER.isEnabledFor(logging.DEBUG):
+ _LOGGER.debug(
+ "Requested Playback Proxy Method %s, Headers: %s",
+ request.method,
+ headers,
+ )
_LOGGER.debug(
"Opening VOD stream from %s: %s",
host.api.camera_name(ch),
@@ -93,6 +102,7 @@ class PlaybackProxyView(HomeAssistantView):
try:
reolink_response = await self.session.get(
reolink_url,
+ headers=headers,
timeout=ClientTimeout(
connect=15, sock_connect=15, sock_read=5, total=None
),
@@ -118,18 +128,25 @@ class PlaybackProxyView(HomeAssistantView):
]:
err_str = f"Reolink playback expected video/mp4 but got {reolink_response.content_type}"
_LOGGER.error(err_str)
+ if reolink_response.content_type == "text/html":
+ text = await reolink_response.text()
+ _LOGGER.debug(text)
return web.Response(body=err_str, status=HTTPStatus.BAD_REQUEST)
- response = web.StreamResponse(
- status=200,
- reason="OK",
- headers={
- "Content-Type": "video/mp4",
- },
+ response_headers = dict(reolink_response.headers)
+ _LOGGER.debug(
+ "Response Playback Proxy Status %s:%s, Headers: %s",
+ reolink_response.status,
+ reolink_response.reason,
+ response_headers,
)
+ response_headers["Content-Type"] = "video/mp4"
- if reolink_response.content_length is not None:
- response.content_length = reolink_response.content_length
+ response = web.StreamResponse(
+ status=reolink_response.status,
+ reason=reolink_response.reason,
+ headers=response_headers,
+ )
await response.prepare(request)
@@ -141,7 +158,8 @@ class PlaybackProxyView(HomeAssistantView):
"Timeout while reading Reolink playback from %s, writing EOF",
host.api.nvr_name,
)
+ finally:
+ reolink_response.release()
- reolink_response.release()
await response.write_eof()
return response
diff --git a/homeassistant/components/rest_command/__init__.py b/homeassistant/components/rest_command/__init__.py
index fe3702510af..c6a4206de4a 100644
--- a/homeassistant/components/rest_command/__init__.py
+++ b/homeassistant/components/rest_command/__init__.py
@@ -34,6 +34,7 @@ from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.reload import async_integration_yaml_config
from homeassistant.helpers.typing import ConfigType
+from homeassistant.util.ssl import SSLCipherList
DOMAIN = "rest_command"
@@ -46,6 +47,7 @@ DEFAULT_VERIFY_SSL = True
SUPPORT_REST_METHODS = ["get", "patch", "post", "put", "delete"]
CONF_CONTENT_TYPE = "content_type"
+CONF_INSECURE_CIPHER = "insecure_cipher"
COMMAND_SCHEMA = vol.Schema(
{
@@ -60,6 +62,7 @@ COMMAND_SCHEMA = vol.Schema(
vol.Optional(CONF_TIMEOUT, default=DEFAULT_TIMEOUT): vol.Coerce(int),
vol.Optional(CONF_CONTENT_TYPE): cv.string,
vol.Optional(CONF_VERIFY_SSL, default=DEFAULT_VERIFY_SSL): cv.boolean,
+ vol.Optional(CONF_INSECURE_CIPHER, default=False): cv.boolean,
}
)
@@ -91,7 +94,15 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
@callback
def async_register_rest_command(name: str, command_config: dict[str, Any]) -> None:
"""Create service for rest command."""
- websession = async_get_clientsession(hass, command_config[CONF_VERIFY_SSL])
+ websession = async_get_clientsession(
+ hass,
+ command_config[CONF_VERIFY_SSL],
+ ssl_cipher=(
+ SSLCipherList.INSECURE
+ if command_config[CONF_INSECURE_CIPHER]
+ else SSLCipherList.PYTHON_DEFAULT
+ ),
+ )
timeout = command_config[CONF_TIMEOUT]
method = command_config[CONF_METHOD]
@@ -135,6 +146,14 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
if content_type:
headers[hdrs.CONTENT_TYPE] = content_type
+ _LOGGER.debug(
+ "Calling %s %s with headers: %s and payload: %s",
+ method,
+ request_url,
+ headers,
+ payload,
+ )
+
try:
async with getattr(websession, method)(
request_url,
diff --git a/homeassistant/components/rflink/entity.py b/homeassistant/components/rflink/entity.py
index 26153acf7ba..0caec4ea2c3 100644
--- a/homeassistant/components/rflink/entity.py
+++ b/homeassistant/components/rflink/entity.py
@@ -105,12 +105,12 @@ class RflinkDevice(Entity):
return self._state
@property
- def assumed_state(self):
+ def assumed_state(self) -> bool:
"""Assume device state until first device event sets state."""
return self._state is None
@property
- def available(self):
+ def available(self) -> bool:
"""Return True if entity is available."""
return self._available
@@ -120,7 +120,7 @@ class RflinkDevice(Entity):
self._available = availability
self.async_write_ha_state()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register update callback."""
await super().async_added_to_hass()
# Remove temporary bogus entity_id if added
@@ -300,7 +300,7 @@ class RflinkCommand(RflinkDevice):
class SwitchableRflinkDevice(RflinkCommand, RestoreEntity):
"""Rflink entity which can switch on/off (eg: light, switch)."""
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Restore RFLink device state (ON/OFF)."""
await super().async_added_to_hass()
if (old_state := await self.async_get_last_state()) is not None:
diff --git a/homeassistant/components/rfxtrx/binary_sensor.py b/homeassistant/components/rfxtrx/binary_sensor.py
index 316cf44ef0d..a86ad5557b4 100644
--- a/homeassistant/components/rfxtrx/binary_sensor.py
+++ b/homeassistant/components/rfxtrx/binary_sensor.py
@@ -17,7 +17,7 @@ from homeassistant.const import CONF_COMMAND_OFF, CONF_COMMAND_ON, STATE_ON
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers import event as evt
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeviceTuple, async_setup_platform_entry, get_pt2262_cmd
from .const import (
@@ -91,7 +91,7 @@ def supported(event: rfxtrxmod.RFXtrxEvent) -> bool:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up config entry."""
diff --git a/homeassistant/components/rfxtrx/cover.py b/homeassistant/components/rfxtrx/cover.py
index 473a0d94056..07443afb38b 100644
--- a/homeassistant/components/rfxtrx/cover.py
+++ b/homeassistant/components/rfxtrx/cover.py
@@ -11,7 +11,7 @@ from homeassistant.components.cover import CoverEntity, CoverEntityFeature, Cove
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeviceTuple, async_setup_platform_entry
from .const import (
@@ -34,7 +34,7 @@ def supported(event: rfxtrxmod.RFXtrxEvent) -> bool:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up config entry."""
diff --git a/homeassistant/components/rfxtrx/event.py b/homeassistant/components/rfxtrx/event.py
index 212d93b5019..40d02953aeb 100644
--- a/homeassistant/components/rfxtrx/event.py
+++ b/homeassistant/components/rfxtrx/event.py
@@ -11,7 +11,7 @@ from homeassistant.components.event import EventEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import slugify
from . import DeviceTuple, async_setup_platform_entry
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up config entry."""
diff --git a/homeassistant/components/rfxtrx/light.py b/homeassistant/components/rfxtrx/light.py
index 0e2f7bef65a..90c0d2eeed7 100644
--- a/homeassistant/components/rfxtrx/light.py
+++ b/homeassistant/components/rfxtrx/light.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_ON
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeviceTuple, async_setup_platform_entry
from .const import COMMAND_OFF_LIST, COMMAND_ON_LIST
@@ -32,7 +32,7 @@ def supported(event: rfxtrxmod.RFXtrxEvent) -> bool:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up config entry."""
diff --git a/homeassistant/components/rfxtrx/sensor.py b/homeassistant/components/rfxtrx/sensor.py
index 13f3c012af8..4b256279445 100644
--- a/homeassistant/components/rfxtrx/sensor.py
+++ b/homeassistant/components/rfxtrx/sensor.py
@@ -36,7 +36,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import DeviceTuple, async_setup_platform_entry, get_rfx_object
@@ -241,7 +241,7 @@ SENSOR_TYPES_DICT = {desc.key: desc for desc in SENSOR_TYPES}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up config entry."""
diff --git a/homeassistant/components/rfxtrx/siren.py b/homeassistant/components/rfxtrx/siren.py
index 1635f1f55a9..1164dafbfce 100644
--- a/homeassistant/components/rfxtrx/siren.py
+++ b/homeassistant/components/rfxtrx/siren.py
@@ -11,7 +11,7 @@ from homeassistant.components.siren import ATTR_TONE, SirenEntity, SirenEntityFe
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from . import DEFAULT_OFF_DELAY, DeviceTuple, async_setup_platform_entry
@@ -47,7 +47,7 @@ def get_first_key(data: dict[int, str], entry: str) -> int:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up config entry."""
diff --git a/homeassistant/components/rfxtrx/switch.py b/homeassistant/components/rfxtrx/switch.py
index cd17e71f4f0..b3eb63fb2b4 100644
--- a/homeassistant/components/rfxtrx/switch.py
+++ b/homeassistant/components/rfxtrx/switch.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_COMMAND_OFF, CONF_COMMAND_ON, STATE_ON
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DeviceTuple, async_setup_platform_entry, get_pt2262_cmd
from .const import (
@@ -41,7 +41,7 @@ def supported(event: rfxtrxmod.RFXtrxEvent) -> bool:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up config entry."""
diff --git a/homeassistant/components/ridwell/__init__.py b/homeassistant/components/ridwell/__init__.py
index 71e80086833..9c9104258a8 100644
--- a/homeassistant/components/ridwell/__init__.py
+++ b/homeassistant/components/ridwell/__init__.py
@@ -17,7 +17,7 @@ PLATFORMS: list[Platform] = [Platform.CALENDAR, Platform.SENSOR, Platform.SWITCH
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Ridwell from a config entry."""
- coordinator = RidwellDataUpdateCoordinator(hass, name=entry.title)
+ coordinator = RidwellDataUpdateCoordinator(hass, entry)
await coordinator.async_initialize()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/ridwell/calendar.py b/homeassistant/components/ridwell/calendar.py
index ecca0366754..bb7982a5391 100644
--- a/homeassistant/components/ridwell/calendar.py
+++ b/homeassistant/components/ridwell/calendar.py
@@ -9,7 +9,7 @@ from aioridwell.model import RidwellAccount, RidwellPickupEvent
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RidwellDataUpdateCoordinator
@@ -36,7 +36,9 @@ def async_get_calendar_event_from_pickup_event(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ridwell calendars based on a config entry."""
coordinator: RidwellDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/ridwell/coordinator.py b/homeassistant/components/ridwell/coordinator.py
index 28190522c76..336a71bc67f 100644
--- a/homeassistant/components/ridwell/coordinator.py
+++ b/homeassistant/components/ridwell/coordinator.py
@@ -29,7 +29,7 @@ class RidwellDataUpdateCoordinator(
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, *, name: str) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize."""
# These will be filled in by async_initialize; we give them these defaults to
# avoid arduous typing checks down the line:
@@ -37,7 +37,13 @@ class RidwellDataUpdateCoordinator(
self.dashboard_url = ""
self.user_id = ""
- super().__init__(hass, LOGGER, name=name, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=config_entry.title,
+ update_interval=UPDATE_INTERVAL,
+ )
async def _async_update_data(self) -> dict[str, list[RidwellPickupEvent]]:
"""Fetch the latest data from the source."""
diff --git a/homeassistant/components/ridwell/sensor.py b/homeassistant/components/ridwell/sensor.py
index 7fc7fdb5348..30f97ecaea8 100644
--- a/homeassistant/components/ridwell/sensor.py
+++ b/homeassistant/components/ridwell/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, SENSOR_TYPE_NEXT_PICKUP
from .coordinator import RidwellDataUpdateCoordinator
@@ -34,7 +34,9 @@ SENSOR_DESCRIPTION = SensorEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ridwell sensors based on a config entry."""
coordinator: RidwellDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/ridwell/switch.py b/homeassistant/components/ridwell/switch.py
index 04e3e4c5ff9..e3be9ea5368 100644
--- a/homeassistant/components/ridwell/switch.py
+++ b/homeassistant/components/ridwell/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RidwellDataUpdateCoordinator
@@ -24,7 +24,9 @@ SWITCH_DESCRIPTION = SwitchEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ridwell sensors based on a config entry."""
coordinator: RidwellDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/ring/binary_sensor.py b/homeassistant/components/ring/binary_sensor.py
index da0e0cc1d9b..49051ee5e11 100644
--- a/homeassistant/components/ring/binary_sensor.py
+++ b/homeassistant/components/ring/binary_sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import Platform
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_at
from . import RingConfigEntry
@@ -67,7 +67,7 @@ BINARY_SENSOR_TYPES: tuple[RingBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ring binary sensors from a config entry."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/button.py b/homeassistant/components/ring/button.py
index 30600237847..09e6c0e413a 100644
--- a/homeassistant/components/ring/button.py
+++ b/homeassistant/components/ring/button.py
@@ -6,7 +6,7 @@ from ring_doorbell import RingOther
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RingConfigEntry
from .coordinator import RingDataCoordinator
@@ -24,7 +24,7 @@ BUTTON_DESCRIPTION = ButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the buttons for the Ring devices."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/camera.py b/homeassistant/components/ring/camera.py
index e0ae2b52fa0..156d82665d2 100644
--- a/homeassistant/components/ring/camera.py
+++ b/homeassistant/components/ring/camera.py
@@ -27,7 +27,7 @@ from homeassistant.components.camera import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import RingConfigEntry
@@ -76,7 +76,7 @@ CAMERA_DESCRIPTIONS: tuple[RingCameraEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Ring Door Bell and StickUp Camera."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/event.py b/homeassistant/components/ring/event.py
index 4d7a6277579..db99a10de74 100644
--- a/homeassistant/components/ring/event.py
+++ b/homeassistant/components/ring/event.py
@@ -12,7 +12,7 @@ from homeassistant.components.event import (
EventEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RingConfigEntry
from .coordinator import RingListenCoordinator
@@ -57,7 +57,7 @@ EVENT_DESCRIPTIONS: tuple[RingEventEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up events for a Ring device."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/light.py b/homeassistant/components/ring/light.py
index 62c5217a89b..34915dd5133 100644
--- a/homeassistant/components/ring/light.py
+++ b/homeassistant/components/ring/light.py
@@ -9,7 +9,7 @@ from ring_doorbell import RingStickUpCam
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import RingConfigEntry
@@ -40,7 +40,7 @@ class OnOffState(StrEnum):
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the lights for the Ring devices."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/number.py b/homeassistant/components/ring/number.py
index b920ff7edc7..68b41451bd0 100644
--- a/homeassistant/components/ring/number.py
+++ b/homeassistant/components/ring/number.py
@@ -13,7 +13,7 @@ from homeassistant.components.number import (
NumberMode,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import RingConfigEntry
@@ -28,7 +28,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a numbers for a Ring device."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/sensor.py b/homeassistant/components/ring/sensor.py
index a2f72b94336..5744ed9a4d8 100644
--- a/homeassistant/components/ring/sensor.py
+++ b/homeassistant/components/ring/sensor.py
@@ -28,7 +28,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import RingConfigEntry
@@ -48,7 +48,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a sensor for a Ring device."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/siren.py b/homeassistant/components/ring/siren.py
index 05fa07c39eb..7f096c0e643 100644
--- a/homeassistant/components/ring/siren.py
+++ b/homeassistant/components/ring/siren.py
@@ -22,7 +22,7 @@ from homeassistant.components.siren import (
)
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RingConfigEntry
from .coordinator import RingDataCoordinator
@@ -85,7 +85,7 @@ SIRENS: tuple[RingSirenEntityDescription[Any], ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the sirens for the Ring devices."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/ring/switch.py b/homeassistant/components/ring/switch.py
index cab5654fc5a..02d98388edc 100644
--- a/homeassistant/components/ring/switch.py
+++ b/homeassistant/components/ring/switch.py
@@ -11,7 +11,7 @@ from ring_doorbell.const import DOORBELL_EXISTING_TYPE
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import RingConfigEntry
@@ -86,7 +86,7 @@ SWITCHES: Sequence[RingSwitchEntityDescription[Any]] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RingConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the switches for the Ring devices."""
ring_data = entry.runtime_data
diff --git a/homeassistant/components/risco/__init__.py b/homeassistant/components/risco/__init__.py
index 7255c724e3f..56c7a509cca 100644
--- a/homeassistant/components/risco/__init__.py
+++ b/homeassistant/components/risco/__init__.py
@@ -16,7 +16,6 @@ from homeassistant.const import (
CONF_PASSWORD,
CONF_PIN,
CONF_PORT,
- CONF_SCAN_INTERVAL,
CONF_TYPE,
CONF_USERNAME,
Platform,
@@ -30,7 +29,6 @@ from .const import (
CONF_CONCURRENCY,
DATA_COORDINATOR,
DEFAULT_CONCURRENCY,
- DEFAULT_SCAN_INTERVAL,
DOMAIN,
EVENTS_COORDINATOR,
SYSTEM_UPDATE_SIGNAL,
@@ -144,12 +142,9 @@ async def _async_setup_cloud_entry(hass: HomeAssistant, entry: ConfigEntry) -> b
except UnauthorizedError as error:
raise ConfigEntryAuthFailed from error
- scan_interval = entry.options.get(CONF_SCAN_INTERVAL, DEFAULT_SCAN_INTERVAL)
- coordinator = RiscoDataUpdateCoordinator(hass, risco, scan_interval)
+ coordinator = RiscoDataUpdateCoordinator(hass, entry, risco)
await coordinator.async_config_entry_first_refresh()
- events_coordinator = RiscoEventsDataUpdateCoordinator(
- hass, risco, entry.entry_id, 60
- )
+ events_coordinator = RiscoEventsDataUpdateCoordinator(hass, entry, risco)
entry.async_on_unload(entry.add_update_listener(_update_listener))
diff --git a/homeassistant/components/risco/alarm_control_panel.py b/homeassistant/components/risco/alarm_control_panel.py
index b1eae8fd917..2472baa932e 100644
--- a/homeassistant/components/risco/alarm_control_panel.py
+++ b/homeassistant/components/risco/alarm_control_panel.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PIN
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LocalData, is_local
from .const import (
@@ -50,7 +50,7 @@ STATES_TO_SUPPORTED_FEATURES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Risco alarm control panel."""
options = {**DEFAULT_OPTIONS, **config_entry.options}
diff --git a/homeassistant/components/risco/binary_sensor.py b/homeassistant/components/risco/binary_sensor.py
index a7ca0129b06..ff61985fef3 100644
--- a/homeassistant/components/risco/binary_sensor.py
+++ b/homeassistant/components/risco/binary_sensor.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LocalData, is_local
from .const import DATA_COORDINATOR, DOMAIN, SYSTEM_UPDATE_SIGNAL
@@ -73,7 +73,7 @@ SYSTEM_ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Risco alarm control panel."""
if is_local(config_entry):
diff --git a/homeassistant/components/risco/const.py b/homeassistant/components/risco/const.py
index 078e26c43b5..ef3280fe232 100644
--- a/homeassistant/components/risco/const.py
+++ b/homeassistant/components/risco/const.py
@@ -30,9 +30,9 @@ RISCO_ARM = "arm"
RISCO_PARTIAL_ARM = "partial_arm"
RISCO_STATES = [RISCO_ARM, RISCO_PARTIAL_ARM, *RISCO_GROUPS]
-DEFAULT_RISCO_GROUPS_TO_HA = {
- group: AlarmControlPanelState.ARMED_HOME for group in RISCO_GROUPS
-}
+DEFAULT_RISCO_GROUPS_TO_HA = dict.fromkeys(
+ RISCO_GROUPS, AlarmControlPanelState.ARMED_HOME
+)
DEFAULT_RISCO_STATES_TO_HA = {
RISCO_ARM: AlarmControlPanelState.ARMED_AWAY,
RISCO_PARTIAL_ARM: AlarmControlPanelState.ARMED_HOME,
diff --git a/homeassistant/components/risco/coordinator.py b/homeassistant/components/risco/coordinator.py
index 8430b6a6172..e7140eb9616 100644
--- a/homeassistant/components/risco/coordinator.py
+++ b/homeassistant/components/risco/coordinator.py
@@ -10,11 +10,13 @@ from pyrisco import CannotConnectError, OperationError, RiscoCloud, Unauthorized
from pyrisco.cloud.alarm import Alarm
from pyrisco.cloud.event import Event
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_SCAN_INTERVAL
from homeassistant.core import HomeAssistant
from homeassistant.helpers.storage import Store
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import DOMAIN
+from .const import DEFAULT_SCAN_INTERVAL, DOMAIN
LAST_EVENT_STORAGE_VERSION = 1
LAST_EVENT_TIMESTAMP_KEY = "last_event_timestamp"
@@ -24,17 +26,26 @@ _LOGGER = logging.getLogger(__name__)
class RiscoDataUpdateCoordinator(DataUpdateCoordinator[Alarm]):
"""Class to manage fetching risco data."""
+ config_entry: ConfigEntry
+
def __init__(
- self, hass: HomeAssistant, risco: RiscoCloud, scan_interval: int
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ risco: RiscoCloud,
) -> None:
"""Initialize global risco data updater."""
self.risco = risco
- interval = timedelta(seconds=scan_interval)
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
- update_interval=interval,
+ update_interval=timedelta(
+ seconds=config_entry.options.get(
+ CONF_SCAN_INTERVAL, DEFAULT_SCAN_INTERVAL
+ )
+ ),
)
async def _async_update_data(self) -> Alarm:
@@ -48,20 +59,27 @@ class RiscoDataUpdateCoordinator(DataUpdateCoordinator[Alarm]):
class RiscoEventsDataUpdateCoordinator(DataUpdateCoordinator[list[Event]]):
"""Class to manage fetching risco data."""
+ config_entry: ConfigEntry
+
def __init__(
- self, hass: HomeAssistant, risco: RiscoCloud, eid: str, scan_interval: int
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ risco: RiscoCloud,
) -> None:
"""Initialize global risco data updater."""
self.risco = risco
self._store = Store[dict[str, Any]](
- hass, LAST_EVENT_STORAGE_VERSION, f"risco_{eid}_last_event_timestamp"
+ hass,
+ LAST_EVENT_STORAGE_VERSION,
+ f"risco_{config_entry.entry_id}_last_event_timestamp",
)
- interval = timedelta(seconds=scan_interval)
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"{DOMAIN}_events",
- update_interval=interval,
+ update_interval=timedelta(seconds=60),
)
async def _async_update_data(self) -> list[Event]:
diff --git a/homeassistant/components/risco/manifest.json b/homeassistant/components/risco/manifest.json
index 149b8761589..43d471172d6 100644
--- a/homeassistant/components/risco/manifest.json
+++ b/homeassistant/components/risco/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/risco",
"iot_class": "local_push",
"loggers": ["pyrisco"],
- "requirements": ["pyrisco==0.6.5"]
+ "requirements": ["pyrisco==0.6.7"]
}
diff --git a/homeassistant/components/risco/sensor.py b/homeassistant/components/risco/sensor.py
index c1495512e62..93683f1aa50 100644
--- a/homeassistant/components/risco/sensor.py
+++ b/homeassistant/components/risco/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -46,7 +46,7 @@ EVENT_ATTRIBUTES = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for device."""
if is_local(config_entry):
diff --git a/homeassistant/components/risco/switch.py b/homeassistant/components/risco/switch.py
index 8bad2c6c15e..547dedd3933 100644
--- a/homeassistant/components/risco/switch.py
+++ b/homeassistant/components/risco/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import LocalData, is_local
from .const import DATA_COORDINATOR, DOMAIN
@@ -21,7 +21,7 @@ from .entity import RiscoCloudZoneEntity, RiscoLocalZoneEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Risco switch."""
if is_local(config_entry):
diff --git a/homeassistant/components/rituals_perfume_genie/__init__.py b/homeassistant/components/rituals_perfume_genie/__init__.py
index d0d16ba6324..e920c2426fe 100644
--- a/homeassistant/components/rituals_perfume_genie/__init__.py
+++ b/homeassistant/components/rituals_perfume_genie/__init__.py
@@ -44,7 +44,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
# Create a coordinator for each diffuser
coordinators = {
- diffuser.hublot: RitualsDataUpdateCoordinator(hass, diffuser, update_interval)
+ diffuser.hublot: RitualsDataUpdateCoordinator(
+ hass, entry, diffuser, update_interval
+ )
for diffuser in account_devices
}
diff --git a/homeassistant/components/rituals_perfume_genie/binary_sensor.py b/homeassistant/components/rituals_perfume_genie/binary_sensor.py
index 63666fc1aca..97e9c8418d1 100644
--- a/homeassistant/components/rituals_perfume_genie/binary_sensor.py
+++ b/homeassistant/components/rituals_perfume_genie/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RitualsDataUpdateCoordinator
@@ -44,7 +44,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the diffuser binary sensors."""
coordinators: dict[str, RitualsDataUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/rituals_perfume_genie/coordinator.py b/homeassistant/components/rituals_perfume_genie/coordinator.py
index a83e823bd4e..bbcb24b3e65 100644
--- a/homeassistant/components/rituals_perfume_genie/coordinator.py
+++ b/homeassistant/components/rituals_perfume_genie/coordinator.py
@@ -5,6 +5,7 @@ import logging
from pyrituals import Diffuser
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -16,9 +17,12 @@ _LOGGER = logging.getLogger(__name__)
class RitualsDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Class to manage fetching Rituals Perfume Genie device data from single endpoint."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
diffuser: Diffuser,
update_interval: timedelta,
) -> None:
@@ -27,6 +31,7 @@ class RitualsDataUpdateCoordinator(DataUpdateCoordinator[None]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"{DOMAIN}-{diffuser.hublot}",
update_interval=update_interval,
)
diff --git a/homeassistant/components/rituals_perfume_genie/number.py b/homeassistant/components/rituals_perfume_genie/number.py
index 0ac9c30f285..98e833ff9bd 100644
--- a/homeassistant/components/rituals_perfume_genie/number.py
+++ b/homeassistant/components/rituals_perfume_genie/number.py
@@ -11,7 +11,7 @@ from pyrituals import Diffuser
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RitualsDataUpdateCoordinator
@@ -41,7 +41,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the diffuser numbers."""
coordinators: dict[str, RitualsDataUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/rituals_perfume_genie/select.py b/homeassistant/components/rituals_perfume_genie/select.py
index 27aff70649b..c239627e9c6 100644
--- a/homeassistant/components/rituals_perfume_genie/select.py
+++ b/homeassistant/components/rituals_perfume_genie/select.py
@@ -11,7 +11,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfArea
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RitualsDataUpdateCoordinator
@@ -44,7 +44,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the diffuser select entities."""
coordinators: dict[str, RitualsDataUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/rituals_perfume_genie/sensor.py b/homeassistant/components/rituals_perfume_genie/sensor.py
index 46faa8d73e9..3921fd0b6c2 100644
--- a/homeassistant/components/rituals_perfume_genie/sensor.py
+++ b/homeassistant/components/rituals_perfume_genie/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RitualsDataUpdateCoordinator
@@ -60,7 +60,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the diffuser sensors."""
coordinators: dict[str, RitualsDataUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/rituals_perfume_genie/switch.py b/homeassistant/components/rituals_perfume_genie/switch.py
index b5828f5ca07..c5331b49078 100644
--- a/homeassistant/components/rituals_perfume_genie/switch.py
+++ b/homeassistant/components/rituals_perfume_genie/switch.py
@@ -11,7 +11,7 @@ from pyrituals import Diffuser
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RitualsDataUpdateCoordinator
@@ -42,7 +42,7 @@ ENTITY_DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the diffuser switch."""
coordinators: dict[str, RitualsDataUpdateCoordinator] = hass.data[DOMAIN][
diff --git a/homeassistant/components/roborock/__init__.py b/homeassistant/components/roborock/__init__.py
index b383c1acfd7..8140b58b86c 100644
--- a/homeassistant/components/roborock/__init__.py
+++ b/homeassistant/components/roborock/__init__.py
@@ -4,7 +4,6 @@ from __future__ import annotations
import asyncio
from collections.abc import Coroutine
-from dataclasses import dataclass
from datetime import timedelta
import logging
from typing import Any
@@ -21,35 +20,25 @@ from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1
from roborock.version_a01_apis import RoborockMqttClientA01
from roborock.web_api import RoborockApiClient
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_USERNAME, EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import CONF_BASE_URL, CONF_USER_DATA, DOMAIN, PLATFORMS
-from .coordinator import RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01
+from .coordinator import (
+ RoborockConfigEntry,
+ RoborockCoordinators,
+ RoborockDataUpdateCoordinator,
+ RoborockDataUpdateCoordinatorA01,
+)
from .roborock_storage import async_remove_map_storage
SCAN_INTERVAL = timedelta(seconds=30)
_LOGGER = logging.getLogger(__name__)
-type RoborockConfigEntry = ConfigEntry[RoborockCoordinators]
-
-
-@dataclass
-class RoborockCoordinators:
- """Roborock coordinators type."""
-
- v1: list[RoborockDataUpdateCoordinator]
- a01: list[RoborockDataUpdateCoordinatorA01]
-
- def values(
- self,
- ) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01]:
- """Return all coordinators."""
- return self.v1 + self.a01
-
async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) -> bool:
"""Set up roborock from a config entry."""
@@ -57,7 +46,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) ->
entry.async_on_unload(entry.add_update_listener(update_listener))
user_data = UserData.from_dict(entry.data[CONF_USER_DATA])
- api_client = RoborockApiClient(entry.data[CONF_USERNAME], entry.data[CONF_BASE_URL])
+ api_client = RoborockApiClient(
+ entry.data[CONF_USERNAME],
+ entry.data[CONF_BASE_URL],
+ session=async_get_clientsession(hass),
+ )
_LOGGER.debug("Getting home data")
try:
home_data = await api_client.get_home_data_v2(user_data)
@@ -78,6 +71,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) ->
translation_key="no_user_agreement",
) from err
except RoborockException as err:
+ _LOGGER.debug("Failed to get Roborock home data: %s", err)
raise ConfigEntryNotReady(
"Failed to get Roborock home data",
translation_domain=DOMAIN,
@@ -95,7 +89,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) ->
# Get a Coordinator if the device is available or if we have connected to the device before
coordinators = await asyncio.gather(
*build_setup_functions(
- hass, device_map, user_data, product_info, home_data.rooms
+ hass,
+ entry,
+ device_map,
+ user_data,
+ product_info,
+ home_data.rooms,
+ api_client,
),
return_exceptions=True,
)
@@ -117,6 +117,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) ->
translation_key="no_coordinators",
)
valid_coordinators = RoborockCoordinators(v1_coords, a01_coords)
+ await asyncio.gather(
+ *(coord.refresh_coordinator_map() for coord in valid_coordinators.v1)
+ )
async def on_stop(_: Any) -> None:
_LOGGER.debug("Shutting down roborock")
@@ -137,15 +140,38 @@ async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) ->
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+ device_registry = dr.async_get(hass)
+ device_entries = dr.async_entries_for_config_entry(
+ device_registry, config_entry_id=entry.entry_id
+ )
+ for device in device_entries:
+ # Remove any devices that are no longer in the account.
+ # The API returns all devices, even if they are offline
+ device_duids = {
+ identifier[1].replace("_dock", "") for identifier in device.identifiers
+ }
+ if any(device_duid in device_map for device_duid in device_duids):
+ continue
+ _LOGGER.info(
+ "Removing device: %s because it is no longer exists in your account",
+ device.name,
+ )
+ device_registry.async_update_device(
+ device_id=device.id,
+ remove_config_entry_id=entry.entry_id,
+ )
+
return True
def build_setup_functions(
hass: HomeAssistant,
+ entry: RoborockConfigEntry,
device_map: dict[str, HomeDataDevice],
user_data: UserData,
product_info: dict[str, HomeDataProduct],
home_data_rooms: list[HomeDataRoom],
+ api_client: RoborockApiClient,
) -> list[
Coroutine[
Any,
@@ -156,7 +182,13 @@ def build_setup_functions(
"""Create a list of setup functions that can later be called asynchronously."""
return [
setup_device(
- hass, user_data, device, product_info[device.product_id], home_data_rooms
+ hass,
+ entry,
+ user_data,
+ device,
+ product_info[device.product_id],
+ home_data_rooms,
+ api_client,
)
for device in device_map.values()
]
@@ -164,18 +196,20 @@ def build_setup_functions(
async def setup_device(
hass: HomeAssistant,
+ entry: RoborockConfigEntry,
user_data: UserData,
device: HomeDataDevice,
product_info: HomeDataProduct,
home_data_rooms: list[HomeDataRoom],
+ api_client: RoborockApiClient,
) -> RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | None:
"""Set up a coordinator for a given device."""
if device.pv == "1.0":
return await setup_device_v1(
- hass, user_data, device, product_info, home_data_rooms
+ hass, entry, user_data, device, product_info, home_data_rooms, api_client
)
if device.pv == "A01":
- return await setup_device_a01(hass, user_data, device, product_info)
+ return await setup_device_a01(hass, entry, user_data, device, product_info)
_LOGGER.warning(
"Not adding device %s because its protocol version %s or category %s is not supported",
device.duid,
@@ -187,10 +221,12 @@ async def setup_device(
async def setup_device_v1(
hass: HomeAssistant,
+ entry: RoborockConfigEntry,
user_data: UserData,
device: HomeDataDevice,
product_info: HomeDataProduct,
home_data_rooms: list[HomeDataRoom],
+ api_client: RoborockApiClient,
) -> RoborockDataUpdateCoordinator | None:
"""Set up a device Coordinator."""
mqtt_client = await hass.async_add_executor_job(
@@ -212,7 +248,15 @@ async def setup_device_v1(
await mqtt_client.async_release()
raise
coordinator = RoborockDataUpdateCoordinator(
- hass, device, networking, product_info, mqtt_client, home_data_rooms
+ hass,
+ entry,
+ device,
+ networking,
+ product_info,
+ mqtt_client,
+ home_data_rooms,
+ api_client,
+ user_data,
)
try:
await coordinator.async_config_entry_first_refresh()
@@ -246,6 +290,7 @@ async def setup_device_v1(
async def setup_device_a01(
hass: HomeAssistant,
+ entry: RoborockConfigEntry,
user_data: UserData,
device: HomeDataDevice,
product_info: HomeDataProduct,
@@ -254,7 +299,9 @@ async def setup_device_a01(
mqtt_client = RoborockMqttClientA01(
user_data, DeviceData(device, product_info.name), product_info.category
)
- coord = RoborockDataUpdateCoordinatorA01(hass, device, product_info, mqtt_client)
+ coord = RoborockDataUpdateCoordinatorA01(
+ hass, entry, device, product_info, mqtt_client
+ )
await coord.async_config_entry_first_refresh()
return coord
diff --git a/homeassistant/components/roborock/binary_sensor.py b/homeassistant/components/roborock/binary_sensor.py
index b88556ea857..a2c34f5c59d 100644
--- a/homeassistant/components/roborock/binary_sensor.py
+++ b/homeassistant/components/roborock/binary_sensor.py
@@ -5,6 +5,7 @@ from __future__ import annotations
from collections.abc import Callable
from dataclasses import dataclass
+from roborock.containers import RoborockStateCode
from roborock.roborock_typing import DeviceProp
from homeassistant.components.binary_sensor import (
@@ -12,20 +13,23 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.const import EntityCategory
+from homeassistant.const import ATTR_BATTERY_CHARGING, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
from .entity import RoborockCoordinatedEntityV1
+PARALLEL_UPDATES = 0
+
@dataclass(frozen=True, kw_only=True)
class RoborockBinarySensorDescription(BinarySensorEntityDescription):
"""A class that describes Roborock binary sensors."""
value_fn: Callable[[DeviceProp], bool | int | None]
+ # If it is a dock entity
+ is_dock_entity: bool = False
BINARY_SENSOR_DESCRIPTIONS = [
@@ -35,6 +39,7 @@ BINARY_SENSOR_DESCRIPTIONS = [
device_class=BinarySensorDeviceClass.RUNNING,
entity_category=EntityCategory.DIAGNOSTIC,
value_fn=lambda data: data.status.dry_status,
+ is_dock_entity=True,
),
RoborockBinarySensorDescription(
key="water_box_carriage_status",
@@ -64,13 +69,20 @@ BINARY_SENSOR_DESCRIPTIONS = [
entity_category=EntityCategory.DIAGNOSTIC,
value_fn=lambda data: data.status.in_cleaning,
),
+ RoborockBinarySensorDescription(
+ key=ATTR_BATTERY_CHARGING,
+ device_class=BinarySensorDeviceClass.BATTERY_CHARGING,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ value_fn=lambda data: data.status.state
+ in (RoborockStateCode.charging, RoborockStateCode.charging_complete),
+ ),
]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Roborock vacuum binary sensors."""
async_add_entities(
@@ -98,6 +110,7 @@ class RoborockBinarySensorEntity(RoborockCoordinatedEntityV1, BinarySensorEntity
super().__init__(
f"{description.key}_{coordinator.duid_slug}",
coordinator,
+ is_dock_entity=description.is_dock_entity,
)
self.entity_description = description
diff --git a/homeassistant/components/roborock/button.py b/homeassistant/components/roborock/button.py
index 2f214c7c51c..fea38524fe0 100644
--- a/homeassistant/components/roborock/button.py
+++ b/homeassistant/components/roborock/button.py
@@ -2,18 +2,22 @@
from __future__ import annotations
+import asyncio
from dataclasses import dataclass
+import itertools
+from typing import Any
from roborock.roborock_typing import RoborockCommand
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator
-from .entity import RoborockEntityV1
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
+from .entity import RoborockEntity, RoborockEntityV1
+
+PARALLEL_UPDATES = 0
@dataclass(frozen=True, kw_only=True)
@@ -63,17 +67,37 @@ CONSUMABLE_BUTTON_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roborock button platform."""
+ routines_lists = await asyncio.gather(
+ *[coordinator.get_routines() for coordinator in config_entry.runtime_data.v1],
+ )
async_add_entities(
- RoborockButtonEntity(
- coordinator,
- description,
+ itertools.chain(
+ (
+ RoborockButtonEntity(
+ coordinator,
+ description,
+ )
+ for coordinator in config_entry.runtime_data.v1
+ for description in CONSUMABLE_BUTTON_DESCRIPTIONS
+ if isinstance(coordinator, RoborockDataUpdateCoordinator)
+ ),
+ (
+ RoborockRoutineButtonEntity(
+ coordinator,
+ ButtonEntityDescription(
+ key=str(routine.id),
+ name=routine.name,
+ ),
+ )
+ for coordinator, routines in zip(
+ config_entry.runtime_data.v1, routines_lists, strict=True
+ )
+ for routine in routines
+ ),
)
- for coordinator in config_entry.runtime_data.v1
- for description in CONSUMABLE_BUTTON_DESCRIPTIONS
- if isinstance(coordinator, RoborockDataUpdateCoordinator)
)
@@ -98,3 +122,28 @@ class RoborockButtonEntity(RoborockEntityV1, ButtonEntity):
async def async_press(self) -> None:
"""Press the button."""
await self.send(self.entity_description.command, self.entity_description.param)
+
+
+class RoborockRoutineButtonEntity(RoborockEntity, ButtonEntity):
+ """A class to define Roborock routines button entities."""
+
+ entity_description: ButtonEntityDescription
+
+ def __init__(
+ self,
+ coordinator: RoborockDataUpdateCoordinator,
+ entity_description: ButtonEntityDescription,
+ ) -> None:
+ """Create a button entity."""
+ super().__init__(
+ f"{entity_description.key}_{coordinator.duid_slug}",
+ coordinator.device_info,
+ coordinator.api,
+ )
+ self._routine_id = int(entity_description.key)
+ self._coordinator = coordinator
+ self.entity_description = entity_description
+
+ async def async_press(self, **kwargs: Any) -> None:
+ """Press the button."""
+ await self._coordinator.execute_routines(self._routine_id)
diff --git a/homeassistant/components/roborock/config_flow.py b/homeassistant/components/roborock/config_flow.py
index 1a6b67286bb..1a359faca10 100644
--- a/homeassistant/components/roborock/config_flow.py
+++ b/homeassistant/components/roborock/config_flow.py
@@ -21,14 +21,17 @@ import voluptuous as vol
from homeassistant.config_entries import (
SOURCE_REAUTH,
- ConfigEntry,
ConfigFlow,
ConfigFlowResult,
OptionsFlow,
)
from homeassistant.const import CONF_USERNAME
from homeassistant.core import callback
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
+from . import RoborockConfigEntry
from .const import (
CONF_BASE_URL,
CONF_ENTRY_CODE,
@@ -63,7 +66,9 @@ class RoborockFlowHandler(ConfigFlow, domain=DOMAIN):
self._abort_if_unique_id_configured(error="already_configured_account")
self._username = username
_LOGGER.debug("Requesting code for Roborock account")
- self._client = RoborockApiClient(username)
+ self._client = RoborockApiClient(
+ username, session=async_get_clientsession(self.hass)
+ )
errors = await self._request_code()
if not errors:
return await self.async_step_code()
@@ -134,13 +139,31 @@ class RoborockFlowHandler(ConfigFlow, domain=DOMAIN):
errors=errors,
)
+ async def async_step_dhcp(
+ self, discovery_info: DhcpServiceInfo
+ ) -> ConfigFlowResult:
+ """Handle a flow started by a dhcp discovery."""
+ device_registry = dr.async_get(self.hass)
+ device = device_registry.async_get_device(
+ connections={
+ (dr.CONNECTION_NETWORK_MAC, dr.format_mac(discovery_info.macaddress))
+ }
+ )
+ if device is not None and any(
+ identifier[0] == DOMAIN for identifier in device.identifiers
+ ):
+ return self.async_abort(reason="already_configured")
+ return await self.async_step_user()
+
async def async_step_reauth(
self, entry_data: Mapping[str, Any]
) -> ConfigFlowResult:
"""Perform reauth upon an API authentication error."""
self._username = entry_data[CONF_USERNAME]
assert self._username
- self._client = RoborockApiClient(self._username)
+ self._client = RoborockApiClient(
+ self._username, session=async_get_clientsession(self.hass)
+ )
return await self.async_step_reauth_confirm()
async def async_step_reauth_confirm(
@@ -170,7 +193,7 @@ class RoborockFlowHandler(ConfigFlow, domain=DOMAIN):
@staticmethod
@callback
def async_get_options_flow(
- config_entry: ConfigEntry,
+ config_entry: RoborockConfigEntry,
) -> RoborockOptionsFlowHandler:
"""Create the options flow."""
return RoborockOptionsFlowHandler(config_entry)
@@ -179,7 +202,7 @@ class RoborockFlowHandler(ConfigFlow, domain=DOMAIN):
class RoborockOptionsFlowHandler(OptionsFlow):
"""Handle an option flow for Roborock."""
- def __init__(self, config_entry: ConfigEntry) -> None:
+ def __init__(self, config_entry: RoborockConfigEntry) -> None:
"""Initialize options flow."""
self.options = deepcopy(dict(config_entry.options))
diff --git a/homeassistant/components/roborock/const.py b/homeassistant/components/roborock/const.py
index cc8d34fbadc..e56fade7078 100644
--- a/homeassistant/components/roborock/const.py
+++ b/homeassistant/components/roborock/const.py
@@ -1,5 +1,7 @@
"""Constants for Roborock."""
+from datetime import timedelta
+
from vacuum_map_parser_base.config.drawable import Drawable
from homeassistant.const import Platform
@@ -43,13 +45,21 @@ PLATFORMS = [
Platform.VACUUM,
]
-
-IMAGE_CACHE_INTERVAL = 90
+# This can be lowered in the future if we do not receive rate limiting issues.
+IMAGE_CACHE_INTERVAL = timedelta(seconds=30)
MAP_SLEEP = 3
GET_MAPS_SERVICE_NAME = "get_maps"
+MAP_SCALE = 4
MAP_FILE_FORMAT = "PNG"
MAP_FILENAME_SUFFIX = ".png"
SET_VACUUM_GOTO_POSITION_SERVICE_NAME = "set_vacuum_goto_position"
GET_VACUUM_CURRENT_POSITION_SERVICE_NAME = "get_vacuum_current_position"
+
+
+A01_UPDATE_INTERVAL = timedelta(minutes=1)
+V1_CLOUD_IN_CLEANING_INTERVAL = timedelta(seconds=30)
+V1_CLOUD_NOT_CLEANING_INTERVAL = timedelta(minutes=1)
+V1_LOCAL_IN_CLEANING_INTERVAL = timedelta(seconds=15)
+V1_LOCAL_NOT_CLEANING_INTERVAL = timedelta(seconds=30)
diff --git a/homeassistant/components/roborock/coordinator.py b/homeassistant/components/roborock/coordinator.py
index 8860a5c1f43..cc0bee1cd5f 100644
--- a/homeassistant/components/roborock/coordinator.py
+++ b/homeassistant/components/roborock/coordinator.py
@@ -3,30 +3,59 @@
from __future__ import annotations
import asyncio
+from dataclasses import dataclass
from datetime import timedelta
+import io
import logging
from propcache.api import cached_property
from roborock import HomeDataRoom
from roborock.code_mappings import RoborockCategory
-from roborock.containers import DeviceData, HomeDataDevice, HomeDataProduct, NetworkInfo
+from roborock.containers import (
+ DeviceData,
+ HomeDataDevice,
+ HomeDataProduct,
+ HomeDataScene,
+ NetworkInfo,
+ UserData,
+)
from roborock.exceptions import RoborockException
from roborock.roborock_message import RoborockDyadDataProtocol, RoborockZeoProtocol
from roborock.roborock_typing import DeviceProp
from roborock.version_1_apis.roborock_local_client_v1 import RoborockLocalClientV1
from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1
from roborock.version_a01_apis import RoborockClientA01
+from roborock.web_api import RoborockApiClient
+from vacuum_map_parser_base.config.color import ColorsPalette
+from vacuum_map_parser_base.config.image_config import ImageConfig
+from vacuum_map_parser_base.config.size import Sizes
+from vacuum_map_parser_base.map_data import MapData
+from vacuum_map_parser_roborock.map_data_parser import RoborockMapDataParser
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_CONNECTIONS
from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from homeassistant.util import slugify
+from homeassistant.util import dt as dt_util, slugify
-from .const import DOMAIN
+from .const import (
+ A01_UPDATE_INTERVAL,
+ DEFAULT_DRAWABLES,
+ DOMAIN,
+ DRAWABLES,
+ IMAGE_CACHE_INTERVAL,
+ MAP_FILE_FORMAT,
+ MAP_SCALE,
+ MAP_SLEEP,
+ V1_CLOUD_IN_CLEANING_INTERVAL,
+ V1_CLOUD_NOT_CLEANING_INTERVAL,
+ V1_LOCAL_IN_CLEANING_INTERVAL,
+ V1_LOCAL_NOT_CLEANING_INTERVAL,
+)
from .models import RoborockA01HassDeviceInfo, RoborockHassDeviceInfo, RoborockMapInfo
from .roborock_storage import RoborockMapStorage
@@ -35,22 +64,49 @@ SCAN_INTERVAL = timedelta(seconds=30)
_LOGGER = logging.getLogger(__name__)
+@dataclass
+class RoborockCoordinators:
+ """Roborock coordinators type."""
+
+ v1: list[RoborockDataUpdateCoordinator]
+ a01: list[RoborockDataUpdateCoordinatorA01]
+
+ def values(
+ self,
+ ) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01]:
+ """Return all coordinators."""
+ return self.v1 + self.a01
+
+
+type RoborockConfigEntry = ConfigEntry[RoborockCoordinators]
+
+
class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
"""Class to manage fetching data from the API."""
- config_entry: ConfigEntry
+ config_entry: RoborockConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: RoborockConfigEntry,
device: HomeDataDevice,
device_networking: NetworkInfo,
product_info: HomeDataProduct,
cloud_api: RoborockMqttClientV1,
home_data_rooms: list[HomeDataRoom],
+ api_client: RoborockApiClient,
+ user_data: UserData,
) -> None:
"""Initialize."""
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ # Assume we can use the local api.
+ update_interval=V1_LOCAL_NOT_CLEANING_INTERVAL,
+ )
self.roborock_device_info = RoborockHassDeviceInfo(
device,
device_networking,
@@ -64,7 +120,7 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
self.cloud_api = cloud_api
self.device_info = DeviceInfo(
name=self.roborock_device_info.device.name,
- identifiers={(DOMAIN, self.roborock_device_info.device.duid)},
+ identifiers={(DOMAIN, self.duid)},
manufacturer="Roborock",
model=self.roborock_device_info.product.model,
model_id=self.roborock_device_info.product.model,
@@ -73,13 +129,62 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
self.current_map: int | None = None
if mac := self.roborock_device_info.network_info.mac:
- self.device_info[ATTR_CONNECTIONS] = {(dr.CONNECTION_NETWORK_MAC, mac)}
+ self.device_info[ATTR_CONNECTIONS] = {
+ (dr.CONNECTION_NETWORK_MAC, dr.format_mac(mac))
+ }
# Maps from map flag to map name
self.maps: dict[int, RoborockMapInfo] = {}
self._home_data_rooms = {str(room.id): room.name for room in home_data_rooms}
self.map_storage = RoborockMapStorage(
- hass, self.config_entry.entry_id, slugify(self.duid)
+ hass, self.config_entry.entry_id, self.duid_slug
)
+ self._user_data = user_data
+ self._api_client = api_client
+ self._is_cloud_api = False
+ drawables = [
+ drawable
+ for drawable, default_value in DEFAULT_DRAWABLES.items()
+ if config_entry.options.get(DRAWABLES, {}).get(drawable, default_value)
+ ]
+ self.map_parser = RoborockMapDataParser(
+ ColorsPalette(),
+ Sizes({k: v * MAP_SCALE for k, v in Sizes.SIZES.items()}),
+ drawables,
+ ImageConfig(scale=MAP_SCALE),
+ [],
+ )
+
+ @cached_property
+ def dock_device_info(self) -> DeviceInfo:
+ """Gets the device info for the dock.
+
+ This must happen after the coordinator does the first update.
+ Which will be the case when this is called.
+ """
+ dock_type = self.roborock_device_info.props.status.dock_type
+ return DeviceInfo(
+ name=f"{self.roborock_device_info.device.name} Dock",
+ identifiers={(DOMAIN, f"{self.duid}_dock")},
+ manufacturer="Roborock",
+ model=f"{self.roborock_device_info.product.model} Dock",
+ model_id=str(dock_type.value) if dock_type is not None else "Unknown",
+ sw_version=self.roborock_device_info.device.fv,
+ )
+
+ def parse_map_data_v1(
+ self, map_bytes: bytes
+ ) -> tuple[bytes | None, MapData | None]:
+ """Parse map_bytes and return MapData and the image."""
+ try:
+ parsed_map = self.map_parser.parse(map_bytes)
+ except (IndexError, ValueError) as err:
+ _LOGGER.debug("Exception when parsing map contents: %s", err)
+ return None, None
+ if parsed_map.image is None:
+ return None, None
+ img_byte_arr = io.BytesIO()
+ parsed_map.image.data.save(img_byte_arr, format=MAP_FILE_FORMAT)
+ return img_byte_arr.getvalue(), parsed_map
async def _async_setup(self) -> None:
"""Set up the coordinator."""
@@ -90,17 +195,68 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
try:
maps = await self.api.get_multi_maps_list()
except RoborockException as err:
- raise UpdateFailed("Failed to get map data: {err}") from err
+ _LOGGER.debug("Failed to get maps: %s", err)
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="map_failure",
+ translation_placeholders={"error": str(err)},
+ ) from err
# Rooms names populated later with calls to `set_current_map_rooms` for each map
+ roborock_maps = maps.map_info if (maps and maps.map_info) else ()
+ stored_images = await asyncio.gather(
+ *[
+ self.map_storage.async_load_map(roborock_map.mapFlag)
+ for roborock_map in roborock_maps
+ ]
+ )
self.maps = {
roborock_map.mapFlag: RoborockMapInfo(
flag=roborock_map.mapFlag,
name=roborock_map.name or f"Map {roborock_map.mapFlag}",
rooms={},
+ image=image,
+ last_updated=dt_util.utcnow() - IMAGE_CACHE_INTERVAL,
+ map_data=None,
)
- for roborock_map in (maps.map_info if (maps and maps.map_info) else ())
+ for image, roborock_map in zip(stored_images, roborock_maps, strict=False)
}
+ async def update_map(self) -> None:
+ """Update the currently selected map."""
+ # The current map was set in the props update, so these can be done without
+ # worry of applying them to the wrong map.
+ if self.current_map is None:
+ # This exists as a safeguard/ to keep mypy happy.
+ return
+ try:
+ response = await self.cloud_api.get_map_v1()
+ except RoborockException as ex:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="map_failure",
+ ) from ex
+ if not isinstance(response, bytes):
+ _LOGGER.debug("Failed to parse map contents: %s", response)
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="map_failure",
+ )
+ parsed_image, parsed_map = self.parse_map_data_v1(response)
+ if parsed_image is None or parsed_map is None:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="map_failure",
+ )
+ current_roborock_map_info = self.maps[self.current_map]
+ if parsed_image != self.maps[self.current_map].image:
+ await self.map_storage.async_save_map(
+ self.current_map,
+ parsed_image,
+ )
+ current_roborock_map_info.image = parsed_image
+ current_roborock_map_info.last_updated = dt_util.utcnow()
+ current_roborock_map_info.map_data = parsed_map
+
async def _verify_api(self) -> None:
"""Verify that the api is reachable. If it is not, switch clients."""
if isinstance(self.api, RoborockLocalClientV1):
@@ -109,11 +265,13 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
except RoborockException:
_LOGGER.warning(
"Using the cloud API for device %s. This is not recommended as it can lead to rate limiting. We recommend making your vacuum accessible by your Home Assistant instance",
- self.roborock_device_info.device.duid,
+ self.duid,
)
await self.api.async_disconnect()
# We use the cloud api if the local api fails to connect.
self.api = self.cloud_api
+ self.update_interval = V1_CLOUD_NOT_CLEANING_INTERVAL
+ self._is_cloud_api = True
# Right now this should never be called if the cloud api is the primary api,
# but in the future if it is, a new else should be added.
@@ -133,15 +291,45 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
async def _async_update_data(self) -> DeviceProp:
"""Update data via library."""
+ previous_state = self.roborock_device_info.props.status.state_name
try:
# Update device props and standard api information
await self._update_device_prop()
# Set the new map id from the updated device props
self._set_current_map()
# Get the rooms for that map id.
+
+ # If the vacuum is currently cleaning and it has been IMAGE_CACHE_INTERVAL
+ # since the last map update, you can update the map.
+ new_status = self.roborock_device_info.props.status
+ if self.current_map is not None and (
+ (
+ new_status.in_cleaning
+ and (dt_util.utcnow() - self.maps[self.current_map].last_updated)
+ > IMAGE_CACHE_INTERVAL
+ )
+ or previous_state != new_status.state_name
+ ):
+ try:
+ await self.update_map()
+ except HomeAssistantError as err:
+ _LOGGER.debug("Failed to update map: %s", err)
await self.set_current_map_rooms()
except RoborockException as ex:
- raise UpdateFailed(ex) from ex
+ _LOGGER.debug("Failed to update data: %s", ex)
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="update_data_fail",
+ ) from ex
+ if self.roborock_device_info.props.status.in_cleaning:
+ if self._is_cloud_api:
+ self.update_interval = V1_CLOUD_IN_CLEANING_INTERVAL
+ else:
+ self.update_interval = V1_LOCAL_IN_CLEANING_INTERVAL
+ elif self._is_cloud_api:
+ self.update_interval = V1_CLOUD_NOT_CLEANING_INTERVAL
+ else:
+ self.update_interval = V1_LOCAL_NOT_CLEANING_INTERVAL
return self.roborock_device_info.props
def _set_current_map(self) -> None:
@@ -168,6 +356,34 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
for room in room_mapping or ()
}
+ async def get_routines(self) -> list[HomeDataScene]:
+ """Get routines."""
+ try:
+ return await self._api_client.get_scenes(self._user_data, self.duid)
+ except RoborockException as err:
+ _LOGGER.error("Failed to get routines %s", err)
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="command_failed",
+ translation_placeholders={
+ "command": "get_scenes",
+ },
+ ) from err
+
+ async def execute_routines(self, routine_id: int) -> None:
+ """Execute routines."""
+ try:
+ await self._api_client.execute_scene(self._user_data, routine_id)
+ except RoborockException as err:
+ _LOGGER.error("Failed to execute routines %s %s", routine_id, err)
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="command_failed",
+ translation_placeholders={
+ "command": "execute_scene",
+ },
+ ) from err
+
@cached_property
def duid(self) -> str:
"""Get the unique id of the device as specified by Roborock."""
@@ -178,6 +394,43 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
"""Get the slug of the duid."""
return slugify(self.duid)
+ async def refresh_coordinator_map(self) -> None:
+ """Get the starting map information for all maps for this device.
+
+ The following steps must be done synchronously.
+ Only one map can be loaded at a time per device.
+ """
+ cur_map = self.current_map
+ # This won't be None at this point as the coordinator will have run first.
+ if cur_map is None:
+ # If we don't have a cur map(shouldn't happen) just
+ # return as we can't do anything.
+ return
+ map_flags = sorted(self.maps, key=lambda data: data == cur_map, reverse=True)
+ for map_flag in map_flags:
+ if map_flag != cur_map:
+ # Only change the map and sleep if we have multiple maps.
+ await self.api.load_multi_map(map_flag)
+ self.current_map = map_flag
+ # We cannot get the map until the roborock servers fully process the
+ # map change.
+ await asyncio.sleep(MAP_SLEEP)
+ tasks = [self.set_current_map_rooms()]
+ # The image is set within async_setup, so if it exists, we have it here.
+ if self.maps[map_flag].image is None:
+ # If we don't have a cached map, let's update it here so that it can be
+ # cached in the future.
+ tasks.append(self.update_map())
+ # If either of these fail, we don't care, and we want to continue.
+ await asyncio.gather(*tasks, return_exceptions=True)
+
+ if len(self.maps) != 1:
+ # Set the map back to the map the user previously had selected so that it
+ # does not change the end user's app.
+ # Only needs to happen when we changed maps above.
+ await self.api.load_multi_map(cur_map)
+ self.current_map = cur_map
+
class RoborockDataUpdateCoordinatorA01(
DataUpdateCoordinator[
@@ -186,15 +439,24 @@ class RoborockDataUpdateCoordinatorA01(
):
"""Class to manage fetching data from the API for A01 devices."""
+ config_entry: RoborockConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: RoborockConfigEntry,
device: HomeDataDevice,
product_info: HomeDataProduct,
api: RoborockClientA01,
) -> None:
"""Initialize."""
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=A01_UPDATE_INTERVAL,
+ )
self.api = api
self.device_info = DeviceInfo(
name=device.name,
diff --git a/homeassistant/components/roborock/diagnostics.py b/homeassistant/components/roborock/diagnostics.py
index e784e4ce837..4602b4bd02a 100644
--- a/homeassistant/components/roborock/diagnostics.py
+++ b/homeassistant/components/roborock/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_UNIQUE_ID
from homeassistant.core import HomeAssistant
-from . import RoborockConfigEntry
+from .coordinator import RoborockConfigEntry
TO_REDACT_CONFIG = ["token", "sn", "rruid", CONF_UNIQUE_ID, "username", "uid"]
diff --git a/homeassistant/components/roborock/entity.py b/homeassistant/components/roborock/entity.py
index 4a16ada5967..404f239c93a 100644
--- a/homeassistant/components/roborock/entity.py
+++ b/homeassistant/components/roborock/entity.py
@@ -8,7 +8,11 @@ from roborock.containers import Consumable, Status
from roborock.exceptions import RoborockException
from roborock.roborock_message import RoborockDataProtocol
from roborock.roborock_typing import RoborockCommand
-from roborock.version_1_apis.roborock_client_v1 import AttributeCache, RoborockClientV1
+from roborock.version_1_apis.roborock_client_v1 import (
+ CLOUD_REQUIRED,
+ AttributeCache,
+ RoborockClientV1,
+)
from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1
from roborock.version_a01_apis import RoborockClientA01
@@ -53,14 +57,16 @@ class RoborockEntityV1(RoborockEntity):
"""Get an item from the api cache."""
return self._api.cache[attribute]
- async def send(
- self,
+ @classmethod
+ async def _send_command(
+ cls,
command: RoborockCommand | str,
+ api: RoborockClientV1,
params: dict[str, Any] | list[Any] | int | None = None,
) -> dict:
- """Send a command to a vacuum cleaner."""
+ """Send a Roborock command with params to a given api."""
try:
- response: dict = await self._api.send_command(command, params)
+ response: dict = await api.send_command(command, params)
except RoborockException as err:
if isinstance(command, RoborockCommand):
command_name = command.name
@@ -75,6 +81,14 @@ class RoborockEntityV1(RoborockEntity):
) from err
return response
+ async def send(
+ self,
+ command: RoborockCommand | str,
+ params: dict[str, Any] | list[Any] | int | None = None,
+ ) -> dict:
+ """Send a command to a vacuum cleaner."""
+ return await self._send_command(command, self._api, params)
+
@property
def api(self) -> RoborockClientV1:
"""Returns the api."""
@@ -107,12 +121,15 @@ class RoborockCoordinatedEntityV1(
listener_request: list[RoborockDataProtocol]
| RoborockDataProtocol
| None = None,
+ is_dock_entity: bool = False,
) -> None:
"""Initialize the coordinated Roborock Device."""
RoborockEntityV1.__init__(
self,
unique_id=unique_id,
- device_info=coordinator.device_info,
+ device_info=coordinator.device_info
+ if not is_dock_entity
+ else coordinator.dock_device_info,
api=coordinator.api,
)
CoordinatorEntity.__init__(self, coordinator=coordinator)
@@ -152,7 +169,10 @@ class RoborockCoordinatedEntityV1(
params: dict[str, Any] | list[Any] | int | None = None,
) -> dict:
"""Overloads normal send command but refreshes coordinator."""
- res = await super().send(command, params)
+ if command in CLOUD_REQUIRED:
+ res = await self._send_command(command, self.coordinator.cloud_api, params)
+ else:
+ res = await self._send_command(command, self._api, params)
await self.coordinator.async_refresh()
return res
diff --git a/homeassistant/components/roborock/image.py b/homeassistant/components/roborock/image.py
index b4776c27164..d1c19331ba4 100644
--- a/homeassistant/components/roborock/image.py
+++ b/homeassistant/components/roborock/image.py
@@ -1,64 +1,29 @@
"""Support for Roborock image."""
-import asyncio
-from collections.abc import Callable
from datetime import datetime
-import io
-
-from roborock import RoborockCommand
-from vacuum_map_parser_base.config.color import ColorsPalette
-from vacuum_map_parser_base.config.image_config import ImageConfig
-from vacuum_map_parser_base.config.size import Sizes
-from vacuum_map_parser_roborock.map_data_parser import RoborockMapDataParser
+import logging
from homeassistant.components.image import ImageEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.util import dt as dt_util
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RoborockConfigEntry
-from .const import (
- DEFAULT_DRAWABLES,
- DOMAIN,
- DRAWABLES,
- IMAGE_CACHE_INTERVAL,
- MAP_FILE_FORMAT,
- MAP_SLEEP,
-)
-from .coordinator import RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
from .entity import RoborockCoordinatedEntityV1
+_LOGGER = logging.getLogger(__name__)
+
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roborock image platform."""
- drawables = [
- drawable
- for drawable, default_value in DEFAULT_DRAWABLES.items()
- if config_entry.options.get(DRAWABLES, {}).get(drawable, default_value)
- ]
- parser = RoborockMapDataParser(
- ColorsPalette(), Sizes(), drawables, ImageConfig(), []
- )
-
- def parse_image(map_bytes: bytes) -> bytes | None:
- parsed_map = parser.parse(map_bytes)
- if parsed_map.image is None:
- return None
- img_byte_arr = io.BytesIO()
- parsed_map.image.data.save(img_byte_arr, format=MAP_FILE_FORMAT)
- return img_byte_arr.getvalue()
-
- await asyncio.gather(
- *(refresh_coordinators(hass, coord) for coord in config_entry.runtime_data.v1)
- )
async_add_entities(
(
RoborockMap(
@@ -67,7 +32,6 @@ async def async_setup_entry(
coord,
map_info.flag,
map_info.name,
- parse_image,
)
for coord in config_entry.runtime_data.v1
for map_info in coord.maps.values()
@@ -89,14 +53,12 @@ class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity):
coordinator: RoborockDataUpdateCoordinator,
map_flag: int,
map_name: str,
- parser: Callable[[bytes], bytes | None],
) -> None:
"""Initialize a Roborock map."""
RoborockCoordinatedEntityV1.__init__(self, unique_id, coordinator)
ImageEntity.__init__(self, coordinator.hass)
self.config_entry = config_entry
self._attr_name = map_name
- self.parser = parser
self.map_flag = map_flag
self.cached_map = b""
self._attr_entity_category = EntityCategory.DIAGNOSTIC
@@ -106,89 +68,22 @@ class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity):
"""Return if this map is the currently selected map."""
return self.map_flag == self.coordinator.current_map
- def is_map_valid(self) -> bool:
- """Update the map if it is valid.
-
- Update this map if it is the currently active map, and the
- vacuum is cleaning, or if it has never been set at all.
- """
- return self.cached_map == b"" or (
- self.is_selected
- and self.image_last_updated is not None
- and self.coordinator.roborock_device_info.props.status is not None
- and bool(self.coordinator.roborock_device_info.props.status.in_cleaning)
- )
-
async def async_added_to_hass(self) -> None:
"""When entity is added to hass load any previously cached maps from disk."""
await super().async_added_to_hass()
- content = await self.coordinator.map_storage.async_load_map(self.map_flag)
- self.cached_map = content or b""
- self._attr_image_last_updated = dt_util.utcnow()
+ self._attr_image_last_updated = self.coordinator.maps[
+ self.map_flag
+ ].last_updated
self.async_write_ha_state()
def _handle_coordinator_update(self) -> None:
- # Bump last updated every third time the coordinator runs, so that async_image
- # will be called and we will evaluate on the new coordinator data if we should
- # update the cache.
- if (
- dt_util.utcnow() - self.image_last_updated
- ).total_seconds() > IMAGE_CACHE_INTERVAL and self.is_map_valid():
- self._attr_image_last_updated = dt_util.utcnow()
+ # If the coordinator has updated the map, we can update the image.
+ self._attr_image_last_updated = self.coordinator.maps[
+ self.map_flag
+ ].last_updated
+
super()._handle_coordinator_update()
async def async_image(self) -> bytes | None:
- """Update the image if it is not cached."""
- if self.is_map_valid():
- response = await asyncio.gather(
- *(
- self.cloud_api.get_map_v1(),
- self.coordinator.set_current_map_rooms(),
- ),
- return_exceptions=True,
- )
- if (
- not isinstance(response[0], bytes)
- or (content := self.parser(response[0])) is None
- ):
- raise HomeAssistantError(
- translation_domain=DOMAIN,
- translation_key="map_failure",
- )
- if self.cached_map != content:
- self.cached_map = content
- await self.coordinator.map_storage.async_save_map(
- self.map_flag,
- content,
- )
- return self.cached_map
-
-
-async def refresh_coordinators(
- hass: HomeAssistant, coord: RoborockDataUpdateCoordinator
-) -> None:
- """Get the starting map information for all maps for this device.
-
- The following steps must be done synchronously.
- Only one map can be loaded at a time per device.
- """
- cur_map = coord.current_map
- # This won't be None at this point as the coordinator will have run first.
- assert cur_map is not None
- map_flags = sorted(coord.maps, key=lambda data: data == cur_map, reverse=True)
- for map_flag in map_flags:
- if map_flag != cur_map:
- # Only change the map and sleep if we have multiple maps.
- await coord.api.send_command(RoborockCommand.LOAD_MULTI_MAP, [map_flag])
- coord.current_map = map_flag
- # We cannot get the map until the roborock servers fully process the
- # map change.
- await asyncio.sleep(MAP_SLEEP)
- await coord.set_current_map_rooms()
-
- if len(coord.maps) != 1:
- # Set the map back to the map the user previously had selected so that it
- # does not change the end user's app.
- # Only needs to happen when we changed maps above.
- await coord.cloud_api.send_command(RoborockCommand.LOAD_MULTI_MAP, [cur_map])
- coord.current_map = cur_map
+ """Get the cached image."""
+ return self.coordinator.maps[self.map_flag].image
diff --git a/homeassistant/components/roborock/manifest.json b/homeassistant/components/roborock/manifest.json
index db2654d4baa..60036edb0bc 100644
--- a/homeassistant/components/roborock/manifest.json
+++ b/homeassistant/components/roborock/manifest.json
@@ -3,11 +3,22 @@
"name": "Roborock",
"codeowners": ["@Lash-L", "@allenporter"],
"config_flow": true,
+ "dhcp": [
+ {
+ "macaddress": "249E7D*"
+ },
+ {
+ "macaddress": "B04A39*"
+ },
+ {
+ "hostname": "roborock-*"
+ }
+ ],
"documentation": "https://www.home-assistant.io/integrations/roborock",
"iot_class": "local_polling",
"loggers": ["roborock"],
"requirements": [
- "python-roborock==2.11.1",
+ "python-roborock==2.16.1",
"vacuum-map-parser-roborock==0.1.2"
]
}
diff --git a/homeassistant/components/roborock/models.py b/homeassistant/components/roborock/models.py
index 4b8ab43b4a1..ab40f23d574 100644
--- a/homeassistant/components/roborock/models.py
+++ b/homeassistant/components/roborock/models.py
@@ -1,10 +1,12 @@
"""Roborock Models."""
from dataclasses import dataclass
+from datetime import datetime
from typing import Any
from roborock.containers import HomeDataDevice, HomeDataProduct, NetworkInfo
from roborock.roborock_typing import DeviceProp
+from vacuum_map_parser_base.map_data import MapData
@dataclass
@@ -48,3 +50,13 @@ class RoborockMapInfo:
flag: int
name: str
rooms: dict[int, str]
+ image: bytes | None
+ last_updated: datetime
+ map_data: MapData | None
+
+ @property
+ def current_room(self) -> str | None:
+ """Get the currently active room for this map if any."""
+ if self.map_data is None or self.map_data.vacuum_room is None:
+ return None
+ return self.rooms.get(self.map_data.vacuum_room)
diff --git a/homeassistant/components/roborock/number.py b/homeassistant/components/roborock/number.py
index 7f568ae824b..73ac14fca71 100644
--- a/homeassistant/components/roborock/number.py
+++ b/homeassistant/components/roborock/number.py
@@ -14,14 +14,16 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescriptio
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import DOMAIN, RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator
+from .const import DOMAIN
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
from .entity import RoborockEntityV1
_LOGGER = logging.getLogger(__name__)
+PARALLEL_UPDATES = 0
+
@dataclass(frozen=True, kw_only=True)
class RoborockNumberDescription(NumberEntityDescription):
@@ -50,7 +52,7 @@ NUMBER_DESCRIPTIONS: list[RoborockNumberDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roborock number platform."""
possible_entities: list[
diff --git a/homeassistant/components/roborock/quality_scale.yaml b/homeassistant/components/roborock/quality_scale.yaml
new file mode 100644
index 00000000000..d064c30ccf6
--- /dev/null
+++ b/homeassistant/components/roborock/quality_scale.yaml
@@ -0,0 +1,75 @@
+rules:
+ # Bronze
+ action-setup: done
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow: done
+ config-flow-test-coverage: done
+ dependency-transparency: done
+ docs-actions: done
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup: done
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure:
+ status: done
+ comment: The config flow verifies credentials and the cloud APIs.
+ test-before-setup: done
+ unique-config-entry: done
+ # Silver
+ action-exceptions: todo
+ config-entry-unloading: done
+ docs-configuration-parameters: done
+ docs-installation-parameters: done
+ entity-unavailable: done
+ integration-owner: done
+ log-when-unavailable: done
+ parallel-updates: done
+ reauthentication-flow: todo
+ test-coverage: done
+ # Gold
+ devices: done
+ diagnostics: done
+ discovery: done
+ discovery-update-info:
+ status: exempt
+ comment: Devices do not support discovery.
+ docs-data-update: done
+ docs-examples: done
+ docs-known-limitations:
+ status: todo
+ comment: Documentation does not describe known limitations like rate limiting
+ docs-supported-devices: todo
+ docs-supported-functions: done
+ docs-troubleshooting:
+ status: todo
+ comment: |
+ There are good troubleshooting steps, however we should update the "cloud vs local"
+ and rate limiting documentation with more information.
+ docs-use-cases:
+ status: todo
+ comment: |
+ The docs describe controlling the vacuum, though does not describe more
+ interesting potential integrations with the homoe assistant ecosystem.
+ dynamic-devices: todo
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default:
+ status: exempt
+ comment: There are no noisy entities.
+ entity-translations: done
+ exception-translations: done
+ icon-translations: todo
+ reconfiguration-flow: todo
+ repair-issues:
+ status: todo
+ comment: The Cloud vs Local API warning should probably be a repair issue.
+ stale-devices: done
+ # Platinum
+ async-dependency: todo
+ inject-websession: done
+ strict-typing: done
diff --git a/homeassistant/components/roborock/select.py b/homeassistant/components/roborock/select.py
index 73cb95d2d7c..208020dccab 100644
--- a/homeassistant/components/roborock/select.py
+++ b/homeassistant/components/roborock/select.py
@@ -4,20 +4,21 @@ import asyncio
from collections.abc import Callable
from dataclasses import dataclass
-from roborock.containers import Status
+from roborock.code_mappings import RoborockDockDustCollectionModeCode
from roborock.roborock_message import RoborockDataProtocol
-from roborock.roborock_typing import RoborockCommand
+from roborock.roborock_typing import DeviceProp, RoborockCommand
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RoborockConfigEntry
from .const import MAP_SLEEP
-from .coordinator import RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
from .entity import RoborockCoordinatedEntityV1
+PARALLEL_UPDATES = 0
+
@dataclass(frozen=True, kw_only=True)
class RoborockSelectDescription(SelectEntityDescription):
@@ -26,13 +27,15 @@ class RoborockSelectDescription(SelectEntityDescription):
# The command that the select entity will send to the api.
api_command: RoborockCommand
# Gets the current value of the select entity.
- value_fn: Callable[[Status], str | None]
+ value_fn: Callable[[DeviceProp], str | None]
# Gets all options of the select entity.
- options_lambda: Callable[[Status], list[str] | None]
+ options_lambda: Callable[[DeviceProp], list[str] | None]
# Takes the value from the select entity and converts it for the api.
- parameter_lambda: Callable[[str, Status], list[int]]
+ parameter_lambda: Callable[[str, DeviceProp], list[int]]
protocol_listener: RoborockDataProtocol | None = None
+ # If it is a dock entity
+ is_dock_entity: bool = False
SELECT_DESCRIPTIONS: list[RoborockSelectDescription] = [
@@ -40,24 +43,38 @@ SELECT_DESCRIPTIONS: list[RoborockSelectDescription] = [
key="water_box_mode",
translation_key="mop_intensity",
api_command=RoborockCommand.SET_WATER_BOX_CUSTOM_MODE,
- value_fn=lambda data: data.water_box_mode_name,
+ value_fn=lambda data: data.status.water_box_mode_name,
entity_category=EntityCategory.CONFIG,
- options_lambda=lambda data: data.water_box_mode.keys()
- if data.water_box_mode is not None
+ options_lambda=lambda data: data.status.water_box_mode.keys()
+ if data.status.water_box_mode is not None
else None,
- parameter_lambda=lambda key, status: [status.get_mop_intensity_code(key)],
+ parameter_lambda=lambda key, prop: [prop.status.get_mop_intensity_code(key)],
protocol_listener=RoborockDataProtocol.WATER_BOX_MODE,
),
RoborockSelectDescription(
key="mop_mode",
translation_key="mop_mode",
api_command=RoborockCommand.SET_MOP_MODE,
- value_fn=lambda data: data.mop_mode_name,
+ value_fn=lambda data: data.status.mop_mode_name,
entity_category=EntityCategory.CONFIG,
- options_lambda=lambda data: data.mop_mode.keys()
- if data.mop_mode is not None
+ options_lambda=lambda data: data.status.mop_mode.keys()
+ if data.status.mop_mode is not None
else None,
- parameter_lambda=lambda key, status: [status.get_mop_mode_code(key)],
+ parameter_lambda=lambda key, prop: [prop.status.get_mop_mode_code(key)],
+ ),
+ RoborockSelectDescription(
+ key="dust_collection_mode",
+ translation_key="dust_collection_mode",
+ api_command=RoborockCommand.SET_DUST_COLLECTION_MODE,
+ value_fn=lambda data: data.dust_collection_mode_name,
+ entity_category=EntityCategory.CONFIG,
+ options_lambda=lambda data: RoborockDockDustCollectionModeCode.keys()
+ if data.dust_collection_mode_name is not None
+ else None,
+ parameter_lambda=lambda key, _: [
+ RoborockDockDustCollectionModeCode.as_dict().get(key)
+ ],
+ is_dock_entity=True,
),
]
@@ -65,7 +82,7 @@ SELECT_DESCRIPTIONS: list[RoborockSelectDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roborock select platform."""
@@ -75,7 +92,7 @@ async def async_setup_entry(
for description in SELECT_DESCRIPTIONS
if (
options := description.options_lambda(
- coordinator.roborock_device_info.props.status
+ coordinator.roborock_device_info.props
)
)
is not None
@@ -105,6 +122,7 @@ class RoborockSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
f"{entity_description.key}_{coordinator.duid_slug}",
coordinator,
entity_description.protocol_listener,
+ is_dock_entity=entity_description.is_dock_entity,
)
self._attr_options = options
@@ -112,27 +130,28 @@ class RoborockSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
"""Set the option."""
await self.send(
self.entity_description.api_command,
- self.entity_description.parameter_lambda(option, self._device_status),
+ self.entity_description.parameter_lambda(option, self.coordinator.data),
)
@property
def current_option(self) -> str | None:
- """Get the current status of the select entity from device_status."""
- return self.entity_description.value_fn(self._device_status)
+ """Get the current status of the select entity from device props."""
+ return self.entity_description.value_fn(self.coordinator.data)
class RoborockCurrentMapSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
"""A class to let you set the selected map on Roborock vacuum."""
- _attr_entity_category = EntityCategory.DIAGNOSTIC
+ _attr_entity_category = EntityCategory.CONFIG
_attr_translation_key = "selected_map"
async def async_select_option(self, option: str) -> None:
"""Set the option."""
for map_id, map_ in self.coordinator.maps.items():
if map_.name == option:
- await self.send(
+ await self._send_command(
RoborockCommand.LOAD_MULTI_MAP,
+ self.api,
[map_id],
)
# Update the current map id manually so that nothing gets broken
@@ -141,6 +160,7 @@ class RoborockCurrentMapSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
# We need to wait after updating the map
# so that other commands will be executed correctly.
await asyncio.sleep(MAP_SLEEP)
+ await self.coordinator.async_refresh()
break
@property
diff --git a/homeassistant/components/roborock/sensor.py b/homeassistant/components/roborock/sensor.py
index e01a03d7720..33ecaf74d4f 100644
--- a/homeassistant/components/roborock/sensor.py
+++ b/homeassistant/components/roborock/sensor.py
@@ -28,12 +28,21 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfArea, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01
-from .entity import RoborockCoordinatedEntityA01, RoborockCoordinatedEntityV1
+from .coordinator import (
+ RoborockConfigEntry,
+ RoborockDataUpdateCoordinator,
+ RoborockDataUpdateCoordinatorA01,
+)
+from .entity import (
+ RoborockCoordinatedEntityA01,
+ RoborockCoordinatedEntityV1,
+ RoborockEntity,
+)
+
+PARALLEL_UPDATES = 0
@dataclass(frozen=True, kw_only=True)
@@ -44,6 +53,9 @@ class RoborockSensorDescription(SensorEntityDescription):
protocol_listener: RoborockDataProtocol | None = None
+ # If it is a dock entity
+ is_dock_entity: bool = False
+
@dataclass(frozen=True, kw_only=True)
class RoborockSensorDescriptionA01(SensorEntityDescription):
@@ -194,6 +206,7 @@ SENSOR_DESCRIPTIONS = [
entity_category=EntityCategory.DIAGNOSTIC,
device_class=SensorDeviceClass.ENUM,
options=RoborockDockErrorCode.keys(),
+ is_dock_entity=True,
),
RoborockSensorDescription(
key="mop_clean_remaining",
@@ -202,6 +215,7 @@ SENSOR_DESCRIPTIONS = [
value_fn=lambda data: data.status.rdt,
translation_key="mop_drying_remaining_time",
entity_category=EntityCategory.DIAGNOSTIC,
+ is_dock_entity=True,
),
]
@@ -292,11 +306,11 @@ A01_SENSOR_DESCRIPTIONS: list[RoborockSensorDescriptionA01] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Roborock vacuum sensors."""
coordinators = config_entry.runtime_data
- async_add_entities(
+ entities: list[RoborockEntity] = [
RoborockSensorEntity(
coordinator,
description,
@@ -304,8 +318,9 @@ async def async_setup_entry(
for coordinator in coordinators.v1
for description in SENSOR_DESCRIPTIONS
if description.value_fn(coordinator.roborock_device_info.props) is not None
- )
- async_add_entities(
+ ]
+ entities.extend(RoborockCurrentRoom(coordinator) for coordinator in coordinators.v1)
+ entities.extend(
RoborockSensorEntityA01(
coordinator,
description,
@@ -314,6 +329,7 @@ async def async_setup_entry(
for description in A01_SENSOR_DESCRIPTIONS
if description.data_protocol in coordinator.data
)
+ async_add_entities(entities)
class RoborockSensorEntity(RoborockCoordinatedEntityV1, SensorEntity):
@@ -332,6 +348,7 @@ class RoborockSensorEntity(RoborockCoordinatedEntityV1, SensorEntity):
f"{description.key}_{coordinator.duid_slug}",
coordinator,
description.protocol_listener,
+ is_dock_entity=description.is_dock_entity,
)
@property
@@ -342,6 +359,42 @@ class RoborockSensorEntity(RoborockCoordinatedEntityV1, SensorEntity):
)
+class RoborockCurrentRoom(RoborockCoordinatedEntityV1, SensorEntity):
+ """Representation of a Current Room Sensor."""
+
+ _attr_device_class = SensorDeviceClass.ENUM
+ _attr_translation_key = "current_room"
+ _attr_entity_category = EntityCategory.DIAGNOSTIC
+
+ def __init__(
+ self,
+ coordinator: RoborockDataUpdateCoordinator,
+ ) -> None:
+ """Initialize the entity."""
+ super().__init__(
+ f"current_room_{coordinator.duid_slug}",
+ coordinator,
+ None,
+ is_dock_entity=False,
+ )
+
+ @property
+ def options(self) -> list[str]:
+ """Return the currently valid rooms."""
+ if self.coordinator.current_map is not None:
+ return list(
+ self.coordinator.maps[self.coordinator.current_map].rooms.values()
+ )
+ return []
+
+ @property
+ def native_value(self) -> str | None:
+ """Return the value reported by the sensor."""
+ if self.coordinator.current_map is not None:
+ return self.coordinator.maps[self.coordinator.current_map].current_room
+ return None
+
+
class RoborockSensorEntityA01(RoborockCoordinatedEntityA01, SensorEntity):
"""Representation of a A01 Roborock sensor."""
diff --git a/homeassistant/components/roborock/strings.json b/homeassistant/components/roborock/strings.json
index 8968ac020a2..caad67e4ce6 100644
--- a/homeassistant/components/roborock/strings.json
+++ b/homeassistant/components/roborock/strings.json
@@ -5,12 +5,18 @@
"description": "Enter your Roborock email address.",
"data": {
"username": "[%key:common::config_flow::data::email%]"
+ },
+ "data_description": {
+ "username": "The email address used to sign in to the Roborock app."
}
},
"code": {
"description": "Type the verification code sent to your email",
"data": {
"code": "Verification code"
+ },
+ "data_description": {
+ "code": "The verification code sent to your email."
}
},
"reauth_confirm": {
@@ -54,6 +60,25 @@
"vacuum_position": "Vacuum position",
"virtual_walls": "Virtual walls",
"zones": "Zones"
+ },
+ "data_description": {
+ "charger": "Show the charger on the map.",
+ "cleaned_area": "Show the area cleaned on the map.",
+ "goto_path": "Show the go-to path on the map.",
+ "ignored_obstacles": "Show ignored obstacles on the map.",
+ "ignored_obstacles_with_photo": "Show ignored obstacles with photos on the map.",
+ "mop_path": "Show the mop path on the map.",
+ "no_carpet_zones": "Show the no carpet zones on the map.",
+ "no_go_zones": "Show the no-go zones on the map.",
+ "no_mopping_zones": "Show the no-mop zones on the map.",
+ "obstacles": "Show obstacles on the map.",
+ "obstacles_with_photo": "Show obstacles with photos on the map.",
+ "path": "Show the path on the map.",
+ "predicted_path": "Show the predicted path on the map.",
+ "room_names": "Show room names on the map.",
+ "vacuum_position": "Show the vacuum position on the map.",
+ "virtual_walls": "Show virtual walls on the map.",
+ "zones": "Show zones on the map."
}
}
}
@@ -128,7 +153,7 @@
"updating": "[%key:component::roborock::entity::sensor::status::state::updating%]",
"washing": "Washing",
"ready": "Ready",
- "charging": "[%key:component::roborock::entity::sensor::status::state::charging%]",
+ "charging": "[%key:common::state::charging%]",
"mop_washing": "Washing mop",
"self_clean_cleaning": "Self clean cleaning",
"self_clean_deep_cleaning": "Self clean deep cleaning",
@@ -156,6 +181,9 @@
"countdown": {
"name": "Countdown"
},
+ "current_room": {
+ "name": "Current room"
+ },
"dock_error": {
"name": "Dock error",
"state": {
@@ -199,7 +227,7 @@
"cleaning": "Cleaning",
"returning_home": "Returning home",
"manual_mode": "Manual mode",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"charging_problem": "Charging problem",
"paused": "[%key:common::state::paused%]",
"spot_cleaning": "Spot cleaning",
@@ -353,6 +381,15 @@
},
"selected_map": {
"name": "Selected map"
+ },
+ "dust_collection_mode": {
+ "name": "Empty mode",
+ "state": {
+ "smart": "Smart",
+ "light": "Light",
+ "balanced": "[%key:component::roborock::entity::vacuum::roborock::state_attributes::fan_speed::state::balanced%]",
+ "max": "[%key:component::roborock::entity::select::mop_intensity::state::max%]"
+ }
}
},
"switch": {
@@ -420,6 +457,12 @@
"map_failure": {
"message": "Something went wrong creating the map"
},
+ "position_not_found": {
+ "message": "Robot position not found"
+ },
+ "update_data_fail": {
+ "message": "Failed to update data"
+ },
"no_coordinators": {
"message": "No devices were able to successfully setup"
},
diff --git a/homeassistant/components/roborock/switch.py b/homeassistant/components/roborock/switch.py
index b0c8c880188..44feccdebac 100644
--- a/homeassistant/components/roborock/switch.py
+++ b/homeassistant/components/roborock/switch.py
@@ -16,14 +16,16 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import DOMAIN, RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator
+from .const import DOMAIN
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
from .entity import RoborockEntityV1
_LOGGER = logging.getLogger(__name__)
+PARALLEL_UPDATES = 0
+
@dataclass(frozen=True, kw_only=True)
class RoborockSwitchDescription(SwitchEntityDescription):
@@ -35,6 +37,8 @@ class RoborockSwitchDescription(SwitchEntityDescription):
update_value: Callable[[AttributeCache, bool], Coroutine[Any, Any, None]]
# Attribute from cache
attribute: str
+ # If it is a dock entity
+ is_dock_entity: bool = False
SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription] = [
@@ -47,6 +51,7 @@ SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription] = [
key="child_lock",
translation_key="child_lock",
entity_category=EntityCategory.CONFIG,
+ is_dock_entity=True,
),
RoborockSwitchDescription(
cache_key=CacheableAttribute.flow_led_status,
@@ -57,6 +62,7 @@ SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription] = [
key="status_indicator",
translation_key="status_indicator",
entity_category=EntityCategory.CONFIG,
+ is_dock_entity=True,
),
RoborockSwitchDescription(
cache_key=CacheableAttribute.dnd_timer,
@@ -99,7 +105,7 @@ SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roborock switch platform."""
possible_entities: list[
@@ -147,7 +153,13 @@ class RoborockSwitch(RoborockEntityV1, SwitchEntity):
) -> None:
"""Initialize the entity."""
self.entity_description = entity_description
- super().__init__(unique_id, coordinator.device_info, coordinator.api)
+ super().__init__(
+ unique_id,
+ coordinator.device_info
+ if not entity_description.is_dock_entity
+ else coordinator.dock_device_info,
+ coordinator.api,
+ )
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn off the switch."""
diff --git a/homeassistant/components/roborock/time.py b/homeassistant/components/roborock/time.py
index 1dd681dff1f..83d341fa2dd 100644
--- a/homeassistant/components/roborock/time.py
+++ b/homeassistant/components/roborock/time.py
@@ -16,14 +16,16 @@ from homeassistant.components.time import TimeEntity, TimeEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import DOMAIN, RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator
+from .const import DOMAIN
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
from .entity import RoborockEntityV1
_LOGGER = logging.getLogger(__name__)
+PARALLEL_UPDATES = 0
+
@dataclass(frozen=True, kw_only=True)
class RoborockTimeDescription(TimeEntityDescription):
@@ -114,7 +116,7 @@ TIME_DESCRIPTIONS: list[RoborockTimeDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roborock time platform."""
possible_entities: list[
diff --git a/homeassistant/components/roborock/vacuum.py b/homeassistant/components/roborock/vacuum.py
index 7582dadad16..058fffbdb1c 100644
--- a/homeassistant/components/roborock/vacuum.py
+++ b/homeassistant/components/roborock/vacuum.py
@@ -1,11 +1,14 @@
"""Support for Roborock vacuum class."""
-from dataclasses import asdict
from typing import Any
from roborock.code_mappings import RoborockStateCode
from roborock.roborock_message import RoborockDataProtocol
from roborock.roborock_typing import RoborockCommand
+from vacuum_map_parser_base.config.color import ColorsPalette
+from vacuum_map_parser_base.config.image_config import ImageConfig
+from vacuum_map_parser_base.config.size import Sizes
+from vacuum_map_parser_roborock.map_data_parser import RoborockMapDataParser
import voluptuous as vol
from homeassistant.components.vacuum import (
@@ -16,18 +19,16 @@ from homeassistant.components.vacuum import (
from homeassistant.core import HomeAssistant, ServiceResponse, SupportsResponse
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RoborockConfigEntry
from .const import (
DOMAIN,
GET_MAPS_SERVICE_NAME,
GET_VACUUM_CURRENT_POSITION_SERVICE_NAME,
SET_VACUUM_GOTO_POSITION_SERVICE_NAME,
)
-from .coordinator import RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry, RoborockDataUpdateCoordinator
from .entity import RoborockCoordinatedEntityV1
-from .image import ColorsPalette, ImageConfig, RoborockMapDataParser, Sizes
STATE_CODE_TO_STATE = {
RoborockStateCode.starting: VacuumActivity.IDLE, # "Starting"
@@ -55,11 +56,13 @@ STATE_CODE_TO_STATE = {
RoborockStateCode.device_offline: VacuumActivity.ERROR, # "Device offline"
}
+PARALLEL_UPDATES = 0
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: RoborockConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Roborock sensor."""
async_add_entities(
@@ -202,7 +205,14 @@ class RoborockVacuum(RoborockCoordinatedEntityV1, StateVacuumEntity):
"""Get map information such as map id and room ids."""
return {
"maps": [
- asdict(vacuum_map) for vacuum_map in self.coordinator.maps.values()
+ {
+ "flag": vacuum_map.flag,
+ "name": vacuum_map.name,
+ # JsonValueType does not accept a int as a key - was not a
+ # issue with previous asdict() implementation.
+ "rooms": vacuum_map.rooms, # type: ignore[dict-item]
+ }
+ for vacuum_map in self.coordinator.maps.values()
]
}
@@ -211,13 +221,18 @@ class RoborockVacuum(RoborockCoordinatedEntityV1, StateVacuumEntity):
map_data = await self.coordinator.cloud_api.get_map_v1()
if not isinstance(map_data, bytes):
- raise HomeAssistantError("Failed to retrieve map data.")
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="map_failure",
+ )
parser = RoborockMapDataParser(ColorsPalette(), Sizes(), [], ImageConfig(), [])
parsed_map = parser.parse(map_data)
robot_position = parsed_map.vacuum_position
if robot_position is None:
- raise HomeAssistantError("Robot position not found")
+ raise HomeAssistantError(
+ translation_domain=DOMAIN, translation_key="position_not_found"
+ )
return {
"x": robot_position.x,
diff --git a/homeassistant/components/roku/__init__.py b/homeassistant/components/roku/__init__.py
index e6b92d91335..be0b20c97fb 100644
--- a/homeassistant/components/roku/__init__.py
+++ b/homeassistant/components/roku/__init__.py
@@ -2,12 +2,10 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_HOST, Platform
+from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .const import CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID
-from .coordinator import RokuDataUpdateCoordinator
+from .coordinator import RokuConfigEntry, RokuDataUpdateCoordinator
PLATFORMS = [
Platform.BINARY_SENSOR,
@@ -17,22 +15,10 @@ PLATFORMS = [
Platform.SENSOR,
]
-type RokuConfigEntry = ConfigEntry[RokuDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry) -> bool:
"""Set up Roku from a config entry."""
- if (device_id := entry.unique_id) is None:
- device_id = entry.entry_id
-
- coordinator = RokuDataUpdateCoordinator(
- hass,
- host=entry.data[CONF_HOST],
- device_id=device_id,
- play_media_app_id=entry.options.get(
- CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID
- ),
- )
+ coordinator = RokuDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/roku/binary_sensor.py b/homeassistant/components/roku/binary_sensor.py
index 2e7fd12788c..31250898055 100644
--- a/homeassistant/components/roku/binary_sensor.py
+++ b/homeassistant/components/roku/binary_sensor.py
@@ -13,9 +13,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RokuConfigEntry
+from .coordinator import RokuConfigEntry
from .entity import RokuEntity
# Coordinator is used to centralize the data updates
@@ -59,7 +59,7 @@ BINARY_SENSORS: tuple[RokuBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RokuConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Roku binary sensors based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/roku/config_flow.py b/homeassistant/components/roku/config_flow.py
index 2fb016b5467..47bc86802d2 100644
--- a/homeassistant/components/roku/config_flow.py
+++ b/homeassistant/components/roku/config_flow.py
@@ -25,8 +25,8 @@ from homeassistant.helpers.service_info.ssdp import (
)
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
-from . import RokuConfigEntry
from .const import CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID, DOMAIN
+from .coordinator import RokuConfigEntry
DATA_SCHEMA = vol.Schema({vol.Required(CONF_HOST): str})
diff --git a/homeassistant/components/roku/coordinator.py b/homeassistant/components/roku/coordinator.py
index 7900669d02f..e3c20d8351f 100644
--- a/homeassistant/components/roku/coordinator.py
+++ b/homeassistant/components/roku/coordinator.py
@@ -8,33 +8,44 @@ import logging
from rokuecp import Roku, RokuError
from rokuecp.models import Device
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util.dt import utcnow
-from .const import DOMAIN
+from .const import CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID, DOMAIN
REQUEST_REFRESH_DELAY = 0.35
SCAN_INTERVAL = timedelta(seconds=10)
_LOGGER = logging.getLogger(__name__)
+type RokuConfigEntry = ConfigEntry[RokuDataUpdateCoordinator]
+
class RokuDataUpdateCoordinator(DataUpdateCoordinator[Device]):
"""Class to manage fetching Roku data."""
+ config_entry: RokuConfigEntry
last_full_update: datetime | None
roku: Roku
def __init__(
- self, hass: HomeAssistant, *, host: str, device_id: str, play_media_app_id: str
+ self,
+ hass: HomeAssistant,
+ config_entry: RokuConfigEntry,
) -> None:
"""Initialize global Roku data updater."""
- self.device_id = device_id
- self.roku = Roku(host=host, session=async_get_clientsession(hass))
- self.play_media_app_id = play_media_app_id
+ self.device_id = config_entry.unique_id or config_entry.entry_id
+ self.roku = Roku(
+ host=config_entry.data[CONF_HOST], session=async_get_clientsession(hass)
+ )
+ self.play_media_app_id = config_entry.options.get(
+ CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID
+ )
self.full_update_interval = timedelta(minutes=15)
self.last_full_update = None
@@ -42,6 +53,7 @@ class RokuDataUpdateCoordinator(DataUpdateCoordinator[Device]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
# We don't want an immediate refresh since the device
diff --git a/homeassistant/components/roku/diagnostics.py b/homeassistant/components/roku/diagnostics.py
index e98837ca442..86e7a7ac1c9 100644
--- a/homeassistant/components/roku/diagnostics.py
+++ b/homeassistant/components/roku/diagnostics.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.core import HomeAssistant
-from . import RokuConfigEntry
+from .coordinator import RokuConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/roku/entity.py b/homeassistant/components/roku/entity.py
index 259cb092cb8..1321e3806d1 100644
--- a/homeassistant/components/roku/entity.py
+++ b/homeassistant/components/roku/entity.py
@@ -6,8 +6,8 @@ from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, Device
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import RokuDataUpdateCoordinator
from .const import DOMAIN
+from .coordinator import RokuDataUpdateCoordinator
class RokuEntity(CoordinatorEntity[RokuDataUpdateCoordinator]):
diff --git a/homeassistant/components/roku/media_player.py b/homeassistant/components/roku/media_player.py
index 0c1f92521af..d0e1e3a53c0 100644
--- a/homeassistant/components/roku/media_player.py
+++ b/homeassistant/components/roku/media_player.py
@@ -26,10 +26,9 @@ from homeassistant.components.stream import FORMAT_CONTENT_TYPE, HLS_PROVIDER
from homeassistant.const import ATTR_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
-from . import RokuConfigEntry
from .browse_media import async_browse_media
from .const import (
ATTR_ARTIST_NAME,
@@ -40,7 +39,7 @@ from .const import (
ATTR_THUMBNAIL,
SERVICE_SEARCH,
)
-from .coordinator import RokuDataUpdateCoordinator
+from .coordinator import RokuConfigEntry, RokuDataUpdateCoordinator
from .entity import RokuEntity
from .helpers import format_channel_name, roku_exception_handler
@@ -83,7 +82,9 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
- hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: RokuConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Roku config entry."""
async_add_entities(
diff --git a/homeassistant/components/roku/remote.py b/homeassistant/components/roku/remote.py
index f7916fb23a2..cc3689c9df3 100644
--- a/homeassistant/components/roku/remote.py
+++ b/homeassistant/components/roku/remote.py
@@ -7,9 +7,9 @@ from typing import Any
from homeassistant.components.remote import ATTR_NUM_REPEATS, RemoteEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RokuConfigEntry
+from .coordinator import RokuConfigEntry
from .entity import RokuEntity
from .helpers import roku_exception_handler
@@ -19,7 +19,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: RokuConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Roku remote based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/roku/select.py b/homeassistant/components/roku/select.py
index 360d4e25415..062e1258ea2 100644
--- a/homeassistant/components/roku/select.py
+++ b/homeassistant/components/roku/select.py
@@ -10,9 +10,9 @@ from rokuecp.models import Device as RokuDevice
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RokuConfigEntry
+from .coordinator import RokuConfigEntry
from .entity import RokuEntity
from .helpers import format_channel_name, roku_exception_handler
@@ -109,7 +109,7 @@ CHANNEL_ENTITY = RokuSelectEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: RokuConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roku select based on a config entry."""
device: RokuDevice = entry.runtime_data.data
diff --git a/homeassistant/components/roku/sensor.py b/homeassistant/components/roku/sensor.py
index 870386945a6..a61a9be6a73 100644
--- a/homeassistant/components/roku/sensor.py
+++ b/homeassistant/components/roku/sensor.py
@@ -10,9 +10,9 @@ from rokuecp.models import Device as RokuDevice
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import RokuConfigEntry
+from .coordinator import RokuConfigEntry
from .entity import RokuEntity
# Coordinator is used to centralize the data updates
@@ -45,7 +45,7 @@ SENSORS: tuple[RokuSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: RokuConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roku sensor based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/romy/__init__.py b/homeassistant/components/romy/__init__.py
index 352f5f3715a..be227645122 100644
--- a/homeassistant/components/romy/__init__.py
+++ b/homeassistant/components/romy/__init__.py
@@ -17,7 +17,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
config_entry.data[CONF_HOST], config_entry.data.get(CONF_PASSWORD, "")
)
- coordinator = RomyVacuumCoordinator(hass, new_romy)
+ coordinator = RomyVacuumCoordinator(hass, config_entry, new_romy)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = coordinator
diff --git a/homeassistant/components/romy/binary_sensor.py b/homeassistant/components/romy/binary_sensor.py
index d8f6216007f..599c0fe023e 100644
--- a/homeassistant/components/romy/binary_sensor.py
+++ b/homeassistant/components/romy/binary_sensor.py
@@ -7,7 +7,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RomyVacuumCoordinator
@@ -39,7 +39,7 @@ BINARY_SENSORS: list[BinarySensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ROMY vacuum cleaner."""
diff --git a/homeassistant/components/romy/coordinator.py b/homeassistant/components/romy/coordinator.py
index 5868eae70e2..d666ec44f80 100644
--- a/homeassistant/components/romy/coordinator.py
+++ b/homeassistant/components/romy/coordinator.py
@@ -2,6 +2,7 @@
from romy import RomyRobot
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -11,9 +12,19 @@ from .const import DOMAIN, LOGGER, UPDATE_INTERVAL
class RomyVacuumCoordinator(DataUpdateCoordinator[None]):
"""ROMY Vacuum Coordinator."""
- def __init__(self, hass: HomeAssistant, romy: RomyRobot) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, romy: RomyRobot
+ ) -> None:
"""Initialize."""
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=UPDATE_INTERVAL,
+ )
self.hass = hass
self.romy = romy
diff --git a/homeassistant/components/romy/sensor.py b/homeassistant/components/romy/sensor.py
index 341125b86ba..85bf0df8f64 100644
--- a/homeassistant/components/romy/sensor.py
+++ b/homeassistant/components/romy/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import RomyVacuumCoordinator
@@ -77,7 +77,7 @@ SENSORS: list[SensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ROMY vacuum cleaner."""
diff --git a/homeassistant/components/romy/vacuum.py b/homeassistant/components/romy/vacuum.py
index 49129daabbd..0e9dd13ffe1 100644
--- a/homeassistant/components/romy/vacuum.py
+++ b/homeassistant/components/romy/vacuum.py
@@ -13,7 +13,7 @@ from homeassistant.components.vacuum import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, LOGGER
from .coordinator import RomyVacuumCoordinator
@@ -51,7 +51,7 @@ SUPPORT_ROMY_ROBOT = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up ROMY vacuum cleaner."""
diff --git a/homeassistant/components/roomba/binary_sensor.py b/homeassistant/components/roomba/binary_sensor.py
index baf66375036..d50535c885a 100644
--- a/homeassistant/components/roomba/binary_sensor.py
+++ b/homeassistant/components/roomba/binary_sensor.py
@@ -3,7 +3,7 @@
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import roomba_reported_state
from .const import DOMAIN
@@ -14,7 +14,7 @@ from .models import RoombaData
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the iRobot Roomba vacuum cleaner."""
domain_data: RoombaData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/roomba/entity.py b/homeassistant/components/roomba/entity.py
index ae5577da4e4..14c7ac3af3e 100644
--- a/homeassistant/components/roomba/entity.py
+++ b/homeassistant/components/roomba/entity.py
@@ -80,7 +80,7 @@ class IRobotEntity(Entity):
return None
return dt_util.utc_from_timestamp(ts)
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register callback function."""
self.vacuum.register_on_message_callback(self.on_message)
diff --git a/homeassistant/components/roomba/sensor.py b/homeassistant/components/roomba/sensor.py
index d358dcb428c..3a98bedcd94 100644
--- a/homeassistant/components/roomba/sensor.py
+++ b/homeassistant/components/roomba/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfArea, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN
@@ -125,7 +125,7 @@ SENSORS: list[RoombaSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the iRobot Roomba vacuum cleaner."""
domain_data: RoombaData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/roomba/vacuum.py b/homeassistant/components/roomba/vacuum.py
index 92063f74afa..10606814a35 100644
--- a/homeassistant/components/roomba/vacuum.py
+++ b/homeassistant/components/roomba/vacuum.py
@@ -14,7 +14,7 @@ from homeassistant.components.vacuum import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from homeassistant.util.unit_system import METRIC_SYSTEM
@@ -89,7 +89,7 @@ SUPPORT_BRAAVA = SUPPORT_IROBOT | VacuumEntityFeature.FAN_SPEED
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the iRobot Roomba vacuum cleaner."""
domain_data: RoombaData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/roon/event.py b/homeassistant/components/roon/event.py
index 7bc6ea27dd9..2f2967c5789 100644
--- a/homeassistant/components/roon/event.py
+++ b/homeassistant/components/roon/event.py
@@ -8,7 +8,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -18,7 +18,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roon Event from Config Entry."""
roon_server = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/roon/media_player.py b/homeassistant/components/roon/media_player.py
index 3b1735cd2fc..4a87601a24f 100644
--- a/homeassistant/components/roon/media_player.py
+++ b/homeassistant/components/roon/media_player.py
@@ -25,7 +25,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import convert
from homeassistant.util.dt import utcnow
@@ -52,7 +52,7 @@ REPEAT_MODE_MAPPING_TO_ROON = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Roon MediaPlayer from Config Entry."""
roon_server = hass.data[DOMAIN][config_entry.entry_id]
@@ -329,6 +329,11 @@ class RoonDevice(MediaPlayerEntity):
"""Album artist of current playing media (Music track only)."""
return self.media_artist
+ @property
+ def media_content_type(self) -> str:
+ """Return the media type."""
+ return MediaType.MUSIC
+
@property
def supports_standby(self):
"""Return power state of source controls."""
diff --git a/homeassistant/components/rova/__init__.py b/homeassistant/components/rova/__init__.py
index 64f0e787a4b..ecde0578772 100644
--- a/homeassistant/components/rova/__init__.py
+++ b/homeassistant/components/rova/__init__.py
@@ -46,7 +46,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
)
raise ConfigEntryError("Rova does not collect garbage in this area")
- coordinator = RovaCoordinator(hass, api)
+ coordinator = RovaCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/rova/coordinator.py b/homeassistant/components/rova/coordinator.py
index ecd91cad823..a48048d32c3 100644
--- a/homeassistant/components/rova/coordinator.py
+++ b/homeassistant/components/rova/coordinator.py
@@ -4,6 +4,7 @@ from datetime import datetime, timedelta
from rova.rova import Rova
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from homeassistant.util.dt import get_time_zone
@@ -16,11 +17,16 @@ EUROPE_AMSTERDAM_ZONE_INFO = get_time_zone("Europe/Amsterdam")
class RovaCoordinator(DataUpdateCoordinator[dict[str, datetime]]):
"""Class to manage fetching Rova data."""
- def __init__(self, hass: HomeAssistant, api: Rova) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: Rova
+ ) -> None:
"""Initialize."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(hours=12),
)
diff --git a/homeassistant/components/rova/sensor.py b/homeassistant/components/rova/sensor.py
index 589183eb7a8..59f9f28f8f5 100644
--- a/homeassistant/components/rova/sensor.py
+++ b/homeassistant/components/rova/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -43,7 +43,7 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Rova entry."""
coordinator: RovaCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/rova/strings.json b/homeassistant/components/rova/strings.json
index 3b89fc789ee..21f4146bf78 100644
--- a/homeassistant/components/rova/strings.json
+++ b/homeassistant/components/rova/strings.json
@@ -4,7 +4,7 @@
"user": {
"title": "Provide your address details",
"data": {
- "zip_code": "Your zip code",
+ "zip_code": "Your ZIP code",
"house_number": "Your house number",
"house_number_suffix": "A suffix for your house number"
}
diff --git a/homeassistant/components/rpi_power/binary_sensor.py b/homeassistant/components/rpi_power/binary_sensor.py
index 00d7ec0e3f4..1424148f554 100644
--- a/homeassistant/components/rpi_power/binary_sensor.py
+++ b/homeassistant/components/rpi_power/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
_LOGGER = logging.getLogger(__name__)
@@ -28,7 +28,7 @@ DESCRIPTION_UNDER_VOLTAGE = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up rpi_power binary sensor."""
under_voltage = await hass.async_add_executor_job(new_under_voltage)
diff --git a/homeassistant/components/ruckus_unleashed/__init__.py b/homeassistant/components/ruckus_unleashed/__init__.py
index 4ee870e8322..8e9219985ce 100644
--- a/homeassistant/components/ruckus_unleashed/__init__.py
+++ b/homeassistant/components/ruckus_unleashed/__init__.py
@@ -46,7 +46,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await ruckus.close()
raise ConfigEntryAuthFailed from autherr
- coordinator = RuckusDataUpdateCoordinator(hass, ruckus=ruckus)
+ coordinator = RuckusDataUpdateCoordinator(hass, entry, ruckus)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/ruckus_unleashed/coordinator.py b/homeassistant/components/ruckus_unleashed/coordinator.py
index d9f20883559..7ffaab2e977 100644
--- a/homeassistant/components/ruckus_unleashed/coordinator.py
+++ b/homeassistant/components/ruckus_unleashed/coordinator.py
@@ -6,6 +6,7 @@ import logging
from aioruckus import AjaxSession
from aioruckus.exceptions import AuthenticationError, SchemaError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -18,17 +19,20 @@ _LOGGER = logging.getLogger(__package__)
class RuckusDataUpdateCoordinator(DataUpdateCoordinator):
"""Coordinator to manage data from Ruckus client."""
- def __init__(self, hass: HomeAssistant, *, ruckus: AjaxSession) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, ruckus: AjaxSession
+ ) -> None:
"""Initialize global Ruckus data updater."""
self.ruckus = ruckus
- update_interval = timedelta(seconds=SCAN_INTERVAL)
-
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
- update_interval=update_interval,
+ update_interval=timedelta(seconds=SCAN_INTERVAL),
)
async def _fetch_clients(self) -> dict:
diff --git a/homeassistant/components/ruckus_unleashed/device_tracker.py b/homeassistant/components/ruckus_unleashed/device_tracker.py
index 8a5e8b79294..890148ec25c 100644
--- a/homeassistant/components/ruckus_unleashed/device_tracker.py
+++ b/homeassistant/components/ruckus_unleashed/device_tracker.py
@@ -8,7 +8,7 @@ from homeassistant.components.device_tracker import ScannerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -25,7 +25,9 @@ _LOGGER = logging.getLogger(__package__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Ruckus component."""
coordinator = hass.data[DOMAIN][entry.entry_id][COORDINATOR]
@@ -69,7 +71,7 @@ def restore_entities(
registry: er.EntityRegistry,
coordinator: RuckusDataUpdateCoordinator,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
tracked: set[str],
) -> None:
"""Restore clients that are not a part of active clients list."""
diff --git a/homeassistant/components/russound_rio/media_player.py b/homeassistant/components/russound_rio/media_player.py
index 346f4903f6a..b40b82862f9 100644
--- a/homeassistant/components/russound_rio/media_player.py
+++ b/homeassistant/components/russound_rio/media_player.py
@@ -20,7 +20,7 @@ from homeassistant.components.media_player import (
MediaType,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import RussoundConfigEntry
from .entity import RussoundBaseEntity, command
@@ -33,7 +33,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: RussoundConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Russound RIO platform."""
client = entry.runtime_data
diff --git a/homeassistant/components/ruuvi_gateway/__init__.py b/homeassistant/components/ruuvi_gateway/__init__.py
index 77b3e9b57de..da93a89a9f3 100644
--- a/homeassistant/components/ruuvi_gateway/__init__.py
+++ b/homeassistant/components/ruuvi_gateway/__init__.py
@@ -5,11 +5,10 @@ from __future__ import annotations
import logging
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_HOST, CONF_TOKEN
from homeassistant.core import HomeAssistant
from .bluetooth import async_connect_scanner
-from .const import DOMAIN, SCAN_INTERVAL
+from .const import DOMAIN
from .coordinator import RuuviGatewayUpdateCoordinator
from .models import RuuviGatewayRuntimeData
@@ -18,14 +17,7 @@ _LOGGER = logging.getLogger(DOMAIN)
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Ruuvi Gateway from a config entry."""
- coordinator = RuuviGatewayUpdateCoordinator(
- hass,
- logger=_LOGGER,
- name=entry.title,
- update_interval=SCAN_INTERVAL,
- host=entry.data[CONF_HOST],
- token=entry.data[CONF_TOKEN],
- )
+ coordinator = RuuviGatewayUpdateCoordinator(hass, entry, _LOGGER)
scanner, unload_scanner = async_connect_scanner(hass, entry, coordinator)
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = RuuviGatewayRuntimeData(
update_coordinator=coordinator,
diff --git a/homeassistant/components/ruuvi_gateway/coordinator.py b/homeassistant/components/ruuvi_gateway/coordinator.py
index ba72dfe4cbc..0c42cd0cb38 100644
--- a/homeassistant/components/ruuvi_gateway/coordinator.py
+++ b/homeassistant/components/ruuvi_gateway/coordinator.py
@@ -2,34 +2,41 @@
from __future__ import annotations
-from datetime import timedelta
import logging
from aioruuvigateway.api import get_gateway_history_data
from aioruuvigateway.models import TagData
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, CONF_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.helpers.httpx_client import get_async_client
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
+from .const import SCAN_INTERVAL
+
class RuuviGatewayUpdateCoordinator(DataUpdateCoordinator[list[TagData]]):
"""Polls the gateway for data and returns a list of TagData objects that have changed since the last poll."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
logger: logging.Logger,
- *,
- name: str,
- update_interval: timedelta | None = None,
- host: str,
- token: str,
) -> None:
"""Initialize the coordinator using the given configuration (host, token)."""
- super().__init__(hass, logger, name=name, update_interval=update_interval)
- self.host = host
- self.token = token
+ super().__init__(
+ hass,
+ logger,
+ config_entry=config_entry,
+ name=config_entry.title,
+ update_interval=SCAN_INTERVAL,
+ )
+ self.host = config_entry.data[CONF_HOST]
+ self.token = config_entry.data[CONF_TOKEN]
self.last_tag_datas: dict[str, TagData] = {}
async def _async_update_data(self) -> list[TagData]:
diff --git a/homeassistant/components/ruuvitag_ble/sensor.py b/homeassistant/components/ruuvitag_ble/sensor.py
index ef287753ed4..57248d547ba 100644
--- a/homeassistant/components/ruuvitag_ble/sensor.py
+++ b/homeassistant/components/ruuvitag_ble/sensor.py
@@ -32,7 +32,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -126,7 +126,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Ruuvitag BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/rympro/coordinator.py b/homeassistant/components/rympro/coordinator.py
index 55e5f0f90df..6b49a065d35 100644
--- a/homeassistant/components/rympro/coordinator.py
+++ b/homeassistant/components/rympro/coordinator.py
@@ -42,6 +42,12 @@ class RymProDataUpdateCoordinator(DataUpdateCoordinator[dict[int, dict]]):
try:
meters = await self.rympro.last_read()
for meter_id, meter in meters.items():
+ meter["monthly_consumption"] = await self.rympro.monthly_consumption(
+ meter_id
+ )
+ meter["daily_consumption"] = await self.rympro.daily_consumption(
+ meter_id
+ )
meter["consumption_forecast"] = await self.rympro.consumption_forecast(
meter_id
)
diff --git a/homeassistant/components/rympro/manifest.json b/homeassistant/components/rympro/manifest.json
index 046e778f05b..51c26b312fb 100644
--- a/homeassistant/components/rympro/manifest.json
+++ b/homeassistant/components/rympro/manifest.json
@@ -5,5 +5,5 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/rympro",
"iot_class": "cloud_polling",
- "requirements": ["pyrympro==0.0.8"]
+ "requirements": ["pyrympro==0.0.9"]
}
diff --git a/homeassistant/components/rympro/sensor.py b/homeassistant/components/rympro/sensor.py
index 8bb0af6e9ff..66ed41a4ce9 100644
--- a/homeassistant/components/rympro/sensor.py
+++ b/homeassistant/components/rympro/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfVolume
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -36,6 +36,20 @@ SENSOR_DESCRIPTIONS: tuple[RymProSensorEntityDescription, ...] = (
suggested_display_precision=3,
value_key="read",
),
+ RymProSensorEntityDescription(
+ key="monthly_consumption",
+ translation_key="monthly_consumption",
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ suggested_display_precision=3,
+ value_key="monthly_consumption",
+ ),
+ RymProSensorEntityDescription(
+ key="daily_consumption",
+ translation_key="daily_consumption",
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ suggested_display_precision=3,
+ value_key="daily_consumption",
+ ),
RymProSensorEntityDescription(
key="monthly_forecast",
translation_key="monthly_forecast",
@@ -48,7 +62,7 @@ SENSOR_DESCRIPTIONS: tuple[RymProSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for device."""
coordinator: RymProDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/rympro/strings.json b/homeassistant/components/rympro/strings.json
index 2c1e2ad93c9..589e91a6c6f 100644
--- a/homeassistant/components/rympro/strings.json
+++ b/homeassistant/components/rympro/strings.json
@@ -23,6 +23,12 @@
"total_consumption": {
"name": "Total consumption"
},
+ "monthly_consumption": {
+ "name": "Monthly consumption"
+ },
+ "daily_consumption": {
+ "name": "Daily consumption"
+ },
"monthly_forecast": {
"name": "Monthly forecast"
}
diff --git a/homeassistant/components/sabnzbd/binary_sensor.py b/homeassistant/components/sabnzbd/binary_sensor.py
index 1d65bf01211..59ef17237e2 100644
--- a/homeassistant/components/sabnzbd/binary_sensor.py
+++ b/homeassistant/components/sabnzbd/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SabnzbdConfigEntry
from .entity import SabnzbdEntity
@@ -40,7 +40,7 @@ BINARY_SENSORS: tuple[SabnzbdBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SabnzbdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Sabnzbd sensor entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/sabnzbd/button.py b/homeassistant/components/sabnzbd/button.py
index 1ff26b41655..25c11f6b2ec 100644
--- a/homeassistant/components/sabnzbd/button.py
+++ b/homeassistant/components/sabnzbd/button.py
@@ -9,7 +9,7 @@ from pysabnzbd import SabnzbdApiException
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
@@ -40,7 +40,7 @@ BUTTON_DESCRIPTIONS: tuple[SabnzbdButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SabnzbdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up buttons from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/sabnzbd/number.py b/homeassistant/components/sabnzbd/number.py
index 53c8d462f11..63b2206ac70 100644
--- a/homeassistant/components/sabnzbd/number.py
+++ b/homeassistant/components/sabnzbd/number.py
@@ -15,7 +15,7 @@ from homeassistant.components.number import (
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SabnzbdConfigEntry, SabnzbdUpdateCoordinator
@@ -48,7 +48,7 @@ NUMBER_DESCRIPTIONS: tuple[SabnzbdNumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SabnzbdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SABnzbd number entity."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/sabnzbd/sensor.py b/homeassistant/components/sabnzbd/sensor.py
index 662ae739d15..5e871b4bf40 100644
--- a/homeassistant/components/sabnzbd/sensor.py
+++ b/homeassistant/components/sabnzbd/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfDataRate, UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .coordinator import SabnzbdConfigEntry
@@ -115,7 +115,7 @@ SENSOR_TYPES: tuple[SabnzbdSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SabnzbdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Sabnzbd sensor entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/samsungtv/__init__.py b/homeassistant/components/samsungtv/__init__.py
index 6d4e491b839..e416cd35765 100644
--- a/homeassistant/components/samsungtv/__init__.py
+++ b/homeassistant/components/samsungtv/__init__.py
@@ -44,14 +44,11 @@ from .const import (
UPNP_SVC_MAIN_TV_AGENT,
UPNP_SVC_RENDERING_CONTROL,
)
-from .coordinator import SamsungTVDataUpdateCoordinator
+from .coordinator import SamsungTVConfigEntry, SamsungTVDataUpdateCoordinator
PLATFORMS = [Platform.MEDIA_PLAYER, Platform.REMOTE]
-SamsungTVConfigEntry = ConfigEntry[SamsungTVDataUpdateCoordinator]
-
-
@callback
def _async_get_device_bridge(
hass: HomeAssistant, data: dict[str, Any]
@@ -165,7 +162,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: SamsungTVConfigEntry) ->
entry.async_on_unload(debounced_reloader.async_shutdown)
entry.async_on_unload(entry.add_update_listener(debounced_reloader.async_call))
- coordinator = SamsungTVDataUpdateCoordinator(hass, bridge)
+ coordinator = SamsungTVDataUpdateCoordinator(hass, entry, bridge)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/samsungtv/coordinator.py b/homeassistant/components/samsungtv/coordinator.py
index 92d8dc8fa84..443e62b13fb 100644
--- a/homeassistant/components/samsungtv/coordinator.py
+++ b/homeassistant/components/samsungtv/coordinator.py
@@ -15,17 +15,25 @@ from .const import DOMAIN, LOGGER
SCAN_INTERVAL = 10
+type SamsungTVConfigEntry = ConfigEntry[SamsungTVDataUpdateCoordinator]
+
class SamsungTVDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Coordinator for the SamsungTV integration."""
- config_entry: ConfigEntry
+ config_entry: SamsungTVConfigEntry
- def __init__(self, hass: HomeAssistant, bridge: SamsungTVBridge) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SamsungTVConfigEntry,
+ bridge: SamsungTVBridge,
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=SCAN_INTERVAL),
)
diff --git a/homeassistant/components/samsungtv/diagnostics.py b/homeassistant/components/samsungtv/diagnostics.py
index ebca8d2543b..667d23ba631 100644
--- a/homeassistant/components/samsungtv/diagnostics.py
+++ b/homeassistant/components/samsungtv/diagnostics.py
@@ -8,8 +8,8 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_TOKEN
from homeassistant.core import HomeAssistant
-from . import SamsungTVConfigEntry
from .const import CONF_SESSION_ID
+from .coordinator import SamsungTVConfigEntry
TO_REDACT = {CONF_TOKEN, CONF_SESSION_ID}
diff --git a/homeassistant/components/samsungtv/helpers.py b/homeassistant/components/samsungtv/helpers.py
index 4e8dd00d486..b4075b8117f 100644
--- a/homeassistant/components/samsungtv/helpers.py
+++ b/homeassistant/components/samsungtv/helpers.py
@@ -7,9 +7,9 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.device_registry import DeviceEntry
-from . import SamsungTVConfigEntry
from .bridge import SamsungTVBridge
from .const import DOMAIN
+from .coordinator import SamsungTVConfigEntry
@callback
diff --git a/homeassistant/components/samsungtv/media_player.py b/homeassistant/components/samsungtv/media_player.py
index 7180e8a0c1a..4e6ecfd3593 100644
--- a/homeassistant/components/samsungtv/media_player.py
+++ b/homeassistant/components/samsungtv/media_player.py
@@ -31,13 +31,12 @@ from homeassistant.components.media_player import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.async_ import create_eager_task
-from . import SamsungTVConfigEntry
from .bridge import SamsungTVWSBridge
from .const import CONF_SSDP_RENDERING_CONTROL_LOCATION, LOGGER
-from .coordinator import SamsungTVDataUpdateCoordinator
+from .coordinator import SamsungTVConfigEntry, SamsungTVDataUpdateCoordinator
from .entity import SamsungTVEntity
SOURCES = {"TV": "KEY_TV", "HDMI": "KEY_HDMI"}
@@ -64,7 +63,7 @@ APP_LIST_DELAY = 3
async def async_setup_entry(
hass: HomeAssistant,
entry: SamsungTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Samsung TV from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/samsungtv/remote.py b/homeassistant/components/samsungtv/remote.py
index 401a5d383f0..d6fef262d91 100644
--- a/homeassistant/components/samsungtv/remote.py
+++ b/homeassistant/components/samsungtv/remote.py
@@ -7,17 +7,17 @@ from typing import Any
from homeassistant.components.remote import ATTR_NUM_REPEATS, RemoteEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SamsungTVConfigEntry
from .const import LOGGER
+from .coordinator import SamsungTVConfigEntry
from .entity import SamsungTVEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: SamsungTVConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Samsung TV from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/sanix/__init__.py b/homeassistant/components/sanix/__init__.py
index c8c5567eedc..60cc5b56f2e 100644
--- a/homeassistant/components/sanix/__init__.py
+++ b/homeassistant/components/sanix/__init__.py
@@ -19,7 +19,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
token = entry.data[CONF_TOKEN]
sanix_api = Sanix(serial_no, token)
- coordinator = SanixCoordinator(hass, sanix_api)
+ coordinator = SanixCoordinator(hass, entry, sanix_api)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/sanix/coordinator.py b/homeassistant/components/sanix/coordinator.py
index d6362337a38..64d28fa9191 100644
--- a/homeassistant/components/sanix/coordinator.py
+++ b/homeassistant/components/sanix/coordinator.py
@@ -21,10 +21,16 @@ class SanixCoordinator(DataUpdateCoordinator[Measurement]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, sanix_api: Sanix) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, sanix_api: Sanix
+ ) -> None:
"""Initialize coordinator."""
super().__init__(
- hass, _LOGGER, name=MANUFACTURER, update_interval=timedelta(hours=1)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=MANUFACTURER,
+ update_interval=timedelta(hours=1),
)
self._sanix_api = sanix_api
diff --git a/homeassistant/components/sanix/sensor.py b/homeassistant/components/sanix/sensor.py
index 39a1c593433..d2a1aecb099 100644
--- a/homeassistant/components/sanix/sensor.py
+++ b/homeassistant/components/sanix/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfLength
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MANUFACTURER
@@ -82,7 +82,9 @@ SENSOR_TYPES: tuple[SanixSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sanix Sensor entities based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/schedule/strings.json b/homeassistant/components/schedule/strings.json
index 8638e4a8a84..bb81c029dbf 100644
--- a/homeassistant/components/schedule/strings.json
+++ b/homeassistant/components/schedule/strings.json
@@ -28,7 +28,7 @@
},
"get_schedule": {
"name": "Get schedule",
- "description": "Retrieve one or multiple schedules."
+ "description": "Retrieves the configured time ranges of one or multiple schedules."
}
}
}
diff --git a/homeassistant/components/schlage/__init__.py b/homeassistant/components/schlage/__init__.py
index 6eae69d9542..509a335aafe 100644
--- a/homeassistant/components/schlage/__init__.py
+++ b/homeassistant/components/schlage/__init__.py
@@ -5,12 +5,11 @@ from __future__ import annotations
from pycognito.exceptions import WarrantException
import pyschlage
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
-from .coordinator import SchlageDataUpdateCoordinator
+from .coordinator import SchlageConfigEntry, SchlageDataUpdateCoordinator
PLATFORMS: list[Platform] = [
Platform.BINARY_SENSOR,
@@ -20,8 +19,6 @@ PLATFORMS: list[Platform] = [
Platform.SWITCH,
]
-type SchlageConfigEntry = ConfigEntry[SchlageDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: SchlageConfigEntry) -> bool:
"""Set up Schlage from a config entry."""
@@ -32,7 +29,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: SchlageConfigEntry) -> b
except WarrantException as ex:
raise ConfigEntryAuthFailed from ex
- coordinator = SchlageDataUpdateCoordinator(hass, username, pyschlage.Schlage(auth))
+ coordinator = SchlageDataUpdateCoordinator(
+ hass, entry, username, pyschlage.Schlage(auth)
+ )
entry.runtime_data = coordinator
await coordinator.async_config_entry_first_refresh()
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/schlage/binary_sensor.py b/homeassistant/components/schlage/binary_sensor.py
index f928d42b3ee..62e69b5cb4a 100644
--- a/homeassistant/components/schlage/binary_sensor.py
+++ b/homeassistant/components/schlage/binary_sensor.py
@@ -12,10 +12,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SchlageConfigEntry
-from .coordinator import LockData, SchlageDataUpdateCoordinator
+from .coordinator import LockData, SchlageConfigEntry, SchlageDataUpdateCoordinator
from .entity import SchlageEntity
@@ -40,7 +39,7 @@ _DESCRIPTIONS: tuple[SchlageBinarySensorEntityDescription] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SchlageConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary_sensors based on a config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/schlage/coordinator.py b/homeassistant/components/schlage/coordinator.py
index 936ef9ee91e..eec143c574f 100644
--- a/homeassistant/components/schlage/coordinator.py
+++ b/homeassistant/components/schlage/coordinator.py
@@ -34,15 +34,28 @@ class SchlageData:
locks: dict[str, LockData]
+type SchlageConfigEntry = ConfigEntry[SchlageDataUpdateCoordinator]
+
+
class SchlageDataUpdateCoordinator(DataUpdateCoordinator[SchlageData]):
"""The Schlage data update coordinator."""
- config_entry: ConfigEntry
+ config_entry: SchlageConfigEntry
- def __init__(self, hass: HomeAssistant, username: str, api: Schlage) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SchlageConfigEntry,
+ username: str,
+ api: Schlage,
+ ) -> None:
"""Initialize the class."""
super().__init__(
- hass, LOGGER, name=f"{DOMAIN} ({username})", update_interval=UPDATE_INTERVAL
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=f"{DOMAIN} ({username})",
+ update_interval=UPDATE_INTERVAL,
)
self.data = SchlageData(locks={})
self.api = api
diff --git a/homeassistant/components/schlage/diagnostics.py b/homeassistant/components/schlage/diagnostics.py
index ec4d9c489e3..357f04f00db 100644
--- a/homeassistant/components/schlage/diagnostics.py
+++ b/homeassistant/components/schlage/diagnostics.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.core import HomeAssistant
-from . import SchlageConfigEntry
+from .coordinator import SchlageConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/schlage/lock.py b/homeassistant/components/schlage/lock.py
index d203913191d..83abf9214e3 100644
--- a/homeassistant/components/schlage/lock.py
+++ b/homeassistant/components/schlage/lock.py
@@ -6,17 +6,16 @@ from typing import Any
from homeassistant.components.lock import LockEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SchlageConfigEntry
-from .coordinator import LockData, SchlageDataUpdateCoordinator
+from .coordinator import LockData, SchlageConfigEntry, SchlageDataUpdateCoordinator
from .entity import SchlageEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SchlageConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Schlage WiFi locks based on a config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/schlage/select.py b/homeassistant/components/schlage/select.py
index 6cf0853835f..4648686aaac 100644
--- a/homeassistant/components/schlage/select.py
+++ b/homeassistant/components/schlage/select.py
@@ -5,10 +5,9 @@ from __future__ import annotations
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SchlageConfigEntry
-from .coordinator import LockData, SchlageDataUpdateCoordinator
+from .coordinator import LockData, SchlageConfigEntry, SchlageDataUpdateCoordinator
from .entity import SchlageEntity
_DESCRIPTIONS = (
@@ -33,7 +32,7 @@ _DESCRIPTIONS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SchlageConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up selects based on a config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/schlage/sensor.py b/homeassistant/components/schlage/sensor.py
index a15d1740b91..494efc7585a 100644
--- a/homeassistant/components/schlage/sensor.py
+++ b/homeassistant/components/schlage/sensor.py
@@ -8,12 +8,11 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .coordinator import LockData, SchlageDataUpdateCoordinator
+from .coordinator import LockData, SchlageConfigEntry, SchlageDataUpdateCoordinator
from .entity import SchlageEntity
_SENSOR_DESCRIPTIONS: list[SensorEntityDescription] = [
@@ -29,8 +28,8 @@ _SENSOR_DESCRIPTIONS: list[SensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: SchlageConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors based on a config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/schlage/switch.py b/homeassistant/components/schlage/switch.py
index 39fe6dbbc99..c40d0c41e88 100644
--- a/homeassistant/components/schlage/switch.py
+++ b/homeassistant/components/schlage/switch.py
@@ -14,12 +14,11 @@ from homeassistant.components.switch import (
SwitchEntity,
SwitchEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .coordinator import LockData, SchlageDataUpdateCoordinator
+from .coordinator import LockData, SchlageConfigEntry, SchlageDataUpdateCoordinator
from .entity import SchlageEntity
@@ -56,8 +55,8 @@ SWITCHES: tuple[SchlageSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: SchlageConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches based on a config entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/scrape/sensor.py b/homeassistant/components/scrape/sensor.py
index 5ee837f32d1..b8ad9cb8a56 100644
--- a/homeassistant/components/scrape/sensor.py
+++ b/homeassistant/components/scrape/sensor.py
@@ -21,7 +21,10 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.template import Template
from homeassistant.helpers.trigger_template_entity import (
CONF_AVAILABILITY,
@@ -92,7 +95,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: ScrapeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Scrape sensor entry."""
entities: list = []
diff --git a/homeassistant/components/screenlogic/binary_sensor.py b/homeassistant/components/screenlogic/binary_sensor.py
index 4a178c60d81..a846a9fa4e3 100644
--- a/homeassistant/components/screenlogic/binary_sensor.py
+++ b/homeassistant/components/screenlogic/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ScreenlogicDataUpdateCoordinator
from .entity import (
@@ -195,7 +195,7 @@ SUPPORTED_SCG_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ScreenLogicConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/screenlogic/climate.py b/homeassistant/components/screenlogic/climate.py
index e44d9b18ae1..03aebadbba6 100644
--- a/homeassistant/components/screenlogic/climate.py
+++ b/homeassistant/components/screenlogic/climate.py
@@ -21,7 +21,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .entity import ScreenLogicPushEntity, ScreenLogicPushEntityDescription
@@ -42,7 +42,7 @@ SUPPORTED_PRESETS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ScreenLogicConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/screenlogic/light.py b/homeassistant/components/screenlogic/light.py
index 412b2df5f81..b0bd154b66d 100644
--- a/homeassistant/components/screenlogic/light.py
+++ b/homeassistant/components/screenlogic/light.py
@@ -12,7 +12,7 @@ from homeassistant.components.light import (
LightEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import LIGHT_CIRCUIT_FUNCTIONS
from .entity import ScreenLogicCircuitEntity, ScreenLogicPushEntityDescription
@@ -22,7 +22,7 @@ from .types import ScreenLogicConfigEntry
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ScreenLogicConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
entities: list[ScreenLogicLight] = []
diff --git a/homeassistant/components/screenlogic/number.py b/homeassistant/components/screenlogic/number.py
index 3634147e509..ea9bf8ac95d 100644
--- a/homeassistant/components/screenlogic/number.py
+++ b/homeassistant/components/screenlogic/number.py
@@ -17,7 +17,7 @@ from homeassistant.components.number import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ScreenlogicDataUpdateCoordinator
from .entity import (
@@ -104,7 +104,7 @@ SUPPORTED_SCG_NUMBERS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ScreenLogicConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
entities: list[ScreenLogicNumber] = []
diff --git a/homeassistant/components/screenlogic/sensor.py b/homeassistant/components/screenlogic/sensor.py
index 7a5e910923c..95a7e3a5c75 100644
--- a/homeassistant/components/screenlogic/sensor.py
+++ b/homeassistant/components/screenlogic/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ScreenlogicDataUpdateCoordinator
from .entity import (
@@ -272,7 +272,7 @@ SUPPORTED_SCG_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ScreenLogicConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/screenlogic/strings.json b/homeassistant/components/screenlogic/strings.json
index 09e64808dfe..97e12277eb6 100644
--- a/homeassistant/components/screenlogic/strings.json
+++ b/homeassistant/components/screenlogic/strings.json
@@ -3,9 +3,9 @@
"service_config_entry_name": "Config entry",
"service_config_entry_description": "The config entry to use for this action.",
"climate_preset_solar": "Solar",
- "climate_preset_solar_preferred": "Solar Preferred",
+ "climate_preset_solar_preferred": "Solar preferred",
"climate_preset_heater": "Heater",
- "climate_preset_dont_change": "Don't Change"
+ "climate_preset_dont_change": "Don't change"
},
"config": {
"flow_title": "{name}",
@@ -15,7 +15,7 @@
"step": {
"gateway_entry": {
"title": "ScreenLogic",
- "description": "Enter your ScreenLogic Gateway information.",
+ "description": "Enter your ScreenLogic gateway information.",
"data": {
"ip_address": "[%key:common::config_flow::data::ip%]",
"port": "[%key:common::config_flow::data::port%]"
@@ -46,7 +46,7 @@
},
"services": {
"set_color_mode": {
- "name": "Set Color Mode",
+ "name": "Set color mode",
"description": "Sets the color mode for all color-capable lights attached to this ScreenLogic gateway.",
"fields": {
"config_entry": {
@@ -54,13 +54,13 @@
"description": "[%key:component::screenlogic::common::service_config_entry_description%]"
},
"color_mode": {
- "name": "Color Mode",
+ "name": "Color mode",
"description": "The ScreenLogic color mode to set."
}
}
},
"start_super_chlorination": {
- "name": "Start Super Chlorination",
+ "name": "Start super chlorination",
"description": "Begins super chlorination, running for the specified period or 24 hours if none is specified.",
"fields": {
"config_entry": {
@@ -68,13 +68,13 @@
"description": "[%key:component::screenlogic::common::service_config_entry_description%]"
},
"runtime": {
- "name": "Run Time",
+ "name": "Run time",
"description": "Number of hours for super chlorination to run."
}
}
},
"stop_super_chlorination": {
- "name": "Stop Super Chlorination",
+ "name": "Stop super chlorination",
"description": "Stops super chlorination.",
"fields": {
"config_entry": {
diff --git a/homeassistant/components/screenlogic/switch.py b/homeassistant/components/screenlogic/switch.py
index 1d36ee00b94..dfbb1c1781d 100644
--- a/homeassistant/components/screenlogic/switch.py
+++ b/homeassistant/components/screenlogic/switch.py
@@ -8,7 +8,7 @@ from screenlogicpy.device_const.circuit import GENERIC_CIRCUIT_NAMES, INTERFACE
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import LIGHT_CIRCUIT_FUNCTIONS
from .entity import (
@@ -29,7 +29,7 @@ class ScreenLogicCircuitSwitchDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ScreenLogicConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
entities: list[ScreenLogicSwitchingEntity] = []
diff --git a/homeassistant/components/search/manifest.json b/homeassistant/components/search/manifest.json
index cd372139451..42a54fe8b55 100644
--- a/homeassistant/components/search/manifest.json
+++ b/homeassistant/components/search/manifest.json
@@ -1,7 +1,6 @@
{
"domain": "search",
"name": "Search",
- "after_dependencies": ["scene", "group", "automation", "script"],
"codeowners": ["@home-assistant/core"],
"dependencies": ["websocket_api"],
"documentation": "https://www.home-assistant.io/integrations/search",
diff --git a/homeassistant/components/season/sensor.py b/homeassistant/components/season/sensor.py
index 96744db1d02..bdc24883c90 100644
--- a/homeassistant/components/season/sensor.py
+++ b/homeassistant/components/season/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TYPE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from .const import DOMAIN, TYPE_ASTRONOMICAL
@@ -37,7 +37,7 @@ HEMISPHERE_SEASON_SWAP = {
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from config entry."""
hemisphere = EQUATOR
diff --git a/homeassistant/components/sense/__init__.py b/homeassistant/components/sense/__init__.py
index e919d48e96d..a5393181057 100644
--- a/homeassistant/components/sense/__init__.py
+++ b/homeassistant/components/sense/__init__.py
@@ -89,8 +89,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: SenseConfigEntry) -> boo
except SENSE_WEBSOCKET_EXCEPTIONS as err:
raise ConfigEntryNotReady(str(err) or "Error during realtime update") from err
- trends_coordinator = SenseTrendCoordinator(hass, gateway)
- realtime_coordinator = SenseRealtimeCoordinator(hass, gateway)
+ trends_coordinator = SenseTrendCoordinator(hass, entry, gateway)
+ realtime_coordinator = SenseRealtimeCoordinator(hass, entry, gateway)
# This can take longer than 60s and we already know
# sense is online since get_discovered_device_data was
diff --git a/homeassistant/components/sense/binary_sensor.py b/homeassistant/components/sense/binary_sensor.py
index d06b3a62937..3bb8a32b8e4 100644
--- a/homeassistant/components/sense/binary_sensor.py
+++ b/homeassistant/components/sense/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SenseConfigEntry
from .const import DOMAIN
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SenseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Sense binary sensor."""
sense_monitor_id = config_entry.runtime_data.data.sense_monitor_id
diff --git a/homeassistant/components/sense/coordinator.py b/homeassistant/components/sense/coordinator.py
index c0029cd79ea..1957352aea6 100644
--- a/homeassistant/components/sense/coordinator.py
+++ b/homeassistant/components/sense/coordinator.py
@@ -1,7 +1,10 @@
"""Sense Coordinators."""
+from __future__ import annotations
+
from datetime import timedelta
import logging
+from typing import TYPE_CHECKING
from sense_energy import (
ASyncSenseable,
@@ -13,6 +16,9 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import SenseConfigEntry
+
from .const import (
ACTIVE_UPDATE_RATE,
SENSE_CONNECT_EXCEPTIONS,
@@ -27,13 +33,21 @@ _LOGGER = logging.getLogger(__name__)
class SenseCoordinator(DataUpdateCoordinator[None]):
"""Sense Trend Coordinator."""
+ config_entry: SenseConfigEntry
+
def __init__(
- self, hass: HomeAssistant, gateway: ASyncSenseable, name: str, update: int
+ self,
+ hass: HomeAssistant,
+ config_entry: SenseConfigEntry,
+ gateway: ASyncSenseable,
+ name: str,
+ update: int,
) -> None:
"""Initialize."""
super().__init__(
hass,
logger=_LOGGER,
+ config_entry=config_entry,
name=f"Sense {name} {gateway.sense_monitor_id}",
update_interval=timedelta(seconds=update),
)
@@ -44,9 +58,14 @@ class SenseCoordinator(DataUpdateCoordinator[None]):
class SenseTrendCoordinator(SenseCoordinator):
"""Sense Trend Coordinator."""
- def __init__(self, hass: HomeAssistant, gateway: ASyncSenseable) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SenseConfigEntry,
+ gateway: ASyncSenseable,
+ ) -> None:
"""Initialize."""
- super().__init__(hass, gateway, "Trends", TREND_UPDATE_RATE)
+ super().__init__(hass, config_entry, gateway, "Trends", TREND_UPDATE_RATE)
async def _async_update_data(self) -> None:
"""Update the trend data."""
@@ -62,9 +81,14 @@ class SenseTrendCoordinator(SenseCoordinator):
class SenseRealtimeCoordinator(SenseCoordinator):
"""Sense Realtime Coordinator."""
- def __init__(self, hass: HomeAssistant, gateway: ASyncSenseable) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SenseConfigEntry,
+ gateway: ASyncSenseable,
+ ) -> None:
"""Initialize."""
- super().__init__(hass, gateway, "Realtime", ACTIVE_UPDATE_RATE)
+ super().__init__(hass, config_entry, gateway, "Realtime", ACTIVE_UPDATE_RATE)
async def _async_update_data(self) -> None:
"""Retrieve latest state."""
diff --git a/homeassistant/components/sense/manifest.json b/homeassistant/components/sense/manifest.json
index 966488b6a48..0a21dbf4cc3 100644
--- a/homeassistant/components/sense/manifest.json
+++ b/homeassistant/components/sense/manifest.json
@@ -20,5 +20,5 @@
"documentation": "https://www.home-assistant.io/integrations/sense",
"iot_class": "cloud_polling",
"loggers": ["sense_energy"],
- "requirements": ["sense-energy==0.13.4"]
+ "requirements": ["sense-energy==0.13.7"]
}
diff --git a/homeassistant/components/sense/sensor.py b/homeassistant/components/sense/sensor.py
index 2f5c82675d5..8cb4bdd3e56 100644
--- a/homeassistant/components/sense/sensor.py
+++ b/homeassistant/components/sense/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SenseConfigEntry
from .const import (
@@ -66,7 +66,7 @@ TREND_SENSOR_VARIANTS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SenseConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Sense sensor."""
data = config_entry.runtime_data.data
diff --git a/homeassistant/components/sensibo/binary_sensor.py b/homeassistant/components/sensibo/binary_sensor.py
index a66ab46c882..c7116db7954 100644
--- a/homeassistant/components/sensibo/binary_sensor.py
+++ b/homeassistant/components/sensibo/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SensiboConfigEntry
from .const import LOGGER
@@ -118,7 +118,7 @@ DESCRIPTION_BY_MODELS = {"pure": PURE_SENSOR_TYPES}
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensibo binary sensor platform."""
@@ -130,9 +130,10 @@ async def async_setup_entry(
"""Handle additions of devices and sensors."""
entities: list[SensiboMotionSensor | SensiboDeviceSensor] = []
nonlocal added_devices
- new_devices, remove_devices, added_devices = coordinator.get_devices(
+ new_devices, remove_devices, new_added_devices = coordinator.get_devices(
added_devices
)
+ added_devices = new_added_devices
if LOGGER.isEnabledFor(logging.DEBUG):
LOGGER.debug(
@@ -168,8 +169,7 @@ async def async_setup_entry(
device_data.model, DEVICE_SENSOR_TYPES
)
)
-
- async_add_entities(entities)
+ async_add_entities(entities)
entry.async_on_unload(coordinator.async_add_listener(_add_remove_devices))
_add_remove_devices()
diff --git a/homeassistant/components/sensibo/button.py b/homeassistant/components/sensibo/button.py
index df8d4625840..d36967dae06 100644
--- a/homeassistant/components/sensibo/button.py
+++ b/homeassistant/components/sensibo/button.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SensiboConfigEntry
from .coordinator import SensiboDataUpdateCoordinator
@@ -35,7 +35,7 @@ DEVICE_BUTTON_TYPES = SensiboButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensibo button platform."""
@@ -46,7 +46,8 @@ async def async_setup_entry(
def _add_remove_devices() -> None:
"""Handle additions of devices and sensors."""
nonlocal added_devices
- new_devices, _, added_devices = coordinator.get_devices(added_devices)
+ new_devices, _, new_added_devices = coordinator.get_devices(added_devices)
+ added_devices = new_added_devices
if new_devices:
async_add_entities(
diff --git a/homeassistant/components/sensibo/climate.py b/homeassistant/components/sensibo/climate.py
index 5d1c6ff9e79..906c4259ce5 100644
--- a/homeassistant/components/sensibo/climate.py
+++ b/homeassistant/components/sensibo/climate.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, SupportsResponse
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import TemperatureConverter
from . import SensiboConfigEntry
@@ -138,7 +138,7 @@ def _find_valid_target_temp(target: float, valid_targets: list[int]) -> int:
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Sensibo climate entry."""
@@ -149,7 +149,8 @@ async def async_setup_entry(
def _add_remove_devices() -> None:
"""Handle additions of devices and sensors."""
nonlocal added_devices
- new_devices, _, added_devices = coordinator.get_devices(added_devices)
+ new_devices, _, new_added_devices = coordinator.get_devices(added_devices)
+ added_devices = new_added_devices
if new_devices:
async_add_entities(
diff --git a/homeassistant/components/sensibo/coordinator.py b/homeassistant/components/sensibo/coordinator.py
index e19f24295b9..3fa8a6e5dae 100644
--- a/homeassistant/components/sensibo/coordinator.py
+++ b/homeassistant/components/sensibo/coordinator.py
@@ -56,18 +56,31 @@ class SensiboDataUpdateCoordinator(DataUpdateCoordinator[SensiboData]):
) -> tuple[set[str], set[str], set[str]]:
"""Addition and removal of devices."""
data = self.data
- motion_sensors = {
+ current_motion_sensors = {
sensor_id
for device_data in data.parsed.values()
if device_data.motion_sensors
for sensor_id in device_data.motion_sensors
}
- devices: set[str] = set(data.parsed)
- new_devices: set[str] = motion_sensors | devices - added_devices
- remove_devices = added_devices - devices - motion_sensors
- added_devices = (added_devices - remove_devices) | new_devices
+ current_devices: set[str] = set(data.parsed)
+ LOGGER.debug(
+ "Current devices: %s, moption sensors: %s",
+ current_devices,
+ current_motion_sensors,
+ )
+ new_devices: set[str] = (
+ current_motion_sensors | current_devices
+ ) - added_devices
+ remove_devices = added_devices - current_devices - current_motion_sensors
+ new_added_devices = (added_devices - remove_devices) | new_devices
- return (new_devices, remove_devices, added_devices)
+ LOGGER.debug(
+ "New devices: %s, Removed devices: %s, Added devices: %s",
+ new_devices,
+ remove_devices,
+ new_added_devices,
+ )
+ return (new_devices, remove_devices, new_added_devices)
async def _async_update_data(self) -> SensiboData:
"""Fetch data from Sensibo."""
diff --git a/homeassistant/components/sensibo/manifest.json b/homeassistant/components/sensibo/manifest.json
index e6398c5076e..610695aaf7b 100644
--- a/homeassistant/components/sensibo/manifest.json
+++ b/homeassistant/components/sensibo/manifest.json
@@ -14,5 +14,6 @@
},
"iot_class": "cloud_polling",
"loggers": ["pysensibo"],
+ "quality_scale": "platinum",
"requirements": ["pysensibo==1.1.0"]
}
diff --git a/homeassistant/components/sensibo/number.py b/homeassistant/components/sensibo/number.py
index aa46c7f8c1e..e71ed6f0235 100644
--- a/homeassistant/components/sensibo/number.py
+++ b/homeassistant/components/sensibo/number.py
@@ -15,7 +15,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SensiboConfigEntry
from .coordinator import SensiboDataUpdateCoordinator
@@ -65,7 +65,7 @@ DEVICE_NUMBER_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensibo number platform."""
@@ -76,7 +76,8 @@ async def async_setup_entry(
def _add_remove_devices() -> None:
"""Handle additions of devices and sensors."""
nonlocal added_devices
- new_devices, _, added_devices = coordinator.get_devices(added_devices)
+ new_devices, _, new_added_devices = coordinator.get_devices(added_devices)
+ added_devices = new_added_devices
if new_devices:
async_add_entities(
diff --git a/homeassistant/components/sensibo/quality_scale.yaml b/homeassistant/components/sensibo/quality_scale.yaml
index c21cf100e9d..3d71d0ad3ba 100644
--- a/homeassistant/components/sensibo/quality_scale.yaml
+++ b/homeassistant/components/sensibo/quality_scale.yaml
@@ -19,9 +19,9 @@ rules:
comment: |
No integrations services.
common-modules: done
- docs-high-level-description: todo
+ docs-high-level-description: done
docs-installation-instructions: done
- docs-removal-instructions: todo
+ docs-removal-instructions: done
docs-actions: done
brands: done
# Silver
@@ -39,9 +39,7 @@ rules:
comment: |
Tests are very complex and needs a rewrite for future additions
integration-owner: done
- docs-installation-parameters:
- status: todo
- comment: configuration_basic
+ docs-installation-parameters: done
docs-configuration-parameters:
status: exempt
comment: |
@@ -71,13 +69,13 @@ rules:
status: exempt
comment: |
This integration doesn't have any cases where raising an issue is needed.
- docs-use-cases: todo
- docs-supported-devices: todo
- docs-supported-functions: todo
- docs-data-update: todo
- docs-known-limitations: todo
- docs-troubleshooting: todo
- docs-examples: todo
+ docs-use-cases: done
+ docs-supported-devices: done
+ docs-supported-functions: done
+ docs-data-update: done
+ docs-known-limitations: done
+ docs-troubleshooting: done
+ docs-examples: done
# Platinum
async-dependency: done
diff --git a/homeassistant/components/sensibo/select.py b/homeassistant/components/sensibo/select.py
index 51521b59f03..5a0546b1aa2 100644
--- a/homeassistant/components/sensibo/select.py
+++ b/homeassistant/components/sensibo/select.py
@@ -17,7 +17,7 @@ from homeassistant.components.select import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import (
IssueSeverity,
async_create_issue,
@@ -67,7 +67,7 @@ DEVICE_SELECT_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensibo select platform."""
@@ -115,7 +115,8 @@ async def async_setup_entry(
def _add_remove_devices() -> None:
"""Handle additions of devices and sensors."""
nonlocal added_devices
- new_devices, _, added_devices = coordinator.get_devices(added_devices)
+ new_devices, _, new_added_devices = coordinator.get_devices(added_devices)
+ added_devices = new_added_devices
if new_devices:
async_add_entities(
diff --git a/homeassistant/components/sensibo/sensor.py b/homeassistant/components/sensibo/sensor.py
index b242f38febe..09f095bfaec 100644
--- a/homeassistant/components/sensibo/sensor.py
+++ b/homeassistant/components/sensibo/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import SensiboConfigEntry
@@ -240,7 +240,7 @@ DESCRIPTION_BY_MODELS = {
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensibo sensor platform."""
@@ -253,9 +253,8 @@ async def async_setup_entry(
entities: list[SensiboMotionSensor | SensiboDeviceSensor] = []
nonlocal added_devices
- new_devices, remove_devices, added_devices = coordinator.get_devices(
- added_devices
- )
+ new_devices, _, new_added_devices = coordinator.get_devices(added_devices)
+ added_devices = new_added_devices
if new_devices:
entities.extend(
diff --git a/homeassistant/components/sensibo/strings.json b/homeassistant/components/sensibo/strings.json
index 6c5210d12bf..6aba2be52fc 100644
--- a/homeassistant/components/sensibo/strings.json
+++ b/homeassistant/components/sensibo/strings.json
@@ -429,16 +429,16 @@
}
},
"enable_pure_boost": {
- "name": "Enable pure boost",
+ "name": "Enable Pure Boost",
"description": "Enables and configures Pure Boost settings.",
"fields": {
"ac_integration": {
"name": "AC integration",
- "description": "Integrate with Air Conditioner."
+ "description": "Integrate with air conditioner."
},
"geo_integration": {
"name": "Geo integration",
- "description": "Integrate with Presence."
+ "description": "Integrate with presence."
},
"indoor_integration": {
"name": "Indoor air quality",
@@ -468,7 +468,7 @@
},
"fan_mode": {
"name": "Fan mode",
- "description": "set fan mode."
+ "description": "Set fan mode."
},
"swing_mode": {
"name": "Swing mode",
diff --git a/homeassistant/components/sensibo/switch.py b/homeassistant/components/sensibo/switch.py
index 0bc2c55a706..03e7c12ec2b 100644
--- a/homeassistant/components/sensibo/switch.py
+++ b/homeassistant/components/sensibo/switch.py
@@ -15,7 +15,7 @@ from homeassistant.components.switch import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SensiboConfigEntry
from .const import DOMAIN
@@ -78,7 +78,7 @@ DESCRIPTION_BY_MODELS = {"pure": PURE_SWITCH_TYPES}
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensibo Switch platform."""
@@ -89,7 +89,8 @@ async def async_setup_entry(
def _add_remove_devices() -> None:
"""Handle additions of devices and sensors."""
nonlocal added_devices
- new_devices, _, added_devices = coordinator.get_devices(added_devices)
+ new_devices, _, new_added_devices = coordinator.get_devices(added_devices)
+ added_devices = new_added_devices
if new_devices:
async_add_entities(
diff --git a/homeassistant/components/sensibo/update.py b/homeassistant/components/sensibo/update.py
index 0b02264b3e0..6f868e5f366 100644
--- a/homeassistant/components/sensibo/update.py
+++ b/homeassistant/components/sensibo/update.py
@@ -14,7 +14,7 @@ from homeassistant.components.update import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SensiboConfigEntry
from .coordinator import SensiboDataUpdateCoordinator
@@ -45,7 +45,7 @@ DEVICE_SENSOR_TYPES: tuple[SensiboDeviceUpdateEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SensiboConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensibo Update platform."""
@@ -56,7 +56,8 @@ async def async_setup_entry(
def _add_remove_devices() -> None:
"""Handle additions of devices and sensors."""
nonlocal added_devices
- new_devices, _, added_devices = coordinator.get_devices(added_devices)
+ new_devices, _, new_added_devices = coordinator.get_devices(added_devices)
+ added_devices = new_added_devices
if new_devices:
async_add_entities(
diff --git a/homeassistant/components/sensirion_ble/sensor.py b/homeassistant/components/sensirion_ble/sensor.py
index a7254fd3609..16f7571f392 100644
--- a/homeassistant/components/sensirion_ble/sensor.py
+++ b/homeassistant/components/sensirion_ble/sensor.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -106,7 +106,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Sensirion BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/sensor/__init__.py b/homeassistant/components/sensor/__init__.py
index 89f39d4fb8c..e3ee566a855 100644
--- a/homeassistant/components/sensor/__init__.py
+++ b/homeassistant/components/sensor/__init__.py
@@ -675,22 +675,13 @@ class SensorEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
):
# Deduce the precision by finding the decimal point, if any
value_s = str(value)
- precision = (
- len(value_s) - value_s.index(".") - 1 if "." in value_s else 0
- )
-
# Scale the precision when converting to a larger unit
# For example 1.1 Wh should be rendered as 0.0011 kWh, not 0.0 kWh
- ratio_log = max(
- 0,
- log10(
- converter.get_unit_ratio(
- native_unit_of_measurement, unit_of_measurement
- )
- ),
+ precision = (
+ len(value_s) - value_s.index(".") - 1 if "." in value_s else 0
+ ) + converter.get_unit_floored_log_ratio(
+ native_unit_of_measurement, unit_of_measurement
)
- precision = precision + floor(ratio_log)
-
value = f"{converted_numerical_value:z.{precision}f}"
else:
value = converted_numerical_value
diff --git a/homeassistant/components/sensor/const.py b/homeassistant/components/sensor/const.py
index c46aca548c8..e1f7dd13d93 100644
--- a/homeassistant/components/sensor/const.py
+++ b/homeassistant/components/sensor/const.py
@@ -11,6 +11,7 @@ from homeassistant.const import (
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
CONCENTRATION_PARTS_PER_BILLION,
CONCENTRATION_PARTS_PER_MILLION,
+ DEGREE,
LIGHT_LUX,
PERCENTAGE,
SIGNAL_STRENGTH_DECIBELS,
@@ -185,7 +186,7 @@ class SensorDeviceClass(StrEnum):
DURATION = "duration"
"""Fixed duration.
- Unit of measurement: `d`, `h`, `min`, `s`, `ms`
+ Unit of measurement: `d`, `h`, `min`, `s`, `ms`, `µs`
"""
ENERGY = "energy"
@@ -454,6 +455,12 @@ class SensorDeviceClass(StrEnum):
- USCS / imperial: `oz`, `lb`
"""
+ WIND_DIRECTION = "wind_direction"
+ """Wind direction.
+
+ Unit of measurement: `°`
+ """
+
WIND_SPEED = "wind_speed"
"""Wind speed.
@@ -551,6 +558,7 @@ DEVICE_CLASS_UNITS: dict[SensorDeviceClass, set[type[StrEnum] | str | None]] = {
UnitOfTime.MINUTES,
UnitOfTime.SECONDS,
UnitOfTime.MILLISECONDS,
+ UnitOfTime.MICROSECONDS,
},
SensorDeviceClass.ENERGY: set(UnitOfEnergy),
SensorDeviceClass.ENERGY_DISTANCE: set(UnitOfEnergyDistance),
@@ -575,6 +583,7 @@ DEVICE_CLASS_UNITS: dict[SensorDeviceClass, set[type[StrEnum] | str | None]] = {
SensorDeviceClass.PM25: {CONCENTRATION_MICROGRAMS_PER_CUBIC_METER},
SensorDeviceClass.POWER_FACTOR: {PERCENTAGE, None},
SensorDeviceClass.POWER: {
+ UnitOfPower.MILLIWATT,
UnitOfPower.WATT,
UnitOfPower.KILO_WATT,
UnitOfPower.MEGA_WATT,
@@ -612,6 +621,7 @@ DEVICE_CLASS_UNITS: dict[SensorDeviceClass, set[type[StrEnum] | str | None]] = {
UnitOfVolume.LITERS,
},
SensorDeviceClass.WEIGHT: set(UnitOfMass),
+ SensorDeviceClass.WIND_DIRECTION: {DEGREE},
SensorDeviceClass.WIND_SPEED: set(UnitOfSpeed),
}
@@ -683,5 +693,6 @@ DEVICE_CLASS_STATE_CLASSES: dict[SensorDeviceClass, set[SensorStateClass]] = {
SensorStateClass.TOTAL,
SensorStateClass.TOTAL_INCREASING,
},
+ SensorDeviceClass.WIND_DIRECTION: set(),
SensorDeviceClass.WIND_SPEED: {SensorStateClass.MEASUREMENT},
}
diff --git a/homeassistant/components/sensor/device_condition.py b/homeassistant/components/sensor/device_condition.py
index 4a68fbabe8f..f52393f28ff 100644
--- a/homeassistant/components/sensor/device_condition.py
+++ b/homeassistant/components/sensor/device_condition.py
@@ -83,6 +83,7 @@ CONF_IS_VOLUME = "is_volume"
CONF_IS_VOLUME_FLOW_RATE = "is_volume_flow_rate"
CONF_IS_WATER = "is_water"
CONF_IS_WEIGHT = "is_weight"
+CONF_IS_WIND_DIRECTION = "is_wind_direction"
CONF_IS_WIND_SPEED = "is_wind_speed"
ENTITY_CONDITIONS = {
@@ -145,6 +146,7 @@ ENTITY_CONDITIONS = {
SensorDeviceClass.VOLUME_FLOW_RATE: [{CONF_TYPE: CONF_IS_VOLUME_FLOW_RATE}],
SensorDeviceClass.WATER: [{CONF_TYPE: CONF_IS_WATER}],
SensorDeviceClass.WEIGHT: [{CONF_TYPE: CONF_IS_WEIGHT}],
+ SensorDeviceClass.WIND_DIRECTION: [{CONF_TYPE: CONF_IS_WIND_DIRECTION}],
SensorDeviceClass.WIND_SPEED: [{CONF_TYPE: CONF_IS_WIND_SPEED}],
DEVICE_CLASS_NONE: [{CONF_TYPE: CONF_IS_VALUE}],
}
@@ -204,6 +206,7 @@ CONDITION_SCHEMA = vol.All(
CONF_IS_VOLUME_FLOW_RATE,
CONF_IS_WATER,
CONF_IS_WEIGHT,
+ CONF_IS_WIND_DIRECTION,
CONF_IS_WIND_SPEED,
CONF_IS_VALUE,
]
diff --git a/homeassistant/components/sensor/device_trigger.py b/homeassistant/components/sensor/device_trigger.py
index 0003b83d05a..dee48434294 100644
--- a/homeassistant/components/sensor/device_trigger.py
+++ b/homeassistant/components/sensor/device_trigger.py
@@ -82,6 +82,7 @@ CONF_VOLUME = "volume"
CONF_VOLUME_FLOW_RATE = "volume_flow_rate"
CONF_WATER = "water"
CONF_WEIGHT = "weight"
+CONF_WIND_DIRECTION = "wind_direction"
CONF_WIND_SPEED = "wind_speed"
ENTITY_TRIGGERS = {
@@ -144,6 +145,7 @@ ENTITY_TRIGGERS = {
SensorDeviceClass.VOLUME_FLOW_RATE: [{CONF_TYPE: CONF_VOLUME_FLOW_RATE}],
SensorDeviceClass.WATER: [{CONF_TYPE: CONF_WATER}],
SensorDeviceClass.WEIGHT: [{CONF_TYPE: CONF_WEIGHT}],
+ SensorDeviceClass.WIND_DIRECTION: [{CONF_TYPE: CONF_WIND_DIRECTION}],
SensorDeviceClass.WIND_SPEED: [{CONF_TYPE: CONF_WIND_SPEED}],
DEVICE_CLASS_NONE: [{CONF_TYPE: CONF_VALUE}],
}
@@ -204,6 +206,7 @@ TRIGGER_SCHEMA = vol.All(
CONF_VOLUME_FLOW_RATE,
CONF_WATER,
CONF_WEIGHT,
+ CONF_WIND_DIRECTION,
CONF_WIND_SPEED,
CONF_VALUE,
]
diff --git a/homeassistant/components/sensor/icons.json b/homeassistant/components/sensor/icons.json
index 5f770765ee3..497c1544b3b 100644
--- a/homeassistant/components/sensor/icons.json
+++ b/homeassistant/components/sensor/icons.json
@@ -156,6 +156,9 @@
"weight": {
"default": "mdi:weight"
},
+ "wind_direction": {
+ "default": "mdi:compass-rose"
+ },
"wind_speed": {
"default": "mdi:weather-windy"
}
diff --git a/homeassistant/components/sensor/recorder.py b/homeassistant/components/sensor/recorder.py
index 675d24b9240..4e8e27e0c79 100644
--- a/homeassistant/components/sensor/recorder.py
+++ b/homeassistant/components/sensor/recorder.py
@@ -134,16 +134,7 @@ def _time_weighted_average(
duration = end - old_start_time
accumulated += old_fstate * duration.total_seconds()
- period_seconds = (end - start).total_seconds()
- if period_seconds == 0:
- # If the only state changed that happened was at the exact moment
- # at the end of the period, we can't calculate a meaningful average
- # so we return 0.0 since it represents a time duration smaller than
- # we can measure. This probably means the precision of statistics
- # column schema in the database is incorrect but it is actually possible
- # to happen if the state change event fired at the exact microsecond
- return 0.0
- return accumulated / period_seconds
+ return accumulated / (end - start).total_seconds()
def _get_units(fstates: list[tuple[float, State]]) -> set[str | None]:
@@ -447,7 +438,11 @@ def compile_statistics( # noqa: C901
entity_id = _state.entity_id
# If there are no recent state changes, the sensor's state may already be pruned
# from the recorder. Get the state from the state machine instead.
- if not (entity_history := history_list.get(entity_id, [_state])):
+ try:
+ entity_history = history_list[entity_id]
+ except KeyError:
+ entity_history = [_state] if _state.last_changed < end else []
+ if not entity_history:
continue
if not (float_states := _entity_history_to_float_and_state(entity_history)):
continue
diff --git a/homeassistant/components/sensor/strings.json b/homeassistant/components/sensor/strings.json
index dcbb4d3c826..ae414a178e9 100644
--- a/homeassistant/components/sensor/strings.json
+++ b/homeassistant/components/sensor/strings.json
@@ -52,6 +52,7 @@
"is_volume_flow_rate": "Current {entity_name} volume flow rate",
"is_water": "Current {entity_name} water",
"is_weight": "Current {entity_name} weight",
+ "is_wind_direction": "Current {entity_name} wind direction",
"is_wind_speed": "Current {entity_name} wind speed"
},
"trigger_type": {
@@ -105,6 +106,7 @@
"volume_flow_rate": "{entity_name} volume flow rate changes",
"water": "{entity_name} water changes",
"weight": "{entity_name} weight changes",
+ "wind_direction": "{entity_name} wind direction changes",
"wind_speed": "{entity_name} wind speed changes"
},
"extra_fields": {
@@ -299,6 +301,9 @@
"weight": {
"name": "Weight"
},
+ "wind_direction": {
+ "name": "Wind direction"
+ },
"wind_speed": {
"name": "Wind speed"
}
diff --git a/homeassistant/components/sensorpro/sensor.py b/homeassistant/components/sensorpro/sensor.py
index b972aac04fb..997fa0db995 100644
--- a/homeassistant/components/sensorpro/sensor.py
+++ b/homeassistant/components/sensorpro/sensor.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -111,7 +111,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SensorPro BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/sensorpush/sensor.py b/homeassistant/components/sensorpush/sensor.py
index 6eea5c10f78..730277350b5 100644
--- a/homeassistant/components/sensorpush/sensor.py
+++ b/homeassistant/components/sensorpush/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from . import SensorPushConfigEntry
@@ -97,7 +97,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: SensorPushConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SensorPush BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/sensorpush_cloud/__init__.py b/homeassistant/components/sensorpush_cloud/__init__.py
new file mode 100644
index 00000000000..2d9d299c132
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/__init__.py
@@ -0,0 +1,28 @@
+"""The SensorPush Cloud integration."""
+
+from __future__ import annotations
+
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant
+
+from .coordinator import SensorPushCloudConfigEntry, SensorPushCloudCoordinator
+
+PLATFORMS: list[Platform] = [Platform.SENSOR]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant, entry: SensorPushCloudConfigEntry
+) -> bool:
+ """Set up SensorPush Cloud from a config entry."""
+ coordinator = SensorPushCloudCoordinator(hass, entry)
+ entry.runtime_data = coordinator
+ await coordinator.async_config_entry_first_refresh()
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+ return True
+
+
+async def async_unload_entry(
+ hass: HomeAssistant, entry: SensorPushCloudConfigEntry
+) -> bool:
+ """Unload a config entry."""
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/sensorpush_cloud/config_flow.py b/homeassistant/components/sensorpush_cloud/config_flow.py
new file mode 100644
index 00000000000..d06fde2eba1
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/config_flow.py
@@ -0,0 +1,64 @@
+"""Config flow for the SensorPush Cloud integration."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from sensorpush_ha import SensorPushCloudApi, SensorPushCloudAuthError
+import voluptuous as vol
+
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.selector import (
+ TextSelector,
+ TextSelectorConfig,
+ TextSelectorType,
+)
+
+from .const import DOMAIN, LOGGER
+
+
+class SensorPushCloudConfigFlow(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow for SensorPush Cloud."""
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle the initial step."""
+ errors: dict[str, str] = {}
+ if user_input is not None:
+ email, password = user_input[CONF_EMAIL], user_input[CONF_PASSWORD]
+ await self.async_set_unique_id(email)
+ self._abort_if_unique_id_configured()
+ clientsession = async_get_clientsession(self.hass)
+ api = SensorPushCloudApi(email, password, clientsession)
+ try:
+ await api.async_authorize()
+ except SensorPushCloudAuthError:
+ errors["base"] = "invalid_auth"
+ except Exception: # noqa: BLE001
+ LOGGER.exception("Unexpected error")
+ errors["base"] = "unknown"
+ else:
+ return self.async_create_entry(title=email, data=user_input)
+
+ return self.async_show_form(
+ step_id="user",
+ data_schema=vol.Schema(
+ {
+ vol.Required(CONF_EMAIL): TextSelector(
+ TextSelectorConfig(
+ type=TextSelectorType.EMAIL, autocomplete="username"
+ )
+ ),
+ vol.Required(CONF_PASSWORD): TextSelector(
+ TextSelectorConfig(
+ type=TextSelectorType.PASSWORD,
+ autocomplete="current-password",
+ )
+ ),
+ }
+ ),
+ errors=errors,
+ )
diff --git a/homeassistant/components/sensorpush_cloud/const.py b/homeassistant/components/sensorpush_cloud/const.py
new file mode 100644
index 00000000000..9e66dacfaba
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/const.py
@@ -0,0 +1,12 @@
+"""Constants for the SensorPush Cloud integration."""
+
+from datetime import timedelta
+import logging
+from typing import Final
+
+LOGGER = logging.getLogger(__package__)
+
+DOMAIN: Final = "sensorpush_cloud"
+
+UPDATE_INTERVAL: Final = timedelta(seconds=60)
+MAX_TIME_BETWEEN_UPDATES: Final = UPDATE_INTERVAL * 60
diff --git a/homeassistant/components/sensorpush_cloud/coordinator.py b/homeassistant/components/sensorpush_cloud/coordinator.py
new file mode 100644
index 00000000000..9885538b55a
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/coordinator.py
@@ -0,0 +1,45 @@
+"""Coordinator for the SensorPush Cloud integration."""
+
+from __future__ import annotations
+
+from sensorpush_ha import (
+ SensorPushCloudApi,
+ SensorPushCloudData,
+ SensorPushCloudError,
+ SensorPushCloudHelper,
+)
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+from .const import LOGGER, UPDATE_INTERVAL
+
+type SensorPushCloudConfigEntry = ConfigEntry[SensorPushCloudCoordinator]
+
+
+class SensorPushCloudCoordinator(DataUpdateCoordinator[dict[str, SensorPushCloudData]]):
+ """SensorPush Cloud coordinator."""
+
+ def __init__(self, hass: HomeAssistant, entry: SensorPushCloudConfigEntry) -> None:
+ """Initialize the coordinator."""
+ super().__init__(
+ hass,
+ LOGGER,
+ name=entry.title,
+ update_interval=UPDATE_INTERVAL,
+ config_entry=entry,
+ )
+ email, password = entry.data[CONF_EMAIL], entry.data[CONF_PASSWORD]
+ clientsession = async_get_clientsession(hass)
+ api = SensorPushCloudApi(email, password, clientsession)
+ self.helper = SensorPushCloudHelper(api)
+
+ async def _async_update_data(self) -> dict[str, SensorPushCloudData]:
+ """Fetch data from API endpoints."""
+ try:
+ return await self.helper.async_get_data()
+ except SensorPushCloudError as e:
+ raise UpdateFailed(e) from e
diff --git a/homeassistant/components/sensorpush_cloud/manifest.json b/homeassistant/components/sensorpush_cloud/manifest.json
new file mode 100644
index 00000000000..ad817251fa1
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/manifest.json
@@ -0,0 +1,11 @@
+{
+ "domain": "sensorpush_cloud",
+ "name": "SensorPush Cloud",
+ "codeowners": ["@sstallion"],
+ "config_flow": true,
+ "documentation": "https://www.home-assistant.io/integrations/sensorpush_cloud",
+ "iot_class": "cloud_polling",
+ "loggers": ["sensorpush_api", "sensorpush_ha"],
+ "quality_scale": "bronze",
+ "requirements": ["sensorpush-api==2.1.1", "sensorpush-ha==1.3.2"]
+}
diff --git a/homeassistant/components/sensorpush_cloud/quality_scale.yaml b/homeassistant/components/sensorpush_cloud/quality_scale.yaml
new file mode 100644
index 00000000000..96816e1d50d
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/quality_scale.yaml
@@ -0,0 +1,68 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: Integration does not register custom actions.
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: Integration does not register custom actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup: done
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions:
+ status: exempt
+ comment: Integration does not register custom actions.
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: Integration does not support options flow.
+ docs-installation-parameters: todo
+ entity-unavailable: done
+ integration-owner: done
+ log-when-unavailable: done
+ parallel-updates: done
+ reauthentication-flow: todo
+ test-coverage: todo
+
+ # Gold
+ devices: done
+ diagnostics: todo
+ discovery-update-info: todo
+ discovery: todo
+ docs-data-update: todo
+ docs-examples: todo
+ docs-known-limitations: done
+ docs-supported-devices: done
+ docs-supported-functions: done
+ docs-troubleshooting: todo
+ docs-use-cases: todo
+ dynamic-devices: todo
+ entity-category: todo
+ entity-device-class: done
+ entity-disabled-by-default: done
+ entity-translations: todo
+ exception-translations: todo
+ icon-translations: todo
+ reconfiguration-flow: todo
+ repair-issues: todo
+ stale-devices: todo
+
+ # Platinum
+ async-dependency: done
+ inject-websession: done
+ strict-typing: done
diff --git a/homeassistant/components/sensorpush_cloud/sensor.py b/homeassistant/components/sensorpush_cloud/sensor.py
new file mode 100644
index 00000000000..d2855f63a62
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/sensor.py
@@ -0,0 +1,158 @@
+"""Support for SensorPush Cloud sensors."""
+
+from __future__ import annotations
+
+from typing import Final
+
+from homeassistant.components.sensor import (
+ SensorDeviceClass,
+ SensorEntity,
+ SensorEntityDescription,
+ SensorStateClass,
+)
+from homeassistant.const import (
+ ATTR_TEMPERATURE,
+ PERCENTAGE,
+ SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+ UnitOfElectricPotential,
+ UnitOfLength,
+ UnitOfPressure,
+ UnitOfTemperature,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.typing import StateType
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+from homeassistant.util import dt as dt_util
+
+from .const import DOMAIN, MAX_TIME_BETWEEN_UPDATES
+from .coordinator import SensorPushCloudConfigEntry, SensorPushCloudCoordinator
+
+ATTR_ALTITUDE: Final = "altitude"
+ATTR_ATMOSPHERIC_PRESSURE: Final = "atmospheric_pressure"
+ATTR_BATTERY_VOLTAGE: Final = "battery_voltage"
+ATTR_DEWPOINT: Final = "dewpoint"
+ATTR_HUMIDITY: Final = "humidity"
+ATTR_SIGNAL_STRENGTH: Final = "signal_strength"
+ATTR_VAPOR_PRESSURE: Final = "vapor_pressure"
+
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
+
+SENSORS: Final[tuple[SensorEntityDescription, ...]] = (
+ SensorEntityDescription(
+ key=ATTR_ALTITUDE,
+ device_class=SensorDeviceClass.DISTANCE,
+ entity_registry_enabled_default=False,
+ translation_key="altitude",
+ native_unit_of_measurement=UnitOfLength.FEET,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=ATTR_ATMOSPHERIC_PRESSURE,
+ device_class=SensorDeviceClass.ATMOSPHERIC_PRESSURE,
+ entity_registry_enabled_default=False,
+ native_unit_of_measurement=UnitOfPressure.INHG,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=ATTR_BATTERY_VOLTAGE,
+ device_class=SensorDeviceClass.VOLTAGE,
+ entity_registry_enabled_default=False,
+ translation_key="battery_voltage",
+ native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=ATTR_DEWPOINT,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ entity_registry_enabled_default=False,
+ translation_key="dewpoint",
+ native_unit_of_measurement=UnitOfTemperature.FAHRENHEIT,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=ATTR_HUMIDITY,
+ device_class=SensorDeviceClass.HUMIDITY,
+ native_unit_of_measurement=PERCENTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=ATTR_SIGNAL_STRENGTH,
+ device_class=SensorDeviceClass.SIGNAL_STRENGTH,
+ entity_registry_enabled_default=False,
+ native_unit_of_measurement=SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=ATTR_TEMPERATURE,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ native_unit_of_measurement=UnitOfTemperature.FAHRENHEIT,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ SensorEntityDescription(
+ key=ATTR_VAPOR_PRESSURE,
+ device_class=SensorDeviceClass.PRESSURE,
+ entity_registry_enabled_default=False,
+ translation_key="vapor_pressure",
+ native_unit_of_measurement=UnitOfPressure.KPA,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SensorPushCloudConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up SensorPush Cloud sensors."""
+ coordinator = entry.runtime_data
+ async_add_entities(
+ SensorPushCloudSensor(coordinator, entity_description, device_id)
+ for entity_description in SENSORS
+ for device_id in coordinator.data
+ )
+
+
+class SensorPushCloudSensor(
+ CoordinatorEntity[SensorPushCloudCoordinator], SensorEntity
+):
+ """SensorPush Cloud sensor."""
+
+ _attr_has_entity_name = True
+
+ def __init__(
+ self,
+ coordinator: SensorPushCloudCoordinator,
+ entity_description: SensorEntityDescription,
+ device_id: str,
+ ) -> None:
+ """Initialize the sensor."""
+ super().__init__(coordinator)
+ self.entity_description = entity_description
+ self.device_id = device_id
+
+ device = coordinator.data[device_id]
+ self._attr_unique_id = f"{device.device_id}_{entity_description.key}"
+ self._attr_device_info = DeviceInfo(
+ identifiers={(DOMAIN, device.device_id)},
+ manufacturer=device.manufacturer,
+ model=device.model,
+ name=device.name,
+ )
+
+ @property
+ def available(self) -> bool:
+ """Return true if entity is available."""
+ if self.device_id in self.coordinator.data:
+ last_update = self.coordinator.data[self.device_id].last_update
+ if dt_util.utcnow() >= (last_update + MAX_TIME_BETWEEN_UPDATES):
+ return False
+ return super().available
+
+ @property
+ def native_value(self) -> StateType:
+ """Return the value reported by the sensor."""
+ return self.coordinator.data[self.device_id][self.entity_description.key]
diff --git a/homeassistant/components/sensorpush_cloud/strings.json b/homeassistant/components/sensorpush_cloud/strings.json
new file mode 100644
index 00000000000..8467a123b6f
--- /dev/null
+++ b/homeassistant/components/sensorpush_cloud/strings.json
@@ -0,0 +1,40 @@
+{
+ "config": {
+ "step": {
+ "user": {
+ "description": "To activate API access, log in to the [Gateway Cloud Dashboard](https://dashboard.sensorpush.com/) and agree to the terms of service. Devices are not available until activated with the SensorPush app on iOS or Android.",
+ "data": {
+ "email": "[%key:common::config_flow::data::email%]",
+ "password": "[%key:common::config_flow::data::password%]"
+ },
+ "data_description": {
+ "email": "The email address used to log in to the SensorPush Gateway Cloud Dashboard",
+ "password": "The password used to log in to the SensorPush Gateway Cloud Dashboard"
+ }
+ }
+ },
+ "error": {
+ "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
+ "unknown": "[%key:common::config_flow::error::unknown%]"
+ },
+ "abort": {
+ "already_configured": "[%key:common::config_flow::abort::already_configured_account%]"
+ }
+ },
+ "entity": {
+ "sensor": {
+ "altitude": {
+ "name": "Altitude"
+ },
+ "battery_voltage": {
+ "name": "Battery voltage"
+ },
+ "dewpoint": {
+ "name": "Dew point"
+ },
+ "vapor_pressure": {
+ "name": "Vapor pressure"
+ }
+ }
+ }
+}
diff --git a/homeassistant/components/sensoterra/__init__.py b/homeassistant/components/sensoterra/__init__.py
index b1428351f09..1559dc10c43 100644
--- a/homeassistant/components/sensoterra/__init__.py
+++ b/homeassistant/components/sensoterra/__init__.py
@@ -4,16 +4,13 @@ from __future__ import annotations
from sensoterra.customerapi import CustomerApi
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TOKEN, Platform
from homeassistant.core import HomeAssistant
-from .coordinator import SensoterraCoordinator
+from .coordinator import SensoterraConfigEntry, SensoterraCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
-type SensoterraConfigEntry = ConfigEntry[SensoterraCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: SensoterraConfigEntry) -> bool:
"""Set up Sensoterra platform based on a configuration entry."""
@@ -24,7 +21,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: SensoterraConfigEntry) -
api.set_language(hass.config.language)
api.set_token(entry.data[CONF_TOKEN])
- coordinator = SensoterraCoordinator(hass, api)
+ coordinator = SensoterraCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/sensoterra/coordinator.py b/homeassistant/components/sensoterra/coordinator.py
index 2dffdceb443..9020633a2a3 100644
--- a/homeassistant/components/sensoterra/coordinator.py
+++ b/homeassistant/components/sensoterra/coordinator.py
@@ -10,21 +10,29 @@ from sensoterra.customerapi import (
)
from sensoterra.probe import Probe, Sensor
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import LOGGER, SCAN_INTERVAL_MINUTES
+type SensoterraConfigEntry = ConfigEntry[SensoterraCoordinator]
+
class SensoterraCoordinator(DataUpdateCoordinator[list[Probe]]):
"""Sensoterra coordinator."""
- def __init__(self, hass: HomeAssistant, api: CustomerApi) -> None:
+ config_entry: SensoterraConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: SensoterraConfigEntry, api: CustomerApi
+ ) -> None:
"""Initialize Sensoterra coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Sensoterra probe",
update_interval=timedelta(minutes=SCAN_INTERVAL_MINUTES),
)
diff --git a/homeassistant/components/sensoterra/sensor.py b/homeassistant/components/sensoterra/sensor.py
index 7e9f4d0840e..56f47ade212 100644
--- a/homeassistant/components/sensoterra/sensor.py
+++ b/homeassistant/components/sensoterra/sensor.py
@@ -21,13 +21,12 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import SensoterraConfigEntry
from .const import CONFIGURATION_URL, DOMAIN, SENSOR_EXPIRATION_DAYS
-from .coordinator import SensoterraCoordinator
+from .coordinator import SensoterraConfigEntry, SensoterraCoordinator
class ProbeSensorType(StrEnum):
@@ -85,7 +84,7 @@ SENSORS: dict[ProbeSensorType, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: SensoterraConfigEntry,
- async_add_devices: AddEntitiesCallback,
+ async_add_devices: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sensoterra sensor."""
diff --git a/homeassistant/components/sentry/manifest.json b/homeassistant/components/sentry/manifest.json
index 425225e07ef..4c3a7518085 100644
--- a/homeassistant/components/sentry/manifest.json
+++ b/homeassistant/components/sentry/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/sentry",
"integration_type": "service",
"iot_class": "cloud_polling",
- "requirements": ["sentry-sdk==1.40.3"]
+ "requirements": ["sentry-sdk==1.45.1"]
}
diff --git a/homeassistant/components/senz/climate.py b/homeassistant/components/senz/climate.py
index d5749a3f040..48eeee54974 100644
--- a/homeassistant/components/senz/climate.py
+++ b/homeassistant/components/senz/climate.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_TENTHS, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import SENZDataUpdateCoordinator
@@ -26,7 +26,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SENZ climate entities from a config entry."""
coordinator: SENZDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/seventeentrack/__init__.py b/homeassistant/components/seventeentrack/__init__.py
index 695ca179966..235a5338cb6 100644
--- a/homeassistant/components/seventeentrack/__init__.py
+++ b/homeassistant/components/seventeentrack/__init__.py
@@ -39,7 +39,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except SeventeenTrackError as err:
raise ConfigEntryNotReady from err
- seventeen_coordinator = SeventeenTrackCoordinator(hass, client)
+ seventeen_coordinator = SeventeenTrackCoordinator(hass, entry, client)
await seventeen_coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/seventeentrack/coordinator.py b/homeassistant/components/seventeentrack/coordinator.py
index 3e27f9f0369..107f1d48a21 100644
--- a/homeassistant/components/seventeentrack/coordinator.py
+++ b/homeassistant/components/seventeentrack/coordinator.py
@@ -34,11 +34,17 @@ class SeventeenTrackCoordinator(DataUpdateCoordinator[SeventeenTrackData]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, client: SeventeenTrackClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ client: SeventeenTrackClient,
+ ) -> None:
"""Initialize."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=DEFAULT_SCAN_INTERVAL,
)
diff --git a/homeassistant/components/seventeentrack/icons.json b/homeassistant/components/seventeentrack/icons.json
index a5cac0a9f84..c48e147e973 100644
--- a/homeassistant/components/seventeentrack/icons.json
+++ b/homeassistant/components/seventeentrack/icons.json
@@ -19,7 +19,7 @@
"delivered": {
"default": "mdi:package"
},
- "returned": {
+ "alert": {
"default": "mdi:package"
},
"package": {
diff --git a/homeassistant/components/seventeentrack/manifest.json b/homeassistant/components/seventeentrack/manifest.json
index a130fbe9aee..34019208a14 100644
--- a/homeassistant/components/seventeentrack/manifest.json
+++ b/homeassistant/components/seventeentrack/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "service",
"iot_class": "cloud_polling",
"loggers": ["pyseventeentrack"],
- "requirements": ["pyseventeentrack==1.0.1"]
+ "requirements": ["pyseventeentrack==1.0.2"]
}
diff --git a/homeassistant/components/seventeentrack/sensor.py b/homeassistant/components/seventeentrack/sensor.py
index dade9efb67c..b0f9d6cd2bd 100644
--- a/homeassistant/components/seventeentrack/sensor.py
+++ b/homeassistant/components/seventeentrack/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_FRIENDLY_NAME, ATTR_LOCATION
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -28,7 +28,7 @@ from .const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a 17Track sensor entry."""
diff --git a/homeassistant/components/seventeentrack/services.yaml b/homeassistant/components/seventeentrack/services.yaml
index d4592dc8aab..45d7c0a530a 100644
--- a/homeassistant/components/seventeentrack/services.yaml
+++ b/homeassistant/components/seventeentrack/services.yaml
@@ -11,7 +11,7 @@ get_packages:
- "ready_to_be_picked_up"
- "undelivered"
- "delivered"
- - "returned"
+ - "alert"
translation_key: package_state
config_entry_id:
required: true
diff --git a/homeassistant/components/seventeentrack/strings.json b/homeassistant/components/seventeentrack/strings.json
index 982b15ab629..c95a553ae7b 100644
--- a/homeassistant/components/seventeentrack/strings.json
+++ b/homeassistant/components/seventeentrack/strings.json
@@ -57,8 +57,8 @@
"delivered": {
"name": "Delivered"
},
- "returned": {
- "name": "Returned"
+ "alert": {
+ "name": "Alert"
},
"package": {
"name": "Package {name}"
@@ -68,7 +68,7 @@
"services": {
"get_packages": {
"name": "Get packages",
- "description": "Get packages from 17Track",
+ "description": "Queries the 17track API for the latest package data.",
"fields": {
"package_state": {
"name": "Package states",
@@ -82,7 +82,7 @@
},
"archive_package": {
"name": "Archive package",
- "description": "Archive a package",
+ "description": "Archives a package using the 17track API.",
"fields": {
"package_tracking_number": {
"name": "Package tracking number",
@@ -104,7 +104,7 @@
"ready_to_be_picked_up": "[%key:component::seventeentrack::entity::sensor::ready_to_be_picked_up::name%]",
"undelivered": "[%key:component::seventeentrack::entity::sensor::undelivered::name%]",
"delivered": "[%key:component::seventeentrack::entity::sensor::delivered::name%]",
- "returned": "[%key:component::seventeentrack::entity::sensor::returned::name%]"
+ "alert": "[%key:component::seventeentrack::entity::sensor::alert::name%]"
}
}
}
diff --git a/homeassistant/components/sfr_box/__init__.py b/homeassistant/components/sfr_box/__init__.py
index 927e3cb0ef2..a56d208d515 100644
--- a/homeassistant/components/sfr_box/__init__.py
+++ b/homeassistant/components/sfr_box/__init__.py
@@ -37,12 +37,18 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
data = DomainData(
box=box,
- dsl=SFRDataUpdateCoordinator(hass, box, "dsl", lambda b: b.dsl_get_info()),
- ftth=SFRDataUpdateCoordinator(hass, box, "ftth", lambda b: b.ftth_get_info()),
- system=SFRDataUpdateCoordinator(
- hass, box, "system", lambda b: b.system_get_info()
+ dsl=SFRDataUpdateCoordinator(
+ hass, entry, box, "dsl", lambda b: b.dsl_get_info()
+ ),
+ ftth=SFRDataUpdateCoordinator(
+ hass, entry, box, "ftth", lambda b: b.ftth_get_info()
+ ),
+ system=SFRDataUpdateCoordinator(
+ hass, entry, box, "system", lambda b: b.system_get_info()
+ ),
+ wan=SFRDataUpdateCoordinator(
+ hass, entry, box, "wan", lambda b: b.wan_get_info()
),
- wan=SFRDataUpdateCoordinator(hass, box, "wan", lambda b: b.wan_get_info()),
)
# Preload system information
await data.system.async_config_entry_first_refresh()
diff --git a/homeassistant/components/sfr_box/binary_sensor.py b/homeassistant/components/sfr_box/binary_sensor.py
index 4ef5e87761d..de40291b0b6 100644
--- a/homeassistant/components/sfr_box/binary_sensor.py
+++ b/homeassistant/components/sfr_box/binary_sensor.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -62,7 +62,9 @@ WAN_SENSOR_TYPES: tuple[SFRBoxBinarySensorEntityDescription[WanInfo], ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
data: DomainData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sfr_box/button.py b/homeassistant/components/sfr_box/button.py
index bddb1e8f926..9798602ef6b 100644
--- a/homeassistant/components/sfr_box/button.py
+++ b/homeassistant/components/sfr_box/button.py
@@ -21,7 +21,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .models import DomainData
@@ -65,7 +65,9 @@ BUTTON_TYPES: tuple[SFRBoxButtonEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the buttons."""
data: DomainData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sfr_box/coordinator.py b/homeassistant/components/sfr_box/coordinator.py
index 5877d5a454a..e9cb3c592e1 100644
--- a/homeassistant/components/sfr_box/coordinator.py
+++ b/homeassistant/components/sfr_box/coordinator.py
@@ -8,6 +8,7 @@ from typing import Any
from sfrbox_api.bridge import SFRBox
from sfrbox_api.exceptions import SFRBoxError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -18,9 +19,12 @@ _SCAN_INTERVAL = timedelta(minutes=1)
class SFRDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT | None]):
"""Coordinator to manage data updates."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
box: SFRBox,
name: str,
method: Callable[[SFRBox], Coroutine[Any, Any, _DataT | None]],
@@ -28,7 +32,13 @@ class SFRDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT | None]):
"""Initialize coordinator."""
self.box = box
self._method = method
- super().__init__(hass, _LOGGER, name=name, update_interval=_SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=name,
+ update_interval=_SCAN_INTERVAL,
+ )
async def _async_update_data(self) -> _DataT | None:
"""Update data."""
diff --git a/homeassistant/components/sfr_box/sensor.py b/homeassistant/components/sfr_box/sensor.py
index ee3285a8f38..8b495da56c3 100644
--- a/homeassistant/components/sfr_box/sensor.py
+++ b/homeassistant/components/sfr_box/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -123,7 +123,7 @@ DSL_SENSOR_TYPES: tuple[SFRBoxSensorEntityDescription[DslInfo], ...] = (
entity_registry_enabled_default=False,
options=[
"no_defect",
- "of_frame",
+ "loss_of_frame",
"loss_of_signal",
"loss_of_power",
"loss_of_signal_quality",
@@ -217,7 +217,9 @@ def _get_temperature(value: float | None) -> float | None:
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
data: DomainData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sfr_box/strings.json b/homeassistant/components/sfr_box/strings.json
index 6f0001e97ce..35e9b1869ff 100644
--- a/homeassistant/components/sfr_box/strings.json
+++ b/homeassistant/components/sfr_box/strings.json
@@ -64,11 +64,11 @@
"dsl_line_status": {
"name": "DSL line status",
"state": {
- "no_defect": "No Defect",
- "of_frame": "Of Frame",
- "loss_of_signal": "Loss Of Signal",
- "loss_of_power": "Loss Of Power",
- "loss_of_signal_quality": "Loss Of Signal Quality",
+ "no_defect": "No defect",
+ "loss_of_frame": "Loss of frame",
+ "loss_of_signal": "Loss of signal",
+ "loss_of_power": "Loss of power",
+ "loss_of_signal_quality": "Loss of signal quality",
"unknown": "Unknown"
}
},
diff --git a/homeassistant/components/sharkiq/coordinator.py b/homeassistant/components/sharkiq/coordinator.py
index 381f6ca1a7d..1a4a819cdf6 100644
--- a/homeassistant/components/sharkiq/coordinator.py
+++ b/homeassistant/components/sharkiq/coordinator.py
@@ -24,6 +24,8 @@ from .const import API_TIMEOUT, DOMAIN, LOGGER, UPDATE_INTERVAL
class SharkIqUpdateCoordinator(DataUpdateCoordinator[bool]):
"""Define a wrapper class to update Shark IQ data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
@@ -36,10 +38,15 @@ class SharkIqUpdateCoordinator(DataUpdateCoordinator[bool]):
self.shark_vacs: dict[str, SharkIqVacuum] = {
sharkiq.serial_number: sharkiq for sharkiq in shark_vacs
}
- self._config_entry = config_entry
self._online_dsns: set[str] = set()
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=UPDATE_INTERVAL,
+ )
@property
def online_dsns(self) -> set[str]:
diff --git a/homeassistant/components/sharkiq/vacuum.py b/homeassistant/components/sharkiq/vacuum.py
index 332d95b0a3e..daea195a770 100644
--- a/homeassistant/components/sharkiq/vacuum.py
+++ b/homeassistant/components/sharkiq/vacuum.py
@@ -18,7 +18,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, LOGGER, SERVICE_CLEAN_ROOM, SHARK
@@ -50,7 +50,7 @@ ATTR_ROOMS = "rooms"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Shark IQ vacuum cleaner."""
coordinator: SharkIqUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/shelly/__init__.py b/homeassistant/components/shelly/__init__.py
index 5ca58ec7d01..a7ee1c029df 100644
--- a/homeassistant/components/shelly/__init__.py
+++ b/homeassistant/components/shelly/__init__.py
@@ -4,6 +4,7 @@ from __future__ import annotations
from typing import Final
+from aioshelly.ble.const import BLE_SCRIPT_NAME
from aioshelly.block_device import BlockDevice
from aioshelly.common import ConnectionOptions
from aioshelly.const import DEFAULT_COAP_PORT, RPC_GENERATIONS
@@ -11,12 +12,19 @@ from aioshelly.exceptions import (
DeviceConnectionError,
InvalidAuthError,
MacAddressMismatchError,
+ RpcCallError,
)
-from aioshelly.rpc_device import RpcDevice
+from aioshelly.rpc_device import RpcDevice, bluetooth_mac_from_primary_mac
import voluptuous as vol
from homeassistant.components.bluetooth import async_remove_scanner
-from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME, Platform
+from homeassistant.const import (
+ CONF_HOST,
+ CONF_MODEL,
+ CONF_PASSWORD,
+ CONF_USERNAME,
+ Platform,
+)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers import (
@@ -53,6 +61,7 @@ from .utils import (
get_coap_context,
get_device_entry_gen,
get_http_port,
+ get_rpc_scripts_event_types,
get_ws_context,
)
@@ -159,7 +168,7 @@ async def _async_setup_block_entry(
# Following code block will force the right value for affected devices
if (
sleep_period == BLOCK_WRONG_SLEEP_PERIOD
- and entry.data["model"] in MODELS_WITH_WRONG_SLEEP_PERIOD
+ and entry.data[CONF_MODEL] in MODELS_WITH_WRONG_SLEEP_PERIOD
):
LOGGER.warning(
"Updating stored sleep period for %s: from %s to %s",
@@ -264,7 +273,10 @@ async def _async_setup_rpc_entry(hass: HomeAssistant, entry: ShellyConfigEntry)
async_create_issue_unsupported_firmware(hass, entry)
await device.shutdown()
raise ConfigEntryNotReady
- except (DeviceConnectionError, MacAddressMismatchError) as err:
+ runtime_data.rpc_script_events = await get_rpc_scripts_event_types(
+ device, ignore_scripts=[BLE_SCRIPT_NAME]
+ )
+ except (DeviceConnectionError, MacAddressMismatchError, RpcCallError) as err:
await device.shutdown()
raise ConfigEntryNotReady(repr(err)) from err
except InvalidAuthError as err:
@@ -339,4 +351,5 @@ async def async_remove_entry(hass: HomeAssistant, entry: ShellyConfigEntry) -> N
if get_device_entry_gen(entry) in RPC_GENERATIONS and (
mac_address := entry.unique_id
):
- async_remove_scanner(hass, mac_address)
+ source = dr.format_mac(bluetooth_mac_from_primary_mac(mac_address)).upper()
+ async_remove_scanner(hass, source)
diff --git a/homeassistant/components/shelly/binary_sensor.py b/homeassistant/components/shelly/binary_sensor.py
index fb253c682d8..b74578f1fb3 100644
--- a/homeassistant/components/shelly/binary_sensor.py
+++ b/homeassistant/components/shelly/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import CONF_SLEEP_PERIOD
@@ -130,6 +130,7 @@ SENSORS: dict[tuple[str, str], BlockBinarySensorDescription] = {
device_class=BinarySensorDeviceClass.GAS,
translation_key="gas",
value=lambda value: value in ["mild", "heavy"],
+ # Deprecated, remove in 2025.10
extra_state_attributes=lambda block: {"detected": block.gas},
),
("sensor", "smoke"): BlockBinarySensorDescription(
@@ -290,7 +291,7 @@ RPC_SENSORS: Final = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
diff --git a/homeassistant/components/shelly/bluetooth/__init__.py b/homeassistant/components/shelly/bluetooth/__init__.py
index 366d5c51d25..2b772bd1b78 100644
--- a/homeassistant/components/shelly/bluetooth/__init__.py
+++ b/homeassistant/components/shelly/bluetooth/__init__.py
@@ -7,15 +7,22 @@ from typing import TYPE_CHECKING
from aioshelly.ble import async_start_scanner, create_scanner
from aioshelly.ble.const import BLE_SCAN_RESULT_EVENT, BLE_SCAN_RESULT_VERSION
-from homeassistant.components.bluetooth import async_register_scanner
+from homeassistant.components.bluetooth import (
+ BluetoothScanningMode,
+ async_register_scanner,
+)
from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback as hass_callback
-from homeassistant.helpers.device_registry import format_mac
from ..const import BLEScannerMode
if TYPE_CHECKING:
from ..coordinator import ShellyRpcCoordinator
+BLE_SCANNER_MODE_TO_BLUETOOTH_SCANNING_MODE = {
+ BLEScannerMode.PASSIVE: BluetoothScanningMode.PASSIVE,
+ BLEScannerMode.ACTIVE: BluetoothScanningMode.ACTIVE,
+}
+
async def async_connect_scanner(
hass: HomeAssistant,
@@ -25,9 +32,14 @@ async def async_connect_scanner(
) -> CALLBACK_TYPE:
"""Connect scanner."""
device = coordinator.device
- entry = coordinator.entry
- source = format_mac(coordinator.mac).upper()
- scanner = create_scanner(source, entry.title)
+ entry = coordinator.config_entry
+ bluetooth_scanning_mode = BLE_SCANNER_MODE_TO_BLUETOOTH_SCANNING_MODE[scanner_mode]
+ scanner = create_scanner(
+ coordinator.bluetooth_source,
+ entry.title,
+ requested_mode=bluetooth_scanning_mode,
+ current_mode=bluetooth_scanning_mode,
+ )
unload_callbacks = [
async_register_scanner(
hass,
diff --git a/homeassistant/components/shelly/button.py b/homeassistant/components/shelly/button.py
index f1e2f8ef885..15bde4fbdff 100644
--- a/homeassistant/components/shelly/button.py
+++ b/homeassistant/components/shelly/button.py
@@ -2,12 +2,13 @@
from __future__ import annotations
-from collections.abc import Callable, Coroutine
+from collections.abc import Callable
from dataclasses import dataclass
from functools import partial
from typing import TYPE_CHECKING, Any, Final
-from aioshelly.const import RPC_GENERATIONS
+from aioshelly.const import BLU_TRV_IDENTIFIER, MODEL_BLU_GATEWAY, RPC_GENERATIONS
+from aioshelly.exceptions import DeviceConnectionError, InvalidAuthError, RpcCallError
from homeassistant.components.button import (
ButtonDeviceClass,
@@ -16,15 +17,20 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.device_registry import (
+ CONNECTION_BLUETOOTH,
+ CONNECTION_NETWORK_MAC,
+ DeviceInfo,
+)
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import slugify
-from .const import LOGGER, SHELLY_GAS_MODELS
+from .const import DOMAIN, LOGGER, SHELLY_GAS_MODELS
from .coordinator import ShellyBlockCoordinator, ShellyConfigEntry, ShellyRpcCoordinator
-from .utils import get_device_entry_gen
+from .utils import get_device_entry_gen, get_rpc_key_ids
@dataclass(frozen=True, kw_only=True)
@@ -33,7 +39,7 @@ class ShellyButtonDescription[
](ButtonEntityDescription):
"""Class to describe a Button entity."""
- press_action: Callable[[_ShellyCoordinatorT], Coroutine[Any, Any, None]]
+ press_action: str
supported: Callable[[_ShellyCoordinatorT], bool] = lambda _: True
@@ -44,14 +50,14 @@ BUTTONS: Final[list[ShellyButtonDescription[Any]]] = [
name="Reboot",
device_class=ButtonDeviceClass.RESTART,
entity_category=EntityCategory.CONFIG,
- press_action=lambda coordinator: coordinator.device.trigger_reboot(),
+ press_action="trigger_reboot",
),
ShellyButtonDescription[ShellyBlockCoordinator](
key="self_test",
name="Self test",
translation_key="self_test",
entity_category=EntityCategory.DIAGNOSTIC,
- press_action=lambda coordinator: coordinator.device.trigger_shelly_gas_self_test(),
+ press_action="trigger_shelly_gas_self_test",
supported=lambda coordinator: coordinator.device.model in SHELLY_GAS_MODELS,
),
ShellyButtonDescription[ShellyBlockCoordinator](
@@ -59,7 +65,7 @@ BUTTONS: Final[list[ShellyButtonDescription[Any]]] = [
name="Mute",
translation_key="mute",
entity_category=EntityCategory.CONFIG,
- press_action=lambda coordinator: coordinator.device.trigger_shelly_gas_mute(),
+ press_action="trigger_shelly_gas_mute",
supported=lambda coordinator: coordinator.device.model in SHELLY_GAS_MODELS,
),
ShellyButtonDescription[ShellyBlockCoordinator](
@@ -67,11 +73,22 @@ BUTTONS: Final[list[ShellyButtonDescription[Any]]] = [
name="Unmute",
translation_key="unmute",
entity_category=EntityCategory.CONFIG,
- press_action=lambda coordinator: coordinator.device.trigger_shelly_gas_unmute(),
+ press_action="trigger_shelly_gas_unmute",
supported=lambda coordinator: coordinator.device.model in SHELLY_GAS_MODELS,
),
]
+BLU_TRV_BUTTONS: Final[list[ShellyButtonDescription]] = [
+ ShellyButtonDescription[ShellyRpcCoordinator](
+ key="calibrate",
+ name="Calibrate",
+ translation_key="calibrate",
+ entity_category=EntityCategory.CONFIG,
+ press_action="trigger_blu_trv_calibration",
+ supported=lambda coordinator: coordinator.device.model == MODEL_BLU_GATEWAY,
+ ),
+]
+
@callback
def async_migrate_unique_ids(
@@ -106,7 +123,7 @@ def async_migrate_unique_ids(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set buttons for device."""
entry_data = config_entry.runtime_data
@@ -123,14 +140,28 @@ async def async_setup_entry(
hass, config_entry.entry_id, partial(async_migrate_unique_ids, coordinator)
)
- async_add_entities(
+ entities: list[ShellyButton | ShellyBluTrvButton] = []
+
+ entities.extend(
ShellyButton(coordinator, button)
for button in BUTTONS
if button.supported(coordinator)
)
+ if blutrv_key_ids := get_rpc_key_ids(coordinator.device.status, BLU_TRV_IDENTIFIER):
+ if TYPE_CHECKING:
+ assert isinstance(coordinator, ShellyRpcCoordinator)
-class ShellyButton(
+ entities.extend(
+ ShellyBluTrvButton(coordinator, button, id_)
+ for id_ in blutrv_key_ids
+ for button in BLU_TRV_BUTTONS
+ )
+
+ async_add_entities(entities)
+
+
+class ShellyBaseButton(
CoordinatorEntity[ShellyRpcCoordinator | ShellyBlockCoordinator], ButtonEntity
):
"""Defines a Shelly base button."""
@@ -148,14 +179,100 @@ class ShellyButton(
) -> None:
"""Initialize Shelly button."""
super().__init__(coordinator)
+
self.entity_description = description
+ async def async_press(self) -> None:
+ """Triggers the Shelly button press service."""
+ try:
+ await self._press_method()
+ except DeviceConnectionError as err:
+ self.coordinator.last_update_success = False
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="device_communication_action_error",
+ translation_placeholders={
+ "entity": self.entity_id,
+ "device": self.coordinator.device.name,
+ "error": repr(err),
+ },
+ ) from err
+ except RpcCallError as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="rpc_call_action_error",
+ translation_placeholders={
+ "entity": self.entity_id,
+ "device": self.coordinator.device.name,
+ "error": repr(err),
+ },
+ ) from err
+ except InvalidAuthError:
+ await self.coordinator.async_shutdown_device_and_start_reauth()
+
+ async def _press_method(self) -> None:
+ """Press method."""
+ raise NotImplementedError
+
+
+class ShellyButton(ShellyBaseButton):
+ """Defines a Shelly button."""
+
+ def __init__(
+ self,
+ coordinator: ShellyRpcCoordinator | ShellyBlockCoordinator,
+ description: ShellyButtonDescription[
+ ShellyRpcCoordinator | ShellyBlockCoordinator
+ ],
+ ) -> None:
+ """Initialize Shelly button."""
+ super().__init__(coordinator, description)
+
self._attr_name = f"{coordinator.device.name} {description.name}"
self._attr_unique_id = f"{coordinator.mac}_{description.key}"
self._attr_device_info = DeviceInfo(
connections={(CONNECTION_NETWORK_MAC, coordinator.mac)}
)
- async def async_press(self) -> None:
- """Triggers the Shelly button press service."""
- await self.entity_description.press_action(self.coordinator)
+ async def _press_method(self) -> None:
+ """Press method."""
+ method = getattr(self.coordinator.device, self.entity_description.press_action)
+
+ if TYPE_CHECKING:
+ assert method is not None
+
+ await method()
+
+
+class ShellyBluTrvButton(ShellyBaseButton):
+ """Represent a Shelly BLU TRV button."""
+
+ def __init__(
+ self,
+ coordinator: ShellyRpcCoordinator,
+ description: ShellyButtonDescription,
+ id_: int,
+ ) -> None:
+ """Initialize."""
+ super().__init__(coordinator, description)
+
+ ble_addr: str = coordinator.device.config[f"{BLU_TRV_IDENTIFIER}:{id_}"]["addr"]
+ device_name = (
+ coordinator.device.config[f"{BLU_TRV_IDENTIFIER}:{id_}"]["name"]
+ or f"shellyblutrv-{ble_addr.replace(':', '')}"
+ )
+ self._attr_name = f"{device_name} {description.name}"
+ self._attr_unique_id = f"{ble_addr}_{description.key}"
+ self._attr_device_info = DeviceInfo(
+ connections={(CONNECTION_BLUETOOTH, ble_addr)}
+ )
+ self._id = id_
+
+ async def _press_method(self) -> None:
+ """Press method."""
+ method = getattr(self.coordinator.device, self.entity_description.press_action)
+
+ if TYPE_CHECKING:
+ assert method is not None
+
+ await method(self._id)
diff --git a/homeassistant/components/shelly/climate.py b/homeassistant/components/shelly/climate.py
index f1491acdd81..c3612ed3f4f 100644
--- a/homeassistant/components/shelly/climate.py
+++ b/homeassistant/components/shelly/climate.py
@@ -7,7 +7,12 @@ from dataclasses import asdict, dataclass
from typing import Any, cast
from aioshelly.block_device import Block
-from aioshelly.const import BLU_TRV_IDENTIFIER, BLU_TRV_MODEL_NAME, RPC_GENERATIONS
+from aioshelly.const import (
+ BLU_TRV_IDENTIFIER,
+ BLU_TRV_MODEL_NAME,
+ BLU_TRV_TIMEOUT,
+ RPC_GENERATIONS,
+)
from aioshelly.exceptions import DeviceConnectionError, InvalidAuthError
from homeassistant.components.climate import (
@@ -27,7 +32,7 @@ from homeassistant.helpers.device_registry import (
CONNECTION_NETWORK_MAC,
DeviceInfo,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.entity_registry import RegistryEntry
from homeassistant.helpers.restore_state import ExtraStoredData, RestoreEntity
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -36,7 +41,6 @@ from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
from .const import (
BLU_TRV_TEMPERATURE_SETTINGS,
- BLU_TRV_TIMEOUT,
DOMAIN,
LOGGER,
NOT_CALIBRATED_ISSUE_ID,
@@ -56,7 +60,7 @@ from .utils import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
@@ -75,7 +79,7 @@ async def async_setup_entry(
@callback
def async_setup_climate_entities(
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
coordinator: ShellyBlockCoordinator,
) -> None:
"""Set up online climate devices."""
@@ -102,7 +106,7 @@ def async_setup_climate_entities(
def async_restore_climate_entities(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
coordinator: ShellyBlockCoordinator,
) -> None:
"""Restore sleeping climate devices."""
@@ -124,7 +128,7 @@ def async_restore_climate_entities(
def async_setup_rpc_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for RPC device."""
coordinator = config_entry.runtime_data.rpc
diff --git a/homeassistant/components/shelly/config_flow.py b/homeassistant/components/shelly/config_flow.py
index f53da8bd766..c7c1cd70a53 100644
--- a/homeassistant/components/shelly/config_flow.py
+++ b/homeassistant/components/shelly/config_flow.py
@@ -7,7 +7,12 @@ from typing import Any, Final
from aioshelly.block_device import BlockDevice
from aioshelly.common import ConnectionOptions, get_info
-from aioshelly.const import BLOCK_GENERATIONS, DEFAULT_HTTP_PORT, RPC_GENERATIONS
+from aioshelly.const import (
+ BLOCK_GENERATIONS,
+ DEFAULT_HTTP_PORT,
+ MODEL_WALL_DISPLAY,
+ RPC_GENERATIONS,
+)
from aioshelly.exceptions import (
CustomPortNotSupported,
DeviceConnectionError,
@@ -17,15 +22,11 @@ from aioshelly.exceptions import (
from aioshelly.rpc_device import RpcDevice
import voluptuous as vol
-from homeassistant.config_entries import (
- ConfigEntry,
- ConfigFlow,
- ConfigFlowResult,
- OptionsFlow,
-)
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult, OptionsFlow
from homeassistant.const import (
CONF_HOST,
CONF_MAC,
+ CONF_MODEL,
CONF_PASSWORD,
CONF_PORT,
CONF_USERNAME,
@@ -41,10 +42,9 @@ from .const import (
CONF_SLEEP_PERIOD,
DOMAIN,
LOGGER,
- MODEL_WALL_DISPLAY,
BLEScannerMode,
)
-from .coordinator import async_reconnect_soon
+from .coordinator import ShellyConfigEntry, async_reconnect_soon
from .utils import (
get_block_device_sleep_period,
get_coap_context,
@@ -112,7 +112,9 @@ async def validate_input(
return {
"title": rpc_device.name,
CONF_SLEEP_PERIOD: sleep_period,
- "model": rpc_device.shelly.get("model"),
+ CONF_MODEL: (
+ rpc_device.xmod_info.get("p") or rpc_device.shelly.get(CONF_MODEL)
+ ),
CONF_GEN: gen,
}
@@ -132,7 +134,7 @@ async def validate_input(
return {
"title": block_device.name,
CONF_SLEEP_PERIOD: sleep_period,
- "model": block_device.model,
+ CONF_MODEL: block_device.model,
CONF_GEN: gen,
}
@@ -164,7 +166,9 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
LOGGER.exception("Unexpected exception")
errors["base"] = "unknown"
else:
- await self.async_set_unique_id(self.info[CONF_MAC])
+ await self.async_set_unique_id(
+ self.info[CONF_MAC], raise_on_progress=False
+ )
self._abort_if_unique_id_configured({CONF_HOST: host})
self.host = host
self.port = port
@@ -185,14 +189,14 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
LOGGER.exception("Unexpected exception")
errors["base"] = "unknown"
else:
- if device_info["model"]:
+ if device_info[CONF_MODEL]:
return self.async_create_entry(
title=device_info["title"],
data={
CONF_HOST: user_input[CONF_HOST],
CONF_PORT: user_input[CONF_PORT],
CONF_SLEEP_PERIOD: device_info[CONF_SLEEP_PERIOD],
- "model": device_info["model"],
+ CONF_MODEL: device_info[CONF_MODEL],
CONF_GEN: device_info[CONF_GEN],
},
)
@@ -224,7 +228,7 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
LOGGER.exception("Unexpected exception")
errors["base"] = "unknown"
else:
- if device_info["model"]:
+ if device_info[CONF_MODEL]:
return self.async_create_entry(
title=device_info["title"],
data={
@@ -232,7 +236,7 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
CONF_HOST: self.host,
CONF_PORT: self.port,
CONF_SLEEP_PERIOD: device_info[CONF_SLEEP_PERIOD],
- "model": device_info["model"],
+ CONF_MODEL: device_info[CONF_MODEL],
CONF_GEN: device_info[CONF_GEN],
},
)
@@ -330,7 +334,7 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle discovery confirm."""
errors: dict[str, str] = {}
- if not self.device_info["model"]:
+ if not self.device_info[CONF_MODEL]:
errors["base"] = "firmware_not_fully_provisioned"
model = "Shelly"
else:
@@ -339,9 +343,9 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
return self.async_create_entry(
title=self.device_info["title"],
data={
- "host": self.host,
+ CONF_HOST: self.host,
CONF_SLEEP_PERIOD: self.device_info[CONF_SLEEP_PERIOD],
- "model": self.device_info["model"],
+ CONF_MODEL: self.device_info[CONF_MODEL],
CONF_GEN: self.device_info[CONF_GEN],
},
)
@@ -350,8 +354,8 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
return self.async_show_form(
step_id="confirm_discovery",
description_placeholders={
- "model": model,
- "host": self.host,
+ CONF_MODEL: model,
+ CONF_HOST: self.host,
},
errors=errors,
)
@@ -449,18 +453,18 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
@staticmethod
@callback
- def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlowHandler:
+ def async_get_options_flow(config_entry: ShellyConfigEntry) -> OptionsFlowHandler:
"""Get the options flow for this handler."""
return OptionsFlowHandler()
@classmethod
@callback
- def async_supports_options_flow(cls, config_entry: ConfigEntry) -> bool:
+ def async_supports_options_flow(cls, config_entry: ShellyConfigEntry) -> bool:
"""Return options flow support for this handler."""
return (
get_device_entry_gen(config_entry) in RPC_GENERATIONS
and not config_entry.data.get(CONF_SLEEP_PERIOD)
- and config_entry.data.get("model") != MODEL_WALL_DISPLAY
+ and config_entry.data.get(CONF_MODEL) != MODEL_WALL_DISPLAY
)
diff --git a/homeassistant/components/shelly/const.py b/homeassistant/components/shelly/const.py
index d47f2b0ae80..c94c827b7db 100644
--- a/homeassistant/components/shelly/const.py
+++ b/homeassistant/components/shelly/const.py
@@ -271,9 +271,6 @@ API_WS_URL = "/api/shelly/ws"
COMPONENT_ID_PATTERN = re.compile(r"[a-z\d]+:\d+")
-# value confirmed by Shelly team
-BLU_TRV_TIMEOUT = 60
-
ROLE_TO_DEVICE_CLASS_MAP = {
"current_humidity": SensorDeviceClass.HUMIDITY,
"current_temperature": SensorDeviceClass.TEMPERATURE,
diff --git a/homeassistant/components/shelly/coordinator.py b/homeassistant/components/shelly/coordinator.py
index f2a01240f70..85cf430bc5d 100644
--- a/homeassistant/components/shelly/coordinator.py
+++ b/homeassistant/components/shelly/coordinator.py
@@ -10,7 +10,7 @@ from typing import Any, cast
from aioshelly.ble import async_ensure_ble_enabled, async_stop_scanner
from aioshelly.block_device import BlockDevice, BlockUpdateType
-from aioshelly.const import MODEL_NAMES, MODEL_VALVE
+from aioshelly.const import MODEL_VALVE
from aioshelly.exceptions import (
DeviceConnectionError,
InvalidAuthError,
@@ -18,6 +18,7 @@ from aioshelly.exceptions import (
RpcCallError,
)
from aioshelly.rpc_device import RpcDevice, RpcUpdateType
+from aioshelly.rpc_device.utils import bluetooth_mac_from_primary_mac
from propcache.api import cached_property
from homeassistant.components.bluetooth import async_remove_scanner
@@ -25,6 +26,7 @@ from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import (
ATTR_DEVICE_ID,
CONF_HOST,
+ CONF_MODEL,
EVENT_HOMEASSISTANT_STOP,
Platform,
)
@@ -72,6 +74,7 @@ from .utils import (
get_http_port,
get_rpc_device_wakeup_period,
get_rpc_ws_url,
+ get_shelly_model_name,
update_device_fw_info,
)
@@ -85,6 +88,7 @@ class ShellyEntryData:
rest: ShellyRestCoordinator | None = None
rpc: ShellyRpcCoordinator | None = None
rpc_poll: ShellyRpcPollingCoordinator | None = None
+ rpc_script_events: dict[int, list[str]] | None = None
type ShellyConfigEntry = ConfigEntry[ShellyEntryData]
@@ -95,6 +99,8 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
):
"""Coordinator for a Shelly device."""
+ config_entry: ShellyConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
@@ -103,7 +109,6 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
update_interval: float,
) -> None:
"""Initialize the Shelly device coordinator."""
- self.entry = entry
self.device = device
self.device_id: str | None = None
self._pending_platforms: list[Platform] | None = None
@@ -112,7 +117,13 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
# The device has come online at least once. In the case of a sleeping RPC
# device, this means that the device has connected to the WS server at least once.
self._came_online_once = False
- super().__init__(hass, LOGGER, name=device_name, update_interval=interval_td)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=entry,
+ name=device_name,
+ update_interval=interval_td,
+ )
self._debounced_reload: Debouncer[Coroutine[Any, Any, None]] = Debouncer(
hass,
@@ -130,12 +141,12 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
@cached_property
def model(self) -> str:
"""Model of the device."""
- return cast(str, self.entry.data["model"])
+ return cast(str, self.config_entry.data[CONF_MODEL])
@cached_property
def mac(self) -> str:
"""Mac address of the device."""
- return cast(str, self.entry.unique_id)
+ return cast(str, self.config_entry.unique_id)
@property
def sw_version(self) -> str:
@@ -145,23 +156,23 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
@property
def sleep_period(self) -> int:
"""Sleep period of the device."""
- return self.entry.data.get(CONF_SLEEP_PERIOD, 0)
+ return self.config_entry.data.get(CONF_SLEEP_PERIOD, 0)
def async_setup(self, pending_platforms: list[Platform] | None = None) -> None:
"""Set up the coordinator."""
self._pending_platforms = pending_platforms
dev_reg = dr.async_get(self.hass)
device_entry = dev_reg.async_get_or_create(
- config_entry_id=self.entry.entry_id,
+ config_entry_id=self.config_entry.entry_id,
name=self.name,
connections={(CONNECTION_NETWORK_MAC, self.mac)},
identifiers={(DOMAIN, self.mac)},
manufacturer="Shelly",
- model=MODEL_NAMES.get(self.model),
+ model=get_shelly_model_name(self.model, self.sleep_period, self.device),
model_id=self.model,
sw_version=self.sw_version,
- hw_version=f"gen{get_device_entry_gen(self.entry)}",
- configuration_url=f"http://{get_host(self.entry.data[CONF_HOST])}:{get_http_port(self.entry.data)}",
+ hw_version=f"gen{get_device_entry_gen(self.config_entry)}",
+ configuration_url=f"http://{get_host(self.config_entry.data[CONF_HOST])}:{get_http_port(self.config_entry.data)}",
)
self.device_id = device_entry.id
@@ -179,18 +190,18 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
LOGGER.debug("Connecting to Shelly Device - %s", self.name)
try:
await self.device.initialize()
- update_device_fw_info(self.hass, self.device, self.entry)
+ update_device_fw_info(self.hass, self.device, self.config_entry)
except (DeviceConnectionError, MacAddressMismatchError) as err:
LOGGER.debug(
"Error connecting to Shelly device %s, error: %r", self.name, err
)
return False
except InvalidAuthError:
- self.entry.async_start_reauth(self.hass)
+ self.config_entry.async_start_reauth(self.hass)
return False
if not self.device.firmware_supported:
- async_create_issue_unsupported_firmware(self.hass, self.entry)
+ async_create_issue_unsupported_firmware(self.hass, self.config_entry)
return False
if not self._pending_platforms:
@@ -200,7 +211,7 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
platforms = self._pending_platforms
self._pending_platforms = None
- data = {**self.entry.data}
+ data = {**self.config_entry.data}
# Update sleep_period
old_sleep_period = data[CONF_SLEEP_PERIOD]
@@ -211,10 +222,12 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
if new_sleep_period != old_sleep_period:
data[CONF_SLEEP_PERIOD] = new_sleep_period
- self.hass.config_entries.async_update_entry(self.entry, data=data)
+ self.hass.config_entries.async_update_entry(self.config_entry, data=data)
# Resume platform setup
- await self.hass.config_entries.async_forward_entry_setups(self.entry, platforms)
+ await self.hass.config_entries.async_forward_entry_setups(
+ self.config_entry, platforms
+ )
return True
@@ -222,7 +235,7 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
"""Reload entry."""
self._debounced_reload.async_cancel()
LOGGER.debug("Reloading entry %s", self.name)
- await self.hass.config_entries.async_reload(self.entry.entry_id)
+ await self.hass.config_entries.async_reload(self.config_entry.entry_id)
async def async_shutdown_device_and_start_reauth(self) -> None:
"""Shutdown Shelly device and start reauth flow."""
@@ -230,7 +243,7 @@ class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](
# and won't be able to send commands to the device
self.last_update_success = False
await self.shutdown()
- self.entry.async_start_reauth(self.hass)
+ self.config_entry.async_start_reauth(self.hass)
class ShellyBlockCoordinator(ShellyCoordinatorBase[BlockDevice]):
@@ -240,9 +253,8 @@ class ShellyBlockCoordinator(ShellyCoordinatorBase[BlockDevice]):
self, hass: HomeAssistant, entry: ShellyConfigEntry, device: BlockDevice
) -> None:
"""Initialize the Shelly block device coordinator."""
- self.entry = entry
- if self.sleep_period:
- update_interval = UPDATE_PERIOD_MULTIPLIER * self.sleep_period
+ if sleep_period := entry.data.get(CONF_SLEEP_PERIOD, 0):
+ update_interval = UPDATE_PERIOD_MULTIPLIER * sleep_period
else:
update_interval = (
UPDATE_PERIOD_MULTIPLIER * device.settings["coiot"]["update_period"]
@@ -385,7 +397,7 @@ class ShellyBlockCoordinator(ShellyCoordinatorBase[BlockDevice]):
LOGGER.debug("Shelly %s handle update, type: %s", self.name, update_type)
if update_type is BlockUpdateType.ONLINE:
self._came_online_once = True
- self.entry.async_create_background_task(
+ self.config_entry.async_create_background_task(
self.hass,
self._async_device_connect_task(),
"block device online",
@@ -415,7 +427,7 @@ class ShellyBlockCoordinator(ShellyCoordinatorBase[BlockDevice]):
learn_more_url="https://www.home-assistant.io/integrations/shelly/#shelly-device-configuration-generation-1",
translation_key="push_update_failure",
translation_placeholders={
- "device_name": self.entry.title,
+ "device_name": self.config_entry.title,
"ip_address": self.device.ip_address,
},
)
@@ -462,7 +474,7 @@ class ShellyRestCoordinator(ShellyCoordinatorBase[BlockDevice]):
except InvalidAuthError:
await self.async_shutdown_device_and_start_reauth()
else:
- update_device_fw_info(self.hass, self.device, self.entry)
+ update_device_fw_info(self.hass, self.device, self.config_entry)
class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
@@ -472,9 +484,8 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
self, hass: HomeAssistant, entry: ShellyConfigEntry, device: RpcDevice
) -> None:
"""Initialize the Shelly RPC device coordinator."""
- self.entry = entry
- if self.sleep_period:
- update_interval = UPDATE_PERIOD_MULTIPLIER * self.sleep_period
+ if sleep_period := entry.data.get(CONF_SLEEP_PERIOD, 0):
+ update_interval = UPDATE_PERIOD_MULTIPLIER * sleep_period
else:
update_interval = RPC_RECONNECT_INTERVAL
super().__init__(hass, entry, device, update_interval)
@@ -488,6 +499,15 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
self._connect_task: asyncio.Task | None = None
entry.async_on_unload(entry.add_update_listener(self._async_update_listener))
+ @cached_property
+ def bluetooth_source(self) -> str:
+ """Return the Bluetooth source address.
+
+ This is the Bluetooth MAC address of the device that is used
+ for the Bluetooth scanner.
+ """
+ return format_mac(bluetooth_mac_from_primary_mac(self.mac)).upper()
+
async def async_device_online(self, source: str) -> None:
"""Handle device going online."""
if not self.sleep_period:
@@ -514,9 +534,9 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
):
return False
- data = {**self.entry.data}
+ data = {**self.config_entry.data}
data[CONF_SLEEP_PERIOD] = wakeup_period
- self.hass.config_entries.async_update_entry(self.entry, data=data)
+ self.hass.config_entries.async_update_entry(self.config_entry, data=data)
update_interval = UPDATE_PERIOD_MULTIPLIER * wakeup_period
self.update_interval = timedelta(seconds=update_interval)
@@ -693,12 +713,12 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
async def _async_connect_ble_scanner(self) -> None:
"""Connect BLE scanner."""
- ble_scanner_mode = self.entry.options.get(
+ ble_scanner_mode = self.config_entry.options.get(
CONF_BLE_SCANNER_MODE, BLEScannerMode.DISABLED
)
if ble_scanner_mode == BLEScannerMode.DISABLED and self.connected:
await async_stop_scanner(self.device)
- async_remove_scanner(self.hass, format_mac(self.mac).upper())
+ async_remove_scanner(self.hass, self.bluetooth_source)
return
if await async_ensure_ble_enabled(self.device):
# BLE enable required a reboot, don't bother connecting
@@ -719,7 +739,7 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
):
LOGGER.debug("Device %s already connected/connecting", self.name)
return
- self._connect_task = self.entry.async_create_background_task(
+ self._connect_task = self.config_entry.async_create_background_task(
self.hass,
self._async_device_connect_task(),
"rpc device online",
@@ -736,13 +756,13 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
self._came_online_once = True
self._async_handle_rpc_device_online()
elif update_type is RpcUpdateType.INITIALIZED:
- self.entry.async_create_background_task(
+ self.config_entry.async_create_background_task(
self.hass, self._async_connected(), "rpc device init", eager_start=True
)
# Make sure entities are marked available
self.async_set_updated_data(None)
elif update_type is RpcUpdateType.DISCONNECTED:
- self.entry.async_create_background_task(
+ self.config_entry.async_create_background_task(
self.hass,
self._async_disconnected(True),
"rpc device disconnected",
@@ -753,7 +773,7 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
elif update_type is RpcUpdateType.STATUS:
self.async_set_updated_data(None)
if self.sleep_period:
- update_device_fw_info(self.hass, self.device, self.entry)
+ update_device_fw_info(self.hass, self.device, self.config_entry)
elif update_type is RpcUpdateType.EVENT and (event := self.device.event):
self._async_device_event_handler(event)
@@ -763,7 +783,7 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
self.device.subscribe_updates(self._async_handle_update)
if self.device.initialized:
# If we are already initialized, we are connected
- self.entry.async_create_task(
+ self.config_entry.async_create_task(
self.hass, self._async_connected(), eager_start=True
)
@@ -775,7 +795,7 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
await async_stop_scanner(self.device)
await super().shutdown()
except InvalidAuthError:
- self.entry.async_start_reauth(self.hass)
+ self.config_entry.async_start_reauth(self.hass)
return
except DeviceConnectionError as err:
# If the device is restarting or has gone offline before
diff --git a/homeassistant/components/shelly/cover.py b/homeassistant/components/shelly/cover.py
index 09e8279bf9b..e9eb5acf161 100644
--- a/homeassistant/components/shelly/cover.py
+++ b/homeassistant/components/shelly/cover.py
@@ -15,7 +15,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ShellyBlockCoordinator, ShellyConfigEntry, ShellyRpcCoordinator
from .entity import ShellyBlockEntity, ShellyRpcEntity
@@ -25,7 +25,7 @@ from .utils import get_device_entry_gen, get_rpc_key_ids
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up covers for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
@@ -38,7 +38,7 @@ async def async_setup_entry(
def async_setup_block_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up cover for device."""
coordinator = config_entry.runtime_data.block
@@ -55,7 +55,7 @@ def async_setup_block_entry(
def async_setup_rpc_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for RPC device."""
coordinator = config_entry.runtime_data.rpc
diff --git a/homeassistant/components/shelly/diagnostics.py b/homeassistant/components/shelly/diagnostics.py
index a5fe1f5b6c0..2a9699e0a08 100644
--- a/homeassistant/components/shelly/diagnostics.py
+++ b/homeassistant/components/shelly/diagnostics.py
@@ -6,9 +6,14 @@ from typing import Any
from homeassistant.components.bluetooth import async_scanner_by_source
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.const import (
+ ATTR_MODEL,
+ ATTR_NAME,
+ ATTR_SW_VERSION,
+ CONF_PASSWORD,
+ CONF_USERNAME,
+)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.device_registry import format_mac
from .coordinator import ShellyConfigEntry
from .utils import get_rpc_ws_url
@@ -31,9 +36,9 @@ async def async_get_config_entry_diagnostics(
block_coordinator = shelly_entry_data.block
assert block_coordinator
device_info = {
- "name": block_coordinator.name,
- "model": block_coordinator.model,
- "sw_version": block_coordinator.sw_version,
+ ATTR_NAME: block_coordinator.name,
+ ATTR_MODEL: block_coordinator.model,
+ ATTR_SW_VERSION: block_coordinator.sw_version,
}
if block_coordinator.device.initialized:
device_settings = {
@@ -66,28 +71,29 @@ async def async_get_config_entry_diagnostics(
rpc_coordinator = shelly_entry_data.rpc
assert rpc_coordinator
device_info = {
- "name": rpc_coordinator.name,
- "model": rpc_coordinator.model,
- "sw_version": rpc_coordinator.sw_version,
+ ATTR_NAME: rpc_coordinator.name,
+ ATTR_MODEL: rpc_coordinator.model,
+ ATTR_SW_VERSION: rpc_coordinator.sw_version,
}
if rpc_coordinator.device.initialized:
device_settings = {
k: v for k, v in rpc_coordinator.device.config.items() if k in ["cloud"]
}
- ws_config = rpc_coordinator.device.config["ws"]
- device_settings["ws_outbound_enabled"] = ws_config["enable"]
- if ws_config["enable"]:
- device_settings["ws_outbound_server_valid"] = bool(
- ws_config["server"] == get_rpc_ws_url(hass)
- )
+ if not (ws_config := rpc_coordinator.device.config.get("ws", {})):
+ device_settings["ws_outbound"] = "not supported"
+ if (ws_outbound_enabled := ws_config.get("enable")) is not None:
+ device_settings["ws_outbound_enabled"] = ws_outbound_enabled
+ if ws_outbound_enabled:
+ device_settings["ws_outbound_server_valid"] = bool(
+ ws_config["server"] == get_rpc_ws_url(hass)
+ )
device_status = {
k: v
for k, v in rpc_coordinator.device.status.items()
if k in ["sys", "wifi"]
}
- source = format_mac(rpc_coordinator.mac).upper()
- if scanner := async_scanner_by_source(hass, source):
+ if scanner := async_scanner_by_source(hass, rpc_coordinator.bluetooth_source):
bluetooth = {
"scanner": await scanner.async_diagnostics(),
}
diff --git a/homeassistant/components/shelly/entity.py b/homeassistant/components/shelly/entity.py
index 001727c74b3..58ac34fc5ca 100644
--- a/homeassistant/components/shelly/entity.py
+++ b/homeassistant/components/shelly/entity.py
@@ -296,7 +296,6 @@ class RpcEntityDescription(EntityDescription):
value: Callable[[Any, Any], Any] | None = None
available: Callable[[dict], bool] | None = None
removal_condition: Callable[[dict, dict, str], bool] | None = None
- extra_state_attributes: Callable[[dict, dict], dict | None] | None = None
use_polling_coordinator: bool = False
supported: Callable = lambda _: False
unit: Callable[[dict], str | None] | None = None
@@ -313,7 +312,6 @@ class RestEntityDescription(EntityDescription):
name: str = ""
value: Callable[[dict, Any], Any] | None = None
- extra_state_attributes: Callable[[dict], dict | None] | None = None
class ShellyBlockEntity(CoordinatorEntity[ShellyBlockCoordinator]):
diff --git a/homeassistant/components/shelly/event.py b/homeassistant/components/shelly/event.py
index 78093bec8aa..ec5810581b1 100644
--- a/homeassistant/components/shelly/event.py
+++ b/homeassistant/components/shelly/event.py
@@ -18,7 +18,7 @@ from homeassistant.components.event import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -34,7 +34,6 @@ from .utils import (
get_device_entry_gen,
get_rpc_entity_name,
get_rpc_key_instances,
- get_rpc_script_event_types,
is_block_momentary_input,
is_rpc_momentary_input,
)
@@ -83,7 +82,7 @@ SCRIPT_EVENT: Final = ShellyRpcEventDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for device."""
entities: list[ShellyBlockEvent | ShellyRpcEvent] = []
@@ -109,18 +108,15 @@ async def async_setup_entry(
script_instances = get_rpc_key_instances(
coordinator.device.status, SCRIPT_EVENT.key
)
+ script_events = config_entry.runtime_data.rpc_script_events
for script in script_instances:
script_name = get_rpc_entity_name(coordinator.device, script)
if script_name == BLE_SCRIPT_NAME:
continue
- event_types = await get_rpc_script_event_types(
- coordinator.device, int(script.split(":")[-1])
- )
- if not event_types:
- continue
-
- entities.append(ShellyRpcScriptEvent(coordinator, script, event_types))
+ script_id = int(script.split(":")[-1])
+ if script_events and (event_types := script_events[script_id]):
+ entities.append(ShellyRpcScriptEvent(coordinator, script, event_types))
# If a script is removed, from the device configuration, we need to remove orphaned entities
async_remove_orphaned_entities(
diff --git a/homeassistant/components/shelly/icons.json b/homeassistant/components/shelly/icons.json
index f93abf6b854..08b269a73c5 100644
--- a/homeassistant/components/shelly/icons.json
+++ b/homeassistant/components/shelly/icons.json
@@ -23,12 +23,18 @@
"gas_concentration": {
"default": "mdi:gauge"
},
+ "gas_detected": {
+ "default": "mdi:gas-burner"
+ },
"lamp_life": {
"default": "mdi:progress-wrench"
},
"operation": {
"default": "mdi:cog-transfer"
},
+ "self_test": {
+ "default": "mdi:progress-wrench"
+ },
"tilt": {
"default": "mdi:angle-acute"
},
diff --git a/homeassistant/components/shelly/light.py b/homeassistant/components/shelly/light.py
index 5d7bad810b4..ce31533b557 100644
--- a/homeassistant/components/shelly/light.py
+++ b/homeassistant/components/shelly/light.py
@@ -21,7 +21,7 @@ from homeassistant.components.light import (
brightness_supported,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
BLOCK_MAX_TRANSITION_TIME_MS,
@@ -53,7 +53,7 @@ from .utils import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up lights for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
@@ -66,7 +66,7 @@ async def async_setup_entry(
def async_setup_block_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for block device."""
coordinator = config_entry.runtime_data.block
@@ -96,7 +96,7 @@ def async_setup_block_entry(
def async_setup_rpc_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for RPC device."""
coordinator = config_entry.runtime_data.rpc
diff --git a/homeassistant/components/shelly/manifest.json b/homeassistant/components/shelly/manifest.json
index 4cfb49b680f..e863720e476 100644
--- a/homeassistant/components/shelly/manifest.json
+++ b/homeassistant/components/shelly/manifest.json
@@ -8,11 +8,14 @@
"integration_type": "device",
"iot_class": "local_push",
"loggers": ["aioshelly"],
- "requirements": ["aioshelly==12.4.1"],
+ "requirements": ["aioshelly==13.4.0"],
"zeroconf": [
{
"type": "_http._tcp.local.",
"name": "shelly*"
+ },
+ {
+ "type": "_shelly._tcp.local."
}
]
}
diff --git a/homeassistant/components/shelly/number.py b/homeassistant/components/shelly/number.py
index 1fc47b23bdb..a8e6de1ca73 100644
--- a/homeassistant/components/shelly/number.py
+++ b/homeassistant/components/shelly/number.py
@@ -7,7 +7,7 @@ from dataclasses import dataclass
from typing import TYPE_CHECKING, Any, Final, cast
from aioshelly.block_device import Block
-from aioshelly.const import RPC_GENERATIONS
+from aioshelly.const import BLU_TRV_TIMEOUT, RPC_GENERATIONS
from aioshelly.exceptions import DeviceConnectionError, InvalidAuthError
from homeassistant.components.number import (
@@ -22,10 +22,10 @@ from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.entity_registry import RegistryEntry
-from .const import BLU_TRV_TIMEOUT, CONF_SLEEP_PERIOD, LOGGER, VIRTUAL_NUMBER_MODE_MAP
+from .const import CONF_SLEEP_PERIOD, LOGGER, VIRTUAL_NUMBER_MODE_MAP
from .coordinator import ShellyBlockCoordinator, ShellyConfigEntry, ShellyRpcCoordinator
from .entity import (
BlockEntityDescription,
@@ -238,7 +238,7 @@ RPC_NUMBERS: Final = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up numbers for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
diff --git a/homeassistant/components/shelly/select.py b/homeassistant/components/shelly/select.py
index 0caf4661240..1fb3dfb3447 100644
--- a/homeassistant/components/shelly/select.py
+++ b/homeassistant/components/shelly/select.py
@@ -13,7 +13,7 @@ from homeassistant.components.select import (
SelectEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ShellyConfigEntry, ShellyRpcCoordinator
from .entity import (
@@ -45,7 +45,7 @@ RPC_SELECT_ENTITIES: Final = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up selectors for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
diff --git a/homeassistant/components/shelly/sensor.py b/homeassistant/components/shelly/sensor.py
index c492fc1de9e..79e4c97aead 100644
--- a/homeassistant/components/shelly/sensor.py
+++ b/homeassistant/components/shelly/sensor.py
@@ -35,11 +35,11 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.entity_registry import RegistryEntry
from homeassistant.helpers.typing import StateType
-from .const import CONF_SLEEP_PERIOD, ROLE_TO_DEVICE_CLASS_MAP, SHAIR_MAX_WORK_HOURS
+from .const import CONF_SLEEP_PERIOD, ROLE_TO_DEVICE_CLASS_MAP
from .coordinator import ShellyBlockCoordinator, ShellyConfigEntry, ShellyRpcCoordinator
from .entity import (
BlockEntityDescription,
@@ -58,6 +58,7 @@ from .utils import (
async_remove_orphaned_entities,
get_device_entry_gen,
get_device_uptime,
+ get_shelly_air_lamp_life,
get_virtual_component_ids,
is_rpc_wifi_stations_disabled,
)
@@ -355,8 +356,9 @@ SENSORS: dict[tuple[str, str], BlockSensorDescription] = {
name="Lamp life",
native_unit_of_measurement=PERCENTAGE,
translation_key="lamp_life",
- value=lambda value: 100 - (value / 3600 / SHAIR_MAX_WORK_HOURS),
+ value=get_shelly_air_lamp_life,
suggested_display_precision=1,
+ # Deprecated, remove in 2025.10
extra_state_attributes=lambda block: {
"Operational hours": round(cast(int, block.totalWorkTime) / 3600, 1)
},
@@ -374,9 +376,10 @@ SENSORS: dict[tuple[str, str], BlockSensorDescription] = {
key="sensor|sensorOp",
name="Operation",
device_class=SensorDeviceClass.ENUM,
- options=["unknown", "warmup", "normal", "fault"],
+ options=["warmup", "normal", "fault"],
translation_key="operation",
- value=lambda value: value,
+ value=lambda value: None if value == "unknown" else value,
+ # Deprecated, remove in 2025.10
extra_state_attributes=lambda block: {"self_test": block.selfTest},
),
("valve", "valve"): BlockSensorDescription(
@@ -391,11 +394,33 @@ SENSORS: dict[tuple[str, str], BlockSensorDescription] = {
"failure",
"opened",
"opening",
- "unknown",
],
+ value=lambda value: None if value == "unknown" else value,
entity_category=EntityCategory.DIAGNOSTIC,
removal_condition=lambda _, block: block.valve == "not_connected",
),
+ ("sensor", "gas"): BlockSensorDescription(
+ key="sensor|gas",
+ name="Gas detected",
+ translation_key="gas_detected",
+ device_class=SensorDeviceClass.ENUM,
+ options=[
+ "none",
+ "mild",
+ "heavy",
+ "test",
+ ],
+ value=lambda value: None if value == "unknown" else value,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ ("sensor", "selfTest"): BlockSensorDescription(
+ key="sensor|selfTest",
+ name="Self test",
+ translation_key="self_test",
+ device_class=SensorDeviceClass.ENUM,
+ options=["not_completed", "completed", "running", "pending"],
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
}
REST_SENSORS: Final = {
@@ -1324,7 +1349,7 @@ RPC_SENSORS: Final = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
diff --git a/homeassistant/components/shelly/strings.json b/homeassistant/components/shelly/strings.json
index eb869b54e4c..22d88928387 100644
--- a/homeassistant/components/shelly/strings.json
+++ b/homeassistant/components/shelly/strings.json
@@ -106,7 +106,6 @@
"state_attributes": {
"detected": {
"state": {
- "unknown": "Unknown",
"none": "None",
"mild": "Mild",
"heavy": "Heavy",
@@ -139,9 +138,26 @@
}
},
"sensor": {
+ "gas_detected": {
+ "state": {
+ "none": "None",
+ "mild": "Mild",
+ "heavy": "Heavy",
+ "test": "Test"
+ },
+ "state_attributes": {
+ "options": {
+ "state": {
+ "none": "[%key:component::shelly::entity::sensor::gas_detected::state::none%]",
+ "mild": "[%key:component::shelly::entity::sensor::gas_detected::state::mild%]",
+ "heavy": "[%key:component::shelly::entity::sensor::gas_detected::state::heavy%]",
+ "test": "[%key:component::shelly::entity::sensor::gas_detected::state::test%]"
+ }
+ }
+ }
+ },
"operation": {
"state": {
- "unknown": "Unknown",
"warmup": "Warm-up",
"normal": "Normal",
"fault": "Fault"
@@ -157,6 +173,24 @@
}
}
},
+ "self_test": {
+ "state": {
+ "not_completed": "Not completed",
+ "completed": "Completed",
+ "running": "Running",
+ "pending": "Pending"
+ },
+ "state_attributes": {
+ "options": {
+ "state": {
+ "not_completed": "[%key:component::shelly::entity::sensor::self_test::state::not_completed%]",
+ "completed": "[%key:component::shelly::entity::sensor::self_test::state::completed%]",
+ "running": "[%key:component::shelly::entity::sensor::self_test::state::running%]",
+ "pending": "[%key:component::shelly::entity::sensor::self_test::state::pending%]"
+ }
+ }
+ }
+ },
"valve_status": {
"state": {
"checking": "Checking",
@@ -164,12 +198,19 @@
"closing": "Closing",
"failure": "Failure",
"opened": "Opened",
- "opening": "Opening",
- "unknown": "[%key:component::shelly::entity::sensor::operation::state::unknown%]"
+ "opening": "Opening"
}
}
}
},
+ "exceptions": {
+ "device_communication_action_error": {
+ "message": "Device communication error occurred while calling the entity {entity} action for {device} device: {error}"
+ },
+ "rpc_call_action_error": {
+ "message": "RPC call error occurred while calling the entity {entity} action for {device} device: {error}"
+ }
+ },
"issues": {
"device_not_calibrated": {
"title": "Shelly device {device_name} is not calibrated",
diff --git a/homeassistant/components/shelly/switch.py b/homeassistant/components/shelly/switch.py
index 8a33dae0938..ce9e4f065fb 100644
--- a/homeassistant/components/shelly/switch.py
+++ b/homeassistant/components/shelly/switch.py
@@ -2,12 +2,14 @@
from __future__ import annotations
+from collections.abc import Callable
from dataclasses import dataclass
from typing import Any, cast
from aioshelly.block_device import Block
-from aioshelly.const import MODEL_2, MODEL_25, MODEL_WALL_DISPLAY, RPC_GENERATIONS
+from aioshelly.const import RPC_GENERATIONS
+from homeassistant.components.climate import DOMAIN as CLIMATE_PLATFORM
from homeassistant.components.switch import (
DOMAIN as SWITCH_PLATFORM,
SwitchEntity,
@@ -15,32 +17,26 @@ from homeassistant.components.switch import (
)
from homeassistant.const import STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant, State, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.entity_registry import RegistryEntry
from homeassistant.helpers.restore_state import RestoreEntity
-from .const import CONF_SLEEP_PERIOD, MOTION_MODELS
from .coordinator import ShellyBlockCoordinator, ShellyConfigEntry, ShellyRpcCoordinator
from .entity import (
BlockEntityDescription,
RpcEntityDescription,
- ShellyBlockEntity,
+ ShellyBlockAttributeEntity,
ShellyRpcAttributeEntity,
- ShellyRpcEntity,
ShellySleepingBlockAttributeEntity,
async_setup_entry_attribute_entities,
- async_setup_rpc_attribute_entities,
+ async_setup_entry_rpc,
)
from .utils import (
async_remove_orphaned_entities,
- async_remove_shelly_entity,
get_device_entry_gen,
- get_rpc_key_ids,
get_virtual_component_ids,
- is_block_channel_type_light,
- is_rpc_channel_type_light,
- is_rpc_thermostat_internal_actuator,
- is_rpc_thermostat_mode,
+ is_block_exclude_from_relay,
+ is_rpc_exclude_from_relay,
)
@@ -49,35 +45,70 @@ class BlockSwitchDescription(BlockEntityDescription, SwitchEntityDescription):
"""Class to describe a BLOCK switch."""
-MOTION_SWITCH = BlockSwitchDescription(
- key="sensor|motionActive",
- name="Motion detection",
- entity_category=EntityCategory.CONFIG,
-)
+BLOCK_RELAY_SWITCHES = {
+ ("relay", "output"): BlockSwitchDescription(
+ key="relay|output",
+ removal_condition=is_block_exclude_from_relay,
+ )
+}
+
+BLOCK_SLEEPING_MOTION_SWITCH = {
+ ("sensor", "motionActive"): BlockSwitchDescription(
+ key="sensor|motionActive",
+ name="Motion detection",
+ entity_category=EntityCategory.CONFIG,
+ )
+}
@dataclass(frozen=True, kw_only=True)
class RpcSwitchDescription(RpcEntityDescription, SwitchEntityDescription):
"""Class to describe a RPC virtual switch."""
+ is_on: Callable[[dict[str, Any]], bool]
+ method_on: str
+ method_off: str
+ method_params_fn: Callable[[int | None, bool], dict]
-RPC_VIRTUAL_SWITCH = RpcSwitchDescription(
- key="boolean",
- sub_key="value",
-)
-RPC_SCRIPT_SWITCH = RpcSwitchDescription(
- key="script",
- sub_key="running",
- entity_registry_enabled_default=False,
- entity_category=EntityCategory.CONFIG,
-)
+RPC_RELAY_SWITCHES = {
+ "switch": RpcSwitchDescription(
+ key="switch",
+ sub_key="output",
+ removal_condition=is_rpc_exclude_from_relay,
+ is_on=lambda status: bool(status["output"]),
+ method_on="Switch.Set",
+ method_off="Switch.Set",
+ method_params_fn=lambda id, value: {"id": id, "on": value},
+ ),
+}
+
+RPC_SWITCHES = {
+ "boolean": RpcSwitchDescription(
+ key="boolean",
+ sub_key="value",
+ is_on=lambda status: bool(status["value"]),
+ method_on="Boolean.Set",
+ method_off="Boolean.Set",
+ method_params_fn=lambda id, value: {"id": id, "value": value},
+ ),
+ "script": RpcSwitchDescription(
+ key="script",
+ sub_key="running",
+ is_on=lambda status: bool(status["running"]),
+ method_on="Script.Start",
+ method_off="Script.Stop",
+ method_params_fn=lambda id, _: {"id": id},
+ entity_registry_enabled_default=False,
+ entity_category=EntityCategory.CONFIG,
+ ),
+}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
@@ -90,104 +121,41 @@ async def async_setup_entry(
def async_setup_block_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for block device."""
coordinator = config_entry.runtime_data.block
assert coordinator
- # Add Shelly Motion as a switch
- if coordinator.model in MOTION_MODELS:
- async_setup_entry_attribute_entities(
- hass,
- config_entry,
- async_add_entities,
- {("sensor", "motionActive"): MOTION_SWITCH},
- BlockSleepingMotionSwitch,
- )
- return
+ async_setup_entry_attribute_entities(
+ hass, config_entry, async_add_entities, BLOCK_RELAY_SWITCHES, BlockRelaySwitch
+ )
- if config_entry.data[CONF_SLEEP_PERIOD]:
- return
-
- # In roller mode the relay blocks exist but do not contain required info
- if (
- coordinator.model in [MODEL_2, MODEL_25]
- and coordinator.device.settings["mode"] != "relay"
- ):
- return
-
- relay_blocks = []
- assert coordinator.device.blocks
- for block in coordinator.device.blocks:
- if block.type != "relay" or (
- block.channel is not None
- and is_block_channel_type_light(
- coordinator.device.settings, int(block.channel)
- )
- ):
- continue
-
- relay_blocks.append(block)
- unique_id = f"{coordinator.mac}-{block.type}_{block.channel}"
- async_remove_shelly_entity(hass, "light", unique_id)
-
- if not relay_blocks:
- return
-
- async_add_entities(BlockRelaySwitch(coordinator, block) for block in relay_blocks)
+ async_setup_entry_attribute_entities(
+ hass,
+ config_entry,
+ async_add_entities,
+ BLOCK_SLEEPING_MOTION_SWITCH,
+ BlockSleepingMotionSwitch,
+ )
@callback
def async_setup_rpc_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for RPC device."""
coordinator = config_entry.runtime_data.rpc
assert coordinator
- switch_key_ids = get_rpc_key_ids(coordinator.device.status, "switch")
- switch_ids = []
- for id_ in switch_key_ids:
- if is_rpc_channel_type_light(coordinator.device.config, id_):
- continue
-
- if coordinator.model == MODEL_WALL_DISPLAY:
- # There are three configuration scenarios for WallDisplay:
- # - relay mode (no thermostat)
- # - thermostat mode using the internal relay as an actuator
- # - thermostat mode using an external (from another device) relay as
- # an actuator
- if not is_rpc_thermostat_mode(id_, coordinator.device.status):
- # The device is not in thermostat mode, we need to remove a climate
- # entity
- unique_id = f"{coordinator.mac}-thermostat:{id_}"
- async_remove_shelly_entity(hass, "climate", unique_id)
- elif is_rpc_thermostat_internal_actuator(coordinator.device.status):
- # The internal relay is an actuator, skip this ID so as not to create
- # a switch entity
- continue
-
- switch_ids.append(id_)
- unique_id = f"{coordinator.mac}-switch:{id_}"
- async_remove_shelly_entity(hass, "light", unique_id)
-
- async_setup_rpc_attribute_entities(
- hass,
- config_entry,
- async_add_entities,
- {"boolean": RPC_VIRTUAL_SWITCH},
- RpcVirtualSwitch,
+ async_setup_entry_rpc(
+ hass, config_entry, async_add_entities, RPC_RELAY_SWITCHES, RpcRelaySwitch
)
- async_setup_rpc_attribute_entities(
- hass,
- config_entry,
- async_add_entities,
- {"script": RPC_SCRIPT_SWITCH},
- RpcScriptSwitch,
+ async_setup_entry_rpc(
+ hass, config_entry, async_add_entities, RPC_SWITCHES, RpcSwitch
)
# the user can remove virtual components from the device configuration, so we need
@@ -215,10 +183,16 @@ def async_setup_rpc_entry(
"script",
)
- if not switch_ids:
- return
-
- async_add_entities(RpcRelaySwitch(coordinator, id_) for id_ in switch_ids)
+ # if the climate is removed, from the device configuration, we need
+ # to remove orphaned entities
+ async_remove_orphaned_entities(
+ hass,
+ config_entry.entry_id,
+ coordinator.mac,
+ CLIMATE_PLATFORM,
+ coordinator.device.status,
+ "thermostat",
+ )
class BlockSleepingMotionSwitch(
@@ -269,13 +243,22 @@ class BlockSleepingMotionSwitch(
self.last_state = last_state
-class BlockRelaySwitch(ShellyBlockEntity, SwitchEntity):
+class BlockRelaySwitch(ShellyBlockAttributeEntity, SwitchEntity):
"""Entity that controls a relay on Block based Shelly devices."""
- def __init__(self, coordinator: ShellyBlockCoordinator, block: Block) -> None:
+ entity_description: BlockSwitchDescription
+
+ def __init__(
+ self,
+ coordinator: ShellyBlockCoordinator,
+ block: Block,
+ attribute: str,
+ description: BlockSwitchDescription,
+ ) -> None:
"""Initialize relay switch."""
- super().__init__(coordinator, block)
+ super().__init__(coordinator, block, attribute, description)
self.control_result: dict[str, Any] | None = None
+ self._attr_unique_id: str = f"{coordinator.mac}-{block.description}"
@property
def is_on(self) -> bool:
@@ -302,30 +285,8 @@ class BlockRelaySwitch(ShellyBlockEntity, SwitchEntity):
super()._update_callback()
-class RpcRelaySwitch(ShellyRpcEntity, SwitchEntity):
- """Entity that controls a relay on RPC based Shelly devices."""
-
- def __init__(self, coordinator: ShellyRpcCoordinator, id_: int) -> None:
- """Initialize relay switch."""
- super().__init__(coordinator, f"switch:{id_}")
- self._id = id_
-
- @property
- def is_on(self) -> bool:
- """If switch is on."""
- return bool(self.status["output"])
-
- async def async_turn_on(self, **kwargs: Any) -> None:
- """Turn on relay."""
- await self.call_rpc("Switch.Set", {"id": self._id, "on": True})
-
- async def async_turn_off(self, **kwargs: Any) -> None:
- """Turn off relay."""
- await self.call_rpc("Switch.Set", {"id": self._id, "on": False})
-
-
-class RpcVirtualSwitch(ShellyRpcAttributeEntity, SwitchEntity):
- """Entity that controls a virtual boolean component on RPC based Shelly devices."""
+class RpcSwitch(ShellyRpcAttributeEntity, SwitchEntity):
+ """Entity that controls a switch on RPC based Shelly devices."""
entity_description: RpcSwitchDescription
_attr_has_entity_name = True
@@ -333,32 +294,36 @@ class RpcVirtualSwitch(ShellyRpcAttributeEntity, SwitchEntity):
@property
def is_on(self) -> bool:
"""If switch is on."""
- return bool(self.attribute_value)
+ return self.entity_description.is_on(self.status)
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn on relay."""
- await self.call_rpc("Boolean.Set", {"id": self._id, "value": True})
+ await self.call_rpc(
+ self.entity_description.method_on,
+ self.entity_description.method_params_fn(self._id, True),
+ )
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn off relay."""
- await self.call_rpc("Boolean.Set", {"id": self._id, "value": False})
+ await self.call_rpc(
+ self.entity_description.method_off,
+ self.entity_description.method_params_fn(self._id, False),
+ )
-class RpcScriptSwitch(ShellyRpcAttributeEntity, SwitchEntity):
- """Entity that controls a script component on RPC based Shelly devices."""
+class RpcRelaySwitch(RpcSwitch):
+ """Entity that controls a switch on RPC based Shelly devices."""
- entity_description: RpcSwitchDescription
- _attr_has_entity_name = True
+ # False to avoid double naming as True is inerithed from base class
+ _attr_has_entity_name = False
- @property
- def is_on(self) -> bool:
- """If switch is on."""
- return bool(self.status["running"])
-
- async def async_turn_on(self, **kwargs: Any) -> None:
- """Turn on relay."""
- await self.call_rpc("Script.Start", {"id": self._id})
-
- async def async_turn_off(self, **kwargs: Any) -> None:
- """Turn off relay."""
- await self.call_rpc("Script.Stop", {"id": self._id})
+ def __init__(
+ self,
+ coordinator: ShellyRpcCoordinator,
+ key: str,
+ attribute: str,
+ description: RpcEntityDescription,
+ ) -> None:
+ """Initialize the switch."""
+ super().__init__(coordinator, key, attribute, description)
+ self._attr_unique_id: str = f"{coordinator.mac}-{key}"
diff --git a/homeassistant/components/shelly/text.py b/homeassistant/components/shelly/text.py
index 66e2ee4c715..f64d1252b7e 100644
--- a/homeassistant/components/shelly/text.py
+++ b/homeassistant/components/shelly/text.py
@@ -13,7 +13,7 @@ from homeassistant.components.text import (
TextEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ShellyConfigEntry
from .entity import (
@@ -45,7 +45,7 @@ RPC_TEXT_ENTITIES: Final = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for device."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
diff --git a/homeassistant/components/shelly/update.py b/homeassistant/components/shelly/update.py
index f22547acf50..b1aa84b2640 100644
--- a/homeassistant/components/shelly/update.py
+++ b/homeassistant/components/shelly/update.py
@@ -22,7 +22,7 @@ from homeassistant.components.update import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import CONF_SLEEP_PERIOD, OTA_BEGIN, OTA_ERROR, OTA_PROGRESS, OTA_SUCCESS
@@ -104,7 +104,7 @@ RPC_UPDATES: Final = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up update entities for Shelly component."""
if get_device_entry_gen(config_entry) in RPC_GENERATIONS:
diff --git a/homeassistant/components/shelly/utils.py b/homeassistant/components/shelly/utils.py
index fa310104424..474e2bb9410 100644
--- a/homeassistant/components/shelly/utils.py
+++ b/homeassistant/components/shelly/utils.py
@@ -6,7 +6,7 @@ from collections.abc import Iterable
from datetime import datetime, timedelta
from ipaddress import IPv4Address, IPv6Address, ip_address
from types import MappingProxyType
-from typing import Any, cast
+from typing import TYPE_CHECKING, Any, cast
from aiohttp.web import Request, WebSocketResponse
from aioshelly.block_device import COAP, Block, BlockDevice
@@ -28,7 +28,12 @@ from yarl import URL
from homeassistant.components import network
from homeassistant.components.http import HomeAssistantView
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_PORT, EVENT_HOMEASSISTANT_STOP
+from homeassistant.const import (
+ CONF_HOST,
+ CONF_MODEL,
+ CONF_PORT,
+ EVENT_HOMEASSISTANT_STOP,
+)
from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.helpers import (
device_registry as dr,
@@ -54,6 +59,7 @@ from .const import (
GEN2_RELEASE_URL,
LOGGER,
RPC_INPUTS_EVENTS_TYPES,
+ SHAIR_MAX_WORK_HOURS,
SHBTN_INPUTS_EVENTS_TYPES,
SHBTN_MODELS,
SHELLY_EMIT_EVENT_PATTERN,
@@ -175,6 +181,18 @@ def is_block_momentary_input(
return button_type in momentary_types
+def is_block_exclude_from_relay(settings: dict[str, Any], block: Block) -> bool:
+ """Return true if block should be excluded from switch platform."""
+
+ if settings.get("mode") == "roller":
+ return True
+
+ if TYPE_CHECKING:
+ assert block.channel is not None
+
+ return is_block_channel_type_light(settings, int(block.channel))
+
+
def get_device_uptime(uptime: float, last_uptime: datetime | None) -> datetime:
"""Return device uptime string, tolerate up to 5 seconds deviation."""
delta_uptime = utcnow() - timedelta(seconds=uptime)
@@ -310,11 +328,32 @@ def get_info_gen(info: dict[str, Any]) -> int:
def get_model_name(info: dict[str, Any]) -> str:
"""Return the device model name."""
if get_info_gen(info) in RPC_GENERATIONS:
- return cast(str, MODEL_NAMES.get(info["model"], info["model"]))
+ return cast(str, MODEL_NAMES.get(info[CONF_MODEL], info[CONF_MODEL]))
return cast(str, MODEL_NAMES.get(info["type"], info["type"]))
+def get_shelly_model_name(
+ model: str,
+ sleep_period: int,
+ device: BlockDevice | RpcDevice,
+) -> str | None:
+ """Get Shelly model name.
+
+ Assume that XMOD devices are not sleepy devices.
+ """
+ if (
+ sleep_period == 0
+ and isinstance(device, RpcDevice)
+ and (model_name := device.xmod_info.get("n"))
+ ):
+ # Use the model name from XMOD data
+ return cast(str, model_name)
+
+ # Use the model name from aioshelly
+ return cast(str, MODEL_NAMES.get(model))
+
+
def get_rpc_channel_name(device: RpcDevice, key: str) -> str:
"""Get name based on device and channel name."""
key = key.replace("emdata", "em")
@@ -481,7 +520,7 @@ def async_create_issue_unsupported_firmware(
translation_key="unsupported_firmware",
translation_placeholders={
"device_name": entry.title,
- "ip_address": entry.data["host"],
+ "ip_address": entry.data[CONF_HOST],
},
)
@@ -606,3 +645,39 @@ async def get_rpc_script_event_types(device: RpcDevice, id: int) -> list[str]:
code_response = await device.script_getcode(id)
matches = SHELLY_EMIT_EVENT_PATTERN.finditer(code_response["data"])
return sorted([*{str(event_type.group(1)) for event_type in matches}])
+
+
+def is_rpc_exclude_from_relay(
+ settings: dict[str, Any], status: dict[str, Any], channel: str
+) -> bool:
+ """Return true if rpc channel should be excludeed from switch platform."""
+ ch = int(channel.split(":")[1])
+ if is_rpc_thermostat_internal_actuator(status):
+ return True
+
+ return is_rpc_channel_type_light(settings, ch)
+
+
+def get_shelly_air_lamp_life(lamp_seconds: int) -> float:
+ """Return Shelly Air lamp life in percentage."""
+ lamp_hours = lamp_seconds / 3600
+ if lamp_hours >= SHAIR_MAX_WORK_HOURS:
+ return 0.0
+ return 100 * (1 - lamp_hours / SHAIR_MAX_WORK_HOURS)
+
+
+async def get_rpc_scripts_event_types(
+ device: RpcDevice, ignore_scripts: list[str]
+) -> dict[int, list[str]]:
+ """Return a dict of all scripts and their event types."""
+ script_instances = get_rpc_key_instances(device.status, "script")
+ script_events = {}
+ for script in script_instances:
+ script_name = get_rpc_entity_name(device, script)
+ if script_name in ignore_scripts:
+ continue
+
+ script_id = int(script.split(":")[-1])
+ script_events[script_id] = await get_rpc_script_event_types(device, script_id)
+
+ return script_events
diff --git a/homeassistant/components/shelly/valve.py b/homeassistant/components/shelly/valve.py
index ea6feaabe69..1829f663b22 100644
--- a/homeassistant/components/shelly/valve.py
+++ b/homeassistant/components/shelly/valve.py
@@ -15,7 +15,7 @@ from homeassistant.components.valve import (
ValveEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import ShellyBlockCoordinator, ShellyConfigEntry
from .entity import (
@@ -42,7 +42,7 @@ GAS_VALVE = BlockValveDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up valves for device."""
if get_device_entry_gen(config_entry) in BLOCK_GENERATIONS:
@@ -53,7 +53,7 @@ async def async_setup_entry(
def async_setup_block_entry(
hass: HomeAssistant,
config_entry: ShellyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up valve for device."""
coordinator = config_entry.runtime_data.block
diff --git a/homeassistant/components/shopping_list/todo.py b/homeassistant/components/shopping_list/todo.py
index 82b6cbfc7f5..2952c283082 100644
--- a/homeassistant/components/shopping_list/todo.py
+++ b/homeassistant/components/shopping_list/todo.py
@@ -11,7 +11,7 @@ from homeassistant.components.todo import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import NoMatchingShoppingListItem, ShoppingData
from .const import DOMAIN
@@ -20,7 +20,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the shopping_list todo platform."""
shopping_data = hass.data[DOMAIN]
diff --git a/homeassistant/components/sia/alarm_control_panel.py b/homeassistant/components/sia/alarm_control_panel.py
index 7ea878f538d..bb6a0669a99 100644
--- a/homeassistant/components/sia/alarm_control_panel.py
+++ b/homeassistant/components/sia/alarm_control_panel.py
@@ -16,7 +16,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN
from homeassistant.core import HomeAssistant, State
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_ACCOUNT, CONF_ACCOUNTS, CONF_ZONES, KEY_ALARM, PREVIOUS_STATE
from .entity import SIABaseEntity, SIAEntityDescription
@@ -69,7 +69,7 @@ ENTITY_DESCRIPTION_ALARM = SIAAlarmControlPanelEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SIA alarm_control_panel(s) from a config entry."""
async_add_entities(
diff --git a/homeassistant/components/sia/binary_sensor.py b/homeassistant/components/sia/binary_sensor.py
index 4c8e4ca6130..e1b40dc2e55 100644
--- a/homeassistant/components/sia/binary_sensor.py
+++ b/homeassistant/components/sia/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNAVAILABLE, EntityCategory
from homeassistant.core import HomeAssistant, State, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_ACCOUNT,
@@ -105,7 +105,7 @@ def generate_binary_sensors(entry: ConfigEntry) -> Iterable[SIABinarySensor]:
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SIA binary sensors from a config entry."""
async_add_entities(generate_binary_sensors(entry))
diff --git a/homeassistant/components/simplefin/__init__.py b/homeassistant/components/simplefin/__init__.py
index c47b3118415..1fe2f2a6189 100644
--- a/homeassistant/components/simplefin/__init__.py
+++ b/homeassistant/components/simplefin/__init__.py
@@ -4,12 +4,11 @@ from __future__ import annotations
from simplefin4py import SimpleFin
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from .const import CONF_ACCESS_URL
-from .coordinator import SimpleFinDataUpdateCoordinator
+from .coordinator import SimpleFinConfigEntry, SimpleFinDataUpdateCoordinator
PLATFORMS: list[str] = [
Platform.BINARY_SENSOR,
@@ -17,20 +16,17 @@ PLATFORMS: list[str] = [
]
-type SimpleFinConfigEntry = ConfigEntry[SimpleFinDataUpdateCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: SimpleFinConfigEntry) -> bool:
"""Set up from a config entry."""
access_url = entry.data[CONF_ACCESS_URL]
sf_client = SimpleFin(access_url)
- sf_coordinator = SimpleFinDataUpdateCoordinator(hass, sf_client)
+ sf_coordinator = SimpleFinDataUpdateCoordinator(hass, entry, sf_client)
await sf_coordinator.async_config_entry_first_refresh()
entry.runtime_data = sf_coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: SimpleFinConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/simplefin/binary_sensor.py b/homeassistant/components/simplefin/binary_sensor.py
index 5805fc370b6..af97fe9a394 100644
--- a/homeassistant/components/simplefin/binary_sensor.py
+++ b/homeassistant/components/simplefin/binary_sensor.py
@@ -12,9 +12,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SimpleFinConfigEntry
+from .coordinator import SimpleFinConfigEntry
from .entity import SimpleFinEntity
@@ -39,7 +39,7 @@ SIMPLEFIN_BINARY_SENSORS: tuple[SimpleFinBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SimpleFinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SimpleFIN sensors for config entries."""
diff --git a/homeassistant/components/simplefin/coordinator.py b/homeassistant/components/simplefin/coordinator.py
index 7fa5aedb7a1..08e9732c6b7 100644
--- a/homeassistant/components/simplefin/coordinator.py
+++ b/homeassistant/components/simplefin/coordinator.py
@@ -15,17 +15,22 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import LOGGER
+type SimpleFinConfigEntry = ConfigEntry[SimpleFinDataUpdateCoordinator]
+
class SimpleFinDataUpdateCoordinator(DataUpdateCoordinator[FinancialData]):
"""Data update coordinator for the SimpleFIN integration."""
- config_entry: ConfigEntry
+ config_entry: SimpleFinConfigEntry
- def __init__(self, hass: HomeAssistant, client: SimpleFin) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: SimpleFinConfigEntry, client: SimpleFin
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name="simplefin",
update_interval=timedelta(hours=4),
)
diff --git a/homeassistant/components/simplefin/sensor.py b/homeassistant/components/simplefin/sensor.py
index b2167a2c014..183a198040b 100644
--- a/homeassistant/components/simplefin/sensor.py
+++ b/homeassistant/components/simplefin/sensor.py
@@ -16,10 +16,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import SimpleFinConfigEntry
+from .coordinator import SimpleFinConfigEntry
from .entity import SimpleFinEntity
@@ -55,7 +55,7 @@ SIMPLEFIN_SENSORS: tuple[SimpleFinSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SimpleFinConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SimpleFIN sensors for config entries."""
diff --git a/homeassistant/components/simplisafe/__init__.py b/homeassistant/components/simplisafe/__init__.py
index 2f19c5117a4..8a75baa69c6 100644
--- a/homeassistant/components/simplisafe/__init__.py
+++ b/homeassistant/components/simplisafe/__init__.py
@@ -39,7 +39,7 @@ from simplipy.websocket import (
)
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ATTR_CODE,
ATTR_DEVICE_ID,
@@ -402,12 +402,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
if unload_ok:
hass.data[DOMAIN].pop(entry.entry_id)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
# If this is the last loaded instance of SimpliSafe, deregister any services
# defined during integration setup:
for service_name in SERVICES:
diff --git a/homeassistant/components/simplisafe/alarm_control_panel.py b/homeassistant/components/simplisafe/alarm_control_panel.py
index 18f2d8ddcd5..c5a1b2bc708 100644
--- a/homeassistant/components/simplisafe/alarm_control_panel.py
+++ b/homeassistant/components/simplisafe/alarm_control_panel.py
@@ -31,7 +31,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SimpliSafe
from .const import (
@@ -103,7 +103,9 @@ WEBSOCKET_EVENTS_TO_LISTEN_FOR = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a SimpliSafe alarm control panel based on a config entry."""
simplisafe = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/simplisafe/binary_sensor.py b/homeassistant/components/simplisafe/binary_sensor.py
index 0310e958e6e..38a80ddd354 100644
--- a/homeassistant/components/simplisafe/binary_sensor.py
+++ b/homeassistant/components/simplisafe/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SimpliSafe
from .const import DOMAIN, LOGGER
@@ -34,6 +34,7 @@ SUPPORTED_BATTERY_SENSOR_TYPES = [
DeviceTypes.PANIC_BUTTON,
DeviceTypes.REMOTE,
DeviceTypes.SIREN,
+ DeviceTypes.OUTDOOR_ALARM_SECURITY_BELL_BOX,
DeviceTypes.SMOKE,
DeviceTypes.SMOKE_AND_CARBON_MONOXIDE,
DeviceTypes.TEMPERATURE,
@@ -47,6 +48,7 @@ TRIGGERED_SENSOR_TYPES = {
DeviceTypes.MOTION: BinarySensorDeviceClass.MOTION,
DeviceTypes.MOTION_V2: BinarySensorDeviceClass.MOTION,
DeviceTypes.SIREN: BinarySensorDeviceClass.SAFETY,
+ DeviceTypes.OUTDOOR_ALARM_SECURITY_BELL_BOX: BinarySensorDeviceClass.SAFETY,
DeviceTypes.SMOKE: BinarySensorDeviceClass.SMOKE,
# Although this sensor can technically apply to both smoke and carbon, we use the
# SMOKE device class for simplicity:
@@ -55,7 +57,9 @@ TRIGGERED_SENSOR_TYPES = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SimpliSafe binary sensors based on a config entry."""
simplisafe = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/simplisafe/button.py b/homeassistant/components/simplisafe/button.py
index f0272d09f61..129209354c3 100644
--- a/homeassistant/components/simplisafe/button.py
+++ b/homeassistant/components/simplisafe/button.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SimpliSafe
from .const import DOMAIN
@@ -46,7 +46,9 @@ BUTTON_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SimpliSafe buttons based on a config entry."""
simplisafe = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/simplisafe/lock.py b/homeassistant/components/simplisafe/lock.py
index c610223bff1..9e29bb2051b 100644
--- a/homeassistant/components/simplisafe/lock.py
+++ b/homeassistant/components/simplisafe/lock.py
@@ -13,7 +13,7 @@ from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SimpliSafe
from .const import DOMAIN, LOGGER
@@ -31,7 +31,9 @@ WEBSOCKET_EVENTS_TO_LISTEN_FOR = (EVENT_LOCK_LOCKED, EVENT_LOCK_UNLOCKED)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SimpliSafe locks based on a config entry."""
simplisafe = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/simplisafe/sensor.py b/homeassistant/components/simplisafe/sensor.py
index a5f46e87a7c..b82162f0fe7 100644
--- a/homeassistant/components/simplisafe/sensor.py
+++ b/homeassistant/components/simplisafe/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SimpliSafe
from .const import DOMAIN, LOGGER
@@ -22,7 +22,9 @@ from .entity import SimpliSafeEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SimpliSafe freeze sensors based on a config entry."""
simplisafe = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sky_remote/config_flow.py b/homeassistant/components/sky_remote/config_flow.py
index 13cddf99332..51cf9c9bf64 100644
--- a/homeassistant/components/sky_remote/config_flow.py
+++ b/homeassistant/components/sky_remote/config_flow.py
@@ -12,6 +12,8 @@ from homeassistant.helpers import config_validation as cv
from .const import DEFAULT_PORT, DOMAIN, LEGACY_PORT
+_LOGGER = logging.getLogger(__name__)
+
DATA_SCHEMA = vol.Schema(
{
vol.Required(CONF_HOST): cv.string,
@@ -21,7 +23,7 @@ DATA_SCHEMA = vol.Schema(
async def async_find_box_port(host: str) -> int:
"""Find port box uses for communication."""
- logging.debug("Attempting to find port to connect to %s on", host)
+ _LOGGER.debug("Attempting to find port to connect to %s on", host)
remote = RemoteControl(host, DEFAULT_PORT)
try:
await remote.check_connectable()
@@ -46,12 +48,12 @@ class SkyRemoteConfigFlow(ConfigFlow, domain=DOMAIN):
errors: dict[str, str] = {}
if user_input is not None:
- logging.debug("user_input: %s", user_input)
+ _LOGGER.debug("user_input: %s", user_input)
self._async_abort_entries_match(user_input)
try:
port = await async_find_box_port(user_input[CONF_HOST])
except SkyBoxConnectionError:
- logging.exception("while finding port of skybox")
+ _LOGGER.exception("While finding port of skybox")
errors["base"] = "cannot_connect"
else:
return self.async_create_entry(
diff --git a/homeassistant/components/sky_remote/remote.py b/homeassistant/components/sky_remote/remote.py
index 05a464f73a6..1ecd6c3716e 100644
--- a/homeassistant/components/sky_remote/remote.py
+++ b/homeassistant/components/sky_remote/remote.py
@@ -10,7 +10,7 @@ from homeassistant.components.remote import RemoteEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SkyRemoteConfigEntry
from .const import DOMAIN
@@ -21,7 +21,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config: SkyRemoteConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Sky remote platform."""
async_add_entities(
diff --git a/homeassistant/components/skybell/__init__.py b/homeassistant/components/skybell/__init__.py
index 0282ad40254..5baa4ad83ad 100644
--- a/homeassistant/components/skybell/__init__.py
+++ b/homeassistant/components/skybell/__init__.py
@@ -45,7 +45,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
raise ConfigEntryNotReady(f"Unable to connect to Skybell service: {ex}") from ex
device_coordinators: list[SkybellDataUpdateCoordinator] = [
- SkybellDataUpdateCoordinator(hass, device) for device in devices
+ SkybellDataUpdateCoordinator(hass, entry, device) for device in devices
]
await asyncio.gather(
*[
diff --git a/homeassistant/components/skybell/binary_sensor.py b/homeassistant/components/skybell/binary_sensor.py
index 3c2d90b2630..cc42da48b26 100644
--- a/homeassistant/components/skybell/binary_sensor.py
+++ b/homeassistant/components/skybell/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
from .coordinator import SkybellDataUpdateCoordinator
@@ -31,7 +31,9 @@ BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Skybell binary sensor."""
async_add_entities(
diff --git a/homeassistant/components/skybell/camera.py b/homeassistant/components/skybell/camera.py
index 683b840debe..4ee873f8350 100644
--- a/homeassistant/components/skybell/camera.py
+++ b/homeassistant/components/skybell/camera.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SkybellDataUpdateCoordinator
@@ -30,7 +30,9 @@ CAMERA_TYPES: tuple[CameraEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Skybell camera."""
entities = []
diff --git a/homeassistant/components/skybell/coordinator.py b/homeassistant/components/skybell/coordinator.py
index 55e34df5c63..48e67c63ac9 100644
--- a/homeassistant/components/skybell/coordinator.py
+++ b/homeassistant/components/skybell/coordinator.py
@@ -16,11 +16,14 @@ class SkybellDataUpdateCoordinator(DataUpdateCoordinator[None]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, device: SkybellDevice) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, device: SkybellDevice
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=device.name,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/skybell/light.py b/homeassistant/components/skybell/light.py
index cba9e70c848..3f924f68da8 100644
--- a/homeassistant/components/skybell/light.py
+++ b/homeassistant/components/skybell/light.py
@@ -15,14 +15,16 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import SkybellEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Skybell switch."""
async_add_entities(
diff --git a/homeassistant/components/skybell/sensor.py b/homeassistant/components/skybell/sensor.py
index 5f0df77ecfa..a67fdae3b35 100644
--- a/homeassistant/components/skybell/sensor.py
+++ b/homeassistant/components/skybell/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .entity import DOMAIN, SkybellEntity
@@ -88,7 +88,9 @@ SENSOR_TYPES: tuple[SkybellSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Skybell sensor."""
async_add_entities(
diff --git a/homeassistant/components/skybell/switch.py b/homeassistant/components/skybell/switch.py
index fa4f723573f..858363043ca 100644
--- a/homeassistant/components/skybell/switch.py
+++ b/homeassistant/components/skybell/switch.py
@@ -7,7 +7,7 @@ from typing import Any, cast
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import SkybellEntity
@@ -29,7 +29,9 @@ SWITCH_TYPES: tuple[SwitchEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SkyBell switch."""
async_add_entities(
diff --git a/homeassistant/components/slack/sensor.py b/homeassistant/components/slack/sensor.py
index ca8c9830818..042ab00916e 100644
--- a/homeassistant/components/slack/sensor.py
+++ b/homeassistant/components/slack/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import ATTR_SNOOZE, DOMAIN, SLACK_DATA
@@ -21,7 +21,7 @@ from .entity import SlackEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Slack select."""
async_add_entities(
diff --git a/homeassistant/components/sleepiq/__init__.py b/homeassistant/components/sleepiq/__init__.py
index 4f54b4cd305..565611fe169 100644
--- a/homeassistant/components/sleepiq/__init__.py
+++ b/homeassistant/components/sleepiq/__init__.py
@@ -94,8 +94,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await _async_migrate_unique_ids(hass, entry, gateway)
- coordinator = SleepIQDataUpdateCoordinator(hass, gateway, email)
- pause_coordinator = SleepIQPauseUpdateCoordinator(hass, gateway, email)
+ coordinator = SleepIQDataUpdateCoordinator(hass, entry, gateway)
+ pause_coordinator = SleepIQPauseUpdateCoordinator(hass, entry, gateway)
# Call the SleepIQ API to refresh data
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/sleepiq/binary_sensor.py b/homeassistant/components/sleepiq/binary_sensor.py
index cb56a516b9b..99fff9c49b0 100644
--- a/homeassistant/components/sleepiq/binary_sensor.py
+++ b/homeassistant/components/sleepiq/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ICON_EMPTY, ICON_OCCUPIED, IS_IN_BED
from .coordinator import SleepIQData, SleepIQDataUpdateCoordinator
@@ -18,7 +18,7 @@ from .entity import SleepIQSleeperEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SleepIQ bed binary sensors."""
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sleepiq/button.py b/homeassistant/components/sleepiq/button.py
index 94b010066c9..74b1bc0789f 100644
--- a/homeassistant/components/sleepiq/button.py
+++ b/homeassistant/components/sleepiq/button.py
@@ -11,7 +11,7 @@ from asyncsleepiq import SleepIQBed
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SleepIQData
@@ -44,7 +44,7 @@ ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sleep number buttons."""
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sleepiq/coordinator.py b/homeassistant/components/sleepiq/coordinator.py
index 7fe4f964b36..46b754976e5 100644
--- a/homeassistant/components/sleepiq/coordinator.py
+++ b/homeassistant/components/sleepiq/coordinator.py
@@ -7,6 +7,8 @@ import logging
from asyncsleepiq import AsyncSleepIQ
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -19,17 +21,20 @@ LONGER_UPDATE_INTERVAL = timedelta(minutes=5)
class SleepIQDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""SleepIQ data update coordinator."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
client: AsyncSleepIQ,
- username: str,
) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
_LOGGER,
- name=f"{username}@SleepIQ",
+ config_entry=config_entry,
+ name=f"{config_entry.data[CONF_USERNAME]}@SleepIQ",
update_interval=UPDATE_INTERVAL,
)
self.client = client
@@ -45,17 +50,20 @@ class SleepIQDataUpdateCoordinator(DataUpdateCoordinator[None]):
class SleepIQPauseUpdateCoordinator(DataUpdateCoordinator[None]):
"""SleepIQ data update coordinator."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
client: AsyncSleepIQ,
- username: str,
) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
_LOGGER,
- name=f"{username}@SleepIQPause",
+ config_entry=config_entry,
+ name=f"{config_entry.data[CONF_USERNAME]}@SleepIQPause",
update_interval=LONGER_UPDATE_INTERVAL,
)
self.client = client
diff --git a/homeassistant/components/sleepiq/light.py b/homeassistant/components/sleepiq/light.py
index 781bd8e600a..542c212df27 100644
--- a/homeassistant/components/sleepiq/light.py
+++ b/homeassistant/components/sleepiq/light.py
@@ -8,7 +8,7 @@ from asyncsleepiq import SleepIQBed, SleepIQLight
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SleepIQData, SleepIQDataUpdateCoordinator
@@ -20,7 +20,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SleepIQ bed lights."""
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sleepiq/number.py b/homeassistant/components/sleepiq/number.py
index 905ceab18bd..53d6c366e46 100644
--- a/homeassistant/components/sleepiq/number.py
+++ b/homeassistant/components/sleepiq/number.py
@@ -17,7 +17,7 @@ from asyncsleepiq import (
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ACTUATOR,
@@ -138,7 +138,7 @@ NUMBER_DESCRIPTIONS: dict[str, SleepIQNumberEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SleepIQ bed sensors."""
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sleepiq/select.py b/homeassistant/components/sleepiq/select.py
index 0a09aa4d657..7d059ba6b59 100644
--- a/homeassistant/components/sleepiq/select.py
+++ b/homeassistant/components/sleepiq/select.py
@@ -13,7 +13,7 @@ from asyncsleepiq import (
from homeassistant.components.select import SelectEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, FOOT_WARMER
from .coordinator import SleepIQData, SleepIQDataUpdateCoordinator
@@ -23,7 +23,7 @@ from .entity import SleepIQBedEntity, SleepIQSleeperEntity, sleeper_for_side
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SleepIQ foundation preset select entities."""
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sleepiq/sensor.py b/homeassistant/components/sleepiq/sensor.py
index 413e8e4d856..ca4fbc186ed 100644
--- a/homeassistant/components/sleepiq/sensor.py
+++ b/homeassistant/components/sleepiq/sensor.py
@@ -7,7 +7,7 @@ from asyncsleepiq import SleepIQBed, SleepIQSleeper
from homeassistant.components.sensor import SensorEntity, SensorStateClass
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, PRESSURE, SLEEP_NUMBER
from .coordinator import SleepIQData, SleepIQDataUpdateCoordinator
@@ -19,7 +19,7 @@ SENSORS = [PRESSURE, SLEEP_NUMBER]
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SleepIQ bed sensors."""
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/sleepiq/switch.py b/homeassistant/components/sleepiq/switch.py
index 9fc8ca9d20e..8363782c064 100644
--- a/homeassistant/components/sleepiq/switch.py
+++ b/homeassistant/components/sleepiq/switch.py
@@ -9,7 +9,7 @@ from asyncsleepiq import SleepIQBed
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SleepIQData, SleepIQPauseUpdateCoordinator
@@ -19,7 +19,7 @@ from .entity import SleepIQBedEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sleep number switches."""
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/slide_local/__init__.py b/homeassistant/components/slide_local/__init__.py
index 5b4867bf337..4690fe8016c 100644
--- a/homeassistant/components/slide_local/__init__.py
+++ b/homeassistant/components/slide_local/__init__.py
@@ -2,14 +2,12 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .coordinator import SlideCoordinator
+from .coordinator import SlideConfigEntry, SlideCoordinator
PLATFORMS = [Platform.BUTTON, Platform.COVER, Platform.SWITCH]
-type SlideConfigEntry = ConfigEntry[SlideCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: SlideConfigEntry) -> bool:
diff --git a/homeassistant/components/slide_local/button.py b/homeassistant/components/slide_local/button.py
index faca7cb3f2b..3d5de33303d 100644
--- a/homeassistant/components/slide_local/button.py
+++ b/homeassistant/components/slide_local/button.py
@@ -13,11 +13,10 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SlideConfigEntry
from .const import DOMAIN
-from .coordinator import SlideCoordinator
+from .coordinator import SlideConfigEntry, SlideCoordinator
from .entity import SlideEntity
PARALLEL_UPDATES = 1
@@ -26,7 +25,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: SlideConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button for Slide platform."""
diff --git a/homeassistant/components/slide_local/config_flow.py b/homeassistant/components/slide_local/config_flow.py
index 4ceb347568f..96aac1a135c 100644
--- a/homeassistant/components/slide_local/config_flow.py
+++ b/homeassistant/components/slide_local/config_flow.py
@@ -20,8 +20,8 @@ from homeassistant.core import callback
from homeassistant.helpers.device_registry import format_mac
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
-from . import SlideConfigEntry
from .const import CONF_INVERT_POSITION, DOMAIN
+from .coordinator import SlideConfigEntry
_LOGGER = logging.getLogger(__name__)
diff --git a/homeassistant/components/slide_local/coordinator.py b/homeassistant/components/slide_local/coordinator.py
index e5311967198..cbc3e653739 100644
--- a/homeassistant/components/slide_local/coordinator.py
+++ b/homeassistant/components/slide_local/coordinator.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from datetime import timedelta
import logging
-from typing import TYPE_CHECKING, Any
+from typing import Any
from goslideapi.goslideapi import (
AuthenticationFailed,
@@ -14,6 +14,7 @@ from goslideapi.goslideapi import (
GoSlideLocal as SlideLocalApi,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_API_VERSION,
CONF_HOST,
@@ -31,23 +32,30 @@ from .const import DEFAULT_OFFSET, DOMAIN
_LOGGER = logging.getLogger(__name__)
-if TYPE_CHECKING:
- from . import SlideConfigEntry
+type SlideConfigEntry = ConfigEntry[SlideCoordinator]
class SlideCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Get and update the latest data."""
- def __init__(self, hass: HomeAssistant, entry: SlideConfigEntry) -> None:
+ config_entry: SlideConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: SlideConfigEntry) -> None:
"""Initialize the data object."""
super().__init__(
- hass, _LOGGER, name="Slide", update_interval=timedelta(seconds=15)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name="Slide",
+ update_interval=timedelta(seconds=15),
)
self.slide = SlideLocalApi()
- self.api_version = entry.data[CONF_API_VERSION]
- self.mac = entry.data[CONF_MAC]
- self.host = entry.data[CONF_HOST]
- self.password = entry.data[CONF_PASSWORD] if self.api_version == 1 else ""
+ self.api_version = config_entry.data[CONF_API_VERSION]
+ self.mac = config_entry.data[CONF_MAC]
+ self.host = config_entry.data[CONF_HOST]
+ self.password = (
+ config_entry.data[CONF_PASSWORD] if self.api_version == 1 else ""
+ )
async def _async_setup(self) -> None:
"""Do initialization logic for Slide coordinator."""
diff --git a/homeassistant/components/slide_local/cover.py b/homeassistant/components/slide_local/cover.py
index cf04f46d139..6bb3f338cb8 100644
--- a/homeassistant/components/slide_local/cover.py
+++ b/homeassistant/components/slide_local/cover.py
@@ -8,11 +8,10 @@ from typing import Any
from homeassistant.components.cover import ATTR_POSITION, CoverDeviceClass, CoverEntity
from homeassistant.const import STATE_CLOSED, STATE_CLOSING, STATE_OPENING
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SlideConfigEntry
from .const import CONF_INVERT_POSITION, DEFAULT_OFFSET
-from .coordinator import SlideCoordinator
+from .coordinator import SlideConfigEntry, SlideCoordinator
from .entity import SlideEntity
_LOGGER = logging.getLogger(__name__)
@@ -23,7 +22,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: SlideConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up cover(s) for Slide platform."""
diff --git a/homeassistant/components/slide_local/diagnostics.py b/homeassistant/components/slide_local/diagnostics.py
index 2655cb5fada..6a70720a14a 100644
--- a/homeassistant/components/slide_local/diagnostics.py
+++ b/homeassistant/components/slide_local/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_PASSWORD
from homeassistant.core import HomeAssistant
-from . import SlideConfigEntry
+from .coordinator import SlideConfigEntry
TO_REDACT = [
CONF_PASSWORD,
diff --git a/homeassistant/components/slide_local/switch.py b/homeassistant/components/slide_local/switch.py
index 0471dfcc4e6..e83924c87ee 100644
--- a/homeassistant/components/slide_local/switch.py
+++ b/homeassistant/components/slide_local/switch.py
@@ -15,11 +15,10 @@ from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SlideConfigEntry
from .const import DOMAIN
-from .coordinator import SlideCoordinator
+from .coordinator import SlideConfigEntry, SlideCoordinator
from .entity import SlideEntity
PARALLEL_UPDATES = 1
@@ -28,7 +27,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: SlideConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch for Slide platform."""
diff --git a/homeassistant/components/slimproto/media_player.py b/homeassistant/components/slimproto/media_player.py
index 42c50d21e75..417444961fe 100644
--- a/homeassistant/components/slimproto/media_player.py
+++ b/homeassistant/components/slimproto/media_player.py
@@ -22,7 +22,7 @@ from homeassistant.components.media_player import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import utcnow
from .const import DEFAULT_NAME, DOMAIN, PLAYER_EVENT
@@ -39,7 +39,7 @@ STATE_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SlimProto MediaPlayer(s) from Config Entry."""
slimserver: SlimServer = hass.data[DOMAIN]
diff --git a/homeassistant/components/sma/sensor.py b/homeassistant/components/sma/sensor.py
index 863f15a9a17..ffef026aaed 100644
--- a/homeassistant/components/sma/sensor.py
+++ b/homeassistant/components/sma/sensor.py
@@ -27,7 +27,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
@@ -838,7 +838,7 @@ SENSOR_ENTITIES: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SMA sensors."""
sma_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/smappee/binary_sensor.py b/homeassistant/components/smappee/binary_sensor.py
index 86bc225dba1..06dcaa62853 100644
--- a/homeassistant/components/smappee/binary_sensor.py
+++ b/homeassistant/components/smappee/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SmappeeConfigEntry
from .const import DOMAIN
@@ -37,7 +37,7 @@ ICON_MAPPING = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SmappeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smappee binary sensor."""
smappee_base = config_entry.runtime_data
diff --git a/homeassistant/components/smappee/sensor.py b/homeassistant/components/smappee/sensor.py
index 2f9d6443568..759dfb34013 100644
--- a/homeassistant/components/smappee/sensor.py
+++ b/homeassistant/components/smappee/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfElectricPotential, UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SmappeeConfigEntry
from .const import DOMAIN
@@ -189,7 +189,7 @@ VOLTAGE_SENSORS: tuple[SmappeeVoltageSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SmappeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smappee sensor."""
smappee_base = config_entry.runtime_data
diff --git a/homeassistant/components/smappee/switch.py b/homeassistant/components/smappee/switch.py
index bccf816c823..cf2ddea5938 100644
--- a/homeassistant/components/smappee/switch.py
+++ b/homeassistant/components/smappee/switch.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SmappeeConfigEntry
from .const import DOMAIN
@@ -16,7 +16,7 @@ SWITCH_PREFIX = "Switch"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SmappeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smappee Comfort Plugs."""
smappee_base = config_entry.runtime_data
diff --git a/homeassistant/components/smart_meter_texas/__init__.py b/homeassistant/components/smart_meter_texas/__init__.py
index 1cd7df68e91..ce87b85c322 100644
--- a/homeassistant/components/smart_meter_texas/__init__.py
+++ b/homeassistant/components/smart_meter_texas/__init__.py
@@ -3,7 +3,7 @@
import logging
import ssl
-from smart_meter_texas import Account, Client, ClientSSLContext
+from smart_meter_texas import Account, Client
from smart_meter_texas.exceptions import (
SmartMeterTexasAPIError,
SmartMeterTexasAuthError,
@@ -16,6 +16,7 @@ from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+from homeassistant.util.ssl import get_default_context
from .const import (
DATA_COORDINATOR,
@@ -38,8 +39,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
account = Account(username, password)
- client_ssl_context = ClientSSLContext()
- ssl_context = await client_ssl_context.get_ssl_context()
+ ssl_context = get_default_context()
smart_meter_texas_data = SmartMeterTexasData(hass, entry, account, ssl_context)
try:
diff --git a/homeassistant/components/smart_meter_texas/config_flow.py b/homeassistant/components/smart_meter_texas/config_flow.py
index b60855b62c8..18a3716e1b9 100644
--- a/homeassistant/components/smart_meter_texas/config_flow.py
+++ b/homeassistant/components/smart_meter_texas/config_flow.py
@@ -4,7 +4,7 @@ import logging
from typing import Any
from aiohttp import ClientError
-from smart_meter_texas import Account, Client, ClientSSLContext
+from smart_meter_texas import Account, Client
from smart_meter_texas.exceptions import (
SmartMeterTexasAPIError,
SmartMeterTexasAuthError,
@@ -16,6 +16,7 @@ from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import aiohttp_client
+from homeassistant.util.ssl import get_default_context
from .const import DOMAIN
@@ -31,8 +32,7 @@ async def validate_input(hass: HomeAssistant, data):
Data has the keys from DATA_SCHEMA with values provided by the user.
"""
- client_ssl_context = ClientSSLContext()
- ssl_context = await client_ssl_context.get_ssl_context()
+ ssl_context = get_default_context()
client_session = aiohttp_client.async_get_clientsession(hass)
account = Account(data["username"], data["password"])
client = Client(client_session, account, ssl_context)
diff --git a/homeassistant/components/smart_meter_texas/sensor.py b/homeassistant/components/smart_meter_texas/sensor.py
index 80fc79671b5..c6e18bf43c1 100644
--- a/homeassistant/components/smart_meter_texas/sensor.py
+++ b/homeassistant/components/smart_meter_texas/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ADDRESS, UnitOfEnergy
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
@@ -30,7 +30,7 @@ from .const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smart Meter Texas sensors."""
coordinator = hass.data[DOMAIN][config_entry.entry_id][DATA_COORDINATOR]
diff --git a/homeassistant/components/smartthings/__init__.py b/homeassistant/components/smartthings/__init__.py
index 2914851ccbf..e5351798219 100644
--- a/homeassistant/components/smartthings/__init__.py
+++ b/homeassistant/components/smartthings/__init__.py
@@ -2,416 +2,379 @@
from __future__ import annotations
-import asyncio
-from collections.abc import Iterable
+from collections.abc import Callable
+from dataclasses import dataclass
from http import HTTPStatus
-import importlib
import logging
+from typing import TYPE_CHECKING, Any, cast
-from aiohttp.client_exceptions import ClientConnectionError, ClientResponseError
-from pysmartapp.event import EVENT_TYPE_DEVICE
-from pysmartthings import APIInvalidGrant, Attribute, Capability, SmartThings
-
-from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
-from homeassistant.const import CONF_ACCESS_TOKEN, CONF_CLIENT_ID, CONF_CLIENT_SECRET
-from homeassistant.core import HomeAssistant
-from homeassistant.exceptions import (
- ConfigEntryAuthFailed,
- ConfigEntryError,
- ConfigEntryNotReady,
+from aiohttp import ClientResponseError
+from pysmartthings import (
+ Attribute,
+ Capability,
+ Device,
+ DeviceEvent,
+ Scene,
+ SmartThings,
+ SmartThingsAuthenticationFailedError,
+ SmartThingsSinkError,
+ Status,
+)
+from pysmartthings.models import Lifecycle
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import (
+ ATTR_CONNECTIONS,
+ ATTR_HW_VERSION,
+ ATTR_MANUFACTURER,
+ ATTR_MODEL,
+ ATTR_SW_VERSION,
+ ATTR_VIA_DEVICE,
+ CONF_ACCESS_TOKEN,
+ CONF_TOKEN,
+ EVENT_HOMEASSISTANT_STOP,
+ Platform,
+)
+from homeassistant.core import Event, HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.config_entry_oauth2_flow import (
+ OAuth2Session,
+ async_get_config_entry_implementation,
)
-from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.dispatcher import async_dispatcher_send
-from homeassistant.helpers.event import async_track_time_interval
-from homeassistant.helpers.typing import ConfigType
-from homeassistant.loader import async_get_loaded_integration
-from homeassistant.setup import SetupPhases, async_pause_setup
-from .config_flow import SmartThingsFlowHandler # noqa: F401
from .const import (
- CONF_APP_ID,
CONF_INSTALLED_APP_ID,
CONF_LOCATION_ID,
- CONF_REFRESH_TOKEN,
- DATA_BROKERS,
- DATA_MANAGER,
+ CONF_SUBSCRIPTION_ID,
DOMAIN,
EVENT_BUTTON,
- PLATFORMS,
- SIGNAL_SMARTTHINGS_UPDATE,
- TOKEN_REFRESH_INTERVAL,
-)
-from .smartapp import (
- format_unique_id,
- setup_smartapp,
- setup_smartapp_endpoint,
- smartapp_sync_subscriptions,
- unload_smartapp_endpoint,
- validate_installed_app,
- validate_webhook_requirements,
+ MAIN,
+ OLD_DATA,
)
_LOGGER = logging.getLogger(__name__)
-CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
+
+@dataclass
+class SmartThingsData:
+ """Define an object to hold SmartThings data."""
+
+ devices: dict[str, FullDevice]
+ scenes: dict[str, Scene]
+ rooms: dict[str, str]
+ client: SmartThings
-async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
- """Initialize the SmartThings platform."""
- await setup_smartapp_endpoint(hass, False)
- return True
+@dataclass
+class FullDevice:
+ """Define an object to hold device data."""
+
+ device: Device
+ status: dict[str, dict[Capability | str, dict[Attribute | str, Status]]]
-async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
- """Handle migration of a previous version config entry.
+type SmartThingsConfigEntry = ConfigEntry[SmartThingsData]
- A config entry created under a previous version must go through the
- integration setup again so we can properly retrieve the needed data
- elements. Force this by removing the entry and triggering a new flow.
- """
- # Remove the entry which will invoke the callback to delete the app.
- hass.async_create_task(hass.config_entries.async_remove(entry.entry_id))
- # only create new flow if there isn't a pending one for SmartThings.
- if not hass.config_entries.flow.async_progress_by_handler(DOMAIN):
- hass.async_create_task(
- hass.config_entries.flow.async_init(
- DOMAIN, context={"source": SOURCE_IMPORT}
- )
- )
-
- # Return False because it could not be migrated.
- return False
+PLATFORMS = [
+ Platform.BINARY_SENSOR,
+ Platform.BUTTON,
+ Platform.CLIMATE,
+ Platform.COVER,
+ Platform.EVENT,
+ Platform.FAN,
+ Platform.LIGHT,
+ Platform.LOCK,
+ Platform.NUMBER,
+ Platform.SCENE,
+ Platform.SELECT,
+ Platform.SENSOR,
+ Platform.SWITCH,
+ Platform.UPDATE,
+ Platform.VALVE,
+]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: SmartThingsConfigEntry) -> bool:
"""Initialize config entry which represents an installed SmartApp."""
- # For backwards compat
- if entry.unique_id is None:
+ # The oauth smartthings entry will have a token, older ones are version 3
+ # after migration but still require reauthentication
+ if CONF_TOKEN not in entry.data:
+ raise ConfigEntryAuthFailed("Config entry missing token")
+ implementation = await async_get_config_entry_implementation(hass, entry)
+ session = OAuth2Session(hass, entry, implementation)
+
+ try:
+ await session.async_ensure_token_valid()
+ except ClientResponseError as err:
+ if err.status == HTTPStatus.BAD_REQUEST:
+ raise ConfigEntryAuthFailed("Token not valid, trigger renewal") from err
+ raise ConfigEntryNotReady from err
+
+ client = SmartThings(session=async_get_clientsession(hass))
+
+ async def _refresh_token() -> str:
+ await session.async_ensure_token_valid()
+ token = session.token[CONF_ACCESS_TOKEN]
+ if TYPE_CHECKING:
+ assert isinstance(token, str)
+ return token
+
+ client.refresh_token_function = _refresh_token
+
+ def _handle_max_connections() -> None:
+ _LOGGER.debug("We hit the limit of max connections")
+ hass.config_entries.async_schedule_reload(entry.entry_id)
+
+ client.max_connections_reached_callback = _handle_max_connections
+
+ def _handle_new_subscription_identifier(identifier: str | None) -> None:
+ """Handle a new subscription identifier."""
hass.config_entries.async_update_entry(
entry,
- unique_id=format_unique_id(
- entry.data[CONF_APP_ID], entry.data[CONF_LOCATION_ID]
- ),
+ data={
+ **entry.data,
+ CONF_SUBSCRIPTION_ID: identifier,
+ },
)
+ if identifier is not None:
+ _LOGGER.debug("Updating subscription ID to %s", identifier)
+ else:
+ _LOGGER.debug("Removing subscription ID")
- if not validate_webhook_requirements(hass):
- _LOGGER.warning(
- "The 'base_url' of the 'http' integration must be configured and start with"
- " 'https://'"
- )
- return False
+ client.new_subscription_id_callback = _handle_new_subscription_identifier
- api = SmartThings(async_get_clientsession(hass), entry.data[CONF_ACCESS_TOKEN])
-
- # Ensure platform modules are loaded since the DeviceBroker will
- # import them below and we want them to be cached ahead of time
- # so the integration does not do blocking I/O in the event loop
- # to import the modules.
- await async_get_loaded_integration(hass, DOMAIN).async_get_platforms(PLATFORMS)
+ if (old_identifier := entry.data.get(CONF_SUBSCRIPTION_ID)) is not None:
+ _LOGGER.debug("Trying to delete old subscription %s", old_identifier)
+ await client.delete_subscription(old_identifier)
+ _LOGGER.debug("Trying to create a new subscription")
try:
- # See if the app is already setup. This occurs when there are
- # installs in multiple SmartThings locations (valid use-case)
- manager = hass.data[DOMAIN][DATA_MANAGER]
- smart_app = manager.smartapps.get(entry.data[CONF_APP_ID])
- if not smart_app:
- # Validate and setup the app.
- app = await api.app(entry.data[CONF_APP_ID])
- smart_app = setup_smartapp(hass, app)
-
- # Validate and retrieve the installed app.
- installed_app = await validate_installed_app(
- api, entry.data[CONF_INSTALLED_APP_ID]
+ subscription = await client.create_subscription(
+ entry.data[CONF_LOCATION_ID],
+ entry.data[CONF_TOKEN][CONF_INSTALLED_APP_ID],
)
+ except SmartThingsSinkError as err:
+ _LOGGER.exception("Couldn't create a new subscription")
+ raise ConfigEntryNotReady from err
+ subscription_id = subscription.subscription_id
+ _handle_new_subscription_identifier(subscription_id)
- # Get scenes
- scenes = await async_get_entry_scenes(entry, api)
+ entry.async_create_background_task(
+ hass,
+ client.subscribe(
+ entry.data[CONF_LOCATION_ID],
+ entry.data[CONF_TOKEN][CONF_INSTALLED_APP_ID],
+ subscription,
+ ),
+ "smartthings_socket",
+ )
- # Get SmartApp token to sync subscriptions
- token = await api.generate_tokens(
- entry.data[CONF_CLIENT_ID],
- entry.data[CONF_CLIENT_SECRET],
- entry.data[CONF_REFRESH_TOKEN],
+ device_status: dict[str, FullDevice] = {}
+ try:
+ rooms = {
+ room.room_id: room.name
+ for room in await client.get_rooms(location_id=entry.data[CONF_LOCATION_ID])
+ }
+ devices = await client.get_devices()
+ for device in devices:
+ status = process_status(await client.get_device_status(device.device_id))
+ device_status[device.device_id] = FullDevice(device=device, status=status)
+ except SmartThingsAuthenticationFailedError as err:
+ raise ConfigEntryAuthFailed from err
+
+ device_registry = dr.async_get(hass)
+ create_devices(device_registry, device_status, entry, rooms)
+
+ scenes = {
+ scene.scene_id: scene
+ for scene in await client.get_scenes(location_id=entry.data[CONF_LOCATION_ID])
+ }
+
+ def handle_deleted_device(device_id: str) -> None:
+ """Handle a deleted device."""
+ dev_entry = device_registry.async_get_device(
+ identifiers={(DOMAIN, device_id)},
)
- hass.config_entries.async_update_entry(
- entry, data={**entry.data, CONF_REFRESH_TOKEN: token.refresh_token}
- )
-
- # Get devices and their current status
- devices = await api.devices(location_ids=[installed_app.location_id])
-
- async def retrieve_device_status(device):
- try:
- await device.status.refresh()
- except ClientResponseError:
- _LOGGER.debug(
- (
- "Unable to update status for device: %s (%s), the device will"
- " be excluded"
- ),
- device.label,
- device.device_id,
- exc_info=True,
- )
- devices.remove(device)
-
- await asyncio.gather(*(retrieve_device_status(d) for d in devices.copy()))
-
- # Sync device subscriptions
- await smartapp_sync_subscriptions(
- hass,
- token.access_token,
- installed_app.location_id,
- installed_app.installed_app_id,
- devices,
- )
-
- # Setup device broker
- with async_pause_setup(hass, SetupPhases.WAIT_IMPORT_PLATFORMS):
- # DeviceBroker has a side effect of importing platform
- # modules when its created. In the future this should be
- # refactored to not do this.
- broker = await hass.async_add_import_executor_job(
- DeviceBroker, hass, entry, token, smart_app, devices, scenes
+ if dev_entry is not None:
+ device_registry.async_update_device(
+ dev_entry.id, remove_config_entry_id=entry.entry_id
)
- broker.connect()
- hass.data[DOMAIN][DATA_BROKERS][entry.entry_id] = broker
- except APIInvalidGrant as ex:
- raise ConfigEntryAuthFailed from ex
- except ClientResponseError as ex:
- if ex.status in (HTTPStatus.UNAUTHORIZED, HTTPStatus.FORBIDDEN):
- raise ConfigEntryError(
- "The access token is no longer valid. Please remove the integration and set up again."
- ) from ex
- _LOGGER.debug(ex, exc_info=True)
- raise ConfigEntryNotReady from ex
- except (ClientConnectionError, RuntimeWarning) as ex:
- _LOGGER.debug(ex, exc_info=True)
- raise ConfigEntryNotReady from ex
+ entry.async_on_unload(
+ client.add_device_lifecycle_event_listener(
+ Lifecycle.DELETE, handle_deleted_device
+ )
+ )
+
+ entry.runtime_data = SmartThingsData(
+ devices={
+ device_id: device
+ for device_id, device in device_status.items()
+ if MAIN in device.status
+ },
+ client=client,
+ scenes=scenes,
+ rooms=rooms,
+ )
+
+ # Events are deprecated and will be removed in 2025.10
+ def handle_button_press(event: DeviceEvent) -> None:
+ """Handle a button press."""
+ if (
+ event.capability is Capability.BUTTON
+ and event.attribute is Attribute.BUTTON
+ ):
+ hass.bus.async_fire(
+ EVENT_BUTTON,
+ {
+ "component_id": event.component_id,
+ "device_id": event.device_id,
+ "location_id": event.location_id,
+ "value": event.value,
+ "name": entry.runtime_data.devices[event.device_id].device.label,
+ "data": event.data,
+ },
+ )
+
+ entry.async_on_unload(
+ client.add_unspecified_device_event_listener(handle_button_press)
+ )
+
+ async def _handle_shutdown(_: Event) -> None:
+ """Handle shutdown."""
+ await client.delete_subscription(subscription_id)
+
+ entry.async_on_unload(
+ hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _handle_shutdown)
+ )
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+
+ device_entries = dr.async_entries_for_config_entry(device_registry, entry.entry_id)
+ for device_entry in device_entries:
+ device_id = next(
+ identifier[1]
+ for identifier in device_entry.identifiers
+ if identifier[0] == DOMAIN
+ )
+ if device_id in device_status:
+ continue
+ device_registry.async_update_device(
+ device_entry.id, remove_config_entry_id=entry.entry_id
+ )
+
return True
-async def async_get_entry_scenes(entry: ConfigEntry, api):
- """Get the scenes within an integration."""
- try:
- return await api.scenes(location_id=entry.data[CONF_LOCATION_ID])
- except ClientResponseError as ex:
- if ex.status == HTTPStatus.FORBIDDEN:
- _LOGGER.exception(
- (
- "Unable to load scenes for configuration entry '%s' because the"
- " access token does not have the required access"
- ),
- entry.title,
- )
- else:
- raise
- return []
-
-
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, entry: SmartThingsConfigEntry
+) -> bool:
"""Unload a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS].pop(entry.entry_id, None)
- if broker:
- broker.disconnect()
-
+ client = entry.runtime_data.client
+ if (subscription_id := entry.data.get(CONF_SUBSCRIPTION_ID)) is not None:
+ await client.delete_subscription(subscription_id)
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
-async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
- """Perform clean-up when entry is being removed."""
- api = SmartThings(async_get_clientsession(hass), entry.data[CONF_ACCESS_TOKEN])
+async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+ """Handle config entry migration."""
- # Remove the installed_app, which if already removed raises a HTTPStatus.FORBIDDEN error.
- installed_app_id = entry.data[CONF_INSTALLED_APP_ID]
- try:
- await api.delete_installed_app(installed_app_id)
- except ClientResponseError as ex:
- if ex.status == HTTPStatus.FORBIDDEN:
- _LOGGER.debug(
- "Installed app %s has already been removed",
- installed_app_id,
- exc_info=True,
+ if entry.version < 3:
+ # We keep the old data around, so we can use that to clean up the webhook in the future
+ hass.config_entries.async_update_entry(
+ entry, version=3, data={OLD_DATA: dict(entry.data)}
+ )
+
+ return True
+
+
+def create_devices(
+ device_registry: dr.DeviceRegistry,
+ devices: dict[str, FullDevice],
+ entry: SmartThingsConfigEntry,
+ rooms: dict[str, str],
+) -> None:
+ """Create devices in the device registry."""
+ for device in devices.values():
+ kwargs: dict[str, Any] = {}
+ if device.device.hub is not None:
+ kwargs = {
+ ATTR_SW_VERSION: device.device.hub.firmware_version,
+ ATTR_MODEL: device.device.hub.hardware_type,
+ }
+ if device.device.hub.mac_address:
+ kwargs[ATTR_CONNECTIONS] = {
+ (dr.CONNECTION_NETWORK_MAC, device.device.hub.mac_address)
+ }
+ if device.device.parent_device_id:
+ kwargs[ATTR_VIA_DEVICE] = (DOMAIN, device.device.parent_device_id)
+ if (ocf := device.device.ocf) is not None:
+ kwargs.update(
+ {
+ ATTR_MANUFACTURER: ocf.manufacturer_name,
+ ATTR_MODEL: (
+ (ocf.model_number.split("|")[0]) if ocf.model_number else None
+ ),
+ ATTR_HW_VERSION: ocf.hardware_version,
+ ATTR_SW_VERSION: ocf.firmware_version,
+ }
)
- else:
- raise
- _LOGGER.debug("Removed installed app %s", installed_app_id)
-
- # Remove the app if not referenced by other entries, which if already
- # removed raises a HTTPStatus.FORBIDDEN error.
- all_entries = hass.config_entries.async_entries(DOMAIN)
- app_id = entry.data[CONF_APP_ID]
- app_count = sum(1 for entry in all_entries if entry.data[CONF_APP_ID] == app_id)
- if app_count > 1:
- _LOGGER.debug(
- (
- "App %s was not removed because it is in use by other configuration"
- " entries"
+ if (viper := device.device.viper) is not None:
+ kwargs.update(
+ {
+ ATTR_MANUFACTURER: viper.manufacturer_name,
+ ATTR_MODEL: viper.model_name,
+ ATTR_HW_VERSION: viper.hardware_version,
+ ATTR_SW_VERSION: viper.software_version,
+ }
+ )
+ device_registry.async_get_or_create(
+ config_entry_id=entry.entry_id,
+ identifiers={(DOMAIN, device.device.device_id)},
+ configuration_url="https://account.smartthings.com",
+ name=device.device.label,
+ suggested_area=(
+ rooms.get(device.device.room_id) if device.device.room_id else None
),
- app_id,
- )
- return
- # Remove the app
- try:
- await api.delete_app(app_id)
- except ClientResponseError as ex:
- if ex.status == HTTPStatus.FORBIDDEN:
- _LOGGER.debug("App %s has already been removed", app_id, exc_info=True)
- else:
- raise
- _LOGGER.debug("Removed app %s", app_id)
-
- if len(all_entries) == 1:
- await unload_smartapp_endpoint(hass)
-
-
-class DeviceBroker:
- """Manages an individual SmartThings config entry."""
-
- def __init__(
- self,
- hass: HomeAssistant,
- entry: ConfigEntry,
- token,
- smart_app,
- devices: Iterable,
- scenes: Iterable,
- ) -> None:
- """Create a new instance of the DeviceBroker."""
- self._hass = hass
- self._entry = entry
- self._installed_app_id = entry.data[CONF_INSTALLED_APP_ID]
- self._smart_app = smart_app
- self._token = token
- self._event_disconnect = None
- self._regenerate_token_remove = None
- self._assignments = self._assign_capabilities(devices)
- self.devices = {device.device_id: device for device in devices}
- self.scenes = {scene.scene_id: scene for scene in scenes}
-
- def _assign_capabilities(self, devices: Iterable):
- """Assign platforms to capabilities."""
- assignments = {}
- for device in devices:
- capabilities = device.capabilities.copy()
- slots = {}
- for platform in PLATFORMS:
- platform_module = importlib.import_module(
- f".{platform}", self.__module__
- )
- if not hasattr(platform_module, "get_capabilities"):
- continue
- assigned = platform_module.get_capabilities(capabilities)
- if not assigned:
- continue
- # Draw-down capabilities and set slot assignment
- for capability in assigned:
- if capability not in capabilities:
- continue
- capabilities.remove(capability)
- slots[capability] = platform
- assignments[device.device_id] = slots
- return assignments
-
- def connect(self):
- """Connect handlers/listeners for device/lifecycle events."""
-
- # Setup interval to regenerate the refresh token on a periodic basis.
- # Tokens expire in 30 days and once expired, cannot be recovered.
- async def regenerate_refresh_token(now):
- """Generate a new refresh token and update the config entry."""
- await self._token.refresh(
- self._entry.data[CONF_CLIENT_ID],
- self._entry.data[CONF_CLIENT_SECRET],
- )
- self._hass.config_entries.async_update_entry(
- self._entry,
- data={
- **self._entry.data,
- CONF_REFRESH_TOKEN: self._token.refresh_token,
- },
- )
- _LOGGER.debug(
- "Regenerated refresh token for installed app: %s",
- self._installed_app_id,
- )
-
- self._regenerate_token_remove = async_track_time_interval(
- self._hass, regenerate_refresh_token, TOKEN_REFRESH_INTERVAL
+ **kwargs,
)
- # Connect handler to incoming device events
- self._event_disconnect = self._smart_app.connect_event(self._event_handler)
- def disconnect(self):
- """Disconnects handlers/listeners for device/lifecycle events."""
- if self._regenerate_token_remove:
- self._regenerate_token_remove()
- if self._event_disconnect:
- self._event_disconnect()
+KEEP_CAPABILITY_QUIRK: dict[
+ Capability | str, Callable[[dict[Attribute | str, Status]], bool]
+] = {
+ Capability.DRYER_OPERATING_STATE: (
+ lambda status: status[Attribute.SUPPORTED_MACHINE_STATES].value is not None
+ ),
+ Capability.WASHER_OPERATING_STATE: (
+ lambda status: status[Attribute.SUPPORTED_MACHINE_STATES].value is not None
+ ),
+ Capability.DEMAND_RESPONSE_LOAD_CONTROL: lambda _: True,
+}
- def get_assigned(self, device_id: str, platform: str):
- """Get the capabilities assigned to the platform."""
- slots = self._assignments.get(device_id, {})
- return [key for key, value in slots.items() if value == platform]
- def any_assigned(self, device_id: str, platform: str):
- """Return True if the platform has any assigned capabilities."""
- slots = self._assignments.get(device_id, {})
- return any(value for value in slots.values() if value == platform)
-
- async def _event_handler(self, req, resp, app):
- """Broker for incoming events."""
- # Do not process events received from a different installed app
- # under the same parent SmartApp (valid use-scenario)
- if req.installed_app_id != self._installed_app_id:
- return
-
- updated_devices = set()
- for evt in req.events:
- if evt.event_type != EVENT_TYPE_DEVICE:
- continue
- if not (device := self.devices.get(evt.device_id)):
- continue
- device.status.apply_attribute_update(
- evt.component_id,
- evt.capability,
- evt.attribute,
- evt.value,
- data=evt.data,
- )
-
- # Fire events for buttons
- if (
- evt.capability == Capability.button
- and evt.attribute == Attribute.button
- ):
- data = {
- "component_id": evt.component_id,
- "device_id": evt.device_id,
- "location_id": evt.location_id,
- "value": evt.value,
- "name": device.label,
- "data": evt.data,
- }
- self._hass.bus.async_fire(EVENT_BUTTON, data)
- _LOGGER.debug("Fired button event: %s", data)
- else:
- data = {
- "location_id": evt.location_id,
- "device_id": evt.device_id,
- "component_id": evt.component_id,
- "capability": evt.capability,
- "attribute": evt.attribute,
- "value": evt.value,
- "data": evt.data,
- }
- _LOGGER.debug("Push update received: %s", data)
-
- updated_devices.add(device.device_id)
-
- async_dispatcher_send(self._hass, SIGNAL_SMARTTHINGS_UPDATE, updated_devices)
+def process_status(
+ status: dict[str, dict[Capability | str, dict[Attribute | str, Status]]],
+) -> dict[str, dict[Capability | str, dict[Attribute | str, Status]]]:
+ """Remove disabled capabilities from status."""
+ if (main_component := status.get(MAIN)) is None:
+ return status
+ if (
+ disabled_capabilities_capability := main_component.get(
+ Capability.CUSTOM_DISABLED_CAPABILITIES
+ )
+ ) is not None:
+ disabled_capabilities = cast(
+ list[Capability | str],
+ disabled_capabilities_capability[Attribute.DISABLED_CAPABILITIES].value,
+ )
+ if disabled_capabilities is not None:
+ for capability in disabled_capabilities:
+ if capability in main_component and (
+ capability not in KEEP_CAPABILITY_QUIRK
+ or not KEEP_CAPABILITY_QUIRK[capability](main_component[capability])
+ ):
+ del main_component[capability]
+ return status
diff --git a/homeassistant/components/smartthings/application_credentials.py b/homeassistant/components/smartthings/application_credentials.py
new file mode 100644
index 00000000000..1e637c6bd12
--- /dev/null
+++ b/homeassistant/components/smartthings/application_credentials.py
@@ -0,0 +1,64 @@
+"""Application credentials platform for SmartThings."""
+
+from json import JSONDecodeError
+import logging
+from typing import cast
+
+from aiohttp import BasicAuth, ClientError
+
+from homeassistant.components.application_credentials import (
+ AuthImplementation,
+ AuthorizationServer,
+ ClientCredential,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.config_entry_oauth2_flow import AbstractOAuth2Implementation
+
+from .const import DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_get_auth_implementation(
+ hass: HomeAssistant, auth_domain: str, credential: ClientCredential
+) -> AbstractOAuth2Implementation:
+ """Return auth implementation."""
+ return SmartThingsOAuth2Implementation(
+ hass,
+ DOMAIN,
+ credential,
+ authorization_server=AuthorizationServer(
+ authorize_url="https://api.smartthings.com/oauth/authorize",
+ token_url="https://auth-global.api.smartthings.com/oauth/token",
+ ),
+ )
+
+
+class SmartThingsOAuth2Implementation(AuthImplementation):
+ """Oauth2 implementation that only uses the external url."""
+
+ async def _token_request(self, data: dict) -> dict:
+ """Make a token request."""
+ session = async_get_clientsession(self.hass)
+
+ resp = await session.post(
+ self.token_url,
+ data=data,
+ auth=BasicAuth(self.client_id, self.client_secret),
+ )
+ if resp.status >= 400:
+ try:
+ error_response = await resp.json()
+ except (ClientError, JSONDecodeError):
+ error_response = {}
+ error_code = error_response.get("error", "unknown")
+ error_description = error_response.get("error_description", "unknown error")
+ _LOGGER.error(
+ "Token request for %s failed (%s): %s",
+ self.domain,
+ error_code,
+ error_description,
+ )
+ resp.raise_for_status()
+ return cast(dict, await resp.json())
diff --git a/homeassistant/components/smartthings/binary_sensor.py b/homeassistant/components/smartthings/binary_sensor.py
index 611473b011d..8479852a6f6 100644
--- a/homeassistant/components/smartthings/binary_sensor.py
+++ b/homeassistant/components/smartthings/binary_sensor.py
@@ -2,84 +2,268 @@
from __future__ import annotations
-from collections.abc import Sequence
+from dataclasses import dataclass
-from pysmartthings import Attribute, Capability
+from pysmartthings import Attribute, Capability, Category, SmartThings
+from homeassistant.components.automation import automations_with_entity
from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
BinarySensorEntity,
+ BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
+from homeassistant.components.script import scripts_with_entity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers import entity_registry as er
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.issue_registry import (
+ IssueSeverity,
+ async_create_issue,
+ async_delete_issue,
+)
-from .const import DATA_BROKERS, DOMAIN
+from . import FullDevice, SmartThingsConfigEntry
+from .const import DOMAIN, MAIN
from .entity import SmartThingsEntity
-CAPABILITY_TO_ATTRIB = {
- Capability.acceleration_sensor: Attribute.acceleration,
- Capability.contact_sensor: Attribute.contact,
- Capability.filter_status: Attribute.filter_status,
- Capability.motion_sensor: Attribute.motion,
- Capability.presence_sensor: Attribute.presence,
- Capability.sound_sensor: Attribute.sound,
- Capability.tamper_alert: Attribute.tamper,
- Capability.valve: Attribute.valve,
- Capability.water_sensor: Attribute.water,
-}
-ATTRIB_TO_CLASS = {
- Attribute.acceleration: BinarySensorDeviceClass.MOVING,
- Attribute.contact: BinarySensorDeviceClass.OPENING,
- Attribute.filter_status: BinarySensorDeviceClass.PROBLEM,
- Attribute.motion: BinarySensorDeviceClass.MOTION,
- Attribute.presence: BinarySensorDeviceClass.PRESENCE,
- Attribute.sound: BinarySensorDeviceClass.SOUND,
- Attribute.tamper: BinarySensorDeviceClass.PROBLEM,
- Attribute.valve: BinarySensorDeviceClass.OPENING,
- Attribute.water: BinarySensorDeviceClass.MOISTURE,
-}
-ATTRIB_TO_ENTTIY_CATEGORY = {
- Attribute.tamper: EntityCategory.DIAGNOSTIC,
+
+@dataclass(frozen=True, kw_only=True)
+class SmartThingsBinarySensorEntityDescription(BinarySensorEntityDescription):
+ """Describe a SmartThings binary sensor entity."""
+
+ is_on_key: str
+ category_device_class: dict[Category | str, BinarySensorDeviceClass] | None = None
+ category: set[Category] | None = None
+
+
+CAPABILITY_TO_SENSORS: dict[
+ Capability, dict[Attribute, SmartThingsBinarySensorEntityDescription]
+] = {
+ Capability.ACCELERATION_SENSOR: {
+ Attribute.ACCELERATION: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.ACCELERATION,
+ translation_key="acceleration",
+ device_class=BinarySensorDeviceClass.MOVING,
+ is_on_key="active",
+ )
+ },
+ Capability.CONTACT_SENSOR: {
+ Attribute.CONTACT: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.CONTACT,
+ device_class=BinarySensorDeviceClass.DOOR,
+ is_on_key="open",
+ category_device_class={
+ Category.GARAGE_DOOR: BinarySensorDeviceClass.GARAGE_DOOR,
+ Category.DOOR: BinarySensorDeviceClass.DOOR,
+ Category.WINDOW: BinarySensorDeviceClass.WINDOW,
+ },
+ )
+ },
+ Capability.FILTER_STATUS: {
+ Attribute.FILTER_STATUS: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.FILTER_STATUS,
+ translation_key="filter_status",
+ device_class=BinarySensorDeviceClass.PROBLEM,
+ is_on_key="replace",
+ )
+ },
+ Capability.SAMSUNG_CE_KIDS_LOCK: {
+ Attribute.LOCK_STATE: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.LOCK_STATE,
+ translation_key="child_lock",
+ is_on_key="locked",
+ )
+ },
+ Capability.MOTION_SENSOR: {
+ Attribute.MOTION: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.MOTION,
+ device_class=BinarySensorDeviceClass.MOTION,
+ is_on_key="active",
+ )
+ },
+ Capability.PRESENCE_SENSOR: {
+ Attribute.PRESENCE: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.PRESENCE,
+ device_class=BinarySensorDeviceClass.PRESENCE,
+ is_on_key="present",
+ )
+ },
+ Capability.REMOTE_CONTROL_STATUS: {
+ Attribute.REMOTE_CONTROL_ENABLED: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.REMOTE_CONTROL_ENABLED,
+ translation_key="remote_control",
+ is_on_key="true",
+ )
+ },
+ Capability.SOUND_SENSOR: {
+ Attribute.SOUND: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.SOUND,
+ device_class=BinarySensorDeviceClass.SOUND,
+ is_on_key="detected",
+ )
+ },
+ Capability.SWITCH: {
+ Attribute.SWITCH: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.SWITCH,
+ device_class=BinarySensorDeviceClass.POWER,
+ is_on_key="on",
+ category={Category.DRYER, Category.WASHER},
+ )
+ },
+ Capability.TAMPER_ALERT: {
+ Attribute.TAMPER: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.TAMPER,
+ device_class=BinarySensorDeviceClass.TAMPER,
+ is_on_key="detected",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ },
+ Capability.VALVE: {
+ Attribute.VALVE: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.VALVE,
+ translation_key="valve",
+ device_class=BinarySensorDeviceClass.OPENING,
+ is_on_key="open",
+ )
+ },
+ Capability.WATER_SENSOR: {
+ Attribute.WATER: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.WATER,
+ device_class=BinarySensorDeviceClass.MOISTURE,
+ is_on_key="wet",
+ )
+ },
+ Capability.SAMSUNG_CE_DOOR_STATE: {
+ Attribute.DOOR_STATE: SmartThingsBinarySensorEntityDescription(
+ key=Attribute.DOOR_STATE,
+ translation_key="door",
+ device_class=BinarySensorDeviceClass.OPENING,
+ is_on_key="open",
+ )
+ },
}
+def get_main_component_category(
+ device: FullDevice,
+) -> Category | str:
+ """Get the main component of a device."""
+ main = next(
+ component for component in device.device.components if component.id == MAIN
+ )
+ return main.user_category or main.manufacturer_category
+
+
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add binary sensors for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
- sensors = []
- for device in broker.devices.values():
- for capability in broker.get_assigned(device.device_id, "binary_sensor"):
- attrib = CAPABILITY_TO_ATTRIB[capability]
- sensors.append(SmartThingsBinarySensor(device, attrib))
- async_add_entities(sensors)
-
-
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
- return [
- capability for capability in CAPABILITY_TO_ATTRIB if capability in capabilities
- ]
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsBinarySensor(
+ entry_data.client,
+ device,
+ description,
+ capability,
+ attribute,
+ )
+ for device in entry_data.devices.values()
+ for capability, attribute_map in CAPABILITY_TO_SENSORS.items()
+ if capability in device.status[MAIN]
+ for attribute, description in attribute_map.items()
+ if (
+ not description.category
+ or get_main_component_category(device) in description.category
+ )
+ )
class SmartThingsBinarySensor(SmartThingsEntity, BinarySensorEntity):
"""Define a SmartThings Binary Sensor."""
- def __init__(self, device, attribute):
+ entity_description: SmartThingsBinarySensorEntityDescription
+
+ def __init__(
+ self,
+ client: SmartThings,
+ device: FullDevice,
+ entity_description: SmartThingsBinarySensorEntityDescription,
+ capability: Capability,
+ attribute: Attribute,
+ ) -> None:
"""Init the class."""
- super().__init__(device)
+ super().__init__(client, device, {capability})
self._attribute = attribute
- self._attr_name = f"{device.label} {attribute}"
- self._attr_unique_id = f"{device.device_id}.{attribute}"
- self._attr_device_class = ATTRIB_TO_CLASS[attribute]
- self._attr_entity_category = ATTRIB_TO_ENTTIY_CATEGORY.get(attribute)
+ self.capability = capability
+ self.entity_description = entity_description
+ self._attr_unique_id = f"{device.device.device_id}.{attribute}"
+ if (
+ entity_description.category_device_class
+ and (category := get_main_component_category(device))
+ in entity_description.category_device_class
+ ):
+ self._attr_device_class = entity_description.category_device_class[category]
+ self._attr_name = None
@property
- def is_on(self):
+ def is_on(self) -> bool:
"""Return true if the binary sensor is on."""
- return self._device.status.is_on(self._attribute)
+ return (
+ self.get_attribute_value(self.capability, self._attribute)
+ == self.entity_description.is_on_key
+ )
+
+ async def async_added_to_hass(self) -> None:
+ """Call when entity is added to hass."""
+ await super().async_added_to_hass()
+ if self.capability is not Capability.VALVE:
+ return
+ automations = automations_with_entity(self.hass, self.entity_id)
+ scripts = scripts_with_entity(self.hass, self.entity_id)
+ items = automations + scripts
+ if not items:
+ return
+
+ entity_reg: er.EntityRegistry = er.async_get(self.hass)
+ entity_automations = [
+ automation_entity
+ for automation_id in automations
+ if (automation_entity := entity_reg.async_get(automation_id))
+ ]
+ entity_scripts = [
+ script_entity
+ for script_id in scripts
+ if (script_entity := entity_reg.async_get(script_id))
+ ]
+
+ items_list = [
+ f"- [{item.original_name}](/config/automation/edit/{item.unique_id})"
+ for item in entity_automations
+ ] + [
+ f"- [{item.original_name}](/config/script/edit/{item.unique_id})"
+ for item in entity_scripts
+ ]
+
+ async_create_issue(
+ self.hass,
+ DOMAIN,
+ f"deprecated_binary_valve_{self.entity_id}",
+ breaks_in_ha_version="2025.10.0",
+ is_fixable=False,
+ severity=IssueSeverity.WARNING,
+ translation_key="deprecated_binary_valve",
+ translation_placeholders={
+ "entity": self.entity_id,
+ "items": "\n".join(items_list),
+ },
+ )
+
+ async def async_will_remove_from_hass(self) -> None:
+ """Call when entity will be removed from hass."""
+ await super().async_will_remove_from_hass()
+ async_delete_issue(
+ self.hass, DOMAIN, f"deprecated_binary_valve_{self.entity_id}"
+ )
diff --git a/homeassistant/components/smartthings/button.py b/homeassistant/components/smartthings/button.py
new file mode 100644
index 00000000000..ad61880f3b1
--- /dev/null
+++ b/homeassistant/components/smartthings/button.py
@@ -0,0 +1,75 @@
+"""Support for button entities through the SmartThings cloud API."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from pysmartthings import Capability, Command, SmartThings
+
+from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
+from .entity import SmartThingsEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SmartThingsButtonDescription(ButtonEntityDescription):
+ """Class describing SmartThings button entities."""
+
+ key: Capability
+ command: Command
+
+
+CAPABILITIES_TO_BUTTONS: dict[Capability | str, SmartThingsButtonDescription] = {
+ Capability.OVEN_OPERATING_STATE: SmartThingsButtonDescription(
+ key=Capability.OVEN_OPERATING_STATE,
+ translation_key="stop",
+ command=Command.STOP,
+ ),
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add button entities for a config entry."""
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsButtonEntity(
+ entry_data.client, device, CAPABILITIES_TO_BUTTONS[capability]
+ )
+ for device in entry_data.devices.values()
+ for capability in device.status[MAIN]
+ if capability in CAPABILITIES_TO_BUTTONS
+ )
+
+
+class SmartThingsButtonEntity(SmartThingsEntity, ButtonEntity):
+ """Define a SmartThings button."""
+
+ entity_description: SmartThingsButtonDescription
+
+ def __init__(
+ self,
+ client: SmartThings,
+ device: FullDevice,
+ entity_description: SmartThingsButtonDescription,
+ ) -> None:
+ """Initialize the instance."""
+ super().__init__(client, device, set())
+ self.entity_description = entity_description
+ self._attr_unique_id = (
+ f"{device.device.device_id}_{MAIN}_{entity_description.key}"
+ )
+
+ async def async_press(self) -> None:
+ """Press the button."""
+ await self.execute_device_command(
+ self.entity_description.key,
+ self.entity_description.command,
+ )
diff --git a/homeassistant/components/smartthings/climate.py b/homeassistant/components/smartthings/climate.py
index d9535272295..e20f191352f 100644
--- a/homeassistant/components/smartthings/climate.py
+++ b/homeassistant/components/smartthings/climate.py
@@ -3,17 +3,15 @@
from __future__ import annotations
import asyncio
-from collections.abc import Iterable, Sequence
import logging
from typing import Any
-from pysmartthings import Attribute, Capability
+from pysmartthings import Attribute, Capability, Command, SmartThings
from homeassistant.components.climate import (
ATTR_HVAC_MODE,
ATTR_TARGET_TEMP_HIGH,
ATTR_TARGET_TEMP_LOW,
- DOMAIN as CLIMATE_DOMAIN,
SWING_BOTH,
SWING_HORIZONTAL,
SWING_OFF,
@@ -23,12 +21,12 @@ from homeassistant.components.climate import (
HVACAction,
HVACMode,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DATA_BROKERS, DOMAIN
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
from .entity import SmartThingsEntity
ATTR_OPERATION_STATE = "operation_state"
@@ -97,124 +95,106 @@ UNIT_MAP = {"C": UnitOfTemperature.CELSIUS, "F": UnitOfTemperature.FAHRENHEIT}
_LOGGER = logging.getLogger(__name__)
+AC_CAPABILITIES = [
+ Capability.AIR_CONDITIONER_MODE,
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ Capability.SWITCH,
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.THERMOSTAT_COOLING_SETPOINT,
+]
+
+THERMOSTAT_CAPABILITIES = [
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.THERMOSTAT_HEATING_SETPOINT,
+ Capability.THERMOSTAT_MODE,
+]
+
+
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add climate entities for a config entry."""
- ac_capabilities = [
- Capability.air_conditioner_mode,
- Capability.air_conditioner_fan_mode,
- Capability.switch,
- Capability.temperature_measurement,
- Capability.thermostat_cooling_setpoint,
+ entry_data = entry.runtime_data
+ entities: list[ClimateEntity] = [
+ SmartThingsAirConditioner(entry_data.client, device)
+ for device in entry_data.devices.values()
+ if all(capability in device.status[MAIN] for capability in AC_CAPABILITIES)
]
-
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
- entities: list[ClimateEntity] = []
- for device in broker.devices.values():
- if not broker.any_assigned(device.device_id, CLIMATE_DOMAIN):
- continue
- if all(capability in device.capabilities for capability in ac_capabilities):
- entities.append(SmartThingsAirConditioner(device))
- else:
- entities.append(SmartThingsThermostat(device))
- async_add_entities(entities, True)
-
-
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
- supported = [
- Capability.air_conditioner_mode,
- Capability.demand_response_load_control,
- Capability.air_conditioner_fan_mode,
- Capability.switch,
- Capability.thermostat,
- Capability.thermostat_cooling_setpoint,
- Capability.thermostat_fan_mode,
- Capability.thermostat_heating_setpoint,
- Capability.thermostat_mode,
- Capability.thermostat_operating_state,
- ]
- # Can have this legacy/deprecated capability
- if Capability.thermostat in capabilities:
- return supported
- # Or must have all of these thermostat capabilities
- thermostat_capabilities = [
- Capability.temperature_measurement,
- Capability.thermostat_heating_setpoint,
- Capability.thermostat_mode,
- ]
- if all(capability in capabilities for capability in thermostat_capabilities):
- return supported
- # Or must have all of these A/C capabilities
- ac_capabilities = [
- Capability.air_conditioner_mode,
- Capability.air_conditioner_fan_mode,
- Capability.switch,
- Capability.temperature_measurement,
- Capability.thermostat_cooling_setpoint,
- ]
- if all(capability in capabilities for capability in ac_capabilities):
- return supported
- return None
+ entities.extend(
+ SmartThingsThermostat(entry_data.client, device)
+ for device in entry_data.devices.values()
+ if all(
+ capability in device.status[MAIN] for capability in THERMOSTAT_CAPABILITIES
+ )
+ )
+ async_add_entities(entities)
class SmartThingsThermostat(SmartThingsEntity, ClimateEntity):
"""Define a SmartThings climate entities."""
- def __init__(self, device):
- """Init the class."""
- super().__init__(device)
- self._attr_supported_features = self._determine_features()
- self._hvac_mode = None
- self._hvac_modes = None
+ _attr_name = None
- def _determine_features(self):
+ def __init__(self, client: SmartThings, device: FullDevice) -> None:
+ """Init the class."""
+ super().__init__(
+ client,
+ device,
+ {
+ Capability.THERMOSTAT_FAN_MODE,
+ Capability.THERMOSTAT_MODE,
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.THERMOSTAT_HEATING_SETPOINT,
+ Capability.THERMOSTAT_OPERATING_STATE,
+ Capability.THERMOSTAT_COOLING_SETPOINT,
+ Capability.RELATIVE_HUMIDITY_MEASUREMENT,
+ },
+ )
+ self._attr_supported_features = self._determine_features()
+
+ def _determine_features(self) -> ClimateEntityFeature:
flags = (
ClimateEntityFeature.TARGET_TEMPERATURE
| ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
| ClimateEntityFeature.TURN_OFF
| ClimateEntityFeature.TURN_ON
)
- if self._device.get_capability(
- Capability.thermostat_fan_mode, Capability.thermostat
- ):
+ if self.supports_capability(Capability.THERMOSTAT_FAN_MODE):
flags |= ClimateEntityFeature.FAN_MODE
return flags
async def async_set_fan_mode(self, fan_mode: str) -> None:
"""Set new target fan mode."""
- await self._device.set_thermostat_fan_mode(fan_mode, set_status=True)
-
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_schedule_update_ha_state(True)
+ await self.execute_device_command(
+ Capability.THERMOSTAT_FAN_MODE,
+ Command.SET_THERMOSTAT_FAN_MODE,
+ argument=fan_mode,
+ )
async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
"""Set new target operation mode."""
- mode = STATE_TO_MODE[hvac_mode]
- await self._device.set_thermostat_mode(mode, set_status=True)
-
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_schedule_update_ha_state(True)
+ await self.execute_device_command(
+ Capability.THERMOSTAT_MODE,
+ Command.SET_THERMOSTAT_MODE,
+ argument=STATE_TO_MODE[hvac_mode],
+ )
async def async_set_temperature(self, **kwargs: Any) -> None:
"""Set new operation mode and target temperatures."""
+ hvac_mode = self.hvac_mode
# Operation state
if operation_state := kwargs.get(ATTR_HVAC_MODE):
- mode = STATE_TO_MODE[operation_state]
- await self._device.set_thermostat_mode(mode, set_status=True)
- await self.async_update()
+ await self.async_set_hvac_mode(operation_state)
+ hvac_mode = operation_state
# Heat/cool setpoint
heating_setpoint = None
cooling_setpoint = None
- if self.hvac_mode == HVACMode.HEAT:
+ if hvac_mode == HVACMode.HEAT:
heating_setpoint = kwargs.get(ATTR_TEMPERATURE)
- elif self.hvac_mode == HVACMode.COOL:
+ elif hvac_mode == HVACMode.COOL:
cooling_setpoint = kwargs.get(ATTR_TEMPERATURE)
else:
heating_setpoint = kwargs.get(ATTR_TARGET_TEMP_LOW)
@@ -222,137 +202,159 @@ class SmartThingsThermostat(SmartThingsEntity, ClimateEntity):
tasks = []
if heating_setpoint is not None:
tasks.append(
- self._device.set_heating_setpoint(
- round(heating_setpoint, 3), set_status=True
+ self.execute_device_command(
+ Capability.THERMOSTAT_HEATING_SETPOINT,
+ Command.SET_HEATING_SETPOINT,
+ argument=round(heating_setpoint, 3),
)
)
if cooling_setpoint is not None:
tasks.append(
- self._device.set_cooling_setpoint(
- round(cooling_setpoint, 3), set_status=True
+ self.execute_device_command(
+ Capability.THERMOSTAT_COOLING_SETPOINT,
+ Command.SET_COOLING_SETPOINT,
+ argument=round(cooling_setpoint, 3),
)
)
await asyncio.gather(*tasks)
- # State is set optimistically in the commands above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_schedule_update_ha_state(True)
-
- async def async_update(self) -> None:
- """Update the attributes of the climate device."""
- thermostat_mode = self._device.status.thermostat_mode
- self._hvac_mode = MODE_TO_STATE.get(thermostat_mode)
- if self._hvac_mode is None:
- _LOGGER.debug(
- "Device %s (%s) returned an invalid hvac mode: %s",
- self._device.label,
- self._device.device_id,
- thermostat_mode,
- )
-
- modes = set()
- supported_modes = self._device.status.supported_thermostat_modes
- if isinstance(supported_modes, Iterable):
- for mode in supported_modes:
- if (state := MODE_TO_STATE.get(mode)) is not None:
- modes.add(state)
- else:
- _LOGGER.debug(
- (
- "Device %s (%s) returned an invalid supported thermostat"
- " mode: %s"
- ),
- self._device.label,
- self._device.device_id,
- mode,
- )
- else:
- _LOGGER.debug(
- "Device %s (%s) returned invalid supported thermostat modes: %s",
- self._device.label,
- self._device.device_id,
- supported_modes,
- )
- self._hvac_modes = list(modes)
-
@property
- def current_humidity(self):
+ def current_humidity(self) -> float | None:
"""Return the current humidity."""
- return self._device.status.humidity
+ if self.supports_capability(Capability.RELATIVE_HUMIDITY_MEASUREMENT):
+ return self.get_attribute_value(
+ Capability.RELATIVE_HUMIDITY_MEASUREMENT, Attribute.HUMIDITY
+ )
+ return None
@property
- def current_temperature(self):
+ def current_temperature(self) -> float | None:
"""Return the current temperature."""
- return self._device.status.temperature
+ return self.get_attribute_value(
+ Capability.TEMPERATURE_MEASUREMENT, Attribute.TEMPERATURE
+ )
@property
- def fan_mode(self):
+ def fan_mode(self) -> str | None:
"""Return the fan setting."""
- return self._device.status.thermostat_fan_mode
+ return self.get_attribute_value(
+ Capability.THERMOSTAT_FAN_MODE, Attribute.THERMOSTAT_FAN_MODE
+ )
@property
- def fan_modes(self):
+ def fan_modes(self) -> list[str]:
"""Return the list of available fan modes."""
- return self._device.status.supported_thermostat_fan_modes
+ return self.get_attribute_value(
+ Capability.THERMOSTAT_FAN_MODE, Attribute.SUPPORTED_THERMOSTAT_FAN_MODES
+ )
@property
def hvac_action(self) -> HVACAction | None:
"""Return the current running hvac operation if supported."""
+ if not self.supports_capability(Capability.THERMOSTAT_OPERATING_STATE):
+ return None
return OPERATING_STATE_TO_ACTION.get(
- self._device.status.thermostat_operating_state
+ self.get_attribute_value(
+ Capability.THERMOSTAT_OPERATING_STATE,
+ Attribute.THERMOSTAT_OPERATING_STATE,
+ )
)
@property
- def hvac_mode(self) -> HVACMode:
+ def hvac_mode(self) -> HVACMode | None:
"""Return current operation ie. heat, cool, idle."""
- return self._hvac_mode
+ return MODE_TO_STATE.get(
+ self.get_attribute_value(
+ Capability.THERMOSTAT_MODE, Attribute.THERMOSTAT_MODE
+ )
+ )
@property
def hvac_modes(self) -> list[HVACMode]:
"""Return the list of available operation modes."""
- return self._hvac_modes
+ if (
+ supported_thermostat_modes := self.get_attribute_value(
+ Capability.THERMOSTAT_MODE, Attribute.SUPPORTED_THERMOSTAT_MODES
+ )
+ ) is None:
+ return []
+ return [
+ state
+ for mode in supported_thermostat_modes
+ if (state := AC_MODE_TO_STATE.get(mode)) is not None
+ ]
@property
- def target_temperature(self):
+ def target_temperature(self) -> float | None:
"""Return the temperature we try to reach."""
if self.hvac_mode == HVACMode.COOL:
- return self._device.status.cooling_setpoint
+ return self.get_attribute_value(
+ Capability.THERMOSTAT_COOLING_SETPOINT, Attribute.COOLING_SETPOINT
+ )
if self.hvac_mode == HVACMode.HEAT:
- return self._device.status.heating_setpoint
+ return self.get_attribute_value(
+ Capability.THERMOSTAT_HEATING_SETPOINT, Attribute.HEATING_SETPOINT
+ )
return None
@property
- def target_temperature_high(self):
+ def target_temperature_high(self) -> float | None:
"""Return the highbound target temperature we try to reach."""
if self.hvac_mode == HVACMode.HEAT_COOL:
- return self._device.status.cooling_setpoint
+ return self.get_attribute_value(
+ Capability.THERMOSTAT_COOLING_SETPOINT, Attribute.COOLING_SETPOINT
+ )
return None
@property
def target_temperature_low(self):
"""Return the lowbound target temperature we try to reach."""
if self.hvac_mode == HVACMode.HEAT_COOL:
- return self._device.status.heating_setpoint
+ return self.get_attribute_value(
+ Capability.THERMOSTAT_HEATING_SETPOINT, Attribute.HEATING_SETPOINT
+ )
return None
@property
- def temperature_unit(self):
+ def temperature_unit(self) -> str:
"""Return the unit of measurement."""
- return UNIT_MAP.get(self._device.status.attributes[Attribute.temperature].unit)
+ # Offline third party thermostats may not have a unit
+ # Since climate always requires a unit, default to Celsius
+ if (
+ unit := self._internal_state[Capability.TEMPERATURE_MEASUREMENT][
+ Attribute.TEMPERATURE
+ ].unit
+ ) is None:
+ return UnitOfTemperature.CELSIUS
+ return UNIT_MAP[unit]
class SmartThingsAirConditioner(SmartThingsEntity, ClimateEntity):
"""Define a SmartThings Air Conditioner."""
- _hvac_modes: list[HVACMode]
+ _attr_name = None
+ _attr_preset_mode = None
- def __init__(self, device) -> None:
+ def __init__(self, client: SmartThings, device: FullDevice) -> None:
"""Init the class."""
- super().__init__(device)
- self._hvac_modes = []
- self._attr_preset_mode = None
+ super().__init__(
+ client,
+ device,
+ {
+ Capability.AIR_CONDITIONER_MODE,
+ Capability.SWITCH,
+ Capability.FAN_OSCILLATION_MODE,
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ Capability.THERMOSTAT_COOLING_SETPOINT,
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.CUSTOM_AIR_CONDITIONER_OPTIONAL_MODE,
+ Capability.DEMAND_RESPONSE_LOAD_CONTROL,
+ },
+ )
+ self._attr_hvac_modes = self._determine_hvac_modes()
self._attr_preset_modes = self._determine_preset_modes()
- self._attr_swing_modes = self._determine_swing_modes()
+ if self.supports_capability(Capability.FAN_OSCILLATION_MODE):
+ self._attr_swing_modes = self._determine_swing_modes()
self._attr_supported_features = self._determine_supported_features()
def _determine_supported_features(self) -> ClimateEntityFeature:
@@ -362,7 +364,7 @@ class SmartThingsAirConditioner(SmartThingsEntity, ClimateEntity):
| ClimateEntityFeature.TURN_OFF
| ClimateEntityFeature.TURN_ON
)
- if self._device.get_capability(Capability.fan_oscillation_mode):
+ if self.supports_capability(Capability.FAN_OSCILLATION_MODE):
features |= ClimateEntityFeature.SWING_MODE
if (self._attr_preset_modes is not None) and len(self._attr_preset_modes) > 0:
features |= ClimateEntityFeature.PRESET_MODE
@@ -370,14 +372,11 @@ class SmartThingsAirConditioner(SmartThingsEntity, ClimateEntity):
async def async_set_fan_mode(self, fan_mode: str) -> None:
"""Set new target fan mode."""
- await self._device.set_fan_mode(fan_mode, set_status=True)
-
- # setting the fan must reset the preset mode (it deactivates the windFree function)
- self._attr_preset_mode = None
-
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
+ await self.execute_device_command(
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ Command.SET_FAN_MODE,
+ argument=fan_mode,
+ )
async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
"""Set new target operation mode."""
@@ -386,23 +385,27 @@ class SmartThingsAirConditioner(SmartThingsEntity, ClimateEntity):
return
tasks = []
# Turn on the device if it's off before setting mode.
- if not self._device.status.switch:
- tasks.append(self._device.switch_on(set_status=True))
+ if self.get_attribute_value(Capability.SWITCH, Attribute.SWITCH) == "off":
+ tasks.append(self.async_turn_on())
mode = STATE_TO_AC_MODE[hvac_mode]
# If new hvac_mode is HVAC_MODE_FAN_ONLY and AirConditioner support "wind" mode the AirConditioner new mode has to be "wind"
# The conversion make the mode change working
# The conversion is made only for device that wrongly has capability "wind" instead "fan_only"
if hvac_mode == HVACMode.FAN_ONLY:
- supported_modes = self._device.status.supported_ac_modes
- if WIND in supported_modes:
+ if WIND in self.get_attribute_value(
+ Capability.AIR_CONDITIONER_MODE, Attribute.SUPPORTED_AC_MODES
+ ):
mode = WIND
- tasks.append(self._device.set_air_conditioner_mode(mode, set_status=True))
+ tasks.append(
+ self.execute_device_command(
+ Capability.AIR_CONDITIONER_MODE,
+ Command.SET_AIR_CONDITIONER_MODE,
+ argument=mode,
+ )
+ )
await asyncio.gather(*tasks)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
async def async_set_temperature(self, **kwargs: Any) -> None:
"""Set new target temperature."""
@@ -410,155 +413,167 @@ class SmartThingsAirConditioner(SmartThingsEntity, ClimateEntity):
# operation mode
if operation_mode := kwargs.get(ATTR_HVAC_MODE):
if operation_mode == HVACMode.OFF:
- tasks.append(self._device.switch_off(set_status=True))
+ tasks.append(self.async_turn_off())
else:
- if not self._device.status.switch:
- tasks.append(self._device.switch_on(set_status=True))
+ if (
+ self.get_attribute_value(Capability.SWITCH, Attribute.SWITCH)
+ == "off"
+ ):
+ tasks.append(self.async_turn_on())
tasks.append(self.async_set_hvac_mode(operation_mode))
# temperature
tasks.append(
- self._device.set_cooling_setpoint(kwargs[ATTR_TEMPERATURE], set_status=True)
+ self.execute_device_command(
+ Capability.THERMOSTAT_COOLING_SETPOINT,
+ Command.SET_COOLING_SETPOINT,
+ argument=kwargs[ATTR_TEMPERATURE],
+ )
)
await asyncio.gather(*tasks)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
async def async_turn_on(self) -> None:
"""Turn device on."""
- await self._device.switch_on(set_status=True)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
+ await self.execute_device_command(
+ Capability.SWITCH,
+ Command.ON,
+ )
async def async_turn_off(self) -> None:
"""Turn device off."""
- await self._device.switch_off(set_status=True)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
-
- async def async_update(self) -> None:
- """Update the calculated fields of the AC."""
- modes = {HVACMode.OFF}
- for mode in self._device.status.supported_ac_modes:
- if (state := AC_MODE_TO_STATE.get(mode)) is not None:
- modes.add(state)
- else:
- _LOGGER.debug(
- "Device %s (%s) returned an invalid supported AC mode: %s",
- self._device.label,
- self._device.device_id,
- mode,
- )
- self._hvac_modes = list(modes)
+ await self.execute_device_command(
+ Capability.SWITCH,
+ Command.OFF,
+ )
@property
def current_temperature(self) -> float | None:
"""Return the current temperature."""
- return self._device.status.temperature
+ return self.get_attribute_value(
+ Capability.TEMPERATURE_MEASUREMENT, Attribute.TEMPERATURE
+ )
@property
- def extra_state_attributes(self) -> dict[str, Any]:
+ def extra_state_attributes(self) -> dict[str, Any] | None:
"""Return device specific state attributes.
Include attributes from the Demand Response Load Control (drlc)
and Power Consumption capabilities.
"""
- attributes = [
- "drlc_status_duration",
- "drlc_status_level",
- "drlc_status_start",
- "drlc_status_override",
- ]
- state_attributes = {}
- for attribute in attributes:
- value = getattr(self._device.status, attribute)
- if value is not None:
- state_attributes[attribute] = value
- return state_attributes
+ if not self.supports_capability(Capability.DEMAND_RESPONSE_LOAD_CONTROL):
+ return None
+
+ drlc_status = self.get_attribute_value(
+ Capability.DEMAND_RESPONSE_LOAD_CONTROL,
+ Attribute.DEMAND_RESPONSE_LOAD_CONTROL_STATUS,
+ )
+ return {
+ "drlc_status_duration": drlc_status["duration"],
+ "drlc_status_level": drlc_status["drlcLevel"],
+ "drlc_status_start": drlc_status["start"],
+ "drlc_status_override": drlc_status["override"],
+ }
@property
def fan_mode(self) -> str:
"""Return the fan setting."""
- return self._device.status.fan_mode
+ return self.get_attribute_value(
+ Capability.AIR_CONDITIONER_FAN_MODE, Attribute.FAN_MODE
+ )
@property
def fan_modes(self) -> list[str]:
"""Return the list of available fan modes."""
- return self._device.status.supported_ac_fan_modes
+ return self.get_attribute_value(
+ Capability.AIR_CONDITIONER_FAN_MODE, Attribute.SUPPORTED_AC_FAN_MODES
+ )
@property
def hvac_mode(self) -> HVACMode | None:
"""Return current operation ie. heat, cool, idle."""
- if not self._device.status.switch:
+ if self.get_attribute_value(Capability.SWITCH, Attribute.SWITCH) == "off":
return HVACMode.OFF
- return AC_MODE_TO_STATE.get(self._device.status.air_conditioner_mode)
-
- @property
- def hvac_modes(self) -> list[HVACMode]:
- """Return the list of available operation modes."""
- return self._hvac_modes
+ return AC_MODE_TO_STATE.get(
+ self.get_attribute_value(
+ Capability.AIR_CONDITIONER_MODE, Attribute.AIR_CONDITIONER_MODE
+ )
+ )
@property
def target_temperature(self) -> float:
"""Return the temperature we try to reach."""
- return self._device.status.cooling_setpoint
+ return self.get_attribute_value(
+ Capability.THERMOSTAT_COOLING_SETPOINT, Attribute.COOLING_SETPOINT
+ )
@property
def temperature_unit(self) -> str:
"""Return the unit of measurement."""
- return UNIT_MAP[self._device.status.attributes[Attribute.temperature].unit]
+ unit = self._internal_state[Capability.TEMPERATURE_MEASUREMENT][
+ Attribute.TEMPERATURE
+ ].unit
+ assert unit
+ return UNIT_MAP[unit]
def _determine_swing_modes(self) -> list[str] | None:
"""Return the list of available swing modes."""
- supported_swings = None
- supported_modes = self._device.status.attributes[
- Attribute.supported_fan_oscillation_modes
- ][0]
- if supported_modes is not None:
- supported_swings = [
- FAN_OSCILLATION_TO_SWING.get(m, SWING_OFF) for m in supported_modes
- ]
- return supported_swings
+ if (
+ supported_modes := self.get_attribute_value(
+ Capability.FAN_OSCILLATION_MODE,
+ Attribute.SUPPORTED_FAN_OSCILLATION_MODES,
+ )
+ ) is None:
+ return None
+ return [FAN_OSCILLATION_TO_SWING.get(m, SWING_OFF) for m in supported_modes]
async def async_set_swing_mode(self, swing_mode: str) -> None:
"""Set swing mode."""
- fan_oscillation_mode = SWING_TO_FAN_OSCILLATION[swing_mode]
- await self._device.set_fan_oscillation_mode(fan_oscillation_mode)
-
- # setting the fan must reset the preset mode (it deactivates the windFree function)
- self._attr_preset_mode = None
-
- self.async_schedule_update_ha_state(True)
+ await self.execute_device_command(
+ Capability.FAN_OSCILLATION_MODE,
+ Command.SET_FAN_OSCILLATION_MODE,
+ argument=SWING_TO_FAN_OSCILLATION[swing_mode],
+ )
@property
def swing_mode(self) -> str:
"""Return the swing setting."""
return FAN_OSCILLATION_TO_SWING.get(
- self._device.status.fan_oscillation_mode, SWING_OFF
+ self.get_attribute_value(
+ Capability.FAN_OSCILLATION_MODE, Attribute.FAN_OSCILLATION_MODE
+ ),
+ SWING_OFF,
)
def _determine_preset_modes(self) -> list[str] | None:
"""Return a list of available preset modes."""
- supported_modes: list | None = self._device.status.attributes[
- "supportedAcOptionalMode"
- ].value
- if supported_modes and WINDFREE in supported_modes:
- return [WINDFREE]
+ if self.supports_capability(Capability.CUSTOM_AIR_CONDITIONER_OPTIONAL_MODE):
+ supported_modes = self.get_attribute_value(
+ Capability.CUSTOM_AIR_CONDITIONER_OPTIONAL_MODE,
+ Attribute.SUPPORTED_AC_OPTIONAL_MODE,
+ )
+ if supported_modes and WINDFREE in supported_modes:
+ return [WINDFREE]
return None
async def async_set_preset_mode(self, preset_mode: str) -> None:
"""Set special modes (currently only windFree is supported)."""
- result = await self._device.command(
- "main",
- "custom.airConditionerOptionalMode",
- "setAcOptionalMode",
- [preset_mode],
+ await self.execute_device_command(
+ Capability.CUSTOM_AIR_CONDITIONER_OPTIONAL_MODE,
+ Command.SET_AC_OPTIONAL_MODE,
+ argument=preset_mode,
)
- if result:
- self._device.status.update_attribute_value("acOptionalMode", preset_mode)
- self._attr_preset_mode = preset_mode
-
- self.async_write_ha_state()
+ def _determine_hvac_modes(self) -> list[HVACMode]:
+ """Determine the supported HVAC modes."""
+ modes = [HVACMode.OFF]
+ if (
+ ac_modes := self.get_attribute_value(
+ Capability.AIR_CONDITIONER_MODE, Attribute.SUPPORTED_AC_MODES
+ )
+ ) is not None:
+ modes.extend(
+ state
+ for mode in ac_modes
+ if (state := AC_MODE_TO_STATE.get(mode)) is not None
+ if state not in modes
+ )
+ return modes
diff --git a/homeassistant/components/smartthings/config_flow.py b/homeassistant/components/smartthings/config_flow.py
index 7b49854740a..d2654348527 100644
--- a/homeassistant/components/smartthings/config_flow.py
+++ b/homeassistant/components/smartthings/config_flow.py
@@ -1,298 +1,96 @@
"""Config flow to configure SmartThings."""
from collections.abc import Mapping
-from http import HTTPStatus
import logging
from typing import Any
-from aiohttp import ClientResponseError
-from pysmartthings import APIResponseError, AppOAuth, SmartThings
-from pysmartthings.installedapp import format_install_url
-import voluptuous as vol
+from pysmartthings import SmartThings
-from homeassistant.config_entries import SOURCE_REAUTH, ConfigFlow, ConfigFlowResult
-from homeassistant.const import CONF_ACCESS_TOKEN, CONF_CLIENT_ID, CONF_CLIENT_SECRET
+from homeassistant.config_entries import SOURCE_REAUTH, ConfigFlowResult
+from homeassistant.const import CONF_ACCESS_TOKEN, CONF_TOKEN
from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.config_entry_oauth2_flow import AbstractOAuth2FlowHandler
-from .const import (
- APP_OAUTH_CLIENT_NAME,
- APP_OAUTH_SCOPES,
- CONF_APP_ID,
- CONF_INSTALLED_APP_ID,
- CONF_LOCATION_ID,
- CONF_REFRESH_TOKEN,
- DOMAIN,
- VAL_UID_MATCHER,
-)
-from .smartapp import (
- create_app,
- find_app,
- format_unique_id,
- get_webhook_url,
- setup_smartapp,
- setup_smartapp_endpoint,
- update_app,
- validate_webhook_requirements,
-)
+from .const import CONF_LOCATION_ID, DOMAIN, OLD_DATA, REQUESTED_SCOPES, SCOPES
_LOGGER = logging.getLogger(__name__)
-class SmartThingsFlowHandler(ConfigFlow, domain=DOMAIN):
+class SmartThingsConfigFlow(AbstractOAuth2FlowHandler, domain=DOMAIN):
"""Handle configuration of SmartThings integrations."""
- VERSION = 2
+ VERSION = 3
+ DOMAIN = DOMAIN
- api: SmartThings
- app_id: str
- location_id: str
+ @property
+ def logger(self) -> logging.Logger:
+ """Return logger."""
+ return logging.getLogger(__name__)
- def __init__(self) -> None:
- """Create a new instance of the flow handler."""
- self.access_token: str | None = None
- self.oauth_client_secret = None
- self.oauth_client_id = None
- self.installed_app_id = None
- self.refresh_token = None
- self.endpoints_initialized = False
-
- async def async_step_import(self, import_data: None) -> ConfigFlowResult:
- """Occurs when a previously entry setup fails and is re-initiated."""
- return await self.async_step_user(import_data)
+ @property
+ def extra_authorize_data(self) -> dict[str, Any]:
+ """Extra data that needs to be appended to the authorize url."""
+ return {"scope": " ".join(REQUESTED_SCOPES)}
async def async_step_user(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
- """Validate and confirm webhook setup."""
- if not self.endpoints_initialized:
- self.endpoints_initialized = True
- await setup_smartapp_endpoint(
- self.hass, len(self._async_current_entries()) == 0
- )
- webhook_url = get_webhook_url(self.hass)
-
- # Abort if the webhook is invalid
- if not validate_webhook_requirements(self.hass):
+ """Check we have the cloud integration set up."""
+ if "cloud" not in self.hass.config.components:
return self.async_abort(
- reason="invalid_webhook_url",
- description_placeholders={
- "webhook_url": webhook_url,
- "component_url": (
- "https://www.home-assistant.io/integrations/smartthings/"
- ),
+ reason="cloud_not_enabled",
+ description_placeholders={"default_config": "default_config"},
+ )
+ return await super().async_step_user(user_input)
+
+ async def async_oauth_create_entry(self, data: dict[str, Any]) -> ConfigFlowResult:
+ """Create an entry for SmartThings."""
+ if not set(data[CONF_TOKEN]["scope"].split()) >= set(SCOPES):
+ return self.async_abort(reason="missing_scopes")
+ client = SmartThings(session=async_get_clientsession(self.hass))
+ client.authenticate(data[CONF_TOKEN][CONF_ACCESS_TOKEN])
+ locations = await client.get_locations()
+ location = locations[0]
+ # We pick to use the location id as unique id rather than the installed app id
+ # as the installed app id could change with the right settings in the SmartApp
+ # or the app used to sign in changed for any reason.
+ await self.async_set_unique_id(location.location_id)
+ if self.source != SOURCE_REAUTH:
+ self._abort_if_unique_id_configured()
+
+ return self.async_create_entry(
+ title=location.name,
+ data={**data, CONF_LOCATION_ID: location.location_id},
+ )
+
+ if (entry := self._get_reauth_entry()) and CONF_TOKEN not in entry.data:
+ if entry.data[OLD_DATA][CONF_LOCATION_ID] != location.location_id:
+ return self.async_abort(reason="reauth_location_mismatch")
+ return self.async_update_reload_and_abort(
+ self._get_reauth_entry(),
+ data_updates={
+ **data,
+ CONF_LOCATION_ID: location.location_id,
},
+ unique_id=location.location_id,
)
-
- # Show the confirmation
- if user_input is None:
- return self.async_show_form(
- step_id="user",
- description_placeholders={"webhook_url": webhook_url},
- )
-
- # Show the next screen
- return await self.async_step_pat()
-
- async def async_step_pat(
- self, user_input: dict[str, str] | None = None
- ) -> ConfigFlowResult:
- """Get the Personal Access Token and validate it."""
- errors: dict[str, str] = {}
- if user_input is None or CONF_ACCESS_TOKEN not in user_input:
- return self._show_step_pat(errors)
-
- self.access_token = user_input[CONF_ACCESS_TOKEN]
-
- # Ensure token is a UUID
- if not VAL_UID_MATCHER.match(self.access_token):
- errors[CONF_ACCESS_TOKEN] = "token_invalid_format"
- return self._show_step_pat(errors)
-
- # Setup end-point
- self.api = SmartThings(async_get_clientsession(self.hass), self.access_token)
- try:
- app = await find_app(self.hass, self.api)
- if app:
- await app.refresh() # load all attributes
- await update_app(self.hass, app)
- # Find an existing entry to copy the oauth client
- existing = next(
- (
- entry
- for entry in self._async_current_entries()
- if entry.data[CONF_APP_ID] == app.app_id
- ),
- None,
- )
- if existing:
- self.oauth_client_id = existing.data[CONF_CLIENT_ID]
- self.oauth_client_secret = existing.data[CONF_CLIENT_SECRET]
- else:
- # Get oauth client id/secret by regenerating it
- app_oauth = AppOAuth(app.app_id)
- app_oauth.client_name = APP_OAUTH_CLIENT_NAME
- app_oauth.scope.extend(APP_OAUTH_SCOPES)
- client = await self.api.generate_app_oauth(app_oauth)
- self.oauth_client_secret = client.client_secret
- self.oauth_client_id = client.client_id
- else:
- app, client = await create_app(self.hass, self.api)
- self.oauth_client_secret = client.client_secret
- self.oauth_client_id = client.client_id
- setup_smartapp(self.hass, app)
- self.app_id = app.app_id
-
- except APIResponseError as ex:
- if ex.is_target_error():
- errors["base"] = "webhook_error"
- else:
- errors["base"] = "app_setup_error"
- _LOGGER.exception(
- "API error setting up the SmartApp: %s", ex.raw_error_response
- )
- return self._show_step_pat(errors)
- except ClientResponseError as ex:
- if ex.status == HTTPStatus.UNAUTHORIZED:
- errors[CONF_ACCESS_TOKEN] = "token_unauthorized"
- _LOGGER.debug(
- "Unauthorized error received setting up SmartApp", exc_info=True
- )
- elif ex.status == HTTPStatus.FORBIDDEN:
- errors[CONF_ACCESS_TOKEN] = "token_forbidden"
- _LOGGER.debug(
- "Forbidden error received setting up SmartApp", exc_info=True
- )
- else:
- errors["base"] = "app_setup_error"
- _LOGGER.exception("Unexpected error setting up the SmartApp")
- return self._show_step_pat(errors)
- except Exception:
- errors["base"] = "app_setup_error"
- _LOGGER.exception("Unexpected error setting up the SmartApp")
- return self._show_step_pat(errors)
-
- return await self.async_step_select_location()
-
- async def async_step_select_location(
- self, user_input: dict[str, str] | None = None
- ) -> ConfigFlowResult:
- """Ask user to select the location to setup."""
- if user_input is None or CONF_LOCATION_ID not in user_input:
- # Get available locations
- existing_locations = [
- entry.data[CONF_LOCATION_ID] for entry in self._async_current_entries()
- ]
- locations = await self.api.locations()
- locations_options = {
- location.location_id: location.name
- for location in locations
- if location.location_id not in existing_locations
- }
- if not locations_options:
- return self.async_abort(reason="no_available_locations")
-
- return self.async_show_form(
- step_id="select_location",
- data_schema=vol.Schema(
- {vol.Required(CONF_LOCATION_ID): vol.In(locations_options)}
- ),
- )
-
- self.location_id = user_input[CONF_LOCATION_ID]
- await self.async_set_unique_id(format_unique_id(self.app_id, self.location_id))
- return await self.async_step_authorize()
-
- async def async_step_authorize(
- self, user_input: dict[str, Any] | None = None
- ) -> ConfigFlowResult:
- """Wait for the user to authorize the app installation."""
- user_input = {} if user_input is None else user_input
- self.installed_app_id = user_input.get(CONF_INSTALLED_APP_ID)
- self.refresh_token = user_input.get(CONF_REFRESH_TOKEN)
- if self.installed_app_id is None:
- # Launch the external setup URL
- url = format_install_url(self.app_id, self.location_id)
- return self.async_external_step(step_id="authorize", url=url)
-
- next_step_id = "install"
- if self.source == SOURCE_REAUTH:
- next_step_id = "update"
- return self.async_external_step_done(next_step_id=next_step_id)
-
- def _show_step_pat(self, errors):
- if self.access_token is None:
- # Get the token from an existing entry to make it easier to setup multiple locations.
- self.access_token = next(
- (
- entry.data.get(CONF_ACCESS_TOKEN)
- for entry in self._async_current_entries()
- ),
- None,
- )
-
- return self.async_show_form(
- step_id="pat",
- data_schema=vol.Schema(
- {vol.Required(CONF_ACCESS_TOKEN, default=self.access_token): str}
- ),
- errors=errors,
- description_placeholders={
- "token_url": "https://account.smartthings.com/tokens",
- "component_url": (
- "https://www.home-assistant.io/integrations/smartthings/"
- ),
- },
+ self._abort_if_unique_id_mismatch(reason="reauth_account_mismatch")
+ return self.async_update_reload_and_abort(
+ self._get_reauth_entry(), data_updates=data
)
async def async_step_reauth(
self, entry_data: Mapping[str, Any]
) -> ConfigFlowResult:
- """Handle re-authentication of an existing config entry."""
+ """Perform reauth upon migration of old entries."""
return await self.async_step_reauth_confirm()
async def async_step_reauth_confirm(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
- """Handle re-authentication of an existing config entry."""
+ """Confirm reauth dialog."""
if user_input is None:
- return self.async_show_form(step_id="reauth_confirm")
- self.app_id = self._get_reauth_entry().data[CONF_APP_ID]
- self.location_id = self._get_reauth_entry().data[CONF_LOCATION_ID]
- self._set_confirm_only()
- return await self.async_step_authorize()
-
- async def async_step_update(
- self, user_input: dict[str, Any] | None = None
- ) -> ConfigFlowResult:
- """Handle re-authentication of an existing config entry."""
- return await self.async_step_update_confirm()
-
- async def async_step_update_confirm(
- self, user_input: dict[str, Any] | None = None
- ) -> ConfigFlowResult:
- """Handle re-authentication of an existing config entry."""
- if user_input is None:
- self._set_confirm_only()
- return self.async_show_form(step_id="update_confirm")
- entry = self._get_reauth_entry()
- return self.async_update_reload_and_abort(
- entry, data_updates={CONF_REFRESH_TOKEN: self.refresh_token}
- )
-
- async def async_step_install(
- self, user_input: dict[str, Any] | None = None
- ) -> ConfigFlowResult:
- """Create a config entry at completion of a flow and authorization of the app."""
- data = {
- CONF_ACCESS_TOKEN: self.access_token,
- CONF_REFRESH_TOKEN: self.refresh_token,
- CONF_CLIENT_ID: self.oauth_client_id,
- CONF_CLIENT_SECRET: self.oauth_client_secret,
- CONF_LOCATION_ID: self.location_id,
- CONF_APP_ID: self.app_id,
- CONF_INSTALLED_APP_ID: self.installed_app_id,
- }
-
- location = await self.api.location(data[CONF_LOCATION_ID])
-
- return self.async_create_entry(title=location.name, data=data)
+ return self.async_show_form(
+ step_id="reauth_confirm",
+ )
+ return await self.async_step_user()
diff --git a/homeassistant/components/smartthings/const.py b/homeassistant/components/smartthings/const.py
index e50837697e7..2ba59ade4e8 100644
--- a/homeassistant/components/smartthings/const.py
+++ b/homeassistant/components/smartthings/const.py
@@ -1,15 +1,27 @@
"""Constants used by the SmartThings component and platforms."""
-from datetime import timedelta
-import re
-
-from homeassistant.const import Platform
-
DOMAIN = "smartthings"
-APP_OAUTH_CLIENT_NAME = "Home Assistant"
-APP_OAUTH_SCOPES = ["r:devices:*"]
-APP_NAME_PREFIX = "homeassistant."
+SCOPES = [
+ "r:devices:*",
+ "w:devices:*",
+ "x:devices:*",
+ "r:hubs:*",
+ "r:locations:*",
+ "w:locations:*",
+ "x:locations:*",
+ "r:scenes:*",
+ "x:scenes:*",
+ "r:rules:*",
+ "w:rules:*",
+ "sse",
+]
+
+REQUESTED_SCOPES = [
+ *SCOPES,
+ "r:installedapps",
+ "w:installedapps",
+]
CONF_APP_ID = "app_id"
CONF_CLOUDHOOK_URL = "cloudhook_url"
@@ -18,41 +30,8 @@ CONF_INSTANCE_ID = "instance_id"
CONF_LOCATION_ID = "location_id"
CONF_REFRESH_TOKEN = "refresh_token"
-DATA_MANAGER = "manager"
-DATA_BROKERS = "brokers"
+MAIN = "main"
+OLD_DATA = "old_data"
+
+CONF_SUBSCRIPTION_ID = "subscription_id"
EVENT_BUTTON = "smartthings.button"
-
-SIGNAL_SMARTTHINGS_UPDATE = "smartthings_update"
-SIGNAL_SMARTAPP_PREFIX = "smartthings_smartap_"
-
-SETTINGS_INSTANCE_ID = "hassInstanceId"
-
-SUBSCRIPTION_WARNING_LIMIT = 40
-
-STORAGE_KEY = DOMAIN
-STORAGE_VERSION = 1
-
-# Ordered 'specific to least-specific platform' in order for capabilities
-# to be drawn-down and represented by the most appropriate platform.
-PLATFORMS = [
- Platform.BINARY_SENSOR,
- Platform.CLIMATE,
- Platform.COVER,
- Platform.FAN,
- Platform.LIGHT,
- Platform.LOCK,
- Platform.SCENE,
- Platform.SENSOR,
- Platform.SWITCH,
-]
-
-IGNORED_CAPABILITIES = [
- "execute",
- "healthCheck",
- "ocf",
-]
-
-TOKEN_REFRESH_INTERVAL = timedelta(days=14)
-
-VAL_UID = "^(?:([0-9a-fA-F]{32})|([0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}))$"
-VAL_UID_MATCHER = re.compile(VAL_UID)
diff --git a/homeassistant/components/smartthings/cover.py b/homeassistant/components/smartthings/cover.py
index 55e86bd582e..0b68409443d 100644
--- a/homeassistant/components/smartthings/cover.py
+++ b/homeassistant/components/smartthings/cover.py
@@ -2,25 +2,23 @@
from __future__ import annotations
-from collections.abc import Sequence
from typing import Any
-from pysmartthings import Attribute, Capability
+from pysmartthings import Attribute, Capability, Command, SmartThings
from homeassistant.components.cover import (
ATTR_POSITION,
- DOMAIN as COVER_DOMAIN,
CoverDeviceClass,
CoverEntity,
CoverEntityFeature,
CoverState,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_BATTERY_LEVEL
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DATA_BROKERS, DOMAIN
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
from .entity import SmartThingsEntity
VALUE_TO_STATE = {
@@ -32,114 +30,108 @@ VALUE_TO_STATE = {
"unknown": None,
}
+CAPABILITIES = (Capability.WINDOW_SHADE, Capability.DOOR_CONTROL)
+
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add covers for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
+ entry_data = entry.runtime_data
async_add_entities(
- [
- SmartThingsCover(device)
- for device in broker.devices.values()
- if broker.any_assigned(device.device_id, COVER_DOMAIN)
- ],
- True,
+ SmartThingsCover(entry_data.client, device, Capability(capability))
+ for device in entry_data.devices.values()
+ for capability in device.status[MAIN]
+ if capability in CAPABILITIES
)
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
- min_required = [
- Capability.door_control,
- Capability.garage_door_control,
- Capability.window_shade,
- ]
- # Must have one of the min_required
- if any(capability in capabilities for capability in min_required):
- # Return all capabilities supported/consumed
- return [
- *min_required,
- Capability.battery,
- Capability.switch_level,
- Capability.window_shade_level,
- ]
-
- return None
-
-
class SmartThingsCover(SmartThingsEntity, CoverEntity):
"""Define a SmartThings cover."""
- def __init__(self, device):
+ _attr_name = None
+ _state: CoverState | None = None
+
+ def __init__(
+ self,
+ client: SmartThings,
+ device: FullDevice,
+ capability: Capability,
+ ) -> None:
"""Initialize the cover class."""
- super().__init__(device)
- self._current_cover_position = None
- self._state = None
+ super().__init__(
+ client,
+ device,
+ {
+ capability,
+ Capability.BATTERY,
+ Capability.WINDOW_SHADE_LEVEL,
+ Capability.SWITCH_LEVEL,
+ },
+ )
+ self.capability = capability
self._attr_supported_features = (
CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE
)
- if (
- Capability.switch_level in device.capabilities
- or Capability.window_shade_level in device.capabilities
- ):
+ if self.supports_capability(Capability.WINDOW_SHADE_LEVEL):
+ self.level_capability = Capability.WINDOW_SHADE_LEVEL
+ self.level_command = Command.SET_SHADE_LEVEL
+ else:
+ self.level_capability = Capability.SWITCH_LEVEL
+ self.level_command = Command.SET_LEVEL
+ if self.supports_capability(
+ Capability.SWITCH_LEVEL
+ ) or self.supports_capability(Capability.WINDOW_SHADE_LEVEL):
self._attr_supported_features |= CoverEntityFeature.SET_POSITION
- if Capability.door_control in device.capabilities:
+ if self.supports_capability(Capability.DOOR_CONTROL):
self._attr_device_class = CoverDeviceClass.DOOR
- elif Capability.window_shade in device.capabilities:
+ elif self.supports_capability(Capability.WINDOW_SHADE):
self._attr_device_class = CoverDeviceClass.SHADE
- elif Capability.garage_door_control in device.capabilities:
- self._attr_device_class = CoverDeviceClass.GARAGE
async def async_close_cover(self, **kwargs: Any) -> None:
"""Close cover."""
- # Same command for all 3 supported capabilities
- await self._device.close(set_status=True)
- # State is set optimistically in the commands above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_schedule_update_ha_state(True)
+ await self.execute_device_command(self.capability, Command.CLOSE)
async def async_open_cover(self, **kwargs: Any) -> None:
"""Open the cover."""
- # Same for all capability types
- await self._device.open(set_status=True)
- # State is set optimistically in the commands above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_schedule_update_ha_state(True)
+ await self.execute_device_command(self.capability, Command.OPEN)
async def async_set_cover_position(self, **kwargs: Any) -> None:
"""Move the cover to a specific position."""
- if not self.supported_features & CoverEntityFeature.SET_POSITION:
- return
- # Do not set_status=True as device will report progress.
- if Capability.window_shade_level in self._device.capabilities:
- await self._device.set_window_shade_level(
- kwargs[ATTR_POSITION], set_status=False
- )
- else:
- await self._device.set_level(kwargs[ATTR_POSITION], set_status=False)
+ await self.execute_device_command(
+ self.level_capability,
+ self.level_command,
+ argument=kwargs[ATTR_POSITION],
+ )
- async def async_update(self) -> None:
+ def _update_attr(self) -> None:
"""Update the attrs of the cover."""
- if Capability.door_control in self._device.capabilities:
- self._state = VALUE_TO_STATE.get(self._device.status.door)
- elif Capability.window_shade in self._device.capabilities:
- self._state = VALUE_TO_STATE.get(self._device.status.window_shade)
- elif Capability.garage_door_control in self._device.capabilities:
- self._state = VALUE_TO_STATE.get(self._device.status.door)
+ attribute = {
+ Capability.WINDOW_SHADE: Attribute.WINDOW_SHADE,
+ Capability.DOOR_CONTROL: Attribute.DOOR,
+ }[self.capability]
+ self._state = VALUE_TO_STATE.get(
+ self.get_attribute_value(self.capability, attribute)
+ )
- if Capability.window_shade_level in self._device.capabilities:
- self._attr_current_cover_position = self._device.status.shade_level
- elif Capability.switch_level in self._device.capabilities:
- self._attr_current_cover_position = self._device.status.level
+ if self.supports_capability(Capability.SWITCH_LEVEL):
+ self._attr_current_cover_position = self.get_attribute_value(
+ Capability.SWITCH_LEVEL, Attribute.LEVEL
+ )
+ elif self.supports_capability(Capability.WINDOW_SHADE_LEVEL):
+ self._attr_current_cover_position = self.get_attribute_value(
+ Capability.WINDOW_SHADE_LEVEL, Attribute.SHADE_LEVEL
+ )
+ # Deprecated, remove in 2025.10
self._attr_extra_state_attributes = {}
- battery = self._device.status.attributes[Attribute.battery].value
- if battery is not None:
- self._attr_extra_state_attributes[ATTR_BATTERY_LEVEL] = battery
+ if self.supports_capability(Capability.BATTERY):
+ self._attr_extra_state_attributes[ATTR_BATTERY_LEVEL] = (
+ self.get_attribute_value(Capability.BATTERY, Attribute.BATTERY)
+ )
@property
def is_opening(self) -> bool:
diff --git a/homeassistant/components/smartthings/diagnostics.py b/homeassistant/components/smartthings/diagnostics.py
new file mode 100644
index 00000000000..04517112802
--- /dev/null
+++ b/homeassistant/components/smartthings/diagnostics.py
@@ -0,0 +1,56 @@
+"""Diagnostics support for SmartThings."""
+
+from __future__ import annotations
+
+import asyncio
+from dataclasses import asdict
+from typing import Any
+
+from pysmartthings import DeviceEvent
+
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntry
+
+from . import SmartThingsConfigEntry
+from .const import DOMAIN
+
+EVENT_WAIT_TIME = 5
+
+
+async def async_get_config_entry_diagnostics(
+ hass: HomeAssistant,
+ entry: SmartThingsConfigEntry,
+) -> dict[str, Any]:
+ """Return diagnostics for a config entry."""
+ client = entry.runtime_data.client
+ return {"devices": await client.get_raw_devices()}
+
+
+async def async_get_device_diagnostics(
+ hass: HomeAssistant, entry: SmartThingsConfigEntry, device: DeviceEntry
+) -> dict[str, Any]:
+ """Return diagnostics for a device entry."""
+ client = entry.runtime_data.client
+ device_id = next(
+ identifier for identifier in device.identifiers if identifier[0] == DOMAIN
+ )[1]
+
+ device_status = await client.get_raw_device_status(device_id)
+ device_info = await client.get_raw_device(device_id)
+
+ events: list[DeviceEvent] = []
+
+ def register_event(event: DeviceEvent) -> None:
+ events.append(event)
+
+ listener = client.add_device_event_listener(device_id, register_event)
+
+ await asyncio.sleep(EVENT_WAIT_TIME)
+
+ listener()
+
+ return {
+ "events": [asdict(event) for event in events],
+ "status": device_status,
+ "info": device_info,
+ }
diff --git a/homeassistant/components/smartthings/entity.py b/homeassistant/components/smartthings/entity.py
index cc63213d122..12c07bea983 100644
--- a/homeassistant/components/smartthings/entity.py
+++ b/homeassistant/components/smartthings/entity.py
@@ -2,49 +2,98 @@
from __future__ import annotations
-from pysmartthings.device import DeviceEntity
+from typing import Any
+
+from pysmartthings import (
+ Attribute,
+ Capability,
+ Command,
+ DeviceEvent,
+ SmartThings,
+ Status,
+)
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity
-from .const import DOMAIN, SIGNAL_SMARTTHINGS_UPDATE
+from . import FullDevice
+from .const import DOMAIN, MAIN
class SmartThingsEntity(Entity):
"""Defines a SmartThings entity."""
_attr_should_poll = False
+ _attr_has_entity_name = True
- def __init__(self, device: DeviceEntity) -> None:
+ def __init__(
+ self,
+ client: SmartThings,
+ device: FullDevice,
+ capabilities: set[Capability],
+ *,
+ component: str = MAIN,
+ ) -> None:
"""Initialize the instance."""
- self._device = device
- self._dispatcher_remove = None
- self._attr_name = device.label
- self._attr_unique_id = device.device_id
+ self.client = client
+ self.capabilities = capabilities
+ self.component = component
+ self._internal_state: dict[Capability | str, dict[Attribute | str, Status]] = {
+ capability: device.status[component][capability]
+ for capability in capabilities
+ if capability in device.status[component]
+ }
+ self.device = device
+ self._attr_unique_id = device.device.device_id
self._attr_device_info = DeviceInfo(
- configuration_url="https://account.smartthings.com",
- identifiers={(DOMAIN, device.device_id)},
- manufacturer=device.status.ocf_manufacturer_name,
- model=device.status.ocf_model_number,
- name=device.label,
- hw_version=device.status.ocf_hardware_version,
- sw_version=device.status.ocf_firmware_version,
+ identifiers={(DOMAIN, device.device.device_id)},
)
- async def async_added_to_hass(self):
- """Device added to hass."""
+ async def async_added_to_hass(self) -> None:
+ """Subscribe to updates."""
+ await super().async_added_to_hass()
+ for capability in self._internal_state:
+ self.async_on_remove(
+ self.client.add_device_capability_event_listener(
+ self.device.device.device_id,
+ self.component,
+ capability,
+ self._update_handler,
+ )
+ )
+ self._update_attr()
- async def async_update_state(devices):
- """Update device state."""
- if self._device.device_id in devices:
- await self.async_update_ha_state(True)
+ def _update_handler(self, event: DeviceEvent) -> None:
+ self._internal_state[event.capability][event.attribute].value = event.value
+ self._internal_state[event.capability][event.attribute].data = event.data
+ self._handle_update()
- self._dispatcher_remove = async_dispatcher_connect(
- self.hass, SIGNAL_SMARTTHINGS_UPDATE, async_update_state
+ def supports_capability(self, capability: Capability) -> bool:
+ """Test if device supports a capability."""
+ return capability in self.device.status[self.component]
+
+ def get_attribute_value(self, capability: Capability, attribute: Attribute) -> Any:
+ """Get the value of a device attribute."""
+ return self._internal_state[capability][attribute].value
+
+ def _update_attr(self) -> None:
+ """Update the attributes."""
+
+ def _handle_update(self) -> None:
+ """Handle updated data from the coordinator."""
+ self._update_attr()
+ self.async_write_ha_state()
+
+ async def execute_device_command(
+ self,
+ capability: Capability,
+ command: Command,
+ argument: int | str | list[Any] | dict[str, Any] | None = None,
+ ) -> None:
+ """Execute a command on the device."""
+ kwargs = {}
+ if argument is not None:
+ kwargs["argument"] = argument
+ await self.client.execute_device_command(
+ self.device.device.device_id, capability, command, self.component, **kwargs
)
-
- async def async_will_remove_from_hass(self) -> None:
- """Disconnect the device when removed."""
- if self._dispatcher_remove:
- self._dispatcher_remove()
diff --git a/homeassistant/components/smartthings/event.py b/homeassistant/components/smartthings/event.py
new file mode 100644
index 00000000000..e22a32c7726
--- /dev/null
+++ b/homeassistant/components/smartthings/event.py
@@ -0,0 +1,60 @@
+"""Support for events through the SmartThings cloud API."""
+
+from __future__ import annotations
+
+from typing import cast
+
+from pysmartthings import Attribute, Capability, Component, DeviceEvent, SmartThings
+
+from homeassistant.components.event import EventDeviceClass, EventEntity
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import FullDevice, SmartThingsConfigEntry
+from .entity import SmartThingsEntity
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add events for a config entry."""
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsButtonEvent(entry_data.client, device, component)
+ for device in entry_data.devices.values()
+ for component in device.device.components
+ if Capability.BUTTON in component.capabilities
+ )
+
+
+class SmartThingsButtonEvent(SmartThingsEntity, EventEntity):
+ """Define a SmartThings event."""
+
+ _attr_device_class = EventDeviceClass.BUTTON
+ _attr_translation_key = "button"
+
+ def __init__(
+ self,
+ client: SmartThings,
+ device: FullDevice,
+ component: Component,
+ ) -> None:
+ """Init the class."""
+ super().__init__(client, device, {Capability.BUTTON}, component=component.id)
+ self._attr_name = component.label
+ self._attr_unique_id = (
+ f"{device.device.device_id}_{component.id}_{Capability.BUTTON}"
+ )
+
+ @property
+ def event_types(self) -> list[str]:
+ """Return the event types."""
+ return self.get_attribute_value(
+ Capability.BUTTON, Attribute.SUPPORTED_BUTTON_VALUES
+ )
+
+ def _update_handler(self, event: DeviceEvent) -> None:
+ self._trigger_event(cast(str, event.value))
+ self.async_write_ha_state()
diff --git a/homeassistant/components/smartthings/fan.py b/homeassistant/components/smartthings/fan.py
index 61e30589273..1c4cb4edc4a 100644
--- a/homeassistant/components/smartthings/fan.py
+++ b/homeassistant/components/smartthings/fan.py
@@ -2,23 +2,22 @@
from __future__ import annotations
-from collections.abc import Sequence
import math
from typing import Any
-from pysmartthings import Capability
+from pysmartthings import Attribute, Capability, Command, SmartThings
from homeassistant.components.fan import FanEntity, FanEntityFeature
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
)
from homeassistant.util.scaling import int_states_in_range
-from .const import DATA_BROKERS, DOMAIN
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
from .entity import SmartThingsEntity
SPEED_RANGE = (1, 3) # off is not included
@@ -26,86 +25,74 @@ SPEED_RANGE = (1, 3) # off is not included
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add fans for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
+ entry_data = entry.runtime_data
async_add_entities(
- SmartThingsFan(device)
- for device in broker.devices.values()
- if broker.any_assigned(device.device_id, "fan")
+ SmartThingsFan(entry_data.client, device)
+ for device in entry_data.devices.values()
+ if Capability.SWITCH in device.status[MAIN]
+ and any(
+ capability in device.status[MAIN]
+ for capability in (
+ Capability.FAN_SPEED,
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ )
+ )
+ and Capability.THERMOSTAT_COOLING_SETPOINT not in device.status[MAIN]
)
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
-
- # MUST support switch as we need a way to turn it on and off
- if Capability.switch not in capabilities:
- return None
-
- # These are all optional but at least one must be supported
- optional = [
- Capability.air_conditioner_fan_mode,
- Capability.fan_speed,
- ]
-
- # At least one of the optional capabilities must be supported
- # to classify this entity as a fan.
- # If they are not then return None and don't setup the platform.
- if not any(capability in capabilities for capability in optional):
- return None
-
- supported = [Capability.switch]
-
- supported.extend(
- capability for capability in optional if capability in capabilities
- )
-
- return supported
-
-
class SmartThingsFan(SmartThingsEntity, FanEntity):
"""Define a SmartThings Fan."""
+ _attr_name = None
_attr_speed_count = int_states_in_range(SPEED_RANGE)
- def __init__(self, device):
+ def __init__(self, client: SmartThings, device: FullDevice) -> None:
"""Init the class."""
- super().__init__(device)
+ super().__init__(
+ client,
+ device,
+ {
+ Capability.SWITCH,
+ Capability.FAN_SPEED,
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ },
+ )
self._attr_supported_features = self._determine_features()
def _determine_features(self):
flags = FanEntityFeature.TURN_OFF | FanEntityFeature.TURN_ON
- if self._device.get_capability(Capability.fan_speed):
+ if self.supports_capability(Capability.FAN_SPEED):
flags |= FanEntityFeature.SET_SPEED
- if self._device.get_capability(Capability.air_conditioner_fan_mode):
+ if self.supports_capability(Capability.AIR_CONDITIONER_FAN_MODE):
flags |= FanEntityFeature.PRESET_MODE
return flags
async def async_set_percentage(self, percentage: int) -> None:
"""Set the speed percentage of the fan."""
- await self._async_set_percentage(percentage)
-
- async def _async_set_percentage(self, percentage: int | None) -> None:
- if percentage is None:
- await self._device.switch_on(set_status=True)
- elif percentage == 0:
- await self._device.switch_off(set_status=True)
+ if percentage == 0:
+ await self.execute_device_command(Capability.SWITCH, Command.OFF)
else:
value = math.ceil(percentage_to_ranged_value(SPEED_RANGE, percentage))
- await self._device.set_fan_speed(value, set_status=True)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
+ await self.execute_device_command(
+ Capability.FAN_SPEED,
+ Command.SET_FAN_SPEED,
+ argument=value,
+ )
async def async_set_preset_mode(self, preset_mode: str) -> None:
"""Set the preset_mode of the fan."""
- await self._device.set_fan_mode(preset_mode, set_status=True)
- self.async_write_ha_state()
+ await self.execute_device_command(
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ Command.SET_FAN_MODE,
+ argument=preset_mode,
+ )
async def async_turn_on(
self,
@@ -114,32 +101,30 @@ class SmartThingsFan(SmartThingsEntity, FanEntity):
**kwargs: Any,
) -> None:
"""Turn the fan on."""
- if FanEntityFeature.SET_SPEED in self._attr_supported_features:
- # If speed is set in features then turn the fan on with the speed.
- await self._async_set_percentage(percentage)
+ if (
+ FanEntityFeature.SET_SPEED in self._attr_supported_features
+ and percentage is not None
+ ):
+ await self.async_set_percentage(percentage)
else:
- # If speed is not valid then turn on the fan with the
- await self._device.switch_on(set_status=True)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
+ await self.execute_device_command(Capability.SWITCH, Command.ON)
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the fan off."""
- await self._device.switch_off(set_status=True)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
+ await self.execute_device_command(Capability.SWITCH, Command.OFF)
@property
def is_on(self) -> bool:
"""Return true if fan is on."""
- return self._device.status.switch
+ return self.get_attribute_value(Capability.SWITCH, Attribute.SWITCH) == "on"
@property
def percentage(self) -> int | None:
"""Return the current speed percentage."""
- return ranged_value_to_percentage(SPEED_RANGE, self._device.status.fan_speed)
+ return ranged_value_to_percentage(
+ SPEED_RANGE,
+ self.get_attribute_value(Capability.FAN_SPEED, Attribute.FAN_SPEED),
+ )
@property
def preset_mode(self) -> str | None:
@@ -147,7 +132,11 @@ class SmartThingsFan(SmartThingsEntity, FanEntity):
Requires FanEntityFeature.PRESET_MODE.
"""
- return self._device.status.fan_mode
+ if not self.supports_capability(Capability.AIR_CONDITIONER_FAN_MODE):
+ return None
+ return self.get_attribute_value(
+ Capability.AIR_CONDITIONER_FAN_MODE, Attribute.FAN_MODE
+ )
@property
def preset_modes(self) -> list[str] | None:
@@ -155,4 +144,8 @@ class SmartThingsFan(SmartThingsEntity, FanEntity):
Requires FanEntityFeature.PRESET_MODE.
"""
- return self._device.status.supported_ac_fan_modes
+ if not self.supports_capability(Capability.AIR_CONDITIONER_FAN_MODE):
+ return None
+ return self.get_attribute_value(
+ Capability.AIR_CONDITIONER_FAN_MODE, Attribute.SUPPORTED_AC_FAN_MODES
+ )
diff --git a/homeassistant/components/smartthings/icons.json b/homeassistant/components/smartthings/icons.json
new file mode 100644
index 00000000000..80ac70edc3f
--- /dev/null
+++ b/homeassistant/components/smartthings/icons.json
@@ -0,0 +1,45 @@
+{
+ "entity": {
+ "binary_sensor": {
+ "remote_control": {
+ "default": "mdi:remote-off",
+ "state": {
+ "on": "mdi:remote"
+ }
+ },
+ "child_lock": {
+ "default": "mdi:lock-open",
+ "state": {
+ "on": "mdi:lock"
+ }
+ }
+ },
+ "button": {
+ "stop": {
+ "default": "mdi:stop"
+ }
+ },
+ "number": {
+ "washer_rinse_cycles": {
+ "default": "mdi:waves-arrow-up"
+ }
+ },
+ "select": {
+ "operating_state": {
+ "state": {
+ "run": "mdi:play",
+ "pause": "mdi:pause",
+ "stop": "mdi:stop"
+ }
+ }
+ },
+ "switch": {
+ "wrinkle_prevent": {
+ "default": "mdi:tumble-dryer",
+ "state": {
+ "off": "mdi:tumble-dryer-off"
+ }
+ }
+ }
+ }
+}
diff --git a/homeassistant/components/smartthings/light.py b/homeassistant/components/smartthings/light.py
index eb7c9af246b..1ad315bcd97 100644
--- a/homeassistant/components/smartthings/light.py
+++ b/homeassistant/components/smartthings/light.py
@@ -3,13 +3,13 @@
from __future__ import annotations
import asyncio
-from collections.abc import Sequence
-from typing import Any
+from typing import Any, cast
-from pysmartthings import Capability
+from pysmartthings import Attribute, Capability, Command, DeviceEvent, SmartThings
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
+ ATTR_COLOR_MODE,
ATTR_COLOR_TEMP_KELVIN,
ATTR_HS_COLOR,
ATTR_TRANSITION,
@@ -18,104 +18,95 @@ from homeassistant.components.light import (
LightEntityFeature,
brightness_supported,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.restore_state import RestoreEntity
-from .const import DATA_BROKERS, DOMAIN
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
from .entity import SmartThingsEntity
+CAPABILITIES = (
+ Capability.SWITCH_LEVEL,
+ Capability.COLOR_CONTROL,
+ Capability.COLOR_TEMPERATURE,
+)
+
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add lights for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
+ entry_data = entry.runtime_data
async_add_entities(
- [
- SmartThingsLight(device)
- for device in broker.devices.values()
- if broker.any_assigned(device.device_id, "light")
- ],
- True,
+ SmartThingsLight(entry_data.client, device)
+ for device in entry_data.devices.values()
+ if Capability.SWITCH in device.status[MAIN]
+ and any(capability in device.status[MAIN] for capability in CAPABILITIES)
)
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
- supported = [
- Capability.switch,
- Capability.switch_level,
- Capability.color_control,
- Capability.color_temperature,
- ]
- # Must be able to be turned on/off.
- if Capability.switch not in capabilities:
- return None
- # Must have one of these
- light_capabilities = [
- Capability.color_control,
- Capability.color_temperature,
- Capability.switch_level,
- ]
- if any(capability in capabilities for capability in light_capabilities):
- return supported
- return None
-
-
-def convert_scale(value, value_scale, target_scale, round_digits=4):
+def convert_scale(
+ value: float, value_scale: int, target_scale: int, round_digits: int = 4
+) -> float:
"""Convert a value to a different scale."""
return round(value * target_scale / value_scale, round_digits)
-class SmartThingsLight(SmartThingsEntity, LightEntity):
+class SmartThingsLight(SmartThingsEntity, LightEntity, RestoreEntity):
"""Define a SmartThings Light."""
+ _attr_name = None
_attr_supported_color_modes: set[ColorMode]
# SmartThings does not expose this attribute, instead it's
- # implemented within each device-type handler. This value is the
+ # implemented within each device-type handler. This value is the
# lowest kelvin found supported across 20+ handlers.
_attr_min_color_temp_kelvin = 2000 # 500 mireds
# SmartThings does not expose this attribute, instead it's
- # implemented within each device-type handler. This value is the
+ # implemented within each device-type handler. This value is the
# highest kelvin found supported across 20+ handlers.
_attr_max_color_temp_kelvin = 9000 # 111 mireds
- def __init__(self, device):
+ def __init__(self, client: SmartThings, device: FullDevice) -> None:
"""Initialize a SmartThingsLight."""
- super().__init__(device)
- self._attr_supported_color_modes = self._determine_color_modes()
- self._attr_supported_features = self._determine_features()
-
- def _determine_color_modes(self):
- """Get features supported by the device."""
+ super().__init__(
+ client,
+ device,
+ {
+ Capability.COLOR_CONTROL,
+ Capability.COLOR_TEMPERATURE,
+ Capability.SWITCH_LEVEL,
+ Capability.SWITCH,
+ },
+ )
color_modes = set()
- # Color Temperature
- if Capability.color_temperature in self._device.capabilities:
+ if self.supports_capability(Capability.COLOR_TEMPERATURE):
color_modes.add(ColorMode.COLOR_TEMP)
- # Color
- if Capability.color_control in self._device.capabilities:
+ self._attr_color_mode = ColorMode.COLOR_TEMP
+ if self.supports_capability(Capability.COLOR_CONTROL):
color_modes.add(ColorMode.HS)
- # Brightness
- if not color_modes and Capability.switch_level in self._device.capabilities:
+ self._attr_color_mode = ColorMode.HS
+ if not color_modes and self.supports_capability(Capability.SWITCH_LEVEL):
color_modes.add(ColorMode.BRIGHTNESS)
if not color_modes:
color_modes.add(ColorMode.ONOFF)
-
- return color_modes
-
- def _determine_features(self) -> LightEntityFeature:
- """Get features supported by the device."""
+ if len(color_modes) == 1:
+ self._attr_color_mode = list(color_modes)[0]
+ self._attr_supported_color_modes = color_modes
features = LightEntityFeature(0)
- # Transition
- if Capability.switch_level in self._device.capabilities:
+ if self.supports_capability(Capability.SWITCH_LEVEL):
features |= LightEntityFeature.TRANSITION
+ self._attr_supported_features = features
- return features
+ async def async_added_to_hass(self) -> None:
+ """Run when entity about to be added to hass."""
+ await super().async_added_to_hass()
+ if (last_state := await self.async_get_last_extra_data()) is not None:
+ self._attr_color_mode = last_state.as_dict()[ATTR_COLOR_MODE]
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn the light on."""
@@ -136,11 +127,10 @@ class SmartThingsLight(SmartThingsEntity, LightEntity):
kwargs[ATTR_BRIGHTNESS], kwargs.get(ATTR_TRANSITION, 0)
)
else:
- await self._device.switch_on(set_status=True)
-
- # State is set optimistically in the commands above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_schedule_update_ha_state(True)
+ await self.execute_device_command(
+ Capability.SWITCH,
+ Command.ON,
+ )
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the light off."""
@@ -148,42 +138,74 @@ class SmartThingsLight(SmartThingsEntity, LightEntity):
if ATTR_TRANSITION in kwargs:
await self.async_set_level(0, int(kwargs[ATTR_TRANSITION]))
else:
- await self._device.switch_off(set_status=True)
+ await self.execute_device_command(
+ Capability.SWITCH,
+ Command.OFF,
+ )
- # State is set optimistically in the commands above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_schedule_update_ha_state(True)
-
- async def async_update(self) -> None:
+ def _update_attr(self) -> None:
"""Update entity attributes when the device status has changed."""
# Brightness and transition
if brightness_supported(self._attr_supported_color_modes):
- self._attr_brightness = int(
- convert_scale(self._device.status.level, 100, 255, 0)
- )
+ if (
+ brightness := self.get_attribute_value(
+ Capability.SWITCH_LEVEL, Attribute.LEVEL
+ )
+ ) is None:
+ self._attr_brightness = None
+ else:
+ self._attr_brightness = int(
+ convert_scale(
+ brightness,
+ 100,
+ 255,
+ 0,
+ )
+ )
# Color Temperature
if ColorMode.COLOR_TEMP in self._attr_supported_color_modes:
- self._attr_color_temp_kelvin = self._device.status.color_temperature
+ self._attr_color_temp_kelvin = self.get_attribute_value(
+ Capability.COLOR_TEMPERATURE, Attribute.COLOR_TEMPERATURE
+ )
# Color
if ColorMode.HS in self._attr_supported_color_modes:
- self._attr_hs_color = (
- convert_scale(self._device.status.hue, 100, 360),
- self._device.status.saturation,
- )
+ if (
+ hue := self.get_attribute_value(Capability.COLOR_CONTROL, Attribute.HUE)
+ ) is None:
+ self._attr_hs_color = None
+ else:
+ self._attr_hs_color = (
+ convert_scale(
+ hue,
+ 100,
+ 360,
+ ),
+ self.get_attribute_value(
+ Capability.COLOR_CONTROL, Attribute.SATURATION
+ ),
+ )
async def async_set_color(self, hs_color):
"""Set the color of the device."""
hue = convert_scale(float(hs_color[0]), 360, 100)
hue = max(min(hue, 100.0), 0.0)
saturation = max(min(float(hs_color[1]), 100.0), 0.0)
- await self._device.set_color(hue, saturation, set_status=True)
+ await self.execute_device_command(
+ Capability.COLOR_CONTROL,
+ Command.SET_COLOR,
+ argument={"hue": hue, "saturation": saturation},
+ )
async def async_set_color_temp(self, value: int):
"""Set the color temperature of the device."""
kelvin = max(min(value, 30000), 1)
- await self._device.set_color_temperature(kelvin, set_status=True)
+ await self.execute_device_command(
+ Capability.COLOR_TEMPERATURE,
+ Command.SET_COLOR_TEMPERATURE,
+ argument=kelvin,
+ )
- async def async_set_level(self, brightness: int, transition: int):
+ async def async_set_level(self, brightness: int, transition: int) -> None:
"""Set the brightness of the light over transition."""
level = int(convert_scale(brightness, 255, 100, 0))
# Due to rounding, set level to 1 (one) so we don't inadvertently
@@ -191,21 +213,26 @@ class SmartThingsLight(SmartThingsEntity, LightEntity):
level = 1 if level == 0 and brightness > 0 else level
level = max(min(level, 100), 0)
duration = int(transition)
- await self._device.set_level(level, duration, set_status=True)
+ await self.execute_device_command(
+ Capability.SWITCH_LEVEL,
+ Command.SET_LEVEL,
+ argument=[level, duration],
+ )
+
+ def _update_handler(self, event: DeviceEvent) -> None:
+ """Handle device updates."""
+ if event.capability in (Capability.COLOR_CONTROL, Capability.COLOR_TEMPERATURE):
+ self._attr_color_mode = {
+ Capability.COLOR_CONTROL: ColorMode.HS,
+ Capability.COLOR_TEMPERATURE: ColorMode.COLOR_TEMP,
+ }[cast(Capability, event.capability)]
+ super()._update_handler(event)
@property
- def color_mode(self) -> ColorMode:
- """Return the color mode of the light."""
- if len(self._attr_supported_color_modes) == 1:
- # The light supports only a single color mode
- return list(self._attr_supported_color_modes)[0]
-
- # The light supports hs + color temp, determine which one it is
- if self._attr_hs_color and self._attr_hs_color[1]:
- return ColorMode.HS
- return ColorMode.COLOR_TEMP
-
- @property
- def is_on(self) -> bool:
+ def is_on(self) -> bool | None:
"""Return true if light is on."""
- return self._device.status.switch
+ if (
+ state := self.get_attribute_value(Capability.SWITCH, Attribute.SWITCH)
+ ) is None:
+ return None
+ return state == "on"
diff --git a/homeassistant/components/smartthings/lock.py b/homeassistant/components/smartthings/lock.py
index a0ae9e50443..f56ecd5d565 100644
--- a/homeassistant/components/smartthings/lock.py
+++ b/homeassistant/components/smartthings/lock.py
@@ -2,17 +2,16 @@
from __future__ import annotations
-from collections.abc import Sequence
from typing import Any
-from pysmartthings import Attribute, Capability
+from pysmartthings import Attribute, Capability, Command
from homeassistant.components.lock import LockEntity
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DATA_BROKERS, DOMAIN
+from . import SmartThingsConfigEntry
+from .const import MAIN
from .entity import SmartThingsEntity
ST_STATE_LOCKED = "locked"
@@ -28,48 +27,49 @@ ST_LOCK_ATTR_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add locks for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
+ entry_data = entry.runtime_data
async_add_entities(
- SmartThingsLock(device)
- for device in broker.devices.values()
- if broker.any_assigned(device.device_id, "lock")
+ SmartThingsLock(entry_data.client, device, {Capability.LOCK})
+ for device in entry_data.devices.values()
+ if Capability.LOCK in device.status[MAIN]
)
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
- if Capability.lock in capabilities:
- return [Capability.lock]
- return None
-
-
class SmartThingsLock(SmartThingsEntity, LockEntity):
"""Define a SmartThings lock."""
+ _attr_name = None
+
async def async_lock(self, **kwargs: Any) -> None:
"""Lock the device."""
- await self._device.lock(set_status=True)
- self.async_write_ha_state()
+ await self.execute_device_command(
+ Capability.LOCK,
+ Command.LOCK,
+ )
async def async_unlock(self, **kwargs: Any) -> None:
"""Unlock the device."""
- await self._device.unlock(set_status=True)
- self.async_write_ha_state()
+ await self.execute_device_command(
+ Capability.LOCK,
+ Command.UNLOCK,
+ )
@property
def is_locked(self) -> bool:
"""Return true if lock is locked."""
- return self._device.status.lock == ST_STATE_LOCKED
+ return (
+ self.get_attribute_value(Capability.LOCK, Attribute.LOCK) == ST_STATE_LOCKED
+ )
@property
def extra_state_attributes(self) -> dict[str, Any]:
"""Return device specific state attributes."""
state_attrs = {}
- status = self._device.status.attributes[Attribute.lock]
+ status = self._internal_state[Capability.LOCK][Attribute.LOCK]
if status.value:
state_attrs["lock_state"] = status.value
if isinstance(status.data, dict):
diff --git a/homeassistant/components/smartthings/manifest.json b/homeassistant/components/smartthings/manifest.json
index be313248eaf..d7133ce7c6d 100644
--- a/homeassistant/components/smartthings/manifest.json
+++ b/homeassistant/components/smartthings/manifest.json
@@ -1,10 +1,9 @@
{
"domain": "smartthings",
"name": "SmartThings",
- "after_dependencies": ["cloud"],
- "codeowners": [],
+ "codeowners": ["@joostlek"],
"config_flow": true,
- "dependencies": ["webhook"],
+ "dependencies": ["application_credentials"],
"dhcp": [
{
"hostname": "st*",
@@ -29,6 +28,7 @@
],
"documentation": "https://www.home-assistant.io/integrations/smartthings",
"iot_class": "cloud_push",
- "loggers": ["httpsig", "pysmartapp", "pysmartthings"],
- "requirements": ["pysmartapp==0.3.5", "pysmartthings==0.7.8"]
+ "loggers": ["pysmartthings"],
+ "quality_scale": "bronze",
+ "requirements": ["pysmartthings==2.7.4"]
}
diff --git a/homeassistant/components/smartthings/number.py b/homeassistant/components/smartthings/number.py
new file mode 100644
index 00000000000..cbd200e20b6
--- /dev/null
+++ b/homeassistant/components/smartthings/number.py
@@ -0,0 +1,77 @@
+"""Support for number entities through the SmartThings cloud API."""
+
+from __future__ import annotations
+
+from pysmartthings import Attribute, Capability, Command, SmartThings
+
+from homeassistant.components.number import NumberEntity
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
+from .entity import SmartThingsEntity
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add number entities for a config entry."""
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsWasherRinseCyclesNumberEntity(entry_data.client, device)
+ for device in entry_data.devices.values()
+ if Capability.CUSTOM_WASHER_RINSE_CYCLES in device.status[MAIN]
+ )
+
+
+class SmartThingsWasherRinseCyclesNumberEntity(SmartThingsEntity, NumberEntity):
+ """Define a SmartThings number."""
+
+ _attr_translation_key = "washer_rinse_cycles"
+ _attr_native_step = 1.0
+
+ def __init__(self, client: SmartThings, device: FullDevice) -> None:
+ """Initialize the instance."""
+ super().__init__(client, device, {Capability.CUSTOM_WASHER_RINSE_CYCLES})
+ self._attr_unique_id = (
+ f"{device.device.device_id}_{MAIN}_{Capability.CUSTOM_WASHER_RINSE_CYCLES}"
+ )
+
+ @property
+ def options(self) -> list[int]:
+ """Return the list of options."""
+ values = self.get_attribute_value(
+ Capability.CUSTOM_WASHER_RINSE_CYCLES,
+ Attribute.SUPPORTED_WASHER_RINSE_CYCLES,
+ )
+ return [int(value) for value in values] if values else []
+
+ @property
+ def native_value(self) -> float | None:
+ """Return the current value."""
+ return int(
+ self.get_attribute_value(
+ Capability.CUSTOM_WASHER_RINSE_CYCLES, Attribute.WASHER_RINSE_CYCLES
+ )
+ )
+
+ @property
+ def native_min_value(self) -> float:
+ """Return the minimum value."""
+ return min(self.options)
+
+ @property
+ def native_max_value(self) -> float:
+ """Return the maximum value."""
+ return max(self.options)
+
+ async def async_set_native_value(self, value: float) -> None:
+ """Set the value."""
+ await self.execute_device_command(
+ Capability.CUSTOM_WASHER_RINSE_CYCLES,
+ Command.SET_WASHER_RINSE_CYCLES,
+ str(int(value)),
+ )
diff --git a/homeassistant/components/smartthings/quality_scale.yaml b/homeassistant/components/smartthings/quality_scale.yaml
new file mode 100644
index 00000000000..be8a9039617
--- /dev/null
+++ b/homeassistant/components/smartthings/quality_scale.yaml
@@ -0,0 +1,80 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: |
+ This integration does not provide additional actions.
+ appropriate-polling:
+ status: exempt
+ comment: |
+ This integration works via push.
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: |
+ This integration does not provide additional actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup: done
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions: todo
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: No options to configure
+ docs-installation-parameters:
+ status: exempt
+ comment: No parameters needed during installation
+ entity-unavailable: todo
+ integration-owner: done
+ log-when-unavailable: todo
+ parallel-updates: todo
+ reauthentication-flow: done
+ test-coverage: todo
+ # Gold
+ devices: done
+ diagnostics: done
+ discovery-update-info:
+ status: exempt
+ comment: |
+ This integration connects via the cloud.
+ discovery: done
+ docs-data-update: todo
+ docs-examples: todo
+ docs-known-limitations: todo
+ docs-supported-devices: todo
+ docs-supported-functions: done
+ docs-troubleshooting: todo
+ docs-use-cases: todo
+ dynamic-devices: todo
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default:
+ status: exempt
+ comment: |
+ This integration does not have any entities that are disabled by default.
+ entity-translations: done
+ exception-translations: todo
+ icon-translations: todo
+ reconfiguration-flow: todo
+ repair-issues:
+ status: exempt
+ comment: |
+ This integration doesn't have any cases where raising an issue is needed.
+ stale-devices: done
+ # Platinum
+ async-dependency: done
+ inject-websession: done
+ strict-typing: todo
diff --git a/homeassistant/components/smartthings/scene.py b/homeassistant/components/smartthings/scene.py
index 9756cef9f04..2b387859f22 100644
--- a/homeassistant/components/smartthings/scene.py
+++ b/homeassistant/components/smartthings/scene.py
@@ -2,39 +2,42 @@
from typing import Any
-from homeassistant.components.scene import Scene
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from pysmartthings import Scene as STScene, SmartThings
-from .const import DATA_BROKERS, DOMAIN
+from homeassistant.components.scene import Scene
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import SmartThingsConfigEntry
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
- """Add switches for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
- async_add_entities(SmartThingsScene(scene) for scene in broker.scenes.values())
+ """Add lights for a config entry."""
+ client = entry.runtime_data.client
+ scenes = entry.runtime_data.scenes
+ async_add_entities(SmartThingsScene(scene, client) for scene in scenes.values())
class SmartThingsScene(Scene):
"""Define a SmartThings scene."""
- def __init__(self, scene):
+ def __init__(self, scene: STScene, client: SmartThings) -> None:
"""Init the scene class."""
+ self.client = client
self._scene = scene
self._attr_name = scene.name
self._attr_unique_id = scene.scene_id
async def async_activate(self, **kwargs: Any) -> None:
"""Activate scene."""
- await self._scene.execute()
+ await self.client.execute_scene(self._scene.scene_id)
@property
- def extra_state_attributes(self):
+ def extra_state_attributes(self) -> dict[str, Any]:
"""Get attributes about the state."""
return {
"icon": self._scene.icon,
diff --git a/homeassistant/components/smartthings/select.py b/homeassistant/components/smartthings/select.py
new file mode 100644
index 00000000000..6011b7947b7
--- /dev/null
+++ b/homeassistant/components/smartthings/select.py
@@ -0,0 +1,120 @@
+"""Support for select entities through the SmartThings cloud API."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from pysmartthings import Attribute, Capability, Command, SmartThings
+
+from homeassistant.components.select import SelectEntity, SelectEntityDescription
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ServiceValidationError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
+from .entity import SmartThingsEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SmartThingsSelectDescription(SelectEntityDescription):
+ """Class describing SmartThings select entities."""
+
+ key: Capability
+ requires_remote_control_status: bool
+ options_attribute: Attribute
+ status_attribute: Attribute
+ command: Command
+
+
+CAPABILITIES_TO_SELECT: dict[Capability | str, SmartThingsSelectDescription] = {
+ Capability.DRYER_OPERATING_STATE: SmartThingsSelectDescription(
+ key=Capability.DRYER_OPERATING_STATE,
+ name=None,
+ translation_key="operating_state",
+ requires_remote_control_status=True,
+ options_attribute=Attribute.SUPPORTED_MACHINE_STATES,
+ status_attribute=Attribute.MACHINE_STATE,
+ command=Command.SET_MACHINE_STATE,
+ ),
+ Capability.WASHER_OPERATING_STATE: SmartThingsSelectDescription(
+ key=Capability.WASHER_OPERATING_STATE,
+ name=None,
+ translation_key="operating_state",
+ requires_remote_control_status=True,
+ options_attribute=Attribute.SUPPORTED_MACHINE_STATES,
+ status_attribute=Attribute.MACHINE_STATE,
+ command=Command.SET_MACHINE_STATE,
+ ),
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add select entities for a config entry."""
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsSelectEntity(
+ entry_data.client, device, CAPABILITIES_TO_SELECT[capability]
+ )
+ for device in entry_data.devices.values()
+ for capability in device.status[MAIN]
+ if capability in CAPABILITIES_TO_SELECT
+ )
+
+
+class SmartThingsSelectEntity(SmartThingsEntity, SelectEntity):
+ """Define a SmartThings select."""
+
+ entity_description: SmartThingsSelectDescription
+
+ def __init__(
+ self,
+ client: SmartThings,
+ device: FullDevice,
+ entity_description: SmartThingsSelectDescription,
+ ) -> None:
+ """Initialize the instance."""
+ capabilities = {entity_description.key}
+ if entity_description.requires_remote_control_status:
+ capabilities.add(Capability.REMOTE_CONTROL_STATUS)
+ super().__init__(client, device, capabilities)
+ self.entity_description = entity_description
+ self._attr_unique_id = (
+ f"{device.device.device_id}_{MAIN}_{entity_description.key}"
+ )
+
+ @property
+ def options(self) -> list[str]:
+ """Return the list of options."""
+ return self.get_attribute_value(
+ self.entity_description.key, self.entity_description.options_attribute
+ )
+
+ @property
+ def current_option(self) -> str | None:
+ """Return the current option."""
+ return self.get_attribute_value(
+ self.entity_description.key, self.entity_description.status_attribute
+ )
+
+ async def async_select_option(self, option: str) -> None:
+ """Select an option."""
+ if (
+ self.entity_description.requires_remote_control_status
+ and self.get_attribute_value(
+ Capability.REMOTE_CONTROL_STATUS, Attribute.REMOTE_CONTROL_ENABLED
+ )
+ == "false"
+ ):
+ raise ServiceValidationError(
+ "Can only be updated when remote control is enabled"
+ )
+ await self.execute_device_command(
+ self.entity_description.key,
+ self.entity_description.command,
+ option,
+ )
diff --git a/homeassistant/components/smartthings/sensor.py b/homeassistant/components/smartthings/sensor.py
index 8bd0421d2bc..ee8550e4f06 100644
--- a/homeassistant/components/smartthings/sensor.py
+++ b/homeassistant/components/smartthings/sensor.py
@@ -2,25 +2,26 @@
from __future__ import annotations
-from collections.abc import Sequence
-from typing import NamedTuple
+from collections.abc import Callable, Mapping
+from dataclasses import dataclass
+from datetime import datetime
+from typing import Any, cast
-from pysmartthings import Attribute, Capability
-from pysmartthings.device import DeviceEntity
+from pysmartthings import Attribute, Capability, SmartThings, Status
from homeassistant.components.sensor import (
SensorDeviceClass,
SensorEntity,
+ SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
+ CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
CONCENTRATION_PARTS_PER_MILLION,
LIGHT_LUX,
PERCENTAGE,
EntityCategory,
UnitOfArea,
- UnitOfElectricPotential,
UnitOfEnergy,
UnitOfMass,
UnitOfPower,
@@ -28,711 +29,1058 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
-from .const import DATA_BROKERS, DOMAIN
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
from .entity import SmartThingsEntity
-
-class Map(NamedTuple):
- """Tuple for mapping Smartthings capabilities to Home Assistant sensors."""
-
- attribute: str
- name: str
- default_unit: str | None
- device_class: SensorDeviceClass | None
- state_class: SensorStateClass | None
- entity_category: EntityCategory | None
-
-
-CAPABILITY_TO_SENSORS: dict[str, list[Map]] = {
- Capability.activity_lighting_mode: [
- Map(
- Attribute.lighting_mode,
- "Activity Lighting Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.air_conditioner_mode: [
- Map(
- Attribute.air_conditioner_mode,
- "Air Conditioner Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.air_quality_sensor: [
- Map(
- Attribute.air_quality,
- "Air Quality",
- "CAQI",
- None,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.alarm: [Map(Attribute.alarm, "Alarm", None, None, None, None)],
- Capability.audio_volume: [
- Map(Attribute.volume, "Volume", PERCENTAGE, None, None, None)
- ],
- Capability.battery: [
- Map(
- Attribute.battery,
- "Battery",
- PERCENTAGE,
- SensorDeviceClass.BATTERY,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.body_mass_index_measurement: [
- Map(
- Attribute.bmi_measurement,
- "Body Mass Index",
- f"{UnitOfMass.KILOGRAMS}/{UnitOfArea.SQUARE_METERS}",
- None,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.body_weight_measurement: [
- Map(
- Attribute.body_weight_measurement,
- "Body Weight",
- UnitOfMass.KILOGRAMS,
- SensorDeviceClass.WEIGHT,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.carbon_dioxide_measurement: [
- Map(
- Attribute.carbon_dioxide,
- "Carbon Dioxide Measurement",
- CONCENTRATION_PARTS_PER_MILLION,
- SensorDeviceClass.CO2,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.carbon_monoxide_detector: [
- Map(
- Attribute.carbon_monoxide,
- "Carbon Monoxide Detector",
- None,
- None,
- None,
- None,
- )
- ],
- Capability.carbon_monoxide_measurement: [
- Map(
- Attribute.carbon_monoxide_level,
- "Carbon Monoxide Measurement",
- CONCENTRATION_PARTS_PER_MILLION,
- SensorDeviceClass.CO,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.dishwasher_operating_state: [
- Map(
- Attribute.machine_state, "Dishwasher Machine State", None, None, None, None
- ),
- Map(
- Attribute.dishwasher_job_state,
- "Dishwasher Job State",
- None,
- None,
- None,
- None,
- ),
- Map(
- Attribute.completion_time,
- "Dishwasher Completion Time",
- None,
- SensorDeviceClass.TIMESTAMP,
- None,
- None,
- ),
- ],
- Capability.dryer_mode: [
- Map(
- Attribute.dryer_mode,
- "Dryer Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.dryer_operating_state: [
- Map(Attribute.machine_state, "Dryer Machine State", None, None, None, None),
- Map(Attribute.dryer_job_state, "Dryer Job State", None, None, None, None),
- Map(
- Attribute.completion_time,
- "Dryer Completion Time",
- None,
- SensorDeviceClass.TIMESTAMP,
- None,
- None,
- ),
- ],
- Capability.dust_sensor: [
- Map(
- Attribute.fine_dust_level,
- "Fine Dust Level",
- None,
- None,
- SensorStateClass.MEASUREMENT,
- None,
- ),
- Map(
- Attribute.dust_level,
- "Dust Level",
- None,
- None,
- SensorStateClass.MEASUREMENT,
- None,
- ),
- ],
- Capability.energy_meter: [
- Map(
- Attribute.energy,
- "Energy Meter",
- UnitOfEnergy.KILO_WATT_HOUR,
- SensorDeviceClass.ENERGY,
- SensorStateClass.TOTAL_INCREASING,
- None,
- )
- ],
- Capability.equivalent_carbon_dioxide_measurement: [
- Map(
- Attribute.equivalent_carbon_dioxide_measurement,
- "Equivalent Carbon Dioxide Measurement",
- CONCENTRATION_PARTS_PER_MILLION,
- SensorDeviceClass.CO2,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.formaldehyde_measurement: [
- Map(
- Attribute.formaldehyde_level,
- "Formaldehyde Measurement",
- CONCENTRATION_PARTS_PER_MILLION,
- None,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.gas_meter: [
- Map(
- Attribute.gas_meter,
- "Gas Meter",
- UnitOfEnergy.KILO_WATT_HOUR,
- SensorDeviceClass.ENERGY,
- SensorStateClass.MEASUREMENT,
- None,
- ),
- Map(
- Attribute.gas_meter_calorific, "Gas Meter Calorific", None, None, None, None
- ),
- Map(
- Attribute.gas_meter_time,
- "Gas Meter Time",
- None,
- SensorDeviceClass.TIMESTAMP,
- None,
- None,
- ),
- Map(
- Attribute.gas_meter_volume,
- "Gas Meter Volume",
- UnitOfVolume.CUBIC_METERS,
- SensorDeviceClass.GAS,
- SensorStateClass.MEASUREMENT,
- None,
- ),
- ],
- Capability.illuminance_measurement: [
- Map(
- Attribute.illuminance,
- "Illuminance",
- LIGHT_LUX,
- SensorDeviceClass.ILLUMINANCE,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.infrared_level: [
- Map(
- Attribute.infrared_level,
- "Infrared Level",
- PERCENTAGE,
- None,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.media_input_source: [
- Map(Attribute.input_source, "Media Input Source", None, None, None, None)
- ],
- Capability.media_playback_repeat: [
- Map(
- Attribute.playback_repeat_mode,
- "Media Playback Repeat",
- None,
- None,
- None,
- None,
- )
- ],
- Capability.media_playback_shuffle: [
- Map(
- Attribute.playback_shuffle, "Media Playback Shuffle", None, None, None, None
- )
- ],
- Capability.media_playback: [
- Map(Attribute.playback_status, "Media Playback Status", None, None, None, None)
- ],
- Capability.odor_sensor: [
- Map(Attribute.odor_level, "Odor Sensor", None, None, None, None)
- ],
- Capability.oven_mode: [
- Map(
- Attribute.oven_mode,
- "Oven Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.oven_operating_state: [
- Map(Attribute.machine_state, "Oven Machine State", None, None, None, None),
- Map(Attribute.oven_job_state, "Oven Job State", None, None, None, None),
- Map(Attribute.completion_time, "Oven Completion Time", None, None, None, None),
- ],
- Capability.oven_setpoint: [
- Map(Attribute.oven_setpoint, "Oven Set Point", None, None, None, None)
- ],
- Capability.power_consumption_report: [],
- Capability.power_meter: [
- Map(
- Attribute.power,
- "Power Meter",
- UnitOfPower.WATT,
- SensorDeviceClass.POWER,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.power_source: [
- Map(
- Attribute.power_source,
- "Power Source",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.refrigeration_setpoint: [
- Map(
- Attribute.refrigeration_setpoint,
- "Refrigeration Setpoint",
- None,
- SensorDeviceClass.TEMPERATURE,
- None,
- None,
- )
- ],
- Capability.relative_humidity_measurement: [
- Map(
- Attribute.humidity,
- "Relative Humidity Measurement",
- PERCENTAGE,
- SensorDeviceClass.HUMIDITY,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.robot_cleaner_cleaning_mode: [
- Map(
- Attribute.robot_cleaner_cleaning_mode,
- "Robot Cleaner Cleaning Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.robot_cleaner_movement: [
- Map(
- Attribute.robot_cleaner_movement,
- "Robot Cleaner Movement",
- None,
- None,
- None,
- None,
- )
- ],
- Capability.robot_cleaner_turbo_mode: [
- Map(
- Attribute.robot_cleaner_turbo_mode,
- "Robot Cleaner Turbo Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.signal_strength: [
- Map(
- Attribute.lqi,
- "LQI Signal Strength",
- None,
- None,
- SensorStateClass.MEASUREMENT,
- EntityCategory.DIAGNOSTIC,
- ),
- Map(
- Attribute.rssi,
- "RSSI Signal Strength",
- None,
- SensorDeviceClass.SIGNAL_STRENGTH,
- SensorStateClass.MEASUREMENT,
- EntityCategory.DIAGNOSTIC,
- ),
- ],
- Capability.smoke_detector: [
- Map(Attribute.smoke, "Smoke Detector", None, None, None, None)
- ],
- Capability.temperature_measurement: [
- Map(
- Attribute.temperature,
- "Temperature Measurement",
- None,
- SensorDeviceClass.TEMPERATURE,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.thermostat_cooling_setpoint: [
- Map(
- Attribute.cooling_setpoint,
- "Thermostat Cooling Setpoint",
- None,
- SensorDeviceClass.TEMPERATURE,
- None,
- None,
- )
- ],
- Capability.thermostat_fan_mode: [
- Map(
- Attribute.thermostat_fan_mode,
- "Thermostat Fan Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.thermostat_heating_setpoint: [
- Map(
- Attribute.heating_setpoint,
- "Thermostat Heating Setpoint",
- None,
- SensorDeviceClass.TEMPERATURE,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.thermostat_mode: [
- Map(
- Attribute.thermostat_mode,
- "Thermostat Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.thermostat_operating_state: [
- Map(
- Attribute.thermostat_operating_state,
- "Thermostat Operating State",
- None,
- None,
- None,
- None,
- )
- ],
- Capability.thermostat_setpoint: [
- Map(
- Attribute.thermostat_setpoint,
- "Thermostat Setpoint",
- None,
- SensorDeviceClass.TEMPERATURE,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.three_axis: [],
- Capability.tv_channel: [
- Map(Attribute.tv_channel, "Tv Channel", None, None, None, None),
- Map(Attribute.tv_channel_name, "Tv Channel Name", None, None, None, None),
- ],
- Capability.tvoc_measurement: [
- Map(
- Attribute.tvoc_level,
- "Tvoc Measurement",
- CONCENTRATION_PARTS_PER_MILLION,
- None,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.ultraviolet_index: [
- Map(
- Attribute.ultraviolet_index,
- "Ultraviolet Index",
- None,
- None,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.voltage_measurement: [
- Map(
- Attribute.voltage,
- "Voltage Measurement",
- UnitOfElectricPotential.VOLT,
- SensorDeviceClass.VOLTAGE,
- SensorStateClass.MEASUREMENT,
- None,
- )
- ],
- Capability.washer_mode: [
- Map(
- Attribute.washer_mode,
- "Washer Mode",
- None,
- None,
- None,
- EntityCategory.DIAGNOSTIC,
- )
- ],
- Capability.washer_operating_state: [
- Map(Attribute.machine_state, "Washer Machine State", None, None, None, None),
- Map(Attribute.washer_job_state, "Washer Job State", None, None, None, None),
- Map(
- Attribute.completion_time,
- "Washer Completion Time",
- None,
- SensorDeviceClass.TIMESTAMP,
- None,
- None,
- ),
- ],
+THERMOSTAT_CAPABILITIES = {
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.THERMOSTAT_HEATING_SETPOINT,
+ Capability.THERMOSTAT_MODE,
}
+JOB_STATE_MAP = {
+ "airWash": "air_wash",
+ "airwash": "air_wash",
+ "aIRinse": "ai_rinse",
+ "aISpin": "ai_spin",
+ "aIWash": "ai_wash",
+ "aIDrying": "ai_drying",
+ "internalCare": "internal_care",
+ "continuousDehumidifying": "continuous_dehumidifying",
+ "thawingFrozenInside": "thawing_frozen_inside",
+ "delayWash": "delay_wash",
+ "weightSensing": "weight_sensing",
+ "freezeProtection": "freeze_protection",
+ "preDrain": "pre_drain",
+ "preWash": "pre_wash",
+ "prewash": "pre_wash",
+ "wrinklePrevent": "wrinkle_prevent",
+ "unknown": None,
+}
+
+OVEN_JOB_STATE_MAP = {
+ "scheduledStart": "scheduled_start",
+ "fastPreheat": "fast_preheat",
+ "scheduledEnd": "scheduled_end",
+ "stone_heating": "stone_heating",
+ "timeHoldPreheat": "time_hold_preheat",
+}
+
+MEDIA_PLAYBACK_STATE_MAP = {
+ "fast forwarding": "fast_forwarding",
+}
+
+ROBOT_CLEANER_TURBO_MODE_STATE_MAP = {
+ "extraSilence": "extra_silence",
+}
+
+ROBOT_CLEANER_MOVEMENT_MAP = {
+ "powerOff": "off",
+}
+
+OVEN_MODE = {
+ "Conventional": "conventional",
+ "Bake": "bake",
+ "BottomHeat": "bottom_heat",
+ "ConvectionBake": "convection_bake",
+ "ConvectionRoast": "convection_roast",
+ "Broil": "broil",
+ "ConvectionBroil": "convection_broil",
+ "SteamCook": "steam_cook",
+ "SteamBake": "steam_bake",
+ "SteamRoast": "steam_roast",
+ "SteamBottomHeatplusConvection": "steam_bottom_heat_plus_convection",
+ "Microwave": "microwave",
+ "MWplusGrill": "microwave_plus_grill",
+ "MWplusConvection": "microwave_plus_convection",
+ "MWplusHotBlast": "microwave_plus_hot_blast",
+ "MWplusHotBlast2": "microwave_plus_hot_blast_2",
+ "SlimMiddle": "slim_middle",
+ "SlimStrong": "slim_strong",
+ "SlowCook": "slow_cook",
+ "Proof": "proof",
+ "Dehydrate": "dehydrate",
+ "Others": "others",
+ "StrongSteam": "strong_steam",
+ "Descale": "descale",
+ "Rinse": "rinse",
+}
+
+WASHER_OPTIONS = ["pause", "run", "stop"]
+
+
+def power_attributes(status: dict[str, Any]) -> dict[str, Any]:
+ """Return the power attributes."""
+ state = {}
+ for attribute in ("start", "end"):
+ if (value := status.get(attribute)) is not None:
+ state[f"power_consumption_{attribute}"] = value
+ return state
+
+
+@dataclass(frozen=True, kw_only=True)
+class SmartThingsSensorEntityDescription(SensorEntityDescription):
+ """Describe a SmartThings sensor entity."""
+
+ value_fn: Callable[[Any], str | float | int | datetime | None] = lambda value: value
+ extra_state_attributes_fn: Callable[[Any], dict[str, Any]] | None = None
+ unique_id_separator: str = "."
+ capability_ignore_list: list[set[Capability]] | None = None
+ options_attribute: Attribute | None = None
+ exists_fn: Callable[[Status], bool] | None = None
+ use_temperature_unit: bool = False
+
+
+CAPABILITY_TO_SENSORS: dict[
+ Capability, dict[Attribute, list[SmartThingsSensorEntityDescription]]
+] = {
+ # Haven't seen at devices yet
+ Capability.ACTIVITY_LIGHTING_MODE: {
+ Attribute.LIGHTING_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.LIGHTING_MODE,
+ translation_key="lighting_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ]
+ },
+ Capability.AIR_CONDITIONER_MODE: {
+ Attribute.AIR_CONDITIONER_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.AIR_CONDITIONER_MODE,
+ translation_key="air_conditioner_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ capability_ignore_list=[
+ {
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.THERMOSTAT_COOLING_SETPOINT,
+ }
+ ],
+ )
+ ]
+ },
+ Capability.AIR_QUALITY_SENSOR: {
+ Attribute.AIR_QUALITY: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.AIR_QUALITY,
+ translation_key="air_quality",
+ native_unit_of_measurement="CAQI",
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.ALARM: {
+ Attribute.ALARM: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ALARM,
+ translation_key="alarm",
+ options=["both", "strobe", "siren", "off"],
+ device_class=SensorDeviceClass.ENUM,
+ )
+ ]
+ },
+ Capability.AUDIO_VOLUME: {
+ Attribute.VOLUME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.VOLUME,
+ translation_key="audio_volume",
+ native_unit_of_measurement=PERCENTAGE,
+ )
+ ]
+ },
+ Capability.BATTERY: {
+ Attribute.BATTERY: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.BATTERY,
+ native_unit_of_measurement=PERCENTAGE,
+ device_class=SensorDeviceClass.BATTERY,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.BODY_MASS_INDEX_MEASUREMENT: {
+ Attribute.BMI_MEASUREMENT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.BMI_MEASUREMENT,
+ translation_key="body_mass_index",
+ native_unit_of_measurement=f"{UnitOfMass.KILOGRAMS}/{UnitOfArea.SQUARE_METERS}",
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.BODY_WEIGHT_MEASUREMENT: {
+ Attribute.BODY_WEIGHT_MEASUREMENT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.BODY_WEIGHT_MEASUREMENT,
+ translation_key="body_weight",
+ native_unit_of_measurement=UnitOfMass.KILOGRAMS,
+ device_class=SensorDeviceClass.WEIGHT,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.CARBON_DIOXIDE_MEASUREMENT: {
+ Attribute.CARBON_DIOXIDE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.CARBON_DIOXIDE,
+ native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
+ device_class=SensorDeviceClass.CO2,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.CARBON_MONOXIDE_DETECTOR: {
+ Attribute.CARBON_MONOXIDE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.CARBON_MONOXIDE,
+ translation_key="carbon_monoxide_detector",
+ options=["detected", "clear", "tested"],
+ device_class=SensorDeviceClass.ENUM,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.CARBON_MONOXIDE_MEASUREMENT: {
+ Attribute.CARBON_MONOXIDE_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.CARBON_MONOXIDE_LEVEL,
+ native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
+ device_class=SensorDeviceClass.CO,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.DISHWASHER_OPERATING_STATE: {
+ Attribute.MACHINE_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.MACHINE_STATE,
+ translation_key="dishwasher_machine_state",
+ options=WASHER_OPTIONS,
+ device_class=SensorDeviceClass.ENUM,
+ )
+ ],
+ Attribute.DISHWASHER_JOB_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.DISHWASHER_JOB_STATE,
+ translation_key="dishwasher_job_state",
+ options=[
+ "air_wash",
+ "cooling",
+ "drying",
+ "finish",
+ "pre_drain",
+ "pre_wash",
+ "rinse",
+ "spin",
+ "wash",
+ "wrinkle_prevent",
+ ],
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: JOB_STATE_MAP.get(value, value),
+ )
+ ],
+ Attribute.COMPLETION_TIME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.COMPLETION_TIME,
+ translation_key="completion_time",
+ device_class=SensorDeviceClass.TIMESTAMP,
+ value_fn=dt_util.parse_datetime,
+ )
+ ],
+ },
+ # part of the proposed spec, Haven't seen at devices yet
+ Capability.DRYER_MODE: {
+ Attribute.DRYER_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.DRYER_MODE,
+ translation_key="dryer_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ]
+ },
+ Capability.DRYER_OPERATING_STATE: {
+ Attribute.MACHINE_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.MACHINE_STATE,
+ translation_key="dryer_machine_state",
+ options=WASHER_OPTIONS,
+ device_class=SensorDeviceClass.ENUM,
+ )
+ ],
+ Attribute.DRYER_JOB_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.DRYER_JOB_STATE,
+ translation_key="dryer_job_state",
+ options=[
+ "cooling",
+ "delay_wash",
+ "drying",
+ "finished",
+ "none",
+ "refreshing",
+ "weight_sensing",
+ "wrinkle_prevent",
+ "dehumidifying",
+ "ai_drying",
+ "sanitizing",
+ "internal_care",
+ "freeze_protection",
+ "continuous_dehumidifying",
+ "thawing_frozen_inside",
+ ],
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: JOB_STATE_MAP.get(value, value),
+ )
+ ],
+ Attribute.COMPLETION_TIME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.COMPLETION_TIME,
+ translation_key="completion_time",
+ device_class=SensorDeviceClass.TIMESTAMP,
+ value_fn=dt_util.parse_datetime,
+ )
+ ],
+ },
+ Capability.DUST_SENSOR: {
+ Attribute.DUST_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.DUST_LEVEL,
+ device_class=SensorDeviceClass.PM10,
+ native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ],
+ Attribute.FINE_DUST_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.FINE_DUST_LEVEL,
+ device_class=SensorDeviceClass.PM25,
+ native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ],
+ },
+ Capability.ENERGY_METER: {
+ Attribute.ENERGY: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ENERGY,
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.EQUIVALENT_CARBON_DIOXIDE_MEASUREMENT: {
+ Attribute.EQUIVALENT_CARBON_DIOXIDE_MEASUREMENT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.EQUIVALENT_CARBON_DIOXIDE_MEASUREMENT,
+ translation_key="equivalent_carbon_dioxide",
+ native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
+ device_class=SensorDeviceClass.CO2,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.FORMALDEHYDE_MEASUREMENT: {
+ Attribute.FORMALDEHYDE_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.FORMALDEHYDE_LEVEL,
+ translation_key="formaldehyde",
+ native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.GAS_METER: {
+ Attribute.GAS_METER: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.GAS_METER,
+ translation_key="gas_meter",
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ],
+ Attribute.GAS_METER_CALORIFIC: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.GAS_METER_CALORIFIC,
+ translation_key="gas_meter_calorific",
+ )
+ ],
+ Attribute.GAS_METER_TIME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.GAS_METER_TIME,
+ translation_key="gas_meter_time",
+ device_class=SensorDeviceClass.TIMESTAMP,
+ value_fn=dt_util.parse_datetime,
+ )
+ ],
+ Attribute.GAS_METER_VOLUME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.GAS_METER_VOLUME,
+ native_unit_of_measurement=UnitOfVolume.CUBIC_METERS,
+ device_class=SensorDeviceClass.GAS,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ],
+ },
+ # Haven't seen at devices yet
+ Capability.ILLUMINANCE_MEASUREMENT: {
+ Attribute.ILLUMINANCE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ILLUMINANCE,
+ native_unit_of_measurement=LIGHT_LUX,
+ device_class=SensorDeviceClass.ILLUMINANCE,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.INFRARED_LEVEL: {
+ Attribute.INFRARED_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.INFRARED_LEVEL,
+ translation_key="infrared_level",
+ native_unit_of_measurement=PERCENTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.MEDIA_INPUT_SOURCE: {
+ Attribute.INPUT_SOURCE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.INPUT_SOURCE,
+ translation_key="media_input_source",
+ device_class=SensorDeviceClass.ENUM,
+ options_attribute=Attribute.SUPPORTED_INPUT_SOURCES,
+ value_fn=lambda value: value.lower() if value else None,
+ )
+ ]
+ },
+ Capability.MEDIA_PLAYBACK_REPEAT: {
+ Attribute.PLAYBACK_REPEAT_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.PLAYBACK_REPEAT_MODE,
+ translation_key="media_playback_repeat",
+ )
+ ]
+ },
+ Capability.MEDIA_PLAYBACK_SHUFFLE: {
+ Attribute.PLAYBACK_SHUFFLE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.PLAYBACK_SHUFFLE,
+ translation_key="media_playback_shuffle",
+ )
+ ]
+ },
+ Capability.MEDIA_PLAYBACK: {
+ Attribute.PLAYBACK_STATUS: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.PLAYBACK_STATUS,
+ translation_key="media_playback_status",
+ options=[
+ "paused",
+ "playing",
+ "stopped",
+ "fast_forwarding",
+ "rewinding",
+ "buffering",
+ ],
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: MEDIA_PLAYBACK_STATE_MAP.get(value, value),
+ )
+ ]
+ },
+ Capability.ODOR_SENSOR: {
+ Attribute.ODOR_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ODOR_LEVEL,
+ translation_key="odor_sensor",
+ )
+ ]
+ },
+ Capability.OVEN_MODE: {
+ Attribute.OVEN_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.OVEN_MODE,
+ translation_key="oven_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ options=list(OVEN_MODE.values()),
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: OVEN_MODE.get(value, value),
+ )
+ ]
+ },
+ Capability.OVEN_OPERATING_STATE: {
+ Attribute.MACHINE_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.MACHINE_STATE,
+ translation_key="oven_machine_state",
+ options=["ready", "running", "paused"],
+ device_class=SensorDeviceClass.ENUM,
+ )
+ ],
+ Attribute.OVEN_JOB_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.OVEN_JOB_STATE,
+ translation_key="oven_job_state",
+ options=[
+ "cleaning",
+ "cooking",
+ "cooling",
+ "draining",
+ "preheat",
+ "ready",
+ "rinsing",
+ "finished",
+ "scheduled_start",
+ "warming",
+ "defrosting",
+ "sensing",
+ "searing",
+ "fast_preheat",
+ "scheduled_end",
+ "stone_heating",
+ "time_hold_preheat",
+ ],
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: OVEN_JOB_STATE_MAP.get(value, value),
+ )
+ ],
+ Attribute.COMPLETION_TIME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.COMPLETION_TIME,
+ translation_key="completion_time",
+ device_class=SensorDeviceClass.TIMESTAMP,
+ value_fn=dt_util.parse_datetime,
+ )
+ ],
+ },
+ Capability.OVEN_SETPOINT: {
+ Attribute.OVEN_SETPOINT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.OVEN_SETPOINT,
+ translation_key="oven_setpoint",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ use_temperature_unit=True,
+ # Set the value to None if it is 0 F (-17 C)
+ value_fn=lambda value: None if value in {0, -17} else value,
+ )
+ ]
+ },
+ Capability.POWER_CONSUMPTION_REPORT: {
+ Attribute.POWER_CONSUMPTION: [
+ SmartThingsSensorEntityDescription(
+ key="energy_meter",
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ device_class=SensorDeviceClass.ENERGY,
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ value_fn=lambda value: value["energy"] / 1000,
+ suggested_display_precision=2,
+ exists_fn=lambda status: (
+ (value := cast(dict | None, status.value)) is not None
+ and "energy" in value
+ ),
+ ),
+ SmartThingsSensorEntityDescription(
+ key="power_meter",
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.POWER,
+ native_unit_of_measurement=UnitOfPower.WATT,
+ value_fn=lambda value: value["power"],
+ extra_state_attributes_fn=power_attributes,
+ suggested_display_precision=2,
+ exists_fn=lambda status: (
+ (value := cast(dict | None, status.value)) is not None
+ and "power" in value
+ ),
+ ),
+ SmartThingsSensorEntityDescription(
+ key="deltaEnergy_meter",
+ translation_key="energy_difference",
+ state_class=SensorStateClass.TOTAL,
+ device_class=SensorDeviceClass.ENERGY,
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ value_fn=lambda value: value["deltaEnergy"] / 1000,
+ suggested_display_precision=2,
+ exists_fn=lambda status: (
+ (value := cast(dict | None, status.value)) is not None
+ and "deltaEnergy" in value
+ ),
+ ),
+ SmartThingsSensorEntityDescription(
+ key="powerEnergy_meter",
+ translation_key="power_energy",
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ device_class=SensorDeviceClass.ENERGY,
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ value_fn=lambda value: value["powerEnergy"] / 1000,
+ suggested_display_precision=2,
+ exists_fn=lambda status: (
+ (value := cast(dict | None, status.value)) is not None
+ and "powerEnergy" in value
+ ),
+ ),
+ SmartThingsSensorEntityDescription(
+ key="energySaved_meter",
+ translation_key="energy_saved",
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ device_class=SensorDeviceClass.ENERGY,
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ value_fn=lambda value: value["energySaved"] / 1000,
+ suggested_display_precision=2,
+ exists_fn=lambda status: (
+ (value := cast(dict | None, status.value)) is not None
+ and "energySaved" in value
+ ),
+ ),
+ ]
+ },
+ Capability.POWER_METER: {
+ Attribute.POWER: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.POWER,
+ native_unit_of_measurement=UnitOfPower.WATT,
+ device_class=SensorDeviceClass.POWER,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.POWER_SOURCE: {
+ Attribute.POWER_SOURCE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.POWER_SOURCE,
+ translation_key="power_source",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ]
+ },
+ # part of the proposed spec
+ Capability.REFRIGERATION_SETPOINT: {
+ Attribute.REFRIGERATION_SETPOINT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.REFRIGERATION_SETPOINT,
+ translation_key="refrigeration_setpoint",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ )
+ ]
+ },
+ Capability.RELATIVE_HUMIDITY_MEASUREMENT: {
+ Attribute.HUMIDITY: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.HUMIDITY,
+ native_unit_of_measurement=PERCENTAGE,
+ device_class=SensorDeviceClass.HUMIDITY,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.ROBOT_CLEANER_CLEANING_MODE: {
+ Attribute.ROBOT_CLEANER_CLEANING_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ROBOT_CLEANER_CLEANING_MODE,
+ translation_key="robot_cleaner_cleaning_mode",
+ options=["auto", "part", "repeat", "manual", "stop", "map"],
+ device_class=SensorDeviceClass.ENUM,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ],
+ },
+ Capability.ROBOT_CLEANER_MOVEMENT: {
+ Attribute.ROBOT_CLEANER_MOVEMENT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ROBOT_CLEANER_MOVEMENT,
+ translation_key="robot_cleaner_movement",
+ options=[
+ "homing",
+ "idle",
+ "charging",
+ "alarm",
+ "off",
+ "reserve",
+ "point",
+ "after",
+ "cleaning",
+ "pause",
+ ],
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: ROBOT_CLEANER_MOVEMENT_MAP.get(value, value),
+ )
+ ]
+ },
+ Capability.ROBOT_CLEANER_TURBO_MODE: {
+ Attribute.ROBOT_CLEANER_TURBO_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ROBOT_CLEANER_TURBO_MODE,
+ translation_key="robot_cleaner_turbo_mode",
+ options=["on", "off", "silence", "extra_silence"],
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: ROBOT_CLEANER_TURBO_MODE_STATE_MAP.get(
+ value, value
+ ),
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.SIGNAL_STRENGTH: {
+ Attribute.LQI: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.LQI,
+ translation_key="link_quality",
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ],
+ Attribute.RSSI: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.RSSI,
+ device_class=SensorDeviceClass.SIGNAL_STRENGTH,
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ],
+ },
+ # Haven't seen at devices yet
+ Capability.SMOKE_DETECTOR: {
+ Attribute.SMOKE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.SMOKE,
+ translation_key="smoke_detector",
+ options=["detected", "clear", "tested"],
+ device_class=SensorDeviceClass.ENUM,
+ )
+ ]
+ },
+ Capability.TEMPERATURE_MEASUREMENT: {
+ Attribute.TEMPERATURE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.TEMPERATURE,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.THERMOSTAT_COOLING_SETPOINT: {
+ Attribute.COOLING_SETPOINT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.COOLING_SETPOINT,
+ translation_key="thermostat_cooling_setpoint",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ capability_ignore_list=[
+ {
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.AIR_CONDITIONER_MODE,
+ },
+ THERMOSTAT_CAPABILITIES,
+ ],
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.THERMOSTAT_FAN_MODE: {
+ Attribute.THERMOSTAT_FAN_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.THERMOSTAT_FAN_MODE,
+ translation_key="thermostat_fan_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ capability_ignore_list=[THERMOSTAT_CAPABILITIES],
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.THERMOSTAT_HEATING_SETPOINT: {
+ Attribute.HEATING_SETPOINT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.HEATING_SETPOINT,
+ translation_key="thermostat_heating_setpoint",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ capability_ignore_list=[THERMOSTAT_CAPABILITIES],
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.THERMOSTAT_MODE: {
+ Attribute.THERMOSTAT_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.THERMOSTAT_MODE,
+ translation_key="thermostat_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ capability_ignore_list=[THERMOSTAT_CAPABILITIES],
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.THERMOSTAT_OPERATING_STATE: {
+ Attribute.THERMOSTAT_OPERATING_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.THERMOSTAT_OPERATING_STATE,
+ translation_key="thermostat_operating_state",
+ capability_ignore_list=[THERMOSTAT_CAPABILITIES],
+ )
+ ]
+ },
+ # deprecated capability
+ Capability.THERMOSTAT_SETPOINT: {
+ Attribute.THERMOSTAT_SETPOINT: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.THERMOSTAT_SETPOINT,
+ translation_key="thermostat_setpoint",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ]
+ },
+ Capability.THREE_AXIS: {
+ Attribute.THREE_AXIS: [
+ SmartThingsSensorEntityDescription(
+ key="X Coordinate",
+ translation_key="x_coordinate",
+ unique_id_separator=" ",
+ value_fn=lambda value: value[0],
+ ),
+ SmartThingsSensorEntityDescription(
+ key="Y Coordinate",
+ translation_key="y_coordinate",
+ unique_id_separator=" ",
+ value_fn=lambda value: value[1],
+ ),
+ SmartThingsSensorEntityDescription(
+ key="Z Coordinate",
+ translation_key="z_coordinate",
+ unique_id_separator=" ",
+ value_fn=lambda value: value[2],
+ ),
+ ]
+ },
+ Capability.TV_CHANNEL: {
+ Attribute.TV_CHANNEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.TV_CHANNEL,
+ translation_key="tv_channel",
+ )
+ ],
+ Attribute.TV_CHANNEL_NAME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.TV_CHANNEL_NAME,
+ translation_key="tv_channel_name",
+ )
+ ],
+ },
+ # Haven't seen at devices yet
+ Capability.TVOC_MEASUREMENT: {
+ Attribute.TVOC_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.TVOC_LEVEL,
+ device_class=SensorDeviceClass.VOLATILE_ORGANIC_COMPOUNDS_PARTS,
+ native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # Haven't seen at devices yet
+ Capability.ULTRAVIOLET_INDEX: {
+ Attribute.ULTRAVIOLET_INDEX: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.ULTRAVIOLET_INDEX,
+ translation_key="uv_index",
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.VERY_FINE_DUST_SENSOR: {
+ Attribute.VERY_FINE_DUST_LEVEL: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.VERY_FINE_DUST_LEVEL,
+ native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
+ device_class=SensorDeviceClass.PM1,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ Capability.VOLTAGE_MEASUREMENT: {
+ Attribute.VOLTAGE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.VOLTAGE,
+ device_class=SensorDeviceClass.VOLTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ )
+ ]
+ },
+ # part of the proposed spec
+ Capability.WASHER_MODE: {
+ Attribute.WASHER_MODE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.WASHER_MODE,
+ translation_key="washer_mode",
+ entity_category=EntityCategory.DIAGNOSTIC,
+ )
+ ]
+ },
+ Capability.WASHER_OPERATING_STATE: {
+ Attribute.MACHINE_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.MACHINE_STATE,
+ translation_key="washer_machine_state",
+ options=WASHER_OPTIONS,
+ device_class=SensorDeviceClass.ENUM,
+ )
+ ],
+ Attribute.WASHER_JOB_STATE: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.WASHER_JOB_STATE,
+ translation_key="washer_job_state",
+ options=[
+ "air_wash",
+ "ai_rinse",
+ "ai_spin",
+ "ai_wash",
+ "cooling",
+ "delay_wash",
+ "drying",
+ "finish",
+ "none",
+ "pre_wash",
+ "rinse",
+ "spin",
+ "wash",
+ "weight_sensing",
+ "wrinkle_prevent",
+ "freeze_protection",
+ ],
+ device_class=SensorDeviceClass.ENUM,
+ value_fn=lambda value: JOB_STATE_MAP.get(value, value),
+ )
+ ],
+ Attribute.COMPLETION_TIME: [
+ SmartThingsSensorEntityDescription(
+ key=Attribute.COMPLETION_TIME,
+ translation_key="completion_time",
+ device_class=SensorDeviceClass.TIMESTAMP,
+ value_fn=dt_util.parse_datetime,
+ )
+ ],
+ },
+}
+
+
UNITS = {
"C": UnitOfTemperature.CELSIUS,
"F": UnitOfTemperature.FAHRENHEIT,
"lux": LIGHT_LUX,
+ "mG": None,
+ "μg/m^3": CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
}
-THREE_AXIS_NAMES = ["X Coordinate", "Y Coordinate", "Z Coordinate"]
-POWER_CONSUMPTION_REPORT_NAMES = [
- "energy",
- "power",
- "deltaEnergy",
- "powerEnergy",
- "energySaved",
-]
-
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add sensors for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
- entities: list[SensorEntity] = []
- for device in broker.devices.values():
- for capability in broker.get_assigned(device.device_id, "sensor"):
- if capability == Capability.three_axis:
- entities.extend(
- [
- SmartThingsThreeAxisSensor(device, index)
- for index in range(len(THREE_AXIS_NAMES))
- ]
- )
- elif capability == Capability.power_consumption_report:
- entities.extend(
- [
- SmartThingsPowerConsumptionSensor(device, report_name)
- for report_name in POWER_CONSUMPTION_REPORT_NAMES
- ]
- )
- else:
- maps = CAPABILITY_TO_SENSORS[capability]
- entities.extend(
- [
- SmartThingsSensor(
- device,
- m.attribute,
- m.name,
- m.default_unit,
- m.device_class,
- m.state_class,
- m.entity_category,
- )
- for m in maps
- ]
- )
-
- if broker.any_assigned(device.device_id, "switch"):
- for capability in (Capability.energy_meter, Capability.power_meter):
- maps = CAPABILITY_TO_SENSORS[capability]
- entities.extend(
- [
- SmartThingsSensor(
- device,
- m.attribute,
- m.name,
- m.default_unit,
- m.device_class,
- m.state_class,
- m.entity_category,
- )
- for m in maps
- ]
- )
-
- async_add_entities(entities)
-
-
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
- return [
- capability for capability in CAPABILITY_TO_SENSORS if capability in capabilities
- ]
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsSensor(
+ entry_data.client,
+ device,
+ description,
+ capability,
+ attribute,
+ )
+ for device in entry_data.devices.values()
+ for capability, attributes in CAPABILITY_TO_SENSORS.items()
+ if capability in device.status[MAIN]
+ for attribute, descriptions in attributes.items()
+ for description in descriptions
+ if (
+ not description.capability_ignore_list
+ or not any(
+ all(capability in device.status[MAIN] for capability in capability_list)
+ for capability_list in description.capability_ignore_list
+ )
+ )
+ and (
+ not description.exists_fn
+ or description.exists_fn(device.status[MAIN][capability][attribute])
+ )
+ )
class SmartThingsSensor(SmartThingsEntity, SensorEntity):
"""Define a SmartThings Sensor."""
+ entity_description: SmartThingsSensorEntityDescription
+
def __init__(
self,
- device: DeviceEntity,
- attribute: str,
- name: str,
- default_unit: str | None,
- device_class: SensorDeviceClass | None,
- state_class: str | None,
- entity_category: EntityCategory | None,
+ client: SmartThings,
+ device: FullDevice,
+ entity_description: SmartThingsSensorEntityDescription,
+ capability: Capability,
+ attribute: Attribute,
) -> None:
"""Init the class."""
- super().__init__(device)
+ capabilities_to_subscribe = {capability}
+ if entity_description.use_temperature_unit:
+ capabilities_to_subscribe.add(Capability.TEMPERATURE_MEASUREMENT)
+ super().__init__(client, device, capabilities_to_subscribe)
+ self._attr_unique_id = f"{device.device.device_id}{entity_description.unique_id_separator}{entity_description.key}"
self._attribute = attribute
- self._attr_name = f"{device.label} {name}"
- self._attr_unique_id = f"{device.device_id}.{attribute}"
- self._attr_device_class = device_class
- self._default_unit = default_unit
- self._attr_state_class = state_class
- self._attr_entity_category = entity_category
+ self.capability = capability
+ self.entity_description = entity_description
@property
- def native_value(self):
+ def native_value(self) -> str | float | datetime | int | None:
"""Return the state of the sensor."""
- value = self._device.status.attributes[self._attribute].value
-
- if self.device_class != SensorDeviceClass.TIMESTAMP:
- return value
-
- return dt_util.parse_datetime(value)
+ res = self.get_attribute_value(self.capability, self._attribute)
+ return self.entity_description.value_fn(res)
@property
- def native_unit_of_measurement(self):
+ def native_unit_of_measurement(self) -> str | None:
"""Return the unit this state is expressed in."""
- unit = self._device.status.attributes[self._attribute].unit
- return UNITS.get(unit, unit) if unit else self._default_unit
-
-
-class SmartThingsThreeAxisSensor(SmartThingsEntity, SensorEntity):
- """Define a SmartThings Three Axis Sensor."""
-
- def __init__(self, device, index):
- """Init the class."""
- super().__init__(device)
- self._index = index
- self._attr_name = f"{device.label} {THREE_AXIS_NAMES[index]}"
- self._attr_unique_id = f"{device.device_id} {THREE_AXIS_NAMES[index]}"
-
- @property
- def native_value(self):
- """Return the state of the sensor."""
- three_axis = self._device.status.attributes[Attribute.three_axis].value
- try:
- return three_axis[self._index]
- except (TypeError, IndexError):
- return None
-
-
-class SmartThingsPowerConsumptionSensor(SmartThingsEntity, SensorEntity):
- """Define a SmartThings Sensor."""
-
- def __init__(
- self,
- device: DeviceEntity,
- report_name: str,
- ) -> None:
- """Init the class."""
- super().__init__(device)
- self.report_name = report_name
- self._attr_name = f"{device.label} {report_name}"
- self._attr_unique_id = f"{device.device_id}.{report_name}_meter"
- if self.report_name == "power":
- self._attr_state_class = SensorStateClass.MEASUREMENT
- self._attr_device_class = SensorDeviceClass.POWER
- self._attr_native_unit_of_measurement = UnitOfPower.WATT
+ if self.entity_description.use_temperature_unit:
+ unit = self._internal_state[Capability.TEMPERATURE_MEASUREMENT][
+ Attribute.TEMPERATURE
+ ].unit
else:
- self._attr_state_class = SensorStateClass.TOTAL_INCREASING
- self._attr_device_class = SensorDeviceClass.ENERGY
- self._attr_native_unit_of_measurement = UnitOfEnergy.KILO_WATT_HOUR
+ unit = self._internal_state[self.capability][self._attribute].unit
+ return (
+ UNITS.get(unit, unit)
+ if unit
+ else self.entity_description.native_unit_of_measurement
+ )
@property
- def native_value(self):
- """Return the state of the sensor."""
- value = self._device.status.attributes[Attribute.power_consumption].value
- if value is None or value.get(self.report_name) is None:
- return None
- if self.report_name == "power":
- return value[self.report_name]
- return value[self.report_name] / 1000
-
- @property
- def extra_state_attributes(self):
- """Return specific state attributes."""
- if self.report_name == "power":
- attributes = [
- "power_consumption_start",
- "power_consumption_end",
- ]
- state_attributes = {}
- for attribute in attributes:
- value = getattr(self._device.status, attribute)
- if value is not None:
- state_attributes[attribute] = value
- return state_attributes
+ def extra_state_attributes(self) -> Mapping[str, Any] | None:
+ """Return the state attributes."""
+ if self.entity_description.extra_state_attributes_fn:
+ return self.entity_description.extra_state_attributes_fn(
+ self.get_attribute_value(self.capability, self._attribute)
+ )
return None
+
+ @property
+ def options(self) -> list[str] | None:
+ """Return the options for this sensor."""
+ if self.entity_description.options_attribute:
+ if (
+ options := self.get_attribute_value(
+ self.capability, self.entity_description.options_attribute
+ )
+ ) is None:
+ return []
+ return [option.lower() for option in options]
+ return super().options
diff --git a/homeassistant/components/smartthings/smartapp.py b/homeassistant/components/smartthings/smartapp.py
deleted file mode 100644
index 76b6804075f..00000000000
--- a/homeassistant/components/smartthings/smartapp.py
+++ /dev/null
@@ -1,545 +0,0 @@
-"""SmartApp functionality to receive cloud-push notifications."""
-
-from __future__ import annotations
-
-import asyncio
-import functools
-import logging
-import secrets
-from typing import Any
-from urllib.parse import urlparse
-from uuid import uuid4
-
-from aiohttp import web
-from pysmartapp import Dispatcher, SmartAppManager
-from pysmartapp.const import SETTINGS_APP_ID
-from pysmartthings import (
- APP_TYPE_WEBHOOK,
- CAPABILITIES,
- CLASSIFICATION_AUTOMATION,
- App,
- AppEntity,
- AppOAuth,
- AppSettings,
- InstalledAppStatus,
- SmartThings,
- SourceType,
- Subscription,
- SubscriptionEntity,
-)
-
-from homeassistant.components import cloud, webhook
-from homeassistant.config_entries import ConfigFlowResult
-from homeassistant.const import CONF_WEBHOOK_ID
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.dispatcher import (
- async_dispatcher_connect,
- async_dispatcher_send,
-)
-from homeassistant.helpers.network import NoURLAvailableError, get_url
-from homeassistant.helpers.storage import Store
-
-from .const import (
- APP_NAME_PREFIX,
- APP_OAUTH_CLIENT_NAME,
- APP_OAUTH_SCOPES,
- CONF_CLOUDHOOK_URL,
- CONF_INSTALLED_APP_ID,
- CONF_INSTANCE_ID,
- CONF_REFRESH_TOKEN,
- DATA_BROKERS,
- DATA_MANAGER,
- DOMAIN,
- IGNORED_CAPABILITIES,
- SETTINGS_INSTANCE_ID,
- SIGNAL_SMARTAPP_PREFIX,
- STORAGE_KEY,
- STORAGE_VERSION,
- SUBSCRIPTION_WARNING_LIMIT,
-)
-
-_LOGGER = logging.getLogger(__name__)
-
-
-def format_unique_id(app_id: str, location_id: str) -> str:
- """Format the unique id for a config entry."""
- return f"{app_id}_{location_id}"
-
-
-async def find_app(hass: HomeAssistant, api: SmartThings) -> AppEntity | None:
- """Find an existing SmartApp for this installation of hass."""
- apps = await api.apps()
- for app in [app for app in apps if app.app_name.startswith(APP_NAME_PREFIX)]:
- # Load settings to compare instance id
- settings = await app.settings()
- if (
- settings.settings.get(SETTINGS_INSTANCE_ID)
- == hass.data[DOMAIN][CONF_INSTANCE_ID]
- ):
- return app
- return None
-
-
-async def validate_installed_app(api, installed_app_id: str):
- """Ensure the specified installed SmartApp is valid and functioning.
-
- Query the API for the installed SmartApp and validate that it is tied to
- the specified app_id and is in an authorized state.
- """
- installed_app = await api.installed_app(installed_app_id)
- if installed_app.installed_app_status != InstalledAppStatus.AUTHORIZED:
- raise RuntimeWarning(
- f"Installed SmartApp instance '{installed_app.display_name}' "
- f"({installed_app.installed_app_id}) is not AUTHORIZED "
- f"but instead {installed_app.installed_app_status}"
- )
- return installed_app
-
-
-def validate_webhook_requirements(hass: HomeAssistant) -> bool:
- """Ensure Home Assistant is setup properly to receive webhooks."""
- if cloud.async_active_subscription(hass):
- return True
- if hass.data[DOMAIN][CONF_CLOUDHOOK_URL] is not None:
- return True
- return get_webhook_url(hass).lower().startswith("https://")
-
-
-def get_webhook_url(hass: HomeAssistant) -> str:
- """Get the URL of the webhook.
-
- Return the cloudhook if available, otherwise local webhook.
- """
- cloudhook_url = hass.data[DOMAIN][CONF_CLOUDHOOK_URL]
- if cloud.async_active_subscription(hass) and cloudhook_url is not None:
- return cloudhook_url
- return webhook.async_generate_url(hass, hass.data[DOMAIN][CONF_WEBHOOK_ID])
-
-
-def _get_app_template(hass: HomeAssistant):
- try:
- endpoint = f"at {get_url(hass, allow_cloud=False, prefer_external=True)}"
- except NoURLAvailableError:
- endpoint = ""
-
- cloudhook_url = hass.data[DOMAIN][CONF_CLOUDHOOK_URL]
- if cloudhook_url is not None:
- endpoint = "via Nabu Casa"
- description = f"{hass.config.location_name} {endpoint}"
-
- return {
- "app_name": APP_NAME_PREFIX + str(uuid4()),
- "display_name": "Home Assistant",
- "description": description,
- "webhook_target_url": get_webhook_url(hass),
- "app_type": APP_TYPE_WEBHOOK,
- "single_instance": True,
- "classifications": [CLASSIFICATION_AUTOMATION],
- }
-
-
-async def create_app(hass: HomeAssistant, api):
- """Create a SmartApp for this instance of hass."""
- # Create app from template attributes
- template = _get_app_template(hass)
- app = App()
- for key, value in template.items():
- setattr(app, key, value)
- app, client = await api.create_app(app)
- _LOGGER.debug("Created SmartApp '%s' (%s)", app.app_name, app.app_id)
-
- # Set unique hass id in settings
- settings = AppSettings(app.app_id)
- settings.settings[SETTINGS_APP_ID] = app.app_id
- settings.settings[SETTINGS_INSTANCE_ID] = hass.data[DOMAIN][CONF_INSTANCE_ID]
- await api.update_app_settings(settings)
- _LOGGER.debug(
- "Updated App Settings for SmartApp '%s' (%s)", app.app_name, app.app_id
- )
-
- # Set oauth scopes
- oauth = AppOAuth(app.app_id)
- oauth.client_name = APP_OAUTH_CLIENT_NAME
- oauth.scope.extend(APP_OAUTH_SCOPES)
- await api.update_app_oauth(oauth)
- _LOGGER.debug("Updated App OAuth for SmartApp '%s' (%s)", app.app_name, app.app_id)
- return app, client
-
-
-async def update_app(hass: HomeAssistant, app):
- """Ensure the SmartApp is up-to-date and update if necessary."""
- template = _get_app_template(hass)
- template.pop("app_name") # don't update this
- update_required = False
- for key, value in template.items():
- if getattr(app, key) != value:
- update_required = True
- setattr(app, key, value)
- if update_required:
- await app.save()
- _LOGGER.debug(
- "SmartApp '%s' (%s) updated with latest settings", app.app_name, app.app_id
- )
-
-
-def setup_smartapp(hass, app):
- """Configure an individual SmartApp in hass.
-
- Register the SmartApp with the SmartAppManager so that hass will service
- lifecycle events (install, event, etc...). A unique SmartApp is created
- for each SmartThings account that is configured in hass.
- """
- manager = hass.data[DOMAIN][DATA_MANAGER]
- if smartapp := manager.smartapps.get(app.app_id):
- # already setup
- return smartapp
- smartapp = manager.register(app.app_id, app.webhook_public_key)
- smartapp.name = app.display_name
- smartapp.description = app.description
- smartapp.permissions.extend(APP_OAUTH_SCOPES)
- return smartapp
-
-
-async def setup_smartapp_endpoint(hass: HomeAssistant, fresh_install: bool):
- """Configure the SmartApp webhook in hass.
-
- SmartApps are an extension point within the SmartThings ecosystem and
- is used to receive push updates (i.e. device updates) from the cloud.
- """
- if hass.data.get(DOMAIN):
- # already setup
- if not fresh_install:
- return
-
- # We're doing a fresh install, clean up
- await unload_smartapp_endpoint(hass)
-
- # Get/create config to store a unique id for this hass instance.
- store = Store[dict[str, Any]](hass, STORAGE_VERSION, STORAGE_KEY)
-
- if fresh_install or not (config := await store.async_load()):
- # Create config
- config = {
- CONF_INSTANCE_ID: str(uuid4()),
- CONF_WEBHOOK_ID: secrets.token_hex(),
- CONF_CLOUDHOOK_URL: None,
- }
- await store.async_save(config)
-
- # Register webhook
- webhook.async_register(
- hass, DOMAIN, "SmartApp", config[CONF_WEBHOOK_ID], smartapp_webhook
- )
-
- # Create webhook if eligible
- cloudhook_url = config.get(CONF_CLOUDHOOK_URL)
- if (
- cloudhook_url is None
- and cloud.async_active_subscription(hass)
- and not hass.config_entries.async_entries(DOMAIN)
- ):
- cloudhook_url = await cloud.async_create_cloudhook(
- hass, config[CONF_WEBHOOK_ID]
- )
- config[CONF_CLOUDHOOK_URL] = cloudhook_url
- await store.async_save(config)
- _LOGGER.debug("Created cloudhook '%s'", cloudhook_url)
-
- # SmartAppManager uses a dispatcher to invoke callbacks when push events
- # occur. Use hass' implementation instead of the built-in one.
- dispatcher = Dispatcher(
- signal_prefix=SIGNAL_SMARTAPP_PREFIX,
- connect=functools.partial(async_dispatcher_connect, hass),
- send=functools.partial(async_dispatcher_send, hass),
- )
- # Path is used in digital signature validation
- path = (
- urlparse(cloudhook_url).path
- if cloudhook_url
- else webhook.async_generate_path(config[CONF_WEBHOOK_ID])
- )
- manager = SmartAppManager(path, dispatcher=dispatcher)
- manager.connect_install(functools.partial(smartapp_install, hass))
- manager.connect_update(functools.partial(smartapp_update, hass))
- manager.connect_uninstall(functools.partial(smartapp_uninstall, hass))
-
- hass.data[DOMAIN] = {
- DATA_MANAGER: manager,
- CONF_INSTANCE_ID: config[CONF_INSTANCE_ID],
- DATA_BROKERS: {},
- CONF_WEBHOOK_ID: config[CONF_WEBHOOK_ID],
- # Will not be present if not enabled
- CONF_CLOUDHOOK_URL: config.get(CONF_CLOUDHOOK_URL),
- }
- _LOGGER.debug(
- "Setup endpoint for %s",
- cloudhook_url
- if cloudhook_url
- else webhook.async_generate_url(hass, config[CONF_WEBHOOK_ID]),
- )
-
-
-async def unload_smartapp_endpoint(hass: HomeAssistant):
- """Tear down the component configuration."""
- if DOMAIN not in hass.data:
- return
- # Remove the cloudhook if it was created
- cloudhook_url = hass.data[DOMAIN][CONF_CLOUDHOOK_URL]
- if cloudhook_url and cloud.async_is_logged_in(hass):
- await cloud.async_delete_cloudhook(hass, hass.data[DOMAIN][CONF_WEBHOOK_ID])
- # Remove cloudhook from storage
- store = Store[dict[str, Any]](hass, STORAGE_VERSION, STORAGE_KEY)
- await store.async_save(
- {
- CONF_INSTANCE_ID: hass.data[DOMAIN][CONF_INSTANCE_ID],
- CONF_WEBHOOK_ID: hass.data[DOMAIN][CONF_WEBHOOK_ID],
- CONF_CLOUDHOOK_URL: None,
- }
- )
- _LOGGER.debug("Cloudhook '%s' was removed", cloudhook_url)
- # Remove the webhook
- webhook.async_unregister(hass, hass.data[DOMAIN][CONF_WEBHOOK_ID])
- # Disconnect all brokers
- for broker in hass.data[DOMAIN][DATA_BROKERS].values():
- broker.disconnect()
- # Remove all handlers from manager
- hass.data[DOMAIN][DATA_MANAGER].dispatcher.disconnect_all()
- # Remove the component data
- hass.data.pop(DOMAIN)
-
-
-async def smartapp_sync_subscriptions(
- hass: HomeAssistant,
- auth_token: str,
- location_id: str,
- installed_app_id: str,
- devices,
-):
- """Synchronize subscriptions of an installed up."""
- api = SmartThings(async_get_clientsession(hass), auth_token)
- tasks = []
-
- async def create_subscription(target: str):
- sub = Subscription()
- sub.installed_app_id = installed_app_id
- sub.location_id = location_id
- sub.source_type = SourceType.CAPABILITY
- sub.capability = target
- try:
- await api.create_subscription(sub)
- _LOGGER.debug(
- "Created subscription for '%s' under app '%s'", target, installed_app_id
- )
- except Exception as error: # noqa: BLE001
- _LOGGER.error(
- "Failed to create subscription for '%s' under app '%s': %s",
- target,
- installed_app_id,
- error,
- )
-
- async def delete_subscription(sub: SubscriptionEntity):
- try:
- await api.delete_subscription(installed_app_id, sub.subscription_id)
- _LOGGER.debug(
- (
- "Removed subscription for '%s' under app '%s' because it was no"
- " longer needed"
- ),
- sub.capability,
- installed_app_id,
- )
- except Exception as error: # noqa: BLE001
- _LOGGER.error(
- "Failed to remove subscription for '%s' under app '%s': %s",
- sub.capability,
- installed_app_id,
- error,
- )
-
- # Build set of capabilities and prune unsupported ones
- capabilities = set()
- for device in devices:
- capabilities.update(device.capabilities)
- # Remove items not defined in the library
- capabilities.intersection_update(CAPABILITIES)
- # Remove unused capabilities
- capabilities.difference_update(IGNORED_CAPABILITIES)
- capability_count = len(capabilities)
- if capability_count > SUBSCRIPTION_WARNING_LIMIT:
- _LOGGER.warning(
- (
- "Some device attributes may not receive push updates and there may be"
- " subscription creation failures under app '%s' because %s"
- " subscriptions are required but there is a limit of %s per app"
- ),
- installed_app_id,
- capability_count,
- SUBSCRIPTION_WARNING_LIMIT,
- )
- _LOGGER.debug(
- "Synchronizing subscriptions for %s capabilities under app '%s': %s",
- capability_count,
- installed_app_id,
- capabilities,
- )
-
- # Get current subscriptions and find differences
- subscriptions = await api.subscriptions(installed_app_id)
- for subscription in subscriptions:
- if subscription.capability in capabilities:
- capabilities.remove(subscription.capability)
- else:
- # Delete the subscription
- tasks.append(delete_subscription(subscription))
-
- # Remaining capabilities need subscriptions created
- tasks.extend([create_subscription(c) for c in capabilities])
-
- if tasks:
- await asyncio.gather(*tasks)
- else:
- _LOGGER.debug("Subscriptions for app '%s' are up-to-date", installed_app_id)
-
-
-async def _find_and_continue_flow(
- hass: HomeAssistant,
- app_id: str,
- location_id: str,
- installed_app_id: str,
- refresh_token: str,
-):
- """Continue a config flow if one is in progress for the specific installed app."""
- unique_id = format_unique_id(app_id, location_id)
- flow = next(
- (
- flow
- for flow in hass.config_entries.flow.async_progress_by_handler(DOMAIN)
- if flow["context"].get("unique_id") == unique_id
- ),
- None,
- )
- if flow is not None:
- await _continue_flow(hass, app_id, installed_app_id, refresh_token, flow)
-
-
-async def _continue_flow(
- hass: HomeAssistant,
- app_id: str,
- installed_app_id: str,
- refresh_token: str,
- flow: ConfigFlowResult,
-) -> None:
- await hass.config_entries.flow.async_configure(
- flow["flow_id"],
- {
- CONF_INSTALLED_APP_ID: installed_app_id,
- CONF_REFRESH_TOKEN: refresh_token,
- },
- )
- _LOGGER.debug(
- "Continued config flow '%s' for SmartApp '%s' under parent app '%s'",
- flow["flow_id"],
- installed_app_id,
- app_id,
- )
-
-
-async def smartapp_install(hass: HomeAssistant, req, resp, app):
- """Handle a SmartApp installation and continue the config flow."""
- await _find_and_continue_flow(
- hass, app.app_id, req.location_id, req.installed_app_id, req.refresh_token
- )
- _LOGGER.debug(
- "Installed SmartApp '%s' under parent app '%s'",
- req.installed_app_id,
- app.app_id,
- )
-
-
-async def smartapp_update(hass: HomeAssistant, req, resp, app):
- """Handle a SmartApp update and either update the entry or continue the flow."""
- unique_id = format_unique_id(app.app_id, req.location_id)
- flow = next(
- (
- flow
- for flow in hass.config_entries.flow.async_progress_by_handler(DOMAIN)
- if flow["context"].get("unique_id") == unique_id
- and flow["step_id"] == "authorize"
- ),
- None,
- )
- if flow is not None:
- await _continue_flow(
- hass, app.app_id, req.installed_app_id, req.refresh_token, flow
- )
- _LOGGER.debug(
- "Continued reauth flow '%s' for SmartApp '%s' under parent app '%s'",
- flow["flow_id"],
- req.installed_app_id,
- app.app_id,
- )
- return
- entry = next(
- (
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.data.get(CONF_INSTALLED_APP_ID) == req.installed_app_id
- ),
- None,
- )
- if entry:
- hass.config_entries.async_update_entry(
- entry, data={**entry.data, CONF_REFRESH_TOKEN: req.refresh_token}
- )
- _LOGGER.debug(
- "Updated config entry '%s' for SmartApp '%s' under parent app '%s'",
- entry.entry_id,
- req.installed_app_id,
- app.app_id,
- )
-
- await _find_and_continue_flow(
- hass, app.app_id, req.location_id, req.installed_app_id, req.refresh_token
- )
- _LOGGER.debug(
- "Updated SmartApp '%s' under parent app '%s'", req.installed_app_id, app.app_id
- )
-
-
-async def smartapp_uninstall(hass: HomeAssistant, req, resp, app):
- """Handle when a SmartApp is removed from a location by the user.
-
- Find and delete the config entry representing the integration.
- """
- entry = next(
- (
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.data.get(CONF_INSTALLED_APP_ID) == req.installed_app_id
- ),
- None,
- )
- if entry:
- # Add as job not needed because the current coroutine was invoked
- # from the dispatcher and is not being awaited.
- await hass.config_entries.async_remove(entry.entry_id)
-
- _LOGGER.debug(
- "Uninstalled SmartApp '%s' under parent app '%s'",
- req.installed_app_id,
- app.app_id,
- )
-
-
-async def smartapp_webhook(hass: HomeAssistant, webhook_id: str, request):
- """Handle a smartapp lifecycle event callback from SmartThings.
-
- Requests from SmartThings are digitally signed and the SmartAppManager
- validates the signature for authenticity.
- """
- manager = hass.data[DOMAIN][DATA_MANAGER]
- data = await request.json()
- result = await manager.handle_request(data, request.headers)
- return web.json_response(result)
diff --git a/homeassistant/components/smartthings/strings.json b/homeassistant/components/smartthings/strings.json
index 31a552be149..7f6e13ab3ba 100644
--- a/homeassistant/components/smartthings/strings.json
+++ b/homeassistant/components/smartthings/strings.json
@@ -1,43 +1,466 @@
{
"config": {
"step": {
- "user": {
- "title": "Confirm Callback URL",
- "description": "SmartThings will be configured to send push updates to Home Assistant at:\n> {webhook_url}\n\nIf this is not correct, please update your configuration, restart Home Assistant, and try again."
+ "pick_implementation": {
+ "title": "[%key:common::config_flow::title::oauth2_pick_implementation%]"
},
- "pat": {
- "title": "Enter Personal Access Token",
- "description": "Please enter a SmartThings [Personal Access Token]({token_url}) that has been created per the [instructions]({component_url}). This will be used to create the Home Assistant integration within your SmartThings account.\n\n**Please note that all Personal Access Tokens created after 30 December 2024 are only valid for 24 hours, after which the integration will stop working. We are working on a fix.**",
- "data": {
- "access_token": "[%key:common::config_flow::data::access_token%]"
- }
- },
- "select_location": {
- "title": "Select Location",
- "description": "Please select the SmartThings Location you wish to add to Home Assistant. We will then open a new window and ask you to login and authorize installation of the Home Assistant integration into the selected location.",
- "data": { "location_id": "[%key:common::config_flow::data::location%]" }
- },
- "authorize": { "title": "Authorize Home Assistant" },
"reauth_confirm": {
- "title": "Reauthorize Home Assistant",
- "description": "You are about to reauthorize Home Assistant with SmartThings. This will require you to log in and authorize the integration again."
- },
- "update_confirm": {
- "title": "Finish reauthentication",
- "description": "You have almost successfully reauthorized Home Assistant with SmartThings. Please press the button down below to finish the process."
+ "title": "[%key:common::config_flow::title::reauth%]",
+ "description": "The SmartThings integration needs to re-authenticate your account"
}
},
- "abort": {
- "invalid_webhook_url": "Home Assistant is not configured correctly to receive updates from SmartThings. The webhook URL is invalid:\n> {webhook_url}\n\nPlease update your configuration per the [instructions]({component_url}), restart Home Assistant, and try again.",
- "no_available_locations": "There are no available SmartThings Locations to set up in Home Assistant.",
- "reauth_successful": "Home Assistant has been successfully reauthorized with SmartThings."
- },
"error": {
- "token_invalid_format": "The token must be in the UID/GUID format",
- "token_unauthorized": "The token is invalid or no longer authorized.",
- "token_forbidden": "The token does not have the required OAuth scopes.",
- "app_setup_error": "Unable to set up the SmartApp. Please try again.",
- "webhook_error": "SmartThings could not validate the webhook URL. Please ensure the webhook URL is reachable from the internet and try again."
+ "already_configured": "[%key:common::config_flow::abort::already_configured_account%]"
+ },
+ "abort": {
+ "authorize_url_timeout": "[%key:common::config_flow::abort::oauth2_authorize_url_timeout%]",
+ "missing_configuration": "[%key:common::config_flow::abort::oauth2_missing_configuration%]",
+ "already_configured": "[%key:common::config_flow::abort::already_configured_account%]",
+ "no_url_available": "[%key:common::config_flow::abort::oauth2_no_url_available%]",
+ "oauth_error": "[%key:common::config_flow::abort::oauth2_error%]",
+ "oauth_timeout": "[%key:common::config_flow::abort::oauth2_timeout%]",
+ "oauth_unauthorized": "[%key:common::config_flow::abort::oauth2_unauthorized%]",
+ "oauth_failed": "[%key:common::config_flow::abort::oauth2_failed%]",
+ "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",
+ "reauth_account_mismatch": "Authenticated account does not match the account to be reauthenticated. Please log in with the correct account and pick the right location.",
+ "reauth_location_mismatch": "Authenticated location does not match the location to be reauthenticated. Please log in with the correct account and pick the right location.",
+ "missing_scopes": "Authentication failed. Please make sure you have granted all required permissions.",
+ "cloud_not_enabled": "Please make sure you run Home Assistant with `{default_config}` enabled in your configuration.yaml."
+ }
+ },
+ "entity": {
+ "binary_sensor": {
+ "acceleration": {
+ "name": "Acceleration"
+ },
+ "door": {
+ "name": "[%key:component::binary_sensor::entity_component::door::name%]"
+ },
+ "filter_status": {
+ "name": "Filter status"
+ },
+ "remote_control": {
+ "name": "Remote control"
+ },
+ "child_lock": {
+ "name": "Child lock"
+ },
+ "valve": {
+ "name": "Valve"
+ }
+ },
+ "button": {
+ "stop": {
+ "name": "Stop"
+ }
+ },
+ "event": {
+ "button": {
+ "state": {
+ "pushed": "Pushed",
+ "held": "Held",
+ "double": "Double",
+ "pushed_2x": "Pushed 2x",
+ "pushed_3x": "Pushed 3x",
+ "pushed_4x": "Pushed 4x",
+ "pushed_5x": "Pushed 5x",
+ "pushed_6x": "Pushed 6x",
+ "down": "Down",
+ "down_2x": "Down 2x",
+ "down_3x": "Down 3x",
+ "down_4x": "Down 4x",
+ "down_5x": "Down 5x",
+ "down_6x": "Down 6x",
+ "down_hold": "Down hold",
+ "up": "Up",
+ "up_2x": "Up 2x",
+ "up_3x": "Up 3x",
+ "up_4x": "Up 4x",
+ "up_5x": "Up 5x",
+ "up_6x": "Up 6x",
+ "up_hold": "Up hold",
+ "swipe_up": "Swipe up",
+ "swipe_down": "Swipe down",
+ "swipe_left": "Swipe left",
+ "swipe_right": "Swipe right"
+ }
+ }
+ },
+ "number": {
+ "washer_rinse_cycles": {
+ "name": "Rinse cycles",
+ "unit_of_measurement": "cycles"
+ }
+ },
+ "select": {
+ "operating_state": {
+ "state": {
+ "run": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::state::run%]",
+ "pause": "[%key:common::state::paused%]",
+ "stop": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::state::stop%]"
+ }
+ }
+ },
+ "sensor": {
+ "lighting_mode": {
+ "name": "Activity lighting mode"
+ },
+ "air_conditioner_mode": {
+ "name": "Air conditioner mode"
+ },
+ "air_quality": {
+ "name": "Air quality"
+ },
+ "alarm": {
+ "name": "Alarm",
+ "state": {
+ "both": "Strobe and siren",
+ "strobe": "Strobe",
+ "siren": "Siren",
+ "off": "[%key:common::state::off%]"
+ }
+ },
+ "audio_volume": {
+ "name": "Volume"
+ },
+ "body_mass_index": {
+ "name": "Body mass index"
+ },
+ "body_weight": {
+ "name": "Body weight"
+ },
+ "carbon_monoxide_detector": {
+ "name": "Carbon monoxide detector",
+ "state": {
+ "detected": "Detected",
+ "clear": "Clear",
+ "tested": "Tested"
+ }
+ },
+ "dishwasher_machine_state": {
+ "name": "Machine state",
+ "state": {
+ "pause": "[%key:common::state::paused%]",
+ "run": "Running",
+ "stop": "Stopped"
+ }
+ },
+ "dishwasher_job_state": {
+ "name": "Job state",
+ "state": {
+ "air_wash": "Air wash",
+ "cooling": "Cooling",
+ "drying": "Drying",
+ "finish": "Finish",
+ "pre_drain": "Pre-drain",
+ "pre_wash": "Pre-wash",
+ "rinse": "Rinse",
+ "spin": "Spin",
+ "wash": "Wash",
+ "wrinkle_prevent": "Wrinkle prevention"
+ }
+ },
+ "completion_time": {
+ "name": "Completion time"
+ },
+ "dryer_mode": {
+ "name": "Dryer mode"
+ },
+ "dryer_machine_state": {
+ "name": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::name%]",
+ "state": {
+ "pause": "[%key:common::state::paused%]",
+ "run": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::state::run%]",
+ "stop": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::state::stop%]"
+ }
+ },
+ "dryer_job_state": {
+ "name": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::name%]",
+ "state": {
+ "cooling": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::cooling%]",
+ "delay_wash": "[%key:component::smartthings::entity::sensor::washer_job_state::state::delay_wash%]",
+ "drying": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::drying%]",
+ "finished": "[%key:component::smartthings::entity::sensor::oven_job_state::state::finished%]",
+ "none": "[%key:component::smartthings::entity::sensor::washer_job_state::state::none%]",
+ "refreshing": "Refreshing",
+ "weight_sensing": "[%key:component::smartthings::entity::sensor::washer_job_state::state::weight_sensing%]",
+ "wrinkle_prevent": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::wrinkle_prevent%]",
+ "dehumidifying": "Dehumidifying",
+ "ai_drying": "AI drying",
+ "sanitizing": "Sanitizing",
+ "internal_care": "Internal care",
+ "freeze_protection": "Freeze protection",
+ "continuous_dehumidifying": "Continuous dehumidifying",
+ "thawing_frozen_inside": "Thawing frozen inside"
+ }
+ },
+ "equivalent_carbon_dioxide": {
+ "name": "Equivalent carbon dioxide"
+ },
+ "formaldehyde": {
+ "name": "Formaldehyde"
+ },
+ "gas_meter": {
+ "name": "Gas meter"
+ },
+ "gas_meter_calorific": {
+ "name": "Gas meter calorific"
+ },
+ "gas_meter_time": {
+ "name": "Gas meter time"
+ },
+ "infrared_level": {
+ "name": "Infrared level"
+ },
+ "media_input_source": {
+ "name": "Media input source",
+ "state": {
+ "am": "AM",
+ "fm": "FM",
+ "cd": "CD",
+ "hdmi": "HDMI",
+ "hdmi1": "HDMI 1",
+ "hdmi2": "HDMI 2",
+ "hdmi3": "HDMI 3",
+ "hdmi4": "HDMI 4",
+ "hdmi5": "HDMI 5",
+ "hdmi6": "HDMI 6",
+ "digitaltv": "Digital TV",
+ "usb": "USB",
+ "youtube": "YouTube",
+ "aux": "AUX",
+ "bluetooth": "Bluetooth",
+ "digital": "Digital",
+ "melon": "Melon",
+ "wifi": "Wi-Fi",
+ "network": "Network",
+ "optical": "Optical",
+ "coaxial": "Coaxial",
+ "analog1": "Analog 1",
+ "analog2": "Analog 2",
+ "analog3": "Analog 3",
+ "phono": "Phono"
+ }
+ },
+ "media_playback_repeat": {
+ "name": "Media playback repeat"
+ },
+ "media_playback_shuffle": {
+ "name": "Media playback shuffle"
+ },
+ "media_playback_status": {
+ "name": "Media playback status"
+ },
+ "odor_sensor": {
+ "name": "Odor sensor"
+ },
+ "oven_mode": {
+ "name": "Oven mode",
+ "state": {
+ "heating": "Heating",
+ "grill": "Grill",
+ "warming": "Warming",
+ "defrosting": "Defrosting",
+ "conventional": "Conventional",
+ "bake": "Bake",
+ "bottom_heat": "Bottom heat",
+ "convection_bake": "Convection bake",
+ "convection_roast": "Convection roast",
+ "broil": "Broil",
+ "convection_broil": "Convection broil",
+ "steam_cook": "Steam cook",
+ "steam_bake": "Steam bake",
+ "steam_roast": "Steam roast",
+ "steam_bottom_heat_plus_convection": "Steam bottom heat plus convection",
+ "microwave": "Microwave",
+ "microwave_plus_grill": "Microwave plus grill",
+ "microwave_plus_convection": "Microwave plus convection",
+ "microwave_plus_hot_blast": "Microwave plus hot blast",
+ "microwave_plus_hot_blast_2": "Microwave plus hot blast 2",
+ "slim_middle": "Slim middle",
+ "slim_strong": "Slim strong",
+ "slow_cook": "Slow cook",
+ "proof": "Proof",
+ "dehydrate": "Dehydrate",
+ "others": "Others",
+ "strong_steam": "Strong steam",
+ "descale": "Descale",
+ "rinse": "Rinse"
+ }
+ },
+ "oven_machine_state": {
+ "name": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::name%]",
+ "state": {
+ "ready": "Ready",
+ "running": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::state::run%]",
+ "paused": "[%key:common::state::paused%]"
+ }
+ },
+ "oven_job_state": {
+ "name": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::name%]",
+ "state": {
+ "cleaning": "Cleaning",
+ "cooking": "Cooking",
+ "cooling": "Cooling",
+ "draining": "Draining",
+ "preheat": "Preheat",
+ "ready": "Ready",
+ "rinsing": "Rinsing",
+ "finished": "Finished",
+ "scheduled_start": "Scheduled start",
+ "warming": "Warming",
+ "defrosting": "Defrosting",
+ "sensing": "Sensing",
+ "searing": "Searing",
+ "fast_preheat": "Fast preheat",
+ "scheduled_end": "Scheduled end",
+ "stone_heating": "Stone heating",
+ "time_hold_preheat": "Time hold preheat"
+ }
+ },
+ "oven_setpoint": {
+ "name": "Set point"
+ },
+ "energy_difference": {
+ "name": "Energy difference"
+ },
+ "power_energy": {
+ "name": "Power energy"
+ },
+ "energy_saved": {
+ "name": "Energy saved"
+ },
+ "power_source": {
+ "name": "Power source"
+ },
+ "refrigeration_setpoint": {
+ "name": "[%key:component::smartthings::entity::sensor::oven_setpoint::name%]"
+ },
+ "robot_cleaner_cleaning_mode": {
+ "name": "Cleaning mode",
+ "state": {
+ "auto": "Auto",
+ "part": "Partial",
+ "repeat": "Repeat",
+ "manual": "Manual",
+ "stop": "[%key:common::action::stop%]",
+ "map": "Map"
+ }
+ },
+ "robot_cleaner_movement": {
+ "name": "Movement",
+ "state": {
+ "homing": "Homing",
+ "idle": "[%key:common::state::idle%]",
+ "charging": "[%key:common::state::charging%]",
+ "alarm": "Alarm",
+ "off": "[%key:common::state::off%]",
+ "reserve": "Reserve",
+ "point": "Point",
+ "after": "After",
+ "cleaning": "Cleaning",
+ "pause": "[%key:common::state::paused%]"
+ }
+ },
+ "robot_cleaner_turbo_mode": {
+ "name": "Turbo mode",
+ "state": {
+ "on": "[%key:common::state::on%]",
+ "off": "[%key:common::state::off%]",
+ "silence": "Silent",
+ "extra_silence": "Extra silent"
+ }
+ },
+ "link_quality": {
+ "name": "Link quality"
+ },
+ "smoke_detector": {
+ "name": "Smoke detector",
+ "state": {
+ "detected": "[%key:component::smartthings::entity::sensor::carbon_monoxide_detector::state::detected%]",
+ "clear": "[%key:component::smartthings::entity::sensor::carbon_monoxide_detector::state::clear%]",
+ "tested": "[%key:component::smartthings::entity::sensor::carbon_monoxide_detector::state::tested%]"
+ }
+ },
+ "thermostat_cooling_setpoint": {
+ "name": "Cooling set point"
+ },
+ "thermostat_fan_mode": {
+ "name": "Fan mode"
+ },
+ "thermostat_heating_setpoint": {
+ "name": "Heating set point"
+ },
+ "thermostat_mode": {
+ "name": "Mode"
+ },
+ "thermostat_operating_state": {
+ "name": "Operating state"
+ },
+ "thermostat_setpoint": {
+ "name": "[%key:component::smartthings::entity::sensor::oven_setpoint::name%]"
+ },
+ "x_coordinate": {
+ "name": "X coordinate"
+ },
+ "y_coordinate": {
+ "name": "Y coordinate"
+ },
+ "z_coordinate": {
+ "name": "Z coordinate"
+ },
+ "tv_channel": {
+ "name": "TV channel"
+ },
+ "tv_channel_name": {
+ "name": "TV channel name"
+ },
+ "uv_index": {
+ "name": "UV index"
+ },
+ "washer_mode": {
+ "name": "Washer mode"
+ },
+ "washer_machine_state": {
+ "name": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::name%]",
+ "state": {
+ "pause": "[%key:common::state::paused%]",
+ "run": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::state::run%]",
+ "stop": "[%key:component::smartthings::entity::sensor::dishwasher_machine_state::state::stop%]"
+ }
+ },
+ "washer_job_state": {
+ "name": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::name%]",
+ "state": {
+ "air_wash": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::air_wash%]",
+ "ai_rinse": "AI rinse",
+ "ai_spin": "AI spin",
+ "ai_wash": "AI wash",
+ "cooling": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::cooling%]",
+ "delay_wash": "Delay wash",
+ "drying": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::drying%]",
+ "finish": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::finish%]",
+ "none": "None",
+ "pre_wash": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::pre_wash%]",
+ "rinse": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::rinse%]",
+ "spin": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::spin%]",
+ "wash": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::wash%]",
+ "weight_sensing": "Weight sensing",
+ "wrinkle_prevent": "[%key:component::smartthings::entity::sensor::dishwasher_job_state::state::wrinkle_prevent%]",
+ "freeze_protection": "Freeze protection"
+ }
+ }
+ },
+ "switch": {
+ "wrinkle_prevent": {
+ "name": "Wrinkle prevent"
+ }
+ }
+ },
+ "issues": {
+ "deprecated_binary_valve": {
+ "title": "Deprecated valve binary sensor detected in some automations or scripts",
+ "description": "The valve binary sensor `{entity}` is deprecated and is used in the following automations or scripts:\n{items}\n\nA valve entity with controls is available and should be used going forward; Please use it on the above automations or scripts to fix this issue."
}
}
}
diff --git a/homeassistant/components/smartthings/switch.py b/homeassistant/components/smartthings/switch.py
index 5cfe4576d6a..6e0dc1ac93d 100644
--- a/homeassistant/components/smartthings/switch.py
+++ b/homeassistant/components/smartthings/switch.py
@@ -2,60 +2,155 @@
from __future__ import annotations
-from collections.abc import Sequence
+from dataclasses import dataclass
from typing import Any
-from pysmartthings import Capability
+from pysmartthings import Attribute, Capability, Command, SmartThings
-from homeassistant.components.switch import SwitchEntity
-from homeassistant.config_entries import ConfigEntry
+from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DATA_BROKERS, DOMAIN
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
from .entity import SmartThingsEntity
+CAPABILITIES = (
+ Capability.SWITCH_LEVEL,
+ Capability.COLOR_CONTROL,
+ Capability.COLOR_TEMPERATURE,
+ Capability.FAN_SPEED,
+)
+
+AC_CAPABILITIES = (
+ Capability.AIR_CONDITIONER_MODE,
+ Capability.AIR_CONDITIONER_FAN_MODE,
+ Capability.TEMPERATURE_MEASUREMENT,
+ Capability.THERMOSTAT_COOLING_SETPOINT,
+)
+
+
+@dataclass(frozen=True, kw_only=True)
+class SmartThingsSwitchEntityDescription(SwitchEntityDescription):
+ """Describe a SmartThings switch entity."""
+
+ status_attribute: Attribute
+
+
+@dataclass(frozen=True, kw_only=True)
+class SmartThingsCommandSwitchEntityDescription(SmartThingsSwitchEntityDescription):
+ """Describe a SmartThings switch entity."""
+
+ command: Command
+
+
+SWITCH = SmartThingsSwitchEntityDescription(
+ key=Capability.SWITCH,
+ status_attribute=Attribute.SWITCH,
+ name=None,
+)
+CAPABILITY_TO_COMMAND_SWITCHES: dict[
+ Capability | str, SmartThingsCommandSwitchEntityDescription
+] = {
+ Capability.CUSTOM_DRYER_WRINKLE_PREVENT: SmartThingsCommandSwitchEntityDescription(
+ key=Capability.CUSTOM_DRYER_WRINKLE_PREVENT,
+ translation_key="wrinkle_prevent",
+ status_attribute=Attribute.DRYER_WRINKLE_PREVENT,
+ command=Command.SET_DRYER_WRINKLE_PREVENT,
+ )
+}
+
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add switches for a config entry."""
- broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id]
- async_add_entities(
- SmartThingsSwitch(device)
- for device in broker.devices.values()
- if broker.any_assigned(device.device_id, "switch")
+ entry_data = entry.runtime_data
+ entities: list[SmartThingsEntity] = [
+ SmartThingsSwitch(entry_data.client, device, SWITCH, Capability.SWITCH)
+ for device in entry_data.devices.values()
+ if Capability.SWITCH in device.status[MAIN]
+ and not any(capability in device.status[MAIN] for capability in CAPABILITIES)
+ and not all(capability in device.status[MAIN] for capability in AC_CAPABILITIES)
+ ]
+ entities.extend(
+ SmartThingsCommandSwitch(
+ entry_data.client,
+ device,
+ description,
+ Capability(capability),
+ )
+ for device in entry_data.devices.values()
+ for capability, description in CAPABILITY_TO_COMMAND_SWITCHES.items()
+ if capability in device.status[MAIN]
)
-
-
-def get_capabilities(capabilities: Sequence[str]) -> Sequence[str] | None:
- """Return all capabilities supported if minimum required are present."""
- # Must be able to be turned on/off.
- if Capability.switch in capabilities:
- return [Capability.switch, Capability.energy_meter, Capability.power_meter]
- return None
+ async_add_entities(entities)
class SmartThingsSwitch(SmartThingsEntity, SwitchEntity):
"""Define a SmartThings switch."""
+ entity_description: SmartThingsSwitchEntityDescription
+
+ def __init__(
+ self,
+ client: SmartThings,
+ device: FullDevice,
+ entity_description: SmartThingsSwitchEntityDescription,
+ capability: Capability,
+ ) -> None:
+ """Initialize the switch."""
+ super().__init__(client, device, {capability})
+ self.entity_description = entity_description
+ self.switch_capability = capability
+ self._attr_unique_id = device.device.device_id
+ if capability is not Capability.SWITCH:
+ self._attr_unique_id = f"{device.device.device_id}_{MAIN}_{capability}"
+
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the switch off."""
- await self._device.switch_off(set_status=True)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
+ await self.execute_device_command(
+ self.switch_capability,
+ Command.OFF,
+ )
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn the switch on."""
- await self._device.switch_on(set_status=True)
- # State is set optimistically in the command above, therefore update
- # the entity state ahead of receiving the confirming push updates
- self.async_write_ha_state()
+ await self.execute_device_command(
+ self.switch_capability,
+ Command.ON,
+ )
@property
def is_on(self) -> bool:
- """Return true if light is on."""
- return self._device.status.switch
+ """Return true if switch is on."""
+ return (
+ self.get_attribute_value(
+ self.switch_capability, self.entity_description.status_attribute
+ )
+ == "on"
+ )
+
+
+class SmartThingsCommandSwitch(SmartThingsSwitch):
+ """Define a SmartThings command switch."""
+
+ entity_description: SmartThingsCommandSwitchEntityDescription
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the switch off."""
+ await self.execute_device_command(
+ self.switch_capability,
+ self.entity_description.command,
+ "off",
+ )
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the switch on."""
+ await self.execute_device_command(
+ self.switch_capability,
+ self.entity_description.command,
+ "on",
+ )
diff --git a/homeassistant/components/smartthings/update.py b/homeassistant/components/smartthings/update.py
new file mode 100644
index 00000000000..bb226918596
--- /dev/null
+++ b/homeassistant/components/smartthings/update.py
@@ -0,0 +1,87 @@
+"""Support for update entities through the SmartThings cloud API."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from awesomeversion import AwesomeVersion
+from pysmartthings import Attribute, Capability, Command
+
+from homeassistant.components.update import (
+ UpdateDeviceClass,
+ UpdateEntity,
+ UpdateEntityFeature,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import SmartThingsConfigEntry
+from .const import MAIN
+from .entity import SmartThingsEntity
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add update entities for a config entry."""
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsUpdateEntity(entry_data.client, device, {Capability.FIRMWARE_UPDATE})
+ for device in entry_data.devices.values()
+ if Capability.FIRMWARE_UPDATE in device.status[MAIN]
+ )
+
+
+def is_hex_version(version: str) -> bool:
+ """Check if the version is a hex version."""
+ return len(version) == 8 and all(c in "0123456789abcdefABCDEF" for c in version)
+
+
+class SmartThingsUpdateEntity(SmartThingsEntity, UpdateEntity):
+ """Define a SmartThings update entity."""
+
+ _attr_device_class = UpdateDeviceClass.FIRMWARE
+ _attr_supported_features = (
+ UpdateEntityFeature.INSTALL | UpdateEntityFeature.PROGRESS
+ )
+
+ @property
+ def installed_version(self) -> str | None:
+ """Return the installed version of the entity."""
+ return self.get_attribute_value(
+ Capability.FIRMWARE_UPDATE, Attribute.CURRENT_VERSION
+ )
+
+ @property
+ def latest_version(self) -> str | None:
+ """Return the available version of the entity."""
+ return self.get_attribute_value(
+ Capability.FIRMWARE_UPDATE, Attribute.AVAILABLE_VERSION
+ )
+
+ @property
+ def in_progress(self) -> bool:
+ """Return if the entity is in progress."""
+ return (
+ self.get_attribute_value(Capability.FIRMWARE_UPDATE, Attribute.STATE)
+ == "updateInProgress"
+ )
+
+ async def async_install(
+ self, version: str | None, backup: bool, **kwargs: Any
+ ) -> None:
+ """Install the firmware update."""
+ await self.execute_device_command(
+ Capability.FIRMWARE_UPDATE,
+ Command.UPDATE_FIRMWARE,
+ )
+
+ def version_is_newer(self, latest_version: str, installed_version: str) -> bool:
+ """Return if the latest version is newer."""
+ if is_hex_version(latest_version):
+ latest_version = f"0x{latest_version}"
+ if is_hex_version(installed_version):
+ installed_version = f"0x{installed_version}"
+ return AwesomeVersion(latest_version) > AwesomeVersion(installed_version)
diff --git a/homeassistant/components/smartthings/valve.py b/homeassistant/components/smartthings/valve.py
new file mode 100644
index 00000000000..3c401c087ec
--- /dev/null
+++ b/homeassistant/components/smartthings/valve.py
@@ -0,0 +1,71 @@
+"""Support for valves through the SmartThings cloud API."""
+
+from __future__ import annotations
+
+from pysmartthings import Attribute, Capability, Category, Command, SmartThings
+
+from homeassistant.components.valve import (
+ ValveDeviceClass,
+ ValveEntity,
+ ValveEntityFeature,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import FullDevice, SmartThingsConfigEntry
+from .const import MAIN
+from .entity import SmartThingsEntity
+
+DEVICE_CLASS_MAP: dict[Category | str, ValveDeviceClass] = {
+ Category.WATER_VALVE: ValveDeviceClass.WATER,
+ Category.GAS_VALVE: ValveDeviceClass.GAS,
+}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SmartThingsConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Add valves for a config entry."""
+ entry_data = entry.runtime_data
+ async_add_entities(
+ SmartThingsValve(entry_data.client, device)
+ for device in entry_data.devices.values()
+ if Capability.VALVE in device.status[MAIN]
+ )
+
+
+class SmartThingsValve(SmartThingsEntity, ValveEntity):
+ """Define a SmartThings valve."""
+
+ _attr_supported_features = ValveEntityFeature.OPEN | ValveEntityFeature.CLOSE
+ _attr_reports_position = False
+ _attr_name = None
+
+ def __init__(self, client: SmartThings, device: FullDevice) -> None:
+ """Init the class."""
+ super().__init__(client, device, {Capability.VALVE})
+ self._attr_device_class = DEVICE_CLASS_MAP.get(
+ device.device.components[0].user_category
+ or device.device.components[0].manufacturer_category
+ )
+
+ async def async_open_valve(self) -> None:
+ """Open the valve."""
+ await self.execute_device_command(
+ Capability.VALVE,
+ Command.OPEN,
+ )
+
+ async def async_close_valve(self) -> None:
+ """Close the valve."""
+ await self.execute_device_command(
+ Capability.VALVE,
+ Command.CLOSE,
+ )
+
+ @property
+ def is_closed(self) -> bool:
+ """Return if the valve is closed."""
+ return self.get_attribute_value(Capability.VALVE, Attribute.VALVE) == "closed"
diff --git a/homeassistant/components/smarttub/binary_sensor.py b/homeassistant/components/smarttub/binary_sensor.py
index f665f5e61b3..2e8792140b0 100644
--- a/homeassistant/components/smarttub/binary_sensor.py
+++ b/homeassistant/components/smarttub/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from .const import ATTR_ERRORS, ATTR_REMINDERS, DOMAIN, SMARTTUB_CONTROLLER
@@ -43,7 +43,9 @@ SNOOZE_REMINDER_SCHEMA: VolDictType = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor entities for the binary sensors in the tub."""
diff --git a/homeassistant/components/smarttub/climate.py b/homeassistant/components/smarttub/climate.py
index 7f3163834e0..f5759f32fa3 100644
--- a/homeassistant/components/smarttub/climate.py
+++ b/homeassistant/components/smarttub/climate.py
@@ -17,7 +17,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import TemperatureConverter
from .const import DEFAULT_MAX_TEMP, DEFAULT_MIN_TEMP, DOMAIN, SMARTTUB_CONTROLLER
@@ -42,7 +42,9 @@ HVAC_ACTIONS = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate entity for the thermostat in the tub."""
diff --git a/homeassistant/components/smarttub/light.py b/homeassistant/components/smarttub/light.py
index 532234f4059..dda936aa56a 100644
--- a/homeassistant/components/smarttub/light.py
+++ b/homeassistant/components/smarttub/light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_LIGHTS,
@@ -28,7 +28,9 @@ from .helpers import get_spa_name
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entities for any lights in the tub."""
diff --git a/homeassistant/components/smarttub/manifest.json b/homeassistant/components/smarttub/manifest.json
index d5102f14437..b8d81db0ea5 100644
--- a/homeassistant/components/smarttub/manifest.json
+++ b/homeassistant/components/smarttub/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/smarttub",
"iot_class": "cloud_polling",
"loggers": ["smarttub"],
- "requirements": ["python-smarttub==0.0.38"]
+ "requirements": ["python-smarttub==0.0.39"]
}
diff --git a/homeassistant/components/smarttub/sensor.py b/homeassistant/components/smarttub/sensor.py
index 585e8859432..b2bb1170d09 100644
--- a/homeassistant/components/smarttub/sensor.py
+++ b/homeassistant/components/smarttub/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from .const import DOMAIN, SMARTTUB_CONTROLLER
@@ -43,7 +43,9 @@ SET_SECONDARY_FILTRATION_SCHEMA: VolDictType = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor entities for the sensors in the tub."""
diff --git a/homeassistant/components/smarttub/strings.json b/homeassistant/components/smarttub/strings.json
index 974e5fb7d37..79fa7a4820f 100644
--- a/homeassistant/components/smarttub/strings.json
+++ b/homeassistant/components/smarttub/strings.json
@@ -49,17 +49,17 @@
},
"snooze_reminder": {
"name": "Snooze a reminder",
- "description": "Delay a reminder, so that it won't trigger again for a period of time.",
+ "description": "Temporarily suppresses the maintenance reminder on a hot tub.",
"fields": {
"days": {
"name": "Days",
- "description": "The number of days to delay the reminder."
+ "description": "The number of days to snooze the reminder."
}
}
},
"reset_reminder": {
"name": "Reset a reminder",
- "description": "Reset a reminder, and set the next time it will be triggered.",
+ "description": "Resets the maintenance reminder on a hot tub.",
"fields": {
"days": {
"name": "[%key:component::smarttub::services::snooze_reminder::fields::days::name%]",
diff --git a/homeassistant/components/smarttub/switch.py b/homeassistant/components/smarttub/switch.py
index 6e1cf9bef2a..2dedad8e18a 100644
--- a/homeassistant/components/smarttub/switch.py
+++ b/homeassistant/components/smarttub/switch.py
@@ -8,7 +8,7 @@ from smarttub import SpaPump
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import API_TIMEOUT, ATTR_PUMPS, DOMAIN, SMARTTUB_CONTROLLER
from .entity import SmartTubEntity
@@ -16,7 +16,9 @@ from .helpers import get_spa_name
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch entities for the pumps on the tub."""
diff --git a/homeassistant/components/smarty/__init__.py b/homeassistant/components/smarty/__init__.py
index 0e1e99aa444..aab8c6ab3c7 100644
--- a/homeassistant/components/smarty/__init__.py
+++ b/homeassistant/components/smarty/__init__.py
@@ -89,7 +89,7 @@ async def _async_import(hass: HomeAssistant, config: ConfigType) -> None:
async def async_setup_entry(hass: HomeAssistant, entry: SmartyConfigEntry) -> bool:
"""Set up the Smarty environment from a config entry."""
- coordinator = SmartyCoordinator(hass)
+ coordinator = SmartyCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/smarty/binary_sensor.py b/homeassistant/components/smarty/binary_sensor.py
index 213cb00d47c..82236a154f0 100644
--- a/homeassistant/components/smarty/binary_sensor.py
+++ b/homeassistant/components/smarty/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SmartyConfigEntry, SmartyCoordinator
from .entity import SmartyEntity
@@ -53,7 +53,7 @@ ENTITIES: tuple[SmartyBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SmartyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smarty Binary Sensor Platform."""
diff --git a/homeassistant/components/smarty/button.py b/homeassistant/components/smarty/button.py
index b8e31cf6fc8..78638561088 100644
--- a/homeassistant/components/smarty/button.py
+++ b/homeassistant/components/smarty/button.py
@@ -11,7 +11,7 @@ from pysmarty2 import Smarty
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SmartyConfigEntry, SmartyCoordinator
from .entity import SmartyEntity
@@ -38,7 +38,7 @@ ENTITIES: tuple[SmartyButtonDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SmartyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smarty Button Platform."""
diff --git a/homeassistant/components/smarty/coordinator.py b/homeassistant/components/smarty/coordinator.py
index d7f3e2452d1..a55c9f2e78f 100644
--- a/homeassistant/components/smarty/coordinator.py
+++ b/homeassistant/components/smarty/coordinator.py
@@ -22,15 +22,16 @@ class SmartyCoordinator(DataUpdateCoordinator[None]):
software_version: str
configuration_version: str
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: SmartyConfigEntry) -> None:
"""Initialize."""
super().__init__(
hass,
logger=_LOGGER,
+ config_entry=config_entry,
name="Smarty",
update_interval=timedelta(seconds=30),
)
- self.client = Smarty(host=self.config_entry.data[CONF_HOST])
+ self.client = Smarty(host=config_entry.data[CONF_HOST])
async def _async_setup(self) -> None:
if not await self.hass.async_add_executor_job(self.client.update):
diff --git a/homeassistant/components/smarty/fan.py b/homeassistant/components/smarty/fan.py
index 2804f14ee15..07dec85ae47 100644
--- a/homeassistant/components/smarty/fan.py
+++ b/homeassistant/components/smarty/fan.py
@@ -9,7 +9,7 @@ from typing import Any
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -29,7 +29,7 @@ SPEED_RANGE = (1, 3) # off is not included
async def async_setup_entry(
hass: HomeAssistant,
entry: SmartyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smarty Fan Platform."""
diff --git a/homeassistant/components/smarty/manifest.json b/homeassistant/components/smarty/manifest.json
index ca3133d8add..c295647b8e5 100644
--- a/homeassistant/components/smarty/manifest.json
+++ b/homeassistant/components/smarty/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "hub",
"iot_class": "local_polling",
"loggers": ["pymodbus", "pysmarty2"],
- "requirements": ["pysmarty2==0.10.1"]
+ "requirements": ["pysmarty2==0.10.2"]
}
diff --git a/homeassistant/components/smarty/sensor.py b/homeassistant/components/smarty/sensor.py
index 48b169c104e..fe35f741380 100644
--- a/homeassistant/components/smarty/sensor.py
+++ b/homeassistant/components/smarty/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import REVOLUTIONS_PER_MINUTE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .coordinator import SmartyConfigEntry, SmartyCoordinator
@@ -85,7 +85,7 @@ ENTITIES: tuple[SmartySensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SmartyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smarty Sensor Platform."""
diff --git a/homeassistant/components/smarty/switch.py b/homeassistant/components/smarty/switch.py
index bf5fe80db44..5781bb11680 100644
--- a/homeassistant/components/smarty/switch.py
+++ b/homeassistant/components/smarty/switch.py
@@ -11,7 +11,7 @@ from pysmarty2 import Smarty
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SmartyConfigEntry, SmartyCoordinator
from .entity import SmartyEntity
@@ -42,7 +42,7 @@ ENTITIES: tuple[SmartySwitchDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SmartyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Smarty Switch Platform."""
diff --git a/homeassistant/components/smhi/__init__.py b/homeassistant/components/smhi/__init__.py
index 59b32948879..1869b333071 100644
--- a/homeassistant/components/smhi/__init__.py
+++ b/homeassistant/components/smhi/__init__.py
@@ -1,6 +1,5 @@
"""Support for the Swedish weather institute weather service."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_LATITUDE,
CONF_LOCATION,
@@ -10,10 +9,12 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
+from .coordinator import SMHIConfigEntry, SMHIDataUpdateCoordinator
+
PLATFORMS = [Platform.WEATHER]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: SMHIConfigEntry) -> bool:
"""Set up SMHI forecast as config entry."""
# Setting unique id where missing
@@ -21,16 +22,20 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
unique_id = f"{entry.data[CONF_LOCATION][CONF_LATITUDE]}-{entry.data[CONF_LOCATION][CONF_LONGITUDE]}"
hass.config_entries.async_update_entry(entry, unique_id=unique_id)
+ coordinator = SMHIDataUpdateCoordinator(hass, entry)
+ await coordinator.async_config_entry_first_refresh()
+ entry.runtime_data = coordinator
+
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: SMHIConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
-async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_migrate_entry(hass: HomeAssistant, entry: SMHIConfigEntry) -> bool:
"""Migrate old entry."""
if entry.version > 3:
diff --git a/homeassistant/components/smhi/const.py b/homeassistant/components/smhi/const.py
index 11401119227..6cbf928d5e6 100644
--- a/homeassistant/components/smhi/const.py
+++ b/homeassistant/components/smhi/const.py
@@ -1,5 +1,7 @@
"""Constants in smhi component."""
+from datetime import timedelta
+import logging
from typing import Final
from homeassistant.components.weather import DOMAIN as WEATHER_DOMAIN
@@ -12,3 +14,8 @@ HOME_LOCATION_NAME = "Home"
DEFAULT_NAME = "Weather"
ENTITY_ID_SENSOR_FORMAT = WEATHER_DOMAIN + ".smhi_{}"
+
+LOGGER = logging.getLogger(__package__)
+
+DEFAULT_SCAN_INTERVAL = timedelta(minutes=31)
+TIMEOUT = 10
diff --git a/homeassistant/components/smhi/coordinator.py b/homeassistant/components/smhi/coordinator.py
new file mode 100644
index 00000000000..511ba8b38d9
--- /dev/null
+++ b/homeassistant/components/smhi/coordinator.py
@@ -0,0 +1,63 @@
+"""DataUpdateCoordinator for the SMHI integration."""
+
+from __future__ import annotations
+
+import asyncio
+from dataclasses import dataclass
+
+from pysmhi import SMHIForecast, SmhiForecastException, SMHIPointForecast
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_LATITUDE, CONF_LOCATION, CONF_LONGITUDE
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import aiohttp_client
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+from .const import DEFAULT_SCAN_INTERVAL, DOMAIN, LOGGER, TIMEOUT
+
+type SMHIConfigEntry = ConfigEntry[SMHIDataUpdateCoordinator]
+
+
+@dataclass
+class SMHIForecastData:
+ """Dataclass for SMHI data."""
+
+ daily: list[SMHIForecast]
+ hourly: list[SMHIForecast]
+
+
+class SMHIDataUpdateCoordinator(DataUpdateCoordinator[SMHIForecastData]):
+ """A SMHI Data Update Coordinator."""
+
+ config_entry: SMHIConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: SMHIConfigEntry) -> None:
+ """Initialize the SMHI coordinator."""
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=DEFAULT_SCAN_INTERVAL,
+ )
+ self._smhi_api = SMHIPointForecast(
+ config_entry.data[CONF_LOCATION][CONF_LONGITUDE],
+ config_entry.data[CONF_LOCATION][CONF_LATITUDE],
+ session=aiohttp_client.async_get_clientsession(hass),
+ )
+
+ async def _async_update_data(self) -> SMHIForecastData:
+ """Fetch data from SMHI."""
+ try:
+ async with asyncio.timeout(TIMEOUT):
+ _forecast_daily = await self._smhi_api.async_get_daily_forecast()
+ _forecast_hourly = await self._smhi_api.async_get_hourly_forecast()
+ except SmhiForecastException as ex:
+ raise UpdateFailed(
+ "Failed to retrieve the forecast from the SMHI API"
+ ) from ex
+
+ return SMHIForecastData(
+ daily=_forecast_daily,
+ hourly=_forecast_hourly,
+ )
diff --git a/homeassistant/components/smhi/entity.py b/homeassistant/components/smhi/entity.py
new file mode 100644
index 00000000000..89dca3360ca
--- /dev/null
+++ b/homeassistant/components/smhi/entity.py
@@ -0,0 +1,41 @@
+"""Support for the Swedish weather institute weather base entities."""
+
+from __future__ import annotations
+
+from abc import abstractmethod
+
+from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+from .coordinator import SMHIDataUpdateCoordinator
+
+
+class SmhiWeatherBaseEntity(CoordinatorEntity[SMHIDataUpdateCoordinator]):
+ """Representation of a base weather entity."""
+
+ _attr_attribution = "Swedish weather institute (SMHI)"
+ _attr_has_entity_name = True
+ _attr_name = None
+
+ def __init__(
+ self,
+ latitude: str,
+ longitude: str,
+ coordinator: SMHIDataUpdateCoordinator,
+ ) -> None:
+ """Initialize the SMHI base weather entity."""
+ super().__init__(coordinator)
+ self._attr_unique_id = f"{latitude}, {longitude}"
+ self._attr_device_info = DeviceInfo(
+ entry_type=DeviceEntryType.SERVICE,
+ identifiers={(DOMAIN, f"{latitude}, {longitude}")},
+ manufacturer="SMHI",
+ model="v2",
+ configuration_url="http://opendata.smhi.se/apidocs/metfcst/parameters.html",
+ )
+ self.update_entity_data()
+
+ @abstractmethod
+ def update_entity_data(self) -> None:
+ """Refresh the entity data."""
diff --git a/homeassistant/components/smhi/weather.py b/homeassistant/components/smhi/weather.py
index 1707afa2fca..5faef04e03d 100644
--- a/homeassistant/components/smhi/weather.py
+++ b/homeassistant/components/smhi/weather.py
@@ -2,14 +2,10 @@
from __future__ import annotations
-import asyncio
from collections.abc import Mapping
-from datetime import datetime, timedelta
-import logging
from typing import Any, Final
-import aiohttp
-from pysmhi import SMHIForecast, SmhiForecastException, SMHIPointForecast
+from pysmhi import SMHIForecast
from homeassistant.components.weather import (
ATTR_CONDITION_CLEAR_NIGHT,
@@ -39,10 +35,9 @@ from homeassistant.components.weather import (
ATTR_FORECAST_TIME,
ATTR_FORECAST_WIND_BEARING,
Forecast,
- WeatherEntity,
+ SingleCoordinatorWeatherEntity,
WeatherEntityFeature,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_LATITUDE,
CONF_LOCATION,
@@ -53,16 +48,13 @@ from homeassistant.const import (
UnitOfSpeed,
UnitOfTemperature,
)
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers import aiohttp_client, sun
-from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.event import async_call_later
-from homeassistant.util import Throttle
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers import sun
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import ATTR_SMHI_THUNDER_PROBABILITY, DOMAIN, ENTITY_ID_SENSOR_FORMAT
-
-_LOGGER = logging.getLogger(__name__)
+from .const import ATTR_SMHI_THUNDER_PROBABILITY, ENTITY_ID_SENSOR_FORMAT
+from .coordinator import SMHIConfigEntry
+from .entity import SmhiWeatherBaseEntity
# Used to map condition from API results
CONDITION_CLASSES: Final[dict[str, list[int]]] = {
@@ -87,112 +79,70 @@ CONDITION_MAP = {
for cond_code in cond_codes
}
-TIMEOUT = 10
-# 5 minutes between retrying connect to API again
-RETRY_TIMEOUT = 5 * 60
-
-MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=31)
-
async def async_setup_entry(
hass: HomeAssistant,
- config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ config_entry: SMHIConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from map location."""
location = config_entry.data
- session = aiohttp_client.async_get_clientsession(hass)
+ coordinator = config_entry.runtime_data
entity = SmhiWeather(
location[CONF_LOCATION][CONF_LATITUDE],
location[CONF_LOCATION][CONF_LONGITUDE],
- session=session,
+ coordinator=coordinator,
)
entity.entity_id = ENTITY_ID_SENSOR_FORMAT.format(config_entry.title)
- async_add_entities([entity], True)
+ async_add_entities([entity])
-class SmhiWeather(WeatherEntity):
+class SmhiWeather(SmhiWeatherBaseEntity, SingleCoordinatorWeatherEntity):
"""Representation of a weather entity."""
- _attr_attribution = "Swedish weather institute (SMHI)"
_attr_native_temperature_unit = UnitOfTemperature.CELSIUS
_attr_native_visibility_unit = UnitOfLength.KILOMETERS
_attr_native_precipitation_unit = UnitOfPrecipitationDepth.MILLIMETERS
_attr_native_wind_speed_unit = UnitOfSpeed.METERS_PER_SECOND
_attr_native_pressure_unit = UnitOfPressure.HPA
-
- _attr_has_entity_name = True
- _attr_name = None
_attr_supported_features = (
WeatherEntityFeature.FORECAST_DAILY | WeatherEntityFeature.FORECAST_HOURLY
)
- def __init__(
- self,
- latitude: str,
- longitude: str,
- session: aiohttp.ClientSession,
- ) -> None:
- """Initialize the SMHI weather entity."""
- self._attr_unique_id = f"{latitude}, {longitude}"
- self._forecast_daily: list[SMHIForecast] | None = None
- self._forecast_hourly: list[SMHIForecast] | None = None
- self._fail_count = 0
- self._smhi_api = SMHIPointForecast(longitude, latitude, session=session)
- self._attr_device_info = DeviceInfo(
- entry_type=DeviceEntryType.SERVICE,
- identifiers={(DOMAIN, f"{latitude}, {longitude}")},
- manufacturer="SMHI",
- model="v2",
- configuration_url="http://opendata.smhi.se/apidocs/metfcst/parameters.html",
- )
+ def update_entity_data(self) -> None:
+ """Refresh the entity data."""
+ if daily_data := self.coordinator.data.daily:
+ self._attr_native_temperature = daily_data[0]["temperature"]
+ self._attr_humidity = daily_data[0]["humidity"]
+ self._attr_native_wind_speed = daily_data[0]["wind_speed"]
+ self._attr_wind_bearing = daily_data[0]["wind_direction"]
+ self._attr_native_visibility = daily_data[0]["visibility"]
+ self._attr_native_pressure = daily_data[0]["pressure"]
+ self._attr_native_wind_gust_speed = daily_data[0]["wind_gust"]
+ self._attr_cloud_coverage = daily_data[0]["total_cloud"]
+ self._attr_condition = CONDITION_MAP.get(daily_data[0]["symbol"])
+ if self._attr_condition == ATTR_CONDITION_SUNNY and not sun.is_up(
+ self.coordinator.hass
+ ):
+ self._attr_condition = ATTR_CONDITION_CLEAR_NIGHT
@property
def extra_state_attributes(self) -> Mapping[str, Any] | None:
"""Return additional attributes."""
- if self._forecast_daily:
+ if daily_data := self.coordinator.data.daily:
return {
- ATTR_SMHI_THUNDER_PROBABILITY: self._forecast_daily[0]["thunder"],
+ ATTR_SMHI_THUNDER_PROBABILITY: daily_data[0]["thunder"],
}
return None
- @Throttle(MIN_TIME_BETWEEN_UPDATES)
- async def async_update(self) -> None:
- """Refresh the forecast data from SMHI weather API."""
- try:
- async with asyncio.timeout(TIMEOUT):
- self._forecast_daily = await self._smhi_api.async_get_daily_forecast()
- self._forecast_hourly = await self._smhi_api.async_get_hourly_forecast()
- self._fail_count = 0
- except (TimeoutError, SmhiForecastException):
- _LOGGER.error("Failed to connect to SMHI API, retry in 5 minutes")
- self._fail_count += 1
- if self._fail_count < 3:
- async_call_later(self.hass, RETRY_TIMEOUT, self.retry_update)
- return
-
- if self._forecast_daily:
- self._attr_native_temperature = self._forecast_daily[0]["temperature"]
- self._attr_humidity = self._forecast_daily[0]["humidity"]
- self._attr_native_wind_speed = self._forecast_daily[0]["wind_speed"]
- self._attr_wind_bearing = self._forecast_daily[0]["wind_direction"]
- self._attr_native_visibility = self._forecast_daily[0]["visibility"]
- self._attr_native_pressure = self._forecast_daily[0]["pressure"]
- self._attr_native_wind_gust_speed = self._forecast_daily[0]["wind_gust"]
- self._attr_cloud_coverage = self._forecast_daily[0]["total_cloud"]
- self._attr_condition = CONDITION_MAP.get(self._forecast_daily[0]["symbol"])
- if self._attr_condition == ATTR_CONDITION_SUNNY and not sun.is_up(
- self.hass
- ):
- self._attr_condition = ATTR_CONDITION_CLEAR_NIGHT
- await self.async_update_listeners(("daily", "hourly"))
-
- async def retry_update(self, _: datetime) -> None:
- """Retry refresh weather forecast."""
- await self.async_update(no_throttle=True)
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
+ self.update_entity_data()
+ super()._handle_coordinator_update()
def _get_forecast_data(
self, forecast_data: list[SMHIForecast] | None
@@ -231,10 +181,10 @@ class SmhiWeather(WeatherEntity):
return data
- async def async_forecast_daily(self) -> list[Forecast] | None:
+ def _async_forecast_daily(self) -> list[Forecast] | None:
"""Service to retrieve the daily forecast."""
- return self._get_forecast_data(self._forecast_daily)
+ return self._get_forecast_data(self.coordinator.data.daily)
- async def async_forecast_hourly(self) -> list[Forecast] | None:
+ def _async_forecast_hourly(self) -> list[Forecast] | None:
"""Service to retrieve the hourly forecast."""
- return self._get_forecast_data(self._forecast_hourly)
+ return self._get_forecast_data(self.coordinator.data.hourly)
diff --git a/homeassistant/components/smlight/__init__.py b/homeassistant/components/smlight/__init__.py
index 11c6ffb73fb..8f3e675ef6b 100644
--- a/homeassistant/components/smlight/__init__.py
+++ b/homeassistant/components/smlight/__init__.py
@@ -2,16 +2,18 @@
from __future__ import annotations
-from dataclasses import dataclass
-
from pysmlight import Api2, Info, Radio
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .coordinator import SmDataUpdateCoordinator, SmFirmwareUpdateCoordinator
+from .coordinator import (
+ SmConfigEntry,
+ SmDataUpdateCoordinator,
+ SmFirmwareUpdateCoordinator,
+ SmlightData,
+)
PLATFORMS: list[Platform] = [
Platform.BINARY_SENSOR,
@@ -22,25 +24,12 @@ PLATFORMS: list[Platform] = [
]
-@dataclass(kw_only=True)
-class SmlightData:
- """Coordinator data class."""
-
- data: SmDataUpdateCoordinator
- firmware: SmFirmwareUpdateCoordinator
-
-
-type SmConfigEntry = ConfigEntry[SmlightData]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: SmConfigEntry) -> bool:
"""Set up SMLIGHT Zigbee from a config entry."""
client = Api2(host=entry.data[CONF_HOST], session=async_get_clientsession(hass))
- data_coordinator = SmDataUpdateCoordinator(hass, entry.data[CONF_HOST], client)
- firmware_coordinator = SmFirmwareUpdateCoordinator(
- hass, entry.data[CONF_HOST], client
- )
+ data_coordinator = SmDataUpdateCoordinator(hass, entry, client)
+ firmware_coordinator = SmFirmwareUpdateCoordinator(hass, entry, client)
await data_coordinator.async_config_entry_first_refresh()
await firmware_coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/smlight/binary_sensor.py b/homeassistant/components/smlight/binary_sensor.py
index b1aba3a52fe..ce3457ae81b 100644
--- a/homeassistant/components/smlight/binary_sensor.py
+++ b/homeassistant/components/smlight/binary_sensor.py
@@ -14,13 +14,12 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SCAN_INTERNET_INTERVAL
-from .coordinator import SmDataUpdateCoordinator
+from .coordinator import SmConfigEntry, SmDataUpdateCoordinator
from .entity import SmEntity
SCAN_INTERVAL = SCAN_INTERNET_INTERVAL
@@ -56,8 +55,8 @@ SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
- entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SMLIGHT sensor based on a config entry."""
coordinator = entry.runtime_data.data
diff --git a/homeassistant/components/smlight/button.py b/homeassistant/components/smlight/button.py
index d82034b87fb..5caf43b7cba 100644
--- a/homeassistant/components/smlight/button.py
+++ b/homeassistant/components/smlight/button.py
@@ -14,14 +14,13 @@ from homeassistant.components.button import (
ButtonEntity,
ButtonEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
-from .coordinator import SmDataUpdateCoordinator
+from .coordinator import SmConfigEntry, SmDataUpdateCoordinator
from .entity import SmEntity
_LOGGER = logging.getLogger(__name__)
@@ -65,8 +64,8 @@ ROUTER = SmButtonDescription(
async def async_setup_entry(
hass: HomeAssistant,
- entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SmConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SMLIGHT buttons based on a config entry."""
coordinator = entry.runtime_data.data
diff --git a/homeassistant/components/smlight/config_flow.py b/homeassistant/components/smlight/config_flow.py
index 667e6e2884b..fcfc364d983 100644
--- a/homeassistant/components/smlight/config_flow.py
+++ b/homeassistant/components/smlight/config_flow.py
@@ -77,12 +77,14 @@ class SmlightConfigFlow(ConfigFlow, domain=DOMAIN):
if user_input is not None:
try:
- info = await self.client.get_info()
-
- if info.model not in Devices:
- return self.async_abort(reason="unsupported_device")
-
if not await self._async_check_auth_required(user_input):
+ info = await self.client.get_info()
+ self._host = str(info.device_ip)
+ self._device_name = str(info.hostname)
+
+ if info.model not in Devices:
+ return self.async_abort(reason="unsupported_device")
+
return await self._async_complete_entry(user_input)
except SmlightConnectionError:
return self.async_abort(reason="cannot_connect")
diff --git a/homeassistant/components/smlight/coordinator.py b/homeassistant/components/smlight/coordinator.py
index 341c627afe5..5a118e7de15 100644
--- a/homeassistant/components/smlight/coordinator.py
+++ b/homeassistant/components/smlight/coordinator.py
@@ -4,14 +4,14 @@ from __future__ import annotations
from abc import abstractmethod
from dataclasses import dataclass
-from typing import TYPE_CHECKING
from pysmlight import Api2, Info, Sensors
from pysmlight.const import Settings, SettingsProp
from pysmlight.exceptions import SmlightAuthError, SmlightConnectionError
from pysmlight.models import FirmwareList
-from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers import issue_registry as ir
@@ -21,8 +21,13 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN, LOGGER, SCAN_FIRMWARE_INTERVAL, SCAN_INTERVAL
-if TYPE_CHECKING:
- from . import SmConfigEntry
+
+@dataclass(kw_only=True)
+class SmlightData:
+ """Coordinator data class."""
+
+ data: SmDataUpdateCoordinator
+ firmware: SmFirmwareUpdateCoordinator
@dataclass
@@ -42,17 +47,23 @@ class SmFwData:
zb_firmware: list[FirmwareList]
+type SmConfigEntry = ConfigEntry[SmlightData]
+
+
class SmBaseDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
"""Base Coordinator for SMLIGHT."""
config_entry: SmConfigEntry
- def __init__(self, hass: HomeAssistant, host: str, client: Api2) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: SmConfigEntry, client: Api2
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
LOGGER,
- name=f"{DOMAIN}_{host}",
+ config_entry=config_entry,
+ name=f"{DOMAIN}_{config_entry.data[CONF_HOST]}",
update_interval=SCAN_INTERVAL,
)
@@ -133,9 +144,11 @@ class SmDataUpdateCoordinator(SmBaseDataUpdateCoordinator[SmData]):
class SmFirmwareUpdateCoordinator(SmBaseDataUpdateCoordinator[SmFwData]):
"""Class to manage fetching SMLIGHT firmware update data from cloud."""
- def __init__(self, hass: HomeAssistant, host: str, client: Api2) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: SmConfigEntry, client: Api2
+ ) -> None:
"""Initialize the coordinator."""
- super().__init__(hass, host, client)
+ super().__init__(hass, config_entry, client)
self.update_interval = SCAN_FIRMWARE_INTERVAL
# only one update can run at a time (core or zibgee)
diff --git a/homeassistant/components/smlight/diagnostics.py b/homeassistant/components/smlight/diagnostics.py
index d303e5803bb..3812175e673 100644
--- a/homeassistant/components/smlight/diagnostics.py
+++ b/homeassistant/components/smlight/diagnostics.py
@@ -8,7 +8,7 @@ from pysmlight.const import Actions
from homeassistant.core import HomeAssistant
-from . import SmConfigEntry
+from .coordinator import SmConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/smlight/sensor.py b/homeassistant/components/smlight/sensor.py
index 1116b99f8c1..57a08d177d4 100644
--- a/homeassistant/components/smlight/sensor.py
+++ b/homeassistant/components/smlight/sensor.py
@@ -17,13 +17,12 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import EntityCategory, UnitOfInformation, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
-from . import SmConfigEntry
from .const import UPTIME_DEVIATION
-from .coordinator import SmDataUpdateCoordinator
+from .coordinator import SmConfigEntry, SmDataUpdateCoordinator
from .entity import SmEntity
@@ -124,7 +123,7 @@ UPTIME: list[SmSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: SmConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SMLIGHT sensor based on a config entry."""
coordinator = entry.runtime_data.data
diff --git a/homeassistant/components/smlight/strings.json b/homeassistant/components/smlight/strings.json
index 21ff5098d27..ca52f6fea38 100644
--- a/homeassistant/components/smlight/strings.json
+++ b/homeassistant/components/smlight/strings.json
@@ -2,7 +2,7 @@
"config": {
"step": {
"user": {
- "description": "Set up SMLIGHT Zigbee Integration",
+ "description": "Set up SMLIGHT Zigbee integration",
"data": {
"host": "[%key:common::config_flow::data::host%]"
},
@@ -111,7 +111,7 @@
"name": "Zigbee flash mode"
},
"reconnect_zigbee_router": {
- "name": "Reconnect zigbee router"
+ "name": "Reconnect Zigbee router"
}
},
"switch": {
diff --git a/homeassistant/components/smlight/switch.py b/homeassistant/components/smlight/switch.py
index 1c591e3dbe8..09d2714956c 100644
--- a/homeassistant/components/smlight/switch.py
+++ b/homeassistant/components/smlight/switch.py
@@ -17,10 +17,9 @@ from homeassistant.components.switch import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SmConfigEntry
-from .coordinator import SmDataUpdateCoordinator
+from .coordinator import SmConfigEntry, SmDataUpdateCoordinator
from .entity import SmEntity
_LOGGER = logging.getLogger(__name__)
@@ -68,7 +67,7 @@ SWITCHES: list[SmSwitchEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
entry: SmConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize switches for SLZB-06 device."""
coordinator = entry.runtime_data.data
diff --git a/homeassistant/components/smlight/update.py b/homeassistant/components/smlight/update.py
index 50a123345c6..10d142e6221 100644
--- a/homeassistant/components/smlight/update.py
+++ b/homeassistant/components/smlight/update.py
@@ -20,11 +20,11 @@ from homeassistant.components.update import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import SmConfigEntry, get_radio
+from . import get_radio
from .const import LOGGER
-from .coordinator import SmFirmwareUpdateCoordinator, SmFwData
+from .coordinator import SmConfigEntry, SmFirmwareUpdateCoordinator, SmFwData
from .entity import SmEntity
@@ -62,7 +62,9 @@ ZB_UPDATE_ENTITY = SmUpdateEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: SmConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: SmConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SMLIGHT update entities."""
coordinator = entry.runtime_data.firmware
diff --git a/homeassistant/components/sms/sensor.py b/homeassistant/components/sms/sensor.py
index 821200f68b1..46ee754a1f1 100644
--- a/homeassistant/components/sms/sensor.py
+++ b/homeassistant/components/sms/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, SIGNAL_STRENGTH_DECIBELS, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, GATEWAY, NETWORK_COORDINATOR, SIGNAL_COORDINATOR, SMS_GATEWAY
@@ -77,7 +77,7 @@ NETWORK_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all device sensors."""
sms_data = hass.data[DOMAIN][SMS_GATEWAY]
diff --git a/homeassistant/components/snapcast/__init__.py b/homeassistant/components/snapcast/__init__.py
index b853535b525..9c1602494e5 100644
--- a/homeassistant/components/snapcast/__init__.py
+++ b/homeassistant/components/snapcast/__init__.py
@@ -11,15 +11,14 @@ from .coordinator import SnapcastUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Snapcast from a config entry."""
- host = entry.data[CONF_HOST]
- port = entry.data[CONF_PORT]
- coordinator = SnapcastUpdateCoordinator(hass, host, port)
+ coordinator = SnapcastUpdateCoordinator(hass, entry)
try:
await coordinator.async_config_entry_first_refresh()
except OSError as ex:
raise ConfigEntryNotReady(
- f"Could not connect to Snapcast server at {host}:{port}"
+ "Could not connect to Snapcast server at "
+ f"{entry.data[CONF_HOST]}:{entry.data[CONF_PORT]}"
) from ex
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/snapcast/coordinator.py b/homeassistant/components/snapcast/coordinator.py
index 5bb9ae4e51f..4c2f0cb81b7 100644
--- a/homeassistant/components/snapcast/coordinator.py
+++ b/homeassistant/components/snapcast/coordinator.py
@@ -6,6 +6,8 @@ import logging
from snapcast.control.server import Snapserver
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, CONF_PORT
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -15,15 +17,20 @@ _LOGGER = logging.getLogger(__name__)
class SnapcastUpdateCoordinator(DataUpdateCoordinator[None]):
"""Data update coordinator for pushed data from Snapcast server."""
- def __init__(self, hass: HomeAssistant, host: str, port: int) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize coordinator."""
+ host = config_entry.data[CONF_HOST]
+ port = config_entry.data[CONF_PORT]
+
super().__init__(
hass,
logger=_LOGGER,
+ config_entry=config_entry,
name=f"{host}:{port}",
update_interval=None, # Disable update interval as server pushes
)
-
self._server = Snapserver(hass.loop, host, port, True)
self.last_update_success = False
diff --git a/homeassistant/components/snapcast/media_player.py b/homeassistant/components/snapcast/media_player.py
index 0ec27c1ad9c..5f011ca41ee 100644
--- a/homeassistant/components/snapcast/media_player.py
+++ b/homeassistant/components/snapcast/media_player.py
@@ -25,7 +25,7 @@ from homeassistant.helpers import (
entity_platform,
entity_registry as er,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_LATENCY,
@@ -73,7 +73,7 @@ def register_services() -> None:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the snapcast config entry."""
diff --git a/homeassistant/components/snips/strings.json b/homeassistant/components/snips/strings.json
index 724e1a86477..23b255b05a9 100644
--- a/homeassistant/components/snips/strings.json
+++ b/homeassistant/components/snips/strings.json
@@ -44,7 +44,7 @@
"fields": {
"can_be_enqueued": {
"name": "Can be enqueued",
- "description": "If True, session waits for an open session to end, if False session is dropped if one is running."
+ "description": "Whether the session should wait for an open session to end. Otherwise it is dropped if another session is already running."
},
"custom_data": {
"name": "[%key:component::snips::services::say::fields::custom_data::name%]",
diff --git a/homeassistant/components/snoo/__init__.py b/homeassistant/components/snoo/__init__.py
new file mode 100644
index 00000000000..54834bf58ce
--- /dev/null
+++ b/homeassistant/components/snoo/__init__.py
@@ -0,0 +1,69 @@
+"""The Happiest Baby Snoo integration."""
+
+from __future__ import annotations
+
+import asyncio
+import logging
+
+from python_snoo.exceptions import InvalidSnooAuth, SnooAuthException, SnooDeviceError
+from python_snoo.snoo import Snoo
+
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+from .coordinator import SnooConfigEntry, SnooCoordinator
+
+_LOGGER = logging.getLogger(__name__)
+
+PLATFORMS: list[Platform] = [
+ Platform.BINARY_SENSOR,
+ Platform.EVENT,
+ Platform.SELECT,
+ Platform.SENSOR,
+ Platform.SWITCH,
+]
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: SnooConfigEntry) -> bool:
+ """Set up Happiest Baby Snoo from a config entry."""
+
+ snoo = Snoo(
+ email=entry.data[CONF_USERNAME],
+ password=entry.data[CONF_PASSWORD],
+ clientsession=async_get_clientsession(hass),
+ )
+
+ try:
+ await snoo.authorize()
+ except (SnooAuthException, InvalidSnooAuth) as ex:
+ raise ConfigEntryNotReady from ex
+ try:
+ devices = await snoo.get_devices()
+ except SnooDeviceError as ex:
+ raise ConfigEntryNotReady from ex
+ coordinators: dict[str, SnooCoordinator] = {}
+ tasks = []
+ for device in devices:
+ coordinators[device.serialNumber] = SnooCoordinator(hass, device, snoo)
+ tasks.append(coordinators[device.serialNumber].setup())
+ await asyncio.gather(*tasks)
+ entry.runtime_data = coordinators
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+
+ return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: SnooConfigEntry) -> bool:
+ """Unload a config entry."""
+ disconnects = await asyncio.gather(
+ *(coordinator.snoo.disconnect() for coordinator in entry.runtime_data.values()),
+ return_exceptions=True,
+ )
+ for disconnect in disconnects:
+ if isinstance(disconnect, Exception):
+ _LOGGER.warning(
+ "Failed to disconnect a logger with exception: %s", disconnect
+ )
+ return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/snoo/binary_sensor.py b/homeassistant/components/snoo/binary_sensor.py
new file mode 100644
index 00000000000..3c91db5b86d
--- /dev/null
+++ b/homeassistant/components/snoo/binary_sensor.py
@@ -0,0 +1,70 @@
+"""Support for Snoo Binary Sensors."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from python_snoo.containers import SnooData
+
+from homeassistant.components.binary_sensor import (
+ BinarySensorDeviceClass,
+ BinarySensorEntity,
+ BinarySensorEntityDescription,
+ EntityCategory,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import SnooConfigEntry
+from .entity import SnooDescriptionEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SnooBinarySensorEntityDescription(BinarySensorEntityDescription):
+ """Describes a Snoo Binary Sensor."""
+
+ value_fn: Callable[[SnooData], bool]
+
+
+BINARY_SENSOR_DESCRIPTIONS: list[SnooBinarySensorEntityDescription] = [
+ SnooBinarySensorEntityDescription(
+ key="left_clip",
+ translation_key="left_clip",
+ value_fn=lambda data: data.left_safety_clip,
+ device_class=BinarySensorDeviceClass.CONNECTIVITY,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+ SnooBinarySensorEntityDescription(
+ key="right_clip",
+ translation_key="right_clip",
+ value_fn=lambda data: data.left_safety_clip,
+ device_class=BinarySensorDeviceClass.CONNECTIVITY,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SnooConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up Snoo device."""
+ coordinators = entry.runtime_data
+ async_add_entities(
+ SnooBinarySensor(coordinator, description)
+ for coordinator in coordinators.values()
+ for description in BINARY_SENSOR_DESCRIPTIONS
+ )
+
+
+class SnooBinarySensor(SnooDescriptionEntity, BinarySensorEntity):
+ """A Binary sensor using Snoo coordinator."""
+
+ entity_description: SnooBinarySensorEntityDescription
+
+ @property
+ def is_on(self) -> bool:
+ """Return true if the binary sensor is on."""
+ return self.entity_description.value_fn(self.coordinator.data)
diff --git a/homeassistant/components/snoo/config_flow.py b/homeassistant/components/snoo/config_flow.py
new file mode 100644
index 00000000000..986ef6a0071
--- /dev/null
+++ b/homeassistant/components/snoo/config_flow.py
@@ -0,0 +1,68 @@
+"""Config flow for the Happiest Baby Snoo integration."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+import jwt
+from python_snoo.exceptions import InvalidSnooAuth, SnooAuthException
+from python_snoo.snoo import Snoo
+import voluptuous as vol
+
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+from .const import DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+STEP_USER_DATA_SCHEMA = vol.Schema(
+ {
+ vol.Required(CONF_USERNAME): str,
+ vol.Required(CONF_PASSWORD): str,
+ }
+)
+
+
+class SnooConfigFlow(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow for Happiest Baby Snoo."""
+
+ VERSION = 1
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle the initial step."""
+ errors: dict[str, str] = {}
+ if user_input is not None:
+ hub = Snoo(
+ email=user_input[CONF_USERNAME],
+ password=user_input[CONF_PASSWORD],
+ clientsession=async_get_clientsession(self.hass),
+ )
+
+ try:
+ tokens = await hub.authorize()
+ except SnooAuthException:
+ errors["base"] = "cannot_connect"
+ except InvalidSnooAuth:
+ errors["base"] = "invalid_auth"
+ except Exception:
+ _LOGGER.exception("Unexpected exception %s")
+ errors["base"] = "unknown"
+ else:
+ user_uuid = jwt.decode(
+ tokens.aws_access, options={"verify_signature": False}
+ )["username"]
+ await self.async_set_unique_id(user_uuid)
+ self._abort_if_unique_id_configured()
+
+ return self.async_create_entry(
+ title=user_input[CONF_USERNAME], data=user_input
+ )
+
+ return self.async_show_form(
+ step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors
+ )
diff --git a/homeassistant/components/snoo/const.py b/homeassistant/components/snoo/const.py
new file mode 100644
index 00000000000..ff8afe25056
--- /dev/null
+++ b/homeassistant/components/snoo/const.py
@@ -0,0 +1,3 @@
+"""Constants for the Happiest Baby Snoo integration."""
+
+DOMAIN = "snoo"
diff --git a/homeassistant/components/snoo/coordinator.py b/homeassistant/components/snoo/coordinator.py
new file mode 100644
index 00000000000..bc06d20955c
--- /dev/null
+++ b/homeassistant/components/snoo/coordinator.py
@@ -0,0 +1,39 @@
+"""Support for Snoo Coordinators."""
+
+import logging
+
+from python_snoo.containers import SnooData, SnooDevice
+from python_snoo.snoo import Snoo
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
+
+type SnooConfigEntry = ConfigEntry[dict[str, SnooCoordinator]]
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class SnooCoordinator(DataUpdateCoordinator[SnooData]):
+ """Snoo coordinator."""
+
+ config_entry: SnooConfigEntry
+
+ def __init__(self, hass: HomeAssistant, device: SnooDevice, snoo: Snoo) -> None:
+ """Set up Snoo Coordinator."""
+ super().__init__(
+ hass,
+ name=device.name,
+ logger=_LOGGER,
+ )
+ self.device_unique_id = device.serialNumber
+ self.device = device
+ self.sensor_data_set: bool = False
+ self.snoo = snoo
+
+ async def setup(self) -> None:
+ """Perform setup needed on every coordintaor creation."""
+ await self.snoo.subscribe(self.device, self.async_set_updated_data)
+ # After we subscribe - get the status so that we have something to start with.
+ # We only need to do this once. The device will auto update otherwise.
+ await self.snoo.get_status(self.device)
diff --git a/homeassistant/components/snoo/entity.py b/homeassistant/components/snoo/entity.py
new file mode 100644
index 00000000000..25f54344674
--- /dev/null
+++ b/homeassistant/components/snoo/entity.py
@@ -0,0 +1,37 @@
+"""Base entity for the Snoo integration."""
+
+from __future__ import annotations
+
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity import EntityDescription
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+from .coordinator import SnooCoordinator
+
+
+class SnooDescriptionEntity(CoordinatorEntity[SnooCoordinator]):
+ """Defines an Snoo entity that uses a description."""
+
+ _attr_has_entity_name = True
+
+ def __init__(
+ self, coordinator: SnooCoordinator, description: EntityDescription
+ ) -> None:
+ """Initialize the Snoo entity."""
+ super().__init__(coordinator)
+ self.device = coordinator.device
+ self.entity_description = description
+ self._attr_unique_id = f"{coordinator.device_unique_id}_{description.key}"
+ self._attr_device_info = DeviceInfo(
+ identifiers={(DOMAIN, coordinator.device_unique_id)},
+ name=self.device.name,
+ manufacturer="Happiest Baby",
+ model="Snoo",
+ serial_number=self.device.serialNumber,
+ )
+
+ @property
+ def available(self) -> bool:
+ """Return if entity is available."""
+ return self.coordinator.data is not None and super().available
diff --git a/homeassistant/components/snoo/event.py b/homeassistant/components/snoo/event.py
new file mode 100644
index 00000000000..5932bfd9862
--- /dev/null
+++ b/homeassistant/components/snoo/event.py
@@ -0,0 +1,62 @@
+"""Support for Snoo Events."""
+
+from homeassistant.components.event import EventEntity, EventEntityDescription
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import SnooConfigEntry
+from .entity import SnooDescriptionEntity
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SnooConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up Snoo device."""
+ coordinators = entry.runtime_data
+ async_add_entities(
+ SnooEvent(
+ coordinator,
+ EventEntityDescription(
+ key="event",
+ translation_key="event",
+ event_types=[
+ "timer",
+ "cry",
+ "command",
+ "safety_clip",
+ "long_activity_press",
+ "activity",
+ "power",
+ "status_requested",
+ "sticky_white_noise_updated",
+ ],
+ ),
+ )
+ for coordinator in coordinators.values()
+ )
+
+
+class SnooEvent(SnooDescriptionEntity, EventEntity):
+ """A event using Snoo coordinator."""
+
+ @callback
+ def _async_handle_event(self) -> None:
+ """Handle the demo button event."""
+ self._trigger_event(
+ self.coordinator.data.event.value,
+ )
+ self.async_write_ha_state()
+
+ async def async_added_to_hass(self) -> None:
+ """Add Event."""
+ await super().async_added_to_hass()
+ if self.coordinator.data:
+ # If we were able to get data on startup - set it
+ # Otherwise, it will update when the coordinator gets data.
+ self._async_handle_event()
+
+ def _handle_coordinator_update(self) -> None:
+ self._async_handle_event()
+ return super()._handle_coordinator_update()
diff --git a/homeassistant/components/snoo/manifest.json b/homeassistant/components/snoo/manifest.json
new file mode 100644
index 00000000000..4084a7e3e79
--- /dev/null
+++ b/homeassistant/components/snoo/manifest.json
@@ -0,0 +1,11 @@
+{
+ "domain": "snoo",
+ "name": "Happiest Baby Snoo",
+ "codeowners": ["@Lash-L"],
+ "config_flow": true,
+ "documentation": "https://www.home-assistant.io/integrations/snoo",
+ "iot_class": "cloud_push",
+ "loggers": ["snoo"],
+ "quality_scale": "bronze",
+ "requirements": ["python-snoo==0.6.4"]
+}
diff --git a/homeassistant/components/snoo/quality_scale.yaml b/homeassistant/components/snoo/quality_scale.yaml
new file mode 100644
index 00000000000..f10bccb131a
--- /dev/null
+++ b/homeassistant/components/snoo/quality_scale.yaml
@@ -0,0 +1,72 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: |
+ This integration does not provide additional actions.
+ appropriate-polling:
+ status: exempt
+ comment: |
+ This integration does not poll.
+ brands: done
+ common-modules:
+ status: done
+ comment: |
+ There are no common patterns currenty.
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: |
+ This integration does not provide additional actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup: done
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions: todo
+ config-entry-unloading: done
+ docs-configuration-parameters: todo
+ docs-installation-parameters: todo
+ entity-unavailable: todo
+ integration-owner: done
+ log-when-unavailable: todo
+ parallel-updates: todo
+ reauthentication-flow: todo
+ test-coverage: todo
+
+ # Gold
+ devices: done
+ diagnostics: todo
+ discovery-update-info: todo
+ discovery: todo
+ docs-data-update: todo
+ docs-examples: todo
+ docs-known-limitations: todo
+ docs-supported-devices: todo
+ docs-supported-functions: todo
+ docs-troubleshooting: todo
+ docs-use-cases: todo
+ dynamic-devices: todo
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default: todo
+ entity-translations: done
+ exception-translations: todo
+ icon-translations: todo
+ reconfiguration-flow: todo
+ repair-issues: todo
+ stale-devices: todo
+
+ # Platinum
+ async-dependency: done
+ inject-websession: done
+ strict-typing: todo
diff --git a/homeassistant/components/snoo/select.py b/homeassistant/components/snoo/select.py
new file mode 100644
index 00000000000..44624ed1a2d
--- /dev/null
+++ b/homeassistant/components/snoo/select.py
@@ -0,0 +1,78 @@
+"""Support for Snoo Select."""
+
+from __future__ import annotations
+
+from collections.abc import Awaitable, Callable
+from dataclasses import dataclass
+
+from python_snoo.containers import SnooData, SnooDevice, SnooLevels
+from python_snoo.exceptions import SnooCommandException
+from python_snoo.snoo import Snoo
+
+from homeassistant.components.select import SelectEntity, SelectEntityDescription
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .const import DOMAIN
+from .coordinator import SnooConfigEntry
+from .entity import SnooDescriptionEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SnooSelectEntityDescription(SelectEntityDescription):
+ """Describes a Snoo Select."""
+
+ value_fn: Callable[[SnooData], str]
+ set_value_fn: Callable[[Snoo, SnooDevice, str], Awaitable[None]]
+
+
+SELECT_DESCRIPTIONS: list[SnooSelectEntityDescription] = [
+ SnooSelectEntityDescription(
+ key="intensity",
+ translation_key="intensity",
+ value_fn=lambda data: data.state_machine.level.name,
+ set_value_fn=lambda snoo_api, device, state: snoo_api.set_level(
+ device, SnooLevels[state]
+ ),
+ options=[level.name for level in SnooLevels],
+ ),
+]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SnooConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up Snoo device."""
+ coordinators = entry.runtime_data
+ async_add_entities(
+ SnooSelect(coordinator, description)
+ for coordinator in coordinators.values()
+ for description in SELECT_DESCRIPTIONS
+ )
+
+
+class SnooSelect(SnooDescriptionEntity, SelectEntity):
+ """A sensor using Snoo coordinator."""
+
+ entity_description: SnooSelectEntityDescription
+
+ @property
+ def current_option(self) -> str | None:
+ """Return the selected entity option to represent the entity state."""
+ return self.entity_description.value_fn(self.coordinator.data)
+
+ async def async_select_option(self, option: str) -> None:
+ """Change the selected option."""
+ try:
+ await self.entity_description.set_value_fn(
+ self.coordinator.snoo, self.device, option
+ )
+ except SnooCommandException as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="select_failed",
+ translation_placeholders={"name": str(self.name), "option": option},
+ ) from err
diff --git a/homeassistant/components/snoo/sensor.py b/homeassistant/components/snoo/sensor.py
new file mode 100644
index 00000000000..e45b2b88592
--- /dev/null
+++ b/homeassistant/components/snoo/sensor.py
@@ -0,0 +1,71 @@
+"""Support for Snoo Sensors."""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from python_snoo.containers import SnooData, SnooStates
+
+from homeassistant.components.sensor import (
+ EntityCategory,
+ SensorDeviceClass,
+ SensorEntity,
+ SensorEntityDescription,
+ StateType,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import SnooConfigEntry
+from .entity import SnooDescriptionEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SnooSensorEntityDescription(SensorEntityDescription):
+ """Describes a Snoo sensor."""
+
+ value_fn: Callable[[SnooData], StateType]
+
+
+SENSOR_DESCRIPTIONS: list[SnooSensorEntityDescription] = [
+ SnooSensorEntityDescription(
+ key="state",
+ translation_key="state",
+ value_fn=lambda data: data.state_machine.state.name,
+ device_class=SensorDeviceClass.ENUM,
+ options=[e.name for e in SnooStates],
+ ),
+ SnooSensorEntityDescription(
+ key="time_left",
+ translation_key="time_left",
+ value_fn=lambda data: data.state_machine.time_left_timestamp,
+ device_class=SensorDeviceClass.TIMESTAMP,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ ),
+]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SnooConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up Snoo device."""
+ coordinators = entry.runtime_data
+ async_add_entities(
+ SnooSensor(coordinator, description)
+ for coordinator in coordinators.values()
+ for description in SENSOR_DESCRIPTIONS
+ )
+
+
+class SnooSensor(SnooDescriptionEntity, SensorEntity):
+ """A sensor using Snoo coordinator."""
+
+ entity_description: SnooSensorEntityDescription
+
+ @property
+ def native_value(self) -> StateType:
+ """Return the value reported by the sensor."""
+ return self.entity_description.value_fn(self.coordinator.data)
diff --git a/homeassistant/components/snoo/strings.json b/homeassistant/components/snoo/strings.json
new file mode 100644
index 00000000000..f7cf6a4820b
--- /dev/null
+++ b/homeassistant/components/snoo/strings.json
@@ -0,0 +1,104 @@
+{
+ "config": {
+ "step": {
+ "user": {
+ "data": {
+ "username": "[%key:common::config_flow::data::username%]",
+ "password": "[%key:common::config_flow::data::password%]"
+ },
+ "data_description": {
+ "username": "Your Snoo username or email",
+ "password": "Your Snoo password"
+ }
+ }
+ },
+ "error": {
+ "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
+ "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
+ "unknown": "[%key:common::config_flow::error::unknown%]"
+ },
+ "abort": {
+ "already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
+ }
+ },
+ "exceptions": {
+ "select_failed": {
+ "message": "Error while updating {name} to {option}"
+ },
+ "switch_on_failed": {
+ "message": "Turning {name} on failed"
+ },
+ "switch_off_failed": {
+ "message": "Turning {name} off failed"
+ }
+ },
+ "entity": {
+ "binary_sensor": {
+ "left_clip": {
+ "name": "Left safety clip"
+ },
+ "right_clip": {
+ "name": "Right safety clip"
+ }
+ },
+ "event": {
+ "event": {
+ "name": "Snoo event",
+ "state_attributes": {
+ "event_type": {
+ "state": {
+ "timer": "Timer",
+ "cry": "Cry",
+ "command": "Command sent",
+ "safety_clip": "Safety clip changed",
+ "long_activity_press": "Long activity press",
+ "activity": "Activity press",
+ "power": "Power button pressed",
+ "status_requested": "Status requested",
+ "sticky_white_noise_updated": "Sleepytime sounds updated"
+ }
+ }
+ }
+ }
+ },
+ "sensor": {
+ "state": {
+ "name": "State",
+ "state": {
+ "baseline": "Baseline",
+ "level1": "Level 1",
+ "level2": "Level 2",
+ "level3": "Level 3",
+ "level4": "Level 4",
+ "stop": "Stopped",
+ "pretimeout": "Pre-timeout",
+ "timeout": "Timeout"
+ }
+ },
+ "time_left": {
+ "name": "Time left"
+ }
+ },
+ "select": {
+ "intensity": {
+ "name": "Intensity",
+ "state": {
+ "baseline": "[%key:component::snoo::entity::sensor::state::state::baseline%]",
+ "level1": "[%key:component::snoo::entity::sensor::state::state::level1%]",
+ "level2": "[%key:component::snoo::entity::sensor::state::state::level2%]",
+ "level3": "[%key:component::snoo::entity::sensor::state::state::level3%]",
+ "level4": "[%key:component::snoo::entity::sensor::state::state::level4%]",
+ "stop": "[%key:component::snoo::entity::sensor::state::state::stop%]"
+ }
+ }
+ },
+ "switch": {
+ "sticky_white_noise": {
+ "name": "Sleepytime sounds"
+ },
+ "hold": {
+ "name": "Level lock"
+ }
+ }
+ }
+}
diff --git a/homeassistant/components/snoo/switch.py b/homeassistant/components/snoo/switch.py
new file mode 100644
index 00000000000..2ed322d5f6b
--- /dev/null
+++ b/homeassistant/components/snoo/switch.py
@@ -0,0 +1,105 @@
+"""Support for Snoo Switches."""
+
+from __future__ import annotations
+
+from collections.abc import Awaitable, Callable
+from dataclasses import dataclass
+from typing import Any
+
+from python_snoo.containers import SnooData, SnooDevice
+from python_snoo.exceptions import SnooCommandException
+from python_snoo.snoo import Snoo
+
+from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .const import DOMAIN
+from .coordinator import SnooConfigEntry
+from .entity import SnooDescriptionEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SnooSwitchEntityDescription(SwitchEntityDescription):
+ """Describes a Snoo sensor."""
+
+ value_fn: Callable[[SnooData], bool]
+ set_value_fn: Callable[[Snoo, SnooDevice, SnooData, bool], Awaitable[None]]
+
+
+BINARY_SENSOR_DESCRIPTIONS: list[SnooSwitchEntityDescription] = [
+ SnooSwitchEntityDescription(
+ key="sticky_white_noise",
+ translation_key="sticky_white_noise",
+ value_fn=lambda data: data.state_machine.sticky_white_noise == "on",
+ set_value_fn=lambda snoo_api, device, _, state: snoo_api.set_sticky_white_noise(
+ device, state
+ ),
+ ),
+ SnooSwitchEntityDescription(
+ key="hold",
+ translation_key="hold",
+ value_fn=lambda data: data.state_machine.hold == "on",
+ set_value_fn=lambda snoo_api, device, data, state: snoo_api.set_level(
+ device, data.state_machine.level, state
+ ),
+ ),
+]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SnooConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up Snoo device."""
+ coordinators = entry.runtime_data
+ async_add_entities(
+ SnooSwitch(coordinator, description)
+ for coordinator in coordinators.values()
+ for description in BINARY_SENSOR_DESCRIPTIONS
+ )
+
+
+class SnooSwitch(SnooDescriptionEntity, SwitchEntity):
+ """A switch using Snoo coordinator."""
+
+ entity_description: SnooSwitchEntityDescription
+
+ @property
+ def is_on(self) -> bool | None:
+ """Return True if entity is on."""
+ return self.entity_description.value_fn(self.coordinator.data)
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the entity on."""
+ try:
+ await self.entity_description.set_value_fn(
+ self.coordinator.snoo,
+ self.coordinator.device,
+ self.coordinator.data,
+ True,
+ )
+ except SnooCommandException as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="switch_on_failed",
+ translation_placeholders={"name": str(self.name), "status": "on"},
+ ) from err
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the entity off."""
+ try:
+ await self.entity_description.set_value_fn(
+ self.coordinator.snoo,
+ self.coordinator.device,
+ self.coordinator.data,
+ False,
+ )
+ except SnooCommandException as err:
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="switch_off_failed",
+ translation_placeholders={"name": str(self.name), "status": "off"},
+ ) from err
diff --git a/homeassistant/components/snooz/fan.py b/homeassistant/components/snooz/fan.py
index bfe773b4780..ce804450cab 100644
--- a/homeassistant/components/snooz/fan.py
+++ b/homeassistant/components/snooz/fan.py
@@ -23,7 +23,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_platform
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import (
@@ -38,7 +38,9 @@ from .models import SnoozConfigurationData
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Snooz device from a config entry."""
diff --git a/homeassistant/components/snooz/strings.json b/homeassistant/components/snooz/strings.json
index 94ca434e589..ca252b2117c 100644
--- a/homeassistant/components/snooz/strings.json
+++ b/homeassistant/components/snooz/strings.json
@@ -27,25 +27,25 @@
"services": {
"transition_on": {
"name": "Transition on",
- "description": "Transitions to a target volume level over time.",
+ "description": "Transitions the volume level over a specified duration. If the device is powered off, the transition will start at the lowest volume level.",
"fields": {
"duration": {
"name": "Transition duration",
- "description": "Time it takes to reach the target volume level."
+ "description": "Time to transition to the target volume."
},
"volume": {
"name": "Target volume",
- "description": "If not specified, the volume level is read from the device."
+ "description": "Relative volume level. If not specified, the setting on the device is used."
}
}
},
"transition_off": {
"name": "Transition off",
- "description": "Transitions volume off over time.",
+ "description": "Transitions the volume level to the lowest setting over a specified duration, then powers off the device.",
"fields": {
"duration": {
"name": "[%key:component::snooz::services::transition_on::fields::duration::name%]",
- "description": "Time it takes to turn off."
+ "description": "Time to complete the transition."
}
}
}
diff --git a/homeassistant/components/solaredge/coordinator.py b/homeassistant/components/solaredge/coordinator.py
index d37cf355fce..44f015eedeb 100644
--- a/homeassistant/components/solaredge/coordinator.py
+++ b/homeassistant/components/solaredge/coordinator.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from abc import ABC, abstractmethod
from datetime import date, datetime, timedelta
-from typing import Any
+from typing import TYPE_CHECKING, Any
from aiosolaredge import SolarEdge
from stringcase import snakecase
@@ -21,13 +21,22 @@ from .const import (
POWER_FLOW_UPDATE_DELAY,
)
+if TYPE_CHECKING:
+ from .types import SolarEdgeConfigEntry
+
class SolarEdgeDataService(ABC):
"""Get and update the latest data."""
coordinator: DataUpdateCoordinator[None]
- def __init__(self, hass: HomeAssistant, api: SolarEdge, site_id: str) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SolarEdgeConfigEntry,
+ api: SolarEdge,
+ site_id: str,
+ ) -> None:
"""Initialize the data object."""
self.api = api
self.site_id = site_id
@@ -36,6 +45,7 @@ class SolarEdgeDataService(ABC):
self.attributes: dict[str, Any] = {}
self.hass = hass
+ self.config_entry = config_entry
@callback
def async_setup(self) -> None:
@@ -43,6 +53,7 @@ class SolarEdgeDataService(ABC):
self.coordinator = DataUpdateCoordinator(
self.hass,
LOGGER,
+ config_entry=self.config_entry,
name=str(self),
update_method=self.async_update_data,
update_interval=self.update_interval,
@@ -174,9 +185,15 @@ class SolarEdgeInventoryDataService(SolarEdgeDataService):
class SolarEdgeEnergyDetailsService(SolarEdgeDataService):
"""Get and update the latest power flow data."""
- def __init__(self, hass: HomeAssistant, api: SolarEdge, site_id: str) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SolarEdgeConfigEntry,
+ api: SolarEdge,
+ site_id: str,
+ ) -> None:
"""Initialize the power flow data service."""
- super().__init__(hass, api, site_id)
+ super().__init__(hass, config_entry, api, site_id)
self.unit = None
@@ -234,9 +251,15 @@ class SolarEdgeEnergyDetailsService(SolarEdgeDataService):
class SolarEdgePowerFlowDataService(SolarEdgeDataService):
"""Get and update the latest power flow data."""
- def __init__(self, hass: HomeAssistant, api: SolarEdge, site_id: str) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SolarEdgeConfigEntry,
+ api: SolarEdge,
+ site_id: str,
+ ) -> None:
"""Initialize the power flow data service."""
- super().__init__(hass, api, site_id)
+ super().__init__(hass, config_entry, api, site_id)
self.unit = None
diff --git a/homeassistant/components/solaredge/sensor.py b/homeassistant/components/solaredge/sensor.py
index 4b2398d15c2..acb86f875c9 100644
--- a/homeassistant/components/solaredge/sensor.py
+++ b/homeassistant/components/solaredge/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import PERCENTAGE, UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -201,12 +201,12 @@ SENSOR_TYPES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: SolarEdgeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add an solarEdge entry."""
# Add the needed sensors to hass
api = entry.runtime_data[DATA_API_CLIENT]
- sensor_factory = SolarEdgeSensorFactory(hass, entry.data[CONF_SITE_ID], api)
+ sensor_factory = SolarEdgeSensorFactory(hass, entry, entry.data[CONF_SITE_ID], api)
for service in sensor_factory.all_services:
service.async_setup()
await service.coordinator.async_refresh()
@@ -222,14 +222,20 @@ async def async_setup_entry(
class SolarEdgeSensorFactory:
"""Factory which creates sensors based on the sensor_key."""
- def __init__(self, hass: HomeAssistant, site_id: str, api: SolarEdge) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SolarEdgeConfigEntry,
+ site_id: str,
+ api: SolarEdge,
+ ) -> None:
"""Initialize the factory."""
- details = SolarEdgeDetailsDataService(hass, api, site_id)
- overview = SolarEdgeOverviewDataService(hass, api, site_id)
- inventory = SolarEdgeInventoryDataService(hass, api, site_id)
- flow = SolarEdgePowerFlowDataService(hass, api, site_id)
- energy = SolarEdgeEnergyDetailsService(hass, api, site_id)
+ details = SolarEdgeDetailsDataService(hass, config_entry, api, site_id)
+ overview = SolarEdgeOverviewDataService(hass, config_entry, api, site_id)
+ inventory = SolarEdgeInventoryDataService(hass, config_entry, api, site_id)
+ flow = SolarEdgePowerFlowDataService(hass, config_entry, api, site_id)
+ energy = SolarEdgeEnergyDetailsService(hass, config_entry, api, site_id)
self.all_services = (details, overview, inventory, flow, energy)
diff --git a/homeassistant/components/solarlog/__init__.py b/homeassistant/components/solarlog/__init__.py
index 5937c8a496d..7ad1ec8e547 100644
--- a/homeassistant/components/solarlog/__init__.py
+++ b/homeassistant/components/solarlog/__init__.py
@@ -2,18 +2,16 @@
import logging
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from .const import CONF_HAS_PWD
-from .coordinator import SolarLogCoordinator
+from .coordinator import SolarlogConfigEntry, SolarLogCoordinator
_LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.SENSOR]
-type SolarlogConfigEntry = ConfigEntry[SolarLogCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: SolarlogConfigEntry) -> bool:
diff --git a/homeassistant/components/solarlog/coordinator.py b/homeassistant/components/solarlog/coordinator.py
index bf2bc849111..48ebeece1ba 100644
--- a/homeassistant/components/solarlog/coordinator.py
+++ b/homeassistant/components/solarlog/coordinator.py
@@ -5,7 +5,6 @@ from __future__ import annotations
from collections.abc import Callable
from datetime import timedelta
import logging
-from typing import TYPE_CHECKING
from urllib.parse import ParseResult, urlparse
from solarlog_cli.solarlog_connector import SolarLogConnector
@@ -16,6 +15,7 @@ from solarlog_cli.solarlog_exceptions import (
)
from solarlog_cli.solarlog_models import SolarlogData
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -28,30 +28,35 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
-if TYPE_CHECKING:
- from . import SolarlogConfigEntry
+type SolarlogConfigEntry = ConfigEntry[SolarLogCoordinator]
class SolarLogCoordinator(DataUpdateCoordinator[SolarlogData]):
"""Get and update the latest data."""
- def __init__(self, hass: HomeAssistant, entry: SolarlogConfigEntry) -> None:
+ config_entry: SolarlogConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: SolarlogConfigEntry) -> None:
"""Initialize the data object."""
super().__init__(
- hass, _LOGGER, name="SolarLog", update_interval=timedelta(seconds=60)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name="SolarLog",
+ update_interval=timedelta(seconds=60),
)
self.new_device_callbacks: list[Callable[[int], None]] = []
self._devices_last_update: set[tuple[int, str]] = set()
- host_entry = entry.data[CONF_HOST]
- password = entry.data.get("password", "")
+ host_entry = config_entry.data[CONF_HOST]
+ password = config_entry.data.get("password", "")
url = urlparse(host_entry, "http")
netloc = url.netloc or url.path
path = url.path if url.netloc else ""
url = ParseResult("http", netloc, path, *url[3:])
- self.unique_id = entry.entry_id
+ self.unique_id = config_entry.entry_id
self.host = url.geturl()
self.solarlog = SolarLogConnector(
@@ -70,7 +75,7 @@ class SolarLogCoordinator(DataUpdateCoordinator[SolarlogData]):
await self.solarlog.test_extended_data_available()
if logged_in or await self.solarlog.test_extended_data_available():
device_list = await self.solarlog.update_device_list()
- self.solarlog.set_enabled_devices({key: True for key in device_list})
+ self.solarlog.set_enabled_devices(dict.fromkeys(device_list, True))
async def _async_update_data(self) -> SolarlogData:
"""Update the data from the SolarLog device."""
diff --git a/homeassistant/components/solarlog/diagnostics.py b/homeassistant/components/solarlog/diagnostics.py
index 02f6c96edc2..c99222542ea 100644
--- a/homeassistant/components/solarlog/diagnostics.py
+++ b/homeassistant/components/solarlog/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
-from . import SolarlogConfigEntry
+from .coordinator import SolarlogConfigEntry
TO_REDACT = [
CONF_HOST,
diff --git a/homeassistant/components/solarlog/sensor.py b/homeassistant/components/solarlog/sensor.py
index bcff5d57e1b..c4bb119c006 100644
--- a/homeassistant/components/solarlog/sensor.py
+++ b/homeassistant/components/solarlog/sensor.py
@@ -21,10 +21,10 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import SolarlogConfigEntry
+from .coordinator import SolarlogConfigEntry
from .entity import SolarLogCoordinatorEntity, SolarLogInverterEntity
@@ -276,7 +276,7 @@ INVERTER_SENSOR_TYPES: tuple[SolarLogInverterSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SolarlogConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add solarlog entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/solax/sensor.py b/homeassistant/components/solax/sensor.py
index 6ca0bac0c38..1cdec0389fe 100644
--- a/homeassistant/components/solax/sensor.py
+++ b/homeassistant/components/solax/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import SolaxConfigEntry
@@ -89,7 +89,7 @@ SENSOR_DESCRIPTIONS: dict[tuple[Units, bool], SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: SolaxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Entry setup."""
api = entry.runtime_data.api
diff --git a/homeassistant/components/soma/cover.py b/homeassistant/components/soma/cover.py
index e64fee00f16..15aa21b1f48 100644
--- a/homeassistant/components/soma/cover.py
+++ b/homeassistant/components/soma/cover.py
@@ -14,7 +14,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import API, DEVICES, DOMAIN
from .entity import SomaEntity
@@ -24,7 +24,7 @@ from .utils import is_api_response_success
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Soma cover platform."""
diff --git a/homeassistant/components/soma/entity.py b/homeassistant/components/soma/entity.py
index f9824d107b1..4b2fcee5405 100644
--- a/homeassistant/components/soma/entity.py
+++ b/homeassistant/components/soma/entity.py
@@ -71,7 +71,7 @@ class SomaEntity(Entity):
self.api_is_available = True
@property
- def available(self):
+ def available(self) -> bool:
"""Return true if the last API commands returned successfully."""
return self.is_available
diff --git a/homeassistant/components/soma/sensor.py b/homeassistant/components/soma/sensor.py
index 806886009f3..839f28e9a65 100644
--- a/homeassistant/components/soma/sensor.py
+++ b/homeassistant/components/soma/sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle
from .const import API, DEVICES, DOMAIN
@@ -18,7 +18,7 @@ MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=30)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Soma sensor platform."""
diff --git a/homeassistant/components/somfy_mylink/cover.py b/homeassistant/components/somfy_mylink/cover.py
index 8c64e58362b..5b888ea4b96 100644
--- a/homeassistant/components/somfy_mylink/cover.py
+++ b/homeassistant/components/somfy_mylink/cover.py
@@ -7,7 +7,7 @@ from homeassistant.components.cover import CoverDeviceClass, CoverEntity, CoverS
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import (
@@ -29,7 +29,7 @@ MYLINK_COVER_TYPE_TO_DEVICE_CLASS = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Discover and configure Somfy covers."""
reversed_target_ids = config_entry.options.get(CONF_REVERSED_TARGET_IDS, {})
diff --git a/homeassistant/components/sonarr/__init__.py b/homeassistant/components/sonarr/__init__.py
index 7718ff799f5..960227ff0da 100644
--- a/homeassistant/components/sonarr/__init__.py
+++ b/homeassistant/components/sonarr/__init__.py
@@ -67,13 +67,19 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
)
entry.async_on_unload(entry.add_update_listener(_async_update_listener))
coordinators: dict[str, SonarrDataUpdateCoordinator[Any]] = {
- "upcoming": CalendarDataUpdateCoordinator(hass, host_configuration, sonarr),
- "commands": CommandsDataUpdateCoordinator(hass, host_configuration, sonarr),
- "diskspace": DiskSpaceDataUpdateCoordinator(hass, host_configuration, sonarr),
- "queue": QueueDataUpdateCoordinator(hass, host_configuration, sonarr),
- "series": SeriesDataUpdateCoordinator(hass, host_configuration, sonarr),
- "status": StatusDataUpdateCoordinator(hass, host_configuration, sonarr),
- "wanted": WantedDataUpdateCoordinator(hass, host_configuration, sonarr),
+ "upcoming": CalendarDataUpdateCoordinator(
+ hass, entry, host_configuration, sonarr
+ ),
+ "commands": CommandsDataUpdateCoordinator(
+ hass, entry, host_configuration, sonarr
+ ),
+ "diskspace": DiskSpaceDataUpdateCoordinator(
+ hass, entry, host_configuration, sonarr
+ ),
+ "queue": QueueDataUpdateCoordinator(hass, entry, host_configuration, sonarr),
+ "series": SeriesDataUpdateCoordinator(hass, entry, host_configuration, sonarr),
+ "status": StatusDataUpdateCoordinator(hass, entry, host_configuration, sonarr),
+ "wanted": WantedDataUpdateCoordinator(hass, entry, host_configuration, sonarr),
}
# Temporary, until we add diagnostic entities
_version = None
diff --git a/homeassistant/components/sonarr/coordinator.py b/homeassistant/components/sonarr/coordinator.py
index 25fc736212b..a73ef838590 100644
--- a/homeassistant/components/sonarr/coordinator.py
+++ b/homeassistant/components/sonarr/coordinator.py
@@ -48,6 +48,7 @@ class SonarrDataUpdateCoordinator(DataUpdateCoordinator[SonarrDataT]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
host_configuration: PyArrHostConfiguration,
api_client: SonarrClient,
) -> None:
@@ -55,6 +56,7 @@ class SonarrDataUpdateCoordinator(DataUpdateCoordinator[SonarrDataT]):
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/sonarr/sensor.py b/homeassistant/components/sonarr/sensor.py
index fa7d0aa7756..983ac76d93e 100644
--- a/homeassistant/components/sonarr/sensor.py
+++ b/homeassistant/components/sonarr/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -90,7 +90,6 @@ SENSOR_TYPES: dict[str, SonarrSensorEntityDescription[Any]] = {
"commands": SonarrSensorEntityDescription[list[Command]](
key="commands",
translation_key="commands",
- native_unit_of_measurement="Commands",
entity_registry_enabled_default=False,
value_fn=len,
attributes_fn=lambda data: {c.name: c.status for c in data},
@@ -107,7 +106,6 @@ SENSOR_TYPES: dict[str, SonarrSensorEntityDescription[Any]] = {
"queue": SonarrSensorEntityDescription[SonarrQueue](
key="queue",
translation_key="queue",
- native_unit_of_measurement="Episodes",
entity_registry_enabled_default=False,
value_fn=lambda data: data.totalRecords,
attributes_fn=get_queue_attr,
@@ -115,7 +113,6 @@ SENSOR_TYPES: dict[str, SonarrSensorEntityDescription[Any]] = {
"series": SonarrSensorEntityDescription[list[SonarrSeries]](
key="series",
translation_key="series",
- native_unit_of_measurement="Series",
entity_registry_enabled_default=False,
value_fn=len,
attributes_fn=lambda data: {
@@ -129,7 +126,6 @@ SENSOR_TYPES: dict[str, SonarrSensorEntityDescription[Any]] = {
"upcoming": SonarrSensorEntityDescription[list[SonarrCalendar]](
key="upcoming",
translation_key="upcoming",
- native_unit_of_measurement="Episodes",
value_fn=len,
attributes_fn=lambda data: {
e.series.title: f"S{e.seasonNumber:02d}E{e.episodeNumber:02d}" for e in data
@@ -138,7 +134,6 @@ SENSOR_TYPES: dict[str, SonarrSensorEntityDescription[Any]] = {
"wanted": SonarrSensorEntityDescription[SonarrWantedMissing](
key="wanted",
translation_key="wanted",
- native_unit_of_measurement="Episodes",
entity_registry_enabled_default=False,
value_fn=lambda data: data.totalRecords,
attributes_fn=get_wanted_attr,
@@ -149,7 +144,7 @@ SENSOR_TYPES: dict[str, SonarrSensorEntityDescription[Any]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sonarr sensors based on a config entry."""
coordinators: dict[str, SonarrDataUpdateCoordinator[Any]] = hass.data[DOMAIN][
diff --git a/homeassistant/components/sonarr/strings.json b/homeassistant/components/sonarr/strings.json
index 5b17f3283e8..940ec650270 100644
--- a/homeassistant/components/sonarr/strings.json
+++ b/homeassistant/components/sonarr/strings.json
@@ -37,22 +37,27 @@
"entity": {
"sensor": {
"commands": {
- "name": "Commands"
+ "name": "Commands",
+ "unit_of_measurement": "commands"
},
"diskspace": {
"name": "Disk space"
},
"queue": {
- "name": "Queue"
+ "name": "Queue",
+ "unit_of_measurement": "episodes"
},
"series": {
- "name": "Shows"
+ "name": "Shows",
+ "unit_of_measurement": "series"
},
"upcoming": {
- "name": "Upcoming"
+ "name": "Upcoming",
+ "unit_of_measurement": "[%key:component::sonarr::entity::sensor::queue::unit_of_measurement%]"
},
"wanted": {
- "name": "Wanted"
+ "name": "Wanted",
+ "unit_of_measurement": "[%key:component::sonarr::entity::sensor::queue::unit_of_measurement%]"
}
}
}
diff --git a/homeassistant/components/songpal/media_player.py b/homeassistant/components/songpal/media_player.py
index b4063b09691..3fc75d712a7 100644
--- a/homeassistant/components/songpal/media_player.py
+++ b/homeassistant/components/songpal/media_player.py
@@ -34,7 +34,10 @@ from homeassistant.helpers import (
entity_platform,
)
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import CONF_ENDPOINT, DOMAIN, ERROR_REQUEST_RETRY, SET_SOUND_SETTING
@@ -63,7 +66,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up songpal media player."""
name = config_entry.data[CONF_NAME]
diff --git a/homeassistant/components/sonos/__init__.py b/homeassistant/components/sonos/__init__.py
index d530fa21e39..24580971ae2 100644
--- a/homeassistant/components/sonos/__init__.py
+++ b/homeassistant/components/sonos/__init__.py
@@ -7,6 +7,7 @@ from collections import OrderedDict
from dataclasses import dataclass, field
import datetime
from functools import partial
+from ipaddress import AddressValueError, IPv4Address
import logging
import socket
from typing import Any, cast
@@ -208,6 +209,14 @@ class SonosDiscoveryManager:
async def async_subscribe_to_zone_updates(self, ip_address: str) -> None:
"""Test subscriptions and create SonosSpeakers based on results."""
+ try:
+ _ = IPv4Address(ip_address)
+ except AddressValueError:
+ _LOGGER.debug(
+ "Sonos integration only supports IPv4 addresses, invalid ip_address received: %s",
+ ip_address,
+ )
+ return
soco = SoCo(ip_address)
# Cache now to avoid household ID lookup during first ZoneGroupState processing
await self.hass.async_add_executor_job(
diff --git a/homeassistant/components/sonos/binary_sensor.py b/homeassistant/components/sonos/binary_sensor.py
index 2c1e8af9961..322beaed092 100644
--- a/homeassistant/components/sonos/binary_sensor.py
+++ b/homeassistant/components/sonos/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SONOS_CREATE_BATTERY, SONOS_CREATE_MIC_SENSOR
from .entity import SonosEntity
@@ -28,7 +28,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sonos from a config entry."""
diff --git a/homeassistant/components/sonos/config_flow.py b/homeassistant/components/sonos/config_flow.py
index 057cdb8ec08..b5e2c684281 100644
--- a/homeassistant/components/sonos/config_flow.py
+++ b/homeassistant/components/sonos/config_flow.py
@@ -31,6 +31,8 @@ class SonosDiscoveryFlowHandler(DiscoveryFlowHandler[Awaitable[bool]], domain=DO
hostname = discovery_info.hostname
if hostname is None or not hostname.lower().startswith("sonos"):
return self.async_abort(reason="not_sonos_device")
+ if discovery_info.ip_address.version != 4:
+ return self.async_abort(reason="not_ipv4_address")
if discovery_manager := self.hass.data.get(DATA_SONOS_DISCOVERY_MANAGER):
host = discovery_info.host
mdns_name = discovery_info.name
diff --git a/homeassistant/components/sonos/const.py b/homeassistant/components/sonos/const.py
index 610a68afedf..cda40729dbc 100644
--- a/homeassistant/components/sonos/const.py
+++ b/homeassistant/components/sonos/const.py
@@ -32,6 +32,7 @@ SONOS_TRACKS = "tracks"
SONOS_COMPOSER = "composers"
SONOS_RADIO = "radio"
SONOS_OTHER_ITEM = "other items"
+SONOS_AUDIO_BOOK = "audio book"
SONOS_STATE_PLAYING = "PLAYING"
SONOS_STATE_TRANSITIONING = "TRANSITIONING"
@@ -67,6 +68,7 @@ SONOS_TO_MEDIA_CLASSES = {
"object.item": MediaClass.TRACK,
"object.item.audioItem.musicTrack": MediaClass.TRACK,
"object.item.audioItem.audioBroadcast": MediaClass.GENRE,
+ "object.item.audioItem.audioBook": MediaClass.TRACK,
}
SONOS_TO_MEDIA_TYPES = {
@@ -84,6 +86,7 @@ SONOS_TO_MEDIA_TYPES = {
"object.container.playlistContainer.sameArtist": MediaType.ARTIST,
"object.container.playlistContainer": MediaType.PLAYLIST,
"object.item.audioItem.musicTrack": MediaType.TRACK,
+ "object.item.audioItem.audioBook": MediaType.TRACK,
}
MEDIA_TYPES_TO_SONOS: dict[MediaType | str, str] = {
@@ -113,6 +116,7 @@ SONOS_TYPES_MAPPING = {
"object.item": SONOS_OTHER_ITEM,
"object.item.audioItem.musicTrack": SONOS_TRACKS,
"object.item.audioItem.audioBroadcast": SONOS_RADIO,
+ "object.item.audioItem.audioBook": SONOS_AUDIO_BOOK,
}
LIBRARY_TITLES_MAPPING = {
@@ -170,6 +174,7 @@ MODELS_TV_ONLY = (
"BEAM",
"PLAYBAR",
"PLAYBASE",
+ "ULTRA",
)
MODELS_LINEIN_AND_TV = ("AMP",)
diff --git a/homeassistant/components/sonos/favorites.py b/homeassistant/components/sonos/favorites.py
index 5050555a7cb..333c4809e62 100644
--- a/homeassistant/components/sonos/favorites.py
+++ b/homeassistant/components/sonos/favorites.py
@@ -105,7 +105,7 @@ class SonosFavorites(SonosHouseholdCoordinator):
@soco_error()
def update_cache(self, soco: SoCo, update_id: int | None = None) -> bool:
"""Update cache of known favorites and return if cache has changed."""
- new_favorites = soco.music_library.get_sonos_favorites()
+ new_favorites = soco.music_library.get_sonos_favorites(full_album_art_uri=True)
# Polled update_id values do not match event_id values
# Each speaker can return a different polled update_id
diff --git a/homeassistant/components/sonos/manifest.json b/homeassistant/components/sonos/manifest.json
index bfdf0da9dbb..5bbfc33ae5b 100644
--- a/homeassistant/components/sonos/manifest.json
+++ b/homeassistant/components/sonos/manifest.json
@@ -7,8 +7,8 @@
"dependencies": ["ssdp"],
"documentation": "https://www.home-assistant.io/integrations/sonos",
"iot_class": "local_push",
- "loggers": ["soco"],
- "requirements": ["soco==0.30.8", "sonos-websocket==0.1.3"],
+ "loggers": ["soco", "sonos_websocket"],
+ "requirements": ["soco==0.30.9", "sonos-websocket==0.1.3"],
"ssdp": [
{
"st": "urn:schemas-upnp-org:device:ZonePlayer:1"
diff --git a/homeassistant/components/sonos/media_browser.py b/homeassistant/components/sonos/media_browser.py
index 995d6cea08c..16b425dae50 100644
--- a/homeassistant/components/sonos/media_browser.py
+++ b/homeassistant/components/sonos/media_browser.py
@@ -165,6 +165,8 @@ async def async_browse_media(
favorites_folder_payload,
speaker.favorites,
media_content_id,
+ media,
+ get_browse_image_url,
)
payload = {
@@ -443,7 +445,10 @@ def favorites_payload(favorites: SonosFavorites) -> BrowseMedia:
def favorites_folder_payload(
- favorites: SonosFavorites, media_content_id: str
+ favorites: SonosFavorites,
+ media_content_id: str,
+ media: SonosMedia,
+ get_browse_image_url: GetBrowseImageUrlType,
) -> BrowseMedia:
"""Create response payload to describe all items of a type of favorite.
@@ -463,7 +468,14 @@ def favorites_folder_payload(
media_content_type="favorite_item_id",
can_play=True,
can_expand=False,
- thumbnail=getattr(favorite, "album_art_uri", None),
+ thumbnail=get_thumbnail_url_full(
+ media=media,
+ is_internal=True,
+ media_content_type="favorite_item_id",
+ media_content_id=favorite.item_id,
+ get_browse_image_url=get_browse_image_url,
+ item=favorite,
+ ),
)
)
diff --git a/homeassistant/components/sonos/media_player.py b/homeassistant/components/sonos/media_player.py
index 8d0917c5dba..a774de0ae5b 100644
--- a/homeassistant/components/sonos/media_player.py
+++ b/homeassistant/components/sonos/media_player.py
@@ -46,7 +46,7 @@ from homeassistant.core import HomeAssistant, ServiceCall, SupportsResponse, cal
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_platform, service
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from . import UnjoinData, media_browser
@@ -108,7 +108,7 @@ ATTR_QUEUE_POSITION = "queue_position"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sonos from a config entry."""
platform = entity_platform.async_get_current_platform()
@@ -462,11 +462,20 @@ class SonosMediaPlayerEntity(SonosEntity, MediaPlayerEntity):
"""Play a favorite."""
uri = favorite.reference.get_uri()
soco = self.coordinator.soco
- if soco.music_source_from_uri(uri) in [
- MUSIC_SRC_RADIO,
- MUSIC_SRC_LINE_IN,
- ]:
- soco.play_uri(uri, title=favorite.title, timeout=LONG_SERVICE_TIMEOUT)
+ if (
+ soco.music_source_from_uri(uri)
+ in [
+ MUSIC_SRC_RADIO,
+ MUSIC_SRC_LINE_IN,
+ ]
+ or favorite.reference.item_class == "object.item.audioItem.audioBook"
+ ):
+ soco.play_uri(
+ uri,
+ title=favorite.title,
+ meta=favorite.resource_meta_data,
+ timeout=LONG_SERVICE_TIMEOUT,
+ )
else:
soco.clear_queue()
soco.add_to_queue(favorite.reference, timeout=LONG_SERVICE_TIMEOUT)
diff --git a/homeassistant/components/sonos/number.py b/homeassistant/components/sonos/number.py
index 272218cc01e..c23ba51a877 100644
--- a/homeassistant/components/sonos/number.py
+++ b/homeassistant/components/sonos/number.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SONOS_CREATE_LEVELS
from .entity import SonosEntity
@@ -70,7 +70,7 @@ LEVEL_FROM_NUMBER = {"balance": _balance_from_number}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Sonos number platform from a config entry."""
diff --git a/homeassistant/components/sonos/sensor.py b/homeassistant/components/sonos/sensor.py
index a089c09b33c..d888ee669bb 100644
--- a/homeassistant/components/sonos/sensor.py
+++ b/homeassistant/components/sonos/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
SONOS_CREATE_AUDIO_FORMAT_SENSOR,
@@ -29,7 +29,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sonos from a config entry."""
diff --git a/homeassistant/components/sonos/strings.json b/homeassistant/components/sonos/strings.json
index 07d2e2db4e0..433bb3cc36a 100644
--- a/homeassistant/components/sonos/strings.json
+++ b/homeassistant/components/sonos/strings.json
@@ -8,7 +8,8 @@
"abort": {
"not_sonos_device": "Discovered device is not a Sonos device",
"single_instance_allowed": "[%key:common::config_flow::abort::single_instance_allowed%]",
- "no_devices_found": "[%key:common::config_flow::abort::no_devices_found%]"
+ "no_devices_found": "[%key:common::config_flow::abort::no_devices_found%]",
+ "not_ipv4_address": "No IPv4 address in SSDP discovery information"
}
},
"issues": {
diff --git a/homeassistant/components/sonos/switch.py b/homeassistant/components/sonos/switch.py
index 4bf5487b1a6..ce4774a4138 100644
--- a/homeassistant/components/sonos/switch.py
+++ b/homeassistant/components/sonos/switch.py
@@ -15,7 +15,7 @@ from homeassistant.const import ATTR_TIME, EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_change
from .const import (
@@ -74,7 +74,7 @@ WEEKEND_DAYS = (0, 6)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sonos from a config entry."""
diff --git a/homeassistant/components/soundtouch/media_player.py b/homeassistant/components/soundtouch/media_player.py
index 5edd42b931a..c540b8dfd64 100644
--- a/homeassistant/components/soundtouch/media_player.py
+++ b/homeassistant/components/soundtouch/media_player.py
@@ -27,7 +27,7 @@ from homeassistant.helpers.device_registry import (
DeviceInfo,
format_mac,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -47,7 +47,7 @@ ATTR_SOUNDTOUCH_ZONE = "soundtouch_zone"
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Bose SoundTouch media player based on a config entry."""
device = hass.data[DOMAIN][entry.entry_id].device
diff --git a/homeassistant/components/speedtestdotnet/__init__.py b/homeassistant/components/speedtestdotnet/__init__.py
index e4c51ab7aa0..e4f439013c6 100644
--- a/homeassistant/components/speedtestdotnet/__init__.py
+++ b/homeassistant/components/speedtestdotnet/__init__.py
@@ -6,18 +6,16 @@ from functools import partial
import speedtest
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.start import async_at_started
-from .coordinator import SpeedTestDataCoordinator
+from .coordinator import SpeedTestConfigEntry, SpeedTestDataCoordinator
PLATFORMS = [Platform.SENSOR]
-type SpeedTestConfigEntry = ConfigEntry[SpeedTestDataCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, config_entry: SpeedTestConfigEntry
@@ -49,11 +47,15 @@ async def async_setup_entry(
return True
-async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, config_entry: SpeedTestConfigEntry
+) -> bool:
"""Unload SpeedTest Entry from config_entry."""
return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
-async def update_listener(hass: HomeAssistant, config_entry: ConfigEntry) -> None:
+async def update_listener(
+ hass: HomeAssistant, config_entry: SpeedTestConfigEntry
+) -> None:
"""Handle options update."""
await hass.config_entries.async_reload(config_entry.entry_id)
diff --git a/homeassistant/components/speedtestdotnet/config_flow.py b/homeassistant/components/speedtestdotnet/config_flow.py
index 3bfd4eb6e4a..4fbca5e0d29 100644
--- a/homeassistant/components/speedtestdotnet/config_flow.py
+++ b/homeassistant/components/speedtestdotnet/config_flow.py
@@ -9,7 +9,6 @@ import voluptuous as vol
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult, OptionsFlow
from homeassistant.core import callback
-from . import SpeedTestConfigEntry
from .const import (
CONF_SERVER_ID,
CONF_SERVER_NAME,
@@ -17,6 +16,7 @@ from .const import (
DEFAULT_SERVER,
DOMAIN,
)
+from .coordinator import SpeedTestConfigEntry
class SpeedTestFlowHandler(ConfigFlow, domain=DOMAIN):
diff --git a/homeassistant/components/speedtestdotnet/coordinator.py b/homeassistant/components/speedtestdotnet/coordinator.py
index 299652ba0bd..1308cb1d825 100644
--- a/homeassistant/components/speedtestdotnet/coordinator.py
+++ b/homeassistant/components/speedtestdotnet/coordinator.py
@@ -14,23 +14,28 @@ from .const import CONF_SERVER_ID, DEFAULT_SCAN_INTERVAL, DEFAULT_SERVER, DOMAIN
_LOGGER = logging.getLogger(__name__)
+type SpeedTestConfigEntry = ConfigEntry[SpeedTestDataCoordinator]
+
class SpeedTestDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Get the latest data from speedtest.net."""
- config_entry: ConfigEntry
+ config_entry: SpeedTestConfigEntry
def __init__(
- self, hass: HomeAssistant, config_entry: ConfigEntry, api: speedtest.Speedtest
+ self,
+ hass: HomeAssistant,
+ config_entry: SpeedTestConfigEntry,
+ api: speedtest.Speedtest,
) -> None:
"""Initialize the data object."""
self.hass = hass
- self.config_entry = config_entry
self.api = api
self.servers: dict[str, dict] = {DEFAULT_SERVER: {}}
super().__init__(
self.hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=DEFAULT_SCAN_INTERVAL),
)
diff --git a/homeassistant/components/speedtestdotnet/sensor.py b/homeassistant/components/speedtestdotnet/sensor.py
index 10da1dc93af..c2b7a6de28c 100644
--- a/homeassistant/components/speedtestdotnet/sensor.py
+++ b/homeassistant/components/speedtestdotnet/sensor.py
@@ -15,11 +15,10 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfDataRate, UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import SpeedTestConfigEntry
from .const import (
ATTR_BYTES_RECEIVED,
ATTR_BYTES_SENT,
@@ -30,7 +29,7 @@ from .const import (
DEFAULT_NAME,
DOMAIN,
)
-from .coordinator import SpeedTestDataCoordinator
+from .coordinator import SpeedTestConfigEntry, SpeedTestDataCoordinator
@dataclass(frozen=True)
@@ -70,7 +69,7 @@ SENSOR_TYPES: tuple[SpeedtestSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SpeedTestConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Speedtestdotnet sensors."""
speedtest_coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/spider/__init__.py b/homeassistant/components/spider/__init__.py
index 4b138ec77a8..c0d85c02dd4 100644
--- a/homeassistant/components/spider/__init__.py
+++ b/homeassistant/components/spider/__init__.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import issue_registry as ir
@@ -29,11 +29,13 @@ async def async_setup_entry(hass: HomeAssistant, _: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
- if all(
- config_entry.state is ConfigEntryState.NOT_LOADED
- for config_entry in hass.config_entries.async_entries(DOMAIN)
- if config_entry.entry_id != entry.entry_id
- ):
- ir.async_delete_issue(hass, DOMAIN, DOMAIN)
-
return True
+
+
+async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
+ """Remove a config entry."""
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
+ ir.async_delete_issue(hass, DOMAIN, DOMAIN)
+ # Remove any remaining disabled or ignored entries
+ for _entry in hass.config_entries.async_entries(DOMAIN):
+ hass.async_create_task(hass.config_entries.async_remove(_entry.entry_id))
diff --git a/homeassistant/components/spotify/__init__.py b/homeassistant/components/spotify/__init__.py
index 663b3f30caa..1c4ea961ce3 100644
--- a/homeassistant/components/spotify/__init__.py
+++ b/homeassistant/components/spotify/__init__.py
@@ -8,7 +8,6 @@ from typing import TYPE_CHECKING
import aiohttp
from spotifyaio import Device, SpotifyClient, SpotifyConnectionError
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -63,7 +62,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: SpotifyConfigEntry) -> b
spotify.refresh_token_function = _refresh_token
- coordinator = SpotifyCoordinator(hass, spotify)
+ coordinator = SpotifyCoordinator(hass, entry, spotify)
await coordinator.async_config_entry_first_refresh()
@@ -92,6 +91,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: SpotifyConfigEntry) -> b
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: SpotifyConfigEntry) -> bool:
"""Unload Spotify config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/spotify/browse_media.py b/homeassistant/components/spotify/browse_media.py
index 458525dde28..686431da249 100644
--- a/homeassistant/components/spotify/browse_media.py
+++ b/homeassistant/components/spotify/browse_media.py
@@ -226,17 +226,17 @@ async def async_browse_media(
if media_content_id is None or not media_content_id.startswith(MEDIA_PLAYER_PREFIX):
raise BrowseError("Invalid Spotify URL specified")
- # Check for config entry specifier, and extract Spotify URI
+ # The config entry id is the host name of the URL, the Spotify URI is the name
parsed_url = yarl.URL(media_content_id)
- host = parsed_url.host
+ config_entry_id = parsed_url.host
if (
- host is None
+ config_entry_id is None
# config entry ids can be upper or lower case. Yarl always returns host
# names in lower case, so we need to look for the config entry in both
or (
- entry := hass.config_entries.async_get_entry(host)
- or hass.config_entries.async_get_entry(host.upper())
+ entry := hass.config_entries.async_get_entry(config_entry_id)
+ or hass.config_entries.async_get_entry(config_entry_id.upper())
)
is None
or entry.state is not ConfigEntryState.LOADED
diff --git a/homeassistant/components/spotify/coordinator.py b/homeassistant/components/spotify/coordinator.py
index 8b8539d715a..2d5fffebb7b 100644
--- a/homeassistant/components/spotify/coordinator.py
+++ b/homeassistant/components/spotify/coordinator.py
@@ -56,11 +56,17 @@ class SpotifyCoordinator(DataUpdateCoordinator[SpotifyCoordinatorData]):
current_user: UserProfile
config_entry: SpotifyConfigEntry
- def __init__(self, hass: HomeAssistant, client: SpotifyClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SpotifyConfigEntry,
+ client: SpotifyClient,
+ ) -> None:
"""Initialize."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/spotify/media_player.py b/homeassistant/components/spotify/media_player.py
index 20a634efb42..d6265cbc39d 100644
--- a/homeassistant/components/spotify/media_player.py
+++ b/homeassistant/components/spotify/media_player.py
@@ -31,7 +31,7 @@ from homeassistant.components.media_player import (
RepeatMode,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .browse_media import async_browse_media_internal
@@ -70,7 +70,7 @@ AFTER_REQUEST_SLEEP = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: SpotifyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Spotify based on a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/sql/manifest.json b/homeassistant/components/sql/manifest.json
index c18b1b9f05f..37b5dc2b647 100644
--- a/homeassistant/components/sql/manifest.json
+++ b/homeassistant/components/sql/manifest.json
@@ -1,9 +1,10 @@
{
"domain": "sql",
"name": "SQL",
+ "after_dependencies": ["recorder"],
"codeowners": ["@gjohansson-ST", "@dougiteixeira"],
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/sql",
"iot_class": "local_polling",
- "requirements": ["SQLAlchemy==2.0.38", "sqlparse==0.5.0"]
+ "requirements": ["SQLAlchemy==2.0.39", "sqlparse==0.5.0"]
}
diff --git a/homeassistant/components/sql/sensor.py b/homeassistant/components/sql/sensor.py
index 312b0cd345e..a7b488dd521 100644
--- a/homeassistant/components/sql/sensor.py
+++ b/homeassistant/components/sql/sensor.py
@@ -36,7 +36,10 @@ from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.exceptions import TemplateError
from homeassistant.helpers import issue_registry as ir
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.template import Template
from homeassistant.helpers.trigger_template_entity import (
CONF_AVAILABILITY,
@@ -101,7 +104,9 @@ async def async_setup_platform(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SQL sensor from config entry."""
@@ -178,7 +183,7 @@ async def async_setup_sensor(
unique_id: str | None,
db_url: str,
yaml: bool,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddEntitiesCallback | AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SQL sensor."""
try:
diff --git a/homeassistant/components/squeezebox/__init__.py b/homeassistant/components/squeezebox/__init__.py
index f94ea118c6a..78a97e38833 100644
--- a/homeassistant/components/squeezebox/__init__.py
+++ b/homeassistant/components/squeezebox/__init__.py
@@ -53,6 +53,7 @@ _LOGGER = logging.getLogger(__name__)
PLATFORMS = [
Platform.BINARY_SENSOR,
+ Platform.BUTTON,
Platform.MEDIA_PLAYER,
Platform.SENSOR,
]
@@ -127,12 +128,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: SqueezeboxConfigEntry) -
)
_LOGGER.debug("LMS Device %s", device)
- server_coordinator = LMSStatusDataUpdateCoordinator(hass, lms)
+ server_coordinator = LMSStatusDataUpdateCoordinator(hass, entry, lms)
- entry.runtime_data = SqueezeboxData(
- coordinator=server_coordinator,
- server=lms,
- )
+ entry.runtime_data = SqueezeboxData(coordinator=server_coordinator, server=lms)
# set up player discovery
known_servers = hass.data.setdefault(DOMAIN, {}).setdefault(KNOWN_SERVERS, {})
@@ -151,7 +149,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: SqueezeboxConfigEntry) -
else:
_LOGGER.debug("Adding new entity: %s", player)
player_coordinator = SqueezeBoxPlayerUpdateCoordinator(
- hass, player, lms.uuid
+ hass, entry, player, lms.uuid
)
known_players.append(player.player_id)
async_dispatcher_send(
diff --git a/homeassistant/components/squeezebox/binary_sensor.py b/homeassistant/components/squeezebox/binary_sensor.py
index ec0bac0fe43..daae8703597 100644
--- a/homeassistant/components/squeezebox/binary_sensor.py
+++ b/homeassistant/components/squeezebox/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SqueezeboxConfigEntry
from .const import STATUS_SENSOR_NEEDSRESTART, STATUS_SENSOR_RESCAN
@@ -35,7 +35,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: SqueezeboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Platform setup using common elements."""
diff --git a/homeassistant/components/squeezebox/browse_media.py b/homeassistant/components/squeezebox/browse_media.py
index 331bf383c70..633f004993f 100644
--- a/homeassistant/components/squeezebox/browse_media.py
+++ b/homeassistant/components/squeezebox/browse_media.py
@@ -3,6 +3,7 @@
from __future__ import annotations
import contextlib
+from dataclasses import dataclass, field
from typing import Any
from pysqueezebox import Player
@@ -18,6 +19,8 @@ from homeassistant.components.media_player import (
from homeassistant.core import HomeAssistant
from homeassistant.helpers.network import is_internal_request
+from .const import UNPLAYABLE_TYPES
+
LIBRARY = [
"Favorites",
"Artists",
@@ -26,9 +29,12 @@ LIBRARY = [
"Playlists",
"Genres",
"New Music",
+ "Album Artists",
+ "Apps",
+ "Radios",
]
-MEDIA_TYPE_TO_SQUEEZEBOX = {
+MEDIA_TYPE_TO_SQUEEZEBOX: dict[str | MediaType, str] = {
"Favorites": "favorites",
"Artists": "artists",
"Albums": "albums",
@@ -36,38 +42,51 @@ MEDIA_TYPE_TO_SQUEEZEBOX = {
"Playlists": "playlists",
"Genres": "genres",
"New Music": "new music",
+ "Album Artists": "album artists",
MediaType.ALBUM: "album",
MediaType.ARTIST: "artist",
MediaType.TRACK: "title",
MediaType.PLAYLIST: "playlist",
MediaType.GENRE: "genre",
+ "Apps": "apps",
+ "Radios": "radios",
}
-SQUEEZEBOX_ID_BY_TYPE = {
+SQUEEZEBOX_ID_BY_TYPE: dict[str | MediaType, str] = {
MediaType.ALBUM: "album_id",
MediaType.ARTIST: "artist_id",
MediaType.TRACK: "track_id",
MediaType.PLAYLIST: "playlist_id",
MediaType.GENRE: "genre_id",
"Favorites": "item_id",
+ MediaType.APPS: "item_id",
}
-CONTENT_TYPE_MEDIA_CLASS: dict[str | MediaType, dict[str, MediaClass | None]] = {
+CONTENT_TYPE_MEDIA_CLASS: dict[str | MediaType, dict[str, MediaClass | str]] = {
"Favorites": {"item": MediaClass.DIRECTORY, "children": MediaClass.TRACK},
+ "Apps": {"item": MediaClass.DIRECTORY, "children": MediaClass.APP},
+ "Radios": {"item": MediaClass.DIRECTORY, "children": MediaClass.APP},
+ "App": {"item": MediaClass.DIRECTORY, "children": MediaClass.TRACK},
"Artists": {"item": MediaClass.DIRECTORY, "children": MediaClass.ARTIST},
"Albums": {"item": MediaClass.DIRECTORY, "children": MediaClass.ALBUM},
"Tracks": {"item": MediaClass.DIRECTORY, "children": MediaClass.TRACK},
"Playlists": {"item": MediaClass.DIRECTORY, "children": MediaClass.PLAYLIST},
"Genres": {"item": MediaClass.DIRECTORY, "children": MediaClass.GENRE},
"New Music": {"item": MediaClass.DIRECTORY, "children": MediaClass.ALBUM},
+ "Album Artists": {"item": MediaClass.DIRECTORY, "children": MediaClass.ARTIST},
MediaType.ALBUM: {"item": MediaClass.ALBUM, "children": MediaClass.TRACK},
MediaType.ARTIST: {"item": MediaClass.ARTIST, "children": MediaClass.ALBUM},
- MediaType.TRACK: {"item": MediaClass.TRACK, "children": None},
+ MediaType.TRACK: {"item": MediaClass.TRACK, "children": ""},
MediaType.GENRE: {"item": MediaClass.GENRE, "children": MediaClass.ARTIST},
MediaType.PLAYLIST: {"item": MediaClass.PLAYLIST, "children": MediaClass.TRACK},
+ MediaType.APP: {"item": MediaClass.DIRECTORY, "children": MediaClass.TRACK},
+ MediaType.APPS: {"item": MediaClass.DIRECTORY, "children": MediaClass.APP},
}
-CONTENT_TYPE_TO_CHILD_TYPE = {
+CONTENT_TYPE_TO_CHILD_TYPE: dict[
+ str | MediaType,
+ str | MediaType | None,
+] = {
MediaType.ALBUM: MediaType.TRACK,
MediaType.PLAYLIST: MediaType.PLAYLIST,
MediaType.ARTIST: MediaType.ALBUM,
@@ -78,14 +97,142 @@ CONTENT_TYPE_TO_CHILD_TYPE = {
"Playlists": MediaType.PLAYLIST,
"Genres": MediaType.GENRE,
"Favorites": None, # can only be determined after inspecting the item
+ "Apps": MediaClass.APP,
+ "Radios": MediaClass.APP,
+ "App": None, # can only be determined after inspecting the item
"New Music": MediaType.ALBUM,
+ "Album Artists": MediaType.ARTIST,
+ MediaType.APPS: MediaType.APP,
+ MediaType.APP: MediaType.TRACK,
}
-BROWSE_LIMIT = 1000
+
+@dataclass
+class BrowseData:
+ """Class for browser to squeezebox mappings and other browse data."""
+
+ content_type_to_child_type: dict[
+ str | MediaType,
+ str | MediaType | None,
+ ] = field(default_factory=dict)
+ content_type_media_class: dict[str | MediaType, dict[str, MediaClass | str]] = (
+ field(default_factory=dict)
+ )
+ squeezebox_id_by_type: dict[str | MediaType, str] = field(default_factory=dict)
+ media_type_to_squeezebox: dict[str | MediaType, str] = field(default_factory=dict)
+ known_apps_radios: set[str] = field(default_factory=set)
+
+ def __post_init__(self) -> None:
+ """Initialise the maps."""
+ self.content_type_media_class.update(CONTENT_TYPE_MEDIA_CLASS)
+ self.content_type_to_child_type.update(CONTENT_TYPE_TO_CHILD_TYPE)
+ self.squeezebox_id_by_type.update(SQUEEZEBOX_ID_BY_TYPE)
+ self.media_type_to_squeezebox.update(MEDIA_TYPE_TO_SQUEEZEBOX)
+
+
+def _add_new_command_to_browse_data(
+ browse_data: BrowseData, cmd: str | MediaType, type: str
+) -> None:
+ """Add items to maps for new apps or radios."""
+ browse_data.media_type_to_squeezebox[cmd] = cmd
+ browse_data.squeezebox_id_by_type[cmd] = type
+ browse_data.content_type_media_class[cmd] = {
+ "item": MediaClass.DIRECTORY,
+ "children": MediaClass.TRACK,
+ }
+ browse_data.content_type_to_child_type[cmd] = MediaType.TRACK
+
+
+def _build_response_apps_radios_category(
+ browse_data: BrowseData, cmd: str | MediaType, item: dict[str, Any]
+) -> BrowseMedia:
+ """Build item for App or radio category."""
+ return BrowseMedia(
+ media_content_id=item.get("id", ""),
+ title=item["title"],
+ media_content_type=cmd,
+ media_class=browse_data.content_type_media_class[cmd]["item"],
+ can_expand=True,
+ can_play=False,
+ )
+
+
+def _build_response_known_app(
+ browse_data: BrowseData, search_type: str, item: dict[str, Any]
+) -> BrowseMedia:
+ """Build item for app or radio."""
+
+ return BrowseMedia(
+ media_content_id=item.get("id", ""),
+ title=item["title"],
+ media_content_type=search_type,
+ media_class=browse_data.content_type_media_class[search_type]["item"],
+ can_play=bool(item["isaudio"] and item.get("url")),
+ can_expand=item["hasitems"],
+ )
+
+
+def _build_response_favorites(item: dict[str, Any]) -> BrowseMedia:
+ """Build item for Favorites."""
+ if "album_id" in item:
+ return BrowseMedia(
+ media_content_id=str(item["album_id"]),
+ title=item["title"],
+ media_content_type=MediaType.ALBUM,
+ media_class=CONTENT_TYPE_MEDIA_CLASS[MediaType.ALBUM]["item"],
+ can_expand=True,
+ can_play=True,
+ )
+ if item["hasitems"] and not item["isaudio"]:
+ return BrowseMedia(
+ media_content_id=item.get("id", ""),
+ title=item["title"],
+ media_content_type="Favorites",
+ media_class=CONTENT_TYPE_MEDIA_CLASS["Favorites"]["item"],
+ can_expand=True,
+ can_play=False,
+ )
+ return BrowseMedia(
+ media_content_id=item.get("id", ""),
+ title=item["title"],
+ media_content_type="Favorites",
+ media_class=CONTENT_TYPE_MEDIA_CLASS[MediaType.TRACK]["item"],
+ can_expand=item["hasitems"],
+ can_play=bool(item["isaudio"] and item.get("url")),
+ )
+
+
+def _get_item_thumbnail(
+ item: dict[str, Any],
+ player: Player,
+ entity: MediaPlayerEntity,
+ item_type: str | MediaType | None,
+ search_type: str,
+ internal_request: bool,
+) -> str | None:
+ """Construct path to thumbnail image."""
+ item_thumbnail: str | None = None
+ if artwork_track_id := item.get("artwork_track_id"):
+ if internal_request:
+ item_thumbnail = player.generate_image_url_from_track_id(artwork_track_id)
+ elif item_type is not None:
+ item_thumbnail = entity.get_browse_image_url(
+ item_type, item.get("id", ""), artwork_track_id
+ )
+
+ elif search_type in ["Apps", "Radios"]:
+ item_thumbnail = player.generate_image_url(item["icon"])
+ if item_thumbnail is None:
+ item_thumbnail = item.get("image_url") # will not be proxied by HA
+ return item_thumbnail
async def build_item_response(
- entity: MediaPlayerEntity, player: Player, payload: dict[str, str | None]
+ entity: MediaPlayerEntity,
+ player: Player,
+ payload: dict[str, str | None],
+ browse_limit: int,
+ browse_data: BrowseData,
) -> BrowseMedia:
"""Create response payload for search described by payload."""
@@ -96,78 +243,78 @@ async def build_item_response(
assert (
search_type is not None
) # async_browse_media will not call this function if search_type is None
- media_class = CONTENT_TYPE_MEDIA_CLASS[search_type]
+ media_class = browse_data.content_type_media_class[search_type]
children = None
if search_id and search_id != search_type:
- browse_id = (SQUEEZEBOX_ID_BY_TYPE[search_type], search_id)
+ browse_id = (browse_data.squeezebox_id_by_type[search_type], search_id)
else:
browse_id = None
result = await player.async_browse(
- MEDIA_TYPE_TO_SQUEEZEBOX[search_type],
- limit=BROWSE_LIMIT,
+ browse_data.media_type_to_squeezebox[search_type],
+ limit=browse_limit,
browse_id=browse_id,
)
if result is not None and result.get("items"):
- item_type = CONTENT_TYPE_TO_CHILD_TYPE[search_type]
+ item_type = browse_data.content_type_to_child_type[search_type]
children = []
- list_playable = []
for item in result["items"]:
- item_id = str(item["id"])
- item_thumbnail: str | None = None
- if item_type:
- child_item_type: MediaType | str = item_type
- child_media_class = CONTENT_TYPE_MEDIA_CLASS[item_type]
- can_expand = child_media_class["children"] is not None
- can_play = True
-
if search_type == "Favorites":
- if "album_id" in item:
- item_id = str(item["album_id"])
- child_item_type = MediaType.ALBUM
- child_media_class = CONTENT_TYPE_MEDIA_CLASS[MediaType.ALBUM]
- can_expand = True
- can_play = True
- elif item["hasitems"] and not item["isaudio"]:
- child_item_type = "Favorites"
- child_media_class = CONTENT_TYPE_MEDIA_CLASS["Favorites"]
- can_expand = True
- can_play = False
- else:
- child_item_type = "Favorites"
- child_media_class = CONTENT_TYPE_MEDIA_CLASS[MediaType.TRACK]
- can_expand = item["hasitems"]
- can_play = item["isaudio"] and item.get("url")
+ child_media = _build_response_favorites(item)
- if artwork_track_id := item.get("artwork_track_id"):
- if internal_request:
- item_thumbnail = player.generate_image_url_from_track_id(
- artwork_track_id
- )
- elif item_type is not None:
- item_thumbnail = entity.get_browse_image_url(
- item_type, item_id, artwork_track_id
- )
- else:
- item_thumbnail = item.get("image_url") # will not be proxied by HA
+ elif search_type in ["Apps", "Radios"]:
+ # item["cmd"] contains the name of the command to use with the cli for the app
+ # add the command to the dictionaries
+ if item["title"] == "Search" or item.get("type") in UNPLAYABLE_TYPES:
+ # Skip searches in apps as they'd need UI or if the link isn't to audio
+ continue
+ app_cmd = "app-" + item["cmd"]
- assert child_media_class["item"] is not None
- children.append(
- BrowseMedia(
- title=item["title"],
- media_class=child_media_class["item"],
- media_content_id=item_id,
- media_content_type=child_item_type,
- can_play=can_play,
- can_expand=can_expand,
- thumbnail=item_thumbnail,
+ if app_cmd not in browse_data.known_apps_radios:
+ browse_data.known_apps_radios.add(app_cmd)
+ _add_new_command_to_browse_data(browse_data, app_cmd, "item_id")
+
+ child_media = _build_response_apps_radios_category(
+ browse_data=browse_data, cmd=app_cmd, item=item
)
+
+ elif search_type in browse_data.known_apps_radios:
+ if (
+ item.get("title") in ["Search", None]
+ or item.get("type") in UNPLAYABLE_TYPES
+ ):
+ # Skip searches in apps as they'd need UI
+ continue
+
+ child_media = _build_response_known_app(browse_data, search_type, item)
+
+ elif item_type:
+ child_media = BrowseMedia(
+ media_content_id=str(item.get("id", "")),
+ title=item["title"],
+ media_content_type=item_type,
+ media_class=CONTENT_TYPE_MEDIA_CLASS[item_type]["item"],
+ can_expand=CONTENT_TYPE_MEDIA_CLASS[item_type]["children"]
+ is not None,
+ can_play=True,
+ )
+
+ assert child_media.media_class is not None
+
+ child_media.thumbnail = _get_item_thumbnail(
+ item=item,
+ player=player,
+ entity=entity,
+ item_type=item_type,
+ search_type=search_type,
+ internal_request=internal_request,
)
- list_playable.append(can_play)
+
+ children.append(child_media)
if children is None:
raise BrowseError(f"Media not found: {search_type} / {search_id}")
@@ -175,19 +322,24 @@ async def build_item_response(
assert media_class["item"] is not None
if not search_id:
search_id = search_type
+
return BrowseMedia(
title=result.get("title"),
media_class=media_class["item"],
children_media_class=media_class["children"],
media_content_id=search_id,
media_content_type=search_type,
- can_play=any(list_playable),
+ can_play=any(child.can_play for child in children),
children=children,
can_expand=True,
)
-async def library_payload(hass: HomeAssistant, player: Player) -> BrowseMedia:
+async def library_payload(
+ hass: HomeAssistant,
+ player: Player,
+ browse_media: BrowseData,
+) -> BrowseMedia:
"""Create response payload to describe contents of library."""
library_info: dict[str, Any] = {
"title": "Music Library",
@@ -200,10 +352,10 @@ async def library_payload(hass: HomeAssistant, player: Player) -> BrowseMedia:
}
for item in LIBRARY:
- media_class = CONTENT_TYPE_MEDIA_CLASS[item]
+ media_class = browse_media.content_type_media_class[item]
result = await player.async_browse(
- MEDIA_TYPE_TO_SQUEEZEBOX[item],
+ browse_media.media_type_to_squeezebox[item],
limit=1,
)
if result is not None and result.get("items") is not None:
@@ -214,7 +366,7 @@ async def library_payload(hass: HomeAssistant, player: Player) -> BrowseMedia:
media_class=media_class["children"],
media_content_id=item,
media_content_type=item,
- can_play=item != "Favorites",
+ can_play=item not in ["Favorites", "Apps", "Radios"],
can_expand=True,
)
)
@@ -237,17 +389,27 @@ def media_source_content_filter(item: BrowseMedia) -> bool:
return item.media_content_type.startswith("audio/")
-async def generate_playlist(player: Player, payload: dict[str, str]) -> list | None:
+async def generate_playlist(
+ player: Player,
+ payload: dict[str, str],
+ browse_limit: int,
+ browse_media: BrowseData,
+) -> list | None:
"""Generate playlist from browsing payload."""
media_type = payload["search_type"]
media_id = payload["search_id"]
- if media_type not in SQUEEZEBOX_ID_BY_TYPE:
+ if media_type not in browse_media.squeezebox_id_by_type:
raise BrowseError(f"Media type not supported: {media_type}")
- browse_id = (SQUEEZEBOX_ID_BY_TYPE[media_type], media_id)
+ browse_id = (browse_media.squeezebox_id_by_type[media_type], media_id)
+ if media_type.startswith("app-"):
+ category = media_type
+ else:
+ category = "titles"
+
result = await player.async_browse(
- "titles", limit=BROWSE_LIMIT, browse_id=browse_id
+ category, limit=browse_limit, browse_id=browse_id
)
if result and "items" in result:
items: list = result["items"]
diff --git a/homeassistant/components/squeezebox/button.py b/homeassistant/components/squeezebox/button.py
new file mode 100644
index 00000000000..098df3a1b5c
--- /dev/null
+++ b/homeassistant/components/squeezebox/button.py
@@ -0,0 +1,155 @@
+"""Platform for button integration for squeezebox."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+import logging
+
+from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import format_mac
+from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import SqueezeboxConfigEntry
+from .const import SIGNAL_PLAYER_DISCOVERED
+from .coordinator import SqueezeBoxPlayerUpdateCoordinator
+from .entity import SqueezeboxEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+HARDWARE_MODELS_WITH_SCREEN = [
+ "Squeezebox Boom",
+ "Squeezebox Radio",
+ "Transporter",
+ "Squeezebox Touch",
+ "Squeezebox",
+ "SliMP3",
+ "Squeezebox 1",
+ "Squeezebox 2",
+ "Squeezebox 3",
+]
+
+HARDWARE_MODELS_WITH_TONE = [
+ *HARDWARE_MODELS_WITH_SCREEN,
+ "Squeezebox Receiver",
+]
+
+
+@dataclass(frozen=True, kw_only=True)
+class SqueezeboxButtonEntityDescription(ButtonEntityDescription):
+ """Squeezebox Button description."""
+
+ press_action: str
+
+
+BUTTON_ENTITIES: tuple[SqueezeboxButtonEntityDescription, ...] = tuple(
+ SqueezeboxButtonEntityDescription(
+ key=f"preset_{i}",
+ translation_key="preset",
+ translation_placeholders={"index": str(i)},
+ press_action=f"preset_{i}.single",
+ )
+ for i in range(1, 7)
+)
+
+SCREEN_BUTTON_ENTITIES: tuple[SqueezeboxButtonEntityDescription, ...] = (
+ SqueezeboxButtonEntityDescription(
+ key="brightness_up",
+ translation_key="brightness_up",
+ press_action="brightness_up",
+ ),
+ SqueezeboxButtonEntityDescription(
+ key="brightness_down",
+ translation_key="brightness_down",
+ press_action="brightness_down",
+ ),
+)
+
+TONE_BUTTON_ENTITIES: tuple[SqueezeboxButtonEntityDescription, ...] = (
+ SqueezeboxButtonEntityDescription(
+ key="bass_up",
+ translation_key="bass_up",
+ press_action="bass_up",
+ ),
+ SqueezeboxButtonEntityDescription(
+ key="bass_down",
+ translation_key="bass_down",
+ press_action="bass_down",
+ ),
+ SqueezeboxButtonEntityDescription(
+ key="treble_up",
+ translation_key="treble_up",
+ press_action="treble_up",
+ ),
+ SqueezeboxButtonEntityDescription(
+ key="treble_down",
+ translation_key="treble_down",
+ press_action="treble_down",
+ ),
+)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: SqueezeboxConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the Squeezebox button platform from a server config entry."""
+
+ # Add button entities when player discovered
+ async def _player_discovered(
+ player_coordinator: SqueezeBoxPlayerUpdateCoordinator,
+ ) -> None:
+ _LOGGER.debug(
+ "Setting up button entity for player %s, model %s",
+ player_coordinator.player.name,
+ player_coordinator.player.model,
+ )
+
+ entities: list[SqueezeboxButtonEntity] = []
+
+ entities.extend(
+ SqueezeboxButtonEntity(player_coordinator, description)
+ for description in BUTTON_ENTITIES
+ )
+
+ entities.extend(
+ SqueezeboxButtonEntity(player_coordinator, description)
+ for description in TONE_BUTTON_ENTITIES
+ if player_coordinator.player.model in HARDWARE_MODELS_WITH_TONE
+ )
+
+ entities.extend(
+ SqueezeboxButtonEntity(player_coordinator, description)
+ for description in SCREEN_BUTTON_ENTITIES
+ if player_coordinator.player.model in HARDWARE_MODELS_WITH_SCREEN
+ )
+
+ async_add_entities(entities)
+
+ entry.async_on_unload(
+ async_dispatcher_connect(hass, SIGNAL_PLAYER_DISCOVERED, _player_discovered)
+ )
+
+
+class SqueezeboxButtonEntity(SqueezeboxEntity, ButtonEntity):
+ """Representation of Buttons for Squeezebox entities."""
+
+ entity_description: SqueezeboxButtonEntityDescription
+
+ def __init__(
+ self,
+ coordinator: SqueezeBoxPlayerUpdateCoordinator,
+ entity_description: SqueezeboxButtonEntityDescription,
+ ) -> None:
+ """Initialize the SqueezeBox Button."""
+ super().__init__(coordinator)
+ self.entity_description = entity_description
+ self._attr_unique_id = (
+ f"{format_mac(self._player.player_id)}_{entity_description.key}"
+ )
+
+ async def async_press(self) -> None:
+ """Execute the button action."""
+ await self._player.async_query("button", self.entity_description.press_action)
diff --git a/homeassistant/components/squeezebox/config_flow.py b/homeassistant/components/squeezebox/config_flow.py
index 97eb848c21c..2853ad14217 100644
--- a/homeassistant/components/squeezebox/config_flow.py
+++ b/homeassistant/components/squeezebox/config_flow.py
@@ -11,15 +11,34 @@ from pysqueezebox import Server, async_discover
import voluptuous as vol
from homeassistant.components.media_player import DOMAIN as MP_DOMAIN
-from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.config_entries import (
+ ConfigEntry,
+ ConfigFlow,
+ ConfigFlowResult,
+ OptionsFlow,
+)
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_USERNAME
+from homeassistant.core import callback
from homeassistant.data_entry_flow import AbortFlow
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.device_registry import format_mac
+from homeassistant.helpers.selector import (
+ NumberSelector,
+ NumberSelectorConfig,
+ NumberSelectorMode,
+)
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
-from .const import CONF_HTTPS, DEFAULT_PORT, DOMAIN
+from .const import (
+ CONF_BROWSE_LIMIT,
+ CONF_HTTPS,
+ CONF_VOLUME_STEP,
+ DEFAULT_BROWSE_LIMIT,
+ DEFAULT_PORT,
+ DEFAULT_VOLUME_STEP,
+ DOMAIN,
+)
_LOGGER = logging.getLogger(__name__)
@@ -77,6 +96,12 @@ class SqueezeboxConfigFlow(ConfigFlow, domain=DOMAIN):
self.data_schema = _base_schema()
self.discovery_info: dict[str, Any] | None = None
+ @staticmethod
+ @callback
+ def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlowHandler:
+ """Get the options flow for this handler."""
+ return OptionsFlowHandler()
+
async def _discover(self, uuid: str | None = None) -> None:
"""Discover an unconfigured LMS server."""
self.discovery_info = None
@@ -222,3 +247,48 @@ class SqueezeboxConfigFlow(ConfigFlow, domain=DOMAIN):
# if the player is unknown, then we likely need to configure its server
return await self.async_step_user()
+
+
+OPTIONS_SCHEMA = vol.Schema(
+ {
+ vol.Required(CONF_BROWSE_LIMIT): vol.All(
+ NumberSelector(
+ NumberSelectorConfig(min=1, max=65534, mode=NumberSelectorMode.BOX)
+ ),
+ vol.Coerce(int),
+ ),
+ vol.Required(CONF_VOLUME_STEP): vol.All(
+ NumberSelector(
+ NumberSelectorConfig(min=1, max=20, mode=NumberSelectorMode.SLIDER)
+ ),
+ vol.Coerce(int),
+ ),
+ }
+)
+
+
+class OptionsFlowHandler(OptionsFlow):
+ """Options Flow Handler."""
+
+ async def async_step_init(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Options Flow Steps."""
+
+ if user_input is not None:
+ return self.async_create_entry(title="", data=user_input)
+
+ return self.async_show_form(
+ step_id="init",
+ data_schema=self.add_suggested_values_to_schema(
+ OPTIONS_SCHEMA,
+ {
+ CONF_BROWSE_LIMIT: self.config_entry.options.get(
+ CONF_BROWSE_LIMIT, DEFAULT_BROWSE_LIMIT
+ ),
+ CONF_VOLUME_STEP: self.config_entry.options.get(
+ CONF_VOLUME_STEP, DEFAULT_VOLUME_STEP
+ ),
+ },
+ ),
+ )
diff --git a/homeassistant/components/squeezebox/const.py b/homeassistant/components/squeezebox/const.py
index 8bc33214170..5ce95d25632 100644
--- a/homeassistant/components/squeezebox/const.py
+++ b/homeassistant/components/squeezebox/const.py
@@ -27,8 +27,20 @@ STATUS_QUERY_LIBRARYNAME = "libraryname"
STATUS_QUERY_MAC = "mac"
STATUS_QUERY_UUID = "uuid"
STATUS_QUERY_VERSION = "version"
-SQUEEZEBOX_SOURCE_STRINGS = ("source:", "wavin:", "spotify:")
+SQUEEZEBOX_SOURCE_STRINGS = (
+ "source:",
+ "wavin:",
+ "spotify:",
+ "loop:",
+)
SIGNAL_PLAYER_DISCOVERED = "squeezebox_player_discovered"
SIGNAL_PLAYER_REDISCOVERED = "squeezebox_player_rediscovered"
DISCOVERY_INTERVAL = 60
PLAYER_UPDATE_INTERVAL = 5
+CONF_BROWSE_LIMIT = "browse_limit"
+CONF_VOLUME_STEP = "volume_step"
+DEFAULT_BROWSE_LIMIT = 1000
+DEFAULT_VOLUME_STEP = 5
+ATTR_ANNOUNCE_VOLUME = "announce_volume"
+ATTR_ANNOUNCE_TIMEOUT = "announce_timeout"
+UNPLAYABLE_TYPES = ("text", "actions")
diff --git a/homeassistant/components/squeezebox/coordinator.py b/homeassistant/components/squeezebox/coordinator.py
index f3aacbc9833..955e2896947 100644
--- a/homeassistant/components/squeezebox/coordinator.py
+++ b/homeassistant/components/squeezebox/coordinator.py
@@ -1,11 +1,13 @@
"""DataUpdateCoordinator for the Squeezebox integration."""
+from __future__ import annotations
+
from asyncio import timeout
from collections.abc import Callable
from datetime import timedelta
import logging
import re
-from typing import Any
+from typing import TYPE_CHECKING, Any
from pysqueezebox import Player, Server
@@ -14,6 +16,9 @@ from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt as dt_util
+if TYPE_CHECKING:
+ from . import SqueezeboxConfigEntry
+
from .const import (
PLAYER_UPDATE_INTERVAL,
SENSOR_UPDATE_INTERVAL,
@@ -30,11 +35,16 @@ _LOGGER = logging.getLogger(__name__)
class LMSStatusDataUpdateCoordinator(DataUpdateCoordinator):
"""LMS Status custom coordinator."""
- def __init__(self, hass: HomeAssistant, lms: Server) -> None:
+ config_entry: SqueezeboxConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: SqueezeboxConfigEntry, lms: Server
+ ) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=lms.name,
update_interval=timedelta(seconds=SENSOR_UPDATE_INTERVAL),
always_update=False,
@@ -80,11 +90,20 @@ class LMSStatusDataUpdateCoordinator(DataUpdateCoordinator):
class SqueezeBoxPlayerUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Coordinator for Squeezebox players."""
- def __init__(self, hass: HomeAssistant, player: Player, server_uuid: str) -> None:
+ config_entry: SqueezeboxConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: SqueezeboxConfigEntry,
+ player: Player,
+ server_uuid: str,
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=player.name,
update_interval=timedelta(seconds=PLAYER_UPDATE_INTERVAL),
always_update=True,
diff --git a/homeassistant/components/squeezebox/entity.py b/homeassistant/components/squeezebox/entity.py
index 027ca68edc6..2c443c24ffd 100644
--- a/homeassistant/components/squeezebox/entity.py
+++ b/homeassistant/components/squeezebox/entity.py
@@ -1,11 +1,37 @@
"""Base class for Squeezebox Sensor entities."""
-from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.device_registry import (
+ CONNECTION_NETWORK_MAC,
+ DeviceInfo,
+ format_mac,
+)
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, STATUS_QUERY_UUID
-from .coordinator import LMSStatusDataUpdateCoordinator
+from .coordinator import (
+ LMSStatusDataUpdateCoordinator,
+ SqueezeBoxPlayerUpdateCoordinator,
+)
+
+
+class SqueezeboxEntity(CoordinatorEntity[SqueezeBoxPlayerUpdateCoordinator]):
+ """Base entity class for Squeezebox entities."""
+
+ _attr_has_entity_name = True
+
+ def __init__(self, coordinator: SqueezeBoxPlayerUpdateCoordinator) -> None:
+ """Initialize the SqueezeBox entity."""
+ super().__init__(coordinator)
+ self._player = coordinator.player
+ self._attr_device_info = DeviceInfo(
+ identifiers={(DOMAIN, format_mac(self._player.player_id))},
+ name=self._player.name,
+ connections={(CONNECTION_NETWORK_MAC, format_mac(self._player.player_id))},
+ via_device=(DOMAIN, coordinator.server_uuid),
+ model=self._player.model,
+ manufacturer=self._player.creator,
+ )
class LMSStatusEntity(CoordinatorEntity[LMSStatusDataUpdateCoordinator]):
diff --git a/homeassistant/components/squeezebox/manifest.json b/homeassistant/components/squeezebox/manifest.json
index 09eaa4026f4..e9b89291749 100644
--- a/homeassistant/components/squeezebox/manifest.json
+++ b/homeassistant/components/squeezebox/manifest.json
@@ -12,5 +12,5 @@
"documentation": "https://www.home-assistant.io/integrations/squeezebox",
"iot_class": "local_polling",
"loggers": ["pysqueezebox"],
- "requirements": ["pysqueezebox==0.11.1"]
+ "requirements": ["pysqueezebox==0.12.0"]
}
diff --git a/homeassistant/components/squeezebox/media_player.py b/homeassistant/components/squeezebox/media_player.py
index 19cd1e36910..40662477745 100644
--- a/homeassistant/components/squeezebox/media_player.py
+++ b/homeassistant/components/squeezebox/media_player.py
@@ -14,6 +14,7 @@ import voluptuous as vol
from homeassistant.components import media_source
from homeassistant.components.media_player import (
ATTR_MEDIA_ENQUEUE,
+ ATTR_MEDIA_EXTRA,
BrowseError,
BrowseMedia,
MediaPlayerEnqueue,
@@ -34,24 +35,26 @@ from homeassistant.helpers import (
entity_platform,
entity_registry as er,
)
-from homeassistant.helpers.device_registry import (
- CONNECTION_NETWORK_MAC,
- DeviceInfo,
- format_mac,
-)
+from homeassistant.helpers.device_registry import format_mac
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.start import async_at_start
-from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.dt import utcnow
from .browse_media import (
+ BrowseData,
build_item_response,
generate_playlist,
library_payload,
media_source_content_filter,
)
from .const import (
+ ATTR_ANNOUNCE_TIMEOUT,
+ ATTR_ANNOUNCE_VOLUME,
+ CONF_BROWSE_LIMIT,
+ CONF_VOLUME_STEP,
+ DEFAULT_BROWSE_LIMIT,
+ DEFAULT_VOLUME_STEP,
DISCOVERY_TASK,
DOMAIN,
KNOWN_PLAYERS,
@@ -60,6 +63,7 @@ from .const import (
SQUEEZEBOX_SOURCE_STRINGS,
)
from .coordinator import SqueezeBoxPlayerUpdateCoordinator
+from .entity import SqueezeboxEntity
if TYPE_CHECKING:
from . import SqueezeboxConfigEntry
@@ -113,7 +117,7 @@ async def start_server_discovery(hass: HomeAssistant) -> None:
async def async_setup_entry(
hass: HomeAssistant,
entry: SqueezeboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Squeezebox media_player platform from a server config entry."""
@@ -153,9 +157,27 @@ async def async_setup_entry(
entry.async_on_unload(async_at_start(hass, start_server_discovery))
-class SqueezeBoxMediaPlayerEntity(
- CoordinatorEntity[SqueezeBoxPlayerUpdateCoordinator], MediaPlayerEntity
-):
+def get_announce_volume(extra: dict) -> float | None:
+ """Get announce volume from extra service data."""
+ if ATTR_ANNOUNCE_VOLUME not in extra:
+ return None
+ announce_volume = float(extra[ATTR_ANNOUNCE_VOLUME])
+ if not (0 < announce_volume <= 1):
+ raise ValueError
+ return announce_volume * 100
+
+
+def get_announce_timeout(extra: dict) -> int | None:
+ """Get announce volume from extra service data."""
+ if ATTR_ANNOUNCE_TIMEOUT not in extra:
+ return None
+ announce_timeout = int(extra[ATTR_ANNOUNCE_TIMEOUT])
+ if announce_timeout < 1:
+ raise ValueError
+ return announce_timeout
+
+
+class SqueezeBoxMediaPlayerEntity(SqueezeboxEntity, MediaPlayerEntity):
"""Representation of the media player features of a SqueezeBox device.
Wraps a pysqueezebox.Player() object.
@@ -166,6 +188,7 @@ class SqueezeBoxMediaPlayerEntity(
| MediaPlayerEntityFeature.PAUSE
| MediaPlayerEntityFeature.VOLUME_SET
| MediaPlayerEntityFeature.VOLUME_MUTE
+ | MediaPlayerEntityFeature.VOLUME_STEP
| MediaPlayerEntityFeature.PREVIOUS_TRACK
| MediaPlayerEntityFeature.NEXT_TRACK
| MediaPlayerEntityFeature.SEEK
@@ -179,41 +202,20 @@ class SqueezeBoxMediaPlayerEntity(
| MediaPlayerEntityFeature.STOP
| MediaPlayerEntityFeature.GROUPING
| MediaPlayerEntityFeature.MEDIA_ENQUEUE
+ | MediaPlayerEntityFeature.MEDIA_ANNOUNCE
)
_attr_has_entity_name = True
_attr_name = None
_last_update: datetime | None = None
- def __init__(
- self,
- coordinator: SqueezeBoxPlayerUpdateCoordinator,
- ) -> None:
+ def __init__(self, coordinator: SqueezeBoxPlayerUpdateCoordinator) -> None:
"""Initialize the SqueezeBox device."""
super().__init__(coordinator)
- player = coordinator.player
- self._player = player
self._query_result: bool | dict = {}
self._remove_dispatcher: Callable | None = None
self._previous_media_position = 0
- self._attr_unique_id = format_mac(player.player_id)
- _manufacturer = None
- if player.model == "SqueezeLite" or "SqueezePlay" in player.model:
- _manufacturer = "Ralph Irving"
- elif (
- "Squeezebox" in player.model
- or "Transporter" in player.model
- or "Slim" in player.model
- ):
- _manufacturer = "Logitech"
-
- self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, self._attr_unique_id)},
- name=player.name,
- connections={(CONNECTION_NETWORK_MAC, self._attr_unique_id)},
- via_device=(DOMAIN, coordinator.server_uuid),
- model=player.model,
- manufacturer=_manufacturer,
- )
+ self._attr_unique_id = format_mac(self._player.player_id)
+ self._browse_data = BrowseData()
@callback
def _handle_coordinator_update(self) -> None:
@@ -223,6 +225,23 @@ class SqueezeBoxMediaPlayerEntity(
self._last_update = utcnow()
self.async_write_ha_state()
+ @property
+ def volume_step(self) -> float:
+ """Return the step to be used for volume up down."""
+ return float(
+ self.coordinator.config_entry.options.get(
+ CONF_VOLUME_STEP, DEFAULT_VOLUME_STEP
+ )
+ / 100
+ )
+
+ @property
+ def browse_limit(self) -> int:
+ """Return the step to be used for volume up down."""
+ return self.coordinator.config_entry.options.get(
+ CONF_BROWSE_LIMIT, DEFAULT_BROWSE_LIMIT
+ )
+
@property
def available(self) -> bool:
"""Return True if entity is available."""
@@ -366,16 +385,6 @@ class SqueezeBoxMediaPlayerEntity(
await self._player.async_set_power(False)
await self.coordinator.async_refresh()
- async def async_volume_up(self) -> None:
- """Volume up media player."""
- await self._player.async_set_volume("+5")
- await self.coordinator.async_refresh()
-
- async def async_volume_down(self) -> None:
- """Volume down media player."""
- await self._player.async_set_volume("-5")
- await self.coordinator.async_refresh()
-
async def async_set_volume_level(self, volume: float) -> None:
"""Set volume level, range 0..1."""
volume_percent = str(int(volume * 100))
@@ -428,7 +437,11 @@ class SqueezeBoxMediaPlayerEntity(
await self.coordinator.async_refresh()
async def async_play_media(
- self, media_type: MediaType | str, media_id: str, **kwargs: Any
+ self,
+ media_type: MediaType | str,
+ media_id: str,
+ announce: bool | None = None,
+ **kwargs: Any,
) -> None:
"""Send the play_media command to the media player."""
index = None
@@ -451,6 +464,32 @@ class SqueezeBoxMediaPlayerEntity(
)
media_id = play_item.url
+ if announce:
+ if media_type not in MediaType.MUSIC:
+ raise ServiceValidationError(
+ "Announcements must have media type of 'music'. Playlists are not supported"
+ )
+
+ extra = kwargs.get(ATTR_MEDIA_EXTRA, {})
+ cmd = "announce"
+ try:
+ announce_volume = get_announce_volume(extra)
+ except ValueError:
+ raise ServiceValidationError(
+ f"{ATTR_ANNOUNCE_VOLUME} must be a number greater than 0 and less than or equal to 1"
+ ) from None
+ else:
+ self._player.set_announce_volume(announce_volume)
+
+ try:
+ announce_timeout = get_announce_timeout(extra)
+ except ValueError:
+ raise ServiceValidationError(
+ f"{ATTR_ANNOUNCE_TIMEOUT} must be a whole number greater than 0"
+ ) from None
+ else:
+ self._player.set_announce_timeout(announce_timeout)
+
if media_type in MediaType.MUSIC:
if not media_id.startswith(SQUEEZEBOX_SOURCE_STRINGS):
# do not process special squeezebox "source" media ids
@@ -466,7 +505,9 @@ class SqueezeBoxMediaPlayerEntity(
"search_id": media_id,
"search_type": MediaType.PLAYLIST,
}
- playlist = await generate_playlist(self._player, payload)
+ playlist = await generate_playlist(
+ self._player, payload, self.browse_limit, self._browse_data
+ )
except BrowseError:
# a list of urls
content = json.loads(media_id)
@@ -477,7 +518,9 @@ class SqueezeBoxMediaPlayerEntity(
"search_id": media_id,
"search_type": media_type,
}
- playlist = await generate_playlist(self._player, payload)
+ playlist = await generate_playlist(
+ self._player, payload, self.browse_limit, self._browse_data
+ )
_LOGGER.debug("Generated playlist: %s", playlist)
@@ -575,7 +618,7 @@ class SqueezeBoxMediaPlayerEntity(
)
if media_content_type in [None, "library"]:
- return await library_payload(self.hass, self._player)
+ return await library_payload(self.hass, self._player, self._browse_data)
if media_content_id and media_source.is_media_source_id(media_content_id):
return await media_source.async_browse_media(
@@ -587,7 +630,13 @@ class SqueezeBoxMediaPlayerEntity(
"search_id": media_content_id,
}
- return await build_item_response(self, self._player, payload)
+ return await build_item_response(
+ self,
+ self._player,
+ payload,
+ self.browse_limit,
+ self._browse_data,
+ )
async def async_get_browse_image(
self,
diff --git a/homeassistant/components/squeezebox/sensor.py b/homeassistant/components/squeezebox/sensor.py
index 0ca33179f9f..9d9490208ea 100644
--- a/homeassistant/components/squeezebox/sensor.py
+++ b/homeassistant/components/squeezebox/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import SqueezeboxConfigEntry
@@ -43,6 +43,7 @@ SENSORS: tuple[SensorEntityDescription, ...] = (
state_class=SensorStateClass.TOTAL,
device_class=SensorDeviceClass.DURATION,
native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.HOURS,
),
SensorEntityDescription(
key=STATUS_SENSOR_INFO_TOTAL_GENRES,
@@ -73,7 +74,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: SqueezeboxConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Platform setup using common elements."""
diff --git a/homeassistant/components/squeezebox/strings.json b/homeassistant/components/squeezebox/strings.json
index bce71ddb5f2..83c5d7dd5d0 100644
--- a/homeassistant/components/squeezebox/strings.json
+++ b/homeassistant/components/squeezebox/strings.json
@@ -63,6 +63,29 @@
}
},
"entity": {
+ "button": {
+ "preset": {
+ "name": "Preset {index}"
+ },
+ "brightness_up": {
+ "name": "Brightness up"
+ },
+ "brightness_down": {
+ "name": "Brightness down"
+ },
+ "bass_up": {
+ "name": "Bass up"
+ },
+ "bass_down": {
+ "name": "Bass down"
+ },
+ "treble_up": {
+ "name": "Treble up"
+ },
+ "treble_down": {
+ "name": "Treble down"
+ }
+ },
"binary_sensor": {
"rescan": {
"name": "Library rescan"
@@ -103,5 +126,20 @@
"unit_of_measurement": "[%key:component::squeezebox::entity::sensor::player_count::unit_of_measurement%]"
}
}
+ },
+ "options": {
+ "step": {
+ "init": {
+ "title": "LMS Configuration",
+ "data": {
+ "browse_limit": "Browse limit",
+ "volume_step": "Volume step"
+ },
+ "data_description": {
+ "browse_limit": "Maximum number of items when browsing or in a playlist.",
+ "volume_step": "Amount to adjust the volume when turning volume up or down."
+ }
+ }
+ }
}
}
diff --git a/homeassistant/components/srp_energy/__init__.py b/homeassistant/components/srp_energy/__init__.py
index 591ba5043e9..13c21709445 100644
--- a/homeassistant/components/srp_energy/__init__.py
+++ b/homeassistant/components/srp_energy/__init__.py
@@ -6,7 +6,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
-from .const import CONF_IS_TOU, DOMAIN, LOGGER
+from .const import DOMAIN, LOGGER
from .coordinator import SRPEnergyDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
@@ -26,9 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
api_password,
)
- coordinator = SRPEnergyDataUpdateCoordinator(
- hass, api_instance, entry.data[CONF_IS_TOU]
- )
+ coordinator = SRPEnergyDataUpdateCoordinator(hass, entry, api_instance)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/srp_energy/coordinator.py b/homeassistant/components/srp_energy/coordinator.py
index e5a72457433..f3821891afa 100644
--- a/homeassistant/components/srp_energy/coordinator.py
+++ b/homeassistant/components/srp_energy/coordinator.py
@@ -12,7 +12,13 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt as dt_util
-from .const import DOMAIN, LOGGER, MIN_TIME_BETWEEN_UPDATES, PHOENIX_TIME_ZONE
+from .const import (
+ CONF_IS_TOU,
+ DOMAIN,
+ LOGGER,
+ MIN_TIME_BETWEEN_UPDATES,
+ PHOENIX_TIME_ZONE,
+)
TIMEOUT = 10
PHOENIX_ZONE_INFO = dt_util.get_time_zone(PHOENIX_TIME_ZONE)
@@ -24,14 +30,15 @@ class SRPEnergyDataUpdateCoordinator(DataUpdateCoordinator[float]):
config_entry: ConfigEntry
def __init__(
- self, hass: HomeAssistant, client: SrpEnergyClient, is_time_of_use: bool
+ self, hass: HomeAssistant, config_entry: ConfigEntry, client: SrpEnergyClient
) -> None:
"""Initialize the srp_energy data coordinator."""
self._client = client
- self._is_time_of_use = is_time_of_use
+ self._is_time_of_use = config_entry.data[CONF_IS_TOU]
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=MIN_TIME_BETWEEN_UPDATES,
)
diff --git a/homeassistant/components/srp_energy/sensor.py b/homeassistant/components/srp_energy/sensor.py
index a9f5c25d6a5..89274390411 100644
--- a/homeassistant/components/srp_energy/sensor.py
+++ b/homeassistant/components/srp_energy/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfEnergy
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -20,7 +20,9 @@ from .const import DEVICE_CONFIG_URL, DEVICE_MANUFACTURER, DEVICE_MODEL, DOMAIN
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SRP Energy Usage sensor."""
coordinator: SRPEnergyDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/srp_energy/strings.json b/homeassistant/components/srp_energy/strings.json
index eca4f465435..5fa97b00b57 100644
--- a/homeassistant/components/srp_energy/strings.json
+++ b/homeassistant/components/srp_energy/strings.json
@@ -3,10 +3,10 @@
"step": {
"user": {
"data": {
- "id": "Account Id",
+ "id": "Account ID",
"username": "[%key:common::config_flow::data::username%]",
"password": "[%key:common::config_flow::data::password%]",
- "is_tou": "Is Time of Use Plan"
+ "is_tou": "Is Time-of-Use Price Plan"
}
}
},
diff --git a/homeassistant/components/starline/binary_sensor.py b/homeassistant/components/starline/binary_sensor.py
index ac1ad4f2b6e..a570b26a0d1 100644
--- a/homeassistant/components/starline/binary_sensor.py
+++ b/homeassistant/components/starline/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .account import StarlineAccount, StarlineDevice
from .const import DOMAIN
@@ -70,7 +70,9 @@ BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the StarLine sensors."""
account: StarlineAccount = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/starline/button.py b/homeassistant/components/starline/button.py
index 6fb307cda74..fa46d2a3773 100644
--- a/homeassistant/components/starline/button.py
+++ b/homeassistant/components/starline/button.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .account import StarlineAccount, StarlineDevice
from .const import DOMAIN
@@ -34,7 +34,9 @@ BUTTON_TYPES: tuple[ButtonEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the StarLine button."""
account: StarlineAccount = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/starline/device_tracker.py b/homeassistant/components/starline/device_tracker.py
index 610317b72c3..0c8418d28fc 100644
--- a/homeassistant/components/starline/device_tracker.py
+++ b/homeassistant/components/starline/device_tracker.py
@@ -3,7 +3,7 @@
from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .account import StarlineAccount, StarlineDevice
@@ -12,7 +12,9 @@ from .entity import StarlineEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up StarLine entry."""
account: StarlineAccount = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/starline/entity.py b/homeassistant/components/starline/entity.py
index 74807996dfb..f8846c2a97f 100644
--- a/homeassistant/components/starline/entity.py
+++ b/homeassistant/components/starline/entity.py
@@ -27,20 +27,20 @@ class StarlineEntity(Entity):
self._unsubscribe_api: Callable | None = None
@property
- def available(self):
+ def available(self) -> bool:
"""Return True if entity is available."""
return self._account.api.available
- def update(self):
+ def update(self) -> None:
"""Read new state data."""
self.schedule_update_ha_state()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Call when entity about to be added to Home Assistant."""
await super().async_added_to_hass()
self._unsubscribe_api = self._account.api.add_update_listener(self.update)
- async def async_will_remove_from_hass(self):
+ async def async_will_remove_from_hass(self) -> None:
"""Call when entity is being removed from Home Assistant."""
await super().async_will_remove_from_hass()
if self._unsubscribe_api is not None:
diff --git a/homeassistant/components/starline/lock.py b/homeassistant/components/starline/lock.py
index 19aad1a19b2..43886d63962 100644
--- a/homeassistant/components/starline/lock.py
+++ b/homeassistant/components/starline/lock.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .account import StarlineAccount, StarlineDevice
from .const import DOMAIN
@@ -15,7 +15,9 @@ from .entity import StarlineEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the StarLine lock."""
account: StarlineAccount = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/starline/sensor.py b/homeassistant/components/starline/sensor.py
index f9bd304c1e1..16988f1a9dc 100644
--- a/homeassistant/components/starline/sensor.py
+++ b/homeassistant/components/starline/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level, icon_for_signal_level
from .account import StarlineAccount, StarlineDevice
@@ -87,7 +87,9 @@ SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the StarLine sensors."""
account: StarlineAccount = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/starline/switch.py b/homeassistant/components/starline/switch.py
index eb71f0b73b5..79d4fa86ddf 100644
--- a/homeassistant/components/starline/switch.py
+++ b/homeassistant/components/starline/switch.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .account import StarlineAccount, StarlineDevice
from .const import DOMAIN
@@ -34,7 +34,9 @@ SWITCH_TYPES: tuple[SwitchEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the StarLine switch."""
account: StarlineAccount = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/starlink/__init__.py b/homeassistant/components/starlink/__init__.py
index 17081a7491e..0c512bb21c5 100644
--- a/homeassistant/components/starlink/__init__.py
+++ b/homeassistant/components/starlink/__init__.py
@@ -2,12 +2,10 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_IP_ADDRESS, Platform
+from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
-from .coordinator import StarlinkUpdateCoordinator
+from .coordinator import StarlinkConfigEntry, StarlinkUpdateCoordinator
PLATFORMS = [
Platform.BINARY_SENSOR,
@@ -19,25 +17,19 @@ PLATFORMS = [
]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(
+ hass: HomeAssistant, config_entry: StarlinkConfigEntry
+) -> bool:
"""Set up Starlink from a config entry."""
- coordinator = StarlinkUpdateCoordinator(
- hass=hass,
- url=entry.data[CONF_IP_ADDRESS],
- name=entry.title,
- )
+ config_entry.runtime_data = StarlinkUpdateCoordinator(hass, config_entry)
+ await config_entry.runtime_data.async_config_entry_first_refresh()
- await coordinator.async_config_entry_first_refresh()
-
- hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
-
- await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+ await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, config_entry: StarlinkConfigEntry
+) -> bool:
"""Unload a config entry."""
- if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
- hass.data[DOMAIN].pop(entry.entry_id)
-
- return unload_ok
+ return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
diff --git a/homeassistant/components/starlink/binary_sensor.py b/homeassistant/components/starlink/binary_sensor.py
index b03648e81c5..e06e79009c3 100644
--- a/homeassistant/components/starlink/binary_sensor.py
+++ b/homeassistant/components/starlink/binary_sensor.py
@@ -10,24 +10,22 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import StarlinkData
+from .coordinator import StarlinkConfigEntry, StarlinkData
from .entity import StarlinkEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config_entry: StarlinkConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all binary sensors for this entry."""
- coordinator = hass.data[DOMAIN][entry.entry_id]
-
async_add_entities(
- StarlinkBinarySensorEntity(coordinator, description)
+ StarlinkBinarySensorEntity(config_entry.runtime_data, description)
for description in BINARY_SENSORS
)
diff --git a/homeassistant/components/starlink/button.py b/homeassistant/components/starlink/button.py
index f8f18763d30..15f35659b49 100644
--- a/homeassistant/components/starlink/button.py
+++ b/homeassistant/components/starlink/button.py
@@ -10,24 +10,23 @@ from homeassistant.components.button import (
ButtonEntity,
ButtonEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import StarlinkUpdateCoordinator
+from .coordinator import StarlinkConfigEntry, StarlinkUpdateCoordinator
from .entity import StarlinkEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config_entry: StarlinkConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all binary sensors for this entry."""
- coordinator = hass.data[DOMAIN][entry.entry_id]
-
async_add_entities(
- StarlinkButtonEntity(coordinator, description) for description in BUTTONS
+ StarlinkButtonEntity(config_entry.runtime_data, description)
+ for description in BUTTONS
)
diff --git a/homeassistant/components/starlink/coordinator.py b/homeassistant/components/starlink/coordinator.py
index 6fcfd8e0bfe..02d51cd805e 100644
--- a/homeassistant/components/starlink/coordinator.py
+++ b/homeassistant/components/starlink/coordinator.py
@@ -26,12 +26,16 @@ from starlink_grpc import (
status_data,
)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_IP_ADDRESS
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
_LOGGER = logging.getLogger(__name__)
+type StarlinkConfigEntry = ConfigEntry[StarlinkUpdateCoordinator]
+
@dataclass
class StarlinkData:
@@ -49,15 +53,18 @@ class StarlinkData:
class StarlinkUpdateCoordinator(DataUpdateCoordinator[StarlinkData]):
"""Coordinates updates between all Starlink sensors defined in this file."""
- def __init__(self, hass: HomeAssistant, name: str, url: str) -> None:
+ config_entry: StarlinkConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: StarlinkConfigEntry) -> None:
"""Initialize an UpdateCoordinator for a group of sensors."""
- self.channel_context = ChannelContext(target=url)
+ self.channel_context = ChannelContext(target=config_entry.data[CONF_IP_ADDRESS])
self.history_stats_start = None
self.timezone = ZoneInfo(hass.config.time_zone)
super().__init__(
hass,
_LOGGER,
- name=name,
+ config_entry=config_entry,
+ name=config_entry.title,
update_interval=timedelta(seconds=5),
)
diff --git a/homeassistant/components/starlink/device_tracker.py b/homeassistant/components/starlink/device_tracker.py
index 5174be19760..dbe31947b55 100644
--- a/homeassistant/components/starlink/device_tracker.py
+++ b/homeassistant/components/starlink/device_tracker.py
@@ -8,23 +8,22 @@ from homeassistant.components.device_tracker import (
TrackerEntity,
TrackerEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import ATTR_ALTITUDE, DOMAIN
-from .coordinator import StarlinkData
+from .const import ATTR_ALTITUDE
+from .coordinator import StarlinkConfigEntry, StarlinkData
from .entity import StarlinkEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config_entry: StarlinkConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all binary sensors for this entry."""
- coordinator = hass.data[DOMAIN][entry.entry_id]
-
async_add_entities(
- StarlinkDeviceTrackerEntity(coordinator, description)
+ StarlinkDeviceTrackerEntity(config_entry.runtime_data, description)
for description in DEVICE_TRACKERS
)
diff --git a/homeassistant/components/starlink/diagnostics.py b/homeassistant/components/starlink/diagnostics.py
index c619458b1dd..543fe9d8dde 100644
--- a/homeassistant/components/starlink/diagnostics.py
+++ b/homeassistant/components/starlink/diagnostics.py
@@ -4,18 +4,15 @@ from dataclasses import asdict
from typing import Any
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
-from .coordinator import StarlinkUpdateCoordinator
+from .coordinator import StarlinkConfigEntry
TO_REDACT = {"id", "latitude", "longitude", "altitude"}
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, entry: ConfigEntry
+ hass: HomeAssistant, config_entry: StarlinkConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for Starlink config entries."""
- coordinator: StarlinkUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
- return async_redact_data(asdict(coordinator.data), TO_REDACT)
+ return async_redact_data(asdict(config_entry.runtime_data.data), TO_REDACT)
diff --git a/homeassistant/components/starlink/sensor.py b/homeassistant/components/starlink/sensor.py
index 5481e310fbd..d07e8174b27 100644
--- a/homeassistant/components/starlink/sensor.py
+++ b/homeassistant/components/starlink/sensor.py
@@ -12,7 +12,6 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
DEGREE,
PERCENTAGE,
@@ -24,23 +23,23 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import now
-from .const import DOMAIN
-from .coordinator import StarlinkData
+from .coordinator import StarlinkConfigEntry, StarlinkData
from .entity import StarlinkEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config_entry: StarlinkConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all sensors for this entry."""
- coordinator = hass.data[DOMAIN][entry.entry_id]
-
async_add_entities(
- StarlinkSensorEntity(coordinator, description) for description in SENSORS
+ StarlinkSensorEntity(config_entry.runtime_data, description)
+ for description in SENSORS
)
diff --git a/homeassistant/components/starlink/switch.py b/homeassistant/components/starlink/switch.py
index 3534748127e..c6dc237643e 100644
--- a/homeassistant/components/starlink/switch.py
+++ b/homeassistant/components/starlink/switch.py
@@ -11,23 +11,22 @@ from homeassistant.components.switch import (
SwitchEntity,
SwitchEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import StarlinkData, StarlinkUpdateCoordinator
+from .coordinator import StarlinkConfigEntry, StarlinkData, StarlinkUpdateCoordinator
from .entity import StarlinkEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config_entry: StarlinkConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all binary sensors for this entry."""
- coordinator = hass.data[DOMAIN][entry.entry_id]
-
async_add_entities(
- StarlinkSwitchEntity(coordinator, description) for description in SWITCHES
+ StarlinkSwitchEntity(config_entry.runtime_data, description)
+ for description in SWITCHES
)
diff --git a/homeassistant/components/starlink/time.py b/homeassistant/components/starlink/time.py
index 7395ec101ba..9f564333218 100644
--- a/homeassistant/components/starlink/time.py
+++ b/homeassistant/components/starlink/time.py
@@ -8,24 +8,23 @@ from datetime import UTC, datetime, time, tzinfo
import math
from homeassistant.components.time import TimeEntity, TimeEntityDescription
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import StarlinkData, StarlinkUpdateCoordinator
+from .coordinator import StarlinkConfigEntry, StarlinkData, StarlinkUpdateCoordinator
from .entity import StarlinkEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config_entry: StarlinkConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all time entities for this entry."""
- coordinator = hass.data[DOMAIN][entry.entry_id]
-
async_add_entities(
- StarlinkTimeEntity(coordinator, description) for description in TIMES
+ StarlinkTimeEntity(config_entry.runtime_data, description)
+ for description in TIMES
)
diff --git a/homeassistant/components/statistics/sensor.py b/homeassistant/components/statistics/sensor.py
index 5252c23fd3d..a5c5f10ecd0 100644
--- a/homeassistant/components/statistics/sensor.py
+++ b/homeassistant/components/statistics/sensor.py
@@ -47,7 +47,10 @@ from homeassistant.core import (
)
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.device import async_device_info_to_link_from_entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import (
async_track_point_in_utc_time,
async_track_state_change_event,
@@ -617,7 +620,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Statistics sensor entry."""
sampling_size = entry.options.get(CONF_SAMPLES_MAX_BUFFER_SIZE)
diff --git a/homeassistant/components/statistics/strings.json b/homeassistant/components/statistics/strings.json
index 51858034340..e1085a016ce 100644
--- a/homeassistant/components/statistics/strings.json
+++ b/homeassistant/components/statistics/strings.json
@@ -21,7 +21,7 @@
}
},
"state_characteristic": {
- "description": "Read the documention for further details on available options and how to use them.",
+ "description": "Read the documentation for further details on available options and how to use them.",
"data": {
"state_characteristic": "Statistic characteristic"
},
@@ -30,7 +30,7 @@
}
},
"options": {
- "description": "Read the documention for further details on how to configure the statistics sensor using these options.",
+ "description": "Read the documentation for further details on how to configure the statistics sensor using these options.",
"data": {
"sampling_size": "Sampling size",
"max_age": "Max age",
diff --git a/homeassistant/components/steam_online/__init__.py b/homeassistant/components/steam_online/__init__.py
index 6e45758fb94..7a2c32cb4d5 100644
--- a/homeassistant/components/steam_online/__init__.py
+++ b/homeassistant/components/steam_online/__init__.py
@@ -2,19 +2,17 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .coordinator import SteamDataUpdateCoordinator
+from .coordinator import SteamConfigEntry, SteamDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
-type SteamConfigEntry = ConfigEntry[SteamDataUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: SteamConfigEntry) -> bool:
"""Set up Steam from a config entry."""
- coordinator = SteamDataUpdateCoordinator(hass)
+ coordinator = SteamDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/steam_online/config_flow.py b/homeassistant/components/steam_online/config_flow.py
index 69009fca8c4..57c75f0a704 100644
--- a/homeassistant/components/steam_online/config_flow.py
+++ b/homeassistant/components/steam_online/config_flow.py
@@ -18,8 +18,8 @@ from homeassistant.const import CONF_API_KEY, Platform
from homeassistant.core import callback
from homeassistant.helpers import config_validation as cv, entity_registry as er
-from . import SteamConfigEntry
from .const import CONF_ACCOUNT, CONF_ACCOUNTS, DOMAIN, LOGGER, PLACEHOLDERS
+from .coordinator import SteamConfigEntry
# To avoid too long request URIs, the amount of ids to request is limited
MAX_IDS_TO_REQUEST = 275
diff --git a/homeassistant/components/steam_online/coordinator.py b/homeassistant/components/steam_online/coordinator.py
index 81a3bb0d898..731154183ed 100644
--- a/homeassistant/components/steam_online/coordinator.py
+++ b/homeassistant/components/steam_online/coordinator.py
@@ -3,11 +3,11 @@
from __future__ import annotations
from datetime import timedelta
-from typing import TYPE_CHECKING
import steam
from steam.api import _interface_method as INTMethod
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
@@ -15,8 +15,7 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import CONF_ACCOUNTS, DOMAIN, LOGGER
-if TYPE_CHECKING:
- from . import SteamConfigEntry
+type SteamConfigEntry = ConfigEntry[SteamDataUpdateCoordinator]
class SteamDataUpdateCoordinator(
@@ -26,11 +25,12 @@ class SteamDataUpdateCoordinator(
config_entry: SteamConfigEntry
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: SteamConfigEntry) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/steam_online/sensor.py b/homeassistant/components/steam_online/sensor.py
index 058bb386383..c1e20933185 100644
--- a/homeassistant/components/steam_online/sensor.py
+++ b/homeassistant/components/steam_online/sensor.py
@@ -8,11 +8,10 @@ from typing import cast
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utc_from_timestamp
-from . import SteamConfigEntry
from .const import (
CONF_ACCOUNTS,
STEAM_API_URL,
@@ -21,7 +20,7 @@ from .const import (
STEAM_MAIN_IMAGE_FILE,
STEAM_STATUSES,
)
-from .coordinator import SteamDataUpdateCoordinator
+from .coordinator import SteamConfigEntry, SteamDataUpdateCoordinator
from .entity import SteamEntity
PARALLEL_UPDATES = 1
@@ -30,7 +29,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: SteamConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Steam platform."""
async_add_entities(
diff --git a/homeassistant/components/steamist/__init__.py b/homeassistant/components/steamist/__init__.py
index 8d8401ec6fd..380f25ea8da 100644
--- a/homeassistant/components/steamist/__init__.py
+++ b/homeassistant/components/steamist/__init__.py
@@ -8,7 +8,7 @@ from typing import Any
from aiosteamist import Steamist
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_HOST, CONF_NAME, Platform
+from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -52,9 +52,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
host = entry.data[CONF_HOST]
coordinator = SteamistDataUpdateCoordinator(
hass,
+ entry,
Steamist(host, async_get_clientsession(hass)),
- host,
- entry.data.get(CONF_NAME), # Only found from discovery
)
await coordinator.async_config_entry_first_refresh()
if not async_get_discovery(hass, host):
diff --git a/homeassistant/components/steamist/coordinator.py b/homeassistant/components/steamist/coordinator.py
index c5aa7be7ddc..3f864364be7 100644
--- a/homeassistant/components/steamist/coordinator.py
+++ b/homeassistant/components/steamist/coordinator.py
@@ -7,6 +7,8 @@ import logging
from aiosteamist import Steamist, SteamistStatus
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -16,20 +18,22 @@ _LOGGER = logging.getLogger(__name__)
class SteamistDataUpdateCoordinator(DataUpdateCoordinator[SteamistStatus]):
"""DataUpdateCoordinator to gather data from a steamist steam shower."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
client: Steamist,
- host: str,
- device_name: str | None,
) -> None:
"""Initialize DataUpdateCoordinator to gather data for specific steamist."""
self.client = client
- self.device_name = device_name
+ self.device_name = config_entry.data.get(CONF_NAME) # Only found from discovery
super().__init__(
hass,
_LOGGER,
- name=f"Steamist {host}",
+ config_entry=config_entry,
+ name=f"Steamist {config_entry.data[CONF_HOST]}",
update_interval=timedelta(seconds=5),
always_update=False,
)
diff --git a/homeassistant/components/steamist/sensor.py b/homeassistant/components/steamist/sensor.py
index 7c24d015513..94e3ff86ee1 100644
--- a/homeassistant/components/steamist/sensor.py
+++ b/homeassistant/components/steamist/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SteamistDataUpdateCoordinator
@@ -58,7 +58,7 @@ SENSORS: tuple[SteamistSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
coordinator: SteamistDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/steamist/switch.py b/homeassistant/components/steamist/switch.py
index 91806f4fa0c..17e1d6d47ac 100644
--- a/homeassistant/components/steamist/switch.py
+++ b/homeassistant/components/steamist/switch.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SteamistDataUpdateCoordinator
@@ -22,7 +22,7 @@ ACTIVE_SWITCH = SwitchEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
coordinator: SteamistDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/stookwijzer/__init__.py b/homeassistant/components/stookwijzer/__init__.py
index d8b9561bde9..9adfc09de0e 100644
--- a/homeassistant/components/stookwijzer/__init__.py
+++ b/homeassistant/components/stookwijzer/__init__.py
@@ -9,7 +9,6 @@ from stookwijzer import Stookwijzer
from homeassistant.const import CONF_LATITUDE, CONF_LOCATION, CONF_LONGITUDE, Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er, issue_registry as ir
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import DOMAIN, LOGGER
from .coordinator import StookwijzerConfigEntry, StookwijzerCoordinator
@@ -43,13 +42,12 @@ async def async_migrate_entry(
LOGGER.debug("Migrating from version %s", entry.version)
if entry.version == 1:
- latitude, longitude = await Stookwijzer.async_transform_coordinates(
- async_get_clientsession(hass),
+ xy = await Stookwijzer.async_transform_coordinates(
entry.data[CONF_LOCATION][CONF_LATITUDE],
entry.data[CONF_LOCATION][CONF_LONGITUDE],
)
- if not latitude or not longitude:
+ if not xy:
ir.async_create_issue(
hass,
DOMAIN,
@@ -67,8 +65,8 @@ async def async_migrate_entry(
entry,
version=2,
data={
- CONF_LATITUDE: latitude,
- CONF_LONGITUDE: longitude,
+ CONF_LATITUDE: xy["x"],
+ CONF_LONGITUDE: xy["y"],
},
)
diff --git a/homeassistant/components/stookwijzer/config_flow.py b/homeassistant/components/stookwijzer/config_flow.py
index 32b4836763f..ff14bce26e6 100644
--- a/homeassistant/components/stookwijzer/config_flow.py
+++ b/homeassistant/components/stookwijzer/config_flow.py
@@ -9,7 +9,6 @@ import voluptuous as vol
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_LATITUDE, CONF_LOCATION, CONF_LONGITUDE
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.selector import LocationSelector
from .const import DOMAIN
@@ -26,15 +25,14 @@ class StookwijzerFlowHandler(ConfigFlow, domain=DOMAIN):
"""Handle a flow initialized by the user."""
errors = {}
if user_input is not None:
- latitude, longitude = await Stookwijzer.async_transform_coordinates(
- async_get_clientsession(self.hass),
+ xy = await Stookwijzer.async_transform_coordinates(
user_input[CONF_LOCATION][CONF_LATITUDE],
user_input[CONF_LOCATION][CONF_LONGITUDE],
)
- if latitude and longitude:
+ if xy:
return self.async_create_entry(
title="Stookwijzer",
- data={CONF_LATITUDE: latitude, CONF_LONGITUDE: longitude},
+ data={CONF_LATITUDE: xy["x"], CONF_LONGITUDE: xy["y"]},
)
errors["base"] = "unknown"
diff --git a/homeassistant/components/stookwijzer/coordinator.py b/homeassistant/components/stookwijzer/coordinator.py
index 23092bed66e..8f81494b7d5 100644
--- a/homeassistant/components/stookwijzer/coordinator.py
+++ b/homeassistant/components/stookwijzer/coordinator.py
@@ -20,18 +20,23 @@ type StookwijzerConfigEntry = ConfigEntry[StookwijzerCoordinator]
class StookwijzerCoordinator(DataUpdateCoordinator[None]):
"""Stookwijzer update coordinator."""
- def __init__(self, hass: HomeAssistant, entry: StookwijzerConfigEntry) -> None:
+ config_entry: StookwijzerConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: StookwijzerConfigEntry
+ ) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
self.client = Stookwijzer(
async_get_clientsession(hass),
- entry.data[CONF_LATITUDE],
- entry.data[CONF_LONGITUDE],
+ config_entry.data[CONF_LATITUDE],
+ config_entry.data[CONF_LONGITUDE],
)
async def _async_update_data(self) -> None:
diff --git a/homeassistant/components/stookwijzer/diagnostics.py b/homeassistant/components/stookwijzer/diagnostics.py
index 2849e0e976a..1f3ef4ee4ba 100644
--- a/homeassistant/components/stookwijzer/diagnostics.py
+++ b/homeassistant/components/stookwijzer/diagnostics.py
@@ -18,4 +18,5 @@ async def async_get_config_entry_diagnostics(
"advice": client.advice,
"air_quality_index": client.lki,
"windspeed_ms": client.windspeed_ms,
+ "forecast": await client.async_get_forecast(),
}
diff --git a/homeassistant/components/stookwijzer/manifest.json b/homeassistant/components/stookwijzer/manifest.json
index 3fe16fb3d33..dd10f57f485 100644
--- a/homeassistant/components/stookwijzer/manifest.json
+++ b/homeassistant/components/stookwijzer/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/stookwijzer",
"integration_type": "service",
"iot_class": "cloud_polling",
- "requirements": ["stookwijzer==1.5.1"]
+ "requirements": ["stookwijzer==1.6.1"]
}
diff --git a/homeassistant/components/stookwijzer/sensor.py b/homeassistant/components/stookwijzer/sensor.py
index 2660ff2ddb2..91224b711be 100644
--- a/homeassistant/components/stookwijzer/sensor.py
+++ b/homeassistant/components/stookwijzer/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfSpeed
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -59,7 +59,7 @@ STOOKWIJZER_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: StookwijzerConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Stookwijzer sensor from a config entry."""
async_add_entities(
diff --git a/homeassistant/components/stookwijzer/strings.json b/homeassistant/components/stookwijzer/strings.json
index 189af89b282..a028f1f19c5 100644
--- a/homeassistant/components/stookwijzer/strings.json
+++ b/homeassistant/components/stookwijzer/strings.json
@@ -2,7 +2,7 @@
"config": {
"step": {
"user": {
- "description": "Select the location you want to recieve the Stookwijzer information for.",
+ "description": "Select the location you want to receive the Stookwijzer information for.",
"data": {
"location": "[%key:common::config_flow::data::location%]"
},
@@ -29,7 +29,7 @@
},
"issues": {
"location_migration_failed": {
- "description": "The Stookwijzer integration was unable to automatically migrate your location to a new format the updated integrations uses.\n\nMake sure you are connected to the internet and restart Home Assistant to try again.\n\nIf this doesn't resolve the error, remove and re-add the integration.",
+ "description": "The Stookwijzer integration was unable to automatically migrate your location to a new format the updated integration uses.\n\nMake sure you are connected to the Internet and restart Home Assistant to try again.\n\nIf this doesn't resolve the error, remove and re-add the integration.",
"title": "Migration of your location failed"
}
},
diff --git a/homeassistant/components/streamlabswater/__init__.py b/homeassistant/components/streamlabswater/__init__.py
index 313fc1f24c5..1c1357a9b2b 100644
--- a/homeassistant/components/streamlabswater/__init__.py
+++ b/homeassistant/components/streamlabswater/__init__.py
@@ -35,7 +35,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
api_key = entry.data[CONF_API_KEY]
client = StreamlabsClient(api_key)
- coordinator = StreamlabsCoordinator(hass, client)
+ coordinator = StreamlabsCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/streamlabswater/binary_sensor.py b/homeassistant/components/streamlabswater/binary_sensor.py
index 5a0073c25d3..e3e966edde0 100644
--- a/homeassistant/components/streamlabswater/binary_sensor.py
+++ b/homeassistant/components/streamlabswater/binary_sensor.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import StreamlabsCoordinator
from .const import DOMAIN
@@ -15,7 +15,7 @@ from .entity import StreamlabsWaterEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Streamlabs water binary sensor from a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/streamlabswater/coordinator.py b/homeassistant/components/streamlabswater/coordinator.py
index 56e67abe222..df4a6056b36 100644
--- a/homeassistant/components/streamlabswater/coordinator.py
+++ b/homeassistant/components/streamlabswater/coordinator.py
@@ -5,6 +5,7 @@ from datetime import timedelta
from streamlabswater.streamlabswater import StreamlabsClient
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -25,15 +26,19 @@ class StreamlabsData:
class StreamlabsCoordinator(DataUpdateCoordinator[dict[str, StreamlabsData]]):
"""Coordinator for Streamlabs."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
client: StreamlabsClient,
) -> None:
"""Coordinator for Streamlabs."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Streamlabs",
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/streamlabswater/sensor.py b/homeassistant/components/streamlabswater/sensor.py
index 412b2187495..dea3f081326 100644
--- a/homeassistant/components/streamlabswater/sensor.py
+++ b/homeassistant/components/streamlabswater/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfVolume
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import StreamlabsCoordinator
@@ -60,7 +60,7 @@ SENSORS: tuple[StreamlabsWaterSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Streamlabs water sensor from a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/subaru/device_tracker.py b/homeassistant/components/subaru/device_tracker.py
index d406234c36e..6bcca848ef2 100644
--- a/homeassistant/components/subaru/device_tracker.py
+++ b/homeassistant/components/subaru/device_tracker.py
@@ -9,7 +9,7 @@ from subarulink.const import LATITUDE, LONGITUDE, TIMESTAMP
from homeassistant.components.device_tracker.config_entry import TrackerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -29,7 +29,7 @@ from .const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Subaru device tracker by config_entry."""
entry: dict = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/subaru/lock.py b/homeassistant/components/subaru/lock.py
index e21102f0b0c..07caa0d6367 100644
--- a/homeassistant/components/subaru/lock.py
+++ b/homeassistant/components/subaru/lock.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import SERVICE_LOCK, SERVICE_UNLOCK
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, get_device_info
from .const import (
@@ -32,7 +32,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Subaru locks by config_entry."""
entry = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/subaru/sensor.py b/homeassistant/components/subaru/sensor.py
index ba9b7d46b06..aa4c4ee16be 100644
--- a/homeassistant/components/subaru/sensor.py
+++ b/homeassistant/components/subaru/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfLength, UnitOfPressure, UnitOfVolume
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -141,7 +141,7 @@ EV_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Subaru sensors by config_entry."""
entry = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/subaru/strings.json b/homeassistant/components/subaru/strings.json
index 00da729dccd..7525e73f802 100644
--- a/homeassistant/components/subaru/strings.json
+++ b/homeassistant/components/subaru/strings.json
@@ -2,7 +2,7 @@
"config": {
"step": {
"user": {
- "title": "Subaru Starlink Configuration",
+ "title": "Subaru Starlink configuration",
"description": "Please enter your MySubaru credentials\nNOTE: Initial setup may take up to 30 seconds",
"data": {
"username": "[%key:common::config_flow::data::username%]",
@@ -49,7 +49,7 @@
"options": {
"step": {
"init": {
- "title": "Subaru Starlink Options",
+ "title": "Subaru Starlink options",
"description": "When enabled, vehicle polling will send a remote command to your vehicle every 2 hours to obtain new sensor data. Without vehicle polling, new sensor data is only received when the vehicle automatically pushes data (normally after engine shutdown).",
"data": {
"update_enabled": "Enable vehicle polling"
@@ -106,7 +106,7 @@
"fields": {
"door": {
"name": "Door",
- "description": "One of the following: 'all', 'driver', 'tailgate'."
+ "description": "Which door(s) to open."
}
}
}
diff --git a/homeassistant/components/suez_water/coordinator.py b/homeassistant/components/suez_water/coordinator.py
index 38f94b8937e..10d4d3cdbcb 100644
--- a/homeassistant/components/suez_water/coordinator.py
+++ b/homeassistant/components/suez_water/coordinator.py
@@ -20,8 +20,8 @@ class SuezWaterAggregatedAttributes:
this_month_consumption: dict[str, float]
previous_month_consumption: dict[str, float]
- last_year_overall: dict[str, float]
- this_year_overall: dict[str, float]
+ last_year_overall: int
+ this_year_overall: int
history: dict[str, float]
highest_monthly_consumption: float
diff --git a/homeassistant/components/suez_water/manifest.json b/homeassistant/components/suez_water/manifest.json
index 5d317ea5ba3..f09d2e22633 100644
--- a/homeassistant/components/suez_water/manifest.json
+++ b/homeassistant/components/suez_water/manifest.json
@@ -7,5 +7,5 @@
"iot_class": "cloud_polling",
"loggers": ["pysuez", "regex"],
"quality_scale": "bronze",
- "requirements": ["pysuezV2==2.0.3"]
+ "requirements": ["pysuezV2==2.0.4"]
}
diff --git a/homeassistant/components/suez_water/sensor.py b/homeassistant/components/suez_water/sensor.py
index 1152ebd551b..a162cc6168d 100644
--- a/homeassistant/components/suez_water/sensor.py
+++ b/homeassistant/components/suez_water/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import CURRENCY_EURO, UnitOfVolume
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_COUNTER_ID, DOMAIN
@@ -53,7 +53,7 @@ SENSORS: tuple[SuezWaterSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SuezWaterConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Suez Water sensor from a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/suez_water/strings.json b/homeassistant/components/suez_water/strings.json
index be2d4849e76..a8632fcb24a 100644
--- a/homeassistant/components/suez_water/strings.json
+++ b/homeassistant/components/suez_water/strings.json
@@ -5,21 +5,21 @@
"data": {
"username": "[%key:common::config_flow::data::username%]",
"password": "[%key:common::config_flow::data::password%]",
- "counter_id": "Meter id"
+ "counter_id": "Meter ID"
},
"data_description": {
"username": "Enter your login associated with your {tout_sur_mon_eau} account",
"password": "Enter your password associated with your {tout_sur_mon_eau} account",
- "counter_id": "Enter your meter id (ex: 12345678). Should be found automatically during setup, if not see integration documentation for more information"
+ "counter_id": "Enter your meter ID (ex: 12345678). Should be found automatically during setup, if not see integration documentation for more information"
},
- "description": "Connect your suez water {tout_sur_mon_eau} account to retrieve your water consumption"
+ "description": "Connect your Suez Water {tout_sur_mon_eau} account to retrieve your water consumption"
}
},
"error": {
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
"invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
"unknown": "[%key:common::config_flow::error::unknown%]",
- "counter_not_found": "Could not find meter id automatically"
+ "counter_not_found": "Could not find meter ID automatically"
},
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
diff --git a/homeassistant/components/sun/sensor.py b/homeassistant/components/sun/sensor.py
index e7e621d06cd..a042adb9b83 100644
--- a/homeassistant/components/sun/sensor.py
+++ b/homeassistant/components/sun/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.const import DEGREE, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN, SIGNAL_EVENTS_CHANGED, SIGNAL_POSITION_CHANGED
@@ -106,7 +106,9 @@ SENSOR_TYPES: tuple[SunSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: SunConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: SunConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sun sensor platform."""
diff --git a/homeassistant/components/sunweg/sensor/__init__.py b/homeassistant/components/sunweg/sensor/__init__.py
index e582b5135d3..f71d992bea9 100644
--- a/homeassistant/components/sunweg/sensor/__init__.py
+++ b/homeassistant/components/sunweg/sensor/__init__.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .. import SunWEGData
from ..const import CONF_PLANT_ID, DEFAULT_PLANT_ID, DOMAIN, DeviceType
@@ -49,7 +49,7 @@ def get_device_list(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the SunWEG sensor."""
name = config_entry.data[CONF_NAME]
diff --git a/homeassistant/components/surepetcare/__init__.py b/homeassistant/components/surepetcare/__init__.py
index e1f846d63a7..130242b7742 100644
--- a/homeassistant/components/surepetcare/__init__.py
+++ b/homeassistant/components/surepetcare/__init__.py
@@ -38,8 +38,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
try:
hass.data[DOMAIN][entry.entry_id] = coordinator = SurePetcareDataCoordinator(
- entry,
hass,
+ entry,
)
except SurePetcareAuthenticationError as error:
_LOGGER.error("Unable to connect to surepetcare.io: Wrong credentials!")
diff --git a/homeassistant/components/surepetcare/binary_sensor.py b/homeassistant/components/surepetcare/binary_sensor.py
index 3acd768cb30..416d56d1bdd 100644
--- a/homeassistant/components/surepetcare/binary_sensor.py
+++ b/homeassistant/components/surepetcare/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SurePetcareDataCoordinator
@@ -23,7 +23,9 @@ from .entity import SurePetcareEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sure PetCare Flaps binary sensors based on a config entry."""
diff --git a/homeassistant/components/surepetcare/coordinator.py b/homeassistant/components/surepetcare/coordinator.py
index a80e96ad185..d8112cebc90 100644
--- a/homeassistant/components/surepetcare/coordinator.py
+++ b/homeassistant/components/surepetcare/coordinator.py
@@ -33,7 +33,9 @@ SCAN_INTERVAL = timedelta(minutes=3)
class SurePetcareDataCoordinator(DataUpdateCoordinator[dict[int, SurepyEntity]]):
"""Handle Surepetcare data."""
- def __init__(self, entry: ConfigEntry, hass: HomeAssistant) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize the data handler."""
self.surepy = Surepy(
entry.data[CONF_USERNAME],
@@ -51,6 +53,7 @@ class SurePetcareDataCoordinator(DataUpdateCoordinator[dict[int, SurepyEntity]])
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/surepetcare/lock.py b/homeassistant/components/surepetcare/lock.py
index f960400bcbc..09fadf8be60 100644
--- a/homeassistant/components/surepetcare/lock.py
+++ b/homeassistant/components/surepetcare/lock.py
@@ -10,7 +10,7 @@ from surepy.enums import EntityType, LockState as SurepyLockState
from homeassistant.components.lock import LockEntity, LockState
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SurePetcareDataCoordinator
@@ -18,7 +18,9 @@ from .entity import SurePetcareEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sure PetCare locks on a config entry."""
diff --git a/homeassistant/components/surepetcare/sensor.py b/homeassistant/components/surepetcare/sensor.py
index b4e7c6203a3..b012878caf7 100644
--- a/homeassistant/components/surepetcare/sensor.py
+++ b/homeassistant/components/surepetcare/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_VOLTAGE, PERCENTAGE, EntityCategory, UnitOfVolume
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, SURE_BATT_VOLTAGE_DIFF, SURE_BATT_VOLTAGE_LOW
from .coordinator import SurePetcareDataCoordinator
@@ -20,7 +20,9 @@ from .entity import SurePetcareEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Sure PetCare Flaps sensors."""
diff --git a/homeassistant/components/swiss_public_transport/__init__.py b/homeassistant/components/swiss_public_transport/__init__.py
index 628f6e95c2a..0d0c4dc6169 100644
--- a/homeassistant/components/swiss_public_transport/__init__.py
+++ b/homeassistant/components/swiss_public_transport/__init__.py
@@ -96,7 +96,9 @@ async def async_setup_entry(
},
) from e
- coordinator = SwissPublicTransportDataUpdateCoordinator(hass, opendata, time_offset)
+ coordinator = SwissPublicTransportDataUpdateCoordinator(
+ hass, entry, opendata, time_offset
+ )
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/swiss_public_transport/coordinator.py b/homeassistant/components/swiss_public_transport/coordinator.py
index 81322117a6f..32b52122c7d 100644
--- a/homeassistant/components/swiss_public_transport/coordinator.py
+++ b/homeassistant/components/swiss_public_transport/coordinator.py
@@ -61,6 +61,7 @@ class SwissPublicTransportDataUpdateCoordinator(
def __init__(
self,
hass: HomeAssistant,
+ config_entry: SwissPublicTransportConfigEntry,
opendata: OpendataTransport,
time_offset: dict[str, int] | None,
) -> None:
@@ -68,6 +69,7 @@ class SwissPublicTransportDataUpdateCoordinator(
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=DEFAULT_UPDATE_TIME),
)
diff --git a/homeassistant/components/swiss_public_transport/sensor.py b/homeassistant/components/swiss_public_transport/sensor.py
index c8075a6746c..6475fe802c2 100644
--- a/homeassistant/components/swiss_public_transport/sensor.py
+++ b/homeassistant/components/swiss_public_transport/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -90,7 +90,7 @@ SENSORS: tuple[SwissPublicTransportSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SwissPublicTransportConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor from a config entry created in the integrations UI."""
unique_id = config_entry.unique_id
diff --git a/homeassistant/components/swiss_public_transport/strings.json b/homeassistant/components/swiss_public_transport/strings.json
index 1cdbd527467..f1b28f5ed14 100644
--- a/homeassistant/components/swiss_public_transport/strings.json
+++ b/homeassistant/components/swiss_public_transport/strings.json
@@ -83,8 +83,8 @@
},
"services": {
"fetch_connections": {
- "name": "Fetch Connections",
- "description": "Fetch a list of connections from the swiss public transport.",
+ "name": "Fetch connections",
+ "description": "Fetches a list of connections from Swiss public transport.",
"fields": {
"config_entry_id": {
"name": "Instance",
@@ -92,7 +92,7 @@
},
"limit": {
"name": "Limit",
- "description": "Number of connections to fetch from [1-15]"
+ "description": "Number of connections to fetch."
}
}
}
diff --git a/homeassistant/components/switch/strings.json b/homeassistant/components/switch/strings.json
index 0663384fe2c..b73cf8f849d 100644
--- a/homeassistant/components/switch/strings.json
+++ b/homeassistant/components/switch/strings.json
@@ -25,10 +25,18 @@
}
},
"switch": {
- "name": "[%key:component::switch::title%]"
+ "name": "[%key:component::switch::title%]",
+ "state": {
+ "off": "[%key:common::state::off%]",
+ "on": "[%key:common::state::on%]"
+ }
},
"outlet": {
- "name": "Outlet"
+ "name": "Outlet",
+ "state": {
+ "off": "[%key:common::state::off%]",
+ "on": "[%key:common::state::on%]"
+ }
}
},
"services": {
diff --git a/homeassistant/components/switch_as_x/cover.py b/homeassistant/components/switch_as_x/cover.py
index 7c6a7ff38ad..8fd9c799bcb 100644
--- a/homeassistant/components/switch_as_x/cover.py
+++ b/homeassistant/components/switch_as_x/cover.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
)
from homeassistant.core import Event, EventStateChangedData, HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_INVERT
from .entity import BaseInvertableEntity
@@ -29,7 +29,7 @@ from .entity import BaseInvertableEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Cover Switch config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/switch_as_x/fan.py b/homeassistant/components/switch_as_x/fan.py
index 858379e71df..846e9ae7e80 100644
--- a/homeassistant/components/switch_as_x/fan.py
+++ b/homeassistant/components/switch_as_x/fan.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ENTITY_ID
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import BaseToggleEntity
@@ -21,7 +21,7 @@ from .entity import BaseToggleEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Fan Switch config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/switch_as_x/light.py b/homeassistant/components/switch_as_x/light.py
index 59b816f7935..c043a354869 100644
--- a/homeassistant/components/switch_as_x/light.py
+++ b/homeassistant/components/switch_as_x/light.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ENTITY_ID
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import BaseToggleEntity
@@ -19,7 +19,7 @@ from .entity import BaseToggleEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Light Switch config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/switch_as_x/lock.py b/homeassistant/components/switch_as_x/lock.py
index 2095b06bd84..946429e0395 100644
--- a/homeassistant/components/switch_as_x/lock.py
+++ b/homeassistant/components/switch_as_x/lock.py
@@ -16,7 +16,7 @@ from homeassistant.const import (
)
from homeassistant.core import Event, EventStateChangedData, HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_INVERT
from .entity import BaseInvertableEntity
@@ -25,7 +25,7 @@ from .entity import BaseInvertableEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Lock Switch config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/switch_as_x/siren.py b/homeassistant/components/switch_as_x/siren.py
index 7d9a41d9cd9..b96c7c6e0ea 100644
--- a/homeassistant/components/switch_as_x/siren.py
+++ b/homeassistant/components/switch_as_x/siren.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ENTITY_ID
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import BaseToggleEntity
@@ -19,7 +19,7 @@ from .entity import BaseToggleEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Siren Switch config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/switch_as_x/valve.py b/homeassistant/components/switch_as_x/valve.py
index 8626ca3cfb4..2b5f252ac2d 100644
--- a/homeassistant/components/switch_as_x/valve.py
+++ b/homeassistant/components/switch_as_x/valve.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
)
from homeassistant.core import Event, EventStateChangedData, HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_INVERT
from .entity import BaseInvertableEntity
@@ -29,7 +29,7 @@ from .entity import BaseInvertableEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Valve Switch config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/switchbee/__init__.py b/homeassistant/components/switchbee/__init__.py
index a2a3ecf0df9..6e4bf004a3d 100644
--- a/homeassistant/components/switchbee/__init__.py
+++ b/homeassistant/components/switchbee/__init__.py
@@ -63,10 +63,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
websession = async_get_clientsession(hass, verify_ssl=False)
api = await get_api_object(central_unit, user, password, websession)
- coordinator = SwitchBeeCoordinator(
- hass,
- api,
- )
+ coordinator = SwitchBeeCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
entry.async_on_unload(entry.add_update_listener(update_listener))
diff --git a/homeassistant/components/switchbee/button.py b/homeassistant/components/switchbee/button.py
index 78b5c0e6888..1ac81ec4e0d 100644
--- a/homeassistant/components/switchbee/button.py
+++ b/homeassistant/components/switchbee/button.py
@@ -7,7 +7,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SwitchBeeCoordinator
@@ -15,7 +15,9 @@ from .entity import SwitchBeeEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbee button."""
coordinator: SwitchBeeCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/switchbee/climate.py b/homeassistant/components/switchbee/climate.py
index d946ed1761b..7837798b0cb 100644
--- a/homeassistant/components/switchbee/climate.py
+++ b/homeassistant/components/switchbee/climate.py
@@ -27,7 +27,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SwitchBeeCoordinator
@@ -74,7 +74,9 @@ SUPPORTED_FAN_MODES = [FAN_AUTO, FAN_HIGH, FAN_MEDIUM, FAN_LOW]
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBee climate."""
coordinator: SwitchBeeCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/switchbee/coordinator.py b/homeassistant/components/switchbee/coordinator.py
index 49400e3c28d..b0ea1707be8 100644
--- a/homeassistant/components/switchbee/coordinator.py
+++ b/homeassistant/components/switchbee/coordinator.py
@@ -10,6 +10,7 @@ from switchbee.api import CentralUnitPolling, CentralUnitWsRPC
from switchbee.api.central_unit import SwitchBeeError
from switchbee.device import DeviceType, SwitchBeeBaseDevice
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import format_mac
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -22,9 +23,12 @@ _LOGGER = logging.getLogger(__name__)
class SwitchBeeCoordinator(DataUpdateCoordinator[Mapping[int, SwitchBeeBaseDevice]]):
"""Class to manage fetching SwitchBee data API."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
swb_api: CentralUnitPolling | CentralUnitWsRPC,
) -> None:
"""Initialize."""
@@ -39,6 +43,7 @@ class SwitchBeeCoordinator(DataUpdateCoordinator[Mapping[int, SwitchBeeBaseDevic
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=SCAN_INTERVAL_SEC[type(self.api)]),
)
diff --git a/homeassistant/components/switchbee/cover.py b/homeassistant/components/switchbee/cover.py
index 02f3d7167e3..247063ab18a 100644
--- a/homeassistant/components/switchbee/cover.py
+++ b/homeassistant/components/switchbee/cover.py
@@ -17,7 +17,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SwitchBeeCoordinator
@@ -25,7 +25,9 @@ from .entity import SwitchBeeDeviceEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBee switch."""
coordinator: SwitchBeeCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/switchbee/light.py b/homeassistant/components/switchbee/light.py
index 0daa6e204aa..228667540df 100644
--- a/homeassistant/components/switchbee/light.py
+++ b/homeassistant/components/switchbee/light.py
@@ -11,7 +11,7 @@ from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEnti
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SwitchBeeCoordinator
@@ -35,7 +35,9 @@ def _switchbee_brightness_to_hass(value: int) -> int:
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBee light."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/switchbee/switch.py b/homeassistant/components/switchbee/switch.py
index c502e6f22f5..41538f6fd71 100644
--- a/homeassistant/components/switchbee/switch.py
+++ b/homeassistant/components/switchbee/switch.py
@@ -17,7 +17,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SwitchBeeCoordinator
@@ -25,7 +25,9 @@ from .entity import SwitchBeeDeviceEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbee switch."""
coordinator: SwitchBeeCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/switchbot/binary_sensor.py b/homeassistant/components/switchbot/binary_sensor.py
index 144872ff315..6d1490c895b 100644
--- a/homeassistant/components/switchbot/binary_sensor.py
+++ b/homeassistant/components/switchbot/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
from .entity import SwitchbotEntity
@@ -75,7 +75,7 @@ BINARY_SENSOR_TYPES: dict[str, BinarySensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbot curtain based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/switchbot/cover.py b/homeassistant/components/switchbot/cover.py
index d2fd073cdcb..3ef0f5625c2 100644
--- a/homeassistant/components/switchbot/cover.py
+++ b/homeassistant/components/switchbot/cover.py
@@ -17,7 +17,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
@@ -31,7 +31,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbot curtain based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/switchbot/entity.py b/homeassistant/components/switchbot/entity.py
index bde69429bc3..282d23bfd1a 100644
--- a/homeassistant/components/switchbot/entity.py
+++ b/homeassistant/components/switchbot/entity.py
@@ -61,7 +61,7 @@ class SwitchbotEntity(
return self.coordinator.device.parsed_data
@property
- def extra_state_attributes(self) -> Mapping[Any, Any]:
+ def extra_state_attributes(self) -> Mapping[str, Any]:
"""Return the state attributes."""
return {"last_run_success": self._last_run_success}
diff --git a/homeassistant/components/switchbot/humidifier.py b/homeassistant/components/switchbot/humidifier.py
index 40f96577842..34a24948df1 100644
--- a/homeassistant/components/switchbot/humidifier.py
+++ b/homeassistant/components/switchbot/humidifier.py
@@ -12,7 +12,7 @@ from homeassistant.components.humidifier import (
HumidifierEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SwitchbotConfigEntry
from .entity import SwitchbotSwitchedEntity
@@ -23,7 +23,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbot based on a config entry."""
async_add_entities([SwitchBotHumidifier(entry.runtime_data)])
diff --git a/homeassistant/components/switchbot/light.py b/homeassistant/components/switchbot/light.py
index 927ad5120c7..0a2c342ecf0 100644
--- a/homeassistant/components/switchbot/light.py
+++ b/homeassistant/components/switchbot/light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import (
LightEntity,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
from .entity import SwitchbotEntity
@@ -30,7 +30,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switchbot light."""
async_add_entities([SwitchbotLightEntity(entry.runtime_data)])
diff --git a/homeassistant/components/switchbot/lock.py b/homeassistant/components/switchbot/lock.py
index a3bee5661b2..6bad154813a 100644
--- a/homeassistant/components/switchbot/lock.py
+++ b/homeassistant/components/switchbot/lock.py
@@ -7,7 +7,7 @@ from switchbot.const import LockStatus
from homeassistant.components.lock import LockEntity, LockEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_LOCK_NIGHTLATCH, DEFAULT_LOCK_NIGHTLATCH
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import SwitchbotEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbot lock based on a config entry."""
force_nightlatch = entry.options.get(CONF_LOCK_NIGHTLATCH, DEFAULT_LOCK_NIGHTLATCH)
diff --git a/homeassistant/components/switchbot/manifest.json b/homeassistant/components/switchbot/manifest.json
index 92a1c25d6f5..85d5bcf6436 100644
--- a/homeassistant/components/switchbot/manifest.json
+++ b/homeassistant/components/switchbot/manifest.json
@@ -39,5 +39,5 @@
"documentation": "https://www.home-assistant.io/integrations/switchbot",
"iot_class": "local_push",
"loggers": ["switchbot"],
- "requirements": ["PySwitchbot==0.56.0"]
+ "requirements": ["PySwitchbot==0.57.1"]
}
diff --git a/homeassistant/components/switchbot/sensor.py b/homeassistant/components/switchbot/sensor.py
index 9787521a5e9..025c40bff9e 100644
--- a/homeassistant/components/switchbot/sensor.py
+++ b/homeassistant/components/switchbot/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
from .entity import SwitchbotEntity
@@ -102,7 +102,7 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbot sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/switchbot/strings.json b/homeassistant/components/switchbot/strings.json
index 9c101204dcb..c9f93cce604 100644
--- a/homeassistant/components/switchbot/strings.json
+++ b/homeassistant/components/switchbot/strings.json
@@ -70,6 +70,10 @@
"data": {
"retry_count": "Retry count",
"lock_force_nightlatch": "Force Nightlatch operation mode"
+ },
+ "data_description": {
+ "retry_count": "How many times to retry sending commands to your SwitchBot devices",
+ "lock_force_nightlatch": "Force Nightlatch operation mode even if Nightlatch is not detected"
}
}
}
diff --git a/homeassistant/components/switchbot/switch.py b/homeassistant/components/switchbot/switch.py
index 427496ef20c..fd1e8bb6393 100644
--- a/homeassistant/components/switchbot/switch.py
+++ b/homeassistant/components/switchbot/switch.py
@@ -9,7 +9,7 @@ import switchbot
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.const import STATE_ON
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
@@ -21,7 +21,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switchbot based on a config entry."""
async_add_entities([SwitchBotSwitch(entry.runtime_data)])
diff --git a/homeassistant/components/switchbot_cloud/__init__.py b/homeassistant/components/switchbot_cloud/__init__.py
index d7812158260..44e130cc7a4 100644
--- a/homeassistant/components/switchbot_cloud/__init__.py
+++ b/homeassistant/components/switchbot_cloud/__init__.py
@@ -47,13 +47,14 @@ class SwitchbotCloudData:
async def coordinator_for_device(
hass: HomeAssistant,
+ entry: ConfigEntry,
api: SwitchBotAPI,
device: Device | Remote,
coordinators_by_id: dict[str, SwitchBotCoordinator],
) -> SwitchBotCoordinator:
"""Instantiate coordinator and adds to list for gathering."""
coordinator = coordinators_by_id.setdefault(
- device.device_id, SwitchBotCoordinator(hass, api, device)
+ device.device_id, SwitchBotCoordinator(hass, entry, api, device)
)
if coordinator.data is None:
@@ -64,6 +65,7 @@ async def coordinator_for_device(
async def make_switchbot_devices(
hass: HomeAssistant,
+ entry: ConfigEntry,
api: SwitchBotAPI,
devices: list[Device | Remote],
coordinators_by_id: dict[str, SwitchBotCoordinator],
@@ -72,7 +74,7 @@ async def make_switchbot_devices(
devices_data = SwitchbotDevices()
await gather(
*[
- make_device_data(hass, api, device, devices_data, coordinators_by_id)
+ make_device_data(hass, entry, api, device, devices_data, coordinators_by_id)
for device in devices
]
)
@@ -82,6 +84,7 @@ async def make_switchbot_devices(
async def make_device_data(
hass: HomeAssistant,
+ entry: ConfigEntry,
api: SwitchBotAPI,
device: Device | Remote,
devices_data: SwitchbotDevices,
@@ -90,7 +93,7 @@ async def make_device_data(
"""Make device data."""
if isinstance(device, Remote) and device.device_type.endswith("Air Conditioner"):
coordinator = await coordinator_for_device(
- hass, api, device, coordinators_by_id
+ hass, entry, api, device, coordinators_by_id
)
devices_data.climates.append((device, coordinator))
if (
@@ -101,7 +104,7 @@ async def make_device_data(
)
) or isinstance(device, Remote):
coordinator = await coordinator_for_device(
- hass, api, device, coordinators_by_id
+ hass, entry, api, device, coordinators_by_id
)
devices_data.switches.append((device, coordinator))
@@ -117,7 +120,7 @@ async def make_device_data(
"Plug Mini (JP)",
]:
coordinator = await coordinator_for_device(
- hass, api, device, coordinators_by_id
+ hass, entry, api, device, coordinators_by_id
)
devices_data.sensors.append((device, coordinator))
@@ -128,19 +131,19 @@ async def make_device_data(
"Robot Vacuum Cleaner S1 Plus",
]:
coordinator = await coordinator_for_device(
- hass, api, device, coordinators_by_id
+ hass, entry, api, device, coordinators_by_id
)
devices_data.vacuums.append((device, coordinator))
if isinstance(device, Device) and device.device_type.startswith("Smart Lock"):
coordinator = await coordinator_for_device(
- hass, api, device, coordinators_by_id
+ hass, entry, api, device, coordinators_by_id
)
devices_data.locks.append((device, coordinator))
if isinstance(device, Device) and device.device_type in ["Bot"]:
coordinator = await coordinator_for_device(
- hass, api, device, coordinators_by_id
+ hass, entry, api, device, coordinators_by_id
)
if coordinator.data is not None:
if coordinator.data.get("deviceMode") == "pressMode":
@@ -149,10 +152,10 @@ async def make_device_data(
devices_data.switches.append((device, coordinator))
-async def async_setup_entry(hass: HomeAssistant, config: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up SwitchBot via API from a config entry."""
- token = config.data[CONF_API_TOKEN]
- secret = config.data[CONF_API_KEY]
+ token = entry.data[CONF_API_TOKEN]
+ secret = entry.data[CONF_API_KEY]
api = SwitchBotAPI(token=token, secret=secret)
try:
@@ -168,13 +171,13 @@ async def async_setup_entry(hass: HomeAssistant, config: ConfigEntry) -> bool:
coordinators_by_id: dict[str, SwitchBotCoordinator] = {}
switchbot_devices = await make_switchbot_devices(
- hass, api, devices, coordinators_by_id
+ hass, entry, api, devices, coordinators_by_id
)
hass.data.setdefault(DOMAIN, {})
- hass.data[DOMAIN][config.entry_id] = SwitchbotCloudData(
+ hass.data[DOMAIN][entry.entry_id] = SwitchbotCloudData(
api=api, devices=switchbot_devices
)
- await hass.config_entries.async_forward_entry_setups(config, PLATFORMS)
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
diff --git a/homeassistant/components/switchbot_cloud/button.py b/homeassistant/components/switchbot_cloud/button.py
index a6eb1a134a5..aae2758f3ca 100644
--- a/homeassistant/components/switchbot_cloud/button.py
+++ b/homeassistant/components/switchbot_cloud/button.py
@@ -7,7 +7,7 @@ from switchbot_api import BotCommands
from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitchbotCloudData
from .const import DOMAIN
@@ -17,7 +17,7 @@ from .entity import SwitchBotCloudEntity
async def async_setup_entry(
hass: HomeAssistant,
config: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBot Cloud entry."""
data: SwitchbotCloudData = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/switchbot_cloud/climate.py b/homeassistant/components/switchbot_cloud/climate.py
index 9e996649e8c..27698420ae9 100644
--- a/homeassistant/components/switchbot_cloud/climate.py
+++ b/homeassistant/components/switchbot_cloud/climate.py
@@ -13,7 +13,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitchbotCloudData
from .const import DOMAIN
@@ -42,7 +42,7 @@ _DEFAULT_SWITCHBOT_FAN_MODE = _SWITCHBOT_FAN_MODES[FanState.FAN_AUTO]
async def async_setup_entry(
hass: HomeAssistant,
config: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBot Cloud entry."""
data: SwitchbotCloudData = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/switchbot_cloud/coordinator.py b/homeassistant/components/switchbot_cloud/coordinator.py
index 0ebd04f7e5a..02ead5940e4 100644
--- a/homeassistant/components/switchbot_cloud/coordinator.py
+++ b/homeassistant/components/switchbot_cloud/coordinator.py
@@ -6,6 +6,7 @@ from typing import Any
from switchbot_api import CannotConnect, Device, Remote, SwitchBotAPI
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -19,16 +20,22 @@ type Status = dict[str, Any] | None
class SwitchBotCoordinator(DataUpdateCoordinator[Status]):
"""SwitchBot Cloud coordinator."""
+ config_entry: ConfigEntry
_api: SwitchBotAPI
_device_id: str
def __init__(
- self, hass: HomeAssistant, api: SwitchBotAPI, device: Device | Remote
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ api: SwitchBotAPI,
+ device: Device | Remote,
) -> None:
"""Initialize SwitchBot Cloud."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=DEFAULT_SCAN_INTERVAL,
)
diff --git a/homeassistant/components/switchbot_cloud/lock.py b/homeassistant/components/switchbot_cloud/lock.py
index 52f48c66d38..74a9e9d8b1e 100644
--- a/homeassistant/components/switchbot_cloud/lock.py
+++ b/homeassistant/components/switchbot_cloud/lock.py
@@ -7,7 +7,7 @@ from switchbot_api import LockCommands
from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitchbotCloudData
from .const import DOMAIN
@@ -17,7 +17,7 @@ from .entity import SwitchBotCloudEntity
async def async_setup_entry(
hass: HomeAssistant,
config: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBot Cloud entry."""
data: SwitchbotCloudData = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/switchbot_cloud/sensor.py b/homeassistant/components/switchbot_cloud/sensor.py
index 1f755c141a2..28384ffd4d5 100644
--- a/homeassistant/components/switchbot_cloud/sensor.py
+++ b/homeassistant/components/switchbot_cloud/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitchbotCloudData
from .const import DOMAIN
@@ -139,7 +139,7 @@ SENSOR_DESCRIPTIONS_BY_DEVICE_TYPES = {
async def async_setup_entry(
hass: HomeAssistant,
config: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBot Cloud entry."""
data: SwitchbotCloudData = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/switchbot_cloud/switch.py b/homeassistant/components/switchbot_cloud/switch.py
index 22d033625f9..ebe20620d3e 100644
--- a/homeassistant/components/switchbot_cloud/switch.py
+++ b/homeassistant/components/switchbot_cloud/switch.py
@@ -7,7 +7,7 @@ from switchbot_api import CommonCommands, Device, PowerState, Remote, SwitchBotA
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitchbotCloudData
from .const import DOMAIN
@@ -18,7 +18,7 @@ from .entity import SwitchBotCloudEntity
async def async_setup_entry(
hass: HomeAssistant,
config: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBot Cloud entry."""
data: SwitchbotCloudData = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/switchbot_cloud/vacuum.py b/homeassistant/components/switchbot_cloud/vacuum.py
index 84db7cfdbb8..9a9ad49626f 100644
--- a/homeassistant/components/switchbot_cloud/vacuum.py
+++ b/homeassistant/components/switchbot_cloud/vacuum.py
@@ -11,7 +11,7 @@ from homeassistant.components.vacuum import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitchbotCloudData
from .const import (
@@ -28,7 +28,7 @@ from .entity import SwitchBotCloudEntity
async def async_setup_entry(
hass: HomeAssistant,
config: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up SwitchBot Cloud entry."""
data: SwitchbotCloudData = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/switcher_kis/button.py b/homeassistant/components/switcher_kis/button.py
index d2686e2e550..30597ed0738 100644
--- a/homeassistant/components/switcher_kis/button.py
+++ b/homeassistant/components/switcher_kis/button.py
@@ -20,7 +20,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitcherConfigEntry
from .const import SIGNAL_DEVICE_ADD
@@ -81,7 +81,7 @@ THERMOSTAT_BUTTONS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SwitcherConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switcher button from config entry."""
diff --git a/homeassistant/components/switcher_kis/climate.py b/homeassistant/components/switcher_kis/climate.py
index 2fc4a331676..c8bf33eca09 100644
--- a/homeassistant/components/switcher_kis/climate.py
+++ b/homeassistant/components/switcher_kis/climate.py
@@ -29,7 +29,7 @@ from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SwitcherConfigEntry
from .const import SIGNAL_DEVICE_ADD
@@ -62,7 +62,7 @@ HA_TO_DEVICE_FAN = {value: key for key, value in DEVICE_FAN_TO_HA.items()}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: SwitcherConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switcher climate from config entry."""
diff --git a/homeassistant/components/switcher_kis/cover.py b/homeassistant/components/switcher_kis/cover.py
index 513b786a033..5d8e4a4b0ac 100644
--- a/homeassistant/components/switcher_kis/cover.py
+++ b/homeassistant/components/switcher_kis/cover.py
@@ -15,7 +15,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_DEVICE_ADD
from .coordinator import SwitcherDataUpdateCoordinator
@@ -28,7 +28,7 @@ API_STOP = "stop_shutter"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switcher cover from config entry."""
diff --git a/homeassistant/components/switcher_kis/light.py b/homeassistant/components/switcher_kis/light.py
index 75156044efa..b9dc78f5bdf 100644
--- a/homeassistant/components/switcher_kis/light.py
+++ b/homeassistant/components/switcher_kis/light.py
@@ -10,7 +10,7 @@ from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import SIGNAL_DEVICE_ADD
from .coordinator import SwitcherDataUpdateCoordinator
@@ -22,7 +22,7 @@ API_SET_LIGHT = "set_light"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switcher light from a config entry."""
diff --git a/homeassistant/components/switcher_kis/sensor.py b/homeassistant/components/switcher_kis/sensor.py
index 0ed60e5a721..029d517bb09 100644
--- a/homeassistant/components/switcher_kis/sensor.py
+++ b/homeassistant/components/switcher_kis/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfElectricCurrent, UnitOfPower
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import SIGNAL_DEVICE_ADD
@@ -61,7 +61,7 @@ THERMOSTAT_SENSORS = TEMPERATURE_SENSORS
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switcher sensor from config entry."""
diff --git a/homeassistant/components/switcher_kis/strings.json b/homeassistant/components/switcher_kis/strings.json
index e380711303d..c3cf111199f 100644
--- a/homeassistant/components/switcher_kis/strings.json
+++ b/homeassistant/components/switcher_kis/strings.json
@@ -9,13 +9,21 @@
"data": {
"username": "[%key:common::config_flow::data::username%]",
"token": "[%key:common::config_flow::data::access_token%]"
+ },
+ "data_description": {
+ "username": "The email address used to sign in to the Switcher app.",
+ "token": "The local control token received from Switcher."
}
},
"reauth_confirm": {
- "description": "Found a Switcher device that requires a token\nEnter your username and token\nFor more information see https://www.home-assistant.io/integrations/switcher_kis/#prerequisites",
+ "description": "[%key:component::switcher_kis::config::step::credentials::description%]",
"data": {
"username": "[%key:common::config_flow::data::username%]",
"token": "[%key:common::config_flow::data::access_token%]"
+ },
+ "data_description": {
+ "username": "[%key:component::switcher_kis::config::step::credentials::data_description::username%]",
+ "token": "[%key:component::switcher_kis::config::step::credentials::data_description::token%]"
}
}
},
diff --git a/homeassistant/components/switcher_kis/switch.py b/homeassistant/components/switcher_kis/switch.py
index 7d3d71a0615..30b0b4161b1 100644
--- a/homeassistant/components/switcher_kis/switch.py
+++ b/homeassistant/components/switcher_kis/switch.py
@@ -16,7 +16,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from .const import (
@@ -49,7 +49,7 @@ SERVICE_TURN_ON_WITH_TIMER_SCHEMA: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Switcher switch from config entry."""
platform = entity_platform.async_get_current_platform()
diff --git a/homeassistant/components/syncthing/sensor.py b/homeassistant/components/syncthing/sensor.py
index fc1f9ae8aea..697ea8aea6e 100644
--- a/homeassistant/components/syncthing/sensor.py
+++ b/homeassistant/components/syncthing/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_interval
from .const import (
@@ -25,7 +25,7 @@ from .const import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Syncthing sensors."""
syncthing = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/syncthru/binary_sensor.py b/homeassistant/components/syncthru/binary_sensor.py
index 2b110c2af1d..e6d26d22433 100644
--- a/homeassistant/components/syncthru/binary_sensor.py
+++ b/homeassistant/components/syncthru/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -35,7 +35,7 @@ SYNCTHRU_STATE_PROBLEM = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up from config entry."""
diff --git a/homeassistant/components/syncthru/sensor.py b/homeassistant/components/syncthru/sensor.py
index df2ffd99803..c2063bf6c0a 100644
--- a/homeassistant/components/syncthru/sensor.py
+++ b/homeassistant/components/syncthru/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, PERCENTAGE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -43,7 +43,7 @@ SYNCTHRU_STATE_HUMAN = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up from config entry."""
diff --git a/homeassistant/components/synology_dsm/__init__.py b/homeassistant/components/synology_dsm/__init__.py
index 0b8b8731f8f..70c7e76a53a 100644
--- a/homeassistant/components/synology_dsm/__init__.py
+++ b/homeassistant/components/synology_dsm/__init__.py
@@ -9,9 +9,8 @@ from synology_dsm.api.surveillance_station import SynoSurveillanceStation
from synology_dsm.api.surveillance_station.camera import SynoCamera
from synology_dsm.exceptions import SynologyDSMNotLoggedInException
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_MAC, CONF_VERIFY_SSL
-from homeassistant.core import HomeAssistant, callback
+from homeassistant.const import CONF_MAC, CONF_SCAN_INTERVAL, CONF_VERIFY_SSL
+from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
@@ -31,15 +30,16 @@ from .const import (
from .coordinator import (
SynologyDSMCameraUpdateCoordinator,
SynologyDSMCentralUpdateCoordinator,
+ SynologyDSMConfigEntry,
+ SynologyDSMData,
SynologyDSMSwitchUpdateCoordinator,
)
-from .models import SynologyDSMData
from .service import async_setup_services
_LOGGER = logging.getLogger(__name__)
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: SynologyDSMConfigEntry) -> bool:
"""Set up Synology DSM sensors."""
# Migrate device identifiers
@@ -68,6 +68,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
entry,
options={**entry.options, CONF_BACKUP_SHARE: None, CONF_BACKUP_PATH: None},
)
+ if CONF_SCAN_INTERVAL in entry.options:
+ current_options = {**entry.options}
+ current_options.pop(CONF_SCAN_INTERVAL)
+ hass.config_entries.async_update_entry(entry, options=current_options)
# Continue setup
api = SynoApi(hass, entry)
@@ -116,52 +120,50 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except SYNOLOGY_CONNECTION_EXCEPTIONS as ex:
raise ConfigEntryNotReady from ex
- synology_data = SynologyDSMData(
+ entry.runtime_data = SynologyDSMData(
api=api,
coordinator_central=coordinator_central,
coordinator_cameras=coordinator_cameras,
coordinator_switches=coordinator_switches,
)
- hass.data.setdefault(DOMAIN, {})[entry.unique_id] = synology_data
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
entry.async_on_unload(entry.add_update_listener(_async_update_listener))
if entry.options[CONF_BACKUP_SHARE]:
- _async_notify_backup_listeners_soon(hass)
+
+ def async_notify_backup_listeners() -> None:
+ for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
+ listener()
+
+ entry.async_on_unload(
+ entry.async_on_state_change(async_notify_backup_listeners)
+ )
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, entry: SynologyDSMConfigEntry
+) -> bool:
"""Unload Synology DSM sensors."""
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
- entry_data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ entry_data = entry.runtime_data
await entry_data.api.async_unload()
- hass.data[DOMAIN].pop(entry.unique_id)
- _async_notify_backup_listeners_soon(hass)
return unload_ok
-def _async_notify_backup_listeners(hass: HomeAssistant) -> None:
- for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
- listener()
-
-
-@callback
-def _async_notify_backup_listeners_soon(hass: HomeAssistant) -> None:
- hass.loop.call_soon(_async_notify_backup_listeners, hass)
-
-
-async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def _async_update_listener(
+ hass: HomeAssistant, entry: SynologyDSMConfigEntry
+) -> None:
"""Handle options update."""
await hass.config_entries.async_reload(entry.entry_id)
async def async_remove_config_entry_device(
- hass: HomeAssistant, entry: ConfigEntry, device_entry: dr.DeviceEntry
+ hass: HomeAssistant, entry: SynologyDSMConfigEntry, device_entry: dr.DeviceEntry
) -> bool:
"""Remove synology_dsm config entry from a device."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
api = data.api
assert api.information is not None
serial = api.information.serial
diff --git a/homeassistant/components/synology_dsm/backup.py b/homeassistant/components/synology_dsm/backup.py
index 83c3455bdf1..11f4287dea2 100644
--- a/homeassistant/components/synology_dsm/backup.py
+++ b/homeassistant/components/synology_dsm/backup.py
@@ -14,9 +14,9 @@ from homeassistant.components.backup import (
AgentBackup,
BackupAgent,
BackupAgentError,
+ BackupNotFound,
suggested_filename,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.aiohttp_client import ChunkAsyncStreamIterator
from homeassistant.helpers.json import json_dumps
@@ -28,7 +28,7 @@ from .const import (
DATA_BACKUP_AGENT_LISTENERS,
DOMAIN,
)
-from .models import SynologyDSMData
+from .coordinator import SynologyDSMConfigEntry
LOGGER = logging.getLogger(__name__)
@@ -46,18 +46,17 @@ async def async_get_backup_agents(
hass: HomeAssistant,
) -> list[BackupAgent]:
"""Return a list of backup agents."""
- if not (
- entries := hass.config_entries.async_loaded_entries(DOMAIN)
- ) or not hass.data.get(DOMAIN):
+ entries: list[SynologyDSMConfigEntry] = hass.config_entries.async_loaded_entries(
+ DOMAIN
+ )
+ if not entries:
LOGGER.debug("No proper config entry found")
return []
- syno_datas: dict[str, SynologyDSMData] = hass.data[DOMAIN]
return [
SynologyDSMBackupAgent(hass, entry, entry.unique_id)
for entry in entries
if entry.unique_id is not None
- and (syno_data := syno_datas.get(entry.unique_id))
- and syno_data.api.file_station
+ and entry.runtime_data.api.file_station
and entry.options.get(CONF_BACKUP_PATH)
]
@@ -90,7 +89,9 @@ class SynologyDSMBackupAgent(BackupAgent):
domain = DOMAIN
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry, unique_id: str) -> None:
+ def __init__(
+ self, hass: HomeAssistant, entry: SynologyDSMConfigEntry, unique_id: str
+ ) -> None:
"""Initialize the Synology DSM backup agent."""
super().__init__()
LOGGER.debug("Initializing Synology DSM backup agent for %s", entry.unique_id)
@@ -99,8 +100,9 @@ class SynologyDSMBackupAgent(BackupAgent):
self.path = (
f"{entry.options[CONF_BACKUP_SHARE]}/{entry.options[CONF_BACKUP_PATH]}"
)
- syno_data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ syno_data = entry.runtime_data
self.api = syno_data.api
+ self.backup_base_names: dict[str, str] = {}
@property
def _file_station(self) -> SynoFileStation:
@@ -109,18 +111,18 @@ class SynologyDSMBackupAgent(BackupAgent):
assert self.api.file_station
return self.api.file_station
- async def _async_suggested_filenames(
+ async def _async_backup_filenames(
self,
backup_id: str,
) -> tuple[str, str]:
- """Suggest filenames for the backup.
+ """Return the actual backup filenames.
:param backup_id: The ID of the backup that was returned in async_list_backups.
:return: A tuple of tar_filename and meta_filename
"""
- if (backup := await self.async_get_backup(backup_id)) is None:
- raise BackupAgentError("Backup not found")
- return suggested_filenames(backup)
+ await self.async_get_backup(backup_id)
+ base_name = self.backup_base_names[backup_id]
+ return (f"{base_name}.tar", f"{base_name}_meta.json")
async def async_download_backup(
self,
@@ -132,7 +134,7 @@ class SynologyDSMBackupAgent(BackupAgent):
:param backup_id: The ID of the backup that was returned in async_list_backups.
:return: An async iterator that yields bytes.
"""
- (filename_tar, _) = await self._async_suggested_filenames(backup_id)
+ (filename_tar, _) = await self._async_backup_filenames(backup_id)
try:
resp = await self._file_station.download_file(
@@ -192,13 +194,7 @@ class SynologyDSMBackupAgent(BackupAgent):
:param backup_id: The ID of the backup that was returned in async_list_backups.
"""
- try:
- (filename_tar, filename_meta) = await self._async_suggested_filenames(
- backup_id
- )
- except BackupAgentError:
- # backup meta data could not be found, so we can't delete the backup
- return
+ (filename_tar, filename_meta) = await self._async_backup_filenames(backup_id)
for filename in (filename_tar, filename_meta):
try:
@@ -247,6 +243,7 @@ class SynologyDSMBackupAgent(BackupAgent):
assert files
backups: dict[str, AgentBackup] = {}
+ backup_base_names: dict[str, str] = {}
for file in files:
if file.name.endswith("_meta.json"):
try:
@@ -255,14 +252,19 @@ class SynologyDSMBackupAgent(BackupAgent):
LOGGER.error("Failed to download meta data: %s", err)
continue
agent_backup = AgentBackup.from_dict(meta_data)
- backups[agent_backup.backup_id] = agent_backup
+ backup_id = agent_backup.backup_id
+ backups[backup_id] = agent_backup
+ backup_base_names[backup_id] = file.name.replace("_meta.json", "")
+ self.backup_base_names = backup_base_names
return backups
async def async_get_backup(
self,
backup_id: str,
**kwargs: Any,
- ) -> AgentBackup | None:
+ ) -> AgentBackup:
"""Return a backup."""
backups = await self._async_list_backups()
- return backups.get(backup_id)
+ if backup_id not in backups:
+ raise BackupNotFound(f"Backup {backup_id} not found")
+ return backups[backup_id]
diff --git a/homeassistant/components/synology_dsm/binary_sensor.py b/homeassistant/components/synology_dsm/binary_sensor.py
index b9c7ff483ea..1ae5fa90760 100644
--- a/homeassistant/components/synology_dsm/binary_sensor.py
+++ b/homeassistant/components/synology_dsm/binary_sensor.py
@@ -12,20 +12,17 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DISKS, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SynoApi
-from .const import DOMAIN
-from .coordinator import SynologyDSMCentralUpdateCoordinator
+from .coordinator import SynologyDSMCentralUpdateCoordinator, SynologyDSMConfigEntry
from .entity import (
SynologyDSMBaseEntity,
SynologyDSMDeviceEntity,
SynologyDSMEntityDescription,
)
-from .models import SynologyDSMData
@dataclass(frozen=True, kw_only=True)
@@ -63,10 +60,12 @@ STORAGE_DISK_BINARY_SENSORS: tuple[SynologyDSMBinarySensorEntityDescription, ...
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: SynologyDSMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Synology NAS binary sensor."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
api = data.api
coordinator = data.coordinator_central
assert api.storage is not None
diff --git a/homeassistant/components/synology_dsm/button.py b/homeassistant/components/synology_dsm/button.py
index fccd0860036..79297b1f1b4 100644
--- a/homeassistant/components/synology_dsm/button.py
+++ b/homeassistant/components/synology_dsm/button.py
@@ -12,15 +12,14 @@ from homeassistant.components.button import (
ButtonEntity,
ButtonEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SynoApi
from .const import DOMAIN
-from .models import SynologyDSMData
+from .coordinator import SynologyDSMConfigEntry
LOGGER = logging.getLogger(__name__)
@@ -52,11 +51,11 @@ BUTTONS: Final = [
async def async_setup_entry(
hass: HomeAssistant,
- entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: SynologyDSMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set buttons for device."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
async_add_entities(SynologyDSMButton(data.api, button) for button in BUTTONS)
diff --git a/homeassistant/components/synology_dsm/camera.py b/homeassistant/components/synology_dsm/camera.py
index cbf17ec05b4..f393b8efb55 100644
--- a/homeassistant/components/synology_dsm/camera.py
+++ b/homeassistant/components/synology_dsm/camera.py
@@ -16,11 +16,10 @@ from homeassistant.components.camera import (
CameraEntityDescription,
CameraEntityFeature,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SynoApi
from .const import (
@@ -29,9 +28,8 @@ from .const import (
DOMAIN,
SIGNAL_CAMERA_SOURCE_CHANGED,
)
-from .coordinator import SynologyDSMCameraUpdateCoordinator
+from .coordinator import SynologyDSMCameraUpdateCoordinator, SynologyDSMConfigEntry
from .entity import SynologyDSMBaseEntity, SynologyDSMEntityDescription
-from .models import SynologyDSMData
_LOGGER = logging.getLogger(__name__)
@@ -46,10 +44,12 @@ class SynologyDSMCameraEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: SynologyDSMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Synology NAS cameras."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
if coordinator := data.coordinator_cameras:
async_add_entities(
SynoDSMCamera(data.api, coordinator, camera_id)
diff --git a/homeassistant/components/synology_dsm/common.py b/homeassistant/components/synology_dsm/common.py
index dfc372e6bde..2e80624ca5d 100644
--- a/homeassistant/components/synology_dsm/common.py
+++ b/homeassistant/components/synology_dsm/common.py
@@ -7,6 +7,7 @@ from collections.abc import Callable
from contextlib import suppress
import logging
+from awesomeversion import AwesomeVersion
from synology_dsm import SynologyDSM
from synology_dsm.api.core.security import SynoCoreSecurity
from synology_dsm.api.core.system import SynoCoreSystem
@@ -35,13 +36,17 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConfigEntryAuthFailed
+from homeassistant.helpers import issue_registry as ir
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import (
+ CONF_BACKUP_PATH,
CONF_DEVICE_TOKEN,
DEFAULT_TIMEOUT,
+ DOMAIN,
EXCEPTION_DETAILS,
EXCEPTION_UNKNOWN,
+ ISSUE_MISSING_BACKUP_SETUP,
SYNOLOGY_CONNECTION_EXCEPTIONS,
)
@@ -131,6 +136,9 @@ class SynoApi:
)
await self.async_login()
+ self.information = self.dsm.information
+ await self.information.update()
+
# check if surveillance station is used
self._with_surveillance_station = bool(
self.dsm.apis.get(SynoSurveillanceStation.CAMERA_API_KEY)
@@ -161,7 +169,10 @@ class SynoApi:
LOGGER.debug("Disabled fetching upgrade data during setup: %s", ex)
# check if file station is used and permitted
- self._with_file_station = bool(self.dsm.apis.get(SynoFileStation.LIST_API_KEY))
+ self._with_file_station = bool(
+ self.information.awesome_version >= AwesomeVersion("6.0")
+ and self.dsm.apis.get(SynoFileStation.LIST_API_KEY)
+ )
if self._with_file_station:
shares: list | None = None
with suppress(*SYNOLOGY_CONNECTION_EXCEPTIONS):
@@ -174,6 +185,19 @@ class SynoApi:
" permissions or no writable shared folders available"
)
+ if shares and not self._entry.options.get(CONF_BACKUP_PATH):
+ ir.async_create_issue(
+ self._hass,
+ DOMAIN,
+ f"{ISSUE_MISSING_BACKUP_SETUP}_{self._entry.unique_id}",
+ data={"entry_id": self._entry.entry_id},
+ is_fixable=True,
+ is_persistent=False,
+ severity=ir.IssueSeverity.WARNING,
+ translation_key=ISSUE_MISSING_BACKUP_SETUP,
+ translation_placeholders={"title": self._entry.title},
+ )
+
LOGGER.debug(
"State of File Station during setup of '%s': %s",
self._entry.unique_id,
@@ -300,7 +324,6 @@ class SynoApi:
async def _fetch_device_configuration(self) -> None:
"""Fetch initial device config."""
- self.information = self.dsm.information
self.network = self.dsm.network
await self.network.update()
diff --git a/homeassistant/components/synology_dsm/config_flow.py b/homeassistant/components/synology_dsm/config_flow.py
index b4453366718..f0da6f8fe47 100644
--- a/homeassistant/components/synology_dsm/config_flow.py
+++ b/homeassistant/components/synology_dsm/config_flow.py
@@ -33,14 +33,12 @@ from homeassistant.const import (
CONF_NAME,
CONF_PASSWORD,
CONF_PORT,
- CONF_SCAN_INTERVAL,
CONF_SSL,
CONF_USERNAME,
CONF_VERIFY_SSL,
)
from homeassistant.core import callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.selector import (
SelectOptionDict,
@@ -67,7 +65,6 @@ from .const import (
DEFAULT_BACKUP_PATH,
DEFAULT_PORT,
DEFAULT_PORT_SSL,
- DEFAULT_SCAN_INTERVAL,
DEFAULT_SNAPSHOT_QUALITY,
DEFAULT_TIMEOUT,
DEFAULT_USE_SSL,
@@ -75,7 +72,7 @@ from .const import (
DOMAIN,
SYNOLOGY_CONNECTION_EXCEPTIONS,
)
-from .models import SynologyDSMData
+from .coordinator import SynologyDSMConfigEntry
_LOGGER = logging.getLogger(__name__)
@@ -134,7 +131,7 @@ class SynologyDSMFlowHandler(ConfigFlow, domain=DOMAIN):
@staticmethod
@callback
def async_get_options_flow(
- config_entry: ConfigEntry,
+ config_entry: SynologyDSMConfigEntry,
) -> SynologyDSMOptionsFlowHandler:
"""Get the options flow for this handler."""
return SynologyDSMOptionsFlowHandler()
@@ -447,6 +444,8 @@ class SynologyDSMFlowHandler(ConfigFlow, domain=DOMAIN):
class SynologyDSMOptionsFlowHandler(OptionsFlow):
"""Handle a option flow."""
+ config_entry: SynologyDSMConfigEntry
+
async def async_step_init(
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
@@ -454,16 +453,10 @@ class SynologyDSMOptionsFlowHandler(OptionsFlow):
if user_input is not None:
return self.async_create_entry(title="", data=user_input)
- syno_data: SynologyDSMData = self.hass.data[DOMAIN][self.config_entry.unique_id]
+ syno_data = self.config_entry.runtime_data
data_schema = vol.Schema(
{
- vol.Required(
- CONF_SCAN_INTERVAL,
- default=self.config_entry.options.get(
- CONF_SCAN_INTERVAL, DEFAULT_SCAN_INTERVAL
- ),
- ): cv.positive_int,
vol.Required(
CONF_SNAPSHOT_QUALITY,
default=self.config_entry.options.get(
diff --git a/homeassistant/components/synology_dsm/const.py b/homeassistant/components/synology_dsm/const.py
index dbee85b99d6..758fad53970 100644
--- a/homeassistant/components/synology_dsm/const.py
+++ b/homeassistant/components/synology_dsm/const.py
@@ -35,6 +35,8 @@ PLATFORMS = [
EXCEPTION_DETAILS = "details"
EXCEPTION_UNKNOWN = "unknown"
+ISSUE_MISSING_BACKUP_SETUP = "missing_backup_setup"
+
# Configuration
CONF_SERIAL = "serial"
CONF_VOLUMES = "volumes"
@@ -48,7 +50,6 @@ DEFAULT_VERIFY_SSL = False
DEFAULT_PORT = 5000
DEFAULT_PORT_SSL = 5001
# Options
-DEFAULT_SCAN_INTERVAL = 15 # min
DEFAULT_TIMEOUT = ClientTimeout(total=60, connect=15)
DEFAULT_SNAPSHOT_QUALITY = SNAPSHOT_PROFILE_BALANCED
DEFAULT_BACKUP_PATH = "ha_backup"
diff --git a/homeassistant/components/synology_dsm/coordinator.py b/homeassistant/components/synology_dsm/coordinator.py
index 30d1260ef32..a35432f0774 100644
--- a/homeassistant/components/synology_dsm/coordinator.py
+++ b/homeassistant/components/synology_dsm/coordinator.py
@@ -3,6 +3,7 @@
from __future__ import annotations
from collections.abc import Awaitable, Callable, Coroutine
+from dataclasses import dataclass
from datetime import timedelta
import logging
from typing import Any, Concatenate
@@ -14,14 +15,12 @@ from synology_dsm.exceptions import (
)
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_SCAN_INTERVAL
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .common import SynoApi, raise_config_entry_auth_error
from .const import (
- DEFAULT_SCAN_INTERVAL,
SIGNAL_CAMERA_SOURCE_CHANGED,
SYNOLOGY_AUTH_FAILED_EXCEPTIONS,
SYNOLOGY_CONNECTION_EXCEPTIONS,
@@ -30,6 +29,19 @@ from .const import (
_LOGGER = logging.getLogger(__name__)
+@dataclass
+class SynologyDSMData:
+ """Data for the synology_dsm integration."""
+
+ api: SynoApi
+ coordinator_central: SynologyDSMCentralUpdateCoordinator
+ coordinator_cameras: SynologyDSMCameraUpdateCoordinator | None
+ coordinator_switches: SynologyDSMSwitchUpdateCoordinator | None
+
+
+type SynologyDSMConfigEntry = ConfigEntry[SynologyDSMData]
+
+
def async_re_login_on_expired[_T: SynologyDSMUpdateCoordinator[Any], **_P, _R](
func: Callable[Concatenate[_T, _P], Awaitable[_R]],
) -> Callable[Concatenate[_T, _P], Coroutine[Any, Any, _R]]:
@@ -59,12 +71,12 @@ def async_re_login_on_expired[_T: SynologyDSMUpdateCoordinator[Any], **_P, _R](
class SynologyDSMUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
"""DataUpdateCoordinator base class for synology_dsm."""
- config_entry: ConfigEntry
+ config_entry: SynologyDSMConfigEntry
def __init__(
self,
hass: HomeAssistant,
- entry: ConfigEntry,
+ entry: SynologyDSMConfigEntry,
api: SynoApi,
update_interval: timedelta,
) -> None:
@@ -87,7 +99,7 @@ class SynologyDSMSwitchUpdateCoordinator(
def __init__(
self,
hass: HomeAssistant,
- entry: ConfigEntry,
+ entry: SynologyDSMConfigEntry,
api: SynoApi,
) -> None:
"""Initialize DataUpdateCoordinator for switch devices."""
@@ -118,18 +130,11 @@ class SynologyDSMCentralUpdateCoordinator(SynologyDSMUpdateCoordinator[None]):
def __init__(
self,
hass: HomeAssistant,
- entry: ConfigEntry,
+ entry: SynologyDSMConfigEntry,
api: SynoApi,
) -> None:
"""Initialize DataUpdateCoordinator for central device."""
- super().__init__(
- hass,
- entry,
- api,
- timedelta(
- minutes=entry.options.get(CONF_SCAN_INTERVAL, DEFAULT_SCAN_INTERVAL)
- ),
- )
+ super().__init__(hass, entry, api, timedelta(minutes=15))
@async_re_login_on_expired
async def _async_update_data(self) -> None:
@@ -145,7 +150,7 @@ class SynologyDSMCameraUpdateCoordinator(
def __init__(
self,
hass: HomeAssistant,
- entry: ConfigEntry,
+ entry: SynologyDSMConfigEntry,
api: SynoApi,
) -> None:
"""Initialize DataUpdateCoordinator for cameras."""
diff --git a/homeassistant/components/synology_dsm/diagnostics.py b/homeassistant/components/synology_dsm/diagnostics.py
index b30955ae682..a673be23096 100644
--- a/homeassistant/components/synology_dsm/diagnostics.py
+++ b/homeassistant/components/synology_dsm/diagnostics.py
@@ -6,21 +6,20 @@ from typing import Any
from homeassistant.components.camera import diagnostics as camera_diagnostics
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
-from .const import CONF_DEVICE_TOKEN, DOMAIN
-from .models import SynologyDSMData
+from .const import CONF_DEVICE_TOKEN
+from .coordinator import SynologyDSMConfigEntry
TO_REDACT = {CONF_USERNAME, CONF_PASSWORD, CONF_DEVICE_TOKEN}
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, entry: ConfigEntry
+ hass: HomeAssistant, entry: SynologyDSMConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
syno_api = data.api
dsm_info = syno_api.dsm.information
diff --git a/homeassistant/components/synology_dsm/manifest.json b/homeassistant/components/synology_dsm/manifest.json
index ab6fc20b5cb..3804de7f3f1 100644
--- a/homeassistant/components/synology_dsm/manifest.json
+++ b/homeassistant/components/synology_dsm/manifest.json
@@ -7,7 +7,7 @@
"documentation": "https://www.home-assistant.io/integrations/synology_dsm",
"iot_class": "local_polling",
"loggers": ["synology_dsm"],
- "requirements": ["py-synologydsm-api==2.6.0"],
+ "requirements": ["py-synologydsm-api==2.7.1"],
"ssdp": [
{
"manufacturer": "Synology",
diff --git a/homeassistant/components/synology_dsm/media_source.py b/homeassistant/components/synology_dsm/media_source.py
index d35b262809c..6234f5e8dd0 100644
--- a/homeassistant/components/synology_dsm/media_source.py
+++ b/homeassistant/components/synology_dsm/media_source.py
@@ -2,6 +2,7 @@
from __future__ import annotations
+from logging import getLogger
import mimetypes
from aiohttp import web
@@ -22,7 +23,9 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from .const import DOMAIN, SHARED_SUFFIX
-from .models import SynologyDSMData
+from .coordinator import SynologyDSMConfigEntry, SynologyDSMData
+
+LOGGER = getLogger(__name__)
async def async_get_media_source(hass: HomeAssistant) -> MediaSource:
@@ -41,15 +44,13 @@ class SynologyPhotosMediaSourceIdentifier:
"""Split identifier into parts."""
parts = identifier.split("/")
- self.unique_id = None
+ self.unique_id = parts[0]
self.album_id = None
self.cache_key = None
self.file_name = None
self.is_shared = False
self.passphrase = ""
- self.unique_id = parts[0]
-
if len(parts) > 1:
album_parts = parts[1].split("_")
self.album_id = album_parts[0]
@@ -82,7 +83,7 @@ class SynologyPhotosMediaSource(MediaSource):
item: MediaSourceItem,
) -> BrowseMediaSource:
"""Return media."""
- if not self.hass.data.get(DOMAIN):
+ if not self.hass.config_entries.async_loaded_entries(DOMAIN):
raise BrowseError("Diskstation not initialized")
return BrowseMediaSource(
domain=DOMAIN,
@@ -116,7 +117,13 @@ class SynologyPhotosMediaSource(MediaSource):
for entry in self.entries
]
identifier = SynologyPhotosMediaSourceIdentifier(item.identifier)
- diskstation: SynologyDSMData = self.hass.data[DOMAIN][identifier.unique_id]
+ entry: SynologyDSMConfigEntry | None = (
+ self.hass.config_entries.async_entry_for_domain_unique_id(
+ DOMAIN, identifier.unique_id
+ )
+ )
+ assert entry
+ diskstation = entry.runtime_data
assert diskstation.api.photos is not None
if identifier.album_id is None:
@@ -244,7 +251,7 @@ class SynologyDsmMediaView(http.HomeAssistantView):
self, request: web.Request, source_dir_id: str, location: str
) -> web.Response:
"""Start a GET request."""
- if not self.hass.data.get(DOMAIN):
+ if not self.hass.config_entries.async_loaded_entries(DOMAIN):
raise web.HTTPNotFound
# location: {cache_key}/{filename}
cache_key, file_name, passphrase = location.split("/")
@@ -257,7 +264,13 @@ class SynologyDsmMediaView(http.HomeAssistantView):
if not isinstance(mime_type, str):
raise web.HTTPNotFound
- diskstation: SynologyDSMData = self.hass.data[DOMAIN][source_dir_id]
+ entry: SynologyDSMConfigEntry | None = (
+ self.hass.config_entries.async_entry_for_domain_unique_id(
+ DOMAIN, source_dir_id
+ )
+ )
+ assert entry
+ diskstation = entry.runtime_data
assert diskstation.api.photos is not None
item = SynoPhotosItem(image_id, "", "", "", cache_key, "xl", shared, passphrase)
try:
diff --git a/homeassistant/components/synology_dsm/models.py b/homeassistant/components/synology_dsm/models.py
deleted file mode 100644
index 4f51d329ded..00000000000
--- a/homeassistant/components/synology_dsm/models.py
+++ /dev/null
@@ -1,22 +0,0 @@
-"""The synology_dsm integration models."""
-
-from __future__ import annotations
-
-from dataclasses import dataclass
-
-from .common import SynoApi
-from .coordinator import (
- SynologyDSMCameraUpdateCoordinator,
- SynologyDSMCentralUpdateCoordinator,
- SynologyDSMSwitchUpdateCoordinator,
-)
-
-
-@dataclass
-class SynologyDSMData:
- """Data for the synology_dsm integration."""
-
- api: SynoApi
- coordinator_central: SynologyDSMCentralUpdateCoordinator
- coordinator_cameras: SynologyDSMCameraUpdateCoordinator | None
- coordinator_switches: SynologyDSMSwitchUpdateCoordinator | None
diff --git a/homeassistant/components/synology_dsm/repairs.py b/homeassistant/components/synology_dsm/repairs.py
new file mode 100644
index 00000000000..8a4e47a32b5
--- /dev/null
+++ b/homeassistant/components/synology_dsm/repairs.py
@@ -0,0 +1,124 @@
+"""Repair flows for the Synology DSM integration."""
+
+from __future__ import annotations
+
+from contextlib import suppress
+import logging
+from typing import cast
+
+from synology_dsm.api.file_station.models import SynoFileSharedFolder
+import voluptuous as vol
+
+from homeassistant import data_entry_flow
+from homeassistant.components.repairs import ConfirmRepairFlow, RepairsFlow
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import issue_registry as ir
+from homeassistant.helpers.selector import (
+ SelectOptionDict,
+ SelectSelector,
+ SelectSelectorConfig,
+ SelectSelectorMode,
+)
+
+from .const import (
+ CONF_BACKUP_PATH,
+ CONF_BACKUP_SHARE,
+ DOMAIN,
+ ISSUE_MISSING_BACKUP_SETUP,
+ SYNOLOGY_CONNECTION_EXCEPTIONS,
+)
+from .coordinator import SynologyDSMConfigEntry
+
+LOGGER = logging.getLogger(__name__)
+
+
+class MissingBackupSetupRepairFlow(RepairsFlow):
+ """Handler for an issue fixing flow."""
+
+ def __init__(self, entry: SynologyDSMConfigEntry, issue_id: str) -> None:
+ """Create flow."""
+ self.entry = entry
+ self.issue_id = issue_id
+ super().__init__()
+
+ async def async_step_init(
+ self, user_input: dict[str, str] | None = None
+ ) -> data_entry_flow.FlowResult:
+ """Handle the first step of a fix flow."""
+
+ return self.async_show_menu(
+ menu_options=["confirm", "ignore"],
+ description_placeholders={
+ "docs_url": "https://www.home-assistant.io/integrations/synology_dsm/#backup-location"
+ },
+ )
+
+ async def async_step_confirm(
+ self, user_input: dict[str, str] | None = None
+ ) -> data_entry_flow.FlowResult:
+ """Handle the confirm step of a fix flow."""
+
+ syno_data = self.entry.runtime_data
+
+ if user_input is not None:
+ self.hass.config_entries.async_update_entry(
+ self.entry, options={**dict(self.entry.options), **user_input}
+ )
+ return self.async_create_entry(data={})
+
+ shares: list[SynoFileSharedFolder] | None = None
+ if syno_data.api.file_station:
+ with suppress(*SYNOLOGY_CONNECTION_EXCEPTIONS):
+ shares = await syno_data.api.file_station.get_shared_folders(
+ only_writable=True
+ )
+
+ if not shares:
+ return self.async_abort(reason="no_shares")
+
+ return self.async_show_form(
+ data_schema=vol.Schema(
+ {
+ vol.Required(
+ CONF_BACKUP_SHARE,
+ default=self.entry.options[CONF_BACKUP_SHARE],
+ ): SelectSelector(
+ SelectSelectorConfig(
+ options=[
+ SelectOptionDict(value=s.path, label=s.name)
+ for s in shares
+ ],
+ mode=SelectSelectorMode.DROPDOWN,
+ ),
+ ),
+ vol.Required(
+ CONF_BACKUP_PATH,
+ default=self.entry.options[CONF_BACKUP_PATH],
+ ): str,
+ }
+ ),
+ )
+
+ async def async_step_ignore(
+ self, _: dict[str, str] | None = None
+ ) -> data_entry_flow.FlowResult:
+ """Handle the confirm step of a fix flow."""
+ ir.async_ignore_issue(self.hass, DOMAIN, self.issue_id, True)
+ return self.async_abort(reason="ignored")
+
+
+async def async_create_fix_flow(
+ hass: HomeAssistant,
+ issue_id: str,
+ data: dict[str, str | int | float | None] | None,
+) -> RepairsFlow:
+ """Create flow."""
+ entry = None
+ if data and (entry_id := data.get("entry_id")):
+ entry_id = cast(str, entry_id)
+ entry = hass.config_entries.async_get_entry(entry_id)
+
+ if entry and issue_id.startswith(ISSUE_MISSING_BACKUP_SETUP):
+ return MissingBackupSetupRepairFlow(entry, issue_id)
+
+ return ConfirmRepairFlow()
diff --git a/homeassistant/components/synology_dsm/sensor.py b/homeassistant/components/synology_dsm/sensor.py
index b29a33f7253..566885e3989 100644
--- a/homeassistant/components/synology_dsm/sensor.py
+++ b/homeassistant/components/synology_dsm/sensor.py
@@ -16,7 +16,6 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
CONF_DISKS,
PERCENTAGE,
@@ -26,19 +25,18 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
from . import SynoApi
-from .const import CONF_VOLUMES, DOMAIN, ENTITY_UNIT_LOAD
-from .coordinator import SynologyDSMCentralUpdateCoordinator
+from .const import CONF_VOLUMES, ENTITY_UNIT_LOAD
+from .coordinator import SynologyDSMCentralUpdateCoordinator, SynologyDSMConfigEntry
from .entity import (
SynologyDSMBaseEntity,
SynologyDSMDeviceEntity,
SynologyDSMEntityDescription,
)
-from .models import SynologyDSMData
@dataclass(frozen=True, kw_only=True)
@@ -286,10 +284,12 @@ INFORMATION_SENSORS: tuple[SynologyDSMSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: SynologyDSMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Synology NAS Sensor."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
api = data.api
coordinator = data.coordinator_central
storage = api.storage
diff --git a/homeassistant/components/synology_dsm/service.py b/homeassistant/components/synology_dsm/service.py
index 366f7d4ba3a..40b6fd4bc30 100644
--- a/homeassistant/components/synology_dsm/service.py
+++ b/homeassistant/components/synology_dsm/service.py
@@ -3,13 +3,14 @@
from __future__ import annotations
import logging
+from typing import cast
from synology_dsm.exceptions import SynologyDSMException
from homeassistant.core import HomeAssistant, ServiceCall
from .const import CONF_SERIAL, DOMAIN, SERVICE_REBOOT, SERVICE_SHUTDOWN, SERVICES
-from .models import SynologyDSMData
+from .coordinator import SynologyDSMConfigEntry
LOGGER = logging.getLogger(__name__)
@@ -19,11 +20,20 @@ async def async_setup_services(hass: HomeAssistant) -> None:
async def service_handler(call: ServiceCall) -> None:
"""Handle service call."""
- serial = call.data.get(CONF_SERIAL)
- dsm_devices = hass.data[DOMAIN]
+ serial: str | None = call.data.get(CONF_SERIAL)
+ entries: list[SynologyDSMConfigEntry] = (
+ hass.config_entries.async_loaded_entries(DOMAIN)
+ )
+ dsm_devices = {
+ cast(str, entry.unique_id): entry.runtime_data for entry in entries
+ }
if serial:
- dsm_device: SynologyDSMData = hass.data[DOMAIN][serial]
+ entry: SynologyDSMConfigEntry | None = (
+ hass.config_entries.async_entry_for_domain_unique_id(DOMAIN, serial)
+ )
+ assert entry
+ dsm_device = entry.runtime_data
elif len(dsm_devices) == 1:
dsm_device = next(iter(dsm_devices.values()))
serial = next(iter(dsm_devices))
@@ -39,7 +49,7 @@ async def async_setup_services(hass: HomeAssistant) -> None:
return
if call.service in [SERVICE_REBOOT, SERVICE_SHUTDOWN]:
- if serial not in hass.data[DOMAIN]:
+ if serial not in dsm_devices:
LOGGER.error("DSM with specified serial %s not found", serial)
return
LOGGER.debug("%s DSM with serial %s", call.service, serial)
@@ -50,7 +60,7 @@ async def async_setup_services(hass: HomeAssistant) -> None:
),
call.service,
)
- dsm_device = hass.data[DOMAIN][serial]
+ dsm_device = dsm_devices[serial]
dsm_api = dsm_device.api
try:
await getattr(dsm_api, f"async_{call.service}")()
diff --git a/homeassistant/components/synology_dsm/strings.json b/homeassistant/components/synology_dsm/strings.json
index d6d40be3fea..f51184ef1cb 100644
--- a/homeassistant/components/synology_dsm/strings.json
+++ b/homeassistant/components/synology_dsm/strings.json
@@ -68,8 +68,6 @@
"step": {
"init": {
"data": {
- "scan_interval": "Minutes between scans",
- "timeout": "Timeout (seconds)",
"snap_profile_type": "Quality level of camera snapshots (0:high 1:medium 2:low)",
"backup_share": "[%key:component::synology_dsm::config::step::backup_share::data::backup_share%]",
"backup_path": "[%key:component::synology_dsm::config::step::backup_share::data::backup_path%]"
@@ -187,6 +185,37 @@
}
}
},
+ "issues": {
+ "missing_backup_setup": {
+ "title": "Backup location not configured for {title}",
+ "fix_flow": {
+ "step": {
+ "init": {
+ "description": "The backup location for {title} is not configured. Do you want to set it up now? Details can be found in the integration documentation under [Backup Location]({docs_url})",
+ "menu_options": {
+ "confirm": "Set up the backup location now",
+ "ignore": "Don't set it up now"
+ }
+ },
+ "confirm": {
+ "title": "[%key:component::synology_dsm::config::step::backup_share::title%]",
+ "data": {
+ "backup_share": "[%key:component::synology_dsm::config::step::backup_share::data::backup_share%]",
+ "backup_path": "[%key:component::synology_dsm::config::step::backup_share::data::backup_path%]"
+ },
+ "data_description": {
+ "backup_share": "[%key:component::synology_dsm::config::step::backup_share::data_description::backup_share%]",
+ "backup_path": "[%key:component::synology_dsm::config::step::backup_share::data_description::backup_path%]"
+ }
+ }
+ },
+ "abort": {
+ "no_shares": "There are no shared folders available for the user.\nPlease check the documentation.",
+ "ignored": "The backup location has not been configured.\nYou can still set it up later via the integration options."
+ }
+ }
+ }
+ },
"services": {
"reboot": {
"name": "Reboot",
diff --git a/homeassistant/components/synology_dsm/switch.py b/homeassistant/components/synology_dsm/switch.py
index facce824bda..91863ff3a26 100644
--- a/homeassistant/components/synology_dsm/switch.py
+++ b/homeassistant/components/synology_dsm/switch.py
@@ -9,16 +9,14 @@ from typing import Any
from synology_dsm.api.surveillance_station import SynoSurveillanceStation
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import SynoApi
from .const import DOMAIN
-from .coordinator import SynologyDSMSwitchUpdateCoordinator
+from .coordinator import SynologyDSMConfigEntry, SynologyDSMSwitchUpdateCoordinator
from .entity import SynologyDSMBaseEntity, SynologyDSMEntityDescription
-from .models import SynologyDSMData
_LOGGER = logging.getLogger(__name__)
@@ -40,10 +38,12 @@ SURVEILLANCE_SWITCH: tuple[SynologyDSMSwitchEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: SynologyDSMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Synology NAS switch."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
if coordinator := data.coordinator_switches:
assert coordinator.version is not None
async_add_entities(
diff --git a/homeassistant/components/synology_dsm/update.py b/homeassistant/components/synology_dsm/update.py
index ed60191f296..3048a38cb9c 100644
--- a/homeassistant/components/synology_dsm/update.py
+++ b/homeassistant/components/synology_dsm/update.py
@@ -9,15 +9,12 @@ from synology_dsm.api.core.upgrade import SynoCoreUpgrade
from yarl import URL
from homeassistant.components.update import UpdateEntity, UpdateEntityDescription
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
-from .coordinator import SynologyDSMCentralUpdateCoordinator
+from .coordinator import SynologyDSMCentralUpdateCoordinator, SynologyDSMConfigEntry
from .entity import SynologyDSMBaseEntity, SynologyDSMEntityDescription
-from .models import SynologyDSMData
@dataclass(frozen=True, kw_only=True)
@@ -38,10 +35,12 @@ UPDATE_ENTITIES: Final = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: SynologyDSMConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Synology DSM update entities."""
- data: SynologyDSMData = hass.data[DOMAIN][entry.unique_id]
+ data = entry.runtime_data
async_add_entities(
SynoDSMUpdateEntity(data.api, data.coordinator_central, description)
for description in UPDATE_ENTITIES
diff --git a/homeassistant/components/system_bridge/binary_sensor.py b/homeassistant/components/system_bridge/binary_sensor.py
index 019b1df4639..0140499a75a 100644
--- a/homeassistant/components/system_bridge/binary_sensor.py
+++ b/homeassistant/components/system_bridge/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PORT
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SystemBridgeDataUpdateCoordinator
@@ -65,7 +65,9 @@ BATTERY_BINARY_SENSOR_TYPES: tuple[SystemBridgeBinarySensorEntityDescription, ..
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up System Bridge binary sensor based on a config entry."""
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/system_bridge/coordinator.py b/homeassistant/components/system_bridge/coordinator.py
index 7151805f154..1690bad4a4d 100644
--- a/homeassistant/components/system_bridge/coordinator.py
+++ b/homeassistant/components/system_bridge/coordinator.py
@@ -40,6 +40,8 @@ from .data import SystemBridgeData
class SystemBridgeDataUpdateCoordinator(DataUpdateCoordinator[SystemBridgeData]):
"""Class to manage fetching System Bridge data from single endpoint."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
@@ -65,6 +67,7 @@ class SystemBridgeDataUpdateCoordinator(DataUpdateCoordinator[SystemBridgeData])
super().__init__(
hass,
LOGGER,
+ config_entry=entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/system_bridge/media_player.py b/homeassistant/components/system_bridge/media_player.py
index aeff3b22fb2..6d3bbd21a05 100644
--- a/homeassistant/components/system_bridge/media_player.py
+++ b/homeassistant/components/system_bridge/media_player.py
@@ -18,7 +18,7 @@ from homeassistant.components.media_player import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PORT
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SystemBridgeDataUpdateCoordinator
@@ -66,7 +66,7 @@ MEDIA_PLAYER_DESCRIPTION: Final[MediaPlayerEntityDescription] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up System Bridge media players based on a config entry."""
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/system_bridge/sensor.py b/homeassistant/components/system_bridge/sensor.py
index 94c73a2ac05..c7cae2f347b 100644
--- a/homeassistant/components/system_bridge/sensor.py
+++ b/homeassistant/components/system_bridge/sensor.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UNDEFINED, StateType
from homeassistant.util import dt as dt_util
@@ -359,7 +359,7 @@ BATTERY_SENSOR_TYPES: tuple[SystemBridgeSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up System Bridge sensor based on a config entry."""
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/system_bridge/strings.json b/homeassistant/components/system_bridge/strings.json
index ef7495ef74f..1c079c1ef0c 100644
--- a/homeassistant/components/system_bridge/strings.json
+++ b/homeassistant/components/system_bridge/strings.json
@@ -109,7 +109,7 @@
"message": "No data received from {host}"
},
"process_not_found": {
- "message": "Could not find process with id {id}."
+ "message": "Could not find process with ID {id}."
},
"timeout": {
"message": "A timeout occurred for {title} ({host})"
@@ -120,7 +120,7 @@
},
"issues": {
"unsupported_version": {
- "title": "System Bridge Upgrade Required",
+ "title": "System Bridge upgrade required",
"description": "Your version of System Bridge for host {host} is not supported.\n\nPlease upgrade to the latest version."
}
},
diff --git a/homeassistant/components/system_bridge/update.py b/homeassistant/components/system_bridge/update.py
index b0d341cee3b..12060c28669 100644
--- a/homeassistant/components/system_bridge/update.py
+++ b/homeassistant/components/system_bridge/update.py
@@ -6,7 +6,7 @@ from homeassistant.components.update import UpdateEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PORT
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import SystemBridgeDataUpdateCoordinator
@@ -16,7 +16,7 @@ from .entity import SystemBridgeEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up System Bridge update based on a config entry."""
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/systemmonitor/binary_sensor.py b/homeassistant/components/systemmonitor/binary_sensor.py
index aecd30765ff..3968e94ec03 100644
--- a/homeassistant/components/systemmonitor/binary_sensor.py
+++ b/homeassistant/components/systemmonitor/binary_sensor.py
@@ -20,7 +20,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import slugify
@@ -91,7 +91,7 @@ SENSOR_TYPES: tuple[SysMonitorBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: SystemMonitorConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up System Monitor binary sensors based on a config entry."""
coordinator = entry.runtime_data.coordinator
diff --git a/homeassistant/components/systemmonitor/sensor.py b/homeassistant/components/systemmonitor/sensor.py
index 048d7cefd6c..e70bccf0833 100644
--- a/homeassistant/components/systemmonitor/sensor.py
+++ b/homeassistant/components/systemmonitor/sensor.py
@@ -31,7 +31,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import slugify
@@ -397,7 +397,7 @@ IF_ADDRS_FAMILY = {"ipv4_address": socket.AF_INET, "ipv6_address": socket.AF_INE
async def async_setup_entry(
hass: HomeAssistant,
entry: SystemMonitorConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up System Monitor sensors based on a config entry."""
entities: list[SystemMonitorSensor] = []
diff --git a/homeassistant/components/tado/__init__.py b/homeassistant/components/tado/__init__.py
index 3e42e33489f..4b0203acda3 100644
--- a/homeassistant/components/tado/__init__.py
+++ b/homeassistant/components/tado/__init__.py
@@ -31,6 +31,7 @@ PLATFORMS = [
Platform.CLIMATE,
Platform.DEVICE_TRACKER,
Platform.SENSOR,
+ Platform.SWITCH,
Platform.WATER_HEATER,
]
@@ -87,7 +88,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: TadoConfigEntry) -> bool
@callback
-def _async_import_options_from_data_if_missing(hass: HomeAssistant, entry: ConfigEntry):
+def _async_import_options_from_data_if_missing(
+ hass: HomeAssistant, entry: TadoConfigEntry
+):
options = dict(entry.options)
if CONF_FALLBACK not in options:
options[CONF_FALLBACK] = entry.data.get(
diff --git a/homeassistant/components/tado/binary_sensor.py b/homeassistant/components/tado/binary_sensor.py
index c969ea34f42..8cec32e20f0 100644
--- a/homeassistant/components/tado/binary_sensor.py
+++ b/homeassistant/components/tado/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TadoConfigEntry
@@ -115,7 +115,9 @@ ZONE_SENSORS = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TadoConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TadoConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tado sensor platform."""
diff --git a/homeassistant/components/tado/climate.py b/homeassistant/components/tado/climate.py
index db7b1823bd9..6a2067ffff1 100644
--- a/homeassistant/components/tado/climate.py
+++ b/homeassistant/components/tado/climate.py
@@ -26,7 +26,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_TENTHS, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from . import TadoConfigEntry
@@ -100,7 +100,9 @@ CLIMATE_TEMP_OFFSET_SCHEMA: VolDictType = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TadoConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TadoConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tado climate platform."""
@@ -155,8 +157,8 @@ async def create_climate_entity(
TADO_TO_HA_HVAC_MODE_MAP[CONST_MODE_OFF],
TADO_TO_HA_HVAC_MODE_MAP[CONST_MODE_SMART_SCHEDULE],
]
- supported_fan_modes = None
- supported_swing_modes = None
+ supported_fan_modes: list[str] | None = None
+ supported_swing_modes: list[str] | None = None
heat_temperatures = None
cool_temperatures = None
diff --git a/homeassistant/components/tado/coordinator.py b/homeassistant/components/tado/coordinator.py
index ddec9e7f292..559bc4a16fb 100644
--- a/homeassistant/components/tado/coordinator.py
+++ b/homeassistant/components/tado/coordinator.py
@@ -4,18 +4,20 @@ from __future__ import annotations
from datetime import datetime, timedelta
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
from PyTado.interface import Tado
from requests import RequestException
from homeassistant.components.climate import PRESET_AWAY, PRESET_HOME
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import TadoConfigEntry
+
from .const import (
CONF_FALLBACK,
CONST_OVERLAY_TADO_DEFAULT,
@@ -31,8 +33,6 @@ MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=4)
SCAN_INTERVAL = timedelta(minutes=5)
SCAN_MOBILE_DEVICE_INTERVAL = timedelta(seconds=30)
-type TadoConfigEntry = ConfigEntry[TadoDataUpdateCoordinator]
-
class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
"""Class to manage API calls from and to Tado via PyTado."""
@@ -45,7 +45,7 @@ class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
def __init__(
self,
hass: HomeAssistant,
- entry: ConfigEntry,
+ config_entry: TadoConfigEntry,
tado: Tado,
debug: bool = False,
) -> None:
@@ -53,13 +53,16 @@ class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
self._tado = tado
- self._username = entry.data[CONF_USERNAME]
- self._password = entry.data[CONF_PASSWORD]
- self._fallback = entry.options.get(CONF_FALLBACK, CONST_OVERLAY_TADO_DEFAULT)
+ self._username = config_entry.data[CONF_USERNAME]
+ self._password = config_entry.data[CONF_PASSWORD]
+ self._fallback = config_entry.options.get(
+ CONF_FALLBACK, CONST_OVERLAY_TADO_DEFAULT
+ )
self._debug = debug
self.home_id: int
@@ -339,20 +342,34 @@ class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
except RequestException as err:
raise UpdateFailed(f"Error setting Tado meter reading: {err}") from err
+ async def set_child_lock(self, device_id: str, enabled: bool) -> None:
+ """Set child lock of device."""
+ try:
+ await self.hass.async_add_executor_job(
+ self._tado.set_child_lock,
+ device_id,
+ enabled,
+ )
+ except RequestException as exc:
+ raise HomeAssistantError(f"Error setting Tado child lock: {exc}") from exc
+
class TadoMobileDeviceUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
"""Class to manage the mobile devices from Tado via PyTado."""
+ config_entry: TadoConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- entry: ConfigEntry,
+ config_entry: TadoConfigEntry,
tado: Tado,
) -> None:
"""Initialize the Tado data update coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_MOBILE_DEVICE_INTERVAL,
)
diff --git a/homeassistant/components/tado/device_tracker.py b/homeassistant/components/tado/device_tracker.py
index a9be560f434..34aca2dd833 100644
--- a/homeassistant/components/tado/device_tracker.py
+++ b/homeassistant/components/tado/device_tracker.py
@@ -11,7 +11,7 @@ from homeassistant.components.device_tracker import (
from homeassistant.const import STATE_HOME, STATE_NOT_HOME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: TadoConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tado device scannery entity."""
_LOGGER.debug("Setting up Tado device scanner entity")
@@ -57,7 +57,7 @@ async def async_setup_entry(
def add_tracked_entities(
hass: HomeAssistant,
coordinator: TadoMobileDeviceUpdateCoordinator,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
tracked: set[str],
) -> None:
"""Add new tracker entities from Tado."""
diff --git a/homeassistant/components/tado/helper.py b/homeassistant/components/tado/helper.py
index 571a757a3e8..5c515e00cf0 100644
--- a/homeassistant/components/tado/helper.py
+++ b/homeassistant/components/tado/helper.py
@@ -53,13 +53,13 @@ def decide_duration(
return duration
-def generate_supported_fanmodes(tado_to_ha_mapping: dict[str, str], options: list[str]):
+def generate_supported_fanmodes(
+ tado_to_ha_mapping: dict[str, str], options: list[str]
+) -> list[str] | None:
"""Return correct list of fan modes or None."""
supported_fanmodes = [
- tado_to_ha_mapping.get(option)
- for option in options
- if tado_to_ha_mapping.get(option) is not None
+ val for option in options if (val := tado_to_ha_mapping.get(option)) is not None
]
if not supported_fanmodes:
return None
diff --git a/homeassistant/components/tado/icons.json b/homeassistant/components/tado/icons.json
index c799bef0260..65b86359950 100644
--- a/homeassistant/components/tado/icons.json
+++ b/homeassistant/components/tado/icons.json
@@ -1,4 +1,14 @@
{
+ "entity": {
+ "switch": {
+ "child_lock": {
+ "default": "mdi:lock-open-variant",
+ "state": {
+ "on": "mdi:lock"
+ }
+ }
+ }
+ },
"services": {
"set_climate_timer": {
"service": "mdi:timer"
diff --git a/homeassistant/components/tado/sensor.py b/homeassistant/components/tado/sensor.py
index 037b33574e7..d0d54e79670 100644
--- a/homeassistant/components/tado/sensor.py
+++ b/homeassistant/components/tado/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TadoConfigEntry
@@ -191,7 +191,9 @@ ZONE_SENSORS = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TadoConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TadoConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tado sensor platform."""
diff --git a/homeassistant/components/tado/strings.json b/homeassistant/components/tado/strings.json
index f1550517457..ff1afc3c03d 100644
--- a/homeassistant/components/tado/strings.json
+++ b/homeassistant/components/tado/strings.json
@@ -64,6 +64,11 @@
}
}
},
+ "switch": {
+ "child_lock": {
+ "name": "Child lock"
+ }
+ },
"sensor": {
"outdoor_temperature": {
"name": "Outdoor temperature"
diff --git a/homeassistant/components/tado/switch.py b/homeassistant/components/tado/switch.py
new file mode 100644
index 00000000000..b3f355462b8
--- /dev/null
+++ b/homeassistant/components/tado/switch.py
@@ -0,0 +1,88 @@
+"""Module for Tado child lock switch entity."""
+
+import logging
+from typing import Any
+
+from homeassistant.components.switch import SwitchEntity
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import TadoConfigEntry
+from .entity import TadoDataUpdateCoordinator, TadoZoneEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: TadoConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the Tado switch platform."""
+
+ tado = entry.runtime_data.coordinator
+ entities: list[TadoChildLockSwitchEntity] = []
+ for zone in tado.zones:
+ zoneChildLockSupported = (
+ len(zone["devices"]) > 0 and "childLockEnabled" in zone["devices"][0]
+ )
+
+ if not zoneChildLockSupported:
+ continue
+
+ entities.append(
+ TadoChildLockSwitchEntity(
+ tado, zone["name"], zone["id"], zone["devices"][0]
+ )
+ )
+ async_add_entities(entities, True)
+
+
+class TadoChildLockSwitchEntity(TadoZoneEntity, SwitchEntity):
+ """Representation of a Tado child lock switch entity."""
+
+ _attr_translation_key = "child_lock"
+
+ def __init__(
+ self,
+ coordinator: TadoDataUpdateCoordinator,
+ zone_name: str,
+ zone_id: int,
+ device_info: dict[str, Any],
+ ) -> None:
+ """Initialize the Tado child lock switch entity."""
+ super().__init__(zone_name, coordinator.home_id, zone_id, coordinator)
+
+ self._device_info = device_info
+ self._device_id = self._device_info["shortSerialNo"]
+ self._attr_unique_id = f"{zone_id} {coordinator.home_id} child-lock"
+
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn the entity on."""
+ await self.coordinator.set_child_lock(self._device_id, True)
+ await self.coordinator.async_request_refresh()
+
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn the entity off."""
+ await self.coordinator.set_child_lock(self._device_id, False)
+ await self.coordinator.async_request_refresh()
+
+ @callback
+ def _handle_coordinator_update(self) -> None:
+ """Handle updated data from the coordinator."""
+ self._async_update_callback()
+ super()._handle_coordinator_update()
+
+ @callback
+ def _async_update_callback(self) -> None:
+ """Handle update callbacks."""
+ try:
+ self._device_info = self.coordinator.data["device"][self._device_id]
+ except KeyError:
+ _LOGGER.error(
+ "Could not update child lock info for device %s in zone %s",
+ self._device_id,
+ self.zone_name,
+ )
+ else:
+ self._attr_is_on = self._device_info.get("childLockEnabled", False) is True
diff --git a/homeassistant/components/tado/water_heater.py b/homeassistant/components/tado/water_heater.py
index 02fbb3f5e23..3d8825b264f 100644
--- a/homeassistant/components/tado/water_heater.py
+++ b/homeassistant/components/tado/water_heater.py
@@ -12,7 +12,7 @@ from homeassistant.components.water_heater import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from . import TadoConfigEntry
@@ -61,7 +61,9 @@ WATER_HEATER_TIMER_SCHEMA: VolDictType = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TadoConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TadoConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tado water heater platform."""
diff --git a/homeassistant/components/tailscale/binary_sensor.py b/homeassistant/components/tailscale/binary_sensor.py
index 981f871de09..6569b40ada2 100644
--- a/homeassistant/components/tailscale/binary_sensor.py
+++ b/homeassistant/components/tailscale/binary_sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import TailscaleEntity
@@ -84,7 +84,7 @@ BINARY_SENSORS: tuple[TailscaleBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Tailscale binary sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tailscale/coordinator.py b/homeassistant/components/tailscale/coordinator.py
index 64ce0147664..1b29cfbf4be 100644
--- a/homeassistant/components/tailscale/coordinator.py
+++ b/homeassistant/components/tailscale/coordinator.py
@@ -19,18 +19,22 @@ class TailscaleDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Device]]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize the Tailscale coordinator."""
- self.config_entry = entry
-
session = async_get_clientsession(hass)
self.tailscale = Tailscale(
session=session,
- api_key=entry.data[CONF_API_KEY],
- tailnet=entry.data[CONF_TAILNET],
+ api_key=config_entry.data[CONF_API_KEY],
+ tailnet=config_entry.data[CONF_TAILNET],
)
- super().__init__(hass, LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
+ )
async def _async_update_data(self) -> dict[str, Device]:
"""Fetch devices from Tailscale."""
diff --git a/homeassistant/components/tailscale/sensor.py b/homeassistant/components/tailscale/sensor.py
index fa4c966a7d7..cf944aa73ef 100644
--- a/homeassistant/components/tailscale/sensor.py
+++ b/homeassistant/components/tailscale/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import TailscaleEntity
@@ -55,7 +55,7 @@ SENSORS: tuple[TailscaleSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Tailscale sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tailwind/binary_sensor.py b/homeassistant/components/tailwind/binary_sensor.py
index d2f8e1e2ced..4d927b0769e 100644
--- a/homeassistant/components/tailwind/binary_sensor.py
+++ b/homeassistant/components/tailwind/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import TailwindConfigEntry
from .entity import TailwindDoorEntity
@@ -41,7 +41,7 @@ DESCRIPTIONS: tuple[TailwindDoorBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TailwindConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tailwind binary sensor based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/tailwind/button.py b/homeassistant/components/tailwind/button.py
index edff3434866..380eb7ccd7e 100644
--- a/homeassistant/components/tailwind/button.py
+++ b/homeassistant/components/tailwind/button.py
@@ -16,7 +16,7 @@ from homeassistant.components.button import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import TailwindConfigEntry
@@ -43,7 +43,7 @@ DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: TailwindConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tailwind button based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/tailwind/cover.py b/homeassistant/components/tailwind/cover.py
index 8ea1c7d4f6d..84f38c7d579 100644
--- a/homeassistant/components/tailwind/cover.py
+++ b/homeassistant/components/tailwind/cover.py
@@ -20,7 +20,7 @@ from homeassistant.components.cover import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, LOGGER
from .coordinator import TailwindConfigEntry
@@ -30,7 +30,7 @@ from .entity import TailwindDoorEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: TailwindConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tailwind cover based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/tailwind/number.py b/homeassistant/components/tailwind/number.py
index b67df9a6a25..ca6b610c351 100644
--- a/homeassistant/components/tailwind/number.py
+++ b/homeassistant/components/tailwind/number.py
@@ -12,7 +12,7 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescriptio
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import TailwindConfigEntry
@@ -47,7 +47,7 @@ DESCRIPTIONS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: TailwindConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tailwind number based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/tami4/__init__.py b/homeassistant/components/tami4/__init__.py
index 8c597409c77..8b9a5e1a90f 100644
--- a/homeassistant/components/tami4/__init__.py
+++ b/homeassistant/components/tami4/__init__.py
@@ -26,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except exceptions.TokenRefreshFailedException as ex:
raise ConfigEntryNotReady("Error connecting to API") from ex
- coordinator = Tami4EdgeCoordinator(hass, api)
+ coordinator = Tami4EdgeCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = {
diff --git a/homeassistant/components/tami4/button.py b/homeassistant/components/tami4/button.py
index 11377a2dcfb..a1b8db79674 100644
--- a/homeassistant/components/tami4/button.py
+++ b/homeassistant/components/tami4/button.py
@@ -11,7 +11,7 @@ from homeassistant.components.button import ButtonEntity, ButtonEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import API, DOMAIN
from .entity import Tami4EdgeBaseEntity
@@ -41,7 +41,9 @@ BOIL_WATER_BUTTON = Tami4EdgeButtonEntityDescription(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Tami4Edge."""
diff --git a/homeassistant/components/tami4/coordinator.py b/homeassistant/components/tami4/coordinator.py
index 4764562bc34..f65c819b3d8 100644
--- a/homeassistant/components/tami4/coordinator.py
+++ b/homeassistant/components/tami4/coordinator.py
@@ -7,6 +7,7 @@ import logging
from Tami4EdgeAPI import Tami4EdgeAPI, exceptions
from Tami4EdgeAPI.water_quality import WaterQuality
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -36,11 +37,16 @@ class FlattenedWaterQuality:
class Tami4EdgeCoordinator(DataUpdateCoordinator[FlattenedWaterQuality]):
"""Tami4Edge water quality coordinator."""
- def __init__(self, hass: HomeAssistant, api: Tami4EdgeAPI) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: Tami4EdgeAPI
+ ) -> None:
"""Initialize the water quality coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Tami4Edge water quality coordinator",
update_interval=timedelta(minutes=60),
)
diff --git a/homeassistant/components/tami4/sensor.py b/homeassistant/components/tami4/sensor.py
index 888acda9372..2bfd3079c19 100644
--- a/homeassistant/components/tami4/sensor.py
+++ b/homeassistant/components/tami4/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfVolume
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import API, COORDINATOR, DOMAIN
@@ -52,7 +52,9 @@ ENTITY_DESCRIPTIONS = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Tami4Edge."""
data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tankerkoenig/binary_sensor.py b/homeassistant/components/tankerkoenig/binary_sensor.py
index 774262a8854..a38266e57e8 100644
--- a/homeassistant/components/tankerkoenig/binary_sensor.py
+++ b/homeassistant/components/tankerkoenig/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import TankerkoenigConfigEntry, TankerkoenigDataUpdateCoordinator
from .entity import TankerkoenigCoordinatorEntity
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: TankerkoenigConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the tankerkoenig binary sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/tankerkoenig/config_flow.py b/homeassistant/components/tankerkoenig/config_flow.py
index 8796ae46ab7..b269eaaaf55 100644
--- a/homeassistant/components/tankerkoenig/config_flow.py
+++ b/homeassistant/components/tankerkoenig/config_flow.py
@@ -39,7 +39,7 @@ from homeassistant.helpers.selector import (
NumberSelectorConfig,
)
-from .const import CONF_FUEL_TYPES, CONF_STATIONS, DEFAULT_RADIUS, DOMAIN, FUEL_TYPES
+from .const import CONF_STATIONS, DEFAULT_RADIUS, DOMAIN
async def async_get_nearby_stations(
@@ -175,10 +175,6 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
vol.Required(
CONF_API_KEY, default=user_input.get(CONF_API_KEY, "")
): cv.string,
- vol.Required(
- CONF_FUEL_TYPES,
- default=user_input.get(CONF_FUEL_TYPES, list(FUEL_TYPES)),
- ): cv.multi_select(FUEL_TYPES),
vol.Required(
CONF_LOCATION,
default=user_input.get(
diff --git a/homeassistant/components/tankerkoenig/const.py b/homeassistant/components/tankerkoenig/const.py
index c2a1dba9b6a..6761d20f4ce 100644
--- a/homeassistant/components/tankerkoenig/const.py
+++ b/homeassistant/components/tankerkoenig/const.py
@@ -3,14 +3,11 @@
DOMAIN = "tankerkoenig"
NAME = "tankerkoenig"
-CONF_FUEL_TYPES = "fuel_types"
CONF_STATIONS = "stations"
DEFAULT_RADIUS = 2
DEFAULT_SCAN_INTERVAL = 30
-FUEL_TYPES = {"e5": "Super", "e10": "Super E10", "diesel": "Diesel"}
-
ATTR_BRAND = "brand"
ATTR_CITY = "city"
ATTR_FUEL_TYPE = "fuel_type"
diff --git a/homeassistant/components/tankerkoenig/coordinator.py b/homeassistant/components/tankerkoenig/coordinator.py
index 1f73d0577b3..f1e6bc8c865 100644
--- a/homeassistant/components/tankerkoenig/coordinator.py
+++ b/homeassistant/components/tankerkoenig/coordinator.py
@@ -24,7 +24,7 @@ from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import CONF_FUEL_TYPES, CONF_STATIONS, DOMAIN
+from .const import CONF_STATIONS, DOMAIN
_LOGGER = logging.getLogger(__name__)
@@ -54,7 +54,6 @@ class TankerkoenigDataUpdateCoordinator(DataUpdateCoordinator[dict[str, PriceInf
self._selected_stations: list[str] = self.config_entry.data[CONF_STATIONS]
self.stations: dict[str, Station] = {}
- self.fuel_types: list[str] = self.config_entry.data[CONF_FUEL_TYPES]
self.show_on_map: bool = self.config_entry.options[CONF_SHOW_ON_MAP]
self._tankerkoenig = Tankerkoenig(
diff --git a/homeassistant/components/tankerkoenig/sensor.py b/homeassistant/components/tankerkoenig/sensor.py
index 5970f3d3b24..b1646489d96 100644
--- a/homeassistant/components/tankerkoenig/sensor.py
+++ b/homeassistant/components/tankerkoenig/sensor.py
@@ -9,7 +9,7 @@ from aiotankerkoenig import GasType, Station
from homeassistant.components.sensor import SensorEntity, SensorStateClass
from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE, CURRENCY_EURO
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_BRAND,
@@ -30,7 +30,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: TankerkoenigConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the tankerkoenig sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/tankerkoenig/strings.json b/homeassistant/components/tankerkoenig/strings.json
index 29f4f439dd5..db620b2b11c 100644
--- a/homeassistant/components/tankerkoenig/strings.json
+++ b/homeassistant/components/tankerkoenig/strings.json
@@ -5,7 +5,6 @@
"data": {
"name": "Region name",
"api_key": "[%key:common::config_flow::data::api_key%]",
- "fuel_types": "Fuel types",
"location": "[%key:common::config_flow::data::location%]",
"stations": "Additional fuel stations",
"radius": "Search radius"
diff --git a/homeassistant/components/tasmota/binary_sensor.py b/homeassistant/components/tasmota/binary_sensor.py
index 22cdf1a5ff0..3b2e640b807 100644
--- a/homeassistant/components/tasmota/binary_sensor.py
+++ b/homeassistant/components/tasmota/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import event as evt
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_REMOVE_DISCOVER_COMPONENT
from .discovery import TASMOTA_DISCOVERY_ENTITY_NEW
@@ -26,7 +26,7 @@ from .entity import TasmotaAvailability, TasmotaDiscoveryUpdate
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tasmota binary sensor dynamically through discovery."""
diff --git a/homeassistant/components/tasmota/cover.py b/homeassistant/components/tasmota/cover.py
index 2cb3cfeea25..1d7aa8316b6 100644
--- a/homeassistant/components/tasmota/cover.py
+++ b/homeassistant/components/tasmota/cover.py
@@ -18,7 +18,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_REMOVE_DISCOVER_COMPONENT
from .discovery import TASMOTA_DISCOVERY_ENTITY_NEW
@@ -28,7 +28,7 @@ from .entity import TasmotaAvailability, TasmotaDiscoveryUpdate
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tasmota cover dynamically through discovery."""
diff --git a/homeassistant/components/tasmota/fan.py b/homeassistant/components/tasmota/fan.py
index e927bd6ad72..c89b36577be 100644
--- a/homeassistant/components/tasmota/fan.py
+++ b/homeassistant/components/tasmota/fan.py
@@ -16,7 +16,7 @@ from homeassistant.components.fan import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -36,7 +36,7 @@ ORDERED_NAMED_FAN_SPEEDS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tasmota fan dynamically through discovery."""
diff --git a/homeassistant/components/tasmota/light.py b/homeassistant/components/tasmota/light.py
index a06e77eceb1..ed66fa128dc 100644
--- a/homeassistant/components/tasmota/light.py
+++ b/homeassistant/components/tasmota/light.py
@@ -31,7 +31,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .const import DATA_REMOVE_DISCOVER_COMPONENT
@@ -45,7 +45,7 @@ TASMOTA_BRIGHTNESS_MAX = 100
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tasmota light dynamically through discovery."""
diff --git a/homeassistant/components/tasmota/manifest.json b/homeassistant/components/tasmota/manifest.json
index 783483c6ffd..2e0d8af2338 100644
--- a/homeassistant/components/tasmota/manifest.json
+++ b/homeassistant/components/tasmota/manifest.json
@@ -8,5 +8,5 @@
"iot_class": "local_push",
"loggers": ["hatasmota"],
"mqtt": ["tasmota/discovery/#"],
- "requirements": ["HATasmota==0.9.2"]
+ "requirements": ["HATasmota==0.10.0"]
}
diff --git a/homeassistant/components/tasmota/sensor.py b/homeassistant/components/tasmota/sensor.py
index 8cc538e706a..ec20e1c0348 100644
--- a/homeassistant/components/tasmota/sensor.py
+++ b/homeassistant/components/tasmota/sensor.py
@@ -40,7 +40,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_REMOVE_DISCOVER_COMPONENT
from .discovery import TASMOTA_DISCOVERY_ENTITY_NEW
@@ -243,7 +243,7 @@ SENSOR_UNIT_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tasmota sensor dynamically through discovery."""
diff --git a/homeassistant/components/tasmota/switch.py b/homeassistant/components/tasmota/switch.py
index b5c19fc2431..03e594b125c 100644
--- a/homeassistant/components/tasmota/switch.py
+++ b/homeassistant/components/tasmota/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_REMOVE_DISCOVER_COMPONENT
from .discovery import TASMOTA_DISCOVERY_ENTITY_NEW
@@ -21,7 +21,7 @@ from .entity import TasmotaAvailability, TasmotaDiscoveryUpdate, TasmotaOnOffEnt
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tasmota switch dynamically through discovery."""
diff --git a/homeassistant/components/tautulli/__init__.py b/homeassistant/components/tautulli/__init__.py
index a031354ae7d..41089016fac 100644
--- a/homeassistant/components/tautulli/__init__.py
+++ b/homeassistant/components/tautulli/__init__.py
@@ -4,15 +4,13 @@ from __future__ import annotations
from pytautulli import PyTautulli, PyTautulliHostConfiguration
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY, CONF_URL, CONF_VERIFY_SSL, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from .coordinator import TautulliDataUpdateCoordinator
+from .coordinator import TautulliConfigEntry, TautulliDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR]
-type TautulliConfigEntry = ConfigEntry[TautulliDataUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: TautulliConfigEntry) -> bool:
@@ -27,7 +25,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TautulliConfigEntry) ->
session=async_get_clientsession(hass, entry.data[CONF_VERIFY_SSL]),
)
entry.runtime_data = TautulliDataUpdateCoordinator(
- hass, host_configuration, api_client
+ hass, entry, host_configuration, api_client
)
await entry.runtime_data.async_config_entry_first_refresh()
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/tautulli/coordinator.py b/homeassistant/components/tautulli/coordinator.py
index f392ab8df03..5d0f26b83b6 100644
--- a/homeassistant/components/tautulli/coordinator.py
+++ b/homeassistant/components/tautulli/coordinator.py
@@ -4,7 +4,6 @@ from __future__ import annotations
import asyncio
from datetime import timedelta
-from typing import TYPE_CHECKING
from pytautulli import (
PyTautulli,
@@ -18,14 +17,14 @@ from pytautulli.exceptions import (
)
from pytautulli.models.host_configuration import PyTautulliHostConfiguration
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN, LOGGER
-if TYPE_CHECKING:
- from . import TautulliConfigEntry
+type TautulliConfigEntry = ConfigEntry[TautulliDataUpdateCoordinator]
class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
@@ -36,6 +35,7 @@ class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: TautulliConfigEntry,
host_configuration: PyTautulliHostConfiguration,
api_client: PyTautulli,
) -> None:
@@ -43,6 +43,7 @@ class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=10),
)
diff --git a/homeassistant/components/tautulli/sensor.py b/homeassistant/components/tautulli/sensor.py
index cd21630031a..c8d35623c21 100644
--- a/homeassistant/components/tautulli/sensor.py
+++ b/homeassistant/components/tautulli/sensor.py
@@ -23,12 +23,14 @@ from homeassistant.config_entries import SOURCE_IMPORT
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfInformation
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, StateType
-from . import TautulliConfigEntry
from .const import ATTR_TOP_USER, DOMAIN
-from .coordinator import TautulliDataUpdateCoordinator
+from .coordinator import TautulliConfigEntry, TautulliDataUpdateCoordinator
from .entity import TautulliEntity
@@ -213,7 +215,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: TautulliConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tautulli sensor."""
data = entry.runtime_data
diff --git a/homeassistant/components/technove/__init__.py b/homeassistant/components/technove/__init__.py
index b886dabc80c..df4fc7713aa 100644
--- a/homeassistant/components/technove/__init__.py
+++ b/homeassistant/components/technove/__init__.py
@@ -2,16 +2,13 @@
from __future__ import annotations
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from .coordinator import TechnoVEDataUpdateCoordinator
+from .coordinator import TechnoVEConfigEntry, TechnoVEDataUpdateCoordinator
PLATFORMS = [Platform.BINARY_SENSOR, Platform.NUMBER, Platform.SENSOR, Platform.SWITCH]
-TechnoVEConfigEntry = ConfigEntry[TechnoVEDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: TechnoVEConfigEntry) -> bool:
"""Set up TechnoVE from a config entry."""
@@ -25,6 +22,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: TechnoVEConfigEntry) ->
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: TechnoVEConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/technove/binary_sensor.py b/homeassistant/components/technove/binary_sensor.py
index f231e206c96..ac52a19884e 100644
--- a/homeassistant/components/technove/binary_sensor.py
+++ b/homeassistant/components/technove/binary_sensor.py
@@ -14,10 +14,9 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import TechnoVEConfigEntry
-from .coordinator import TechnoVEDataUpdateCoordinator
+from .coordinator import TechnoVEConfigEntry, TechnoVEDataUpdateCoordinator
from .entity import TechnoVEEntity
@@ -66,7 +65,7 @@ BINARY_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: TechnoVEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary sensor platform."""
async_add_entities(
diff --git a/homeassistant/components/technove/coordinator.py b/homeassistant/components/technove/coordinator.py
index 8527c6e543a..53108463301 100644
--- a/homeassistant/components/technove/coordinator.py
+++ b/homeassistant/components/technove/coordinator.py
@@ -2,10 +2,9 @@
from __future__ import annotations
-from typing import TYPE_CHECKING
-
from technove import Station as TechnoVEStation, TechnoVE, TechnoVEError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -13,22 +12,24 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN, LOGGER, SCAN_INTERVAL
-if TYPE_CHECKING:
- from . import TechnoVEConfigEntry
+type TechnoVEConfigEntry = ConfigEntry[TechnoVEDataUpdateCoordinator]
class TechnoVEDataUpdateCoordinator(DataUpdateCoordinator[TechnoVEStation]):
"""Class to manage fetching TechnoVE data from single endpoint."""
- def __init__(self, hass: HomeAssistant, entry: TechnoVEConfigEntry) -> None:
+ config_entry: TechnoVEConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: TechnoVEConfigEntry) -> None:
"""Initialize global TechnoVE data updater."""
self.technove = TechnoVE(
- entry.data[CONF_HOST],
+ config_entry.data[CONF_HOST],
session=async_get_clientsession(hass),
)
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/technove/diagnostics.py b/homeassistant/components/technove/diagnostics.py
index f070d58ab6f..7ac0f6f44fd 100644
--- a/homeassistant/components/technove/diagnostics.py
+++ b/homeassistant/components/technove/diagnostics.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant
-from . import TechnoVEConfigEntry
+from .coordinator import TechnoVEConfigEntry
TO_REDACT = {"unique_id", "mac_address"}
diff --git a/homeassistant/components/technove/manifest.json b/homeassistant/components/technove/manifest.json
index 722aa4004e1..746c2280aaa 100644
--- a/homeassistant/components/technove/manifest.json
+++ b/homeassistant/components/technove/manifest.json
@@ -6,6 +6,6 @@
"documentation": "https://www.home-assistant.io/integrations/technove",
"integration_type": "device",
"iot_class": "local_polling",
- "requirements": ["python-technove==1.3.1"],
+ "requirements": ["python-technove==2.0.0"],
"zeroconf": ["_technove-stations._tcp.local."]
}
diff --git a/homeassistant/components/technove/number.py b/homeassistant/components/technove/number.py
index a1cf094c6bf..11d8f281276 100644
--- a/homeassistant/components/technove/number.py
+++ b/homeassistant/components/technove/number.py
@@ -17,11 +17,10 @@ from homeassistant.components.number import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import TechnoVEConfigEntry
from .const import DOMAIN
-from .coordinator import TechnoVEDataUpdateCoordinator
+from .coordinator import TechnoVEConfigEntry, TechnoVEDataUpdateCoordinator
from .entity import TechnoVEEntity
from .helpers import technove_exception_handler
@@ -66,7 +65,7 @@ NUMBERS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: TechnoVEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up TechnoVE number entity based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/technove/sensor.py b/homeassistant/components/technove/sensor.py
index e16ac23f89c..398c1911cd4 100644
--- a/homeassistant/components/technove/sensor.py
+++ b/homeassistant/components/technove/sensor.py
@@ -21,11 +21,10 @@ from homeassistant.const import (
UnitOfEnergy,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import TechnoVEConfigEntry
-from .coordinator import TechnoVEDataUpdateCoordinator
+from .coordinator import TechnoVEConfigEntry, TechnoVEDataUpdateCoordinator
from .entity import TechnoVEEntity
STATUS_TYPE = [s.value for s in Status if s != Status.UNKNOWN]
@@ -122,7 +121,7 @@ SENSORS: tuple[TechnoVESensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TechnoVEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
async_add_entities(
diff --git a/homeassistant/components/technove/strings.json b/homeassistant/components/technove/strings.json
index 9976f0b3c59..05260845a03 100644
--- a/homeassistant/components/technove/strings.json
+++ b/homeassistant/components/technove/strings.json
@@ -70,7 +70,7 @@
"plugged_waiting": "Plugged, waiting",
"plugged_charging": "Plugged, charging",
"out_of_activation_period": "Out of activation period",
- "high_charge_period": "High charge period"
+ "high_tariff_period": "High tariff period"
}
}
},
diff --git a/homeassistant/components/technove/switch.py b/homeassistant/components/technove/switch.py
index a8ad7581da5..19688075b35 100644
--- a/homeassistant/components/technove/switch.py
+++ b/homeassistant/components/technove/switch.py
@@ -12,11 +12,10 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import TechnoVEConfigEntry
from .const import DOMAIN
-from .coordinator import TechnoVEDataUpdateCoordinator
+from .coordinator import TechnoVEConfigEntry, TechnoVEDataUpdateCoordinator
from .entity import TechnoVEEntity
from .helpers import technove_exception_handler
@@ -80,7 +79,7 @@ SWITCHES = [
async def async_setup_entry(
hass: HomeAssistant,
entry: TechnoVEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up TechnoVE switch based on a config entry."""
diff --git a/homeassistant/components/tedee/binary_sensor.py b/homeassistant/components/tedee/binary_sensor.py
index 4f167619f04..a01b889ef8f 100644
--- a/homeassistant/components/tedee/binary_sensor.py
+++ b/homeassistant/components/tedee/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import TedeeConfigEntry
from .entity import TedeeDescriptionEntity
@@ -64,7 +64,7 @@ ENTITIES: tuple[TedeeBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TedeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tedee sensor entity."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/tedee/lock.py b/homeassistant/components/tedee/lock.py
index 482cd039a98..da6db242db3 100644
--- a/homeassistant/components/tedee/lock.py
+++ b/homeassistant/components/tedee/lock.py
@@ -7,7 +7,7 @@ from aiotedee import TedeeClientException, TedeeLock, TedeeLockState
from homeassistant.components.lock import LockEntity, LockEntityFeature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import TedeeApiCoordinator, TedeeConfigEntry
@@ -19,7 +19,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: TedeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tedee lock entity."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/tedee/sensor.py b/homeassistant/components/tedee/sensor.py
index 828793b4458..a697d36be50 100644
--- a/homeassistant/components/tedee/sensor.py
+++ b/homeassistant/components/tedee/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import TedeeConfigEntry
from .entity import TedeeDescriptionEntity
@@ -53,7 +53,7 @@ ENTITIES: tuple[TedeeSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TedeeConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tedee sensor entity."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/telegram_bot/__init__.py b/homeassistant/components/telegram_bot/__init__.py
index fa3ec1dc4f7..15e1f7d4f0e 100644
--- a/homeassistant/components/telegram_bot/__init__.py
+++ b/homeassistant/components/telegram_bot/__init__.py
@@ -548,7 +548,7 @@ class TelegramNotificationService:
"""Initialize the service."""
self.allowed_chat_ids = allowed_chat_ids
self._default_user = self.allowed_chat_ids[0]
- self._last_message_id = {user: None for user in self.allowed_chat_ids}
+ self._last_message_id = dict.fromkeys(self.allowed_chat_ids)
self._parsers = {
PARSER_HTML: ParseMode.HTML,
PARSER_MD: ParseMode.MARKDOWN,
@@ -756,7 +756,8 @@ class TelegramNotificationService:
message_thread_id=params[ATTR_MESSAGE_THREAD_ID],
context=context,
)
- msg_ids[chat_id] = msg.id
+ if msg is not None:
+ msg_ids[chat_id] = msg.id
return msg_ids
async def delete_message(self, chat_id=None, context=None, **kwargs):
diff --git a/homeassistant/components/telegram_bot/strings.json b/homeassistant/components/telegram_bot/strings.json
index 714e7b74db0..8f4894f42a7 100644
--- a/homeassistant/components/telegram_bot/strings.json
+++ b/homeassistant/components/telegram_bot/strings.json
@@ -96,7 +96,7 @@
},
"verify_ssl": {
"name": "Verify SSL",
- "description": "Enable or disable SSL certificate verification. Set to false if you're downloading the file from a URL and you don't want to validate the SSL certificate of the server."
+ "description": "Enable or disable SSL certificate verification. Disable if you're downloading the file from a URL and you don't want to validate the SSL certificate of the server."
},
"timeout": {
"name": "Read timeout",
@@ -530,11 +530,11 @@
},
"is_anonymous": {
"name": "Is anonymous",
- "description": "If the poll needs to be anonymous, defaults to True."
+ "description": "If the poll needs to be anonymous. This is the default."
},
"allows_multiple_answers": {
"name": "Allow multiple answers",
- "description": "If the poll allows multiple answers, defaults to False."
+ "description": "If the poll allows multiple answers."
},
"open_period": {
"name": "Open period",
diff --git a/homeassistant/components/tellduslive/binary_sensor.py b/homeassistant/components/tellduslive/binary_sensor.py
index 33f936beb54..65301708646 100644
--- a/homeassistant/components/tellduslive/binary_sensor.py
+++ b/homeassistant/components/tellduslive/binary_sensor.py
@@ -5,7 +5,7 @@ from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TELLDUS_DISCOVERY_NEW
from .entity import TelldusLiveEntity
@@ -14,7 +14,7 @@ from .entity import TelldusLiveEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tellduslive sensors dynamically."""
diff --git a/homeassistant/components/tellduslive/cover.py b/homeassistant/components/tellduslive/cover.py
index d55a72cd633..2554acc428c 100644
--- a/homeassistant/components/tellduslive/cover.py
+++ b/homeassistant/components/tellduslive/cover.py
@@ -7,7 +7,7 @@ from homeassistant.components.cover import CoverEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TelldusLiveClient
from .const import DOMAIN, TELLDUS_DISCOVERY_NEW
@@ -17,7 +17,7 @@ from .entity import TelldusLiveEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tellduslive sensors dynamically."""
diff --git a/homeassistant/components/tellduslive/entity.py b/homeassistant/components/tellduslive/entity.py
index a71fcb685c0..5366e4c27df 100644
--- a/homeassistant/components/tellduslive/entity.py
+++ b/homeassistant/components/tellduslive/entity.py
@@ -33,7 +33,7 @@ class TelldusLiveEntity(Entity):
self._id = device_id
self._client = client
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Call when entity is added to hass."""
_LOGGER.debug("Created device %s", self)
self.async_on_remove(
@@ -58,12 +58,12 @@ class TelldusLiveEntity(Entity):
return self.device.state
@property
- def assumed_state(self):
+ def assumed_state(self) -> bool:
"""Return true if unable to access real state of entity."""
return True
@property
- def available(self):
+ def available(self) -> bool:
"""Return true if device is not offline."""
return self._client.is_available(self.device_id)
diff --git a/homeassistant/components/tellduslive/light.py b/homeassistant/components/tellduslive/light.py
index 005bf97d8c0..9f291bb845a 100644
--- a/homeassistant/components/tellduslive/light.py
+++ b/homeassistant/components/tellduslive/light.py
@@ -8,7 +8,7 @@ from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEnti
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TELLDUS_DISCOVERY_NEW
from .entity import TelldusLiveEntity
@@ -19,7 +19,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tellduslive sensors dynamically."""
diff --git a/homeassistant/components/tellduslive/sensor.py b/homeassistant/components/tellduslive/sensor.py
index 9bd2b1fe599..782f240cc41 100644
--- a/homeassistant/components/tellduslive/sensor.py
+++ b/homeassistant/components/tellduslive/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TELLDUS_DISCOVERY_NEW
from .entity import TelldusLiveEntity
@@ -121,7 +121,7 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tellduslive sensors dynamically."""
diff --git a/homeassistant/components/tellduslive/switch.py b/homeassistant/components/tellduslive/switch.py
index bd770ab08f5..3ca2ba066ab 100644
--- a/homeassistant/components/tellduslive/switch.py
+++ b/homeassistant/components/tellduslive/switch.py
@@ -7,7 +7,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TELLDUS_DISCOVERY_NEW
from .entity import TelldusLiveEntity
@@ -16,7 +16,7 @@ from .entity import TelldusLiveEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tellduslive sensors dynamically."""
diff --git a/homeassistant/components/tellstick/entity.py b/homeassistant/components/tellstick/entity.py
index 746c7f4dd4d..5be3d1f48f4 100644
--- a/homeassistant/components/tellstick/entity.py
+++ b/homeassistant/components/tellstick/entity.py
@@ -40,7 +40,7 @@ class TellstickDevice(Entity):
self._attr_name = tellcore_device.name
self._attr_unique_id = tellcore_device.id
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register callbacks."""
self.async_on_remove(
async_dispatcher_connect(
@@ -146,6 +146,6 @@ class TellstickDevice(Entity):
except TelldusError as err:
_LOGGER.error(err)
- def update(self):
+ def update(self) -> None:
"""Poll the current state of the device."""
self._update_from_tellcore()
diff --git a/homeassistant/components/template/alarm_control_panel.py b/homeassistant/components/template/alarm_control_panel.py
index a67e2969f9a..40206a5ccbb 100644
--- a/homeassistant/components/template/alarm_control_panel.py
+++ b/homeassistant/components/template/alarm_control_panel.py
@@ -31,9 +31,11 @@ from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv, selector
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
from homeassistant.helpers.entity import async_generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.restore_state import RestoreEntity
-from homeassistant.helpers.script import Script
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import slugify
@@ -146,7 +148,7 @@ async def _async_create_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
@@ -196,70 +198,31 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity, Restore
name = self._attr_name
assert name is not None
self._template = config.get(CONF_VALUE_TEMPLATE)
- self._disarm_script = None
self._attr_code_arm_required: bool = config[CONF_CODE_ARM_REQUIRED]
self._attr_code_format = config[CONF_CODE_FORMAT].value
- if (disarm_action := config.get(CONF_DISARM_ACTION)) is not None:
- self._disarm_script = Script(hass, disarm_action, name, DOMAIN)
- self._arm_away_script = None
- if (arm_away_action := config.get(CONF_ARM_AWAY_ACTION)) is not None:
- self._arm_away_script = Script(hass, arm_away_action, name, DOMAIN)
- self._arm_home_script = None
- if (arm_home_action := config.get(CONF_ARM_HOME_ACTION)) is not None:
- self._arm_home_script = Script(hass, arm_home_action, name, DOMAIN)
- self._arm_night_script = None
- if (arm_night_action := config.get(CONF_ARM_NIGHT_ACTION)) is not None:
- self._arm_night_script = Script(hass, arm_night_action, name, DOMAIN)
- self._arm_vacation_script = None
- if (arm_vacation_action := config.get(CONF_ARM_VACATION_ACTION)) is not None:
- self._arm_vacation_script = Script(hass, arm_vacation_action, name, DOMAIN)
- self._arm_custom_bypass_script = None
- if (
- arm_custom_bypass_action := config.get(CONF_ARM_CUSTOM_BYPASS_ACTION)
- ) is not None:
- self._arm_custom_bypass_script = Script(
- hass, arm_custom_bypass_action, name, DOMAIN
- )
- self._trigger_script = None
- if (trigger_action := config.get(CONF_TRIGGER_ACTION)) is not None:
- self._trigger_script = Script(hass, trigger_action, name, DOMAIN)
+
+ self._attr_supported_features = AlarmControlPanelEntityFeature(0)
+ for action_id, supported_feature in (
+ (CONF_DISARM_ACTION, 0),
+ (CONF_ARM_AWAY_ACTION, AlarmControlPanelEntityFeature.ARM_AWAY),
+ (CONF_ARM_HOME_ACTION, AlarmControlPanelEntityFeature.ARM_HOME),
+ (CONF_ARM_NIGHT_ACTION, AlarmControlPanelEntityFeature.ARM_NIGHT),
+ (CONF_ARM_VACATION_ACTION, AlarmControlPanelEntityFeature.ARM_VACATION),
+ (
+ CONF_ARM_CUSTOM_BYPASS_ACTION,
+ AlarmControlPanelEntityFeature.ARM_CUSTOM_BYPASS,
+ ),
+ (CONF_TRIGGER_ACTION, AlarmControlPanelEntityFeature.TRIGGER),
+ ):
+ if action_config := config.get(action_id):
+ self.add_script(action_id, action_config, name, DOMAIN)
+ self._attr_supported_features |= supported_feature
self._state: AlarmControlPanelState | None = None
self._attr_device_info = async_device_info_to_link_from_device_id(
hass,
config.get(CONF_DEVICE_ID),
)
- supported_features = AlarmControlPanelEntityFeature(0)
- if self._arm_night_script is not None:
- supported_features = (
- supported_features | AlarmControlPanelEntityFeature.ARM_NIGHT
- )
-
- if self._arm_home_script is not None:
- supported_features = (
- supported_features | AlarmControlPanelEntityFeature.ARM_HOME
- )
-
- if self._arm_away_script is not None:
- supported_features = (
- supported_features | AlarmControlPanelEntityFeature.ARM_AWAY
- )
-
- if self._arm_vacation_script is not None:
- supported_features = (
- supported_features | AlarmControlPanelEntityFeature.ARM_VACATION
- )
-
- if self._arm_custom_bypass_script is not None:
- supported_features = (
- supported_features | AlarmControlPanelEntityFeature.ARM_CUSTOM_BYPASS
- )
-
- if self._trigger_script is not None:
- supported_features = (
- supported_features | AlarmControlPanelEntityFeature.TRIGGER
- )
- self._attr_supported_features = supported_features
async def async_added_to_hass(self) -> None:
"""Restore last state."""
@@ -327,7 +290,7 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity, Restore
"""Arm the panel to Away."""
await self._async_alarm_arm(
AlarmControlPanelState.ARMED_AWAY,
- script=self._arm_away_script,
+ script=self._action_scripts.get(CONF_ARM_AWAY_ACTION),
code=code,
)
@@ -335,7 +298,7 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity, Restore
"""Arm the panel to Home."""
await self._async_alarm_arm(
AlarmControlPanelState.ARMED_HOME,
- script=self._arm_home_script,
+ script=self._action_scripts.get(CONF_ARM_HOME_ACTION),
code=code,
)
@@ -343,7 +306,7 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity, Restore
"""Arm the panel to Night."""
await self._async_alarm_arm(
AlarmControlPanelState.ARMED_NIGHT,
- script=self._arm_night_script,
+ script=self._action_scripts.get(CONF_ARM_NIGHT_ACTION),
code=code,
)
@@ -351,7 +314,7 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity, Restore
"""Arm the panel to Vacation."""
await self._async_alarm_arm(
AlarmControlPanelState.ARMED_VACATION,
- script=self._arm_vacation_script,
+ script=self._action_scripts.get(CONF_ARM_VACATION_ACTION),
code=code,
)
@@ -359,20 +322,22 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity, Restore
"""Arm the panel to Custom Bypass."""
await self._async_alarm_arm(
AlarmControlPanelState.ARMED_CUSTOM_BYPASS,
- script=self._arm_custom_bypass_script,
+ script=self._action_scripts.get(CONF_ARM_CUSTOM_BYPASS_ACTION),
code=code,
)
async def async_alarm_disarm(self, code: str | None = None) -> None:
"""Disarm the panel."""
await self._async_alarm_arm(
- AlarmControlPanelState.DISARMED, script=self._disarm_script, code=code
+ AlarmControlPanelState.DISARMED,
+ script=self._action_scripts.get(CONF_DISARM_ACTION),
+ code=code,
)
async def async_alarm_trigger(self, code: str | None = None) -> None:
"""Trigger the panel."""
await self._async_alarm_arm(
AlarmControlPanelState.TRIGGERED,
- script=self._trigger_script,
+ script=self._action_scripts.get(CONF_TRIGGER_ACTION),
code=code,
)
diff --git a/homeassistant/components/template/binary_sensor.py b/homeassistant/components/template/binary_sensor.py
index 3c6e4899502..7ef64e8077b 100644
--- a/homeassistant/components/template/binary_sensor.py
+++ b/homeassistant/components/template/binary_sensor.py
@@ -43,7 +43,10 @@ from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv, selector, template
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
from homeassistant.helpers.entity import async_generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_call_later, async_track_point_in_utc_time
from homeassistant.helpers.restore_state import ExtraStoredData, RestoreEntity
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -150,7 +153,7 @@ PLATFORM_SCHEMA = BINARY_SENSOR_PLATFORM_SCHEMA.extend(
@callback
def _async_create_template_tracking_entities(
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddEntitiesCallback | AddConfigEntryEntitiesCallback,
hass: HomeAssistant,
definitions: list[dict],
unique_id_prefix: str | None,
@@ -209,7 +212,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
diff --git a/homeassistant/components/template/button.py b/homeassistant/components/template/button.py
index 67ce7e7a16b..7a205446585 100644
--- a/homeassistant/components/template/button.py
+++ b/homeassistant/components/template/button.py
@@ -19,8 +19,10 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers import config_validation as cv, selector
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import CONF_PRESS, DOMAIN
@@ -93,7 +95,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
@@ -118,11 +120,8 @@ class TemplateButtonEntity(TemplateEntity, ButtonEntity):
"""Initialize the button."""
super().__init__(hass, config=config, unique_id=unique_id)
assert self._attr_name is not None
- self._command_press = (
- Script(hass, config.get(CONF_PRESS), self._attr_name, DOMAIN)
- if config.get(CONF_PRESS, None) is not None
- else None
- )
+ if action := config.get(CONF_PRESS):
+ self.add_script(CONF_PRESS, action, self._attr_name, DOMAIN)
self._attr_device_class = config.get(CONF_DEVICE_CLASS)
self._attr_state = None
self._attr_device_info = async_device_info_to_link_from_device_id(
@@ -132,5 +131,5 @@ class TemplateButtonEntity(TemplateEntity, ButtonEntity):
async def async_press(self) -> None:
"""Press the button."""
- if self._command_press:
- await self.async_run_script(self._command_press, context=self._context)
+ if script := self._action_scripts.get(CONF_PRESS):
+ await self.async_run_script(script, context=self._context)
diff --git a/homeassistant/components/template/config.py b/homeassistant/components/template/config.py
index e0c5514def9..4e07d67f6e9 100644
--- a/homeassistant/components/template/config.py
+++ b/homeassistant/components/template/config.py
@@ -1,5 +1,6 @@
"""Template config validator."""
+from collections.abc import Callable
from contextlib import suppress
import logging
@@ -12,9 +13,11 @@ from homeassistant.components.blueprint import (
)
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN
from homeassistant.components.image import DOMAIN as IMAGE_DOMAIN
+from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN
from homeassistant.components.number import DOMAIN as NUMBER_DOMAIN
from homeassistant.components.select import DOMAIN as SELECT_DOMAIN
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
+from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.components.weather import DOMAIN as WEATHER_DOMAIN
from homeassistant.config import async_log_schema_error, config_without_domain
from homeassistant.const import (
@@ -35,9 +38,11 @@ from . import (
binary_sensor as binary_sensor_platform,
button as button_platform,
image as image_platform,
+ light as light_platform,
number as number_platform,
select as select_platform,
sensor as sensor_platform,
+ switch as switch_platform,
weather as weather_platform,
)
from .const import (
@@ -52,41 +57,71 @@ from .helpers import async_get_blueprints
PACKAGE_MERGE_HINT = "list"
+
+def ensure_domains_do_not_have_trigger_or_action(*keys: str) -> Callable[[dict], dict]:
+ """Validate that config does not contain trigger and action."""
+ domains = set(keys)
+
+ def validate(obj: dict):
+ options = set(obj.keys())
+ if found_domains := domains.intersection(options):
+ invalid = {CONF_TRIGGER, CONF_ACTION}
+ if found_invalid := invalid.intersection(set(obj.keys())):
+ raise vol.Invalid(
+ f"Unsupported option(s) found for domain {found_domains.pop()}, please remove ({', '.join(found_invalid)}) from your configuration",
+ )
+
+ return obj
+
+ return validate
+
+
CONFIG_SECTION_SCHEMA = vol.Schema(
- {
- vol.Optional(CONF_UNIQUE_ID): cv.string,
- vol.Optional(CONF_TRIGGER): cv.TRIGGER_SCHEMA,
- vol.Optional(CONF_CONDITION): cv.CONDITIONS_SCHEMA,
- vol.Optional(CONF_ACTION): cv.SCRIPT_SCHEMA,
- vol.Optional(CONF_VARIABLES): cv.SCRIPT_VARIABLES_SCHEMA,
- vol.Optional(NUMBER_DOMAIN): vol.All(
- cv.ensure_list, [number_platform.NUMBER_SCHEMA]
+ vol.All(
+ {
+ vol.Optional(CONF_UNIQUE_ID): cv.string,
+ vol.Optional(CONF_TRIGGER): cv.TRIGGER_SCHEMA,
+ vol.Optional(CONF_CONDITION): cv.CONDITIONS_SCHEMA,
+ vol.Optional(CONF_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_VARIABLES): cv.SCRIPT_VARIABLES_SCHEMA,
+ vol.Optional(NUMBER_DOMAIN): vol.All(
+ cv.ensure_list, [number_platform.NUMBER_SCHEMA]
+ ),
+ vol.Optional(SENSOR_DOMAIN): vol.All(
+ cv.ensure_list, [sensor_platform.SENSOR_SCHEMA]
+ ),
+ vol.Optional(CONF_SENSORS): cv.schema_with_slug_keys(
+ sensor_platform.LEGACY_SENSOR_SCHEMA
+ ),
+ vol.Optional(BINARY_SENSOR_DOMAIN): vol.All(
+ cv.ensure_list, [binary_sensor_platform.BINARY_SENSOR_SCHEMA]
+ ),
+ vol.Optional(CONF_BINARY_SENSORS): cv.schema_with_slug_keys(
+ binary_sensor_platform.LEGACY_BINARY_SENSOR_SCHEMA
+ ),
+ vol.Optional(SELECT_DOMAIN): vol.All(
+ cv.ensure_list, [select_platform.SELECT_SCHEMA]
+ ),
+ vol.Optional(BUTTON_DOMAIN): vol.All(
+ cv.ensure_list, [button_platform.BUTTON_SCHEMA]
+ ),
+ vol.Optional(IMAGE_DOMAIN): vol.All(
+ cv.ensure_list, [image_platform.IMAGE_SCHEMA]
+ ),
+ vol.Optional(LIGHT_DOMAIN): vol.All(
+ cv.ensure_list, [light_platform.LIGHT_SCHEMA]
+ ),
+ vol.Optional(WEATHER_DOMAIN): vol.All(
+ cv.ensure_list, [weather_platform.WEATHER_SCHEMA]
+ ),
+ vol.Optional(SWITCH_DOMAIN): vol.All(
+ cv.ensure_list, [switch_platform.SWITCH_SCHEMA]
+ ),
+ },
+ ensure_domains_do_not_have_trigger_or_action(
+ BUTTON_DOMAIN, LIGHT_DOMAIN, SWITCH_DOMAIN
),
- vol.Optional(SENSOR_DOMAIN): vol.All(
- cv.ensure_list, [sensor_platform.SENSOR_SCHEMA]
- ),
- vol.Optional(CONF_SENSORS): cv.schema_with_slug_keys(
- sensor_platform.LEGACY_SENSOR_SCHEMA
- ),
- vol.Optional(BINARY_SENSOR_DOMAIN): vol.All(
- cv.ensure_list, [binary_sensor_platform.BINARY_SENSOR_SCHEMA]
- ),
- vol.Optional(CONF_BINARY_SENSORS): cv.schema_with_slug_keys(
- binary_sensor_platform.LEGACY_BINARY_SENSOR_SCHEMA
- ),
- vol.Optional(SELECT_DOMAIN): vol.All(
- cv.ensure_list, [select_platform.SELECT_SCHEMA]
- ),
- vol.Optional(BUTTON_DOMAIN): vol.All(
- cv.ensure_list, [button_platform.BUTTON_SCHEMA]
- ),
- vol.Optional(IMAGE_DOMAIN): vol.All(
- cv.ensure_list, [image_platform.IMAGE_SCHEMA]
- ),
- vol.Optional(WEATHER_DOMAIN): vol.All(
- cv.ensure_list, [weather_platform.WEATHER_SCHEMA]
- ),
- },
+ )
)
TEMPLATE_BLUEPRINT_INSTANCE_SCHEMA = vol.Schema(
@@ -122,9 +157,15 @@ async def _async_resolve_blueprints(
raise vol.Invalid("more than one platform defined per blueprint")
if len(platforms) == 1:
platform = platforms.pop()
- for prop in (CONF_NAME, CONF_UNIQUE_ID, CONF_VARIABLES):
+ for prop in (CONF_NAME, CONF_UNIQUE_ID):
if prop in config:
config[platform][prop] = config.pop(prop)
+ # For regular template entities, CONF_VARIABLES should be removed because they just
+ # house input results for template entities. For Trigger based template entities
+ # CONF_VARIABLES should not be removed because the variables are always
+ # executed between the trigger and action.
+ if CONF_TRIGGER not in config and CONF_VARIABLES in config:
+ config[platform][CONF_VARIABLES] = config.pop(CONF_VARIABLES)
raw_config = dict(config)
template_config = TemplateConfig(CONFIG_SECTION_SCHEMA(config))
diff --git a/homeassistant/components/template/coordinator.py b/homeassistant/components/template/coordinator.py
index 4d8fe78f2b5..c11e9b6101b 100644
--- a/homeassistant/components/template/coordinator.py
+++ b/homeassistant/components/template/coordinator.py
@@ -2,12 +2,14 @@
from collections.abc import Callable, Mapping
import logging
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, cast
-from homeassistant.const import EVENT_HOMEASSISTANT_START
+from homeassistant.components.blueprint import CONF_USE_BLUEPRINT
+from homeassistant.const import CONF_PATH, CONF_VARIABLES, EVENT_HOMEASSISTANT_START
from homeassistant.core import Context, CoreState, Event, HomeAssistant, callback
from homeassistant.helpers import condition, discovery, trigger as trigger_helper
from homeassistant.helpers.script import Script
+from homeassistant.helpers.script_variables import ScriptVariables
from homeassistant.helpers.trace import trace_get
from homeassistant.helpers.typing import ConfigType, TemplateVarsType
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -22,7 +24,7 @@ class TriggerUpdateCoordinator(DataUpdateCoordinator):
REMOVE_TRIGGER = object()
- def __init__(self, hass: HomeAssistant, config: dict[str, Any]) -> None:
+ def __init__(self, hass: HomeAssistant, config: ConfigType) -> None:
"""Instantiate trigger data."""
super().__init__(
hass, _LOGGER, config_entry=None, name="Trigger Update Coordinator"
@@ -32,6 +34,18 @@ class TriggerUpdateCoordinator(DataUpdateCoordinator):
self._unsub_start: Callable[[], None] | None = None
self._unsub_trigger: Callable[[], None] | None = None
self._script: Script | None = None
+ self._run_variables: ScriptVariables | None = None
+ self._blueprint_inputs: dict | None = None
+ if config is not None:
+ self._run_variables = config.get(CONF_VARIABLES)
+ self._blueprint_inputs = getattr(config, "raw_blueprint_inputs", None)
+
+ @property
+ def referenced_blueprint(self) -> str | None:
+ """Return referenced blueprint or None."""
+ if self._blueprint_inputs is None:
+ return None
+ return cast(str, self._blueprint_inputs[CONF_USE_BLUEPRINT][CONF_PATH])
@property
def unique_id(self) -> str | None:
@@ -104,6 +118,10 @@ class TriggerUpdateCoordinator(DataUpdateCoordinator):
async def _handle_triggered_with_script(
self, run_variables: TemplateVarsType, context: Context | None = None
) -> None:
+ # Render run variables after the trigger, before checking conditions.
+ if self._run_variables:
+ run_variables = self._run_variables.async_render(self.hass, run_variables)
+
if not self._check_condition(run_variables):
return
# Create a context referring to the trigger context.
@@ -119,6 +137,9 @@ class TriggerUpdateCoordinator(DataUpdateCoordinator):
async def _handle_triggered(
self, run_variables: TemplateVarsType, context: Context | None = None
) -> None:
+ if self._run_variables:
+ run_variables = self._run_variables.async_render(self.hass, run_variables)
+
if not self._check_condition(run_variables):
return
self._execute_update(run_variables, context)
diff --git a/homeassistant/components/template/cover.py b/homeassistant/components/template/cover.py
index 306b4405c6a..7a8e347ee8f 100644
--- a/homeassistant/components/template/cover.py
+++ b/homeassistant/components/template/cover.py
@@ -3,7 +3,7 @@
from __future__ import annotations
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
import voluptuous as vol
@@ -30,7 +30,6 @@ from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity import async_generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import DOMAIN
@@ -103,7 +102,7 @@ PLATFORM_SCHEMA = COVER_PLATFORM_SCHEMA.extend(
)
-async def _async_create_entities(hass, config):
+async def _async_create_entities(hass: HomeAssistant, config):
"""Create the Template cover."""
covers = []
@@ -141,11 +140,11 @@ class CoverTemplate(TemplateEntity, CoverEntity):
def __init__(
self,
- hass,
+ hass: HomeAssistant,
object_id,
- config,
+ config: dict[str, Any],
unique_id,
- ):
+ ) -> None:
"""Initialize the Template cover."""
super().__init__(
hass, config=config, fallback_name=object_id, unique_id=unique_id
@@ -153,45 +152,40 @@ class CoverTemplate(TemplateEntity, CoverEntity):
self.entity_id = async_generate_entity_id(
ENTITY_ID_FORMAT, object_id, hass=hass
)
- friendly_name = self._attr_name
+ name = self._attr_name
+ if TYPE_CHECKING:
+ assert name is not None
self._template = config.get(CONF_VALUE_TEMPLATE)
self._position_template = config.get(CONF_POSITION_TEMPLATE)
self._tilt_template = config.get(CONF_TILT_TEMPLATE)
self._attr_device_class = config.get(CONF_DEVICE_CLASS)
- self._open_script = None
- if (open_action := config.get(OPEN_ACTION)) is not None:
- self._open_script = Script(hass, open_action, friendly_name, DOMAIN)
- self._close_script = None
- if (close_action := config.get(CLOSE_ACTION)) is not None:
- self._close_script = Script(hass, close_action, friendly_name, DOMAIN)
- self._stop_script = None
- if (stop_action := config.get(STOP_ACTION)) is not None:
- self._stop_script = Script(hass, stop_action, friendly_name, DOMAIN)
- self._position_script = None
- if (position_action := config.get(POSITION_ACTION)) is not None:
- self._position_script = Script(hass, position_action, friendly_name, DOMAIN)
- self._tilt_script = None
- if (tilt_action := config.get(TILT_ACTION)) is not None:
- self._tilt_script = Script(hass, tilt_action, friendly_name, DOMAIN)
+
+ # The config requires (open and close scripts) or a set position script,
+ # therefore the base supported features will always include them.
+ self._attr_supported_features = (
+ CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE
+ )
+ for action_id, supported_feature in (
+ (OPEN_ACTION, 0),
+ (CLOSE_ACTION, 0),
+ (STOP_ACTION, CoverEntityFeature.STOP),
+ (POSITION_ACTION, CoverEntityFeature.SET_POSITION),
+ (TILT_ACTION, TILT_FEATURES),
+ ):
+ if action_config := config.get(action_id):
+ self.add_script(action_id, action_config, name, DOMAIN)
+ self._attr_supported_features |= supported_feature
+
optimistic = config.get(CONF_OPTIMISTIC)
self._optimistic = optimistic or (
optimistic is None and not self._template and not self._position_template
)
tilt_optimistic = config.get(CONF_TILT_OPTIMISTIC)
self._tilt_optimistic = tilt_optimistic or not self._tilt_template
- self._position = None
+ self._position: int | None = None
self._is_opening = False
self._is_closing = False
- self._tilt_value = None
-
- supported_features = CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE
- if self._stop_script is not None:
- supported_features |= CoverEntityFeature.STOP
- if self._position_script is not None:
- supported_features |= CoverEntityFeature.SET_POSITION
- if self._tilt_script is not None:
- supported_features |= TILT_FEATURES
- self._attr_supported_features = supported_features
+ self._tilt_value: int | None = None
@callback
def _async_setup_templates(self) -> None:
@@ -317,7 +311,7 @@ class CoverTemplate(TemplateEntity, CoverEntity):
None is unknown, 0 is closed, 100 is fully open.
"""
- if self._position_template or self._position_script:
+ if self._position_template or self._action_scripts.get(POSITION_ACTION):
return self._position
return None
@@ -331,11 +325,11 @@ class CoverTemplate(TemplateEntity, CoverEntity):
async def async_open_cover(self, **kwargs: Any) -> None:
"""Move the cover up."""
- if self._open_script:
- await self.async_run_script(self._open_script, context=self._context)
- elif self._position_script:
+ if open_script := self._action_scripts.get(OPEN_ACTION):
+ await self.async_run_script(open_script, context=self._context)
+ elif position_script := self._action_scripts.get(POSITION_ACTION):
await self.async_run_script(
- self._position_script,
+ position_script,
run_variables={"position": 100},
context=self._context,
)
@@ -345,11 +339,11 @@ class CoverTemplate(TemplateEntity, CoverEntity):
async def async_close_cover(self, **kwargs: Any) -> None:
"""Move the cover down."""
- if self._close_script:
- await self.async_run_script(self._close_script, context=self._context)
- elif self._position_script:
+ if close_script := self._action_scripts.get(CLOSE_ACTION):
+ await self.async_run_script(close_script, context=self._context)
+ elif position_script := self._action_scripts.get(POSITION_ACTION):
await self.async_run_script(
- self._position_script,
+ position_script,
run_variables={"position": 0},
context=self._context,
)
@@ -359,14 +353,14 @@ class CoverTemplate(TemplateEntity, CoverEntity):
async def async_stop_cover(self, **kwargs: Any) -> None:
"""Fire the stop action."""
- if self._stop_script:
- await self.async_run_script(self._stop_script, context=self._context)
+ if stop_script := self._action_scripts.get(STOP_ACTION):
+ await self.async_run_script(stop_script, context=self._context)
async def async_set_cover_position(self, **kwargs: Any) -> None:
"""Set cover position."""
self._position = kwargs[ATTR_POSITION]
await self.async_run_script(
- self._position_script,
+ self._action_scripts[POSITION_ACTION],
run_variables={"position": self._position},
context=self._context,
)
@@ -377,7 +371,7 @@ class CoverTemplate(TemplateEntity, CoverEntity):
"""Tilt the cover open."""
self._tilt_value = 100
await self.async_run_script(
- self._tilt_script,
+ self._action_scripts[TILT_ACTION],
run_variables={"tilt": self._tilt_value},
context=self._context,
)
@@ -388,7 +382,7 @@ class CoverTemplate(TemplateEntity, CoverEntity):
"""Tilt the cover closed."""
self._tilt_value = 0
await self.async_run_script(
- self._tilt_script,
+ self._action_scripts[TILT_ACTION],
run_variables={"tilt": self._tilt_value},
context=self._context,
)
@@ -399,7 +393,7 @@ class CoverTemplate(TemplateEntity, CoverEntity):
"""Move the cover tilt to a specific position."""
self._tilt_value = kwargs[ATTR_TILT_POSITION]
await self.async_run_script(
- self._tilt_script,
+ self._action_scripts[TILT_ACTION],
run_variables={"tilt": self._tilt_value},
context=self._context,
)
diff --git a/homeassistant/components/template/entity.py b/homeassistant/components/template/entity.py
new file mode 100644
index 00000000000..3617d9acdee
--- /dev/null
+++ b/homeassistant/components/template/entity.py
@@ -0,0 +1,64 @@
+"""Template entity base class."""
+
+from collections.abc import Sequence
+from typing import Any
+
+from homeassistant.core import Context, HomeAssistant, callback
+from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.script import Script, _VarsType
+from homeassistant.helpers.template import TemplateStateFromEntityId
+
+
+class AbstractTemplateEntity(Entity):
+ """Actions linked to a template entity."""
+
+ def __init__(self, hass: HomeAssistant) -> None:
+ """Initialize the entity."""
+
+ self.hass = hass
+ self._action_scripts: dict[str, Script] = {}
+
+ @property
+ def referenced_blueprint(self) -> str | None:
+ """Return referenced blueprint or None."""
+ raise NotImplementedError
+
+ @callback
+ def _render_script_variables(self) -> dict:
+ """Render configured variables."""
+ raise NotImplementedError
+
+ def add_script(
+ self,
+ script_id: str,
+ config: Sequence[dict[str, Any]],
+ name: str,
+ domain: str,
+ ):
+ """Add an action script."""
+
+ self._action_scripts[script_id] = Script(
+ self.hass,
+ config,
+ f"{name} {script_id}",
+ domain,
+ )
+
+ async def async_run_script(
+ self,
+ script: Script,
+ *,
+ run_variables: _VarsType | None = None,
+ context: Context | None = None,
+ ) -> None:
+ """Run an action script."""
+ if run_variables is None:
+ run_variables = {}
+ await script.async_run(
+ run_variables={
+ "this": TemplateStateFromEntityId(self.hass, self.entity_id),
+ **self._render_script_variables(),
+ **run_variables,
+ },
+ context=context,
+ )
diff --git a/homeassistant/components/template/fan.py b/homeassistant/components/template/fan.py
index 6ed525fd45f..6e0f9fe5e0c 100644
--- a/homeassistant/components/template/fan.py
+++ b/homeassistant/components/template/fan.py
@@ -3,7 +3,7 @@
from __future__ import annotations
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
import voluptuous as vol
@@ -32,7 +32,6 @@ from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity import async_generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import DOMAIN
@@ -89,7 +88,7 @@ PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend(
)
-async def _async_create_entities(hass, config):
+async def _async_create_entities(hass: HomeAssistant, config):
"""Create the Template Fans."""
fans = []
@@ -127,11 +126,11 @@ class TemplateFan(TemplateEntity, FanEntity):
def __init__(
self,
- hass,
+ hass: HomeAssistant,
object_id,
- config,
+ config: dict[str, Any],
unique_id,
- ):
+ ) -> None:
"""Initialize the fan."""
super().__init__(
hass, config=config, fallback_name=object_id, unique_id=unique_id
@@ -140,7 +139,9 @@ class TemplateFan(TemplateEntity, FanEntity):
self.entity_id = async_generate_entity_id(
ENTITY_ID_FORMAT, object_id, hass=hass
)
- friendly_name = self._attr_name
+ name = self._attr_name
+ if TYPE_CHECKING:
+ assert name is not None
self._template = config.get(CONF_VALUE_TEMPLATE)
self._percentage_template = config.get(CONF_PERCENTAGE_TEMPLATE)
@@ -148,44 +149,28 @@ class TemplateFan(TemplateEntity, FanEntity):
self._oscillating_template = config.get(CONF_OSCILLATING_TEMPLATE)
self._direction_template = config.get(CONF_DIRECTION_TEMPLATE)
- self._on_script = Script(hass, config[CONF_ON_ACTION], friendly_name, DOMAIN)
- self._off_script = Script(hass, config[CONF_OFF_ACTION], friendly_name, DOMAIN)
-
- self._set_percentage_script = None
- if set_percentage_action := config.get(CONF_SET_PERCENTAGE_ACTION):
- self._set_percentage_script = Script(
- hass, set_percentage_action, friendly_name, DOMAIN
- )
-
- self._set_preset_mode_script = None
- if set_preset_mode_action := config.get(CONF_SET_PRESET_MODE_ACTION):
- self._set_preset_mode_script = Script(
- hass, set_preset_mode_action, friendly_name, DOMAIN
- )
-
- self._set_oscillating_script = None
- if set_oscillating_action := config.get(CONF_SET_OSCILLATING_ACTION):
- self._set_oscillating_script = Script(
- hass, set_oscillating_action, friendly_name, DOMAIN
- )
-
- self._set_direction_script = None
- if set_direction_action := config.get(CONF_SET_DIRECTION_ACTION):
- self._set_direction_script = Script(
- hass, set_direction_action, friendly_name, DOMAIN
- )
+ for action_id in (
+ CONF_ON_ACTION,
+ CONF_OFF_ACTION,
+ CONF_SET_PERCENTAGE_ACTION,
+ CONF_SET_PRESET_MODE_ACTION,
+ CONF_SET_OSCILLATING_ACTION,
+ CONF_SET_DIRECTION_ACTION,
+ ):
+ if action_config := config.get(action_id):
+ self.add_script(action_id, action_config, name, DOMAIN)
self._state: bool | None = False
- self._percentage = None
- self._preset_mode = None
- self._oscillating = None
- self._direction = None
+ self._percentage: int | None = None
+ self._preset_mode: str | None = None
+ self._oscillating: bool | None = None
+ self._direction: str | None = None
# Number of valid speeds
self._speed_count = config.get(CONF_SPEED_COUNT)
# List of valid preset modes
- self._preset_modes = config.get(CONF_PRESET_MODES)
+ self._preset_modes: list[str] | None = config.get(CONF_PRESET_MODES)
if self._percentage_template:
self._attr_supported_features |= FanEntityFeature.SET_SPEED
@@ -207,7 +192,7 @@ class TemplateFan(TemplateEntity, FanEntity):
return self._speed_count or 100
@property
- def preset_modes(self) -> list[str]:
+ def preset_modes(self) -> list[str] | None:
"""Get the list of available preset modes."""
return self._preset_modes
@@ -244,7 +229,7 @@ class TemplateFan(TemplateEntity, FanEntity):
) -> None:
"""Turn on the fan."""
await self.async_run_script(
- self._on_script,
+ self._action_scripts[CONF_ON_ACTION],
run_variables={
ATTR_PERCENTAGE: percentage,
ATTR_PRESET_MODE: preset_mode,
@@ -263,7 +248,9 @@ class TemplateFan(TemplateEntity, FanEntity):
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn off the fan."""
- await self.async_run_script(self._off_script, context=self._context)
+ await self.async_run_script(
+ self._action_scripts[CONF_OFF_ACTION], context=self._context
+ )
if self._template is None:
self._state = False
@@ -273,9 +260,9 @@ class TemplateFan(TemplateEntity, FanEntity):
"""Set the percentage speed of the fan."""
self._percentage = percentage
- if self._set_percentage_script:
+ if script := self._action_scripts.get(CONF_SET_PERCENTAGE_ACTION):
await self.async_run_script(
- self._set_percentage_script,
+ script,
run_variables={ATTR_PERCENTAGE: self._percentage},
context=self._context,
)
@@ -288,9 +275,9 @@ class TemplateFan(TemplateEntity, FanEntity):
"""Set the preset_mode of the fan."""
self._preset_mode = preset_mode
- if self._set_preset_mode_script:
+ if script := self._action_scripts.get(CONF_SET_PRESET_MODE_ACTION):
await self.async_run_script(
- self._set_preset_mode_script,
+ script,
run_variables={ATTR_PRESET_MODE: self._preset_mode},
context=self._context,
)
@@ -301,25 +288,25 @@ class TemplateFan(TemplateEntity, FanEntity):
async def async_oscillate(self, oscillating: bool) -> None:
"""Set oscillation of the fan."""
- if self._set_oscillating_script is None:
+ if (script := self._action_scripts.get(CONF_SET_OSCILLATING_ACTION)) is None:
return
self._oscillating = oscillating
await self.async_run_script(
- self._set_oscillating_script,
+ script,
run_variables={ATTR_OSCILLATING: self.oscillating},
context=self._context,
)
async def async_set_direction(self, direction: str) -> None:
"""Set the direction of the fan."""
- if self._set_direction_script is None:
+ if (script := self._action_scripts.get(CONF_SET_DIRECTION_ACTION)) is None:
return
if direction in _VALID_DIRECTIONS:
self._direction = direction
await self.async_run_script(
- self._set_direction_script,
+ script,
run_variables={ATTR_DIRECTION: direction},
context=self._context,
)
diff --git a/homeassistant/components/template/helpers.py b/homeassistant/components/template/helpers.py
index b320f2128cd..d74a4a4ed00 100644
--- a/homeassistant/components/template/helpers.py
+++ b/homeassistant/components/template/helpers.py
@@ -9,7 +9,7 @@ from homeassistant.helpers.entity_platform import async_get_platforms
from homeassistant.helpers.singleton import singleton
from .const import DOMAIN, TEMPLATE_BLUEPRINT_SCHEMA
-from .template_entity import TemplateEntity
+from .entity import AbstractTemplateEntity
DATA_BLUEPRINTS = "template_blueprints"
@@ -23,7 +23,7 @@ def templates_with_blueprint(hass: HomeAssistant, blueprint_path: str) -> list[s
entity_id
for platform in async_get_platforms(hass, DOMAIN)
for entity_id, template_entity in platform.entities.items()
- if isinstance(template_entity, TemplateEntity)
+ if isinstance(template_entity, AbstractTemplateEntity)
and template_entity.referenced_blueprint == blueprint_path
]
@@ -33,7 +33,8 @@ def blueprint_in_template(hass: HomeAssistant, entity_id: str) -> str | None:
"""Return the blueprint the template entity is based on or None."""
for platform in async_get_platforms(hass, DOMAIN):
if isinstance(
- (template_entity := platform.entities.get(entity_id)), TemplateEntity
+ (template_entity := platform.entities.get(entity_id)),
+ AbstractTemplateEntity,
):
return template_entity.referenced_blueprint
return None
diff --git a/homeassistant/components/template/image.py b/homeassistant/components/template/image.py
index ba85418c339..5afbca55cbb 100644
--- a/homeassistant/components/template/image.py
+++ b/homeassistant/components/template/image.py
@@ -20,7 +20,10 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv, selector
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
@@ -96,7 +99,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
diff --git a/homeassistant/components/template/light.py b/homeassistant/components/template/light.py
index 9391e368e2b..1cc47c74aa0 100644
--- a/homeassistant/components/template/light.py
+++ b/homeassistant/components/template/light.py
@@ -3,7 +3,7 @@
from __future__ import annotations
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
import voluptuous as vol
@@ -26,9 +26,13 @@ from homeassistant.components.light import (
filter_supported_color_modes,
)
from homeassistant.const import (
+ CONF_EFFECT,
CONF_ENTITY_ID,
CONF_FRIENDLY_NAME,
CONF_LIGHTS,
+ CONF_NAME,
+ CONF_RGB,
+ CONF_STATE,
CONF_UNIQUE_ID,
CONF_VALUE_TEMPLATE,
STATE_OFF,
@@ -36,16 +40,18 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import TemplateError
-from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers import config_validation as cv, template
from homeassistant.helpers.entity import async_generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import color as color_util
-from .const import DOMAIN
+from .const import CONF_OBJECT_ID, CONF_PICTURE, DOMAIN
from .template_entity import (
+ LEGACY_FIELDS as TEMPLATE_ENTITY_LEGACY_FIELDS,
+ TEMPLATE_ENTITY_AVAILABILITY_SCHEMA,
TEMPLATE_ENTITY_COMMON_SCHEMA_LEGACY,
+ TEMPLATE_ENTITY_ICON_SCHEMA,
TemplateEntity,
rewrite_common_legacy_to_modern_conf,
)
@@ -57,33 +63,96 @@ _VALID_STATES = [STATE_ON, STATE_OFF, "true", "false"]
CONF_COLOR_ACTION = "set_color"
CONF_COLOR_TEMPLATE = "color_template"
+CONF_HS = "hs"
CONF_HS_ACTION = "set_hs"
CONF_HS_TEMPLATE = "hs_template"
CONF_RGB_ACTION = "set_rgb"
CONF_RGB_TEMPLATE = "rgb_template"
+CONF_RGBW = "rgbw"
CONF_RGBW_ACTION = "set_rgbw"
CONF_RGBW_TEMPLATE = "rgbw_template"
+CONF_RGBWW = "rgbww"
CONF_RGBWW_ACTION = "set_rgbww"
CONF_RGBWW_TEMPLATE = "rgbww_template"
CONF_EFFECT_ACTION = "set_effect"
+CONF_EFFECT_LIST = "effect_list"
CONF_EFFECT_LIST_TEMPLATE = "effect_list_template"
CONF_EFFECT_TEMPLATE = "effect_template"
+CONF_LEVEL = "level"
CONF_LEVEL_ACTION = "set_level"
CONF_LEVEL_TEMPLATE = "level_template"
+CONF_MAX_MIREDS = "max_mireds"
CONF_MAX_MIREDS_TEMPLATE = "max_mireds_template"
+CONF_MIN_MIREDS = "min_mireds"
CONF_MIN_MIREDS_TEMPLATE = "min_mireds_template"
CONF_OFF_ACTION = "turn_off"
CONF_ON_ACTION = "turn_on"
-CONF_SUPPORTS_TRANSITION = "supports_transition_template"
+CONF_SUPPORTS_TRANSITION = "supports_transition"
+CONF_SUPPORTS_TRANSITION_TEMPLATE = "supports_transition_template"
CONF_TEMPERATURE_ACTION = "set_temperature"
+CONF_TEMPERATURE = "temperature"
CONF_TEMPERATURE_TEMPLATE = "temperature_template"
CONF_WHITE_VALUE_ACTION = "set_white_value"
+CONF_WHITE_VALUE = "white_value"
CONF_WHITE_VALUE_TEMPLATE = "white_value_template"
DEFAULT_MIN_MIREDS = 153
DEFAULT_MAX_MIREDS = 500
-LIGHT_SCHEMA = vol.All(
+LEGACY_FIELDS = TEMPLATE_ENTITY_LEGACY_FIELDS | {
+ CONF_COLOR_ACTION: CONF_HS_ACTION,
+ CONF_COLOR_TEMPLATE: CONF_HS,
+ CONF_EFFECT_LIST_TEMPLATE: CONF_EFFECT_LIST,
+ CONF_EFFECT_TEMPLATE: CONF_EFFECT,
+ CONF_HS_TEMPLATE: CONF_HS,
+ CONF_LEVEL_TEMPLATE: CONF_LEVEL,
+ CONF_MAX_MIREDS_TEMPLATE: CONF_MAX_MIREDS,
+ CONF_MIN_MIREDS_TEMPLATE: CONF_MIN_MIREDS,
+ CONF_RGB_TEMPLATE: CONF_RGB,
+ CONF_RGBW_TEMPLATE: CONF_RGBW,
+ CONF_RGBWW_TEMPLATE: CONF_RGBWW,
+ CONF_SUPPORTS_TRANSITION_TEMPLATE: CONF_SUPPORTS_TRANSITION,
+ CONF_TEMPERATURE_TEMPLATE: CONF_TEMPERATURE,
+ CONF_VALUE_TEMPLATE: CONF_STATE,
+ CONF_WHITE_VALUE_TEMPLATE: CONF_WHITE_VALUE,
+}
+
+DEFAULT_NAME = "Template Light"
+
+LIGHT_SCHEMA = (
+ vol.Schema(
+ {
+ vol.Inclusive(CONF_EFFECT_ACTION, "effect"): cv.SCRIPT_SCHEMA,
+ vol.Inclusive(CONF_EFFECT_LIST, "effect"): cv.template,
+ vol.Inclusive(CONF_EFFECT, "effect"): cv.template,
+ vol.Optional(CONF_HS_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_HS): cv.template,
+ vol.Optional(CONF_LEVEL_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_LEVEL): cv.template,
+ vol.Optional(CONF_MAX_MIREDS): cv.template,
+ vol.Optional(CONF_MIN_MIREDS): cv.template,
+ vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.template,
+ vol.Optional(CONF_PICTURE): cv.template,
+ vol.Optional(CONF_RGB_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_RGB): cv.template,
+ vol.Optional(CONF_RGBW_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_RGBW): cv.template,
+ vol.Optional(CONF_RGBWW_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_RGBWW): cv.template,
+ vol.Optional(CONF_STATE): cv.template,
+ vol.Optional(CONF_SUPPORTS_TRANSITION): cv.template,
+ vol.Optional(CONF_TEMPERATURE_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_TEMPERATURE): cv.template,
+ vol.Optional(CONF_UNIQUE_ID): cv.string,
+ vol.Required(CONF_OFF_ACTION): cv.SCRIPT_SCHEMA,
+ vol.Required(CONF_ON_ACTION): cv.SCRIPT_SCHEMA,
+ }
+ )
+ .extend(TEMPLATE_ENTITY_AVAILABILITY_SCHEMA.schema)
+ .extend(TEMPLATE_ENTITY_ICON_SCHEMA.schema)
+)
+
+LEGACY_LIGHT_SCHEMA = vol.All(
cv.deprecated(CONF_ENTITY_ID),
vol.Schema(
{
@@ -108,7 +177,7 @@ LIGHT_SCHEMA = vol.All(
vol.Optional(CONF_MIN_MIREDS_TEMPLATE): cv.template,
vol.Required(CONF_OFF_ACTION): cv.SCRIPT_SCHEMA,
vol.Required(CONF_ON_ACTION): cv.SCRIPT_SCHEMA,
- vol.Optional(CONF_SUPPORTS_TRANSITION): cv.template,
+ vol.Optional(CONF_SUPPORTS_TRANSITION_TEMPLATE): cv.template,
vol.Optional(CONF_TEMPERATURE_ACTION): cv.SCRIPT_SCHEMA,
vol.Optional(CONF_TEMPERATURE_TEMPLATE): cv.template,
vol.Optional(CONF_UNIQUE_ID): cv.string,
@@ -122,29 +191,50 @@ PLATFORM_SCHEMA = vol.All(
cv.removed(CONF_WHITE_VALUE_ACTION),
cv.removed(CONF_WHITE_VALUE_TEMPLATE),
LIGHT_PLATFORM_SCHEMA.extend(
- {vol.Required(CONF_LIGHTS): cv.schema_with_slug_keys(LIGHT_SCHEMA)}
+ {vol.Required(CONF_LIGHTS): cv.schema_with_slug_keys(LEGACY_LIGHT_SCHEMA)}
),
)
-async def _async_create_entities(hass, config):
+def rewrite_legacy_to_modern_conf(
+ hass: HomeAssistant, config: dict[str, dict]
+) -> list[dict]:
+ """Rewrite legacy switch configuration definitions to modern ones."""
+ lights = []
+ for object_id, entity_conf in config.items():
+ entity_conf = {**entity_conf, CONF_OBJECT_ID: object_id}
+
+ entity_conf = rewrite_common_legacy_to_modern_conf(
+ hass, entity_conf, LEGACY_FIELDS
+ )
+
+ if CONF_NAME not in entity_conf:
+ entity_conf[CONF_NAME] = template.Template(object_id, hass)
+
+ lights.append(entity_conf)
+
+ return lights
+
+
+@callback
+def _async_create_template_tracking_entities(
+ async_add_entities: AddEntitiesCallback,
+ hass: HomeAssistant,
+ definitions: list[dict],
+ unique_id_prefix: str | None,
+) -> None:
"""Create the Template Lights."""
lights = []
- for object_id, entity_config in config[CONF_LIGHTS].items():
- entity_config = rewrite_common_legacy_to_modern_conf(hass, entity_config)
- unique_id = entity_config.get(CONF_UNIQUE_ID)
+ for entity_conf in definitions:
+ unique_id = entity_conf.get(CONF_UNIQUE_ID)
- lights.append(
- LightTemplate(
- hass,
- object_id,
- entity_config,
- unique_id,
- )
- )
+ if unique_id and unique_id_prefix:
+ unique_id = f"{unique_id_prefix}-{unique_id}"
- return lights
+ lights.append(LightTemplate(hass, entity_conf, unique_id))
+
+ async_add_entities(lights)
async def async_setup_platform(
@@ -154,7 +244,21 @@ async def async_setup_platform(
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up the template lights."""
- async_add_entities(await _async_create_entities(hass, config))
+ if discovery_info is None:
+ _async_create_template_tracking_entities(
+ async_add_entities,
+ hass,
+ rewrite_legacy_to_modern_conf(hass, config[CONF_LIGHTS]),
+ None,
+ )
+ return
+
+ _async_create_template_tracking_entities(
+ async_add_entities,
+ hass,
+ discovery_info["entities"],
+ discovery_info["unique_id"],
+ )
class LightTemplate(TemplateEntity, LightEntity):
@@ -164,64 +268,40 @@ class LightTemplate(TemplateEntity, LightEntity):
def __init__(
self,
- hass,
- object_id,
- config,
- unique_id,
- ):
+ hass: HomeAssistant,
+ config: dict[str, Any],
+ unique_id: str | None,
+ ) -> None:
"""Initialize the light."""
- super().__init__(
- hass, config=config, fallback_name=object_id, unique_id=unique_id
- )
- self.entity_id = async_generate_entity_id(
- ENTITY_ID_FORMAT, object_id, hass=hass
- )
- friendly_name = self._attr_name
- self._template = config.get(CONF_VALUE_TEMPLATE)
- self._on_script = Script(hass, config[CONF_ON_ACTION], friendly_name, DOMAIN)
- self._off_script = Script(hass, config[CONF_OFF_ACTION], friendly_name, DOMAIN)
- self._level_script = None
- if (level_action := config.get(CONF_LEVEL_ACTION)) is not None:
- self._level_script = Script(hass, level_action, friendly_name, DOMAIN)
- self._level_template = config.get(CONF_LEVEL_TEMPLATE)
- self._temperature_script = None
- if (temperature_action := config.get(CONF_TEMPERATURE_ACTION)) is not None:
- self._temperature_script = Script(
- hass, temperature_action, friendly_name, DOMAIN
+ super().__init__(hass, config=config, fallback_name=None, unique_id=unique_id)
+ if (object_id := config.get(CONF_OBJECT_ID)) is not None:
+ self.entity_id = async_generate_entity_id(
+ ENTITY_ID_FORMAT, object_id, hass=hass
)
- self._temperature_template = config.get(CONF_TEMPERATURE_TEMPLATE)
- self._color_script = None
- if (color_action := config.get(CONF_COLOR_ACTION)) is not None:
- self._color_script = Script(hass, color_action, friendly_name, DOMAIN)
- self._color_template = config.get(CONF_COLOR_TEMPLATE)
- self._hs_script = None
- if (hs_action := config.get(CONF_HS_ACTION)) is not None:
- self._hs_script = Script(hass, hs_action, friendly_name, DOMAIN)
- self._hs_template = config.get(CONF_HS_TEMPLATE)
- self._rgb_script = None
- if (rgb_action := config.get(CONF_RGB_ACTION)) is not None:
- self._rgb_script = Script(hass, rgb_action, friendly_name, DOMAIN)
- self._rgb_template = config.get(CONF_RGB_TEMPLATE)
- self._rgbw_script = None
- if (rgbw_action := config.get(CONF_RGBW_ACTION)) is not None:
- self._rgbw_script = Script(hass, rgbw_action, friendly_name, DOMAIN)
- self._rgbw_template = config.get(CONF_RGBW_TEMPLATE)
- self._rgbww_script = None
- if (rgbww_action := config.get(CONF_RGBWW_ACTION)) is not None:
- self._rgbww_script = Script(hass, rgbww_action, friendly_name, DOMAIN)
- self._rgbww_template = config.get(CONF_RGBWW_TEMPLATE)
- self._effect_script = None
- if (effect_action := config.get(CONF_EFFECT_ACTION)) is not None:
- self._effect_script = Script(hass, effect_action, friendly_name, DOMAIN)
- self._effect_list_template = config.get(CONF_EFFECT_LIST_TEMPLATE)
- self._effect_template = config.get(CONF_EFFECT_TEMPLATE)
- self._max_mireds_template = config.get(CONF_MAX_MIREDS_TEMPLATE)
- self._min_mireds_template = config.get(CONF_MIN_MIREDS_TEMPLATE)
+ name = self._attr_name
+ if TYPE_CHECKING:
+ assert name is not None
+
+ self._template = config.get(CONF_STATE)
+ self._level_template = config.get(CONF_LEVEL)
+ self._temperature_template = config.get(CONF_TEMPERATURE)
+ self._hs_template = config.get(CONF_HS)
+ self._rgb_template = config.get(CONF_RGB)
+ self._rgbw_template = config.get(CONF_RGBW)
+ self._rgbww_template = config.get(CONF_RGBWW)
+ self._effect_list_template = config.get(CONF_EFFECT_LIST)
+ self._effect_template = config.get(CONF_EFFECT)
+ self._max_mireds_template = config.get(CONF_MAX_MIREDS)
+ self._min_mireds_template = config.get(CONF_MIN_MIREDS)
self._supports_transition_template = config.get(CONF_SUPPORTS_TRANSITION)
+ for action_id in (CONF_ON_ACTION, CONF_OFF_ACTION, CONF_EFFECT_ACTION):
+ if action_config := config.get(action_id):
+ self.add_script(action_id, action_config, name, DOMAIN)
+
self._state = False
self._brightness = None
- self._temperature = None
+ self._temperature: int | None = None
self._hs_color = None
self._rgb_color = None
self._rgbw_color = None
@@ -235,21 +315,17 @@ class LightTemplate(TemplateEntity, LightEntity):
self._supported_color_modes = None
color_modes = {ColorMode.ONOFF}
- if self._level_script is not None:
- color_modes.add(ColorMode.BRIGHTNESS)
- if self._temperature_script is not None:
- color_modes.add(ColorMode.COLOR_TEMP)
- if self._hs_script is not None:
- color_modes.add(ColorMode.HS)
- if self._color_script is not None:
- color_modes.add(ColorMode.HS)
- if self._rgb_script is not None:
- color_modes.add(ColorMode.RGB)
- if self._rgbw_script is not None:
- color_modes.add(ColorMode.RGBW)
- if self._rgbww_script is not None:
- color_modes.add(ColorMode.RGBWW)
-
+ for action_id, color_mode in (
+ (CONF_TEMPERATURE_ACTION, ColorMode.COLOR_TEMP),
+ (CONF_LEVEL_ACTION, ColorMode.BRIGHTNESS),
+ (CONF_HS_ACTION, ColorMode.HS),
+ (CONF_RGB_ACTION, ColorMode.RGB),
+ (CONF_RGBW_ACTION, ColorMode.RGBW),
+ (CONF_RGBWW_ACTION, ColorMode.RGBWW),
+ ):
+ if action_config := config.get(action_id):
+ self.add_script(action_id, action_config, name, DOMAIN)
+ color_modes.add(color_mode)
self._supported_color_modes = filter_supported_color_modes(color_modes)
if len(self._supported_color_modes) > 1:
self._color_mode = ColorMode.UNKNOWN
@@ -257,7 +333,7 @@ class LightTemplate(TemplateEntity, LightEntity):
self._color_mode = next(iter(self._supported_color_modes))
self._attr_supported_features = LightEntityFeature(0)
- if self._effect_script is not None:
+ if self._action_scripts.get(CONF_EFFECT_ACTION):
self._attr_supported_features |= LightEntityFeature.EFFECT
if self._supports_transition is True:
self._attr_supported_features |= LightEntityFeature.TRANSITION
@@ -321,12 +397,12 @@ class LightTemplate(TemplateEntity, LightEntity):
return self._effect_list
@property
- def color_mode(self):
+ def color_mode(self) -> ColorMode | None:
"""Return current color mode."""
return self._color_mode
@property
- def supported_color_modes(self):
+ def supported_color_modes(self) -> set[ColorMode] | None:
"""Flag supported color modes."""
return self._supported_color_modes
@@ -374,14 +450,6 @@ class LightTemplate(TemplateEntity, LightEntity):
self._update_temperature,
none_on_template_error=True,
)
- if self._color_template:
- self.add_template_attribute(
- "_hs_color",
- self._color_template,
- None,
- self._update_hs,
- none_on_template_error=True,
- )
if self._hs_template:
self.add_template_attribute(
"_hs_color",
@@ -465,7 +533,7 @@ class LightTemplate(TemplateEntity, LightEntity):
)
self._color_mode = ColorMode.COLOR_TEMP
self._temperature = color_temp
- if self._hs_template is None and self._color_template is None:
+ if self._hs_template is None:
self._hs_color = None
if self._rgb_template is None:
self._rgb_color = None
@@ -475,11 +543,7 @@ class LightTemplate(TemplateEntity, LightEntity):
self._rgbww_color = None
optimistic_set = True
- if (
- self._hs_template is None
- and self._color_template is None
- and ATTR_HS_COLOR in kwargs
- ):
+ if self._hs_template is None and ATTR_HS_COLOR in kwargs:
_LOGGER.debug(
"Optimistically setting hs color to %s",
kwargs[ATTR_HS_COLOR],
@@ -505,7 +569,7 @@ class LightTemplate(TemplateEntity, LightEntity):
self._rgb_color = kwargs[ATTR_RGB_COLOR]
if self._temperature_template is None:
self._temperature = None
- if self._hs_template is None and self._color_template is None:
+ if self._hs_template is None:
self._hs_color = None
if self._rgbw_template is None:
self._rgbw_color = None
@@ -522,7 +586,7 @@ class LightTemplate(TemplateEntity, LightEntity):
self._rgbw_color = kwargs[ATTR_RGBW_COLOR]
if self._temperature_template is None:
self._temperature = None
- if self._hs_template is None and self._color_template is None:
+ if self._hs_template is None:
self._hs_color = None
if self._rgb_template is None:
self._rgb_color = None
@@ -539,7 +603,7 @@ class LightTemplate(TemplateEntity, LightEntity):
self._rgbww_color = kwargs[ATTR_RGBWW_COLOR]
if self._temperature_template is None:
self._temperature = None
- if self._hs_template is None and self._color_template is None:
+ if self._hs_template is None:
self._hs_color = None
if self._rgb_template is None:
self._rgb_color = None
@@ -555,17 +619,22 @@ class LightTemplate(TemplateEntity, LightEntity):
if ATTR_TRANSITION in kwargs and self._supports_transition is True:
common_params["transition"] = kwargs[ATTR_TRANSITION]
- if ATTR_COLOR_TEMP_KELVIN in kwargs and self._temperature_script:
+ if ATTR_COLOR_TEMP_KELVIN in kwargs and (
+ temperature_script := self._action_scripts.get(CONF_TEMPERATURE_ACTION)
+ ):
common_params["color_temp"] = color_util.color_temperature_kelvin_to_mired(
kwargs[ATTR_COLOR_TEMP_KELVIN]
)
await self.async_run_script(
- self._temperature_script,
+ temperature_script,
run_variables=common_params,
context=self._context,
)
- elif ATTR_EFFECT in kwargs and self._effect_script:
+ elif ATTR_EFFECT in kwargs and (
+ effect_script := self._action_scripts.get(CONF_EFFECT_ACTION)
+ ):
+ assert self._effect_list is not None
effect = kwargs[ATTR_EFFECT]
if effect not in self._effect_list:
_LOGGER.error(
@@ -579,27 +648,22 @@ class LightTemplate(TemplateEntity, LightEntity):
common_params["effect"] = effect
await self.async_run_script(
- self._effect_script, run_variables=common_params, context=self._context
+ effect_script, run_variables=common_params, context=self._context
)
- elif ATTR_HS_COLOR in kwargs and self._color_script:
+ elif ATTR_HS_COLOR in kwargs and (
+ hs_script := self._action_scripts.get(CONF_HS_ACTION)
+ ):
hs_value = kwargs[ATTR_HS_COLOR]
common_params["hs"] = hs_value
common_params["h"] = int(hs_value[0])
common_params["s"] = int(hs_value[1])
await self.async_run_script(
- self._color_script, run_variables=common_params, context=self._context
+ hs_script, run_variables=common_params, context=self._context
)
- elif ATTR_HS_COLOR in kwargs and self._hs_script:
- hs_value = kwargs[ATTR_HS_COLOR]
- common_params["hs"] = hs_value
- common_params["h"] = int(hs_value[0])
- common_params["s"] = int(hs_value[1])
-
- await self.async_run_script(
- self._hs_script, run_variables=common_params, context=self._context
- )
- elif ATTR_RGBWW_COLOR in kwargs and self._rgbww_script:
+ elif ATTR_RGBWW_COLOR in kwargs and (
+ rgbww_script := self._action_scripts.get(CONF_RGBWW_ACTION)
+ ):
rgbww_value = kwargs[ATTR_RGBWW_COLOR]
common_params["rgbww"] = rgbww_value
common_params["rgb"] = (
@@ -614,9 +678,11 @@ class LightTemplate(TemplateEntity, LightEntity):
common_params["ww"] = int(rgbww_value[4])
await self.async_run_script(
- self._rgbww_script, run_variables=common_params, context=self._context
+ rgbww_script, run_variables=common_params, context=self._context
)
- elif ATTR_RGBW_COLOR in kwargs and self._rgbw_script:
+ elif ATTR_RGBW_COLOR in kwargs and (
+ rgbw_script := self._action_scripts.get(CONF_RGBW_ACTION)
+ ):
rgbw_value = kwargs[ATTR_RGBW_COLOR]
common_params["rgbw"] = rgbw_value
common_params["rgb"] = (
@@ -630,9 +696,11 @@ class LightTemplate(TemplateEntity, LightEntity):
common_params["w"] = int(rgbw_value[3])
await self.async_run_script(
- self._rgbw_script, run_variables=common_params, context=self._context
+ rgbw_script, run_variables=common_params, context=self._context
)
- elif ATTR_RGB_COLOR in kwargs and self._rgb_script:
+ elif ATTR_RGB_COLOR in kwargs and (
+ rgb_script := self._action_scripts.get(CONF_RGB_ACTION)
+ ):
rgb_value = kwargs[ATTR_RGB_COLOR]
common_params["rgb"] = rgb_value
common_params["r"] = int(rgb_value[0])
@@ -640,15 +708,19 @@ class LightTemplate(TemplateEntity, LightEntity):
common_params["b"] = int(rgb_value[2])
await self.async_run_script(
- self._rgb_script, run_variables=common_params, context=self._context
+ rgb_script, run_variables=common_params, context=self._context
)
- elif ATTR_BRIGHTNESS in kwargs and self._level_script:
+ elif ATTR_BRIGHTNESS in kwargs and (
+ level_script := self._action_scripts.get(CONF_LEVEL_ACTION)
+ ):
await self.async_run_script(
- self._level_script, run_variables=common_params, context=self._context
+ level_script, run_variables=common_params, context=self._context
)
else:
await self.async_run_script(
- self._on_script, run_variables=common_params, context=self._context
+ self._action_scripts[CONF_ON_ACTION],
+ run_variables=common_params,
+ context=self._context,
)
if optimistic_set:
@@ -656,14 +728,15 @@ class LightTemplate(TemplateEntity, LightEntity):
async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the light off."""
+ off_script = self._action_scripts[CONF_OFF_ACTION]
if ATTR_TRANSITION in kwargs and self._supports_transition is True:
await self.async_run_script(
- self._off_script,
+ off_script,
run_variables={"transition": kwargs[ATTR_TRANSITION]},
context=self._context,
)
else:
- await self.async_run_script(self._off_script, context=self._context)
+ await self.async_run_script(off_script, context=self._context)
if self._template is None:
self._state = False
self.async_write_ha_state()
@@ -1013,7 +1086,7 @@ class LightTemplate(TemplateEntity, LightEntity):
if render in (None, "None", ""):
self._supports_transition = False
return
- self._attr_supported_features &= LightEntityFeature.EFFECT
+ self._attr_supported_features &= ~LightEntityFeature.TRANSITION
self._supports_transition = bool(render)
if self._supports_transition:
self._attr_supported_features |= LightEntityFeature.TRANSITION
diff --git a/homeassistant/components/template/lock.py b/homeassistant/components/template/lock.py
index 0804f92e46d..b19cadff26c 100644
--- a/homeassistant/components/template/lock.py
+++ b/homeassistant/components/template/lock.py
@@ -23,7 +23,6 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError, TemplateError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import DOMAIN
@@ -90,13 +89,18 @@ class TemplateLock(TemplateEntity, LockEntity):
)
self._state: LockState | None = None
name = self._attr_name
- assert name
+ if TYPE_CHECKING:
+ assert name is not None
+
self._state_template = config.get(CONF_VALUE_TEMPLATE)
- self._command_lock = Script(hass, config[CONF_LOCK], name, DOMAIN)
- self._command_unlock = Script(hass, config[CONF_UNLOCK], name, DOMAIN)
- if CONF_OPEN in config:
- self._command_open = Script(hass, config[CONF_OPEN], name, DOMAIN)
- self._attr_supported_features |= LockEntityFeature.OPEN
+ for action_id, supported_feature in (
+ (CONF_LOCK, 0),
+ (CONF_UNLOCK, 0),
+ (CONF_OPEN, LockEntityFeature.OPEN),
+ ):
+ if action_config := config.get(action_id):
+ self.add_script(action_id, action_config, name, DOMAIN)
+ self._attr_supported_features |= supported_feature
self._code_format_template = config.get(CONF_CODE_FORMAT_TEMPLATE)
self._code_format: str | None = None
self._code_format_template_error: TemplateError | None = None
@@ -210,7 +214,9 @@ class TemplateLock(TemplateEntity, LockEntity):
tpl_vars = {ATTR_CODE: kwargs.get(ATTR_CODE) if kwargs else None}
await self.async_run_script(
- self._command_lock, run_variables=tpl_vars, context=self._context
+ self._action_scripts[CONF_LOCK],
+ run_variables=tpl_vars,
+ context=self._context,
)
async def async_unlock(self, **kwargs: Any) -> None:
@@ -226,7 +232,9 @@ class TemplateLock(TemplateEntity, LockEntity):
tpl_vars = {ATTR_CODE: kwargs.get(ATTR_CODE) if kwargs else None}
await self.async_run_script(
- self._command_unlock, run_variables=tpl_vars, context=self._context
+ self._action_scripts[CONF_UNLOCK],
+ run_variables=tpl_vars,
+ context=self._context,
)
async def async_open(self, **kwargs: Any) -> None:
@@ -242,7 +250,9 @@ class TemplateLock(TemplateEntity, LockEntity):
tpl_vars = {ATTR_CODE: kwargs.get(ATTR_CODE) if kwargs else None}
await self.async_run_script(
- self._command_open, run_variables=tpl_vars, context=self._context
+ self._action_scripts[CONF_OPEN],
+ run_variables=tpl_vars,
+ context=self._context,
)
def _raise_template_error_if_available(self):
diff --git a/homeassistant/components/template/manifest.json b/homeassistant/components/template/manifest.json
index f1225f74f06..32bfd8ce02e 100644
--- a/homeassistant/components/template/manifest.json
+++ b/homeassistant/components/template/manifest.json
@@ -2,7 +2,7 @@
"domain": "template",
"name": "Template",
"after_dependencies": ["group"],
- "codeowners": ["@PhracturedBlue", "@home-assistant/core"],
+ "codeowners": ["@Petro31", "@PhracturedBlue", "@home-assistant/core"],
"config_flow": true,
"dependencies": ["blueprint"],
"documentation": "https://www.home-assistant.io/integrations/template",
diff --git a/homeassistant/components/template/number.py b/homeassistant/components/template/number.py
index 90dd555ca42..3ecf1db565a 100644
--- a/homeassistant/components/template/number.py
+++ b/homeassistant/components/template/number.py
@@ -27,8 +27,10 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, selector
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import TriggerUpdateCoordinator
@@ -121,7 +123,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
@@ -154,9 +156,7 @@ class TemplateNumber(TemplateEntity, NumberEntity):
super().__init__(hass, config=config, unique_id=unique_id)
assert self._attr_name is not None
self._value_template = config[CONF_STATE]
- self._command_set_value = Script(
- hass, config[CONF_SET_VALUE], self._attr_name, DOMAIN
- )
+ self.add_script(CONF_SET_VALUE, config[CONF_SET_VALUE], self._attr_name, DOMAIN)
self._step_template = config[CONF_STEP]
self._min_value_template = config[CONF_MIN]
@@ -207,9 +207,9 @@ class TemplateNumber(TemplateEntity, NumberEntity):
if self._optimistic:
self._attr_native_value = value
self.async_write_ha_state()
- if self._command_set_value:
+ if set_value := self._action_scripts.get(CONF_SET_VALUE):
await self.async_run_script(
- self._command_set_value,
+ set_value,
run_variables={ATTR_VALUE: value},
context=self._context,
)
@@ -235,12 +235,8 @@ class TriggerNumberEntity(TriggerEntity, NumberEntity):
"""Initialize the entity."""
super().__init__(hass, coordinator, config)
- self._command_set_value = Script(
- hass,
- config[CONF_SET_VALUE],
- self._rendered.get(CONF_NAME, DEFAULT_NAME),
- DOMAIN,
- )
+ name = self._rendered.get(CONF_NAME, DEFAULT_NAME)
+ self.add_script(CONF_SET_VALUE, config[CONF_SET_VALUE], name, DOMAIN)
self._attr_native_unit_of_measurement = config.get(CONF_UNIT_OF_MEASUREMENT)
@@ -275,6 +271,9 @@ class TriggerNumberEntity(TriggerEntity, NumberEntity):
if self._config[CONF_OPTIMISTIC]:
self._attr_native_value = value
self.async_write_ha_state()
- await self._command_set_value.async_run(
- {ATTR_VALUE: value}, context=self._context
- )
+ if set_value := self._action_scripts.get(CONF_SET_VALUE):
+ await self.async_run_script(
+ set_value,
+ run_variables={ATTR_VALUE: value},
+ context=self._context,
+ )
diff --git a/homeassistant/components/template/select.py b/homeassistant/components/template/select.py
index bd37ca1015c..eb60a3dbfe4 100644
--- a/homeassistant/components/template/select.py
+++ b/homeassistant/components/template/select.py
@@ -24,8 +24,10 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, selector
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import TriggerUpdateCoordinator
@@ -115,7 +117,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
@@ -139,10 +141,8 @@ class TemplateSelect(TemplateEntity, SelectEntity):
super().__init__(hass, config=config, unique_id=unique_id)
assert self._attr_name is not None
self._value_template = config[CONF_STATE]
- if (selection_option := config.get(CONF_SELECT_OPTION)) is not None:
- self._command_select_option = Script(
- hass, selection_option, self._attr_name, DOMAIN
- )
+ if select_option := config.get(CONF_SELECT_OPTION):
+ self.add_script(CONF_SELECT_OPTION, select_option, self._attr_name, DOMAIN)
self._options_template = config[ATTR_OPTIONS]
self._attr_assumed_state = self._optimistic = config.get(CONF_OPTIMISTIC, False)
self._attr_options = []
@@ -174,9 +174,9 @@ class TemplateSelect(TemplateEntity, SelectEntity):
if self._optimistic:
self._attr_current_option = option
self.async_write_ha_state()
- if self._command_select_option:
+ if select_option := self._action_scripts.get(CONF_SELECT_OPTION):
await self.async_run_script(
- self._command_select_option,
+ select_option,
run_variables={ATTR_OPTION: option},
context=self._context,
)
@@ -197,12 +197,13 @@ class TriggerSelectEntity(TriggerEntity, SelectEntity):
) -> None:
"""Initialize the entity."""
super().__init__(hass, coordinator, config)
- self._command_select_option = Script(
- hass,
- config[CONF_SELECT_OPTION],
- self._rendered.get(CONF_NAME, DEFAULT_NAME),
- DOMAIN,
- )
+ if select_option := config.get(CONF_SELECT_OPTION):
+ self.add_script(
+ CONF_SELECT_OPTION,
+ select_option,
+ self._rendered.get(CONF_NAME, DEFAULT_NAME),
+ DOMAIN,
+ )
@property
def current_option(self) -> str | None:
@@ -219,6 +220,9 @@ class TriggerSelectEntity(TriggerEntity, SelectEntity):
if self._config[CONF_OPTIMISTIC]:
self._attr_current_option = option
self.async_write_ha_state()
- await self._command_select_option.async_run(
- {ATTR_OPTION: option}, context=self._context
- )
+ if select_option := self._action_scripts.get(CONF_SELECT_OPTION):
+ await self.async_run_script(
+ select_option,
+ run_variables={ATTR_OPTION: option},
+ context=self._context,
+ )
diff --git a/homeassistant/components/template/sensor.py b/homeassistant/components/template/sensor.py
index ee24407699d..ca3736ebf76 100644
--- a/homeassistant/components/template/sensor.py
+++ b/homeassistant/components/template/sensor.py
@@ -44,7 +44,10 @@ from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv, selector, template
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
from homeassistant.helpers.entity import async_generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.trigger_template_entity import TEMPLATE_SENSOR_BASE_SCHEMA
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
@@ -178,7 +181,7 @@ _LOGGER = logging.getLogger(__name__)
@callback
def _async_create_template_tracking_entities(
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddEntitiesCallback | AddConfigEntryEntitiesCallback,
hass: HomeAssistant,
definitions: list[dict],
unique_id_prefix: str | None,
@@ -237,7 +240,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
diff --git a/homeassistant/components/template/switch.py b/homeassistant/components/template/switch.py
index bddb51e5e67..b76fc28b83c 100644
--- a/homeassistant/components/template/switch.py
+++ b/homeassistant/components/template/switch.py
@@ -2,7 +2,7 @@
from __future__ import annotations
-from typing import Any
+from typing import TYPE_CHECKING, Any
import voluptuous as vol
@@ -17,6 +17,7 @@ from homeassistant.const import (
ATTR_FRIENDLY_NAME,
CONF_DEVICE_ID,
CONF_NAME,
+ CONF_STATE,
CONF_SWITCHES,
CONF_UNIQUE_ID,
CONF_VALUE_TEMPLATE,
@@ -25,24 +26,51 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import TemplateError
-from homeassistant.helpers import config_validation as cv, selector
+from homeassistant.helpers import config_validation as cv, selector, template
from homeassistant.helpers.device import async_device_info_to_link_from_device_id
from homeassistant.helpers.entity import async_generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.restore_state import RestoreEntity
-from homeassistant.helpers.script import Script
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
-from .const import CONF_TURN_OFF, CONF_TURN_ON, DOMAIN
+from .const import CONF_OBJECT_ID, CONF_PICTURE, CONF_TURN_OFF, CONF_TURN_ON, DOMAIN
from .template_entity import (
+ LEGACY_FIELDS as TEMPLATE_ENTITY_LEGACY_FIELDS,
+ TEMPLATE_ENTITY_AVAILABILITY_SCHEMA,
TEMPLATE_ENTITY_COMMON_SCHEMA_LEGACY,
+ TEMPLATE_ENTITY_ICON_SCHEMA,
TemplateEntity,
rewrite_common_legacy_to_modern_conf,
)
_VALID_STATES = [STATE_ON, STATE_OFF, "true", "false"]
-SWITCH_SCHEMA = vol.All(
+LEGACY_FIELDS = TEMPLATE_ENTITY_LEGACY_FIELDS | {
+ CONF_VALUE_TEMPLATE: CONF_STATE,
+}
+
+DEFAULT_NAME = "Template Switch"
+
+
+SWITCH_SCHEMA = (
+ vol.Schema(
+ {
+ vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.template,
+ vol.Optional(CONF_STATE): cv.template,
+ vol.Required(CONF_TURN_ON): cv.SCRIPT_SCHEMA,
+ vol.Required(CONF_TURN_OFF): cv.SCRIPT_SCHEMA,
+ vol.Optional(CONF_UNIQUE_ID): cv.string,
+ vol.Optional(CONF_PICTURE): cv.template,
+ }
+ )
+ .extend(TEMPLATE_ENTITY_AVAILABILITY_SCHEMA.schema)
+ .extend(TEMPLATE_ENTITY_ICON_SCHEMA.schema)
+)
+
+LEGACY_SWITCH_SCHEMA = vol.All(
cv.deprecated(ATTR_ENTITY_ID),
vol.Schema(
{
@@ -57,13 +85,13 @@ SWITCH_SCHEMA = vol.All(
)
PLATFORM_SCHEMA = SWITCH_PLATFORM_SCHEMA.extend(
- {vol.Required(CONF_SWITCHES): cv.schema_with_slug_keys(SWITCH_SCHEMA)}
+ {vol.Required(CONF_SWITCHES): cv.schema_with_slug_keys(LEGACY_SWITCH_SCHEMA)}
)
SWITCH_CONFIG_SCHEMA = vol.Schema(
{
vol.Required(CONF_NAME): cv.template,
- vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
+ vol.Optional(CONF_STATE): cv.template,
vol.Optional(CONF_TURN_ON): cv.SCRIPT_SCHEMA,
vol.Optional(CONF_TURN_OFF): cv.SCRIPT_SCHEMA,
vol.Optional(CONF_DEVICE_ID): selector.DeviceSelector(),
@@ -71,24 +99,62 @@ SWITCH_CONFIG_SCHEMA = vol.Schema(
)
-async def _async_create_entities(hass, config):
- """Create the Template switches."""
+def rewrite_legacy_to_modern_conf(
+ hass: HomeAssistant, config: dict[str, dict]
+) -> list[dict]:
+ """Rewrite legacy switch configuration definitions to modern ones."""
switches = []
- for object_id, entity_config in config[CONF_SWITCHES].items():
- entity_config = rewrite_common_legacy_to_modern_conf(hass, entity_config)
- unique_id = entity_config.get(CONF_UNIQUE_ID)
+ for object_id, entity_conf in config.items():
+ entity_conf = {**entity_conf, CONF_OBJECT_ID: object_id}
+
+ entity_conf = rewrite_common_legacy_to_modern_conf(
+ hass, entity_conf, LEGACY_FIELDS
+ )
+
+ if CONF_NAME not in entity_conf:
+ entity_conf[CONF_NAME] = template.Template(object_id, hass)
+
+ switches.append(entity_conf)
+
+ return switches
+
+
+def rewrite_options_to_moder_conf(option_config: dict[str, dict]) -> dict[str, dict]:
+ """Rewrite option configuration to modern configuration."""
+ option_config = {**option_config}
+
+ if CONF_VALUE_TEMPLATE in option_config:
+ option_config[CONF_STATE] = option_config.pop(CONF_VALUE_TEMPLATE)
+
+ return option_config
+
+
+@callback
+def _async_create_template_tracking_entities(
+ async_add_entities: AddEntitiesCallback,
+ hass: HomeAssistant,
+ definitions: list[dict],
+ unique_id_prefix: str | None,
+) -> None:
+ """Create the template switches."""
+ switches = []
+
+ for entity_conf in definitions:
+ unique_id = entity_conf.get(CONF_UNIQUE_ID)
+
+ if unique_id and unique_id_prefix:
+ unique_id = f"{unique_id_prefix}-{unique_id}"
switches.append(
SwitchTemplate(
hass,
- object_id,
- entity_config,
+ entity_conf,
unique_id,
)
)
- return switches
+ async_add_entities(switches)
async def async_setup_platform(
@@ -98,21 +164,34 @@ async def async_setup_platform(
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up the template switches."""
- async_add_entities(await _async_create_entities(hass, config))
+ if discovery_info is None:
+ _async_create_template_tracking_entities(
+ async_add_entities,
+ hass,
+ rewrite_legacy_to_modern_conf(hass, config[CONF_SWITCHES]),
+ None,
+ )
+ return
+
+ _async_create_template_tracking_entities(
+ async_add_entities,
+ hass,
+ discovery_info["entities"],
+ discovery_info["unique_id"],
+ )
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize config entry."""
_options = dict(config_entry.options)
_options.pop("template_type")
+ _options = rewrite_options_to_moder_conf(_options)
validated_config = SWITCH_CONFIG_SCHEMA(_options)
- async_add_entities(
- [SwitchTemplate(hass, None, validated_config, config_entry.entry_id)]
- )
+ async_add_entities([SwitchTemplate(hass, validated_config, config_entry.entry_id)])
@callback
@@ -121,7 +200,7 @@ def async_create_preview_switch(
) -> SwitchTemplate:
"""Create a preview switch."""
validated_config = SWITCH_CONFIG_SCHEMA(config | {CONF_NAME: name})
- return SwitchTemplate(hass, None, validated_config, None)
+ return SwitchTemplate(hass, validated_config, None)
class SwitchTemplate(TemplateEntity, SwitchEntity, RestoreEntity):
@@ -131,31 +210,26 @@ class SwitchTemplate(TemplateEntity, SwitchEntity, RestoreEntity):
def __init__(
self,
- hass,
- object_id,
- config,
- unique_id,
- ):
+ hass: HomeAssistant,
+ config: ConfigType,
+ unique_id: str | None,
+ ) -> None:
"""Initialize the Template switch."""
- super().__init__(
- hass, config=config, fallback_name=object_id, unique_id=unique_id
- )
- if object_id is not None:
+ super().__init__(hass, config=config, fallback_name=None, unique_id=unique_id)
+ if (object_id := config.get(CONF_OBJECT_ID)) is not None:
self.entity_id = async_generate_entity_id(
ENTITY_ID_FORMAT, object_id, hass=hass
)
- friendly_name = self._attr_name
- self._template = config.get(CONF_VALUE_TEMPLATE)
- self._on_script = (
- Script(hass, config.get(CONF_TURN_ON), friendly_name, DOMAIN)
- if config.get(CONF_TURN_ON) is not None
- else None
- )
- self._off_script = (
- Script(hass, config.get(CONF_TURN_OFF), friendly_name, DOMAIN)
- if config.get(CONF_TURN_OFF) is not None
- else None
- )
+ name = self._attr_name
+ if TYPE_CHECKING:
+ assert name is not None
+ self._template = config.get(CONF_STATE)
+
+ if on_action := config.get(CONF_TURN_ON):
+ self.add_script(CONF_TURN_ON, on_action, name, DOMAIN)
+ if off_action := config.get(CONF_TURN_OFF):
+ self.add_script(CONF_TURN_OFF, off_action, name, DOMAIN)
+
self._state: bool | None = False
self._attr_assumed_state = self._template is None
self._attr_device_info = async_device_info_to_link_from_device_id(
@@ -206,16 +280,16 @@ class SwitchTemplate(TemplateEntity, SwitchEntity, RestoreEntity):
async def async_turn_on(self, **kwargs: Any) -> None:
"""Fire the on action."""
- if self._on_script:
- await self.async_run_script(self._on_script, context=self._context)
+ if on_script := self._action_scripts.get(CONF_TURN_ON):
+ await self.async_run_script(on_script, context=self._context)
if self._template is None:
self._state = True
self.async_write_ha_state()
async def async_turn_off(self, **kwargs: Any) -> None:
"""Fire the off action."""
- if self._off_script:
- await self.async_run_script(self._off_script, context=self._context)
+ if off_script := self._action_scripts.get(CONF_TURN_OFF):
+ await self.async_run_script(off_script, context=self._context)
if self._template is None:
self._state = False
self.async_write_ha_state()
diff --git a/homeassistant/components/template/template_entity.py b/homeassistant/components/template/template_entity.py
index 8f9edca5976..88708278758 100644
--- a/homeassistant/components/template/template_entity.py
+++ b/homeassistant/components/template/template_entity.py
@@ -24,7 +24,6 @@ from homeassistant.const import (
)
from homeassistant.core import (
CALLBACK_TYPE,
- Context,
Event,
EventStateChangedData,
HomeAssistant,
@@ -41,7 +40,7 @@ from homeassistant.helpers.event import (
TrackTemplateResultInfo,
async_track_template_result,
)
-from homeassistant.helpers.script import Script, _VarsType
+from homeassistant.helpers.script_variables import ScriptVariables
from homeassistant.helpers.start import async_at_start
from homeassistant.helpers.template import (
Template,
@@ -61,6 +60,7 @@ from .const import (
CONF_AVAILABILITY_TEMPLATE,
CONF_PICTURE,
)
+from .entity import AbstractTemplateEntity
_LOGGER = logging.getLogger(__name__)
@@ -248,7 +248,7 @@ class _TemplateAttribute:
return
-class TemplateEntity(Entity): # pylint: disable=hass-enforce-class-module
+class TemplateEntity(AbstractTemplateEntity):
"""Entity that uses templates to calculate attributes."""
_attr_available = True
@@ -268,6 +268,7 @@ class TemplateEntity(Entity): # pylint: disable=hass-enforce-class-module
unique_id: str | None = None,
) -> None:
"""Template Entity."""
+ super().__init__(hass)
self._template_attrs: dict[Template, list[_TemplateAttribute]] = {}
self._template_result_info: TrackTemplateResultInfo | None = None
self._attr_extra_state_attributes = {}
@@ -285,6 +286,7 @@ class TemplateEntity(Entity): # pylint: disable=hass-enforce-class-module
]
| None
) = None
+ self._run_variables: ScriptVariables | dict
if config is None:
self._attribute_templates = attribute_templates
self._availability_template = availability_template
@@ -339,18 +341,6 @@ class TemplateEntity(Entity): # pylint: disable=hass-enforce-class-module
variables=variables, parse_result=False
)
- @callback
- def _render_variables(self) -> dict:
- if isinstance(self._run_variables, dict):
- return self._run_variables
-
- return self._run_variables.async_render(
- self.hass,
- {
- "this": TemplateStateFromEntityId(self.hass, self.entity_id),
- },
- )
-
@callback
def _update_available(self, result: str | TemplateError) -> None:
if isinstance(result, TemplateError):
@@ -387,6 +377,18 @@ class TemplateEntity(Entity): # pylint: disable=hass-enforce-class-module
return None
return cast(str, self._blueprint_inputs[CONF_USE_BLUEPRINT][CONF_PATH])
+ def _render_script_variables(self) -> dict[str, Any]:
+ """Render configured variables."""
+ if isinstance(self._run_variables, dict):
+ return self._run_variables
+
+ return self._run_variables.async_render(
+ self.hass,
+ {
+ "this": TemplateStateFromEntityId(self.hass, self.entity_id),
+ },
+ )
+
def add_template_attribute(
self,
attribute: str,
@@ -488,7 +490,7 @@ class TemplateEntity(Entity): # pylint: disable=hass-enforce-class-module
variables = {
"this": TemplateStateFromEntityId(self.hass, self.entity_id),
- **self._render_variables(),
+ **self._render_script_variables(),
}
for template, attributes in self._template_attrs.items():
@@ -581,22 +583,3 @@ class TemplateEntity(Entity): # pylint: disable=hass-enforce-class-module
"""Call for forced update."""
assert self._template_result_info
self._template_result_info.async_refresh()
-
- async def async_run_script(
- self,
- script: Script,
- *,
- run_variables: _VarsType | None = None,
- context: Context | None = None,
- ) -> None:
- """Run an action script."""
- if run_variables is None:
- run_variables = {}
- await script.async_run(
- run_variables={
- "this": TemplateStateFromEntityId(self.hass, self.entity_id),
- **self._render_variables(),
- **run_variables,
- },
- context=context,
- )
diff --git a/homeassistant/components/template/trigger_entity.py b/homeassistant/components/template/trigger_entity.py
index 5130f332d5b..87c93b6143b 100644
--- a/homeassistant/components/template/trigger_entity.py
+++ b/homeassistant/components/template/trigger_entity.py
@@ -8,10 +8,13 @@ from homeassistant.helpers.trigger_template_entity import TriggerBaseEntity
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import TriggerUpdateCoordinator
+from .entity import AbstractTemplateEntity
class TriggerEntity( # pylint: disable=hass-enforce-class-module
- TriggerBaseEntity, CoordinatorEntity[TriggerUpdateCoordinator]
+ TriggerBaseEntity,
+ CoordinatorEntity[TriggerUpdateCoordinator],
+ AbstractTemplateEntity,
):
"""Template entity based on trigger data."""
@@ -24,6 +27,7 @@ class TriggerEntity( # pylint: disable=hass-enforce-class-module
"""Initialize the entity."""
CoordinatorEntity.__init__(self, coordinator)
TriggerBaseEntity.__init__(self, hass, config)
+ AbstractTemplateEntity.__init__(self, hass)
async def async_added_to_hass(self) -> None:
"""Handle being added to Home Assistant."""
@@ -38,6 +42,16 @@ class TriggerEntity( # pylint: disable=hass-enforce-class-module
else:
self._unique_id = unique_id
+ @property
+ def referenced_blueprint(self) -> str | None:
+ """Return referenced blueprint or None."""
+ return self.coordinator.referenced_blueprint
+
+ @callback
+ def _render_script_variables(self) -> dict:
+ """Render configured variables."""
+ return self.coordinator.data["run_variables"]
+
@callback
def _process_data(self) -> None:
"""Process new data."""
diff --git a/homeassistant/components/template/vacuum.py b/homeassistant/components/template/vacuum.py
index b977f4e659a..c4d41b52f31 100644
--- a/homeassistant/components/template/vacuum.py
+++ b/homeassistant/components/template/vacuum.py
@@ -3,7 +3,7 @@
from __future__ import annotations
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
import voluptuous as vol
@@ -33,7 +33,6 @@ from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity import async_generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.script import Script
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import DOMAIN
@@ -90,7 +89,7 @@ PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend(
)
-async def _async_create_entities(hass, config):
+async def _async_create_entities(hass: HomeAssistant, config: ConfigType):
"""Create the Template Vacuums."""
vacuums = []
@@ -127,11 +126,11 @@ class TemplateVacuum(TemplateEntity, StateVacuumEntity):
def __init__(
self,
- hass,
+ hass: HomeAssistant,
object_id,
- config,
+ config: ConfigType,
unique_id,
- ):
+ ) -> None:
"""Initialize the vacuum."""
super().__init__(
hass, config=config, fallback_name=object_id, unique_id=unique_id
@@ -139,7 +138,9 @@ class TemplateVacuum(TemplateEntity, StateVacuumEntity):
self.entity_id = async_generate_entity_id(
ENTITY_ID_FORMAT, object_id, hass=hass
)
- friendly_name = self._attr_name
+ name = self._attr_name
+ if TYPE_CHECKING:
+ assert name is not None
self._template = config.get(CONF_VALUE_TEMPLATE)
self._battery_level_template = config.get(CONF_BATTERY_LEVEL_TEMPLATE)
@@ -148,43 +149,18 @@ class TemplateVacuum(TemplateEntity, StateVacuumEntity):
VacuumEntityFeature.START | VacuumEntityFeature.STATE
)
- self._start_script = Script(hass, config[SERVICE_START], friendly_name, DOMAIN)
-
- self._pause_script = None
- if pause_action := config.get(SERVICE_PAUSE):
- self._pause_script = Script(hass, pause_action, friendly_name, DOMAIN)
- self._attr_supported_features |= VacuumEntityFeature.PAUSE
-
- self._stop_script = None
- if stop_action := config.get(SERVICE_STOP):
- self._stop_script = Script(hass, stop_action, friendly_name, DOMAIN)
- self._attr_supported_features |= VacuumEntityFeature.STOP
-
- self._return_to_base_script = None
- if return_to_base_action := config.get(SERVICE_RETURN_TO_BASE):
- self._return_to_base_script = Script(
- hass, return_to_base_action, friendly_name, DOMAIN
- )
- self._attr_supported_features |= VacuumEntityFeature.RETURN_HOME
-
- self._clean_spot_script = None
- if clean_spot_action := config.get(SERVICE_CLEAN_SPOT):
- self._clean_spot_script = Script(
- hass, clean_spot_action, friendly_name, DOMAIN
- )
- self._attr_supported_features |= VacuumEntityFeature.CLEAN_SPOT
-
- self._locate_script = None
- if locate_action := config.get(SERVICE_LOCATE):
- self._locate_script = Script(hass, locate_action, friendly_name, DOMAIN)
- self._attr_supported_features |= VacuumEntityFeature.LOCATE
-
- self._set_fan_speed_script = None
- if set_fan_speed_action := config.get(SERVICE_SET_FAN_SPEED):
- self._set_fan_speed_script = Script(
- hass, set_fan_speed_action, friendly_name, DOMAIN
- )
- self._attr_supported_features |= VacuumEntityFeature.FAN_SPEED
+ for action_id, supported_feature in (
+ (SERVICE_START, 0),
+ (SERVICE_PAUSE, VacuumEntityFeature.PAUSE),
+ (SERVICE_STOP, VacuumEntityFeature.STOP),
+ (SERVICE_RETURN_TO_BASE, VacuumEntityFeature.RETURN_HOME),
+ (SERVICE_CLEAN_SPOT, VacuumEntityFeature.CLEAN_SPOT),
+ (SERVICE_LOCATE, VacuumEntityFeature.LOCATE),
+ (SERVICE_SET_FAN_SPEED, VacuumEntityFeature.FAN_SPEED),
+ ):
+ if action_config := config.get(action_id):
+ self.add_script(action_id, action_config, name, DOMAIN)
+ self._attr_supported_features |= supported_feature
self._state = None
self._battery_level = None
@@ -203,62 +179,50 @@ class TemplateVacuum(TemplateEntity, StateVacuumEntity):
async def async_start(self) -> None:
"""Start or resume the cleaning task."""
- await self.async_run_script(self._start_script, context=self._context)
+ await self.async_run_script(
+ self._action_scripts[SERVICE_START], context=self._context
+ )
async def async_pause(self) -> None:
"""Pause the cleaning task."""
- if self._pause_script is None:
- return
-
- await self.async_run_script(self._pause_script, context=self._context)
+ if script := self._action_scripts.get(SERVICE_PAUSE):
+ await self.async_run_script(script, context=self._context)
async def async_stop(self, **kwargs: Any) -> None:
"""Stop the cleaning task."""
- if self._stop_script is None:
- return
-
- await self.async_run_script(self._stop_script, context=self._context)
+ if script := self._action_scripts.get(SERVICE_STOP):
+ await self.async_run_script(script, context=self._context)
async def async_return_to_base(self, **kwargs: Any) -> None:
"""Set the vacuum cleaner to return to the dock."""
- if self._return_to_base_script is None:
- return
-
- await self.async_run_script(self._return_to_base_script, context=self._context)
+ if script := self._action_scripts.get(SERVICE_RETURN_TO_BASE):
+ await self.async_run_script(script, context=self._context)
async def async_clean_spot(self, **kwargs: Any) -> None:
"""Perform a spot clean-up."""
- if self._clean_spot_script is None:
- return
-
- await self.async_run_script(self._clean_spot_script, context=self._context)
+ if script := self._action_scripts.get(SERVICE_CLEAN_SPOT):
+ await self.async_run_script(script, context=self._context)
async def async_locate(self, **kwargs: Any) -> None:
"""Locate the vacuum cleaner."""
- if self._locate_script is None:
- return
-
- await self.async_run_script(self._locate_script, context=self._context)
+ if script := self._action_scripts.get(SERVICE_LOCATE):
+ await self.async_run_script(script, context=self._context)
async def async_set_fan_speed(self, fan_speed: str, **kwargs: Any) -> None:
"""Set fan speed."""
- if self._set_fan_speed_script is None:
- return
-
- if fan_speed in self._attr_fan_speed_list:
- self._attr_fan_speed = fan_speed
- await self.async_run_script(
- self._set_fan_speed_script,
- run_variables={ATTR_FAN_SPEED: fan_speed},
- context=self._context,
- )
- else:
+ if fan_speed not in self._attr_fan_speed_list:
_LOGGER.error(
"Received invalid fan speed: %s for entity %s. Expected: %s",
fan_speed,
self.entity_id,
self._attr_fan_speed_list,
)
+ return
+
+ if script := self._action_scripts.get(SERVICE_SET_FAN_SPEED):
+ await self.async_run_script(
+ script, run_variables={ATTR_FAN_SPEED: fan_speed}, context=self._context
+ )
@callback
def _async_setup_templates(self) -> None:
diff --git a/homeassistant/components/tesla_fleet/__init__.py b/homeassistant/components/tesla_fleet/__init__.py
index 634e8f845f9..27bfb9134ab 100644
--- a/homeassistant/components/tesla_fleet/__init__.py
+++ b/homeassistant/components/tesla_fleet/__init__.py
@@ -139,7 +139,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TeslaFleetConfigEntry) -
api = VehicleSigned(tesla.vehicle, vin)
else:
api = VehicleSpecific(tesla.vehicle, vin)
- coordinator = TeslaFleetVehicleDataCoordinator(hass, api, product)
+ coordinator = TeslaFleetVehicleDataCoordinator(hass, entry, api, product)
await coordinator.async_config_entry_first_refresh()
@@ -175,9 +175,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: TeslaFleetConfigEntry) -
api = EnergySpecific(tesla.energy, site_id)
- live_coordinator = TeslaFleetEnergySiteLiveCoordinator(hass, api)
- history_coordinator = TeslaFleetEnergySiteHistoryCoordinator(hass, api)
- info_coordinator = TeslaFleetEnergySiteInfoCoordinator(hass, api, product)
+ live_coordinator = TeslaFleetEnergySiteLiveCoordinator(hass, entry, api)
+ history_coordinator = TeslaFleetEnergySiteHistoryCoordinator(
+ hass, entry, api
+ )
+ info_coordinator = TeslaFleetEnergySiteInfoCoordinator(
+ hass, entry, api, product
+ )
await live_coordinator.async_config_entry_first_refresh()
await history_coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/tesla_fleet/binary_sensor.py b/homeassistant/components/tesla_fleet/binary_sensor.py
index b92ef9233d1..886fe304c91 100644
--- a/homeassistant/components/tesla_fleet/binary_sensor.py
+++ b/homeassistant/components/tesla_fleet/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TeslaFleetConfigEntry
@@ -179,7 +179,7 @@ ENERGY_INFO_DESCRIPTIONS: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tesla Fleet binary sensor platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/button.py b/homeassistant/components/tesla_fleet/button.py
index aea0f91a97c..2ddce2d517b 100644
--- a/homeassistant/components/tesla_fleet/button.py
+++ b/homeassistant/components/tesla_fleet/button.py
@@ -10,7 +10,7 @@ from tesla_fleet_api.const import Scope
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslaFleetConfigEntry
from .entity import TeslaFleetVehicleEntity
@@ -61,7 +61,7 @@ DESCRIPTIONS: tuple[TeslaFleetButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the TeslaFleet Button platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/climate.py b/homeassistant/components/tesla_fleet/climate.py
index 06e9c9d7c64..f752509ee17 100644
--- a/homeassistant/components/tesla_fleet/climate.py
+++ b/homeassistant/components/tesla_fleet/climate.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslaFleetConfigEntry
from .const import DOMAIN, TeslaFleetClimateSide
@@ -38,7 +38,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tesla Fleet Climate platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/coordinator.py b/homeassistant/components/tesla_fleet/coordinator.py
index 4d99319d49f..6f881d0feba 100644
--- a/homeassistant/components/tesla_fleet/coordinator.py
+++ b/homeassistant/components/tesla_fleet/coordinator.py
@@ -1,9 +1,11 @@
"""Tesla Fleet Data Coordinator."""
+from __future__ import annotations
+
from datetime import datetime, timedelta
from random import randint
from time import time
-from typing import Any
+from typing import TYPE_CHECKING, Any
from tesla_fleet_api import EnergySpecific, VehicleSpecific
from tesla_fleet_api.const import TeslaEnergyPeriod, VehicleDataEndpoint
@@ -15,12 +17,14 @@ from tesla_fleet_api.exceptions import (
TeslaFleetError,
VehicleOffline,
)
-from tesla_fleet_api.ratecalculator import RateCalculator
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import TeslaFleetConfigEntry
+
from .const import ENERGY_HISTORY_FIELDS, LOGGER, TeslaFleetState
VEHICLE_INTERVAL_SECONDS = 300
@@ -57,18 +61,23 @@ def flatten(data: dict[str, Any], parent: str | None = None) -> dict[str, Any]:
class TeslaFleetVehicleDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching data from the TeslaFleet API."""
+ config_entry: TeslaFleetConfigEntry
updated_once: bool
pre2021: bool
last_active: datetime
- rate: RateCalculator
def __init__(
- self, hass: HomeAssistant, api: VehicleSpecific, product: dict
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslaFleetConfigEntry,
+ api: VehicleSpecific,
+ product: dict,
) -> None:
"""Initialize TeslaFleet Vehicle Update Coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Tesla Fleet Vehicle",
update_interval=VEHICLE_INTERVAL,
)
@@ -76,44 +85,36 @@ class TeslaFleetVehicleDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
self.data = flatten(product)
self.updated_once = False
self.last_active = datetime.now()
- self.rate = RateCalculator(100, 86400, VEHICLE_INTERVAL_SECONDS, 3600, 5)
async def _async_update_data(self) -> dict[str, Any]:
"""Update vehicle data using TeslaFleet API."""
try:
- # Check if the vehicle is awake using a non-rate limited API call
- if self.data["state"] != TeslaFleetState.ONLINE:
- response = await self.api.vehicle()
- self.data["state"] = response["response"]["state"]
+ # Check if the vehicle is awake using a free API call
+ response = await self.api.vehicle()
+ self.data["state"] = response["response"]["state"]
if self.data["state"] != TeslaFleetState.ONLINE:
return self.data
- # This is a rated limited API call
- self.rate.consume()
response = await self.api.vehicle_data(endpoints=ENDPOINTS)
data = response["response"]
except VehicleOffline:
self.data["state"] = TeslaFleetState.ASLEEP
return self.data
- except RateLimited as e:
+ except RateLimited:
LOGGER.warning(
- "%s rate limited, will retry in %s seconds",
+ "%s rate limited, will skip refresh",
self.name,
- e.data.get("after"),
)
- if "after" in e.data:
- self.update_interval = timedelta(seconds=int(e.data["after"]))
return self.data
except (InvalidToken, OAuthExpired, LoginRequired) as e:
raise ConfigEntryAuthFailed from e
except TeslaFleetError as e:
raise UpdateFailed(e.message) from e
- # Calculate ideal refresh interval
- self.update_interval = timedelta(seconds=self.rate.calculate())
+ self.update_interval = VEHICLE_INTERVAL
self.updated_once = True
@@ -141,13 +142,20 @@ class TeslaFleetVehicleDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class TeslaFleetEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site live status from the TeslaFleet API."""
+ config_entry: TeslaFleetConfigEntry
updated_once: bool
- def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslaFleetConfigEntry,
+ api: EnergySpecific,
+ ) -> None:
"""Initialize TeslaFleet Energy Site Live coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Tesla Fleet Energy Site Live",
update_interval=timedelta(seconds=10),
)
@@ -188,11 +196,19 @@ class TeslaFleetEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]])
class TeslaFleetEnergySiteHistoryCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site history import and export from the Tesla Fleet API."""
- def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
+ config_entry: TeslaFleetConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslaFleetConfigEntry,
+ api: EnergySpecific,
+ ) -> None:
"""Initialize Tesla Fleet Energy Site History coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=f"Tesla Fleet Energy History {api.energy_site_id}",
update_interval=timedelta(seconds=300),
)
@@ -232,7 +248,7 @@ class TeslaFleetEnergySiteHistoryCoordinator(DataUpdateCoordinator[dict[str, Any
self.updated_once = True
# Add all time periods together
- output = {key: 0 for key in ENERGY_HISTORY_FIELDS}
+ output = dict.fromkeys(ENERGY_HISTORY_FIELDS, 0)
for period in data.get("time_series", []):
for key in ENERGY_HISTORY_FIELDS:
output[key] += period.get(key, 0)
@@ -243,13 +259,21 @@ class TeslaFleetEnergySiteHistoryCoordinator(DataUpdateCoordinator[dict[str, Any
class TeslaFleetEnergySiteInfoCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site info from the TeslaFleet API."""
+ config_entry: TeslaFleetConfigEntry
updated_once: bool
- def __init__(self, hass: HomeAssistant, api: EnergySpecific, product: dict) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslaFleetConfigEntry,
+ api: EnergySpecific,
+ product: dict,
+ ) -> None:
"""Initialize TeslaFleet Energy Info coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Tesla Fleet Energy Site Info",
update_interval=timedelta(seconds=15),
)
diff --git a/homeassistant/components/tesla_fleet/cover.py b/homeassistant/components/tesla_fleet/cover.py
index f270734424f..701b107f9f9 100644
--- a/homeassistant/components/tesla_fleet/cover.py
+++ b/homeassistant/components/tesla_fleet/cover.py
@@ -12,7 +12,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslaFleetConfigEntry
from .entity import TeslaFleetVehicleEntity
@@ -28,7 +28,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the TeslaFleet cover platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/device_tracker.py b/homeassistant/components/tesla_fleet/device_tracker.py
index d6dcef895a6..19bf353c62d 100644
--- a/homeassistant/components/tesla_fleet/device_tracker.py
+++ b/homeassistant/components/tesla_fleet/device_tracker.py
@@ -6,7 +6,7 @@ from homeassistant.components.device_tracker.config_entry import TrackerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_HOME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .entity import TeslaFleetVehicleEntity
@@ -14,7 +14,9 @@ from .models import TeslaFleetVehicleData
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tesla Fleet device tracker platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/lock.py b/homeassistant/components/tesla_fleet/lock.py
index 32998d409be..cdb1d4b066b 100644
--- a/homeassistant/components/tesla_fleet/lock.py
+++ b/homeassistant/components/tesla_fleet/lock.py
@@ -9,7 +9,7 @@ from tesla_fleet_api.const import Scope
from homeassistant.components.lock import LockEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslaFleetConfigEntry
from .const import DOMAIN
@@ -25,7 +25,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the TeslaFleet lock platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/manifest.json b/homeassistant/components/tesla_fleet/manifest.json
index 330745316d7..010197ccbd9 100644
--- a/homeassistant/components/tesla_fleet/manifest.json
+++ b/homeassistant/components/tesla_fleet/manifest.json
@@ -7,5 +7,5 @@
"documentation": "https://www.home-assistant.io/integrations/tesla_fleet",
"iot_class": "cloud_polling",
"loggers": ["tesla-fleet-api"],
- "requirements": ["tesla-fleet-api==0.9.8"]
+ "requirements": ["tesla-fleet-api==0.9.13"]
}
diff --git a/homeassistant/components/tesla_fleet/media_player.py b/homeassistant/components/tesla_fleet/media_player.py
index 455c990077d..89f0768f082 100644
--- a/homeassistant/components/tesla_fleet/media_player.py
+++ b/homeassistant/components/tesla_fleet/media_player.py
@@ -11,7 +11,7 @@ from homeassistant.components.media_player import (
MediaPlayerState,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslaFleetConfigEntry
from .entity import TeslaFleetVehicleEntity
@@ -33,7 +33,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tesla Fleet Media platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/number.py b/homeassistant/components/tesla_fleet/number.py
index b806b4dbc77..a1123ab9553 100644
--- a/homeassistant/components/tesla_fleet/number.py
+++ b/homeassistant/components/tesla_fleet/number.py
@@ -18,7 +18,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import PERCENTAGE, PRECISION_WHOLE, UnitOfElectricCurrent
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level
from . import TeslaFleetConfigEntry
@@ -95,7 +95,7 @@ ENERGY_INFO_DESCRIPTIONS: tuple[TeslaFleetNumberBatteryEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the TeslaFleet number platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/select.py b/homeassistant/components/tesla_fleet/select.py
index 515a0e7c2e7..1c495657bc1 100644
--- a/homeassistant/components/tesla_fleet/select.py
+++ b/homeassistant/components/tesla_fleet/select.py
@@ -10,7 +10,7 @@ from tesla_fleet_api.const import EnergyExportMode, EnergyOperationMode, Scope,
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslaFleetConfigEntry
from .entity import TeslaFleetEnergyInfoEntity, TeslaFleetVehicleEntity
@@ -78,7 +78,7 @@ SEAT_HEATER_DESCRIPTIONS: tuple[SeatHeaterDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the TeslaFleet select platform from a config entry."""
diff --git a/homeassistant/components/tesla_fleet/sensor.py b/homeassistant/components/tesla_fleet/sensor.py
index c1d38bf85c5..bdd5ce2c001 100644
--- a/homeassistant/components/tesla_fleet/sensor.py
+++ b/homeassistant/components/tesla_fleet/sensor.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
from homeassistant.util.variance import ignore_variance
@@ -446,7 +446,7 @@ ENERGY_INFO_DESCRIPTIONS: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tesla Fleet sensor platform from a config entry."""
async_add_entities(
@@ -466,6 +466,7 @@ async def async_setup_entry(
for energysite in entry.runtime_data.energysites
for description in ENERGY_LIVE_DESCRIPTIONS
if description.key in energysite.live_coordinator.data
+ or description.key == "percentage_charged"
),
( # Add energy site history
TeslaFleetEnergyHistorySensorEntity(energysite, description)
diff --git a/homeassistant/components/tesla_fleet/strings.json b/homeassistant/components/tesla_fleet/strings.json
index 540ea2b7135..331885893fe 100644
--- a/homeassistant/components/tesla_fleet/strings.json
+++ b/homeassistant/components/tesla_fleet/strings.json
@@ -329,7 +329,7 @@
"name": "Charging",
"state": {
"starting": "Starting",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"disconnected": "Disconnected",
"stopped": "Stopped",
"complete": "Complete",
diff --git a/homeassistant/components/tesla_fleet/switch.py b/homeassistant/components/tesla_fleet/switch.py
index 054ea84cbe1..614af8772cc 100644
--- a/homeassistant/components/tesla_fleet/switch.py
+++ b/homeassistant/components/tesla_fleet/switch.py
@@ -15,7 +15,7 @@ from homeassistant.components.switch import (
SwitchEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TeslaFleetConfigEntry
@@ -94,7 +94,7 @@ VEHICLE_DESCRIPTIONS: tuple[TeslaFleetSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslaFleetConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the TeslaFleet Switch platform from a config entry."""
diff --git a/homeassistant/components/tesla_wall_connector/binary_sensor.py b/homeassistant/components/tesla_wall_connector/binary_sensor.py
index f7ef385b8ed..6d60162412e 100644
--- a/homeassistant/components/tesla_wall_connector/binary_sensor.py
+++ b/homeassistant/components/tesla_wall_connector/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WallConnectorData
from .const import DOMAIN, WALLCONNECTOR_DATA_VITALS
@@ -48,7 +48,7 @@ WALL_CONNECTOR_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the Wall Connector sensor devices."""
wall_connector_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/tesla_wall_connector/sensor.py b/homeassistant/components/tesla_wall_connector/sensor.py
index a50c81c912e..c6c63a93edb 100644
--- a/homeassistant/components/tesla_wall_connector/sensor.py
+++ b/homeassistant/components/tesla_wall_connector/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WallConnectorData
from .const import DOMAIN, WALLCONNECTOR_DATA_LIFETIME, WALLCONNECTOR_DATA_VITALS
@@ -187,7 +187,7 @@ WALL_CONNECTOR_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the Wall Connector sensor devices."""
wall_connector_data = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/tesla_wall_connector/strings.json b/homeassistant/components/tesla_wall_connector/strings.json
index 1a03207a012..b356a9f3ebc 100644
--- a/homeassistant/components/tesla_wall_connector/strings.json
+++ b/homeassistant/components/tesla_wall_connector/strings.json
@@ -42,7 +42,7 @@
"charging_finished": "Charging finished",
"waiting_car": "Waiting for car",
"charging_reduced": "Charging (reduced)",
- "charging": "Charging"
+ "charging": "[%key:common::state::charging%]"
}
},
"status_code": {
diff --git a/homeassistant/components/teslemetry/__init__.py b/homeassistant/components/teslemetry/__init__.py
index 6e60b34825f..eef974cc5a7 100644
--- a/homeassistant/components/teslemetry/__init__.py
+++ b/homeassistant/components/teslemetry/__init__.py
@@ -112,7 +112,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TeslemetryConfigEntry) -
product.pop("cached_data", None)
vin = product["vin"]
api = VehicleSpecific(teslemetry.vehicle, vin)
- coordinator = TeslemetryVehicleDataCoordinator(hass, api, product)
+ coordinator = TeslemetryVehicleDataCoordinator(hass, entry, api, product)
device = DeviceInfo(
identifiers={(DOMAIN, vin)},
manufacturer="Tesla",
@@ -177,15 +177,17 @@ async def async_setup_entry(hass: HomeAssistant, entry: TeslemetryConfigEntry) -
TeslemetryEnergyData(
api=api,
live_coordinator=(
- TeslemetryEnergySiteLiveCoordinator(hass, api, live_status)
+ TeslemetryEnergySiteLiveCoordinator(
+ hass, entry, api, live_status
+ )
if isinstance(live_status, dict)
else None
),
info_coordinator=TeslemetryEnergySiteInfoCoordinator(
- hass, api, product
+ hass, entry, api, product
),
history_coordinator=(
- TeslemetryEnergyHistoryCoordinator(hass, api)
+ TeslemetryEnergyHistoryCoordinator(hass, entry, api)
if powerwall
else None
),
@@ -242,7 +244,9 @@ async def async_unload_entry(hass: HomeAssistant, entry: TeslemetryConfigEntry)
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
-async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, config_entry: TeslemetryConfigEntry
+) -> bool:
"""Migrate config entry."""
if config_entry.version > 1:
return False
@@ -282,7 +286,7 @@ def create_handle_vehicle_stream(vin: str, coordinator) -> Callable[[dict], None
async def async_setup_stream(
- hass: HomeAssistant, entry: ConfigEntry, vehicle: TeslemetryVehicleData
+ hass: HomeAssistant, entry: TeslemetryConfigEntry, vehicle: TeslemetryVehicleData
):
"""Set up the stream for a vehicle."""
diff --git a/homeassistant/components/teslemetry/binary_sensor.py b/homeassistant/components/teslemetry/binary_sensor.py
index 0b6823f8b61..9d14df4501b 100644
--- a/homeassistant/components/teslemetry/binary_sensor.py
+++ b/homeassistant/components/teslemetry/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import StateType
@@ -377,7 +377,7 @@ ENERGY_INFO_DESCRIPTIONS: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry binary sensor platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/button.py b/homeassistant/components/teslemetry/button.py
index ceeda265795..4ca2fd9b166 100644
--- a/homeassistant/components/teslemetry/button.py
+++ b/homeassistant/components/teslemetry/button.py
@@ -10,7 +10,7 @@ from tesla_fleet_api.const import Scope
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslemetryConfigEntry
from .entity import TeslemetryVehicleEntity
@@ -61,7 +61,7 @@ DESCRIPTIONS: tuple[TeslemetryButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry Button platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/climate.py b/homeassistant/components/teslemetry/climate.py
index 95b769a1c2d..86811131ab6 100644
--- a/homeassistant/components/teslemetry/climate.py
+++ b/homeassistant/components/teslemetry/climate.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TeslemetryConfigEntry
from .const import DOMAIN, TeslemetryClimateSide
@@ -38,7 +38,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry Climate platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/coordinator.py b/homeassistant/components/teslemetry/coordinator.py
index d39402c622c..f902fb4cc1b 100644
--- a/homeassistant/components/teslemetry/coordinator.py
+++ b/homeassistant/components/teslemetry/coordinator.py
@@ -1,7 +1,9 @@
"""Teslemetry Data Coordinator."""
+from __future__ import annotations
+
from datetime import datetime, timedelta
-from typing import Any
+from typing import TYPE_CHECKING, Any
from tesla_fleet_api import EnergySpecific, VehicleSpecific
from tesla_fleet_api.const import TeslaEnergyPeriod, VehicleDataEndpoint
@@ -15,6 +17,9 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import TeslemetryConfigEntry
+
from .const import ENERGY_HISTORY_FIELDS, LOGGER
from .helpers import flatten
@@ -37,15 +42,21 @@ ENDPOINTS = [
class TeslemetryVehicleDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching data from the Teslemetry API."""
+ config_entry: TeslemetryConfigEntry
last_active: datetime
def __init__(
- self, hass: HomeAssistant, api: VehicleSpecific, product: dict
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslemetryConfigEntry,
+ api: VehicleSpecific,
+ product: dict,
) -> None:
"""Initialize Teslemetry Vehicle Update Coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Teslemetry Vehicle",
update_interval=VEHICLE_INTERVAL,
)
@@ -69,13 +80,21 @@ class TeslemetryVehicleDataCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class TeslemetryEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site live status from the Teslemetry API."""
+ config_entry: TeslemetryConfigEntry
updated_once: bool
- def __init__(self, hass: HomeAssistant, api: EnergySpecific, data: dict) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslemetryConfigEntry,
+ api: EnergySpecific,
+ data: dict,
+ ) -> None:
"""Initialize Teslemetry Energy Site Live coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Teslemetry Energy Site Live",
update_interval=ENERGY_LIVE_INTERVAL,
)
@@ -108,11 +127,20 @@ class TeslemetryEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]])
class TeslemetryEnergySiteInfoCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site info from the Teslemetry API."""
- def __init__(self, hass: HomeAssistant, api: EnergySpecific, product: dict) -> None:
+ config_entry: TeslemetryConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslemetryConfigEntry,
+ api: EnergySpecific,
+ product: dict,
+ ) -> None:
"""Initialize Teslemetry Energy Info coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="Teslemetry Energy Site Info",
update_interval=ENERGY_INFO_INTERVAL,
)
@@ -135,11 +163,19 @@ class TeslemetryEnergySiteInfoCoordinator(DataUpdateCoordinator[dict[str, Any]])
class TeslemetryEnergyHistoryCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site info from the Teslemetry API."""
- def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
+ config_entry: TeslemetryConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: TeslemetryConfigEntry,
+ api: EnergySpecific,
+ ) -> None:
"""Initialize Teslemetry Energy Info coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=f"Teslemetry Energy History {api.energy_site_id}",
update_interval=ENERGY_HISTORY_INTERVAL,
)
@@ -156,7 +192,7 @@ class TeslemetryEnergyHistoryCoordinator(DataUpdateCoordinator[dict[str, Any]]):
raise UpdateFailed(e.message) from e
# Add all time periods together
- output = {key: 0 for key in ENERGY_HISTORY_FIELDS}
+ output = dict.fromkeys(ENERGY_HISTORY_FIELDS, 0)
for period in data.get("time_series", []):
for key in ENERGY_HISTORY_FIELDS:
output[key] += period.get(key, 0)
diff --git a/homeassistant/components/teslemetry/cover.py b/homeassistant/components/teslemetry/cover.py
index 4cc15b6feb8..de91f43f084 100644
--- a/homeassistant/components/teslemetry/cover.py
+++ b/homeassistant/components/teslemetry/cover.py
@@ -15,7 +15,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import TeslemetryConfigEntry
@@ -36,7 +36,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry cover platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/device_tracker.py b/homeassistant/components/teslemetry/device_tracker.py
index 42c8fea8d09..6a758e68497 100644
--- a/homeassistant/components/teslemetry/device_tracker.py
+++ b/homeassistant/components/teslemetry/device_tracker.py
@@ -14,7 +14,7 @@ from homeassistant.components.device_tracker.config_entry import (
)
from homeassistant.const import STATE_HOME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import TeslemetryConfigEntry
@@ -68,7 +68,7 @@ DESCRIPTIONS: tuple[TeslemetryDeviceTrackerEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry device tracker platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/diagnostics.py b/homeassistant/components/teslemetry/diagnostics.py
index fc601a58ae6..755935951fc 100644
--- a/homeassistant/components/teslemetry/diagnostics.py
+++ b/homeassistant/components/teslemetry/diagnostics.py
@@ -35,7 +35,9 @@ async def async_get_config_entry_diagnostics(
vehicles = [
{
"data": async_redact_data(x.coordinator.data, VEHICLE_REDACT),
- # Stream diag will go here when implemented
+ "stream": {
+ "config": x.stream_vehicle.config,
+ },
}
for x in entry.runtime_data.vehicles
]
@@ -45,6 +47,7 @@ async def async_get_config_entry_diagnostics(
if x.live_coordinator
else None,
"info": async_redact_data(x.info_coordinator.data, ENERGY_INFO_REDACT),
+ "history": x.history_coordinator.data if x.history_coordinator else None,
}
for x in entry.runtime_data.energysites
]
diff --git a/homeassistant/components/teslemetry/lock.py b/homeassistant/components/teslemetry/lock.py
index 18b88273bec..68505a12a13 100644
--- a/homeassistant/components/teslemetry/lock.py
+++ b/homeassistant/components/teslemetry/lock.py
@@ -10,7 +10,7 @@ from tesla_fleet_api.const import Scope
from homeassistant.components.lock import LockEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import TeslemetryConfigEntry
@@ -31,7 +31,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry lock platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/manifest.json b/homeassistant/components/teslemetry/manifest.json
index bfa0d831a16..3d37ced8cff 100644
--- a/homeassistant/components/teslemetry/manifest.json
+++ b/homeassistant/components/teslemetry/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/teslemetry",
"iot_class": "cloud_polling",
"loggers": ["tesla-fleet-api"],
- "requirements": ["tesla-fleet-api==0.9.8", "teslemetry-stream==0.6.10"]
+ "requirements": ["tesla-fleet-api==0.9.13", "teslemetry-stream==0.6.12"]
}
diff --git a/homeassistant/components/teslemetry/media_player.py b/homeassistant/components/teslemetry/media_player.py
index e0e144ffe3a..409b409e325 100644
--- a/homeassistant/components/teslemetry/media_player.py
+++ b/homeassistant/components/teslemetry/media_player.py
@@ -2,6 +2,7 @@
from __future__ import annotations
+from tesla_fleet_api import VehicleSpecific
from tesla_fleet_api.const import Scope
from homeassistant.components.media_player import (
@@ -11,10 +12,15 @@ from homeassistant.components.media_player import (
MediaPlayerState,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.restore_state import RestoreEntity
from . import TeslemetryConfigEntry
-from .entity import TeslemetryVehicleEntity
+from .entity import (
+ TeslemetryRootEntity,
+ TeslemetryVehicleEntity,
+ TeslemetryVehicleStreamEntity,
+)
from .helpers import handle_vehicle_command
from .models import TeslemetryVehicleData
@@ -24,8 +30,16 @@ STATES = {
"Stopped": MediaPlayerState.IDLE,
"Off": MediaPlayerState.OFF,
}
-VOLUME_MAX = 11.0
-VOLUME_STEP = 1.0 / 3
+DISPLAY_STATES = {
+ "On": MediaPlayerState.IDLE,
+ "Accessory": MediaPlayerState.IDLE,
+ "Charging": MediaPlayerState.OFF,
+ "Sentry": MediaPlayerState.OFF,
+ "Off": MediaPlayerState.OFF,
+}
+# Tesla uses 31 steps, in 0.333 increments up to 10.333
+VOLUME_STEP = 1 / 31
+VOLUME_FACTOR = 31 / 3 # 10.333
PARALLEL_UPDATES = 0
@@ -33,73 +47,104 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry Media platform from a config entry."""
async_add_entities(
- TeslemetryMediaEntity(vehicle, Scope.VEHICLE_CMDS in entry.runtime_data.scopes)
+ TeslemetryPollingMediaEntity(vehicle, entry.runtime_data.scopes)
+ if vehicle.api.pre2021 or vehicle.firmware < "2025.2.6"
+ else TeslemetryStreamingMediaEntity(vehicle, entry.runtime_data.scopes)
for vehicle in entry.runtime_data.vehicles
)
-class TeslemetryMediaEntity(TeslemetryVehicleEntity, MediaPlayerEntity):
- """Vehicle media player class."""
+class TeslemetryMediaEntity(TeslemetryRootEntity, MediaPlayerEntity):
+ """Base vehicle media player class."""
+
+ api: VehicleSpecific
_attr_device_class = MediaPlayerDeviceClass.SPEAKER
- _attr_supported_features = (
- MediaPlayerEntityFeature.NEXT_TRACK
- | MediaPlayerEntityFeature.PAUSE
- | MediaPlayerEntityFeature.PLAY
- | MediaPlayerEntityFeature.PREVIOUS_TRACK
- | MediaPlayerEntityFeature.VOLUME_SET
- )
- _volume_max: float = VOLUME_MAX
+ _attr_volume_step = VOLUME_STEP
+
+ async def async_set_volume_level(self, volume: float) -> None:
+ """Set volume level, range 0..1."""
+ self.raise_for_scope(Scope.VEHICLE_CMDS)
+
+ await handle_vehicle_command(self.api.adjust_volume(volume * VOLUME_FACTOR))
+ self._attr_volume_level = volume
+ self.async_write_ha_state()
+
+ async def async_media_play(self) -> None:
+ """Send play command."""
+ if self.state != MediaPlayerState.PLAYING:
+ self.raise_for_scope(Scope.VEHICLE_CMDS)
+
+ await handle_vehicle_command(self.api.media_toggle_playback())
+ self._attr_state = MediaPlayerState.PLAYING
+ self.async_write_ha_state()
+
+ async def async_media_pause(self) -> None:
+ """Send pause command."""
+
+ if self.state == MediaPlayerState.PLAYING:
+ self.raise_for_scope(Scope.VEHICLE_CMDS)
+
+ await handle_vehicle_command(self.api.media_toggle_playback())
+ self._attr_state = MediaPlayerState.PAUSED
+ self.async_write_ha_state()
+
+ async def async_media_next_track(self) -> None:
+ """Send next track command."""
+
+ self.raise_for_scope(Scope.VEHICLE_CMDS)
+ await handle_vehicle_command(self.api.media_next_track())
+
+ async def async_media_previous_track(self) -> None:
+ """Send previous track command."""
+
+ self.raise_for_scope(Scope.VEHICLE_CMDS)
+ await handle_vehicle_command(self.api.media_prev_track())
+
+
+class TeslemetryPollingMediaEntity(TeslemetryVehicleEntity, TeslemetryMediaEntity):
+ """Polling vehicle media player class."""
def __init__(
self,
data: TeslemetryVehicleData,
- scoped: bool,
+ scopes: list[Scope],
) -> None:
"""Initialize the media player entity."""
super().__init__(data, "media")
- self.scoped = scoped
- if not scoped:
+
+ self._attr_supported_features = (
+ MediaPlayerEntityFeature.NEXT_TRACK
+ | MediaPlayerEntityFeature.PAUSE
+ | MediaPlayerEntityFeature.PLAY
+ | MediaPlayerEntityFeature.PREVIOUS_TRACK
+ | MediaPlayerEntityFeature.VOLUME_SET
+ )
+ self.scoped = Scope.VEHICLE_CMDS in scopes
+ if not self.scoped:
self._attr_supported_features = MediaPlayerEntityFeature(0)
def _async_update_attrs(self) -> None:
"""Update entity attributes."""
- self._volume_max = (
- self.get("vehicle_state_media_info_audio_volume_max") or VOLUME_MAX
- )
- self._attr_state = STATES.get(
- self.get("vehicle_state_media_info_media_playback_status") or "Off",
- )
- self._attr_volume_step = (
- 1.0
- / self._volume_max
- / (
- self.get("vehicle_state_media_info_audio_volume_increment")
- or VOLUME_STEP
- )
- )
+ state = self.get("vehicle_state_media_info_media_playback_status")
+ self._attr_state = STATES.get(state) if state else None
+ self._attr_volume_level = (
+ self.get("vehicle_state_media_info_audio_volume") or 0
+ ) / VOLUME_FACTOR
- if volume := self.get("vehicle_state_media_info_audio_volume"):
- self._attr_volume_level = volume / self._volume_max
- else:
- self._attr_volume_level = None
+ duration = self.get("vehicle_state_media_info_now_playing_duration")
+ self._attr_media_duration = duration / 1000 if duration is not None else None
- if duration := self.get("vehicle_state_media_info_now_playing_duration"):
- self._attr_media_duration = duration / 1000
- else:
- self._attr_media_duration = None
-
- if duration and (
- position := self.get("vehicle_state_media_info_now_playing_elapsed")
- ):
- self._attr_media_position = position / 1000
- else:
- self._attr_media_position = None
+ # Return media position only when a media duration is > 0.
+ elapsed = self.get("vehicle_state_media_info_now_playing_elapsed")
+ self._attr_media_position = (
+ elapsed / 1000 if duration and elapsed is not None else None
+ )
self._attr_media_title = self.get("vehicle_state_media_info_now_playing_title")
self._attr_media_artist = self.get(
@@ -113,42 +158,151 @@ class TeslemetryMediaEntity(TeslemetryVehicleEntity, MediaPlayerEntity):
)
self._attr_source = self.get("vehicle_state_media_info_now_playing_source")
- async def async_set_volume_level(self, volume: float) -> None:
- """Set volume level, range 0..1."""
- self.raise_for_scope(Scope.VEHICLE_CMDS)
- await self.wake_up_if_asleep()
- await handle_vehicle_command(
- self.api.adjust_volume(int(volume * self._volume_max))
+
+class TeslemetryStreamingMediaEntity(
+ TeslemetryVehicleStreamEntity, TeslemetryMediaEntity, RestoreEntity
+):
+ """Streaming vehicle media player class."""
+
+ def __init__(
+ self,
+ data: TeslemetryVehicleData,
+ scopes: list[Scope],
+ ) -> None:
+ """Initialize the media player entity."""
+ super().__init__(data, "media")
+
+ self._attr_supported_features = (
+ MediaPlayerEntityFeature.NEXT_TRACK
+ | MediaPlayerEntityFeature.PAUSE
+ | MediaPlayerEntityFeature.PLAY
+ | MediaPlayerEntityFeature.PREVIOUS_TRACK
+ | MediaPlayerEntityFeature.VOLUME_SET
)
- self._attr_volume_level = volume
+ self.scoped = Scope.VEHICLE_CMDS in scopes
+ if not self.scoped:
+ self._attr_supported_features = MediaPlayerEntityFeature(0)
+
+ async def async_added_to_hass(self) -> None:
+ """Call when entity is added to hass."""
+
+ await super().async_added_to_hass()
+ if (state := await self.async_get_last_state()) is not None:
+ try:
+ self._attr_state = MediaPlayerState(state.state)
+ except ValueError:
+ self._attr_state = None
+ self._attr_volume_level = state.attributes.get("volume_level")
+ self._attr_media_title = state.attributes.get("media_title")
+ self._attr_media_artist = state.attributes.get("media_artist")
+ self._attr_media_album_name = state.attributes.get("media_album_name")
+ self._attr_media_playlist = state.attributes.get("media_playlist")
+ self._attr_media_duration = state.attributes.get("media_duration")
+ self._attr_media_position = state.attributes.get("media_position")
+ self._attr_source = state.attributes.get("source")
+
+ self.async_write_ha_state()
+
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_CenterDisplay(
+ self._async_handle_center_display
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaPlaybackStatus(
+ self._async_handle_media_playback_status
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaPlaybackSource(
+ self._async_handle_media_playback_source
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaAudioVolume(
+ self._async_handle_media_audio_volume
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaNowPlayingDuration(
+ self._async_handle_media_now_playing_duration
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaNowPlayingElapsed(
+ self._async_handle_media_now_playing_elapsed
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaNowPlayingArtist(
+ self._async_handle_media_now_playing_artist
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaNowPlayingAlbum(
+ self._async_handle_media_now_playing_album
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaNowPlayingTitle(
+ self._async_handle_media_now_playing_title
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_MediaNowPlayingStation(
+ self._async_handle_media_now_playing_station
+ )
+ )
+
+ def _async_handle_center_display(self, value: str | None) -> None:
+ """Update entity attributes."""
+ if value is not None:
+ self._attr_state = DISPLAY_STATES.get(value)
+ self.async_write_ha_state()
+
+ def _async_handle_media_playback_status(self, value: str | None) -> None:
+ """Update entity attributes."""
+ self._attr_state = MediaPlayerState.OFF if value is None else STATES.get(value)
self.async_write_ha_state()
- async def async_media_play(self) -> None:
- """Send play command."""
- if self.state != MediaPlayerState.PLAYING:
- self.raise_for_scope(Scope.VEHICLE_CMDS)
- await self.wake_up_if_asleep()
- await handle_vehicle_command(self.api.media_toggle_playback())
- self._attr_state = MediaPlayerState.PLAYING
- self.async_write_ha_state()
+ def _async_handle_media_playback_source(self, value: str | None) -> None:
+ """Update entity attributes."""
+ self._attr_source = value
+ self.async_write_ha_state()
- async def async_media_pause(self) -> None:
- """Send pause command."""
- if self.state == MediaPlayerState.PLAYING:
- self.raise_for_scope(Scope.VEHICLE_CMDS)
- await self.wake_up_if_asleep()
- await handle_vehicle_command(self.api.media_toggle_playback())
- self._attr_state = MediaPlayerState.PAUSED
- self.async_write_ha_state()
+ def _async_handle_media_audio_volume(self, value: float | None) -> None:
+ """Update entity attributes."""
+ self._attr_volume_level = None if value is None else value / VOLUME_FACTOR
+ self.async_write_ha_state()
- async def async_media_next_track(self) -> None:
- """Send next track command."""
- self.raise_for_scope(Scope.VEHICLE_CMDS)
- await self.wake_up_if_asleep()
- await handle_vehicle_command(self.api.media_next_track())
+ def _async_handle_media_now_playing_duration(self, value: int | None) -> None:
+ """Update entity attributes."""
+ self._attr_media_duration = None if value is None else int(value / 1000)
+ self.async_write_ha_state()
- async def async_media_previous_track(self) -> None:
- """Send previous track command."""
- self.raise_for_scope(Scope.VEHICLE_CMDS)
- await self.wake_up_if_asleep()
- await handle_vehicle_command(self.api.media_prev_track())
+ def _async_handle_media_now_playing_elapsed(self, value: int | None) -> None:
+ """Update entity attributes."""
+ self._attr_media_position = None if value is None else int(value / 1000)
+ self.async_write_ha_state()
+
+ def _async_handle_media_now_playing_artist(self, value: str | None) -> None:
+ """Update entity attributes."""
+ self._attr_media_artist = value # Check if this is album artist or not
+ self.async_write_ha_state()
+
+ def _async_handle_media_now_playing_album(self, value: str | None) -> None:
+ """Update entity attributes."""
+ self._attr_media_album_name = value
+ self.async_write_ha_state()
+
+ def _async_handle_media_now_playing_title(self, value: str | None) -> None:
+ """Update entity attributes."""
+ self._attr_media_title = value
+ self.async_write_ha_state()
+
+ def _async_handle_media_now_playing_station(self, value: str | None) -> None:
+ """Update entity attributes."""
+ self._attr_media_channel = (
+ value # could also be _attr_media_playlist when Spotify
+ )
+ self.async_write_ha_state()
diff --git a/homeassistant/components/teslemetry/number.py b/homeassistant/components/teslemetry/number.py
index c44028f2da7..10c15a68b09 100644
--- a/homeassistant/components/teslemetry/number.py
+++ b/homeassistant/components/teslemetry/number.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
UnitOfElectricCurrent,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level
from . import TeslemetryConfigEntry
@@ -133,7 +133,7 @@ ENERGY_INFO_DESCRIPTIONS: tuple[TeslemetryNumberBatteryEntityDescription, ...] =
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry number platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/select.py b/homeassistant/components/teslemetry/select.py
index d2e90a4f5c9..0d268e302de 100644
--- a/homeassistant/components/teslemetry/select.py
+++ b/homeassistant/components/teslemetry/select.py
@@ -13,7 +13,7 @@ from teslemetry_stream import TeslemetryStreamVehicle
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import TeslemetryConfigEntry
@@ -170,7 +170,7 @@ VEHICLE_DESCRIPTIONS: tuple[TeslemetrySelectEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry select platform from a config entry."""
diff --git a/homeassistant/components/teslemetry/sensor.py b/homeassistant/components/teslemetry/sensor.py
index dd83ad04ed6..b1c6b487bf9 100644
--- a/homeassistant/components/teslemetry/sensor.py
+++ b/homeassistant/components/teslemetry/sensor.py
@@ -7,7 +7,8 @@ from dataclasses import dataclass
from datetime import datetime, timedelta
from propcache.api import cached_property
-from teslemetry_stream import Signal
+from teslemetry_stream import Signal, TeslemetryStreamVehicle
+from teslemetry_stream.const import ShiftState
from homeassistant.components.sensor import (
RestoreSensor,
@@ -30,7 +31,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
from homeassistant.util.variance import ignore_variance
@@ -49,6 +50,7 @@ from .models import TeslemetryEnergyData, TeslemetryVehicleData
PARALLEL_UPDATES = 0
+
CHARGE_STATES = {
"Starting": "starting",
"Charging": "charging",
@@ -67,9 +69,9 @@ class TeslemetryVehicleSensorEntityDescription(SensorEntityDescription):
polling: bool = False
polling_value_fn: Callable[[StateType], StateType] = lambda x: x
- polling_available_fn: Callable[[StateType], bool] = lambda x: x is not None
+ nullable: bool = False
streaming_key: Signal | None = None
- streaming_value_fn: Callable[[StateType], StateType] = lambda x: x
+ streaming_value_fn: Callable[[str | int | float], StateType] = lambda x: x
streaming_firmware: str = "2024.26"
@@ -209,10 +211,10 @@ VEHICLE_DESCRIPTIONS: tuple[TeslemetryVehicleSensorEntityDescription, ...] = (
TeslemetryVehicleSensorEntityDescription(
key="drive_state_shift_state",
polling=True,
- polling_available_fn=lambda x: True,
+ nullable=True,
polling_value_fn=lambda x: SHIFT_STATES.get(str(x), "p"),
streaming_key=Signal.GEAR,
- streaming_value_fn=lambda x: SHIFT_STATES.get(str(x)),
+ streaming_value_fn=lambda x: str(ShiftState.get(x, "P")).lower(),
options=list(SHIFT_STATES.values()),
device_class=SensorDeviceClass.ENUM,
entity_registry_enabled_default=False,
@@ -349,21 +351,26 @@ class TeslemetryTimeEntityDescription(SensorEntityDescription):
"""Describes Teslemetry Sensor entity."""
variance: int
- streaming_key: Signal
+ streaming_listener: Callable[
+ [TeslemetryStreamVehicle, Callable[[float | None], None]],
+ Callable[[], None],
+ ]
streaming_firmware: str = "2024.26"
+ streaming_value_fn: Callable[[float], float] = lambda x: x
VEHICLE_TIME_DESCRIPTIONS: tuple[TeslemetryTimeEntityDescription, ...] = (
TeslemetryTimeEntityDescription(
key="charge_state_minutes_to_full_charge",
- streaming_key=Signal.TIME_TO_FULL_CHARGE,
+ streaming_value_fn=lambda x: x * 60,
+ streaming_listener=lambda x, y: x.listen_TimeToFullCharge(y),
device_class=SensorDeviceClass.TIMESTAMP,
entity_category=EntityCategory.DIAGNOSTIC,
variance=4,
),
TeslemetryTimeEntityDescription(
key="drive_state_active_route_minutes_to_arrival",
- streaming_key=Signal.MINUTES_TO_ARRIVAL,
+ streaming_listener=lambda x, y: x.listen_MinutesToArrival(y),
device_class=SensorDeviceClass.TIMESTAMP,
variance=1,
),
@@ -529,7 +536,7 @@ ENERGY_HISTORY_DESCRIPTIONS: tuple[SensorEntityDescription, ...] = tuple(
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry sensor platform from a config entry."""
@@ -621,10 +628,10 @@ class TeslemetryStreamSensorEntity(TeslemetryVehicleStreamEntity, RestoreSensor)
def _async_value_from_stream(self, value) -> None:
"""Update the value of the entity."""
- if value is None:
- self._attr_native_value = None
- else:
+ if self.entity_description.nullable or value is not None:
self._attr_native_value = self.entity_description.streaming_value_fn(value)
+ else:
+ self._attr_native_value = None
class TeslemetryVehicleSensorEntity(TeslemetryVehicleEntity, SensorEntity):
@@ -643,7 +650,7 @@ class TeslemetryVehicleSensorEntity(TeslemetryVehicleEntity, SensorEntity):
def _async_update_attrs(self) -> None:
"""Update the attributes of the sensor."""
- if self.entity_description.polling_available_fn(self._value):
+ if self.entity_description.nullable or self._value is not None:
self._attr_available = True
self._attr_native_value = self.entity_description.polling_value_fn(
self._value
@@ -666,18 +673,22 @@ class TeslemetryStreamTimeSensorEntity(TeslemetryVehicleStreamEntity, SensorEnti
"""Initialize the sensor."""
self.entity_description = description
self._get_timestamp = ignore_variance(
- func=lambda value: dt_util.now() + timedelta(minutes=value),
+ func=lambda value: dt_util.now()
+ + timedelta(minutes=description.streaming_value_fn(value)),
ignored_variance=timedelta(minutes=description.variance),
)
- assert description.streaming_key
- super().__init__(data, description.key, description.streaming_key)
+ super().__init__(data, description.key)
- @cached_property
- def available(self) -> bool:
- """Return True if entity is available."""
- return self.stream.connected
+ async def async_added_to_hass(self) -> None:
+ """When entity is added to hass."""
+ await super().async_added_to_hass()
+ self.async_on_remove(
+ self.entity_description.streaming_listener(
+ self.vehicle.stream_vehicle, self._value_callback
+ )
+ )
- def _async_value_from_stream(self, value) -> None:
+ def _value_callback(self, value: float | None) -> None:
"""Update the value of the entity."""
if value is None:
self._attr_native_value = None
diff --git a/homeassistant/components/teslemetry/strings.json b/homeassistant/components/teslemetry/strings.json
index 68ad12a46b6..c1df7d5aa57 100644
--- a/homeassistant/components/teslemetry/strings.json
+++ b/homeassistant/components/teslemetry/strings.json
@@ -132,7 +132,7 @@
"name": "Tire pressure warning rear right"
},
"pin_to_drive_enabled": {
- "name": "Pin to drive enabled"
+ "name": "PIN to Drive enabled"
},
"drive_rail": {
"name": "Drive rail"
@@ -415,7 +415,7 @@
"name": "Charging",
"state": {
"starting": "Starting",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"disconnected": "Disconnected",
"stopped": "Stopped",
"complete": "Complete",
@@ -712,7 +712,7 @@
"name": "Navigate to coordinates"
},
"set_scheduled_charging": {
- "description": "Sets a time at which charging should be completed.",
+ "description": "Sets a time at which charging should be started.",
"fields": {
"device_id": {
"description": "Vehicle to schedule.",
diff --git a/homeassistant/components/teslemetry/switch.py b/homeassistant/components/teslemetry/switch.py
index f810dee8554..516a6f9852f 100644
--- a/homeassistant/components/teslemetry/switch.py
+++ b/homeassistant/components/teslemetry/switch.py
@@ -7,7 +7,8 @@ from dataclasses import dataclass
from itertools import chain
from typing import Any
-from tesla_fleet_api.const import Scope, Seat
+from tesla_fleet_api.const import Scope
+from teslemetry_stream import TeslemetryStreamVehicle
from homeassistant.components.switch import (
SwitchDeviceClass,
@@ -15,11 +16,17 @@ from homeassistant.components.switch import (
SwitchEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import StateType
from . import TeslemetryConfigEntry
-from .entity import TeslemetryEnergyInfoEntity, TeslemetryVehicleEntity
+from .entity import (
+ TeslemetryEnergyInfoEntity,
+ TeslemetryRootEntity,
+ TeslemetryVehicleEntity,
+ TeslemetryVehicleStreamEntity,
+)
from .helpers import handle_command, handle_vehicle_command
from .models import TeslemetryEnergyData, TeslemetryVehicleData
@@ -34,36 +41,41 @@ class TeslemetrySwitchEntityDescription(SwitchEntityDescription):
off_func: Callable
scopes: list[Scope]
value_func: Callable[[StateType], bool] = bool
+ streaming_listener: Callable[
+ [TeslemetryStreamVehicle, Callable[[StateType], None]],
+ Callable[[], None],
+ ]
+ streaming_value_fn: Callable[[StateType], bool] = bool
+ streaming_firmware: str = "2024.26"
unique_id: str | None = None
VEHICLE_DESCRIPTIONS: tuple[TeslemetrySwitchEntityDescription, ...] = (
TeslemetrySwitchEntityDescription(
key="vehicle_state_sentry_mode",
+ streaming_listener=lambda x, y: x.listen_SentryMode(y),
+ streaming_value_fn=lambda x: x != "Off",
on_func=lambda api: api.set_sentry_mode(on=True),
off_func=lambda api: api.set_sentry_mode(on=False),
scopes=[Scope.VEHICLE_CMDS],
),
TeslemetrySwitchEntityDescription(
key="climate_state_auto_seat_climate_left",
- on_func=lambda api: api.remote_auto_seat_climate_request(Seat.FRONT_LEFT, True),
- off_func=lambda api: api.remote_auto_seat_climate_request(
- Seat.FRONT_LEFT, False
- ),
+ streaming_listener=lambda x, y: x.listen_AutoSeatClimateLeft(y),
+ on_func=lambda api: api.remote_auto_seat_climate_request(1, True),
+ off_func=lambda api: api.remote_auto_seat_climate_request(1, False),
scopes=[Scope.VEHICLE_CMDS],
),
TeslemetrySwitchEntityDescription(
key="climate_state_auto_seat_climate_right",
- on_func=lambda api: api.remote_auto_seat_climate_request(
- Seat.FRONT_RIGHT, True
- ),
- off_func=lambda api: api.remote_auto_seat_climate_request(
- Seat.FRONT_RIGHT, False
- ),
+ streaming_listener=lambda x, y: x.listen_AutoSeatClimateRight(y),
+ on_func=lambda api: api.remote_auto_seat_climate_request(2, True),
+ off_func=lambda api: api.remote_auto_seat_climate_request(2, False),
scopes=[Scope.VEHICLE_CMDS],
),
TeslemetrySwitchEntityDescription(
key="climate_state_auto_steering_wheel_heat",
+ streaming_listener=lambda x, y: x.listen_HvacSteeringWheelHeatAuto(y),
on_func=lambda api: api.remote_auto_steering_wheel_heat_climate_request(
on=True
),
@@ -74,6 +86,8 @@ VEHICLE_DESCRIPTIONS: tuple[TeslemetrySwitchEntityDescription, ...] = (
),
TeslemetrySwitchEntityDescription(
key="climate_state_defrost_mode",
+ streaming_listener=lambda x, y: x.listen_DefrostMode(y),
+ streaming_value_fn=lambda x: x != "Off",
on_func=lambda api: api.set_preconditioning_max(on=True, manual_override=False),
off_func=lambda api: api.set_preconditioning_max(
on=False, manual_override=False
@@ -83,9 +97,11 @@ VEHICLE_DESCRIPTIONS: tuple[TeslemetrySwitchEntityDescription, ...] = (
TeslemetrySwitchEntityDescription(
key="charge_state_charging_state",
unique_id="charge_state_user_charge_enable_request",
+ value_func=lambda state: state in {"Starting", "Charging"},
+ streaming_listener=lambda x, y: x.listen_DetailedChargeState(y),
+ streaming_value_fn=lambda x: x in {"Starting", "Charging"},
on_func=lambda api: api.charge_start(),
off_func=lambda api: api.charge_stop(),
- value_func=lambda state: state in {"Starting", "Charging"},
scopes=[Scope.VEHICLE_CMDS, Scope.VEHICLE_CHARGING_CMDS],
),
)
@@ -94,19 +110,23 @@ VEHICLE_DESCRIPTIONS: tuple[TeslemetrySwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry Switch platform from a config entry."""
async_add_entities(
chain(
(
- TeslemetryVehicleSwitchEntity(
+ TeslemetryPollingVehicleSwitchEntity(
+ vehicle, description, entry.runtime_data.scopes
+ )
+ if vehicle.api.pre2021
+ or vehicle.firmware < description.streaming_firmware
+ else TeslemetryStreamingVehicleSwitchEntity(
vehicle, description, entry.runtime_data.scopes
)
for vehicle in entry.runtime_data.vehicles
for description in VEHICLE_DESCRIPTIONS
- if description.key in vehicle.coordinator.data
),
(
TeslemetryChargeFromGridSwitchEntity(
@@ -126,15 +146,31 @@ async def async_setup_entry(
)
-class TeslemetrySwitchEntity(SwitchEntity):
+class TeslemetryVehicleSwitchEntity(TeslemetryRootEntity, SwitchEntity):
"""Base class for all Teslemetry switch entities."""
_attr_device_class = SwitchDeviceClass.SWITCH
entity_description: TeslemetrySwitchEntityDescription
+ async def async_turn_on(self, **kwargs: Any) -> None:
+ """Turn on the Switch."""
+ self.raise_for_scope(self.entity_description.scopes[0])
+ await handle_vehicle_command(self.entity_description.on_func(self.api))
+ self._attr_is_on = True
+ self.async_write_ha_state()
-class TeslemetryVehicleSwitchEntity(TeslemetryVehicleEntity, TeslemetrySwitchEntity):
- """Base class for Teslemetry vehicle switch entities."""
+ async def async_turn_off(self, **kwargs: Any) -> None:
+ """Turn off the Switch."""
+ self.raise_for_scope(self.entity_description.scopes[0])
+ await handle_vehicle_command(self.entity_description.off_func(self.api))
+ self._attr_is_on = False
+ self.async_write_ha_state()
+
+
+class TeslemetryPollingVehicleSwitchEntity(
+ TeslemetryVehicleEntity, TeslemetryVehicleSwitchEntity
+):
+ """Base class for Teslemetry polling vehicle switch entities."""
def __init__(
self,
@@ -151,30 +187,63 @@ class TeslemetryVehicleSwitchEntity(TeslemetryVehicleEntity, TeslemetrySwitchEnt
def _async_update_attrs(self) -> None:
"""Update the attributes of the sensor."""
- self._attr_is_on = self.entity_description.value_func(self._value)
-
- async def async_turn_on(self, **kwargs: Any) -> None:
- """Turn on the Switch."""
- self.raise_for_scope(self.entity_description.scopes[0])
- await self.wake_up_if_asleep()
- await handle_vehicle_command(self.entity_description.on_func(self.api))
- self._attr_is_on = True
- self.async_write_ha_state()
-
- async def async_turn_off(self, **kwargs: Any) -> None:
- """Turn off the Switch."""
- self.raise_for_scope(self.entity_description.scopes[0])
- await self.wake_up_if_asleep()
- await handle_vehicle_command(self.entity_description.off_func(self.api))
- self._attr_is_on = False
- self.async_write_ha_state()
+ self._attr_is_on = (
+ None
+ if self._value is None
+ else self.entity_description.value_func(self._value)
+ )
-class TeslemetryChargeFromGridSwitchEntity(
- TeslemetryEnergyInfoEntity, TeslemetrySwitchEntity
+class TeslemetryStreamingVehicleSwitchEntity(
+ TeslemetryVehicleStreamEntity, TeslemetryVehicleSwitchEntity, RestoreEntity
):
+ """Base class for Teslemetry streaming vehicle switch entities."""
+
+ def __init__(
+ self,
+ data: TeslemetryVehicleData,
+ description: TeslemetrySwitchEntityDescription,
+ scopes: list[Scope],
+ ) -> None:
+ """Initialize the Switch."""
+
+ self.entity_description = description
+ self.scoped = any(scope in scopes for scope in description.scopes)
+ super().__init__(data, description.key)
+ if description.unique_id:
+ self._attr_unique_id = f"{data.vin}-{description.unique_id}"
+
+ async def async_added_to_hass(self) -> None:
+ """Handle entity which will be added."""
+ await super().async_added_to_hass()
+
+ # Restore previous state
+ if (state := await self.async_get_last_state()) is not None:
+ if state.state == "on":
+ self._attr_is_on = True
+ elif state.state == "off":
+ self._attr_is_on = False
+
+ # Add listener
+ self.async_on_remove(
+ self.entity_description.streaming_listener(
+ self.vehicle.stream_vehicle, self._value_callback
+ )
+ )
+
+ def _value_callback(self, value: StateType) -> None:
+ """Update the value of the entity."""
+ self._attr_is_on = (
+ None if value is None else self.entity_description.streaming_value_fn(value)
+ )
+ self.async_write_ha_state()
+
+
+class TeslemetryChargeFromGridSwitchEntity(TeslemetryEnergyInfoEntity, SwitchEntity):
"""Entity class for Charge From Grid switch."""
+ _attr_device_class = SwitchDeviceClass.SWITCH
+
def __init__(
self,
data: TeslemetryEnergyData,
@@ -215,11 +284,11 @@ class TeslemetryChargeFromGridSwitchEntity(
self.async_write_ha_state()
-class TeslemetryStormModeSwitchEntity(
- TeslemetryEnergyInfoEntity, TeslemetrySwitchEntity
-):
+class TeslemetryStormModeSwitchEntity(TeslemetryEnergyInfoEntity, SwitchEntity):
"""Entity class for Storm Mode switch."""
+ _attr_device_class = SwitchDeviceClass.SWITCH
+
def __init__(
self,
data: TeslemetryEnergyData,
diff --git a/homeassistant/components/teslemetry/update.py b/homeassistant/components/teslemetry/update.py
index 670cd0e0eda..0b0255508e0 100644
--- a/homeassistant/components/teslemetry/update.py
+++ b/homeassistant/components/teslemetry/update.py
@@ -2,16 +2,22 @@
from __future__ import annotations
-from typing import Any, cast
+from typing import Any
from tesla_fleet_api.const import Scope
+from tesla_fleet_api.vehiclespecific import VehicleSpecific
from homeassistant.components.update import UpdateEntity, UpdateEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.helpers.restore_state import RestoreEntity
from . import TeslemetryConfigEntry
-from .entity import TeslemetryVehicleEntity
+from .entity import (
+ TeslemetryRootEntity,
+ TeslemetryVehicleEntity,
+ TeslemetryVehicleStreamEntity,
+)
from .helpers import handle_vehicle_command
from .models import TeslemetryVehicleData
@@ -27,17 +33,36 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TeslemetryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Teslemetry update platform from a config entry."""
async_add_entities(
- TeslemetryUpdateEntity(vehicle, entry.runtime_data.scopes)
+ TeslemetryPollingUpdateEntity(vehicle, entry.runtime_data.scopes)
+ if vehicle.api.pre2021 or vehicle.firmware < "2024.44.25"
+ else TeslemetryStreamingUpdateEntity(vehicle, entry.runtime_data.scopes)
for vehicle in entry.runtime_data.vehicles
)
-class TeslemetryUpdateEntity(TeslemetryVehicleEntity, UpdateEntity):
+class TeslemetryUpdateEntity(TeslemetryRootEntity, UpdateEntity):
+ """Teslemetry Updates entity."""
+
+ api: VehicleSpecific
+ _attr_supported_features = UpdateEntityFeature.PROGRESS
+
+ async def async_install(
+ self, version: str | None, backup: bool, **kwargs: Any
+ ) -> None:
+ """Install an update."""
+ self.raise_for_scope(Scope.VEHICLE_CMDS)
+
+ await handle_vehicle_command(self.api.schedule_software_update(offset_sec=0))
+ self._attr_in_progress = True
+ self.async_write_ha_state()
+
+
+class TeslemetryPollingUpdateEntity(TeslemetryVehicleEntity, TeslemetryUpdateEntity):
"""Teslemetry Updates entity."""
def __init__(
@@ -94,18 +119,125 @@ class TeslemetryUpdateEntity(TeslemetryVehicleEntity, UpdateEntity):
):
self._attr_in_progress = True
if install_perc := self.get("vehicle_state_software_update_install_perc"):
- self._attr_update_percentage = cast(int, install_perc)
+ self._attr_update_percentage = install_perc
else:
self._attr_in_progress = False
self._attr_update_percentage = None
- async def async_install(
- self, version: str | None, backup: bool, **kwargs: Any
+
+class TeslemetryStreamingUpdateEntity(
+ TeslemetryVehicleStreamEntity, TeslemetryUpdateEntity, RestoreEntity
+):
+ """Teslemetry Updates entity."""
+
+ _download_percentage: int = 0
+ _install_percentage: int = 0
+
+ def __init__(
+ self,
+ data: TeslemetryVehicleData,
+ scopes: list[Scope],
) -> None:
- """Install an update."""
- self.raise_for_scope(Scope.ENERGY_CMDS)
- await self.wake_up_if_asleep()
- await handle_vehicle_command(self.api.schedule_software_update(offset_sec=60))
- self._attr_in_progress = True
- self._attr_update_percentage = None
+ """Initialize the Update."""
+ self.scoped = Scope.VEHICLE_CMDS in scopes
+ super().__init__(
+ data,
+ "vehicle_state_software_update_status",
+ )
+
+ async def async_added_to_hass(self) -> None:
+ """Handle entity which will be added."""
+ await super().async_added_to_hass()
+ if (state := await self.async_get_last_state()) is not None:
+ self._attr_in_progress = state.attributes.get("in_progress", False)
+ self._install_percentage = state.attributes.get("install_percentage", False)
+ self._attr_installed_version = state.attributes.get("installed_version")
+ self._attr_latest_version = state.attributes.get("latest_version")
+ self._attr_supported_features = UpdateEntityFeature(
+ state.attributes.get(
+ "supported_features", self._attr_supported_features
+ )
+ )
+ self.async_write_ha_state()
+
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_SoftwareUpdateDownloadPercentComplete(
+ self._async_handle_software_update_download_percent_complete
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_SoftwareUpdateInstallationPercentComplete(
+ self._async_handle_software_update_installation_percent_complete
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_SoftwareUpdateScheduledStartTime(
+ self._async_handle_software_update_scheduled_start_time
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_SoftwareUpdateVersion(
+ self._async_handle_software_update_version
+ )
+ )
+ self.async_on_remove(
+ self.vehicle.stream_vehicle.listen_Version(self._async_handle_version)
+ )
+
+ def _async_handle_software_update_download_percent_complete(
+ self, value: float | None
+ ):
+ """Handle software update download percent complete."""
+
+ self._download_percentage = round(value) if value is not None else 0
+ if self.scoped and self._download_percentage == 100:
+ self._attr_supported_features = (
+ UpdateEntityFeature.PROGRESS | UpdateEntityFeature.INSTALL
+ )
+ else:
+ self._attr_supported_features = UpdateEntityFeature.PROGRESS
+ self._async_update_progress()
self.async_write_ha_state()
+
+ def _async_handle_software_update_installation_percent_complete(
+ self, value: float | None
+ ):
+ """Handle software update installation percent complete."""
+
+ self._install_percentage = round(value) if value is not None else 0
+ self._async_update_progress()
+ self.async_write_ha_state()
+
+ def _async_handle_software_update_scheduled_start_time(self, value: str | None):
+ """Handle software update scheduled start time."""
+
+ self._attr_in_progress = value is not None
+ self.async_write_ha_state()
+
+ def _async_handle_software_update_version(self, value: str | None):
+ """Handle software update version."""
+
+ self._attr_latest_version = (
+ value if value and value != " " else self._attr_installed_version
+ )
+ self.async_write_ha_state()
+
+ def _async_handle_version(self, value: str | None):
+ """Handle version."""
+
+ if value is not None:
+ self._attr_installed_version = value.split(" ")[0]
+ self.async_write_ha_state()
+
+ def _async_update_progress(self) -> None:
+ """Update the progress of the update."""
+
+ if self._download_percentage > 1 and self._download_percentage < 100:
+ self._attr_in_progress = True
+ self._attr_update_percentage = self._download_percentage
+ elif self._install_percentage > 1:
+ self._attr_in_progress = True
+ self._attr_update_percentage = self._install_percentage
+ else:
+ self._attr_in_progress = False
+ self._attr_update_percentage = None
diff --git a/homeassistant/components/tessie/__init__.py b/homeassistant/components/tessie/__init__.py
index a0bc58896e4..f73ecc7a729 100644
--- a/homeassistant/components/tessie/__init__.py
+++ b/homeassistant/components/tessie/__init__.py
@@ -69,6 +69,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TessieConfigEntry) -> bo
vin=vehicle["vin"],
data_coordinator=TessieStateUpdateCoordinator(
hass,
+ entry,
api_key=api_key,
vin=vehicle["vin"],
data=vehicle["last_state"],
@@ -127,8 +128,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: TessieConfigEntry) -> bo
TessieEnergyData(
api=api,
id=site_id,
- live_coordinator=TessieEnergySiteLiveCoordinator(hass, api),
- info_coordinator=TessieEnergySiteInfoCoordinator(hass, api),
+ live_coordinator=TessieEnergySiteLiveCoordinator(
+ hass, entry, api
+ ),
+ info_coordinator=TessieEnergySiteInfoCoordinator(
+ hass, entry, api
+ ),
device=DeviceInfo(
identifiers={(DOMAIN, str(site_id))},
manufacturer="Tesla",
diff --git a/homeassistant/components/tessie/binary_sensor.py b/homeassistant/components/tessie/binary_sensor.py
index fd6565b62b7..515339c3da8 100644
--- a/homeassistant/components/tessie/binary_sensor.py
+++ b/homeassistant/components/tessie/binary_sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .const import TessieState
@@ -177,7 +177,7 @@ ENERGY_INFO_DESCRIPTIONS: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie binary sensor platform from a config entry."""
async_add_entities(
diff --git a/homeassistant/components/tessie/button.py b/homeassistant/components/tessie/button.py
index bef9c2585f6..a370f504323 100644
--- a/homeassistant/components/tessie/button.py
+++ b/homeassistant/components/tessie/button.py
@@ -16,7 +16,7 @@ from tessie_api import (
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .entity import TessieEntity
@@ -50,7 +50,7 @@ DESCRIPTIONS: tuple[TessieButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie Button platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/tessie/climate.py b/homeassistant/components/tessie/climate.py
index 1d26926aeaa..a8aa18132ee 100644
--- a/homeassistant/components/tessie/climate.py
+++ b/homeassistant/components/tessie/climate.py
@@ -19,7 +19,7 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_HALVES, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .const import TessieClimateKeeper
@@ -32,7 +32,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie Climate platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/tessie/coordinator.py b/homeassistant/components/tessie/coordinator.py
index 4582260bfb2..b06fe6123a5 100644
--- a/homeassistant/components/tessie/coordinator.py
+++ b/homeassistant/components/tessie/coordinator.py
@@ -1,9 +1,11 @@
"""Tessie Data Coordinator."""
+from __future__ import annotations
+
from datetime import timedelta
from http import HTTPStatus
import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
from aiohttp import ClientResponseError
from tesla_fleet_api import EnergySpecific
@@ -15,6 +17,9 @@ from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import TessieConfigEntry
+
from .const import TessieStatus
# This matches the update interval Tessie performs server side
@@ -40,9 +45,12 @@ def flatten(data: dict[str, Any], parent: str | None = None) -> dict[str, Any]:
class TessieStateUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching data from the Tessie API."""
+ config_entry: TessieConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: TessieConfigEntry,
api_key: str,
vin: str,
data: dict[str, Any],
@@ -51,6 +59,7 @@ class TessieStateUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Tessie",
update_interval=timedelta(seconds=TESSIE_SYNC_INTERVAL),
)
@@ -90,11 +99,16 @@ class TessieStateUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class TessieEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site live status from the Tessie API."""
- def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
+ config_entry: TessieConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: TessieConfigEntry, api: EnergySpecific
+ ) -> None:
"""Initialize Tessie Energy Site Live coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Tessie Energy Site Live",
update_interval=TESSIE_FLEET_API_SYNC_INTERVAL,
)
@@ -121,11 +135,16 @@ class TessieEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class TessieEnergySiteInfoCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site info from the Tessie API."""
- def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
+ config_entry: TessieConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: TessieConfigEntry, api: EnergySpecific
+ ) -> None:
"""Initialize Tessie Energy Info coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="Tessie Energy Site Info",
update_interval=TESSIE_FLEET_API_SYNC_INTERVAL,
)
diff --git a/homeassistant/components/tessie/cover.py b/homeassistant/components/tessie/cover.py
index e739f8c074d..bfd7b1b816c 100644
--- a/homeassistant/components/tessie/cover.py
+++ b/homeassistant/components/tessie/cover.py
@@ -22,7 +22,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .const import TessieCoverStates
@@ -35,7 +35,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie sensor platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/tessie/device_tracker.py b/homeassistant/components/tessie/device_tracker.py
index df74cd2a7a7..fe81ed67337 100644
--- a/homeassistant/components/tessie/device_tracker.py
+++ b/homeassistant/components/tessie/device_tracker.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from homeassistant.components.device_tracker.config_entry import TrackerEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TessieConfigEntry
@@ -17,7 +17,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie device tracker platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/tessie/lock.py b/homeassistant/components/tessie/lock.py
index 76d58a9070c..66cb813b995 100644
--- a/homeassistant/components/tessie/lock.py
+++ b/homeassistant/components/tessie/lock.py
@@ -9,7 +9,7 @@ from tessie_api import lock, open_unlock_charge_port, unlock
from homeassistant.components.lock import LockEntity
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .const import DOMAIN, TessieChargeCableLockStates
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie sensor platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/tessie/manifest.json b/homeassistant/components/tessie/manifest.json
index ef4d366c779..4ddd63552f0 100644
--- a/homeassistant/components/tessie/manifest.json
+++ b/homeassistant/components/tessie/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/tessie",
"iot_class": "cloud_polling",
"loggers": ["tessie", "tesla-fleet-api"],
- "requirements": ["tessie-api==0.1.1", "tesla-fleet-api==0.9.8"]
+ "requirements": ["tessie-api==0.1.1", "tesla-fleet-api==0.9.13"]
}
diff --git a/homeassistant/components/tessie/media_player.py b/homeassistant/components/tessie/media_player.py
index 7dfe568926b..139ee07ca5b 100644
--- a/homeassistant/components/tessie/media_player.py
+++ b/homeassistant/components/tessie/media_player.py
@@ -8,7 +8,7 @@ from homeassistant.components.media_player import (
MediaPlayerState,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .entity import TessieEntity
@@ -26,7 +26,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie Media platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/tessie/number.py b/homeassistant/components/tessie/number.py
index 74249d392a7..1e857345278 100644
--- a/homeassistant/components/tessie/number.py
+++ b/homeassistant/components/tessie/number.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfSpeed,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level
from . import TessieConfigEntry
@@ -111,7 +111,7 @@ ENERGY_INFO_DESCRIPTIONS: tuple[TessieNumberBatteryEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie sensor platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/tessie/select.py b/homeassistant/components/tessie/select.py
index 4dfe7088439..471372a68bd 100644
--- a/homeassistant/components/tessie/select.py
+++ b/homeassistant/components/tessie/select.py
@@ -9,7 +9,7 @@ from tessie_api import set_seat_cool, set_seat_heat
from homeassistant.components.select import SelectEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .const import TessieSeatCoolerOptions, TessieSeatHeaterOptions
@@ -38,7 +38,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie select platform from a config entry."""
diff --git a/homeassistant/components/tessie/sensor.py b/homeassistant/components/tessie/sensor.py
index 323fa76ef1f..e5b476057fa 100644
--- a/homeassistant/components/tessie/sensor.py
+++ b/homeassistant/components/tessie/sensor.py
@@ -28,7 +28,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
from homeassistant.util.variance import ignore_variance
@@ -148,7 +148,7 @@ DESCRIPTIONS: tuple[TessieSensorEntityDescription, ...] = (
key="drive_state_shift_state",
options=["p", "d", "r", "n"],
device_class=SensorDeviceClass.ENUM,
- value_fn=lambda x: x.lower() if isinstance(x, str) else x,
+ value_fn=lambda x: x.lower() if isinstance(x, str) else "p",
),
TessieSensorEntityDescription(
key="vehicle_state_odometer",
@@ -375,7 +375,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie sensor platform from a config entry."""
@@ -397,6 +397,7 @@ async def async_setup_entry(
for energysite in entry.runtime_data.energysites
for description in ENERGY_LIVE_DESCRIPTIONS
if description.key in energysite.live_coordinator.data
+ or description.key == "percentage_charged"
),
( # Add wall connectors
TessieWallConnectorSensorEntity(energysite, din, description)
@@ -449,7 +450,6 @@ class TessieEnergyLiveSensorEntity(TessieEnergyEntity, SensorEntity):
def _async_update_attrs(self) -> None:
"""Update the attributes of the sensor."""
- self._attr_available = self._value is not None
self._attr_native_value = self.entity_description.value_fn(self._value)
diff --git a/homeassistant/components/tessie/strings.json b/homeassistant/components/tessie/strings.json
index 8384bb3d8fb..4f0f5f67ebd 100644
--- a/homeassistant/components/tessie/strings.json
+++ b/homeassistant/components/tessie/strings.json
@@ -75,7 +75,7 @@
"name": "Charging",
"state": {
"starting": "Starting",
- "charging": "Charging",
+ "charging": "[%key:common::state::charging%]",
"disconnected": "Disconnected",
"stopped": "Stopped",
"complete": "Complete",
@@ -212,7 +212,7 @@
"name": "State",
"state": {
"booting": "Booting",
- "charging": "[%key:component::tessie::entity::sensor::charge_state_charging_state::state::charging%]",
+ "charging": "[%key:common::state::charging%]",
"disconnected": "[%key:common::state::disconnected%]",
"connected": "[%key:common::state::connected%]",
"scheduled": "Scheduled",
@@ -506,7 +506,7 @@
},
"exceptions": {
"unknown": {
- "message": "An unknown issue occured changing {name}."
+ "message": "An unknown issue occurred changing {name}."
},
"not_supported": {
"message": "{name} is not supported."
diff --git a/homeassistant/components/tessie/switch.py b/homeassistant/components/tessie/switch.py
index dba00a85bb2..41134b38fda 100644
--- a/homeassistant/components/tessie/switch.py
+++ b/homeassistant/components/tessie/switch.py
@@ -26,7 +26,7 @@ from homeassistant.components.switch import (
SwitchEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TessieConfigEntry
@@ -81,7 +81,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie Switch platform from a config entry."""
diff --git a/homeassistant/components/tessie/update.py b/homeassistant/components/tessie/update.py
index f6198fa6c03..e9af673b1f4 100644
--- a/homeassistant/components/tessie/update.py
+++ b/homeassistant/components/tessie/update.py
@@ -8,7 +8,7 @@ from tessie_api import schedule_software_update
from homeassistant.components.update import UpdateEntity, UpdateEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TessieConfigEntry
from .const import TessieUpdateStatus
@@ -21,7 +21,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TessieConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tessie Update platform from a config entry."""
data = entry.runtime_data
diff --git a/homeassistant/components/thermobeacon/config_flow.py b/homeassistant/components/thermobeacon/config_flow.py
index 08994a41008..6fa502716ca 100644
--- a/homeassistant/components/thermobeacon/config_flow.py
+++ b/homeassistant/components/thermobeacon/config_flow.py
@@ -72,7 +72,7 @@ class ThermoBeaconConfigFlow(ConfigFlow, domain=DOMAIN):
title=self._discovered_devices[address], data={}
)
- current_addresses = self._async_current_ids()
+ current_addresses = self._async_current_ids(include_ignore=False)
for discovery_info in async_discovered_service_info(self.hass, False):
address = discovery_info.address
if address in current_addresses or address in self._discovered_devices:
diff --git a/homeassistant/components/thermobeacon/manifest.json b/homeassistant/components/thermobeacon/manifest.json
index ce6a3f71ef3..b231137d335 100644
--- a/homeassistant/components/thermobeacon/manifest.json
+++ b/homeassistant/components/thermobeacon/manifest.json
@@ -14,6 +14,12 @@
"manufacturer_data_start": [0],
"connectable": false
},
+ {
+ "service_uuid": "0000fff0-0000-1000-8000-00805f9b34fb",
+ "manufacturer_id": 20,
+ "manufacturer_data_start": [0],
+ "connectable": false
+ },
{
"service_uuid": "0000fff0-0000-1000-8000-00805f9b34fb",
"manufacturer_id": 21,
@@ -48,5 +54,5 @@
"dependencies": ["bluetooth_adapters"],
"documentation": "https://www.home-assistant.io/integrations/thermobeacon",
"iot_class": "local_push",
- "requirements": ["thermobeacon-ble==0.7.0"]
+ "requirements": ["thermobeacon-ble==0.8.1"]
}
diff --git a/homeassistant/components/thermobeacon/sensor.py b/homeassistant/components/thermobeacon/sensor.py
index 53e86f37f11..916ec91359a 100644
--- a/homeassistant/components/thermobeacon/sensor.py
+++ b/homeassistant/components/thermobeacon/sensor.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -113,7 +113,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ThermoBeacon BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/thermopro/__init__.py b/homeassistant/components/thermopro/__init__.py
index 2cd207818c5..742449cffbe 100644
--- a/homeassistant/components/thermopro/__init__.py
+++ b/homeassistant/components/thermopro/__init__.py
@@ -2,25 +2,47 @@
from __future__ import annotations
+from functools import partial
import logging
-from thermopro_ble import ThermoProBluetoothDeviceData
+from thermopro_ble import SensorUpdate, ThermoProBluetoothDeviceData
-from homeassistant.components.bluetooth import BluetoothScanningMode
+from homeassistant.components.bluetooth import (
+ BluetoothScanningMode,
+ BluetoothServiceInfoBleak,
+)
from homeassistant.components.bluetooth.passive_update_processor import (
PassiveBluetoothProcessorCoordinator,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
+from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.dispatcher import async_dispatcher_send
-from .const import DOMAIN
+from .const import DOMAIN, SIGNAL_DATA_UPDATED
-PLATFORMS: list[Platform] = [Platform.SENSOR]
+CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
+
+PLATFORMS: list[Platform] = [Platform.BUTTON, Platform.SENSOR]
_LOGGER = logging.getLogger(__name__)
+def process_service_info(
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ data: ThermoProBluetoothDeviceData,
+ service_info: BluetoothServiceInfoBleak,
+) -> SensorUpdate:
+ """Process a BluetoothServiceInfoBleak, running side effects and returning sensor data."""
+ update = data.update(service_info)
+ async_dispatcher_send(
+ hass, f"{SIGNAL_DATA_UPDATED}_{entry.entry_id}", data, service_info, update
+ )
+ return update
+
+
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up ThermoPro BLE device from a config entry."""
address = entry.unique_id
@@ -32,13 +54,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
_LOGGER,
address=address,
mode=BluetoothScanningMode.ACTIVE,
- update_method=data.update,
+ update_method=partial(process_service_info, hass, entry, data),
)
)
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
- entry.async_on_unload(
- coordinator.async_start()
- ) # only start after all platforms have had a chance to subscribe
+ # only start after all platforms have had a chance to subscribe
+ entry.async_on_unload(coordinator.async_start())
return True
diff --git a/homeassistant/components/thermopro/button.py b/homeassistant/components/thermopro/button.py
new file mode 100644
index 00000000000..9faa9f22c4c
--- /dev/null
+++ b/homeassistant/components/thermopro/button.py
@@ -0,0 +1,157 @@
+"""Thermopro button platform."""
+
+from __future__ import annotations
+
+from collections.abc import Callable, Coroutine
+from dataclasses import dataclass
+from typing import Any
+
+from thermopro_ble import SensorUpdate, ThermoProBluetoothDeviceData, ThermoProDevice
+
+from homeassistant.components.bluetooth import (
+ BluetoothServiceInfoBleak,
+ async_ble_device_from_address,
+ async_track_unavailable,
+)
+from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.dispatcher import (
+ async_dispatcher_connect,
+ async_dispatcher_send,
+)
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+from homeassistant.util.dt import now
+
+from .const import DOMAIN, SIGNAL_AVAILABILITY_UPDATED, SIGNAL_DATA_UPDATED
+
+PARALLEL_UPDATES = 1 # one connection at a time
+
+
+@dataclass(kw_only=True, frozen=True)
+class ThermoProButtonEntityDescription(ButtonEntityDescription):
+ """Describe a ThermoPro button entity."""
+
+ press_action_fn: Callable[[HomeAssistant, str], Coroutine[None, Any, Any]]
+
+
+async def _async_set_datetime(hass: HomeAssistant, address: str) -> None:
+ """Set Date&Time for a given device."""
+ ble_device = async_ble_device_from_address(hass, address, connectable=True)
+ assert ble_device is not None
+ await ThermoProDevice(ble_device).set_datetime(now(), am_pm=False)
+
+
+BUTTON_ENTITIES: tuple[ThermoProButtonEntityDescription, ...] = (
+ ThermoProButtonEntityDescription(
+ key="datetime",
+ translation_key="set_datetime",
+ icon="mdi:calendar-clock",
+ entity_category=EntityCategory.CONFIG,
+ press_action_fn=_async_set_datetime,
+ ),
+)
+
+MODELS_THAT_SUPPORT_BUTTONS = {"TP358", "TP393"}
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up the thermopro button platform."""
+ address = entry.unique_id
+ assert address is not None
+ availability_signal = f"{SIGNAL_AVAILABILITY_UPDATED}_{entry.entry_id}"
+ entity_added = False
+
+ @callback
+ def _async_on_data_updated(
+ data: ThermoProBluetoothDeviceData,
+ service_info: BluetoothServiceInfoBleak,
+ update: SensorUpdate,
+ ) -> None:
+ nonlocal entity_added
+ sensor_device_info = update.devices[data.primary_device_id]
+ if sensor_device_info.model not in MODELS_THAT_SUPPORT_BUTTONS:
+ return
+
+ if not entity_added:
+ name = sensor_device_info.name
+ assert name is not None
+ entity_added = True
+ async_add_entities(
+ ThermoProButtonEntity(
+ description=description,
+ data=data,
+ availability_signal=availability_signal,
+ address=address,
+ )
+ for description in BUTTON_ENTITIES
+ )
+
+ if service_info.connectable:
+ async_dispatcher_send(hass, availability_signal, True)
+
+ entry.async_on_unload(
+ async_dispatcher_connect(
+ hass, f"{SIGNAL_DATA_UPDATED}_{entry.entry_id}", _async_on_data_updated
+ )
+ )
+
+
+class ThermoProButtonEntity(ButtonEntity):
+ """Representation of a ThermoPro button entity."""
+
+ _attr_has_entity_name = True
+ entity_description: ThermoProButtonEntityDescription
+
+ def __init__(
+ self,
+ description: ThermoProButtonEntityDescription,
+ data: ThermoProBluetoothDeviceData,
+ availability_signal: str,
+ address: str,
+ ) -> None:
+ """Initialize the thermopro button entity."""
+ self.entity_description = description
+ self._address = address
+ self._availability_signal = availability_signal
+ self._attr_unique_id = f"{address}-{description.key}"
+ self._attr_device_info = dr.DeviceInfo(
+ name=data.get_device_name(),
+ identifiers={(DOMAIN, address)},
+ connections={(dr.CONNECTION_BLUETOOTH, address)},
+ )
+
+ async def async_added_to_hass(self) -> None:
+ """Connect availability dispatcher."""
+ await super().async_added_to_hass()
+ self.async_on_remove(
+ async_dispatcher_connect(
+ self.hass,
+ self._availability_signal,
+ self._async_on_availability_changed,
+ )
+ )
+ self.async_on_remove(
+ async_track_unavailable(
+ self.hass, self._async_on_unavailable, self._address, connectable=True
+ )
+ )
+
+ @callback
+ def _async_on_unavailable(self, _: BluetoothServiceInfoBleak) -> None:
+ self._async_on_availability_changed(False)
+
+ @callback
+ def _async_on_availability_changed(self, available: bool) -> None:
+ self._attr_available = available
+ self.async_write_ha_state()
+
+ async def async_press(self) -> None:
+ """Execute the press action for the entity."""
+ await self.entity_description.press_action_fn(self.hass, self._address)
diff --git a/homeassistant/components/thermopro/const.py b/homeassistant/components/thermopro/const.py
index 343729442cf..7d2170f8cf9 100644
--- a/homeassistant/components/thermopro/const.py
+++ b/homeassistant/components/thermopro/const.py
@@ -1,3 +1,6 @@
"""Constants for the ThermoPro Bluetooth integration."""
DOMAIN = "thermopro"
+
+SIGNAL_DATA_UPDATED = f"{DOMAIN}_service_info_updated"
+SIGNAL_AVAILABILITY_UPDATED = f"{DOMAIN}_availability_updated"
diff --git a/homeassistant/components/thermopro/sensor.py b/homeassistant/components/thermopro/sensor.py
index 4aca6101685..853f00f2dd5 100644
--- a/homeassistant/components/thermopro/sensor.py
+++ b/homeassistant/components/thermopro/sensor.py
@@ -9,7 +9,6 @@ from thermopro_ble import (
Units,
)
-from homeassistant import config_entries
from homeassistant.components.bluetooth.passive_update_processor import (
PassiveBluetoothDataProcessor,
PassiveBluetoothDataUpdate,
@@ -23,6 +22,7 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
PERCENTAGE,
SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
@@ -30,7 +30,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -110,8 +110,8 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
- entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ThermoPro BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/thermopro/strings.json b/homeassistant/components/thermopro/strings.json
index 4e12a84b653..5789de410b2 100644
--- a/homeassistant/components/thermopro/strings.json
+++ b/homeassistant/components/thermopro/strings.json
@@ -17,5 +17,12 @@
"already_in_progress": "[%key:common::config_flow::abort::already_in_progress%]",
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
}
+ },
+ "entity": {
+ "button": {
+ "set_datetime": {
+ "name": "Set Date&Time"
+ }
+ }
}
}
diff --git a/homeassistant/components/thethingsnetwork/coordinator.py b/homeassistant/components/thethingsnetwork/coordinator.py
index 64608c2f064..78ffceecf84 100644
--- a/homeassistant/components/thethingsnetwork/coordinator.py
+++ b/homeassistant/components/thethingsnetwork/coordinator.py
@@ -19,11 +19,14 @@ _LOGGER = logging.getLogger(__name__)
class TTNCoordinator(DataUpdateCoordinator[TTNClient.DATA_TYPE]):
"""TTN coordinator."""
+ config_entry: ConfigEntry
+
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize my coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
# Name of the data. For logging purposes.
name=f"TheThingsNetwork_{entry.data[CONF_APP_ID]}",
# Polling interval. Will only be polled if there are subscribers.
diff --git a/homeassistant/components/thethingsnetwork/sensor.py b/homeassistant/components/thethingsnetwork/sensor.py
index 25dd2f1e1eb..5aa851d99ae 100644
--- a/homeassistant/components/thethingsnetwork/sensor.py
+++ b/homeassistant/components/thethingsnetwork/sensor.py
@@ -7,7 +7,7 @@ from ttn_client import TTNSensorValue
from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import CONF_APP_ID, DOMAIN
@@ -17,7 +17,9 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add entities for TTN."""
@@ -38,7 +40,7 @@ async def async_setup_entry(
if (device_id, field_id) not in sensors
and isinstance(ttn_value, TTNSensorValue)
}
- if len(new_sensors):
+ if new_sensors:
async_add_entities(new_sensors.values())
sensors.update(new_sensors.keys())
diff --git a/homeassistant/components/threshold/binary_sensor.py b/homeassistant/components/threshold/binary_sensor.py
index 3d52d2225be..3227f030812 100644
--- a/homeassistant/components/threshold/binary_sensor.py
+++ b/homeassistant/components/threshold/binary_sensor.py
@@ -33,7 +33,10 @@ from homeassistant.core import (
from homeassistant.helpers import config_validation as cv, entity_registry as er
from homeassistant.helpers.device import async_device_info_to_link_from_entity
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -90,7 +93,7 @@ PLATFORM_SCHEMA = vol.All(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize threshold config entry."""
registry = er.async_get(hass)
diff --git a/homeassistant/components/tibber/coordinator.py b/homeassistant/components/tibber/coordinator.py
index 78841f9db91..2de9ebd1ec6 100644
--- a/homeassistant/components/tibber/coordinator.py
+++ b/homeassistant/components/tibber/coordinator.py
@@ -33,11 +33,17 @@ class TibberDataCoordinator(DataUpdateCoordinator[None]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, tibber_connection: tibber.Tibber) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ tibber_connection: tibber.Tibber,
+ ) -> None:
"""Initialize the data handler."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"Tibber {tibber_connection.name}",
update_interval=timedelta(minutes=20),
)
diff --git a/homeassistant/components/tibber/notify.py b/homeassistant/components/tibber/notify.py
index fdeeeba68ef..df6541591e0 100644
--- a/homeassistant/components/tibber/notify.py
+++ b/homeassistant/components/tibber/notify.py
@@ -12,13 +12,15 @@ from homeassistant.components.notify import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN as TIBBER_DOMAIN
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tibber notification entity."""
async_add_entities([TibberNotificationEntity(entry.entry_id)])
diff --git a/homeassistant/components/tibber/sensor.py b/homeassistant/components/tibber/sensor.py
index c1ec7bf2a9e..9f87b8a8490 100644
--- a/homeassistant/components/tibber/sensor.py
+++ b/homeassistant/components/tibber/sensor.py
@@ -33,7 +33,7 @@ from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
@@ -261,7 +261,9 @@ SENSORS: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tibber sensor."""
@@ -285,7 +287,7 @@ async def async_setup_entry(
if home.has_active_subscription:
entities.append(TibberSensorElPrice(home))
if coordinator is None:
- coordinator = TibberDataCoordinator(hass, tibber_connection)
+ coordinator = TibberDataCoordinator(hass, entry, tibber_connection)
entities.extend(
TibberDataSensor(home, coordinator, entity_description)
for entity_description in SENSORS
@@ -531,7 +533,7 @@ class TibberRtEntityCreator:
def __init__(
self,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
tibber_home: tibber.TibberHome,
entity_registry: er.EntityRegistry,
) -> None:
diff --git a/homeassistant/components/tibber/strings.json b/homeassistant/components/tibber/strings.json
index 05b98b97995..ec2c005d4e3 100644
--- a/homeassistant/components/tibber/strings.json
+++ b/homeassistant/components/tibber/strings.json
@@ -87,7 +87,7 @@
"services": {
"get_prices": {
"name": "Get energy prices",
- "description": "Get hourly energy prices from Tibber",
+ "description": "Fetches hourly energy prices including price level.",
"fields": {
"start": {
"name": "Start",
diff --git a/homeassistant/components/tile/binary_sensor.py b/homeassistant/components/tile/binary_sensor.py
index 1719c793c0e..6abc80732a6 100644
--- a/homeassistant/components/tile/binary_sensor.py
+++ b/homeassistant/components/tile/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import TileConfigEntry, TileCoordinator
from .entity import TileEntity
@@ -35,7 +35,9 @@ ENTITIES: tuple[TileBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: TileConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TileConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tile binary sensors."""
diff --git a/homeassistant/components/tile/device_tracker.py b/homeassistant/components/tile/device_tracker.py
index 6a0aae1bdf9..66a3b8b0e27 100644
--- a/homeassistant/components/tile/device_tracker.py
+++ b/homeassistant/components/tile/device_tracker.py
@@ -6,7 +6,7 @@ import logging
from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import as_utc
from .coordinator import TileConfigEntry, TileCoordinator
@@ -26,7 +26,9 @@ ATTR_VOIP_STATE = "voip_state"
async def async_setup_entry(
- hass: HomeAssistant, entry: TileConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TileConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tile device trackers."""
diff --git a/homeassistant/components/tilt_ble/sensor.py b/homeassistant/components/tilt_ble/sensor.py
index e8e1f902cd9..411484cf2fe 100644
--- a/homeassistant/components/tilt_ble/sensor.py
+++ b/homeassistant/components/tilt_ble/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .const import DOMAIN
@@ -86,7 +86,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Tilt Hydrometer BLE sensors."""
coordinator: PassiveBluetoothProcessorCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/time_date/sensor.py b/homeassistant/components/time_date/sensor.py
index 1e86a1ba6c6..f05244e7680 100644
--- a/homeassistant/components/time_date/sensor.py
+++ b/homeassistant/components/time_date/sensor.py
@@ -18,7 +18,10 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DISPLAY_OPTIONS, EVENT_CORE_CONFIG_UPDATE
from homeassistant.core import CALLBACK_TYPE, Event, HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_point_in_utc_time
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
@@ -56,7 +59,9 @@ async def async_setup_platform(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Time & Date sensor."""
diff --git a/homeassistant/components/timer/__init__.py b/homeassistant/components/timer/__init__.py
index b0ade17b9c9..3cf8307e9b3 100644
--- a/homeassistant/components/timer/__init__.py
+++ b/homeassistant/components/timer/__init__.py
@@ -374,6 +374,9 @@ class Timer(collection.CollectionEntity, RestoreEntity):
@callback
def async_cancel(self) -> None:
"""Cancel a timer."""
+ if self._state == STATUS_IDLE:
+ return
+
if self._listener:
self._listener()
self._listener = None
@@ -389,13 +392,15 @@ class Timer(collection.CollectionEntity, RestoreEntity):
@callback
def async_finish(self) -> None:
"""Reset and updates the states, fire finished event."""
- if self._state != STATUS_ACTIVE or self._end is None:
+ if self._state == STATUS_IDLE:
return
if self._listener:
self._listener()
self._listener = None
end = self._end
+ if end is None:
+ end = dt_util.utcnow().replace(microsecond=0)
self._state = STATUS_IDLE
self._end = None
self._remaining = None
diff --git a/homeassistant/components/tod/binary_sensor.py b/homeassistant/components/tod/binary_sensor.py
index 3ac90b5578c..1ab34861a6e 100644
--- a/homeassistant/components/tod/binary_sensor.py
+++ b/homeassistant/components/tod/binary_sensor.py
@@ -24,7 +24,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, event
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.sun import get_astral_event_date, get_astral_event_next
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util
@@ -59,7 +62,7 @@ PLATFORM_SCHEMA = BINARY_SENSOR_PLATFORM_SCHEMA.extend(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Times of the Day config entry."""
if hass.config.time_zone is None:
diff --git a/homeassistant/components/todo/intent.py b/homeassistant/components/todo/intent.py
index c678408a576..d679a57bf96 100644
--- a/homeassistant/components/todo/intent.py
+++ b/homeassistant/components/todo/intent.py
@@ -11,11 +11,13 @@ from . import TodoItem, TodoItemStatus, TodoListEntity
from .const import DATA_COMPONENT, DOMAIN
INTENT_LIST_ADD_ITEM = "HassListAddItem"
+INTENT_LIST_COMPLETE_ITEM = "HassListCompleteItem"
async def async_setup_intents(hass: HomeAssistant) -> None:
"""Set up the todo intents."""
intent.async_register(hass, ListAddItemIntent())
+ intent.async_register(hass, ListCompleteItemIntent())
class ListAddItemIntent(intent.IntentHandler):
@@ -53,14 +55,92 @@ class ListAddItemIntent(intent.IntentHandler):
match_result.states[0].entity_id
)
if target_list is None:
- raise intent.IntentHandleError(f"No to-do list: {list_name}")
+ raise intent.IntentHandleError(
+ f"No to-do list: {list_name}", "list_not_found"
+ )
# Add to list
await target_list.async_create_todo_item(
TodoItem(summary=item, status=TodoItemStatus.NEEDS_ACTION)
)
- response = intent_obj.create_response()
+ response: intent.IntentResponse = intent_obj.create_response()
+ response.response_type = intent.IntentResponseType.ACTION_DONE
+ response.async_set_results(
+ [
+ intent.IntentResponseTarget(
+ type=intent.IntentResponseTargetType.ENTITY,
+ name=list_name,
+ id=match_result.states[0].entity_id,
+ )
+ ]
+ )
+ return response
+
+
+class ListCompleteItemIntent(intent.IntentHandler):
+ """Handle ListCompleteItem intents."""
+
+ intent_type = INTENT_LIST_COMPLETE_ITEM
+ description = "Complete item on a todo list"
+ slot_schema = {
+ vol.Required("item"): intent.non_empty_string,
+ vol.Required("name"): intent.non_empty_string,
+ }
+ platforms = {DOMAIN}
+
+ async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse:
+ """Handle the intent."""
+ hass = intent_obj.hass
+
+ slots = self.async_validate_slots(intent_obj.slots)
+ item = slots["item"]["value"]
+ list_name = slots["name"]["value"]
+
+ target_list: TodoListEntity | None = None
+
+ # Find matching list
+ match_constraints = intent.MatchTargetsConstraints(
+ name=list_name, domains=[DOMAIN], assistant=intent_obj.assistant
+ )
+ match_result = intent.async_match_targets(hass, match_constraints)
+ if not match_result.is_match:
+ raise intent.MatchFailedError(
+ result=match_result, constraints=match_constraints
+ )
+
+ target_list = hass.data[DATA_COMPONENT].get_entity(
+ match_result.states[0].entity_id
+ )
+ if target_list is None:
+ raise intent.IntentHandleError(
+ f"No to-do list: {list_name}", "list_not_found"
+ )
+
+ # Find item in list
+ matching_item = None
+ for todo_item in target_list.todo_items or ():
+ if (
+ item in (todo_item.uid, todo_item.summary)
+ and todo_item.status == TodoItemStatus.NEEDS_ACTION
+ ):
+ matching_item = todo_item
+ break
+ if not matching_item or not matching_item.uid:
+ raise intent.IntentHandleError(
+ f"Item '{item}' not found on list", "item_not_found"
+ )
+
+ # Mark as completed
+ await target_list.async_update_todo_item(
+ TodoItem(
+ uid=matching_item.uid,
+ summary=matching_item.summary,
+ status=TodoItemStatus.COMPLETED,
+ )
+ )
+
+ response: intent.IntentResponse = intent_obj.create_response()
response.response_type = intent.IntentResponseType.ACTION_DONE
response.async_set_results(
[
diff --git a/homeassistant/components/todoist/calendar.py b/homeassistant/components/todoist/calendar.py
index 8c61394d300..2e2873353c6 100644
--- a/homeassistant/components/todoist/calendar.py
+++ b/homeassistant/components/todoist/calendar.py
@@ -24,7 +24,10 @@ from homeassistant.core import Event, HomeAssistant, ServiceCall, callback
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -113,7 +116,9 @@ SCAN_INTERVAL = timedelta(minutes=1)
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Todoist calendar platform config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/todoist/todo.py b/homeassistant/components/todoist/todo.py
index 490e4ad9f1a..202c51fb4c0 100644
--- a/homeassistant/components/todoist/todo.py
+++ b/homeassistant/components/todoist/todo.py
@@ -14,7 +14,7 @@ from homeassistant.components.todo import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -23,7 +23,9 @@ from .coordinator import TodoistCoordinator
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Todoist todo platform config entry."""
coordinator: TodoistCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/binary_sensor.py b/homeassistant/components/tolo/binary_sensor.py
index 845f8ed22e3..cb3ba46b604 100644
--- a/homeassistant/components/tolo/binary_sensor.py
+++ b/homeassistant/components/tolo/binary_sensor.py
@@ -7,7 +7,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import ToloSaunaCoordinatorEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/button.py b/homeassistant/components/tolo/button.py
index b7c4362ca7b..9e4c8c84be9 100644
--- a/homeassistant/components/tolo/button.py
+++ b/homeassistant/components/tolo/button.py
@@ -6,7 +6,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -16,7 +16,7 @@ from .entity import ToloSaunaCoordinatorEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up buttons for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/climate.py b/homeassistant/components/tolo/climate.py
index 5e6428525c1..0df8635fca9 100644
--- a/homeassistant/components/tolo/climate.py
+++ b/homeassistant/components/tolo/climate.py
@@ -23,7 +23,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -33,7 +33,7 @@ from .entity import ToloSaunaCoordinatorEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate controls for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/coordinator.py b/homeassistant/components/tolo/coordinator.py
index 632cc819f5a..729073b16c4 100644
--- a/homeassistant/components/tolo/coordinator.py
+++ b/homeassistant/components/tolo/coordinator.py
@@ -28,6 +28,8 @@ class ToloSaunaData(NamedTuple):
class ToloSaunaUpdateCoordinator(DataUpdateCoordinator[ToloSaunaData]):
"""DataUpdateCoordinator for TOLO Sauna."""
+ config_entry: ConfigEntry
+
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize ToloSaunaUpdateCoordinator."""
self.client = ToloClient(
@@ -38,6 +40,7 @@ class ToloSaunaUpdateCoordinator(DataUpdateCoordinator[ToloSaunaData]):
super().__init__(
hass=hass,
logger=_LOGGER,
+ config_entry=entry,
name=f"{entry.title} ({entry.data[CONF_HOST]}) Data Update Coordinator",
update_interval=timedelta(seconds=5),
)
diff --git a/homeassistant/components/tolo/fan.py b/homeassistant/components/tolo/fan.py
index 9e48778b507..7bddf775143 100644
--- a/homeassistant/components/tolo/fan.py
+++ b/homeassistant/components/tolo/fan.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import ToloSaunaCoordinatorEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up fan controls for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/light.py b/homeassistant/components/tolo/light.py
index eeb37305fe8..9ccd4a8e407 100644
--- a/homeassistant/components/tolo/light.py
+++ b/homeassistant/components/tolo/light.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import ToloSaunaCoordinatorEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up light controls for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/number.py b/homeassistant/components/tolo/number.py
index 73505c5b251..902fb749d23 100644
--- a/homeassistant/components/tolo/number.py
+++ b/homeassistant/components/tolo/number.py
@@ -18,7 +18,7 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -68,7 +68,7 @@ NUMBERS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number controls for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/select.py b/homeassistant/components/tolo/select.py
index fee1ac1774e..b08f37e40ae 100644
--- a/homeassistant/components/tolo/select.py
+++ b/homeassistant/components/tolo/select.py
@@ -11,7 +11,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, AromaTherapySlot, LampMode
from .coordinator import ToloSaunaUpdateCoordinator
@@ -54,7 +54,7 @@ SELECTS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select entities for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/sensor.py b/homeassistant/components/tolo/sensor.py
index 0e94ec0ae1e..e97211c8e40 100644
--- a/homeassistant/components/tolo/sensor.py
+++ b/homeassistant/components/tolo/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -89,7 +89,7 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up (non-binary, general) sensors for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tolo/switch.py b/homeassistant/components/tolo/switch.py
index d39dd17f0f3..ce863053e26 100644
--- a/homeassistant/components/tolo/switch.py
+++ b/homeassistant/components/tolo/switch.py
@@ -11,7 +11,7 @@ from tololib import ToloClient, ToloStatus
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToloSaunaUpdateCoordinator
@@ -45,7 +45,7 @@ SWITCHES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch controls for TOLO Sauna."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tomorrowio/__init__.py b/homeassistant/components/tomorrowio/__init__.py
index 73f62735e06..7d6b9ed3f73 100644
--- a/homeassistant/components/tomorrowio/__init__.py
+++ b/homeassistant/components/tomorrowio/__init__.py
@@ -29,7 +29,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
# we will not use the class's lat and long so we can pass in garbage
# lats and longs
api = TomorrowioV4(api_key, 361.0, 361.0, unit_system="metric", session=session)
- coordinator = TomorrowioDataUpdateCoordinator(hass, api)
+ coordinator = TomorrowioDataUpdateCoordinator(hass, entry, api)
hass.data[DOMAIN][api_key] = coordinator
await coordinator.async_setup_entry(entry)
diff --git a/homeassistant/components/tomorrowio/coordinator.py b/homeassistant/components/tomorrowio/coordinator.py
index 60b997e4c0d..2a6b3675792 100644
--- a/homeassistant/components/tomorrowio/coordinator.py
+++ b/homeassistant/components/tomorrowio/coordinator.py
@@ -116,14 +116,23 @@ def async_set_update_interval(
class TomorrowioDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Define an object to hold Tomorrow.io data."""
- def __init__(self, hass: HomeAssistant, api: TomorrowioV4) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, api: TomorrowioV4
+ ) -> None:
"""Initialize."""
self._api = api
self.data = {CURRENT: {}, FORECASTS: {}}
self.entry_id_to_location_dict: dict[str, str] = {}
self._coordinator_ready: asyncio.Event | None = None
- super().__init__(hass, LOGGER, name=f"{DOMAIN}_{self._api.api_key_masked}")
+ super().__init__(
+ hass,
+ LOGGER,
+ config_entry=config_entry,
+ name=f"{DOMAIN}_{self._api.api_key_masked}",
+ )
def add_entry_to_location_dict(self, entry: ConfigEntry) -> None:
"""Add an entry to the location dict."""
diff --git a/homeassistant/components/tomorrowio/sensor.py b/homeassistant/components/tomorrowio/sensor.py
index 7ff17961b58..08e1991d831 100644
--- a/homeassistant/components/tomorrowio/sensor.py
+++ b/homeassistant/components/tomorrowio/sensor.py
@@ -34,7 +34,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import DistanceConverter, SpeedConverter
from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
@@ -328,7 +328,7 @@ SENSOR_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
coordinator = hass.data[DOMAIN][config_entry.data[CONF_API_KEY]]
diff --git a/homeassistant/components/tomorrowio/weather.py b/homeassistant/components/tomorrowio/weather.py
index 92b09500e7b..0a070a1b33b 100644
--- a/homeassistant/components/tomorrowio/weather.py
+++ b/homeassistant/components/tomorrowio/weather.py
@@ -33,7 +33,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sun import is_up
from homeassistant.util import dt as dt_util
@@ -66,7 +66,7 @@ from .entity import TomorrowioEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a config entry."""
coordinator = hass.data[DOMAIN][config_entry.data[CONF_API_KEY]]
diff --git a/homeassistant/components/toon/__init__.py b/homeassistant/components/toon/__init__.py
index 43c787b2301..1c56b780c0f 100644
--- a/homeassistant/components/toon/__init__.py
+++ b/homeassistant/components/toon/__init__.py
@@ -89,7 +89,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
implementation = await async_get_config_entry_implementation(hass, entry)
session = OAuth2Session(hass, entry, implementation)
- coordinator = ToonDataUpdateCoordinator(hass, entry=entry, session=session)
+ coordinator = ToonDataUpdateCoordinator(hass, entry, session)
await coordinator.toon.activate_agreement(
agreement_id=entry.data[CONF_AGREEMENT_ID]
)
diff --git a/homeassistant/components/toon/binary_sensor.py b/homeassistant/components/toon/binary_sensor.py
index 11b13a32ee5..eff8aed0a20 100644
--- a/homeassistant/components/toon/binary_sensor.py
+++ b/homeassistant/components/toon/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToonDataUpdateCoordinator
@@ -25,7 +25,9 @@ from .entity import (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Toon binary sensor based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/toon/climate.py b/homeassistant/components/toon/climate.py
index 0c2e5b9b232..5538a0abd91 100644
--- a/homeassistant/components/toon/climate.py
+++ b/homeassistant/components/toon/climate.py
@@ -24,7 +24,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ToonDataUpdateCoordinator
from .const import DEFAULT_MAX_TEMP, DEFAULT_MIN_TEMP, DOMAIN
@@ -33,7 +33,9 @@ from .helpers import toon_exception_handler
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Toon binary sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/toon/coordinator.py b/homeassistant/components/toon/coordinator.py
index 586eca34959..894b4c91334 100644
--- a/homeassistant/components/toon/coordinator.py
+++ b/homeassistant/components/toon/coordinator.py
@@ -28,12 +28,13 @@ _LOGGER = logging.getLogger(__name__)
class ToonDataUpdateCoordinator(DataUpdateCoordinator[Status]):
"""Class to manage fetching Toon data from single endpoint."""
+ config_entry: ConfigEntry
+
def __init__(
- self, hass: HomeAssistant, *, entry: ConfigEntry, session: OAuth2Session
+ self, hass: HomeAssistant, entry: ConfigEntry, session: OAuth2Session
) -> None:
"""Initialize global Toon data updater."""
self.session = session
- self.entry = entry
async def async_token_refresh() -> str:
await session.async_ensure_token_valid()
@@ -46,49 +47,55 @@ class ToonDataUpdateCoordinator(DataUpdateCoordinator[Status]):
)
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=DEFAULT_SCAN_INTERVAL
+ hass,
+ _LOGGER,
+ config_entry=entry,
+ name=DOMAIN,
+ update_interval=DEFAULT_SCAN_INTERVAL,
)
async def register_webhook(self, event: Event | None = None) -> None:
"""Register a webhook with Toon to get live updates."""
- if CONF_WEBHOOK_ID not in self.entry.data:
- data = {**self.entry.data, CONF_WEBHOOK_ID: secrets.token_hex()}
- self.hass.config_entries.async_update_entry(self.entry, data=data)
+ if CONF_WEBHOOK_ID not in self.config_entry.data:
+ data = {**self.config_entry.data, CONF_WEBHOOK_ID: secrets.token_hex()}
+ self.hass.config_entries.async_update_entry(self.config_entry, data=data)
if cloud.async_active_subscription(self.hass):
- if CONF_CLOUDHOOK_URL not in self.entry.data:
+ if CONF_CLOUDHOOK_URL not in self.config_entry.data:
try:
webhook_url = await cloud.async_create_cloudhook(
- self.hass, self.entry.data[CONF_WEBHOOK_ID]
+ self.hass, self.config_entry.data[CONF_WEBHOOK_ID]
)
except cloud.CloudNotConnected:
webhook_url = webhook.async_generate_url(
- self.hass, self.entry.data[CONF_WEBHOOK_ID]
+ self.hass, self.config_entry.data[CONF_WEBHOOK_ID]
)
else:
- data = {**self.entry.data, CONF_CLOUDHOOK_URL: webhook_url}
- self.hass.config_entries.async_update_entry(self.entry, data=data)
+ data = {**self.config_entry.data, CONF_CLOUDHOOK_URL: webhook_url}
+ self.hass.config_entries.async_update_entry(
+ self.config_entry, data=data
+ )
else:
- webhook_url = self.entry.data[CONF_CLOUDHOOK_URL]
+ webhook_url = self.config_entry.data[CONF_CLOUDHOOK_URL]
else:
webhook_url = webhook.async_generate_url(
- self.hass, self.entry.data[CONF_WEBHOOK_ID]
+ self.hass, self.config_entry.data[CONF_WEBHOOK_ID]
)
# Ensure the webhook is not registered already
- webhook_unregister(self.hass, self.entry.data[CONF_WEBHOOK_ID])
+ webhook_unregister(self.hass, self.config_entry.data[CONF_WEBHOOK_ID])
webhook_register(
self.hass,
DOMAIN,
"Toon",
- self.entry.data[CONF_WEBHOOK_ID],
+ self.config_entry.data[CONF_WEBHOOK_ID],
self.handle_webhook,
)
try:
await self.toon.subscribe_webhook(
- application_id=self.entry.entry_id, url=webhook_url
+ application_id=self.config_entry.entry_id, url=webhook_url
)
_LOGGER.debug("Registered Toon webhook: %s", webhook_url)
except ToonError as err:
@@ -131,14 +138,14 @@ class ToonDataUpdateCoordinator(DataUpdateCoordinator[Status]):
async def unregister_webhook(self, event: Event | None = None) -> None:
"""Remove / Unregister webhook for toon."""
_LOGGER.debug(
- "Unregistering Toon webhook (%s)", self.entry.data[CONF_WEBHOOK_ID]
+ "Unregistering Toon webhook (%s)", self.config_entry.data[CONF_WEBHOOK_ID]
)
try:
- await self.toon.unsubscribe_webhook(self.entry.entry_id)
+ await self.toon.unsubscribe_webhook(self.config_entry.entry_id)
except ToonError as err:
_LOGGER.error("Failed unregistering Toon webhook - %s", err)
- webhook_unregister(self.hass, self.entry.data[CONF_WEBHOOK_ID])
+ webhook_unregister(self.hass, self.config_entry.data[CONF_WEBHOOK_ID])
async def _async_update_data(self) -> Status:
"""Fetch data from Toon."""
diff --git a/homeassistant/components/toon/sensor.py b/homeassistant/components/toon/sensor.py
index 09f36c88079..e5b155b409b 100644
--- a/homeassistant/components/toon/sensor.py
+++ b/homeassistant/components/toon/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CURRENCY_EUR, DOMAIN, VOLUME_CM3, VOLUME_LMIN
from .coordinator import ToonDataUpdateCoordinator
@@ -36,7 +36,9 @@ from .entity import (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Toon sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/toon/switch.py b/homeassistant/components/toon/switch.py
index deb2a12f2d0..d59a542d4d8 100644
--- a/homeassistant/components/toon/switch.py
+++ b/homeassistant/components/toon/switch.py
@@ -15,7 +15,7 @@ from toonapi import (
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ToonDataUpdateCoordinator
@@ -24,7 +24,9 @@ from .helpers import toon_exception_handler
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Toon switches based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/totalconnect/__init__.py b/homeassistant/components/totalconnect/__init__.py
index 9f291ea15a6..a481fd41c84 100644
--- a/homeassistant/components/totalconnect/__init__.py
+++ b/homeassistant/components/totalconnect/__init__.py
@@ -3,18 +3,15 @@
from total_connect_client.client import TotalConnectClient
from total_connect_client.exceptions import AuthenticationError
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from .const import AUTO_BYPASS, CONF_USERCODES
-from .coordinator import TotalConnectDataUpdateCoordinator
+from .coordinator import TotalConnectConfigEntry, TotalConnectDataUpdateCoordinator
PLATFORMS = [Platform.ALARM_CONTROL_PANEL, Platform.BINARY_SENSOR, Platform.BUTTON]
-type TotalConnectConfigEntry = ConfigEntry[TotalConnectDataUpdateCoordinator]
-
async def async_setup_entry(
hass: HomeAssistant, entry: TotalConnectConfigEntry
@@ -41,7 +38,7 @@ async def async_setup_entry(
"TotalConnect authentication failed during setup"
) from exception
- coordinator = TotalConnectDataUpdateCoordinator(hass, client)
+ coordinator = TotalConnectDataUpdateCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/totalconnect/alarm_control_panel.py b/homeassistant/components/totalconnect/alarm_control_panel.py
index 021d1c7b886..9ed29ea01c8 100644
--- a/homeassistant/components/totalconnect/alarm_control_panel.py
+++ b/homeassistant/components/totalconnect/alarm_control_panel.py
@@ -12,14 +12,13 @@ from homeassistant.components.alarm_control_panel import (
AlarmControlPanelState,
CodeFormat,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CODE_REQUIRED, DOMAIN
-from .coordinator import TotalConnectDataUpdateCoordinator
+from .coordinator import TotalConnectConfigEntry, TotalConnectDataUpdateCoordinator
from .entity import TotalConnectLocationEntity
SERVICE_ALARM_ARM_AWAY_INSTANT = "arm_away_instant"
@@ -27,7 +26,9 @@ SERVICE_ALARM_ARM_HOME_INSTANT = "arm_home_instant"
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TotalConnectConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up TotalConnect alarm panels based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/totalconnect/binary_sensor.py b/homeassistant/components/totalconnect/binary_sensor.py
index 9a3c2558999..2f3802dc9a6 100644
--- a/homeassistant/components/totalconnect/binary_sensor.py
+++ b/homeassistant/components/totalconnect/binary_sensor.py
@@ -12,12 +12,11 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
BinarySensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .coordinator import TotalConnectDataUpdateCoordinator
+from .coordinator import TotalConnectConfigEntry, TotalConnectDataUpdateCoordinator
from .entity import TotalConnectLocationEntity, TotalConnectZoneEntity
LOW_BATTERY = "low_battery"
@@ -119,7 +118,9 @@ LOCATION_BINARY_SENSORS: tuple[TotalConnectAlarmBinarySensorEntityDescription, .
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TotalConnectConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up TotalConnect device sensors based on a config entry."""
sensors: list = []
diff --git a/homeassistant/components/totalconnect/button.py b/homeassistant/components/totalconnect/button.py
index e228f03ec6b..eb85dcce1bf 100644
--- a/homeassistant/components/totalconnect/button.py
+++ b/homeassistant/components/totalconnect/button.py
@@ -7,12 +7,11 @@ from total_connect_client.location import TotalConnectLocation
from total_connect_client.zone import TotalConnectZone
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .coordinator import TotalConnectDataUpdateCoordinator
+from .coordinator import TotalConnectConfigEntry, TotalConnectDataUpdateCoordinator
from .entity import TotalConnectLocationEntity, TotalConnectZoneEntity
@@ -38,7 +37,9 @@ PANEL_BUTTONS: tuple[TotalConnectButtonEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TotalConnectConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up TotalConnect buttons based on a config entry."""
buttons: list = []
diff --git a/homeassistant/components/totalconnect/coordinator.py b/homeassistant/components/totalconnect/coordinator.py
index 9b500db1951..673c168d204 100644
--- a/homeassistant/components/totalconnect/coordinator.py
+++ b/homeassistant/components/totalconnect/coordinator.py
@@ -20,17 +20,28 @@ from .const import DOMAIN
SCAN_INTERVAL = timedelta(seconds=30)
_LOGGER = logging.getLogger(__name__)
+type TotalConnectConfigEntry = ConfigEntry[TotalConnectDataUpdateCoordinator]
+
class TotalConnectDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Class to fetch data from TotalConnect."""
- config_entry: ConfigEntry
+ config_entry: TotalConnectConfigEntry
- def __init__(self, hass: HomeAssistant, client: TotalConnectClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: TotalConnectConfigEntry,
+ client: TotalConnectClient,
+ ) -> None:
"""Initialize."""
self.client = client
super().__init__(
- hass, logger=_LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL
+ hass,
+ logger=_LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
)
async def _async_update_data(self) -> None:
diff --git a/homeassistant/components/totalconnect/diagnostics.py b/homeassistant/components/totalconnect/diagnostics.py
index 85f52ccc670..fc310bf850c 100644
--- a/homeassistant/components/totalconnect/diagnostics.py
+++ b/homeassistant/components/totalconnect/diagnostics.py
@@ -5,9 +5,10 @@ from __future__ import annotations
from typing import Any
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
+from .coordinator import TotalConnectConfigEntry
+
TO_REDACT = [
"username",
"Password",
@@ -22,7 +23,7 @@ TO_REDACT = [
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, config_entry: ConfigEntry
+ hass: HomeAssistant, config_entry: TotalConnectConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
client = config_entry.runtime_data.client
@@ -82,6 +83,7 @@ async def async_get_config_entry_diagnostics(
"is_new_partition": partition.is_new_partition,
"is_night_stay_enabled": partition.is_night_stay_enabled,
"exit_delay_timer": partition.exit_delay_timer,
+ "arming_state": partition.arming_state,
}
new_location["partitions"].append(new_partition)
diff --git a/homeassistant/components/touchline/climate.py b/homeassistant/components/touchline/climate.py
index f7eec7c54f9..86526f4718b 100644
--- a/homeassistant/components/touchline/climate.py
+++ b/homeassistant/components/touchline/climate.py
@@ -4,7 +4,7 @@ from __future__ import annotations
from typing import Any, NamedTuple
-from pytouchline import PyTouchline
+from pytouchline_extended import PyTouchline
import voluptuous as vol
from homeassistant.components.climate import (
@@ -53,12 +53,13 @@ def setup_platform(
"""Set up the Touchline devices."""
host = config[CONF_HOST]
- py_touchline = PyTouchline()
- number_of_devices = int(py_touchline.get_number_of_devices(host))
- add_entities(
- (Touchline(PyTouchline(device_id)) for device_id in range(number_of_devices)),
- True,
- )
+ py_touchline = PyTouchline(url=host)
+ number_of_devices = int(py_touchline.get_number_of_devices())
+ devices = [
+ Touchline(PyTouchline(id=device_id, url=host))
+ for device_id in range(number_of_devices)
+ ]
+ add_entities(devices, True)
class Touchline(ClimateEntity):
diff --git a/homeassistant/components/touchline/manifest.json b/homeassistant/components/touchline/manifest.json
index c003cca97a4..6d25462408b 100644
--- a/homeassistant/components/touchline/manifest.json
+++ b/homeassistant/components/touchline/manifest.json
@@ -6,5 +6,5 @@
"iot_class": "local_polling",
"loggers": ["pytouchline"],
"quality_scale": "legacy",
- "requirements": ["pytouchline==0.7"]
+ "requirements": ["pytouchline_extended==0.4.5"]
}
diff --git a/homeassistant/components/touchline_sl/__init__.py b/homeassistant/components/touchline_sl/__init__.py
index 45a85185673..ba1da06ed5a 100644
--- a/homeassistant/components/touchline_sl/__init__.py
+++ b/homeassistant/components/touchline_sl/__init__.py
@@ -6,18 +6,15 @@ import asyncio
from pytouchlinesl import TouchlineSL
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
from .const import DOMAIN
-from .coordinator import TouchlineSLModuleCoordinator
+from .coordinator import TouchlineSLConfigEntry, TouchlineSLModuleCoordinator
PLATFORMS: list[Platform] = [Platform.CLIMATE]
-type TouchlineSLConfigEntry = ConfigEntry[list[TouchlineSLModuleCoordinator]]
-
async def async_setup_entry(hass: HomeAssistant, entry: TouchlineSLConfigEntry) -> bool:
"""Set up Roth Touchline SL from a config entry."""
@@ -26,7 +23,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: TouchlineSLConfigEntry)
)
coordinators: list[TouchlineSLModuleCoordinator] = [
- TouchlineSLModuleCoordinator(hass, module) for module in await account.modules()
+ TouchlineSLModuleCoordinator(hass, entry, module)
+ for module in await account.modules()
]
await asyncio.gather(
diff --git a/homeassistant/components/touchline_sl/climate.py b/homeassistant/components/touchline_sl/climate.py
index 8a0ffc4cd86..7c5ea4ea9ca 100644
--- a/homeassistant/components/touchline_sl/climate.py
+++ b/homeassistant/components/touchline_sl/climate.py
@@ -10,17 +10,16 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import TouchlineSLConfigEntry
-from .coordinator import TouchlineSLModuleCoordinator
+from .coordinator import TouchlineSLConfigEntry, TouchlineSLModuleCoordinator
from .entity import TouchlineSLZoneEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: TouchlineSLConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Touchline devices."""
coordinators = entry.runtime_data
diff --git a/homeassistant/components/touchline_sl/coordinator.py b/homeassistant/components/touchline_sl/coordinator.py
index cd74ba6130f..dce616a81b3 100644
--- a/homeassistant/components/touchline_sl/coordinator.py
+++ b/homeassistant/components/touchline_sl/coordinator.py
@@ -10,6 +10,7 @@ from pytouchlinesl import Module, Zone
from pytouchlinesl.client import RothAPIError
from pytouchlinesl.client.models import GlobalScheduleModel
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -26,14 +27,22 @@ class TouchlineSLModuleData:
schedules: dict[str, GlobalScheduleModel]
+type TouchlineSLConfigEntry = ConfigEntry[list[TouchlineSLModuleCoordinator]]
+
+
class TouchlineSLModuleCoordinator(DataUpdateCoordinator[TouchlineSLModuleData]):
"""A coordinator to manage the fetching of Touchline SL data."""
- def __init__(self, hass: HomeAssistant, module: Module) -> None:
+ config_entry: TouchlineSLConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: TouchlineSLConfigEntry, module: Module
+ ) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
logger=_LOGGER,
+ config_entry=config_entry,
name=f"Touchline SL ({module.name})",
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/tplink/binary_sensor.py b/homeassistant/components/tplink/binary_sensor.py
index 6986765b110..38935595fe2 100644
--- a/homeassistant/components/tplink/binary_sensor.py
+++ b/homeassistant/components/tplink/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .entity import CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription
@@ -73,7 +73,7 @@ BINARYSENSOR_DESCRIPTIONS_MAP = {desc.key: desc for desc in BINARY_SENSOR_DESCRI
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/button.py b/homeassistant/components/tplink/button.py
index 4279a233d21..145adb79185 100644
--- a/homeassistant/components/tplink/button.py
+++ b/homeassistant/components/tplink/button.py
@@ -15,7 +15,7 @@ from homeassistant.components.button import (
)
from homeassistant.components.siren import DOMAIN as SIREN_DOMAIN
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .deprecate import DeprecatedInfo
@@ -95,7 +95,7 @@ BUTTON_DESCRIPTIONS_MAP = {desc.key: desc for desc in BUTTON_DESCRIPTIONS}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up buttons."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/camera.py b/homeassistant/components/tplink/camera.py
index b0f1f1a62c1..7b59678da8e 100644
--- a/homeassistant/components/tplink/camera.py
+++ b/homeassistant/components/tplink/camera.py
@@ -19,7 +19,7 @@ from homeassistant.components.camera import (
from homeassistant.config_entries import ConfigFlowContext
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .const import CONF_CAMERA_CREDENTIALS
@@ -59,7 +59,7 @@ CAMERA_DESCRIPTIONS: tuple[TPLinkCameraEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up camera entities."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/climate.py b/homeassistant/components/tplink/climate.py
index 7204c2a7665..66037d7476e 100644
--- a/homeassistant/components/tplink/climate.py
+++ b/homeassistant/components/tplink/climate.py
@@ -22,7 +22,7 @@ from homeassistant.components.climate import (
from homeassistant.const import PRECISION_TENTHS, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry, legacy_device_id
from .const import DOMAIN, UNIT_MAPPING
@@ -71,7 +71,7 @@ CLIMATE_DESCRIPTIONS: tuple[TPLinkClimateEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up climate entities."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/coordinator.py b/homeassistant/components/tplink/coordinator.py
index d1b4694779d..1a7b40457f0 100644
--- a/homeassistant/components/tplink/coordinator.py
+++ b/homeassistant/components/tplink/coordinator.py
@@ -9,6 +9,7 @@ import logging
from kasa import AuthenticationError, Credentials, Device, KasaException
from kasa.iot import IotStrip
+from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
@@ -123,11 +124,14 @@ class TPLinkDataUpdateCoordinator(DataUpdateCoordinator[None]):
def get_child_coordinator(
self,
child: Device,
+ platform_domain: str,
) -> TPLinkDataUpdateCoordinator:
"""Get separate child coordinator for a device or self if not needed."""
# The iot HS300 allows a limited number of concurrent requests and fetching the
# emeter information requires separate ones so create child coordinators here.
- if isinstance(self.device, IotStrip):
+ # This does not happen for switches as the state is available on the
+ # parent device info.
+ if isinstance(self.device, IotStrip) and platform_domain != SWITCH_DOMAIN:
if not (child_coordinator := self._child_coordinators.get(child.device_id)):
# The child coordinators only update energy data so we can
# set a longer update interval to avoid flooding the device
diff --git a/homeassistant/components/tplink/entity.py b/homeassistant/components/tplink/entity.py
index 15c07655e69..7c1e9e72b85 100644
--- a/homeassistant/components/tplink/entity.py
+++ b/homeassistant/components/tplink/entity.py
@@ -508,7 +508,9 @@ class CoordinatedTPLinkFeatureEntity(CoordinatedTPLinkEntity, ABC):
)
for child in children:
- child_coordinator = coordinator.get_child_coordinator(child)
+ child_coordinator = coordinator.get_child_coordinator(
+ child, platform_domain
+ )
child_entities = cls._entities_for_device(
hass,
@@ -651,7 +653,9 @@ class CoordinatedTPLinkModuleEntity(CoordinatedTPLinkEntity, ABC):
device.host,
)
for child in children:
- child_coordinator = coordinator.get_child_coordinator(child)
+ child_coordinator = coordinator.get_child_coordinator(
+ child, platform_domain
+ )
child_entities: list[_E] = cls._entities_for_device(
hass,
diff --git a/homeassistant/components/tplink/fan.py b/homeassistant/components/tplink/fan.py
index 1c31d84b778..88396742b36 100644
--- a/homeassistant/components/tplink/fan.py
+++ b/homeassistant/components/tplink/fan.py
@@ -15,7 +15,7 @@ from homeassistant.components.fan import (
FanEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -59,7 +59,7 @@ FAN_DESCRIPTIONS: tuple[TPLinkFanEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up fans."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/light.py b/homeassistant/components/tplink/light.py
index 718b5ed7120..b3cee1d3baf 100644
--- a/homeassistant/components/tplink/light.py
+++ b/homeassistant/components/tplink/light.py
@@ -29,7 +29,7 @@ from homeassistant.components.light import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from . import TPLinkConfigEntry, legacy_device_id
@@ -196,7 +196,7 @@ LIGHT_EFFECT_DESCRIPTIONS: tuple[TPLinkLightEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up lights."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/manifest.json b/homeassistant/components/tplink/manifest.json
index ff65211c9b3..cdd6ab57c6a 100644
--- a/homeassistant/components/tplink/manifest.json
+++ b/homeassistant/components/tplink/manifest.json
@@ -301,5 +301,5 @@
"iot_class": "local_polling",
"loggers": ["kasa"],
"quality_scale": "platinum",
- "requirements": ["python-kasa[speedups]==0.10.1"]
+ "requirements": ["python-kasa[speedups]==0.10.2"]
}
diff --git a/homeassistant/components/tplink/number.py b/homeassistant/components/tplink/number.py
index a9d002c0083..252c4888d26 100644
--- a/homeassistant/components/tplink/number.py
+++ b/homeassistant/components/tplink/number.py
@@ -15,7 +15,7 @@ from homeassistant.components.number import (
NumberMode,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .entity import (
@@ -81,7 +81,7 @@ NUMBER_DESCRIPTIONS_MAP = {desc.key: desc for desc in NUMBER_DESCRIPTIONS}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number entities."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/select.py b/homeassistant/components/tplink/select.py
index 8e9dee7b964..72042f571e6 100644
--- a/homeassistant/components/tplink/select.py
+++ b/homeassistant/components/tplink/select.py
@@ -13,7 +13,7 @@ from homeassistant.components.select import (
SelectEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .entity import (
@@ -53,7 +53,7 @@ SELECT_DESCRIPTIONS_MAP = {desc.key: desc for desc in SELECT_DESCRIPTIONS}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up select entities."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/sensor.py b/homeassistant/components/tplink/sensor.py
index 9b21ba775a9..cc35b1fd142 100644
--- a/homeassistant/components/tplink/sensor.py
+++ b/homeassistant/components/tplink/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import UnitOfTime
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .const import UNIT_MAPPING
@@ -271,7 +271,7 @@ SENSOR_DESCRIPTIONS_MAP = {desc.key: desc for desc in SENSOR_DESCRIPTIONS}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/siren.py b/homeassistant/components/tplink/siren.py
index 027fa2dd58f..65cb722052f 100644
--- a/homeassistant/components/tplink/siren.py
+++ b/homeassistant/components/tplink/siren.py
@@ -21,7 +21,7 @@ from homeassistant.components.siren import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry, legacy_device_id
from .const import DOMAIN
@@ -61,7 +61,7 @@ SIREN_DESCRIPTIONS: tuple[TPLinkSirenEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up siren entities."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/switch.py b/homeassistant/components/tplink/switch.py
index f08753def26..3cb20d63cd7 100644
--- a/homeassistant/components/tplink/switch.py
+++ b/homeassistant/components/tplink/switch.py
@@ -14,7 +14,7 @@ from homeassistant.components.switch import (
SwitchEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .entity import (
@@ -85,7 +85,7 @@ SWITCH_DESCRIPTIONS_MAP = {desc.key: desc for desc in SWITCH_DESCRIPTIONS}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink/vacuum.py b/homeassistant/components/tplink/vacuum.py
index c62cd1d27c8..e948e778be4 100644
--- a/homeassistant/components/tplink/vacuum.py
+++ b/homeassistant/components/tplink/vacuum.py
@@ -16,7 +16,7 @@ from homeassistant.components.vacuum import (
VacuumEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TPLinkConfigEntry
from .coordinator import TPLinkDataUpdateCoordinator
@@ -63,7 +63,7 @@ VACUUM_DESCRIPTIONS: tuple[TPLinkVacuumEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TPLinkConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up vacuum entities."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/tplink_omada/__init__.py b/homeassistant/components/tplink_omada/__init__.py
index 2d33a890510..7ea7fd95fef 100644
--- a/homeassistant/components/tplink_omada/__init__.py
+++ b/homeassistant/components/tplink_omada/__init__.py
@@ -11,7 +11,7 @@ from tplink_omada_client.exceptions import (
UnsupportedControllerVersion,
)
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -55,7 +55,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: OmadaConfigEntry) -> boo
) from ex
site_client = await client.get_site_client(OmadaSite("", entry.data[CONF_SITE]))
- controller = OmadaSiteController(hass, site_client)
+ controller = OmadaSiteController(hass, entry, site_client)
await controller.initialize_first_refresh()
entry.runtime_data = controller
@@ -80,12 +80,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: OmadaConfigEntry) -> boo
async def async_unload_entry(hass: HomeAssistant, entry: OmadaConfigEntry) -> bool:
"""Unload a config entry."""
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
# This is the last loaded instance of Omada, deregister any services
hass.services.async_remove(DOMAIN, "reconnect_client")
diff --git a/homeassistant/components/tplink_omada/binary_sensor.py b/homeassistant/components/tplink_omada/binary_sensor.py
index 73d5f54b8b3..fb179634fd1 100644
--- a/homeassistant/components/tplink_omada/binary_sensor.py
+++ b/homeassistant/components/tplink_omada/binary_sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OmadaConfigEntry
from .controller import OmadaGatewayCoordinator
@@ -28,7 +28,7 @@ from .entity import OmadaDeviceEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OmadaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/tplink_omada/controller.py b/homeassistant/components/tplink_omada/controller.py
index 658286981f9..60a07f76b23 100644
--- a/homeassistant/components/tplink_omada/controller.py
+++ b/homeassistant/components/tplink_omada/controller.py
@@ -1,10 +1,17 @@
"""Controller for sharing Omada API coordinators between platforms."""
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
from tplink_omada_client import OmadaSiteClient
from tplink_omada_client.devices import OmadaSwitch
from homeassistant.core import HomeAssistant
+if TYPE_CHECKING:
+ from . import OmadaConfigEntry
+
from .coordinator import (
OmadaClientsCoordinator,
OmadaDevicesCoordinator,
@@ -21,15 +28,21 @@ class OmadaSiteController:
def __init__(
self,
hass: HomeAssistant,
+ config_entry: OmadaConfigEntry,
omada_client: OmadaSiteClient,
) -> None:
"""Create the controller."""
self._hass = hass
+ self._config_entry = config_entry
self._omada_client = omada_client
self._switch_port_coordinators: dict[str, OmadaSwitchPortCoordinator] = {}
- self._devices_coordinator = OmadaDevicesCoordinator(hass, omada_client)
- self._clients_coordinator = OmadaClientsCoordinator(hass, omada_client)
+ self._devices_coordinator = OmadaDevicesCoordinator(
+ hass, config_entry, omada_client
+ )
+ self._clients_coordinator = OmadaClientsCoordinator(
+ hass, config_entry, omada_client
+ )
async def initialize_first_refresh(self) -> None:
"""Initialize the all coordinators, and perform first refresh."""
@@ -39,7 +52,7 @@ class OmadaSiteController:
gateway = next((d for d in devices if d.type == "gateway"), None)
if gateway:
self._gateway_coordinator = OmadaGatewayCoordinator(
- self._hass, self._omada_client, gateway.mac
+ self._hass, self._config_entry, self._omada_client, gateway.mac
)
await self._gateway_coordinator.async_config_entry_first_refresh()
@@ -56,7 +69,7 @@ class OmadaSiteController:
"""Get coordinator for network port information of a given switch."""
if switch.mac not in self._switch_port_coordinators:
self._switch_port_coordinators[switch.mac] = OmadaSwitchPortCoordinator(
- self._hass, self._omada_client, switch
+ self._hass, self._config_entry, self._omada_client, switch
)
return self._switch_port_coordinators[switch.mac]
diff --git a/homeassistant/components/tplink_omada/coordinator.py b/homeassistant/components/tplink_omada/coordinator.py
index a80bedeb65e..1552b568297 100644
--- a/homeassistant/components/tplink_omada/coordinator.py
+++ b/homeassistant/components/tplink_omada/coordinator.py
@@ -1,8 +1,11 @@
"""Generic Omada API coordinator."""
+from __future__ import annotations
+
import asyncio
from datetime import timedelta
import logging
+from typing import TYPE_CHECKING
from tplink_omada_client import OmadaSiteClient, OmadaSwitchPortDetails
from tplink_omada_client.clients import OmadaWirelessClient
@@ -12,6 +15,9 @@ from tplink_omada_client.exceptions import OmadaClientException
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+if TYPE_CHECKING:
+ from . import OmadaConfigEntry
+
_LOGGER = logging.getLogger(__name__)
POLL_SWITCH_PORT = 300
@@ -23,9 +29,12 @@ POLL_DEVICES = 300
class OmadaCoordinator[_T](DataUpdateCoordinator[dict[str, _T]]):
"""Coordinator for synchronizing bulk Omada data."""
+ config_entry: OmadaConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: OmadaConfigEntry,
omada_client: OmadaSiteClient,
name: str,
poll_delay: int | None = 300,
@@ -34,6 +43,7 @@ class OmadaCoordinator[_T](DataUpdateCoordinator[dict[str, _T]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=f"Omada API Data - {name}",
update_interval=timedelta(seconds=poll_delay) if poll_delay else None,
)
@@ -58,12 +68,17 @@ class OmadaSwitchPortCoordinator(OmadaCoordinator[OmadaSwitchPortDetails]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: OmadaConfigEntry,
omada_client: OmadaSiteClient,
network_switch: OmadaSwitch,
) -> None:
"""Initialize my coordinator."""
super().__init__(
- hass, omada_client, f"{network_switch.name} Ports", POLL_SWITCH_PORT
+ hass,
+ config_entry,
+ omada_client,
+ f"{network_switch.name} Ports",
+ POLL_SWITCH_PORT,
)
self._network_switch = network_switch
@@ -79,11 +94,12 @@ class OmadaGatewayCoordinator(OmadaCoordinator[OmadaGateway]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: OmadaConfigEntry,
omada_client: OmadaSiteClient,
mac: str,
) -> None:
"""Initialize my coordinator."""
- super().__init__(hass, omada_client, "Gateway", POLL_GATEWAY)
+ super().__init__(hass, config_entry, omada_client, "Gateway", POLL_GATEWAY)
self.mac = mac
async def poll_update(self) -> dict[str, OmadaGateway]:
@@ -98,10 +114,11 @@ class OmadaDevicesCoordinator(OmadaCoordinator[OmadaListDevice]):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: OmadaConfigEntry,
omada_client: OmadaSiteClient,
) -> None:
"""Initialize my coordinator."""
- super().__init__(hass, omada_client, "DeviceList", POLL_CLIENTS)
+ super().__init__(hass, config_entry, omada_client, "DeviceList", POLL_CLIENTS)
async def poll_update(self) -> dict[str, OmadaListDevice]:
"""Poll the site's current registered Omada devices."""
@@ -111,9 +128,14 @@ class OmadaDevicesCoordinator(OmadaCoordinator[OmadaListDevice]):
class OmadaClientsCoordinator(OmadaCoordinator[OmadaWirelessClient]):
"""Coordinator for getting details about the site's connected clients."""
- def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: OmadaConfigEntry,
+ omada_client: OmadaSiteClient,
+ ) -> None:
"""Initialize my coordinator."""
- super().__init__(hass, omada_client, "ClientsList", POLL_CLIENTS)
+ super().__init__(hass, config_entry, omada_client, "ClientsList", POLL_CLIENTS)
async def poll_update(self) -> dict[str, OmadaWirelessClient]:
"""Poll the site's current active wi-fi clients."""
diff --git a/homeassistant/components/tplink_omada/device_tracker.py b/homeassistant/components/tplink_omada/device_tracker.py
index fe78adf8847..ce1c8ba40e1 100644
--- a/homeassistant/components/tplink_omada/device_tracker.py
+++ b/homeassistant/components/tplink_omada/device_tracker.py
@@ -6,7 +6,7 @@ from tplink_omada_client.clients import OmadaWirelessClient
from homeassistant.components.device_tracker import ScannerEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import OmadaConfigEntry
@@ -19,7 +19,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OmadaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device trackers and scanners."""
diff --git a/homeassistant/components/tplink_omada/manifest.json b/homeassistant/components/tplink_omada/manifest.json
index af20b54675b..274f2815330 100644
--- a/homeassistant/components/tplink_omada/manifest.json
+++ b/homeassistant/components/tplink_omada/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/tplink_omada",
"integration_type": "hub",
"iot_class": "local_polling",
- "requirements": ["tplink-omada-client==1.4.3"]
+ "requirements": ["tplink-omada-client==1.4.4"]
}
diff --git a/homeassistant/components/tplink_omada/sensor.py b/homeassistant/components/tplink_omada/sensor.py
index 272334d1b52..b41f3da2f33 100644
--- a/homeassistant/components/tplink_omada/sensor.py
+++ b/homeassistant/components/tplink_omada/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import OmadaConfigEntry
@@ -57,7 +57,7 @@ def _map_device_status(device: OmadaListDevice) -> str | None:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OmadaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/tplink_omada/switch.py b/homeassistant/components/tplink_omada/switch.py
index f99d8aaedde..37c73a9e11f 100644
--- a/homeassistant/components/tplink_omada/switch.py
+++ b/homeassistant/components/tplink_omada/switch.py
@@ -22,7 +22,7 @@ from tplink_omada_client.omadasiteclient import GatewayPortSettings
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OmadaConfigEntry
from .controller import OmadaGatewayCoordinator, OmadaSwitchPortCoordinator
@@ -37,7 +37,7 @@ TCoordinator = TypeVar("TCoordinator", bound="OmadaCoordinator[Any]")
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OmadaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/tplink_omada/update.py b/homeassistant/components/tplink_omada/update.py
index 54b586794be..8a8531c10b6 100644
--- a/homeassistant/components/tplink_omada/update.py
+++ b/homeassistant/components/tplink_omada/update.py
@@ -16,7 +16,7 @@ from homeassistant.components.update import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import OmadaConfigEntry
from .coordinator import POLL_DEVICES, OmadaCoordinator, OmadaDevicesCoordinator
@@ -43,7 +43,9 @@ class OmadaFirmwareUpdateCoordinator(OmadaCoordinator[FirmwareUpdateStatus]): #
devices_coordinator: OmadaDevicesCoordinator,
) -> None:
"""Initialize my coordinator."""
- super().__init__(hass, omada_client, "Firmware Updates", poll_delay=None)
+ super().__init__(
+ hass, config_entry, omada_client, "Firmware Updates", poll_delay=None
+ )
self._devices_coordinator = devices_coordinator
self._config_entry = config_entry
@@ -91,7 +93,7 @@ class OmadaFirmwareUpdateCoordinator(OmadaCoordinator[FirmwareUpdateStatus]): #
async def async_setup_entry(
hass: HomeAssistant,
config_entry: OmadaConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches."""
controller = config_entry.runtime_data
diff --git a/homeassistant/components/traccar/device_tracker.py b/homeassistant/components/traccar/device_tracker.py
index 0fa7fc344ea..43210ee92ea 100644
--- a/homeassistant/components/traccar/device_tracker.py
+++ b/homeassistant/components/traccar/device_tracker.py
@@ -11,7 +11,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from . import DOMAIN, TRACKER_UPDATE
@@ -69,7 +69,9 @@ EVENTS = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure a dispatcher connection based on a config entry."""
diff --git a/homeassistant/components/traccar_server/__init__.py b/homeassistant/components/traccar_server/__init__.py
index c7a65d2d4a8..44aeedc3376 100644
--- a/homeassistant/components/traccar_server/__init__.py
+++ b/homeassistant/components/traccar_server/__init__.py
@@ -21,13 +21,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_create_clientsession
from homeassistant.helpers.event import async_track_time_interval
-from .const import (
- CONF_CUSTOM_ATTRIBUTES,
- CONF_EVENTS,
- CONF_MAX_ACCURACY,
- CONF_SKIP_ACCURACY_FILTER_FOR,
- DOMAIN,
-)
+from .const import CONF_EVENTS, DOMAIN
from .coordinator import TraccarServerCoordinator
PLATFORMS: list[Platform] = [
@@ -47,6 +41,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
)
coordinator = TraccarServerCoordinator(
hass=hass,
+ config_entry=entry,
client=ApiClient(
client_session=client_session,
host=entry.data[CONF_HOST],
@@ -56,10 +51,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
ssl=entry.data[CONF_SSL],
verify_ssl=entry.data[CONF_VERIFY_SSL],
),
- events=entry.options.get(CONF_EVENTS, []),
- max_accuracy=entry.options.get(CONF_MAX_ACCURACY, 0.0),
- skip_accuracy_filter_for=entry.options.get(CONF_SKIP_ACCURACY_FILTER_FOR, []),
- custom_attributes=entry.options.get(CONF_CUSTOM_ATTRIBUTES, []),
)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/traccar_server/binary_sensor.py b/homeassistant/components/traccar_server/binary_sensor.py
index 58c46502b53..6d81ba84ed4 100644
--- a/homeassistant/components/traccar_server/binary_sensor.py
+++ b/homeassistant/components/traccar_server/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import TraccarServerCoordinator
@@ -55,7 +55,7 @@ TRACCAR_SERVER_BINARY_SENSOR_ENTITY_DESCRIPTIONS: tuple[
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor entities."""
coordinator: TraccarServerCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/traccar_server/coordinator.py b/homeassistant/components/traccar_server/coordinator.py
index 95ce42469f1..2c878856cc2 100644
--- a/homeassistant/components/traccar_server/coordinator.py
+++ b/homeassistant/components/traccar_server/coordinator.py
@@ -22,7 +22,15 @@ from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt as dt_util
-from .const import DOMAIN, EVENTS, LOGGER
+from .const import (
+ CONF_CUSTOM_ATTRIBUTES,
+ CONF_EVENTS,
+ CONF_MAX_ACCURACY,
+ CONF_SKIP_ACCURACY_FILTER_FOR,
+ DOMAIN,
+ EVENTS,
+ LOGGER,
+)
from .helpers import get_device, get_first_geofence
@@ -46,25 +54,24 @@ class TraccarServerCoordinator(DataUpdateCoordinator[TraccarServerCoordinatorDat
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
client: ApiClient,
- *,
- events: list[str],
- max_accuracy: float,
- skip_accuracy_filter_for: list[str],
- custom_attributes: list[str],
) -> None:
"""Initialize global Traccar Server data updater."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=None,
)
self.client = client
- self.custom_attributes = custom_attributes
- self.events = events
- self.max_accuracy = max_accuracy
- self.skip_accuracy_filter_for = skip_accuracy_filter_for
+ self.custom_attributes = config_entry.options.get(CONF_CUSTOM_ATTRIBUTES, [])
+ self.events = config_entry.options.get(CONF_EVENTS, [])
+ self.max_accuracy = config_entry.options.get(CONF_MAX_ACCURACY, 0.0)
+ self.skip_accuracy_filter_for = config_entry.options.get(
+ CONF_SKIP_ACCURACY_FILTER_FOR, []
+ )
self._geofences: list[GeofenceModel] = []
self._last_event_import: datetime | None = None
self._should_log_subscription_error: bool = True
diff --git a/homeassistant/components/traccar_server/device_tracker.py b/homeassistant/components/traccar_server/device_tracker.py
index 9e5a3c0ee9f..7f2a6dd7c40 100644
--- a/homeassistant/components/traccar_server/device_tracker.py
+++ b/homeassistant/components/traccar_server/device_tracker.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_CATEGORY, ATTR_TRACCAR_ID, ATTR_TRACKER, DOMAIN
from .coordinator import TraccarServerCoordinator
@@ -17,7 +17,7 @@ from .entity import TraccarServerEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker entities."""
coordinator: TraccarServerCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/traccar_server/sensor.py b/homeassistant/components/traccar_server/sensor.py
index bb3c4ed4401..9aee6f28489 100644
--- a/homeassistant/components/traccar_server/sensor.py
+++ b/homeassistant/components/traccar_server/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfLength, UnitOfSpeed
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN
@@ -83,7 +83,7 @@ TRACCAR_SERVER_SENSOR_ENTITY_DESCRIPTIONS: tuple[
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensor entities."""
coordinator: TraccarServerCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/tractive/binary_sensor.py b/homeassistant/components/tractive/binary_sensor.py
index 80219154d81..2978d369344 100644
--- a/homeassistant/components/tractive/binary_sensor.py
+++ b/homeassistant/components/tractive/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import ATTR_BATTERY_CHARGING, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Trackables, TractiveClient, TractiveConfigEntry
from .const import TRACKER_HARDWARE_STATUS_UPDATED
@@ -58,7 +58,7 @@ SENSOR_TYPE = BinarySensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: TractiveConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tractive device trackers."""
client = entry.runtime_data.client
diff --git a/homeassistant/components/tractive/device_tracker.py b/homeassistant/components/tractive/device_tracker.py
index f31afaf92f6..73be7216a2f 100644
--- a/homeassistant/components/tractive/device_tracker.py
+++ b/homeassistant/components/tractive/device_tracker.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.device_tracker import SourceType, TrackerEntity
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Trackables, TractiveClient, TractiveConfigEntry
from .const import (
@@ -21,7 +21,7 @@ from .entity import TractiveEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: TractiveConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tractive device trackers."""
client = entry.runtime_data.client
diff --git a/homeassistant/components/tractive/sensor.py b/homeassistant/components/tractive/sensor.py
index a3c1893267c..18d7e4c23ab 100644
--- a/homeassistant/components/tractive/sensor.py
+++ b/homeassistant/components/tractive/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import Trackables, TractiveClient, TractiveConfigEntry
@@ -182,7 +182,7 @@ SENSOR_TYPES: tuple[TractiveSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TractiveConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tractive device trackers."""
client = entry.runtime_data.client
diff --git a/homeassistant/components/tractive/switch.py b/homeassistant/components/tractive/switch.py
index 3bf6887e99c..da2c8e35ff7 100644
--- a/homeassistant/components/tractive/switch.py
+++ b/homeassistant/components/tractive/switch.py
@@ -11,7 +11,7 @@ from aiotractive.exceptions import TractiveError
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Trackables, TractiveClient, TractiveConfigEntry
from .const import (
@@ -57,7 +57,7 @@ SWITCH_TYPES: tuple[TractiveSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TractiveConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tractive switches."""
client = entry.runtime_data.client
diff --git a/homeassistant/components/tradfri/__init__.py b/homeassistant/components/tradfri/__init__.py
index 92ed2ea8b82..c3e8938b244 100644
--- a/homeassistant/components/tradfri/__init__.py
+++ b/homeassistant/components/tradfri/__init__.py
@@ -106,7 +106,7 @@ async def async_setup_entry(
for device in devices:
coordinator = TradfriDeviceDataUpdateCoordinator(
- hass=hass, api=api, device=device
+ hass=hass, config_entry=entry, api=api, device=device
)
await coordinator.async_config_entry_first_refresh()
@@ -159,7 +159,7 @@ def remove_stale_devices(
device_entries = dr.async_entries_for_config_entry(
device_registry, config_entry.entry_id
)
- all_device_ids = {device.id for device in devices}
+ all_device_ids = {str(device.id) for device in devices}
for device_entry in device_entries:
device_id: str | None = None
@@ -176,7 +176,7 @@ def remove_stale_devices(
gateway_id = _id
break
- device_id = _id
+ device_id = _id.replace(f"{config_entry.data[CONF_GATEWAY_ID]}-", "")
break
if gateway_id is not None:
@@ -190,3 +190,93 @@ def remove_stale_devices(
device_registry.async_update_device(
device_entry.id, remove_config_entry_id=config_entry.entry_id
)
+
+
+async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+ """Migrate old entry."""
+ LOGGER.debug(
+ "Migrating Tradfri configuration from version %s.%s",
+ config_entry.version,
+ config_entry.minor_version,
+ )
+
+ if config_entry.version > 1:
+ # This means the user has downgraded from a future version
+ return False
+
+ if config_entry.version == 1:
+ # Migrate to version 2
+ migrate_config_entry_and_identifiers(hass, config_entry)
+
+ hass.config_entries.async_update_entry(config_entry, version=2)
+
+ LOGGER.debug(
+ "Migration to Tradfri configuration version %s.%s successful",
+ config_entry.version,
+ config_entry.minor_version,
+ )
+
+ return True
+
+
+def migrate_config_entry_and_identifiers(
+ hass: HomeAssistant, config_entry: ConfigEntry
+) -> None:
+ """Migrate old non-unique identifiers to new unique identifiers."""
+
+ related_device_flag: bool
+ device_id: str
+
+ device_reg = dr.async_get(hass)
+ # Get all devices associated to contextual gateway config_entry
+ # and loop through list of devices.
+ for device in dr.async_entries_for_config_entry(device_reg, config_entry.entry_id):
+ related_device_flag = False
+ for identifier in device.identifiers:
+ if identifier[0] != DOMAIN:
+ continue
+
+ related_device_flag = True
+
+ _id = identifier[1]
+
+ # Identify gateway device.
+ if _id == config_entry.data[CONF_GATEWAY_ID]:
+ # Using this to avoid updating gateway's own device registry entry
+ related_device_flag = False
+ break
+
+ device_id = str(_id)
+ break
+
+ # Check that device is related to tradfri domain (and is not the gateway itself)
+ if not related_device_flag:
+ continue
+
+ # Loop through list of config_entry_ids for device
+ config_entry_ids = device.config_entries
+ for config_entry_id in config_entry_ids:
+ # Check that the config entry in list is not the device's primary config entry
+ if config_entry_id == device.primary_config_entry:
+ continue
+
+ # Check that the 'other' config entry is also a tradfri config entry
+ other_entry = hass.config_entries.async_get_entry(config_entry_id)
+
+ if other_entry is None or other_entry.domain != DOMAIN:
+ continue
+
+ # Remove non-primary 'tradfri' config entry from device's config_entry_ids
+ device_reg.async_update_device(
+ device.id, remove_config_entry_id=config_entry_id
+ )
+
+ if config_entry.data[CONF_GATEWAY_ID] in device_id:
+ continue
+
+ device_reg.async_update_device(
+ device.id,
+ new_identifiers={
+ (DOMAIN, f"{config_entry.data[CONF_GATEWAY_ID]}-{device_id}")
+ },
+ )
diff --git a/homeassistant/components/tradfri/config_flow.py b/homeassistant/components/tradfri/config_flow.py
index 29d876346a7..9f5b39a9657 100644
--- a/homeassistant/components/tradfri/config_flow.py
+++ b/homeassistant/components/tradfri/config_flow.py
@@ -35,7 +35,7 @@ class AuthError(Exception):
class FlowHandler(ConfigFlow, domain=DOMAIN):
"""Handle a config flow."""
- VERSION = 1
+ VERSION = 2
def __init__(self) -> None:
"""Initialize flow."""
diff --git a/homeassistant/components/tradfri/coordinator.py b/homeassistant/components/tradfri/coordinator.py
index 5246545ae65..4c5c186626e 100644
--- a/homeassistant/components/tradfri/coordinator.py
+++ b/homeassistant/components/tradfri/coordinator.py
@@ -10,6 +10,7 @@ from pytradfri.command import Command
from pytradfri.device import Device
from pytradfri.error import RequestError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -21,10 +22,12 @@ SCAN_INTERVAL = 60 # Interval for updating the coordinator
class TradfriDeviceDataUpdateCoordinator(DataUpdateCoordinator[Device]):
"""Coordinator to manage data for a specific Tradfri device."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- *,
+ config_entry: ConfigEntry,
api: Callable[[Command | list[Command]], Any],
device: Device,
) -> None:
@@ -36,6 +39,7 @@ class TradfriDeviceDataUpdateCoordinator(DataUpdateCoordinator[Device]):
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=f"Update coordinator for {device}",
update_interval=timedelta(seconds=SCAN_INTERVAL),
)
diff --git a/homeassistant/components/tradfri/cover.py b/homeassistant/components/tradfri/cover.py
index 92d10320327..b1fb9b153ad 100644
--- a/homeassistant/components/tradfri/cover.py
+++ b/homeassistant/components/tradfri/cover.py
@@ -10,7 +10,7 @@ from pytradfri.command import Command
from homeassistant.components.cover import ATTR_POSITION, CoverEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_GATEWAY_ID, COORDINATOR, COORDINATOR_LIST, DOMAIN, KEY_API
from .coordinator import TradfriDeviceDataUpdateCoordinator
@@ -20,7 +20,7 @@ from .entity import TradfriBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Tradfri covers based on a config entry."""
gateway_id = config_entry.data[CONF_GATEWAY_ID]
diff --git a/homeassistant/components/tradfri/entity.py b/homeassistant/components/tradfri/entity.py
index b06d0081477..41c20b19de5 100644
--- a/homeassistant/components/tradfri/entity.py
+++ b/homeassistant/components/tradfri/entity.py
@@ -58,7 +58,7 @@ class TradfriBaseEntity(CoordinatorEntity[TradfriDeviceDataUpdateCoordinator]):
info = self._device.device_info
self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, self._device_id)},
+ identifiers={(DOMAIN, f"{gateway_id}-{self._device_id}")},
manufacturer=info.manufacturer,
model=info.model_number,
name=self._device.name,
diff --git a/homeassistant/components/tradfri/fan.py b/homeassistant/components/tradfri/fan.py
index 3f45ee3e1eb..e8fb7c050ed 100644
--- a/homeassistant/components/tradfri/fan.py
+++ b/homeassistant/components/tradfri/fan.py
@@ -10,7 +10,7 @@ from pytradfri.command import Command
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_GATEWAY_ID, COORDINATOR, COORDINATOR_LIST, DOMAIN, KEY_API
from .coordinator import TradfriDeviceDataUpdateCoordinator
@@ -33,7 +33,7 @@ def _from_fan_speed(fan_speed: int) -> int:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Tradfri switches based on a config entry."""
gateway_id = config_entry.data[CONF_GATEWAY_ID]
diff --git a/homeassistant/components/tradfri/light.py b/homeassistant/components/tradfri/light.py
index e464d1a8142..b945c7f2bec 100644
--- a/homeassistant/components/tradfri/light.py
+++ b/homeassistant/components/tradfri/light.py
@@ -19,7 +19,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .const import CONF_GATEWAY_ID, COORDINATOR, COORDINATOR_LIST, DOMAIN, KEY_API
@@ -30,7 +30,7 @@ from .entity import TradfriBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Tradfri lights based on a config entry."""
gateway_id = config_entry.data[CONF_GATEWAY_ID]
diff --git a/homeassistant/components/tradfri/sensor.py b/homeassistant/components/tradfri/sensor.py
index 4e560f0e7b5..b4a7c335481 100644
--- a/homeassistant/components/tradfri/sensor.py
+++ b/homeassistant/components/tradfri/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_GATEWAY_ID,
@@ -128,7 +128,7 @@ def _migrate_old_unique_ids(hass: HomeAssistant, old_unique_id: str, key: str) -
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Tradfri config entry."""
gateway_id = config_entry.data[CONF_GATEWAY_ID]
diff --git a/homeassistant/components/tradfri/switch.py b/homeassistant/components/tradfri/switch.py
index 088b775b9fd..a2a1a5b4623 100644
--- a/homeassistant/components/tradfri/switch.py
+++ b/homeassistant/components/tradfri/switch.py
@@ -10,7 +10,7 @@ from pytradfri.command import Command
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_GATEWAY_ID, COORDINATOR, COORDINATOR_LIST, DOMAIN, KEY_API
from .coordinator import TradfriDeviceDataUpdateCoordinator
@@ -20,7 +20,7 @@ from .entity import TradfriBaseEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Load Tradfri switches based on a config entry."""
gateway_id = config_entry.data[CONF_GATEWAY_ID]
diff --git a/homeassistant/components/trafikverket_camera/binary_sensor.py b/homeassistant/components/trafikverket_camera/binary_sensor.py
index b367fa0fb45..92112b41466 100644
--- a/homeassistant/components/trafikverket_camera/binary_sensor.py
+++ b/homeassistant/components/trafikverket_camera/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TVCameraConfigEntry
from .coordinator import CameraData
@@ -36,7 +36,7 @@ BINARY_SENSOR_TYPE = TVCameraSensorEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: TVCameraConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Trafikverket Camera binary sensor platform."""
diff --git a/homeassistant/components/trafikverket_camera/camera.py b/homeassistant/components/trafikverket_camera/camera.py
index ece02cacf70..b4eddb0890f 100644
--- a/homeassistant/components/trafikverket_camera/camera.py
+++ b/homeassistant/components/trafikverket_camera/camera.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.camera import Camera
from homeassistant.const import ATTR_LOCATION
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TVCameraConfigEntry
from .const import ATTR_DESCRIPTION, ATTR_TYPE
@@ -21,7 +21,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: TVCameraConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Trafikverket Camera."""
diff --git a/homeassistant/components/trafikverket_camera/sensor.py b/homeassistant/components/trafikverket_camera/sensor.py
index cb5c458f742..726fcb6f901 100644
--- a/homeassistant/components/trafikverket_camera/sensor.py
+++ b/homeassistant/components/trafikverket_camera/sensor.py
@@ -13,7 +13,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import DEGREE
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TVCameraConfigEntry
@@ -74,7 +74,7 @@ SENSOR_TYPES: tuple[TVCameraSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TVCameraConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Trafikverket Camera sensor platform."""
diff --git a/homeassistant/components/trafikverket_ferry/sensor.py b/homeassistant/components/trafikverket_ferry/sensor.py
index 44176ab82b7..b908bc5f550 100644
--- a/homeassistant/components/trafikverket_ferry/sensor.py
+++ b/homeassistant/components/trafikverket_ferry/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.dt import as_utc
@@ -92,7 +92,7 @@ SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TVFerryConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Trafikverket sensor entry."""
diff --git a/homeassistant/components/trafikverket_train/config_flow.py b/homeassistant/components/trafikverket_train/config_flow.py
index 57d74eef78a..f6a58e464a1 100644
--- a/homeassistant/components/trafikverket_train/config_flow.py
+++ b/homeassistant/components/trafikverket_train/config_flow.py
@@ -101,6 +101,9 @@ class TVTrainConfigFlow(ConfigFlow, domain=DOMAIN):
_from_stations: list[StationInfoModel]
_to_stations: list[StationInfoModel]
+ _time: str | None
+ _days: list
+ _product: str | None
_data: dict[str, Any]
@staticmethod
@@ -243,8 +246,10 @@ class TVTrainConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle the select station step."""
if user_input is not None:
api_key: str = self._data[CONF_API_KEY]
- train_from: str = user_input[CONF_FROM]
- train_to: str = user_input[CONF_TO]
+ train_from: str = (
+ user_input.get(CONF_FROM) or self._from_stations[0].signature
+ )
+ train_to: str = user_input.get(CONF_TO) or self._to_stations[0].signature
train_time: str | None = self._data.get(CONF_TIME)
train_days: list = self._data[CONF_WEEKDAY]
filter_product: str | None = self._data[CONF_FILTER_PRODUCT]
diff --git a/homeassistant/components/trafikverket_train/sensor.py b/homeassistant/components/trafikverket_train/sensor.py
index a4de8c1ef26..150b5ee7abb 100644
--- a/homeassistant/components/trafikverket_train/sensor.py
+++ b/homeassistant/components/trafikverket_train/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.const import CONF_NAME, UnitOfTime
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -110,7 +110,7 @@ SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TVTrainConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Trafikverket sensor entry."""
diff --git a/homeassistant/components/trafikverket_weatherstation/sensor.py b/homeassistant/components/trafikverket_weatherstation/sensor.py
index bc17c82748a..cb923037a24 100644
--- a/homeassistant/components/trafikverket_weatherstation/sensor.py
+++ b/homeassistant/components/trafikverket_weatherstation/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -204,7 +204,7 @@ SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TVWeatherConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Trafikverket sensor entry."""
diff --git a/homeassistant/components/transmission/__init__.py b/homeassistant/components/transmission/__init__.py
index 1a8ffdea0c2..6d23017ab75 100644
--- a/homeassistant/components/transmission/__init__.py
+++ b/homeassistant/components/transmission/__init__.py
@@ -15,7 +15,7 @@ from transmission_rpc.error import (
)
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import (
CONF_HOST,
CONF_ID,
@@ -54,7 +54,7 @@ from .const import (
SERVICE_START_TORRENT,
SERVICE_STOP_TORRENT,
)
-from .coordinator import TransmissionDataUpdateCoordinator
+from .coordinator import TransmissionConfigEntry, TransmissionDataUpdateCoordinator
from .errors import AuthenticationError, CannotConnect, UnknownError
_LOGGER = logging.getLogger(__name__)
@@ -78,7 +78,9 @@ MIGRATION_NAME_TO_KEY = {
SERVICE_BASE_SCHEMA = vol.Schema(
{
- vol.Required(CONF_ENTRY_ID): selector.ConfigEntrySelector(),
+ vol.Required(CONF_ENTRY_ID): selector.ConfigEntrySelector(
+ {"integration": DOMAIN}
+ ),
}
)
@@ -115,8 +117,6 @@ SERVICE_STOP_TORRENT_SCHEMA = vol.All(
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
-type TransmissionConfigEntry = ConfigEntry[TransmissionDataUpdateCoordinator]
-
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the Transmission component."""
@@ -165,12 +165,16 @@ async def async_setup_entry(
return True
-async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_unload_entry(
+ hass: HomeAssistant, config_entry: TransmissionConfigEntry
+) -> bool:
"""Unload Transmission Entry from config_entry."""
return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
-async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+ hass: HomeAssistant, config_entry: TransmissionConfigEntry
+) -> bool:
"""Migrate an old config entry."""
_LOGGER.debug(
"Migrating from version %s.%s",
diff --git a/homeassistant/components/transmission/coordinator.py b/homeassistant/components/transmission/coordinator.py
index b998ab6fbdd..afe2660e711 100644
--- a/homeassistant/components/transmission/coordinator.py
+++ b/homeassistant/components/transmission/coordinator.py
@@ -27,17 +27,21 @@ from .const import (
_LOGGER = logging.getLogger(__name__)
+type TransmissionConfigEntry = ConfigEntry[TransmissionDataUpdateCoordinator]
+
class TransmissionDataUpdateCoordinator(DataUpdateCoordinator[SessionStats]):
"""Transmission dataupdate coordinator class."""
- config_entry: ConfigEntry
+ config_entry: TransmissionConfigEntry
def __init__(
- self, hass: HomeAssistant, entry: ConfigEntry, api: transmission_rpc.Client
+ self,
+ hass: HomeAssistant,
+ entry: TransmissionConfigEntry,
+ api: transmission_rpc.Client,
) -> None:
"""Initialize the Transmission RPC API."""
- self.config_entry = entry
self.api = api
self.host = entry.data[CONF_HOST]
self._session: transmission_rpc.Session | None = None
@@ -47,6 +51,7 @@ class TransmissionDataUpdateCoordinator(DataUpdateCoordinator[SessionStats]):
self.torrents: list[transmission_rpc.Torrent] = []
super().__init__(
hass,
+ config_entry=entry,
name=f"{DOMAIN} - {self.host}",
logger=_LOGGER,
update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL),
diff --git a/homeassistant/components/transmission/sensor.py b/homeassistant/components/transmission/sensor.py
index 652f5d51fbb..a0babe7464a 100644
--- a/homeassistant/components/transmission/sensor.py
+++ b/homeassistant/components/transmission/sensor.py
@@ -17,11 +17,10 @@ from homeassistant.components.sensor import (
from homeassistant.const import STATE_IDLE, UnitOfDataRate
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import TransmissionConfigEntry
from .const import (
DOMAIN,
STATE_ATTR_TORRENT_INFO,
@@ -30,7 +29,7 @@ from .const import (
STATE_UP_DOWN,
SUPPORTED_ORDER_MODES,
)
-from .coordinator import TransmissionDataUpdateCoordinator
+from .coordinator import TransmissionConfigEntry, TransmissionDataUpdateCoordinator
MODES: dict[str, list[str] | None] = {
"started_torrents": ["downloading"],
@@ -130,7 +129,7 @@ SENSOR_TYPES: tuple[TransmissionSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TransmissionConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Transmission sensors."""
diff --git a/homeassistant/components/transmission/switch.py b/homeassistant/components/transmission/switch.py
index d88f794cb10..9ca8a197344 100644
--- a/homeassistant/components/transmission/switch.py
+++ b/homeassistant/components/transmission/switch.py
@@ -7,12 +7,11 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from . import TransmissionConfigEntry
from .const import DOMAIN
-from .coordinator import TransmissionDataUpdateCoordinator
+from .coordinator import TransmissionConfigEntry, TransmissionDataUpdateCoordinator
@dataclass(frozen=True, kw_only=True)
@@ -45,7 +44,7 @@ SWITCH_TYPES: tuple[TransmissionSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TransmissionConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Transmission switch."""
diff --git a/homeassistant/components/trend/binary_sensor.py b/homeassistant/components/trend/binary_sensor.py
index e5ff5c64a8b..4261f96bbe6 100644
--- a/homeassistant/components/trend/binary_sensor.py
+++ b/homeassistant/components/trend/binary_sensor.py
@@ -35,7 +35,10 @@ from homeassistant.core import Event, EventStateChangedData, HomeAssistant, call
from homeassistant.helpers import config_validation as cv, device_registry as dr
from homeassistant.helpers.device import async_device_info_to_link_from_entity
from homeassistant.helpers.entity import generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers.reload import async_setup_reload_service
from homeassistant.helpers.restore_state import RestoreEntity
@@ -130,7 +133,7 @@ async def async_setup_platform(
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up trend sensor from config entry."""
diff --git a/homeassistant/components/triggercmd/switch.py b/homeassistant/components/triggercmd/switch.py
index 94566fe301d..e04cf5ee7e8 100644
--- a/homeassistant/components/triggercmd/switch.py
+++ b/homeassistant/components/triggercmd/switch.py
@@ -9,7 +9,7 @@ from triggercmd import client, ha
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TriggercmdConfigEntry
from .const import DOMAIN
@@ -20,7 +20,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TriggercmdConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add switch for passed config_entry in HA."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/tts/__init__.py b/homeassistant/components/tts/__init__.py
index 6c7e521f3ef..cb207643471 100644
--- a/homeassistant/components/tts/__init__.py
+++ b/homeassistant/components/tts/__init__.py
@@ -3,9 +3,9 @@
from __future__ import annotations
import asyncio
-from collections.abc import Mapping
+from collections.abc import AsyncGenerator
+from dataclasses import dataclass
from datetime import datetime
-from functools import partial
import hashlib
from http import HTTPStatus
import io
@@ -16,7 +16,8 @@ import re
import secrets
import subprocess
import tempfile
-from typing import Any, Final, TypedDict, final
+from time import monotonic
+from typing import Any, Final
from aiohttp import web
import mutagen
@@ -26,30 +27,24 @@ import voluptuous as vol
from homeassistant.components import ffmpeg, websocket_api
from homeassistant.components.http import HomeAssistantView
-from homeassistant.components.media_player import (
- ATTR_MEDIA_ANNOUNCE,
- ATTR_MEDIA_CONTENT_ID,
- ATTR_MEDIA_CONTENT_TYPE,
- DOMAIN as DOMAIN_MP,
- SERVICE_PLAY_MEDIA,
- MediaType,
-)
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import (
- ATTR_ENTITY_ID,
- PLATFORM_FORMAT,
- STATE_UNAVAILABLE,
- STATE_UNKNOWN,
+from homeassistant.const import EVENT_HOMEASSISTANT_STOP, PLATFORM_FORMAT
+from homeassistant.core import (
+ CALLBACK_TYPE,
+ Event,
+ HassJob,
+ HassJobType,
+ HomeAssistant,
+ ServiceCall,
+ callback,
)
-from homeassistant.core import HassJob, HomeAssistant, ServiceCall, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_component import EntityComponent
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.network import get_url
-from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import UNDEFINED, ConfigType
-from homeassistant.util import dt as dt_util, language as language_util
+from homeassistant.util import language as language_util
from .const import (
ATTR_CACHE,
@@ -67,6 +62,7 @@ from .const import (
DOMAIN,
TtsAudioType,
)
+from .entity import TextToSpeechEntity, TTSAudioRequest
from .helper import get_engine_instance
from .legacy import PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE, Provider, async_setup_legacy
from .media_source import generate_media_source_id, media_source_id_to_kwargs
@@ -83,12 +79,13 @@ __all__ = [
"PLATFORM_SCHEMA",
"PLATFORM_SCHEMA_BASE",
"Provider",
+ "ResultStream",
"SampleFormat",
+ "TextToSpeechEntity",
"TtsAudioType",
"Voice",
"async_default_engine",
"async_get_media_source_audio",
- "async_support_options",
"generate_media_source_id",
]
@@ -126,12 +123,94 @@ KEY_PATTERN = "{0}_{1}_{2}_{3}"
SCHEMA_SERVICE_CLEAR_CACHE = vol.Schema({})
-class TTSCache(TypedDict):
- """Cached TTS file."""
+class TTSCache:
+ """Cached bytes of a TTS result."""
- filename: str
- voice: bytes
- pending: asyncio.Task | None
+ _result_data: bytes | None = None
+ """When fully loaded, contains the result data."""
+
+ _partial_data: list[bytes] | None = None
+ """While loading, contains the data already received from the generator."""
+
+ _loading_error: Exception | None = None
+ """If an error occurred while loading, contains the error."""
+
+ _consumers: list[asyncio.Queue[bytes | None]] | None = None
+ """A queue for each current consumer to notify of new data while the generator is loading."""
+
+ def __init__(
+ self,
+ cache_key: str,
+ extension: str,
+ data_gen: AsyncGenerator[bytes],
+ ) -> None:
+ """Initialize the TTS cache."""
+ self.cache_key = cache_key
+ self.extension = extension
+ self.last_used = monotonic()
+ self._data_gen = data_gen
+
+ async def async_load_data(self) -> bytes:
+ """Load the data from the generator."""
+ if self._result_data is not None or self._partial_data is not None:
+ raise RuntimeError("Data already being loaded")
+
+ self._partial_data = []
+ self._consumers = []
+
+ try:
+ async for chunk in self._data_gen:
+ self._partial_data.append(chunk)
+ for queue in self._consumers:
+ queue.put_nowait(chunk)
+ except Exception as err:
+ self._loading_error = err
+ raise
+ finally:
+ for queue in self._consumers:
+ queue.put_nowait(None)
+ self._consumers = None
+
+ self._result_data = b"".join(self._partial_data)
+ self._partial_data = None
+ return self._result_data
+
+ async def async_stream_data(self) -> AsyncGenerator[bytes]:
+ """Stream the data.
+
+ Will return all data already returned from the generator.
+ Will listen for future data returned from the generator.
+ Raises error if one occurred.
+ """
+ if self._result_data is not None:
+ yield self._result_data
+ return
+ if self._loading_error:
+ raise self._loading_error
+
+ if self._partial_data is None:
+ raise RuntimeError("Data not being loaded")
+
+ queue: asyncio.Queue[bytes | None] | None = None
+ # Check if generator is still feeding data
+ if self._consumers is not None:
+ queue = asyncio.Queue()
+ self._consumers.append(queue)
+
+ for chunk in list(self._partial_data):
+ yield chunk
+
+ if self._loading_error:
+ raise self._loading_error
+
+ if queue is not None:
+ while (chunk2 := await queue.get()) is not None:
+ yield chunk2
+
+ if self._loading_error:
+ raise self._loading_error
+
+ self.last_used = monotonic()
@callback
@@ -169,22 +248,25 @@ def async_resolve_engine(hass: HomeAssistant, engine: str | None) -> str | None:
return async_default_engine(hass)
-async def async_support_options(
+@callback
+def async_create_stream(
hass: HomeAssistant,
engine: str,
language: str | None = None,
options: dict | None = None,
-) -> bool:
- """Return if an engine supports options."""
- if (engine_instance := get_engine_instance(hass, engine)) is None:
- raise HomeAssistantError(f"Provider {engine} not found")
+) -> ResultStream:
+ """Create a streaming URL where the rendered TTS can be retrieved."""
+ return hass.data[DATA_TTS_MANAGER].async_create_result_stream(
+ engine=engine,
+ language=language,
+ options=options,
+ )
- try:
- hass.data[DATA_TTS_MANAGER].process_options(engine_instance, language, options)
- except HomeAssistantError:
- return False
- return True
+@callback
+def async_get_stream(hass: HomeAssistant, token: str) -> ResultStream | None:
+ """Return a result stream given a token."""
+ return hass.data[DATA_TTS_MANAGER].token_to_stream.get(token)
async def async_get_media_source_audio(
@@ -192,9 +274,12 @@ async def async_get_media_source_audio(
media_source_id: str,
) -> tuple[str, bytes]:
"""Get TTS audio as extension, data."""
- return await hass.data[DATA_TTS_MANAGER].async_get_tts_audio(
- **media_source_id_to_kwargs(media_source_id),
+ manager = hass.data[DATA_TTS_MANAGER]
+ cache = manager.async_cache_message_in_memory(
+ **media_source_id_to_kwargs(media_source_id)
)
+ data = b"".join([chunk async for chunk in cache.async_stream_data()])
+ return cache.extension, data
@callback
@@ -213,18 +298,19 @@ def async_get_text_to_speech_languages(hass: HomeAssistant) -> set[str]:
return languages
-async def async_convert_audio(
+async def _async_convert_audio(
hass: HomeAssistant,
from_extension: str,
- audio_bytes: bytes,
+ audio_bytes_gen: AsyncGenerator[bytes],
to_extension: str,
to_sample_rate: int | None = None,
to_sample_channels: int | None = None,
to_sample_bytes: int | None = None,
-) -> bytes:
+) -> AsyncGenerator[bytes]:
"""Convert audio to a preferred format using ffmpeg."""
ffmpeg_manager = ffmpeg.get_ffmpeg_manager(hass)
- return await hass.async_add_executor_job(
+ audio_bytes = b"".join([chunk async for chunk in audio_bytes_gen])
+ data = await hass.async_add_executor_job(
lambda: _convert_audio(
ffmpeg_manager.binary,
from_extension,
@@ -235,6 +321,7 @@ async def async_convert_audio(
to_sample_bytes=to_sample_bytes,
)
)
+ yield data
def _convert_audio(
@@ -306,11 +393,11 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
# Legacy config options
conf = config[DOMAIN][0] if config.get(DOMAIN) else {}
- use_cache: bool = conf.get(CONF_CACHE, DEFAULT_CACHE)
+ use_file_cache: bool = conf.get(CONF_CACHE, DEFAULT_CACHE)
cache_dir: str = conf.get(CONF_CACHE_DIR, DEFAULT_CACHE_DIR)
- time_memory: int = conf.get(CONF_TIME_MEMORY, DEFAULT_TIME_MEMORY)
+ memory_cache_maxage: int = conf.get(CONF_TIME_MEMORY, DEFAULT_TIME_MEMORY)
- tts = SpeechManager(hass, use_cache, cache_dir, time_memory)
+ tts = SpeechManager(hass, use_file_cache, cache_dir, memory_cache_maxage)
try:
await tts.async_init_cache()
@@ -375,140 +462,56 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return await hass.data[DATA_COMPONENT].async_unload_entry(entry)
-CACHED_PROPERTIES_WITH_ATTR_ = {
- "default_language",
- "default_options",
- "supported_languages",
- "supported_options",
-}
+@dataclass
+class ResultStream:
+ """Class that will stream the result when available."""
+ # Streaming/conversion properties
+ token: str
+ extension: str
+ content_type: str
-class TextToSpeechEntity(RestoreEntity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
- """Represent a single TTS engine."""
+ # TTS properties
+ engine: str
+ use_file_cache: bool
+ language: str
+ options: dict
- _attr_should_poll = False
- __last_tts_loaded: str | None = None
-
- _attr_default_language: str
- _attr_default_options: Mapping[str, Any] | None = None
- _attr_supported_languages: list[str]
- _attr_supported_options: list[str] | None = None
-
- @property
- @final
- def state(self) -> str | None:
- """Return the state of the entity."""
- if self.__last_tts_loaded is None:
- return None
- return self.__last_tts_loaded
+ _manager: SpeechManager
@cached_property
- def supported_languages(self) -> list[str]:
- """Return a list of supported languages."""
- return self._attr_supported_languages
+ def url(self) -> str:
+ """Get the URL to stream the result."""
+ return f"/api/tts_proxy/{self.token}"
@cached_property
- def default_language(self) -> str:
- """Return the default language."""
- return self._attr_default_language
-
- @cached_property
- def supported_options(self) -> list[str] | None:
- """Return a list of supported options like voice, emotions."""
- return self._attr_supported_options
-
- @cached_property
- def default_options(self) -> Mapping[str, Any] | None:
- """Return a mapping with the default options."""
- return self._attr_default_options
+ def _result_cache(self) -> asyncio.Future[TTSCache]:
+ """Get the future that returns the cache."""
+ return asyncio.Future()
@callback
- def async_get_supported_voices(self, language: str) -> list[Voice] | None:
- """Return a list of supported voices for a language."""
- return None
+ def async_set_message_cache(self, cache: TTSCache) -> None:
+ """Set cache containing message audio to be streamed."""
+ self._result_cache.set_result(cache)
- async def async_internal_added_to_hass(self) -> None:
- """Call when the entity is added to hass."""
- await super().async_internal_added_to_hass()
- try:
- _ = self.default_language
- except AttributeError as err:
- raise AttributeError(
- "TTS entities must either set the '_attr_default_language' attribute or override the 'default_language' property"
- ) from err
- try:
- _ = self.supported_languages
- except AttributeError as err:
- raise AttributeError(
- "TTS entities must either set the '_attr_supported_languages' attribute or override the 'supported_languages' property"
- ) from err
- state = await self.async_get_last_state()
- if (
- state is not None
- and state.state is not None
- and state.state not in (STATE_UNAVAILABLE, STATE_UNKNOWN)
- ):
- self.__last_tts_loaded = state.state
-
- async def async_speak(
- self,
- media_player_entity_id: list[str],
- message: str,
- cache: bool,
- language: str | None = None,
- options: dict | None = None,
- ) -> None:
- """Speak via a Media Player."""
- await self.hass.services.async_call(
- DOMAIN_MP,
- SERVICE_PLAY_MEDIA,
- {
- ATTR_ENTITY_ID: media_player_entity_id,
- ATTR_MEDIA_CONTENT_ID: generate_media_source_id(
- self.hass,
- message=message,
- engine=self.entity_id,
- language=language,
- options=options,
- cache=cache,
- ),
- ATTR_MEDIA_CONTENT_TYPE: MediaType.MUSIC,
- ATTR_MEDIA_ANNOUNCE: True,
- },
- blocking=True,
- context=self._context,
+ @callback
+ def async_set_message(self, message: str) -> None:
+ """Set message to be generated."""
+ self._result_cache.set_result(
+ self._manager.async_cache_message_in_memory(
+ engine=self.engine,
+ message=message,
+ use_file_cache=self.use_file_cache,
+ language=self.language,
+ options=self.options,
+ )
)
- @final
- async def internal_async_get_tts_audio(
- self, message: str, language: str, options: dict[str, Any]
- ) -> TtsAudioType:
- """Process an audio stream to TTS service.
-
- Only streaming content is allowed!
- """
- self.__last_tts_loaded = dt_util.utcnow().isoformat()
- self.async_write_ha_state()
- return await self.async_get_tts_audio(
- message=message, language=language, options=options
- )
-
- def get_tts_audio(
- self, message: str, language: str, options: dict[str, Any]
- ) -> TtsAudioType:
- """Load tts audio file from the engine."""
- raise NotImplementedError
-
- async def async_get_tts_audio(
- self, message: str, language: str, options: dict[str, Any]
- ) -> TtsAudioType:
- """Load tts audio file from the engine.
-
- Return a tuple of file extension and data as bytes.
- """
- return await self.hass.async_add_executor_job(
- partial(self.get_tts_audio, message, language, options=options)
- )
+ async def async_stream_result(self) -> AsyncGenerator[bytes]:
+ """Get the stream of this result."""
+ cache = await self._result_cache
+ async for chunk in cache.async_stream_data():
+ yield chunk
def _hash_options(options: dict) -> str:
@@ -521,29 +524,82 @@ def _hash_options(options: dict) -> str:
return opts_hash.hexdigest()
+class MemcacheCleanup:
+ """Helper to clean up the stale sessions."""
+
+ unsub: CALLBACK_TYPE | None = None
+
+ def __init__(
+ self, hass: HomeAssistant, maxage: float, memcache: dict[str, TTSCache]
+ ) -> None:
+ """Initialize the cleanup."""
+ self.hass = hass
+ self.maxage = maxage
+ self.memcache = memcache
+ hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, self._on_hass_stop)
+ self.cleanup_job = HassJob(
+ self._cleanup, "chat_session_cleanup", job_type=HassJobType.Callback
+ )
+
+ @callback
+ def schedule(self) -> None:
+ """Schedule the cleanup."""
+ if self.unsub:
+ return
+ self.unsub = async_call_later(
+ self.hass,
+ self.maxage + 1,
+ self.cleanup_job,
+ )
+
+ @callback
+ def _on_hass_stop(self, event: Event) -> None:
+ """Cancel the cleanup on shutdown."""
+ if self.unsub:
+ self.unsub()
+ self.unsub = None
+
+ @callback
+ def _cleanup(self, _now: datetime) -> None:
+ """Clean up and schedule follow-up if necessary."""
+ self.unsub = None
+ memcache = self.memcache
+ maxage = self.maxage
+ now = monotonic()
+
+ for cache_key, info in list(memcache.items()):
+ if info.last_used + maxage < now:
+ _LOGGER.debug("Cleaning up %s", cache_key)
+ del memcache[cache_key]
+
+ # Still items left, check again in timeout time.
+ if memcache:
+ self.schedule()
+
+
class SpeechManager:
"""Representation of a speech store."""
def __init__(
self,
hass: HomeAssistant,
- use_cache: bool,
+ use_file_cache: bool,
cache_dir: str,
- time_memory: int,
+ memory_cache_maxage: int,
) -> None:
"""Initialize a speech store."""
self.hass = hass
self.providers: dict[str, Provider] = {}
- self.use_cache = use_cache
+ self.use_file_cache = use_file_cache
self.cache_dir = cache_dir
- self.time_memory = time_memory
+ self.memory_cache_maxage = memory_cache_maxage
self.file_cache: dict[str, str] = {}
self.mem_cache: dict[str, TTSCache] = {}
-
- # filename <-> token
- self.filename_to_token: dict[str, str] = {}
- self.token_to_filename: dict[str, str] = {}
+ self.token_to_stream: dict[str, ResultStream] = {}
+ self.memcache_cleanup = MemcacheCleanup(
+ hass, memory_cache_maxage, self.mem_cache
+ )
def _init_cache(self) -> dict[str, str]:
"""Init cache folder and fetch files."""
@@ -563,18 +619,21 @@ class SpeechManager:
async def async_clear_cache(self) -> None:
"""Read file cache and delete files."""
- self.mem_cache = {}
+ self.mem_cache.clear()
- def remove_files() -> None:
+ def remove_files(files: list[str]) -> None:
"""Remove files from filesystem."""
- for filename in self.file_cache.values():
+ for filename in files:
try:
os.remove(os.path.join(self.cache_dir, filename))
except OSError as err:
_LOGGER.warning("Can't remove cache file '%s': %s", filename, err)
- await self.hass.async_add_executor_job(remove_files)
- self.file_cache = {}
+ task = self.hass.async_add_executor_job(
+ remove_files, list(self.file_cache.values())
+ )
+ self.file_cache.clear()
+ await task
@callback
def async_register_legacy_engine(
@@ -629,110 +688,198 @@ class SpeechManager:
return language, merged_options
- async def async_get_url_path(
+ @callback
+ def async_create_result_stream(
self,
engine: str,
- message: str,
- cache: bool | None = None,
+ message: str | None = None,
+ use_file_cache: bool | None = None,
language: str | None = None,
options: dict | None = None,
- ) -> str:
- """Get URL for play message.
-
- This method is a coroutine.
- """
+ ) -> ResultStream:
+ """Create a streaming URL where the rendered TTS can be retrieved."""
if (engine_instance := get_engine_instance(self.hass, engine)) is None:
raise HomeAssistantError(f"Provider {engine} not found")
language, options = self.process_options(engine_instance, language, options)
- cache_key = self._generate_cache_key(message, language, options, engine)
- use_cache = cache if cache is not None else self.use_cache
+ if use_file_cache is None:
+ use_file_cache = self.use_file_cache
- # Is speech already in memory
- if cache_key in self.mem_cache:
- filename = self.mem_cache[cache_key]["filename"]
- # Is file store in file cache
- elif use_cache and cache_key in self.file_cache:
- filename = self.file_cache[cache_key]
- self.hass.async_create_task(self._async_file_to_mem(cache_key))
- # Load speech from engine into memory
- else:
- filename = await self._async_get_tts_audio(
- engine_instance, cache_key, message, use_cache, language, options
+ extension = options.get(ATTR_PREFERRED_FORMAT, _DEFAULT_FORMAT)
+ token = f"{secrets.token_urlsafe(16)}.{extension}"
+ content, _ = mimetypes.guess_type(token)
+ result_stream = ResultStream(
+ token=token,
+ extension=extension,
+ content_type=content or "audio/mpeg",
+ use_file_cache=use_file_cache,
+ engine=engine,
+ language=language,
+ options=options,
+ _manager=self,
+ )
+ self.token_to_stream[token] = result_stream
+
+ if message is None:
+ return result_stream
+
+ # We added this method as an alternative to stream.async_set_message
+ # to avoid the options being processed twice
+ result_stream.async_set_message_cache(
+ self._async_ensure_cached_in_memory(
+ engine=engine,
+ engine_instance=engine_instance,
+ message=message,
+ use_file_cache=use_file_cache,
+ language=language,
+ options=options,
)
+ )
- # Use a randomly generated token instead of exposing the filename
- token = self.filename_to_token.get(filename)
- if not token:
- # Keep extension (.mp3, etc.)
- token = secrets.token_urlsafe(16) + os.path.splitext(filename)[1]
-
- # Map token <-> filename
- self.filename_to_token[filename] = token
- self.token_to_filename[token] = filename
-
- return f"/api/tts_proxy/{token}"
-
- async def async_get_tts_audio(
- self,
- engine: str,
- message: str,
- cache: bool | None = None,
- language: str | None = None,
- options: dict | None = None,
- ) -> tuple[str, bytes]:
- """Fetch TTS audio."""
- if (engine_instance := get_engine_instance(self.hass, engine)) is None:
- raise HomeAssistantError(f"Provider {engine} not found")
-
- language, options = self.process_options(engine_instance, language, options)
- cache_key = self._generate_cache_key(message, language, options, engine)
- use_cache = cache if cache is not None else self.use_cache
-
- # If we have the file, load it into memory if necessary
- if cache_key not in self.mem_cache:
- if use_cache and cache_key in self.file_cache:
- await self._async_file_to_mem(cache_key)
- else:
- await self._async_get_tts_audio(
- engine_instance, cache_key, message, use_cache, language, options
- )
-
- extension = os.path.splitext(self.mem_cache[cache_key]["filename"])[1][1:]
- cached = self.mem_cache[cache_key]
- if pending := cached.get("pending"):
- await pending
- cached = self.mem_cache[cache_key]
- return extension, cached["voice"]
+ return result_stream
@callback
- def _generate_cache_key(
+ def async_cache_message_in_memory(
self,
- message: str,
- language: str,
- options: dict | None,
engine: str,
- ) -> str:
- """Generate a cache key for a message."""
+ message: str,
+ use_file_cache: bool | None = None,
+ language: str | None = None,
+ options: dict | None = None,
+ ) -> TTSCache:
+ """Make sure a message is cached in memory and returns cache key."""
+ if (engine_instance := get_engine_instance(self.hass, engine)) is None:
+ raise HomeAssistantError(f"Provider {engine} not found")
+
+ language, options = self.process_options(engine_instance, language, options)
+ if use_file_cache is None:
+ use_file_cache = self.use_file_cache
+
+ return self._async_ensure_cached_in_memory(
+ engine=engine,
+ engine_instance=engine_instance,
+ message=message,
+ use_file_cache=use_file_cache,
+ language=language,
+ options=options,
+ )
+
+ @callback
+ def _async_ensure_cached_in_memory(
+ self,
+ engine: str,
+ engine_instance: TextToSpeechEntity | Provider,
+ message: str,
+ use_file_cache: bool,
+ language: str,
+ options: dict,
+ ) -> TTSCache:
+ """Ensure a message is cached.
+
+ Requires options, language to be processed.
+ """
options_key = _hash_options(options) if options else "-"
msg_hash = hashlib.sha1(bytes(message, "utf-8")).hexdigest()
- return KEY_PATTERN.format(
+ cache_key = KEY_PATTERN.format(
msg_hash, language.replace("_", "-"), options_key, engine
).lower()
- async def _async_get_tts_audio(
+ # Is speech already in memory
+ if cache := self.mem_cache.get(cache_key):
+ _LOGGER.debug("Found audio in cache for %s", message[0:32])
+ return cache
+
+ store_to_disk = use_file_cache
+
+ if use_file_cache and (filename := self.file_cache.get(cache_key)):
+ _LOGGER.debug("Loading audio from disk for %s", message[0:32])
+ extension = os.path.splitext(filename)[1][1:]
+ data_gen = self._async_load_file(cache_key)
+ store_to_disk = False
+ else:
+ _LOGGER.debug("Generating audio for %s", message[0:32])
+ extension = options.get(ATTR_PREFERRED_FORMAT, _DEFAULT_FORMAT)
+ data_gen = self._async_generate_tts_audio(
+ engine_instance, message, language, options
+ )
+
+ cache = TTSCache(
+ cache_key=cache_key,
+ extension=extension,
+ data_gen=data_gen,
+ )
+
+ self.mem_cache[cache_key] = cache
+ self.hass.async_create_background_task(
+ self._load_data_into_cache(
+ cache, engine_instance, message, store_to_disk, language, options
+ ),
+ f"tts_load_data_into_cache_{engine_instance.name}",
+ )
+ self.memcache_cleanup.schedule()
+ return cache
+
+ async def _load_data_into_cache(
+ self,
+ cache: TTSCache,
+ engine_instance: TextToSpeechEntity | Provider,
+ message: str,
+ store_to_disk: bool,
+ language: str,
+ options: dict,
+ ) -> None:
+ """Load and process a finished loading TTS Cache."""
+ try:
+ data = await cache.async_load_data()
+ except Exception as err: # pylint: disable=broad-except # noqa: BLE001
+ # Truncate message so we don't flood the logs. Cutting off at 32 chars
+ # but since we add 3 dots to truncated message, we cut off at 35.
+ trunc_msg = message if len(message) < 35 else f"{message[0:32]}…"
+ _LOGGER.error("Error getting audio for %s: %s", trunc_msg, err)
+ self.mem_cache.pop(cache.cache_key, None)
+ return
+
+ if not store_to_disk:
+ return
+
+ filename = f"{cache.cache_key}.{cache.extension}".lower()
+
+ # Validate filename
+ if not _RE_VOICE_FILE.match(filename) and not _RE_LEGACY_VOICE_FILE.match(
+ filename
+ ):
+ raise HomeAssistantError(
+ f"TTS filename '{filename}' from {engine_instance.name} is invalid!"
+ )
+
+ if cache.extension == "mp3":
+ name = (
+ engine_instance.name if isinstance(engine_instance.name, str) else "-"
+ )
+ data = self.write_tags(filename, data, name, message, language, options)
+
+ voice_file = os.path.join(self.cache_dir, filename)
+
+ def save_speech() -> None:
+ """Store speech to filesystem."""
+ with open(voice_file, "wb") as speech:
+ speech.write(data)
+
+ try:
+ await self.hass.async_add_executor_job(save_speech)
+ except OSError as err:
+ _LOGGER.error("Can't write %s: %s", filename, err)
+ else:
+ self.file_cache[cache.cache_key] = filename
+
+ async def _async_generate_tts_audio(
self,
engine_instance: TextToSpeechEntity | Provider,
- cache_key: str,
message: str,
- cache: bool,
language: str,
options: dict[str, Any],
- ) -> str:
- """Receive TTS, store for view in cache and return filename.
-
- This method is a coroutine.
- """
+ ) -> AsyncGenerator[bytes]:
+ """Generate TTS audio from an engine."""
options = dict(options or {})
supported_options = engine_instance.supported_options or []
@@ -773,114 +920,61 @@ class SpeechManager:
if sample_bytes is not None:
sample_bytes = int(sample_bytes)
- async def get_tts_data() -> str:
- """Handle data available."""
- if engine_instance.name is None or engine_instance.name is UNDEFINED:
- raise HomeAssistantError("TTS engine name is not set.")
+ if engine_instance.name is None or engine_instance.name is UNDEFINED:
+ raise HomeAssistantError("TTS engine name is not set.")
- if isinstance(engine_instance, Provider):
- extension, data = await engine_instance.async_get_tts_audio(
- message, language, options
- )
- else:
- extension, data = await engine_instance.internal_async_get_tts_audio(
- message, language, options
- )
+ if isinstance(engine_instance, Provider):
+ extension, data = await engine_instance.async_get_tts_audio(
+ message, language, options
+ )
if data is None or extension is None:
raise HomeAssistantError(
f"No TTS from {engine_instance.name} for '{message}'"
)
- # Only convert if we have a preferred format different than the
- # expected format from the TTS system, or if a specific sample
- # rate/format/channel count is requested.
- needs_conversion = (
- (final_extension != extension)
- or (sample_rate is not None)
- or (sample_channels is not None)
- or (sample_bytes is not None)
+ async def make_data_generator(data: bytes) -> AsyncGenerator[bytes]:
+ yield data
+
+ data_gen = make_data_generator(data)
+
+ else:
+
+ async def message_gen() -> AsyncGenerator[str]:
+ yield message
+
+ tts_result = await engine_instance.internal_async_stream_tts_audio(
+ TTSAudioRequest(language, options, message_gen())
+ )
+ extension = tts_result.extension
+ data_gen = tts_result.data_gen
+
+ # Only convert if we have a preferred format different than the
+ # expected format from the TTS system, or if a specific sample
+ # rate/format/channel count is requested.
+ needs_conversion = (
+ (final_extension != extension)
+ or (sample_rate is not None)
+ or (sample_channels is not None)
+ or (sample_bytes is not None)
+ )
+
+ if needs_conversion:
+ data_gen = _async_convert_audio(
+ self.hass,
+ extension,
+ data_gen,
+ to_extension=final_extension,
+ to_sample_rate=sample_rate,
+ to_sample_channels=sample_channels,
+ to_sample_bytes=sample_bytes,
)
- if needs_conversion:
- data = await async_convert_audio(
- self.hass,
- extension,
- data,
- to_extension=final_extension,
- to_sample_rate=sample_rate,
- to_sample_channels=sample_channels,
- to_sample_bytes=sample_bytes,
- )
+ async for chunk in data_gen:
+ yield chunk
- # Create file infos
- filename = f"{cache_key}.{final_extension}".lower()
-
- # Validate filename
- if not _RE_VOICE_FILE.match(filename) and not _RE_LEGACY_VOICE_FILE.match(
- filename
- ):
- raise HomeAssistantError(
- f"TTS filename '{filename}' from {engine_instance.name} is invalid!"
- )
-
- # Save to memory
- if final_extension == "mp3":
- data = self.write_tags(
- filename, data, engine_instance.name, message, language, options
- )
-
- self._async_store_to_memcache(cache_key, filename, data)
-
- if cache:
- self.hass.async_create_task(
- self._async_save_tts_audio(cache_key, filename, data)
- )
-
- return filename
-
- audio_task = self.hass.async_create_task(get_tts_data(), eager_start=False)
-
- def handle_error(_future: asyncio.Future) -> None:
- """Handle error."""
- if audio_task.exception():
- self.mem_cache.pop(cache_key, None)
-
- audio_task.add_done_callback(handle_error)
-
- filename = f"{cache_key}.{final_extension}".lower()
- self.mem_cache[cache_key] = {
- "filename": filename,
- "voice": b"",
- "pending": audio_task,
- }
- return filename
-
- async def _async_save_tts_audio(
- self, cache_key: str, filename: str, data: bytes
- ) -> None:
- """Store voice data to file and file_cache.
-
- This method is a coroutine.
- """
- voice_file = os.path.join(self.cache_dir, filename)
-
- def save_speech() -> None:
- """Store speech to filesystem."""
- with open(voice_file, "wb") as speech:
- speech.write(data)
-
- try:
- await self.hass.async_add_executor_job(save_speech)
- self.file_cache[cache_key] = filename
- except OSError as err:
- _LOGGER.error("Can't write %s: %s", filename, err)
-
- async def _async_file_to_mem(self, cache_key: str) -> None:
- """Load voice from file cache into memory.
-
- This method is a coroutine.
- """
+ async def _async_load_file(self, cache_key: str) -> AsyncGenerator[bytes]:
+ """Load TTS audio from disk."""
if not (filename := self.file_cache.get(cache_key)):
raise HomeAssistantError(f"Key {cache_key} not in file cache!")
@@ -897,64 +991,7 @@ class SpeechManager:
del self.file_cache[cache_key]
raise HomeAssistantError(f"Can't read {voice_file}") from err
- self._async_store_to_memcache(cache_key, filename, data)
-
- @callback
- def _async_store_to_memcache(
- self, cache_key: str, filename: str, data: bytes
- ) -> None:
- """Store data to memcache and set timer to remove it."""
- self.mem_cache[cache_key] = {
- "filename": filename,
- "voice": data,
- "pending": None,
- }
-
- @callback
- def async_remove_from_mem(_: datetime) -> None:
- """Cleanup memcache."""
- self.mem_cache.pop(cache_key, None)
-
- async_call_later(
- self.hass,
- self.time_memory,
- HassJob(
- async_remove_from_mem,
- name="tts remove_from_mem",
- cancel_on_shutdown=True,
- ),
- )
-
- async def async_read_tts(self, token: str) -> tuple[str | None, bytes]:
- """Read a voice file and return binary.
-
- This method is a coroutine.
- """
- filename = self.token_to_filename.get(token)
- if not filename:
- raise HomeAssistantError(f"{token} was not recognized!")
-
- if not (record := _RE_VOICE_FILE.match(filename.lower())) and not (
- record := _RE_LEGACY_VOICE_FILE.match(filename.lower())
- ):
- raise HomeAssistantError("Wrong tts file format!")
-
- cache_key = KEY_PATTERN.format(
- record.group(1), record.group(2), record.group(3), record.group(4)
- )
-
- if cache_key not in self.mem_cache:
- if cache_key not in self.file_cache:
- raise HomeAssistantError(f"{cache_key} not in cache!")
- await self._async_file_to_mem(cache_key)
-
- cached = self.mem_cache[cache_key]
- if pending := cached.get("pending"):
- await pending
- cached = self.mem_cache[cache_key]
-
- content, _ = mimetypes.guess_type(filename)
- return content, cached["voice"]
+ yield data
@staticmethod
def write_tags(
@@ -1042,9 +1079,9 @@ class TextToSpeechUrlView(HomeAssistantView):
url = "/api/tts_get_url"
name = "api:tts:geturl"
- def __init__(self, tts: SpeechManager) -> None:
+ def __init__(self, manager: SpeechManager) -> None:
"""Initialize a tts view."""
- self.tts = tts
+ self.manager = manager
async def post(self, request: web.Request) -> web.Response:
"""Generate speech and provide url."""
@@ -1061,45 +1098,65 @@ class TextToSpeechUrlView(HomeAssistantView):
engine = data.get("engine_id") or data[ATTR_PLATFORM]
message = data[ATTR_MESSAGE]
- cache = data.get(ATTR_CACHE)
+ use_file_cache = data.get(ATTR_CACHE)
language = data.get(ATTR_LANGUAGE)
options = data.get(ATTR_OPTIONS)
try:
- path = await self.tts.async_get_url_path(
- engine, message, cache=cache, language=language, options=options
+ stream = self.manager.async_create_result_stream(
+ engine,
+ message,
+ use_file_cache=use_file_cache,
+ language=language,
+ options=options,
)
except HomeAssistantError as err:
_LOGGER.error("Error on init tts: %s", err)
return self.json({"error": err}, HTTPStatus.BAD_REQUEST)
- base = get_url(self.tts.hass)
- url = base + path
+ base = get_url(self.manager.hass)
+ url = base + stream.url
- return self.json({"url": url, "path": path})
+ return self.json({"url": url, "path": stream.url})
class TextToSpeechView(HomeAssistantView):
"""TTS view to serve a speech audio."""
requires_auth = False
- url = "/api/tts_proxy/{filename}"
+ url = "/api/tts_proxy/{token}"
name = "api:tts_speech"
- def __init__(self, tts: SpeechManager) -> None:
+ def __init__(self, manager: SpeechManager) -> None:
"""Initialize a tts view."""
- self.tts = tts
+ self.manager = manager
- async def get(self, request: web.Request, filename: str) -> web.Response:
+ async def get(self, request: web.Request, token: str) -> web.StreamResponse:
"""Start a get request."""
- try:
- # filename is actually token, but we keep its name for compatibility
- content, data = await self.tts.async_read_tts(filename)
- except HomeAssistantError as err:
- _LOGGER.error("Error on load tts: %s", err)
+ stream = self.manager.token_to_stream.get(token)
+
+ if stream is None:
return web.Response(status=HTTPStatus.NOT_FOUND)
- return web.Response(body=data, content_type=content)
+ response: web.StreamResponse | None = None
+ try:
+ async for data in stream.async_stream_result():
+ if response is None:
+ response = web.StreamResponse()
+ response.content_type = stream.content_type
+ await response.prepare(request)
+
+ await response.write(data)
+ # pylint: disable=broad-except
+ except Exception as err: # noqa: BLE001
+ _LOGGER.error("Error streaming tts: %s", err)
+
+ # Empty result or exception happened
+ if response is None:
+ return web.Response(status=HTTPStatus.INTERNAL_SERVER_ERROR)
+
+ await response.write_eof()
+ return response
@websocket_api.websocket_command(
diff --git a/homeassistant/components/tts/entity.py b/homeassistant/components/tts/entity.py
new file mode 100644
index 00000000000..199d673398e
--- /dev/null
+++ b/homeassistant/components/tts/entity.py
@@ -0,0 +1,196 @@
+"""Entity for Text-to-Speech."""
+
+from collections.abc import AsyncGenerator, Mapping
+from dataclasses import dataclass
+from functools import partial
+from typing import Any, final
+
+from propcache.api import cached_property
+
+from homeassistant.components.media_player import (
+ ATTR_MEDIA_ANNOUNCE,
+ ATTR_MEDIA_CONTENT_ID,
+ ATTR_MEDIA_CONTENT_TYPE,
+ DOMAIN as DOMAIN_MP,
+ SERVICE_PLAY_MEDIA,
+ MediaType,
+)
+from homeassistant.const import ATTR_ENTITY_ID, STATE_UNAVAILABLE, STATE_UNKNOWN
+from homeassistant.core import callback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.restore_state import RestoreEntity
+from homeassistant.util import dt as dt_util
+
+from .const import TtsAudioType
+from .media_source import generate_media_source_id
+from .models import Voice
+
+CACHED_PROPERTIES_WITH_ATTR_ = {
+ "default_language",
+ "default_options",
+ "supported_languages",
+ "supported_options",
+}
+
+
+@dataclass
+class TTSAudioRequest:
+ """Request to get TTS audio."""
+
+ language: str
+ options: dict[str, Any]
+ message_gen: AsyncGenerator[str]
+
+
+@dataclass
+class TTSAudioResponse:
+ """Response containing TTS audio stream."""
+
+ extension: str
+ data_gen: AsyncGenerator[bytes]
+
+
+class TextToSpeechEntity(RestoreEntity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
+ """Represent a single TTS engine."""
+
+ _attr_should_poll = False
+ __last_tts_loaded: str | None = None
+
+ _attr_default_language: str
+ _attr_default_options: Mapping[str, Any] | None = None
+ _attr_supported_languages: list[str]
+ _attr_supported_options: list[str] | None = None
+
+ @property
+ @final
+ def state(self) -> str | None:
+ """Return the state of the entity."""
+ if self.__last_tts_loaded is None:
+ return None
+ return self.__last_tts_loaded
+
+ @cached_property
+ def supported_languages(self) -> list[str]:
+ """Return a list of supported languages."""
+ return self._attr_supported_languages
+
+ @cached_property
+ def default_language(self) -> str:
+ """Return the default language."""
+ return self._attr_default_language
+
+ @cached_property
+ def supported_options(self) -> list[str] | None:
+ """Return a list of supported options like voice, emotions."""
+ return self._attr_supported_options
+
+ @cached_property
+ def default_options(self) -> Mapping[str, Any] | None:
+ """Return a mapping with the default options."""
+ return self._attr_default_options
+
+ @callback
+ def async_get_supported_voices(self, language: str) -> list[Voice] | None:
+ """Return a list of supported voices for a language."""
+ return None
+
+ async def async_internal_added_to_hass(self) -> None:
+ """Call when the entity is added to hass."""
+ await super().async_internal_added_to_hass()
+ try:
+ _ = self.default_language
+ except AttributeError as err:
+ raise AttributeError(
+ "TTS entities must either set the '_attr_default_language' attribute or override the 'default_language' property"
+ ) from err
+ try:
+ _ = self.supported_languages
+ except AttributeError as err:
+ raise AttributeError(
+ "TTS entities must either set the '_attr_supported_languages' attribute or override the 'supported_languages' property"
+ ) from err
+ state = await self.async_get_last_state()
+ if (
+ state is not None
+ and state.state is not None
+ and state.state not in (STATE_UNAVAILABLE, STATE_UNKNOWN)
+ ):
+ self.__last_tts_loaded = state.state
+
+ async def async_speak(
+ self,
+ media_player_entity_id: list[str],
+ message: str,
+ cache: bool,
+ language: str | None = None,
+ options: dict | None = None,
+ ) -> None:
+ """Speak via a Media Player."""
+ await self.hass.services.async_call(
+ DOMAIN_MP,
+ SERVICE_PLAY_MEDIA,
+ {
+ ATTR_ENTITY_ID: media_player_entity_id,
+ ATTR_MEDIA_CONTENT_ID: generate_media_source_id(
+ self.hass,
+ message=message,
+ engine=self.entity_id,
+ language=language,
+ options=options,
+ cache=cache,
+ ),
+ ATTR_MEDIA_CONTENT_TYPE: MediaType.MUSIC,
+ ATTR_MEDIA_ANNOUNCE: True,
+ },
+ blocking=True,
+ context=self._context,
+ )
+
+ @final
+ async def internal_async_stream_tts_audio(
+ self, request: TTSAudioRequest
+ ) -> TTSAudioResponse:
+ """Process an audio stream to TTS service.
+
+ Only streaming content is allowed!
+ """
+ self.__last_tts_loaded = dt_util.utcnow().isoformat()
+ self.async_write_ha_state()
+ return await self.async_stream_tts_audio(request)
+
+ async def async_stream_tts_audio(
+ self, request: TTSAudioRequest
+ ) -> TTSAudioResponse:
+ """Generate speech from an incoming message.
+
+ The default implementation is backwards compatible with async_get_tts_audio.
+ """
+ message = "".join([chunk async for chunk in request.message_gen])
+ extension, data = await self.async_get_tts_audio(
+ message, request.language, request.options
+ )
+
+ if extension is None or data is None:
+ raise HomeAssistantError(f"No TTS from {self.entity_id} for '{message}'")
+
+ async def data_gen() -> AsyncGenerator[bytes]:
+ yield data
+
+ return TTSAudioResponse(extension, data_gen())
+
+ def get_tts_audio(
+ self, message: str, language: str, options: dict[str, Any]
+ ) -> TtsAudioType:
+ """Load tts audio file from the engine."""
+ raise NotImplementedError
+
+ async def async_get_tts_audio(
+ self, message: str, language: str, options: dict[str, Any]
+ ) -> TtsAudioType:
+ """Load tts audio file from the engine.
+
+ Return a tuple of file extension and data as bytes.
+ """
+ return await self.hass.async_add_executor_job(
+ partial(self.get_tts_audio, message, language, options=options)
+ )
diff --git a/homeassistant/components/tts/media_source.py b/homeassistant/components/tts/media_source.py
index 4f1fa59f001..aa2cd6e7555 100644
--- a/homeassistant/components/tts/media_source.py
+++ b/homeassistant/components/tts/media_source.py
@@ -3,7 +3,6 @@
from __future__ import annotations
import json
-import mimetypes
from typing import TypedDict
from yarl import URL
@@ -73,7 +72,7 @@ class MediaSourceOptions(TypedDict):
message: str
language: str | None
options: dict | None
- cache: bool | None
+ use_file_cache: bool | None
@callback
@@ -98,10 +97,10 @@ def media_source_id_to_kwargs(media_source_id: str) -> MediaSourceOptions:
"message": parsed.query["message"],
"language": parsed.query.get("language"),
"options": options,
- "cache": None,
+ "use_file_cache": None,
}
if "cache" in parsed.query:
- kwargs["cache"] = parsed.query["cache"] == "true"
+ kwargs["use_file_cache"] = parsed.query["cache"] == "true"
return kwargs
@@ -119,7 +118,7 @@ class TTSMediaSource(MediaSource):
async def async_resolve_media(self, item: MediaSourceItem) -> PlayMedia:
"""Resolve media to a url."""
try:
- url = await self.hass.data[DATA_TTS_MANAGER].async_get_url_path(
+ stream = self.hass.data[DATA_TTS_MANAGER].async_create_result_stream(
**media_source_id_to_kwargs(item.identifier)
)
except Unresolvable:
@@ -127,9 +126,7 @@ class TTSMediaSource(MediaSource):
except HomeAssistantError as err:
raise Unresolvable(str(err)) from err
- mime_type = mimetypes.guess_type(url)[0] or "audio/mpeg"
-
- return PlayMedia(url, mime_type)
+ return PlayMedia(stream.url, stream.content_type)
async def async_browse_media(
self,
diff --git a/homeassistant/components/tuya/__init__.py b/homeassistant/components/tuya/__init__.py
index c8a639cd239..32119add5f4 100644
--- a/homeassistant/components/tuya/__init__.py
+++ b/homeassistant/components/tuya/__init__.py
@@ -3,7 +3,8 @@
from __future__ import annotations
import logging
-from typing import Any, NamedTuple
+from typing import TYPE_CHECKING, Any, NamedTuple
+from urllib.parse import urlsplit
from tuya_sharing import (
CustomerDevice,
@@ -11,6 +12,7 @@ from tuya_sharing import (
SharingDeviceListener,
SharingTokenListener,
)
+from tuya_sharing.mq import SharingMQ, SharingMQConfig
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
@@ -45,13 +47,81 @@ class HomeAssistantTuyaData(NamedTuple):
listener: SharingDeviceListener
+if TYPE_CHECKING:
+ import paho.mqtt.client as mqtt
+
+
+class ManagerCompat(Manager):
+ """Extended Manager class from the Tuya device sharing SDK.
+
+ The extension ensures compatibility a paho-mqtt client version >= 2.1.0.
+ It overrides extend refresh_mq method to ensure correct paho.mqtt client calls.
+
+ This code can be removed when a version of tuya-device-sharing with
+ https://github.com/tuya/tuya-device-sharing-sdk/pull/25 is available.
+ """
+
+ def refresh_mq(self):
+ """Refresh the MQTT connection."""
+ if self.mq is not None:
+ self.mq.stop()
+ self.mq = None
+
+ home_ids = [home.id for home in self.user_homes]
+ device = [
+ device
+ for device in self.device_map.values()
+ if hasattr(device, "id") and getattr(device, "set_up", False)
+ ]
+
+ sharing_mq = SharingMQCompat(self.customer_api, home_ids, device)
+ sharing_mq.start()
+ sharing_mq.add_message_listener(self.on_message)
+ self.mq = sharing_mq
+
+
+class SharingMQCompat(SharingMQ):
+ """Extended SharingMQ class from the Tuya device sharing SDK.
+
+ The extension ensures compatibility a paho-mqtt client version >= 2.1.0.
+ It overrides _start method to ensure correct paho.mqtt client calls.
+
+ This code can be removed when a version of tuya-device-sharing with
+ https://github.com/tuya/tuya-device-sharing-sdk/pull/25 is available.
+ """
+
+ def _start(self, mq_config: SharingMQConfig) -> mqtt.Client:
+ """Start the MQTT client."""
+ # We don't import on the top because some integrations
+ # should be able to optionally rely on MQTT.
+ import paho.mqtt.client as mqtt # pylint: disable=import-outside-toplevel
+
+ mqttc = mqtt.Client(client_id=mq_config.client_id)
+ mqttc.username_pw_set(mq_config.username, mq_config.password)
+ mqttc.user_data_set({"mqConfig": mq_config})
+ mqttc.on_connect = self._on_connect
+ mqttc.on_message = self._on_message
+ mqttc.on_subscribe = self._on_subscribe
+ mqttc.on_log = self._on_log
+ mqttc.on_disconnect = self._on_disconnect
+
+ url = urlsplit(mq_config.url)
+ if url.scheme == "ssl":
+ mqttc.tls_set()
+
+ mqttc.connect(url.hostname, url.port)
+
+ mqttc.loop_start()
+ return mqttc
+
+
async def async_setup_entry(hass: HomeAssistant, entry: TuyaConfigEntry) -> bool:
"""Async setup hass config entry."""
if CONF_APP_TYPE in entry.data:
raise ConfigEntryAuthFailed("Authentication failed. Please re-authenticate.")
token_listener = TokenListener(hass, entry)
- manager = Manager(
+ manager = ManagerCompat(
TUYA_CLIENT_ID,
entry.data[CONF_USER_CODE],
entry.data[CONF_TERMINAL_ID],
diff --git a/homeassistant/components/tuya/alarm_control_panel.py b/homeassistant/components/tuya/alarm_control_panel.py
index 56bccc73581..96f7d3a1e1c 100644
--- a/homeassistant/components/tuya/alarm_control_panel.py
+++ b/homeassistant/components/tuya/alarm_control_panel.py
@@ -14,7 +14,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode, DPType
@@ -53,7 +53,9 @@ ALARM: dict[str, tuple[AlarmControlPanelEntityDescription, ...]] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya alarm dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/binary_sensor.py b/homeassistant/components/tuya/binary_sensor.py
index 12661a26fd1..486dd6e1387 100644
--- a/homeassistant/components/tuya/binary_sensor.py
+++ b/homeassistant/components/tuya/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode
@@ -256,7 +256,7 @@ BINARY_SENSORS: dict[str, tuple[TuyaBinarySensorEntityDescription, ...]] = {
TuyaBinarySensorEntityDescription(
key=DPCode.WATERSENSOR_STATE,
device_class=BinarySensorDeviceClass.MOISTURE,
- on_value="alarm",
+ on_value={"1", "alarm"},
),
TAMPER_BINARY_SENSOR,
),
@@ -291,6 +291,9 @@ BINARY_SENSORS: dict[str, tuple[TuyaBinarySensorEntityDescription, ...]] = {
# Temperature and Humidity Sensor
# https://developer.tuya.com/en/docs/iot/categorywsdcg?id=Kaiuz3hinij34
"wsdcg": (TAMPER_BINARY_SENSOR,),
+ # Temperature and Humidity Sensor with External Probe
+ # New undocumented category qxj, see https://github.com/home-assistant/core/issues/136472
+ "qxj": (TAMPER_BINARY_SENSOR,),
# Pressure Sensor
# https://developer.tuya.com/en/docs/iot/categoryylcg?id=Kaiuz3kc2e4gm
"ylcg": (
@@ -341,7 +344,9 @@ BINARY_SENSORS: dict[str, tuple[TuyaBinarySensorEntityDescription, ...]] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya binary sensor dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/button.py b/homeassistant/components/tuya/button.py
index f77fed776b0..8e538b07309 100644
--- a/homeassistant/components/tuya/button.py
+++ b/homeassistant/components/tuya/button.py
@@ -8,7 +8,7 @@ from homeassistant.components.button import ButtonEntity, ButtonEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode
@@ -58,7 +58,9 @@ BUTTONS: dict[str, tuple[ButtonEntityDescription, ...]] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya buttons dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/camera.py b/homeassistant/components/tuya/camera.py
index 9e66531dd51..c04a8a043dc 100644
--- a/homeassistant/components/tuya/camera.py
+++ b/homeassistant/components/tuya/camera.py
@@ -8,7 +8,7 @@ from homeassistant.components import ffmpeg
from homeassistant.components.camera import Camera as CameraEntity, CameraEntityFeature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode
@@ -20,11 +20,16 @@ CAMERAS: tuple[str, ...] = (
# Smart Camera (including doorbells)
# https://developer.tuya.com/en/docs/iot/categorysgbj?id=Kaiuz37tlpbnu
"sp",
+ # Smart Camera - Low power consumption camera
+ # Undocumented, see https://github.com/home-assistant/core/issues/132844
+ "dghsxj",
)
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya cameras dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/climate.py b/homeassistant/components/tuya/climate.py
index 1780256a740..deccb08c5aa 100644
--- a/homeassistant/components/tuya/climate.py
+++ b/homeassistant/components/tuya/climate.py
@@ -21,7 +21,7 @@ from homeassistant.components.climate import (
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode, DPType
@@ -84,7 +84,9 @@ CLIMATE_DESCRIPTIONS: dict[str, TuyaClimateEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya climate dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/const.py b/homeassistant/components/tuya/const.py
index 08bdef474ef..a40260ed787 100644
--- a/homeassistant/components/tuya/const.py
+++ b/homeassistant/components/tuya/const.py
@@ -333,6 +333,12 @@ class DPCode(StrEnum):
TEMP_CONTROLLER = "temp_controller"
TEMP_CURRENT = "temp_current" # Current temperature in °C
TEMP_CURRENT_F = "temp_current_f" # Current temperature in °F
+ TEMP_CURRENT_EXTERNAL = (
+ "temp_current_external" # Current external temperature in Celsius
+ )
+ TEMP_CURRENT_EXTERNAL_F = (
+ "temp_current_external_f" # Current external temperature in Fahrenheit
+ )
TEMP_INDOOR = "temp_indoor" # Indoor temperature in °C
TEMP_SET = "temp_set" # Set the temperature in °C
TEMP_SET_F = "temp_set_f" # Set the temperature in °F
diff --git a/homeassistant/components/tuya/cover.py b/homeassistant/components/tuya/cover.py
index 9c3269c27f2..315075e7f37 100644
--- a/homeassistant/components/tuya/cover.py
+++ b/homeassistant/components/tuya/cover.py
@@ -17,7 +17,7 @@ from homeassistant.components.cover import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode, DPType
@@ -142,7 +142,9 @@ COVERS: dict[str, tuple[TuyaCoverEntityDescription, ...]] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya cover dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/fan.py b/homeassistant/components/tuya/fan.py
index ffab9efdde8..3b951e75da1 100644
--- a/homeassistant/components/tuya/fan.py
+++ b/homeassistant/components/tuya/fan.py
@@ -14,7 +14,7 @@ from homeassistant.components.fan import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -34,7 +34,9 @@ TUYA_SUPPORT_TYPE = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tuya fan dynamically through tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/humidifier.py b/homeassistant/components/tuya/humidifier.py
index cb872d67719..6c47148eeda 100644
--- a/homeassistant/components/tuya/humidifier.py
+++ b/homeassistant/components/tuya/humidifier.py
@@ -14,7 +14,7 @@ from homeassistant.components.humidifier import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode, DPType
@@ -55,7 +55,9 @@ HUMIDIFIERS: dict[str, TuyaHumidifierEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya (de)humidifier dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/light.py b/homeassistant/components/tuya/light.py
index d7dffc16b58..67a94c4e267 100644
--- a/homeassistant/components/tuya/light.py
+++ b/homeassistant/components/tuya/light.py
@@ -20,7 +20,7 @@ from homeassistant.components.light import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from . import TuyaConfigEntry
@@ -327,6 +327,18 @@ LIGHTS: dict[str, tuple[TuyaLightEntityDescription, ...]] = {
brightness_min=DPCode.BRIGHTNESS_MIN_1,
),
),
+ # Outdoor Flood Light
+ # Not documented
+ "tyd": (
+ TuyaLightEntityDescription(
+ key=DPCode.SWITCH_LED,
+ name=None,
+ color_mode=DPCode.WORK_MODE,
+ brightness=DPCode.BRIGHT_VALUE,
+ color_temp=DPCode.TEMP_VALUE,
+ color_data=DPCode.COLOUR_DATA,
+ ),
+ ),
# Solar Light
# https://developer.tuya.com/en/docs/iot/tynd?id=Kaof8j02e1t98
"tyndj": (
@@ -392,6 +404,10 @@ LIGHTS["cz"] = LIGHTS["kg"]
# https://developer.tuya.com/en/docs/iot/s?id=K9gf7o5prgf7s
LIGHTS["pc"] = LIGHTS["kg"]
+# Smart Camera - Low power consumption camera (duplicate of `sp`)
+# Undocumented, see https://github.com/home-assistant/core/issues/132844
+LIGHTS["dghsxj"] = LIGHTS["sp"]
+
# Dimmer (duplicate of `tgq`)
# https://developer.tuya.com/en/docs/iot/tgq?id=Kaof8ke9il4k4
LIGHTS["tdq"] = LIGHTS["tgq"]
@@ -421,7 +437,9 @@ class ColorData:
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tuya light dynamically through tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/number.py b/homeassistant/components/tuya/number.py
index 8d5b5dbfa19..d4fe7836daa 100644
--- a/homeassistant/components/tuya/number.py
+++ b/homeassistant/components/tuya/number.py
@@ -12,7 +12,7 @@ from homeassistant.components.number import (
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import DEVICE_CLASS_UNITS, DOMAIN, TUYA_DISCOVERY_NEW, DPCode, DPType
@@ -305,9 +305,15 @@ NUMBERS: dict[str, tuple[NumberEntityDescription, ...]] = {
),
}
+# Smart Camera - Low power consumption camera (duplicate of `sp`)
+# Undocumented, see https://github.com/home-assistant/core/issues/132844
+NUMBERS["dghsxj"] = NUMBERS["sp"]
+
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya number dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/scene.py b/homeassistant/components/tuya/scene.py
index dbc849356b2..4ad027d39ee 100644
--- a/homeassistant/components/tuya/scene.py
+++ b/homeassistant/components/tuya/scene.py
@@ -9,14 +9,16 @@ from tuya_sharing import Manager, SharingScene
from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import DOMAIN
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya scenes."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/select.py b/homeassistant/components/tuya/select.py
index 831d3cb3e0c..553191b7d45 100644
--- a/homeassistant/components/tuya/select.py
+++ b/homeassistant/components/tuya/select.py
@@ -8,7 +8,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode, DPType
@@ -326,9 +326,15 @@ SELECTS["cz"] = SELECTS["kg"]
# https://developer.tuya.com/en/docs/iot/s?id=K9gf7o5prgf7s
SELECTS["pc"] = SELECTS["kg"]
+# Smart Camera - Low power consumption camera (duplicate of `sp`)
+# Undocumented, see https://github.com/home-assistant/core/issues/132844
+SELECTS["dghsxj"] = SELECTS["sp"]
+
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya select dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/sensor.py b/homeassistant/components/tuya/sensor.py
index 756564c6a03..b1150be306a 100644
--- a/homeassistant/components/tuya/sensor.py
+++ b/homeassistant/components/tuya/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import TuyaConfigEntry
@@ -715,6 +715,47 @@ SENSORS: dict[str, tuple[TuyaSensorEntityDescription, ...]] = {
),
*BATTERY_SENSORS,
),
+ # Temperature and Humidity Sensor with External Probe
+ # New undocumented category qxj, see https://github.com/home-assistant/core/issues/136472
+ "qxj": (
+ TuyaSensorEntityDescription(
+ key=DPCode.VA_TEMPERATURE,
+ translation_key="temperature",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ TuyaSensorEntityDescription(
+ key=DPCode.TEMP_CURRENT,
+ translation_key="temperature",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ TuyaSensorEntityDescription(
+ key=DPCode.TEMP_CURRENT_EXTERNAL,
+ translation_key="temperature_external",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ TuyaSensorEntityDescription(
+ key=DPCode.VA_HUMIDITY,
+ translation_key="humidity",
+ device_class=SensorDeviceClass.HUMIDITY,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ TuyaSensorEntityDescription(
+ key=DPCode.HUMIDITY_VALUE,
+ translation_key="humidity",
+ device_class=SensorDeviceClass.HUMIDITY,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ TuyaSensorEntityDescription(
+ key=DPCode.BRIGHT_VALUE,
+ translation_key="illuminance",
+ device_class=SensorDeviceClass.ILLUMINANCE,
+ state_class=SensorStateClass.MEASUREMENT,
+ ),
+ *BATTERY_SENSORS,
+ ),
# Pressure Sensor
# https://developer.tuya.com/en/docs/iot/categoryylcg?id=Kaiuz3kc2e4gm
"ylcg": (
@@ -1220,9 +1261,15 @@ SENSORS["cz"] = SENSORS["kg"]
# https://developer.tuya.com/en/docs/iot/s?id=K9gf7o5prgf7s
SENSORS["pc"] = SENSORS["kg"]
+# Smart Camera - Low power consumption camera (duplicate of `sp`)
+# Undocumented, see https://github.com/home-assistant/core/issues/132844
+SENSORS["dghsxj"] = SENSORS["sp"]
+
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya sensor dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/siren.py b/homeassistant/components/tuya/siren.py
index 6f7dfe4c96c..039442dafe5 100644
--- a/homeassistant/components/tuya/siren.py
+++ b/homeassistant/components/tuya/siren.py
@@ -14,7 +14,7 @@ from homeassistant.components.siren import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode
@@ -54,9 +54,15 @@ SIRENS: dict[str, tuple[SirenEntityDescription, ...]] = {
),
}
+# Smart Camera - Low power consumption camera (duplicate of `sp`)
+# Undocumented, see https://github.com/home-assistant/core/issues/132844
+SIRENS["dghsxj"] = SIRENS["sp"]
+
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya siren dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/strings.json b/homeassistant/components/tuya/strings.json
index 8ec61cc8aa5..83847d32fb5 100644
--- a/homeassistant/components/tuya/strings.json
+++ b/homeassistant/components/tuya/strings.json
@@ -469,6 +469,9 @@
"temperature": {
"name": "[%key:component::sensor::entity_component::temperature::name%]"
},
+ "temperature_external": {
+ "name": "Probe temperature"
+ },
"humidity": {
"name": "[%key:component::sensor::entity_component::humidity::name%]"
},
diff --git a/homeassistant/components/tuya/switch.py b/homeassistant/components/tuya/switch.py
index 2b5e6fec4a6..4000e8d9b24 100644
--- a/homeassistant/components/tuya/switch.py
+++ b/homeassistant/components/tuya/switch.py
@@ -14,7 +14,7 @@ from homeassistant.components.switch import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode
@@ -612,6 +612,15 @@ SWITCHES: dict[str, tuple[SwitchEntityDescription, ...]] = {
device_class=SwitchDeviceClass.OUTLET,
),
),
+ # SIREN: Siren (switch) with Temperature and Humidity Sensor with External Probe
+ # New undocumented category qxj, see https://github.com/home-assistant/core/issues/136472
+ "qxj": (
+ SwitchEntityDescription(
+ key=DPCode.SWITCH,
+ translation_key="switch",
+ device_class=SwitchDeviceClass.OUTLET,
+ ),
+ ),
# Ceiling Light
# https://developer.tuya.com/en/docs/iot/ceiling-light?id=Kaiuz03xxfc4r
"xdd": (
@@ -726,9 +735,15 @@ SWITCHES: dict[str, tuple[SwitchEntityDescription, ...]] = {
# https://developer.tuya.com/en/docs/iot/s?id=K9gf7o5prgf7s
SWITCHES["cz"] = SWITCHES["pc"]
+# Smart Camera - Low power consumption camera (duplicate of `sp`)
+# Undocumented, see https://github.com/home-assistant/core/issues/132844
+SWITCHES["dghsxj"] = SWITCHES["sp"]
+
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up tuya sensors dynamically through tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/tuya/vacuum.py b/homeassistant/components/tuya/vacuum.py
index bab9ac309ec..e36a682fa4e 100644
--- a/homeassistant/components/tuya/vacuum.py
+++ b/homeassistant/components/tuya/vacuum.py
@@ -13,7 +13,7 @@ from homeassistant.components.vacuum import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import TuyaConfigEntry
from .const import TUYA_DISCOVERY_NEW, DPCode, DPType
@@ -48,7 +48,9 @@ TUYA_STATUS_TO_HA = {
async def async_setup_entry(
- hass: HomeAssistant, entry: TuyaConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: TuyaConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Tuya vacuum dynamically through Tuya discovery."""
hass_data = entry.runtime_data
diff --git a/homeassistant/components/twentemilieu/calendar.py b/homeassistant/components/twentemilieu/calendar.py
index 606fb4913d1..19e3f4f3337 100644
--- a/homeassistant/components/twentemilieu/calendar.py
+++ b/homeassistant/components/twentemilieu/calendar.py
@@ -7,7 +7,7 @@ from datetime import datetime, timedelta
from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.const import CONF_ID
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import WASTE_TYPE_TO_DESCRIPTION
@@ -18,7 +18,7 @@ from .entity import TwenteMilieuEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: TwenteMilieuConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Twente Milieu calendar based on a config entry."""
async_add_entities([TwenteMilieuCalendar(entry)])
diff --git a/homeassistant/components/twentemilieu/sensor.py b/homeassistant/components/twentemilieu/sensor.py
index 4605ede1f87..81751d10a81 100644
--- a/homeassistant/components/twentemilieu/sensor.py
+++ b/homeassistant/components/twentemilieu/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import CONF_ID
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import TwenteMilieuConfigEntry
@@ -65,7 +65,7 @@ SENSORS: tuple[TwenteMilieuSensorDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: TwenteMilieuConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Twente Milieu sensor based on a config entry."""
async_add_entities(
diff --git a/homeassistant/components/twilio/strings.json b/homeassistant/components/twilio/strings.json
index 871711ff087..f4b7dee707f 100644
--- a/homeassistant/components/twilio/strings.json
+++ b/homeassistant/components/twilio/strings.json
@@ -12,7 +12,7 @@
"webhook_not_internet_accessible": "[%key:common::config_flow::abort::webhook_not_internet_accessible%]"
},
"create_entry": {
- "default": "To send events to Home Assistant, you will need to setup [Webhooks with Twilio]({twilio_url}).\n\nFill in the following info:\n\n- URL: `{webhook_url}`\n- Method: POST\n- Content Type: application/x-www-form-urlencoded\n\nSee [the documentation]({docs_url}) on how to configure automations to handle incoming data."
+ "default": "To send events to Home Assistant, you will need to set up [webhooks with Twilio]({twilio_url}).\n\nFill in the following info:\n\n- URL: `{webhook_url}`\n- Method: POST\n- Content Type: application/x-www-form-urlencoded\n\nSee [the documentation]({docs_url}) on how to configure automations to handle incoming data."
}
}
}
diff --git a/homeassistant/components/twinkly/__init__.py b/homeassistant/components/twinkly/__init__.py
index cd29ffaf423..e3b53bba6c9 100644
--- a/homeassistant/components/twinkly/__init__.py
+++ b/homeassistant/components/twinkly/__init__.py
@@ -5,23 +5,19 @@ import logging
from aiohttp import ClientError
from ttls.client import Twinkly
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import DOMAIN
-from .coordinator import TwinklyCoordinator
+from .coordinator import TwinklyConfigEntry, TwinklyCoordinator
PLATFORMS = [Platform.LIGHT, Platform.SELECT]
_LOGGER = logging.getLogger(__name__)
-type TwinklyConfigEntry = ConfigEntry[TwinklyCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: TwinklyConfigEntry) -> bool:
"""Set up entries from config flow."""
# We setup the client here so if at some point we add any other entity for this device,
@@ -30,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TwinklyConfigEntry) -> b
client = Twinkly(host, async_get_clientsession(hass))
- coordinator = TwinklyCoordinator(hass, client)
+ coordinator = TwinklyCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/twinkly/coordinator.py b/homeassistant/components/twinkly/coordinator.py
index 627fb0b39ba..2c2fc2a41d4 100644
--- a/homeassistant/components/twinkly/coordinator.py
+++ b/homeassistant/components/twinkly/coordinator.py
@@ -9,6 +9,7 @@ from aiohttp import ClientError
from awesomeversion import AwesomeVersion
from ttls.client import Twinkly, TwinklyError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -17,6 +18,8 @@ from .const import DEV_NAME, DOMAIN, MIN_EFFECT_VERSION
_LOGGER = logging.getLogger(__name__)
+type TwinklyConfigEntry = ConfigEntry[TwinklyCoordinator]
+
@dataclass
class TwinklyData:
@@ -33,15 +36,19 @@ class TwinklyData:
class TwinklyCoordinator(DataUpdateCoordinator[TwinklyData]):
"""Class to manage fetching Twinkly data from API."""
+ config_entry: TwinklyConfigEntry
software_version: str
supports_effects: bool
device_name: str
- def __init__(self, hass: HomeAssistant, client: Twinkly) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: TwinklyConfigEntry, client: Twinkly
+ ) -> None:
"""Initialize global Twinkly data updater."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=30),
)
diff --git a/homeassistant/components/twinkly/diagnostics.py b/homeassistant/components/twinkly/diagnostics.py
index d732ce14929..2bf46a208e8 100644
--- a/homeassistant/components/twinkly/diagnostics.py
+++ b/homeassistant/components/twinkly/diagnostics.py
@@ -10,8 +10,8 @@ from homeassistant.const import ATTR_SW_VERSION, CONF_HOST, CONF_IP_ADDRESS, CON
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from . import TwinklyConfigEntry
from .const import DOMAIN
+from .coordinator import TwinklyConfigEntry
TO_REDACT = [CONF_HOST, CONF_IP_ADDRESS, CONF_MAC]
diff --git a/homeassistant/components/twinkly/icons.json b/homeassistant/components/twinkly/icons.json
index 82c95aebce6..d57d54aa507 100644
--- a/homeassistant/components/twinkly/icons.json
+++ b/homeassistant/components/twinkly/icons.json
@@ -4,6 +4,11 @@
"light": {
"default": "mdi:string-lights"
}
+ },
+ "select": {
+ "mode": {
+ "default": "mdi:cogs"
+ }
}
}
}
diff --git a/homeassistant/components/twinkly/light.py b/homeassistant/components/twinkly/light.py
index 31e95d70fc0..c270421d8cd 100644
--- a/homeassistant/components/twinkly/light.py
+++ b/homeassistant/components/twinkly/light.py
@@ -15,10 +15,10 @@ from homeassistant.components.light import (
LightEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import TwinklyConfigEntry, TwinklyCoordinator
from .const import DEV_LED_PROFILE, DEV_PROFILE_RGB, DEV_PROFILE_RGBW
+from .coordinator import TwinklyConfigEntry, TwinklyCoordinator
from .entity import TwinklyEntity
_LOGGER = logging.getLogger(__name__)
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TwinklyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Setups an entity from a config entry (UI config flow)."""
entity = TwinklyLight(config_entry.runtime_data)
diff --git a/homeassistant/components/twinkly/select.py b/homeassistant/components/twinkly/select.py
index 38e5c9a6fc7..a5283b3f91d 100644
--- a/homeassistant/components/twinkly/select.py
+++ b/homeassistant/components/twinkly/select.py
@@ -8,9 +8,9 @@ from ttls.client import TWINKLY_MODES
from homeassistant.components.select import SelectEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import TwinklyConfigEntry, TwinklyCoordinator
+from .coordinator import TwinklyConfigEntry, TwinklyCoordinator
from .entity import TwinklyEntity
_LOGGER = logging.getLogger(__name__)
@@ -19,7 +19,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: TwinklyConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a mode select from a config entry."""
entity = TwinklyModeSelect(config_entry.runtime_data)
@@ -29,7 +29,7 @@ async def async_setup_entry(
class TwinklyModeSelect(TwinklyEntity, SelectEntity):
"""Twinkly Mode Selection."""
- _attr_name = "Mode"
+ _attr_translation_key = "mode"
_attr_options = TWINKLY_MODES
def __init__(self, coordinator: TwinklyCoordinator) -> None:
diff --git a/homeassistant/components/twinkly/strings.json b/homeassistant/components/twinkly/strings.json
index bbc3d67373d..c2e0efef92c 100644
--- a/homeassistant/components/twinkly/strings.json
+++ b/homeassistant/components/twinkly/strings.json
@@ -20,5 +20,21 @@
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_device%]"
}
+ },
+ "entity": {
+ "select": {
+ "mode": {
+ "name": "Mode",
+ "state": {
+ "color": "Color",
+ "demo": "Demo",
+ "effect": "Effect",
+ "movie": "Uploaded effect",
+ "off": "[%key:common::state::off%]",
+ "playlist": "Playlist",
+ "rt": "Real time"
+ }
+ }
+ }
}
}
diff --git a/homeassistant/components/twitch/sensor.py b/homeassistant/components/twitch/sensor.py
index b407eae0319..deec319e5cf 100644
--- a/homeassistant/components/twitch/sensor.py
+++ b/homeassistant/components/twitch/sensor.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -32,7 +32,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: TwitchConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize entries."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/uk_transport/sensor.py b/homeassistant/components/uk_transport/sensor.py
index b06d0e24891..594d46c74ab 100644
--- a/homeassistant/components/uk_transport/sensor.py
+++ b/homeassistant/components/uk_transport/sensor.py
@@ -32,6 +32,7 @@ ATTR_NEXT_BUSES = "next_buses"
ATTR_STATION_CODE = "station_code"
ATTR_CALLING_AT = "calling_at"
ATTR_NEXT_TRAINS = "next_trains"
+ATTR_LAST_UPDATED = "last_updated"
CONF_API_APP_KEY = "app_key"
CONF_API_APP_ID = "app_id"
@@ -199,7 +200,9 @@ class UkTransportLiveBusTimeSensor(UkTransportSensor):
def extra_state_attributes(self) -> dict[str, Any] | None:
"""Return other details about the sensor state."""
if self._data is not None:
- attrs = {ATTR_NEXT_BUSES: self._next_buses}
+ attrs = {
+ ATTR_NEXT_BUSES: self._next_buses,
+ }
for key in (
ATTR_ATCOCODE,
ATTR_LOCALITY,
@@ -272,6 +275,7 @@ class UkTransportLiveTrainTimeSensor(UkTransportSensor):
attrs = {
ATTR_STATION_CODE: self._station_code,
ATTR_CALLING_AT: self._calling_at,
+ ATTR_LAST_UPDATED: self._data[ATTR_REQUEST_TIME],
}
if self._next_trains:
attrs[ATTR_NEXT_TRAINS] = self._next_trains
diff --git a/homeassistant/components/ukraine_alarm/__init__.py b/homeassistant/components/ukraine_alarm/__init__.py
index d850ed6eba8..3658b821625 100644
--- a/homeassistant/components/ukraine_alarm/__init__.py
+++ b/homeassistant/components/ukraine_alarm/__init__.py
@@ -3,7 +3,6 @@
from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_REGION
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -13,11 +12,9 @@ from .coordinator import UkraineAlarmDataUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Ukraine Alarm as config entry."""
- region_id = entry.data[CONF_REGION]
-
websession = async_get_clientsession(hass)
- coordinator = UkraineAlarmDataUpdateCoordinator(hass, websession, region_id)
+ coordinator = UkraineAlarmDataUpdateCoordinator(hass, entry, websession)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
diff --git a/homeassistant/components/ukraine_alarm/binary_sensor.py b/homeassistant/components/ukraine_alarm/binary_sensor.py
index 30cb8e0f553..9009031ea14 100644
--- a/homeassistant/components/ukraine_alarm/binary_sensor.py
+++ b/homeassistant/components/ukraine_alarm/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import (
@@ -64,7 +64,7 @@ BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Ukraine Alarm binary sensor entities based on a config entry."""
name = config_entry.data[CONF_NAME]
diff --git a/homeassistant/components/ukraine_alarm/coordinator.py b/homeassistant/components/ukraine_alarm/coordinator.py
index fbf7c9f81c2..b4e1decb1a1 100644
--- a/homeassistant/components/ukraine_alarm/coordinator.py
+++ b/homeassistant/components/ukraine_alarm/coordinator.py
@@ -10,6 +10,8 @@ import aiohttp
from aiohttp import ClientSession
from uasiren.client import Client
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_REGION
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -23,17 +25,25 @@ UPDATE_INTERVAL = timedelta(seconds=10)
class UkraineAlarmDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching Ukraine Alarm API."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
session: ClientSession,
- region_id: str,
) -> None:
"""Initialize."""
- self.region_id = region_id
+ self.region_id = config_entry.data[CONF_REGION]
self.uasiren = Client(session)
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=UPDATE_INTERVAL,
+ )
async def _async_update_data(self) -> dict[str, Any]:
"""Update data via library."""
@@ -42,7 +52,7 @@ class UkraineAlarmDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
except aiohttp.ClientError as error:
raise UpdateFailed(f"Error fetching alerts from API: {error}") from error
- current = {alert_type: False for alert_type in ALERT_TYPES}
+ current = dict.fromkeys(ALERT_TYPES, False)
for alert in res[0]["activeAlerts"]:
current[alert["type"]] = True
diff --git a/homeassistant/components/unifi/button.py b/homeassistant/components/unifi/button.py
index 25c6816d794..3e5ef62f49e 100644
--- a/homeassistant/components/unifi/button.py
+++ b/homeassistant/components/unifi/button.py
@@ -31,7 +31,7 @@ from homeassistant.components.button import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import UnifiConfigEntry
from .entity import (
@@ -135,7 +135,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiButtonEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UnifiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up button platform for UniFi Network integration."""
config_entry.runtime_data.entity_loader.register_platform(
diff --git a/homeassistant/components/unifi/device_tracker.py b/homeassistant/components/unifi/device_tracker.py
index da5ca74fc37..a26232664a8 100644
--- a/homeassistant/components/unifi/device_tracker.py
+++ b/homeassistant/components/unifi/device_tracker.py
@@ -26,7 +26,7 @@ from homeassistant.components.device_tracker import (
from homeassistant.core import Event as core_Event, HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import UnifiConfigEntry
@@ -222,7 +222,7 @@ def async_update_unique_id(hass: HomeAssistant, config_entry: UnifiConfigEntry)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UnifiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for UniFi Network integration."""
async_update_unique_id(hass, config_entry)
diff --git a/homeassistant/components/unifi/hub/entity_loader.py b/homeassistant/components/unifi/hub/entity_loader.py
index 64403152b0c..84948a92e98 100644
--- a/homeassistant/components/unifi/hub/entity_loader.py
+++ b/homeassistant/components/unifi/hub/entity_loader.py
@@ -46,6 +46,7 @@ class UnifiEntityLoader:
hub.api.port_forwarding.update,
hub.api.sites.update,
hub.api.system_information.update,
+ hub.api.firewall_policies.update,
hub.api.traffic_rules.update,
hub.api.traffic_routes.update,
hub.api.wlans.update,
diff --git a/homeassistant/components/unifi/icons.json b/homeassistant/components/unifi/icons.json
index 6874bb5ae03..616d7cb185f 100644
--- a/homeassistant/components/unifi/icons.json
+++ b/homeassistant/components/unifi/icons.json
@@ -55,6 +55,9 @@
"off": "mdi:network-off"
}
},
+ "firewall_policy_control": {
+ "default": "mdi:security-network"
+ },
"port_forward_control": {
"default": "mdi:upload-network"
},
diff --git a/homeassistant/components/unifi/image.py b/homeassistant/components/unifi/image.py
index f1ada9a01e0..f3045d5fc1c 100644
--- a/homeassistant/components/unifi/image.py
+++ b/homeassistant/components/unifi/image.py
@@ -16,7 +16,7 @@ from aiounifi.models.wlan import Wlan
from homeassistant.components.image import ImageEntity, ImageEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import UnifiConfigEntry
@@ -67,7 +67,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiImageEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UnifiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up image platform for UniFi Network integration."""
config_entry.runtime_data.entity_loader.register_platform(
diff --git a/homeassistant/components/unifi/manifest.json b/homeassistant/components/unifi/manifest.json
index ce573592153..dd255c57c13 100644
--- a/homeassistant/components/unifi/manifest.json
+++ b/homeassistant/components/unifi/manifest.json
@@ -7,7 +7,7 @@
"integration_type": "hub",
"iot_class": "local_push",
"loggers": ["aiounifi"],
- "requirements": ["aiounifi==81"],
+ "requirements": ["aiounifi==83"],
"ssdp": [
{
"manufacturer": "Ubiquiti Networks",
diff --git a/homeassistant/components/unifi/sensor.py b/homeassistant/components/unifi/sensor.py
index fd78c606043..47a2c2ba62e 100644
--- a/homeassistant/components/unifi/sensor.py
+++ b/homeassistant/components/unifi/sensor.py
@@ -46,7 +46,7 @@ from homeassistant.const import (
)
from homeassistant.core import Event as core_Event, HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util, slugify
@@ -644,7 +644,7 @@ ENTITY_DESCRIPTIONS += make_wan_latency_sensors() + make_device_temperatur_senso
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UnifiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for UniFi Network integration."""
config_entry.runtime_data.entity_loader.register_platform(
diff --git a/homeassistant/components/unifi/services.py b/homeassistant/components/unifi/services.py
index fc63c092d56..9d4d92839fc 100644
--- a/homeassistant/components/unifi/services.py
+++ b/homeassistant/components/unifi/services.py
@@ -6,7 +6,6 @@ from typing import Any
from aiounifi.models.client import ClientReconnectRequest, ClientRemoveRequest
import voluptuous as vol
-from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import ATTR_DEVICE_ID
from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.helpers import device_registry as dr
@@ -67,9 +66,9 @@ async def async_reconnect_client(hass: HomeAssistant, data: Mapping[str, Any]) -
if mac == "":
return
- for config_entry in hass.config_entries.async_entries(UNIFI_DOMAIN):
- if config_entry.state is not ConfigEntryState.LOADED or (
- ((hub := config_entry.runtime_data) and not hub.available)
+ for config_entry in hass.config_entries.async_loaded_entries(UNIFI_DOMAIN):
+ if (
+ (not (hub := config_entry.runtime_data).available)
or (client := hub.api.clients.get(mac)) is None
or client.is_wired
):
@@ -85,10 +84,8 @@ async def async_remove_clients(hass: HomeAssistant, data: Mapping[str, Any]) ->
- Total time between first seen and last seen is less than 15 minutes.
- Neither IP, hostname nor name is configured.
"""
- for config_entry in hass.config_entries.async_entries(UNIFI_DOMAIN):
- if config_entry.state is not ConfigEntryState.LOADED or (
- (hub := config_entry.runtime_data) and not hub.available
- ):
+ for config_entry in hass.config_entries.async_loaded_entries(UNIFI_DOMAIN):
+ if not (hub := config_entry.runtime_data).available:
continue
clients_to_remove = []
diff --git a/homeassistant/components/unifi/switch.py b/homeassistant/components/unifi/switch.py
index 91e4a0222f6..282d0c9ae93 100644
--- a/homeassistant/components/unifi/switch.py
+++ b/homeassistant/components/unifi/switch.py
@@ -4,6 +4,7 @@ Support for controlling power supply of clients which are powered over Ethernet
Support for controlling network access of clients selected in option flow.
Support for controlling deep packet inspection (DPI) restriction groups.
Support for controlling WLAN availability.
+Support for controlling zone based traffic rules.
"""
from __future__ import annotations
@@ -17,6 +18,7 @@ import aiounifi
from aiounifi.interfaces.api_handlers import ItemEvent
from aiounifi.interfaces.clients import Clients
from aiounifi.interfaces.dpi_restriction_groups import DPIRestrictionGroups
+from aiounifi.interfaces.firewall_policies import FirewallPolicies
from aiounifi.interfaces.outlets import Outlets
from aiounifi.interfaces.port_forwarding import PortForwarding
from aiounifi.interfaces.ports import Ports
@@ -29,6 +31,7 @@ from aiounifi.models.device import DeviceSetOutletRelayRequest
from aiounifi.models.dpi_restriction_app import DPIRestrictionAppEnableRequest
from aiounifi.models.dpi_restriction_group import DPIRestrictionGroup
from aiounifi.models.event import Event, EventKey
+from aiounifi.models.firewall_policy import FirewallPolicy, FirewallPolicyUpdateRequest
from aiounifi.models.outlet import Outlet
from aiounifi.models.port import Port
from aiounifi.models.port_forward import PortForward, PortForwardEnableRequest
@@ -46,7 +49,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import UnifiConfigEntry
from .const import ATTR_MANUFACTURER, DOMAIN as UNIFI_DOMAIN
@@ -129,6 +132,24 @@ async def async_dpi_group_control_fn(hub: UnifiHub, obj_id: str, target: bool) -
)
+async def async_firewall_policy_control_fn(
+ hub: UnifiHub, obj_id: str, target: bool
+) -> None:
+ """Control firewall policy state."""
+ policy = hub.api.firewall_policies[obj_id].raw
+ policy["enabled"] = target
+ await hub.api.request(FirewallPolicyUpdateRequest.create(policy))
+ # Update the policies so the UI is updated appropriately
+ await hub.api.firewall_policies.update()
+
+
+@callback
+def async_firewall_policy_supported_fn(hub: UnifiHub, obj_id: str) -> bool:
+ """Check if firewall policy is able to be controlled. Predefined policies are unable to be turned off."""
+ policy = hub.api.firewall_policies[obj_id]
+ return not policy.predefined
+
+
@callback
def async_outlet_switching_supported_fn(hub: UnifiHub, obj_id: str) -> bool:
"""Determine if an outlet supports switching."""
@@ -236,6 +257,20 @@ ENTITY_DESCRIPTIONS: tuple[UnifiSwitchEntityDescription, ...] = (
supported_fn=lambda hub, obj_id: bool(hub.api.dpi_groups[obj_id].dpiapp_ids),
unique_id_fn=lambda hub, obj_id: obj_id,
),
+ UnifiSwitchEntityDescription[FirewallPolicies, FirewallPolicy](
+ key="Firewall policy control",
+ translation_key="firewall_policy_control",
+ device_class=SwitchDeviceClass.SWITCH,
+ entity_category=EntityCategory.CONFIG,
+ api_handler_fn=lambda api: api.firewall_policies,
+ control_fn=async_firewall_policy_control_fn,
+ device_info_fn=async_unifi_network_device_info_fn,
+ is_on_fn=lambda hub, firewall_policy: firewall_policy.enabled,
+ name_fn=lambda firewall_policy: firewall_policy.name,
+ object_fn=lambda api, obj_id: api.firewall_policies[obj_id],
+ unique_id_fn=lambda hub, obj_id: f"firewall_policy-{obj_id}",
+ supported_fn=async_firewall_policy_supported_fn,
+ ),
UnifiSwitchEntityDescription[Outlets, Outlet](
key="Outlet control",
device_class=SwitchDeviceClass.OUTLET,
@@ -352,7 +387,7 @@ def async_update_unique_id(hass: HomeAssistant, config_entry: UnifiConfigEntry)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UnifiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches for UniFi Network integration."""
async_update_unique_id(hass, config_entry)
diff --git a/homeassistant/components/unifi/update.py b/homeassistant/components/unifi/update.py
index 65202045a05..589b2ff1215 100644
--- a/homeassistant/components/unifi/update.py
+++ b/homeassistant/components/unifi/update.py
@@ -19,7 +19,7 @@ from homeassistant.components.update import (
UpdateEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import UnifiConfigEntry
from .entity import (
@@ -68,7 +68,7 @@ ENTITY_DESCRIPTIONS: tuple[UnifiUpdateEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UnifiConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up update entities for UniFi Network integration."""
config_entry.runtime_data.entity_loader.register_platform(
diff --git a/homeassistant/components/unifiprotect/binary_sensor.py b/homeassistant/components/unifiprotect/binary_sensor.py
index a88d4b65678..0d904d3c3ba 100644
--- a/homeassistant/components/unifiprotect/binary_sensor.py
+++ b/homeassistant/components/unifiprotect/binary_sensor.py
@@ -23,7 +23,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .data import ProtectData, ProtectDeviceType, UFPConfigEntry
from .entity import (
@@ -769,7 +769,7 @@ def _async_nvr_entities(
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensors for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/button.py b/homeassistant/components/unifiprotect/button.py
index b24c90be3ec..7b766299946 100644
--- a/homeassistant/components/unifiprotect/button.py
+++ b/homeassistant/components/unifiprotect/button.py
@@ -19,7 +19,7 @@ from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DEVICES_THAT_ADOPT, DOMAIN
from .data import ProtectDeviceType, UFPConfigEntry
@@ -120,7 +120,7 @@ def _async_remove_adopt_button(
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Discover devices on a UniFi Protect NVR."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/camera.py b/homeassistant/components/unifiprotect/camera.py
index 0b1c03b8dd6..3947324fd73 100644
--- a/homeassistant/components/unifiprotect/camera.py
+++ b/homeassistant/components/unifiprotect/camera.py
@@ -16,7 +16,7 @@ from homeassistant.components.camera import Camera, CameraEntityFeature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import issue_registry as ir
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.issue_registry import IssueSeverity
from .const import (
@@ -138,7 +138,7 @@ def _async_camera_entities(
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Discover cameras on a UniFi Protect NVR."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/event.py b/homeassistant/components/unifiprotect/event.py
index 78fdf7746de..cb9090dd530 100644
--- a/homeassistant/components/unifiprotect/event.py
+++ b/homeassistant/components/unifiprotect/event.py
@@ -10,7 +10,7 @@ from homeassistant.components.event import (
EventEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import Bootstrap
from .const import (
@@ -218,7 +218,7 @@ def _async_event_entities(
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up event entities for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/light.py b/homeassistant/components/unifiprotect/light.py
index fcdfe5e85b8..873f715de58 100644
--- a/homeassistant/components/unifiprotect/light.py
+++ b/homeassistant/components/unifiprotect/light.py
@@ -9,7 +9,7 @@ from uiprotect.data import Light, ModelType, ProtectAdoptableDeviceModel
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .data import ProtectDeviceType, UFPConfigEntry
from .entity import ProtectDeviceEntity
@@ -20,7 +20,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up lights for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/lock.py b/homeassistant/components/unifiprotect/lock.py
index 3e9372db0e5..79ed47a6c3b 100644
--- a/homeassistant/components/unifiprotect/lock.py
+++ b/homeassistant/components/unifiprotect/lock.py
@@ -14,7 +14,7 @@ from uiprotect.data import (
from homeassistant.components.lock import LockEntity, LockEntityDescription
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .data import ProtectDeviceType, UFPConfigEntry
from .entity import ProtectDeviceEntity
@@ -25,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up locks on a UniFi Protect NVR."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/media_player.py b/homeassistant/components/unifiprotect/media_player.py
index 5f9991b257b..a1e60931026 100644
--- a/homeassistant/components/unifiprotect/media_player.py
+++ b/homeassistant/components/unifiprotect/media_player.py
@@ -21,7 +21,7 @@ from homeassistant.components.media_player import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .data import ProtectDeviceType, UFPConfigEntry
from .entity import ProtectDeviceEntity
@@ -36,7 +36,7 @@ _SPEAKER_DESCRIPTION = MediaPlayerEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Discover cameras with speakers on a UniFi Protect NVR."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/number.py b/homeassistant/components/unifiprotect/number.py
index 767128337ba..5dbf9f2b00e 100644
--- a/homeassistant/components/unifiprotect/number.py
+++ b/homeassistant/components/unifiprotect/number.py
@@ -17,7 +17,7 @@ from uiprotect.data import (
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .data import ProtectData, ProtectDeviceType, UFPConfigEntry
from .entity import (
@@ -227,7 +227,7 @@ _MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number entities for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/select.py b/homeassistant/components/unifiprotect/select.py
index 00c277c957e..054c9430387 100644
--- a/homeassistant/components/unifiprotect/select.py
+++ b/homeassistant/components/unifiprotect/select.py
@@ -29,7 +29,7 @@ from uiprotect.data import (
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import TYPE_EMPTY_VALUE
from .data import ProtectData, ProtectDeviceType, UFPConfigEntry
@@ -334,7 +334,9 @@ _MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: UFPConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number entities for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/sensor.py b/homeassistant/components/unifiprotect/sensor.py
index 09187e023a1..a719f36c2b3 100644
--- a/homeassistant/components/unifiprotect/sensor.py
+++ b/homeassistant/components/unifiprotect/sensor.py
@@ -38,7 +38,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .data import ProtectData, ProtectDeviceType, UFPConfigEntry
from .entity import (
@@ -640,7 +640,7 @@ _MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/switch.py b/homeassistant/components/unifiprotect/switch.py
index fa960261cf2..fce92912a52 100644
--- a/homeassistant/components/unifiprotect/switch.py
+++ b/homeassistant/components/unifiprotect/switch.py
@@ -18,7 +18,7 @@ from uiprotect.data import (
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .data import ProtectData, ProtectDeviceType, UFPConfigEntry
@@ -568,7 +568,7 @@ class ProtectPrivacyModeSwitch(RestoreEntity, ProtectSwitch):
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/unifiprotect/text.py b/homeassistant/components/unifiprotect/text.py
index 0c7e1322f23..1c468d44cc6 100644
--- a/homeassistant/components/unifiprotect/text.py
+++ b/homeassistant/components/unifiprotect/text.py
@@ -15,7 +15,7 @@ from uiprotect.data import (
from homeassistant.components.text import TextEntity, TextEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .data import ProtectDeviceType, UFPConfigEntry
from .entity import (
@@ -63,7 +63,7 @@ _MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]] = {
async def async_setup_entry(
hass: HomeAssistant,
entry: UFPConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up sensors for UniFi Protect integration."""
data = entry.runtime_data
diff --git a/homeassistant/components/upb/entity.py b/homeassistant/components/upb/entity.py
index 13037adf680..8a9afa453b1 100644
--- a/homeassistant/components/upb/entity.py
+++ b/homeassistant/components/upb/entity.py
@@ -30,7 +30,7 @@ class UpbEntity(Entity):
return self._element.as_dict()
@property
- def available(self):
+ def available(self) -> bool:
"""Is the entity available to be updated."""
return self._upb.is_connected()
@@ -43,7 +43,7 @@ class UpbEntity(Entity):
self._element_changed(element, changeset)
self.async_write_ha_state()
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Register callback for UPB changes and update entity state."""
self._element.add_callback(self._element_callback)
self._element_callback(self._element, {})
diff --git a/homeassistant/components/upb/light.py b/homeassistant/components/upb/light.py
index 07bd50b7d9f..0838ec3ef01 100644
--- a/homeassistant/components/upb/light.py
+++ b/homeassistant/components/upb/light.py
@@ -13,7 +13,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, UPB_BLINK_RATE_SCHEMA, UPB_BRIGHTNESS_RATE_SCHEMA
from .entity import UpbAttachedEntity
@@ -26,7 +26,7 @@ SERVICE_LIGHT_BLINK = "light_blink"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UPB light based on a config entry."""
diff --git a/homeassistant/components/upb/manifest.json b/homeassistant/components/upb/manifest.json
index e5da4c4d621..b40388be71b 100644
--- a/homeassistant/components/upb/manifest.json
+++ b/homeassistant/components/upb/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/upb",
"iot_class": "local_push",
"loggers": ["upb_lib"],
- "requirements": ["upb-lib==0.6.0"]
+ "requirements": ["upb-lib==0.6.1"]
}
diff --git a/homeassistant/components/upb/scene.py b/homeassistant/components/upb/scene.py
index 5a5e17b3e4c..45a1d664b15 100644
--- a/homeassistant/components/upb/scene.py
+++ b/homeassistant/components/upb/scene.py
@@ -6,7 +6,7 @@ from homeassistant.components.scene import Scene
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, UPB_BLINK_RATE_SCHEMA, UPB_BRIGHTNESS_RATE_SCHEMA
from .entity import UpbEntity
@@ -21,7 +21,7 @@ SERVICE_LINK_BLINK = "link_blink"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UPB link based on a config entry."""
upb = hass.data[DOMAIN][config_entry.entry_id]["upb"]
diff --git a/homeassistant/components/upcloud/binary_sensor.py b/homeassistant/components/upcloud/binary_sensor.py
index bca313d306f..923d8f2d896 100644
--- a/homeassistant/components/upcloud/binary_sensor.py
+++ b/homeassistant/components/upcloud/binary_sensor.py
@@ -5,7 +5,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import UpCloudConfigEntry
from .entity import UpCloudServerEntity
@@ -14,7 +14,7 @@ from .entity import UpCloudServerEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UpCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UpCloud server binary sensor."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/upcloud/switch.py b/homeassistant/components/upcloud/switch.py
index 97c08b19188..de180907919 100644
--- a/homeassistant/components/upcloud/switch.py
+++ b/homeassistant/components/upcloud/switch.py
@@ -6,7 +6,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.const import STATE_OFF
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import dispatcher_send
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .coordinator import UpCloudConfigEntry
from .entity import UpCloudServerEntity
@@ -17,7 +17,7 @@ SIGNAL_UPDATE_UPCLOUD = "upcloud_update"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UpCloudConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UpCloud server switch."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/update/__init__.py b/homeassistant/components/update/__init__.py
index 0ff8c448197..47cc5aa369b 100644
--- a/homeassistant/components/update/__init__.py
+++ b/homeassistant/components/update/__init__.py
@@ -226,7 +226,7 @@ class UpdateEntity(
_attr_installed_version: str | None = None
_attr_device_class: UpdateDeviceClass | None
_attr_display_precision: int
- _attr_in_progress: bool | int = False
+ _attr_in_progress: bool = False
_attr_latest_version: str | None = None
_attr_release_summary: str | None = None
_attr_release_url: str | None = None
@@ -295,7 +295,7 @@ class UpdateEntity(
)
@cached_property
- def in_progress(self) -> bool | int | None:
+ def in_progress(self) -> bool | None:
"""Update installation progress.
Needs UpdateEntityFeature.PROGRESS flag to be set for it to be used.
@@ -442,7 +442,7 @@ class UpdateEntity(
in_progress = self.in_progress
update_percentage = self.update_percentage if in_progress else None
if type(in_progress) is not bool and isinstance(in_progress, int):
- update_percentage = in_progress
+ update_percentage = in_progress # type: ignore[unreachable]
in_progress = True
else:
in_progress = self.__in_progress
diff --git a/homeassistant/components/upnp/__init__.py b/homeassistant/components/upnp/__init__.py
index aacb7538b61..757cad221b5 100644
--- a/homeassistant/components/upnp/__init__.py
+++ b/homeassistant/components/upnp/__init__.py
@@ -8,7 +8,6 @@ from datetime import timedelta
from async_upnp_client.exceptions import UpnpConnectionError
from homeassistant.components import ssdp
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@@ -28,7 +27,7 @@ from .const import (
IDENTIFIER_SERIAL_NUMBER,
LOGGER,
)
-from .coordinator import UpnpDataUpdateCoordinator
+from .coordinator import UpnpConfigEntry, UpnpDataUpdateCoordinator
from .device import async_create_device, get_preferred_location
NOTIFICATION_ID = "upnp_notification"
@@ -37,9 +36,6 @@ NOTIFICATION_TITLE = "UPnP/IGD Setup"
PLATFORMS = [Platform.BINARY_SENSOR, Platform.SENSOR]
-type UpnpConfigEntry = ConfigEntry[UpnpDataUpdateCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: UpnpConfigEntry) -> bool:
"""Set up UPnP/IGD device from a config entry."""
LOGGER.debug("Setting up config entry: %s", entry.entry_id)
@@ -176,6 +172,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: UpnpConfigEntry) -> bool
update_interval = timedelta(seconds=DEFAULT_SCAN_INTERVAL)
coordinator = UpnpDataUpdateCoordinator(
hass,
+ config_entry=entry,
device=device,
device_entry=device_entry,
update_interval=update_interval,
@@ -193,7 +190,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: UpnpConfigEntry) -> bool
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: UpnpConfigEntry) -> bool:
"""Unload a UPnP/IGD device from a config entry."""
LOGGER.debug("Unloading config entry: %s", entry.entry_id)
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/upnp/binary_sensor.py b/homeassistant/components/upnp/binary_sensor.py
index fb32946bf7d..0c7b7aa5dc2 100644
--- a/homeassistant/components/upnp/binary_sensor.py
+++ b/homeassistant/components/upnp/binary_sensor.py
@@ -11,10 +11,10 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import UpnpConfigEntry, UpnpDataUpdateCoordinator
from .const import LOGGER, WAN_STATUS
+from .coordinator import UpnpConfigEntry, UpnpDataUpdateCoordinator
from .entity import UpnpEntity, UpnpEntityDescription
@@ -38,7 +38,7 @@ SENSOR_DESCRIPTIONS: tuple[UpnpBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UpnpConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UPnP/IGD sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/upnp/coordinator.py b/homeassistant/components/upnp/coordinator.py
index 37ff700bfe2..03e4c53f143 100644
--- a/homeassistant/components/upnp/coordinator.py
+++ b/homeassistant/components/upnp/coordinator.py
@@ -6,6 +6,7 @@ from datetime import datetime, timedelta
from async_upnp_client.exceptions import UpnpCommunicationError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntry
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -13,15 +14,20 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import LOGGER
from .device import Device
+type UpnpConfigEntry = ConfigEntry[UpnpDataUpdateCoordinator]
+
class UpnpDataUpdateCoordinator(
DataUpdateCoordinator[dict[str, str | datetime | int | float | None]]
):
"""Define an object to update data from UPNP device."""
+ config_entry: UpnpConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: UpnpConfigEntry,
device: Device,
device_entry: DeviceEntry,
update_interval: timedelta,
@@ -34,6 +40,7 @@ class UpnpDataUpdateCoordinator(
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=device.name,
update_interval=update_interval,
)
diff --git a/homeassistant/components/upnp/sensor.py b/homeassistant/components/upnp/sensor.py
index aae2f8308c1..c7e343d36b5 100644
--- a/homeassistant/components/upnp/sensor.py
+++ b/homeassistant/components/upnp/sensor.py
@@ -18,9 +18,8 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import UpnpConfigEntry
from .const import (
BYTES_RECEIVED,
BYTES_SENT,
@@ -38,6 +37,7 @@ from .const import (
ROUTER_UPTIME,
WAN_STATUS,
)
+from .coordinator import UpnpConfigEntry
from .entity import UpnpEntity, UpnpEntityDescription
@@ -153,7 +153,7 @@ SENSOR_DESCRIPTIONS: tuple[UpnpSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: UpnpConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UPnP/IGD sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/uptime/sensor.py b/homeassistant/components/uptime/sensor.py
index 25917d09096..488682a79c6 100644
--- a/homeassistant/components/uptime/sensor.py
+++ b/homeassistant/components/uptime/sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import DOMAIN
@@ -15,7 +15,7 @@ from .const import DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from config_entry."""
async_add_entities([UptimeSensor(entry)])
diff --git a/homeassistant/components/uptimerobot/__init__.py b/homeassistant/components/uptimerobot/__init__.py
index afff0c8fe03..b8619b1fe39 100644
--- a/homeassistant/components/uptimerobot/__init__.py
+++ b/homeassistant/components/uptimerobot/__init__.py
@@ -8,7 +8,6 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_KEY
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
-from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import DOMAIN, PLATFORMS
@@ -24,12 +23,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"Wrong API key type detected, use the 'main' API key"
)
uptime_robot_api = UptimeRobot(key, async_get_clientsession(hass))
- dev_reg = dr.async_get(hass)
hass.data[DOMAIN][entry.entry_id] = coordinator = UptimeRobotDataUpdateCoordinator(
hass,
- config_entry_id=entry.entry_id,
- dev_reg=dev_reg,
+ entry,
api=uptime_robot_api,
)
diff --git a/homeassistant/components/uptimerobot/binary_sensor.py b/homeassistant/components/uptimerobot/binary_sensor.py
index 0c1bd972387..73f9400c013 100644
--- a/homeassistant/components/uptimerobot/binary_sensor.py
+++ b/homeassistant/components/uptimerobot/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import UptimeRobotDataUpdateCoordinator
@@ -19,7 +19,7 @@ from .entity import UptimeRobotEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UptimeRobot binary_sensors."""
coordinator: UptimeRobotDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/uptimerobot/coordinator.py b/homeassistant/components/uptimerobot/coordinator.py
index 3069884eb99..fbadc237965 100644
--- a/homeassistant/components/uptimerobot/coordinator.py
+++ b/homeassistant/components/uptimerobot/coordinator.py
@@ -26,19 +26,18 @@ class UptimeRobotDataUpdateCoordinator(DataUpdateCoordinator[list[UptimeRobotMon
def __init__(
self,
hass: HomeAssistant,
- config_entry_id: str,
- dev_reg: dr.DeviceRegistry,
+ config_entry: ConfigEntry,
api: UptimeRobot,
) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=COORDINATOR_UPDATE_INTERVAL,
)
- self._config_entry_id = config_entry_id
- self._device_registry = dev_reg
+ self._device_registry = dr.async_get(hass)
self.api = api
async def _async_update_data(self) -> list[UptimeRobotMonitor]:
@@ -58,7 +57,7 @@ class UptimeRobotDataUpdateCoordinator(DataUpdateCoordinator[list[UptimeRobotMon
current_monitors = {
list(device.identifiers)[0][1]
for device in dr.async_entries_for_config_entry(
- self._device_registry, self._config_entry_id
+ self._device_registry, self.config_entry.entry_id
)
}
new_monitors = {str(monitor.id) for monitor in monitors}
@@ -73,7 +72,7 @@ class UptimeRobotDataUpdateCoordinator(DataUpdateCoordinator[list[UptimeRobotMon
# create new devices and entities.
if self.data and new_monitors - {str(monitor.id) for monitor in self.data}:
self.hass.async_create_task(
- self.hass.config_entries.async_reload(self._config_entry_id)
+ self.hass.config_entries.async_reload(self.config_entry.entry_id)
)
return monitors
diff --git a/homeassistant/components/uptimerobot/sensor.py b/homeassistant/components/uptimerobot/sensor.py
index c5ff8abf5d9..724c3075a3b 100644
--- a/homeassistant/components/uptimerobot/sensor.py
+++ b/homeassistant/components/uptimerobot/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import UptimeRobotDataUpdateCoordinator
@@ -28,7 +28,7 @@ SENSORS_INFO = {
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UptimeRobot sensors."""
coordinator: UptimeRobotDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/uptimerobot/switch.py b/homeassistant/components/uptimerobot/switch.py
index aa7d07e10fd..31401ac7eb4 100644
--- a/homeassistant/components/uptimerobot/switch.py
+++ b/homeassistant/components/uptimerobot/switch.py
@@ -13,7 +13,7 @@ from homeassistant.components.switch import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import API_ATTR_OK, DOMAIN, LOGGER
from .coordinator import UptimeRobotDataUpdateCoordinator
@@ -21,7 +21,9 @@ from .entity import UptimeRobotEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the UptimeRobot switches."""
coordinator: UptimeRobotDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/usb/__init__.py b/homeassistant/components/usb/__init__.py
index d68742522a0..994f4f71c35 100644
--- a/homeassistant/components/usb/__init__.py
+++ b/homeassistant/components/usb/__init__.py
@@ -265,8 +265,15 @@ class USBDiscovery:
async def async_setup(self) -> None:
"""Set up USB Discovery."""
- if self._async_supports_monitoring():
- await self._async_start_monitor()
+ try:
+ await self._async_start_aiousbwatcher()
+ except InotifyNotAvailableError as ex:
+ _LOGGER.info(
+ "Falling back to periodic filesystem polling for development, "
+ "aiousbwatcher is not available on this system: %s",
+ ex,
+ )
+ self._async_start_monitor_polling()
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STARTED, self.async_start)
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, self.async_stop)
@@ -281,22 +288,6 @@ class USBDiscovery:
if self._request_debouncer:
self._request_debouncer.async_shutdown()
- @hass_callback
- def _async_supports_monitoring(self) -> bool:
- return sys.platform == "linux"
-
- async def _async_start_monitor(self) -> None:
- """Start monitoring hardware."""
- try:
- await self._async_start_aiousbwatcher()
- except InotifyNotAvailableError as ex:
- _LOGGER.info(
- "Falling back to periodic filesystem polling for development, aiousbwatcher "
- "is not available on this system: %s",
- ex,
- )
- self._async_start_monitor_polling()
-
@hass_callback
def _async_start_monitor_polling(self) -> None:
"""Start monitoring hardware with polling (for development only!)."""
diff --git a/homeassistant/components/utility_meter/select.py b/homeassistant/components/utility_meter/select.py
index 5815ce7ec95..0c818525c8d 100644
--- a/homeassistant/components/utility_meter/select.py
+++ b/homeassistant/components/utility_meter/select.py
@@ -10,7 +10,10 @@ from homeassistant.const import CONF_NAME, CONF_UNIQUE_ID
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device import async_device_info_to_link_from_entity
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -22,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Utility Meter config entry."""
name = config_entry.title
diff --git a/homeassistant/components/utility_meter/sensor.py b/homeassistant/components/utility_meter/sensor.py
index cd65c42b22a..425dfa2c3fd 100644
--- a/homeassistant/components/utility_meter/sensor.py
+++ b/homeassistant/components/utility_meter/sensor.py
@@ -41,7 +41,10 @@ from homeassistant.core import (
from homeassistant.helpers import entity_platform, entity_registry as er
from homeassistant.helpers.device import async_device_info_to_link_from_entity
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.event import (
async_track_point_in_time,
async_track_state_change_event,
@@ -116,7 +119,7 @@ def validate_is_number(value):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize Utility Meter config entry."""
entry_id = config_entry.entry_id
diff --git a/homeassistant/components/v2c/__init__.py b/homeassistant/components/v2c/__init__.py
index 0c07891df72..7cd5e71f3ae 100644
--- a/homeassistant/components/v2c/__init__.py
+++ b/homeassistant/components/v2c/__init__.py
@@ -4,12 +4,11 @@ from __future__ import annotations
from pytrydan import Trydan
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.httpx_client import get_async_client
-from .coordinator import V2CUpdateCoordinator
+from .coordinator import V2CConfigEntry, V2CUpdateCoordinator
PLATFORMS: list[Platform] = [
Platform.BINARY_SENSOR,
@@ -19,15 +18,11 @@ PLATFORMS: list[Platform] = [
]
-type V2CConfigEntry = ConfigEntry[V2CUpdateCoordinator]
-
-
async def async_setup_entry(hass: HomeAssistant, entry: V2CConfigEntry) -> bool:
"""Set up V2C from a config entry."""
- host = entry.data[CONF_HOST]
- trydan = Trydan(host, get_async_client(hass, verify_ssl=False))
- coordinator = V2CUpdateCoordinator(hass, trydan, host)
+ trydan = Trydan(entry.data[CONF_HOST], get_async_client(hass, verify_ssl=False))
+ coordinator = V2CUpdateCoordinator(hass, entry, trydan)
await coordinator.async_config_entry_first_refresh()
@@ -41,6 +36,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: V2CConfigEntry) -> bool:
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: V2CConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
diff --git a/homeassistant/components/v2c/binary_sensor.py b/homeassistant/components/v2c/binary_sensor.py
index 28ad3665996..85f03d6b4fb 100644
--- a/homeassistant/components/v2c/binary_sensor.py
+++ b/homeassistant/components/v2c/binary_sensor.py
@@ -13,10 +13,9 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import V2CConfigEntry
-from .coordinator import V2CUpdateCoordinator
+from .coordinator import V2CConfigEntry, V2CUpdateCoordinator
from .entity import V2CBaseEntity
@@ -51,7 +50,7 @@ TRYDAN_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: V2CConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up V2C binary sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/v2c/coordinator.py b/homeassistant/components/v2c/coordinator.py
index b121c84563c..de8015985f9 100644
--- a/homeassistant/components/v2c/coordinator.py
+++ b/homeassistant/components/v2c/coordinator.py
@@ -9,6 +9,7 @@ from pytrydan import Trydan, TrydanData
from pytrydan.exceptions import TrydanError
from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -16,19 +17,24 @@ SCAN_INTERVAL = timedelta(seconds=5)
_LOGGER = logging.getLogger(__name__)
+type V2CConfigEntry = ConfigEntry[V2CUpdateCoordinator]
+
class V2CUpdateCoordinator(DataUpdateCoordinator[TrydanData]):
"""DataUpdateCoordinator to gather data from any v2c."""
- config_entry: ConfigEntry
+ config_entry: V2CConfigEntry
- def __init__(self, hass: HomeAssistant, evse: Trydan, host: str) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: V2CConfigEntry, evse: Trydan
+ ) -> None:
"""Initialize DataUpdateCoordinator for a v2c evse."""
self.evse = evse
super().__init__(
hass,
_LOGGER,
- name=f"EVSE {host}",
+ config_entry=config_entry,
+ name=f"EVSE {config_entry.data[CONF_HOST]}",
update_interval=SCAN_INTERVAL,
)
diff --git a/homeassistant/components/v2c/diagnostics.py b/homeassistant/components/v2c/diagnostics.py
index 289d585b164..994f702a7bd 100644
--- a/homeassistant/components/v2c/diagnostics.py
+++ b/homeassistant/components/v2c/diagnostics.py
@@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
-from . import V2CConfigEntry
+from .coordinator import V2CConfigEntry
TO_REDACT = {CONF_HOST, "title"}
diff --git a/homeassistant/components/v2c/number.py b/homeassistant/components/v2c/number.py
index 1540b098cf1..e52242f0ce0 100644
--- a/homeassistant/components/v2c/number.py
+++ b/homeassistant/components/v2c/number.py
@@ -15,10 +15,9 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory, UnitOfElectricCurrent
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import V2CConfigEntry
-from .coordinator import V2CUpdateCoordinator
+from .coordinator import V2CConfigEntry, V2CUpdateCoordinator
from .entity import V2CBaseEntity
MIN_INTENSITY = 6
@@ -72,7 +71,7 @@ TRYDAN_NUMBER_SETTINGS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: V2CConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up V2C Trydan number platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/v2c/sensor.py b/homeassistant/components/v2c/sensor.py
index 97853740e9d..cfccaacda18 100644
--- a/homeassistant/components/v2c/sensor.py
+++ b/homeassistant/components/v2c/sensor.py
@@ -23,11 +23,10 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import V2CConfigEntry
-from .coordinator import V2CUpdateCoordinator
+from .coordinator import V2CConfigEntry, V2CUpdateCoordinator
from .entity import V2CBaseEntity
_LOGGER = logging.getLogger(__name__)
@@ -143,7 +142,7 @@ TRYDAN_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: V2CConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up V2C sensor platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/v2c/switch.py b/homeassistant/components/v2c/switch.py
index cca7da70e48..20bc3419757 100644
--- a/homeassistant/components/v2c/switch.py
+++ b/homeassistant/components/v2c/switch.py
@@ -18,10 +18,9 @@ from pytrydan.models.trydan import (
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import V2CConfigEntry
-from .coordinator import V2CUpdateCoordinator
+from .coordinator import V2CConfigEntry, V2CUpdateCoordinator
from .entity import V2CBaseEntity
_LOGGER = logging.getLogger(__name__)
@@ -80,7 +79,7 @@ TRYDAN_SWITCHES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: V2CConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up V2C switch platform."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/vallox/__init__.py b/homeassistant/components/vallox/__init__.py
index ceb34bc6ff9..785ecd09fb1 100644
--- a/homeassistant/components/vallox/__init__.py
+++ b/homeassistant/components/vallox/__init__.py
@@ -111,7 +111,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
client = Vallox(host)
- coordinator = ValloxDataUpdateCoordinator(hass, name, client)
+ coordinator = ValloxDataUpdateCoordinator(hass, entry, client)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/vallox/binary_sensor.py b/homeassistant/components/vallox/binary_sensor.py
index 4a0efc7b101..a205dd2039e 100644
--- a/homeassistant/components/vallox/binary_sensor.py
+++ b/homeassistant/components/vallox/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ValloxDataUpdateCoordinator
@@ -62,7 +62,7 @@ BINARY_SENSOR_ENTITIES: tuple[ValloxBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
diff --git a/homeassistant/components/vallox/coordinator.py b/homeassistant/components/vallox/coordinator.py
index c2485c7b4fd..2fe7fa533db 100644
--- a/homeassistant/components/vallox/coordinator.py
+++ b/homeassistant/components/vallox/coordinator.py
@@ -6,6 +6,8 @@ import logging
from vallox_websocket_api import MetricData, Vallox, ValloxApiException
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -17,17 +19,20 @@ _LOGGER = logging.getLogger(__name__)
class ValloxDataUpdateCoordinator(DataUpdateCoordinator[MetricData]):
"""The DataUpdateCoordinator for Vallox."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- name: str,
+ config_entry: ConfigEntry,
client: Vallox,
) -> None:
"""Initialize Vallox data coordinator."""
super().__init__(
hass,
_LOGGER,
- name=f"{name} DataUpdateCoordinator",
+ config_entry=config_entry,
+ name=f"{config_entry.data[CONF_NAME]} DataUpdateCoordinator",
update_interval=STATE_SCAN_INTERVAL,
)
self.client = client
diff --git a/homeassistant/components/vallox/date.py b/homeassistant/components/vallox/date.py
index 33c3ebb253c..da2906c02c2 100644
--- a/homeassistant/components/vallox/date.py
+++ b/homeassistant/components/vallox/date.py
@@ -10,7 +10,7 @@ from homeassistant.components.date import DateEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ValloxDataUpdateCoordinator
@@ -51,7 +51,7 @@ class ValloxFilterChangeDateEntity(ValloxEntity, DateEntity):
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Vallox filter change date entity."""
diff --git a/homeassistant/components/vallox/fan.py b/homeassistant/components/vallox/fan.py
index 3a21ef060a7..8519b4cb913 100644
--- a/homeassistant/components/vallox/fan.py
+++ b/homeassistant/components/vallox/fan.py
@@ -11,7 +11,7 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -57,7 +57,9 @@ def _convert_to_int(value: StateType) -> int | None:
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the fan device."""
data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/vallox/number.py b/homeassistant/components/vallox/number.py
index 96bc07b5a93..ce3b9c72a6d 100644
--- a/homeassistant/components/vallox/number.py
+++ b/homeassistant/components/vallox/number.py
@@ -14,7 +14,7 @@ from homeassistant.components.number import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ValloxDataUpdateCoordinator
@@ -102,7 +102,9 @@ NUMBER_ENTITIES: tuple[ValloxNumberEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
data = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/vallox/sensor.py b/homeassistant/components/vallox/sensor.py
index 7165947861a..e9194a8254c 100644
--- a/homeassistant/components/vallox/sensor.py
+++ b/homeassistant/components/vallox/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -278,7 +278,9 @@ SENSOR_ENTITIES: tuple[ValloxSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors."""
name = hass.data[DOMAIN][entry.entry_id]["name"]
diff --git a/homeassistant/components/vallox/strings.json b/homeassistant/components/vallox/strings.json
index 8a30ed4ad01..f00206826d3 100644
--- a/homeassistant/components/vallox/strings.json
+++ b/homeassistant/components/vallox/strings.json
@@ -110,7 +110,7 @@
"fields": {
"fan_speed": {
"name": "Fan speed",
- "description": "Fan speed."
+ "description": "Relative speed of the built-in fans."
}
}
},
@@ -119,7 +119,7 @@
"description": "Sets the fan speed of the Away profile.",
"fields": {
"fan_speed": {
- "name": "Fan speed",
+ "name": "[%key:component::vallox::services::set_profile_fan_speed_home::fields::fan_speed::name%]",
"description": "[%key:component::vallox::services::set_profile_fan_speed_home::fields::fan_speed::description%]"
}
}
@@ -129,7 +129,7 @@
"description": "Sets the fan speed of the Boost profile.",
"fields": {
"fan_speed": {
- "name": "Fan speed",
+ "name": "[%key:component::vallox::services::set_profile_fan_speed_home::fields::fan_speed::name%]",
"description": "[%key:component::vallox::services::set_profile_fan_speed_home::fields::fan_speed::description%]"
}
}
diff --git a/homeassistant/components/vallox/switch.py b/homeassistant/components/vallox/switch.py
index 20b270f8f18..9386f914f58 100644
--- a/homeassistant/components/vallox/switch.py
+++ b/homeassistant/components/vallox/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity, SwitchEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ValloxDataUpdateCoordinator
@@ -82,7 +82,7 @@ SWITCH_ENTITIES: tuple[ValloxSwitchEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switches."""
diff --git a/homeassistant/components/velbus/__init__.py b/homeassistant/components/velbus/__init__.py
index 41b8730eeb0..35c61892964 100644
--- a/homeassistant/components/velbus/__init__.py
+++ b/homeassistant/components/velbus/__init__.py
@@ -135,15 +135,39 @@ async def async_migrate_entry(
hass: HomeAssistant, config_entry: VelbusConfigEntry
) -> bool:
"""Migrate old entry."""
- _LOGGER.debug("Migrating from version %s", config_entry.version)
- cache_path = hass.config.path(STORAGE_DIR, f"velbuscache-{config_entry.entry_id}/")
- if config_entry.version == 1:
- # This is the config entry migration for adding the new program selection
+ _LOGGER.error(
+ "Migrating from version %s.%s", config_entry.version, config_entry.minor_version
+ )
+
+ # This is the config entry migration for adding the new program selection
+ # migrate from 1.x to 2.1
+ if config_entry.version < 2:
# clean the velbusCache
+ cache_path = hass.config.path(
+ STORAGE_DIR, f"velbuscache-{config_entry.entry_id}/"
+ )
if os.path.isdir(cache_path):
await hass.async_add_executor_job(shutil.rmtree, cache_path)
- # set the new version
- hass.config_entries.async_update_entry(config_entry, version=2)
- _LOGGER.debug("Migration to version %s successful", config_entry.version)
+ # This is the config entry migration for swapping the usb unique id to the serial number
+ # migrate from 2.1 to 2.2
+ if (
+ config_entry.version < 3
+ and config_entry.minor_version == 1
+ and config_entry.unique_id is not None
+ ):
+ # not all velbus devices have a unique id, so handle this correctly
+ parts = config_entry.unique_id.split("_")
+ # old one should have 4 item
+ if len(parts) == 4:
+ hass.config_entries.async_update_entry(config_entry, unique_id=parts[1])
+
+ # update the config entry
+ hass.config_entries.async_update_entry(config_entry, version=2, minor_version=2)
+
+ _LOGGER.error(
+ "Migration to version %s.%s successful",
+ config_entry.version,
+ config_entry.minor_version,
+ )
return True
diff --git a/homeassistant/components/velbus/binary_sensor.py b/homeassistant/components/velbus/binary_sensor.py
index 88dc994efe8..2ddf6605c19 100644
--- a/homeassistant/components/velbus/binary_sensor.py
+++ b/homeassistant/components/velbus/binary_sensor.py
@@ -4,7 +4,7 @@ from velbusaio.channels import Button as VelbusButton
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .entity import VelbusEntity
@@ -15,7 +15,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velbus/button.py b/homeassistant/components/velbus/button.py
index fc943159123..8f736dcd35b 100644
--- a/homeassistant/components/velbus/button.py
+++ b/homeassistant/components/velbus/button.py
@@ -10,7 +10,7 @@ from velbusaio.channels import (
from homeassistant.components.button import ButtonEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .entity import VelbusEntity, api_call
@@ -21,7 +21,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velbus/climate.py b/homeassistant/components/velbus/climate.py
index b2f3077ecee..e31d9a97416 100644
--- a/homeassistant/components/velbus/climate.py
+++ b/homeassistant/components/velbus/climate.py
@@ -14,7 +14,7 @@ from homeassistant.components.climate import (
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .const import DOMAIN, PRESET_MODES
@@ -26,7 +26,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velbus/config_flow.py b/homeassistant/components/velbus/config_flow.py
index 9e99b2631d4..7c93d8784ad 100644
--- a/homeassistant/components/velbus/config_flow.py
+++ b/homeassistant/components/velbus/config_flow.py
@@ -4,22 +4,23 @@ from __future__ import annotations
from typing import Any
+import serial.tools.list_ports
import velbusaio.controller
from velbusaio.exceptions import VelbusConnectionFailed
import voluptuous as vol
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
-from homeassistant.const import CONF_NAME, CONF_PORT
+from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, CONF_PORT
from homeassistant.helpers.service_info.usb import UsbServiceInfo
-from homeassistant.util import slugify
-from .const import DOMAIN
+from .const import CONF_TLS, DOMAIN
class VelbusConfigFlow(ConfigFlow, domain=DOMAIN):
"""Handle a config flow."""
VERSION = 2
+ MINOR_VERSION = 2
def __init__(self) -> None:
"""Initialize the velbus config flow."""
@@ -27,14 +28,16 @@ class VelbusConfigFlow(ConfigFlow, domain=DOMAIN):
self._device: str = ""
self._title: str = ""
- def _create_device(self, name: str, prt: str) -> ConfigFlowResult:
+ def _create_device(self) -> ConfigFlowResult:
"""Create an entry async."""
- return self.async_create_entry(title=name, data={CONF_PORT: prt})
+ return self.async_create_entry(
+ title=self._title, data={CONF_PORT: self._device}
+ )
- async def _test_connection(self, prt: str) -> bool:
+ async def _test_connection(self) -> bool:
"""Try to connect to the velbus with the port specified."""
try:
- controller = velbusaio.controller.Velbus(prt)
+ controller = velbusaio.controller.Velbus(self._device)
await controller.connect()
await controller.stop()
except VelbusConnectionFailed:
@@ -46,43 +49,86 @@ class VelbusConfigFlow(ConfigFlow, domain=DOMAIN):
self, user_input: dict[str, Any] | None = None
) -> ConfigFlowResult:
"""Step when user initializes a integration."""
- self._errors = {}
+ return self.async_show_menu(
+ step_id="user", menu_options=["network", "usbselect"]
+ )
+
+ async def async_step_network(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle network step."""
if user_input is not None:
- name = slugify(user_input[CONF_NAME])
- prt = user_input[CONF_PORT]
- self._async_abort_entries_match({CONF_PORT: prt})
- if await self._test_connection(prt):
- return self._create_device(name, prt)
+ self._title = "Velbus Network"
+ if user_input[CONF_TLS]:
+ self._device = "tls://"
+ else:
+ self._device = ""
+ if CONF_PASSWORD in user_input and user_input[CONF_PASSWORD] != "":
+ self._device += f"{user_input[CONF_PASSWORD]}@"
+ self._device += f"{user_input[CONF_HOST]}:{user_input[CONF_PORT]}"
+ self._async_abort_entries_match({CONF_PORT: self._device})
+ if await self._test_connection():
+ return self._create_device()
+ else:
+ user_input = {
+ CONF_TLS: True,
+ CONF_PORT: 27015,
+ }
+
+ return self.async_show_form(
+ step_id="network",
+ data_schema=self.add_suggested_values_to_schema(
+ vol.Schema(
+ {
+ vol.Required(CONF_TLS): bool,
+ vol.Required(CONF_HOST): str,
+ vol.Required(CONF_PORT): int,
+ vol.Optional(CONF_PASSWORD): str,
+ }
+ ),
+ suggested_values=user_input,
+ ),
+ errors=self._errors,
+ )
+
+ async def async_step_usbselect(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle usb select step."""
+ ports = await self.hass.async_add_executor_job(serial.tools.list_ports.comports)
+ list_of_ports = [
+ f"{p}{', s/n: ' + p.serial_number if p.serial_number else ''}"
+ + (f" - {p.manufacturer}" if p.manufacturer else "")
+ for p in ports
+ ]
+
+ if user_input is not None:
+ self._title = "Velbus USB"
+ self._device = ports[list_of_ports.index(user_input[CONF_PORT])].device
+ self._async_abort_entries_match({CONF_PORT: self._device})
+ if await self._test_connection():
+ return self._create_device()
else:
user_input = {}
- user_input[CONF_NAME] = ""
user_input[CONF_PORT] = ""
return self.async_show_form(
- step_id="user",
- data_schema=vol.Schema(
- {
- vol.Required(CONF_NAME, default=user_input[CONF_NAME]): str,
- vol.Required(CONF_PORT, default=user_input[CONF_PORT]): str,
- }
+ step_id="usbselect",
+ data_schema=self.add_suggested_values_to_schema(
+ vol.Schema({vol.Required(CONF_PORT): vol.In(list_of_ports)}),
+ suggested_values=user_input,
),
errors=self._errors,
)
async def async_step_usb(self, discovery_info: UsbServiceInfo) -> ConfigFlowResult:
"""Handle USB Discovery."""
- await self.async_set_unique_id(
- f"{discovery_info.vid}:{discovery_info.pid}_{discovery_info.serial_number}_{discovery_info.manufacturer}_{discovery_info.description}"
- )
- dev_path = discovery_info.device
- # check if this device is not already configured
- self._async_abort_entries_match({CONF_PORT: dev_path})
- # check if we can make a valid velbus connection
- if not await self._test_connection(dev_path):
- return self.async_abort(reason="cannot_connect")
- # store the data for the config step
- self._device = dev_path
+ await self.async_set_unique_id(discovery_info.serial_number)
+ self._device = discovery_info.device
self._title = "Velbus USB"
+ self._async_abort_entries_match({CONF_PORT: self._device})
+ if not await self._test_connection():
+ return self.async_abort(reason="cannot_connect")
# call the config step
self._set_confirm_only()
return await self.async_step_discovery_confirm()
@@ -92,7 +138,7 @@ class VelbusConfigFlow(ConfigFlow, domain=DOMAIN):
) -> ConfigFlowResult:
"""Handle Discovery confirmation."""
if user_input is not None:
- return self._create_device(self._title, self._device)
+ return self._create_device()
return self.async_show_form(
step_id="discovery_confirm",
diff --git a/homeassistant/components/velbus/const.py b/homeassistant/components/velbus/const.py
index b40f64e8607..f42e449bdcc 100644
--- a/homeassistant/components/velbus/const.py
+++ b/homeassistant/components/velbus/const.py
@@ -14,6 +14,7 @@ DOMAIN: Final = "velbus"
CONF_CONFIG_ENTRY: Final = "config_entry"
CONF_INTERFACE: Final = "interface"
CONF_MEMO_TEXT: Final = "memo_text"
+CONF_TLS: Final = "tls"
SERVICE_SCAN: Final = "scan"
SERVICE_SYNC: Final = "sync_clock"
diff --git a/homeassistant/components/velbus/cover.py b/homeassistant/components/velbus/cover.py
index 2ddea37f2d6..995b7e9d59c 100644
--- a/homeassistant/components/velbus/cover.py
+++ b/homeassistant/components/velbus/cover.py
@@ -12,7 +12,7 @@ from homeassistant.components.cover import (
CoverEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .entity import VelbusEntity, api_call
@@ -23,7 +23,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velbus/light.py b/homeassistant/components/velbus/light.py
index c134095c2ff..5037e2b1ced 100644
--- a/homeassistant/components/velbus/light.py
+++ b/homeassistant/components/velbus/light.py
@@ -23,7 +23,7 @@ from homeassistant.components.light import (
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .entity import VelbusEntity, api_call
@@ -34,7 +34,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velbus/manifest.json b/homeassistant/components/velbus/manifest.json
index 960f127d16e..1cb540b22ec 100644
--- a/homeassistant/components/velbus/manifest.json
+++ b/homeassistant/components/velbus/manifest.json
@@ -13,7 +13,8 @@
"velbus-packet",
"velbus-protocol"
],
- "requirements": ["velbus-aio==2025.1.1"],
+ "quality_scale": "bronze",
+ "requirements": ["velbus-aio==2025.3.1"],
"usb": [
{
"vid": "10CF",
diff --git a/homeassistant/components/velbus/quality_scale.yaml b/homeassistant/components/velbus/quality_scale.yaml
index 0ad3e3ce485..829f48e6f52 100644
--- a/homeassistant/components/velbus/quality_scale.yaml
+++ b/homeassistant/components/velbus/quality_scale.yaml
@@ -8,10 +8,7 @@ rules:
brands: done
common-modules: done
config-flow-test-coverage: done
- config-flow:
- status: todo
- comment: |
- Dynamically build up the port parameter based on inputs provided by the user, do not fill-in a name parameter, build it up in the config flow
+ config-flow: done
dependency-transparency: done
docs-actions: done
docs-high-level-description: done
diff --git a/homeassistant/components/velbus/select.py b/homeassistant/components/velbus/select.py
index 6c2dfe0a3b1..1d52b8d4afc 100644
--- a/homeassistant/components/velbus/select.py
+++ b/homeassistant/components/velbus/select.py
@@ -5,7 +5,7 @@ from velbusaio.channels import SelectedProgram
from homeassistant.components.select import SelectEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .entity import VelbusEntity, api_call
@@ -16,7 +16,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus select based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velbus/sensor.py b/homeassistant/components/velbus/sensor.py
index 77833da3ee1..96ef91e8174 100644
--- a/homeassistant/components/velbus/sensor.py
+++ b/homeassistant/components/velbus/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .entity import VelbusEntity
@@ -21,7 +21,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velbus/strings.json b/homeassistant/components/velbus/strings.json
index 69fc3d661e9..a50395af115 100644
--- a/homeassistant/components/velbus/strings.json
+++ b/homeassistant/components/velbus/strings.json
@@ -7,6 +7,32 @@
"name": "The name for this Velbus connection",
"port": "Connection string"
}
+ },
+ "network": {
+ "title": "TCP/IP configuration",
+ "data": {
+ "tls": "Use TLS (secure connection)",
+ "host": "[%key:common::config_flow::data::host%]",
+ "port": "[%key:common::config_flow::data::port%]",
+ "password": "[%key:common::config_flow::data::password%]"
+ },
+ "data_description": {
+ "tls": "Enable this if you use a secure connection to your Velbus interface, like a Signum.",
+ "host": "The IP address or hostname of the Velbus interface.",
+ "port": "The port number of the Velbus interface.",
+ "password": "The password of the Velbus interface, this is only needed if the interface is password protected."
+ },
+ "description": "TCP/IP configuration, in case you use a Signum, VelServ, velbus-tcp or any other Velbus to TCP/IP interface."
+ },
+ "usbselect": {
+ "title": "USB configuration",
+ "data": {
+ "port": "[%key:common::config_flow::data::port%]"
+ },
+ "data_description": {
+ "port": "Select the serial port for your Velbus USB interface."
+ },
+ "description": "Select the serial port for your Velbus USB interface."
}
},
"error": {
diff --git a/homeassistant/components/velbus/switch.py b/homeassistant/components/velbus/switch.py
index 8256e716d4f..40dc3c09f73 100644
--- a/homeassistant/components/velbus/switch.py
+++ b/homeassistant/components/velbus/switch.py
@@ -6,7 +6,7 @@ from velbusaio.channels import Relay as VelbusRelay
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VelbusConfigEntry
from .entity import VelbusEntity, api_call
@@ -17,7 +17,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
entry: VelbusConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Velbus switch based on config_entry."""
await entry.runtime_data.scan_task
diff --git a/homeassistant/components/velux/cover.py b/homeassistant/components/velux/cover.py
index 90745f601b4..d6bf8905d91 100644
--- a/homeassistant/components/velux/cover.py
+++ b/homeassistant/components/velux/cover.py
@@ -16,7 +16,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import VeluxEntity
@@ -25,7 +25,9 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
- hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up cover(s) for Velux platform."""
module = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/velux/entity.py b/homeassistant/components/velux/entity.py
index 674ba5dde45..1231a98e0a8 100644
--- a/homeassistant/components/velux/entity.py
+++ b/homeassistant/components/velux/entity.py
@@ -31,6 +31,6 @@ class VeluxEntity(Entity):
self.node.register_device_updated_cb(after_update_callback)
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Store register state change callback."""
self.async_register_callbacks()
diff --git a/homeassistant/components/velux/light.py b/homeassistant/components/velux/light.py
index 14f12a01060..b991239b7a4 100644
--- a/homeassistant/components/velux/light.py
+++ b/homeassistant/components/velux/light.py
@@ -9,7 +9,7 @@ from pyvlx import Intensity, LighteningDevice
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import VeluxEntity
@@ -18,7 +18,9 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
- hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up light(s) for Velux platform."""
module = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/velux/scene.py b/homeassistant/components/velux/scene.py
index 54888413613..636ab82e819 100644
--- a/homeassistant/components/velux/scene.py
+++ b/homeassistant/components/velux/scene.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.scene import Scene
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
@@ -15,7 +15,9 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
- hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ config: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the scenes for Velux platform."""
module = hass.data[DOMAIN][config.entry_id]
diff --git a/homeassistant/components/venstar/__init__.py b/homeassistant/components/venstar/__init__.py
index 3243c7a6f47..faa47bfc8e4 100644
--- a/homeassistant/components/venstar/__init__.py
+++ b/homeassistant/components/venstar/__init__.py
@@ -21,14 +21,14 @@ from .coordinator import VenstarDataUpdateCoordinator
PLATFORMS = [Platform.BINARY_SENSOR, Platform.CLIMATE, Platform.SENSOR]
-async def async_setup_entry(hass: HomeAssistant, config: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
"""Set up the Venstar thermostat."""
- username = config.data.get(CONF_USERNAME)
- password = config.data.get(CONF_PASSWORD)
- pin = config.data.get(CONF_PIN)
- host = config.data[CONF_HOST]
+ username = config_entry.data.get(CONF_USERNAME)
+ password = config_entry.data.get(CONF_PASSWORD)
+ pin = config_entry.data.get(CONF_PIN)
+ host = config_entry.data[CONF_HOST]
timeout = VENSTAR_TIMEOUT
- protocol = "https" if config.data[CONF_SSL] else "http"
+ protocol = "https" if config_entry.data[CONF_SSL] else "http"
client = VenstarColorTouch(
addr=host,
@@ -41,19 +41,22 @@ async def async_setup_entry(hass: HomeAssistant, config: ConfigEntry) -> bool:
venstar_data_coordinator = VenstarDataUpdateCoordinator(
hass,
+ config_entry,
venstar_connection=client,
)
await venstar_data_coordinator.async_config_entry_first_refresh()
- hass.data.setdefault(DOMAIN, {})[config.entry_id] = venstar_data_coordinator
- await hass.config_entries.async_forward_entry_setups(config, PLATFORMS)
+ hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = venstar_data_coordinator
+ await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
return True
-async def async_unload_entry(hass: HomeAssistant, config: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
"""Unload the config and platforms."""
- unload_ok = await hass.config_entries.async_unload_platforms(config, PLATFORMS)
+ unload_ok = await hass.config_entries.async_unload_platforms(
+ config_entry, PLATFORMS
+ )
if unload_ok:
- hass.data[DOMAIN].pop(config.entry_id)
+ hass.data[DOMAIN].pop(config_entry.entry_id)
return unload_ok
diff --git a/homeassistant/components/venstar/binary_sensor.py b/homeassistant/components/venstar/binary_sensor.py
index 315df09b625..672db463791 100644
--- a/homeassistant/components/venstar/binary_sensor.py
+++ b/homeassistant/components/venstar/binary_sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import VenstarEntity
@@ -15,7 +15,7 @@ from .entity import VenstarEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Vensar device binary_sensors based on a config entry."""
coordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/venstar/climate.py b/homeassistant/components/venstar/climate.py
index 50f6508e7ed..ade86e8dd71 100644
--- a/homeassistant/components/venstar/climate.py
+++ b/homeassistant/components/venstar/climate.py
@@ -33,7 +33,10 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import (
@@ -66,7 +69,7 @@ PLATFORM_SCHEMA = CLIMATE_PLATFORM_SCHEMA.extend(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Venstar thermostat."""
venstar_data_coordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/venstar/coordinator.py b/homeassistant/components/venstar/coordinator.py
index b825775de7f..1d0ff60c1e0 100644
--- a/homeassistant/components/venstar/coordinator.py
+++ b/homeassistant/components/venstar/coordinator.py
@@ -8,6 +8,7 @@ from datetime import timedelta
from requests import RequestException
from venstarcolortouch import VenstarColorTouch
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import update_coordinator
@@ -17,16 +18,19 @@ from .const import _LOGGER, DOMAIN, VENSTAR_SLEEP
class VenstarDataUpdateCoordinator(update_coordinator.DataUpdateCoordinator[None]):
"""Class to manage fetching Venstar data."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
- *,
+ config_entry: ConfigEntry,
venstar_connection: VenstarColorTouch,
) -> None:
"""Initialize global Venstar data updater."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/venstar/sensor.py b/homeassistant/components/venstar/sensor.py
index 94180f6ad79..14e7103a83f 100644
--- a/homeassistant/components/venstar/sensor.py
+++ b/homeassistant/components/venstar/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import VenstarDataUpdateCoordinator
@@ -81,7 +81,7 @@ class VenstarSensorEntityDescription(SensorEntityDescription):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Venstar device sensors based on a config entry."""
coordinator: VenstarDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/vera/binary_sensor.py b/homeassistant/components/vera/binary_sensor.py
index 3438ee81d4a..00780fec8ce 100644
--- a/homeassistant/components/vera/binary_sensor.py
+++ b/homeassistant/components/vera/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import ENTITY_ID_FORMAT, BinarySenso
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import ControllerData, get_controller_data
from .entity import VeraEntity
@@ -17,7 +17,7 @@ from .entity import VeraEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/vera/climate.py b/homeassistant/components/vera/climate.py
index eb2a5206f30..084725f484e 100644
--- a/homeassistant/components/vera/climate.py
+++ b/homeassistant/components/vera/climate.py
@@ -17,7 +17,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import ControllerData, get_controller_data
from .entity import VeraEntity
@@ -30,7 +30,7 @@ SUPPORT_HVAC = [HVACMode.COOL, HVACMode.HEAT, HVACMode.HEAT_COOL, HVACMode.OFF]
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/vera/cover.py b/homeassistant/components/vera/cover.py
index b5b57f43c0c..8256804b8a3 100644
--- a/homeassistant/components/vera/cover.py
+++ b/homeassistant/components/vera/cover.py
@@ -10,7 +10,7 @@ from homeassistant.components.cover import ATTR_POSITION, ENTITY_ID_FORMAT, Cove
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import ControllerData, get_controller_data
from .entity import VeraEntity
@@ -19,7 +19,7 @@ from .entity import VeraEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/vera/entity.py b/homeassistant/components/vera/entity.py
index 84e21e54983..b3013c288c1 100644
--- a/homeassistant/components/vera/entity.py
+++ b/homeassistant/components/vera/entity.py
@@ -52,7 +52,7 @@ class VeraEntity[_DeviceTypeT: veraApi.VeraDevice](Entity):
"""Update the state."""
self.schedule_update_ha_state(True)
- def update(self):
+ def update(self) -> None:
"""Force a refresh from the device if the device is unavailable."""
refresh_needed = self.vera_device.should_poll or not self.available
_LOGGER.debug("%s: update called (refresh=%s)", self._name, refresh_needed)
@@ -90,7 +90,7 @@ class VeraEntity[_DeviceTypeT: veraApi.VeraDevice](Entity):
return attr
@property
- def available(self):
+ def available(self) -> bool:
"""If device communications have failed return false."""
return not self.vera_device.comm_failure
diff --git a/homeassistant/components/vera/light.py b/homeassistant/components/vera/light.py
index 9b8ae42f620..f573fcd94ea 100644
--- a/homeassistant/components/vera/light.py
+++ b/homeassistant/components/vera/light.py
@@ -16,7 +16,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .common import ControllerData, get_controller_data
@@ -26,7 +26,7 @@ from .entity import VeraEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/vera/lock.py b/homeassistant/components/vera/lock.py
index 18f0b9de3e2..3f76f3a6106 100644
--- a/homeassistant/components/vera/lock.py
+++ b/homeassistant/components/vera/lock.py
@@ -10,7 +10,7 @@ from homeassistant.components.lock import ENTITY_ID_FORMAT, LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import ControllerData, get_controller_data
from .entity import VeraEntity
@@ -22,7 +22,7 @@ ATTR_LOW_BATTERY = "low_battery"
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/vera/scene.py b/homeassistant/components/vera/scene.py
index 22061f98929..0e504b12303 100644
--- a/homeassistant/components/vera/scene.py
+++ b/homeassistant/components/vera/scene.py
@@ -9,7 +9,7 @@ import pyvera as veraApi
from homeassistant.components.scene import Scene
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import slugify
from .common import ControllerData, get_controller_data
@@ -19,7 +19,7 @@ from .const import VERA_ID_FORMAT
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/vera/sensor.py b/homeassistant/components/vera/sensor.py
index 95f1fa0bd89..d778b4c2e5d 100644
--- a/homeassistant/components/vera/sensor.py
+++ b/homeassistant/components/vera/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import ControllerData, get_controller_data
from .entity import VeraEntity
@@ -32,7 +32,7 @@ SCAN_INTERVAL = timedelta(seconds=5)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/vera/switch.py b/homeassistant/components/vera/switch.py
index ad7fbe68458..67be4a7849a 100644
--- a/homeassistant/components/vera/switch.py
+++ b/homeassistant/components/vera/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import ENTITY_ID_FORMAT, SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import ControllerData, get_controller_data
from .entity import VeraEntity
@@ -19,7 +19,7 @@ from .entity import VeraEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
controller_data = get_controller_data(hass, entry)
diff --git a/homeassistant/components/verisure/alarm_control_panel.py b/homeassistant/components/verisure/alarm_control_panel.py
index 5f34b587163..7ead1f014c8 100644
--- a/homeassistant/components/verisure/alarm_control_panel.py
+++ b/homeassistant/components/verisure/alarm_control_panel.py
@@ -13,7 +13,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ALARM_STATE_TO_HA, CONF_GIID, DOMAIN, LOGGER
@@ -23,7 +23,7 @@ from .coordinator import VerisureDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Verisure alarm control panel from a config entry."""
async_add_entities([VerisureAlarm(coordinator=hass.data[DOMAIN][entry.entry_id])])
@@ -49,14 +49,14 @@ class VerisureAlarm(
name="Verisure Alarm",
manufacturer="Verisure",
model="VBox",
- identifiers={(DOMAIN, self.coordinator.entry.data[CONF_GIID])},
+ identifiers={(DOMAIN, self.coordinator.config_entry.data[CONF_GIID])},
configuration_url="https://mypages.verisure.com",
)
@property
def unique_id(self) -> str:
"""Return the unique ID for this entity."""
- return self.coordinator.entry.data[CONF_GIID]
+ return self.coordinator.config_entry.data[CONF_GIID]
async def _async_set_arm_state(
self, state: str, command_data: dict[str, str | dict[str, str]]
diff --git a/homeassistant/components/verisure/binary_sensor.py b/homeassistant/components/verisure/binary_sensor.py
index 542ee3485ce..4d9221c3ca9 100644
--- a/homeassistant/components/verisure/binary_sensor.py
+++ b/homeassistant/components/verisure/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.const import ATTR_LAST_TRIP_TIME, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util
@@ -22,7 +22,7 @@ from .coordinator import VerisureDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Verisure binary sensors based on a config entry."""
coordinator: VerisureDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
@@ -62,7 +62,7 @@ class VerisureDoorWindowSensor(
manufacturer="Verisure",
model="Shock Sensor Detector",
identifiers={(DOMAIN, self.serial_number)},
- via_device=(DOMAIN, self.coordinator.entry.data[CONF_GIID]),
+ via_device=(DOMAIN, self.coordinator.config_entry.data[CONF_GIID]),
configuration_url="https://mypages.verisure.com",
)
@@ -104,7 +104,7 @@ class VerisureEthernetStatus(
@property
def unique_id(self) -> str:
"""Return the unique ID for this entity."""
- return f"{self.coordinator.entry.data[CONF_GIID]}_ethernet"
+ return f"{self.coordinator.config_entry.data[CONF_GIID]}_ethernet"
@property
def device_info(self) -> DeviceInfo:
@@ -113,7 +113,7 @@ class VerisureEthernetStatus(
name="Verisure Alarm",
manufacturer="Verisure",
model="VBox",
- identifiers={(DOMAIN, self.coordinator.entry.data[CONF_GIID])},
+ identifiers={(DOMAIN, self.coordinator.config_entry.data[CONF_GIID])},
configuration_url="https://mypages.verisure.com",
)
diff --git a/homeassistant/components/verisure/camera.py b/homeassistant/components/verisure/camera.py
index 70cd436d24c..1f5d48ea197 100644
--- a/homeassistant/components/verisure/camera.py
+++ b/homeassistant/components/verisure/camera.py
@@ -13,7 +13,7 @@ from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -25,7 +25,7 @@ from .coordinator import VerisureDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Verisure sensors based on a config entry."""
coordinator: VerisureDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
@@ -75,7 +75,7 @@ class VerisureSmartcam(CoordinatorEntity[VerisureDataUpdateCoordinator], Camera)
manufacturer="Verisure",
model="SmartCam",
identifiers={(DOMAIN, self.serial_number)},
- via_device=(DOMAIN, self.coordinator.entry.data[CONF_GIID]),
+ via_device=(DOMAIN, self.coordinator.config_entry.data[CONF_GIID]),
configuration_url="https://mypages.verisure.com",
)
diff --git a/homeassistant/components/verisure/coordinator.py b/homeassistant/components/verisure/coordinator.py
index 930d862257b..5165ddc6d3d 100644
--- a/homeassistant/components/verisure/coordinator.py
+++ b/homeassistant/components/verisure/coordinator.py
@@ -25,10 +25,11 @@ from .const import CONF_GIID, DEFAULT_SCAN_INTERVAL, DOMAIN, LOGGER
class VerisureDataUpdateCoordinator(DataUpdateCoordinator):
"""A Verisure Data Update Coordinator."""
+ config_entry: ConfigEntry
+
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize the Verisure hub."""
self.imageseries: list[dict[str, str]] = []
- self.entry = entry
self._overview: list[dict] = []
self.verisure = Verisure(
@@ -40,7 +41,11 @@ class VerisureDataUpdateCoordinator(DataUpdateCoordinator):
)
super().__init__(
- hass, LOGGER, name=DOMAIN, update_interval=DEFAULT_SCAN_INTERVAL
+ hass,
+ LOGGER,
+ config_entry=entry,
+ name=DOMAIN,
+ update_interval=DEFAULT_SCAN_INTERVAL,
)
async def async_login(self) -> bool:
@@ -55,7 +60,7 @@ class VerisureDataUpdateCoordinator(DataUpdateCoordinator):
return False
await self.hass.async_add_executor_job(
- self.verisure.set_giid, self.entry.data[CONF_GIID]
+ self.verisure.set_giid, self.config_entry.data[CONF_GIID]
)
return True
diff --git a/homeassistant/components/verisure/lock.py b/homeassistant/components/verisure/lock.py
index 87f5c53880e..76aeedd05fa 100644
--- a/homeassistant/components/verisure/lock.py
+++ b/homeassistant/components/verisure/lock.py
@@ -13,7 +13,7 @@ from homeassistant.const import ATTR_CODE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -33,7 +33,7 @@ from .coordinator import VerisureDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Verisure alarm control panel from a config entry."""
coordinator: VerisureDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
@@ -81,7 +81,7 @@ class VerisureDoorlock(CoordinatorEntity[VerisureDataUpdateCoordinator], LockEnt
manufacturer="Verisure",
model="Lockguard Smartlock",
identifiers={(DOMAIN, self.serial_number)},
- via_device=(DOMAIN, self.coordinator.entry.data[CONF_GIID]),
+ via_device=(DOMAIN, self.coordinator.config_entry.data[CONF_GIID]),
configuration_url="https://mypages.verisure.com",
)
@@ -109,7 +109,7 @@ class VerisureDoorlock(CoordinatorEntity[VerisureDataUpdateCoordinator], LockEnt
@property
def code_format(self) -> str:
"""Return the configured code format."""
- digits = self.coordinator.entry.options.get(
+ digits = self.coordinator.config_entry.options.get(
CONF_LOCK_CODE_DIGITS, DEFAULT_LOCK_CODE_DIGITS
)
return f"^\\d{{{digits}}}$"
diff --git a/homeassistant/components/verisure/sensor.py b/homeassistant/components/verisure/sensor.py
index 4f6e6b3d3c5..6ed4784bffb 100644
--- a/homeassistant/components/verisure/sensor.py
+++ b/homeassistant/components/verisure/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.const import PERCENTAGE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_GIID, DEVICE_TYPE_NAME, DOMAIN
@@ -22,7 +22,7 @@ from .coordinator import VerisureDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Verisure sensors based on a config entry."""
coordinator: VerisureDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
@@ -72,7 +72,7 @@ class VerisureThermometer(
manufacturer="Verisure",
model=DEVICE_TYPE_NAME.get(device_type, device_type),
identifiers={(DOMAIN, self.serial_number)},
- via_device=(DOMAIN, self.coordinator.entry.data[CONF_GIID]),
+ via_device=(DOMAIN, self.coordinator.config_entry.data[CONF_GIID]),
configuration_url="https://mypages.verisure.com",
)
@@ -122,7 +122,7 @@ class VerisureHygrometer(
manufacturer="Verisure",
model=DEVICE_TYPE_NAME.get(device_type, device_type),
identifiers={(DOMAIN, self.serial_number)},
- via_device=(DOMAIN, self.coordinator.entry.data[CONF_GIID]),
+ via_device=(DOMAIN, self.coordinator.config_entry.data[CONF_GIID]),
configuration_url="https://mypages.verisure.com",
)
diff --git a/homeassistant/components/verisure/switch.py b/homeassistant/components/verisure/switch.py
index e0238097e01..0deb1da5e95 100644
--- a/homeassistant/components/verisure/switch.py
+++ b/homeassistant/components/verisure/switch.py
@@ -9,7 +9,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_GIID, DOMAIN
@@ -19,7 +19,7 @@ from .coordinator import VerisureDataUpdateCoordinator
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Verisure alarm control panel from a config entry."""
coordinator: VerisureDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
@@ -57,7 +57,7 @@ class VerisureSmartplug(CoordinatorEntity[VerisureDataUpdateCoordinator], Switch
manufacturer="Verisure",
model="SmartPlug",
identifiers={(DOMAIN, self.serial_number)},
- via_device=(DOMAIN, self.coordinator.entry.data[CONF_GIID]),
+ via_device=(DOMAIN, self.coordinator.config_entry.data[CONF_GIID]),
configuration_url="https://mypages.verisure.com",
)
diff --git a/homeassistant/components/version/__init__.py b/homeassistant/components/version/__init__.py
index cf13821dc8a..6fabf97c8dd 100644
--- a/homeassistant/components/version/__init__.py
+++ b/homeassistant/components/version/__init__.py
@@ -6,7 +6,6 @@ import logging
from pyhaversion import HaVersion
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -18,12 +17,10 @@ from .const import (
CONF_SOURCE,
PLATFORMS,
)
-from .coordinator import VersionDataUpdateCoordinator
+from .coordinator import VersionConfigEntry, VersionDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__)
-type VersionConfigEntry = ConfigEntry[VersionDataUpdateCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: VersionConfigEntry) -> bool:
"""Set up the version integration from a config entry."""
@@ -40,6 +37,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: VersionConfigEntry) -> b
coordinator = VersionDataUpdateCoordinator(
hass=hass,
+ config_entry=entry,
api=HaVersion(
session=async_get_clientsession(hass),
source=entry.data[CONF_SOURCE],
diff --git a/homeassistant/components/version/binary_sensor.py b/homeassistant/components/version/binary_sensor.py
index 827029e1d8c..900daa7aba1 100644
--- a/homeassistant/components/version/binary_sensor.py
+++ b/homeassistant/components/version/binary_sensor.py
@@ -11,10 +11,10 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import CONF_NAME, EntityCategory, __version__ as HA_VERSION
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import VersionConfigEntry
from .const import CONF_SOURCE, DEFAULT_NAME
+from .coordinator import VersionConfigEntry
from .entity import VersionEntity
HA_VERSION_OBJECT = AwesomeVersion(HA_VERSION)
@@ -23,7 +23,7 @@ HA_VERSION_OBJECT = AwesomeVersion(HA_VERSION)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: VersionConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up version binary_sensors."""
coordinator = config_entry.runtime_data
diff --git a/homeassistant/components/version/coordinator.py b/homeassistant/components/version/coordinator.py
index 05adf07642b..349ede53d33 100644
--- a/homeassistant/components/version/coordinator.py
+++ b/homeassistant/components/version/coordinator.py
@@ -14,21 +14,25 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN, LOGGER, UPDATE_COORDINATOR_UPDATE_INTERVAL
+type VersionConfigEntry = ConfigEntry[VersionDataUpdateCoordinator]
+
class VersionDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Data update coordinator for Version entities."""
- config_entry: ConfigEntry
+ config_entry: VersionConfigEntry
def __init__(
self,
hass: HomeAssistant,
+ config_entry: VersionConfigEntry,
api: HaVersion,
) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=UPDATE_COORDINATOR_UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/version/diagnostics.py b/homeassistant/components/version/diagnostics.py
index ca7318f468b..bcc94bd8da4 100644
--- a/homeassistant/components/version/diagnostics.py
+++ b/homeassistant/components/version/diagnostics.py
@@ -9,7 +9,7 @@ from attr import asdict
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
-from . import VersionConfigEntry
+from .coordinator import VersionConfigEntry
async def async_get_config_entry_diagnostics(
diff --git a/homeassistant/components/version/sensor.py b/homeassistant/components/version/sensor.py
index e1d552bcd36..7e173b46d36 100644
--- a/homeassistant/components/version/sensor.py
+++ b/homeassistant/components/version/sensor.py
@@ -7,18 +7,18 @@ from typing import Any
from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import VersionConfigEntry
from .const import CONF_SOURCE, DEFAULT_NAME
+from .coordinator import VersionConfigEntry
from .entity import VersionEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: VersionConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up version sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/vesync/__init__.py b/homeassistant/components/vesync/__init__.py
index 4951bdb2dc1..dddf7857545 100644
--- a/homeassistant/components/vesync/__init__.py
+++ b/homeassistant/components/vesync/__init__.py
@@ -5,8 +5,14 @@ import logging
from pyvesync import VeSync
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
-from homeassistant.core import HomeAssistant, ServiceCall
+from homeassistant.const import (
+ CONF_PASSWORD,
+ CONF_USERNAME,
+ EVENT_LOGGING_CHANGED,
+ Platform,
+)
+from homeassistant.core import Event, HomeAssistant, ServiceCall, callback
+from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send
@@ -17,6 +23,7 @@ from .const import (
VS_COORDINATOR,
VS_DEVICES,
VS_DISCOVERY,
+ VS_LISTENERS,
VS_MANAGER,
)
from .coordinator import VeSyncDataCoordinator
@@ -27,6 +34,7 @@ PLATFORMS = [
Platform.HUMIDIFIER,
Platform.LIGHT,
Platform.NUMBER,
+ Platform.SELECT,
Platform.SENSOR,
Platform.SWITCH,
]
@@ -41,13 +49,18 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
time_zone = str(hass.config.time_zone)
- manager = VeSync(username, password, time_zone)
+ manager = VeSync(
+ username=username,
+ password=password,
+ time_zone=time_zone,
+ debug=logging.getLogger("pyvesync.vesync").level == logging.DEBUG,
+ redact=True,
+ )
login = await hass.async_add_executor_job(manager.login)
if not login:
- _LOGGER.error("Unable to login to the VeSync server")
- return False
+ raise ConfigEntryAuthFailed
hass.data[DOMAIN] = {}
hass.data[DOMAIN][VS_MANAGER] = manager
@@ -61,6 +74,17 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
+ @callback
+ def _async_handle_logging_changed(_event: Event) -> None:
+ """Handle when the logging level changes."""
+ manager.debug = logging.getLogger("pyvesync.vesync").level == logging.DEBUG
+
+ cleanup = hass.bus.async_listen(
+ EVENT_LOGGING_CHANGED, _async_handle_logging_changed
+ )
+
+ hass.data[DOMAIN][VS_LISTENERS] = cleanup
+
async def async_new_device_discovery(service: ServiceCall) -> None:
"""Discover if new devices should be added."""
manager = hass.data[DOMAIN][VS_MANAGER]
@@ -86,7 +110,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
-
+ hass.data[DOMAIN][VS_LISTENERS]()
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
if unload_ok:
hass.data.pop(DOMAIN)
diff --git a/homeassistant/components/vesync/binary_sensor.py b/homeassistant/components/vesync/binary_sensor.py
index dd1b6398c06..7b6f14e04dc 100644
--- a/homeassistant/components/vesync/binary_sensor.py
+++ b/homeassistant/components/vesync/binary_sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import rgetattr
from .const import DOMAIN, VS_COORDINATOR, VS_DEVICES, VS_DISCOVERY
@@ -52,7 +52,7 @@ SENSOR_DESCRIPTIONS: tuple[VeSyncBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary_sensor platform."""
@@ -102,5 +102,4 @@ class VeSyncBinarySensor(BinarySensorEntity, VeSyncBaseEntity):
@property
def is_on(self) -> bool:
"""Return true if the binary sensor is on."""
- _LOGGER.debug(rgetattr(self.device, self.entity_description.key))
return self.entity_description.is_on(self.device)
diff --git a/homeassistant/components/vesync/config_flow.py b/homeassistant/components/vesync/config_flow.py
index 07543440e91..e5537d8fcc9 100644
--- a/homeassistant/components/vesync/config_flow.py
+++ b/homeassistant/components/vesync/config_flow.py
@@ -1,5 +1,6 @@
"""Config flow utilities."""
+from collections.abc import Mapping
from typing import Any
from pyvesync import VeSync
@@ -57,3 +58,36 @@ class VeSyncFlowHandler(ConfigFlow, domain=DOMAIN):
title=username,
data={CONF_USERNAME: username, CONF_PASSWORD: password},
)
+
+ async def async_step_reauth(
+ self, entry_data: Mapping[str, Any]
+ ) -> ConfigFlowResult:
+ """Handle re-authentication with vesync."""
+ return await self.async_step_reauth_confirm()
+
+ async def async_step_reauth_confirm(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Confirm re-authentication with vesync."""
+
+ if user_input:
+ username = user_input[CONF_USERNAME]
+ password = user_input[CONF_PASSWORD]
+
+ manager = VeSync(username, password)
+ login = await self.hass.async_add_executor_job(manager.login)
+ if login:
+ return self.async_update_reload_and_abort(
+ self._get_reauth_entry(),
+ data_updates={
+ CONF_USERNAME: username,
+ CONF_PASSWORD: password,
+ },
+ )
+
+ return self.async_show_form(
+ step_id="reauth_confirm",
+ data_schema=DATA_SCHEMA,
+ description_placeholders={"name": "VeSync"},
+ errors={"base": "invalid_auth"},
+ )
diff --git a/homeassistant/components/vesync/const.py b/homeassistant/components/vesync/const.py
index 34454081567..4e39fe40f2d 100644
--- a/homeassistant/components/vesync/const.py
+++ b/homeassistant/components/vesync/const.py
@@ -22,6 +22,7 @@ exceeds the quota of 7700.
VS_DEVICES = "devices"
VS_COORDINATOR = "coordinator"
VS_MANAGER = "manager"
+VS_LISTENERS = "listeners"
VS_NUMBERS = "numbers"
VS_HUMIDIFIER_MODE_AUTO = "auto"
@@ -29,6 +30,14 @@ VS_HUMIDIFIER_MODE_HUMIDITY = "humidity"
VS_HUMIDIFIER_MODE_MANUAL = "manual"
VS_HUMIDIFIER_MODE_SLEEP = "sleep"
+FAN_NIGHT_LIGHT_LEVEL_DIM = "dim"
+FAN_NIGHT_LIGHT_LEVEL_OFF = "off"
+FAN_NIGHT_LIGHT_LEVEL_ON = "on"
+
+HUMIDIFIER_NIGHT_LIGHT_LEVEL_BRIGHT = "bright"
+HUMIDIFIER_NIGHT_LIGHT_LEVEL_DIM = "dim"
+HUMIDIFIER_NIGHT_LIGHT_LEVEL_OFF = "off"
+
VeSyncHumidifierDevice = VeSyncHumid200300S | VeSyncSuperior6000S
"""Humidifier device types"""
@@ -59,6 +68,7 @@ SKU_TO_BASE_DEVICE = {
# Air Purifiers
"LV-PUR131S": "LV-PUR131S",
"LV-RH131S": "LV-PUR131S", # Alt ID Model LV-PUR131S
+ "LV-RH131S-WM": "LV-PUR131S", # Alt ID Model LV-PUR131S
"Core200S": "Core200S",
"LAP-C201S-AUSR": "Core200S", # Alt ID Model Core200S
"LAP-C202S-WUSR": "Core200S", # Alt ID Model Core200S
diff --git a/homeassistant/components/vesync/fan.py b/homeassistant/components/vesync/fan.py
index 21a92a22db2..daf734d50a8 100644
--- a/homeassistant/components/vesync/fan.py
+++ b/homeassistant/components/vesync/fan.py
@@ -12,7 +12,7 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -72,7 +72,7 @@ SPEED_RANGE = { # off is not included
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the VeSync fan platform."""
diff --git a/homeassistant/components/vesync/humidifier.py b/homeassistant/components/vesync/humidifier.py
index 5afe7360673..9a98a39aa8c 100644
--- a/homeassistant/components/vesync/humidifier.py
+++ b/homeassistant/components/vesync/humidifier.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import is_humidifier
from .const import (
@@ -50,7 +50,7 @@ VS_TO_HA_MODE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the VeSync humidifier platform."""
@@ -71,7 +71,7 @@ async def async_setup_entry(
@callback
def _setup_entities(
devices: list[VeSyncBaseDevice],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
coordinator: VeSyncDataCoordinator,
):
"""Add humidifier entities."""
diff --git a/homeassistant/components/vesync/light.py b/homeassistant/components/vesync/light.py
index 40f68986145..887400b2cf0 100644
--- a/homeassistant/components/vesync/light.py
+++ b/homeassistant/components/vesync/light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .const import DEV_TYPE_TO_HA, DOMAIN, VS_COORDINATOR, VS_DEVICES, VS_DISCOVERY
@@ -29,7 +29,7 @@ MIN_MIREDS = 153 # 1,000,000 divided by 6500 Kelvin = 153 Mireds
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up lights."""
diff --git a/homeassistant/components/vesync/manifest.json b/homeassistant/components/vesync/manifest.json
index b3697844f19..571c6ee0036 100644
--- a/homeassistant/components/vesync/manifest.json
+++ b/homeassistant/components/vesync/manifest.json
@@ -11,6 +11,6 @@
"config_flow": true,
"documentation": "https://www.home-assistant.io/integrations/vesync",
"iot_class": "cloud_polling",
- "loggers": ["pyvesync"],
- "requirements": ["pyvesync==2.1.17"]
+ "loggers": ["pyvesync.vesync"],
+ "requirements": ["pyvesync==2.1.18"]
}
diff --git a/homeassistant/components/vesync/number.py b/homeassistant/components/vesync/number.py
index 3c43cce28cf..707dd6ab30e 100644
--- a/homeassistant/components/vesync/number.py
+++ b/homeassistant/components/vesync/number.py
@@ -14,7 +14,7 @@ from homeassistant.components.number import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import is_humidifier
from .const import DOMAIN, VS_COORDINATOR, VS_DEVICES, VS_DISCOVERY
@@ -51,7 +51,7 @@ NUMBER_DESCRIPTIONS: list[VeSyncNumberEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up number entities."""
@@ -72,7 +72,7 @@ async def async_setup_entry(
@callback
def _setup_entities(
devices: list[VeSyncBaseDevice],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
coordinator: VeSyncDataCoordinator,
):
"""Add number entities."""
diff --git a/homeassistant/components/vesync/select.py b/homeassistant/components/vesync/select.py
new file mode 100644
index 00000000000..a9d2e1b533a
--- /dev/null
+++ b/homeassistant/components/vesync/select.py
@@ -0,0 +1,155 @@
+"""Support for VeSync numeric entities."""
+
+from collections.abc import Callable
+from dataclasses import dataclass
+import logging
+
+from pyvesync.vesyncbasedevice import VeSyncBaseDevice
+
+from homeassistant.components.select import SelectEntity, SelectEntityDescription
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .common import rgetattr
+from .const import (
+ DOMAIN,
+ FAN_NIGHT_LIGHT_LEVEL_DIM,
+ FAN_NIGHT_LIGHT_LEVEL_OFF,
+ FAN_NIGHT_LIGHT_LEVEL_ON,
+ HUMIDIFIER_NIGHT_LIGHT_LEVEL_BRIGHT,
+ HUMIDIFIER_NIGHT_LIGHT_LEVEL_DIM,
+ HUMIDIFIER_NIGHT_LIGHT_LEVEL_OFF,
+ VS_COORDINATOR,
+ VS_DEVICES,
+ VS_DISCOVERY,
+)
+from .coordinator import VeSyncDataCoordinator
+from .entity import VeSyncBaseEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+VS_TO_HA_HUMIDIFIER_NIGHT_LIGHT_LEVEL_MAP = {
+ 100: HUMIDIFIER_NIGHT_LIGHT_LEVEL_BRIGHT,
+ 50: HUMIDIFIER_NIGHT_LIGHT_LEVEL_DIM,
+ 0: HUMIDIFIER_NIGHT_LIGHT_LEVEL_OFF,
+}
+
+HA_TO_VS_HUMIDIFIER_NIGHT_LIGHT_LEVEL_MAP = {
+ v: k for k, v in VS_TO_HA_HUMIDIFIER_NIGHT_LIGHT_LEVEL_MAP.items()
+}
+
+
+@dataclass(frozen=True, kw_only=True)
+class VeSyncSelectEntityDescription(SelectEntityDescription):
+ """Class to describe a Vesync select entity."""
+
+ exists_fn: Callable[[VeSyncBaseDevice], bool]
+ current_option_fn: Callable[[VeSyncBaseDevice], str]
+ select_option_fn: Callable[[VeSyncBaseDevice, str], bool]
+
+
+SELECT_DESCRIPTIONS: list[VeSyncSelectEntityDescription] = [
+ # night_light for humidifier
+ VeSyncSelectEntityDescription(
+ key="night_light_level",
+ translation_key="night_light_level",
+ options=list(VS_TO_HA_HUMIDIFIER_NIGHT_LIGHT_LEVEL_MAP.values()),
+ icon="mdi:brightness-6",
+ exists_fn=lambda device: rgetattr(device, "set_night_light_brightness"),
+ # The select_option service framework ensures that only options specified are
+ # accepted. ServiceValidationError gets raised for invalid value.
+ select_option_fn=lambda device, value: device.set_night_light_brightness(
+ HA_TO_VS_HUMIDIFIER_NIGHT_LIGHT_LEVEL_MAP.get(value, 0)
+ ),
+ # Reporting "off" as the choice for unhandled level.
+ current_option_fn=lambda device: VS_TO_HA_HUMIDIFIER_NIGHT_LIGHT_LEVEL_MAP.get(
+ device.details.get("night_light_brightness"),
+ HUMIDIFIER_NIGHT_LIGHT_LEVEL_OFF,
+ ),
+ ),
+ # night_light for fan devices based on pyvesync.VeSyncAirBypass
+ VeSyncSelectEntityDescription(
+ key="night_light_level",
+ translation_key="night_light_level",
+ options=[
+ FAN_NIGHT_LIGHT_LEVEL_OFF,
+ FAN_NIGHT_LIGHT_LEVEL_DIM,
+ FAN_NIGHT_LIGHT_LEVEL_ON,
+ ],
+ icon="mdi:brightness-6",
+ exists_fn=lambda device: rgetattr(device, "set_night_light"),
+ select_option_fn=lambda device, value: device.set_night_light(value),
+ current_option_fn=lambda device: VS_TO_HA_HUMIDIFIER_NIGHT_LIGHT_LEVEL_MAP.get(
+ device.details.get("night_light"),
+ FAN_NIGHT_LIGHT_LEVEL_OFF,
+ ),
+ ),
+]
+
+
+async def async_setup_entry(
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+ """Set up select entities."""
+
+ coordinator = hass.data[DOMAIN][VS_COORDINATOR]
+
+ @callback
+ def discover(devices):
+ """Add new devices to platform."""
+ _setup_entities(devices, async_add_entities, coordinator)
+
+ config_entry.async_on_unload(
+ async_dispatcher_connect(hass, VS_DISCOVERY.format(VS_DEVICES), discover)
+ )
+
+ _setup_entities(hass.data[DOMAIN][VS_DEVICES], async_add_entities, coordinator)
+
+
+@callback
+def _setup_entities(
+ devices: list[VeSyncBaseDevice],
+ async_add_entities: AddConfigEntryEntitiesCallback,
+ coordinator: VeSyncDataCoordinator,
+):
+ """Add select entities."""
+
+ async_add_entities(
+ VeSyncSelectEntity(dev, description, coordinator)
+ for dev in devices
+ for description in SELECT_DESCRIPTIONS
+ if description.exists_fn(dev)
+ )
+
+
+class VeSyncSelectEntity(VeSyncBaseEntity, SelectEntity):
+ """A class to set numeric options on Vesync device."""
+
+ entity_description: VeSyncSelectEntityDescription
+
+ def __init__(
+ self,
+ device: VeSyncBaseDevice,
+ description: VeSyncSelectEntityDescription,
+ coordinator: VeSyncDataCoordinator,
+ ) -> None:
+ """Initialize the VeSync select device."""
+ super().__init__(device, coordinator)
+ self.entity_description = description
+ self._attr_unique_id = f"{super().unique_id}-{description.key}"
+
+ @property
+ def current_option(self) -> str | None:
+ """Return an option."""
+ return self.entity_description.current_option_fn(self.device)
+
+ async def async_select_option(self, option: str) -> None:
+ """Set an option."""
+ if await self.hass.async_add_executor_job(
+ self.entity_description.select_option_fn, self.device, option
+ ):
+ await self.coordinator.async_request_refresh()
diff --git a/homeassistant/components/vesync/sensor.py b/homeassistant/components/vesync/sensor.py
index bf52050d745..3bc6608989a 100644
--- a/homeassistant/components/vesync/sensor.py
+++ b/homeassistant/components/vesync/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .common import is_humidifier
@@ -194,7 +194,7 @@ SENSORS: tuple[VeSyncSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switches."""
@@ -215,7 +215,7 @@ async def async_setup_entry(
@callback
def _setup_entities(
devices: list[VeSyncBaseDevice],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
coordinator: VeSyncDataCoordinator,
):
"""Check if device is online and add entity."""
diff --git a/homeassistant/components/vesync/strings.json b/homeassistant/components/vesync/strings.json
index 3eb2a0c3fd5..9b63bf3e614 100644
--- a/homeassistant/components/vesync/strings.json
+++ b/homeassistant/components/vesync/strings.json
@@ -2,7 +2,15 @@
"config": {
"step": {
"user": {
- "title": "Enter Username and Password",
+ "title": "Enter username and password",
+ "data": {
+ "username": "[%key:common::config_flow::data::email%]",
+ "password": "[%key:common::config_flow::data::password%]"
+ }
+ },
+ "reauth_confirm": {
+ "title": "[%key:common::config_flow::title::reauth%]",
+ "description": "The VeSync integration needs to re-authenticate your account",
"data": {
"username": "[%key:common::config_flow::data::email%]",
"password": "[%key:common::config_flow::data::password%]"
@@ -13,7 +21,8 @@
"invalid_auth": "[%key:common::config_flow::error::invalid_auth%]"
},
"abort": {
- "single_instance_allowed": "[%key:common::config_flow::abort::single_instance_allowed%]"
+ "single_instance_allowed": "[%key:common::config_flow::abort::single_instance_allowed%]",
+ "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]"
}
},
"entity": {
@@ -56,6 +65,17 @@
"name": "Mist level"
}
},
+ "select": {
+ "night_light_level": {
+ "name": "Night light level",
+ "state": {
+ "bright": "Bright",
+ "dim": "Dim",
+ "off": "[%key:common::state::off%]",
+ "on": "[%key:common::state::on%]"
+ }
+ }
+ },
"fan": {
"vesync": {
"state_attributes": {
diff --git a/homeassistant/components/vesync/switch.py b/homeassistant/components/vesync/switch.py
index 3d2dc8a8e96..3e8deedb4ad 100644
--- a/homeassistant/components/vesync/switch.py
+++ b/homeassistant/components/vesync/switch.py
@@ -15,7 +15,7 @@ from homeassistant.components.switch import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import is_outlet, is_wall_switch
from .const import DOMAIN, VS_COORDINATOR, VS_DEVICES, VS_DISCOVERY
@@ -51,7 +51,7 @@ SENSOR_DESCRIPTIONS: Final[tuple[VeSyncSwitchEntityDescription, ...]] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up switch platform."""
diff --git a/homeassistant/components/vicare/binary_sensor.py b/homeassistant/components/vicare/binary_sensor.py
index 9d216404156..902dfd18d30 100644
--- a/homeassistant/components/vicare/binary_sensor.py
+++ b/homeassistant/components/vicare/binary_sensor.py
@@ -25,7 +25,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ViCareEntity
from .types import ViCareConfigEntry, ViCareDevice, ViCareRequiredKeysMixin
@@ -157,7 +157,7 @@ def _build_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ViCareConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the ViCare binary sensor devices."""
async_add_entities(
diff --git a/homeassistant/components/vicare/button.py b/homeassistant/components/vicare/button.py
index 65182990bfb..9c30a9e68ee 100644
--- a/homeassistant/components/vicare/button.py
+++ b/homeassistant/components/vicare/button.py
@@ -18,7 +18,7 @@ import requests
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ViCareEntity
from .types import ViCareConfigEntry, ViCareDevice, ViCareRequiredKeysMixinWithSet
@@ -66,7 +66,7 @@ def _build_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ViCareConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the ViCare button entities."""
async_add_entities(
diff --git a/homeassistant/components/vicare/climate.py b/homeassistant/components/vicare/climate.py
index f62fdc363a6..9fba83c5700 100644
--- a/homeassistant/components/vicare/climate.py
+++ b/homeassistant/components/vicare/climate.py
@@ -33,7 +33,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import ViCareEntity
@@ -98,7 +98,7 @@ def _build_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ViCareConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ViCare climate platform."""
diff --git a/homeassistant/components/vicare/entity.py b/homeassistant/components/vicare/entity.py
index 11955a94b94..7b73d2e5ba3 100644
--- a/homeassistant/components/vicare/entity.py
+++ b/homeassistant/components/vicare/entity.py
@@ -28,6 +28,7 @@ class ViCareEntity(Entity):
"""Initialize the entity."""
gateway_serial = device_config.getConfig().serial
device_id = device_config.getId()
+ model = device_config.getModel().replace("_", " ")
identifier = (
f"{gateway_serial}_{device_serial.replace('zigbee-', 'zigbee_')}"
@@ -45,8 +46,8 @@ class ViCareEntity(Entity):
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, identifier)},
serial_number=device_serial,
- name=device_config.getModel(),
+ name=model,
manufacturer="Viessmann",
- model=device_config.getModel(),
+ model=model,
configuration_url="https://developer.viessmann.com/",
)
diff --git a/homeassistant/components/vicare/fan.py b/homeassistant/components/vicare/fan.py
index c5e24f46c33..d84b2038dde 100644
--- a/homeassistant/components/vicare/fan.py
+++ b/homeassistant/components/vicare/fan.py
@@ -18,7 +18,7 @@ from requests.exceptions import ConnectionError as RequestConnectionError
from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -111,7 +111,7 @@ def _build_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ViCareConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ViCare fan platform."""
async_add_entities(
@@ -196,7 +196,10 @@ class ViCareFan(ViCareEntity, FanEntity):
@property
def is_on(self) -> bool | None:
"""Return true if the entity is on."""
- if self._api.getVentilationQuickmode(VentilationQuickmode.STANDBY):
+ if (
+ self._attr_supported_features & FanEntityFeature.TURN_OFF
+ and self._api.getVentilationQuickmode(VentilationQuickmode.STANDBY)
+ ):
return False
return self.percentage is not None and self.percentage > 0
@@ -209,7 +212,10 @@ class ViCareFan(ViCareEntity, FanEntity):
@property
def icon(self) -> str | None:
"""Return the icon to use in the frontend."""
- if self._api.getVentilationQuickmode(VentilationQuickmode.STANDBY):
+ if (
+ self._attr_supported_features & FanEntityFeature.TURN_OFF
+ and self._api.getVentilationQuickmode(VentilationQuickmode.STANDBY)
+ ):
return "mdi:fan-off"
if hasattr(self, "_attr_preset_mode"):
if self._attr_preset_mode == VentilationMode.VENTILATION:
diff --git a/homeassistant/components/vicare/manifest.json b/homeassistant/components/vicare/manifest.json
index 766cf22cb94..e39adaf6c4c 100644
--- a/homeassistant/components/vicare/manifest.json
+++ b/homeassistant/components/vicare/manifest.json
@@ -11,5 +11,5 @@
"documentation": "https://www.home-assistant.io/integrations/vicare",
"iot_class": "cloud_polling",
"loggers": ["PyViCare"],
- "requirements": ["PyViCare==2.41.0"]
+ "requirements": ["PyViCare==2.43.1"]
}
diff --git a/homeassistant/components/vicare/number.py b/homeassistant/components/vicare/number.py
index 534c0752cc1..04c4088bd3e 100644
--- a/homeassistant/components/vicare/number.py
+++ b/homeassistant/components/vicare/number.py
@@ -27,7 +27,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ViCareEntity
from .types import (
@@ -374,7 +374,7 @@ def _build_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ViCareConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the ViCare number devices."""
async_add_entities(
diff --git a/homeassistant/components/vicare/quality_scale.yaml b/homeassistant/components/vicare/quality_scale.yaml
index 55b7590a092..81b03364142 100644
--- a/homeassistant/components/vicare/quality_scale.yaml
+++ b/homeassistant/components/vicare/quality_scale.yaml
@@ -1,43 +1,70 @@
rules:
# Bronze
- config-flow: done
- test-before-configure: done
- unique-config-entry:
- status: todo
- comment: Uniqueness is not checked yet.
- config-flow-test-coverage: done
- runtime-data: done
- test-before-setup: done
- appropriate-polling: done
- entity-unique-id: done
- has-entity-name: done
- entity-event-setup:
- status: exempt
- comment: Entities of this integration does not explicitly subscribe to events.
- dependency-transparency: done
action-setup:
status: todo
comment: service registered in climate async_setup_entry.
+ appropriate-polling: done
+ brands: done
common-modules:
status: done
comment: No coordinator is used, data update is centrally handled by the library.
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions: done
docs-high-level-description: done
docs-installation-instructions: done
docs-removal-instructions: done
- docs-actions: done
- brands: done
+ entity-event-setup:
+ status: exempt
+ comment: Entities of this integration does not explicitly subscribe to events.
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry:
+ status: todo
+ comment: Uniqueness is not checked yet.
+
# Silver
- integration-owner: done
- reauthentication-flow: done
+ action-exceptions: todo
config-entry-unloading: done
+ docs-configuration-parameters: todo
+ docs-installation-parameters: todo
+ entity-unavailable: todo
+ integration-owner: done
+ log-when-unavailable: todo
+ parallel-updates: todo
+ reauthentication-flow: done
+ test-coverage: todo
+
# Gold
devices: done
diagnostics: done
- entity-category: done
+ discovery-update-info: todo
+ discovery: todo
+ docs-data-update: todo
+ docs-examples: todo
+ docs-known-limitations: todo
+ docs-supported-devices: todo
+ docs-supported-functions: todo
+ docs-troubleshooting: todo
+ docs-use-cases: todo
dynamic-devices: done
+ entity-category: done
entity-device-class: done
- entity-translations: done
entity-disabled-by-default: done
+ entity-translations: done
+ exception-translations: todo
+ icon-translations: todo
+ reconfiguration-flow: todo
repair-issues:
status: exempt
comment: This integration does not raise any repairable issues.
+ stale-devices: todo
+
+ # Platinum
+ async-dependency: todo
+ inject-websession: todo
+ strict-typing: todo
diff --git a/homeassistant/components/vicare/sensor.py b/homeassistant/components/vicare/sensor.py
index c99e7857d9b..cddc5ca021a 100644
--- a/homeassistant/components/vicare/sensor.py
+++ b/homeassistant/components/vicare/sensor.py
@@ -29,6 +29,7 @@ from homeassistant.const import (
PERCENTAGE,
EntityCategory,
UnitOfEnergy,
+ UnitOfMass,
UnitOfPower,
UnitOfPressure,
UnitOfTemperature,
@@ -37,7 +38,7 @@ from homeassistant.const import (
UnitOfVolumeFlowRate,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
VICARE_CUBIC_METER,
@@ -635,6 +636,38 @@ GLOBAL_SENSORS: tuple[ViCareSensorEntityDescription, ...] = (
device_class=SensorDeviceClass.TEMPERATURE,
state_class=SensorStateClass.MEASUREMENT,
),
+ ViCareSensorEntityDescription(
+ key="buffer_mid_top_temperature",
+ translation_key="buffer_mid_top_temperature",
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+ value_getter=lambda api: api.getBufferMidTopTemperature(),
+ ),
+ ViCareSensorEntityDescription(
+ key="buffer_middle_temperature",
+ translation_key="buffer_middle_temperature",
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+ value_getter=lambda api: api.getBufferMiddleTemperature(),
+ ),
+ ViCareSensorEntityDescription(
+ key="buffer_mid_bottom_temperature",
+ translation_key="buffer_mid_bottom_temperature",
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+ value_getter=lambda api: api.getBufferMidBottomTemperature(),
+ ),
+ ViCareSensorEntityDescription(
+ key="buffer_bottom_temperature",
+ translation_key="buffer_bottom_temperature",
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.TEMPERATURE,
+ native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+ value_getter=lambda api: api.getBufferBottomTemperature(),
+ ),
ViCareSensorEntityDescription(
key="buffer main temperature",
translation_key="buffer_main_temperature",
@@ -883,6 +916,23 @@ GLOBAL_SENSORS: tuple[ViCareSensorEntityDescription, ...] = (
entity_category=EntityCategory.DIAGNOSTIC,
value_getter=lambda api: api.getSeasonalPerformanceFactorHeating(),
),
+ ViCareSensorEntityDescription(
+ key="battery_level",
+ native_unit_of_measurement=PERCENTAGE,
+ state_class=SensorStateClass.MEASUREMENT,
+ device_class=SensorDeviceClass.BATTERY,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ value_getter=lambda api: api.getBatteryLevel(),
+ ),
+ ViCareSensorEntityDescription(
+ key="fuel_need",
+ translation_key="fuel_need",
+ state_class=SensorStateClass.MEASUREMENT,
+ entity_category=EntityCategory.DIAGNOSTIC,
+ native_unit_of_measurement=UnitOfMass.KILOGRAMS,
+ value_getter=lambda api: api.getFuelNeed(),
+ unit_getter=lambda api: api.getFuelUnit(),
+ ),
)
CIRCUIT_SENSORS: tuple[ViCareSensorEntityDescription, ...] = (
@@ -1033,7 +1083,7 @@ def _build_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ViCareConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create the ViCare sensor devices."""
async_add_entities(
diff --git a/homeassistant/components/vicare/strings.json b/homeassistant/components/vicare/strings.json
index 50eeaf038e0..04049f026bd 100644
--- a/homeassistant/components/vicare/strings.json
+++ b/homeassistant/components/vicare/strings.json
@@ -338,6 +338,18 @@
"buffer_top_temperature": {
"name": "Buffer top temperature"
},
+ "buffer_mid_top_temperature": {
+ "name": "Buffer mid top temperature"
+ },
+ "buffer_middle_temperature": {
+ "name": "Buffer middle temperature"
+ },
+ "buffer_mid_bottom_temperature": {
+ "name": "Buffer mid bottom temperature"
+ },
+ "buffer_bottom_temperature": {
+ "name": "Buffer bottom temperature"
+ },
"buffer_main_temperature": {
"name": "Buffer main temperature"
},
@@ -478,6 +490,9 @@
},
"spf_heating": {
"name": "Seasonal performance factor - heating"
+ },
+ "fuel_need": {
+ "name": "Fuel need"
}
},
"water_heater": {
@@ -500,11 +515,11 @@
"services": {
"set_vicare_mode": {
"name": "Set ViCare mode",
- "description": "Set a ViCare mode.",
+ "description": "Sets the mode of the climate device as defined by Viessmann.",
"fields": {
"vicare_mode": {
"name": "ViCare mode",
- "description": "ViCare mode."
+ "description": "For supported values, see the `vicare_modes` attribute of the climate entity."
}
}
}
diff --git a/homeassistant/components/vicare/water_heater.py b/homeassistant/components/vicare/water_heater.py
index 114ff620c3f..f92c9e3e1af 100644
--- a/homeassistant/components/vicare/water_heater.py
+++ b/homeassistant/components/vicare/water_heater.py
@@ -22,7 +22,7 @@ from homeassistant.components.water_heater import (
)
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_TENTHS, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ViCareEntity
from .types import ViCareConfigEntry, ViCareDevice
@@ -80,7 +80,7 @@ def _build_entities(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ViCareConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ViCare water heater platform."""
async_add_entities(
diff --git a/homeassistant/components/vilfo/sensor.py b/homeassistant/components/vilfo/sensor.py
index 77a7df7a0a8..fa2d5cae196 100644
--- a/homeassistant/components/vilfo/sensor.py
+++ b/homeassistant/components/vilfo/sensor.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_API_DATA_FIELD_BOOT_TIME,
@@ -51,7 +51,7 @@ SENSOR_TYPES: tuple[VilfoSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add Vilfo Router entities from a config_entry."""
vilfo = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/vizio/__init__.py b/homeassistant/components/vizio/__init__.py
index 4af42d76b62..10a71695e05 100644
--- a/homeassistant/components/vizio/__init__.py
+++ b/homeassistant/components/vizio/__init__.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from typing import Any
from homeassistant.components.media_player import MediaPlayerDeviceClass
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE_CLASS, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.storage import Store
@@ -25,7 +25,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
and entry.data[CONF_DEVICE_CLASS] == MediaPlayerDeviceClass.TV
):
store: Store[list[dict[str, Any]]] = Store(hass, 1, DOMAIN)
- coordinator = VizioAppsDataUpdateCoordinator(hass, store)
+ coordinator = VizioAppsDataUpdateCoordinator(hass, entry, store)
await coordinator.async_config_entry_first_refresh()
hass.data[DOMAIN][CONF_APPS] = coordinator
@@ -39,12 +39,9 @@ async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
unload_ok = await hass.config_entries.async_unload_platforms(
config_entry, PLATFORMS
)
- # Exclude this config entry because its not unloaded yet
if not any(
- entry.state is ConfigEntryState.LOADED
- and entry.entry_id != config_entry.entry_id
- and entry.data[CONF_DEVICE_CLASS] == MediaPlayerDeviceClass.TV
- for entry in hass.config_entries.async_entries(DOMAIN)
+ entry.data[CONF_DEVICE_CLASS] == MediaPlayerDeviceClass.TV
+ for entry in hass.config_entries.async_loaded_entries(DOMAIN)
):
hass.data[DOMAIN].pop(CONF_APPS, None)
diff --git a/homeassistant/components/vizio/coordinator.py b/homeassistant/components/vizio/coordinator.py
index a7ca7d7f9ed..0f95c8a53b7 100644
--- a/homeassistant/components/vizio/coordinator.py
+++ b/homeassistant/components/vizio/coordinator.py
@@ -9,6 +9,7 @@ from typing import Any
from pyvizio.const import APPS
from pyvizio.util import gen_apps_list_from_url
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.storage import Store
@@ -22,11 +23,19 @@ _LOGGER = logging.getLogger(__name__)
class VizioAppsDataUpdateCoordinator(DataUpdateCoordinator[list[dict[str, Any]]]):
"""Define an object to hold Vizio app config data."""
- def __init__(self, hass: HomeAssistant, store: Store[list[dict[str, Any]]]) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: ConfigEntry,
+ store: Store[list[dict[str, Any]]],
+ ) -> None:
"""Initialize."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(days=1),
)
diff --git a/homeassistant/components/vizio/media_player.py b/homeassistant/components/vizio/media_player.py
index 5711d8fbac9..d44db5e45ee 100644
--- a/homeassistant/components/vizio/media_player.py
+++ b/homeassistant/components/vizio/media_player.py
@@ -32,7 +32,7 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_ADDITIONAL_CONFIGS,
@@ -63,7 +63,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Vizio media player entry."""
host = config_entry.data[CONF_HOST]
diff --git a/homeassistant/components/vlc_telnet/media_player.py b/homeassistant/components/vlc_telnet/media_player.py
index 9597c706570..6ae9fbb9f5a 100644
--- a/homeassistant/components/vlc_telnet/media_player.py
+++ b/homeassistant/components/vlc_telnet/media_player.py
@@ -22,7 +22,7 @@ from homeassistant.config_entries import SOURCE_HASSIO, ConfigEntry
from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import VlcConfigEntry
@@ -39,7 +39,9 @@ def _get_str(data: dict, key: str) -> str | None:
async def async_setup_entry(
- hass: HomeAssistant, entry: VlcConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: VlcConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the vlc platform."""
# CONF_NAME is only present in imported YAML.
diff --git a/homeassistant/components/vodafone_station/__init__.py b/homeassistant/components/vodafone_station/__init__.py
index 871afe09a2e..9f118fe4fbd 100644
--- a/homeassistant/components/vodafone_station/__init__.py
+++ b/homeassistant/components/vodafone_station/__init__.py
@@ -1,16 +1,15 @@
"""Vodafone Station integration."""
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from .const import DOMAIN
-from .coordinator import VodafoneStationRouter
+from .coordinator import VodafoneConfigEntry, VodafoneStationRouter
PLATFORMS = [Platform.BUTTON, Platform.DEVICE_TRACKER, Platform.SENSOR]
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_setup_entry(hass: HomeAssistant, entry: VodafoneConfigEntry) -> bool:
"""Set up Vodafone Station platform."""
coordinator = VodafoneStationRouter(
hass,
@@ -22,7 +21,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await coordinator.async_config_entry_first_refresh()
- hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
+ entry.runtime_data = coordinator
entry.async_on_unload(entry.add_update_listener(update_listener))
@@ -31,10 +30,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: VodafoneConfigEntry) -> bool:
"""Unload a config entry."""
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
- coordinator: VodafoneStationRouter = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
await coordinator.api.logout()
await coordinator.api.close()
hass.data[DOMAIN].pop(entry.entry_id)
@@ -42,7 +41,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return unload_ok
-async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
+async def update_listener(hass: HomeAssistant, entry: VodafoneConfigEntry) -> None:
"""Update when config_entry options update."""
if entry.options:
await hass.config_entries.async_reload(entry.entry_id)
diff --git a/homeassistant/components/vodafone_station/button.py b/homeassistant/components/vodafone_station/button.py
index efea011a541..8dda4d49c7b 100644
--- a/homeassistant/components/vodafone_station/button.py
+++ b/homeassistant/components/vodafone_station/button.py
@@ -4,21 +4,32 @@ from __future__ import annotations
from collections.abc import Callable
from dataclasses import dataclass
+from json.decoder import JSONDecodeError
from typing import Any, Final
+from aiovodafone.exceptions import (
+ AlreadyLogged,
+ CannotAuthenticate,
+ CannotConnect,
+ GenericLoginError,
+)
+
from homeassistant.components.button import (
ButtonDeviceClass,
ButtonEntity,
ButtonEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import _LOGGER, DOMAIN
-from .coordinator import VodafoneStationRouter
+from .coordinator import VodafoneConfigEntry, VodafoneStationRouter
+
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
@dataclass(frozen=True, kw_only=True)
@@ -67,12 +78,14 @@ BUTTON_TYPES: Final = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: VodafoneConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
_LOGGER.debug("Setting up Vodafone Station buttons")
- coordinator: VodafoneStationRouter = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
sensors_data = coordinator.data.sensors
@@ -104,4 +117,25 @@ class VodafoneStationSensorEntity(
async def async_press(self) -> None:
"""Triggers the Shelly button press service."""
- await self.entity_description.press_action(self.coordinator)
+
+ try:
+ await self.entity_description.press_action(self.coordinator)
+ except CannotAuthenticate as err:
+ self.coordinator.config_entry.async_start_reauth(self.hass)
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="cannot_authenticate",
+ translation_placeholders={"error": repr(err)},
+ ) from err
+ except (
+ CannotConnect,
+ AlreadyLogged,
+ GenericLoginError,
+ JSONDecodeError,
+ ) as err:
+ self.coordinator.last_update_success = False
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="cannot_execute_action",
+ translation_placeholders={"error": repr(err)},
+ ) from err
diff --git a/homeassistant/components/vodafone_station/config_flow.py b/homeassistant/components/vodafone_station/config_flow.py
index 7a80244f8d6..fd0683bdacc 100644
--- a/homeassistant/components/vodafone_station/config_flow.py
+++ b/homeassistant/components/vodafone_station/config_flow.py
@@ -12,16 +12,12 @@ from homeassistant.components.device_tracker import (
CONF_CONSIDER_HOME,
DEFAULT_CONSIDER_HOME,
)
-from homeassistant.config_entries import (
- ConfigEntry,
- ConfigFlow,
- ConfigFlowResult,
- OptionsFlow,
-)
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult, OptionsFlow
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant, callback
from .const import _LOGGER, DEFAULT_HOST, DEFAULT_USERNAME, DOMAIN
+from .coordinator import VodafoneConfigEntry
def user_form_schema(user_input: dict[str, Any] | None) -> vol.Schema:
@@ -63,7 +59,7 @@ class VodafoneStationConfigFlow(ConfigFlow, domain=DOMAIN):
@staticmethod
@callback
def async_get_options_flow(
- config_entry: ConfigEntry,
+ config_entry: VodafoneConfigEntry,
) -> VodafoneStationOptionsFlowHandler:
"""Get the options flow for this handler."""
return VodafoneStationOptionsFlowHandler()
diff --git a/homeassistant/components/vodafone_station/coordinator.py b/homeassistant/components/vodafone_station/coordinator.py
index cd640d10cb6..cee66bd2e7c 100644
--- a/homeassistant/components/vodafone_station/coordinator.py
+++ b/homeassistant/components/vodafone_station/coordinator.py
@@ -3,7 +3,7 @@
from dataclasses import dataclass
from datetime import datetime, timedelta
from json.decoder import JSONDecodeError
-from typing import Any
+from typing import Any, cast
from aiovodafone import VodafoneStationDevice, VodafoneStationSercommApi, exceptions
@@ -21,6 +21,8 @@ from .helpers import cleanup_device_tracker
CONSIDER_HOME_SECONDS = DEFAULT_CONSIDER_HOME.total_seconds()
+type VodafoneConfigEntry = ConfigEntry[VodafoneStationRouter]
+
@dataclass(slots=True)
class VodafoneStationDeviceInfo:
@@ -42,7 +44,7 @@ class UpdateCoordinatorDataType:
class VodafoneStationRouter(DataUpdateCoordinator[UpdateCoordinatorDataType]):
"""Queries router running Vodafone Station firmware."""
- config_entry: ConfigEntry
+ config_entry: VodafoneConfigEntry
def __init__(
self,
@@ -50,7 +52,7 @@ class VodafoneStationRouter(DataUpdateCoordinator[UpdateCoordinatorDataType]):
host: str,
username: str,
password: str,
- config_entry: ConfigEntry,
+ config_entry: VodafoneConfigEntry,
) -> None:
"""Initialize the scanner."""
@@ -120,14 +122,22 @@ class VodafoneStationRouter(DataUpdateCoordinator[UpdateCoordinatorDataType]):
data_sensors = await self.api.get_sensor_data()
await self.api.logout()
except exceptions.CannotAuthenticate as err:
- raise ConfigEntryAuthFailed from err
+ raise ConfigEntryAuthFailed(
+ translation_domain=DOMAIN,
+ translation_key="cannot_authenticate",
+ translation_placeholders={"error": repr(err)},
+ ) from err
except (
exceptions.CannotConnect,
exceptions.AlreadyLogged,
exceptions.GenericLoginError,
JSONDecodeError,
) as err:
- raise UpdateFailed(f"Error fetching data: {err!r}") from err
+ raise UpdateFailed(
+ translation_domain=DOMAIN,
+ translation_key="update_failed",
+ translation_placeholders={"error": repr(err)},
+ ) from err
except (ConfigEntryAuthFailed, UpdateFailed):
await self.api.close()
raise
@@ -164,7 +174,7 @@ class VodafoneStationRouter(DataUpdateCoordinator[UpdateCoordinatorDataType]):
@property
def serial_number(self) -> str:
"""Device serial number."""
- return self.data.sensors["sys_serial_number"]
+ return cast(str, self.data.sensors["sys_serial_number"])
@property
def device_info(self) -> DeviceInfo:
diff --git a/homeassistant/components/vodafone_station/device_tracker.py b/homeassistant/components/vodafone_station/device_tracker.py
index 4af0b85e003..4efa26cda8c 100644
--- a/homeassistant/components/vodafone_station/device_tracker.py
+++ b/homeassistant/components/vodafone_station/device_tracker.py
@@ -3,23 +3,31 @@
from __future__ import annotations
from homeassistant.components.device_tracker import ScannerEntity
-from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .const import _LOGGER, DOMAIN
-from .coordinator import VodafoneStationDeviceInfo, VodafoneStationRouter
+from .const import _LOGGER
+from .coordinator import (
+ VodafoneConfigEntry,
+ VodafoneStationDeviceInfo,
+ VodafoneStationRouter,
+)
+
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: VodafoneConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device tracker for Vodafone Station component."""
_LOGGER.debug("Start device trackers setup")
- coordinator: VodafoneStationRouter = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
tracked: set = set()
@@ -40,7 +48,7 @@ async def async_setup_entry(
@callback
def async_add_new_tracked_entities(
coordinator: VodafoneStationRouter,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
tracked: set[str],
) -> None:
"""Add new tracker entities from the router."""
diff --git a/homeassistant/components/vodafone_station/diagnostics.py b/homeassistant/components/vodafone_station/diagnostics.py
index e306d6caca2..4778e7d5a4e 100644
--- a/homeassistant/components/vodafone_station/diagnostics.py
+++ b/homeassistant/components/vodafone_station/diagnostics.py
@@ -5,22 +5,20 @@ from __future__ import annotations
from typing import Any
from homeassistant.components.diagnostics import async_redact_data
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
-from .const import DOMAIN
-from .coordinator import VodafoneStationRouter
+from .coordinator import VodafoneConfigEntry
TO_REDACT = {CONF_USERNAME, CONF_PASSWORD}
async def async_get_config_entry_diagnostics(
- hass: HomeAssistant, entry: ConfigEntry
+ hass: HomeAssistant, entry: VodafoneConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
- coordinator: VodafoneStationRouter = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
sensors_data = coordinator.data.sensors
return {
diff --git a/homeassistant/components/vodafone_station/manifest.json b/homeassistant/components/vodafone_station/manifest.json
index 4acafc8df3a..29cb3c070ab 100644
--- a/homeassistant/components/vodafone_station/manifest.json
+++ b/homeassistant/components/vodafone_station/manifest.json
@@ -7,5 +7,6 @@
"integration_type": "hub",
"iot_class": "local_polling",
"loggers": ["aiovodafone"],
+ "quality_scale": "silver",
"requirements": ["aiovodafone==0.6.1"]
}
diff --git a/homeassistant/components/vodafone_station/quality_scale.yaml b/homeassistant/components/vodafone_station/quality_scale.yaml
new file mode 100644
index 00000000000..fe114b4b324
--- /dev/null
+++ b/homeassistant/components/vodafone_station/quality_scale.yaml
@@ -0,0 +1,84 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: no actions
+ appropriate-polling: done
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: no actions
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: exempt
+ comment: no events
+ entity-unique-id: done
+ has-entity-name: done
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions: done
+ config-entry-unloading: done
+ docs-configuration-parameters: done
+ docs-installation-parameters: done
+ entity-unavailable: done
+ integration-owner: done
+ log-when-unavailable: done
+ parallel-updates: done
+ reauthentication-flow: done
+ test-coverage: done
+
+ # Gold
+ devices: done
+ diagnostics: done
+ discovery-update-info:
+ status: exempt
+ comment: device not discoverable
+ discovery:
+ status: exempt
+ comment: device not discoverable
+ docs-data-update: done
+ docs-examples:
+ status: todo
+ comment: add some automation example
+ docs-known-limitations:
+ status: exempt
+ comment: no known limitations, yet
+ docs-supported-devices: done
+ docs-supported-functions: done
+ docs-troubleshooting:
+ status: todo
+ comment: add some info for troubleshooting
+ docs-use-cases:
+ status: todo
+ comment: add some use caes
+ dynamic-devices: done
+ entity-category: done
+ entity-device-class: done
+ entity-disabled-by-default:
+ status: exempt
+ comment: no known use case
+ entity-translations: done
+ exception-translations: done
+ icon-translations: done
+ reconfiguration-flow:
+ status: todo
+ comment: handle host change
+ repair-issues:
+ status: exempt
+ comment: no known use cases for repair issues or flows, yet
+ stale-devices: done
+
+ # Platinum
+ async-dependency: done
+ inject-websession: done
+ strict-typing: done
diff --git a/homeassistant/components/vodafone_station/sensor.py b/homeassistant/components/vodafone_station/sensor.py
index 307fcaf0ea8..2573864330d 100644
--- a/homeassistant/components/vodafone_station/sensor.py
+++ b/homeassistant/components/vodafone_station/sensor.py
@@ -12,14 +12,16 @@ from homeassistant.components.sensor import (
SensorEntity,
SensorEntityDescription,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfDataRate
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from .const import _LOGGER, DOMAIN, LINE_TYPES
-from .coordinator import VodafoneStationRouter
+from .const import _LOGGER, LINE_TYPES
+from .coordinator import VodafoneConfigEntry, VodafoneStationRouter
+
+# Coordinator is used to centralize the data updates
+PARALLEL_UPDATES = 0
NOT_AVAILABLE: list = ["", "N/A", "0.0.0.0"]
UPTIME_DEVIATION = 60
@@ -165,12 +167,14 @@ SENSOR_TYPES: Final = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: VodafoneConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
_LOGGER.debug("Setting up Vodafone Station sensors")
- coordinator: VodafoneStationRouter = hass.data[DOMAIN][entry.entry_id]
+ coordinator = entry.runtime_data
sensors_data = coordinator.data.sensors
diff --git a/homeassistant/components/vodafone_station/strings.json b/homeassistant/components/vodafone_station/strings.json
index 8910d7178b7..6e308c35e4f 100644
--- a/homeassistant/components/vodafone_station/strings.json
+++ b/homeassistant/components/vodafone_station/strings.json
@@ -3,9 +3,11 @@
"flow_title": "{host}",
"step": {
"reauth_confirm": {
- "description": "Please enter the correct password for host: {host}",
"data": {
"password": "[%key:common::config_flow::data::password%]"
+ },
+ "data_description": {
+ "password": "[%key:component::vodafone_station::config::step::user::data_description::password%]"
}
},
"user": {
@@ -15,7 +17,9 @@
"password": "[%key:common::config_flow::data::password%]"
},
"data_description": {
- "host": "The hostname or IP address of your Vodafone Station."
+ "host": "The hostname or IP address of your Vodafone Station.",
+ "username": "The username for your Vodafone Station.",
+ "password": "The password for your Vodafone Station."
}
}
},
@@ -29,10 +33,10 @@
"unknown": "[%key:common::config_flow::error::unknown%]"
},
"error": {
- "already_logged": "User already logged-in, please try again later.",
+ "already_logged": "[%key:component::vodafone_station::config::abort::already_logged%]",
"cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
"invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
- "model_not_supported": "The device model is currently unsupported.",
+ "model_not_supported": "[%key:component::vodafone_station::config::abort::model_not_supported%]",
"unknown": "[%key:common::config_flow::error::unknown%]"
}
},
@@ -41,20 +45,35 @@
"init": {
"data": {
"consider_home": "Seconds to consider a device at 'home'"
+ },
+ "data_description": {
+ "consider_home": "The number of seconds to wait until marking a device as not home after it disconnects from the network."
}
}
}
},
"entity": {
"button": {
- "dsl_reconnect": { "name": "DSL reconnect" },
- "fiber_reconnect": { "name": "Fiber reconnect" },
- "internet_key_reconnect": { "name": "Internet key reconnect" }
+ "dsl_reconnect": {
+ "name": "DSL reconnect"
+ },
+ "fiber_reconnect": {
+ "name": "Fiber reconnect"
+ },
+ "internet_key_reconnect": {
+ "name": "Internet key reconnect"
+ }
},
"sensor": {
- "external_ipv4": { "name": "WAN IPv4 address" },
- "external_ipv6": { "name": "WAN IPv6 address" },
- "external_ip_key": { "name": "WAN internet key address" },
+ "external_ipv4": {
+ "name": "WAN IPv4 address"
+ },
+ "external_ipv6": {
+ "name": "WAN IPv6 address"
+ },
+ "external_ip_key": {
+ "name": "WAN internet key address"
+ },
"active_connection": {
"name": "Active connection",
"state": {
@@ -64,15 +83,44 @@
"internet_key": "Internet key"
}
},
- "down_stream": { "name": "WAN download rate" },
- "up_stream": { "name": "WAN upload rate" },
- "fw_version": { "name": "Firmware version" },
- "phone_num1": { "name": "Phone number (1)" },
- "phone_num2": { "name": "Phone number (2)" },
- "sys_uptime": { "name": "Uptime" },
- "sys_cpu_usage": { "name": "CPU usage" },
- "sys_memory_usage": { "name": "Memory usage" },
- "sys_reboot_cause": { "name": "Reboot cause" }
+ "down_stream": {
+ "name": "WAN download rate"
+ },
+ "up_stream": {
+ "name": "WAN upload rate"
+ },
+ "fw_version": {
+ "name": "Firmware version"
+ },
+ "phone_num1": {
+ "name": "Phone number (1)"
+ },
+ "phone_num2": {
+ "name": "Phone number (2)"
+ },
+ "sys_uptime": {
+ "name": "Uptime"
+ },
+ "sys_cpu_usage": {
+ "name": "CPU usage"
+ },
+ "sys_memory_usage": {
+ "name": "Memory usage"
+ },
+ "sys_reboot_cause": {
+ "name": "Reboot cause"
+ }
+ }
+ },
+ "exceptions": {
+ "update_failed": {
+ "message": "Error fetching data: {error}"
+ },
+ "cannot_execute_action": {
+ "message": "Cannot execute requested action: {error}"
+ },
+ "cannot_authenticate": {
+ "message": "Error authenticating: {error}"
}
}
}
diff --git a/homeassistant/components/voip/assist_satellite.py b/homeassistant/components/voip/assist_satellite.py
index 1877b8c655c..2c0a3b9641a 100644
--- a/homeassistant/components/voip/assist_satellite.py
+++ b/homeassistant/components/voip/assist_satellite.py
@@ -3,6 +3,7 @@
from __future__ import annotations
import asyncio
+from datetime import timedelta
from enum import IntFlag
from functools import partial
import io
@@ -16,7 +17,7 @@ import wave
from voip_utils import SIP_PORT, RtpDatagramProtocol
from voip_utils.sip import SipDatagramProtocol, SipEndpoint, get_sip_endpoint
-from homeassistant.components import tts
+from homeassistant.components import intent, tts
from homeassistant.components.assist_pipeline import PipelineEvent, PipelineEventType
from homeassistant.components.assist_satellite import (
AssistSatelliteAnnouncement,
@@ -25,12 +26,22 @@ from homeassistant.components.assist_satellite import (
AssistSatelliteEntityDescription,
AssistSatelliteEntityFeature,
)
+from homeassistant.components.intent import TimerEventType, TimerInfo
from homeassistant.components.network import async_get_source_ip
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import Context, HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import CHANNELS, CONF_SIP_PORT, DOMAIN, RATE, RTP_AUDIO_SETTINGS, WIDTH
+from .const import (
+ CHANNELS,
+ CONF_SIP_PORT,
+ CONF_SIP_USER,
+ DOMAIN,
+ RATE,
+ RTP_AUDIO_SETTINGS,
+ WIDTH,
+)
from .devices import VoIPDevice
from .entity import VoIPEntity
@@ -64,7 +75,7 @@ _TONE_FILENAMES: dict[Tones, str] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up VoIP Assist satellite entity."""
domain_data: DomainData = hass.data[DOMAIN]
@@ -152,6 +163,13 @@ class VoipAssistSatellite(VoIPEntity, AssistSatelliteEntity, RtpDatagramProtocol
await super().async_added_to_hass()
self.voip_device.protocol = self
+ assert self.device_entry is not None
+ self.async_on_remove(
+ intent.async_register_timer_handler(
+ self.hass, self.device_entry.id, self.async_handle_timer_event
+ )
+ )
+
async def async_will_remove_from_hass(self) -> None:
"""Run when entity will be removed from hass."""
await super().async_will_remove_from_hass()
@@ -165,6 +183,29 @@ class VoipAssistSatellite(VoIPEntity, AssistSatelliteEntity, RtpDatagramProtocol
"""Get the current satellite configuration."""
raise NotImplementedError
+ @callback
+ def async_handle_timer_event(
+ self,
+ event_type: TimerEventType,
+ timer_info: TimerInfo,
+ ) -> None:
+ """Handle timer event."""
+ if event_type != TimerEventType.FINISHED:
+ return
+
+ if timer_info.name:
+ message = f"{timer_info.name} finished"
+ else:
+ message = f"{timedelta(seconds=timer_info.created_seconds)} timer finished"
+
+ async def announce_message():
+ announcement = await self._resolve_announcement_media_id(message, None)
+ await self.async_announce(announcement)
+
+ self.config_entry.async_create_background_task(
+ self.hass, announce_message(), "voip_announce_timer"
+ )
+
async def async_set_configuration(
self, config: AssistSatelliteConfiguration
) -> None:
@@ -185,6 +226,12 @@ class VoipAssistSatellite(VoIPEntity, AssistSatelliteEntity, RtpDatagramProtocol
Optionally run a voice pipeline after the announcement has finished.
"""
+ if announcement.media_id_source != "tts":
+ raise HomeAssistantError(
+ translation_domain=DOMAIN,
+ translation_key="non_tts_announcement",
+ )
+
self._announcement_future = asyncio.Future()
self._run_pipeline_after_announce = run_pipeline_after
@@ -199,7 +246,10 @@ class VoipAssistSatellite(VoIPEntity, AssistSatelliteEntity, RtpDatagramProtocol
# HA SIP server
source_ip = await async_get_source_ip(self.hass)
sip_port = self.config_entry.options.get(CONF_SIP_PORT, SIP_PORT)
- source_endpoint = get_sip_endpoint(host=source_ip, port=sip_port)
+ sip_user = self.config_entry.options.get(CONF_SIP_USER)
+ source_endpoint = get_sip_endpoint(
+ host=source_ip, port=sip_port, username=sip_user
+ )
try:
# VoIP ID is SIP header
diff --git a/homeassistant/components/voip/binary_sensor.py b/homeassistant/components/voip/binary_sensor.py
index f38b228c46c..34dac4b6068 100644
--- a/homeassistant/components/voip/binary_sensor.py
+++ b/homeassistant/components/voip/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import issue_registry as ir
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .devices import VoIPDevice
@@ -24,7 +24,7 @@ if TYPE_CHECKING:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up VoIP binary sensor entities."""
domain_data: DomainData = hass.data[DOMAIN]
diff --git a/homeassistant/components/voip/config_flow.py b/homeassistant/components/voip/config_flow.py
index 63dcb8f86ee..7ae603f0f6a 100644
--- a/homeassistant/components/voip/config_flow.py
+++ b/homeassistant/components/voip/config_flow.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import (
from homeassistant.core import callback
from homeassistant.helpers import config_validation as cv
-from .const import CONF_SIP_PORT, DOMAIN
+from .const import CONF_SIP_PORT, CONF_SIP_USER, DOMAIN
class VoIPConfigFlow(ConfigFlow, domain=DOMAIN):
@@ -58,7 +58,15 @@ class VoipOptionsFlowHandler(OptionsFlow):
) -> ConfigFlowResult:
"""Manage the options."""
if user_input is not None:
- return self.async_create_entry(title="", data=user_input)
+ if CONF_SIP_USER in user_input and not user_input[CONF_SIP_USER]:
+ del user_input[CONF_SIP_USER]
+ self.hass.config_entries.async_update_entry(
+ self.config_entry, options=user_input
+ )
+ return self.async_create_entry(
+ title="",
+ data=user_input,
+ )
return self.async_show_form(
step_id="init",
@@ -70,7 +78,15 @@ class VoipOptionsFlowHandler(OptionsFlow):
CONF_SIP_PORT,
SIP_PORT,
),
- ): cv.port
+ ): cv.port,
+ vol.Optional(
+ CONF_SIP_USER,
+ description={
+ "suggested_value": self.config_entry.options.get(
+ CONF_SIP_USER, None
+ )
+ },
+ ): vol.Any(None, cv.string),
}
),
)
diff --git a/homeassistant/components/voip/const.py b/homeassistant/components/voip/const.py
index b4ee5d8ce7a..9a4403f9df2 100644
--- a/homeassistant/components/voip/const.py
+++ b/homeassistant/components/voip/const.py
@@ -13,3 +13,4 @@ RTP_AUDIO_SETTINGS = {
}
CONF_SIP_PORT = "sip_port"
+CONF_SIP_USER = "sip_user"
diff --git a/homeassistant/components/voip/manifest.json b/homeassistant/components/voip/manifest.json
index e3b2861dbe5..dfd397fde14 100644
--- a/homeassistant/components/voip/manifest.json
+++ b/homeassistant/components/voip/manifest.json
@@ -3,9 +3,10 @@
"name": "Voice over IP",
"codeowners": ["@balloob", "@synesthesiam"],
"config_flow": true,
- "dependencies": ["assist_pipeline", "assist_satellite", "network"],
+ "dependencies": ["assist_pipeline", "assist_satellite", "intent", "network"],
"documentation": "https://www.home-assistant.io/integrations/voip",
"iot_class": "local_push",
+ "loggers": ["voip_utils"],
"quality_scale": "internal",
"requirements": ["voip-utils==0.3.1"]
}
diff --git a/homeassistant/components/voip/select.py b/homeassistant/components/voip/select.py
index f145f866ae3..bfce112d0c5 100644
--- a/homeassistant/components/voip/select.py
+++ b/homeassistant/components/voip/select.py
@@ -10,7 +10,7 @@ from homeassistant.components.assist_pipeline.select import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .devices import VoIPDevice
@@ -23,7 +23,7 @@ if TYPE_CHECKING:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up VoIP switch entities."""
domain_data: DomainData = hass.data[DOMAIN]
diff --git a/homeassistant/components/voip/strings.json b/homeassistant/components/voip/strings.json
index c25c22f3f80..4f37ad1d6f7 100644
--- a/homeassistant/components/voip/strings.json
+++ b/homeassistant/components/voip/strings.json
@@ -53,9 +53,15 @@
"step": {
"init": {
"data": {
- "sip_port": "SIP port"
+ "sip_port": "SIP port",
+ "sip_user": "SIP user"
}
}
}
+ },
+ "exceptions": {
+ "non_tts_announcement": {
+ "message": "VoIP does not currently support non-TTS announcements"
+ }
}
}
diff --git a/homeassistant/components/voip/switch.py b/homeassistant/components/voip/switch.py
index f8484241fc5..7690b8f125c 100644
--- a/homeassistant/components/voip/switch.py
+++ b/homeassistant/components/voip/switch.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import restore_state
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .devices import VoIPDevice
@@ -22,7 +22,7 @@ if TYPE_CHECKING:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up VoIP switch entities."""
domain_data: DomainData = hass.data[DOMAIN]
diff --git a/homeassistant/components/volumio/media_player.py b/homeassistant/components/volumio/media_player.py
index 5ba67d7974f..773a125d483 100644
--- a/homeassistant/components/volumio/media_player.py
+++ b/homeassistant/components/volumio/media_player.py
@@ -21,7 +21,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import Throttle
from .browse_media import browse_node, browse_top_level
@@ -33,7 +33,7 @@ PLAYLIST_UPDATE_INTERVAL = timedelta(seconds=15)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Volumio media player platform."""
@@ -70,7 +70,6 @@ class Volumio(MediaPlayerEntity):
| MediaPlayerEntityFeature.CLEAR_PLAYLIST
| MediaPlayerEntityFeature.BROWSE_MEDIA
)
- _attr_source_list = []
def __init__(self, volumio, uid, name, info):
"""Initialize the media player."""
@@ -78,6 +77,7 @@ class Volumio(MediaPlayerEntity):
unique_id = uid
self._state = {}
self.thumbnail_cache = {}
+ self._attr_source_list = []
self._attr_unique_id = unique_id
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, unique_id)},
diff --git a/homeassistant/components/volvooncall/__init__.py b/homeassistant/components/volvooncall/__init__.py
index 9fc07dd92b0..1a53f9a5dc4 100644
--- a/homeassistant/components/volvooncall/__init__.py
+++ b/homeassistant/components/volvooncall/__init__.py
@@ -52,7 +52,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
volvo_data = VolvoData(hass, connection, entry)
- coordinator = VolvoUpdateCoordinator(hass, volvo_data)
+ coordinator = VolvoUpdateCoordinator(hass, entry, volvo_data)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/volvooncall/binary_sensor.py b/homeassistant/components/volvooncall/binary_sensor.py
index e6104f8d87c..2ba8d19e3db 100644
--- a/homeassistant/components/volvooncall/binary_sensor.py
+++ b/homeassistant/components/volvooncall/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, VOLVO_DISCOVERY_NEW
from .coordinator import VolvoUpdateCoordinator
@@ -24,7 +24,7 @@ from .entity import VolvoEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure binary_sensors from a config entry created in the integrations UI."""
coordinator: VolvoUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/volvooncall/coordinator.py b/homeassistant/components/volvooncall/coordinator.py
index 5ac6a58acb0..2c3e2ba365f 100644
--- a/homeassistant/components/volvooncall/coordinator.py
+++ b/homeassistant/components/volvooncall/coordinator.py
@@ -3,6 +3,7 @@
import asyncio
import logging
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -15,12 +16,17 @@ _LOGGER = logging.getLogger(__name__)
class VolvoUpdateCoordinator(DataUpdateCoordinator[None]):
"""Volvo coordinator."""
- def __init__(self, hass: HomeAssistant, volvo_data: VolvoData) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, volvo_data: VolvoData
+ ) -> None:
"""Initialize the data update coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="volvooncall",
update_interval=DEFAULT_UPDATE_INTERVAL,
)
diff --git a/homeassistant/components/volvooncall/device_tracker.py b/homeassistant/components/volvooncall/device_tracker.py
index 96fe5a644bb..018acb02d49 100644
--- a/homeassistant/components/volvooncall/device_tracker.py
+++ b/homeassistant/components/volvooncall/device_tracker.py
@@ -8,7 +8,7 @@ from homeassistant.components.device_tracker import TrackerEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, VOLVO_DISCOVERY_NEW
from .coordinator import VolvoUpdateCoordinator
@@ -18,7 +18,7 @@ from .entity import VolvoEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure device_trackers from a config entry created in the integrations UI."""
coordinator: VolvoUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/volvooncall/entity.py b/homeassistant/components/volvooncall/entity.py
index 6ebc4bdc754..5a1194e8b1a 100644
--- a/homeassistant/components/volvooncall/entity.py
+++ b/homeassistant/components/volvooncall/entity.py
@@ -57,7 +57,7 @@ class VolvoEntity(CoordinatorEntity[VolvoUpdateCoordinator]):
return f"{self._vehicle_name} {self._entity_name}"
@property
- def assumed_state(self):
+ def assumed_state(self) -> bool:
"""Return true if unable to access real state of entity."""
return True
diff --git a/homeassistant/components/volvooncall/lock.py b/homeassistant/components/volvooncall/lock.py
index cff5df35750..75b54e9dbbc 100644
--- a/homeassistant/components/volvooncall/lock.py
+++ b/homeassistant/components/volvooncall/lock.py
@@ -10,7 +10,7 @@ from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, VOLVO_DISCOVERY_NEW
from .coordinator import VolvoUpdateCoordinator
@@ -20,7 +20,7 @@ from .entity import VolvoEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure locks from a config entry created in the integrations UI."""
coordinator: VolvoUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/volvooncall/sensor.py b/homeassistant/components/volvooncall/sensor.py
index 9916d37197b..feb7248ccaf 100644
--- a/homeassistant/components/volvooncall/sensor.py
+++ b/homeassistant/components/volvooncall/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, VOLVO_DISCOVERY_NEW
from .coordinator import VolvoUpdateCoordinator
@@ -18,7 +18,7 @@ from .entity import VolvoEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure sensors from a config entry created in the integrations UI."""
coordinator: VolvoUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/volvooncall/switch.py b/homeassistant/components/volvooncall/switch.py
index 7e60f47fb44..ff321577348 100644
--- a/homeassistant/components/volvooncall/switch.py
+++ b/homeassistant/components/volvooncall/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, VOLVO_DISCOVERY_NEW
from .coordinator import VolvoUpdateCoordinator
@@ -20,7 +20,7 @@ from .entity import VolvoEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Configure binary_sensors from a config entry created in the integrations UI."""
coordinator: VolvoUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/vulcan/calendar.py b/homeassistant/components/vulcan/calendar.py
index a89b6b4a116..c2ef8b70d46 100644
--- a/homeassistant/components/vulcan/calendar.py
+++ b/homeassistant/components/vulcan/calendar.py
@@ -20,7 +20,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity import generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN
from .fetch_data import get_lessons, get_student_info
@@ -31,7 +31,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the calendar platform for entity."""
client = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wake_on_lan/button.py b/homeassistant/components/wake_on_lan/button.py
index 4d6b19bdd8e..e9cf69b1fe7 100644
--- a/homeassistant/components/wake_on_lan/button.py
+++ b/homeassistant/components/wake_on_lan/button.py
@@ -13,7 +13,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_BROADCAST_ADDRESS, CONF_BROADCAST_PORT, CONF_MAC
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
_LOGGER = logging.getLogger(__name__)
@@ -21,7 +21,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Wake on LAN button entry."""
broadcast_address: str | None = entry.options.get(CONF_BROADCAST_ADDRESS)
diff --git a/homeassistant/components/wallbox/__init__.py b/homeassistant/components/wallbox/__init__.py
index b2f8ac7fd5d..fc8c6e00e84 100644
--- a/homeassistant/components/wallbox/__init__.py
+++ b/homeassistant/components/wallbox/__init__.py
@@ -9,7 +9,7 @@ from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
-from .const import CONF_STATION, DOMAIN, UPDATE_INTERVAL
+from .const import DOMAIN, UPDATE_INTERVAL
from .coordinator import InvalidAuth, WallboxCoordinator, async_validate_input
PLATFORMS = [Platform.LOCK, Platform.NUMBER, Platform.SENSOR, Platform.SWITCH]
@@ -27,11 +27,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except InvalidAuth as ex:
raise ConfigEntryAuthFailed from ex
- wallbox_coordinator = WallboxCoordinator(
- entry.data[CONF_STATION],
- wallbox,
- hass,
- )
+ wallbox_coordinator = WallboxCoordinator(hass, entry, wallbox)
await wallbox_coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = wallbox_coordinator
diff --git a/homeassistant/components/wallbox/coordinator.py b/homeassistant/components/wallbox/coordinator.py
index 99c565d9c0c..4f20f5c406d 100644
--- a/homeassistant/components/wallbox/coordinator.py
+++ b/homeassistant/components/wallbox/coordinator.py
@@ -11,6 +11,7 @@ from typing import Any, Concatenate
import requests
from wallbox import Wallbox
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, HomeAssistantError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -28,6 +29,7 @@ from .const import (
CHARGER_STATUS_DESCRIPTION_KEY,
CHARGER_STATUS_ID_KEY,
CODE_KEY,
+ CONF_STATION,
DOMAIN,
UPDATE_INTERVAL,
ChargerStatus,
@@ -107,14 +109,19 @@ async def async_validate_input(hass: HomeAssistant, wallbox: Wallbox) -> None:
class WallboxCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Wallbox Coordinator class."""
- def __init__(self, station: str, wallbox: Wallbox, hass: HomeAssistant) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, wallbox: Wallbox
+ ) -> None:
"""Initialize."""
- self._station = station
+ self._station = config_entry.data[CONF_STATION]
self._wallbox = wallbox
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=UPDATE_INTERVAL),
)
diff --git a/homeassistant/components/wallbox/lock.py b/homeassistant/components/wallbox/lock.py
index 4853a9104f2..ef35734ed7e 100644
--- a/homeassistant/components/wallbox/lock.py
+++ b/homeassistant/components/wallbox/lock.py
@@ -8,7 +8,7 @@ from homeassistant.components.lock import LockEntity, LockEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CHARGER_DATA_KEY,
@@ -28,7 +28,9 @@ LOCK_TYPES: dict[str, LockEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create wallbox lock entities in HASS."""
coordinator: WallboxCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/wallbox/manifest.json b/homeassistant/components/wallbox/manifest.json
index 63102646508..d217a018303 100644
--- a/homeassistant/components/wallbox/manifest.json
+++ b/homeassistant/components/wallbox/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/wallbox",
"iot_class": "cloud_polling",
"loggers": ["wallbox"],
- "requirements": ["wallbox==0.7.0"]
+ "requirements": ["wallbox==0.8.0"]
}
diff --git a/homeassistant/components/wallbox/number.py b/homeassistant/components/wallbox/number.py
index 24cdd16f99d..a5880f6e0f7 100644
--- a/homeassistant/components/wallbox/number.py
+++ b/homeassistant/components/wallbox/number.py
@@ -13,7 +13,7 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
BIDIRECTIONAL_MODEL_PREFIXES,
@@ -71,9 +71,7 @@ NUMBER_TYPES: dict[str, WallboxNumberEntityDescription] = {
CHARGER_MAX_ICP_CURRENT_KEY: WallboxNumberEntityDescription(
key=CHARGER_MAX_ICP_CURRENT_KEY,
translation_key="maximum_icp_current",
- max_value_fn=lambda coordinator: cast(
- float, coordinator.data[CHARGER_MAX_AVAILABLE_POWER_KEY]
- ),
+ max_value_fn=lambda _: 255,
min_value_fn=lambda _: 6,
set_value_fn=lambda coordinator: coordinator.async_set_icp_current,
native_step=1,
@@ -82,7 +80,9 @@ NUMBER_TYPES: dict[str, WallboxNumberEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create wallbox number entities in HASS."""
coordinator: WallboxCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/wallbox/sensor.py b/homeassistant/components/wallbox/sensor.py
index 18d8afb5612..78b26520bec 100644
--- a/homeassistant/components/wallbox/sensor.py
+++ b/homeassistant/components/wallbox/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import (
@@ -157,7 +157,9 @@ SENSOR_TYPES: dict[str, WallboxSensorEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create wallbox sensor entities in HASS."""
coordinator: WallboxCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/wallbox/switch.py b/homeassistant/components/wallbox/switch.py
index 06c2674579d..30275951ab2 100644
--- a/homeassistant/components/wallbox/switch.py
+++ b/homeassistant/components/wallbox/switch.py
@@ -7,7 +7,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CHARGER_DATA_KEY,
@@ -29,7 +29,9 @@ SWITCH_TYPES: dict[str, SwitchEntityDescription] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Create wallbox sensor entities in HASS."""
coordinator: WallboxCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/waqi/__init__.py b/homeassistant/components/waqi/__init__.py
index e9feca75ee7..9821b5435d9 100644
--- a/homeassistant/components/waqi/__init__.py
+++ b/homeassistant/components/waqi/__init__.py
@@ -21,7 +21,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
client = WAQIClient(session=async_get_clientsession(hass))
client.authenticate(entry.data[CONF_API_KEY])
- waqi_coordinator = WAQIDataUpdateCoordinator(hass, client)
+ waqi_coordinator = WAQIDataUpdateCoordinator(hass, entry, client)
await waqi_coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = waqi_coordinator
diff --git a/homeassistant/components/waqi/coordinator.py b/homeassistant/components/waqi/coordinator.py
index d1a44e9f5b8..86f553a86cd 100644
--- a/homeassistant/components/waqi/coordinator.py
+++ b/homeassistant/components/waqi/coordinator.py
@@ -18,11 +18,14 @@ class WAQIDataUpdateCoordinator(DataUpdateCoordinator[WAQIAirQuality]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, client: WAQIClient) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, client: WAQIClient
+ ) -> None:
"""Initialize the WAQI data coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=5),
)
diff --git a/homeassistant/components/waqi/sensor.py b/homeassistant/components/waqi/sensor.py
index 4c921c68336..59daf60392e 100644
--- a/homeassistant/components/waqi/sensor.py
+++ b/homeassistant/components/waqi/sensor.py
@@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfPressure, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -138,7 +138,9 @@ SENSORS: list[WAQISensorEntityDescription] = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WAQI sensor."""
coordinator: WAQIDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/waqi/strings.json b/homeassistant/components/waqi/strings.json
index a1feb217249..f455e3ead33 100644
--- a/homeassistant/components/waqi/strings.json
+++ b/homeassistant/components/waqi/strings.json
@@ -57,7 +57,7 @@
"name": "[%key:component::sensor::entity_component::pm25::name%]"
},
"neph": {
- "name": "Visbility using nephelometry"
+ "name": "Visibility using nephelometry"
},
"dominant_pollutant": {
"name": "Dominant pollutant",
diff --git a/homeassistant/components/water_heater/__init__.py b/homeassistant/components/water_heater/__init__.py
index c9155950680..f2038def79c 100644
--- a/homeassistant/components/water_heater/__init__.py
+++ b/homeassistant/components/water_heater/__init__.py
@@ -77,6 +77,7 @@ ATTR_OPERATION_MODE = "operation_mode"
ATTR_OPERATION_LIST = "operation_list"
ATTR_TARGET_TEMP_HIGH = "target_temp_high"
ATTR_TARGET_TEMP_LOW = "target_temp_low"
+ATTR_TARGET_TEMP_STEP = "target_temp_step"
ATTR_CURRENT_TEMPERATURE = "current_temperature"
CONVERTIBLE_ATTRIBUTE = [ATTR_TEMPERATURE]
@@ -154,6 +155,7 @@ CACHED_PROPERTIES_WITH_ATTR_ = {
"target_temperature",
"target_temperature_high",
"target_temperature_low",
+ "target_temperature_step",
"is_away_mode_on",
}
@@ -162,7 +164,12 @@ class WaterHeaterEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
"""Base class for water heater entities."""
_entity_component_unrecorded_attributes = frozenset(
- {ATTR_OPERATION_LIST, ATTR_MIN_TEMP, ATTR_MAX_TEMP}
+ {
+ ATTR_OPERATION_LIST,
+ ATTR_MIN_TEMP,
+ ATTR_MAX_TEMP,
+ ATTR_TARGET_TEMP_STEP,
+ }
)
entity_description: WaterHeaterEntityDescription
@@ -179,6 +186,7 @@ class WaterHeaterEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
_attr_target_temperature_low: float | None = None
_attr_target_temperature: float | None = None
_attr_temperature_unit: str
+ _attr_target_temperature_step: float | None = None
@final
@property
@@ -206,6 +214,8 @@ class WaterHeaterEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
self.hass, self.max_temp, self.temperature_unit, self.precision
),
}
+ if target_temperature_step := self.target_temperature_step:
+ data[ATTR_TARGET_TEMP_STEP] = target_temperature_step
if WaterHeaterEntityFeature.OPERATION_MODE in self.supported_features:
data[ATTR_OPERATION_LIST] = self.operation_list
@@ -289,6 +299,11 @@ class WaterHeaterEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
"""Return the lowbound target temperature we try to reach."""
return self._attr_target_temperature_low
+ @cached_property
+ def target_temperature_step(self) -> float | None:
+ """Return the supported step of target temperature."""
+ return self._attr_target_temperature_step
+
@cached_property
def is_away_mode_on(self) -> bool | None:
"""Return true if away mode is on."""
diff --git a/homeassistant/components/watergate/__init__.py b/homeassistant/components/watergate/__init__.py
index fa761110339..c1747af1f11 100644
--- a/homeassistant/components/watergate/__init__.py
+++ b/homeassistant/components/watergate/__init__.py
@@ -16,12 +16,11 @@ from homeassistant.components.webhook import (
async_generate_url,
async_register,
)
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_IP_ADDRESS, CONF_WEBHOOK_ID, Platform
from homeassistant.core import HomeAssistant
from .const import DOMAIN
-from .coordinator import WatergateDataCoordinator
+from .coordinator import WatergateConfigEntry, WatergateDataCoordinator
_LOGGER = logging.getLogger(__name__)
@@ -35,8 +34,6 @@ PLATFORMS: list[Platform] = [
Platform.VALVE,
]
-type WatergateConfigEntry = ConfigEntry[WatergateDataCoordinator]
-
async def async_setup_entry(hass: HomeAssistant, entry: WatergateConfigEntry) -> bool:
"""Set up Watergate from a config entry."""
@@ -52,7 +49,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: WatergateConfigEntry) ->
sonic_address if sonic_address.startswith("http") else f"http://{sonic_address}"
)
- coordinator = WatergateDataCoordinator(hass, watergate_client)
+ coordinator = WatergateDataCoordinator(hass, entry, watergate_client)
entry.runtime_data = coordinator
async_register(
diff --git a/homeassistant/components/watergate/coordinator.py b/homeassistant/components/watergate/coordinator.py
index 1d83b7a3ccb..e3f198c144d 100644
--- a/homeassistant/components/watergate/coordinator.py
+++ b/homeassistant/components/watergate/coordinator.py
@@ -7,6 +7,7 @@ import logging
from watergate_local_api import WatergateApiException, WatergateLocalApiClient
from watergate_local_api.models import DeviceState, NetworkingData, TelemetryData
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -24,14 +25,25 @@ class WatergateAgregatedRequests:
networking: NetworkingData
+type WatergateConfigEntry = ConfigEntry[WatergateDataCoordinator]
+
+
class WatergateDataCoordinator(DataUpdateCoordinator[WatergateAgregatedRequests]):
"""Class to manage fetching watergate data."""
- def __init__(self, hass: HomeAssistant, api: WatergateLocalApiClient) -> None:
+ config_entry: WatergateConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WatergateConfigEntry,
+ api: WatergateLocalApiClient,
+ ) -> None:
"""Initialize."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=2),
)
diff --git a/homeassistant/components/watergate/sensor.py b/homeassistant/components/watergate/sensor.py
index 6782a93541b..5aced8b7488 100644
--- a/homeassistant/components/watergate/sensor.py
+++ b/homeassistant/components/watergate/sensor.py
@@ -22,12 +22,15 @@ from homeassistant.const import (
UnitOfVolume,
UnitOfVolumeFlowRate,
)
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
-from . import WatergateConfigEntry
-from .coordinator import WatergateAgregatedRequests, WatergateDataCoordinator
+from .coordinator import (
+ WatergateAgregatedRequests,
+ WatergateConfigEntry,
+ WatergateDataCoordinator,
+)
from .entity import WatergateEntity
_LOGGER = logging.getLogger(__name__)
@@ -179,7 +182,7 @@ DESCRIPTIONS: list[WatergateSensorEntityDescription] = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: WatergateConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all entries for Watergate Platform."""
diff --git a/homeassistant/components/watergate/valve.py b/homeassistant/components/watergate/valve.py
index 556b53e1d3c..cb6bfa7bd59 100644
--- a/homeassistant/components/watergate/valve.py
+++ b/homeassistant/components/watergate/valve.py
@@ -8,10 +8,9 @@ from homeassistant.components.valve import (
ValveState,
)
from homeassistant.core import callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from . import WatergateConfigEntry
-from .coordinator import WatergateDataCoordinator
+from .coordinator import WatergateConfigEntry, WatergateDataCoordinator
from .entity import WatergateEntity
ENTITY_NAME = "valve"
@@ -21,7 +20,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: WatergateConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all entries for Watergate Platform."""
diff --git a/homeassistant/components/watttime/sensor.py b/homeassistant/components/watttime/sensor.py
index c6cc81580d7..d3aa9d8f895 100644
--- a/homeassistant/components/watttime/sensor.py
+++ b/homeassistant/components/watttime/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
@@ -52,7 +52,9 @@ REALTIME_EMISSIONS_SENSOR_DESCRIPTIONS = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WattTime sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/waze_travel_time/__init__.py b/homeassistant/components/waze_travel_time/__init__.py
index 34f22c9218f..3a91690ef07 100644
--- a/homeassistant/components/waze_travel_time/__init__.py
+++ b/homeassistant/components/waze_travel_time/__init__.py
@@ -17,6 +17,7 @@ from homeassistant.core import (
SupportsResponse,
)
from homeassistant.helpers.httpx_client import get_async_client
+from homeassistant.helpers.location import find_coordinates
from homeassistant.helpers.selector import (
BooleanSelector,
SelectSelector,
@@ -115,10 +116,21 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
client = WazeRouteCalculator(
region=service.data[CONF_REGION].upper(), client=httpx_client
)
+
+ origin_coordinates = find_coordinates(hass, service.data[CONF_ORIGIN])
+ destination_coordinates = find_coordinates(hass, service.data[CONF_DESTINATION])
+
+ origin = origin_coordinates if origin_coordinates else service.data[CONF_ORIGIN]
+ destination = (
+ destination_coordinates
+ if destination_coordinates
+ else service.data[CONF_DESTINATION]
+ )
+
response = await async_get_travel_times(
client=client,
- origin=service.data[CONF_ORIGIN],
- destination=service.data[CONF_DESTINATION],
+ origin=origin,
+ destination=destination,
vehicle_type=service.data[CONF_VEHICLE_TYPE],
avoid_toll_roads=service.data[CONF_AVOID_TOLL_ROADS],
avoid_subscription_roads=service.data[CONF_AVOID_SUBSCRIPTION_ROADS],
diff --git a/homeassistant/components/waze_travel_time/sensor.py b/homeassistant/components/waze_travel_time/sensor.py
index a216a02f61e..1f21cc2ea78 100644
--- a/homeassistant/components/waze_travel_time/sensor.py
+++ b/homeassistant/components/waze_travel_time/sensor.py
@@ -24,7 +24,7 @@ from homeassistant.const import (
)
from homeassistant.core import CoreState, HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.httpx_client import get_async_client
from homeassistant.helpers.location import find_coordinates
@@ -57,7 +57,7 @@ SECONDS_BETWEEN_API_CALLS = 0.5
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a Waze travel time sensor entry."""
destination = config_entry.data[CONF_DESTINATION]
diff --git a/homeassistant/components/weather/strings.json b/homeassistant/components/weather/strings.json
index 85d331f5bd0..31e644b32e3 100644
--- a/homeassistant/components/weather/strings.json
+++ b/homeassistant/components/weather/strings.json
@@ -90,17 +90,17 @@
"services": {
"get_forecasts": {
"name": "Get forecasts",
- "description": "Get weather forecasts.",
+ "description": "Retrieves the forecast from selected weather services.",
"fields": {
"type": {
"name": "Forecast type",
- "description": "Forecast type: daily, hourly or twice daily."
+ "description": "The scope of the weather forecast."
}
}
},
"get_forecast": {
"name": "Get forecast",
- "description": "Get weather forecast.",
+ "description": "Retrieves the forecast from a selected weather service.",
"fields": {
"type": {
"name": "[%key:component::weather::services::get_forecasts::fields::type::name%]",
@@ -111,12 +111,12 @@
},
"issues": {
"deprecated_service_weather_get_forecast": {
- "title": "Detected use of deprecated service weather.get_forecast",
+ "title": "Detected use of deprecated action weather.get_forecast",
"fix_flow": {
"step": {
"confirm": {
"title": "[%key:component::weather::issues::deprecated_service_weather_get_forecast::title%]",
- "description": "Use `weather.get_forecasts` instead which supports multiple entities.\n\nPlease replace this service and adjust your automations and scripts and select **Submit** to close this issue."
+ "description": "Use `weather.get_forecasts` instead which supports multiple entities.\n\nPlease replace this action and adjust your automations and scripts and select **Submit** to close this issue."
}
}
}
diff --git a/homeassistant/components/weatherflow/sensor.py b/homeassistant/components/weatherflow/sensor.py
index cacede55c42..8eee472fe5c 100644
--- a/homeassistant/components/weatherflow/sensor.py
+++ b/homeassistant/components/weatherflow/sensor.py
@@ -40,7 +40,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.unit_system import METRIC_SYSTEM
@@ -267,16 +267,16 @@ SENSORS: tuple[WeatherFlowSensorEntityDescription, ...] = (
WeatherFlowSensorEntityDescription(
key="wind_direction",
translation_key="wind_direction",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
native_unit_of_measurement=DEGREE,
- state_class=SensorStateClass.MEASUREMENT,
event_subscriptions=[EVENT_RAPID_WIND, EVENT_OBSERVATION],
raw_data_conv_fn=lambda raw_data: raw_data.magnitude,
),
WeatherFlowSensorEntityDescription(
key="wind_direction_average",
translation_key="wind_direction_average",
+ device_class=SensorDeviceClass.WIND_DIRECTION,
native_unit_of_measurement=DEGREE,
- state_class=SensorStateClass.MEASUREMENT,
raw_data_conv_fn=lambda raw_data: raw_data.magnitude,
),
)
@@ -285,7 +285,7 @@ SENSORS: tuple[WeatherFlowSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WeatherFlow sensors using config entry."""
diff --git a/homeassistant/components/weatherflow_cloud/__init__.py b/homeassistant/components/weatherflow_cloud/__init__.py
index 8dc26f9b9c6..94c65b7c0a1 100644
--- a/homeassistant/components/weatherflow_cloud/__init__.py
+++ b/homeassistant/components/weatherflow_cloud/__init__.py
@@ -3,7 +3,7 @@
from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_API_TOKEN, Platform
+from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from .const import DOMAIN
@@ -15,10 +15,7 @@ PLATFORMS: list[Platform] = [Platform.SENSOR, Platform.WEATHER]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up WeatherFlowCloud from a config entry."""
- data_coordinator = WeatherFlowCloudDataUpdateCoordinator(
- hass=hass,
- api_token=entry.data[CONF_API_TOKEN],
- )
+ data_coordinator = WeatherFlowCloudDataUpdateCoordinator(hass, entry)
await data_coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = data_coordinator
diff --git a/homeassistant/components/weatherflow_cloud/coordinator.py b/homeassistant/components/weatherflow_cloud/coordinator.py
index 8b8a916262f..b6d2bfd5af2 100644
--- a/homeassistant/components/weatherflow_cloud/coordinator.py
+++ b/homeassistant/components/weatherflow_cloud/coordinator.py
@@ -6,6 +6,8 @@ from aiohttp import ClientResponseError
from weatherflow4py.api import WeatherFlowRestAPI
from weatherflow4py.models.rest.unified import WeatherFlowDataREST
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_API_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -18,12 +20,17 @@ class WeatherFlowCloudDataUpdateCoordinator(
):
"""Class to manage fetching REST Based WeatherFlow Forecast data."""
- def __init__(self, hass: HomeAssistant, api_token: str) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
"""Initialize global WeatherFlow forecast data updater."""
- self.weather_api = WeatherFlowRestAPI(api_token=api_token)
+ self.weather_api = WeatherFlowRestAPI(
+ api_token=config_entry.data[CONF_API_TOKEN]
+ )
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=60),
)
diff --git a/homeassistant/components/weatherflow_cloud/manifest.json b/homeassistant/components/weatherflow_cloud/manifest.json
index 98c98cfbac7..9ffa457a355 100644
--- a/homeassistant/components/weatherflow_cloud/manifest.json
+++ b/homeassistant/components/weatherflow_cloud/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/weatherflow_cloud",
"iot_class": "cloud_polling",
"loggers": ["weatherflow4py"],
- "requirements": ["weatherflow4py==1.0.6"]
+ "requirements": ["weatherflow4py==1.3.1"]
}
diff --git a/homeassistant/components/weatherflow_cloud/sensor.py b/homeassistant/components/weatherflow_cloud/sensor.py
index aeab955878f..d2c62b5f281 100644
--- a/homeassistant/components/weatherflow_cloud/sensor.py
+++ b/homeassistant/components/weatherflow_cloud/sensor.py
@@ -17,7 +17,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfLength, UnitOfPressure, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .const import DOMAIN
@@ -172,7 +172,7 @@ WF_SENSORS: tuple[WeatherFlowCloudSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WeatherFlow sensors based on a config entry."""
diff --git a/homeassistant/components/weatherflow_cloud/weather.py b/homeassistant/components/weatherflow_cloud/weather.py
index c475f2974a9..3cb1f477095 100644
--- a/homeassistant/components/weatherflow_cloud/weather.py
+++ b/homeassistant/components/weatherflow_cloud/weather.py
@@ -17,7 +17,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, STATE_MAP
from .coordinator import WeatherFlowCloudDataUpdateCoordinator
@@ -27,7 +27,7 @@ from .entity import WeatherFlowCloudEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
coordinator: WeatherFlowCloudDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/weatherkit/__init__.py b/homeassistant/components/weatherkit/__init__.py
index 49158182696..4cbac2b32d8 100644
--- a/homeassistant/components/weatherkit/__init__.py
+++ b/homeassistant/components/weatherkit/__init__.py
@@ -32,6 +32,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
hass.data.setdefault(DOMAIN, {})
coordinator = WeatherKitDataUpdateCoordinator(
hass=hass,
+ config_entry=entry,
client=WeatherKitApiClient(
key_id=entry.data[CONF_KEY_ID],
service_id=entry.data[CONF_SERVICE_ID],
diff --git a/homeassistant/components/weatherkit/coordinator.py b/homeassistant/components/weatherkit/coordinator.py
index 6438d7503db..6c7119d6fb0 100644
--- a/homeassistant/components/weatherkit/coordinator.py
+++ b/homeassistant/components/weatherkit/coordinator.py
@@ -33,6 +33,7 @@ class WeatherKitDataUpdateCoordinator(DataUpdateCoordinator):
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
client: WeatherKitApiClient,
) -> None:
"""Initialize."""
@@ -40,6 +41,7 @@ class WeatherKitDataUpdateCoordinator(DataUpdateCoordinator):
super().__init__(
hass=hass,
logger=LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=5),
)
diff --git a/homeassistant/components/weatherkit/sensor.py b/homeassistant/components/weatherkit/sensor.py
index d9c17bb855a..b3639fa5356 100644
--- a/homeassistant/components/weatherkit/sensor.py
+++ b/homeassistant/components/weatherkit/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfVolumetricFlux
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -36,7 +36,7 @@ SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add sensor entities from a config_entry."""
coordinator: WeatherKitDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/weatherkit/weather.py b/homeassistant/components/weatherkit/weather.py
index 98816d520ba..b57e488d06a 100644
--- a/homeassistant/components/weatherkit/weather.py
+++ b/homeassistant/components/weatherkit/weather.py
@@ -29,7 +29,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_CURRENT_WEATHER,
@@ -45,7 +45,7 @@ from .entity import WeatherKitEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Add a weather entity from a config_entry."""
coordinator: WeatherKitDataUpdateCoordinator = hass.data[DOMAIN][
diff --git a/homeassistant/components/webdav/__init__.py b/homeassistant/components/webdav/__init__.py
new file mode 100644
index 00000000000..36a03dce4d7
--- /dev/null
+++ b/homeassistant/components/webdav/__init__.py
@@ -0,0 +1,75 @@
+"""The WebDAV integration."""
+
+from __future__ import annotations
+
+import logging
+
+from aiowebdav2.client import Client
+from aiowebdav2.exceptions import UnauthorizedError
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_PASSWORD, CONF_URL, CONF_USERNAME, CONF_VERIFY_SSL
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryError, ConfigEntryNotReady
+
+from .const import CONF_BACKUP_PATH, DATA_BACKUP_AGENT_LISTENERS, DOMAIN
+from .helpers import (
+ async_create_client,
+ async_ensure_path_exists,
+ async_migrate_wrong_folder_path,
+)
+
+type WebDavConfigEntry = ConfigEntry[Client]
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: WebDavConfigEntry) -> bool:
+ """Set up WebDAV from a config entry."""
+ client = async_create_client(
+ hass=hass,
+ url=entry.data[CONF_URL],
+ username=entry.data[CONF_USERNAME],
+ password=entry.data[CONF_PASSWORD],
+ verify_ssl=entry.data.get(CONF_VERIFY_SSL, True),
+ )
+
+ try:
+ result = await client.check()
+ except UnauthorizedError as err:
+ raise ConfigEntryError(
+ translation_domain=DOMAIN,
+ translation_key="invalid_username_password",
+ ) from err
+
+ # Check if we can connect to the WebDAV server
+ # and access the root directory
+ if not result:
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="cannot_connect",
+ )
+
+ path = entry.data.get(CONF_BACKUP_PATH, "/")
+ await async_migrate_wrong_folder_path(client, path)
+
+ # Ensure the backup directory exists
+ if not await async_ensure_path_exists(client, path):
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="cannot_access_or_create_backup_path",
+ )
+
+ entry.runtime_data = client
+
+ def async_notify_backup_listeners() -> None:
+ for listener in hass.data.get(DATA_BACKUP_AGENT_LISTENERS, []):
+ listener()
+
+ entry.async_on_unload(entry.async_on_state_change(async_notify_backup_listeners))
+ return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: WebDavConfigEntry) -> bool:
+ """Unload a WebDAV config entry."""
+ return True
diff --git a/homeassistant/components/webdav/backup.py b/homeassistant/components/webdav/backup.py
new file mode 100644
index 00000000000..fb2927a58bb
--- /dev/null
+++ b/homeassistant/components/webdav/backup.py
@@ -0,0 +1,248 @@
+"""Support for WebDAV backup."""
+
+from __future__ import annotations
+
+from collections.abc import AsyncIterator, Callable, Coroutine
+from functools import wraps
+import logging
+from time import time
+from typing import Any, Concatenate
+
+from aiohttp import ClientTimeout
+from aiowebdav2.exceptions import UnauthorizedError, WebDavError
+from propcache.api import cached_property
+
+from homeassistant.components.backup import (
+ AgentBackup,
+ BackupAgent,
+ BackupAgentError,
+ BackupNotFound,
+ suggested_filename,
+)
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.json import json_dumps
+from homeassistant.util.json import json_loads_object
+
+from . import WebDavConfigEntry
+from .const import CONF_BACKUP_PATH, DATA_BACKUP_AGENT_LISTENERS, DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+BACKUP_TIMEOUT = ClientTimeout(connect=10, total=43200)
+CACHE_TTL = 300
+
+
+async def async_get_backup_agents(
+ hass: HomeAssistant,
+) -> list[BackupAgent]:
+ """Return a list of backup agents."""
+ entries: list[WebDavConfigEntry] = hass.config_entries.async_loaded_entries(DOMAIN)
+ return [WebDavBackupAgent(hass, entry) for entry in entries]
+
+
+@callback
+def async_register_backup_agents_listener(
+ hass: HomeAssistant,
+ *,
+ listener: Callable[[], None],
+ **kwargs: Any,
+) -> Callable[[], None]:
+ """Register a listener to be called when agents are added or removed.
+
+ :return: A function to unregister the listener.
+ """
+ hass.data.setdefault(DATA_BACKUP_AGENT_LISTENERS, []).append(listener)
+
+ @callback
+ def remove_listener() -> None:
+ """Remove the listener."""
+ hass.data[DATA_BACKUP_AGENT_LISTENERS].remove(listener)
+ if not hass.data[DATA_BACKUP_AGENT_LISTENERS]:
+ del hass.data[DATA_BACKUP_AGENT_LISTENERS]
+
+ return remove_listener
+
+
+def handle_backup_errors[_R, **P](
+ func: Callable[Concatenate[WebDavBackupAgent, P], Coroutine[Any, Any, _R]],
+) -> Callable[Concatenate[WebDavBackupAgent, P], Coroutine[Any, Any, _R]]:
+ """Handle backup errors."""
+
+ @wraps(func)
+ async def wrapper(self: WebDavBackupAgent, *args: P.args, **kwargs: P.kwargs) -> _R:
+ try:
+ return await func(self, *args, **kwargs)
+ except UnauthorizedError as err:
+ raise BackupAgentError("Authentication error") from err
+ except WebDavError as err:
+ _LOGGER.debug("Full error: %s", err, exc_info=True)
+ raise BackupAgentError(
+ f"Backup operation failed: {err}",
+ ) from err
+ except TimeoutError as err:
+ _LOGGER.error(
+ "Error during backup in %s: Timeout",
+ func.__name__,
+ )
+ raise BackupAgentError("Backup operation timed out") from err
+
+ return wrapper
+
+
+def suggested_filenames(backup: AgentBackup) -> tuple[str, str]:
+ """Return the suggested filenames for the backup and metadata."""
+ base_name = suggested_filename(backup).rsplit(".", 1)[0]
+ return f"{base_name}.tar", f"{base_name}.metadata.json"
+
+
+class WebDavBackupAgent(BackupAgent):
+ """Backup agent interface."""
+
+ domain = DOMAIN
+
+ def __init__(self, hass: HomeAssistant, entry: WebDavConfigEntry) -> None:
+ """Initialize the WebDAV backup agent."""
+ super().__init__()
+ self._hass = hass
+ self._entry = entry
+ self._client = entry.runtime_data
+ self.name = entry.title
+ self.unique_id = entry.entry_id
+ self._cache_metadata_files: dict[str, AgentBackup] = {}
+ self._cache_expiration = time()
+
+ @cached_property
+ def _backup_path(self) -> str:
+ """Return the path to the backup."""
+ return self._entry.data.get(CONF_BACKUP_PATH, "")
+
+ @handle_backup_errors
+ async def async_download_backup(
+ self,
+ backup_id: str,
+ **kwargs: Any,
+ ) -> AsyncIterator[bytes]:
+ """Download a backup file.
+
+ :param backup_id: The ID of the backup that was returned in async_list_backups.
+ :return: An async iterator that yields bytes.
+ """
+ backup = await self._find_backup_by_id(backup_id)
+
+ return await self._client.download_iter(
+ f"{self._backup_path}/{suggested_filename(backup)}",
+ timeout=BACKUP_TIMEOUT,
+ )
+
+ @handle_backup_errors
+ async def async_upload_backup(
+ self,
+ *,
+ open_stream: Callable[[], Coroutine[Any, Any, AsyncIterator[bytes]]],
+ backup: AgentBackup,
+ **kwargs: Any,
+ ) -> None:
+ """Upload a backup.
+
+ :param open_stream: A function returning an async iterator that yields bytes.
+ :param backup: Metadata about the backup that should be uploaded.
+ """
+ (filename_tar, filename_meta) = suggested_filenames(backup)
+
+ await self._client.upload_iter(
+ await open_stream(),
+ f"{self._backup_path}/{filename_tar}",
+ timeout=BACKUP_TIMEOUT,
+ content_length=backup.size,
+ )
+
+ _LOGGER.debug(
+ "Uploaded backup to %s",
+ f"{self._backup_path}/{filename_tar}",
+ )
+
+ await self._client.upload_iter(
+ json_dumps(backup.as_dict()),
+ f"{self._backup_path}/{filename_meta}",
+ )
+
+ _LOGGER.debug(
+ "Uploaded metadata file for %s",
+ f"{self._backup_path}/{filename_meta}",
+ )
+
+ # reset cache
+ self._cache_expiration = time()
+
+ @handle_backup_errors
+ async def async_delete_backup(
+ self,
+ backup_id: str,
+ **kwargs: Any,
+ ) -> None:
+ """Delete a backup file.
+
+ :param backup_id: The ID of the backup that was returned in async_list_backups.
+ """
+ backup = await self._find_backup_by_id(backup_id)
+
+ (filename_tar, filename_meta) = suggested_filenames(backup)
+ backup_path = f"{self._backup_path}/{filename_tar}"
+
+ await self._client.clean(backup_path)
+ await self._client.clean(f"{self._backup_path}/{filename_meta}")
+
+ _LOGGER.debug(
+ "Deleted backup at %s",
+ backup_path,
+ )
+
+ # reset cache
+ self._cache_expiration = time()
+
+ @handle_backup_errors
+ async def async_list_backups(self, **kwargs: Any) -> list[AgentBackup]:
+ """List backups."""
+ return list((await self._list_cached_metadata_files()).values())
+
+ @handle_backup_errors
+ async def async_get_backup(
+ self,
+ backup_id: str,
+ **kwargs: Any,
+ ) -> AgentBackup:
+ """Return a backup."""
+ return await self._find_backup_by_id(backup_id)
+
+ async def _list_cached_metadata_files(self) -> dict[str, AgentBackup]:
+ """List metadata files with a cache."""
+ if time() <= self._cache_expiration:
+ return self._cache_metadata_files
+
+ async def _download_metadata(path: str) -> AgentBackup:
+ """Download metadata file."""
+ iterator = await self._client.download_iter(path)
+ metadata = await anext(iterator)
+ return AgentBackup.from_dict(json_loads_object(metadata))
+
+ async def _list_metadata_files() -> dict[str, AgentBackup]:
+ """List metadata files."""
+ files = await self._client.list_files(self._backup_path)
+ return {
+ metadata_content.backup_id: metadata_content
+ for file_name in files
+ if file_name.endswith(".json")
+ if (metadata_content := await _download_metadata(file_name))
+ }
+
+ self._cache_metadata_files = await _list_metadata_files()
+ self._cache_expiration = time() + CACHE_TTL
+ return self._cache_metadata_files
+
+ async def _find_backup_by_id(self, backup_id: str) -> AgentBackup:
+ """Find a backup by its backup ID on remote."""
+ metadata_files = await self._list_cached_metadata_files()
+ if metadata_file := metadata_files.get(backup_id):
+ return metadata_file
+
+ raise BackupNotFound(f"Backup {backup_id} not found")
diff --git a/homeassistant/components/webdav/config_flow.py b/homeassistant/components/webdav/config_flow.py
new file mode 100644
index 00000000000..fa1a4fe3ca9
--- /dev/null
+++ b/homeassistant/components/webdav/config_flow.py
@@ -0,0 +1,92 @@
+"""Config flow for the WebDAV integration."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from aiowebdav2.exceptions import MethodNotSupportedError, UnauthorizedError
+import voluptuous as vol
+import yarl
+
+from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
+from homeassistant.const import CONF_PASSWORD, CONF_URL, CONF_USERNAME, CONF_VERIFY_SSL
+from homeassistant.helpers.selector import (
+ TextSelector,
+ TextSelectorConfig,
+ TextSelectorType,
+)
+
+from .const import CONF_BACKUP_PATH, DOMAIN
+from .helpers import async_create_client
+
+_LOGGER = logging.getLogger(__name__)
+
+STEP_USER_DATA_SCHEMA = vol.Schema(
+ {
+ vol.Required(CONF_URL): TextSelector(
+ TextSelectorConfig(
+ type=TextSelectorType.URL,
+ )
+ ),
+ vol.Required(CONF_USERNAME): str,
+ vol.Required(CONF_PASSWORD): TextSelector(
+ TextSelectorConfig(
+ type=TextSelectorType.PASSWORD,
+ )
+ ),
+ vol.Optional(CONF_BACKUP_PATH, default="/"): str,
+ vol.Optional(CONF_VERIFY_SSL, default=True): bool,
+ }
+)
+
+
+class WebDavConfigFlow(ConfigFlow, domain=DOMAIN):
+ """Handle a config flow for WebDAV."""
+
+ async def async_step_user(
+ self, user_input: dict[str, Any] | None = None
+ ) -> ConfigFlowResult:
+ """Handle the initial step."""
+ errors: dict[str, str] = {}
+ if user_input is not None:
+ client = async_create_client(
+ hass=self.hass,
+ url=user_input[CONF_URL],
+ username=user_input[CONF_USERNAME],
+ password=user_input[CONF_PASSWORD],
+ verify_ssl=user_input.get(CONF_VERIFY_SSL, True),
+ )
+
+ # Check if we can connect to the WebDAV server
+ # .check() already does the most of the error handling and will return True
+ # if we can access the root directory
+ try:
+ result = await client.check()
+ except UnauthorizedError:
+ errors["base"] = "invalid_auth"
+ except MethodNotSupportedError:
+ errors["base"] = "invalid_method"
+ except Exception: # pylint: disable=broad-except
+ _LOGGER.exception("Unexpected error")
+ errors["base"] = "unknown"
+ else:
+ if result:
+ self._async_abort_entries_match(
+ {
+ CONF_URL: user_input[CONF_URL],
+ CONF_USERNAME: user_input[CONF_USERNAME],
+ }
+ )
+
+ parsed_url = yarl.URL(user_input[CONF_URL])
+ return self.async_create_entry(
+ title=f"{user_input[CONF_USERNAME]}@{parsed_url.host}",
+ data=user_input,
+ )
+
+ errors["base"] = "cannot_connect"
+
+ return self.async_show_form(
+ step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors
+ )
diff --git a/homeassistant/components/webdav/const.py b/homeassistant/components/webdav/const.py
new file mode 100644
index 00000000000..faf8ce77ca5
--- /dev/null
+++ b/homeassistant/components/webdav/const.py
@@ -0,0 +1,13 @@
+"""Constants for the WebDAV integration."""
+
+from collections.abc import Callable
+
+from homeassistant.util.hass_dict import HassKey
+
+DOMAIN = "webdav"
+
+DATA_BACKUP_AGENT_LISTENERS: HassKey[list[Callable[[], None]]] = HassKey(
+ f"{DOMAIN}.backup_agent_listeners"
+)
+
+CONF_BACKUP_PATH = "backup_path"
diff --git a/homeassistant/components/webdav/helpers.py b/homeassistant/components/webdav/helpers.py
new file mode 100644
index 00000000000..442f69b4d3c
--- /dev/null
+++ b/homeassistant/components/webdav/helpers.py
@@ -0,0 +1,68 @@
+"""Helper functions for the WebDAV component."""
+
+import logging
+
+from aiowebdav2.client import Client, ClientOptions
+from aiowebdav2.exceptions import WebDavError
+
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import ConfigEntryNotReady
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+from .const import DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+
+@callback
+def async_create_client(
+ *,
+ hass: HomeAssistant,
+ url: str,
+ username: str,
+ password: str,
+ verify_ssl: bool = False,
+) -> Client:
+ """Create a WebDAV client."""
+ return Client(
+ url=url,
+ username=username,
+ password=password,
+ options=ClientOptions(
+ verify_ssl=verify_ssl,
+ session=async_get_clientsession(hass),
+ ),
+ )
+
+
+async def async_ensure_path_exists(client: Client, path: str) -> bool:
+ """Ensure that a path exists recursively on the WebDAV server."""
+ parts = path.strip("/").split("/")
+ for i in range(1, len(parts) + 1):
+ sub_path = "/".join(parts[:i])
+ if not await client.check(sub_path) and not await client.mkdir(sub_path):
+ return False
+
+ return True
+
+
+async def async_migrate_wrong_folder_path(client: Client, path: str) -> None:
+ """Migrate the wrong encoded folder path to the correct one."""
+ wrong_path = path.replace(" ", "%20")
+ # migrate folder when the old folder exists
+ if wrong_path != path and await client.check(wrong_path):
+ try:
+ await client.move(wrong_path, path)
+ except WebDavError as err:
+ raise ConfigEntryNotReady(
+ translation_domain=DOMAIN,
+ translation_key="failed_to_migrate_folder",
+ translation_placeholders={
+ "wrong_path": wrong_path,
+ "correct_path": path,
+ },
+ ) from err
+
+ _LOGGER.debug(
+ "Migrated wrong encoded folder path from %s to %s", wrong_path, path
+ )
diff --git a/homeassistant/components/webdav/manifest.json b/homeassistant/components/webdav/manifest.json
new file mode 100644
index 00000000000..30028cb28c9
--- /dev/null
+++ b/homeassistant/components/webdav/manifest.json
@@ -0,0 +1,12 @@
+{
+ "domain": "webdav",
+ "name": "WebDAV",
+ "codeowners": ["@jpbede"],
+ "config_flow": true,
+ "documentation": "https://www.home-assistant.io/integrations/webdav",
+ "integration_type": "service",
+ "iot_class": "cloud_polling",
+ "loggers": ["aiowebdav2"],
+ "quality_scale": "bronze",
+ "requirements": ["aiowebdav2==0.4.2"]
+}
diff --git a/homeassistant/components/webdav/quality_scale.yaml b/homeassistant/components/webdav/quality_scale.yaml
new file mode 100644
index 00000000000..560626fda7e
--- /dev/null
+++ b/homeassistant/components/webdav/quality_scale.yaml
@@ -0,0 +1,145 @@
+rules:
+ # Bronze
+ action-setup:
+ status: exempt
+ comment: Integration does not register custom actions.
+ appropriate-polling:
+ status: exempt
+ comment: |
+ This integration does not poll.
+ brands: done
+ common-modules: done
+ config-flow-test-coverage: done
+ config-flow: done
+ dependency-transparency: done
+ docs-actions:
+ status: exempt
+ comment: |
+ This integration does not have any custom actions.
+ docs-high-level-description: done
+ docs-installation-instructions: done
+ docs-removal-instructions: done
+ entity-event-setup:
+ status: exempt
+ comment: |
+ Entities of this integration does not explicitly subscribe to events.
+ entity-unique-id:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ has-entity-name:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ runtime-data: done
+ test-before-configure: done
+ test-before-setup: done
+ unique-config-entry: done
+
+ # Silver
+ action-exceptions:
+ status: exempt
+ comment: Integration does not register custom actions.
+ config-entry-unloading: done
+ docs-configuration-parameters:
+ status: exempt
+ comment: |
+ No Options flow.
+ docs-installation-parameters: done
+ entity-unavailable:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ integration-owner: done
+ log-when-unavailable:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ parallel-updates:
+ status: exempt
+ comment: |
+ This integration does not have platforms.
+ reauthentication-flow: todo
+ test-coverage: todo
+
+ # Gold
+ devices:
+ status: exempt
+ comment: |
+ This integration connects to a single service.
+ diagnostics:
+ status: exempt
+ comment: |
+ There is no data to diagnose.
+ discovery-update-info:
+ status: exempt
+ comment: |
+ This integration is a cloud service and does not support discovery.
+ discovery:
+ status: exempt
+ comment: |
+ This integration is a cloud service and does not support discovery.
+ docs-data-update:
+ status: exempt
+ comment: |
+ This integration does not poll or push.
+ docs-examples:
+ status: exempt
+ comment: |
+ This integration only serves backup.
+ docs-known-limitations:
+ status: done
+ comment: |
+ No known limitations.
+ docs-supported-devices:
+ status: exempt
+ comment: |
+ This integration is a cloud service.
+ docs-supported-functions:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ docs-troubleshooting:
+ status: exempt
+ comment: |
+ No issues known to troubleshoot.
+ docs-use-cases: todo
+ dynamic-devices:
+ status: exempt
+ comment: |
+ This integration connects to a single service.
+ entity-category:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ entity-device-class:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ entity-disabled-by-default:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ entity-translations:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ exception-translations: done
+ icon-translations:
+ status: exempt
+ comment: |
+ This integration does not have entities.
+ reconfiguration-flow:
+ status: exempt
+ comment: |
+ Nothing to reconfigure.
+ repair-issues: todo
+ stale-devices:
+ status: exempt
+ comment: |
+ This integration connects to a single service.
+
+ # Platinum
+ async-dependency: done
+ inject-websession: todo
+ strict-typing: todo
diff --git a/homeassistant/components/webdav/strings.json b/homeassistant/components/webdav/strings.json
new file mode 100644
index 00000000000..ac6418f1239
--- /dev/null
+++ b/homeassistant/components/webdav/strings.json
@@ -0,0 +1,45 @@
+{
+ "config": {
+ "step": {
+ "user": {
+ "data": {
+ "url": "[%key:common::config_flow::data::url%]",
+ "username": "[%key:common::config_flow::data::username%]",
+ "password": "[%key:common::config_flow::data::password%]",
+ "backup_path": "Backup path",
+ "verify_ssl": "[%key:common::config_flow::data::verify_ssl%]"
+ },
+ "data_description": {
+ "url": "The URL of the WebDAV server. Check with your provider for the correct URL.",
+ "username": "The username for the WebDAV server.",
+ "password": "The password for the WebDAV server.",
+ "backup_path": "Define the path where the backups should be located (will be created automatically if it does not exist).",
+ "verify_ssl": "Whether to verify the SSL certificate of the server. If you are using a self-signed certificate, do not select this option."
+ }
+ }
+ },
+ "error": {
+ "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
+ "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
+ "invalid_method": "The server does not support the required methods. Please check whether you have the correct URL. Check with your provider for the correct URL.",
+ "unknown": "[%key:common::config_flow::error::unknown%]"
+ },
+ "abort": {
+ "already_configured": "[%key:common::config_flow::abort::already_configured_service%]"
+ }
+ },
+ "exceptions": {
+ "invalid_username_password": {
+ "message": "Invalid username or password"
+ },
+ "cannot_connect": {
+ "message": "Cannot connect to WebDAV server"
+ },
+ "cannot_access_or_create_backup_path": {
+ "message": "Cannot access or create backup path. Please check the path and permissions."
+ },
+ "failed_to_migrate_folder": {
+ "message": "Failed to migrate wrong encoded folder \"{wrong_path}\" to \"{correct_path}\"."
+ }
+ }
+}
diff --git a/homeassistant/components/webmin/sensor.py b/homeassistant/components/webmin/sensor.py
index 785140393a2..a21c73bed13 100644
--- a/homeassistant/components/webmin/sensor.py
+++ b/homeassistant/components/webmin/sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.sensor import (
)
from homeassistant.const import PERCENTAGE, UnitOfInformation
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import WebminConfigEntry
@@ -200,7 +200,7 @@ def generate_filesystem_sensor_description(
async def async_setup_entry(
hass: HomeAssistant,
entry: WebminConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Webmin sensors based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/webmin/strings.json b/homeassistant/components/webmin/strings.json
index 9a6d6d4fbe4..b92986f917a 100644
--- a/homeassistant/components/webmin/strings.json
+++ b/homeassistant/components/webmin/strings.json
@@ -29,13 +29,13 @@
"entity": {
"sensor": {
"load_1m": {
- "name": "Load (1m)"
+ "name": "Load (1 min)"
},
"load_5m": {
- "name": "Load (5m)"
+ "name": "Load (5 min)"
},
"load_15m": {
- "name": "Load (15m)"
+ "name": "Load (15 min)"
},
"mem_total": {
"name": "Memory total"
diff --git a/homeassistant/components/webostv/config_flow.py b/homeassistant/components/webostv/config_flow.py
index fbc3eb958dd..80c8fb7f8f2 100644
--- a/homeassistant/components/webostv/config_flow.py
+++ b/homeassistant/components/webostv/config_flow.py
@@ -92,13 +92,13 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
errors["base"] = "cannot_connect"
else:
await self.async_set_unique_id(
- client.hello_info["deviceUUID"], raise_on_progress=False
+ client.tv_info.hello["deviceUUID"], raise_on_progress=False
)
self._abort_if_unique_id_configured({CONF_HOST: self._host})
data = {CONF_HOST: self._host, CONF_CLIENT_SECRET: client.client_key}
if not self._name:
- self._name = f"{DEFAULT_NAME} {client.system_info['modelName']}"
+ self._name = f"{DEFAULT_NAME} {client.tv_info.system['modelName']}"
return self.async_create_entry(title=self._name, data=data)
return self.async_show_form(step_id="pairing", errors=errors)
@@ -176,7 +176,7 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
except WEBOSTV_EXCEPTIONS:
errors["base"] = "cannot_connect"
else:
- await self.async_set_unique_id(client.hello_info["deviceUUID"])
+ await self.async_set_unique_id(client.tv_info.hello["deviceUUID"])
self._abort_if_unique_id_mismatch(reason="wrong_device")
data = {CONF_HOST: host, CONF_CLIENT_SECRET: client.client_key}
return self.async_update_reload_and_abort(reconfigure_entry, data=data)
@@ -214,7 +214,7 @@ class OptionsFlowHandler(OptionsFlow):
sources_list = []
try:
client = await async_control_connect(self.hass, self.host, self.key)
- sources_list = get_sources(client)
+ sources_list = get_sources(client.tv_state)
except WebOsTvPairError:
errors["base"] = "error_pairing"
except WEBOSTV_EXCEPTIONS:
diff --git a/homeassistant/components/webostv/const.py b/homeassistant/components/webostv/const.py
index e505611db52..118ea7b32db 100644
--- a/homeassistant/components/webostv/const.py
+++ b/homeassistant/components/webostv/const.py
@@ -31,6 +31,7 @@ WEBOSTV_EXCEPTIONS = (
WebOsTvCommandError,
aiohttp.ClientConnectorError,
aiohttp.ServerDisconnectedError,
+ aiohttp.WSMessageTypeError,
asyncio.CancelledError,
asyncio.TimeoutError,
)
diff --git a/homeassistant/components/webostv/diagnostics.py b/homeassistant/components/webostv/diagnostics.py
index 7fb64a2cb8f..e4ea38064a8 100644
--- a/homeassistant/components/webostv/diagnostics.py
+++ b/homeassistant/components/webostv/diagnostics.py
@@ -32,15 +32,8 @@ async def async_get_config_entry_diagnostics(
client_data = {
"is_registered": client.is_registered(),
"is_connected": client.is_connected(),
- "current_app_id": client.current_app_id,
- "current_channel": client.current_channel,
- "apps": client.apps,
- "inputs": client.inputs,
- "system_info": client.system_info,
- "software_info": client.software_info,
- "hello_info": client.hello_info,
- "sound_output": client.sound_output,
- "is_on": client.is_on,
+ "tv_info": client.tv_info.__dict__,
+ "tv_state": client.tv_state.__dict__,
}
return async_redact_data(
diff --git a/homeassistant/components/webostv/helpers.py b/homeassistant/components/webostv/helpers.py
index 3c509a56d1e..f70f250f91d 100644
--- a/homeassistant/components/webostv/helpers.py
+++ b/homeassistant/components/webostv/helpers.py
@@ -4,7 +4,7 @@ from __future__ import annotations
import logging
-from aiowebostv import WebOsClient
+from aiowebostv import WebOsClient, WebOsTvState
from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import CONF_CLIENT_SECRET, CONF_HOST
@@ -83,16 +83,16 @@ def async_get_client_by_device_entry(
)
-def get_sources(client: WebOsClient) -> list[str]:
+def get_sources(tv_state: WebOsTvState) -> list[str]:
"""Construct sources list."""
sources = []
found_live_tv = False
- for app in client.apps.values():
+ for app in tv_state.apps.values():
sources.append(app["title"])
if app["id"] == LIVE_TV_APP_ID:
found_live_tv = True
- for source in client.inputs.values():
+ for source in tv_state.inputs.values():
sources.append(source["label"])
if source["appId"] == LIVE_TV_APP_ID:
found_live_tv = True
diff --git a/homeassistant/components/webostv/manifest.json b/homeassistant/components/webostv/manifest.json
index 174e8025dd0..8ac470ae922 100644
--- a/homeassistant/components/webostv/manifest.json
+++ b/homeassistant/components/webostv/manifest.json
@@ -6,7 +6,7 @@
"documentation": "https://www.home-assistant.io/integrations/webostv",
"iot_class": "local_push",
"loggers": ["aiowebostv"],
- "requirements": ["aiowebostv==0.6.1"],
+ "requirements": ["aiowebostv==0.7.3"],
"ssdp": [
{
"st": "urn:lge-com:service:webos-second-screen:1"
diff --git a/homeassistant/components/webostv/media_player.py b/homeassistant/components/webostv/media_player.py
index 5c47a5e775f..780e9f418a5 100644
--- a/homeassistant/components/webostv/media_player.py
+++ b/homeassistant/components/webostv/media_player.py
@@ -11,7 +11,7 @@ from http import HTTPStatus
import logging
from typing import Any, Concatenate, cast
-from aiowebostv import WebOsClient, WebOsTvPairError
+from aiowebostv import WebOsTvPairError, WebOsTvState
import voluptuous as vol
from homeassistant import util
@@ -28,7 +28,7 @@ from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.trigger import PluggableAction
from homeassistant.helpers.typing import VolDictType
@@ -102,7 +102,7 @@ SERVICES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: WebOsTvConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the LG webOS TV platform."""
platform = entity_platform.async_get_current_platform()
@@ -205,51 +205,52 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
"""Call disconnect on removal."""
self._client.unregister_state_update_callback(self.async_handle_state_update)
- async def async_handle_state_update(self, _client: WebOsClient) -> None:
+ async def async_handle_state_update(self, tv_state: WebOsTvState) -> None:
"""Update state from WebOsClient."""
self._update_states()
self.async_write_ha_state()
def _update_states(self) -> None:
"""Update entity state attributes."""
+ tv_state = self._client.tv_state
self._update_sources()
self._attr_state = (
- MediaPlayerState.ON if self._client.is_on else MediaPlayerState.OFF
+ MediaPlayerState.ON if tv_state.is_on else MediaPlayerState.OFF
)
- self._attr_is_volume_muted = cast(bool, self._client.muted)
+ self._attr_is_volume_muted = cast(bool, tv_state.muted)
self._attr_volume_level = None
- if self._client.volume is not None:
- self._attr_volume_level = self._client.volume / 100.0
+ if tv_state.volume is not None:
+ self._attr_volume_level = tv_state.volume / 100.0
self._attr_source = self._current_source
self._attr_source_list = sorted(self._source_list)
self._attr_media_content_type = None
- if self._client.current_app_id == LIVE_TV_APP_ID:
+ if tv_state.current_app_id == LIVE_TV_APP_ID:
self._attr_media_content_type = MediaType.CHANNEL
self._attr_media_title = None
- if (self._client.current_app_id == LIVE_TV_APP_ID) and (
- self._client.current_channel is not None
+ if (tv_state.current_app_id == LIVE_TV_APP_ID) and (
+ tv_state.current_channel is not None
):
self._attr_media_title = cast(
- str, self._client.current_channel.get("channelName")
+ str, tv_state.current_channel.get("channelName")
)
self._attr_media_image_url = None
- if self._client.current_app_id in self._client.apps:
- icon: str = self._client.apps[self._client.current_app_id]["largeIcon"]
+ if tv_state.current_app_id in tv_state.apps:
+ icon: str = tv_state.apps[tv_state.current_app_id]["largeIcon"]
if not icon.startswith("http"):
- icon = self._client.apps[self._client.current_app_id]["icon"]
+ icon = tv_state.apps[tv_state.current_app_id]["icon"]
self._attr_media_image_url = icon
if self.state != MediaPlayerState.OFF or not self._supported_features:
supported = SUPPORT_WEBOSTV
- if self._client.sound_output == "external_speaker":
+ if tv_state.sound_output == "external_speaker":
supported = supported | SUPPORT_WEBOSTV_VOLUME
- elif self._client.sound_output != "lineout":
+ elif tv_state.sound_output != "lineout":
supported = (
supported
| SUPPORT_WEBOSTV_VOLUME
@@ -265,9 +266,9 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
)
self._attr_assumed_state = True
- if self._client.is_on and self._client.media_state:
+ if tv_state.is_on and tv_state.media_state:
self._attr_assumed_state = False
- for entry in self._client.media_state:
+ for entry in tv_state.media_state:
if entry.get("playState") == "playing":
self._attr_state = MediaPlayerState.PLAYING
elif entry.get("playState") == "paused":
@@ -275,35 +276,37 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
elif entry.get("playState") == "unloaded":
self._attr_state = MediaPlayerState.IDLE
+ tv_info = self._client.tv_info
if self.state != MediaPlayerState.OFF:
- maj_v = self._client.software_info.get("major_ver")
- min_v = self._client.software_info.get("minor_ver")
+ maj_v = tv_info.software.get("major_ver")
+ min_v = tv_info.software.get("minor_ver")
if maj_v and min_v:
self._attr_device_info["sw_version"] = f"{maj_v}.{min_v}"
- if model := self._client.system_info.get("modelName"):
+ if model := tv_info.system.get("modelName"):
self._attr_device_info["model"] = model
- if serial_number := self._client.system_info.get("serialNumber"):
+ if serial_number := tv_info.system.get("serialNumber"):
self._attr_device_info["serial_number"] = serial_number
self._attr_extra_state_attributes = {}
- if self._client.sound_output is not None or self.state != MediaPlayerState.OFF:
+ if tv_state.sound_output is not None or self.state != MediaPlayerState.OFF:
self._attr_extra_state_attributes = {
- ATTR_SOUND_OUTPUT: self._client.sound_output
+ ATTR_SOUND_OUTPUT: tv_state.sound_output
}
def _update_sources(self) -> None:
"""Update list of sources from current source, apps, inputs and configured list."""
+ tv_state = self._client.tv_state
source_list = self._source_list
self._source_list = {}
conf_sources = self._sources
found_live_tv = False
- for app in self._client.apps.values():
+ for app in tv_state.apps.values():
if app["id"] == LIVE_TV_APP_ID:
found_live_tv = True
- if app["id"] == self._client.current_app_id:
+ if app["id"] == tv_state.current_app_id:
self._current_source = app["title"]
self._source_list[app["title"]] = app
elif (
@@ -314,10 +317,10 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
):
self._source_list[app["title"]] = app
- for source in self._client.inputs.values():
+ for source in tv_state.inputs.values():
if source["appId"] == LIVE_TV_APP_ID:
found_live_tv = True
- if source["appId"] == self._client.current_app_id:
+ if source["appId"] == tv_state.current_app_id:
self._current_source = source["label"]
self._source_list[source["label"]] = source
elif (
@@ -334,7 +337,7 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
# not appear in the app or input lists in some cases
elif not found_live_tv:
app = {"id": LIVE_TV_APP_ID, "title": "Live TV"}
- if self._client.current_app_id == LIVE_TV_APP_ID:
+ if tv_state.current_app_id == LIVE_TV_APP_ID:
self._current_source = app["title"]
self._source_list["Live TV"] = app
elif (
@@ -434,12 +437,12 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
"""Play a piece of media."""
_LOGGER.debug("Call play media type <%s>, Id <%s>", media_type, media_id)
- if media_type == MediaType.CHANNEL and self._client.channels:
+ if media_type == MediaType.CHANNEL and self._client.tv_state.channels:
_LOGGER.debug("Searching channel")
partial_match_channel_id = None
perfect_match_channel_id = None
- for channel in self._client.channels:
+ for channel in self._client.tv_state.channels:
if media_id == channel["channelNumber"]:
perfect_match_channel_id = channel["channelId"]
continue
@@ -484,7 +487,7 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
@cmd
async def async_media_next_track(self) -> None:
"""Send next track command."""
- if self._client.current_app_id == LIVE_TV_APP_ID:
+ if self._client.tv_state.current_app_id == LIVE_TV_APP_ID:
await self._client.channel_up()
else:
await self._client.fast_forward()
@@ -492,7 +495,7 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
@cmd
async def async_media_previous_track(self) -> None:
"""Send the previous track command."""
- if self._client.current_app_id == LIVE_TV_APP_ID:
+ if self._client.tv_state.current_app_id == LIVE_TV_APP_ID:
await self._client.channel_down()
else:
await self._client.rewind()
diff --git a/homeassistant/components/webostv/notify.py b/homeassistant/components/webostv/notify.py
index 2393cb4cd07..3966cea5e92 100644
--- a/homeassistant/components/webostv/notify.py
+++ b/homeassistant/components/webostv/notify.py
@@ -49,7 +49,7 @@ class LgWebOSNotificationService(BaseNotificationService):
data = kwargs[ATTR_DATA]
icon_path = data.get(ATTR_ICON) if data else None
- if not client.is_on:
+ if not client.tv_state.is_on:
raise HomeAssistantError(
translation_domain=DOMAIN,
translation_key="notify_device_off",
diff --git a/homeassistant/components/weheat/__init__.py b/homeassistant/components/weheat/__init__.py
index d8d8616c867..15935f3e418 100644
--- a/homeassistant/components/weheat/__init__.py
+++ b/homeassistant/components/weheat/__init__.py
@@ -2,13 +2,13 @@
from __future__ import annotations
+import asyncio
from http import HTTPStatus
import aiohttp
from weheat.abstractions.discovery import HeatPumpDiscovery
from weheat.exceptions import UnauthorizedException
-from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
@@ -19,12 +19,16 @@ from homeassistant.helpers.config_entry_oauth2_flow import (
)
from .const import API_URL, LOGGER
-from .coordinator import WeheatDataUpdateCoordinator
+from .coordinator import (
+ HeatPumpInfo,
+ WeheatConfigEntry,
+ WeheatData,
+ WeheatDataUpdateCoordinator,
+ WeheatEnergyUpdateCoordinator,
+)
PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SENSOR]
-type WeheatConfigEntry = ConfigEntry[list[WeheatDataUpdateCoordinator]]
-
async def async_setup_entry(hass: HomeAssistant, entry: WeheatConfigEntry) -> bool:
"""Set up Weheat from a config entry."""
@@ -55,14 +59,38 @@ async def async_setup_entry(hass: HomeAssistant, entry: WeheatConfigEntry) -> bo
except UnauthorizedException as error:
raise ConfigEntryAuthFailed from error
+ nr_of_pumps = len(discovered_heat_pumps)
+
for pump_info in discovered_heat_pumps:
LOGGER.debug("Adding %s", pump_info)
- # for each pump, add a coordinator
- new_coordinator = WeheatDataUpdateCoordinator(hass, session, pump_info)
+ # for each pump, add the coordinators
- await new_coordinator.async_config_entry_first_refresh()
+ new_heat_pump = HeatPumpInfo(pump_info)
+ new_data_coordinator = WeheatDataUpdateCoordinator(
+ hass, entry, session, pump_info, nr_of_pumps
+ )
+ new_energy_coordinator = WeheatEnergyUpdateCoordinator(
+ hass, entry, session, pump_info
+ )
- entry.runtime_data.append(new_coordinator)
+ entry.runtime_data.append(
+ WeheatData(
+ heat_pump_info=new_heat_pump,
+ data_coordinator=new_data_coordinator,
+ energy_coordinator=new_energy_coordinator,
+ )
+ )
+
+ await asyncio.gather(
+ *[
+ data.data_coordinator.async_config_entry_first_refresh()
+ for data in entry.runtime_data
+ ],
+ *[
+ data.energy_coordinator.async_config_entry_first_refresh()
+ for data in entry.runtime_data
+ ],
+ )
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/weheat/binary_sensor.py b/homeassistant/components/weheat/binary_sensor.py
index 1fb8f614a40..5e4c91fde60 100644
--- a/homeassistant/components/weheat/binary_sensor.py
+++ b/homeassistant/components/weheat/binary_sensor.py
@@ -11,11 +11,10 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import WeheatConfigEntry
-from .coordinator import WeheatDataUpdateCoordinator
+from .coordinator import HeatPumpInfo, WeheatConfigEntry, WeheatDataUpdateCoordinator
from .entity import WeheatEntity
# Coordinator is used to centralize the data updates
@@ -65,14 +64,18 @@ BINARY_SENSORS = [
async def async_setup_entry(
hass: HomeAssistant,
entry: WeheatConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors for weheat heat pump."""
entities = [
- WeheatHeatPumpBinarySensor(coordinator, entity_description)
+ WeheatHeatPumpBinarySensor(
+ weheatdata.heat_pump_info,
+ weheatdata.data_coordinator,
+ entity_description,
+ )
+ for weheatdata in entry.runtime_data
for entity_description in BINARY_SENSORS
- for coordinator in entry.runtime_data
- if entity_description.value_fn(coordinator.data) is not None
+ if entity_description.value_fn(weheatdata.data_coordinator.data) is not None
]
async_add_entities(entities)
@@ -81,20 +84,21 @@ async def async_setup_entry(
class WeheatHeatPumpBinarySensor(WeheatEntity, BinarySensorEntity):
"""Defines a Weheat heat pump binary sensor."""
+ heat_pump_info: HeatPumpInfo
coordinator: WeheatDataUpdateCoordinator
entity_description: WeHeatBinarySensorEntityDescription
def __init__(
self,
+ heat_pump_info: HeatPumpInfo,
coordinator: WeheatDataUpdateCoordinator,
entity_description: WeHeatBinarySensorEntityDescription,
) -> None:
"""Pass coordinator to CoordinatorEntity."""
- super().__init__(coordinator)
-
+ super().__init__(heat_pump_info, coordinator)
self.entity_description = entity_description
- self._attr_unique_id = f"{coordinator.heatpump_id}_{entity_description.key}"
+ self._attr_unique_id = f"{heat_pump_info.heatpump_id}_{entity_description.key}"
@property
def is_on(self) -> bool | None:
diff --git a/homeassistant/components/weheat/const.py b/homeassistant/components/weheat/const.py
index e33fd983572..ee9b77281e6 100644
--- a/homeassistant/components/weheat/const.py
+++ b/homeassistant/components/weheat/const.py
@@ -17,7 +17,8 @@ API_URL = "https://api.weheat.nl"
OAUTH2_SCOPES = ["openid", "offline_access"]
-UPDATE_INTERVAL = 30
+LOG_UPDATE_INTERVAL = 120
+ENERGY_UPDATE_INTERVAL = 1800
LOGGER: Logger = getLogger(__package__)
diff --git a/homeassistant/components/weheat/coordinator.py b/homeassistant/components/weheat/coordinator.py
index 4a85380e4a3..30ca61d0387 100644
--- a/homeassistant/components/weheat/coordinator.py
+++ b/homeassistant/components/weheat/coordinator.py
@@ -1,5 +1,6 @@
"""Define a custom coordinator for the Weheat heatpump integration."""
+from dataclasses import dataclass
from datetime import timedelta
from weheat.abstractions.discovery import HeatPumpDiscovery
@@ -10,9 +11,11 @@ from weheat.exceptions import (
ForbiddenException,
NotFoundException,
ServiceException,
+ TooManyRequestsException,
UnauthorizedException,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
@@ -20,7 +23,9 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.config_entry_oauth2_flow import OAuth2Session
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-from .const import API_URL, DOMAIN, LOGGER, UPDATE_INTERVAL
+from .const import API_URL, DOMAIN, ENERGY_UPDATE_INTERVAL, LOG_UPDATE_INTERVAL, LOGGER
+
+type WeheatConfigEntry = ConfigEntry[list[WeheatData]]
EXCEPTIONS = (
ServiceException,
@@ -28,55 +33,78 @@ EXCEPTIONS = (
ForbiddenException,
BadRequestException,
ApiException,
+ TooManyRequestsException,
)
+class HeatPumpInfo(HeatPumpDiscovery.HeatPumpInfo):
+ """Heat pump info with additional properties."""
+
+ def __init__(self, pump_info: HeatPumpDiscovery.HeatPumpInfo) -> None:
+ """Initialize the HeatPump object with the provided pump information.
+
+ Args:
+ pump_info (HeatPumpDiscovery.HeatPumpInfo): An object containing the heat pump's discovery information, including:
+ - uuid (str): Unique identifier for the heat pump.
+ - uuid (str): Unique identifier for the heat pump.
+ - device_name (str): Name of the heat pump device.
+ - model (str): Model of the heat pump.
+ - sn (str): Serial number of the heat pump.
+ - has_dhw (bool): Indicates if the heat pump has domestic hot water functionality.
+
+ """
+ super().__init__(
+ pump_info.uuid,
+ pump_info.device_name,
+ pump_info.model,
+ pump_info.sn,
+ pump_info.has_dhw,
+ )
+
+ @property
+ def readable_name(self) -> str | None:
+ """Return the readable name of the heat pump."""
+ return self.device_name if self.device_name else self.model
+
+ @property
+ def heatpump_id(self) -> str:
+ """Return the heat pump id."""
+ return self.uuid
+
+
class WeheatDataUpdateCoordinator(DataUpdateCoordinator[HeatPump]):
"""A custom coordinator for the Weheat heatpump integration."""
+ config_entry: WeheatConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: WeheatConfigEntry,
session: OAuth2Session,
heat_pump: HeatPumpDiscovery.HeatPumpInfo,
+ nr_of_heat_pumps: int,
) -> None:
"""Initialize the data coordinator."""
super().__init__(
hass,
+ config_entry=config_entry,
logger=LOGGER,
name=DOMAIN,
- update_interval=timedelta(seconds=UPDATE_INTERVAL),
+ update_interval=timedelta(seconds=LOG_UPDATE_INTERVAL * nr_of_heat_pumps),
)
- self.heat_pump_info = heat_pump
self._heat_pump_data = HeatPump(
API_URL, heat_pump.uuid, async_get_clientsession(hass)
)
self.session = session
- @property
- def heatpump_id(self) -> str:
- """Return the heat pump id."""
- return self.heat_pump_info.uuid
-
- @property
- def readable_name(self) -> str | None:
- """Return the readable name of the heat pump."""
- if self.heat_pump_info.name:
- return self.heat_pump_info.name
- return self.heat_pump_info.model
-
- @property
- def model(self) -> str:
- """Return the model of the heat pump."""
- return self.heat_pump_info.model
-
async def _async_update_data(self) -> HeatPump:
"""Fetch data from the API."""
await self.session.async_ensure_token_valid()
try:
- await self._heat_pump_data.async_get_status(
+ await self._heat_pump_data.async_get_logs(
self.session.token[CONF_ACCESS_TOKEN]
)
except UnauthorizedException as error:
@@ -85,3 +113,54 @@ class WeheatDataUpdateCoordinator(DataUpdateCoordinator[HeatPump]):
raise UpdateFailed(error) from error
return self._heat_pump_data
+
+
+class WeheatEnergyUpdateCoordinator(DataUpdateCoordinator[HeatPump]):
+ """A custom Energy coordinator for the Weheat heatpump integration."""
+
+ config_entry: WeheatConfigEntry
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WeheatConfigEntry,
+ session: OAuth2Session,
+ heat_pump: HeatPumpDiscovery.HeatPumpInfo,
+ ) -> None:
+ """Initialize the data coordinator."""
+ super().__init__(
+ hass,
+ config_entry=config_entry,
+ logger=LOGGER,
+ name=DOMAIN,
+ update_interval=timedelta(seconds=ENERGY_UPDATE_INTERVAL),
+ )
+ self._heat_pump_data = HeatPump(
+ API_URL, heat_pump.uuid, async_get_clientsession(hass)
+ )
+
+ self.session = session
+
+ async def _async_update_data(self) -> HeatPump:
+ """Fetch data from the API."""
+ await self.session.async_ensure_token_valid()
+
+ try:
+ await self._heat_pump_data.async_get_energy(
+ self.session.token[CONF_ACCESS_TOKEN]
+ )
+ except UnauthorizedException as error:
+ raise ConfigEntryAuthFailed from error
+ except EXCEPTIONS as error:
+ raise UpdateFailed(error) from error
+
+ return self._heat_pump_data
+
+
+@dataclass
+class WeheatData:
+ """Data for the Weheat integration."""
+
+ heat_pump_info: HeatPumpInfo
+ data_coordinator: WeheatDataUpdateCoordinator
+ energy_coordinator: WeheatEnergyUpdateCoordinator
diff --git a/homeassistant/components/weheat/entity.py b/homeassistant/components/weheat/entity.py
index 079db596e19..7a12b2edcfa 100644
--- a/homeassistant/components/weheat/entity.py
+++ b/homeassistant/components/weheat/entity.py
@@ -3,25 +3,30 @@
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
+from . import HeatPumpInfo
from .const import DOMAIN, MANUFACTURER
-from .coordinator import WeheatDataUpdateCoordinator
+from .coordinator import WeheatDataUpdateCoordinator, WeheatEnergyUpdateCoordinator
-class WeheatEntity(CoordinatorEntity[WeheatDataUpdateCoordinator]):
+class WeheatEntity[
+ _WeheatEntityT: WeheatDataUpdateCoordinator | WeheatEnergyUpdateCoordinator
+](CoordinatorEntity[_WeheatEntityT]):
"""Defines a base Weheat entity."""
_attr_has_entity_name = True
def __init__(
self,
- coordinator: WeheatDataUpdateCoordinator,
+ heat_pump_info: HeatPumpInfo,
+ coordinator: _WeheatEntityT,
) -> None:
"""Initialize the Weheat entity."""
super().__init__(coordinator)
+ self.heat_pump_info = heat_pump_info
self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, coordinator.heatpump_id)},
- name=coordinator.readable_name,
+ identifiers={(DOMAIN, heat_pump_info.heatpump_id)},
+ name=heat_pump_info.readable_name,
manufacturer=MANUFACTURER,
- model=coordinator.model,
+ model=heat_pump_info.model,
)
diff --git a/homeassistant/components/weheat/manifest.json b/homeassistant/components/weheat/manifest.json
index 1d60f66afba..7297c601213 100644
--- a/homeassistant/components/weheat/manifest.json
+++ b/homeassistant/components/weheat/manifest.json
@@ -6,5 +6,5 @@
"dependencies": ["application_credentials"],
"documentation": "https://www.home-assistant.io/integrations/weheat",
"iot_class": "cloud_polling",
- "requirements": ["weheat==2025.1.15"]
+ "requirements": ["weheat==2025.2.26"]
}
diff --git a/homeassistant/components/weheat/sensor.py b/homeassistant/components/weheat/sensor.py
index 2d840aec86a..d3b758e41eb 100644
--- a/homeassistant/components/weheat/sensor.py
+++ b/homeassistant/components/weheat/sensor.py
@@ -19,16 +19,20 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
-from . import WeheatConfigEntry
from .const import (
DISPLAY_PRECISION_COP,
DISPLAY_PRECISION_WATER_TEMP,
DISPLAY_PRECISION_WATTS,
)
-from .coordinator import WeheatDataUpdateCoordinator
+from .coordinator import (
+ HeatPumpInfo,
+ WeheatConfigEntry,
+ WeheatDataUpdateCoordinator,
+ WeheatEnergyUpdateCoordinator,
+)
from .entity import WeheatEntity
# Coordinator is used to centralize the data updates
@@ -143,22 +147,6 @@ SENSORS = [
else None
),
),
- WeHeatSensorEntityDescription(
- translation_key="electricity_used",
- key="electricity_used",
- native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
- device_class=SensorDeviceClass.ENERGY,
- state_class=SensorStateClass.TOTAL_INCREASING,
- value_fn=lambda status: status.energy_total,
- ),
- WeHeatSensorEntityDescription(
- translation_key="energy_output",
- key="energy_output",
- native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
- device_class=SensorDeviceClass.ENERGY,
- state_class=SensorStateClass.TOTAL_INCREASING,
- value_fn=lambda status: status.energy_output,
- ),
WeHeatSensorEntityDescription(
translation_key="compressor_rpm",
key="compressor_rpm",
@@ -175,7 +163,6 @@ SENSORS = [
),
]
-
DHW_SENSORS = [
WeHeatSensorEntityDescription(
translation_key="dhw_top_temperature",
@@ -197,24 +184,65 @@ DHW_SENSORS = [
),
]
+ENERGY_SENSORS = [
+ WeHeatSensorEntityDescription(
+ translation_key="electricity_used",
+ key="electricity_used",
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ value_fn=lambda status: status.energy_total,
+ ),
+ WeHeatSensorEntityDescription(
+ translation_key="energy_output",
+ key="energy_output",
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ device_class=SensorDeviceClass.ENERGY,
+ state_class=SensorStateClass.TOTAL_INCREASING,
+ value_fn=lambda status: status.energy_output,
+ ),
+]
+
async def async_setup_entry(
hass: HomeAssistant,
entry: WeheatConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensors for weheat heat pump."""
- entities = [
- WeheatHeatPumpSensor(coordinator, entity_description)
- for entity_description in SENSORS
- for coordinator in entry.runtime_data
- ]
- entities.extend(
- WeheatHeatPumpSensor(coordinator, entity_description)
- for entity_description in DHW_SENSORS
- for coordinator in entry.runtime_data
- if coordinator.heat_pump_info.has_dhw
- )
+
+ entities: list[WeheatHeatPumpSensor] = []
+ for weheatdata in entry.runtime_data:
+ entities.extend(
+ WeheatHeatPumpSensor(
+ weheatdata.heat_pump_info,
+ weheatdata.data_coordinator,
+ entity_description,
+ )
+ for entity_description in SENSORS
+ if entity_description.value_fn(weheatdata.data_coordinator.data) is not None
+ )
+ if weheatdata.heat_pump_info.has_dhw:
+ entities.extend(
+ WeheatHeatPumpSensor(
+ weheatdata.heat_pump_info,
+ weheatdata.data_coordinator,
+ entity_description,
+ )
+ for entity_description in DHW_SENSORS
+ if entity_description.value_fn(weheatdata.data_coordinator.data)
+ is not None
+ )
+ entities.extend(
+ WeheatHeatPumpSensor(
+ weheatdata.heat_pump_info,
+ weheatdata.energy_coordinator,
+ entity_description,
+ )
+ for entity_description in ENERGY_SENSORS
+ if entity_description.value_fn(weheatdata.energy_coordinator.data)
+ is not None
+ )
async_add_entities(entities)
@@ -222,20 +250,21 @@ async def async_setup_entry(
class WeheatHeatPumpSensor(WeheatEntity, SensorEntity):
"""Defines a Weheat heat pump sensor."""
- coordinator: WeheatDataUpdateCoordinator
+ heat_pump_info: HeatPumpInfo
+ coordinator: WeheatDataUpdateCoordinator | WeheatEnergyUpdateCoordinator
entity_description: WeHeatSensorEntityDescription
def __init__(
self,
- coordinator: WeheatDataUpdateCoordinator,
+ heat_pump_info: HeatPumpInfo,
+ coordinator: WeheatDataUpdateCoordinator | WeheatEnergyUpdateCoordinator,
entity_description: WeHeatSensorEntityDescription,
) -> None:
"""Pass coordinator to CoordinatorEntity."""
- super().__init__(coordinator)
-
+ super().__init__(heat_pump_info, coordinator)
self.entity_description = entity_description
- self._attr_unique_id = f"{coordinator.heatpump_id}_{entity_description.key}"
+ self._attr_unique_id = f"{heat_pump_info.heatpump_id}_{entity_description.key}"
@property
def native_value(self) -> StateType:
diff --git a/homeassistant/components/weheat/strings.json b/homeassistant/components/weheat/strings.json
index 2a208c2f8ca..3959acad053 100644
--- a/homeassistant/components/weheat/strings.json
+++ b/homeassistant/components/weheat/strings.json
@@ -37,7 +37,7 @@
"name": "Indoor unit water pump"
},
"indoor_unit_auxiliary_pump_state": {
- "name": "Indoor unit auxilary water pump"
+ "name": "Indoor unit auxiliary water pump"
},
"indoor_unit_dhw_valve_or_pump_state": {
"name": "Indoor unit DHW valve or water pump"
diff --git a/homeassistant/components/wemo/binary_sensor.py b/homeassistant/components/wemo/binary_sensor.py
index f2bcb04d96f..4ed361b18ba 100644
--- a/homeassistant/components/wemo/binary_sensor.py
+++ b/homeassistant/components/wemo/binary_sensor.py
@@ -5,7 +5,7 @@ from pywemo import Insight, Maker, StandbyState
from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import async_wemo_dispatcher_connect
from .coordinator import DeviceCoordinator
@@ -15,7 +15,7 @@ from .entity import WemoBinaryStateEntity, WemoEntity
async def async_setup_entry(
hass: HomeAssistant,
_config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WeMo binary sensors."""
diff --git a/homeassistant/components/wemo/coordinator.py b/homeassistant/components/wemo/coordinator.py
index 1f25c12f7ca..0aaedf598d2 100644
--- a/homeassistant/components/wemo/coordinator.py
+++ b/homeassistant/components/wemo/coordinator.py
@@ -88,13 +88,17 @@ class Options:
class DeviceCoordinator(DataUpdateCoordinator[None]):
"""Home Assistant wrapper for a pyWeMo device."""
+ config_entry: ConfigEntry
options: Options | None = None
- def __init__(self, hass: HomeAssistant, wemo: WeMoDevice) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, wemo: WeMoDevice
+ ) -> None:
"""Initialize DeviceCoordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=wemo.name,
update_interval=timedelta(seconds=30),
)
@@ -285,7 +289,7 @@ async def async_register_device(
hass: HomeAssistant, config_entry: ConfigEntry, wemo: WeMoDevice
) -> DeviceCoordinator:
"""Register a device with home assistant and enable pywemo event callbacks."""
- device = DeviceCoordinator(hass, wemo)
+ device = DeviceCoordinator(hass, config_entry, wemo)
await device.async_refresh()
if not device.last_update_success and device.last_exception:
raise device.last_exception
diff --git a/homeassistant/components/wemo/fan.py b/homeassistant/components/wemo/fan.py
index 42dae679aa5..edfdfc1c78c 100644
--- a/homeassistant/components/wemo/fan.py
+++ b/homeassistant/components/wemo/fan.py
@@ -13,7 +13,7 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from homeassistant.util.percentage import (
percentage_to_ranged_value,
@@ -48,7 +48,7 @@ SET_HUMIDITY_SCHEMA: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
_config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WeMo binary sensors."""
diff --git a/homeassistant/components/wemo/light.py b/homeassistant/components/wemo/light.py
index 619e0952457..838073be84a 100644
--- a/homeassistant/components/wemo/light.py
+++ b/homeassistant/components/wemo/light.py
@@ -20,7 +20,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import CONNECTION_ZIGBEE, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from . import async_wemo_dispatcher_connect
@@ -35,7 +35,7 @@ WEMO_OFF = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WeMo lights."""
@@ -53,7 +53,7 @@ async def async_setup_entry(
def async_setup_bridge(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
coordinator: DeviceCoordinator,
) -> None:
"""Set up a WeMo link."""
diff --git a/homeassistant/components/wemo/sensor.py b/homeassistant/components/wemo/sensor.py
index 90e3546eaf7..76a0265d7da 100644
--- a/homeassistant/components/wemo/sensor.py
+++ b/homeassistant/components/wemo/sensor.py
@@ -15,7 +15,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import async_wemo_dispatcher_connect
@@ -59,7 +59,7 @@ ATTRIBUTE_SENSORS = (
async def async_setup_entry(
hass: HomeAssistant,
_config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WeMo sensors."""
diff --git a/homeassistant/components/wemo/switch.py b/homeassistant/components/wemo/switch.py
index 3f7bb08b704..7b87b3147d0 100644
--- a/homeassistant/components/wemo/switch.py
+++ b/homeassistant/components/wemo/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_OFF, STATE_ON, STATE_STANDBY, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import async_wemo_dispatcher_connect
from .coordinator import DeviceCoordinator
@@ -36,7 +36,7 @@ MAKER_SWITCH_TOGGLE = "toggle"
async def async_setup_entry(
hass: HomeAssistant,
_config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WeMo switches."""
diff --git a/homeassistant/components/whirlpool/__init__.py b/homeassistant/components/whirlpool/__init__.py
index 6231324bb0d..cb073779379 100644
--- a/homeassistant/components/whirlpool/__init__.py
+++ b/homeassistant/components/whirlpool/__init__.py
@@ -1,6 +1,5 @@
"""The Whirlpool Appliances integration."""
-from dataclasses import dataclass
import logging
from aiohttp import ClientError
@@ -20,7 +19,7 @@ _LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.CLIMATE, Platform.SENSOR]
-type WhirlpoolConfigEntry = ConfigEntry[WhirlpoolData]
+type WhirlpoolConfigEntry = ConfigEntry[AppliancesManager]
async def async_setup_entry(hass: HomeAssistant, entry: WhirlpoolConfigEntry) -> bool:
@@ -52,8 +51,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: WhirlpoolConfigEntry) ->
if not await appliances_manager.fetch_appliances():
_LOGGER.error("Cannot fetch appliances")
return False
+ await appliances_manager.connect()
- entry.runtime_data = WhirlpoolData(appliances_manager, auth, backend_selector)
+ entry.runtime_data = appliances_manager
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
@@ -61,13 +61,5 @@ async def async_setup_entry(hass: HomeAssistant, entry: WhirlpoolConfigEntry) ->
async def async_unload_entry(hass: HomeAssistant, entry: WhirlpoolConfigEntry) -> bool:
"""Unload a config entry."""
+ await entry.runtime_data.disconnect()
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
-
-
-@dataclass
-class WhirlpoolData:
- """Whirlpool integaration shared data."""
-
- appliances_manager: AppliancesManager
- auth: Auth
- backend_selector: BackendSelector
diff --git a/homeassistant/components/whirlpool/climate.py b/homeassistant/components/whirlpool/climate.py
index 943c5d1c956..84a2c0d52ca 100644
--- a/homeassistant/components/whirlpool/climate.py
+++ b/homeassistant/components/whirlpool/climate.py
@@ -5,10 +5,7 @@ from __future__ import annotations
import logging
from typing import Any
-from aiohttp import ClientSession
from whirlpool.aircon import Aircon, FanSpeed as AirconFanSpeed, Mode as AirconMode
-from whirlpool.auth import Auth
-from whirlpool.backendselector import BackendSelector
from homeassistant.components.climate import (
ENTITY_ID_FORMAT,
@@ -25,10 +22,9 @@ from homeassistant.components.climate import (
)
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import generate_entity_id
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WhirlpoolConfigEntry
from .const import DOMAIN
@@ -70,22 +66,11 @@ SUPPORTED_TARGET_TEMPERATURE_STEP = 1
async def async_setup_entry(
hass: HomeAssistant,
config_entry: WhirlpoolConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry."""
- whirlpool_data = config_entry.runtime_data
-
- aircons = [
- AirConEntity(
- hass,
- ac_data["SAID"],
- ac_data["NAME"],
- whirlpool_data.backend_selector,
- whirlpool_data.auth,
- async_get_clientsession(hass),
- )
- for ac_data in whirlpool_data.appliances_manager.aircons
- ]
+ appliances_manager = config_entry.runtime_data
+ aircons = [AirConEntity(hass, aircon) for aircon in appliances_manager.aircons]
async_add_entities(aircons, True)
@@ -110,36 +95,26 @@ class AirConEntity(ClimateEntity):
_attr_target_temperature_step = SUPPORTED_TARGET_TEMPERATURE_STEP
_attr_temperature_unit = UnitOfTemperature.CELSIUS
- def __init__(
- self,
- hass: HomeAssistant,
- said: str,
- name: str | None,
- backend_selector: BackendSelector,
- auth: Auth,
- session: ClientSession,
- ) -> None:
+ def __init__(self, hass: HomeAssistant, aircon: Aircon) -> None:
"""Initialize the entity."""
- self._aircon = Aircon(backend_selector, auth, said, session)
- self.entity_id = generate_entity_id(ENTITY_ID_FORMAT, said, hass=hass)
- self._attr_unique_id = said
+ self._aircon = aircon
+ self.entity_id = generate_entity_id(ENTITY_ID_FORMAT, aircon.said, hass=hass)
+ self._attr_unique_id = aircon.said
self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, said)},
- name=name if name is not None else said,
+ identifiers={(DOMAIN, aircon.said)},
+ name=aircon.name if aircon.name is not None else aircon.said,
manufacturer="Whirlpool",
model="Sixth Sense",
)
async def async_added_to_hass(self) -> None:
- """Connect aircon to the cloud."""
+ """Register updates callback."""
self._aircon.register_attr_callback(self.async_write_ha_state)
- await self._aircon.connect()
async def async_will_remove_from_hass(self) -> None:
- """Close Whrilpool Appliance sockets before removing."""
+ """Unregister updates callback."""
self._aircon.unregister_attr_callback(self.async_write_ha_state)
- await self._aircon.disconnect()
@property
def available(self) -> bool:
diff --git a/homeassistant/components/whirlpool/diagnostics.py b/homeassistant/components/whirlpool/diagnostics.py
index 87d6ea827e2..09338396de4 100644
--- a/homeassistant/components/whirlpool/diagnostics.py
+++ b/homeassistant/components/whirlpool/diagnostics.py
@@ -4,6 +4,8 @@ from __future__ import annotations
from typing import Any
+from whirlpool.appliance import Appliance
+
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant
@@ -26,18 +28,25 @@ async def async_get_config_entry_diagnostics(
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
- whirlpool = config_entry.runtime_data
+ def get_appliance_diagnostics(appliance: Appliance) -> dict[str, Any]:
+ return {
+ "data_model": appliance.appliance_info.data_model,
+ "category": appliance.appliance_info.category,
+ "model_number": appliance.appliance_info.model_number,
+ }
+
+ appliances_manager = config_entry.runtime_data
diagnostics_data = {
- "Washer_dryers": {
- wd["NAME"]: dict(wd.items())
- for wd in whirlpool.appliances_manager.washer_dryers
+ "washer_dryers": {
+ wd.name: get_appliance_diagnostics(wd)
+ for wd in appliances_manager.washer_dryers
},
"aircons": {
- ac["NAME"]: dict(ac.items()) for ac in whirlpool.appliances_manager.aircons
+ ac.name: get_appliance_diagnostics(ac) for ac in appliances_manager.aircons
},
"ovens": {
- oven["NAME"]: dict(oven.items())
- for oven in whirlpool.appliances_manager.ovens
+ oven.name: get_appliance_diagnostics(oven)
+ for oven in appliances_manager.ovens
},
}
diff --git a/homeassistant/components/whirlpool/manifest.json b/homeassistant/components/whirlpool/manifest.json
index 67901eea482..ace2e31791d 100644
--- a/homeassistant/components/whirlpool/manifest.json
+++ b/homeassistant/components/whirlpool/manifest.json
@@ -7,5 +7,5 @@
"integration_type": "hub",
"iot_class": "cloud_push",
"loggers": ["whirlpool"],
- "requirements": ["whirlpool-sixth-sense==0.18.12"]
+ "requirements": ["whirlpool-sixth-sense==0.19.1"]
}
diff --git a/homeassistant/components/whirlpool/sensor.py b/homeassistant/components/whirlpool/sensor.py
index 9180164c272..d0d13a128e2 100644
--- a/homeassistant/components/whirlpool/sensor.py
+++ b/homeassistant/components/whirlpool/sensor.py
@@ -16,9 +16,8 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
@@ -132,39 +131,18 @@ SENSOR_TIMER: tuple[SensorEntityDescription] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: WhirlpoolConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
- """Config flow entry for Whrilpool Laundry."""
+ """Config flow entry for Whirlpool sensors."""
entities: list = []
- whirlpool_data = config_entry.runtime_data
- for appliance in whirlpool_data.appliances_manager.washer_dryers:
- _wd = WasherDryer(
- whirlpool_data.backend_selector,
- whirlpool_data.auth,
- appliance["SAID"],
- async_get_clientsession(hass),
- )
- await _wd.connect()
-
+ appliances_manager = config_entry.runtime_data
+ for washer_dryer in appliances_manager.washer_dryers:
entities.extend(
- [
- WasherDryerClass(
- appliance["SAID"],
- appliance["NAME"],
- description,
- _wd,
- )
- for description in SENSORS
- ]
+ [WasherDryerClass(washer_dryer, description) for description in SENSORS]
)
entities.extend(
[
- WasherDryerTimeClass(
- appliance["SAID"],
- appliance["NAME"],
- description,
- _wd,
- )
+ WasherDryerTimeClass(washer_dryer, description)
for description in SENSOR_TIMER
]
)
@@ -178,34 +156,30 @@ class WasherDryerClass(SensorEntity):
_attr_has_entity_name = True
def __init__(
- self,
- said: str,
- name: str,
- description: WhirlpoolSensorEntityDescription,
- washdry: WasherDryer,
+ self, washer_dryer: WasherDryer, description: WhirlpoolSensorEntityDescription
) -> None:
"""Initialize the washer sensor."""
- self._wd: WasherDryer = washdry
+ self._wd: WasherDryer = washer_dryer
- if name == "dryer":
+ if washer_dryer.name == "dryer":
self._attr_icon = ICON_D
else:
self._attr_icon = ICON_W
self.entity_description: WhirlpoolSensorEntityDescription = description
self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, said)},
- name=name.capitalize(),
+ identifiers={(DOMAIN, washer_dryer.said)},
+ name=washer_dryer.name.capitalize(),
manufacturer="Whirlpool",
)
- self._attr_unique_id = f"{said}-{description.key}"
+ self._attr_unique_id = f"{washer_dryer.said}-{description.key}"
async def async_added_to_hass(self) -> None:
- """Connect washer/dryer to the cloud."""
+ """Register updates callback."""
self._wd.register_attr_callback(self.async_write_ha_state)
async def async_will_remove_from_hass(self) -> None:
- """Close Whirlpool Appliance sockets before removing."""
+ """Unregister updates callback."""
self._wd.unregister_attr_callback(self.async_write_ha_state)
@property
@@ -226,16 +200,12 @@ class WasherDryerTimeClass(RestoreSensor):
_attr_has_entity_name = True
def __init__(
- self,
- said: str,
- name: str,
- description: SensorEntityDescription,
- washdry: WasherDryer,
+ self, washer_dryer: WasherDryer, description: SensorEntityDescription
) -> None:
"""Initialize the washer sensor."""
- self._wd: WasherDryer = washdry
+ self._wd: WasherDryer = washer_dryer
- if name == "dryer":
+ if washer_dryer.name == "dryer":
self._attr_icon = ICON_D
else:
self._attr_icon = ICON_W
@@ -243,11 +213,11 @@ class WasherDryerTimeClass(RestoreSensor):
self.entity_description: SensorEntityDescription = description
self._running: bool | None = None
self._attr_device_info = DeviceInfo(
- identifiers={(DOMAIN, said)},
- name=name.capitalize(),
+ identifiers={(DOMAIN, washer_dryer.said)},
+ name=washer_dryer.name.capitalize(),
manufacturer="Whirlpool",
)
- self._attr_unique_id = f"{said}-{description.key}"
+ self._attr_unique_id = f"{washer_dryer.said}-{description.key}"
async def async_added_to_hass(self) -> None:
"""Connect washer/dryer to the cloud."""
@@ -259,7 +229,6 @@ class WasherDryerTimeClass(RestoreSensor):
async def async_will_remove_from_hass(self) -> None:
"""Close Whrilpool Appliance sockets before removing."""
self._wd.unregister_attr_callback(self.update_from_latest_data)
- await self._wd.disconnect()
@property
def available(self) -> bool:
diff --git a/homeassistant/components/whois/sensor.py b/homeassistant/components/whois/sensor.py
index fe193b16eea..8098e052575 100644
--- a/homeassistant/components/whois/sensor.py
+++ b/homeassistant/components/whois/sensor.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DOMAIN, EntityCategory, UnitOfTime
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -127,7 +127,7 @@ SENSORS: tuple[WhoisSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the platform from config_entry."""
coordinator: DataUpdateCoordinator[Domain | None] = hass.data[DOMAIN][
diff --git a/homeassistant/components/wiffi/binary_sensor.py b/homeassistant/components/wiffi/binary_sensor.py
index b7431b2555c..93fdb7cce1c 100644
--- a/homeassistant/components/wiffi/binary_sensor.py
+++ b/homeassistant/components/wiffi/binary_sensor.py
@@ -4,7 +4,7 @@ from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CREATE_ENTITY_SIGNAL
from .entity import WiffiEntity
@@ -13,7 +13,7 @@ from .entity import WiffiEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform for a new integration.
diff --git a/homeassistant/components/wiffi/entity.py b/homeassistant/components/wiffi/entity.py
index fd774c930c8..84bbc9b3df1 100644
--- a/homeassistant/components/wiffi/entity.py
+++ b/homeassistant/components/wiffi/entity.py
@@ -41,7 +41,7 @@ class WiffiEntity(Entity):
self._value = None
self._timeout = options.get(CONF_TIMEOUT, DEFAULT_TIMEOUT)
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Entity has been added to hass."""
self.async_on_remove(
async_dispatcher_connect(
diff --git a/homeassistant/components/wiffi/sensor.py b/homeassistant/components/wiffi/sensor.py
index 699a760685a..9afcc719c9b 100644
--- a/homeassistant/components/wiffi/sensor.py
+++ b/homeassistant/components/wiffi/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import DEGREE, LIGHT_LUX, UnitOfPressure, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CREATE_ENTITY_SIGNAL
from .entity import WiffiEntity
@@ -41,7 +41,7 @@ UOM_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up platform for a new integration.
diff --git a/homeassistant/components/wilight/cover.py b/homeassistant/components/wilight/cover.py
index 8a5cb45d909..2e9b92e7a21 100644
--- a/homeassistant/components/wilight/cover.py
+++ b/homeassistant/components/wilight/cover.py
@@ -18,7 +18,7 @@ from pywilight.const import (
from homeassistant.components.cover import ATTR_POSITION, CoverEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import WiLightDevice
@@ -26,7 +26,9 @@ from .parent_device import WiLightParent
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WiLight covers from a config entry."""
parent: WiLightParent = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/wilight/fan.py b/homeassistant/components/wilight/fan.py
index a14198e3b5d..6a22da5879e 100644
--- a/homeassistant/components/wilight/fan.py
+++ b/homeassistant/components/wilight/fan.py
@@ -19,7 +19,7 @@ from pywilight.wilight_device import PyWiLightDevice
from homeassistant.components.fan import DIRECTION_FORWARD, FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
@@ -33,7 +33,9 @@ ORDERED_NAMED_FAN_SPEEDS = [WL_SPEED_LOW, WL_SPEED_MEDIUM, WL_SPEED_HIGH]
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WiLight lights from a config entry."""
parent: WiLightParent = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/wilight/light.py b/homeassistant/components/wilight/light.py
index fbe2499798d..7df0eb1a4c6 100644
--- a/homeassistant/components/wilight/light.py
+++ b/homeassistant/components/wilight/light.py
@@ -15,7 +15,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import WiLightDevice
@@ -41,7 +41,9 @@ def entities_from_discovered_wilight(api_device: PyWiLightDevice) -> list[LightE
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WiLight lights from a config entry."""
parent: WiLightParent = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/wilight/switch.py b/homeassistant/components/wilight/switch.py
index f2a1ce8b0c5..148ea65dd94 100644
--- a/homeassistant/components/wilight/switch.py
+++ b/homeassistant/components/wilight/switch.py
@@ -12,7 +12,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import WiLightDevice
@@ -75,7 +75,9 @@ def entities_from_discovered_wilight(api_device: PyWiLightDevice) -> tuple[Any]:
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WiLight switches from a config entry."""
parent: WiLightParent = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/wirelesstag/entity.py b/homeassistant/components/wirelesstag/entity.py
index 31f8ee99d0d..73b13cdc397 100644
--- a/homeassistant/components/wirelesstag/entity.py
+++ b/homeassistant/components/wirelesstag/entity.py
@@ -60,11 +60,11 @@ class WirelessTagBaseSensor(Entity):
return f"{value:.1f}"
@property
- def available(self):
+ def available(self) -> bool:
"""Return True if entity is available."""
return self._tag.is_alive
- def update(self):
+ def update(self) -> None:
"""Update state."""
if not self.should_poll:
return
diff --git a/homeassistant/components/withings/__init__.py b/homeassistant/components/withings/__init__.py
index 59c3ed8433f..1392b72f16b 100644
--- a/homeassistant/components/withings/__init__.py
+++ b/homeassistant/components/withings/__init__.py
@@ -120,13 +120,17 @@ async def async_setup_entry(hass: HomeAssistant, entry: WithingsConfigEntry) ->
client.refresh_token_function = _refresh_token
withings_data = WithingsData(
client=client,
- measurement_coordinator=WithingsMeasurementDataUpdateCoordinator(hass, client),
- sleep_coordinator=WithingsSleepDataUpdateCoordinator(hass, client),
- bed_presence_coordinator=WithingsBedPresenceDataUpdateCoordinator(hass, client),
- goals_coordinator=WithingsGoalsDataUpdateCoordinator(hass, client),
- activity_coordinator=WithingsActivityDataUpdateCoordinator(hass, client),
- workout_coordinator=WithingsWorkoutDataUpdateCoordinator(hass, client),
- device_coordinator=WithingsDeviceDataUpdateCoordinator(hass, client),
+ measurement_coordinator=WithingsMeasurementDataUpdateCoordinator(
+ hass, entry, client
+ ),
+ sleep_coordinator=WithingsSleepDataUpdateCoordinator(hass, entry, client),
+ bed_presence_coordinator=WithingsBedPresenceDataUpdateCoordinator(
+ hass, entry, client
+ ),
+ goals_coordinator=WithingsGoalsDataUpdateCoordinator(hass, entry, client),
+ activity_coordinator=WithingsActivityDataUpdateCoordinator(hass, entry, client),
+ workout_coordinator=WithingsWorkoutDataUpdateCoordinator(hass, entry, client),
+ device_coordinator=WithingsDeviceDataUpdateCoordinator(hass, entry, client),
)
for coordinator in withings_data.coordinators:
diff --git a/homeassistant/components/withings/binary_sensor.py b/homeassistant/components/withings/binary_sensor.py
index 856aeeffc5c..457bbe59bcc 100644
--- a/homeassistant/components/withings/binary_sensor.py
+++ b/homeassistant/components/withings/binary_sensor.py
@@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WithingsConfigEntry
from .const import DOMAIN
@@ -22,7 +22,7 @@ from .entity import WithingsEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: WithingsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
coordinator = entry.runtime_data.bed_presence_coordinator
diff --git a/homeassistant/components/withings/calendar.py b/homeassistant/components/withings/calendar.py
index ac867fbfdca..8dcad9d73ba 100644
--- a/homeassistant/components/withings/calendar.py
+++ b/homeassistant/components/withings/calendar.py
@@ -11,7 +11,7 @@ from homeassistant.components.calendar import CalendarEntity, CalendarEvent
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import DOMAIN, WithingsConfigEntry
from .coordinator import WithingsWorkoutDataUpdateCoordinator
@@ -21,7 +21,7 @@ from .entity import WithingsEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: WithingsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the calendar platform for entity."""
ent_reg = er.async_get(hass)
diff --git a/homeassistant/components/withings/coordinator.py b/homeassistant/components/withings/coordinator.py
index 79419ae23ff..13789816d85 100644
--- a/homeassistant/components/withings/coordinator.py
+++ b/homeassistant/components/withings/coordinator.py
@@ -44,11 +44,17 @@ class WithingsDataUpdateCoordinator[_DataT](DataUpdateCoordinator[_DataT]):
webhooks_connected: bool = False
coordinator_name: str = ""
- def __init__(self, hass: HomeAssistant, client: WithingsClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WithingsConfigEntry,
+ client: WithingsClient,
+ ) -> None:
"""Initialize the Withings data coordinator."""
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name="",
update_interval=self._default_update_interval,
)
@@ -95,9 +101,14 @@ class WithingsMeasurementDataUpdateCoordinator(
coordinator_name: str = "measurements"
- def __init__(self, hass: HomeAssistant, client: WithingsClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WithingsConfigEntry,
+ client: WithingsClient,
+ ) -> None:
"""Initialize the Withings data coordinator."""
- super().__init__(hass, client)
+ super().__init__(hass, config_entry, client)
self.notification_categories = {
NotificationCategory.WEIGHT,
NotificationCategory.PRESSURE,
@@ -133,9 +144,14 @@ class WithingsSleepDataUpdateCoordinator(
coordinator_name: str = "sleep"
- def __init__(self, hass: HomeAssistant, client: WithingsClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WithingsConfigEntry,
+ client: WithingsClient,
+ ) -> None:
"""Initialize the Withings data coordinator."""
- super().__init__(hass, client)
+ super().__init__(hass, config_entry, client)
self.notification_categories = {
NotificationCategory.SLEEP,
}
@@ -184,9 +200,14 @@ class WithingsBedPresenceDataUpdateCoordinator(WithingsDataUpdateCoordinator[Non
in_bed: bool | None = None
_default_update_interval = None
- def __init__(self, hass: HomeAssistant, client: WithingsClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WithingsConfigEntry,
+ client: WithingsClient,
+ ) -> None:
"""Initialize the Withings data coordinator."""
- super().__init__(hass, client)
+ super().__init__(hass, config_entry, client)
self.notification_categories = {
NotificationCategory.IN_BED,
NotificationCategory.OUT_BED,
@@ -226,9 +247,14 @@ class WithingsActivityDataUpdateCoordinator(
coordinator_name: str = "activity"
_previous_data: Activity | None = None
- def __init__(self, hass: HomeAssistant, client: WithingsClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WithingsConfigEntry,
+ client: WithingsClient,
+ ) -> None:
"""Initialize the Withings data coordinator."""
- super().__init__(hass, client)
+ super().__init__(hass, config_entry, client)
self.notification_categories = {
NotificationCategory.ACTIVITY,
}
@@ -265,9 +291,14 @@ class WithingsWorkoutDataUpdateCoordinator(
coordinator_name: str = "workout"
_previous_data: Workout | None = None
- def __init__(self, hass: HomeAssistant, client: WithingsClient) -> None:
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ config_entry: WithingsConfigEntry,
+ client: WithingsClient,
+ ) -> None:
"""Initialize the Withings data coordinator."""
- super().__init__(hass, client)
+ super().__init__(hass, config_entry, client)
self.notification_categories = {
NotificationCategory.ACTIVITY,
}
diff --git a/homeassistant/components/withings/manifest.json b/homeassistant/components/withings/manifest.json
index 4c78e077d21..232997da054 100644
--- a/homeassistant/components/withings/manifest.json
+++ b/homeassistant/components/withings/manifest.json
@@ -13,5 +13,5 @@
"documentation": "https://www.home-assistant.io/integrations/withings",
"iot_class": "cloud_push",
"loggers": ["aiowithings"],
- "requirements": ["aiowithings==3.1.5"]
+ "requirements": ["aiowithings==3.1.6"]
}
diff --git a/homeassistant/components/withings/sensor.py b/homeassistant/components/withings/sensor.py
index 1005b5995a5..f20145f8bf9 100644
--- a/homeassistant/components/withings/sensor.py
+++ b/homeassistant/components/withings/sensor.py
@@ -36,7 +36,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util import dt as dt_util
@@ -122,7 +122,7 @@ MEASUREMENT_SENSORS: dict[
measurement_type=MeasurementType.HEIGHT,
translation_key="height",
native_unit_of_measurement=UnitOfLength.METERS,
- suggested_display_precision=1,
+ suggested_display_precision=2,
device_class=SensorDeviceClass.DISTANCE,
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
@@ -326,6 +326,7 @@ SLEEP_SENSORS = [
value_fn=lambda sleep_summary: sleep_summary.deep_sleep_duration,
translation_key="deep_sleep",
native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.HOURS,
device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT,
),
@@ -334,6 +335,7 @@ SLEEP_SENSORS = [
value_fn=lambda sleep_summary: sleep_summary.sleep_latency,
translation_key="time_to_sleep",
native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.HOURS,
device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
@@ -343,6 +345,7 @@ SLEEP_SENSORS = [
value_fn=lambda sleep_summary: sleep_summary.wake_up_latency,
translation_key="time_to_wakeup",
native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.HOURS,
device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
@@ -376,6 +379,7 @@ SLEEP_SENSORS = [
value_fn=lambda sleep_summary: sleep_summary.light_sleep_duration,
translation_key="light_sleep",
native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.HOURS,
device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
@@ -385,6 +389,7 @@ SLEEP_SENSORS = [
value_fn=lambda sleep_summary: sleep_summary.rem_sleep_duration,
translation_key="rem_sleep",
native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.HOURS,
device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
@@ -425,6 +430,9 @@ SLEEP_SENSORS = [
key="sleep_snoring",
value_fn=lambda sleep_summary: sleep_summary.snoring,
translation_key="snoring",
+ native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.MINUTES,
+ device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
),
@@ -448,6 +456,7 @@ SLEEP_SENSORS = [
value_fn=lambda sleep_summary: sleep_summary.total_time_awake,
translation_key="wakeup_time",
native_unit_of_measurement=UnitOfTime.SECONDS,
+ suggested_unit_of_measurement=UnitOfTime.HOURS,
device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT,
entity_registry_enabled_default=False,
@@ -683,7 +692,7 @@ def get_current_goals(goals: Goals) -> set[str]:
async def async_setup_entry(
hass: HomeAssistant,
entry: WithingsConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor config entry."""
ent_reg = er.async_get(hass)
diff --git a/homeassistant/components/wiz/binary_sensor.py b/homeassistant/components/wiz/binary_sensor.py
index 3411ee200b9..385e6827d77 100644
--- a/homeassistant/components/wiz/binary_sensor.py
+++ b/homeassistant/components/wiz/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WizConfigEntry
from .const import DOMAIN, SIGNAL_WIZ_PIR
@@ -27,7 +27,7 @@ OCCUPANCY_UNIQUE_ID = "{}_occupancy"
async def async_setup_entry(
hass: HomeAssistant,
entry: WizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WiZ binary sensor platform."""
mac = entry.runtime_data.bulb.mac
diff --git a/homeassistant/components/wiz/light.py b/homeassistant/components/wiz/light.py
index 9ef4cd57b3d..e38d518f6bc 100644
--- a/homeassistant/components/wiz/light.py
+++ b/homeassistant/components/wiz/light.py
@@ -20,7 +20,7 @@ from homeassistant.components.light import (
filter_supported_color_modes,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WizConfigEntry
from .entity import WizToggleEntity
@@ -57,7 +57,7 @@ def _async_pilot_builder(**kwargs: Any) -> PilotBuilder:
async def async_setup_entry(
hass: HomeAssistant,
entry: WizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WiZ Platform from config_flow."""
if entry.runtime_data.bulb.bulbtype.bulb_type != BulbClass.SOCKET:
diff --git a/homeassistant/components/wiz/number.py b/homeassistant/components/wiz/number.py
index 0591e854d7d..0c8ee3f2bf4 100644
--- a/homeassistant/components/wiz/number.py
+++ b/homeassistant/components/wiz/number.py
@@ -15,7 +15,7 @@ from homeassistant.components.number import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WizConfigEntry
from .entity import WizEntity
@@ -68,7 +68,7 @@ NUMBERS: tuple[WizNumberEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: WizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the wiz speed number."""
async_add_entities(
diff --git a/homeassistant/components/wiz/sensor.py b/homeassistant/components/wiz/sensor.py
index eb77686a5cf..217dae9e8fb 100644
--- a/homeassistant/components/wiz/sensor.py
+++ b/homeassistant/components/wiz/sensor.py
@@ -14,7 +14,7 @@ from homeassistant.const import (
UnitOfPower,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WizConfigEntry
from .entity import WizEntity
@@ -45,7 +45,7 @@ POWER_SENSORS: tuple[SensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: WizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the wiz sensor."""
entities = [
diff --git a/homeassistant/components/wiz/switch.py b/homeassistant/components/wiz/switch.py
index 4c089d2d6d2..a57834bc18d 100644
--- a/homeassistant/components/wiz/switch.py
+++ b/homeassistant/components/wiz/switch.py
@@ -9,7 +9,7 @@ from pywizlight.bulblibrary import BulbClass
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WizConfigEntry
from .entity import WizToggleEntity
@@ -19,7 +19,7 @@ from .models import WizData
async def async_setup_entry(
hass: HomeAssistant,
entry: WizConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WiZ switch platform."""
if entry.runtime_data.bulb.bulbtype.bulb_type == BulbClass.SOCKET:
diff --git a/homeassistant/components/wled/button.py b/homeassistant/components/wled/button.py
index 74799b4dcc4..119b2dc9b9f 100644
--- a/homeassistant/components/wled/button.py
+++ b/homeassistant/components/wled/button.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
@@ -16,7 +16,7 @@ from .helpers import wled_exception_handler
async def async_setup_entry(
hass: HomeAssistant,
entry: WLEDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WLED button based on a config entry."""
async_add_entities([WLEDRestartButton(entry.runtime_data)])
diff --git a/homeassistant/components/wled/light.py b/homeassistant/components/wled/light.py
index b4edf10dc58..5e2ff117580 100644
--- a/homeassistant/components/wled/light.py
+++ b/homeassistant/components/wled/light.py
@@ -17,7 +17,7 @@ from homeassistant.components.light import (
LightEntityFeature,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WLEDConfigEntry
from .const import (
@@ -39,7 +39,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: WLEDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WLED light based on a config entry."""
coordinator = entry.runtime_data
@@ -284,7 +284,7 @@ class WLEDSegmentLight(WLEDEntity, LightEntity):
def async_update_segments(
coordinator: WLEDDataUpdateCoordinator,
current_ids: set[int],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Update segments."""
segment_ids = {
diff --git a/homeassistant/components/wled/number.py b/homeassistant/components/wled/number.py
index 225d783bfdb..e4ff184fd4b 100644
--- a/homeassistant/components/wled/number.py
+++ b/homeassistant/components/wled/number.py
@@ -11,7 +11,7 @@ from wled import Segment
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WLEDConfigEntry
from .const import ATTR_INTENSITY, ATTR_SPEED
@@ -25,7 +25,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: WLEDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WLED number based on a config entry."""
coordinator = entry.runtime_data
@@ -130,7 +130,7 @@ class WLEDNumber(WLEDEntity, NumberEntity):
def async_update_segments(
coordinator: WLEDDataUpdateCoordinator,
current_ids: set[int],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Update segments."""
segment_ids = {
diff --git a/homeassistant/components/wled/select.py b/homeassistant/components/wled/select.py
index a645b04573c..e340c323151 100644
--- a/homeassistant/components/wled/select.py
+++ b/homeassistant/components/wled/select.py
@@ -9,7 +9,7 @@ from wled import LiveDataOverride
from homeassistant.components.select import SelectEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: WLEDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WLED select based on a config entry."""
coordinator = entry.runtime_data
@@ -191,7 +191,7 @@ class WLEDPaletteSelect(WLEDEntity, SelectEntity):
def async_update_segments(
coordinator: WLEDDataUpdateCoordinator,
current_ids: set[int],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Update segments."""
segment_ids = {
diff --git a/homeassistant/components/wled/sensor.py b/homeassistant/components/wled/sensor.py
index 4f97c367612..06f96782019 100644
--- a/homeassistant/components/wled/sensor.py
+++ b/homeassistant/components/wled/sensor.py
@@ -22,7 +22,7 @@ from homeassistant.const import (
UnitOfInformation,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
@@ -128,7 +128,7 @@ SENSORS: tuple[WLEDSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: WLEDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WLED sensor based on a config entry."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/wled/switch.py b/homeassistant/components/wled/switch.py
index 643834dcdec..8ed6ed56114 100644
--- a/homeassistant/components/wled/switch.py
+++ b/homeassistant/components/wled/switch.py
@@ -8,7 +8,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WLEDConfigEntry
from .const import ATTR_DURATION, ATTR_TARGET_BRIGHTNESS, ATTR_UDP_PORT
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
entry: WLEDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WLED switch based on a config entry."""
coordinator = entry.runtime_data
@@ -195,7 +195,7 @@ class WLEDReverseSwitch(WLEDEntity, SwitchEntity):
def async_update_segments(
coordinator: WLEDDataUpdateCoordinator,
current_ids: set[int],
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Update segments."""
segment_ids = {
diff --git a/homeassistant/components/wled/update.py b/homeassistant/components/wled/update.py
index 384b394ac50..ccf72425b77 100644
--- a/homeassistant/components/wled/update.py
+++ b/homeassistant/components/wled/update.py
@@ -10,7 +10,7 @@ from homeassistant.components.update import (
UpdateEntityFeature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WLED_KEY, WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator, WLEDReleasesDataUpdateCoordinator
@@ -21,7 +21,7 @@ from .helpers import wled_exception_handler
async def async_setup_entry(
hass: HomeAssistant,
entry: WLEDConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up WLED update based on a config entry."""
async_add_entities([WLEDUpdateEntity(entry.runtime_data, hass.data[WLED_KEY])])
diff --git a/homeassistant/components/wmspro/cover.py b/homeassistant/components/wmspro/cover.py
index a36b34642b7..715add3023f 100644
--- a/homeassistant/components/wmspro/cover.py
+++ b/homeassistant/components/wmspro/cover.py
@@ -12,7 +12,7 @@ from wmspro.const import (
from homeassistant.components.cover import ATTR_POSITION, CoverDeviceClass, CoverEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WebControlProConfigEntry
from .entity import WebControlProGenericEntity
@@ -24,7 +24,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
config_entry: WebControlProConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WMS based covers from a config entry."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/wmspro/light.py b/homeassistant/components/wmspro/light.py
index 9242982bcf9..d181beb1eaa 100644
--- a/homeassistant/components/wmspro/light.py
+++ b/homeassistant/components/wmspro/light.py
@@ -9,7 +9,7 @@ from wmspro.const import WMS_WebControl_pro_API_actionDescription
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.color import brightness_to_value, value_to_brightness
from . import WebControlProConfigEntry
@@ -23,7 +23,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
config_entry: WebControlProConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WMS based lights from a config entry."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/wmspro/scene.py b/homeassistant/components/wmspro/scene.py
index de18106b7f0..7edd7a2b186 100644
--- a/homeassistant/components/wmspro/scene.py
+++ b/homeassistant/components/wmspro/scene.py
@@ -9,7 +9,7 @@ from wmspro.scene import Scene as WMS_Scene
from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import WebControlProConfigEntry
from .const import ATTRIBUTION, DOMAIN, MANUFACTURER
@@ -18,7 +18,7 @@ from .const import ATTRIBUTION, DOMAIN, MANUFACTURER
async def async_setup_entry(
hass: HomeAssistant,
config_entry: WebControlProConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WMS based scenes from a config entry."""
hub = config_entry.runtime_data
diff --git a/homeassistant/components/wolflink/manifest.json b/homeassistant/components/wolflink/manifest.json
index 4bfc0e6dd83..5f3a6366fe1 100644
--- a/homeassistant/components/wolflink/manifest.json
+++ b/homeassistant/components/wolflink/manifest.json
@@ -6,5 +6,5 @@
"documentation": "https://www.home-assistant.io/integrations/wolflink",
"iot_class": "cloud_polling",
"loggers": ["wolf_comm"],
- "requirements": ["wolf-comm==0.0.15"]
+ "requirements": ["wolf-comm==0.0.23"]
}
diff --git a/homeassistant/components/wolflink/sensor.py b/homeassistant/components/wolflink/sensor.py
index 1f6e6c42464..9380c28de89 100644
--- a/homeassistant/components/wolflink/sensor.py
+++ b/homeassistant/components/wolflink/sensor.py
@@ -2,52 +2,150 @@
from __future__ import annotations
+from collections.abc import Callable
+from dataclasses import dataclass
+
from wolf_comm.models import (
+ EnergyParameter,
+ FlowParameter,
+ FrequencyParameter,
HoursParameter,
ListItemParameter,
Parameter,
PercentageParameter,
+ PowerParameter,
Pressure,
+ RPMParameter,
SimpleParameter,
Temperature,
)
-from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
+from homeassistant.components.sensor import (
+ SensorDeviceClass,
+ SensorEntity,
+ SensorEntityDescription,
+ SensorStateClass,
+)
from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import UnitOfPressure, UnitOfTemperature, UnitOfTime
+from homeassistant.const import (
+ PERCENTAGE,
+ REVOLUTIONS_PER_MINUTE,
+ UnitOfEnergy,
+ UnitOfFrequency,
+ UnitOfPower,
+ UnitOfPressure,
+ UnitOfTemperature,
+ UnitOfTime,
+ UnitOfVolumeFlowRate,
+)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import COORDINATOR, DEVICE_ID, DOMAIN, MANUFACTURER, PARAMETERS, STATES
+def get_listitem_resolve_state(wolf_object, state):
+ """Resolve list item state."""
+ resolved_state = [item for item in wolf_object.items if item.value == int(state)]
+ if resolved_state:
+ resolved_name = resolved_state[0].name
+ state = STATES.get(resolved_name, resolved_name)
+ return state
+
+
+@dataclass(kw_only=True, frozen=True)
+class WolflinkSensorEntityDescription(SensorEntityDescription):
+ """Describes Wolflink sensor entity."""
+
+ value_fn: Callable[[Parameter, str], str | None] = lambda param, value: value
+ supported_fn: Callable[[Parameter], bool]
+
+
+SENSOR_DESCRIPTIONS = [
+ WolflinkSensorEntityDescription(
+ key="temperature",
+ device_class=SensorDeviceClass.TEMPERATURE,
+ native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+ supported_fn=lambda param: isinstance(param, Temperature),
+ ),
+ WolflinkSensorEntityDescription(
+ key="pressure",
+ device_class=SensorDeviceClass.PRESSURE,
+ native_unit_of_measurement=UnitOfPressure.BAR,
+ supported_fn=lambda param: isinstance(param, Pressure),
+ ),
+ WolflinkSensorEntityDescription(
+ key="energy",
+ device_class=SensorDeviceClass.ENERGY,
+ native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
+ supported_fn=lambda param: isinstance(param, EnergyParameter),
+ ),
+ WolflinkSensorEntityDescription(
+ key="power",
+ device_class=SensorDeviceClass.POWER,
+ native_unit_of_measurement=UnitOfPower.KILO_WATT,
+ supported_fn=lambda param: isinstance(param, PowerParameter),
+ ),
+ WolflinkSensorEntityDescription(
+ key="percentage",
+ native_unit_of_measurement=PERCENTAGE,
+ supported_fn=lambda param: isinstance(param, PercentageParameter),
+ ),
+ WolflinkSensorEntityDescription(
+ key="list_item",
+ translation_key="state",
+ supported_fn=lambda param: isinstance(param, ListItemParameter),
+ value_fn=get_listitem_resolve_state,
+ ),
+ WolflinkSensorEntityDescription(
+ key="hours",
+ icon="mdi:clock",
+ native_unit_of_measurement=UnitOfTime.HOURS,
+ supported_fn=lambda param: isinstance(param, HoursParameter),
+ ),
+ WolflinkSensorEntityDescription(
+ key="flow",
+ device_class=SensorDeviceClass.VOLUME_FLOW_RATE,
+ native_unit_of_measurement=UnitOfVolumeFlowRate.LITERS_PER_MINUTE,
+ supported_fn=lambda param: isinstance(param, FlowParameter),
+ ),
+ WolflinkSensorEntityDescription(
+ key="frequency",
+ device_class=SensorDeviceClass.FREQUENCY,
+ native_unit_of_measurement=UnitOfFrequency.HERTZ,
+ supported_fn=lambda param: isinstance(param, FrequencyParameter),
+ ),
+ WolflinkSensorEntityDescription(
+ key="rpm",
+ state_class=SensorStateClass.MEASUREMENT,
+ native_unit_of_measurement=REVOLUTIONS_PER_MINUTE,
+ supported_fn=lambda param: isinstance(param, RPMParameter),
+ ),
+ WolflinkSensorEntityDescription(
+ key="default",
+ supported_fn=lambda param: isinstance(param, SimpleParameter),
+ ),
+]
+
+
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up all entries for Wolf Platform."""
-
coordinator = hass.data[DOMAIN][config_entry.entry_id][COORDINATOR]
parameters = hass.data[DOMAIN][config_entry.entry_id][PARAMETERS]
device_id = hass.data[DOMAIN][config_entry.entry_id][DEVICE_ID]
- entities: list[WolfLinkSensor] = []
- for parameter in parameters:
- if isinstance(parameter, Temperature):
- entities.append(WolfLinkTemperature(coordinator, parameter, device_id))
- if isinstance(parameter, Pressure):
- entities.append(WolfLinkPressure(coordinator, parameter, device_id))
- if isinstance(parameter, PercentageParameter):
- entities.append(WolfLinkPercentage(coordinator, parameter, device_id))
- if isinstance(parameter, ListItemParameter):
- entities.append(WolfLinkState(coordinator, parameter, device_id))
- if isinstance(parameter, HoursParameter):
- entities.append(WolfLinkHours(coordinator, parameter, device_id))
- if isinstance(parameter, SimpleParameter):
- entities.append(WolfLinkSensor(coordinator, parameter, device_id))
+ entities: list[WolfLinkSensor] = [
+ WolfLinkSensor(coordinator, parameter, device_id, description)
+ for parameter in parameters
+ for description in SENSOR_DESCRIPTIONS
+ if description.supported_fn(parameter)
+ ]
async_add_entities(entities, True)
@@ -55,9 +153,18 @@ async def async_setup_entry(
class WolfLinkSensor(CoordinatorEntity, SensorEntity):
"""Base class for all Wolf entities."""
- def __init__(self, coordinator, wolf_object: Parameter, device_id) -> None:
+ entity_description: WolflinkSensorEntityDescription
+
+ def __init__(
+ self,
+ coordinator,
+ wolf_object: Parameter,
+ device_id: str,
+ description: WolflinkSensorEntityDescription,
+ ) -> None:
"""Initialize."""
super().__init__(coordinator)
+ self.entity_description = description
self.wolf_object = wolf_object
self._attr_name = wolf_object.name
self._attr_unique_id = f"{device_id}:{wolf_object.parameter_id}"
@@ -69,68 +176,26 @@ class WolfLinkSensor(CoordinatorEntity, SensorEntity):
)
@property
- def native_value(self):
+ def native_value(self) -> str | None:
"""Return the state. Wolf Client is returning only changed values so we need to store old value here."""
if self.wolf_object.parameter_id in self.coordinator.data:
new_state = self.coordinator.data[self.wolf_object.parameter_id]
self.wolf_object.value_id = new_state[0]
self._state = new_state[1]
+ if (
+ isinstance(self.wolf_object, ListItemParameter)
+ and self._state is not None
+ ):
+ self._state = self.entity_description.value_fn(
+ self.wolf_object, self._state
+ )
return self._state
@property
- def extra_state_attributes(self):
+ def extra_state_attributes(self) -> dict[str, str | None]:
"""Return the state attributes."""
return {
"parameter_id": self.wolf_object.parameter_id,
"value_id": self.wolf_object.value_id,
"parent": self.wolf_object.parent,
}
-
-
-class WolfLinkHours(WolfLinkSensor):
- """Class for hour based entities."""
-
- _attr_icon = "mdi:clock"
- _attr_native_unit_of_measurement = UnitOfTime.HOURS
-
-
-class WolfLinkTemperature(WolfLinkSensor):
- """Class for temperature based entities."""
-
- _attr_device_class = SensorDeviceClass.TEMPERATURE
- _attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
-
-
-class WolfLinkPressure(WolfLinkSensor):
- """Class for pressure based entities."""
-
- _attr_device_class = SensorDeviceClass.PRESSURE
- _attr_native_unit_of_measurement = UnitOfPressure.BAR
-
-
-class WolfLinkPercentage(WolfLinkSensor):
- """Class for percentage based entities."""
-
- @property
- def native_unit_of_measurement(self):
- """Return the unit the value is expressed in."""
- return self.wolf_object.unit
-
-
-class WolfLinkState(WolfLinkSensor):
- """Class for entities which has defined list of state."""
-
- _attr_translation_key = "state"
-
- @property
- def native_value(self):
- """Return the state converting with supported values."""
- state = super().native_value
- if state is not None:
- resolved_state = [
- item for item in self.wolf_object.items if item.value == int(state)
- ]
- if resolved_state:
- resolved_name = resolved_state[0].name
- return STATES.get(resolved_name, resolved_name)
- return state
diff --git a/homeassistant/components/workday/binary_sensor.py b/homeassistant/components/workday/binary_sensor.py
index 3aad6d805d0..6b878db8159 100644
--- a/homeassistant/components/workday/binary_sensor.py
+++ b/homeassistant/components/workday/binary_sensor.py
@@ -26,7 +26,7 @@ from homeassistant.core import (
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
from homeassistant.helpers.event import async_track_point_in_utc_time
@@ -113,7 +113,9 @@ def _get_obj_holidays(
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Workday sensor."""
add_holidays: list[str] = entry.options[CONF_ADD_HOLIDAYS]
diff --git a/homeassistant/components/workday/manifest.json b/homeassistant/components/workday/manifest.json
index cbb11a06aec..cc6b0f30002 100644
--- a/homeassistant/components/workday/manifest.json
+++ b/homeassistant/components/workday/manifest.json
@@ -7,5 +7,5 @@
"iot_class": "local_polling",
"loggers": ["holidays"],
"quality_scale": "internal",
- "requirements": ["holidays==0.66"]
+ "requirements": ["holidays==0.68"]
}
diff --git a/homeassistant/components/worldclock/sensor.py b/homeassistant/components/worldclock/sensor.py
index 88e5a317cdd..9b52993919c 100644
--- a/homeassistant/components/worldclock/sensor.py
+++ b/homeassistant/components/worldclock/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, CONF_TIME_ZONE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from .const import CONF_TIME_FORMAT, DOMAIN
@@ -18,7 +18,7 @@ from .const import CONF_TIME_FORMAT, DOMAIN
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the World clock sensor entry."""
time_zone = await dt_util.async_get_time_zone(entry.options[CONF_TIME_ZONE])
diff --git a/homeassistant/components/ws66i/__init__.py b/homeassistant/components/ws66i/__init__.py
index 83ad7bbf070..32c6a11f25c 100644
--- a/homeassistant/components/ws66i/__init__.py
+++ b/homeassistant/components/ws66i/__init__.py
@@ -78,6 +78,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
# Create the coordinator for the WS66i
coordinator: Ws66iDataUpdateCoordinator = Ws66iDataUpdateCoordinator(
hass,
+ entry,
ws66i,
zones,
)
diff --git a/homeassistant/components/ws66i/coordinator.py b/homeassistant/components/ws66i/coordinator.py
index 013e4d02b15..1b2b43963fc 100644
--- a/homeassistant/components/ws66i/coordinator.py
+++ b/homeassistant/components/ws66i/coordinator.py
@@ -6,6 +6,7 @@ import logging
from pyws66i import WS66i, ZoneStatus
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -17,9 +18,12 @@ _LOGGER = logging.getLogger(__name__)
class Ws66iDataUpdateCoordinator(DataUpdateCoordinator[list[ZoneStatus]]):
"""DataUpdateCoordinator to gather data for WS66i Zones."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
my_api: WS66i,
zones: list[int],
) -> None:
@@ -27,6 +31,7 @@ class Ws66iDataUpdateCoordinator(DataUpdateCoordinator[list[ZoneStatus]]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name="WS66i",
update_interval=POLL_INTERVAL,
)
diff --git a/homeassistant/components/ws66i/media_player.py b/homeassistant/components/ws66i/media_player.py
index a2cd7ba471b..fb8ba5ae996 100644
--- a/homeassistant/components/ws66i/media_player.py
+++ b/homeassistant/components/ws66i/media_player.py
@@ -10,7 +10,7 @@ from homeassistant.components.media_player import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN, MAX_VOL
@@ -23,7 +23,7 @@ PARALLEL_UPDATES = 1
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the WS66i 6-zone amplifier platform from a config entry."""
ws66i_data: Ws66iData = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/assist_satellite.py b/homeassistant/components/wyoming/assist_satellite.py
index 615084bcbf3..5440b2bebeb 100644
--- a/homeassistant/components/wyoming/assist_satellite.py
+++ b/homeassistant/components/wyoming/assist_satellite.py
@@ -24,18 +24,20 @@ from wyoming.tts import Synthesize, SynthesizeVoice
from wyoming.vad import VoiceStarted, VoiceStopped
from wyoming.wake import Detect, Detection
-from homeassistant.components import assist_pipeline, intent, tts
+from homeassistant.components import assist_pipeline, ffmpeg, intent, tts
from homeassistant.components.assist_pipeline import PipelineEvent
from homeassistant.components.assist_satellite import (
+ AssistSatelliteAnnouncement,
AssistSatelliteConfiguration,
AssistSatelliteEntity,
AssistSatelliteEntityDescription,
+ AssistSatelliteEntityFeature,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from .const import DOMAIN
+from .const import DOMAIN, SAMPLE_CHANNELS, SAMPLE_WIDTH
from .data import WyomingService
from .devices import SatelliteDevice
from .entity import WyomingSatelliteEntity
@@ -49,6 +51,8 @@ _RESTART_SECONDS: Final = 3
_PING_TIMEOUT: Final = 5
_PING_SEND_DELAY: Final = 2
_PIPELINE_FINISH_TIMEOUT: Final = 1
+_TTS_SAMPLE_RATE: Final = 22050
+_ANNOUNCE_CHUNK_BYTES: Final = 2048 # 1024 samples
# Wyoming stage -> Assist stage
_STAGES: dict[PipelineStage, assist_pipeline.PipelineStage] = {
@@ -62,7 +66,7 @@ _STAGES: dict[PipelineStage, assist_pipeline.PipelineStage] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Wyoming Assist satellite entity."""
domain_data: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
@@ -83,6 +87,7 @@ class WyomingAssistSatellite(WyomingSatelliteEntity, AssistSatelliteEntity):
entity_description = AssistSatelliteEntityDescription(key="assist_satellite")
_attr_translation_key = "assist_satellite"
_attr_name = None
+ _attr_supported_features = AssistSatelliteEntityFeature.ANNOUNCE
def __init__(
self,
@@ -116,6 +121,10 @@ class WyomingAssistSatellite(WyomingSatelliteEntity, AssistSatelliteEntity):
self.device.set_pipeline_listener(self._pipeline_changed)
self.device.set_audio_settings_listener(self._audio_settings_changed)
+ # For announcements
+ self._ffmpeg_manager: ffmpeg.FFmpegManager | None = None
+ self._played_event_received: asyncio.Event | None = None
+
@property
def pipeline_entity_id(self) -> str | None:
"""Return the entity ID of the pipeline to use for the next conversation."""
@@ -131,9 +140,9 @@ class WyomingAssistSatellite(WyomingSatelliteEntity, AssistSatelliteEntity):
"""Options passed for text-to-speech."""
return {
tts.ATTR_PREFERRED_FORMAT: "wav",
- tts.ATTR_PREFERRED_SAMPLE_RATE: 16000,
- tts.ATTR_PREFERRED_SAMPLE_CHANNELS: 1,
- tts.ATTR_PREFERRED_SAMPLE_BYTES: 2,
+ tts.ATTR_PREFERRED_SAMPLE_RATE: _TTS_SAMPLE_RATE,
+ tts.ATTR_PREFERRED_SAMPLE_CHANNELS: SAMPLE_CHANNELS,
+ tts.ATTR_PREFERRED_SAMPLE_BYTES: SAMPLE_WIDTH,
}
async def async_added_to_hass(self) -> None:
@@ -244,6 +253,76 @@ class WyomingAssistSatellite(WyomingSatelliteEntity, AssistSatelliteEntity):
)
)
+ async def async_announce(self, announcement: AssistSatelliteAnnouncement) -> None:
+ """Announce media on the satellite.
+
+ Should block until the announcement is done playing.
+ """
+ assert self._client is not None
+
+ if self._ffmpeg_manager is None:
+ self._ffmpeg_manager = ffmpeg.get_ffmpeg_manager(self.hass)
+
+ if self._played_event_received is None:
+ self._played_event_received = asyncio.Event()
+
+ self._played_event_received.clear()
+ await self._client.write_event(
+ AudioStart(
+ rate=_TTS_SAMPLE_RATE,
+ width=SAMPLE_WIDTH,
+ channels=SAMPLE_CHANNELS,
+ timestamp=0,
+ ).event()
+ )
+
+ timestamp = 0
+ try:
+ # Use ffmpeg to convert to raw PCM audio with the appropriate format
+ proc = await asyncio.create_subprocess_exec(
+ self._ffmpeg_manager.binary,
+ "-i",
+ announcement.media_id,
+ "-f",
+ "s16le",
+ "-ac",
+ str(SAMPLE_CHANNELS),
+ "-ar",
+ str(_TTS_SAMPLE_RATE),
+ "-nostats",
+ "pipe:",
+ stdout=asyncio.subprocess.PIPE,
+ stderr=asyncio.subprocess.PIPE,
+ close_fds=False, # use posix_spawn in CPython < 3.13
+ )
+ assert proc.stdout is not None
+ while True:
+ chunk_bytes = await proc.stdout.read(_ANNOUNCE_CHUNK_BYTES)
+ if not chunk_bytes:
+ break
+
+ chunk = AudioChunk(
+ rate=_TTS_SAMPLE_RATE,
+ width=SAMPLE_WIDTH,
+ channels=SAMPLE_CHANNELS,
+ audio=chunk_bytes,
+ timestamp=timestamp,
+ )
+ await self._client.write_event(chunk.event())
+
+ timestamp += chunk.milliseconds
+ finally:
+ await self._client.write_event(AudioStop().event())
+ if timestamp > 0:
+ # Wait the length of the audio or until we receive a played event
+ audio_seconds = timestamp / 1000
+ try:
+ async with asyncio.timeout(audio_seconds + 0.5):
+ await self._played_event_received.wait()
+ except TimeoutError:
+ # Older satellite clients will wait longer than necessary
+ _LOGGER.debug("Did not receive played event for announcement")
+
# -------------------------------------------------------------------------
def start_satellite(self) -> None:
@@ -511,6 +590,9 @@ class WyomingAssistSatellite(WyomingSatelliteEntity, AssistSatelliteEntity):
elif Played.is_type(client_event.type):
# TTS response has finished playing on satellite
self.tts_response_finished()
+
+ if self._played_event_received is not None:
+ self._played_event_received.set()
else:
_LOGGER.debug("Unexpected event from satellite: %s", client_event)
diff --git a/homeassistant/components/wyoming/binary_sensor.py b/homeassistant/components/wyoming/binary_sensor.py
index 24ee073ec4d..a3652e7f70f 100644
--- a/homeassistant/components/wyoming/binary_sensor.py
+++ b/homeassistant/components/wyoming/binary_sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import WyomingSatelliteEntity
@@ -22,7 +22,7 @@ if TYPE_CHECKING:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up binary sensor entities."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/conversation.py b/homeassistant/components/wyoming/conversation.py
index 988d47925ac..5760d04bfc2 100644
--- a/homeassistant/components/wyoming/conversation.py
+++ b/homeassistant/components/wyoming/conversation.py
@@ -12,7 +12,7 @@ from homeassistant.components import conversation
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import intent
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import ulid as ulid_util
from .const import DOMAIN
@@ -26,7 +26,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Wyoming conversation."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/manifest.json b/homeassistant/components/wyoming/manifest.json
index b837d2a9e76..d75b70dffa8 100644
--- a/homeassistant/components/wyoming/manifest.json
+++ b/homeassistant/components/wyoming/manifest.json
@@ -7,7 +7,8 @@
"assist_satellite",
"assist_pipeline",
"intent",
- "conversation"
+ "conversation",
+ "ffmpeg"
],
"documentation": "https://www.home-assistant.io/integrations/wyoming",
"integration_type": "service",
diff --git a/homeassistant/components/wyoming/number.py b/homeassistant/components/wyoming/number.py
index d9a58cc3333..96ec5877545 100644
--- a/homeassistant/components/wyoming/number.py
+++ b/homeassistant/components/wyoming/number.py
@@ -8,7 +8,7 @@ from homeassistant.components.number import NumberEntityDescription, RestoreNumb
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import WyomingSatelliteEntity
@@ -24,7 +24,7 @@ _MAX_VOLUME_MULTIPLIER: Final = 10.0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Wyoming number entities."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/select.py b/homeassistant/components/wyoming/select.py
index bbcaab81710..2af0438e35f 100644
--- a/homeassistant/components/wyoming/select.py
+++ b/homeassistant/components/wyoming/select.py
@@ -14,7 +14,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers import restore_state
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .devices import SatelliteDevice
@@ -36,7 +36,7 @@ _DEFAULT_NOISE_SUPPRESSION_LEVEL: Final = "off"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Wyoming select entities."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/stt.py b/homeassistant/components/wyoming/stt.py
index a28e5fdb527..2851004a854 100644
--- a/homeassistant/components/wyoming/stt.py
+++ b/homeassistant/components/wyoming/stt.py
@@ -10,7 +10,7 @@ from wyoming.client import AsyncTcpClient
from homeassistant.components import stt
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, SAMPLE_CHANNELS, SAMPLE_RATE, SAMPLE_WIDTH
from .data import WyomingService
@@ -23,7 +23,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Wyoming speech-to-text."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/switch.py b/homeassistant/components/wyoming/switch.py
index 308429331c3..9eb91d5ef39 100644
--- a/homeassistant/components/wyoming/switch.py
+++ b/homeassistant/components/wyoming/switch.py
@@ -9,7 +9,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers import restore_state
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .entity import WyomingSatelliteEntity
@@ -21,7 +21,7 @@ if TYPE_CHECKING:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up VoIP switch entities."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/tts.py b/homeassistant/components/wyoming/tts.py
index 65ce4d942f1..79e431fee98 100644
--- a/homeassistant/components/wyoming/tts.py
+++ b/homeassistant/components/wyoming/tts.py
@@ -12,7 +12,7 @@ from wyoming.tts import Synthesize, SynthesizeVoice
from homeassistant.components import tts
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_SPEAKER, DOMAIN
from .data import WyomingService
@@ -25,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Wyoming speech-to-text."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/wyoming/wake_word.py b/homeassistant/components/wyoming/wake_word.py
index 64dfd60c068..2a21b7303e5 100644
--- a/homeassistant/components/wyoming/wake_word.py
+++ b/homeassistant/components/wyoming/wake_word.py
@@ -11,7 +11,7 @@ from wyoming.wake import Detect, Detection
from homeassistant.components import wake_word
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .data import WyomingService, load_wyoming_info
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Wyoming wake-word-detection."""
item: DomainDataItem = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/xbox/__init__.py b/homeassistant/components/xbox/__init__.py
index ab0d510a709..30bc7d59417 100644
--- a/homeassistant/components/xbox/__init__.py
+++ b/homeassistant/components/xbox/__init__.py
@@ -48,7 +48,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
consoles.dict(),
)
- coordinator = XboxUpdateCoordinator(hass, client, consoles)
+ coordinator = XboxUpdateCoordinator(hass, entry, client, consoles)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = {
diff --git a/homeassistant/components/xbox/binary_sensor.py b/homeassistant/components/xbox/binary_sensor.py
index af95834425a..5339c4d7a8e 100644
--- a/homeassistant/components/xbox/binary_sensor.py
+++ b/homeassistant/components/xbox/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import XboxUpdateCoordinator
@@ -18,7 +18,9 @@ PRESENCE_ATTRIBUTES = ["online", "in_party", "in_game", "in_multiplayer"]
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Xbox Live friends."""
coordinator: XboxUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
diff --git a/homeassistant/components/xbox/coordinator.py b/homeassistant/components/xbox/coordinator.py
index 4012820c43c..62c7a35e88b 100644
--- a/homeassistant/components/xbox/coordinator.py
+++ b/homeassistant/components/xbox/coordinator.py
@@ -20,6 +20,7 @@ from xbox.webapi.api.provider.smartglass.models import (
SmartglassConsoleStatus,
)
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -64,9 +65,12 @@ class XboxData:
class XboxUpdateCoordinator(DataUpdateCoordinator[XboxData]):
"""Store Xbox Console Status."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
client: XboxLiveClient,
consoles: SmartglassConsoleList,
) -> None:
@@ -74,6 +78,7 @@ class XboxUpdateCoordinator(DataUpdateCoordinator[XboxData]):
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(seconds=10),
)
diff --git a/homeassistant/components/xbox/media_player.py b/homeassistant/components/xbox/media_player.py
index 7298c7e2da3..6464b2417cc 100644
--- a/homeassistant/components/xbox/media_player.py
+++ b/homeassistant/components/xbox/media_player.py
@@ -24,7 +24,7 @@ from homeassistant.components.media_player import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .browse_media import build_item_response
@@ -56,7 +56,9 @@ XBOX_STATE_MAP: dict[PlaybackState | PowerState, MediaPlayerState | None] = {
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Xbox media_player from a config entry."""
client: XboxLiveClient = hass.data[DOMAIN][entry.entry_id]["client"]
diff --git a/homeassistant/components/xbox/remote.py b/homeassistant/components/xbox/remote.py
index 1b4ffdf35cc..4e5893ddb13 100644
--- a/homeassistant/components/xbox/remote.py
+++ b/homeassistant/components/xbox/remote.py
@@ -24,7 +24,7 @@ from homeassistant.components.remote import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN
@@ -32,7 +32,9 @@ from .coordinator import ConsoleData, XboxUpdateCoordinator
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Xbox media_player from a config entry."""
client: XboxLiveClient = hass.data[DOMAIN][entry.entry_id]["client"]
diff --git a/homeassistant/components/xbox/sensor.py b/homeassistant/components/xbox/sensor.py
index f269e0a5bb9..da53557a2d3 100644
--- a/homeassistant/components/xbox/sensor.py
+++ b/homeassistant/components/xbox/sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import XboxUpdateCoordinator
@@ -20,7 +20,7 @@ SENSOR_ATTRIBUTES = ["status", "gamer_score", "account_tier", "gold_tenure"]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Xbox Live friends."""
coordinator: XboxUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id][
diff --git a/homeassistant/components/xiaomi_aqara/__init__.py b/homeassistant/components/xiaomi_aqara/__init__.py
index 579994aaf6b..6e4d143d84e 100644
--- a/homeassistant/components/xiaomi_aqara/__init__.py
+++ b/homeassistant/components/xiaomi_aqara/__init__.py
@@ -7,7 +7,7 @@ import voluptuous as vol
from xiaomi_gateway import AsyncXiaomiGatewayMulticast, XiaomiGateway
from homeassistant.components import persistent_notification
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ATTR_DEVICE_ID,
CONF_HOST,
@@ -216,12 +216,7 @@ async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
if unload_ok:
hass.data[DOMAIN][GATEWAYS_KEY].pop(config_entry.entry_id)
- loaded_entries = [
- entry
- for entry in hass.config_entries.async_entries(DOMAIN)
- if entry.state == ConfigEntryState.LOADED
- ]
- if len(loaded_entries) == 1:
+ if not hass.config_entries.async_loaded_entries(DOMAIN):
# No gateways left, stop Xiaomi socket
unsub_stop = hass.data[DOMAIN].pop(KEY_UNSUB_STOP)
unsub_stop()
diff --git a/homeassistant/components/xiaomi_aqara/binary_sensor.py b/homeassistant/components/xiaomi_aqara/binary_sensor.py
index ad91dda2173..47cc823ad7f 100644
--- a/homeassistant/components/xiaomi_aqara/binary_sensor.py
+++ b/homeassistant/components/xiaomi_aqara/binary_sensor.py
@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.restore_state import RestoreEntity
@@ -32,7 +32,7 @@ ATTR_DENSITY = "Density"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Xiaomi devices."""
entities: list[XiaomiBinarySensor] = []
diff --git a/homeassistant/components/xiaomi_aqara/cover.py b/homeassistant/components/xiaomi_aqara/cover.py
index e073ef6b683..82d5129ac5e 100644
--- a/homeassistant/components/xiaomi_aqara/cover.py
+++ b/homeassistant/components/xiaomi_aqara/cover.py
@@ -5,7 +5,7 @@ from typing import Any
from homeassistant.components.cover import ATTR_POSITION, CoverEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, GATEWAYS_KEY
from .entity import XiaomiDevice
@@ -19,7 +19,7 @@ DATA_KEY_PROTO_V2 = "curtain_status"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Xiaomi devices."""
entities = []
diff --git a/homeassistant/components/xiaomi_aqara/entity.py b/homeassistant/components/xiaomi_aqara/entity.py
index db47015c0cf..59107984ddf 100644
--- a/homeassistant/components/xiaomi_aqara/entity.py
+++ b/homeassistant/components/xiaomi_aqara/entity.py
@@ -57,7 +57,7 @@ class XiaomiDevice(Entity):
self._is_gateway = False
self._device_id = self._sid
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Start unavailability tracking."""
self._xiaomi_hub.callbacks[self._sid].append(self.push_data)
self._async_track_unavailable()
@@ -100,7 +100,7 @@ class XiaomiDevice(Entity):
return device_info
@property
- def available(self):
+ def available(self) -> bool:
"""Return True if entity is available."""
return self._is_available
diff --git a/homeassistant/components/xiaomi_aqara/light.py b/homeassistant/components/xiaomi_aqara/light.py
index 11ce7a0107b..ef1f06695f9 100644
--- a/homeassistant/components/xiaomi_aqara/light.py
+++ b/homeassistant/components/xiaomi_aqara/light.py
@@ -13,7 +13,7 @@ from homeassistant.components.light import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .const import DOMAIN, GATEWAYS_KEY
@@ -25,7 +25,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Xiaomi devices."""
entities = []
diff --git a/homeassistant/components/xiaomi_aqara/lock.py b/homeassistant/components/xiaomi_aqara/lock.py
index 5e538f25699..b3f4e9f4caf 100644
--- a/homeassistant/components/xiaomi_aqara/lock.py
+++ b/homeassistant/components/xiaomi_aqara/lock.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from homeassistant.components.lock import LockEntity, LockState
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from .const import DOMAIN, GATEWAYS_KEY
@@ -24,7 +24,7 @@ UNLOCK_MAINTAIN_TIME = 5
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Xiaomi devices."""
gateway = hass.data[DOMAIN][GATEWAYS_KEY][config_entry.entry_id]
diff --git a/homeassistant/components/xiaomi_aqara/sensor.py b/homeassistant/components/xiaomi_aqara/sensor.py
index 49358276a48..59ccee5a1a8 100644
--- a/homeassistant/components/xiaomi_aqara/sensor.py
+++ b/homeassistant/components/xiaomi_aqara/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import BATTERY_MODELS, DOMAIN, GATEWAYS_KEY, POWER_MODELS
from .entity import XiaomiDevice
@@ -85,7 +85,7 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Xiaomi devices."""
entities: list[XiaomiSensor | XiaomiBatterySensor] = []
diff --git a/homeassistant/components/xiaomi_aqara/switch.py b/homeassistant/components/xiaomi_aqara/switch.py
index f66cf8c7603..7d3abf47bd1 100644
--- a/homeassistant/components/xiaomi_aqara/switch.py
+++ b/homeassistant/components/xiaomi_aqara/switch.py
@@ -6,7 +6,7 @@ from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, GATEWAYS_KEY
from .entity import XiaomiDevice
@@ -29,7 +29,7 @@ IN_USE = "inuse"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Perform the setup for Xiaomi devices."""
entities = []
diff --git a/homeassistant/components/xiaomi_ble/binary_sensor.py b/homeassistant/components/xiaomi_ble/binary_sensor.py
index b853f83b967..8956e207253 100644
--- a/homeassistant/components/xiaomi_ble/binary_sensor.py
+++ b/homeassistant/components/xiaomi_ble/binary_sensor.py
@@ -18,7 +18,7 @@ from homeassistant.components.bluetooth.passive_update_processor import (
PassiveBluetoothProcessorEntity,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .coordinator import XiaomiPassiveBluetoothDataProcessor
@@ -135,7 +135,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: XiaomiBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/xiaomi_ble/event.py b/homeassistant/components/xiaomi_ble/event.py
index 7265bcd112c..c5f6e01e575 100644
--- a/homeassistant/components/xiaomi_ble/event.py
+++ b/homeassistant/components/xiaomi_ble/event.py
@@ -12,7 +12,7 @@ from homeassistant.components.event import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import format_discovered_event_class, format_event_dispatcher_name
from .const import (
@@ -183,7 +183,7 @@ class XiaomiEventEntity(EventEntity):
async def async_setup_entry(
hass: HomeAssistant,
entry: XiaomiBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Xiaomi event."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/xiaomi_ble/sensor.py b/homeassistant/components/xiaomi_ble/sensor.py
index ba8f64383ee..01f15ff09b8 100644
--- a/homeassistant/components/xiaomi_ble/sensor.py
+++ b/homeassistant/components/xiaomi_ble/sensor.py
@@ -31,7 +31,7 @@ from homeassistant.const import (
UnitOfTime,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.sensor import sensor_device_info_to_hass_device_info
from .coordinator import XiaomiPassiveBluetoothDataProcessor
@@ -208,7 +208,7 @@ def sensor_update_to_bluetooth_data_update(
async def async_setup_entry(
hass: HomeAssistant,
entry: XiaomiBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi BLE sensors."""
coordinator = entry.runtime_data
diff --git a/homeassistant/components/xiaomi_miio/air_quality.py b/homeassistant/components/xiaomi_miio/air_quality.py
index 199d9161353..1ce37c661a2 100644
--- a/homeassistant/components/xiaomi_miio/air_quality.py
+++ b/homeassistant/components/xiaomi_miio/air_quality.py
@@ -9,7 +9,7 @@ from homeassistant.components.air_quality import AirQualityEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE, CONF_HOST, CONF_MODEL, CONF_TOKEN
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_FLOW_TYPE,
@@ -242,7 +242,7 @@ DEVICE_MAP: dict[str, dict[str, Callable]] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi Air Quality from a config entry."""
entities = []
diff --git a/homeassistant/components/xiaomi_miio/alarm_control_panel.py b/homeassistant/components/xiaomi_miio/alarm_control_panel.py
index 9c06198bc7e..ecab5228f6e 100644
--- a/homeassistant/components/xiaomi_miio/alarm_control_panel.py
+++ b/homeassistant/components/xiaomi_miio/alarm_control_panel.py
@@ -15,7 +15,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import CONF_GATEWAY, DOMAIN
@@ -29,7 +29,7 @@ XIAOMI_STATE_ARMING_VALUE = "oning"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi Gateway Alarm from a config entry."""
entities = []
diff --git a/homeassistant/components/xiaomi_miio/binary_sensor.py b/homeassistant/components/xiaomi_miio/binary_sensor.py
index a5ab7e56e6b..213886691f0 100644
--- a/homeassistant/components/xiaomi_miio/binary_sensor.py
+++ b/homeassistant/components/xiaomi_miio/binary_sensor.py
@@ -14,7 +14,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE, CONF_MODEL, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import VacuumCoordinatorDataAttributes
from .const import (
@@ -171,7 +171,7 @@ def _setup_vacuum_sensors(hass, config_entry, async_add_entities):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi sensor from a config entry."""
entities = []
diff --git a/homeassistant/components/xiaomi_miio/button.py b/homeassistant/components/xiaomi_miio/button.py
index 9a64941f398..a7bcb3a12fe 100644
--- a/homeassistant/components/xiaomi_miio/button.py
+++ b/homeassistant/components/xiaomi_miio/button.py
@@ -14,7 +14,7 @@ from homeassistant.components.button import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_MODEL, EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
@@ -117,14 +117,14 @@ MODEL_TO_BUTTON_MAP: dict[str, tuple[str, ...]] = {
ATTR_RESET_DUST_FILTER,
ATTR_RESET_UPPER_FILTER,
),
- **{model: BUTTONS_FOR_VACUUM for model in MODELS_VACUUM},
+ **dict.fromkeys(MODELS_VACUUM, BUTTONS_FOR_VACUUM),
}
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the button from a config entry."""
model = config_entry.data[CONF_MODEL]
diff --git a/homeassistant/components/xiaomi_miio/entity.py b/homeassistant/components/xiaomi_miio/entity.py
index 0343a7526d7..ba1148985ba 100644
--- a/homeassistant/components/xiaomi_miio/entity.py
+++ b/homeassistant/components/xiaomi_miio/entity.py
@@ -185,7 +185,7 @@ class XiaomiGatewayDevice(CoordinatorEntity, Entity):
)
@property
- def available(self):
+ def available(self) -> bool:
"""Return if entity is available."""
if self.coordinator.data is None:
return False
diff --git a/homeassistant/components/xiaomi_miio/fan.py b/homeassistant/components/xiaomi_miio/fan.py
index 12ed9f7195b..31d5dd9de2c 100644
--- a/homeassistant/components/xiaomi_miio/fan.py
+++ b/homeassistant/components/xiaomi_miio/fan.py
@@ -34,7 +34,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_ENTITY_ID, CONF_DEVICE, CONF_MODEL
from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -205,7 +205,7 @@ FAN_DIRECTIONS_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Fan from a config entry."""
entities: list[FanEntity] = []
diff --git a/homeassistant/components/xiaomi_miio/humidifier.py b/homeassistant/components/xiaomi_miio/humidifier.py
index 4701345756a..f19fbec5e78 100644
--- a/homeassistant/components/xiaomi_miio/humidifier.py
+++ b/homeassistant/components/xiaomi_miio/humidifier.py
@@ -23,7 +23,7 @@ from homeassistant.components.humidifier import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_MODE, CONF_DEVICE, CONF_MODEL
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import percentage_to_ranged_value
from .const import (
@@ -71,7 +71,7 @@ AVAILABLE_MODES_OTHER = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Humidifier from a config entry."""
if config_entry.data[CONF_FLOW_TYPE] != CONF_DEVICE:
diff --git a/homeassistant/components/xiaomi_miio/light.py b/homeassistant/components/xiaomi_miio/light.py
index c1f778928d9..81f68306cbc 100644
--- a/homeassistant/components/xiaomi_miio/light.py
+++ b/homeassistant/components/xiaomi_miio/light.py
@@ -44,7 +44,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util, dt as dt_util
from .const import (
@@ -132,7 +132,7 @@ SERVICE_TO_METHOD = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi light from a config entry."""
entities: list[LightEntity] = []
diff --git a/homeassistant/components/xiaomi_miio/number.py b/homeassistant/components/xiaomi_miio/number.py
index a3c501aad3f..f30d4728275 100644
--- a/homeassistant/components/xiaomi_miio/number.py
+++ b/homeassistant/components/xiaomi_miio/number.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .const import (
@@ -289,7 +289,7 @@ FAVORITE_LEVEL_VALUES = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Selectors from a config entry."""
entities = []
diff --git a/homeassistant/components/xiaomi_miio/select.py b/homeassistant/components/xiaomi_miio/select.py
index 6729ce2e0f4..94a93fc1fae 100644
--- a/homeassistant/components/xiaomi_miio/select.py
+++ b/homeassistant/components/xiaomi_miio/select.py
@@ -32,7 +32,7 @@ from homeassistant.components.select import SelectEntity, SelectEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE, CONF_MODEL, EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_FLOW_TYPE,
@@ -205,7 +205,7 @@ SELECTOR_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Selectors from a config entry."""
if config_entry.data[CONF_FLOW_TYPE] != CONF_DEVICE:
diff --git a/homeassistant/components/xiaomi_miio/sensor.py b/homeassistant/components/xiaomi_miio/sensor.py
index aafcba97487..6f623c46af8 100644
--- a/homeassistant/components/xiaomi_miio/sensor.py
+++ b/homeassistant/components/xiaomi_miio/sensor.py
@@ -45,7 +45,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util
from . import VacuumCoordinatorDataAttributes
@@ -755,7 +755,7 @@ def _setup_vacuum_sensors(hass, config_entry, async_add_entities):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi sensor from a config entry."""
entities: list[SensorEntity] = []
diff --git a/homeassistant/components/xiaomi_miio/strings.json b/homeassistant/components/xiaomi_miio/strings.json
index bafc1ec543b..bd3b3499689 100644
--- a/homeassistant/components/xiaomi_miio/strings.json
+++ b/homeassistant/components/xiaomi_miio/strings.json
@@ -331,7 +331,7 @@
"fields": {
"entity_id": {
"name": "Entity ID",
- "description": "Name of the xiaomi miio entity."
+ "description": "Name of the Xiaomi Miio entity."
}
}
},
@@ -365,7 +365,7 @@
},
"light_set_delayed_turn_off": {
"name": "Light set delayed turn off",
- "description": "Delayed turn off.",
+ "description": "Sets the delayed turning off of a light.",
"fields": {
"entity_id": {
"name": "Entity ID",
@@ -373,7 +373,7 @@
},
"time_period": {
"name": "Time period",
- "description": "Time period for the delayed turn off."
+ "description": "Time period for the delayed turning off."
}
}
},
@@ -398,8 +398,8 @@
}
},
"light_night_light_mode_on": {
- "name": "Night light mode on",
- "description": "Turns the eyecare mode on (EYECARE SMART LAMP 2 ONLY).",
+ "name": "Light night light mode on",
+ "description": "Turns on the night light mode of a light (EYECARE SMART LAMP 2 ONLY).",
"fields": {
"entity_id": {
"name": "Entity ID",
@@ -408,8 +408,8 @@
}
},
"light_night_light_mode_off": {
- "name": "Night light mode off",
- "description": "Turns the eyecare mode fan_set_dry_off (EYECARE SMART LAMP 2 ONLY).",
+ "name": "Light night light mode off",
+ "description": "Turns off the night light mode of a light (EYECARE SMART LAMP 2 ONLY).",
"fields": {
"entity_id": {
"name": "Entity ID",
@@ -419,7 +419,7 @@
},
"light_eyecare_mode_on": {
"name": "Light eyecare mode on",
- "description": "[%key:component::xiaomi_miio::services::light_reminder_on::description%]",
+ "description": "Turns on the eyecare mode of a light (EYECARE SMART LAMP 2 ONLY).",
"fields": {
"entity_id": {
"name": "Entity ID",
@@ -429,7 +429,7 @@
},
"light_eyecare_mode_off": {
"name": "Light eyecare mode off",
- "description": "[%key:component::xiaomi_miio::services::light_reminder_off::description%]",
+ "description": "Turns off the eyecare mode of a light (EYECARE SMART LAMP 2 ONLY).",
"fields": {
"entity_id": {
"name": "Entity ID",
@@ -439,7 +439,7 @@
},
"remote_learn_command": {
"name": "Remote learn command",
- "description": "Learns an IR command, select **Perform action**, point the remote at the IR device, and the learned command will be shown as a notification in Overview.",
+ "description": "Learns an IR command. Select **Perform action**, point the remote at the IR device, and the learned command will be shown as a notification in Overview.",
"fields": {
"slot": {
"name": "Slot",
@@ -447,21 +447,21 @@
},
"timeout": {
"name": "Timeout",
- "description": "Define the timeout, before which the command must be learned."
+ "description": "Define the timeout before which the command must be learned."
}
}
},
"remote_set_led_on": {
"name": "Remote set LED on",
- "description": "Turns on blue LED."
+ "description": "Turns on the remote’s blue LED."
},
"remote_set_led_off": {
"name": "Remote set LED off",
- "description": "Turns off blue LED."
+ "description": "Turns off the remote’s blue LED."
},
"switch_set_wifi_led_on": {
"name": "Switch set Wi-Fi LED on",
- "description": "Turns the wifi led on.",
+ "description": "Turns on the Wi-Fi LED of a switch.",
"fields": {
"entity_id": {
"name": "Entity ID",
@@ -471,7 +471,7 @@
},
"switch_set_wifi_led_off": {
"name": "Switch set Wi-Fi LED off",
- "description": "Turn the Wi-Fi led off.",
+ "description": "Turns off the Wi-Fi LED of a switch.",
"fields": {
"entity_id": {
"name": "Entity ID",
@@ -509,7 +509,7 @@
},
"vacuum_remote_control_start": {
"name": "Vacuum remote control start",
- "description": "Starts remote control of the vacuum cleaner. You can then move it with `remote_control_move`, when done call `remote_control_stop`."
+ "description": "Starts remote control of the vacuum cleaner. You can then move it with the 'Vacuum remote control move' action, when done use 'Vacuum remote control stop'."
},
"vacuum_remote_control_stop": {
"name": "Vacuum remote control stop",
@@ -517,7 +517,7 @@
},
"vacuum_remote_control_move": {
"name": "Vacuum remote control move",
- "description": "Remote controls the vacuum cleaner, make sure you first set it in remote control mode with `remote_control_start`.",
+ "description": "Remote controls the vacuum cleaner, make sure you first set it in remote control mode with the 'Vacuum remote control start' action.",
"fields": {
"velocity": {
"name": "Velocity",
@@ -567,7 +567,7 @@
},
"vacuum_goto": {
"name": "Vacuum go to",
- "description": "Go to the specified coordinates.",
+ "description": "Sends the robot to the specified coordinates.",
"fields": {
"x_coord": {
"name": "X coordinate",
diff --git a/homeassistant/components/xiaomi_miio/switch.py b/homeassistant/components/xiaomi_miio/switch.py
index b4c4300dbe8..e4b94aebc20 100644
--- a/homeassistant/components/xiaomi_miio/switch.py
+++ b/homeassistant/components/xiaomi_miio/switch.py
@@ -30,7 +30,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.helpers import config_validation as cv
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
CONF_FLOW_TYPE,
@@ -341,7 +341,7 @@ SWITCH_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switch from a config entry."""
model = config_entry.data[CONF_MODEL]
diff --git a/homeassistant/components/xiaomi_miio/vacuum.py b/homeassistant/components/xiaomi_miio/vacuum.py
index 532eb9581cd..1cbc79b89f3 100644
--- a/homeassistant/components/xiaomi_miio/vacuum.py
+++ b/homeassistant/components/xiaomi_miio/vacuum.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from homeassistant.util.dt import as_utc
@@ -79,7 +79,7 @@ STATE_CODE_TO_STATE = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Xiaomi vacuum cleaner robot from a config entry."""
entities = []
diff --git a/homeassistant/components/xs1/entity.py b/homeassistant/components/xs1/entity.py
index 7239a6fd446..c1ec43ec33c 100644
--- a/homeassistant/components/xs1/entity.py
+++ b/homeassistant/components/xs1/entity.py
@@ -17,7 +17,7 @@ class XS1DeviceEntity(Entity):
"""Initialize the XS1 device."""
self.device = device
- async def async_update(self):
+ async def async_update(self) -> None:
"""Retrieve latest device state."""
async with UPDATE_LOCK:
await self.hass.async_add_executor_job(self.device.update)
diff --git a/homeassistant/components/yale/binary_sensor.py b/homeassistant/components/yale/binary_sensor.py
index dbb00ad7d42..bb9acb16644 100644
--- a/homeassistant/components/yale/binary_sensor.py
+++ b/homeassistant/components/yale/binary_sensor.py
@@ -21,7 +21,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from . import YaleConfigEntry, YaleData
@@ -92,7 +92,7 @@ SENSOR_TYPES_DOORBELL: tuple[YaleDoorbellBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: YaleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Yale binary sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/yale/button.py b/homeassistant/components/yale/button.py
index b04ad638f0c..005d477e4ca 100644
--- a/homeassistant/components/yale/button.py
+++ b/homeassistant/components/yale/button.py
@@ -2,7 +2,7 @@
from homeassistant.components.button import ButtonEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .entity import YaleEntity
@@ -11,7 +11,7 @@ from .entity import YaleEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: YaleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Yale lock wake buttons."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/yale/camera.py b/homeassistant/components/yale/camera.py
index 217e8f5f6fd..acabba23b59 100644
--- a/homeassistant/components/yale/camera.py
+++ b/homeassistant/components/yale/camera.py
@@ -12,7 +12,7 @@ from yalexs.util import update_doorbell_image_from_activity
from homeassistant.components.camera import Camera
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import aiohttp_client
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry, YaleData
from .const import DEFAULT_NAME, DEFAULT_TIMEOUT
@@ -24,7 +24,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: YaleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Yale cameras."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/yale/event.py b/homeassistant/components/yale/event.py
index 935ba7376f8..0ea7694be6d 100644
--- a/homeassistant/components/yale/event.py
+++ b/homeassistant/components/yale/event.py
@@ -16,7 +16,7 @@ from homeassistant.components.event import (
EventEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry, YaleData
from .entity import YaleDescriptionEntity
@@ -59,7 +59,7 @@ TYPES_DOORBELL: tuple[YaleEventEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: YaleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the yale event platform."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/yale/lock.py b/homeassistant/components/yale/lock.py
index 7fdad118cde..079c1dcd3dd 100644
--- a/homeassistant/components/yale/lock.py
+++ b/homeassistant/components/yale/lock.py
@@ -14,7 +14,7 @@ from yalexs.util import get_latest_activity, update_lock_detail_from_activity
from homeassistant.components.lock import ATTR_CHANGED_BY, LockEntity, LockEntityFeature
from homeassistant.const import ATTR_BATTERY_LEVEL
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.util import dt as dt_util
@@ -29,7 +29,7 @@ LOCK_JAMMED_ERR = 531
async def async_setup_entry(
hass: HomeAssistant,
config_entry: YaleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Yale locks."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/yale/sensor.py b/homeassistant/components/yale/sensor.py
index bb3d4317277..91ecbea704d 100644
--- a/homeassistant/components/yale/sensor.py
+++ b/homeassistant/components/yale/sensor.py
@@ -25,7 +25,7 @@ from homeassistant.const import (
EntityCategory,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .const import (
@@ -82,7 +82,7 @@ SENSOR_TYPE_KEYPAD_BATTERY = YaleSensorEntityDescription[KeypadDetail](
async def async_setup_entry(
hass: HomeAssistant,
config_entry: YaleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Yale sensors."""
data = config_entry.runtime_data
diff --git a/homeassistant/components/yale_smart_alarm/alarm_control_panel.py b/homeassistant/components/yale_smart_alarm/alarm_control_panel.py
index 8244d96064a..b443ba016d6 100644
--- a/homeassistant/components/yale_smart_alarm/alarm_control_panel.py
+++ b/homeassistant/components/yale_smart_alarm/alarm_control_panel.py
@@ -17,7 +17,7 @@ from homeassistant.components.alarm_control_panel import (
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .const import DOMAIN, STATE_MAP, YALE_ALL_ERRORS
@@ -26,7 +26,9 @@ from .entity import YaleAlarmEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: YaleConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the alarm entry."""
diff --git a/homeassistant/components/yale_smart_alarm/binary_sensor.py b/homeassistant/components/yale_smart_alarm/binary_sensor.py
index fa9584505e2..20fe3648eed 100644
--- a/homeassistant/components/yale_smart_alarm/binary_sensor.py
+++ b/homeassistant/components/yale_smart_alarm/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .coordinator import YaleDataUpdateCoordinator
@@ -44,7 +44,9 @@ SENSOR_TYPES = (
async def async_setup_entry(
- hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: YaleConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Yale binary sensor entry."""
diff --git a/homeassistant/components/yale_smart_alarm/button.py b/homeassistant/components/yale_smart_alarm/button.py
index 0e53c814fd4..0875ab4514d 100644
--- a/homeassistant/components/yale_smart_alarm/button.py
+++ b/homeassistant/components/yale_smart_alarm/button.py
@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .const import DOMAIN, YALE_ALL_ERRORS
@@ -25,7 +25,7 @@ BUTTON_TYPES = (
async def async_setup_entry(
hass: HomeAssistant,
entry: YaleConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the button from a config entry."""
diff --git a/homeassistant/components/yale_smart_alarm/lock.py b/homeassistant/components/yale_smart_alarm/lock.py
index 7a93baf0827..f4fae531b67 100644
--- a/homeassistant/components/yale_smart_alarm/lock.py
+++ b/homeassistant/components/yale_smart_alarm/lock.py
@@ -10,7 +10,7 @@ from homeassistant.components.lock import LockEntity, LockState
from homeassistant.const import ATTR_CODE
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .const import (
@@ -30,7 +30,9 @@ LOCK_STATE_MAP = {
async def async_setup_entry(
- hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: YaleConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Yale lock entry."""
diff --git a/homeassistant/components/yale_smart_alarm/select.py b/homeassistant/components/yale_smart_alarm/select.py
index 55b56dd8e54..0b443e762e6 100644
--- a/homeassistant/components/yale_smart_alarm/select.py
+++ b/homeassistant/components/yale_smart_alarm/select.py
@@ -6,7 +6,7 @@ from yalesmartalarmclient import YaleLock, YaleLockVolume
from homeassistant.components.select import SelectEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .coordinator import YaleDataUpdateCoordinator
@@ -16,7 +16,9 @@ VOLUME_OPTIONS = {value.name.lower(): str(value.value) for value in YaleLockVolu
async def async_setup_entry(
- hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: YaleConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Yale select entry."""
diff --git a/homeassistant/components/yale_smart_alarm/sensor.py b/homeassistant/components/yale_smart_alarm/sensor.py
index 50343f2e41f..14301d0c6b5 100644
--- a/homeassistant/components/yale_smart_alarm/sensor.py
+++ b/homeassistant/components/yale_smart_alarm/sensor.py
@@ -7,7 +7,7 @@ from typing import cast
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import YaleConfigEntry
@@ -15,7 +15,9 @@ from .entity import YaleEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: YaleConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Yale sensor entry."""
diff --git a/homeassistant/components/yale_smart_alarm/switch.py b/homeassistant/components/yale_smart_alarm/switch.py
index e8c0817c2de..e4523a66802 100644
--- a/homeassistant/components/yale_smart_alarm/switch.py
+++ b/homeassistant/components/yale_smart_alarm/switch.py
@@ -8,7 +8,7 @@ from yalesmartalarmclient import YaleLock
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YaleConfigEntry
from .coordinator import YaleDataUpdateCoordinator
@@ -16,7 +16,9 @@ from .entity import YaleLockEntity
async def async_setup_entry(
- hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: YaleConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Yale switch entry."""
diff --git a/homeassistant/components/yalexs_ble/binary_sensor.py b/homeassistant/components/yalexs_ble/binary_sensor.py
index 7cd142bb9ba..dc924486df2 100644
--- a/homeassistant/components/yalexs_ble/binary_sensor.py
+++ b/homeassistant/components/yalexs_ble/binary_sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YALEXSBLEConfigEntry
from .entity import YALEXSBLEEntity
@@ -18,7 +18,7 @@ from .entity import YALEXSBLEEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: YALEXSBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YALE XS binary sensors."""
data = entry.runtime_data
diff --git a/homeassistant/components/yalexs_ble/lock.py b/homeassistant/components/yalexs_ble/lock.py
index 6eb32e3f78a..78b92ab9eb1 100644
--- a/homeassistant/components/yalexs_ble/lock.py
+++ b/homeassistant/components/yalexs_ble/lock.py
@@ -8,7 +8,7 @@ from yalexs_ble import ConnectionInfo, LockInfo, LockState, LockStatus
from homeassistant.components.lock import LockEntity
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YALEXSBLEConfigEntry
from .entity import YALEXSBLEEntity
@@ -17,7 +17,7 @@ from .entity import YALEXSBLEEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: YALEXSBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up locks."""
async_add_entities([YaleXSBLELock(entry.runtime_data)])
diff --git a/homeassistant/components/yalexs_ble/sensor.py b/homeassistant/components/yalexs_ble/sensor.py
index 90f61219e0b..bc9312effe3 100644
--- a/homeassistant/components/yalexs_ble/sensor.py
+++ b/homeassistant/components/yalexs_ble/sensor.py
@@ -20,7 +20,7 @@ from homeassistant.const import (
UnitOfElectricPotential,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import YALEXSBLEConfigEntry
from .entity import YALEXSBLEEntity
@@ -75,7 +75,7 @@ SENSORS: tuple[YaleXSBLESensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
entry: YALEXSBLEConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YALE XS Bluetooth sensors."""
data = entry.runtime_data
diff --git a/homeassistant/components/yamaha_musiccast/__init__.py b/homeassistant/components/yamaha_musiccast/__init__.py
index a2ce98dde56..3e890c8b943 100644
--- a/homeassistant/components/yamaha_musiccast/__init__.py
+++ b/homeassistant/components/yamaha_musiccast/__init__.py
@@ -55,7 +55,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async_get_clientsession(hass),
entry.data[CONF_UPNP_DESC],
)
- coordinator = MusicCastDataUpdateCoordinator(hass, client=client)
+ coordinator = MusicCastDataUpdateCoordinator(hass, entry, client=client)
await coordinator.async_config_entry_first_refresh()
coordinator.musiccast.build_capabilities()
diff --git a/homeassistant/components/yamaha_musiccast/coordinator.py b/homeassistant/components/yamaha_musiccast/coordinator.py
index d5e0c67310a..13afbe3aa5e 100644
--- a/homeassistant/components/yamaha_musiccast/coordinator.py
+++ b/homeassistant/components/yamaha_musiccast/coordinator.py
@@ -9,6 +9,7 @@ from typing import TYPE_CHECKING
from aiomusiccast import MusicCastConnectionException
from aiomusiccast.musiccast_device import MusicCastData, MusicCastDevice
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -25,11 +26,21 @@ SCAN_INTERVAL = timedelta(seconds=60)
class MusicCastDataUpdateCoordinator(DataUpdateCoordinator[MusicCastData]):
"""Class to manage fetching data from the API."""
- def __init__(self, hass: HomeAssistant, client: MusicCastDevice) -> None:
+ config_entry: ConfigEntry
+
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, client: MusicCastDevice
+ ) -> None:
"""Initialize."""
self.musiccast = client
- super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
+ super().__init__(
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=SCAN_INTERVAL,
+ )
self.entities: list[MusicCastDeviceEntity] = []
async def _async_update_data(self) -> MusicCastData:
diff --git a/homeassistant/components/yamaha_musiccast/entity.py b/homeassistant/components/yamaha_musiccast/entity.py
index 4f1add825e4..8023b13c10a 100644
--- a/homeassistant/components/yamaha_musiccast/entity.py
+++ b/homeassistant/components/yamaha_musiccast/entity.py
@@ -78,13 +78,13 @@ class MusicCastDeviceEntity(MusicCastEntity):
return device_info
- async def async_added_to_hass(self):
+ async def async_added_to_hass(self) -> None:
"""Run when this Entity has been added to HA."""
await super().async_added_to_hass()
# All entities should register callbacks to update HA when their state changes
self.coordinator.musiccast.register_callback(self.async_write_ha_state)
- async def async_will_remove_from_hass(self):
+ async def async_will_remove_from_hass(self) -> None:
"""Entity being removed from hass."""
await super().async_will_remove_from_hass()
self.coordinator.musiccast.remove_callback(self.async_write_ha_state)
diff --git a/homeassistant/components/yamaha_musiccast/media_player.py b/homeassistant/components/yamaha_musiccast/media_player.py
index cff14f2b67d..7bf139e9c3b 100644
--- a/homeassistant/components/yamaha_musiccast/media_player.py
+++ b/homeassistant/components/yamaha_musiccast/media_player.py
@@ -24,7 +24,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import uuid as uuid_util
from .const import (
@@ -55,7 +55,7 @@ MUSIC_PLAYER_BASE_SUPPORT = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MusicCast sensor based on a config entry."""
coordinator: MusicCastDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/yamaha_musiccast/number.py b/homeassistant/components/yamaha_musiccast/number.py
index 02dd6720d91..0de14ef142d 100644
--- a/homeassistant/components/yamaha_musiccast/number.py
+++ b/homeassistant/components/yamaha_musiccast/number.py
@@ -7,7 +7,7 @@ from aiomusiccast.capabilities import NumberSetter
from homeassistant.components.number import NumberEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MusicCastDataUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import MusicCastCapabilityEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MusicCast number entities based on a config entry."""
coordinator: MusicCastDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/yamaha_musiccast/select.py b/homeassistant/components/yamaha_musiccast/select.py
index 3a4649b9ae5..133cb4c4d7b 100644
--- a/homeassistant/components/yamaha_musiccast/select.py
+++ b/homeassistant/components/yamaha_musiccast/select.py
@@ -7,7 +7,7 @@ from aiomusiccast.capabilities import OptionSetter
from homeassistant.components.select import SelectEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, TRANSLATION_KEY_MAPPING
from .coordinator import MusicCastDataUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import MusicCastCapabilityEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MusicCast select entities based on a config entry."""
coordinator: MusicCastDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/yamaha_musiccast/switch.py b/homeassistant/components/yamaha_musiccast/switch.py
index 49d031a02b5..148f09930f3 100644
--- a/homeassistant/components/yamaha_musiccast/switch.py
+++ b/homeassistant/components/yamaha_musiccast/switch.py
@@ -7,7 +7,7 @@ from aiomusiccast.capabilities import BinarySetter
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import MusicCastDataUpdateCoordinator
@@ -17,7 +17,7 @@ from .entity import MusicCastCapabilityEntity
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up MusicCast sensor based on a config entry."""
coordinator: MusicCastDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/yardian/coordinator.py b/homeassistant/components/yardian/coordinator.py
index b0c8a882474..da016ca4ec4 100644
--- a/homeassistant/components/yardian/coordinator.py
+++ b/homeassistant/components/yardian/coordinator.py
@@ -28,6 +28,8 @@ SCAN_INTERVAL = datetime.timedelta(seconds=30)
class YardianUpdateCoordinator(DataUpdateCoordinator[YardianDeviceState]):
"""Coordinator for Yardian API calls."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
@@ -38,6 +40,7 @@ class YardianUpdateCoordinator(DataUpdateCoordinator[YardianDeviceState]):
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=entry.title,
update_interval=SCAN_INTERVAL,
always_update=False,
diff --git a/homeassistant/components/yardian/switch.py b/homeassistant/components/yardian/switch.py
index 910bacc1c2e..6531a48dc82 100644
--- a/homeassistant/components/yardian/switch.py
+++ b/homeassistant/components/yardian/switch.py
@@ -10,7 +10,7 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv, entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import VolDictType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -26,7 +26,7 @@ SERVICE_SCHEMA_START_IRRIGATION: VolDictType = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up entry for a Yardian irrigation switches."""
coordinator = hass.data[DOMAIN][config_entry.entry_id]
diff --git a/homeassistant/components/yeelight/binary_sensor.py b/homeassistant/components/yeelight/binary_sensor.py
index 9993272d510..69427c65fd5 100644
--- a/homeassistant/components/yeelight/binary_sensor.py
+++ b/homeassistant/components/yeelight/binary_sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_CONFIG_ENTRIES, DATA_DEVICE, DATA_UPDATED, DOMAIN
from .entity import YeelightEntity
@@ -17,7 +17,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Yeelight from a config entry."""
device = hass.data[DOMAIN][DATA_CONFIG_ENTRIES][config_entry.entry_id][DATA_DEVICE]
diff --git a/homeassistant/components/yeelight/light.py b/homeassistant/components/yeelight/light.py
index 92ee3976f7f..a2f705298c9 100644
--- a/homeassistant/components/yeelight/light.py
+++ b/homeassistant/components/yeelight/light.py
@@ -34,7 +34,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.typing import VolDictType
from homeassistant.util import color as color_util
@@ -278,7 +278,7 @@ def _async_cmd[_YeelightBaseLightT: YeelightBaseLight, **_P, _R](
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Yeelight from a config entry."""
custom_effects = _parse_custom_effects(hass.data[DOMAIN][DATA_CUSTOM_EFFECTS])
diff --git a/homeassistant/components/yeelight/strings.json b/homeassistant/components/yeelight/strings.json
index 72e400b7cf3..d53c28cb64a 100644
--- a/homeassistant/components/yeelight/strings.json
+++ b/homeassistant/components/yeelight/strings.json
@@ -161,11 +161,11 @@
},
"set_music_mode": {
"name": "Set music mode",
- "description": "Enables or disables music_mode.",
+ "description": "Enables or disables music mode.",
"fields": {
"music_mode": {
"name": "Music mode",
- "description": "Use true or false to enable / disable music_mode."
+ "description": "Whether to enable or disable music mode."
}
}
}
diff --git a/homeassistant/components/yolink/__init__.py b/homeassistant/components/yolink/__init__.py
index 004c5a70cc1..7ba7433f53f 100644
--- a/homeassistant/components/yolink/__init__.py
+++ b/homeassistant/components/yolink/__init__.py
@@ -7,7 +7,7 @@ from dataclasses import dataclass
from datetime import timedelta
from typing import Any
-from yolink.const import ATTR_DEVICE_SMART_REMOTER
+from yolink.const import ATTR_DEVICE_SMART_REMOTER, ATTR_DEVICE_SWITCH
from yolink.device import YoLinkDevice
from yolink.exception import YoLinkAuthFailError, YoLinkClientError
from yolink.home_manager import YoLinkHome
@@ -75,7 +75,8 @@ class YoLinkHomeMessageListener(MessageListener):
device_coordinator.async_set_updated_data(msg_data)
# handling events
if (
- device_coordinator.device.device_type == ATTR_DEVICE_SMART_REMOTER
+ device_coordinator.device.device_type
+ in [ATTR_DEVICE_SMART_REMOTER, ATTR_DEVICE_SWITCH]
and msg_data.get("event") is not None
):
device_registry = dr.async_get(self._hass)
@@ -152,7 +153,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
paried_device_id := device_pairing_mapping.get(device.device_id)
) is not None:
paried_device = yolink_home.get_device(paried_device_id)
- device_coordinator = YoLinkCoordinator(hass, device, paried_device)
+ device_coordinator = YoLinkCoordinator(hass, entry, device, paried_device)
try:
await device_coordinator.async_config_entry_first_refresh()
except ConfigEntryNotReady:
diff --git a/homeassistant/components/yolink/binary_sensor.py b/homeassistant/components/yolink/binary_sensor.py
index fa4c2202b03..30c04d3a424 100644
--- a/homeassistant/components/yolink/binary_sensor.py
+++ b/homeassistant/components/yolink/binary_sensor.py
@@ -23,7 +23,7 @@ from homeassistant.components.binary_sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import YoLinkCoordinator
@@ -101,7 +101,7 @@ SENSOR_TYPES: tuple[YoLinkBinarySensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink Sensor from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/yolink/climate.py b/homeassistant/components/yolink/climate.py
index ff3bbf0d93b..65253094fa9 100644
--- a/homeassistant/components/yolink/climate.py
+++ b/homeassistant/components/yolink/climate.py
@@ -22,7 +22,7 @@ from homeassistant.components.climate import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import YoLinkCoordinator
@@ -47,7 +47,7 @@ YOLINK_ACTION_2_HA = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink Thermostat from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/yolink/const.py b/homeassistant/components/yolink/const.py
index eb6169eccad..8879ef15125 100644
--- a/homeassistant/components/yolink/const.py
+++ b/homeassistant/components/yolink/const.py
@@ -33,3 +33,7 @@ DEV_MODEL_PLUG_YS6602_UC = "YS6602-UC"
DEV_MODEL_PLUG_YS6602_EC = "YS6602-EC"
DEV_MODEL_PLUG_YS6803_UC = "YS6803-UC"
DEV_MODEL_PLUG_YS6803_EC = "YS6803-EC"
+DEV_MODEL_SWITCH_YS5708_UC = "YS5708-UC"
+DEV_MODEL_SWITCH_YS5708_EC = "YS5708-EC"
+DEV_MODEL_SWITCH_YS5709_UC = "YS5709-UC"
+DEV_MODEL_SWITCH_YS5709_EC = "YS5709-EC"
diff --git a/homeassistant/components/yolink/coordinator.py b/homeassistant/components/yolink/coordinator.py
index b7db36541b1..d18a37bd276 100644
--- a/homeassistant/components/yolink/coordinator.py
+++ b/homeassistant/components/yolink/coordinator.py
@@ -9,6 +9,7 @@ import logging
from yolink.device import YoLinkDevice
from yolink.exception import YoLinkAuthFailError, YoLinkClientError
+from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -21,9 +22,12 @@ _LOGGER = logging.getLogger(__name__)
class YoLinkCoordinator(DataUpdateCoordinator[dict]):
"""YoLink DataUpdateCoordinator."""
+ config_entry: ConfigEntry
+
def __init__(
self,
hass: HomeAssistant,
+ config_entry: ConfigEntry,
device: YoLinkDevice,
paired_device: YoLinkDevice | None = None,
) -> None:
@@ -34,7 +38,11 @@ class YoLinkCoordinator(DataUpdateCoordinator[dict]):
data at first update
"""
super().__init__(
- hass, _LOGGER, name=DOMAIN, update_interval=timedelta(minutes=30)
+ hass,
+ _LOGGER,
+ config_entry=config_entry,
+ name=DOMAIN,
+ update_interval=timedelta(minutes=30),
)
self.device = device
self.paired_device = paired_device
diff --git a/homeassistant/components/yolink/cover.py b/homeassistant/components/yolink/cover.py
index b2454bd0d4a..b1cfc3681cc 100644
--- a/homeassistant/components/yolink/cover.py
+++ b/homeassistant/components/yolink/cover.py
@@ -14,7 +14,7 @@ from homeassistant.components.cover import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import YoLinkCoordinator
@@ -24,7 +24,7 @@ from .entity import YoLinkEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink garage door from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/yolink/device_trigger.py b/homeassistant/components/yolink/device_trigger.py
index 6e247bf858e..6f5ed8b24fa 100644
--- a/homeassistant/components/yolink/device_trigger.py
+++ b/homeassistant/components/yolink/device_trigger.py
@@ -5,7 +5,7 @@ from __future__ import annotations
from typing import Any
import voluptuous as vol
-from yolink.const import ATTR_DEVICE_SMART_REMOTER
+from yolink.const import ATTR_DEVICE_SMART_REMOTER, ATTR_DEVICE_SWITCH
from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA
from homeassistant.components.homeassistant.triggers import event as event_trigger
@@ -21,6 +21,10 @@ from .const import (
DEV_MODEL_FLEX_FOB_YS3604_UC,
DEV_MODEL_FLEX_FOB_YS3614_EC,
DEV_MODEL_FLEX_FOB_YS3614_UC,
+ DEV_MODEL_SWITCH_YS5708_EC,
+ DEV_MODEL_SWITCH_YS5708_UC,
+ DEV_MODEL_SWITCH_YS5709_EC,
+ DEV_MODEL_SWITCH_YS5709_UC,
)
CONF_BUTTON_1 = "button_1"
@@ -30,7 +34,7 @@ CONF_BUTTON_4 = "button_4"
CONF_SHORT_PRESS = "short_press"
CONF_LONG_PRESS = "long_press"
-FLEX_FOB_4_BUTTONS = {
+FLEX_BUTTONS_4 = {
f"{CONF_BUTTON_1}_{CONF_SHORT_PRESS}",
f"{CONF_BUTTON_1}_{CONF_LONG_PRESS}",
f"{CONF_BUTTON_2}_{CONF_SHORT_PRESS}",
@@ -41,7 +45,7 @@ FLEX_FOB_4_BUTTONS = {
f"{CONF_BUTTON_4}_{CONF_LONG_PRESS}",
}
-FLEX_FOB_2_BUTTONS = {
+FLEX_BUTTONS_2 = {
f"{CONF_BUTTON_1}_{CONF_SHORT_PRESS}",
f"{CONF_BUTTON_1}_{CONF_LONG_PRESS}",
f"{CONF_BUTTON_2}_{CONF_SHORT_PRESS}",
@@ -49,16 +53,19 @@ FLEX_FOB_2_BUTTONS = {
}
TRIGGER_SCHEMA = DEVICE_TRIGGER_BASE_SCHEMA.extend(
- {vol.Required(CONF_TYPE): vol.In(FLEX_FOB_4_BUTTONS)}
+ {vol.Required(CONF_TYPE): vol.In(FLEX_BUTTONS_4)}
)
-
-# YoLink Remotes YS3604/YS3614
-FLEX_FOB_TRIGGER_TYPES: dict[str, set[str]] = {
- DEV_MODEL_FLEX_FOB_YS3604_EC: FLEX_FOB_4_BUTTONS,
- DEV_MODEL_FLEX_FOB_YS3604_UC: FLEX_FOB_4_BUTTONS,
- DEV_MODEL_FLEX_FOB_YS3614_UC: FLEX_FOB_2_BUTTONS,
- DEV_MODEL_FLEX_FOB_YS3614_EC: FLEX_FOB_2_BUTTONS,
+# YoLink Remotes YS3604/YS3614, Switch YS5708/YS5709
+TRIGGER_MAPPINGS: dict[str, set[str]] = {
+ DEV_MODEL_FLEX_FOB_YS3604_EC: FLEX_BUTTONS_4,
+ DEV_MODEL_FLEX_FOB_YS3604_UC: FLEX_BUTTONS_4,
+ DEV_MODEL_FLEX_FOB_YS3614_UC: FLEX_BUTTONS_2,
+ DEV_MODEL_FLEX_FOB_YS3614_EC: FLEX_BUTTONS_2,
+ DEV_MODEL_SWITCH_YS5708_EC: FLEX_BUTTONS_2,
+ DEV_MODEL_SWITCH_YS5708_UC: FLEX_BUTTONS_2,
+ DEV_MODEL_SWITCH_YS5709_EC: FLEX_BUTTONS_2,
+ DEV_MODEL_SWITCH_YS5709_UC: FLEX_BUTTONS_2,
}
@@ -68,9 +75,12 @@ async def async_get_triggers(
"""List device triggers for YoLink devices."""
device_registry = dr.async_get(hass)
registry_device = device_registry.async_get(device_id)
- if not registry_device or registry_device.model != ATTR_DEVICE_SMART_REMOTER:
+ if not registry_device or registry_device.model not in [
+ ATTR_DEVICE_SMART_REMOTER,
+ ATTR_DEVICE_SWITCH,
+ ]:
return []
- if registry_device.model_id not in list(FLEX_FOB_TRIGGER_TYPES.keys()):
+ if registry_device.model_id not in list(TRIGGER_MAPPINGS.keys()):
return []
return [
{
@@ -79,7 +89,7 @@ async def async_get_triggers(
CONF_PLATFORM: "device",
CONF_TYPE: trigger,
}
- for trigger in FLEX_FOB_TRIGGER_TYPES[registry_device.model_id]
+ for trigger in TRIGGER_MAPPINGS[registry_device.model_id]
]
diff --git a/homeassistant/components/yolink/light.py b/homeassistant/components/yolink/light.py
index e07d17f7d74..54470673fa5 100644
--- a/homeassistant/components/yolink/light.py
+++ b/homeassistant/components/yolink/light.py
@@ -10,7 +10,7 @@ from yolink.const import ATTR_DEVICE_DIMMER
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import YoLinkCoordinator
@@ -20,7 +20,7 @@ from .entity import YoLinkEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink Dimmer from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/yolink/lock.py b/homeassistant/components/yolink/lock.py
index d675fd8cf06..5e244dd08f2 100644
--- a/homeassistant/components/yolink/lock.py
+++ b/homeassistant/components/yolink/lock.py
@@ -10,7 +10,7 @@ from yolink.const import ATTR_DEVICE_LOCK, ATTR_DEVICE_LOCK_V2
from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import YoLinkCoordinator
@@ -20,7 +20,7 @@ from .entity import YoLinkEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink lock from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
@@ -51,15 +51,16 @@ class YoLinkLockEntity(YoLinkEntity, LockEntity):
def update_entity_state(self, state: dict[str, Any]) -> None:
"""Update HA Entity State."""
state_value = state.get("state")
- if self.coordinator.device.device_type == ATTR_DEVICE_LOCK_V2:
- self._attr_is_locked = (
- state_value["lock"] == "locked" if state_value is not None else None
- )
- else:
- self._attr_is_locked = (
- state_value == "locked" if state_value is not None else None
- )
- self.async_write_ha_state()
+ if state_value is not None:
+ if self.coordinator.device.device_type == ATTR_DEVICE_LOCK_V2:
+ self._attr_is_locked = (
+ state_value["lock"] == "locked" if state_value is not None else None
+ )
+ else:
+ self._attr_is_locked = (
+ state_value == "locked" if state_value is not None else None
+ )
+ self.async_write_ha_state()
async def call_lock_state_change(self, state: str) -> None:
"""Call setState api to change lock state."""
@@ -69,7 +70,7 @@ class YoLinkLockEntity(YoLinkEntity, LockEntity):
)
else:
await self.call_device(ClientRequest("setState", {"state": state}))
- self._attr_is_locked = state == "lock"
+ self._attr_is_locked = state in ["locked", "lock"]
self.async_write_ha_state()
async def async_lock(self, **kwargs: Any) -> None:
diff --git a/homeassistant/components/yolink/manifest.json b/homeassistant/components/yolink/manifest.json
index 78b553d7978..52ae8281f59 100644
--- a/homeassistant/components/yolink/manifest.json
+++ b/homeassistant/components/yolink/manifest.json
@@ -6,5 +6,5 @@
"dependencies": ["auth", "application_credentials"],
"documentation": "https://www.home-assistant.io/integrations/yolink",
"iot_class": "cloud_push",
- "requirements": ["yolink-api==0.4.7"]
+ "requirements": ["yolink-api==0.4.8"]
}
diff --git a/homeassistant/components/yolink/number.py b/homeassistant/components/yolink/number.py
index 7b7b582382b..c643a20d0ea 100644
--- a/homeassistant/components/yolink/number.py
+++ b/homeassistant/components/yolink/number.py
@@ -17,7 +17,7 @@ from homeassistant.components.number import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import YoLinkCoordinator
@@ -66,7 +66,7 @@ DEVICE_CONFIG_DESCRIPTIONS: tuple[YoLinkNumberTypeConfigEntityDescription, ...]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up device number type config option entity from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/yolink/sensor.py b/homeassistant/components/yolink/sensor.py
index 8f263cdae07..511b7718e26 100644
--- a/homeassistant/components/yolink/sensor.py
+++ b/homeassistant/components/yolink/sensor.py
@@ -47,7 +47,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import percentage
from .const import (
@@ -280,7 +280,7 @@ SENSOR_TYPES: tuple[YoLinkSensorEntityDescription, ...] = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink Sensor from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/yolink/services.py b/homeassistant/components/yolink/services.py
index 8d622de70e7..f17408a7005 100644
--- a/homeassistant/components/yolink/services.py
+++ b/homeassistant/components/yolink/services.py
@@ -39,7 +39,7 @@ def async_register_services(hass: HomeAssistant) -> None:
continue
if entry.domain == DOMAIN:
break
- if entry is None or entry.state == ConfigEntryState.NOT_LOADED:
+ if entry is None or entry.state != ConfigEntryState.LOADED:
raise ServiceValidationError(
translation_domain=DOMAIN,
translation_key="invalid_config_entry",
diff --git a/homeassistant/components/yolink/siren.py b/homeassistant/components/yolink/siren.py
index 9e02f50bb70..d13e2dc6573 100644
--- a/homeassistant/components/yolink/siren.py
+++ b/homeassistant/components/yolink/siren.py
@@ -17,7 +17,7 @@ from homeassistant.components.siren import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import YoLinkCoordinator
@@ -46,7 +46,7 @@ DEVICE_TYPE = [ATTR_DEVICE_SIREN]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink siren from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/yolink/switch.py b/homeassistant/components/yolink/switch.py
index c999f04d90d..2af7a3c9ddc 100644
--- a/homeassistant/components/yolink/switch.py
+++ b/homeassistant/components/yolink/switch.py
@@ -23,7 +23,7 @@ from homeassistant.components.switch import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DEV_MODEL_MULTI_OUTLET_YS6801, DOMAIN
from .coordinator import YoLinkCoordinator
@@ -116,7 +116,7 @@ DEVICE_TYPE = [
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink switch from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
@@ -162,11 +162,12 @@ class YoLinkSwitchEntity(YoLinkEntity, SwitchEntity):
@callback
def update_entity_state(self, state: dict[str, str | list[str]]) -> None:
"""Update HA Entity State."""
- self._attr_is_on = self._get_state(
- state.get("state"),
- self.entity_description.plug_index_fn(self.coordinator.device),
- )
- self.async_write_ha_state()
+ if (state_value := state.get("state")) is not None:
+ self._attr_is_on = self._get_state(
+ state_value,
+ self.entity_description.plug_index_fn(self.coordinator.device),
+ )
+ self.async_write_ha_state()
async def call_state_change(self, state: str) -> None:
"""Call setState api to change switch state."""
diff --git a/homeassistant/components/yolink/valve.py b/homeassistant/components/yolink/valve.py
index d8c199697c3..26ce72a53d1 100644
--- a/homeassistant/components/yolink/valve.py
+++ b/homeassistant/components/yolink/valve.py
@@ -17,7 +17,7 @@ from homeassistant.components.valve import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DEV_MODEL_WATER_METER_YS5007, DOMAIN
from .coordinator import YoLinkCoordinator
@@ -50,7 +50,7 @@ DEVICE_TYPE = [ATTR_DEVICE_WATER_METER_CONTROLLER]
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up YoLink valve from a config entry."""
device_coordinators = hass.data[DOMAIN][config_entry.entry_id].device_coordinators
diff --git a/homeassistant/components/youless/sensor.py b/homeassistant/components/youless/sensor.py
index db8244c0b06..be17bed4352 100644
--- a/homeassistant/components/youless/sensor.py
+++ b/homeassistant/components/youless/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
UnitOfVolume,
)
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import DOMAIN
@@ -302,7 +302,9 @@ SENSOR_TYPES: tuple[YouLessSensorEntityDescription, ...] = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize the integration."""
coordinator: YouLessCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/youtube/__init__.py b/homeassistant/components/youtube/__init__.py
index aee4b83508c..ec8a3f325cb 100644
--- a/homeassistant/components/youtube/__init__.py
+++ b/homeassistant/components/youtube/__init__.py
@@ -36,7 +36,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
raise ConfigEntryNotReady from err
except ClientError as err:
raise ConfigEntryNotReady from err
- coordinator = YouTubeDataUpdateCoordinator(hass, auth)
+ coordinator = YouTubeDataUpdateCoordinator(hass, entry, auth)
await coordinator.async_config_entry_first_refresh()
diff --git a/homeassistant/components/youtube/coordinator.py b/homeassistant/components/youtube/coordinator.py
index 0da480f1169..476e5bb4022 100644
--- a/homeassistant/components/youtube/coordinator.py
+++ b/homeassistant/components/youtube/coordinator.py
@@ -35,12 +35,15 @@ class YouTubeDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
config_entry: ConfigEntry
- def __init__(self, hass: HomeAssistant, auth: AsyncConfigEntryAuth) -> None:
+ def __init__(
+ self, hass: HomeAssistant, config_entry: ConfigEntry, auth: AsyncConfigEntryAuth
+ ) -> None:
"""Initialize the YouTube data coordinator."""
self._auth = auth
super().__init__(
hass,
LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=timedelta(minutes=15),
)
diff --git a/homeassistant/components/youtube/sensor.py b/homeassistant/components/youtube/sensor.py
index 8832382508c..128c23f7082 100644
--- a/homeassistant/components/youtube/sensor.py
+++ b/homeassistant/components/youtube/sensor.py
@@ -10,7 +10,7 @@ from homeassistant.components.sensor import SensorEntity, SensorEntityDescriptio
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_ICON
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import YouTubeDataUpdateCoordinator
@@ -72,7 +72,9 @@ SENSOR_TYPES = [
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the YouTube sensor."""
coordinator: YouTubeDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
diff --git a/homeassistant/components/zamg/coordinator.py b/homeassistant/components/zamg/coordinator.py
index d53c743f500..a88c97ad267 100644
--- a/homeassistant/components/zamg/coordinator.py
+++ b/homeassistant/components/zamg/coordinator.py
@@ -32,6 +32,7 @@ class ZamgDataUpdateCoordinator(DataUpdateCoordinator[ZamgDevice]):
super().__init__(
hass,
LOGGER,
+ config_entry=entry,
name=DOMAIN,
update_interval=MIN_TIME_BETWEEN_UPDATES,
)
diff --git a/homeassistant/components/zamg/sensor.py b/homeassistant/components/zamg/sensor.py
index 7c7f5fd6c16..5846092e555 100644
--- a/homeassistant/components/zamg/sensor.py
+++ b/homeassistant/components/zamg/sensor.py
@@ -23,7 +23,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@@ -163,7 +163,9 @@ API_FIELDS: list[str] = [desc.para_name for desc in SENSOR_TYPES]
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ZAMG sensor platform."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/zamg/weather.py b/homeassistant/components/zamg/weather.py
index 286a6460f19..ac376577ade 100644
--- a/homeassistant/components/zamg/weather.py
+++ b/homeassistant/components/zamg/weather.py
@@ -12,7 +12,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import ATTRIBUTION, CONF_STATION_ID, DOMAIN, MANUFACTURER_URL
@@ -20,7 +20,9 @@ from .coordinator import ZamgDataUpdateCoordinator
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the ZAMG weather platform."""
coordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/zeroconf/__init__.py b/homeassistant/components/zeroconf/__init__.py
index b748006336c..e80b6b8cfdb 100644
--- a/homeassistant/components/zeroconf/__init__.py
+++ b/homeassistant/components/zeroconf/__init__.py
@@ -141,13 +141,13 @@ def async_get_async_zeroconf(hass: HomeAssistant) -> HaAsyncZeroconf:
return _async_get_instance(hass)
-def _async_get_instance(hass: HomeAssistant, **zcargs: Any) -> HaAsyncZeroconf:
+def _async_get_instance(hass: HomeAssistant) -> HaAsyncZeroconf:
if DOMAIN in hass.data:
return cast(HaAsyncZeroconf, hass.data[DOMAIN])
logging.getLogger("zeroconf").setLevel(logging.NOTSET)
- zeroconf = HaZeroconf(**zcargs)
+ zeroconf = HaZeroconf(**_async_get_zc_args(hass))
aio_zc = HaAsyncZeroconf(zc=zeroconf)
install_multiple_zeroconf_catcher(zeroconf)
@@ -175,12 +175,10 @@ def _async_zc_has_functional_dual_stack() -> bool:
)
-async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
- """Set up Zeroconf and make Home Assistant discoverable."""
- zc_args: dict = {"ip_version": IPVersion.V4Only}
-
- adapters = await network.async_get_adapters(hass)
-
+def _async_get_zc_args(hass: HomeAssistant) -> dict[str, Any]:
+ """Get zeroconf arguments from config."""
+ zc_args: dict[str, Any] = {"ip_version": IPVersion.V4Only}
+ adapters = network.async_get_loaded_adapters(hass)
ipv6 = False
if _async_zc_has_functional_dual_stack():
if any(adapter["enabled"] and adapter["ipv6"] for adapter in adapters):
@@ -195,7 +193,9 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
else:
zc_args["interfaces"] = [
str(source_ip)
- for source_ip in await network.async_get_enabled_source_ips(hass)
+ for source_ip in network.async_get_enabled_source_ips_from_adapters(
+ adapters
+ )
if not source_ip.is_loopback
and not (isinstance(source_ip, IPv6Address) and source_ip.is_global)
and not (
@@ -207,8 +207,12 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
and zc_args["ip_version"] == IPVersion.V6Only
)
]
+ return zc_args
- aio_zc = _async_get_instance(hass, **zc_args)
+
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
+ """Set up Zeroconf and make Home Assistant discoverable."""
+ aio_zc = _async_get_instance(hass)
zeroconf = cast(HaZeroconf, aio_zc.zeroconf)
zeroconf_types = await async_get_zeroconf(hass)
homekit_models = await async_get_homekit(hass)
diff --git a/homeassistant/components/zeroconf/manifest.json b/homeassistant/components/zeroconf/manifest.json
index f4a78cd99e9..a7fbfdfeada 100644
--- a/homeassistant/components/zeroconf/manifest.json
+++ b/homeassistant/components/zeroconf/manifest.json
@@ -8,5 +8,5 @@
"iot_class": "local_push",
"loggers": ["zeroconf"],
"quality_scale": "internal",
- "requirements": ["zeroconf==0.143.0"]
+ "requirements": ["zeroconf==0.146.0"]
}
diff --git a/homeassistant/components/zerproc/light.py b/homeassistant/components/zerproc/light.py
index 36a964a46ab..19175ae3084 100644
--- a/homeassistant/components/zerproc/light.py
+++ b/homeassistant/components/zerproc/light.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import Event, HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_time_interval
from homeassistant.util import color as color_util
@@ -51,7 +51,7 @@ async def discover_entities(hass: HomeAssistant) -> list[ZerprocLight]:
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Zerproc light devices."""
warned = False
diff --git a/homeassistant/components/zeversolar/coordinator.py b/homeassistant/components/zeversolar/coordinator.py
index 9f6ff49eaf8..ec68cf4b56f 100644
--- a/homeassistant/components/zeversolar/coordinator.py
+++ b/homeassistant/components/zeversolar/coordinator.py
@@ -20,11 +20,14 @@ _LOGGER = logging.getLogger(__name__)
class ZeversolarCoordinator(DataUpdateCoordinator[zeversolar.ZeverSolarData]):
"""Data update coordinator."""
+ config_entry: ConfigEntry
+
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=entry,
name=DOMAIN,
update_interval=timedelta(minutes=1),
)
diff --git a/homeassistant/components/zeversolar/sensor.py b/homeassistant/components/zeversolar/sensor.py
index 5023e274267..330e5bb72d8 100644
--- a/homeassistant/components/zeversolar/sensor.py
+++ b/homeassistant/components/zeversolar/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory, UnitOfEnergy, UnitOfPower
from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN
from .coordinator import ZeversolarCoordinator
@@ -52,7 +52,9 @@ SENSOR_TYPES = (
async def async_setup_entry(
- hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+ hass: HomeAssistant,
+ entry: ConfigEntry,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zeversolar sensor."""
coordinator: ZeversolarCoordinator = hass.data[DOMAIN][entry.entry_id]
diff --git a/homeassistant/components/zha/alarm_control_panel.py b/homeassistant/components/zha/alarm_control_panel.py
index 734683e5497..ff61ce07d23 100644
--- a/homeassistant/components/zha/alarm_control_panel.py
+++ b/homeassistant/components/zha/alarm_control_panel.py
@@ -18,7 +18,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -46,7 +46,7 @@ ZHA_STATE_TO_ALARM_STATE_MAP = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation alarm control panel from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/binary_sensor.py b/homeassistant/components/zha/binary_sensor.py
index f45ebf0c5a5..f8146026384 100644
--- a/homeassistant/components/zha/binary_sensor.py
+++ b/homeassistant/components/zha/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -26,7 +26,7 @@ from .helpers import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation binary sensor from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/button.py b/homeassistant/components/zha/button.py
index ecd5cd51f61..dd90bcd29b1 100644
--- a/homeassistant/components/zha/button.py
+++ b/homeassistant/components/zha/button.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation button from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/climate.py b/homeassistant/components/zha/climate.py
index af9f56cd7dc..a3f60420a38 100644
--- a/homeassistant/components/zha/climate.py
+++ b/homeassistant/components/zha/climate.py
@@ -30,7 +30,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PRECISION_TENTHS, Platform, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -66,7 +66,7 @@ ZHA_TO_HA_HVAC_ACTION = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation sensor from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/cover.py b/homeassistant/components/zha/cover.py
index 0d6be2dbb35..d058f37ff6b 100644
--- a/homeassistant/components/zha/cover.py
+++ b/homeassistant/components/zha/cover.py
@@ -23,7 +23,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -40,7 +40,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation cover from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/device_tracker.py b/homeassistant/components/zha/device_tracker.py
index 7bdfc54c986..c86bb3352b5 100644
--- a/homeassistant/components/zha/device_tracker.py
+++ b/homeassistant/components/zha/device_tracker.py
@@ -10,7 +10,7 @@ from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -23,7 +23,7 @@ from .helpers import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation device tracker from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/entity.py b/homeassistant/components/zha/entity.py
index 499721722fa..e3339661d15 100644
--- a/homeassistant/components/zha/entity.py
+++ b/homeassistant/components/zha/entity.py
@@ -59,6 +59,10 @@ class ZHAEntity(LogMixin, RestoreEntity, Entity):
def name(self) -> str | UndefinedType | None:
"""Return the name of the entity."""
meta = self.entity_data.entity.info_object
+ if meta.primary:
+ self._attr_name = None
+ return super().name
+
original_name = super().name
if original_name not in (UNDEFINED, None) or meta.fallback_name is None:
diff --git a/homeassistant/components/zha/fan.py b/homeassistant/components/zha/fan.py
index 73b23e97387..81206f8819e 100644
--- a/homeassistant/components/zha/fan.py
+++ b/homeassistant/components/zha/fan.py
@@ -12,7 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -27,7 +27,7 @@ from .helpers import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation fan from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/helpers.py b/homeassistant/components/zha/helpers.py
index c31627d3dc3..700e2833705 100644
--- a/homeassistant/components/zha/helpers.py
+++ b/homeassistant/components/zha/helpers.py
@@ -11,6 +11,7 @@ import enum
import functools
import itertools
import logging
+import queue
import re
import time
from types import MappingProxyType
@@ -111,9 +112,10 @@ from homeassistant.helpers import (
device_registry as dr,
entity_registry as er,
)
-from homeassistant.helpers.dispatcher import async_dispatcher_send
+from homeassistant.helpers.dispatcher import async_dispatcher_send, dispatcher_send
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType
+from homeassistant.util.logging import HomeAssistantQueueHandler
from .const import (
ATTR_ACTIVE_COORDINATOR,
@@ -505,7 +507,14 @@ class ZHAGatewayProxy(EventBase):
DEBUG_LEVEL_CURRENT: async_capture_log_levels(),
}
self.debug_enabled: bool = False
- self._log_relay_handler: LogRelayHandler = LogRelayHandler(hass, self)
+
+ log_relay_handler: LogRelayHandler = LogRelayHandler(hass, self)
+ log_simple_queue: queue.SimpleQueue[logging.Handler] = queue.SimpleQueue()
+ self._log_queue_handler = HomeAssistantQueueHandler(log_simple_queue)
+ self._log_queue_handler.listener = logging.handlers.QueueListener(
+ log_simple_queue, log_relay_handler
+ )
+
self._unsubs: list[Callable[[], None]] = []
self._unsubs.append(self.gateway.on_all_events(self._handle_event_protocol))
self._reload_task: asyncio.Task | None = None
@@ -736,10 +745,13 @@ class ZHAGatewayProxy(EventBase):
self._log_levels[DEBUG_LEVEL_CURRENT] = async_capture_log_levels()
if filterer:
- self._log_relay_handler.addFilter(filterer)
+ self._log_queue_handler.addFilter(filterer)
+
+ if self._log_queue_handler.listener:
+ self._log_queue_handler.listener.start()
for logger_name in DEBUG_RELAY_LOGGERS:
- logging.getLogger(logger_name).addHandler(self._log_relay_handler)
+ logging.getLogger(logger_name).addHandler(self._log_queue_handler)
self.debug_enabled = True
@@ -749,9 +761,14 @@ class ZHAGatewayProxy(EventBase):
async_set_logger_levels(self._log_levels[DEBUG_LEVEL_ORIGINAL])
self._log_levels[DEBUG_LEVEL_CURRENT] = async_capture_log_levels()
for logger_name in DEBUG_RELAY_LOGGERS:
- logging.getLogger(logger_name).removeHandler(self._log_relay_handler)
+ logging.getLogger(logger_name).removeHandler(self._log_queue_handler)
+
+ if self._log_queue_handler.listener:
+ self._log_queue_handler.listener.stop()
+
if filterer:
- self._log_relay_handler.removeFilter(filterer)
+ self._log_queue_handler.removeFilter(filterer)
+
self.debug_enabled = False
async def shutdown(self) -> None:
@@ -978,7 +995,7 @@ class LogRelayHandler(logging.Handler):
entry = LogEntry(
record, self.paths_re, figure_out_source=record.levelno >= logging.WARNING
)
- async_dispatcher_send(
+ dispatcher_send(
self.hass,
ZHA_GW_MSG,
{ATTR_TYPE: ZHA_GW_MSG_LOG_OUTPUT, ZHA_GW_MSG_LOG_ENTRY: entry.to_dict()},
diff --git a/homeassistant/components/zha/light.py b/homeassistant/components/zha/light.py
index 2f5d9e9e4c9..a2fb61dc019 100644
--- a/homeassistant/components/zha/light.py
+++ b/homeassistant/components/zha/light.py
@@ -28,7 +28,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_ON, Platform
from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .entity import ZHAEntity
@@ -59,7 +59,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation light from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/lock.py b/homeassistant/components/zha/lock.py
index ebac03eb7b8..dc27ec7a6fa 100644
--- a/homeassistant/components/zha/lock.py
+++ b/homeassistant/components/zha/lock.py
@@ -12,7 +12,7 @@ from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_platform import (
- AddEntitiesCallback,
+ AddConfigEntryEntitiesCallback,
async_get_current_platform,
)
@@ -33,7 +33,7 @@ SERVICE_CLEAR_LOCK_USER_CODE = "clear_lock_user_code"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation Door Lock from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/manifest.json b/homeassistant/components/zha/manifest.json
index 821159afb22..6ed8b253e75 100644
--- a/homeassistant/components/zha/manifest.json
+++ b/homeassistant/components/zha/manifest.json
@@ -21,7 +21,7 @@
"zha",
"universal_silabs_flasher"
],
- "requirements": ["zha==0.0.48"],
+ "requirements": ["zha==0.0.53"],
"usb": [
{
"vid": "10C4",
diff --git a/homeassistant/components/zha/number.py b/homeassistant/components/zha/number.py
index 263f5262994..567e2a5b37a 100644
--- a/homeassistant/components/zha/number.py
+++ b/homeassistant/components/zha/number.py
@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UndefinedType
from .entity import ZHAEntity
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation Analog Output from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/select.py b/homeassistant/components/zha/select.py
index fdb47b550fe..4a38738b7dd 100644
--- a/homeassistant/components/zha/select.py
+++ b/homeassistant/components/zha/select.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN, Platform
from homeassistant.core import HomeAssistant, State, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -28,7 +28,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation siren from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/sensor.py b/homeassistant/components/zha/sensor.py
index 0506496f447..a8383857e57 100644
--- a/homeassistant/components/zha/sensor.py
+++ b/homeassistant/components/zha/sensor.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import StateType
from .entity import ZHAEntity
@@ -80,7 +80,7 @@ _EXTRA_STATE_ATTRIBUTES: set[str] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation sensor from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/siren.py b/homeassistant/components/zha/siren.py
index 9d876d9ca4d..0c8b447cb37 100644
--- a/homeassistant/components/zha/siren.py
+++ b/homeassistant/components/zha/siren.py
@@ -26,7 +26,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -41,7 +41,7 @@ from .helpers import (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation siren from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/strings.json b/homeassistant/components/zha/strings.json
index c73a0989faa..23bb9ae051e 100644
--- a/homeassistant/components/zha/strings.json
+++ b/homeassistant/components/zha/strings.json
@@ -3,11 +3,11 @@
"flow_title": "{name}",
"step": {
"choose_serial_port": {
- "title": "Select a Serial Port",
+ "title": "Select a serial port",
+ "description": "Select the serial port for your Zigbee radio",
"data": {
- "path": "Serial Device Path"
- },
- "description": "Select the serial port for your Zigbee radio"
+ "path": "Serial device path"
+ }
},
"confirm": {
"description": "Do you want to set up {name}?"
@@ -16,14 +16,14 @@
"description": "Do you want to set up {name}?"
},
"manual_pick_radio_type": {
+ "title": "Select a radio type",
+ "description": "Pick your Zigbee radio type",
"data": {
- "radio_type": "Radio Type"
- },
- "title": "[%key:component::zha::config::step::manual_pick_radio_type::data::radio_type%]",
- "description": "Pick your Zigbee radio type"
+ "radio_type": "Radio type"
+ }
},
"manual_port_config": {
- "title": "Serial Port Settings",
+ "title": "Serial port settings",
"description": "Enter the serial port settings",
"data": {
"path": "Serial device path",
@@ -36,7 +36,7 @@
"description": "The radio you are using ({name}) is not recommended and support for it may be removed in the future. Please see the Zigbee Home Automation integration's documentation for [a list of recommended adapters]({docs_recommended_adapters_url})."
},
"choose_formation_strategy": {
- "title": "Network Formation",
+ "title": "Network formation",
"description": "Choose the network settings for your radio.",
"menu_options": {
"form_new_network": "Erase network settings and create a new network",
@@ -47,21 +47,21 @@
}
},
"choose_automatic_backup": {
- "title": "Restore Automatic Backup",
+ "title": "Restore automatic backup",
"description": "Restore your network settings from an automatic backup",
"data": {
"choose_automatic_backup": "Choose an automatic backup"
}
},
"upload_manual_backup": {
- "title": "Upload a Manual Backup",
+ "title": "Upload a manual backup",
"description": "Restore your network settings from an uploaded backup JSON file. You can download one from a different ZHA installation from **Network Settings**, or use a Zigbee2MQTT `coordinator_backup.json` file.",
"data": {
"uploaded_backup_file": "Upload a file"
}
},
"maybe_confirm_ezsp_restore": {
- "title": "Overwrite Radio IEEE Address",
+ "title": "Overwrite radio IEEE address",
"description": "Your backup has a different IEEE address than your radio. For your network to function properly, the IEEE address of your radio should also be changed.\n\nThis is a permanent operation.",
"data": {
"overwrite_coordinator_ieee": "Permanently replace the radio IEEE address"
@@ -74,10 +74,10 @@
},
"abort": {
"single_instance_allowed": "[%key:common::config_flow::abort::single_instance_allowed%]",
- "not_zha_device": "This device is not a zha device",
- "usb_probe_failed": "Failed to probe the usb device",
+ "not_zha_device": "This device is not a ZHA device",
+ "usb_probe_failed": "Failed to probe the USB device",
"wrong_firmware_installed": "Your device is running the wrong firmware and cannot be used with ZHA until the correct firmware is installed. [A repair has been created]({repair_url}) with more information and instructions for how to fix this.",
- "invalid_zeroconf_data": "The coordinator has invalid zeroconf service info and cannot be identified by ZHA"
+ "invalid_zeroconf_data": "The coordinator has invalid Zeroconf service info and cannot be identified by ZHA"
}
},
"options": {
@@ -176,7 +176,7 @@
},
"config_panel": {
"zha_options": {
- "title": "Global Options",
+ "title": "Global options",
"enhanced_light_transition": "Enable enhanced light color/temperature transition from an off-state",
"light_transitioning_flag": "Enable enhanced brightness slider during light transition",
"group_members_assume_state": "Group members assume state of group",
@@ -187,7 +187,7 @@
"consider_unavailable_battery": "Consider battery powered devices unavailable after (seconds)"
},
"zha_alarm_options": {
- "title": "Alarm Control Panel Options",
+ "title": "Alarm control panel options",
"alarm_master_code": "Master code for the alarm control panel(s)",
"alarm_failed_tries": "The number of consecutive failed code entries to trigger an alarm",
"alarm_arm_requires_code": "Code required for arming actions"
@@ -274,15 +274,15 @@
},
"source_ieee": {
"name": "Source IEEE",
- "description": "IEEE address of the joining device (must be used with the install code)."
+ "description": "IEEE address of the joining device (must be combined with the 'Install code' field)."
},
"install_code": {
"name": "Install code",
- "description": "Install code of the joining device (must be used with the source_ieee)."
+ "description": "Install code of the joining device (must be combined with the 'Source IEEE' field)."
},
"qr_code": {
"name": "QR code",
- "description": "Value of the QR install code (different between vendors)."
+ "description": "Provides both the IEEE address and the install code of the joining device (different between vendors)."
}
}
},
@@ -307,7 +307,7 @@
}
},
"set_zigbee_cluster_attribute": {
- "name": "Set zigbee cluster attribute",
+ "name": "Set Zigbee cluster attribute",
"description": "Sets an attribute value for the specified cluster on the specified entity.",
"fields": {
"ieee": {
@@ -323,7 +323,7 @@
"description": "ZCL cluster to retrieve attributes for."
},
"cluster_type": {
- "name": "Cluster Type",
+ "name": "Cluster type",
"description": "Type of the cluster."
},
"attribute": {
@@ -341,7 +341,7 @@
}
},
"issue_zigbee_cluster_command": {
- "name": "Issue zigbee cluster command",
+ "name": "Issue Zigbee cluster command",
"description": "Issues a command on the specified cluster on the specified entity.",
"fields": {
"ieee": {
@@ -383,8 +383,8 @@
}
},
"issue_zigbee_group_command": {
- "name": "Issue zigbee group command",
- "description": "Issue command on the specified cluster on the specified group.",
+ "name": "Issue Zigbee group command",
+ "description": "Issues a command on the specified cluster on the specified group.",
"fields": {
"group": {
"name": "Group",
@@ -1044,6 +1044,63 @@
},
"valve_duration": {
"name": "Irrigation duration"
+ },
+ "down_movement": {
+ "name": "Down movement"
+ },
+ "sustain_time": {
+ "name": "Sustain time"
+ },
+ "up_movement": {
+ "name": "Up movement"
+ },
+ "large_motion_detection_sensitivity": {
+ "name": "Motion detection sensitivity"
+ },
+ "large_motion_detection_distance": {
+ "name": "Motion detection distance"
+ },
+ "medium_motion_detection_distance": {
+ "name": "Medium motion detection distance"
+ },
+ "medium_motion_detection_sensitivity": {
+ "name": "Medium motion detection sensitivity"
+ },
+ "small_motion_detection_distance": {
+ "name": "Small motion detection distance"
+ },
+ "small_motion_detection_sensitivity": {
+ "name": "Small motion detection sensitivity"
+ },
+ "static_detection_sensitivity": {
+ "name": "Static detection sensitivity"
+ },
+ "static_detection_distance": {
+ "name": "Static detection distance"
+ },
+ "motion_detection_sensitivity": {
+ "name": "Motion detection sensitivity"
+ },
+ "holiday_temperature": {
+ "name": "Holiday temperature"
+ },
+ "boost_time": {
+ "name": "Boost time"
+ },
+ "antifrost_temperature": {
+ "name": "Antifrost temperature"
+ },
+ "eco_temperature": {
+ "name": "Eco temperature"
+ },
+ "comfort_temperature": {
+ "name": "Comfort temperature"
+ },
+ "valve_state_auto_shutdown": {
+ "name": "Valve state auto shutdown"
+ },
+ "shutdown_timer": {
+ "name": "Shutdown timer"
}
},
"select": {
@@ -1087,10 +1144,10 @@
"name": "Switch type"
},
"led_scaling_mode": {
- "name": "Led scaling mode"
+ "name": "LED scaling mode"
},
"smart_fan_led_display_levels": {
- "name": "Smart fan led display levels"
+ "name": "Smart fan LED display levels"
},
"increased_non_neutral_output": {
"name": "Non neutral output"
@@ -1235,6 +1292,33 @@
},
"eco_mode": {
"name": "Eco mode"
+ },
+ "mode": {
+ "name": "Mode"
+ },
+ "reverse": {
+ "name": "Reverse"
+ },
+ "motion_state": {
+ "name": "Motion state"
+ },
+ "motion_detection_mode": {
+ "name": "Motion detection mode"
+ },
+ "screen_orientation": {
+ "name": "Screen orientation"
+ },
+ "motor_thrust": {
+ "name": "Motor thrust"
+ },
+ "display_brightness": {
+ "name": "Display brightness"
+ },
+ "display_orientation": {
+ "name": "Display orientation"
+ },
+ "hysteresis_mode": {
+ "name": "Hysteresis mode"
}
},
"sensor": {
@@ -1561,6 +1645,27 @@
},
"error_status": {
"name": "Error status"
+ },
+ "brightness_level": {
+ "name": "Brightness level"
+ },
+ "average_light_intensity_20mins": {
+ "name": "Average light intensity last 20 min"
+ },
+ "todays_max_light_intensity": {
+ "name": "Today's max light intensity"
+ },
+ "fault_code": {
+ "name": "Fault code"
+ },
+ "water_flow": {
+ "name": "Water flow"
+ },
+ "remaining_watering_time": {
+ "name": "Remaining watering time"
+ },
+ "last_watering_duration": {
+ "name": "Last watering duration"
}
},
"switch": {
@@ -1746,6 +1851,30 @@
},
"total_flow_reset_switch": {
"name": "Total flow reset switch"
+ },
+ "touch_control": {
+ "name": "Touch control"
+ },
+ "sound_enabled": {
+ "name": "Sound enabled"
+ },
+ "invert_relay": {
+ "name": "Invert relay"
+ },
+ "boost_heating": {
+ "name": "Boost heating"
+ },
+ "holiday_mode": {
+ "name": "Holiday mode"
+ },
+ "heating_stop": {
+ "name": "Heating stop"
+ },
+ "schedule_mode": {
+ "name": "Schedule mode"
+ },
+ "auto_clean": {
+ "name": "Auto clean"
}
}
}
diff --git a/homeassistant/components/zha/switch.py b/homeassistant/components/zha/switch.py
index cb0268f98e0..dc150e2407d 100644
--- a/homeassistant/components/zha/switch.py
+++ b/homeassistant/components/zha/switch.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .entity import ZHAEntity
from .helpers import (
@@ -27,7 +27,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation switch from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zha/update.py b/homeassistant/components/zha/update.py
index 2f540da5ea7..062581fd259 100644
--- a/homeassistant/components/zha/update.py
+++ b/homeassistant/components/zha/update.py
@@ -19,7 +19,7 @@ from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
@@ -52,7 +52,7 @@ OTA_MESSAGE_RELIABILITY = (
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the Zigbee Home Automation update from config entry."""
zha_data = get_zha_data(hass)
diff --git a/homeassistant/components/zodiac/sensor.py b/homeassistant/components/zodiac/sensor.py
index d7cac07a322..41f200366ae 100644
--- a/homeassistant/components/zodiac/sensor.py
+++ b/homeassistant/components/zodiac/sensor.py
@@ -6,7 +6,7 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import as_local, utcnow
from .const import (
@@ -150,7 +150,7 @@ ZODIAC_BY_DATE = (
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize the entries."""
diff --git a/homeassistant/components/zone/__init__.py b/homeassistant/components/zone/__init__.py
index 1c43a79e10e..813425c95f2 100644
--- a/homeassistant/components/zone/__init__.py
+++ b/homeassistant/components/zone/__init__.py
@@ -363,7 +363,7 @@ class Zone(collection.CollectionEntity):
"""Return entity instance initialized from storage."""
zone = cls(config)
zone.editable = True
- zone._generate_attrs() # noqa: SLF001
+ zone._generate_attrs()
return zone
@classmethod
@@ -371,7 +371,7 @@ class Zone(collection.CollectionEntity):
"""Return entity instance initialized from yaml."""
zone = cls(config)
zone.editable = False
- zone._generate_attrs() # noqa: SLF001
+ zone._generate_attrs()
return zone
@property
diff --git a/homeassistant/components/zwave_js/__init__.py b/homeassistant/components/zwave_js/__init__.py
index c8503b1f4c6..a7b8f9ed665 100644
--- a/homeassistant/components/zwave_js/__init__.py
+++ b/homeassistant/components/zwave_js/__init__.py
@@ -4,7 +4,7 @@ from __future__ import annotations
import asyncio
from collections import defaultdict
-from contextlib import suppress
+import contextlib
import logging
from typing import Any
@@ -12,7 +12,11 @@ from awesomeversion import AwesomeVersion
import voluptuous as vol
from zwave_js_server.client import Client as ZwaveClient
from zwave_js_server.const import CommandClass, RemoveNodeReason
-from zwave_js_server.exceptions import BaseZwaveJSServerError, InvalidServerVersion
+from zwave_js_server.exceptions import (
+ BaseZwaveJSServerError,
+ InvalidServerVersion,
+ NotConnected,
+)
from zwave_js_server.model.driver import Driver
from zwave_js_server.model.node import Node as ZwaveNode
from zwave_js_server.model.notification import (
@@ -25,7 +29,7 @@ from zwave_js_server.model.value import Value, ValueNotification
from homeassistant.components.hassio import AddonError, AddonManager, AddonState
from homeassistant.components.persistent_notification import async_create
-from homeassistant.config_entries import ConfigEntry
+from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import (
ATTR_DEVICE_ID,
ATTR_DOMAIN,
@@ -36,7 +40,7 @@ from homeassistant.const import (
Platform,
)
from homeassistant.core import Event, HomeAssistant, callback
-from homeassistant.exceptions import ConfigEntryNotReady
+from homeassistant.exceptions import ConfigEntryNotReady, HomeAssistantError
from homeassistant.helpers import (
config_validation as cv,
device_registry as dr,
@@ -130,9 +134,8 @@ from .migrate import async_migrate_discovered_value
from .services import ZWaveServices
CONNECT_TIMEOUT = 10
-DATA_CLIENT_LISTEN_TASK = "client_listen_task"
DATA_DRIVER_EVENTS = "driver_events"
-DATA_START_CLIENT_TASK = "start_client_task"
+DRIVER_READY_TIMEOUT = 60
CONFIG_SCHEMA = vol.Schema(
{
@@ -145,6 +148,24 @@ CONFIG_SCHEMA = vol.Schema(
extra=vol.ALLOW_EXTRA,
)
+PLATFORMS = [
+ Platform.BINARY_SENSOR,
+ Platform.BUTTON,
+ Platform.CLIMATE,
+ Platform.COVER,
+ Platform.EVENT,
+ Platform.FAN,
+ Platform.HUMIDIFIER,
+ Platform.LIGHT,
+ Platform.LOCK,
+ Platform.NUMBER,
+ Platform.SELECT,
+ Platform.SENSOR,
+ Platform.SIREN,
+ Platform.SWITCH,
+ Platform.UPDATE,
+]
+
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the Z-Wave JS component."""
@@ -196,53 +217,99 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
raise ConfigEntryNotReady(f"Failed to connect: {err}") from err
async_delete_issue(hass, DOMAIN, "invalid_server_version")
- LOGGER.info("Connected to Zwave JS Server")
+ LOGGER.debug("Connected to Zwave JS Server")
# Set up websocket API
async_register_api(hass)
- entry.runtime_data = {}
- # Create a task to allow the config entry to be unloaded before the driver is ready.
- # Unloading the config entry is needed if the client listen task errors.
- start_client_task = hass.async_create_task(start_client(hass, entry, client))
- entry.runtime_data[DATA_START_CLIENT_TASK] = start_client_task
+ driver_ready = asyncio.Event()
+ listen_task = entry.async_create_background_task(
+ hass,
+ client_listen(hass, entry, client, driver_ready),
+ f"{DOMAIN}_{entry.title}_client_listen",
+ )
- return True
-
-
-async def start_client(
- hass: HomeAssistant, entry: ConfigEntry, client: ZwaveClient
-) -> None:
- """Start listening with the client."""
- entry.runtime_data[DATA_CLIENT] = client
- driver_events = entry.runtime_data[DATA_DRIVER_EVENTS] = DriverEvents(hass, entry)
+ entry.async_on_unload(client.disconnect)
async def handle_ha_shutdown(event: Event) -> None:
"""Handle HA shutdown."""
- await disconnect_client(hass, entry)
+ await client.disconnect()
- listen_task = asyncio.create_task(
- client_listen(hass, entry, client, driver_events.ready)
- )
- entry.runtime_data[DATA_CLIENT_LISTEN_TASK] = listen_task
entry.async_on_unload(
hass.bus.async_listen(EVENT_HOMEASSISTANT_STOP, handle_ha_shutdown)
)
- try:
- await driver_events.ready.wait()
- except asyncio.CancelledError:
- LOGGER.debug("Cancelling start client")
- return
-
- LOGGER.info("Connection to Zwave JS Server initialized")
-
- assert client.driver
- async_dispatcher_send(
- hass, f"{DOMAIN}_{client.driver.controller.home_id}_connected_to_server"
+ driver_ready_task = entry.async_create_task(
+ hass,
+ driver_ready.wait(),
+ f"{DOMAIN}_{entry.title}_driver_ready",
+ )
+ done, pending = await asyncio.wait(
+ (driver_ready_task, listen_task),
+ return_when=asyncio.FIRST_COMPLETED,
+ timeout=DRIVER_READY_TIMEOUT,
)
- await driver_events.setup(client.driver)
+ if driver_ready_task in pending or listen_task in done:
+ error_message = "Driver ready timed out"
+ listen_error: BaseException | None = None
+ if listen_task.done():
+ listen_error, error_message = _get_listen_task_error(listen_task)
+ else:
+ listen_task.cancel()
+ driver_ready_task.cancel()
+ raise ConfigEntryNotReady(error_message) from listen_error
+
+ LOGGER.debug("Connection to Zwave JS Server initialized")
+
+ entry_runtime_data = entry.runtime_data = {
+ DATA_CLIENT: client,
+ }
+ entry_runtime_data[DATA_DRIVER_EVENTS] = driver_events = DriverEvents(hass, entry)
+
+ driver = client.driver
+ # When the driver is ready we know it's set on the client.
+ assert driver is not None
+
+ await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+
+ with contextlib.suppress(NotConnected):
+ # If the client isn't connected the listen task may have an exception
+ # and we'll handle the clean up below.
+ await driver_events.setup(driver)
+
+ # If the listen task is already failed, we need to raise ConfigEntryNotReady
+ if listen_task.done():
+ listen_error, error_message = _get_listen_task_error(listen_task)
+ await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+ raise ConfigEntryNotReady(error_message) from listen_error
+
+ # Re-attach trigger listeners.
+ # Schedule this call to make sure the config entry is loaded first.
+
+ @callback
+ def on_config_entry_loaded() -> None:
+ """Signal that server connection and driver are ready."""
+ if entry.state is ConfigEntryState.LOADED:
+ async_dispatcher_send(
+ hass,
+ f"{DOMAIN}_{driver.controller.home_id}_connected_to_server",
+ )
+
+ entry.async_on_unload(entry.async_on_state_change(on_config_entry_loaded))
+
+ return True
+
+
+def _get_listen_task_error(
+ listen_task: asyncio.Task,
+) -> tuple[BaseException | None, str]:
+ """Check the listen task for errors."""
+ if listen_error := listen_task.exception():
+ error_message = f"Client listen failed: {listen_error}"
+ else:
+ error_message = "Client connection was closed"
+ return listen_error, error_message
class DriverEvents:
@@ -255,8 +322,6 @@ class DriverEvents:
self.config_entry = entry
self.dev_reg = dr.async_get(hass)
self.hass = hass
- self.platform_setup_tasks: dict[str, asyncio.Task] = {}
- self.ready = asyncio.Event()
# Make sure to not pass self to ControllerEvents until all attributes are set.
self.controller_events = ControllerEvents(hass, self)
@@ -339,16 +404,6 @@ class DriverEvents:
controller.on("identify", self.controller_events.async_on_identify)
)
- async def async_setup_platform(self, platform: Platform) -> None:
- """Set up platform if needed."""
- if platform not in self.platform_setup_tasks:
- self.platform_setup_tasks[platform] = self.hass.async_create_task(
- self.hass.config_entries.async_forward_entry_setups(
- self.config_entry, [platform]
- )
- )
- await self.platform_setup_tasks[platform]
-
class ControllerEvents:
"""Represent controller events.
@@ -380,9 +435,6 @@ class ControllerEvents:
async def async_on_node_added(self, node: ZwaveNode) -> None:
"""Handle node added event."""
- # Every node including the controller will have at least one sensor
- await self.driver_events.async_setup_platform(Platform.SENSOR)
-
# Remove stale entities that may exist from a previous interview when an
# interview is started.
base_unique_id = get_valueless_base_unique_id(self.driver_events.driver, node)
@@ -411,7 +463,6 @@ class ControllerEvents:
)
# Create a ping button for each device
- await self.driver_events.async_setup_platform(Platform.BUTTON)
async_dispatcher_send(
self.hass,
f"{DOMAIN}_{self.config_entry.entry_id}_add_ping_button_entity",
@@ -668,9 +719,6 @@ class NodeEvents:
cc.id == CommandClass.FIRMWARE_UPDATE_MD.value
for cc in node.command_classes
):
- await self.controller_events.driver_events.async_setup_platform(
- Platform.UPDATE
- )
async_dispatcher_send(
self.hass,
f"{DOMAIN}_{self.config_entry.entry_id}_add_firmware_update_entity",
@@ -701,21 +749,19 @@ class NodeEvents:
value_updates_disc_info: dict[str, ZwaveDiscoveryInfo],
) -> None:
"""Handle discovery info and all dependent tasks."""
+ platform = disc_info.platform
# This migration logic was added in 2021.3 to handle a breaking change to
# the value_id format. Some time in the future, this call (as well as the
# helper functions) can be removed.
async_migrate_discovered_value(
self.hass,
self.ent_reg,
- self.controller_events.registered_unique_ids[device.id][disc_info.platform],
+ self.controller_events.registered_unique_ids[device.id][platform],
device,
self.controller_events.driver_events.driver,
disc_info,
)
- platform = disc_info.platform
- await self.controller_events.driver_events.async_setup_platform(platform)
-
LOGGER.debug("Discovered entity: %s", disc_info)
async_dispatcher_send(
self.hass,
@@ -930,63 +976,37 @@ async def client_listen(
driver_ready: asyncio.Event,
) -> None:
"""Listen with the client."""
- should_reload = True
try:
await client.listen(driver_ready)
- except asyncio.CancelledError:
- should_reload = False
except BaseZwaveJSServerError as err:
- LOGGER.error("Failed to listen: %s", err)
- except Exception as err: # noqa: BLE001
+ if entry.state is not ConfigEntryState.LOADED:
+ raise
+ LOGGER.error("Client listen failed: %s", err)
+ except Exception as err:
# We need to guard against unknown exceptions to not crash this task.
LOGGER.exception("Unexpected exception: %s", err)
+ if entry.state is not ConfigEntryState.LOADED:
+ raise
# The entry needs to be reloaded since a new driver state
# will be acquired on reconnect.
# All model instances will be replaced when the new state is acquired.
- if should_reload:
- LOGGER.info("Disconnected from server. Reloading integration")
- hass.async_create_task(hass.config_entries.async_reload(entry.entry_id))
-
-
-async def disconnect_client(hass: HomeAssistant, entry: ConfigEntry) -> None:
- """Disconnect client."""
- client: ZwaveClient = entry.runtime_data[DATA_CLIENT]
- listen_task: asyncio.Task = entry.runtime_data[DATA_CLIENT_LISTEN_TASK]
- start_client_task: asyncio.Task = entry.runtime_data[DATA_START_CLIENT_TASK]
- driver_events: DriverEvents = entry.runtime_data[DATA_DRIVER_EVENTS]
- listen_task.cancel()
- start_client_task.cancel()
- platform_setup_tasks = driver_events.platform_setup_tasks.values()
- for task in platform_setup_tasks:
- task.cancel()
-
- tasks = (listen_task, start_client_task, *platform_setup_tasks)
- await asyncio.gather(*tasks, return_exceptions=True)
- for task in tasks:
- with suppress(asyncio.CancelledError):
- await task
-
- if client.connected:
- await client.disconnect()
- LOGGER.info("Disconnected from Zwave JS Server")
+ if not hass.is_stopping:
+ if entry.state is not ConfigEntryState.LOADED:
+ raise HomeAssistantError("Listen task ended unexpectedly")
+ LOGGER.debug("Disconnected from server. Reloading integration")
+ hass.config_entries.async_schedule_reload(entry.entry_id)
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
- client: ZwaveClient = entry.runtime_data[DATA_CLIENT]
- driver_events: DriverEvents = entry.runtime_data[DATA_DRIVER_EVENTS]
- platforms = [
- platform
- for platform, task in driver_events.platform_setup_tasks.items()
- if not task.cancel()
- ]
- unload_ok = await hass.config_entries.async_unload_platforms(entry, platforms)
+ unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
- if client.connected and client.driver:
- await async_disable_server_logging_if_needed(hass, entry, client.driver)
- if DATA_CLIENT_LISTEN_TASK in entry.runtime_data:
- await disconnect_client(hass, entry)
+ entry_runtime_data = entry.runtime_data
+ client: ZwaveClient = entry_runtime_data[DATA_CLIENT]
+
+ if client.connected and (driver := client.driver):
+ await async_disable_server_logging_if_needed(hass, entry, driver)
if entry.data.get(CONF_USE_ADDON) and entry.disabled_by:
addon_manager: AddonManager = get_addon_manager(hass)
diff --git a/homeassistant/components/zwave_js/api.py b/homeassistant/components/zwave_js/api.py
index 37ce9a51c91..dd698d9ed66 100644
--- a/homeassistant/components/zwave_js/api.py
+++ b/homeassistant/components/zwave_js/api.py
@@ -405,6 +405,7 @@ def async_register_api(hass: HomeAssistant) -> None:
websocket_api.async_register_command(
hass, websocket_try_parse_dsk_from_qr_code_string
)
+ websocket_api.async_register_command(hass, websocket_lookup_device)
websocket_api.async_register_command(hass, websocket_supports_feature)
websocket_api.async_register_command(hass, websocket_stop_inclusion)
websocket_api.async_register_command(hass, websocket_stop_exclusion)
@@ -454,6 +455,8 @@ def async_register_api(hass: HomeAssistant) -> None:
websocket_api.async_register_command(hass, websocket_node_capabilities)
websocket_api.async_register_command(hass, websocket_invoke_cc_api)
websocket_api.async_register_command(hass, websocket_get_integration_settings)
+ websocket_api.async_register_command(hass, websocket_backup_nvm)
+ websocket_api.async_register_command(hass, websocket_restore_nvm)
hass.http.register_view(FirmwareUploadView(dr.async_get(hass)))
@@ -518,6 +521,7 @@ async def websocket_network_status(
"supported_function_types": controller.supported_function_types,
"suc_node_id": controller.suc_node_id,
"supports_timers": controller.supports_timers,
+ "supports_long_range": controller.supports_long_range,
"is_rebuilding_routes": controller.is_rebuilding_routes,
"inclusion_state": controller.inclusion_state,
"rf_region": controller.rf_region,
@@ -805,7 +809,7 @@ async def websocket_add_node(
]
msg[DATA_UNSUBSCRIBE] = unsubs
- if controller.inclusion_state == InclusionState.INCLUDING:
+ if controller.inclusion_state in (InclusionState.INCLUDING, InclusionState.BUSY):
connection.send_result(
msg[ID],
True, # Inclusion is already in progress
@@ -883,6 +887,11 @@ async def websocket_subscribe_s2_inclusion(
) -> None:
"""Subscribe to S2 inclusion initiated by the controller."""
+ @callback
+ def async_cleanup() -> None:
+ for unsub in unsubs:
+ unsub()
+
@callback
def forward_dsk(event: dict) -> None:
connection.send_message(
@@ -891,9 +900,18 @@ async def websocket_subscribe_s2_inclusion(
)
)
- unsub = driver.controller.on("validate dsk and enter pin", forward_dsk)
- connection.subscriptions[msg["id"]] = unsub
- msg[DATA_UNSUBSCRIBE] = [unsub]
+ @callback
+ def handle_requested_grant(event: dict) -> None:
+ """Accept the requested security classes without user interaction."""
+ hass.async_create_task(
+ driver.controller.async_grant_security_classes(event["requested_grant"])
+ )
+
+ connection.subscriptions[msg["id"]] = async_cleanup
+ msg[DATA_UNSUBSCRIBE] = unsubs = [
+ driver.controller.on("grant security classes", handle_requested_grant),
+ driver.controller.on("validate dsk and enter pin", forward_dsk),
+ ]
connection.send_result(msg[ID])
@@ -958,13 +976,7 @@ async def websocket_validate_dsk_and_enter_pin(
{
vol.Required(TYPE): "zwave_js/provision_smart_start_node",
vol.Required(ENTRY_ID): str,
- vol.Exclusive(
- PLANNED_PROVISIONING_ENTRY, "options"
- ): PLANNED_PROVISIONING_ENTRY_SCHEMA,
- vol.Exclusive(
- QR_PROVISIONING_INFORMATION, "options"
- ): QR_PROVISIONING_INFORMATION_SCHEMA,
- vol.Exclusive(QR_CODE_STRING, "options"): QR_CODE_STRING_SCHEMA,
+ vol.Required(QR_PROVISIONING_INFORMATION): QR_PROVISIONING_INFORMATION_SCHEMA,
}
)
@websocket_api.async_response
@@ -979,28 +991,10 @@ async def websocket_provision_smart_start_node(
driver: Driver,
) -> None:
"""Pre-provision a smart start node."""
- try:
- cv.has_at_least_one_key(
- PLANNED_PROVISIONING_ENTRY, QR_PROVISIONING_INFORMATION, QR_CODE_STRING
- )(msg)
- except vol.Invalid as err:
- connection.send_error(
- msg[ID],
- ERR_INVALID_FORMAT,
- err.args[0],
- )
- return
- provisioning_info = (
- msg.get(PLANNED_PROVISIONING_ENTRY)
- or msg.get(QR_PROVISIONING_INFORMATION)
- or msg[QR_CODE_STRING]
- )
+ provisioning_info = msg[QR_PROVISIONING_INFORMATION]
- if (
- QR_PROVISIONING_INFORMATION in msg
- and provisioning_info.version == QRCodeVersion.S2
- ):
+ if provisioning_info.version == QRCodeVersion.S2:
connection.send_error(
msg[ID],
ERR_INVALID_FORMAT,
@@ -1121,6 +1115,41 @@ async def websocket_try_parse_dsk_from_qr_code_string(
)
+@websocket_api.require_admin
+@websocket_api.websocket_command(
+ {
+ vol.Required(TYPE): "zwave_js/lookup_device",
+ vol.Required(ENTRY_ID): str,
+ vol.Required(MANUFACTURER_ID): int,
+ vol.Required(PRODUCT_TYPE): int,
+ vol.Required(PRODUCT_ID): int,
+ vol.Optional(APPLICATION_VERSION): str,
+ }
+)
+@websocket_api.async_response
+@async_handle_failed_command
+@async_get_entry
+async def websocket_lookup_device(
+ hass: HomeAssistant,
+ connection: ActiveConnection,
+ msg: dict[str, Any],
+ entry: ConfigEntry,
+ client: Client,
+ driver: Driver,
+) -> None:
+ """Look up the definition of a given device in the configuration DB."""
+ device = await driver.config_manager.lookup_device(
+ msg[MANUFACTURER_ID],
+ msg[PRODUCT_TYPE],
+ msg[PRODUCT_ID],
+ msg.get(APPLICATION_VERSION),
+ )
+ if device is None:
+ connection.send_error(msg[ID], ERR_NOT_FOUND, "Device not found")
+ else:
+ connection.send_result(msg[ID], device.to_dict())
+
+
@websocket_api.require_admin
@websocket_api.websocket_command(
{
@@ -2765,3 +2794,126 @@ def websocket_get_integration_settings(
CONF_INSTALLER_MODE: hass.data[DOMAIN].get(CONF_INSTALLER_MODE, False),
},
)
+
+
+@websocket_api.require_admin
+@websocket_api.websocket_command(
+ {
+ vol.Required(TYPE): "zwave_js/backup_nvm",
+ vol.Required(ENTRY_ID): str,
+ }
+)
+@websocket_api.async_response
+@async_handle_failed_command
+@async_get_entry
+async def websocket_backup_nvm(
+ hass: HomeAssistant,
+ connection: ActiveConnection,
+ msg: dict[str, Any],
+ entry: ConfigEntry,
+ client: Client,
+ driver: Driver,
+) -> None:
+ """Backup NVM data."""
+ controller = driver.controller
+
+ @callback
+ def async_cleanup() -> None:
+ """Remove signal listeners."""
+ for unsub in unsubs:
+ unsub()
+
+ @callback
+ def forward_progress(event: dict) -> None:
+ """Forward progress events to websocket."""
+ connection.send_message(
+ websocket_api.event_message(
+ msg[ID],
+ {
+ "event": event["event"],
+ "bytesRead": event["bytesRead"],
+ "total": event["total"],
+ },
+ )
+ )
+
+ # Set up subscription for progress events
+ connection.subscriptions[msg["id"]] = async_cleanup
+ msg[DATA_UNSUBSCRIBE] = unsubs = [
+ controller.on("nvm backup progress", forward_progress),
+ ]
+
+ result = await controller.async_backup_nvm_raw_base64()
+ # Send the finished event with the backup data
+ connection.send_message(
+ websocket_api.event_message(
+ msg[ID],
+ {
+ "event": "finished",
+ "data": result,
+ },
+ )
+ )
+ connection.send_result(msg[ID])
+
+
+@websocket_api.require_admin
+@websocket_api.websocket_command(
+ {
+ vol.Required(TYPE): "zwave_js/restore_nvm",
+ vol.Required(ENTRY_ID): str,
+ vol.Required("data"): str,
+ }
+)
+@websocket_api.async_response
+@async_handle_failed_command
+@async_get_entry
+async def websocket_restore_nvm(
+ hass: HomeAssistant,
+ connection: ActiveConnection,
+ msg: dict[str, Any],
+ entry: ConfigEntry,
+ client: Client,
+ driver: Driver,
+) -> None:
+ """Restore NVM data."""
+ controller = driver.controller
+
+ @callback
+ def async_cleanup() -> None:
+ """Remove signal listeners."""
+ for unsub in unsubs:
+ unsub()
+
+ @callback
+ def forward_progress(event: dict) -> None:
+ """Forward progress events to websocket."""
+ connection.send_message(
+ websocket_api.event_message(
+ msg[ID],
+ {
+ "event": event["event"],
+ "bytesRead": event.get("bytesRead"),
+ "bytesWritten": event.get("bytesWritten"),
+ "total": event["total"],
+ },
+ )
+ )
+
+ # Set up subscription for progress events
+ connection.subscriptions[msg["id"]] = async_cleanup
+ msg[DATA_UNSUBSCRIBE] = unsubs = [
+ controller.on("nvm convert progress", forward_progress),
+ controller.on("nvm restore progress", forward_progress),
+ ]
+
+ await controller.async_restore_nvm_base64(msg["data"])
+ connection.send_message(
+ websocket_api.event_message(
+ msg[ID],
+ {
+ "event": "finished",
+ },
+ )
+ )
+ connection.send_result(msg[ID])
diff --git a/homeassistant/components/zwave_js/binary_sensor.py b/homeassistant/components/zwave_js/binary_sensor.py
index 0f1495fc6e6..d07846c8dcc 100644
--- a/homeassistant/components/zwave_js/binary_sensor.py
+++ b/homeassistant/components/zwave_js/binary_sensor.py
@@ -22,7 +22,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_CLIENT, DOMAIN
from .discovery import ZwaveDiscoveryInfo
@@ -261,7 +261,7 @@ def is_valid_notification_binary_sensor(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave binary sensor from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/button.py b/homeassistant/components/zwave_js/button.py
index 7fd42700a05..f3a1d5af04d 100644
--- a/homeassistant/components/zwave_js/button.py
+++ b/homeassistant/components/zwave_js/button.py
@@ -11,7 +11,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_CLIENT, DOMAIN, LOGGER
from .discovery import ZwaveDiscoveryInfo
@@ -24,7 +24,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave button from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/climate.py b/homeassistant/components/zwave_js/climate.py
index 580694cae11..b27dbdad1a0 100644
--- a/homeassistant/components/zwave_js/climate.py
+++ b/homeassistant/components/zwave_js/climate.py
@@ -35,7 +35,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_TENTHS, UnitOfTemperature
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.unit_conversion import TemperatureConverter
from .const import DATA_CLIENT, DOMAIN
@@ -97,7 +97,7 @@ ATTR_FAN_STATE = "fan_state"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave climate from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/config_flow.py b/homeassistant/components/zwave_js/config_flow.py
index 44adf6a12ab..aed0dd839be 100644
--- a/homeassistant/components/zwave_js/config_flow.py
+++ b/homeassistant/components/zwave_js/config_flow.py
@@ -42,7 +42,6 @@ from homeassistant.helpers.service_info.usb import UsbServiceInfo
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from homeassistant.helpers.typing import VolDictType
-from . import disconnect_client
from .addon import get_addon_manager
from .const import (
ADDON_SLUG,
@@ -861,7 +860,7 @@ class OptionsFlowHandler(BaseZwaveJSFlow, OptionsFlow):
and self.config_entry.state == ConfigEntryState.LOADED
):
# Disconnect integration before restarting add-on.
- await disconnect_client(self.hass, self.config_entry)
+ await self.hass.config_entries.async_unload(self.config_entry.entry_id)
return await self.async_step_start_addon()
diff --git a/homeassistant/components/zwave_js/cover.py b/homeassistant/components/zwave_js/cover.py
index 218c5cc82fe..dc44f46a3ce 100644
--- a/homeassistant/components/zwave_js/cover.py
+++ b/homeassistant/components/zwave_js/cover.py
@@ -37,7 +37,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
COVER_POSITION_PROPERTY_KEYS,
@@ -55,7 +55,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave Cover from Config Entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/event.py b/homeassistant/components/zwave_js/event.py
index 8dae66c26ac..66959aa9b75 100644
--- a/homeassistant/components/zwave_js/event.py
+++ b/homeassistant/components/zwave_js/event.py
@@ -10,7 +10,7 @@ from homeassistant.components.event import DOMAIN as EVENT_DOMAIN, EventEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_VALUE, DATA_CLIENT, DOMAIN
from .discovery import ZwaveDiscoveryInfo
@@ -22,7 +22,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave Event entity from Config Entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/fan.py b/homeassistant/components/zwave_js/fan.py
index d83132e4b95..ae36e0afb42 100644
--- a/homeassistant/components/zwave_js/fan.py
+++ b/homeassistant/components/zwave_js/fan.py
@@ -24,7 +24,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.percentage import (
percentage_to_ranged_value,
ranged_value_to_percentage,
@@ -46,7 +46,7 @@ ATTR_FAN_STATE = "fan_state"
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave Fan from Config Entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/helpers.py b/homeassistant/components/zwave_js/helpers.py
index 904a26acc78..8a90ebf6f88 100644
--- a/homeassistant/components/zwave_js/helpers.py
+++ b/homeassistant/components/zwave_js/helpers.py
@@ -187,7 +187,7 @@ async def async_disable_server_logging_if_needed(
old_server_log_level,
)
await driver.async_update_log_config(LogConfig(level=old_server_log_level))
- await driver.client.disable_server_logging()
+ driver.client.disable_server_logging()
LOGGER.info("Zwave-js-server logging is enabled")
diff --git a/homeassistant/components/zwave_js/humidifier.py b/homeassistant/components/zwave_js/humidifier.py
index e883858036b..2b85bd4449f 100644
--- a/homeassistant/components/zwave_js/humidifier.py
+++ b/homeassistant/components/zwave_js/humidifier.py
@@ -26,7 +26,7 @@ from homeassistant.components.humidifier import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_CLIENT, DOMAIN
from .discovery import ZwaveDiscoveryInfo
@@ -70,7 +70,7 @@ DEHUMIDIFIER_ENTITY_DESCRIPTION = ZwaveHumidifierEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave humidifier from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/light.py b/homeassistant/components/zwave_js/light.py
index 0a2ca95a2b0..a610bbcb91e 100644
--- a/homeassistant/components/zwave_js/light.py
+++ b/homeassistant/components/zwave_js/light.py
@@ -41,7 +41,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import color as color_util
from .const import DATA_CLIENT, DOMAIN
@@ -67,7 +67,7 @@ MAX_MIREDS = 370 # 2700K as a safe default
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave Light from Config Entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/lock.py b/homeassistant/components/zwave_js/lock.py
index c14517f4b03..f609084955c 100644
--- a/homeassistant/components/zwave_js/lock.py
+++ b/homeassistant/components/zwave_js/lock.py
@@ -25,7 +25,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv, entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
ATTR_AUTO_RELOCK_TIME,
@@ -62,7 +62,7 @@ UNIT16_SCHEMA = vol.All(vol.Coerce(int), vol.Range(min=0, max=65535))
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave lock from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/manifest.json b/homeassistant/components/zwave_js/manifest.json
index 011776f4556..7e8b473922f 100644
--- a/homeassistant/components/zwave_js/manifest.json
+++ b/homeassistant/components/zwave_js/manifest.json
@@ -9,7 +9,7 @@
"integration_type": "hub",
"iot_class": "local_push",
"loggers": ["zwave_js_server"],
- "requirements": ["pyserial==3.5", "zwave-js-server-python==0.60.0"],
+ "requirements": ["pyserial==3.5", "zwave-js-server-python==0.62.0"],
"usb": [
{
"vid": "0658",
diff --git a/homeassistant/components/zwave_js/number.py b/homeassistant/components/zwave_js/number.py
index 54162488d89..2e2d93bbdbe 100644
--- a/homeassistant/components/zwave_js/number.py
+++ b/homeassistant/components/zwave_js/number.py
@@ -16,7 +16,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ATTR_RESERVED_VALUES, DATA_CLIENT, DOMAIN
from .discovery import ZwaveDiscoveryInfo
@@ -28,7 +28,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave Number entity from Config Entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/select.py b/homeassistant/components/zwave_js/select.py
index 49ad1868005..8a6ccc57c17 100644
--- a/homeassistant/components/zwave_js/select.py
+++ b/homeassistant/components/zwave_js/select.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_CLIENT, DOMAIN
from .discovery import ZwaveDiscoveryInfo
@@ -27,7 +27,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave Select entity from Config Entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/sensor.py b/homeassistant/components/zwave_js/sensor.py
index b259711d21b..4db14d003b1 100644
--- a/homeassistant/components/zwave_js/sensor.py
+++ b/homeassistant/components/zwave_js/sensor.py
@@ -48,7 +48,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UNDEFINED, StateType
from .binary_sensor import is_valid_notification_binary_sensor
@@ -552,7 +552,7 @@ def get_entity_description(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave sensor from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/siren.py b/homeassistant/components/zwave_js/siren.py
index 3a09049def3..f0526171a70 100644
--- a/homeassistant/components/zwave_js/siren.py
+++ b/homeassistant/components/zwave_js/siren.py
@@ -18,7 +18,7 @@ from homeassistant.components.siren import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_CLIENT, DOMAIN
from .discovery import ZwaveDiscoveryInfo
@@ -30,7 +30,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave Siren entity from Config Entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/strings.json b/homeassistant/components/zwave_js/strings.json
index e2d7720189d..8f23fee4447 100644
--- a/homeassistant/components/zwave_js/strings.json
+++ b/homeassistant/components/zwave_js/strings.json
@@ -344,8 +344,8 @@
"name": "[%key:component::zwave_js::services::set_value::fields::area_id::name%]"
},
"broadcast": {
- "description": "Whether command should be broadcast to all devices on the network.",
- "name": "Broadcast?"
+ "description": "Whether the command should be broadcast to all devices on the network.",
+ "name": "Broadcast"
},
"command_class": {
"description": "[%key:component::zwave_js::services::set_value::fields::command_class::description%]",
@@ -434,8 +434,8 @@
"name": "Entities"
},
"refresh_all_values": {
- "description": "Whether to refresh all values (true) or just the primary value (false).",
- "name": "Refresh all values?"
+ "description": "Whether to refresh all values or just the primary value.",
+ "name": "Refresh all values"
}
},
"name": "Refresh values"
@@ -516,8 +516,8 @@
"name": "Auto relock time"
},
"block_to_block": {
- "description": "Enable block-to-block functionality.",
- "name": "Block to block"
+ "description": "Whether the lock should run the motor until it hits resistance.",
+ "name": "Block to Block"
},
"hold_and_release_time": {
"description": "Duration in seconds the latch stays retracted.",
@@ -529,11 +529,11 @@
},
"operation_type": {
"description": "The operation type of the lock.",
- "name": "Operation Type"
+ "name": "Operation type"
},
"twist_assist": {
- "description": "Enable Twist Assist.",
- "name": "Twist assist"
+ "description": "Whether the motor should help in locking and unlocking.",
+ "name": "Twist Assist"
}
},
"name": "Set lock configuration"
@@ -592,8 +592,8 @@
"name": "[%key:component::zwave_js::services::set_config_parameter::fields::value::name%]"
},
"wait_for_result": {
- "description": "Whether or not to wait for a response from the node. If not included in the payload, the integration will decide whether to wait or not. If set to `true`, note that the action can take a while if setting a value on an asleep battery device.",
- "name": "Wait for result?"
+ "description": "Whether to wait for a response from the node. If not included in the payload, the integration will decide whether to wait or not. If enabled, the action can take a while if setting a value on an asleep battery device.",
+ "name": "Wait for result"
}
},
"name": "Set a value (advanced)"
diff --git a/homeassistant/components/zwave_js/switch.py b/homeassistant/components/zwave_js/switch.py
index ef769209b31..2ff80d8505e 100644
--- a/homeassistant/components/zwave_js/switch.py
+++ b/homeassistant/components/zwave_js/switch.py
@@ -16,7 +16,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DATA_CLIENT, DOMAIN
from .discovery import ZwaveDiscoveryInfo
@@ -28,7 +28,7 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave sensor from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_js/update.py b/homeassistant/components/zwave_js/update.py
index d060abe007d..985c4a86813 100644
--- a/homeassistant/components/zwave_js/update.py
+++ b/homeassistant/components/zwave_js/update.py
@@ -32,7 +32,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import CoreState, HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_call_later
from homeassistant.helpers.restore_state import ExtraStoredData
@@ -77,7 +77,7 @@ class ZWaveNodeFirmwareUpdateExtraStoredData(ExtraStoredData):
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up Z-Wave update entity from config entry."""
client: ZwaveClient = config_entry.runtime_data[DATA_CLIENT]
diff --git a/homeassistant/components/zwave_me/binary_sensor.py b/homeassistant/components/zwave_me/binary_sensor.py
index d121c17770b..8563ef76ce1 100644
--- a/homeassistant/components/zwave_me/binary_sensor.py
+++ b/homeassistant/components/zwave_me/binary_sensor.py
@@ -12,7 +12,7 @@ from homeassistant.components.binary_sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ZWaveMeController
from .const import DOMAIN, ZWaveMePlatform
@@ -33,7 +33,7 @@ DEVICE_NAME = ZWaveMePlatform.BINARY_SENSOR
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the binary sensor platform."""
diff --git a/homeassistant/components/zwave_me/button.py b/homeassistant/components/zwave_me/button.py
index 50ddf01aeab..27d95a14199 100644
--- a/homeassistant/components/zwave_me/button.py
+++ b/homeassistant/components/zwave_me/button.py
@@ -4,7 +4,7 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -15,7 +15,7 @@ DEVICE_NAME = ZWaveMePlatform.BUTTON
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the number platform."""
diff --git a/homeassistant/components/zwave_me/climate.py b/homeassistant/components/zwave_me/climate.py
index b8eed88b505..d54cc6a9310 100644
--- a/homeassistant/components/zwave_me/climate.py
+++ b/homeassistant/components/zwave_me/climate.py
@@ -15,7 +15,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -28,7 +28,7 @@ DEVICE_NAME = ZWaveMePlatform.CLIMATE
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the climate platform."""
diff --git a/homeassistant/components/zwave_me/cover.py b/homeassistant/components/zwave_me/cover.py
index c9359402c01..3ae8ec894e1 100644
--- a/homeassistant/components/zwave_me/cover.py
+++ b/homeassistant/components/zwave_me/cover.py
@@ -12,7 +12,7 @@ from homeassistant.components.cover import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -23,7 +23,7 @@ DEVICE_NAME = ZWaveMePlatform.COVER
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the cover platform."""
diff --git a/homeassistant/components/zwave_me/fan.py b/homeassistant/components/zwave_me/fan.py
index bd0feba0dfb..6ab1df618cb 100644
--- a/homeassistant/components/zwave_me/fan.py
+++ b/homeassistant/components/zwave_me/fan.py
@@ -8,7 +8,7 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -19,7 +19,7 @@ DEVICE_NAME = ZWaveMePlatform.FAN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the fan platform."""
diff --git a/homeassistant/components/zwave_me/light.py b/homeassistant/components/zwave_me/light.py
index ef3eca5d389..f8ed397ea25 100644
--- a/homeassistant/components/zwave_me/light.py
+++ b/homeassistant/components/zwave_me/light.py
@@ -15,7 +15,7 @@ from homeassistant.components.light import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ZWaveMeController
from .const import DOMAIN, ZWaveMePlatform
@@ -25,7 +25,7 @@ from .entity import ZWaveMeEntity
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the rgb platform."""
diff --git a/homeassistant/components/zwave_me/lock.py b/homeassistant/components/zwave_me/lock.py
index 0bcc8f092ae..cdc8b6471c1 100644
--- a/homeassistant/components/zwave_me/lock.py
+++ b/homeassistant/components/zwave_me/lock.py
@@ -10,7 +10,7 @@ from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -21,7 +21,7 @@ DEVICE_NAME = ZWaveMePlatform.LOCK
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the lock platform."""
diff --git a/homeassistant/components/zwave_me/number.py b/homeassistant/components/zwave_me/number.py
index 9a98a4f8d00..2d6b88840f4 100644
--- a/homeassistant/components/zwave_me/number.py
+++ b/homeassistant/components/zwave_me/number.py
@@ -4,7 +4,7 @@ from homeassistant.components.number import NumberEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -15,7 +15,7 @@ DEVICE_NAME = ZWaveMePlatform.NUMBER
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the number platform."""
diff --git a/homeassistant/components/zwave_me/sensor.py b/homeassistant/components/zwave_me/sensor.py
index be0b0bae284..fa9ccdfee99 100644
--- a/homeassistant/components/zwave_me/sensor.py
+++ b/homeassistant/components/zwave_me/sensor.py
@@ -26,7 +26,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import ZWaveMeController
from .const import DOMAIN, ZWaveMePlatform
@@ -118,7 +118,7 @@ DEVICE_NAME = ZWaveMePlatform.SENSOR
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the sensor platform."""
diff --git a/homeassistant/components/zwave_me/siren.py b/homeassistant/components/zwave_me/siren.py
index 443b2cc7b37..7bfbf2b2cd4 100644
--- a/homeassistant/components/zwave_me/siren.py
+++ b/homeassistant/components/zwave_me/siren.py
@@ -6,7 +6,7 @@ from homeassistant.components.siren import SirenEntity, SirenEntityFeature
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -17,7 +17,7 @@ DEVICE_NAME = ZWaveMePlatform.SIREN
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the siren platform."""
diff --git a/homeassistant/components/zwave_me/switch.py b/homeassistant/components/zwave_me/switch.py
index 05cf06484e9..26d832ca022 100644
--- a/homeassistant/components/zwave_me/switch.py
+++ b/homeassistant/components/zwave_me/switch.py
@@ -11,7 +11,7 @@ from homeassistant.components.switch import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import DOMAIN, ZWaveMePlatform
from .entity import ZWaveMeEntity
@@ -30,7 +30,7 @@ SWITCH_MAP: dict[str, SwitchEntityDescription] = {
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up the switch platform."""
diff --git a/homeassistant/config_entries.py b/homeassistant/config_entries.py
index 620e4bc8197..9336ead633a 100644
--- a/homeassistant/config_entries.py
+++ b/homeassistant/config_entries.py
@@ -15,6 +15,7 @@ from collections.abc import (
)
from contextvars import ContextVar
from copy import deepcopy
+from dataclasses import dataclass, field
from datetime import datetime
from enum import Enum, StrEnum
import functools
@@ -22,7 +23,7 @@ from functools import cache
import logging
from random import randint
from types import MappingProxyType
-from typing import TYPE_CHECKING, Any, Self, cast
+from typing import TYPE_CHECKING, Any, Self, TypedDict, cast
from async_interrupt import interrupt
from propcache.api import cached_property
@@ -127,7 +128,7 @@ HANDLERS: Registry[str, type[ConfigFlow]] = Registry()
STORAGE_KEY = "core.config_entries"
STORAGE_VERSION = 1
-STORAGE_VERSION_MINOR = 4
+STORAGE_VERSION_MINOR = 5
SAVE_DELAY = 1
@@ -154,6 +155,8 @@ class ConfigEntryState(Enum):
"""An error occurred when trying to unload the entry"""
SETUP_IN_PROGRESS = "setup_in_progress", False
"""The config entry is setting up."""
+ UNLOAD_IN_PROGRESS = "unload_in_progress", False
+ """The config entry is being unloaded."""
_recoverable: bool
@@ -253,6 +256,10 @@ class UnknownEntry(ConfigError):
"""Unknown entry specified."""
+class UnknownSubEntry(ConfigError):
+ """Unknown subentry specified."""
+
+
class OperationNotAllowed(ConfigError):
"""Raised when a config entry operation is not allowed."""
@@ -297,6 +304,7 @@ class ConfigFlowResult(FlowResult[ConfigFlowContext, str], total=False):
minor_version: int
options: Mapping[str, Any]
+ subentries: Iterable[ConfigSubentryData]
version: int
@@ -310,6 +318,61 @@ def _validate_item(*, disabled_by: ConfigEntryDisabler | Any | None = None) -> N
)
+class ConfigSubentryData(TypedDict):
+ """Container for configuration subentry data.
+
+ Returned by integrations, a subentry_id will be assigned automatically.
+ """
+
+ data: Mapping[str, Any]
+ subentry_type: str
+ title: str
+ unique_id: str | None
+
+
+class ConfigSubentryDataWithId(ConfigSubentryData):
+ """Container for configuration subentry data.
+
+ This type is used when loading existing subentries from storage.
+ """
+
+ subentry_id: str
+
+
+class SubentryFlowContext(FlowContext, total=False):
+ """Typed context dict for subentry flow."""
+
+ entry_id: str
+ subentry_id: str
+
+
+class SubentryFlowResult(FlowResult[SubentryFlowContext, tuple[str, str]], total=False):
+ """Typed result dict for subentry flow."""
+
+ unique_id: str | None
+
+
+@dataclass(frozen=True, kw_only=True)
+class ConfigSubentry:
+ """Container for a configuration subentry."""
+
+ data: MappingProxyType[str, Any]
+ subentry_id: str = field(default_factory=ulid_util.ulid_now)
+ subentry_type: str
+ title: str
+ unique_id: str | None
+
+ def as_dict(self) -> ConfigSubentryDataWithId:
+ """Return dictionary version of this subentry."""
+ return {
+ "data": dict(self.data),
+ "subentry_id": self.subentry_id,
+ "subentry_type": self.subentry_type,
+ "title": self.title,
+ "unique_id": self.unique_id,
+ }
+
+
class ConfigEntry[_DataT = Any]:
"""Hold a configuration entry."""
@@ -319,6 +382,7 @@ class ConfigEntry[_DataT = Any]:
data: MappingProxyType[str, Any]
runtime_data: _DataT
options: MappingProxyType[str, Any]
+ subentries: MappingProxyType[str, ConfigSubentry]
unique_id: str | None
state: ConfigEntryState
reason: str | None
@@ -334,9 +398,11 @@ class ConfigEntry[_DataT = Any]:
supports_remove_device: bool | None
_supports_options: bool | None
_supports_reconfigure: bool | None
+ _supported_subentry_types: dict[str, dict[str, bool]] | None
update_listeners: list[UpdateListenerType]
_async_cancel_retry_setup: Callable[[], Any] | None
_on_unload: list[Callable[[], Coroutine[Any, Any, None] | None]] | None
+ _on_state_change: list[CALLBACK_TYPE] | None
setup_lock: asyncio.Lock
_reauth_lock: asyncio.Lock
_tasks: set[asyncio.Future[Any]]
@@ -363,6 +429,7 @@ class ConfigEntry[_DataT = Any]:
pref_disable_polling: bool | None = None,
source: str,
state: ConfigEntryState = ConfigEntryState.NOT_LOADED,
+ subentries_data: Iterable[ConfigSubentryData | ConfigSubentryDataWithId] | None,
title: str,
unique_id: str | None,
version: int,
@@ -388,6 +455,25 @@ class ConfigEntry[_DataT = Any]:
# Entry options
_setter(self, "options", MappingProxyType(options or {}))
+ # Subentries
+ subentries_data = subentries_data or ()
+ subentries = {}
+ for subentry_data in subentries_data:
+ subentry_kwargs = {}
+ if "subentry_id" in subentry_data:
+ # If subentry_data has key "subentry_id", we're loading from storage
+ subentry_kwargs["subentry_id"] = subentry_data["subentry_id"] # type: ignore[typeddict-item]
+ subentry = ConfigSubentry(
+ data=MappingProxyType(subentry_data["data"]),
+ subentry_type=subentry_data["subentry_type"],
+ title=subentry_data["title"],
+ unique_id=subentry_data.get("unique_id"),
+ **subentry_kwargs,
+ )
+ subentries[subentry.subentry_id] = subentry
+
+ _setter(self, "subentries", MappingProxyType(subentries))
+
# Entry system options
if pref_disable_new_entities is None:
pref_disable_new_entities = False
@@ -424,6 +510,9 @@ class ConfigEntry[_DataT = Any]:
# Supports reconfigure
_setter(self, "_supports_reconfigure", None)
+ # Supports subentries
+ _setter(self, "_supported_subentry_types", None)
+
# Listeners to call on update
_setter(self, "update_listeners", [])
@@ -438,6 +527,9 @@ class ConfigEntry[_DataT = Any]:
# Hold list for actions to call on unload.
_setter(self, "_on_unload", None)
+ # Hold list for actions to call on state change.
+ _setter(self, "_on_state_change", None)
+
# Reload lock to prevent conflicting reloads
_setter(self, "setup_lock", asyncio.Lock())
# Reauth lock to prevent concurrent reauth flows
@@ -496,6 +588,28 @@ class ConfigEntry[_DataT = Any]:
)
return self._supports_reconfigure or False
+ @property
+ def supported_subentry_types(self) -> dict[str, dict[str, bool]]:
+ """Return supported subentry types."""
+ if self._supported_subentry_types is None and (
+ handler := HANDLERS.get(self.domain)
+ ):
+ # work out sub entries supported by the handler
+ supported_flows = handler.async_get_supported_subentry_types(self)
+ object.__setattr__(
+ self,
+ "_supported_subentry_types",
+ {
+ subentry_flow_type: {
+ "supports_reconfigure": hasattr(
+ subentry_flow_handler, "async_step_reconfigure"
+ )
+ }
+ for subentry_flow_type, subentry_flow_handler in supported_flows.items()
+ },
+ )
+ return self._supported_subentry_types or {}
+
def clear_state_cache(self) -> None:
"""Clear cached properties that are included in as_json_fragment."""
self.__dict__.pop("as_json_fragment", None)
@@ -515,12 +629,14 @@ class ConfigEntry[_DataT = Any]:
"supports_remove_device": self.supports_remove_device or False,
"supports_unload": self.supports_unload or False,
"supports_reconfigure": self.supports_reconfigure,
+ "supported_subentry_types": self.supported_subentry_types,
"pref_disable_new_entities": self.pref_disable_new_entities,
"pref_disable_polling": self.pref_disable_polling,
"disabled_by": self.disabled_by,
"reason": self.reason,
"error_reason_translation_key": self.error_reason_translation_key,
"error_reason_translation_placeholders": self.error_reason_translation_placeholders,
+ "num_subentries": len(self.subentries),
}
return json_fragment(json_bytes(json_repr))
@@ -845,18 +961,25 @@ class ConfigEntry[_DataT = Any]:
)
return False
+ if domain_is_integration:
+ self._async_set_state(hass, ConfigEntryState.UNLOAD_IN_PROGRESS, None)
try:
result = await component.async_unload_entry(hass, self)
assert isinstance(result, bool)
- # Only adjust state if we unloaded the component
- if domain_is_integration and result:
- await self._async_process_on_unload(hass)
- if hasattr(self, "runtime_data"):
- object.__delattr__(self, "runtime_data")
+ # Only do side effects if we unloaded the integration
+ if domain_is_integration:
+ if result:
+ await self._async_process_on_unload(hass)
+ if hasattr(self, "runtime_data"):
+ object.__delattr__(self, "runtime_data")
- self._async_set_state(hass, ConfigEntryState.NOT_LOADED, None)
+ self._async_set_state(hass, ConfigEntryState.NOT_LOADED, None)
+ else:
+ self._async_set_state(
+ hass, ConfigEntryState.FAILED_UNLOAD, "Unload failed"
+ )
except Exception as exc:
_LOGGER.exception(
@@ -939,6 +1062,8 @@ class ConfigEntry[_DataT = Any]:
hass, SIGNAL_CONFIG_ENTRY_CHANGED, ConfigEntryChange.UPDATED, self
)
+ self._async_process_on_state_change()
+
async def async_migrate(self, hass: HomeAssistant) -> bool:
"""Migrate an entry.
@@ -1012,6 +1137,7 @@ class ConfigEntry[_DataT = Any]:
"pref_disable_new_entities": self.pref_disable_new_entities,
"pref_disable_polling": self.pref_disable_polling,
"source": self.source,
+ "subentries": [subentry.as_dict() for subentry in self.subentries.values()],
"title": self.title,
"unique_id": self.unique_id,
"version": self.version,
@@ -1052,6 +1178,28 @@ class ConfigEntry[_DataT = Any]:
task,
)
+ @callback
+ def async_on_state_change(self, func: CALLBACK_TYPE) -> CALLBACK_TYPE:
+ """Add a function to call when a config entry changes its state."""
+ if self._on_state_change is None:
+ self._on_state_change = []
+ self._on_state_change.append(func)
+ return lambda: cast(list, self._on_state_change).remove(func)
+
+ def _async_process_on_state_change(self) -> None:
+ """Process the on_state_change callbacks and wait for pending tasks."""
+ if self._on_state_change is None:
+ return
+ for func in self._on_state_change:
+ try:
+ func()
+ except Exception:
+ _LOGGER.exception(
+ "Error calling on_state_change callback for %s (%s)",
+ self.title,
+ self.domain,
+ )
+
@callback
def async_start_reauth(
self,
@@ -1480,6 +1628,23 @@ class ConfigEntriesFlowManager(
result["handler"], flow.unique_id
)
+ if existing_entry is not None and flow.handler != "mobile_app":
+ # This causes the old entry to be removed and replaced, when the flow
+ # should instead be aborted.
+ # In case of manual flows, integrations should implement options, reauth,
+ # reconfigure to allow the user to change settings.
+ # In case of non user visible flows, the integration should optionally
+ # update the existing entry before aborting.
+ # see https://developers.home-assistant.io/blog/2025/03/01/config-flow-unique-id/
+ report_usage(
+ "creates a config entry when another entry with the same unique ID "
+ "exists",
+ core_behavior=ReportBehavior.LOG,
+ core_integration_behavior=ReportBehavior.LOG,
+ custom_integration_behavior=ReportBehavior.LOG,
+ integration_domain=flow.handler,
+ )
+
# Unload the entry before setting up the new one.
if existing_entry is not None and existing_entry.state.recoverable:
await self.config_entries.async_unload(existing_entry.entry_id)
@@ -1497,6 +1662,7 @@ class ConfigEntriesFlowManager(
minor_version=result["minor_version"],
options=result["options"],
source=flow.context["source"],
+ subentries_data=result["subentries"],
title=result["title"],
unique_id=flow.unique_id,
version=result["version"],
@@ -1787,6 +1953,11 @@ class ConfigEntryStore(storage.Store[dict[str, list[dict[str, Any]]]]):
for entry in data["entries"]:
entry["discovery_keys"] = {}
+ if old_minor_version < 5:
+ # Version 1.4 adds config subentries
+ for entry in data["entries"]:
+ entry.setdefault("subentries", entry.get("subentries", {}))
+
if old_major_version > 1:
raise NotImplementedError
return data
@@ -1803,6 +1974,7 @@ class ConfigEntries:
self.hass = hass
self.flow = ConfigEntriesFlowManager(hass, self, hass_config)
self.options = OptionsFlowManager(hass)
+ self.subentries = ConfigSubentryFlowManager(hass)
self._hass_config = hass_config
self._entries = ConfigEntryItems(hass)
self._store = ConfigEntryStore(hass)
@@ -1834,7 +2006,7 @@ class ConfigEntries:
Raises UnknownEntry if entry is not found.
"""
if (entry := self.async_get_entry(entry_id)) is None:
- raise UnknownEntry
+ raise UnknownEntry(entry_id)
return entry
@callback
@@ -1934,9 +2106,9 @@ class ConfigEntries:
else:
unload_success = await self.async_unload(entry_id, _lock=False)
+ del self._entries[entry.entry_id]
await entry.async_remove(self.hass)
- del self._entries[entry.entry_id]
self.async_update_issues()
self._async_schedule_save()
@@ -2005,6 +2177,7 @@ class ConfigEntries:
pref_disable_new_entities=entry["pref_disable_new_entities"],
pref_disable_polling=entry["pref_disable_polling"],
source=entry["source"],
+ subentries_data=entry["subentries"],
title=entry["title"],
unique_id=entry["unique_id"],
version=entry["version"],
@@ -2164,6 +2337,44 @@ class ConfigEntries:
If the entry was changed, the update_listeners are
fired and this function returns True
+ If the entry was not changed, the update_listeners are
+ not fired and this function returns False
+ """
+ return self._async_update_entry(
+ entry,
+ data=data,
+ discovery_keys=discovery_keys,
+ minor_version=minor_version,
+ options=options,
+ pref_disable_new_entities=pref_disable_new_entities,
+ pref_disable_polling=pref_disable_polling,
+ title=title,
+ unique_id=unique_id,
+ version=version,
+ )
+
+ @callback
+ def _async_update_entry(
+ self,
+ entry: ConfigEntry,
+ *,
+ data: Mapping[str, Any] | UndefinedType = UNDEFINED,
+ discovery_keys: MappingProxyType[str, tuple[DiscoveryKey, ...]]
+ | UndefinedType = UNDEFINED,
+ minor_version: int | UndefinedType = UNDEFINED,
+ options: Mapping[str, Any] | UndefinedType = UNDEFINED,
+ pref_disable_new_entities: bool | UndefinedType = UNDEFINED,
+ pref_disable_polling: bool | UndefinedType = UNDEFINED,
+ subentries: dict[str, ConfigSubentry] | UndefinedType = UNDEFINED,
+ title: str | UndefinedType = UNDEFINED,
+ unique_id: str | None | UndefinedType = UNDEFINED,
+ version: int | UndefinedType = UNDEFINED,
+ ) -> bool:
+ """Update a config entry.
+
+ If the entry was changed, the update_listeners are
+ fired and this function returns True
+
If the entry was not changed, the update_listeners are
not fired and this function returns False
"""
@@ -2226,11 +2437,21 @@ class ConfigEntries:
changed = True
_setter(entry, "options", MappingProxyType(options))
+ if subentries is not UNDEFINED:
+ if entry.subentries != subentries:
+ changed = True
+ _setter(entry, "subentries", MappingProxyType(subentries))
+
if not changed:
return False
_setter(entry, "modified_at", utcnow())
+ self._async_save_and_notify(entry)
+ return True
+
+ @callback
+ def _async_save_and_notify(self, entry: ConfigEntry) -> None:
for listener in entry.update_listeners:
self.hass.async_create_task(
listener(self.hass, entry),
@@ -2241,8 +2462,92 @@ class ConfigEntries:
entry.clear_state_cache()
entry.clear_storage_cache()
self._async_dispatch(ConfigEntryChange.UPDATED, entry)
+
+ @callback
+ def async_add_subentry(self, entry: ConfigEntry, subentry: ConfigSubentry) -> bool:
+ """Add a subentry to a config entry."""
+ self._raise_if_subentry_unique_id_exists(entry, subentry.unique_id)
+
+ return self._async_update_entry(
+ entry,
+ subentries=entry.subentries | {subentry.subentry_id: subentry},
+ )
+
+ @callback
+ def async_remove_subentry(self, entry: ConfigEntry, subentry_id: str) -> bool:
+ """Remove a subentry from a config entry."""
+ subentries = dict(entry.subentries)
+ try:
+ subentries.pop(subentry_id)
+ except KeyError as err:
+ raise UnknownSubEntry from err
+
+ result = self._async_update_entry(entry, subentries=subentries)
+ dev_reg = dr.async_get(self.hass)
+ ent_reg = er.async_get(self.hass)
+
+ dev_reg.async_clear_config_subentry(entry.entry_id, subentry_id)
+ ent_reg.async_clear_config_subentry(entry.entry_id, subentry_id)
+ return result
+
+ @callback
+ def async_update_subentry(
+ self,
+ entry: ConfigEntry,
+ subentry: ConfigSubentry,
+ *,
+ data: Mapping[str, Any] | UndefinedType = UNDEFINED,
+ title: str | UndefinedType = UNDEFINED,
+ unique_id: str | None | UndefinedType = UNDEFINED,
+ ) -> bool:
+ """Update a config subentry.
+
+ If the subentry was changed, the update_listeners are
+ fired and this function returns True
+
+ If the subentry was not changed, the update_listeners are
+ not fired and this function returns False
+ """
+ if entry.entry_id not in self._entries:
+ raise UnknownEntry(entry.entry_id)
+ if subentry.subentry_id not in entry.subentries:
+ raise UnknownSubEntry(subentry.subentry_id)
+
+ self.hass.verify_event_loop_thread("hass.config_entries.async_update_subentry")
+ changed = False
+ _setter = object.__setattr__
+
+ if unique_id is not UNDEFINED and subentry.unique_id != unique_id:
+ self._raise_if_subentry_unique_id_exists(entry, unique_id)
+ changed = True
+ _setter(subentry, "unique_id", unique_id)
+
+ if title is not UNDEFINED and subentry.title != title:
+ changed = True
+ _setter(subentry, "title", title)
+
+ if data is not UNDEFINED and subentry.data != data:
+ changed = True
+ _setter(subentry, "data", MappingProxyType(data))
+
+ if not changed:
+ return False
+
+ _setter(entry, "modified_at", utcnow())
+
+ self._async_save_and_notify(entry)
return True
+ def _raise_if_subentry_unique_id_exists(
+ self, entry: ConfigEntry, unique_id: str | None
+ ) -> None:
+ """Raise if a subentry with the same unique_id exists."""
+ if unique_id is None:
+ return
+ for existing_subentry in entry.subentries.values():
+ if existing_subentry.unique_id == unique_id:
+ raise data_entry_flow.AbortFlow("already_configured")
+
@callback
def _async_dispatch(
self, change_type: ConfigEntryChange, entry: ConfigEntry
@@ -2579,6 +2884,14 @@ class ConfigFlow(ConfigEntryBaseFlow):
"""Return options flow support for this handler."""
return cls.async_get_options_flow is not ConfigFlow.async_get_options_flow
+ @classmethod
+ @callback
+ def async_get_supported_subentry_types(
+ cls, config_entry: ConfigEntry
+ ) -> dict[str, type[ConfigSubentryFlow]]:
+ """Return subentries supported by this handler."""
+ return {}
+
@callback
def _async_abort_entries_match(
self, match_dict: dict[str, Any] | None = None
@@ -2673,8 +2986,11 @@ class ConfigFlow(ConfigEntryBaseFlow):
return None
if raise_on_progress:
- if self._async_in_progress(
- include_uninitialized=True, match_context={"unique_id": unique_id}
+ if any(
+ flow["context"]["source"] != SOURCE_REAUTH
+ for flow in self._async_in_progress(
+ include_uninitialized=True, match_context={"unique_id": unique_id}
+ )
):
raise data_entry_flow.AbortFlow("already_in_progress")
@@ -2887,6 +3203,7 @@ class ConfigFlow(ConfigEntryBaseFlow):
description: str | None = None,
description_placeholders: Mapping[str, str] | None = None,
options: Mapping[str, Any] | None = None,
+ subentries: Iterable[ConfigSubentryData] | None = None,
) -> ConfigFlowResult:
"""Finish config flow and create a config entry."""
if self.source in {SOURCE_REAUTH, SOURCE_RECONFIGURE}:
@@ -2906,6 +3223,7 @@ class ConfigFlow(ConfigEntryBaseFlow):
result["minor_version"] = self.MINOR_VERSION
result["options"] = options or {}
+ result["subentries"] = subentries or ()
result["version"] = self.VERSION
return result
@@ -3020,17 +3338,199 @@ class ConfigFlow(ConfigEntryBaseFlow):
)
-class OptionsFlowManager(
- data_entry_flow.FlowManager[ConfigFlowContext, ConfigFlowResult]
-):
- """Flow to set options for a configuration entry."""
+class _ConfigSubFlowManager:
+ """Mixin class for flow managers which manage flows tied to a config entry."""
- _flow_result = ConfigFlowResult
+ hass: HomeAssistant
def _async_get_config_entry(self, config_entry_id: str) -> ConfigEntry:
"""Return config entry or raise if not found."""
return self.hass.config_entries.async_get_known_entry(config_entry_id)
+
+class ConfigSubentryFlowManager(
+ data_entry_flow.FlowManager[
+ SubentryFlowContext, SubentryFlowResult, tuple[str, str]
+ ],
+ _ConfigSubFlowManager,
+):
+ """Manage all the config subentry flows that are in progress."""
+
+ _flow_result = SubentryFlowResult
+
+ async def async_create_flow(
+ self,
+ handler_key: tuple[str, str],
+ *,
+ context: FlowContext | None = None,
+ data: dict[str, Any] | None = None,
+ ) -> ConfigSubentryFlow:
+ """Create a subentry flow for a config entry.
+
+ The entry_id and flow.handler[0] is the same thing to map entry with flow.
+ """
+ if not context or "source" not in context:
+ raise KeyError("Context not set or doesn't have a source set")
+
+ entry_id, subentry_type = handler_key
+ entry = self._async_get_config_entry(entry_id)
+ handler = await _async_get_flow_handler(self.hass, entry.domain, {})
+ subentry_types = handler.async_get_supported_subentry_types(entry)
+ if subentry_type not in subentry_types:
+ raise data_entry_flow.UnknownHandler(
+ f"Config entry '{entry.domain}' does not support subentry '{subentry_type}'"
+ )
+ subentry_flow = subentry_types[subentry_type]()
+ subentry_flow.init_step = context["source"]
+ return subentry_flow
+
+ async def async_finish_flow(
+ self,
+ flow: data_entry_flow.FlowHandler[
+ SubentryFlowContext, SubentryFlowResult, tuple[str, str]
+ ],
+ result: SubentryFlowResult,
+ ) -> SubentryFlowResult:
+ """Finish a subentry flow and add a new subentry to the configuration entry.
+
+ The flow.handler[0] and entry_id is the same thing to map flow with entry.
+ """
+ flow = cast(ConfigSubentryFlow, flow)
+
+ if result["type"] != data_entry_flow.FlowResultType.CREATE_ENTRY:
+ return result
+
+ entry_id, subentry_type = flow.handler
+ entry = self.hass.config_entries.async_get_entry(entry_id)
+ if entry is None:
+ raise UnknownEntry(entry_id)
+
+ unique_id = result.get("unique_id")
+ if unique_id is not None and not isinstance(unique_id, str):
+ raise HomeAssistantError("unique_id must be a string")
+
+ self.hass.config_entries.async_add_subentry(
+ entry,
+ ConfigSubentry(
+ data=MappingProxyType(result["data"]),
+ subentry_type=subentry_type,
+ title=result["title"],
+ unique_id=unique_id,
+ ),
+ )
+
+ result["result"] = True
+ return result
+
+
+class ConfigSubentryFlow(
+ data_entry_flow.FlowHandler[
+ SubentryFlowContext, SubentryFlowResult, tuple[str, str]
+ ]
+):
+ """Base class for config subentry flows."""
+
+ _flow_result = SubentryFlowResult
+ handler: tuple[str, str]
+
+ @callback
+ def async_create_entry(
+ self,
+ *,
+ title: str | None = None,
+ data: Mapping[str, Any],
+ description: str | None = None,
+ description_placeholders: Mapping[str, str] | None = None,
+ unique_id: str | None = None,
+ ) -> SubentryFlowResult:
+ """Finish config flow and create a config entry."""
+ if self.source != SOURCE_USER:
+ raise ValueError(f"Source is {self.source}, expected {SOURCE_USER}")
+
+ result = super().async_create_entry(
+ title=title,
+ data=data,
+ description=description,
+ description_placeholders=description_placeholders,
+ )
+
+ result["unique_id"] = unique_id
+
+ return result
+
+ @callback
+ def async_update_and_abort(
+ self,
+ entry: ConfigEntry,
+ subentry: ConfigSubentry,
+ *,
+ unique_id: str | None | UndefinedType = UNDEFINED,
+ title: str | UndefinedType = UNDEFINED,
+ data: Mapping[str, Any] | UndefinedType = UNDEFINED,
+ data_updates: Mapping[str, Any] | UndefinedType = UNDEFINED,
+ ) -> SubentryFlowResult:
+ """Update config subentry and finish subentry flow.
+
+ :param data: replace the subentry data with new data
+ :param data_updates: add items from data_updates to subentry data - existing
+ keys are overridden
+ :param title: replace the title of the subentry
+ :param unique_id: replace the unique_id of the subentry
+ """
+ if data_updates is not UNDEFINED:
+ if data is not UNDEFINED:
+ raise ValueError("Cannot set both data and data_updates")
+ data = subentry.data | data_updates
+ self.hass.config_entries.async_update_subentry(
+ entry=entry,
+ subentry=subentry,
+ unique_id=unique_id,
+ title=title,
+ data=data,
+ )
+ return self.async_abort(reason="reconfigure_successful")
+
+ @property
+ def _reconfigure_entry_id(self) -> str:
+ """Return reconfigure entry id."""
+ if self.source != SOURCE_RECONFIGURE:
+ raise ValueError(f"Source is {self.source}, expected {SOURCE_RECONFIGURE}")
+ return self.handler[0]
+
+ @callback
+ def _get_reconfigure_entry(self) -> ConfigEntry:
+ """Return the reconfigure config entry linked to the current context."""
+ return self.hass.config_entries.async_get_known_entry(
+ self._reconfigure_entry_id
+ )
+
+ @property
+ def _reconfigure_subentry_id(self) -> str:
+ """Return reconfigure subentry id."""
+ if self.source != SOURCE_RECONFIGURE:
+ raise ValueError(f"Source is {self.source}, expected {SOURCE_RECONFIGURE}")
+ return self.context["subentry_id"]
+
+ @callback
+ def _get_reconfigure_subentry(self) -> ConfigSubentry:
+ """Return the reconfigure config subentry linked to the current context."""
+ entry = self.hass.config_entries.async_get_known_entry(
+ self._reconfigure_entry_id
+ )
+ subentry_id = self._reconfigure_subentry_id
+ if subentry_id not in entry.subentries:
+ raise UnknownSubEntry(subentry_id)
+ return entry.subentries[subentry_id]
+
+
+class OptionsFlowManager(
+ data_entry_flow.FlowManager[ConfigFlowContext, ConfigFlowResult],
+ _ConfigSubFlowManager,
+):
+ """Manage all the config entry option flows that are in progress."""
+
+ _flow_result = ConfigFlowResult
+
async def async_create_flow(
self,
handler_key: str,
@@ -3040,7 +3540,7 @@ class OptionsFlowManager(
) -> OptionsFlow:
"""Create an options flow for a config entry.
- Entry_id and flow.handler is the same thing to map entry with flow.
+ The entry_id and the flow.handler is the same thing to map entry with flow.
"""
entry = self._async_get_config_entry(handler_key)
handler = await _async_get_flow_handler(self.hass, entry.domain, {})
@@ -3056,7 +3556,7 @@ class OptionsFlowManager(
This method is called when a flow step returns FlowResultType.ABORT or
FlowResultType.CREATE_ENTRY.
- Flow.handler and entry_id is the same thing to map flow with entry.
+ The flow.handler and the entry_id is the same thing to map flow with entry.
"""
flow = cast(OptionsFlow, flow)
diff --git a/homeassistant/const.py b/homeassistant/const.py
index 7775b618795..b9695c350a7 100644
--- a/homeassistant/const.py
+++ b/homeassistant/const.py
@@ -24,7 +24,7 @@ if TYPE_CHECKING:
APPLICATION_NAME: Final = "HomeAssistant"
MAJOR_VERSION: Final = 2025
-MINOR_VERSION: Final = 3
+MINOR_VERSION: Final = 4
PATCH_VERSION: Final = "0.dev0"
__short_version__: Final = f"{MAJOR_VERSION}.{MINOR_VERSION}"
__version__: Final = f"{__short_version__}.{PATCH_VERSION}"
diff --git a/homeassistant/data_entry_flow.py b/homeassistant/data_entry_flow.py
index e5ee5a79922..251e22e7990 100644
--- a/homeassistant/data_entry_flow.py
+++ b/homeassistant/data_entry_flow.py
@@ -561,7 +561,7 @@ class FlowManager(abc.ABC, Generic[_FlowContextT, _FlowResultT, _HandlerT]):
if not hasattr(flow, method):
self._async_remove_flow_progress(flow.flow_id)
raise UnknownStep(
- f"Handler {self.__class__.__name__} doesn't support step {step_id}"
+ f"Handler {flow.__class__.__name__} doesn't support step {step_id}"
)
async def _async_setup_preview(
diff --git a/homeassistant/generated/application_credentials.py b/homeassistant/generated/application_credentials.py
index 08fe28e4df5..68c6de405e6 100644
--- a/homeassistant/generated/application_credentials.py
+++ b/homeassistant/generated/application_credentials.py
@@ -25,9 +25,11 @@ APPLICATION_CREDENTIALS = [
"neato",
"nest",
"netatmo",
+ "ondilo_ico",
"onedrive",
"point",
"senz",
+ "smartthings",
"spotify",
"tesla_fleet",
"twitch",
diff --git a/homeassistant/generated/bluetooth.py b/homeassistant/generated/bluetooth.py
index 447b6d284f0..1ff444ca25f 100644
--- a/homeassistant/generated/bluetooth.py
+++ b/homeassistant/generated/bluetooth.py
@@ -356,6 +356,21 @@ BLUETOOTH: Final[list[dict[str, bool | str | int | list[int]]]] = [
"domain": "inkbird",
"local_name": "tps",
},
+ {
+ "connectable": False,
+ "domain": "inkbird",
+ "local_name": "ITH-11-B",
+ },
+ {
+ "connectable": False,
+ "domain": "inkbird",
+ "local_name": "ITH-13-B",
+ },
+ {
+ "connectable": False,
+ "domain": "inkbird",
+ "local_name": "ITH-21-B",
+ },
{
"connectable": True,
"domain": "iron_os",
@@ -688,6 +703,15 @@ BLUETOOTH: Final[list[dict[str, bool | str | int | list[int]]]] = [
"manufacturer_id": 17,
"service_uuid": "0000fff0-0000-1000-8000-00805f9b34fb",
},
+ {
+ "connectable": False,
+ "domain": "thermobeacon",
+ "manufacturer_data_start": [
+ 0,
+ ],
+ "manufacturer_id": 20,
+ "service_uuid": "0000fff0-0000-1000-8000-00805f9b34fb",
+ },
{
"connectable": False,
"domain": "thermobeacon",
diff --git a/homeassistant/generated/config_flows.py b/homeassistant/generated/config_flows.py
index d0a8e821f8d..a9c4a6b0a93 100644
--- a/homeassistant/generated/config_flows.py
+++ b/homeassistant/generated/config_flows.py
@@ -79,6 +79,7 @@ FLOWS = {
"azure_data_explorer",
"azure_devops",
"azure_event_hub",
+ "azure_storage",
"baf",
"balboa",
"bang_olufsen",
@@ -467,6 +468,7 @@ FLOWS = {
"peco",
"pegel_online",
"permobil",
+ "pglab",
"philips_js",
"pi_hole",
"picnic",
@@ -511,6 +513,7 @@ FLOWS = {
"rdw",
"recollect_waste",
"refoss",
+ "remote_calendar",
"renault",
"renson",
"reolink",
@@ -545,6 +548,7 @@ FLOWS = {
"sensirion_ble",
"sensorpro",
"sensorpush",
+ "sensorpush_cloud",
"sensoterra",
"sentry",
"senz",
@@ -573,6 +577,7 @@ FLOWS = {
"smlight",
"sms",
"snapcast",
+ "snoo",
"snooz",
"solaredge",
"solarlog",
@@ -689,6 +694,7 @@ FLOWS = {
"weatherflow",
"weatherflow_cloud",
"weatherkit",
+ "webdav",
"webmin",
"webostv",
"weheat",
diff --git a/homeassistant/generated/dhcp.py b/homeassistant/generated/dhcp.py
index 3dba5a98f3c..8ee1ea270f3 100644
--- a/homeassistant/generated/dhcp.py
+++ b/homeassistant/generated/dhcp.py
@@ -498,6 +498,18 @@ DHCP: Final[list[dict[str, str | bool]]] = [
"hostname": "ring*",
"macaddress": "341513*",
},
+ {
+ "domain": "roborock",
+ "macaddress": "249E7D*",
+ },
+ {
+ "domain": "roborock",
+ "macaddress": "B04A39*",
+ },
+ {
+ "domain": "roborock",
+ "hostname": "roborock-*",
+ },
{
"domain": "roomba",
"hostname": "irobot-*",
diff --git a/homeassistant/generated/integrations.json b/homeassistant/generated/integrations.json
index 026eab30f8f..55fcb08ba92 100644
--- a/homeassistant/generated/integrations.json
+++ b/homeassistant/generated/integrations.json
@@ -345,6 +345,11 @@
"config_flow": true,
"iot_class": "local_polling"
},
+ "apollo_automation": {
+ "name": "Apollo Automation",
+ "integration_type": "virtual",
+ "supported_by": "esphome"
+ },
"appalachianpower": {
"name": "Appalachian Power",
"integration_type": "virtual",
@@ -850,6 +855,11 @@
"config_flow": true,
"iot_class": "cloud_polling"
},
+ "burbank_water_and_power": {
+ "name": "Burbank Water and Power (BWP)",
+ "integration_type": "virtual",
+ "supported_by": "opower"
+ },
"caldav": {
"name": "CalDAV",
"integration_type": "hub",
@@ -2041,6 +2051,11 @@
"config_flow": false,
"iot_class": "cloud_push"
},
+ "frankever": {
+ "name": "FrankEver",
+ "integration_type": "virtual",
+ "supported_by": "shelly"
+ },
"free_mobile": {
"name": "Free Mobile",
"integration_type": "hub",
@@ -2609,7 +2624,8 @@
"name": "Home Connect",
"integration_type": "hub",
"config_flow": true,
- "iot_class": "cloud_push"
+ "iot_class": "cloud_push",
+ "single_config_entry": true
},
"home_plus_control": {
"name": "Legrand Home+ Control",
@@ -3391,12 +3407,22 @@
"config_flow": false,
"iot_class": "assumed_state"
},
+ "linak": {
+ "name": "LINAK",
+ "integration_type": "virtual",
+ "supported_by": "idasen_desk"
+ },
"linear_garage_door": {
"name": "Linear Garage Door",
"integration_type": "hub",
"config_flow": true,
"iot_class": "cloud_polling"
},
+ "linkedgo": {
+ "name": "LinkedGo",
+ "integration_type": "virtual",
+ "supported_by": "shelly"
+ },
"linkplay": {
"name": "LinkPlay",
"integration_type": "hub",
@@ -3624,7 +3650,7 @@
"iot_class": "cloud_push"
},
"matter": {
- "name": "Matter (BETA)",
+ "name": "Matter",
"integration_type": "hub",
"config_flow": true,
"iot_class": "local_push"
@@ -3794,6 +3820,12 @@
"iot_class": "cloud_push",
"name": "Azure Service Bus"
},
+ "azure_storage": {
+ "integration_type": "service",
+ "config_flow": true,
+ "iot_class": "cloud_polling",
+ "name": "Azure Storage"
+ },
"microsoft_face_detect": {
"integration_type": "hub",
"config_flow": false,
@@ -4394,6 +4426,11 @@
"config_flow": false,
"iot_class": "local_polling"
},
+ "ogemray": {
+ "name": "Ogemray",
+ "integration_type": "virtual",
+ "supported_by": "shelly"
+ },
"ohmconnect": {
"name": "OhmConnect",
"integration_type": "hub",
@@ -4738,6 +4775,13 @@
"integration_type": "virtual",
"supported_by": "opower"
},
+ "pglab": {
+ "name": "PG LAB Electronics",
+ "integration_type": "hub",
+ "config_flow": true,
+ "iot_class": "local_push",
+ "single_config_entry": true
+ },
"philips": {
"name": "Philips",
"integrations": {
@@ -5221,6 +5265,11 @@
"config_flow": false,
"iot_class": "cloud_push"
},
+ "remote_calendar": {
+ "integration_type": "service",
+ "config_flow": true,
+ "iot_class": "cloud_polling"
+ },
"renault": {
"name": "Renault",
"integration_type": "hub",
@@ -5574,9 +5623,20 @@
},
"sensorpush": {
"name": "SensorPush",
- "integration_type": "hub",
- "config_flow": true,
- "iot_class": "local_push"
+ "integrations": {
+ "sensorpush": {
+ "integration_type": "hub",
+ "config_flow": true,
+ "iot_class": "local_push",
+ "name": "SensorPush"
+ },
+ "sensorpush_cloud": {
+ "integration_type": "hub",
+ "config_flow": true,
+ "iot_class": "cloud_polling",
+ "name": "SensorPush Cloud"
+ }
+ }
},
"sensoterra": {
"name": "Sensoterra",
@@ -5892,6 +5952,12 @@
"config_flow": false,
"iot_class": "local_polling"
},
+ "snoo": {
+ "name": "Happiest Baby Snoo",
+ "integration_type": "hub",
+ "config_flow": true,
+ "iot_class": "cloud_push"
+ },
"snooz": {
"name": "Snooz",
"integration_type": "hub",
@@ -7062,6 +7128,12 @@
}
}
},
+ "webdav": {
+ "name": "WebDAV",
+ "integration_type": "service",
+ "config_flow": true,
+ "iot_class": "cloud_polling"
+ },
"webmin": {
"name": "Webmin",
"integration_type": "device",
@@ -7623,6 +7695,7 @@
"plant",
"proximity",
"random",
+ "remote_calendar",
"rpi_power",
"schedule",
"season",
diff --git a/homeassistant/generated/mqtt.py b/homeassistant/generated/mqtt.py
index 72f160ee2ec..c4eb8708b0e 100644
--- a/homeassistant/generated/mqtt.py
+++ b/homeassistant/generated/mqtt.py
@@ -16,6 +16,9 @@ MQTT = {
"fully_kiosk": [
"fully/deviceInfo/+",
],
+ "pglab": [
+ "pglab/discovery/#",
+ ],
"qbus": [
"cloudapp/QBUSMQTTGW/state",
"cloudapp/QBUSMQTTGW/config",
diff --git a/homeassistant/generated/ssdp.py b/homeassistant/generated/ssdp.py
index 89d1aa30cb8..5bbc178ba17 100644
--- a/homeassistant/generated/ssdp.py
+++ b/homeassistant/generated/ssdp.py
@@ -211,6 +211,12 @@ SSDP = {
{
"st": "nanoleaf:nl52",
},
+ {
+ "st": "nanoleaf:nl69",
+ },
+ {
+ "st": "inanoleaf:nl81",
+ },
],
"netgear": [
{
diff --git a/homeassistant/generated/zeroconf.py b/homeassistant/generated/zeroconf.py
index 8244f19660f..cc1683a3603 100644
--- a/homeassistant/generated/zeroconf.py
+++ b/homeassistant/generated/zeroconf.py
@@ -208,6 +208,14 @@ HOMEKIT = {
"always_discover": False,
"domain": "nanoleaf",
},
+ "NL69": {
+ "always_discover": False,
+ "domain": "nanoleaf",
+ },
+ "NL81": {
+ "always_discover": False,
+ "domain": "nanoleaf",
+ },
"Netatmo Relay": {
"always_discover": True,
"domain": "netatmo",
@@ -795,6 +803,11 @@ ZEROCONF = {
"domain": "russound_rio",
},
],
+ "_shelly._tcp.local.": [
+ {
+ "domain": "shelly",
+ },
+ ],
"_sideplay._tcp.local.": [
{
"domain": "ecobee",
diff --git a/homeassistant/helpers/backup.py b/homeassistant/helpers/backup.py
new file mode 100644
index 00000000000..4ab302749a1
--- /dev/null
+++ b/homeassistant/helpers/backup.py
@@ -0,0 +1,70 @@
+"""Helpers for the backup integration."""
+
+from __future__ import annotations
+
+import asyncio
+from collections.abc import Callable
+from dataclasses import dataclass, field
+from typing import TYPE_CHECKING
+
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.util.hass_dict import HassKey
+
+if TYPE_CHECKING:
+ from homeassistant.components.backup import BackupManager, ManagerStateEvent
+
+DATA_BACKUP: HassKey[BackupData] = HassKey("backup_data")
+DATA_MANAGER: HassKey[BackupManager] = HassKey("backup")
+
+
+@dataclass(slots=True)
+class BackupData:
+ """Backup data stored in hass.data."""
+
+ backup_event_subscriptions: list[Callable[[ManagerStateEvent], None]] = field(
+ default_factory=list
+ )
+ manager_ready: asyncio.Future[None] = field(default_factory=asyncio.Future)
+
+
+@callback
+def async_initialize_backup(hass: HomeAssistant) -> None:
+ """Initialize backup data.
+
+ This creates the BackupData instance stored in hass.data[DATA_BACKUP] and
+ registers the basic backup websocket API which is used by frontend to subscribe
+ to backup events.
+ """
+ # pylint: disable-next=import-outside-toplevel
+ from homeassistant.components.backup import basic_websocket
+
+ hass.data[DATA_BACKUP] = BackupData()
+ basic_websocket.async_register_websocket_handlers(hass)
+
+
+async def async_get_manager(hass: HomeAssistant) -> BackupManager:
+ """Get the backup manager instance.
+
+ Raises HomeAssistantError if the backup integration is not available.
+ """
+ if DATA_BACKUP not in hass.data:
+ raise HomeAssistantError("Backup integration is not available")
+
+ await hass.data[DATA_BACKUP].manager_ready
+ return hass.data[DATA_MANAGER]
+
+
+@callback
+def async_subscribe_events(
+ hass: HomeAssistant,
+ on_event: Callable[[ManagerStateEvent], None],
+) -> Callable[[], None]:
+ """Subscribe to backup events."""
+ backup_event_subscriptions = hass.data[DATA_BACKUP].backup_event_subscriptions
+
+ def remove_subscription() -> None:
+ backup_event_subscriptions.remove(on_event)
+
+ backup_event_subscriptions.append(on_event)
+ return remove_subscription
diff --git a/homeassistant/helpers/chat_session.py b/homeassistant/helpers/chat_session.py
index 686272dd834..e7a4ecd2ca9 100644
--- a/homeassistant/helpers/chat_session.py
+++ b/homeassistant/helpers/chat_session.py
@@ -7,6 +7,7 @@ from contextlib import contextmanager
from contextvars import ContextVar
from dataclasses import dataclass, field
from datetime import datetime, timedelta
+import logging
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import (
@@ -27,6 +28,7 @@ DATA_CHAT_SESSION: HassKey[dict[str, ChatSession]] = HassKey("chat_session")
DATA_CHAT_SESSION_CLEANUP: HassKey[SessionCleanup] = HassKey("chat_session_cleanup")
CONVERSATION_TIMEOUT = timedelta(minutes=5)
+LOGGER = logging.getLogger(__name__)
current_session: ContextVar[ChatSession | None] = ContextVar(
"current_session", default=None
@@ -100,6 +102,7 @@ class SessionCleanup:
# yielding session based on it.
for conversation_id, session in list(all_sessions.items()):
if session.last_updated + CONVERSATION_TIMEOUT < now:
+ LOGGER.debug("Cleaning up session %s", conversation_id)
del all_sessions[conversation_id]
session.async_cleanup()
@@ -150,6 +153,7 @@ def async_get_chat_session(
pass
if session is None:
+ LOGGER.debug("Creating new session %s", conversation_id)
session = ChatSession(conversation_id)
current_session.set(session)
diff --git a/homeassistant/helpers/check_config.py b/homeassistant/helpers/check_config.py
index 0841585e1a1..836536da9ee 100644
--- a/homeassistant/helpers/check_config.py
+++ b/homeassistant/helpers/check_config.py
@@ -8,6 +8,7 @@ import os
from pathlib import Path
from typing import NamedTuple, Self
+from annotatedyaml import loader as yaml_loader
import voluptuous as vol
from homeassistant import loader
@@ -29,7 +30,6 @@ from homeassistant.requirements import (
async_clear_install_history,
async_get_integration_with_requirements,
)
-from homeassistant.util.yaml import loader as yaml_loader
from . import config_validation as cv
from .typing import ConfigType
diff --git a/homeassistant/helpers/config_entry_oauth2_flow.py b/homeassistant/helpers/config_entry_oauth2_flow.py
index 24a9de5b562..84728978ede 100644
--- a/homeassistant/helpers/config_entry_oauth2_flow.py
+++ b/homeassistant/helpers/config_entry_oauth2_flow.py
@@ -11,7 +11,9 @@ from __future__ import annotations
from abc import ABC, ABCMeta, abstractmethod
import asyncio
from asyncio import Lock
+import base64
from collections.abc import Awaitable, Callable
+import hashlib
from http import HTTPStatus
from json import JSONDecodeError
import logging
@@ -166,6 +168,11 @@ class LocalOAuth2Implementation(AbstractOAuth2Implementation):
"""Extra data that needs to be appended to the authorize url."""
return {}
+ @property
+ def extra_token_resolve_data(self) -> dict:
+ """Extra data for the token resolve request."""
+ return {}
+
async def async_generate_authorize_url(self, flow_id: str) -> str:
"""Generate a url for the user to authorize."""
redirect_uri = self.redirect_uri
@@ -186,13 +193,13 @@ class LocalOAuth2Implementation(AbstractOAuth2Implementation):
async def async_resolve_external_data(self, external_data: Any) -> dict:
"""Resolve the authorization code to tokens."""
- return await self._token_request(
- {
- "grant_type": "authorization_code",
- "code": external_data["code"],
- "redirect_uri": external_data["state"]["redirect_uri"],
- }
- )
+ request_data: dict = {
+ "grant_type": "authorization_code",
+ "code": external_data["code"],
+ "redirect_uri": external_data["state"]["redirect_uri"],
+ }
+ request_data.update(self.extra_token_resolve_data)
+ return await self._token_request(request_data)
async def _async_refresh_token(self, token: dict) -> dict:
"""Refresh tokens."""
@@ -211,7 +218,7 @@ class LocalOAuth2Implementation(AbstractOAuth2Implementation):
data["client_id"] = self.client_id
- if self.client_secret is not None:
+ if self.client_secret:
data["client_secret"] = self.client_secret
_LOGGER.debug("Sending token request to %s", self.token_url)
@@ -233,6 +240,100 @@ class LocalOAuth2Implementation(AbstractOAuth2Implementation):
return cast(dict, await resp.json())
+class LocalOAuth2ImplementationWithPkce(LocalOAuth2Implementation):
+ """Local OAuth2 implementation with PKCE."""
+
+ def __init__(
+ self,
+ hass: HomeAssistant,
+ domain: str,
+ client_id: str,
+ authorize_url: str,
+ token_url: str,
+ client_secret: str = "",
+ code_verifier_length: int = 128,
+ ) -> None:
+ """Initialize local auth implementation."""
+ super().__init__(
+ hass,
+ domain,
+ client_id,
+ client_secret,
+ authorize_url,
+ token_url,
+ )
+
+ # Generate code verifier
+ self.code_verifier = LocalOAuth2ImplementationWithPkce.generate_code_verifier(
+ code_verifier_length
+ )
+
+ @property
+ def extra_authorize_data(self) -> dict:
+ """Extra data that needs to be appended to the authorize url.
+
+ If you want to override this method,
+ calling super is mandatory (for adding scopes):
+ ```
+ @def extra_authorize_data(self) -> dict:
+ data: dict = {
+ "scope": "openid profile email",
+ }
+ data.update(super().extra_authorize_data)
+ return data
+ ```
+ """
+ return {
+ "code_challenge": LocalOAuth2ImplementationWithPkce.compute_code_challenge(
+ self.code_verifier
+ ),
+ "code_challenge_method": "S256",
+ }
+
+ @property
+ def extra_token_resolve_data(self) -> dict:
+ """Extra data that needs to be included in the token resolve request.
+
+ If you want to override this method,
+ calling super is mandatory (for adding `someKey`):
+ ```
+ @def extra_token_resolve_data(self) -> dict:
+ data: dict = {
+ "someKey": "someValue",
+ }
+ data.update(super().extra_token_resolve_data)
+ return data
+ ```
+ """
+
+ return {"code_verifier": self.code_verifier}
+
+ @staticmethod
+ def generate_code_verifier(code_verifier_length: int = 128) -> str:
+ """Generate a code verifier."""
+ if not 43 <= code_verifier_length <= 128:
+ msg = (
+ "Parameter `code_verifier_length` must validate"
+ "`43 <= code_verifier_length <= 128`."
+ )
+ raise ValueError(msg)
+ return secrets.token_urlsafe(96)[:code_verifier_length]
+
+ @staticmethod
+ def compute_code_challenge(code_verifier: str) -> str:
+ """Compute the code challenge."""
+ if not 43 <= len(code_verifier) <= 128:
+ msg = (
+ "Parameter `code_verifier` must validate "
+ "`43 <= len(code_verifier) <= 128`."
+ )
+ raise ValueError(msg)
+
+ hashed = hashlib.sha256(code_verifier.encode("ascii")).digest()
+ encoded = base64.urlsafe_b64encode(hashed)
+ return encoded.decode("ascii").replace("=", "")
+
+
class AbstractOAuth2FlowHandler(config_entries.ConfigFlow, metaclass=ABCMeta):
"""Handle a config flow."""
diff --git a/homeassistant/helpers/data_entry_flow.py b/homeassistant/helpers/data_entry_flow.py
index b15d8b9e607..65eb2786aaf 100644
--- a/homeassistant/helpers/data_entry_flow.py
+++ b/homeassistant/helpers/data_entry_flow.py
@@ -17,7 +17,7 @@ from . import config_validation as cv
_FlowManagerT = TypeVar(
"_FlowManagerT",
- bound=data_entry_flow.FlowManager[Any, Any],
+ bound=data_entry_flow.FlowManager[Any, Any, Any],
default=data_entry_flow.FlowManager,
)
@@ -70,7 +70,7 @@ class FlowManagerIndexView(_BaseFlowManagerView[_FlowManagerT]):
async def post(self, request: web.Request, data: dict[str, Any]) -> web.Response:
"""Initialize a POST request.
- Override `_post_impl` in subclasses which need
+ Override `post` and call `_post_impl` in subclasses which need
to implement their own `RequestDataValidator`
"""
return await self._post_impl(request, data)
diff --git a/homeassistant/helpers/deprecation.py b/homeassistant/helpers/deprecation.py
index f02c6507d02..101b9731caf 100644
--- a/homeassistant/helpers/deprecation.py
+++ b/homeassistant/helpers/deprecation.py
@@ -244,35 +244,35 @@ def _print_deprecation_warning_internal_impl(
)
-class DeprecatedConstant(NamedTuple):
+class DeprecatedConstant[T](NamedTuple):
"""Deprecated constant."""
- value: Any
+ value: T
replacement: str
breaks_in_ha_version: str | None
-class DeprecatedConstantEnum(NamedTuple):
+class DeprecatedConstantEnum[T: (StrEnum | IntEnum | IntFlag)](NamedTuple):
"""Deprecated constant."""
- enum: StrEnum | IntEnum | IntFlag
+ enum: T
breaks_in_ha_version: str | None
-class DeprecatedAlias(NamedTuple):
+class DeprecatedAlias[T](NamedTuple):
"""Deprecated alias."""
- value: Any
+ value: T
replacement: str
breaks_in_ha_version: str | None
-class DeferredDeprecatedAlias:
+class DeferredDeprecatedAlias[T]:
"""Deprecated alias with deferred evaluation of the value."""
def __init__(
self,
- value_fn: Callable[[], Any],
+ value_fn: Callable[[], T],
replacement: str,
breaks_in_ha_version: str | None,
) -> None:
@@ -282,7 +282,7 @@ class DeferredDeprecatedAlias:
self._value_fn = value_fn
@functools.cached_property
- def value(self) -> Any:
+ def value(self) -> T:
"""Return the value."""
return self._value_fn()
@@ -369,7 +369,7 @@ class EnumWithDeprecatedMembers(EnumType):
"""Enum with deprecated members."""
def __new__(
- mcs, # noqa: N804 ruff bug, ruff does not understand this is a metaclass
+ mcs,
cls: str,
bases: tuple[type, ...],
classdict: _EnumDict,
diff --git a/homeassistant/helpers/device_registry.py b/homeassistant/helpers/device_registry.py
index 92101dd0e21..991a6cf5a57 100644
--- a/homeassistant/helpers/device_registry.py
+++ b/homeassistant/helpers/device_registry.py
@@ -56,7 +56,7 @@ EVENT_DEVICE_REGISTRY_UPDATED: EventType[EventDeviceRegistryUpdatedData] = Event
)
STORAGE_KEY = "core.device_registry"
STORAGE_VERSION_MAJOR = 1
-STORAGE_VERSION_MINOR = 8
+STORAGE_VERSION_MINOR = 9
CLEANUP_DELAY = 10
@@ -272,6 +272,7 @@ class DeviceEntry:
area_id: str | None = attr.ib(default=None)
config_entries: set[str] = attr.ib(converter=set, factory=set)
+ config_entries_subentries: dict[str, set[str | None]] = attr.ib(factory=dict)
configuration_url: str | None = attr.ib(default=None)
connections: set[tuple[str, str]] = attr.ib(converter=set, factory=set)
created_at: datetime = attr.ib(factory=utcnow)
@@ -311,6 +312,10 @@ class DeviceEntry:
"area_id": self.area_id,
"configuration_url": self.configuration_url,
"config_entries": list(self.config_entries),
+ "config_entries_subentries": {
+ config_entry_id: list(subentries)
+ for config_entry_id, subentries in self.config_entries_subentries.items()
+ },
"connections": list(self.connections),
"created_at": self.created_at.timestamp(),
"disabled_by": self.disabled_by,
@@ -354,7 +359,13 @@ class DeviceEntry:
json_bytes(
{
"area_id": self.area_id,
+ # The config_entries list can be removed from the storage
+ # representation in HA Core 2026.2
"config_entries": list(self.config_entries),
+ "config_entries_subentries": {
+ config_entry_id: list(subentries)
+ for config_entry_id, subentries in self.config_entries_subentries.items()
+ },
"configuration_url": self.configuration_url,
"connections": list(self.connections),
"created_at": self.created_at,
@@ -384,6 +395,7 @@ class DeletedDeviceEntry:
"""Deleted Device Registry Entry."""
config_entries: set[str] = attr.ib()
+ config_entries_subentries: dict[str, set[str | None]] = attr.ib()
connections: set[tuple[str, str]] = attr.ib()
identifiers: set[tuple[str, str]] = attr.ib()
id: str = attr.ib()
@@ -395,6 +407,7 @@ class DeletedDeviceEntry:
def to_device_entry(
self,
config_entry_id: str,
+ config_subentry_id: str | None,
connections: set[tuple[str, str]],
identifiers: set[tuple[str, str]],
) -> DeviceEntry:
@@ -402,6 +415,7 @@ class DeletedDeviceEntry:
return DeviceEntry(
# type ignores: likely https://github.com/python/mypy/issues/8625
config_entries={config_entry_id}, # type: ignore[arg-type]
+ config_entries_subentries={config_entry_id: {config_subentry_id}},
connections=self.connections & connections, # type: ignore[arg-type]
created_at=self.created_at,
identifiers=self.identifiers & identifiers, # type: ignore[arg-type]
@@ -415,7 +429,13 @@ class DeletedDeviceEntry:
return json_fragment(
json_bytes(
{
+ # The config_entries list can be removed from the storage
+ # representation in HA Core 2026.2
"config_entries": list(self.config_entries),
+ "config_entries_subentries": {
+ config_entry_id: list(subentries)
+ for config_entry_id, subentries in self.config_entries_subentries.items()
+ },
"connections": list(self.connections),
"created_at": self.created_at,
"identifiers": list(self.identifiers),
@@ -458,7 +478,10 @@ class DeviceRegistryStore(storage.Store[dict[str, list[dict[str, Any]]]]):
old_data: dict[str, list[dict[str, Any]]],
) -> dict[str, Any]:
"""Migrate to the new version."""
- if old_major_version < 2:
+ # Support for a future major version bump to 2 added in HA Core 2025.2.
+ # Major versions 1 and 2 will be the same, except that version 2 will no
+ # longer store a list of config_entries.
+ if old_major_version < 3:
if old_minor_version < 2:
# Version 1.2 implements migration and freezes the available keys,
# populate keys which were introduced before version 1.2
@@ -505,8 +528,20 @@ class DeviceRegistryStore(storage.Store[dict[str, list[dict[str, Any]]]]):
device["created_at"] = device["modified_at"] = created_at
for device in old_data["deleted_devices"]:
device["created_at"] = device["modified_at"] = created_at
+ if old_minor_version < 9:
+ # Introduced in 2025.2
+ for device in old_data["devices"]:
+ device["config_entries_subentries"] = {
+ config_entry_id: {None}
+ for config_entry_id in device["config_entries"]
+ }
+ for device in old_data["deleted_devices"]:
+ device["config_entries_subentries"] = {
+ config_entry_id: {None}
+ for config_entry_id in device["config_entries"]
+ }
- if old_major_version > 1:
+ if old_major_version > 2:
raise NotImplementedError
return old_data
@@ -722,6 +757,7 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
self,
*,
config_entry_id: str,
+ config_subentry_id: str | None | UndefinedType = UNDEFINED,
configuration_url: str | URL | None | UndefinedType = UNDEFINED,
connections: set[tuple[str, str]] | None | UndefinedType = UNDEFINED,
created_at: str | datetime | UndefinedType = UNDEFINED, # will be ignored
@@ -812,7 +848,11 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
else:
self.deleted_devices.pop(deleted_device.id)
device = deleted_device.to_device_entry(
- config_entry_id, connections, identifiers
+ config_entry_id,
+ # Interpret not specifying a subentry as None
+ config_subentry_id if config_subentry_id is not UNDEFINED else None,
+ connections,
+ identifiers,
)
self.devices[device.id] = device
# If creating a new device, default to the config entry name
@@ -846,6 +886,7 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
device.id,
allow_collisions=True,
add_config_entry_id=config_entry_id,
+ add_config_subentry_id=config_subentry_id,
configuration_url=configuration_url,
device_info_type=device_info_type,
disabled_by=disabled_by,
@@ -874,6 +915,7 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
device_id: str,
*,
add_config_entry_id: str | UndefinedType = UNDEFINED,
+ add_config_subentry_id: str | None | UndefinedType = UNDEFINED,
# Temporary flag so we don't blow up when collisions are implicitly introduced
# by calls to async_get_or_create. Must not be set by integrations.
allow_collisions: bool = False,
@@ -894,25 +936,58 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
new_connections: set[tuple[str, str]] | UndefinedType = UNDEFINED,
new_identifiers: set[tuple[str, str]] | UndefinedType = UNDEFINED,
remove_config_entry_id: str | UndefinedType = UNDEFINED,
+ remove_config_subentry_id: str | None | UndefinedType = UNDEFINED,
serial_number: str | None | UndefinedType = UNDEFINED,
suggested_area: str | None | UndefinedType = UNDEFINED,
sw_version: str | None | UndefinedType = UNDEFINED,
via_device_id: str | None | UndefinedType = UNDEFINED,
) -> DeviceEntry | None:
- """Update device attributes."""
+ """Update device attributes.
+
+ :param add_config_subentry_id: Add the device to a specific subentry of add_config_entry_id
+ :param remove_config_subentry_id: Remove the device from a specific subentry of remove_config_subentry_id
+ """
old = self.devices[device_id]
new_values: dict[str, Any] = {} # Dict with new key/value pairs
old_values: dict[str, Any] = {} # Dict with old key/value pairs
config_entries = old.config_entries
+ config_entries_subentries = old.config_entries_subentries
if add_config_entry_id is not UNDEFINED:
- if self.hass.config_entries.async_get_entry(add_config_entry_id) is None:
+ if (
+ add_config_entry := self.hass.config_entries.async_get_entry(
+ add_config_entry_id
+ )
+ ) is None:
raise HomeAssistantError(
f"Can't link device to unknown config entry {add_config_entry_id}"
)
+ if add_config_subentry_id is not UNDEFINED:
+ if add_config_entry_id is UNDEFINED:
+ raise HomeAssistantError(
+ "Can't add config subentry without specifying config entry"
+ )
+ if (
+ add_config_subentry_id
+ # mypy says add_config_entry can be None. That's impossible, because we
+ # raise above if that happens
+ and add_config_subentry_id not in add_config_entry.subentries # type: ignore[union-attr]
+ ):
+ raise HomeAssistantError(
+ f"Config entry {add_config_entry_id} has no subentry {add_config_subentry_id}"
+ )
+
+ if (
+ remove_config_subentry_id is not UNDEFINED
+ and remove_config_entry_id is UNDEFINED
+ ):
+ raise HomeAssistantError(
+ "Can't remove config subentry without specifying config entry"
+ )
+
if not new_connections and not new_identifiers:
raise HomeAssistantError(
"A device must have at least one of identifiers or connections"
@@ -943,6 +1018,10 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
area_id = area.id
if add_config_entry_id is not UNDEFINED:
+ if add_config_subentry_id is UNDEFINED:
+ # Interpret not specifying a subentry as None (the main entry)
+ add_config_subentry_id = None
+
primary_entry_id = old.primary_config_entry
if (
device_info_type == "primary"
@@ -962,25 +1041,59 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
if add_config_entry_id not in old.config_entries:
config_entries = old.config_entries | {add_config_entry_id}
+ config_entries_subentries = old.config_entries_subentries | {
+ add_config_entry_id: {add_config_subentry_id}
+ }
+ elif (
+ add_config_subentry_id
+ not in old.config_entries_subentries[add_config_entry_id]
+ ):
+ config_entries_subentries = old.config_entries_subentries | {
+ add_config_entry_id: old.config_entries_subentries[
+ add_config_entry_id
+ ]
+ | {add_config_subentry_id}
+ }
if (
remove_config_entry_id is not UNDEFINED
and remove_config_entry_id in config_entries
):
- if config_entries == {remove_config_entry_id}:
- self.async_remove_device(device_id)
- return None
+ if remove_config_subentry_id is UNDEFINED:
+ config_entries_subentries = dict(old.config_entries_subentries)
+ del config_entries_subentries[remove_config_entry_id]
+ elif (
+ remove_config_subentry_id
+ in old.config_entries_subentries[remove_config_entry_id]
+ ):
+ config_entries_subentries = old.config_entries_subentries | {
+ remove_config_entry_id: old.config_entries_subentries[
+ remove_config_entry_id
+ ]
+ - {remove_config_subentry_id}
+ }
+ if not config_entries_subentries[remove_config_entry_id]:
+ del config_entries_subentries[remove_config_entry_id]
- if remove_config_entry_id == old.primary_config_entry:
- new_values["primary_config_entry"] = None
- old_values["primary_config_entry"] = old.primary_config_entry
+ if remove_config_entry_id not in config_entries_subentries:
+ if config_entries == {remove_config_entry_id}:
+ self.async_remove_device(device_id)
+ return None
- config_entries = config_entries - {remove_config_entry_id}
+ if remove_config_entry_id == old.primary_config_entry:
+ new_values["primary_config_entry"] = None
+ old_values["primary_config_entry"] = old.primary_config_entry
+
+ config_entries = config_entries - {remove_config_entry_id}
if config_entries != old.config_entries:
new_values["config_entries"] = config_entries
old_values["config_entries"] = old.config_entries
+ if config_entries_subentries != old.config_entries_subentries:
+ new_values["config_entries_subentries"] = config_entries_subentries
+ old_values["config_entries_subentries"] = old.config_entries_subentries
+
added_connections: set[tuple[str, str]] | None = None
added_identifiers: set[tuple[str, str]] | None = None
@@ -1138,6 +1251,7 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
device = self.devices.pop(device_id)
self.deleted_devices[device_id] = DeletedDeviceEntry(
config_entries=device.config_entries,
+ config_entries_subentries=device.config_entries_subentries,
connections=device.connections,
created_at=device.created_at,
identifiers=device.identifiers,
@@ -1168,7 +1282,13 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
for device in data["devices"]:
devices[device["id"]] = DeviceEntry(
area_id=device["area_id"],
- config_entries=set(device["config_entries"]),
+ config_entries=set(device["config_entries_subentries"]),
+ config_entries_subentries={
+ config_entry_id: set(subentries)
+ for config_entry_id, subentries in device[
+ "config_entries_subentries"
+ ].items()
+ },
configuration_url=device["configuration_url"],
# type ignores (if tuple arg was cast): likely https://github.com/python/mypy/issues/8625
connections={
@@ -1208,6 +1328,12 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
for device in data["deleted_devices"]:
deleted_devices[device["id"]] = DeletedDeviceEntry(
config_entries=set(device["config_entries"]),
+ config_entries_subentries={
+ config_entry_id: set(subentries)
+ for config_entry_id, subentries in device[
+ "config_entries_subentries"
+ ].items()
+ },
connections={tuple(conn) for conn in device["connections"]},
created_at=datetime.fromisoformat(device["created_at"]),
identifiers={tuple(iden) for iden in device["identifiers"]},
@@ -1243,14 +1369,70 @@ class DeviceRegistry(BaseRegistry[dict[str, list[dict[str, Any]]]]):
if config_entries == {config_entry_id}:
# Add a time stamp when the deleted device became orphaned
self.deleted_devices[deleted_device.id] = attr.evolve(
- deleted_device, orphaned_timestamp=now_time, config_entries=set()
+ deleted_device,
+ orphaned_timestamp=now_time,
+ config_entries=set(),
+ config_entries_subentries={},
)
else:
config_entries = config_entries - {config_entry_id}
+ config_entries_subentries = dict(
+ deleted_device.config_entries_subentries
+ )
+ del config_entries_subentries[config_entry_id]
# No need to reindex here since we currently
# do not have a lookup by config entry
self.deleted_devices[deleted_device.id] = attr.evolve(
- deleted_device, config_entries=config_entries
+ deleted_device,
+ config_entries=config_entries,
+ config_entries_subentries=config_entries_subentries,
+ )
+ self.async_schedule_save()
+
+ @callback
+ def async_clear_config_subentry(
+ self, config_entry_id: str, config_subentry_id: str
+ ) -> None:
+ """Clear config entry from registry entries."""
+ now_time = time.time()
+ now_time = time.time()
+ for device in self.devices.get_devices_for_config_entry_id(config_entry_id):
+ self.async_update_device(
+ device.id,
+ remove_config_entry_id=config_entry_id,
+ remove_config_subentry_id=config_subentry_id,
+ )
+ for deleted_device in list(self.deleted_devices.values()):
+ config_entries = deleted_device.config_entries
+ config_entries_subentries = deleted_device.config_entries_subentries
+ if (
+ config_entry_id not in config_entries_subentries
+ or config_subentry_id not in config_entries_subentries[config_entry_id]
+ ):
+ continue
+ if config_entries_subentries == {config_entry_id: {config_subentry_id}}:
+ # We're removing the last config subentry from the last config
+ # entry, add a time stamp when the deleted device became orphaned
+ self.deleted_devices[deleted_device.id] = attr.evolve(
+ deleted_device,
+ orphaned_timestamp=now_time,
+ config_entries=set(),
+ config_entries_subentries={},
+ )
+ else:
+ config_entries_subentries = config_entries_subentries | {
+ config_entry_id: config_entries_subentries[config_entry_id]
+ - {config_subentry_id}
+ }
+ if not config_entries_subentries[config_entry_id]:
+ del config_entries_subentries[config_entry_id]
+ config_entries = config_entries - {config_entry_id}
+ # No need to reindex here since we currently
+ # do not have a lookup by config entry
+ self.deleted_devices[deleted_device.id] = attr.evolve(
+ deleted_device,
+ config_entries=config_entries,
+ config_entries_subentries=config_entries_subentries,
)
self.async_schedule_save()
diff --git a/homeassistant/helpers/entity.py b/homeassistant/helpers/entity.py
index 2b9f2d7069e..bdcda58c054 100644
--- a/homeassistant/helpers/entity.py
+++ b/homeassistant/helpers/entity.py
@@ -281,7 +281,7 @@ class CachedProperties(type):
"""
def __new__(
- mcs, # noqa: N804 ruff bug, ruff does not understand this is a metaclass
+ mcs,
name: str,
bases: tuple[type, ...],
namespace: dict[Any, Any],
@@ -1085,9 +1085,9 @@ class Entity(
state = self._stringify_state(available)
if available:
if state_attributes := self.state_attributes:
- attr.update(state_attributes)
+ attr |= state_attributes
if extra_state_attributes := self.extra_state_attributes:
- attr.update(extra_state_attributes)
+ attr |= extra_state_attributes
if (unit_of_measurement := self.unit_of_measurement) is not None:
attr[ATTR_UNIT_OF_MEASUREMENT] = unit_of_measurement
@@ -1214,7 +1214,7 @@ class Entity(
else:
# Overwrite properties that have been set in the config file.
if custom := customize.get(entity_id):
- attr.update(custom)
+ attr |= custom
if (
self._context_set is not None
diff --git a/homeassistant/helpers/entity_platform.py b/homeassistant/helpers/entity_platform.py
index c8cc6979226..11a9786f86e 100644
--- a/homeassistant/helpers/entity_platform.py
+++ b/homeassistant/helpers/entity_platform.py
@@ -80,6 +80,22 @@ class AddEntitiesCallback(Protocol):
"""Define add_entities type."""
+class AddConfigEntryEntitiesCallback(Protocol):
+ """Protocol type for EntityPlatform.add_entities callback."""
+
+ def __call__(
+ self,
+ new_entities: Iterable[Entity],
+ update_before_add: bool = False,
+ *,
+ config_subentry_id: str | None = None,
+ ) -> None:
+ """Define add_entities type.
+
+ :param config_subentry_id: subentry which the entities should be added to
+ """
+
+
class EntityPlatformModule(Protocol):
"""Protocol type for entity platform modules."""
@@ -105,7 +121,7 @@ class EntityPlatformModule(Protocol):
self,
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up an integration platform from a config entry."""
@@ -517,13 +533,21 @@ class EntityPlatform:
@callback
def _async_schedule_add_entities_for_entry(
- self, new_entities: Iterable[Entity], update_before_add: bool = False
+ self,
+ new_entities: Iterable[Entity],
+ update_before_add: bool = False,
+ *,
+ config_subentry_id: str | None = None,
) -> None:
"""Schedule adding entities for a single platform async and track the task."""
assert self.config_entry
task = self.config_entry.async_create_task(
self.hass,
- self.async_add_entities(new_entities, update_before_add=update_before_add),
+ self.async_add_entities(
+ new_entities,
+ update_before_add=update_before_add,
+ config_subentry_id=config_subentry_id,
+ ),
f"EntityPlatform async_add_entities_for_entry {self.domain}.{self.platform_name}",
eager_start=True,
)
@@ -625,12 +649,27 @@ class EntityPlatform:
)
async def async_add_entities(
- self, new_entities: Iterable[Entity], update_before_add: bool = False
+ self,
+ new_entities: Iterable[Entity],
+ update_before_add: bool = False,
+ *,
+ config_subentry_id: str | None = None,
) -> None:
"""Add entities for a single platform async.
This method must be run in the event loop.
+
+ :param config_subentry_id: subentry which the entities should be added to
"""
+ if config_subentry_id and (
+ not self.config_entry
+ or config_subentry_id not in self.config_entry.subentries
+ ):
+ raise HomeAssistantError(
+ f"Can't add entities to unknown subentry {config_subentry_id} of config "
+ f"entry {self.config_entry.entry_id if self.config_entry else None}"
+ )
+
# handle empty list from component/platform
if not new_entities: # type: ignore[truthy-iterable]
return
@@ -641,7 +680,9 @@ class EntityPlatform:
entities: list[Entity] = []
for entity in new_entities:
coros.append(
- self._async_add_entity(entity, update_before_add, entity_registry)
+ self._async_add_entity(
+ entity, update_before_add, entity_registry, config_subentry_id
+ )
)
entities.append(entity)
@@ -720,6 +761,7 @@ class EntityPlatform:
entity: Entity,
update_before_add: bool,
entity_registry: EntityRegistry,
+ config_subentry_id: str | None,
) -> None:
"""Add an entity to the platform."""
if entity is None:
@@ -779,6 +821,7 @@ class EntityPlatform:
try:
device = dev_reg.async_get(self.hass).async_get_or_create(
config_entry_id=self.config_entry.entry_id,
+ config_subentry_id=config_subentry_id,
**device_info,
)
except dev_reg.DeviceInfoError as exc:
@@ -825,6 +868,7 @@ class EntityPlatform:
entity.unique_id,
capabilities=entity.capability_attributes,
config_entry=self.config_entry,
+ config_subentry_id=config_subentry_id,
device_id=device.id if device else None,
disabled_by=disabled_by,
entity_category=entity.entity_category,
diff --git a/homeassistant/helpers/entity_registry.py b/homeassistant/helpers/entity_registry.py
index 95a32696228..684d00fe344 100644
--- a/homeassistant/helpers/entity_registry.py
+++ b/homeassistant/helpers/entity_registry.py
@@ -79,7 +79,7 @@ EVENT_ENTITY_REGISTRY_UPDATED: EventType[EventEntityRegistryUpdatedData] = Event
_LOGGER = logging.getLogger(__name__)
STORAGE_VERSION_MAJOR = 1
-STORAGE_VERSION_MINOR = 15
+STORAGE_VERSION_MINOR = 16
STORAGE_KEY = "core.entity_registry"
CLEANUP_INTERVAL = 3600 * 24
@@ -177,6 +177,7 @@ class RegistryEntry:
categories: dict[str, str] = attr.ib(factory=dict)
capabilities: Mapping[str, Any] | None = attr.ib(default=None)
config_entry_id: str | None = attr.ib(default=None)
+ config_subentry_id: str | None = attr.ib(default=None)
created_at: datetime = attr.ib(factory=utcnow)
device_class: str | None = attr.ib(default=None)
device_id: str | None = attr.ib(default=None)
@@ -280,6 +281,7 @@ class RegistryEntry:
"area_id": self.area_id,
"categories": self.categories,
"config_entry_id": self.config_entry_id,
+ "config_subentry_id": self.config_subentry_id,
"created_at": self.created_at.timestamp(),
"device_id": self.device_id,
"disabled_by": self.disabled_by,
@@ -341,6 +343,7 @@ class RegistryEntry:
"categories": self.categories,
"capabilities": self.capabilities,
"config_entry_id": self.config_entry_id,
+ "config_subentry_id": self.config_subentry_id,
"created_at": self.created_at,
"device_class": self.device_class,
"device_id": self.device_id,
@@ -405,6 +408,7 @@ class DeletedRegistryEntry:
unique_id: str = attr.ib()
platform: str = attr.ib()
config_entry_id: str | None = attr.ib()
+ config_subentry_id: str | None = attr.ib()
domain: str = attr.ib(init=False, repr=False)
id: str = attr.ib()
orphaned_timestamp: float | None = attr.ib()
@@ -424,6 +428,7 @@ class DeletedRegistryEntry:
json_bytes(
{
"config_entry_id": self.config_entry_id,
+ "config_subentry_id": self.config_subentry_id,
"created_at": self.created_at,
"entity_id": self.entity_id,
"id": self.id,
@@ -539,6 +544,13 @@ class EntityRegistryStore(storage.Store[dict[str, list[dict[str, Any]]]]):
for entity in data["deleted_entities"]:
entity["created_at"] = entity["modified_at"] = created_at
+ if old_minor_version < 16:
+ # Version 1.16 adds config_subentry_id
+ for entity in data["entities"]:
+ entity["config_subentry_id"] = None
+ for entity in data["deleted_entities"]:
+ entity["config_subentry_id"] = None
+
if old_major_version > 1:
raise NotImplementedError
return data
@@ -647,10 +659,12 @@ def _validate_item(
platform: str,
*,
config_entry_id: str | None | UndefinedType = None,
+ config_subentry_id: str | None | UndefinedType = None,
device_id: str | None | UndefinedType = None,
disabled_by: RegistryEntryDisabler | None | UndefinedType = None,
entity_category: EntityCategory | None | UndefinedType = None,
hidden_by: RegistryEntryHider | None | UndefinedType = None,
+ old_config_subentry_id: str | None = None,
report_non_string_unique_id: bool = True,
unique_id: str | Hashable | UndefinedType | Any,
) -> None:
@@ -676,6 +690,26 @@ def _validate_item(
raise ValueError(
f"Can't link entity to unknown config entry {config_entry_id}"
)
+ if (
+ config_entry_id
+ and config_entry_id is not UNDEFINED
+ and old_config_subentry_id
+ and config_subentry_id is UNDEFINED
+ ):
+ raise ValueError("Can't change config entry without changing subentry")
+ if (
+ config_entry_id
+ and config_entry_id is not UNDEFINED
+ and config_subentry_id
+ and config_subentry_id is not UNDEFINED
+ ):
+ if (
+ not (config_entry := hass.config_entries.async_get_entry(config_entry_id))
+ or config_subentry_id not in config_entry.subentries
+ ):
+ raise ValueError(
+ f"Config entry {config_entry_id} has no subentry {config_subentry_id}"
+ )
if device_id and device_id is not UNDEFINED:
device_registry = dr.async_get(hass)
if not device_registry.async_get(device_id):
@@ -826,6 +860,7 @@ class EntityRegistry(BaseRegistry):
# Data that we want entry to have
capabilities: Mapping[str, Any] | None | UndefinedType = UNDEFINED,
config_entry: ConfigEntry | None | UndefinedType = UNDEFINED,
+ config_subentry_id: str | None | UndefinedType = UNDEFINED,
device_id: str | None | UndefinedType = UNDEFINED,
entity_category: EntityCategory | UndefinedType | None = UNDEFINED,
has_entity_name: bool | UndefinedType = UNDEFINED,
@@ -852,6 +887,7 @@ class EntityRegistry(BaseRegistry):
entity_id,
capabilities=capabilities,
config_entry_id=config_entry_id,
+ config_subentry_id=config_subentry_id,
device_id=device_id,
entity_category=entity_category,
has_entity_name=has_entity_name,
@@ -869,6 +905,7 @@ class EntityRegistry(BaseRegistry):
domain,
platform,
config_entry_id=config_entry_id,
+ config_subentry_id=config_subentry_id,
device_id=device_id,
disabled_by=disabled_by,
entity_category=entity_category,
@@ -907,6 +944,7 @@ class EntityRegistry(BaseRegistry):
entry = RegistryEntry(
capabilities=none_if_undefined(capabilities),
config_entry_id=none_if_undefined(config_entry_id),
+ config_subentry_id=none_if_undefined(config_subentry_id),
created_at=created_at,
device_id=none_if_undefined(device_id),
disabled_by=disabled_by,
@@ -949,6 +987,7 @@ class EntityRegistry(BaseRegistry):
orphaned_timestamp = None if config_entry_id else time.time()
self.deleted_entities[key] = DeletedRegistryEntry(
config_entry_id=config_entry_id,
+ config_subentry_id=entity.config_subentry_id,
created_at=entity.created_at,
entity_id=entity_id,
id=entity.id,
@@ -1008,6 +1047,20 @@ class EntityRegistry(BaseRegistry):
):
self.async_remove(entity.entity_id)
+ # Remove entities which belong to config subentries no longer associated with the
+ # device
+ entities = async_entries_for_device(
+ self, event.data["device_id"], include_disabled_entities=True
+ )
+ for entity in entities:
+ if (
+ (config_entry_id := entity.config_entry_id) is not None
+ and config_entry_id in device.config_entries
+ and entity.config_subentry_id
+ not in device.config_entries_subentries[config_entry_id]
+ ):
+ self.async_remove(entity.entity_id)
+
# Re-enable disabled entities if the device is no longer disabled
if not device.disabled:
entities = async_entries_for_device(
@@ -1041,6 +1094,7 @@ class EntityRegistry(BaseRegistry):
categories: dict[str, str] | UndefinedType = UNDEFINED,
capabilities: Mapping[str, Any] | None | UndefinedType = UNDEFINED,
config_entry_id: str | None | UndefinedType = UNDEFINED,
+ config_subentry_id: str | None | UndefinedType = UNDEFINED,
device_class: str | None | UndefinedType = UNDEFINED,
device_id: str | None | UndefinedType = UNDEFINED,
disabled_by: RegistryEntryDisabler | None | UndefinedType = UNDEFINED,
@@ -1073,6 +1127,7 @@ class EntityRegistry(BaseRegistry):
("categories", categories),
("capabilities", capabilities),
("config_entry_id", config_entry_id),
+ ("config_subentry_id", config_subentry_id),
("device_class", device_class),
("device_id", device_id),
("disabled_by", disabled_by),
@@ -1102,10 +1157,12 @@ class EntityRegistry(BaseRegistry):
old.domain,
old.platform,
config_entry_id=config_entry_id,
+ config_subentry_id=config_subentry_id,
device_id=device_id,
disabled_by=disabled_by,
entity_category=entity_category,
hidden_by=hidden_by,
+ old_config_subentry_id=old.config_subentry_id,
unique_id=new_unique_id,
)
@@ -1170,6 +1227,7 @@ class EntityRegistry(BaseRegistry):
categories: dict[str, str] | UndefinedType = UNDEFINED,
capabilities: Mapping[str, Any] | None | UndefinedType = UNDEFINED,
config_entry_id: str | None | UndefinedType = UNDEFINED,
+ config_subentry_id: str | None | UndefinedType = UNDEFINED,
device_class: str | None | UndefinedType = UNDEFINED,
device_id: str | None | UndefinedType = UNDEFINED,
disabled_by: RegistryEntryDisabler | None | UndefinedType = UNDEFINED,
@@ -1196,6 +1254,7 @@ class EntityRegistry(BaseRegistry):
categories=categories,
capabilities=capabilities,
config_entry_id=config_entry_id,
+ config_subentry_id=config_subentry_id,
device_class=device_class,
device_id=device_id,
disabled_by=disabled_by,
@@ -1222,6 +1281,7 @@ class EntityRegistry(BaseRegistry):
new_platform: str,
*,
new_config_entry_id: str | UndefinedType = UNDEFINED,
+ new_config_subentry_id: str | UndefinedType = UNDEFINED,
new_unique_id: str | UndefinedType = UNDEFINED,
new_device_id: str | None | UndefinedType = UNDEFINED,
) -> RegistryEntry:
@@ -1246,6 +1306,7 @@ class EntityRegistry(BaseRegistry):
entity_id,
new_unique_id=new_unique_id,
config_entry_id=new_config_entry_id,
+ config_subentry_id=new_config_subentry_id,
device_id=new_device_id,
platform=new_platform,
)
@@ -1308,6 +1369,7 @@ class EntityRegistry(BaseRegistry):
categories=entity["categories"],
capabilities=entity["capabilities"],
config_entry_id=entity["config_entry_id"],
+ config_subentry_id=entity["config_subentry_id"],
created_at=datetime.fromisoformat(entity["created_at"]),
device_class=entity["device_class"],
device_id=entity["device_id"],
@@ -1357,6 +1419,7 @@ class EntityRegistry(BaseRegistry):
)
deleted_entities[key] = DeletedRegistryEntry(
config_entry_id=entity["config_entry_id"],
+ config_subentry_id=entity["config_subentry_id"],
created_at=datetime.fromisoformat(entity["created_at"]),
entity_id=entity["entity_id"],
id=entity["id"],
@@ -1415,6 +1478,30 @@ class EntityRegistry(BaseRegistry):
)
self.async_schedule_save()
+ @callback
+ def async_clear_config_subentry(
+ self, config_entry_id: str, config_subentry_id: str
+ ) -> None:
+ """Clear config subentry from registry entries."""
+ now_time = time.time()
+ for entity_id in [
+ entry.entity_id
+ for entry in self.entities.get_entries_for_config_entry_id(config_entry_id)
+ if entry.config_subentry_id == config_subentry_id
+ ]:
+ self.async_remove(entity_id)
+ for key, deleted_entity in list(self.deleted_entities.items()):
+ if config_subentry_id != deleted_entity.config_subentry_id:
+ continue
+ # Add a time stamp when the deleted entity became orphaned
+ self.deleted_entities[key] = attr.evolve(
+ deleted_entity,
+ orphaned_timestamp=now_time,
+ config_entry_id=None,
+ config_subentry_id=None,
+ )
+ self.async_schedule_save()
+
@callback
def async_purge_expired_orphaned_entities(self) -> None:
"""Purge expired orphaned entities from the registry.
diff --git a/homeassistant/helpers/frame.py b/homeassistant/helpers/frame.py
index f33f8407e47..ca7b097d90d 100644
--- a/homeassistant/helpers/frame.py
+++ b/homeassistant/helpers/frame.py
@@ -10,18 +10,20 @@ import functools
import linecache
import logging
import sys
+import threading
from types import FrameType
from typing import Any, cast
from propcache.api import cached_property
-from homeassistant.core import HomeAssistant, async_get_hass_or_none
+from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.loader import (
Integration,
async_get_issue_integration,
async_suggest_report_issue,
)
+from homeassistant.util.async_ import run_callback_threadsafe
_LOGGER = logging.getLogger(__name__)
@@ -29,6 +31,21 @@ _LOGGER = logging.getLogger(__name__)
_REPORTED_INTEGRATIONS: set[str] = set()
+class _Hass:
+ """Container which makes a HomeAssistant instance available to frame helper."""
+
+ hass: HomeAssistant | None = None
+
+
+_hass = _Hass()
+
+
+@callback
+def async_setup(hass: HomeAssistant) -> None:
+ """Set up the frame helper."""
+ _hass.hass = hass
+
+
@dataclass(kw_only=True)
class IntegrationFrame:
"""Integration frame container."""
@@ -133,44 +150,6 @@ class MissingIntegrationFrame(HomeAssistantError):
"""Raised when no integration is found in the frame."""
-def report(
- what: str,
- *,
- exclude_integrations: set[str] | None = None,
- error_if_core: bool = True,
- error_if_integration: bool = False,
- level: int = logging.WARNING,
- log_custom_component_only: bool = False,
-) -> None:
- """Report incorrect usage.
-
- If error_if_core is True, raise instead of log if an integration is not found
- when unwinding the stack frame.
- If error_if_integration is True, raise instead of log if an integration is found
- when unwinding the stack frame.
- """
- core_behavior = ReportBehavior.ERROR if error_if_core else ReportBehavior.LOG
- core_integration_behavior = (
- ReportBehavior.ERROR if error_if_integration else ReportBehavior.LOG
- )
- custom_integration_behavior = core_integration_behavior
-
- if log_custom_component_only:
- if core_behavior is ReportBehavior.LOG:
- core_behavior = ReportBehavior.IGNORE
- if core_integration_behavior is ReportBehavior.LOG:
- core_integration_behavior = ReportBehavior.IGNORE
-
- report_usage(
- what,
- core_behavior=core_behavior,
- core_integration_behavior=core_integration_behavior,
- custom_integration_behavior=custom_integration_behavior,
- exclude_integrations=exclude_integrations,
- level=level,
- )
-
-
class ReportBehavior(enum.Enum):
"""Enum for behavior on code usage."""
@@ -201,18 +180,49 @@ def report_usage(
breaking version
:param exclude_integrations: skip specified integration when reviewing the stack.
If no integration is found, the core behavior will be applied
- :param integration_domain: fallback for identifying the integration if the
- frame is not found
+ :param integration_domain: domain of the integration causing the issue. If None, the
+ stack frame will be searched to identify the integration causing the issue.
"""
- try:
- integration_frame = get_integration_frame(
- exclude_integrations=exclude_integrations
- )
- except MissingIntegrationFrame as err:
- if integration := async_get_issue_integration(
- hass := async_get_hass_or_none(), integration_domain
- ):
- _report_integration_domain(
+ if (hass := _hass.hass) is None:
+ raise RuntimeError("Frame helper not set up")
+ _report_usage_partial = functools.partial(
+ _report_usage,
+ hass,
+ what,
+ breaks_in_ha_version=breaks_in_ha_version,
+ core_behavior=core_behavior,
+ core_integration_behavior=core_integration_behavior,
+ custom_integration_behavior=custom_integration_behavior,
+ exclude_integrations=exclude_integrations,
+ integration_domain=integration_domain,
+ level=level,
+ )
+ if hass.loop_thread_id != threading.get_ident():
+ future = run_callback_threadsafe(hass.loop, _report_usage_partial)
+ future.result()
+ return
+ _report_usage_partial()
+
+
+def _report_usage(
+ hass: HomeAssistant,
+ what: str,
+ *,
+ breaks_in_ha_version: str | None,
+ core_behavior: ReportBehavior,
+ core_integration_behavior: ReportBehavior,
+ custom_integration_behavior: ReportBehavior,
+ exclude_integrations: set[str] | None,
+ integration_domain: str | None,
+ level: int,
+) -> None:
+ """Report incorrect code usage.
+
+ Must be called from the event loop.
+ """
+ if integration_domain:
+ if integration := async_get_issue_integration(hass, integration_domain):
+ _report_usage_integration_domain(
hass,
what,
breaks_in_ha_version,
@@ -222,16 +232,15 @@ def report_usage(
level,
)
return
- msg = f"Detected code that {what}. Please report this issue"
- if core_behavior is ReportBehavior.ERROR:
- raise RuntimeError(msg) from err
- if core_behavior is ReportBehavior.LOG:
- if breaks_in_ha_version:
- msg = (
- f"Detected code that {what}. This will stop working in Home "
- f"Assistant {breaks_in_ha_version}, please report this issue"
- )
- _LOGGER.warning(msg, stack_info=True)
+ _report_usage_no_integration(what, core_behavior, breaks_in_ha_version, None)
+ return
+
+ try:
+ integration_frame = get_integration_frame(
+ exclude_integrations=exclude_integrations
+ )
+ except MissingIntegrationFrame as err:
+ _report_usage_no_integration(what, core_behavior, breaks_in_ha_version, err)
return
integration_behavior = core_integration_behavior
@@ -239,7 +248,8 @@ def report_usage(
integration_behavior = custom_integration_behavior
if integration_behavior is not ReportBehavior.IGNORE:
- _report_integration_frame(
+ _report_usage_integration_frame(
+ hass,
what,
breaks_in_ha_version,
integration_frame,
@@ -248,7 +258,7 @@ def report_usage(
)
-def _report_integration_domain(
+def _report_usage_integration_domain(
hass: HomeAssistant | None,
what: str,
breaks_in_ha_version: str | None,
@@ -298,7 +308,8 @@ def _report_integration_domain(
)
-def _report_integration_frame(
+def _report_usage_integration_frame(
+ hass: HomeAssistant,
what: str,
breaks_in_ha_version: str | None,
integration_frame: IntegrationFrame,
@@ -316,7 +327,7 @@ def _report_integration_frame(
_REPORTED_INTEGRATIONS.add(key)
report_issue = async_suggest_report_issue(
- async_get_hass_or_none(),
+ hass,
integration_domain=integration_frame.integration,
module=integration_frame.module,
)
@@ -346,31 +357,55 @@ def _report_integration_frame(
)
+def _report_usage_no_integration(
+ what: str,
+ core_behavior: ReportBehavior,
+ breaks_in_ha_version: str | None,
+ err: MissingIntegrationFrame | None,
+) -> None:
+ """Report incorrect usage without an integration.
+
+ This could happen because the offending call happened outside of an integration,
+ or because the integration could not be identified.
+ """
+ msg = f"Detected code that {what}. Please report this issue"
+ if core_behavior is ReportBehavior.ERROR:
+ raise RuntimeError(msg) from err
+ if core_behavior is ReportBehavior.LOG:
+ if breaks_in_ha_version:
+ msg = (
+ f"Detected code that {what}. This will stop working in Home "
+ f"Assistant {breaks_in_ha_version}, please report this issue"
+ )
+ _LOGGER.warning(msg, stack_info=True)
+
+
def warn_use[_CallableT: Callable](func: _CallableT, what: str) -> _CallableT:
"""Mock a function to warn when it was about to be used."""
if asyncio.iscoroutinefunction(func):
@functools.wraps(func)
async def report_use(*args: Any, **kwargs: Any) -> None:
- report(what)
+ report_usage(what)
else:
@functools.wraps(func)
def report_use(*args: Any, **kwargs: Any) -> None:
- report(what)
+ report_usage(what)
return cast(_CallableT, report_use)
def report_non_thread_safe_operation(what: str) -> None:
"""Report a non-thread safe operation."""
- report(
+ report_usage(
f"calls {what} from a thread other than the event loop, "
"which may cause Home Assistant to crash or data to corrupt. "
"For more information, see "
"https://developers.home-assistant.io/docs/asyncio_thread_safety/"
f"#{what.replace('.', '')}",
- error_if_core=True,
- error_if_integration=True,
+ core_behavior=ReportBehavior.ERROR,
+ core_integration_behavior=ReportBehavior.ERROR,
+ custom_integration_behavior=ReportBehavior.ERROR,
)
diff --git a/homeassistant/helpers/intent.py b/homeassistant/helpers/intent.py
index c93545ed414..75572194bb8 100644
--- a/homeassistant/helpers/intent.py
+++ b/homeassistant/helpers/intent.py
@@ -38,7 +38,7 @@ from .typing import VolSchemaType
_LOGGER = logging.getLogger(__name__)
type _SlotsType = dict[str, Any]
type _IntentSlotsType = dict[
- str | tuple[str, str], VolSchemaType | Callable[[Any], Any]
+ str | tuple[str, str], IntentSlotInfo | VolSchemaType | Callable[[Any], Any]
]
INTENT_TURN_OFF = "HassTurnOff"
@@ -59,6 +59,7 @@ INTENT_GET_CURRENT_DATE = "HassGetCurrentDate"
INTENT_GET_CURRENT_TIME = "HassGetCurrentTime"
INTENT_RESPOND = "HassRespond"
INTENT_BROADCAST = "HassBroadcast"
+INTENT_GET_TEMPERATURE = "HassClimateGetTemperature"
SLOT_SCHEMA = vol.Schema({}, extra=vol.ALLOW_EXTRA)
@@ -506,12 +507,22 @@ def _add_areas(
candidate.area = areas.async_get_area(candidate.device.area_id)
+def _default_area_candidate_filter(
+ candidate: MatchTargetsCandidate, possible_area_ids: Collection[str]
+) -> bool:
+ """Keep candidates in the possible areas."""
+ return (candidate.area is not None) and (candidate.area.id in possible_area_ids)
+
+
@callback
def async_match_targets( # noqa: C901
hass: HomeAssistant,
constraints: MatchTargetsConstraints,
preferences: MatchTargetsPreferences | None = None,
states: list[State] | None = None,
+ area_candidate_filter: Callable[
+ [MatchTargetsCandidate, Collection[str]], bool
+ ] = _default_area_candidate_filter,
) -> MatchTargetsResult:
"""Match entities based on constraints in order to handle an intent."""
preferences = preferences or MatchTargetsPreferences()
@@ -622,9 +633,7 @@ def async_match_targets( # noqa: C901
}
candidates = [
- c
- for c in candidates
- if (c.area is not None) and (c.area.id in possible_area_ids)
+ c for c in candidates if area_candidate_filter(c, possible_area_ids)
]
if not candidates:
return MatchTargetsResult(
@@ -648,9 +657,7 @@ def async_match_targets( # noqa: C901
# May be constrained by floors above
possible_area_ids.intersection_update(matching_area_ids)
candidates = [
- c
- for c in candidates
- if (c.area is not None) and (c.area.id in possible_area_ids)
+ c for c in candidates if area_candidate_filter(c, possible_area_ids)
]
if not candidates:
return MatchTargetsResult(
@@ -700,7 +707,7 @@ def async_match_targets( # noqa: C901
group_candidates = [
c
for c in group_candidates
- if (c.area is not None) and (c.area.id == preferences.area_id)
+ if area_candidate_filter(c, {preferences.area_id})
]
if len(group_candidates) < 2:
# Disambiguated by area
@@ -746,7 +753,7 @@ def async_match_targets( # noqa: C901
if preferences.area_id:
# Filter by area
filtered_candidates = [
- c for c in candidates if c.area and (c.area.id == preferences.area_id)
+ c for c in candidates if area_candidate_filter(c, {preferences.area_id})
]
if (len(filtered_candidates) > 1) and preferences.floor_id:
@@ -867,6 +874,34 @@ def non_empty_string(value: Any) -> str:
return value_str
+@dataclass(kw_only=True)
+class IntentSlotInfo:
+ """Details about how intent slots are processed and validated."""
+
+ service_data_name: str | None = None
+ """Optional name of the service data input to map to this slot."""
+
+ description: str | None = None
+ """Human readable description of the slot."""
+
+ value_schema: VolSchemaType | Callable[[Any], Any] = vol.Any
+ """Validator for the slot."""
+
+
+def _convert_slot_info(
+ key: str | tuple[str, str],
+ value: IntentSlotInfo | VolSchemaType | Callable[[Any], Any],
+) -> tuple[str, IntentSlotInfo]:
+ """Create an IntentSlotInfo from the various supported input arguments."""
+ if isinstance(value, IntentSlotInfo):
+ if not isinstance(key, str):
+ raise TypeError("Tuple key and IntentSlotDescription value not supported")
+ return key, value
+ if isinstance(key, tuple):
+ return key[0], IntentSlotInfo(service_data_name=key[1], value_schema=value)
+ return key, IntentSlotInfo(value_schema=value)
+
+
class DynamicServiceIntentHandler(IntentHandler):
"""Service Intent handler registration (dynamic).
@@ -900,23 +935,14 @@ class DynamicServiceIntentHandler(IntentHandler):
self.platforms = platforms
self.device_classes = device_classes
- self.required_slots: _IntentSlotsType = {}
- if required_slots:
- for key, value_schema in required_slots.items():
- if isinstance(key, str):
- # Slot name/service data key
- key = (key, key)
-
- self.required_slots[key] = value_schema
-
- self.optional_slots: _IntentSlotsType = {}
- if optional_slots:
- for key, value_schema in optional_slots.items():
- if isinstance(key, str):
- # Slot name/service data key
- key = (key, key)
-
- self.optional_slots[key] = value_schema
+ self.required_slots: dict[str, IntentSlotInfo] = dict(
+ _convert_slot_info(key, value)
+ for key, value in (required_slots or {}).items()
+ )
+ self.optional_slots: dict[str, IntentSlotInfo] = dict(
+ _convert_slot_info(key, value)
+ for key, value in (optional_slots or {}).items()
+ )
@cached_property
def slot_schema(self) -> dict:
@@ -957,16 +983,20 @@ class DynamicServiceIntentHandler(IntentHandler):
if self.required_slots:
slot_schema.update(
{
- vol.Required(key[0]): validator
- for key, validator in self.required_slots.items()
+ vol.Required(
+ key, description=slot_info.description
+ ): slot_info.value_schema
+ for key, slot_info in self.required_slots.items()
}
)
if self.optional_slots:
slot_schema.update(
{
- vol.Optional(key[0]): validator
- for key, validator in self.optional_slots.items()
+ vol.Optional(
+ key, description=slot_info.description
+ ): slot_info.value_schema
+ for key, slot_info in self.optional_slots.items()
}
)
@@ -1149,18 +1179,15 @@ class DynamicServiceIntentHandler(IntentHandler):
service_data: dict[str, Any] = {ATTR_ENTITY_ID: state.entity_id}
if self.required_slots:
- service_data.update(
- {
- key[1]: intent_obj.slots[key[0]]["value"]
- for key in self.required_slots
- }
- )
+ for key, slot_info in self.required_slots.items():
+ service_data[slot_info.service_data_name or key] = intent_obj.slots[
+ key
+ ]["value"]
if self.optional_slots:
- for key in self.optional_slots:
- value = intent_obj.slots.get(key[0])
- if value:
- service_data[key[1]] = value["value"]
+ for key, slot_info in self.optional_slots.items():
+ if value := intent_obj.slots.get(key):
+ service_data[slot_info.service_data_name or key] = value["value"]
await self._run_then_background(
hass.async_create_task_internal(
diff --git a/homeassistant/helpers/llm.py b/homeassistant/helpers/llm.py
index 2ef785e7f71..7f6fe22ec70 100644
--- a/homeassistant/helpers/llm.py
+++ b/homeassistant/helpers/llm.py
@@ -19,7 +19,6 @@ from homeassistant.components.calendar import (
DOMAIN as CALENDAR_DOMAIN,
SERVICE_GET_EVENTS,
)
-from homeassistant.components.climate import INTENT_GET_TEMPERATURE
from homeassistant.components.cover import INTENT_CLOSE_COVER, INTENT_OPEN_COVER
from homeassistant.components.homeassistant import async_should_expose
from homeassistant.components.intent import async_device_supports_timers
@@ -67,6 +66,11 @@ Answer questions about the world truthfully.
Answer in plain text. Keep it simple and to the point.
"""
+NO_ENTITIES_PROMPT = (
+ "Only if the user wants to control a device, tell them to expose entities "
+ "to their voice assistant in Home Assistant."
+)
+
@callback
def async_render_no_api_prompt(hass: HomeAssistant) -> str:
@@ -285,7 +289,7 @@ class AssistAPI(API):
"""API exposing Assist API to LLMs."""
IGNORE_INTENTS = {
- INTENT_GET_TEMPERATURE,
+ intent.INTENT_GET_TEMPERATURE,
INTENT_GET_WEATHER,
INTENT_OPEN_COVER, # deprecated
INTENT_CLOSE_COVER, # deprecated
@@ -312,7 +316,7 @@ class AssistAPI(API):
"""Return the instance of the API."""
if llm_context.assistant:
exposed_entities: dict | None = _get_exposed_entities(
- self.hass, llm_context.assistant
+ self.hass, llm_context.assistant, include_state=False
)
else:
exposed_entities = None
@@ -330,10 +334,7 @@ class AssistAPI(API):
self, llm_context: LLMContext, exposed_entities: dict | None
) -> str:
if not exposed_entities or not exposed_entities["entities"]:
- return (
- "Only if the user wants to control a device, tell them to expose entities "
- "to their voice assistant in Home Assistant."
- )
+ return NO_ENTITIES_PROMPT
return "\n".join(
[
*self._async_get_preable(llm_context),
@@ -455,11 +456,16 @@ class AssistAPI(API):
for script_entity_id in exposed_entities[SCRIPT_DOMAIN]
)
+ if exposed_domains:
+ tools.append(GetHomeStateTool())
+
return tools
def _get_exposed_entities(
- hass: HomeAssistant, assistant: str
+ hass: HomeAssistant,
+ assistant: str,
+ include_state: bool = True,
) -> dict[str, dict[str, dict[str, Any]]]:
"""Get exposed entities.
@@ -520,22 +526,28 @@ def _get_exposed_entities(
info: dict[str, Any] = {
"names": ", ".join(names),
"domain": state.domain,
- "state": state.state,
}
+ if include_state:
+ info["state"] = state.state
+
if description:
info["description"] = description
if area_names:
info["areas"] = ", ".join(area_names)
- if attributes := {
- attr_name: str(attr_value)
- if isinstance(attr_value, (Enum, Decimal, int))
- else attr_value
- for attr_name, attr_value in state.attributes.items()
- if attr_name in interesting_attributes
- }:
+ if include_state and (
+ attributes := {
+ attr_name: (
+ str(attr_value)
+ if isinstance(attr_value, (Enum, Decimal, int))
+ else attr_value
+ )
+ for attr_name, attr_value in state.attributes.items()
+ if attr_name in interesting_attributes
+ }
+ ):
info["attributes"] = attributes
if state.domain in data:
@@ -884,3 +896,39 @@ class CalendarGetEventsTool(Tool):
]
return {"success": True, "result": events}
+
+
+class GetHomeStateTool(Tool):
+ """Tool for getting the current state of exposed entities.
+
+ This returns state for all entities that have been exposed to
+ the assistant. This is different than the GetState intent, which
+ returns state for entities based on intent parameters.
+ """
+
+ name = "get_home_state"
+ description = "Get the current state of all devices in the home. "
+
+ async def async_call(
+ self,
+ hass: HomeAssistant,
+ tool_input: ToolInput,
+ llm_context: LLMContext,
+ ) -> JsonObjectType:
+ """Get the current state of exposed entities."""
+ if llm_context.assistant is None:
+ # Note this doesn't happen in practice since this tool won't be
+ # exposed if no assistant is configured.
+ return {"success": False, "error": "No assistant configured"}
+
+ exposed_entities = _get_exposed_entities(hass, llm_context.assistant)
+ if not exposed_entities["entities"]:
+ return {"success": False, "error": NO_ENTITIES_PROMPT}
+ prompt = [
+ "An overview of the areas and the devices in this smart home:",
+ yaml_util.dump(list(exposed_entities["entities"].values())),
+ ]
+ return {
+ "success": True,
+ "result": "\n".join(prompt),
+ }
diff --git a/homeassistant/helpers/recorder.py b/homeassistant/helpers/recorder.py
index 59604944eeb..7ad319419c1 100644
--- a/homeassistant/helpers/recorder.py
+++ b/homeassistant/helpers/recorder.py
@@ -20,7 +20,7 @@ if TYPE_CHECKING:
_LOGGER = logging.getLogger(__name__)
-DOMAIN: HassKey[RecorderData] = HassKey("recorder")
+DATA_RECORDER: HassKey[RecorderData] = HassKey("recorder")
DATA_INSTANCE: HassKey[Recorder] = HassKey("recorder_instance")
@@ -52,24 +52,19 @@ def async_migration_is_live(hass: HomeAssistant) -> bool:
@callback
def async_initialize_recorder(hass: HomeAssistant) -> None:
- """Initialize recorder data."""
+ """Initialize recorder data.
+
+ This creates the RecorderData instance stored in hass.data[DATA_RECORDER] and
+ registers the basic recorder websocket API which is used by frontend to determine
+ if the recorder is migrating the database.
+ """
# pylint: disable-next=import-outside-toplevel
from homeassistant.components.recorder.basic_websocket_api import async_setup
- hass.data[DOMAIN] = RecorderData()
+ hass.data[DATA_RECORDER] = RecorderData()
async_setup(hass)
-async def async_wait_recorder(hass: HomeAssistant) -> bool:
- """Wait for recorder to initialize and return connection status.
-
- Returns False immediately if the recorder is not enabled.
- """
- if DOMAIN not in hass.data:
- return False
- return await hass.data[DOMAIN].db_connected
-
-
@functools.lru_cache(maxsize=1)
def get_instance(hass: HomeAssistant) -> Recorder:
"""Get the recorder instance."""
diff --git a/homeassistant/helpers/script.py b/homeassistant/helpers/script.py
index bd3babc8793..1242ef3e4d5 100644
--- a/homeassistant/helpers/script.py
+++ b/homeassistant/helpers/script.py
@@ -12,7 +12,6 @@ from datetime import datetime, timedelta
from functools import partial
import itertools
import logging
-from types import MappingProxyType
from typing import Any, Literal, TypedDict, cast, overload
import async_interrupt
@@ -90,7 +89,7 @@ from . import condition, config_validation as cv, service, template
from .condition import ConditionCheckerType, trace_condition_function
from .dispatcher import async_dispatcher_connect, async_dispatcher_send_internal
from .event import async_call_later, async_track_template
-from .script_variables import ScriptVariables
+from .script_variables import ScriptRunVariables, ScriptVariables
from .template import Template
from .trace import (
TraceElement,
@@ -177,7 +176,7 @@ def _set_result_unless_done(future: asyncio.Future[None]) -> None:
future.set_result(None)
-def action_trace_append(variables: dict[str, Any], path: str) -> TraceElement:
+def action_trace_append(variables: TemplateVarsType, path: str) -> TraceElement:
"""Append a TraceElement to trace[path]."""
trace_element = TraceElement(variables, path)
trace_append_element(trace_element, ACTION_TRACE_NODE_MAX_LEN)
@@ -189,7 +188,7 @@ async def trace_action(
hass: HomeAssistant,
script_run: _ScriptRun,
stop: asyncio.Future[None],
- variables: dict[str, Any],
+ variables: TemplateVarsType,
) -> AsyncGenerator[TraceElement]:
"""Trace action execution."""
path = trace_path_get()
@@ -411,7 +410,7 @@ class _ScriptRun:
self,
hass: HomeAssistant,
script: Script,
- variables: dict[str, Any],
+ variables: ScriptRunVariables,
context: Context | None,
log_exceptions: bool,
) -> None:
@@ -430,9 +429,6 @@ class _ScriptRun:
if not self._stop.done():
self._script._changed() # noqa: SLF001
- async def _async_get_condition(self, config: ConfigType) -> ConditionCheckerType:
- return await self._script._async_get_condition(config) # noqa: SLF001
-
def _log(
self, msg: str, *args: Any, level: int = logging.INFO, **kwargs: Any
) -> None:
@@ -488,14 +484,16 @@ class _ScriptRun:
script_stack.pop()
self._finish()
- return ScriptRunResult(self._conversation_response, response, self._variables)
+ return ScriptRunResult(
+ self._conversation_response, response, self._variables.local_scope
+ )
async def _async_step(self, log_exceptions: bool) -> None:
continue_on_error = self._action.get(CONF_CONTINUE_ON_ERROR, False)
with trace_path(str(self._step)):
async with trace_action(
- self._hass, self, self._stop, self._variables
+ self._hass, self, self._stop, self._variables.non_parallel_scope
) as trace_element:
if self._stop.done():
return
@@ -521,7 +519,7 @@ class _ScriptRun:
trace_set_result(enabled=False)
return
- handler = f"_async_{action}_step"
+ handler = f"_async_step_{action}"
try:
await getattr(self, handler)()
except Exception as ex: # noqa: BLE001
@@ -529,7 +527,7 @@ class _ScriptRun:
ex, continue_on_error, self._log_exceptions or log_exceptions
)
finally:
- trace_element.update_variables(self._variables)
+ trace_element.update_variables(self._variables.non_parallel_scope)
def _finish(self) -> None:
self._script._runs.remove(self) # noqa: SLF001
@@ -614,107 +612,6 @@ class _ScriptRun:
level=level,
)
- def _get_pos_time_period_template(self, key: str) -> timedelta:
- try:
- return cv.positive_time_period( # type: ignore[no-any-return]
- template.render_complex(self._action[key], self._variables)
- )
- except (exceptions.TemplateError, vol.Invalid) as ex:
- self._log(
- "Error rendering %s %s template: %s",
- self._script.name,
- key,
- ex,
- level=logging.ERROR,
- )
- raise _AbortScript from ex
-
- async def _async_delay_step(self) -> None:
- """Handle delay."""
- delay_delta = self._get_pos_time_period_template(CONF_DELAY)
-
- self._step_log(f"delay {delay_delta}")
-
- delay = delay_delta.total_seconds()
- self._changed()
- if not delay:
- # Handle an empty delay
- trace_set_result(delay=delay, done=True)
- return
-
- trace_set_result(delay=delay, done=False)
- futures, timeout_handle, timeout_future = self._async_futures_with_timeout(
- delay
- )
-
- try:
- await asyncio.wait(futures, return_when=asyncio.FIRST_COMPLETED)
- finally:
- if timeout_future.done():
- trace_set_result(delay=delay, done=True)
- else:
- timeout_handle.cancel()
-
- def _get_timeout_seconds_from_action(self) -> float | None:
- """Get the timeout from the action."""
- if CONF_TIMEOUT in self._action:
- return self._get_pos_time_period_template(CONF_TIMEOUT).total_seconds()
- return None
-
- async def _async_wait_template_step(self) -> None:
- """Handle a wait template."""
- timeout = self._get_timeout_seconds_from_action()
- self._step_log("wait template", timeout)
-
- self._variables["wait"] = {"remaining": timeout, "completed": False}
- trace_set_result(wait=self._variables["wait"])
-
- wait_template = self._action[CONF_WAIT_TEMPLATE]
-
- # check if condition already okay
- if condition.async_template(self._hass, wait_template, self._variables, False):
- self._variables["wait"]["completed"] = True
- self._changed()
- return
-
- if timeout == 0:
- self._changed()
- self._async_handle_timeout()
- return
-
- futures, timeout_handle, timeout_future = self._async_futures_with_timeout(
- timeout
- )
- done = self._hass.loop.create_future()
- futures.append(done)
-
- @callback
- def async_script_wait(
- entity_id: str, from_s: State | None, to_s: State | None
- ) -> None:
- """Handle script after template condition is true."""
- self._async_set_remaining_time_var(timeout_handle)
- self._variables["wait"]["completed"] = True
- _set_result_unless_done(done)
-
- unsub = async_track_template(
- self._hass, wait_template, async_script_wait, self._variables
- )
- self._changed()
- await self._async_wait_with_optional_timeout(
- futures, timeout_handle, timeout_future, unsub
- )
-
- def _async_set_remaining_time_var(
- self, timeout_handle: asyncio.TimerHandle | None
- ) -> None:
- """Set the remaining time variable for a wait step."""
- wait_var = self._variables["wait"]
- if timeout_handle:
- wait_var["remaining"] = timeout_handle.when() - self._hass.loop.time()
- else:
- wait_var["remaining"] = None
-
async def _async_run_long_action[_T](
self, long_task: asyncio.Task[_T]
) -> _T | None:
@@ -728,111 +625,54 @@ class _ScriptRun:
except ScriptStoppedError as ex:
raise asyncio.CancelledError from ex
- async def _async_call_service_step(self) -> None:
- """Call the service specified in the action."""
- self._step_log("call service")
-
- params = service.async_prepare_call_from_config(
- self._hass, self._action, self._variables
- )
-
- # Validate response data parameters. This check ignores services that do
- # not exist which will raise an appropriate error in the service call below.
- response_variable = self._action.get(CONF_RESPONSE_VARIABLE)
- return_response = response_variable is not None
- if self._hass.services.has_service(params[CONF_DOMAIN], params[CONF_SERVICE]):
- supports_response = self._hass.services.supports_response(
- params[CONF_DOMAIN], params[CONF_SERVICE]
- )
- if supports_response == SupportsResponse.ONLY and not return_response:
- raise vol.Invalid(
- f"Script requires '{CONF_RESPONSE_VARIABLE}' for response data "
- f"for service call {params[CONF_DOMAIN]}.{params[CONF_SERVICE]}"
- )
- if supports_response == SupportsResponse.NONE and return_response:
- raise vol.Invalid(
- f"Script does not support '{CONF_RESPONSE_VARIABLE}' for service "
- f"'{CONF_RESPONSE_VARIABLE}' which does not support response data."
- )
-
- running_script = (
- params[CONF_DOMAIN] == "automation" and params[CONF_SERVICE] == "trigger"
- ) or params[CONF_DOMAIN] in ("python_script", "script")
- trace_set_result(params=params, running_script=running_script)
- response_data = await self._async_run_long_action(
+ async def _async_run_script(
+ self, script: Script, *, parallel: bool = False
+ ) -> None:
+ """Execute a script."""
+ result = await self._async_run_long_action(
self._hass.async_create_task_internal(
- self._hass.services.async_call(
- **params,
- blocking=True,
- context=self._context,
- return_response=return_response,
+ script.async_run(
+ self._variables.enter_scope(parallel=parallel), self._context
),
eager_start=True,
)
)
- if response_variable:
- self._variables[response_variable] = response_data
+ if result and result.conversation_response is not UNDEFINED:
+ self._conversation_response = result.conversation_response
- async def _async_device_step(self) -> None:
- """Perform the device automation specified in the action."""
- self._step_log("device automation")
- await device_action.async_call_action_from_config(
- self._hass, self._action, self._variables, self._context
+ ## Flow control actions ##
+
+ ### Sequence actions ###
+
+ @async_trace_path("parallel")
+ async def _async_step_parallel(self) -> None:
+ """Run a sequence in parallel."""
+ scripts = await self._script._async_get_parallel_scripts(self._step) # noqa: SLF001
+
+ async def async_run_with_trace(idx: int, script: Script) -> None:
+ """Run a script with a trace path."""
+ trace_path_stack_cv.set(copy(trace_path_stack_cv.get()))
+ with trace_path([str(idx), "sequence"]):
+ await self._async_run_script(script, parallel=True)
+
+ results = await asyncio.gather(
+ *(async_run_with_trace(idx, script) for idx, script in enumerate(scripts)),
+ return_exceptions=True,
)
+ for result in results:
+ if isinstance(result, Exception):
+ raise result
- async def _async_scene_step(self) -> None:
- """Activate the scene specified in the action."""
- self._step_log("activate scene")
- trace_set_result(scene=self._action[CONF_SCENE])
- await self._hass.services.async_call(
- scene.DOMAIN,
- SERVICE_TURN_ON,
- {ATTR_ENTITY_ID: self._action[CONF_SCENE]},
- blocking=True,
- context=self._context,
- )
+ @async_trace_path("sequence")
+ async def _async_step_sequence(self) -> None:
+ """Run a sequence."""
+ sequence = await self._script._async_get_sequence_script(self._step) # noqa: SLF001
+ await self._async_run_script(sequence)
- async def _async_event_step(self) -> None:
- """Fire an event."""
- self._step_log(self._action.get(CONF_ALIAS, self._action[CONF_EVENT]))
- event_data = {}
- for conf in (CONF_EVENT_DATA, CONF_EVENT_DATA_TEMPLATE):
- if conf not in self._action:
- continue
+ ### Condition actions ###
- try:
- event_data.update(
- template.render_complex(self._action[conf], self._variables)
- )
- except exceptions.TemplateError as ex:
- self._log(
- "Error rendering event data template: %s", ex, level=logging.ERROR
- )
-
- trace_set_result(event=self._action[CONF_EVENT], event_data=event_data)
- self._hass.bus.async_fire_internal(
- self._action[CONF_EVENT], event_data, context=self._context
- )
-
- async def _async_condition_step(self) -> None:
- """Test if condition is matching."""
- self._script.last_action = self._action.get(
- CONF_ALIAS, self._action[CONF_CONDITION]
- )
- cond = await self._async_get_condition(self._action)
- try:
- trace_element = trace_stack_top(trace_stack_cv)
- if trace_element:
- trace_element.reuse_by_child = True
- check = cond(self._hass, self._variables)
- except exceptions.ConditionError as ex:
- _LOGGER.warning("Error in 'condition' evaluation:\n%s", ex)
- check = False
-
- self._log("Test condition %s: %s", self._script.last_action, check)
- trace_update_result(result=check)
- if not check:
- raise _ConditionFail
+ async def _async_get_condition(self, config: ConfigType) -> ConditionCheckerType:
+ return await self._script._async_get_condition(config) # noqa: SLF001
def _test_conditions(
self,
@@ -861,14 +701,76 @@ class _ScriptRun:
return traced_test_conditions(self._hass, self._variables)
- @async_trace_path("repeat")
- async def _async_repeat_step(self) -> None: # noqa: C901
- """Repeat a sequence."""
+ async def _async_step_choose(self) -> None:
+ """Choose a sequence."""
+ choose_data = await self._script._async_get_choose_data(self._step) # noqa: SLF001
+
+ with trace_path("choose"):
+ for idx, (conditions, script) in enumerate(choose_data["choices"]):
+ with trace_path(str(idx)):
+ try:
+ if self._test_conditions(conditions, "choose", "conditions"):
+ trace_set_result(choice=idx)
+ with trace_path("sequence"):
+ await self._async_run_script(script)
+ return
+ except exceptions.ConditionError as ex:
+ _LOGGER.warning("Error in 'choose' evaluation:\n%s", ex)
+
+ if choose_data["default"] is not None:
+ trace_set_result(choice="default")
+ with trace_path(["default"]):
+ await self._async_run_script(choose_data["default"])
+
+ async def _async_step_condition(self) -> None:
+ """Test if condition is matching."""
+ self._script.last_action = self._action.get(
+ CONF_ALIAS, self._action[CONF_CONDITION]
+ )
+ cond = await self._async_get_condition(self._action)
+ try:
+ trace_element = trace_stack_top(trace_stack_cv)
+ if trace_element:
+ trace_element.reuse_by_child = True
+ check = cond(self._hass, self._variables)
+ except exceptions.ConditionError as ex:
+ _LOGGER.warning("Error in 'condition' evaluation:\n%s", ex)
+ check = False
+
+ self._log("Test condition %s: %s", self._script.last_action, check)
+ trace_update_result(result=check)
+ if not check:
+ raise _ConditionFail
+
+ async def _async_step_if(self) -> None:
+ """If sequence."""
+ if_data = await self._script._async_get_if_data(self._step) # noqa: SLF001
+
+ test_conditions: bool | None = False
+ try:
+ with trace_path("if"):
+ test_conditions = self._test_conditions(
+ if_data["if_conditions"], "if", "condition"
+ )
+ except exceptions.ConditionError as ex:
+ _LOGGER.warning("Error in 'if' evaluation:\n%s", ex)
+
+ if test_conditions:
+ trace_set_result(choice="then")
+ with trace_path("then"):
+ await self._async_run_script(if_data["if_then"])
+ return
+
+ if if_data["if_else"] is not None:
+ trace_set_result(choice="else")
+ with trace_path("else"):
+ await self._async_run_script(if_data["if_else"])
+
+ async def _async_do_step_repeat(self) -> None: # noqa: C901
+ """Repeat a sequence helper."""
description = self._action.get(CONF_ALIAS, "sequence")
repeat = self._action[CONF_REPEAT]
- saved_repeat_vars = self._variables.get("repeat")
-
def set_repeat_var(
iteration: int, count: int | None = None, item: Any = None
) -> None:
@@ -877,7 +779,7 @@ class _ScriptRun:
repeat_vars["last"] = iteration == count
if item is not None:
repeat_vars["item"] = item
- self._variables["repeat"] = repeat_vars
+ self._variables.define_local("repeat", repeat_vars)
script = self._script._get_repeat_script(self._step) # noqa: SLF001
warned_too_many_loops = False
@@ -1028,55 +930,137 @@ class _ScriptRun:
# while all the cpu time is consumed.
await asyncio.sleep(0)
- if saved_repeat_vars:
- self._variables["repeat"] = saved_repeat_vars
- else:
- self._variables.pop("repeat", None) # Not set if count = 0
-
- async def _async_choose_step(self) -> None:
- """Choose a sequence."""
- choose_data = await self._script._async_get_choose_data(self._step) # noqa: SLF001
-
- with trace_path("choose"):
- for idx, (conditions, script) in enumerate(choose_data["choices"]):
- with trace_path(str(idx)):
- try:
- if self._test_conditions(conditions, "choose", "conditions"):
- trace_set_result(choice=idx)
- with trace_path("sequence"):
- await self._async_run_script(script)
- return
- except exceptions.ConditionError as ex:
- _LOGGER.warning("Error in 'choose' evaluation:\n%s", ex)
-
- if choose_data["default"] is not None:
- trace_set_result(choice="default")
- with trace_path(["default"]):
- await self._async_run_script(choose_data["default"])
-
- async def _async_if_step(self) -> None:
- """If sequence."""
- if_data = await self._script._async_get_if_data(self._step) # noqa: SLF001
-
- test_conditions: bool | None = False
+ @async_trace_path("repeat")
+ async def _async_step_repeat(self) -> None:
+ """Repeat a sequence."""
+ self._variables = self._variables.enter_scope()
try:
- with trace_path("if"):
- test_conditions = self._test_conditions(
- if_data["if_conditions"], "if", "condition"
+ await self._async_do_step_repeat()
+ finally:
+ self._variables = self._variables.exit_scope()
+
+ ### Stop actions ###
+
+ async def _async_step_stop(self) -> None:
+ """Stop script execution."""
+ stop = self._action[CONF_STOP]
+ error = self._action.get(CONF_ERROR, False)
+ trace_set_result(stop=stop, error=error)
+ if error:
+ self._log("Error script sequence: %s", stop)
+ raise _AbortScript(stop)
+
+ self._log("Stop script sequence: %s", stop)
+ if CONF_RESPONSE_VARIABLE in self._action:
+ try:
+ response = self._variables[self._action[CONF_RESPONSE_VARIABLE]]
+ except KeyError as ex:
+ raise _AbortScript(
+ f"Response variable '{self._action[CONF_RESPONSE_VARIABLE]}' "
+ "is not defined"
+ ) from ex
+ else:
+ response = None
+ raise _StopScript(stop, response)
+
+ ## Variable actions ##
+
+ async def _async_step_variables(self) -> None:
+ """Assign values to variables."""
+ self._step_log("assigning variables")
+ self._variables.update(
+ self._action[CONF_VARIABLES].async_simple_render(self._variables)
+ )
+
+ ## External actions ##
+
+ async def _async_step_call_service(self) -> None:
+ """Call the service specified in the action."""
+ self._step_log("call service")
+
+ params = service.async_prepare_call_from_config(
+ self._hass, self._action, self._variables
+ )
+
+ # Validate response data parameters. This check ignores services that do
+ # not exist which will raise an appropriate error in the service call below.
+ response_variable = self._action.get(CONF_RESPONSE_VARIABLE)
+ return_response = response_variable is not None
+ if self._hass.services.has_service(params[CONF_DOMAIN], params[CONF_SERVICE]):
+ supports_response = self._hass.services.supports_response(
+ params[CONF_DOMAIN], params[CONF_SERVICE]
+ )
+ if supports_response == SupportsResponse.ONLY and not return_response:
+ raise vol.Invalid(
+ f"Script requires '{CONF_RESPONSE_VARIABLE}' for response data "
+ f"for service call {params[CONF_DOMAIN]}.{params[CONF_SERVICE]}"
+ )
+ if supports_response == SupportsResponse.NONE and return_response:
+ raise vol.Invalid(
+ f"Script does not support '{CONF_RESPONSE_VARIABLE}' for service "
+ f"'{CONF_RESPONSE_VARIABLE}' which does not support response data."
)
- except exceptions.ConditionError as ex:
- _LOGGER.warning("Error in 'if' evaluation:\n%s", ex)
- if test_conditions:
- trace_set_result(choice="then")
- with trace_path("then"):
- await self._async_run_script(if_data["if_then"])
- return
+ running_script = (
+ params[CONF_DOMAIN] == "automation" and params[CONF_SERVICE] == "trigger"
+ ) or params[CONF_DOMAIN] in ("python_script", "script")
+ trace_set_result(params=params, running_script=running_script)
+ response_data = await self._async_run_long_action(
+ self._hass.async_create_task_internal(
+ self._hass.services.async_call(
+ **params,
+ blocking=True,
+ context=self._context,
+ return_response=return_response,
+ ),
+ eager_start=True,
+ )
+ )
+ if response_variable:
+ self._variables[response_variable] = response_data
- if if_data["if_else"] is not None:
- trace_set_result(choice="else")
- with trace_path("else"):
- await self._async_run_script(if_data["if_else"])
+ async def _async_step_device(self) -> None:
+ """Perform the device automation specified in the action."""
+ self._step_log("device automation")
+ await device_action.async_call_action_from_config(
+ self._hass, self._action, dict(self._variables), self._context
+ )
+
+ async def _async_step_event(self) -> None:
+ """Fire an event."""
+ self._step_log(self._action.get(CONF_ALIAS, self._action[CONF_EVENT]))
+ event_data = {}
+ for conf in (CONF_EVENT_DATA, CONF_EVENT_DATA_TEMPLATE):
+ if conf not in self._action:
+ continue
+
+ try:
+ event_data.update(
+ template.render_complex(self._action[conf], self._variables)
+ )
+ except exceptions.TemplateError as ex:
+ self._log(
+ "Error rendering event data template: %s", ex, level=logging.ERROR
+ )
+
+ trace_set_result(event=self._action[CONF_EVENT], event_data=event_data)
+ self._hass.bus.async_fire_internal(
+ self._action[CONF_EVENT], event_data, context=self._context
+ )
+
+ async def _async_step_scene(self) -> None:
+ """Activate the scene specified in the action."""
+ self._step_log("activate scene")
+ trace_set_result(scene=self._action[CONF_SCENE])
+ await self._hass.services.async_call(
+ scene.DOMAIN,
+ SERVICE_TURN_ON,
+ {ATTR_ENTITY_ID: self._action[CONF_SCENE]},
+ blocking=True,
+ context=self._context,
+ )
+
+ ## Time-based actions ##
@overload
def _async_futures_with_timeout(
@@ -1124,18 +1108,103 @@ class _ScriptRun:
futures.append(timeout_future)
return futures, timeout_handle, timeout_future
- async def _async_wait_for_trigger_step(self) -> None:
+ def _get_pos_time_period_template(self, key: str) -> timedelta:
+ try:
+ return cv.positive_time_period( # type: ignore[no-any-return]
+ template.render_complex(self._action[key], self._variables)
+ )
+ except (exceptions.TemplateError, vol.Invalid) as ex:
+ self._log(
+ "Error rendering %s %s template: %s",
+ self._script.name,
+ key,
+ ex,
+ level=logging.ERROR,
+ )
+ raise _AbortScript from ex
+
+ async def _async_step_delay(self) -> None:
+ """Handle delay."""
+ delay_delta = self._get_pos_time_period_template(CONF_DELAY)
+
+ self._step_log(f"delay {delay_delta}")
+
+ delay = delay_delta.total_seconds()
+ self._changed()
+ if not delay:
+ # Handle an empty delay
+ trace_set_result(delay=delay, done=True)
+ return
+
+ trace_set_result(delay=delay, done=False)
+ futures, timeout_handle, timeout_future = self._async_futures_with_timeout(
+ delay
+ )
+
+ try:
+ await asyncio.wait(futures, return_when=asyncio.FIRST_COMPLETED)
+ finally:
+ if timeout_future.done():
+ trace_set_result(delay=delay, done=True)
+ else:
+ timeout_handle.cancel()
+
+ def _get_timeout_seconds_from_action(self) -> float | None:
+ """Get the timeout from the action."""
+ if CONF_TIMEOUT in self._action:
+ return self._get_pos_time_period_template(CONF_TIMEOUT).total_seconds()
+ return None
+
+ def _async_handle_timeout(self) -> None:
+ """Handle timeout."""
+ self._variables["wait"]["remaining"] = 0.0
+ if not self._action.get(CONF_CONTINUE_ON_TIMEOUT, True):
+ self._log(_TIMEOUT_MSG)
+ trace_set_result(wait=self._variables["wait"], timeout=True)
+ raise _AbortScript from TimeoutError()
+
+ async def _async_wait_with_optional_timeout(
+ self,
+ futures: list[asyncio.Future[None]],
+ timeout_handle: asyncio.TimerHandle | None,
+ timeout_future: asyncio.Future[None] | None,
+ unsub: Callable[[], None],
+ ) -> None:
+ try:
+ await asyncio.wait(futures, return_when=asyncio.FIRST_COMPLETED)
+ if timeout_future and timeout_future.done():
+ self._async_handle_timeout()
+ finally:
+ if timeout_future and not timeout_future.done() and timeout_handle:
+ timeout_handle.cancel()
+
+ unsub()
+
+ def _async_set_remaining_time_var(
+ self, timeout_handle: asyncio.TimerHandle | None
+ ) -> None:
+ """Set the remaining time variable for a wait step."""
+ wait_var = self._variables["wait"]
+ if timeout_handle:
+ wait_var["remaining"] = timeout_handle.when() - self._hass.loop.time()
+ else:
+ wait_var["remaining"] = None
+
+ async def _async_step_wait_for_trigger(self) -> None:
"""Wait for a trigger event."""
timeout = self._get_timeout_seconds_from_action()
self._step_log("wait for trigger", timeout)
- variables = {**self._variables}
- self._variables["wait"] = {
- "remaining": timeout,
- "completed": False,
- "trigger": None,
- }
+ variables = dict(self._variables)
+ self._variables.assign_parallel_protected(
+ "wait",
+ {
+ "remaining": timeout,
+ "completed": False,
+ "trigger": None,
+ },
+ )
trace_set_result(wait=self._variables["wait"])
if timeout == 0:
@@ -1176,39 +1245,55 @@ class _ScriptRun:
futures, timeout_handle, timeout_future, remove_triggers
)
- def _async_handle_timeout(self) -> None:
- """Handle timeout."""
- self._variables["wait"]["remaining"] = 0.0
- if not self._action.get(CONF_CONTINUE_ON_TIMEOUT, True):
- self._log(_TIMEOUT_MSG)
- trace_set_result(wait=self._variables["wait"], timeout=True)
- raise _AbortScript from TimeoutError()
+ async def _async_step_wait_template(self) -> None:
+ """Handle a wait template."""
+ timeout = self._get_timeout_seconds_from_action()
+ self._step_log("wait template", timeout)
- async def _async_wait_with_optional_timeout(
- self,
- futures: list[asyncio.Future[None]],
- timeout_handle: asyncio.TimerHandle | None,
- timeout_future: asyncio.Future[None] | None,
- unsub: Callable[[], None],
- ) -> None:
- try:
- await asyncio.wait(futures, return_when=asyncio.FIRST_COMPLETED)
- if timeout_future and timeout_future.done():
- self._async_handle_timeout()
- finally:
- if timeout_future and not timeout_future.done() and timeout_handle:
- timeout_handle.cancel()
+ self._variables.assign_parallel_protected(
+ "wait", {"remaining": timeout, "completed": False}
+ )
+ trace_set_result(wait=self._variables["wait"])
- unsub()
+ wait_template = self._action[CONF_WAIT_TEMPLATE]
- async def _async_variables_step(self) -> None:
- """Set a variable value."""
- self._step_log("setting variables")
- self._variables = self._action[CONF_VARIABLES].async_render(
- self._hass, self._variables, render_as_defaults=False
+ # check if condition already okay
+ if condition.async_template(self._hass, wait_template, self._variables, False):
+ self._variables["wait"]["completed"] = True
+ self._changed()
+ return
+
+ if timeout == 0:
+ self._changed()
+ self._async_handle_timeout()
+ return
+
+ futures, timeout_handle, timeout_future = self._async_futures_with_timeout(
+ timeout
+ )
+ done = self._hass.loop.create_future()
+ futures.append(done)
+
+ @callback
+ def async_script_wait(
+ entity_id: str, from_s: State | None, to_s: State | None
+ ) -> None:
+ """Handle script after template condition is true."""
+ self._async_set_remaining_time_var(timeout_handle)
+ self._variables["wait"]["completed"] = True
+ _set_result_unless_done(done)
+
+ unsub = async_track_template(
+ self._hass, wait_template, async_script_wait, self._variables
+ )
+ self._changed()
+ await self._async_wait_with_optional_timeout(
+ futures, timeout_handle, timeout_future, unsub
)
- async def _async_set_conversation_response_step(self) -> None:
+ ## Conversation actions ##
+
+ async def _async_step_set_conversation_response(self) -> None:
"""Set conversation response."""
self._step_log("setting conversation response")
resp: template.Template | None = self._action[CONF_SET_CONVERSATION_RESPONSE]
@@ -1220,63 +1305,6 @@ class _ScriptRun:
)
trace_set_result(conversation_response=self._conversation_response)
- async def _async_stop_step(self) -> None:
- """Stop script execution."""
- stop = self._action[CONF_STOP]
- error = self._action.get(CONF_ERROR, False)
- trace_set_result(stop=stop, error=error)
- if error:
- self._log("Error script sequence: %s", stop)
- raise _AbortScript(stop)
-
- self._log("Stop script sequence: %s", stop)
- if CONF_RESPONSE_VARIABLE in self._action:
- try:
- response = self._variables[self._action[CONF_RESPONSE_VARIABLE]]
- except KeyError as ex:
- raise _AbortScript(
- f"Response variable '{self._action[CONF_RESPONSE_VARIABLE]}' "
- "is not defined"
- ) from ex
- else:
- response = None
- raise _StopScript(stop, response)
-
- @async_trace_path("sequence")
- async def _async_sequence_step(self) -> None:
- """Run a sequence."""
- sequence = await self._script._async_get_sequence_script(self._step) # noqa: SLF001
- await self._async_run_script(sequence)
-
- @async_trace_path("parallel")
- async def _async_parallel_step(self) -> None:
- """Run a sequence in parallel."""
- scripts = await self._script._async_get_parallel_scripts(self._step) # noqa: SLF001
-
- async def async_run_with_trace(idx: int, script: Script) -> None:
- """Run a script with a trace path."""
- trace_path_stack_cv.set(copy(trace_path_stack_cv.get()))
- with trace_path([str(idx), "sequence"]):
- await self._async_run_script(script)
-
- results = await asyncio.gather(
- *(async_run_with_trace(idx, script) for idx, script in enumerate(scripts)),
- return_exceptions=True,
- )
- for result in results:
- if isinstance(result, Exception):
- raise result
-
- async def _async_run_script(self, script: Script) -> None:
- """Execute a script."""
- result = await self._async_run_long_action(
- self._hass.async_create_task_internal(
- script.async_run(self._variables, self._context), eager_start=True
- )
- )
- if result and result.conversation_response is not UNDEFINED:
- self._conversation_response = result.conversation_response
-
class _QueuedScriptRun(_ScriptRun):
"""Manage queued Script sequence run."""
@@ -1353,7 +1381,7 @@ async def _async_stop_scripts_at_shutdown(hass: HomeAssistant, event: Event) ->
)
-type _VarsType = dict[str, Any] | Mapping[str, Any] | MappingProxyType[str, Any]
+type _VarsType = dict[str, Any] | Mapping[str, Any] | ScriptRunVariables
def _referenced_extract_ids(data: Any, key: str, found: set[str]) -> None:
@@ -1391,7 +1419,7 @@ class ScriptRunResult:
conversation_response: str | None | UndefinedType
service_response: ServiceResponse
- variables: dict[str, Any]
+ variables: Mapping[str, Any]
class Script:
@@ -1406,7 +1434,6 @@ class Script:
*,
# Used in "Running " log message
change_listener: Callable[[], Any] | None = None,
- copy_variables: bool = False,
log_exceptions: bool = True,
logger: logging.Logger | None = None,
max_exceeded: str = DEFAULT_MAX_EXCEEDED,
@@ -1460,8 +1487,6 @@ class Script:
self._parallel_scripts: dict[int, list[Script]] = {}
self._sequence_scripts: dict[int, Script] = {}
self.variables = variables
- self._variables_dynamic = template.is_complex(variables)
- self._copy_variables_on_run = copy_variables
@property
def change_listener(self) -> Callable[..., Any] | None:
@@ -1739,25 +1764,19 @@ class Script:
if self.top_level:
if self.variables:
try:
- variables = self.variables.async_render(
+ run_variables = self.variables.async_render(
self._hass,
run_variables,
)
except exceptions.TemplateError as err:
self._log("Error rendering variables: %s", err, level=logging.ERROR)
raise
- elif run_variables:
- variables = dict(run_variables)
- else:
- variables = {}
+ variables = ScriptRunVariables.create_top_level(run_variables)
variables["context"] = context
- elif self._copy_variables_on_run:
- # This is not the top level script, variables have been turned to a dict
- variables = cast(dict[str, Any], copy(run_variables))
else:
- # This is not the top level script, variables have been turned to a dict
- variables = cast(dict[str, Any], run_variables)
+ # This is not the top level script, run_variables is an instance of ScriptRunVariables
+ variables = cast(ScriptRunVariables, run_variables)
# Prevent non-allowed recursive calls which will cause deadlocks when we try to
# stop (restart) or wait for (queued) our own script run.
@@ -1983,7 +2002,6 @@ class Script:
max_runs=self.max_runs,
logger=self._logger,
top_level=False,
- copy_variables=True,
)
parallel_script.change_listener = partial(
self._chain_change_listener, parallel_script
diff --git a/homeassistant/helpers/script_variables.py b/homeassistant/helpers/script_variables.py
index 2b4507abd64..54200e094e6 100644
--- a/homeassistant/helpers/script_variables.py
+++ b/homeassistant/helpers/script_variables.py
@@ -2,8 +2,10 @@
from __future__ import annotations
+from collections import ChainMap, UserDict
from collections.abc import Mapping
-from typing import Any
+from dataclasses import dataclass, field
+from typing import Any, cast
from homeassistant.core import HomeAssistant, callback
@@ -24,30 +26,23 @@ class ScriptVariables:
hass: HomeAssistant,
run_variables: Mapping[str, Any] | None,
*,
- render_as_defaults: bool = True,
limited: bool = False,
) -> dict[str, Any]:
"""Render script variables.
- The run variables are used to compute the static variables.
-
- If `render_as_defaults` is True, the run variables will not be overridden.
-
+ The run variables are included in the result.
+ The run variables are used to compute the rendered variable values.
+ The run variables will not be overridden.
+ The rendering happens one at a time, with previous results influencing the next.
"""
if self._has_template is None:
self._has_template = template.is_complex(self.variables)
if not self._has_template:
- if render_as_defaults:
- rendered_variables = dict(self.variables)
+ rendered_variables = dict(self.variables)
- if run_variables is not None:
- rendered_variables.update(run_variables)
- else:
- rendered_variables = (
- {} if run_variables is None else dict(run_variables)
- )
- rendered_variables.update(self.variables)
+ if run_variables is not None:
+ rendered_variables.update(run_variables)
return rendered_variables
@@ -56,7 +51,7 @@ class ScriptVariables:
for key, value in self.variables.items():
# We can skip if we're going to override this key with
# run variables anyway
- if render_as_defaults and key in rendered_variables:
+ if key in rendered_variables:
continue
rendered_variables[key] = template.render_complex(
@@ -65,6 +60,197 @@ class ScriptVariables:
return rendered_variables
+ @callback
+ def async_simple_render(self, run_variables: Mapping[str, Any]) -> dict[str, Any]:
+ """Render script variables.
+
+ Simply renders the variables, the run variables are not included in the result.
+ The run variables are used to compute the rendered variable values.
+ The rendering happens one at a time, with previous results influencing the next.
+ """
+ if self._has_template is None:
+ self._has_template = template.is_complex(self.variables)
+
+ if not self._has_template:
+ return self.variables
+
+ run_variables = dict(run_variables)
+ rendered_variables = {}
+
+ for key, value in self.variables.items():
+ rendered_variable = template.render_complex(value, run_variables)
+ rendered_variables[key] = rendered_variable
+ run_variables[key] = rendered_variable
+
+ return rendered_variables
+
def as_dict(self) -> dict[str, Any]:
"""Return dict version of this class."""
return self.variables
+
+
+@dataclass
+class _ParallelData:
+ """Data used in each parallel sequence."""
+
+ # `protected` is for variables that need special protection in parallel sequences.
+ # What this means is that such a variable defined in one parallel sequence will not be
+ # clobbered by the variable with the same name assigned in another parallel sequence.
+ # It also means that such a variable will not be visible in the outer scope.
+ # Currently the only such variable is `wait`.
+ protected: dict[str, Any] = field(default_factory=dict)
+ # `outer_scope_writes` is for variables that are written to the outer scope from
+ # a parallel sequence. This is used for generating correct traces of changed variables
+ # for each of the parallel sequences, isolating them from one another.
+ outer_scope_writes: dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass(kw_only=True)
+class ScriptRunVariables(UserDict[str, Any]):
+ """Class to hold script run variables.
+
+ The purpose of this class is to provide proper variable scoping semantics for scripts.
+ Each instance institutes a new local scope, in which variables can be defined.
+ Each instance has a reference to the previous instance, except for the top-level instance.
+ The instances therefore form a chain, in which variable lookup and assignment is performed.
+ The variables defined lower in the chain naturally override those defined higher up.
+ """
+
+ # _previous is the previous ScriptRunVariables in the chain
+ _previous: ScriptRunVariables | None = None
+ # _parent is the previous non-empty ScriptRunVariables in the chain
+ _parent: ScriptRunVariables | None = None
+
+ # _local_data is the store for local variables
+ _local_data: dict[str, Any] | None = None
+ # _parallel_data is used for each parallel sequence
+ _parallel_data: _ParallelData | None = None
+
+ # _non_parallel_scope includes all scopes all the way to the most recent parallel split
+ _non_parallel_scope: ChainMap[str, Any]
+ # _full_scope includes all scopes (all the way to the top-level)
+ _full_scope: ChainMap[str, Any]
+
+ @classmethod
+ def create_top_level(
+ cls,
+ initial_data: Mapping[str, Any] | None = None,
+ ) -> ScriptRunVariables:
+ """Create a new top-level ScriptRunVariables."""
+ local_data: dict[str, Any] = {}
+ non_parallel_scope = full_scope = ChainMap(local_data)
+ self = cls(
+ _local_data=local_data,
+ _non_parallel_scope=non_parallel_scope,
+ _full_scope=full_scope,
+ )
+ if initial_data is not None:
+ self.update(initial_data)
+ return self
+
+ def enter_scope(self, *, parallel: bool = False) -> ScriptRunVariables:
+ """Return a new child scope.
+
+ :param parallel: Whether the new scope starts a parallel sequence.
+ """
+ if self._local_data is not None or self._parallel_data is not None:
+ parent = self
+ else:
+ parent = cast( # top level always has local data, so we can cast safely
+ ScriptRunVariables, self._parent
+ )
+
+ parallel_data: _ParallelData | None
+ if not parallel:
+ parallel_data = None
+ non_parallel_scope = self._non_parallel_scope
+ full_scope = self._full_scope
+ else:
+ parallel_data = _ParallelData()
+ non_parallel_scope = ChainMap(
+ parallel_data.protected, parallel_data.outer_scope_writes
+ )
+ full_scope = self._full_scope.new_child(parallel_data.protected)
+
+ return ScriptRunVariables(
+ _previous=self,
+ _parent=parent,
+ _parallel_data=parallel_data,
+ _non_parallel_scope=non_parallel_scope,
+ _full_scope=full_scope,
+ )
+
+ def exit_scope(self) -> ScriptRunVariables:
+ """Exit the current scope.
+
+ Does no clean-up, but simply returns the previous scope.
+ """
+ if self._previous is None:
+ raise ValueError("Cannot exit top-level scope")
+ return self._previous
+
+ def __delitem__(self, key: str) -> None:
+ """Delete a variable (disallowed)."""
+ raise TypeError("Deleting items is not allowed in ScriptRunVariables.")
+
+ def __setitem__(self, key: str, value: Any) -> None:
+ """Assign value to a variable."""
+ self._assign(key, value, parallel_protected=False)
+
+ def assign_parallel_protected(self, key: str, value: Any) -> None:
+ """Assign value to a variable which is to be protected in parallel sequences."""
+ self._assign(key, value, parallel_protected=True)
+
+ def _assign(self, key: str, value: Any, *, parallel_protected: bool) -> None:
+ """Assign value to a variable.
+
+ Value is always assigned to the variable in the nearest scope, in which it is defined.
+ If the variable is not defined at all, it is created in the top-level scope.
+
+ :param parallel_protected: Whether variable is to be protected in parallel sequences.
+ """
+ if self._local_data is not None and key in self._local_data:
+ self._local_data[key] = value
+ return
+
+ if self._parent is None:
+ assert self._local_data is not None # top level always has local data
+ self._local_data[key] = value
+ return
+
+ if self._parallel_data is not None:
+ if parallel_protected:
+ self._parallel_data.protected[key] = value
+ return
+ self._parallel_data.protected.pop(key, None)
+ self._parallel_data.outer_scope_writes[key] = value
+
+ self._parent._assign(key, value, parallel_protected=parallel_protected) # noqa: SLF001
+
+ def define_local(self, key: str, value: Any) -> None:
+ """Define a local variable and assign value to it."""
+ if self._local_data is None:
+ self._local_data = {}
+ self._non_parallel_scope = self._non_parallel_scope.new_child(
+ self._local_data
+ )
+ self._full_scope = self._full_scope.new_child(self._local_data)
+ self._local_data[key] = value
+
+ @property
+ def data(self) -> Mapping[str, Any]: # type: ignore[override]
+ """Return variables in full scope.
+
+ Defined here for UserDict compatibility.
+ """
+ return self._full_scope
+
+ @property
+ def non_parallel_scope(self) -> Mapping[str, Any]:
+ """Return variables in non-parallel scope."""
+ return self._non_parallel_scope
+
+ @property
+ def local_scope(self) -> Mapping[str, Any]:
+ """Return variables in local scope."""
+ return self._local_data if self._local_data is not None else {}
diff --git a/homeassistant/helpers/selector.py b/homeassistant/helpers/selector.py
index 025b8de8896..f2c76d1d019 100644
--- a/homeassistant/helpers/selector.py
+++ b/homeassistant/helpers/selector.py
@@ -164,6 +164,8 @@ DEVICE_FILTER_SELECTOR_CONFIG_SCHEMA = vol.Schema(
vol.Optional("manufacturer"): str,
# Model of device
vol.Optional("model"): str,
+ # Model ID of device
+ vol.Optional("model_id"): str,
# Device has to contain entities matching this selector
vol.Optional("entity"): vol.All(
cv.ensure_list, [ENTITY_FILTER_SELECTOR_CONFIG_SCHEMA]
@@ -178,6 +180,7 @@ class DeviceFilterSelectorConfig(TypedDict, total=False):
integration: str
manufacturer: str
model: str
+ model_id: str
class ActionSelectorConfig(TypedDict):
@@ -1133,7 +1136,7 @@ class SelectOptionDict(TypedDict):
class SelectSelectorMode(StrEnum):
- """Possible modes for a number selector."""
+ """Possible modes for a select selector."""
LIST = "list"
DROPDOWN = "dropdown"
diff --git a/homeassistant/helpers/template.py b/homeassistant/helpers/template.py
index 7866250d658..0d017dda64f 100644
--- a/homeassistant/helpers/template.py
+++ b/homeassistant/helpers/template.py
@@ -6,12 +6,13 @@ from ast import literal_eval
import asyncio
import base64
import collections.abc
-from collections.abc import Callable, Generator, Iterable
+from collections.abc import Callable, Generator, Iterable, MutableSequence
from contextlib import AbstractContextManager
from contextvars import ContextVar
from copy import deepcopy
from datetime import date, datetime, time, timedelta
from functools import cache, lru_cache, partial, wraps
+import hashlib
import json
import logging
import math
@@ -1525,6 +1526,15 @@ def floor_areas(hass: HomeAssistant, floor_id_or_name: str) -> Iterable[str]:
return [entry.id for entry in entries if entry.id]
+def floor_entities(hass: HomeAssistant, floor_id_or_name: str) -> Iterable[str]:
+ """Return entity_ids for a given floor ID or name."""
+ return [
+ entity_id
+ for area_id in floor_areas(hass, floor_id_or_name)
+ for entity_id in area_entities(hass, area_id)
+ ]
+
+
def areas(hass: HomeAssistant) -> Iterable[str | None]:
"""Return all areas."""
return list(area_registry.async_get(hass).areas)
@@ -2727,6 +2737,100 @@ def iif(
return if_false
+def shuffle(*args: Any, seed: Any = None) -> MutableSequence[Any]:
+ """Shuffle a list, either with a seed or without."""
+ if not args:
+ raise TypeError("shuffle expected at least 1 argument, got 0")
+
+ # If first argument is iterable and more than 1 argument provided
+ # but not a named seed, then use 2nd argument as seed.
+ if isinstance(args[0], Iterable):
+ items = list(args[0])
+ if len(args) > 1 and seed is None:
+ seed = args[1]
+ elif len(args) == 1:
+ raise TypeError(f"'{type(args[0]).__name__}' object is not iterable")
+ else:
+ items = list(args)
+
+ if seed:
+ r = random.Random(seed)
+ r.shuffle(items)
+ else:
+ random.shuffle(items)
+ return items
+
+
+def typeof(value: Any) -> Any:
+ """Return the type of value passed to debug types."""
+ return value.__class__.__name__
+
+
+def flatten(value: Iterable[Any], levels: int | None = None) -> list[Any]:
+ """Flattens list of lists."""
+ if not isinstance(value, Iterable) or isinstance(value, str):
+ raise TypeError(f"flatten expected a list, got {type(value).__name__}")
+
+ flattened: list[Any] = []
+ for item in value:
+ if isinstance(item, Iterable) and not isinstance(item, str):
+ if levels is None:
+ flattened.extend(flatten(item))
+ elif levels >= 1:
+ flattened.extend(flatten(item, levels=(levels - 1)))
+ else:
+ flattened.append(item)
+ else:
+ flattened.append(item)
+ return flattened
+
+
+def combine(*args: Any, recursive: bool = False) -> dict[Any, Any]:
+ """Combine multiple dictionaries into one."""
+ if not args:
+ raise TypeError("combine expected at least 1 argument, got 0")
+
+ result: dict[Any, Any] = {}
+ for arg in args:
+ if not isinstance(arg, dict):
+ raise TypeError(f"combine expected a dict, got {type(arg).__name__}")
+
+ if recursive:
+ for key, value in arg.items():
+ if (
+ key in result
+ and isinstance(result[key], dict)
+ and isinstance(value, dict)
+ ):
+ result[key] = combine(result[key], value, recursive=True)
+ else:
+ result[key] = value
+ else:
+ result |= arg
+
+ return result
+
+
+def md5(value: str) -> str:
+ """Generate md5 hash from a string."""
+ return hashlib.md5(value.encode()).hexdigest()
+
+
+def sha1(value: str) -> str:
+ """Generate sha1 hash from a string."""
+ return hashlib.sha1(value.encode()).hexdigest()
+
+
+def sha256(value: str) -> str:
+ """Generate sha256 hash from a string."""
+ return hashlib.sha256(value.encode()).hexdigest()
+
+
+def sha512(value: str) -> str:
+ """Generate sha512 hash from a string."""
+ return hashlib.sha512(value.encode()).hexdigest()
+
+
class TemplateContextManager(AbstractContextManager):
"""Context manager to store template being parsed or rendered in a ContextVar."""
@@ -2879,100 +2983,119 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
str | jinja2.nodes.Template, CodeType | None
] = weakref.WeakValueDictionary()
self.add_extension("jinja2.ext.loopcontrols")
- self.filters["round"] = forgiving_round
- self.filters["multiply"] = multiply
- self.filters["add"] = add
- self.filters["log"] = logarithm
- self.filters["sin"] = sine
- self.filters["cos"] = cosine
- self.filters["tan"] = tangent
- self.filters["asin"] = arc_sine
- self.filters["acos"] = arc_cosine
- self.filters["atan"] = arc_tangent
- self.filters["atan2"] = arc_tangent2
- self.filters["sqrt"] = square_root
- self.filters["as_datetime"] = as_datetime
- self.filters["as_timedelta"] = as_timedelta
- self.filters["as_timestamp"] = forgiving_as_timestamp
- self.filters["as_local"] = dt_util.as_local
- self.filters["timestamp_custom"] = timestamp_custom
- self.filters["timestamp_local"] = timestamp_local
- self.filters["timestamp_utc"] = timestamp_utc
- self.filters["to_json"] = to_json
- self.filters["from_json"] = from_json
- self.filters["is_defined"] = fail_when_undefined
- self.filters["average"] = average
- self.filters["median"] = median
- self.filters["statistical_mode"] = statistical_mode
- self.filters["random"] = random_every_time
- self.filters["base64_encode"] = base64_encode
- self.filters["base64_decode"] = base64_decode
- self.filters["ordinal"] = ordinal
- self.filters["regex_match"] = regex_match
- self.filters["regex_replace"] = regex_replace
- self.filters["regex_search"] = regex_search
- self.filters["regex_findall"] = regex_findall
- self.filters["regex_findall_index"] = regex_findall_index
- self.filters["bitwise_and"] = bitwise_and
- self.filters["bitwise_or"] = bitwise_or
- self.filters["bitwise_xor"] = bitwise_xor
- self.filters["pack"] = struct_pack
- self.filters["unpack"] = struct_unpack
- self.filters["ord"] = ord
- self.filters["is_number"] = is_number
- self.filters["float"] = forgiving_float_filter
- self.filters["int"] = forgiving_int_filter
- self.filters["slugify"] = slugify
- self.filters["iif"] = iif
- self.filters["bool"] = forgiving_boolean
- self.filters["version"] = version
- self.filters["contains"] = contains
- self.globals["log"] = logarithm
- self.globals["sin"] = sine
- self.globals["cos"] = cosine
- self.globals["tan"] = tangent
- self.globals["sqrt"] = square_root
- self.globals["pi"] = math.pi
- self.globals["tau"] = math.pi * 2
- self.globals["e"] = math.e
- self.globals["asin"] = arc_sine
+
self.globals["acos"] = arc_cosine
- self.globals["atan"] = arc_tangent
- self.globals["atan2"] = arc_tangent2
- self.globals["float"] = forgiving_float
self.globals["as_datetime"] = as_datetime
self.globals["as_local"] = dt_util.as_local
self.globals["as_timedelta"] = as_timedelta
self.globals["as_timestamp"] = forgiving_as_timestamp
- self.globals["timedelta"] = timedelta
- self.globals["merge_response"] = merge_response
- self.globals["strptime"] = strptime
- self.globals["urlencode"] = urlencode
+ self.globals["asin"] = arc_sine
+ self.globals["atan"] = arc_tangent
+ self.globals["atan2"] = arc_tangent2
self.globals["average"] = average
- self.globals["median"] = median
- self.globals["statistical_mode"] = statistical_mode
- self.globals["max"] = min_max_from_filter(self.filters["max"], "max")
- self.globals["min"] = min_max_from_filter(self.filters["min"], "min")
- self.globals["is_number"] = is_number
- self.globals["set"] = _to_set
- self.globals["tuple"] = _to_tuple
- self.globals["int"] = forgiving_int
- self.globals["pack"] = struct_pack
- self.globals["unpack"] = struct_unpack
- self.globals["slugify"] = slugify
- self.globals["iif"] = iif
self.globals["bool"] = forgiving_boolean
+ self.globals["combine"] = combine
+ self.globals["cos"] = cosine
+ self.globals["e"] = math.e
+ self.globals["flatten"] = flatten
+ self.globals["float"] = forgiving_float
+ self.globals["iif"] = iif
+ self.globals["int"] = forgiving_int
+ self.globals["is_number"] = is_number
+ self.globals["log"] = logarithm
+ self.globals["max"] = min_max_from_filter(self.filters["max"], "max")
+ self.globals["md5"] = md5
+ self.globals["median"] = median
+ self.globals["merge_response"] = merge_response
+ self.globals["min"] = min_max_from_filter(self.filters["min"], "min")
+ self.globals["pack"] = struct_pack
+ self.globals["pi"] = math.pi
+ self.globals["set"] = _to_set
+ self.globals["sha1"] = sha1
+ self.globals["sha256"] = sha256
+ self.globals["sha512"] = sha512
+ self.globals["shuffle"] = shuffle
+ self.globals["sin"] = sine
+ self.globals["slugify"] = slugify
+ self.globals["sqrt"] = square_root
+ self.globals["statistical_mode"] = statistical_mode
+ self.globals["strptime"] = strptime
+ self.globals["tan"] = tangent
+ self.globals["tau"] = math.pi * 2
+ self.globals["timedelta"] = timedelta
+ self.globals["tuple"] = _to_tuple
+ self.globals["typeof"] = typeof
+ self.globals["unpack"] = struct_unpack
+ self.globals["urlencode"] = urlencode
self.globals["version"] = version
self.globals["zip"] = zip
+
+ self.filters["acos"] = arc_cosine
+ self.filters["add"] = add
+ self.filters["as_datetime"] = as_datetime
+ self.filters["as_local"] = dt_util.as_local
+ self.filters["as_timedelta"] = as_timedelta
+ self.filters["as_timestamp"] = forgiving_as_timestamp
+ self.filters["asin"] = arc_sine
+ self.filters["atan"] = arc_tangent
+ self.filters["atan2"] = arc_tangent2
+ self.filters["average"] = average
+ self.filters["base64_decode"] = base64_decode
+ self.filters["base64_encode"] = base64_encode
+ self.filters["bitwise_and"] = bitwise_and
+ self.filters["bitwise_or"] = bitwise_or
+ self.filters["bitwise_xor"] = bitwise_xor
+ self.filters["bool"] = forgiving_boolean
+ self.filters["combine"] = combine
+ self.filters["contains"] = contains
+ self.filters["cos"] = cosine
+ self.filters["flatten"] = flatten
+ self.filters["float"] = forgiving_float_filter
+ self.filters["from_json"] = from_json
+ self.filters["iif"] = iif
+ self.filters["int"] = forgiving_int_filter
+ self.filters["is_defined"] = fail_when_undefined
+ self.filters["is_number"] = is_number
+ self.filters["log"] = logarithm
+ self.filters["md5"] = md5
+ self.filters["median"] = median
+ self.filters["multiply"] = multiply
+ self.filters["ord"] = ord
+ self.filters["ordinal"] = ordinal
+ self.filters["pack"] = struct_pack
+ self.filters["random"] = random_every_time
+ self.filters["regex_findall_index"] = regex_findall_index
+ self.filters["regex_findall"] = regex_findall
+ self.filters["regex_match"] = regex_match
+ self.filters["regex_replace"] = regex_replace
+ self.filters["regex_search"] = regex_search
+ self.filters["round"] = forgiving_round
+ self.filters["sha1"] = sha1
+ self.filters["sha256"] = sha256
+ self.filters["sha512"] = sha512
+ self.filters["shuffle"] = shuffle
+ self.filters["sin"] = sine
+ self.filters["slugify"] = slugify
+ self.filters["sqrt"] = square_root
+ self.filters["statistical_mode"] = statistical_mode
+ self.filters["tan"] = tangent
+ self.filters["timestamp_custom"] = timestamp_custom
+ self.filters["timestamp_local"] = timestamp_local
+ self.filters["timestamp_utc"] = timestamp_utc
+ self.filters["to_json"] = to_json
+ self.filters["typeof"] = typeof
+ self.filters["unpack"] = struct_unpack
+ self.filters["version"] = version
+
+ self.tests["contains"] = contains
+ self.tests["datetime"] = _is_datetime
self.tests["is_number"] = is_number
self.tests["list"] = _is_list
- self.tests["set"] = _is_set
- self.tests["tuple"] = _is_tuple
- self.tests["datetime"] = _is_datetime
- self.tests["string_like"] = _is_string_like
self.tests["match"] = regex_match
self.tests["search"] = regex_search
- self.tests["contains"] = contains
+ self.tests["set"] = _is_set
+ self.tests["string_like"] = _is_string_like
+ self.tests["tuple"] = _is_tuple
if hass is None:
return
@@ -2999,28 +3122,7 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
return jinja_context(wrapper)
- self.globals["device_entities"] = hassfunction(device_entities)
- self.filters["device_entities"] = self.globals["device_entities"]
-
- self.globals["device_attr"] = hassfunction(device_attr)
- self.filters["device_attr"] = self.globals["device_attr"]
-
- self.globals["config_entry_attr"] = hassfunction(config_entry_attr)
- self.filters["config_entry_attr"] = self.globals["config_entry_attr"]
-
- self.globals["is_device_attr"] = hassfunction(is_device_attr)
- self.tests["is_device_attr"] = hassfunction(is_device_attr, pass_eval_context)
-
- self.globals["config_entry_id"] = hassfunction(config_entry_id)
- self.filters["config_entry_id"] = self.globals["config_entry_id"]
-
- self.globals["device_id"] = hassfunction(device_id)
- self.filters["device_id"] = self.globals["device_id"]
-
- self.globals["issues"] = hassfunction(issues)
-
- self.globals["issue"] = hassfunction(issue)
- self.filters["issue"] = self.globals["issue"]
+ # Area extensions
self.globals["areas"] = hassfunction(areas)
@@ -3036,6 +3138,8 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
self.globals["area_devices"] = hassfunction(area_devices)
self.filters["area_devices"] = self.globals["area_devices"]
+ # Floor extensions
+
self.globals["floors"] = hassfunction(floors)
self.filters["floors"] = self.globals["floors"]
@@ -3048,9 +3152,38 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
self.globals["floor_areas"] = hassfunction(floor_areas)
self.filters["floor_areas"] = self.globals["floor_areas"]
+ self.globals["floor_entities"] = hassfunction(floor_entities)
+ self.filters["floor_entities"] = self.globals["floor_entities"]
+
+ # Integration extensions
+
self.globals["integration_entities"] = hassfunction(integration_entities)
self.filters["integration_entities"] = self.globals["integration_entities"]
+ # Config entry extensions
+
+ self.globals["config_entry_attr"] = hassfunction(config_entry_attr)
+ self.filters["config_entry_attr"] = self.globals["config_entry_attr"]
+
+ self.globals["config_entry_id"] = hassfunction(config_entry_id)
+ self.filters["config_entry_id"] = self.globals["config_entry_id"]
+
+ # Device extensions
+
+ self.globals["device_attr"] = hassfunction(device_attr)
+ self.filters["device_attr"] = self.globals["device_attr"]
+
+ self.globals["device_entities"] = hassfunction(device_entities)
+ self.filters["device_entities"] = self.globals["device_entities"]
+
+ self.globals["is_device_attr"] = hassfunction(is_device_attr)
+ self.tests["is_device_attr"] = hassfunction(is_device_attr, pass_eval_context)
+
+ self.globals["device_id"] = hassfunction(device_id)
+ self.filters["device_id"] = self.globals["device_id"]
+
+ # Label extensions
+
self.globals["labels"] = hassfunction(labels)
self.filters["labels"] = self.globals["labels"]
@@ -3069,6 +3202,12 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
self.globals["label_entities"] = hassfunction(label_entities)
self.filters["label_entities"] = self.globals["label_entities"]
+ # Issue extensions
+
+ self.globals["issues"] = hassfunction(issues)
+ self.globals["issue"] = hassfunction(issue)
+ self.filters["issue"] = self.globals["issue"]
+
if limited:
# Only device_entities is available to limited templates, mark other
# functions and filters as unsupported.
@@ -3081,38 +3220,38 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
return warn_unsupported
hass_globals = [
- "closest",
- "distance",
- "expand",
- "is_hidden_entity",
- "is_state",
- "is_state_attr",
- "state_attr",
- "states",
- "state_translated",
- "has_value",
- "utcnow",
- "now",
- "device_attr",
- "is_device_attr",
- "device_id",
"area_id",
"area_name",
+ "closest",
+ "device_attr",
+ "device_id",
+ "distance",
+ "expand",
"floor_id",
"floor_name",
+ "has_value",
+ "is_device_attr",
+ "is_hidden_entity",
+ "is_state_attr",
+ "is_state",
+ "label_id",
+ "label_name",
+ "now",
"relative_time",
+ "state_attr",
+ "state_translated",
+ "states",
"time_since",
"time_until",
"today_at",
- "label_id",
- "label_name",
+ "utcnow",
]
hass_filters = [
- "closest",
- "expand",
- "device_id",
"area_id",
"area_name",
+ "closest",
+ "device_id",
+ "expand",
"floor_id",
"floor_name",
"has_value",
@@ -3122,8 +3261,8 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
hass_tests = [
"has_value",
"is_hidden_entity",
- "is_state",
"is_state_attr",
+ "is_state",
]
for glob in hass_globals:
self.globals[glob] = unsupported(glob)
@@ -3133,38 +3272,46 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
self.filters[test] = unsupported(test)
return
- self.globals["expand"] = hassfunction(expand)
- self.filters["expand"] = self.globals["expand"]
self.globals["closest"] = hassfunction(closest)
- self.filters["closest"] = hassfunction(closest_filter)
self.globals["distance"] = hassfunction(distance)
+ self.globals["expand"] = hassfunction(expand)
+ self.globals["has_value"] = hassfunction(has_value)
+ self.globals["now"] = hassfunction(now)
+ self.globals["relative_time"] = hassfunction(relative_time)
+ self.globals["time_since"] = hassfunction(time_since)
+ self.globals["time_until"] = hassfunction(time_until)
+ self.globals["today_at"] = hassfunction(today_at)
+ self.globals["utcnow"] = hassfunction(utcnow)
+
+ self.filters["closest"] = hassfunction(closest_filter)
+ self.filters["expand"] = self.globals["expand"]
+ self.filters["has_value"] = self.globals["has_value"]
+ self.filters["relative_time"] = self.globals["relative_time"]
+ self.filters["time_since"] = self.globals["time_since"]
+ self.filters["time_until"] = self.globals["time_until"]
+ self.filters["today_at"] = self.globals["today_at"]
+
+ self.tests["has_value"] = hassfunction(has_value, pass_eval_context)
+
+ # Entity extensions
+
self.globals["is_hidden_entity"] = hassfunction(is_hidden_entity)
self.tests["is_hidden_entity"] = hassfunction(
is_hidden_entity, pass_eval_context
)
- self.globals["is_state"] = hassfunction(is_state)
- self.tests["is_state"] = hassfunction(is_state, pass_eval_context)
+
+ # State extensions
+
self.globals["is_state_attr"] = hassfunction(is_state_attr)
- self.tests["is_state_attr"] = hassfunction(is_state_attr, pass_eval_context)
+ self.globals["is_state"] = hassfunction(is_state)
self.globals["state_attr"] = hassfunction(state_attr)
- self.filters["state_attr"] = self.globals["state_attr"]
- self.globals["states"] = AllStates(hass)
- self.filters["states"] = self.globals["states"]
self.globals["state_translated"] = StateTranslated(hass)
+ self.globals["states"] = AllStates(hass)
+ self.filters["state_attr"] = self.globals["state_attr"]
self.filters["state_translated"] = self.globals["state_translated"]
- self.globals["has_value"] = hassfunction(has_value)
- self.filters["has_value"] = self.globals["has_value"]
- self.tests["has_value"] = hassfunction(has_value, pass_eval_context)
- self.globals["utcnow"] = hassfunction(utcnow)
- self.globals["now"] = hassfunction(now)
- self.globals["relative_time"] = hassfunction(relative_time)
- self.filters["relative_time"] = self.globals["relative_time"]
- self.globals["time_since"] = hassfunction(time_since)
- self.filters["time_since"] = self.globals["time_since"]
- self.globals["time_until"] = hassfunction(time_until)
- self.filters["time_until"] = self.globals["time_until"]
- self.globals["today_at"] = hassfunction(today_at)
- self.filters["today_at"] = self.globals["today_at"]
+ self.filters["states"] = self.globals["states"]
+ self.tests["is_state_attr"] = hassfunction(is_state_attr, pass_eval_context)
+ self.tests["is_state"] = hassfunction(is_state, pass_eval_context)
def is_safe_callable(self, obj):
"""Test if callback is safe."""
diff --git a/homeassistant/helpers/trigger.py b/homeassistant/helpers/trigger.py
index 67e9010df79..a27c85a5c58 100644
--- a/homeassistant/helpers/trigger.py
+++ b/homeassistant/helpers/trigger.py
@@ -265,18 +265,18 @@ def _trigger_action_wrapper(
while isinstance(check_func, functools.partial):
check_func = check_func.func
- wrapper_func: Callable[..., None] | Callable[..., Coroutine[Any, Any, None]]
+ wrapper_func: Callable[..., Any] | Callable[..., Coroutine[Any, Any, Any]]
if asyncio.iscoroutinefunction(check_func):
- async_action = cast(Callable[..., Coroutine[Any, Any, None]], action)
+ async_action = cast(Callable[..., Coroutine[Any, Any, Any]], action)
@functools.wraps(async_action)
async def async_with_vars(
run_variables: dict[str, Any], context: Context | None = None
- ) -> None:
+ ) -> Any:
"""Wrap action with extra vars."""
trigger_variables = conf[CONF_VARIABLES]
run_variables.update(trigger_variables.async_render(hass, run_variables))
- await action(run_variables, context)
+ return await action(run_variables, context)
wrapper_func = async_with_vars
@@ -285,11 +285,11 @@ def _trigger_action_wrapper(
@functools.wraps(action)
async def with_vars(
run_variables: dict[str, Any], context: Context | None = None
- ) -> None:
+ ) -> Any:
"""Wrap action with extra vars."""
trigger_variables = conf[CONF_VARIABLES]
run_variables.update(trigger_variables.async_render(hass, run_variables))
- action(run_variables, context)
+ return action(run_variables, context)
if is_callback(check_func):
with_vars = callback(with_vars)
diff --git a/homeassistant/helpers/update_coordinator.py b/homeassistant/helpers/update_coordinator.py
index be765ff422d..7130264eb0d 100644
--- a/homeassistant/helpers/update_coordinator.py
+++ b/homeassistant/helpers/update_coordinator.py
@@ -348,8 +348,8 @@ class DataUpdateCoordinator(BaseDataUpdateCoordinatorProtocol, Generic[_DataT]):
only once during the first refresh.
"""
if self.setup_method is None:
- return None
- return await self.setup_method()
+ return
+ await self.setup_method()
async def async_refresh(self) -> None:
"""Refresh data and log errors."""
diff --git a/homeassistant/loader.py b/homeassistant/loader.py
index 92b588dbe15..20763dc7b30 100644
--- a/homeassistant/loader.py
+++ b/homeassistant/loader.py
@@ -40,7 +40,8 @@ from .generated.ssdp import SSDP
from .generated.usb import USB
from .generated.zeroconf import HOMEKIT, ZEROCONF
from .helpers.json import json_bytes, json_fragment
-from .helpers.typing import UNDEFINED
+from .helpers.typing import UNDEFINED, UndefinedType
+from .util.async_ import create_eager_task
from .util.hass_dict import HassKey
from .util.json import JSON_DECODE_EXCEPTIONS, json_loads
@@ -125,9 +126,9 @@ BLOCKED_CUSTOM_INTEGRATIONS: dict[str, BlockedIntegration] = {
DATA_COMPONENTS: HassKey[dict[str, ModuleType | ComponentProtocol]] = HassKey(
"components"
)
-DATA_INTEGRATIONS: HassKey[dict[str, Integration | asyncio.Future[None]]] = HassKey(
- "integrations"
-)
+DATA_INTEGRATIONS: HassKey[
+ dict[str, Integration | asyncio.Future[Integration | IntegrationNotFound]]
+] = HassKey("integrations")
DATA_MISSING_PLATFORMS: HassKey[dict[str, bool]] = HassKey("missing_platforms")
DATA_CUSTOM_COMPONENTS: HassKey[
dict[str, Integration] | asyncio.Future[dict[str, Integration]]
@@ -759,10 +760,8 @@ class Integration:
manifest["overwrites_built_in"] = self.overwrites_built_in
if self.dependencies:
- self._all_dependencies_resolved: bool | None = None
- self._all_dependencies: set[str] | None = None
+ self._all_dependencies: set[str] | Exception | None = None
else:
- self._all_dependencies_resolved = True
self._all_dependencies = set()
self._platforms_to_preload = hass.data[DATA_PRELOAD_PLATFORMS]
@@ -934,47 +933,25 @@ class Integration:
"""Return all dependencies including sub-dependencies."""
if self._all_dependencies is None:
raise RuntimeError("Dependencies not resolved!")
+ if isinstance(self._all_dependencies, Exception):
+ raise self._all_dependencies
return self._all_dependencies
@property
def all_dependencies_resolved(self) -> bool:
"""Return if all dependencies have been resolved."""
- return self._all_dependencies_resolved is not None
+ return self._all_dependencies is not None
- async def resolve_dependencies(self) -> bool:
+ async def resolve_dependencies(self) -> set[str] | None:
"""Resolve all dependencies."""
- if self._all_dependencies_resolved is not None:
- return self._all_dependencies_resolved
+ if self._all_dependencies is not None:
+ if isinstance(self._all_dependencies, Exception):
+ return None
+ return self._all_dependencies
- self._all_dependencies_resolved = False
- try:
- dependencies = await _async_component_dependencies(self.hass, self)
- except IntegrationNotFound as err:
- _LOGGER.error(
- (
- "Unable to resolve dependencies for %s: unable to resolve"
- " (sub)dependency %s"
- ),
- self.domain,
- err.domain,
- )
- except CircularDependency as err:
- _LOGGER.error(
- (
- "Unable to resolve dependencies for %s: it contains a circular"
- " dependency: %s -> %s"
- ),
- self.domain,
- err.from_domain,
- err.to_domain,
- )
- else:
- dependencies.discard(self.domain)
- self._all_dependencies = dependencies
- self._all_dependencies_resolved = True
-
- return self._all_dependencies_resolved
+ result = await resolve_integrations_dependencies(self.hass, (self,))
+ return result.get(self.domain)
async def async_get_component(self) -> ComponentProtocol:
"""Return the component.
@@ -1345,7 +1322,7 @@ def async_get_loaded_integration(hass: HomeAssistant, domain: str) -> Integratio
Raises IntegrationNotLoaded if the integration is not loaded.
"""
cache = hass.data[DATA_INTEGRATIONS]
- int_or_fut = cache.get(domain, UNDEFINED)
+ int_or_fut = cache.get(domain)
# Integration is never subclassed, so we can check for type
if type(int_or_fut) is Integration:
return int_or_fut
@@ -1355,7 +1332,7 @@ def async_get_loaded_integration(hass: HomeAssistant, domain: str) -> Integratio
async def async_get_integration(hass: HomeAssistant, domain: str) -> Integration:
"""Get integration."""
cache = hass.data[DATA_INTEGRATIONS]
- if type(int_or_fut := cache.get(domain, UNDEFINED)) is Integration:
+ if type(int_or_fut := cache.get(domain)) is Integration:
return int_or_fut
integrations_or_excs = await async_get_integrations(hass, [domain])
int_or_exc = integrations_or_excs[domain]
@@ -1370,15 +1347,17 @@ async def async_get_integrations(
"""Get integrations."""
cache = hass.data[DATA_INTEGRATIONS]
results: dict[str, Integration | Exception] = {}
- needed: dict[str, asyncio.Future[None]] = {}
- in_progress: dict[str, asyncio.Future[None]] = {}
+ needed: dict[str, asyncio.Future[Integration | IntegrationNotFound]] = {}
+ in_progress: dict[str, asyncio.Future[Integration | IntegrationNotFound]] = {}
for domain in domains:
- int_or_fut = cache.get(domain, UNDEFINED)
+ int_or_fut = cache.get(domain)
# Integration is never subclassed, so we can check for type
if type(int_or_fut) is Integration:
results[domain] = int_or_fut
- elif int_or_fut is not UNDEFINED:
- in_progress[domain] = cast(asyncio.Future[None], int_or_fut)
+ elif int_or_fut:
+ if TYPE_CHECKING:
+ assert isinstance(int_or_fut, asyncio.Future)
+ in_progress[domain] = int_or_fut
elif "." in domain:
results[domain] = ValueError(f"Invalid domain {domain}")
else:
@@ -1386,14 +1365,13 @@ async def async_get_integrations(
if in_progress:
await asyncio.wait(in_progress.values())
- for domain in in_progress:
- # When we have waited and it's UNDEFINED, it doesn't exist
- # We don't cache that it doesn't exist, or else people can't fix it
- # and then restart, because their config will never be valid.
- if (int_or_fut := cache.get(domain, UNDEFINED)) is UNDEFINED:
- results[domain] = IntegrationNotFound(domain)
- else:
- results[domain] = cast(Integration, int_or_fut)
+ # Here we retrieve the results we waited for
+ # instead of reading them from the cache since
+ # reading from the cache will have a race if
+ # the integration gets removed from the cache
+ # because it was not found.
+ for domain, future in in_progress.items():
+ results[domain] = future.result()
if not needed:
return results
@@ -1405,7 +1383,7 @@ async def async_get_integrations(
for domain, future in needed.items():
if integration := custom.get(domain):
results[domain] = cache[domain] = integration
- future.set_result(None)
+ future.set_result(integration)
for domain in results:
if domain in needed:
@@ -1419,22 +1397,211 @@ async def async_get_integrations(
_resolve_integrations_from_root, hass, components, needed
)
for domain, future in needed.items():
- int_or_exc = integrations.get(domain)
- if not int_or_exc:
- cache.pop(domain)
- results[domain] = IntegrationNotFound(domain)
- elif isinstance(int_or_exc, Exception):
- cache.pop(domain)
- exc = IntegrationNotFound(domain)
- exc.__cause__ = int_or_exc
- results[domain] = exc
+ if integration := integrations.get(domain):
+ results[domain] = cache[domain] = integration
+ future.set_result(integration)
else:
- results[domain] = cache[domain] = int_or_exc
- future.set_result(None)
+ # We don't cache that it doesn't exist as configuration
+ # validation that relies on integrations being loaded
+ # would be unfixable. For example if a custom integration
+ # was temporarily removed.
+ # This allows restoring a missing integration to fix the
+ # validation error so the config validations checks do not
+ # block restarting.
+ del cache[domain]
+ exc = IntegrationNotFound(domain)
+ results[domain] = exc
+ # We don't use set_exception because
+ # we expect there will be cases where
+ # the future exception is never retrieved
+ future.set_result(exc)
return results
+class _ResolveDependenciesCacheProtocol(Protocol):
+ def get(self, itg: Integration) -> set[str] | Exception | None: ...
+
+ def __setitem__(
+ self, itg: Integration, all_dependencies: set[str] | Exception
+ ) -> None: ...
+
+
+class _ResolveDependenciesCache(_ResolveDependenciesCacheProtocol):
+ """Cache for resolve_integrations_dependencies."""
+
+ def get(self, itg: Integration) -> set[str] | Exception | None:
+ return itg._all_dependencies # noqa: SLF001
+
+ def __setitem__(
+ self, itg: Integration, all_dependencies: set[str] | Exception
+ ) -> None:
+ itg._all_dependencies = all_dependencies # noqa: SLF001
+
+
+async def resolve_integrations_dependencies(
+ hass: HomeAssistant, integrations: Iterable[Integration]
+) -> dict[str, set[str]]:
+ """Resolve all dependencies for integrations.
+
+ Detects circular dependencies and missing integrations.
+ """
+ resolved = _ResolveDependenciesCache()
+
+ async def _resolve_deps_catch_exceptions(itg: Integration) -> set[str] | None:
+ try:
+ return await _do_resolve_dependencies(itg, cache=resolved)
+ except Exception as exc: # noqa: BLE001
+ _LOGGER.error("Unable to resolve dependencies for %s: %s", itg.domain, exc)
+ return None
+
+ resolve_dependencies_tasks = {
+ itg.domain: create_eager_task(
+ _resolve_deps_catch_exceptions(itg),
+ name=f"resolve dependencies {itg.domain}",
+ loop=hass.loop,
+ )
+ for itg in integrations
+ }
+
+ result = await asyncio.gather(*resolve_dependencies_tasks.values())
+
+ return {
+ domain: deps
+ for domain, deps in zip(resolve_dependencies_tasks, result, strict=True)
+ if deps is not None
+ }
+
+
+async def resolve_integrations_after_dependencies(
+ hass: HomeAssistant,
+ integrations: Iterable[Integration],
+ possible_after_dependencies: set[str] | None = None,
+ *,
+ ignore_exceptions: bool = False,
+) -> dict[str, set[str]]:
+ """Resolve all dependencies, including after_dependencies, for integrations.
+
+ Detects circular dependencies and missing integrations.
+ """
+ resolved: dict[Integration, set[str] | Exception] = {}
+
+ async def _resolve_deps_catch_exceptions(itg: Integration) -> set[str] | None:
+ try:
+ return await _do_resolve_dependencies(
+ itg,
+ cache=resolved,
+ possible_after_dependencies=possible_after_dependencies,
+ ignore_exceptions=ignore_exceptions,
+ )
+ except Exception as exc: # noqa: BLE001
+ _LOGGER.error(
+ "Unable to resolve (after) dependencies for %s: %s", itg.domain, exc
+ )
+ return None
+
+ resolve_dependencies_tasks = {
+ itg.domain: create_eager_task(
+ _resolve_deps_catch_exceptions(itg),
+ name=f"resolve after dependencies {itg.domain}",
+ loop=hass.loop,
+ )
+ for itg in integrations
+ }
+
+ result = await asyncio.gather(*resolve_dependencies_tasks.values())
+
+ return {
+ domain: deps
+ for domain, deps in zip(resolve_dependencies_tasks, result, strict=True)
+ if deps is not None
+ }
+
+
+async def _do_resolve_dependencies(
+ itg: Integration,
+ *,
+ cache: _ResolveDependenciesCacheProtocol,
+ possible_after_dependencies: set[str] | None | UndefinedType = UNDEFINED,
+ ignore_exceptions: bool = False,
+) -> set[str]:
+ """Recursively resolve all dependencies.
+
+ Uses `cache` to cache the results.
+
+ If `possible_after_dependencies` is not UNDEFINED,
+ listed after dependencies are also considered.
+ If `possible_after_dependencies` is None,
+ all the possible after dependencies are considered.
+
+ If `ignore_exceptions` is True, exceptions are caught and ignored
+ and the normal resolution algorithm continues.
+ Otherwise, exceptions are raised.
+ """
+ resolved = cache
+ resolving: set[str] = set()
+
+ async def do_resolve_dependencies_impl(itg: Integration) -> set[str]:
+ domain = itg.domain
+
+ # If it's already resolved, no point doing it again.
+ if (result := resolved.get(itg)) is not None:
+ if isinstance(result, Exception):
+ raise result
+ return result
+
+ # If we are already resolving it, we have a circular dependency.
+ if domain in resolving:
+ if ignore_exceptions:
+ resolved[itg] = set()
+ return set()
+ exc = CircularDependency([domain])
+ resolved[itg] = exc
+ raise exc
+
+ resolving.add(domain)
+
+ dependencies_domains = set(itg.dependencies)
+ if possible_after_dependencies is not UNDEFINED:
+ if possible_after_dependencies is None:
+ after_dependencies: Iterable[str] = itg.after_dependencies
+ else:
+ after_dependencies = (
+ set(itg.after_dependencies) & possible_after_dependencies
+ )
+ dependencies_domains.update(after_dependencies)
+ dependencies = await async_get_integrations(itg.hass, dependencies_domains)
+
+ all_dependencies: set[str] = set()
+ for dep_domain, dep_integration in dependencies.items():
+ if isinstance(dep_integration, Exception):
+ if ignore_exceptions:
+ continue
+ resolved[itg] = dep_integration
+ raise dep_integration
+
+ all_dependencies.add(dep_domain)
+
+ try:
+ dep_dependencies = await do_resolve_dependencies_impl(dep_integration)
+ except CircularDependency as exc:
+ exc.extend_cycle(domain)
+ resolved[itg] = exc
+ raise
+ except Exception as exc:
+ resolved[itg] = exc
+ raise
+
+ all_dependencies.update(dep_dependencies)
+
+ resolving.remove(domain)
+
+ resolved[itg] = all_dependencies
+ return all_dependencies
+
+ return await do_resolve_dependencies_impl(itg)
+
+
class LoaderError(Exception):
"""Loader base error."""
@@ -1460,11 +1627,13 @@ class IntegrationNotLoaded(LoaderError):
class CircularDependency(LoaderError):
"""Raised when a circular dependency is found when resolving components."""
- def __init__(self, from_domain: str | set[str], to_domain: str) -> None:
+ def __init__(self, domain_cycle: list[str]) -> None:
"""Initialize circular dependency error."""
- super().__init__(f"Circular dependency detected: {from_domain} -> {to_domain}.")
- self.from_domain = from_domain
- self.to_domain = to_domain
+ super().__init__("Circular dependency detected", domain_cycle)
+
+ def extend_cycle(self, domain: str) -> None:
+ """Extend the cycle with the domain."""
+ self.args[1].insert(0, domain)
def _load_file(
@@ -1618,50 +1787,6 @@ def bind_hass[_CallableT: Callable[..., Any]](func: _CallableT) -> _CallableT:
return func
-async def _async_component_dependencies(
- hass: HomeAssistant,
- integration: Integration,
-) -> set[str]:
- """Get component dependencies."""
- loading: set[str] = set()
- loaded: set[str] = set()
-
- async def component_dependencies_impl(integration: Integration) -> None:
- """Recursively get component dependencies."""
- domain = integration.domain
- if not (dependencies := integration.dependencies):
- loaded.add(domain)
- return
-
- loading.add(domain)
- dep_integrations = await async_get_integrations(hass, dependencies)
- for dependency_domain, dep_integration in dep_integrations.items():
- if isinstance(dep_integration, Exception):
- raise dep_integration
-
- # If we are already loading it, we have a circular dependency.
- # We have to check it here to make sure that every integration that
- # depends on us, does not appear in our own after_dependencies.
- if conflict := loading.intersection(dep_integration.after_dependencies):
- raise CircularDependency(conflict, dependency_domain)
-
- # If we have already loaded it, no point doing it again.
- if dependency_domain in loaded:
- continue
-
- # If we are already loading it, we have a circular dependency.
- if dependency_domain in loading:
- raise CircularDependency(dependency_domain, domain)
-
- await component_dependencies_impl(dep_integration)
- loading.remove(domain)
- loaded.add(domain)
-
- await component_dependencies_impl(integration)
-
- return loaded
-
-
def _async_mount_config_dir(hass: HomeAssistant) -> None:
"""Mount config dir in order to load custom_component.
diff --git a/homeassistant/package_constraints.txt b/homeassistant/package_constraints.txt
index 82e5a025eef..fec703761d5 100644
--- a/homeassistant/package_constraints.txt
+++ b/homeassistant/package_constraints.txt
@@ -1,17 +1,18 @@
# Automatically generated by gen_requirements_all.py, do not edit
-aiodhcpwatcher==1.1.0
-aiodiscover==2.2.2
+aiodhcpwatcher==1.1.1
+aiodiscover==2.6.1
aiodns==3.2.0
aiohasupervisor==0.3.0
-aiohttp-asyncmdnsresolver==0.1.0
-aiohttp-fast-zlib==0.2.2
-aiohttp==3.11.12
+aiohttp-asyncmdnsresolver==0.1.1
+aiohttp-fast-zlib==0.2.3
+aiohttp==3.11.14
aiohttp_cors==0.7.0
aiousbwatcher==1.1.1
aiozoneinfo==0.2.3
+annotatedyaml==0.4.4
astral==2.2
-async-interrupt==1.2.1
+async-interrupt==1.2.2
async-upnp-client==0.43.0
atomicwrites-homeassistant==1.4.1
attrs==25.1.0
@@ -19,36 +20,36 @@ audioop-lts==0.2.1
av==13.1.0
awesomeversion==24.6.0
bcrypt==4.2.0
-bleak-retry-connector==3.8.1
+bleak-retry-connector==3.9.0
bleak==0.22.3
bluetooth-adapters==0.21.4
-bluetooth-auto-recovery==1.4.2
-bluetooth-data-tools==1.23.4
-cached-ipaddress==0.8.0
+bluetooth-auto-recovery==1.4.5
+bluetooth-data-tools==1.26.1
+cached-ipaddress==0.10.0
certifi>=2021.5.30
ciso8601==2.3.2
cronsim==2.6
-cryptography==44.0.0
-dbus-fast==2.33.0
-fnv-hash-fast==1.2.2
+cryptography==44.0.1
+dbus-fast==2.41.1
+fnv-hash-fast==1.4.0
go2rtc-client==0.1.2
ha-ffmpeg==3.2.2
-habluetooth==3.21.1
-hass-nabucasa==0.89.0
+habluetooth==3.37.0
+hass-nabucasa==0.94.0
hassil==2.2.3
home-assistant-bluetooth==1.13.1
-home-assistant-frontend==20250205.0
-home-assistant-intents==2025.2.5
+home-assistant-frontend==20250306.0
+home-assistant-intents==2025.3.5
httpx==0.28.1
ifaddr==0.2.0
-Jinja2==3.1.5
+Jinja2==3.1.6
lru-dict==1.3.0
mutagen==1.47.0
-orjson==3.10.12
+orjson==3.10.15
packaging>=23.1
paho-mqtt==2.1.0
Pillow==11.1.0
-propcache==0.2.1
+propcache==0.3.0
psutil-home-assistant==0.0.1
PyJWT==2.10.1
pymicro-vad==1.0.1
@@ -60,20 +61,20 @@ python-slugify==8.0.4
PyTurboJPEG==1.7.5
PyYAML==6.0.2
requests==2.32.3
-securetar==2025.1.4
-SQLAlchemy==2.0.38
+securetar==2025.2.1
+SQLAlchemy==2.0.39
standard-aifc==3.13.0
standard-telnetlib==3.13.0
typing-extensions>=4.12.2,<5.0
-ulid-transform==1.2.0
+ulid-transform==1.4.0
urllib3>=1.26.5,<2
-uv==0.5.27
+uv==0.6.8
voluptuous-openapi==0.0.6
voluptuous-serialize==2.6.0
voluptuous==0.15.2
webrtc-models==0.3.0
yarl==1.18.3
-zeroconf==0.143.0
+zeroconf==0.146.0
# Constrain pycryptodome to avoid vulnerability
# see https://github.com/home-assistant/core/pull/16238
@@ -108,7 +109,7 @@ uuid==1000000000.0.0
# these requirements are quite loose. As the entire stack has some outstanding issues, and
# even newer versions seem to introduce new issues, it's useful for us to pin all these
# requirements so we can directly link HA versions to these library versions.
-anyio==4.8.0
+anyio==4.9.0
h11==0.14.0
httpcore==1.0.7
diff --git a/homeassistant/scripts/check_config.py b/homeassistant/scripts/check_config.py
index a24568e9a6f..ca3df5080b5 100644
--- a/homeassistant/scripts/check_config.py
+++ b/homeassistant/scripts/check_config.py
@@ -12,6 +12,9 @@ import os
from typing import Any
from unittest.mock import patch
+from annotatedyaml import loader as yaml_loader
+from annotatedyaml.loader import Secrets
+
from homeassistant import core, loader
from homeassistant.config import get_default_config_dir
from homeassistant.config_entries import ConfigEntries
@@ -23,7 +26,6 @@ from homeassistant.helpers import (
issue_registry as ir,
)
from homeassistant.helpers.check_config import async_check_ha_config_file
-from homeassistant.util.yaml import Secrets, loader as yaml_loader
# mypy: allow-untyped-calls, allow-untyped-defs
@@ -31,9 +33,9 @@ REQUIREMENTS = ("colorlog==6.8.2",)
_LOGGER = logging.getLogger(__name__)
MOCKS: dict[str, tuple[str, Callable]] = {
- "load": ("homeassistant.util.yaml.loader.load_yaml", yaml_loader.load_yaml),
+ "load": ("annotatedyaml.loader.load_yaml", yaml_loader.load_yaml),
"load*": ("homeassistant.config.load_yaml_dict", yaml_loader.load_yaml_dict),
- "secrets": ("homeassistant.util.yaml.loader.secret_yaml", yaml_loader.secret_yaml),
+ "secrets": ("annotatedyaml.loader.secret_yaml", yaml_loader.secret_yaml),
}
PATCHES: dict[str, Any] = {}
diff --git a/homeassistant/setup.py b/homeassistant/setup.py
index dc4d0988b91..9572136559a 100644
--- a/homeassistant/setup.py
+++ b/homeassistant/setup.py
@@ -323,7 +323,7 @@ async def _async_setup_component(
translation.async_load_integrations(hass, integration_set), loop=hass.loop
)
# Validate all dependencies exist and there are no circular dependencies
- if not await integration.resolve_dependencies():
+ if await integration.resolve_dependencies() is None:
return False
# Process requirements as soon as possible, so we can import the component
diff --git a/homeassistant/strings.json b/homeassistant/strings.json
index fca55353aa0..29b7db7a011 100644
--- a/homeassistant/strings.json
+++ b/homeassistant/strings.json
@@ -1,13 +1,101 @@
{
"common": {
- "generic": {
- "model": "Model",
- "ui_managed": "Managed via UI"
+ "action": {
+ "close": "Close",
+ "connect": "Connect",
+ "disable": "Disable",
+ "disconnect": "Disconnect",
+ "enable": "Enable",
+ "open": "Open",
+ "pause": "Pause",
+ "reload": "Reload",
+ "restart": "Restart",
+ "start": "Start",
+ "stop": "Stop",
+ "toggle": "Toggle",
+ "turn_off": "Turn off",
+ "turn_on": "Turn on"
+ },
+ "config_flow": {
+ "abort": {
+ "already_configured_account": "Account is already configured",
+ "already_configured_device": "Device is already configured",
+ "already_configured_location": "Location is already configured",
+ "already_configured_service": "Service is already configured",
+ "already_in_progress": "Configuration flow is already in progress",
+ "cloud_not_connected": "Not connected to Home Assistant Cloud.",
+ "no_devices_found": "No devices found on the network",
+ "oauth2_authorize_url_timeout": "Timeout generating authorize URL.",
+ "oauth2_error": "Received invalid token data.",
+ "oauth2_failed": "Error while obtaining access token.",
+ "oauth2_missing_configuration": "The component is not configured. Please follow the documentation.",
+ "oauth2_missing_credentials": "The integration requires application credentials.",
+ "oauth2_no_url_available": "No URL available. For information about this error, [check the help section]({docs_url})",
+ "oauth2_timeout": "Timeout resolving OAuth token.",
+ "oauth2_unauthorized": "OAuth authorization error while obtaining access token.",
+ "oauth2_user_rejected_authorize": "Account linking rejected: {error}",
+ "reauth_successful": "Re-authentication was successful",
+ "reconfigure_successful": "Re-configuration was successful",
+ "single_instance_allowed": "Already configured. Only a single configuration possible.",
+ "unknown_authorize_url_generation": "Unknown error generating an authorize URL.",
+ "webhook_not_internet_accessible": "Your Home Assistant instance needs to be accessible from the internet to receive webhook messages."
+ },
+ "create_entry": {
+ "authenticated": "Successfully authenticated"
+ },
+ "data": {
+ "access_token": "Access token",
+ "api_key": "API key",
+ "api_token": "API token",
+ "device": "Device",
+ "elevation": "Elevation",
+ "email": "Email",
+ "host": "Host",
+ "ip": "IP address",
+ "language": "Language",
+ "latitude": "Latitude",
+ "llm_hass_api": "Control Home Assistant",
+ "location": "Location",
+ "longitude": "Longitude",
+ "mode": "Mode",
+ "name": "Name",
+ "password": "Password",
+ "path": "Path",
+ "pin": "PIN code",
+ "port": "Port",
+ "ssl": "Uses an SSL certificate",
+ "url": "URL",
+ "usb_path": "USB device path",
+ "username": "Username",
+ "verify_ssl": "Verify SSL certificate"
+ },
+ "description": {
+ "confirm_setup": "Do you want to start setup?"
+ },
+ "error": {
+ "cannot_connect": "Failed to connect",
+ "invalid_access_token": "Invalid access token",
+ "invalid_api_key": "Invalid API key",
+ "invalid_auth": "Invalid authentication",
+ "invalid_host": "Invalid hostname or IP address",
+ "timeout_connect": "Timeout establishing connection",
+ "unknown": "Unexpected error"
+ },
+ "title": {
+ "oauth2_pick_implementation": "Pick authentication method",
+ "reauth": "Authentication expired for {name}",
+ "via_hassio_addon": "{name} via Home Assistant add-on"
+ }
},
"device_automation": {
+ "action_type": {
+ "toggle": "Toggle {entity_name}",
+ "turn_off": "Turn off {entity_name}",
+ "turn_on": "Turn on {entity_name}"
+ },
"condition_type": {
- "is_on": "{entity_name} is on",
- "is_off": "{entity_name} is off"
+ "is_off": "{entity_name} is off",
+ "is_on": "{entity_name} is on"
},
"extra_fields": {
"above": "Above",
@@ -19,30 +107,35 @@
},
"trigger_type": {
"changed_states": "{entity_name} turned on or off",
- "turned_on": "{entity_name} turned on",
- "turned_off": "{entity_name} turned off"
- },
- "action_type": {
- "toggle": "Toggle {entity_name}",
- "turn_on": "Turn on {entity_name}",
- "turn_off": "Turn off {entity_name}"
+ "turned_off": "{entity_name} turned off",
+ "turned_on": "{entity_name} turned on"
}
},
- "action": {
- "connect": "Connect",
- "disconnect": "Disconnect",
- "enable": "Enable",
- "disable": "Disable",
+ "generic": {
+ "model": "Model",
+ "ui_managed": "Managed via UI"
+ },
+ "state": {
+ "active": "Active",
+ "charging": "Charging",
+ "closed": "Closed",
+ "connected": "Connected",
+ "disabled": "Disabled",
+ "discharging": "Discharging",
+ "disconnected": "Disconnected",
+ "enabled": "Enabled",
+ "home": "Home",
+ "idle": "Idle",
+ "locked": "Locked",
+ "no": "No",
+ "not_home": "Away",
+ "off": "Off",
+ "on": "On",
"open": "Open",
- "close": "Close",
- "reload": "Reload",
- "restart": "Restart",
- "start": "Start",
- "stop": "Stop",
- "pause": "Pause",
- "turn_on": "Turn on",
- "turn_off": "Turn off",
- "toggle": "Toggle"
+ "paused": "Paused",
+ "standby": "Standby",
+ "unlocked": "Unlocked",
+ "yes": "Yes"
},
"time": {
"monday": "Monday",
@@ -52,97 +145,6 @@
"friday": "Friday",
"saturday": "Saturday",
"sunday": "Sunday"
- },
- "state": {
- "off": "Off",
- "on": "On",
- "yes": "Yes",
- "no": "No",
- "open": "Open",
- "closed": "Closed",
- "enabled": "Enabled",
- "disabled": "Disabled",
- "connected": "Connected",
- "disconnected": "Disconnected",
- "locked": "Locked",
- "unlocked": "Unlocked",
- "active": "Active",
- "idle": "Idle",
- "standby": "Standby",
- "paused": "Paused",
- "home": "Home",
- "not_home": "Away"
- },
- "config_flow": {
- "title": {
- "oauth2_pick_implementation": "Pick authentication method",
- "reauth": "Authentication expired for {name}",
- "via_hassio_addon": "{name} via Home Assistant add-on"
- },
- "description": {
- "confirm_setup": "Do you want to start setup?"
- },
- "data": {
- "device": "Device",
- "name": "Name",
- "email": "Email",
- "username": "Username",
- "password": "Password",
- "host": "Host",
- "ip": "IP address",
- "port": "Port",
- "url": "URL",
- "usb_path": "USB device path",
- "access_token": "Access token",
- "api_key": "API key",
- "api_token": "API token",
- "llm_hass_api": "Control Home Assistant",
- "ssl": "Uses an SSL certificate",
- "verify_ssl": "Verify SSL certificate",
- "elevation": "Elevation",
- "longitude": "Longitude",
- "latitude": "Latitude",
- "location": "Location",
- "pin": "PIN code",
- "mode": "Mode",
- "path": "Path",
- "language": "Language"
- },
- "create_entry": {
- "authenticated": "Successfully authenticated"
- },
- "error": {
- "cannot_connect": "Failed to connect",
- "invalid_access_token": "Invalid access token",
- "invalid_api_key": "Invalid API key",
- "invalid_auth": "Invalid authentication",
- "invalid_host": "Invalid hostname or IP address",
- "unknown": "Unexpected error",
- "timeout_connect": "Timeout establishing connection"
- },
- "abort": {
- "single_instance_allowed": "Already configured. Only a single configuration possible.",
- "already_configured_account": "Account is already configured",
- "already_configured_device": "Device is already configured",
- "already_configured_location": "Location is already configured",
- "already_configured_service": "Service is already configured",
- "already_in_progress": "Configuration flow is already in progress",
- "no_devices_found": "No devices found on the network",
- "webhook_not_internet_accessible": "Your Home Assistant instance needs to be accessible from the internet to receive webhook messages.",
- "oauth2_error": "Received invalid token data.",
- "oauth2_timeout": "Timeout resolving OAuth token.",
- "oauth2_missing_configuration": "The component is not configured. Please follow the documentation.",
- "oauth2_missing_credentials": "The integration requires application credentials.",
- "oauth2_authorize_url_timeout": "Timeout generating authorize URL.",
- "oauth2_no_url_available": "No URL available. For information about this error, [check the help section]({docs_url})",
- "oauth2_user_rejected_authorize": "Account linking rejected: {error}",
- "oauth2_unauthorized": "OAuth authorization error while obtaining access token.",
- "oauth2_failed": "Error while obtaining access token.",
- "reauth_successful": "Re-authentication was successful",
- "reconfigure_successful": "Re-configuration was successful",
- "unknown_authorize_url_generation": "Unknown error generating an authorize URL.",
- "cloud_not_connected": "Not connected to Home Assistant Cloud."
- }
}
}
}
diff --git a/homeassistant/util/frozen_dataclass_compat.py b/homeassistant/util/frozen_dataclass_compat.py
index 81ce9961a0b..518515d4f85 100644
--- a/homeassistant/util/frozen_dataclass_compat.py
+++ b/homeassistant/util/frozen_dataclass_compat.py
@@ -63,7 +63,7 @@ class FrozenOrThawed(type):
)
def __new__(
- mcs, # noqa: N804 ruff bug, ruff does not understand this is a metaclass
+ mcs,
name: str,
bases: tuple[type, ...],
namespace: dict[Any, Any],
diff --git a/homeassistant/util/logging.py b/homeassistant/util/logging.py
index 2c4eb744614..1e516742bfe 100644
--- a/homeassistant/util/logging.py
+++ b/homeassistant/util/logging.py
@@ -2,14 +2,16 @@
from __future__ import annotations
+from collections import defaultdict
from collections.abc import Callable, Coroutine
from functools import partial, wraps
import inspect
import logging
import logging.handlers
-import queue
+from queue import SimpleQueue
+import time
import traceback
-from typing import Any, cast, overload
+from typing import Any, cast, overload, override
from homeassistant.core import (
HassJobType,
@@ -18,6 +20,59 @@ from homeassistant.core import (
get_hassjob_callable_job_type,
)
+_LOGGER = logging.getLogger(__name__)
+
+
+class HomeAssistantQueueListener(logging.handlers.QueueListener):
+ """Custom QueueListener to watch for noisy loggers."""
+
+ LOG_COUNTS_RESET_INTERVAL = 300
+ MAX_LOGS_COUNT = 200
+
+ _last_reset: float
+ _log_counts: dict[str, int]
+ _warned_modules: set[str]
+
+ def __init__(
+ self, queue: SimpleQueue[logging.Handler], *handlers: logging.Handler
+ ) -> None:
+ """Initialize the handler."""
+ super().__init__(queue, *handlers)
+ self._warned_modules = set()
+ self._reset_counters(time.time())
+
+ @override
+ def handle(self, record: logging.LogRecord) -> None:
+ """Handle the record."""
+ super().handle(record)
+
+ if record.levelno < logging.INFO:
+ return
+
+ if (record.created - self._last_reset) > self.LOG_COUNTS_RESET_INTERVAL:
+ self._reset_counters(record.created)
+
+ module_name = record.name
+ if module_name == __name__ or module_name in self._warned_modules:
+ return
+
+ self._log_counts[module_name] += 1
+ module_count = self._log_counts[module_name]
+ if module_count < self.MAX_LOGS_COUNT:
+ return
+
+ _LOGGER.warning(
+ "Module %s is logging too frequently. %d messages since last count",
+ module_name,
+ module_count,
+ )
+ self._warned_modules.add(module_name)
+
+ def _reset_counters(self, time_sec: float) -> None:
+ _LOGGER.debug("Resetting log counters")
+ self._last_reset = time_sec
+ self._log_counts = defaultdict(int)
+
class HomeAssistantQueueHandler(logging.handlers.QueueHandler):
"""Process the log in another thread."""
@@ -60,7 +115,7 @@ def async_activate_log_queue_handler(hass: HomeAssistant) -> None:
This allows us to avoid blocking I/O and formatting messages
in the event loop as log messages are written in another thread.
"""
- simple_queue: queue.SimpleQueue[logging.Handler] = queue.SimpleQueue()
+ simple_queue: SimpleQueue[logging.Handler] = SimpleQueue()
queue_handler = HomeAssistantQueueHandler(simple_queue)
logging.root.addHandler(queue_handler)
@@ -71,7 +126,7 @@ def async_activate_log_queue_handler(hass: HomeAssistant) -> None:
logging.root.removeHandler(handler)
migrated_handlers.append(handler)
- listener = logging.handlers.QueueListener(simple_queue, *migrated_handlers)
+ listener = HomeAssistantQueueListener(simple_queue, *migrated_handlers)
queue_handler.listener = listener
listener.start()
diff --git a/homeassistant/util/unit_conversion.py b/homeassistant/util/unit_conversion.py
index 67258c9cd09..f2619c5dd61 100644
--- a/homeassistant/util/unit_conversion.py
+++ b/homeassistant/util/unit_conversion.py
@@ -4,6 +4,7 @@ from __future__ import annotations
from collections.abc import Callable
from functools import lru_cache
+from math import floor, log10
from homeassistant.const import (
CONCENTRATION_PARTS_PER_BILLION,
@@ -144,6 +145,15 @@ class BaseUnitConverter:
from_ratio, to_ratio = cls._get_from_to_ratio(from_unit, to_unit)
return from_ratio / to_ratio
+ @classmethod
+ @lru_cache
+ def get_unit_floored_log_ratio(
+ cls, from_unit: str | None, to_unit: str | None
+ ) -> float:
+ """Get floored base10 log ratio between units of measurement."""
+ from_ratio, to_ratio = cls._get_from_to_ratio(from_unit, to_unit)
+ return floor(max(0, log10(from_ratio / to_ratio)))
+
@classmethod
@lru_cache
def _are_unit_inverses(cls, from_unit: str | None, to_unit: str | None) -> bool:
diff --git a/homeassistant/util/yaml/__init__.py b/homeassistant/util/yaml/__init__.py
index 3b1f5c4cc0a..323383ef53f 100644
--- a/homeassistant/util/yaml/__init__.py
+++ b/homeassistant/util/yaml/__init__.py
@@ -1,17 +1,10 @@
"""YAML utility functions."""
-from .const import SECRET_YAML
+from annotatedyaml import SECRET_YAML, Input, YamlTypeError
+from annotatedyaml.input import UndefinedSubstitution, extract_inputs, substitute
+
from .dumper import dump, save_yaml
-from .input import UndefinedSubstitution, extract_inputs, substitute
-from .loader import (
- Secrets,
- YamlTypeError,
- load_yaml,
- load_yaml_dict,
- parse_yaml,
- secret_yaml,
-)
-from .objects import Input
+from .loader import Secrets, load_yaml, load_yaml_dict, parse_yaml, secret_yaml
__all__ = [
"SECRET_YAML",
diff --git a/homeassistant/util/yaml/const.py b/homeassistant/util/yaml/const.py
deleted file mode 100644
index 811c7d149f7..00000000000
--- a/homeassistant/util/yaml/const.py
+++ /dev/null
@@ -1,3 +0,0 @@
-"""Constants."""
-
-SECRET_YAML = "secrets.yaml"
diff --git a/homeassistant/util/yaml/dumper.py b/homeassistant/util/yaml/dumper.py
index 61772b6989d..059be2c1c5b 100644
--- a/homeassistant/util/yaml/dumper.py
+++ b/homeassistant/util/yaml/dumper.py
@@ -1,96 +1,5 @@
"""Custom dumper and representers."""
-from collections import OrderedDict
-from typing import Any
+from annotatedyaml.dumper import add_representer, dump, represent_odict, save_yaml
-import yaml
-
-from .objects import Input, NodeDictClass, NodeListClass, NodeStrClass
-
-# mypy: allow-untyped-calls, no-warn-return-any
-
-
-try:
- from yaml import CSafeDumper as FastestAvailableSafeDumper
-except ImportError:
- from yaml import ( # type: ignore[assignment]
- SafeDumper as FastestAvailableSafeDumper,
- )
-
-
-def dump(_dict: dict | list) -> str:
- """Dump YAML to a string and remove null."""
- return yaml.dump(
- _dict,
- default_flow_style=False,
- allow_unicode=True,
- sort_keys=False,
- Dumper=FastestAvailableSafeDumper,
- ).replace(": null\n", ":\n")
-
-
-def save_yaml(path: str, data: dict) -> None:
- """Save YAML to a file."""
- # Dump before writing to not truncate the file if dumping fails
- str_data = dump(data)
- with open(path, "w", encoding="utf-8") as outfile:
- outfile.write(str_data)
-
-
-# From: https://gist.github.com/miracle2k/3184458
-def represent_odict( # type: ignore[no-untyped-def]
- dumper, tag, mapping, flow_style=None
-) -> yaml.MappingNode:
- """Like BaseRepresenter.represent_mapping but does not issue the sort()."""
- value: list = []
- node = yaml.MappingNode(tag, value, flow_style=flow_style)
- if dumper.alias_key is not None:
- dumper.represented_objects[dumper.alias_key] = node
- best_style = True
- if hasattr(mapping, "items"):
- mapping = mapping.items()
- for item_key, item_value in mapping:
- node_key = dumper.represent_data(item_key)
- node_value = dumper.represent_data(item_value)
- if not (isinstance(node_key, yaml.ScalarNode) and not node_key.style):
- best_style = False
- if not (isinstance(node_value, yaml.ScalarNode) and not node_value.style):
- best_style = False
- value.append((node_key, node_value))
- if flow_style is None:
- if dumper.default_flow_style is not None:
- node.flow_style = dumper.default_flow_style
- else:
- node.flow_style = best_style
- return node
-
-
-def add_representer(klass: Any, representer: Any) -> None:
- """Add to representer to the dumper."""
- FastestAvailableSafeDumper.add_representer(klass, representer)
-
-
-add_representer(
- OrderedDict,
- lambda dumper, value: represent_odict(dumper, "tag:yaml.org,2002:map", value),
-)
-
-add_representer(
- NodeDictClass,
- lambda dumper, value: represent_odict(dumper, "tag:yaml.org,2002:map", value),
-)
-
-add_representer(
- NodeListClass,
- lambda dumper, value: dumper.represent_sequence("tag:yaml.org,2002:seq", value),
-)
-
-add_representer(
- NodeStrClass,
- lambda dumper, value: dumper.represent_scalar("tag:yaml.org,2002:str", str(value)),
-)
-
-add_representer(
- Input,
- lambda dumper, value: dumper.represent_scalar("!input", value.name),
-)
+__all__ = ["add_representer", "dump", "represent_odict", "save_yaml"]
diff --git a/homeassistant/util/yaml/input.py b/homeassistant/util/yaml/input.py
index ff9b37f18f1..5dad8a63ae5 100644
--- a/homeassistant/util/yaml/input.py
+++ b/homeassistant/util/yaml/input.py
@@ -2,55 +2,8 @@
from __future__ import annotations
-from typing import Any
+from annotatedyaml.input import UndefinedSubstitution, extract_inputs, substitute
from .objects import Input
-
-class UndefinedSubstitution(Exception):
- """Error raised when we find a substitution that is not defined."""
-
- def __init__(self, input_name: str) -> None:
- """Initialize the undefined substitution exception."""
- super().__init__(f"No substitution found for input {input_name}")
- self.input = input
-
-
-def extract_inputs(obj: Any) -> set[str]:
- """Extract input from a structure."""
- found: set[str] = set()
- _extract_inputs(obj, found)
- return found
-
-
-def _extract_inputs(obj: Any, found: set[str]) -> None:
- """Extract input from a structure."""
- if isinstance(obj, Input):
- found.add(obj.name)
- return
-
- if isinstance(obj, list):
- for val in obj:
- _extract_inputs(val, found)
- return
-
- if isinstance(obj, dict):
- for val in obj.values():
- _extract_inputs(val, found)
- return
-
-
-def substitute(obj: Any, substitutions: dict[str, Any]) -> Any:
- """Substitute values."""
- if isinstance(obj, Input):
- if obj.name not in substitutions:
- raise UndefinedSubstitution(obj.name)
- return substitutions[obj.name]
-
- if isinstance(obj, list):
- return [substitute(val, substitutions) for val in obj]
-
- if isinstance(obj, dict):
- return {key: substitute(val, substitutions) for key, val in obj.items()}
-
- return obj
+__all__ = ["Input", "UndefinedSubstitution", "extract_inputs", "substitute"]
diff --git a/homeassistant/util/yaml/loader.py b/homeassistant/util/yaml/loader.py
index 3911d62040b..1f8338a1ff7 100644
--- a/homeassistant/util/yaml/loader.py
+++ b/homeassistant/util/yaml/loader.py
@@ -2,157 +2,37 @@
from __future__ import annotations
-from collections.abc import Callable, Iterator
-import fnmatch
-from io import StringIO, TextIOWrapper
-import logging
+from io import StringIO
import os
-from pathlib import Path
-from typing import Any, TextIO, overload
+from typing import TextIO
+from annotatedyaml import YAMLException, YamlTypeError
+from annotatedyaml.loader import (
+ HAS_C_LOADER,
+ JSON_TYPE,
+ LoaderType,
+ Secrets,
+ add_constructor,
+ load_yaml as load_annotated_yaml,
+ load_yaml_dict as load_annotated_yaml_dict,
+ parse_yaml as parse_annotated_yaml,
+ secret_yaml as annotated_secret_yaml,
+)
import yaml
-try:
- from yaml import CSafeLoader as FastestAvailableSafeLoader
-
- HAS_C_LOADER = True
-except ImportError:
- HAS_C_LOADER = False
- from yaml import ( # type: ignore[assignment]
- SafeLoader as FastestAvailableSafeLoader,
- )
-
-from propcache.api import cached_property
-
from homeassistant.exceptions import HomeAssistantError
-from .const import SECRET_YAML
-from .objects import Input, NodeDictClass, NodeListClass, NodeStrClass
-
-# mypy: allow-untyped-calls, no-warn-return-any
-
-JSON_TYPE = list | dict | str
-
-_LOGGER = logging.getLogger(__name__)
-
-
-class YamlTypeError(HomeAssistantError):
- """Raised by load_yaml_dict if top level data is not a dict."""
-
-
-class Secrets:
- """Store secrets while loading YAML."""
-
- def __init__(self, config_dir: Path) -> None:
- """Initialize secrets."""
- self.config_dir = config_dir
- self._cache: dict[Path, dict[str, str]] = {}
-
- def get(self, requester_path: str, secret: str) -> str:
- """Return the value of a secret."""
- current_path = Path(requester_path)
-
- secret_dir = current_path
- while True:
- secret_dir = secret_dir.parent
-
- try:
- secret_dir.relative_to(self.config_dir)
- except ValueError:
- # We went above the config dir
- break
-
- secrets = self._load_secret_yaml(secret_dir)
-
- if secret in secrets:
- _LOGGER.debug(
- "Secret %s retrieved from secrets.yaml in folder %s",
- secret,
- secret_dir,
- )
- return secrets[secret]
-
- raise HomeAssistantError(f"Secret {secret} not defined")
-
- def _load_secret_yaml(self, secret_dir: Path) -> dict[str, str]:
- """Load the secrets yaml from path."""
- if (secret_path := secret_dir / SECRET_YAML) in self._cache:
- return self._cache[secret_path]
-
- _LOGGER.debug("Loading %s", secret_path)
- try:
- secrets = load_yaml(str(secret_path))
-
- if not isinstance(secrets, dict):
- raise HomeAssistantError("Secrets is not a dictionary")
-
- if "logger" in secrets:
- logger = str(secrets["logger"]).lower()
- if logger == "debug":
- _LOGGER.setLevel(logging.DEBUG)
- else:
- _LOGGER.error(
- (
- "Error in secrets.yaml: 'logger: debug' expected, but"
- " 'logger: %s' found"
- ),
- logger,
- )
- del secrets["logger"]
- except FileNotFoundError:
- secrets = {}
-
- self._cache[secret_path] = secrets
-
- return secrets
-
-
-class _LoaderMixin:
- """Mixin class with extensions for YAML loader."""
-
- name: str
- stream: Any
-
- @cached_property
- def get_name(self) -> str:
- """Get the name of the loader."""
- return self.name
-
- @cached_property
- def get_stream_name(self) -> str:
- """Get the name of the stream."""
- return getattr(self.stream, "name", "")
-
-
-class FastSafeLoader(FastestAvailableSafeLoader, _LoaderMixin):
- """The fastest available safe loader, either C or Python."""
-
- def __init__(self, stream: Any, secrets: Secrets | None = None) -> None:
- """Initialize a safe line loader."""
- self.stream = stream
-
- # Set name in same way as the Python loader does in yaml.reader.__init__
- if isinstance(stream, str):
- self.name = ""
- elif isinstance(stream, bytes):
- self.name = ""
- else:
- self.name = getattr(stream, "name", "")
-
- super().__init__(stream)
- self.secrets = secrets
-
-
-class PythonSafeLoader(yaml.SafeLoader, _LoaderMixin):
- """Python safe loader."""
-
- def __init__(self, stream: Any, secrets: Secrets | None = None) -> None:
- """Initialize a safe line loader."""
- super().__init__(stream)
- self.secrets = secrets
-
-
-type LoaderType = FastSafeLoader | PythonSafeLoader
+__all__ = [
+ "HAS_C_LOADER",
+ "JSON_TYPE",
+ "Secrets",
+ "YamlTypeError",
+ "add_constructor",
+ "load_yaml",
+ "load_yaml_dict",
+ "parse_yaml",
+ "secret_yaml",
+]
def load_yaml(
@@ -164,15 +44,9 @@ def load_yaml(
except for FileNotFoundError which will be re-raised.
"""
try:
- with open(fname, encoding="utf-8") as conf_file:
- return parse_yaml(conf_file, secrets)
- except UnicodeDecodeError as exc:
- _LOGGER.error("Unable to read file %s: %s", fname, exc)
- raise HomeAssistantError(exc) from exc
- except FileNotFoundError:
- raise
- except OSError as exc:
- raise HomeAssistantError(exc) from exc
+ return load_annotated_yaml(fname, secrets)
+ except YAMLException as exc:
+ raise HomeAssistantError(str(exc)) from exc
def load_yaml_dict(
@@ -183,320 +57,27 @@ def load_yaml_dict(
Raise if the top level is not a dict.
Return an empty dict if the file is empty.
"""
- loaded_yaml = load_yaml(fname, secrets)
- if loaded_yaml is None:
- loaded_yaml = {}
- if not isinstance(loaded_yaml, dict):
- raise YamlTypeError(f"YAML file {fname} does not contain a dict")
- return loaded_yaml
+ try:
+ return load_annotated_yaml_dict(fname, secrets)
+ except YamlTypeError:
+ raise
+ except YAMLException as exc:
+ raise HomeAssistantError(str(exc)) from exc
def parse_yaml(
content: str | TextIO | StringIO, secrets: Secrets | None = None
) -> JSON_TYPE:
"""Parse YAML with the fastest available loader."""
- if not HAS_C_LOADER:
- return _parse_yaml_python(content, secrets)
try:
- return _parse_yaml(FastSafeLoader, content, secrets)
- except yaml.YAMLError:
- # Loading failed, so we now load with the Python loader which has more
- # readable exceptions
- if isinstance(content, (StringIO, TextIO, TextIOWrapper)):
- # Rewind the stream so we can try again
- content.seek(0, 0)
- return _parse_yaml_python(content, secrets)
-
-
-def _parse_yaml_python(
- content: str | TextIO | StringIO, secrets: Secrets | None = None
-) -> JSON_TYPE:
- """Parse YAML with the python loader (this is very slow)."""
- try:
- return _parse_yaml(PythonSafeLoader, content, secrets)
- except yaml.YAMLError as exc:
- _LOGGER.error(str(exc))
- raise HomeAssistantError(exc) from exc
-
-
-def _parse_yaml(
- loader: type[FastSafeLoader | PythonSafeLoader],
- content: str | TextIO,
- secrets: Secrets | None = None,
-) -> JSON_TYPE:
- """Load a YAML file."""
- return yaml.load(content, Loader=lambda stream: loader(stream, secrets)) # type: ignore[arg-type]
-
-
-@overload
-def _add_reference(
- obj: list | NodeListClass, loader: LoaderType, node: yaml.nodes.Node
-) -> NodeListClass: ...
-
-
-@overload
-def _add_reference(
- obj: str | NodeStrClass, loader: LoaderType, node: yaml.nodes.Node
-) -> NodeStrClass: ...
-
-
-@overload
-def _add_reference(
- obj: dict | NodeDictClass, loader: LoaderType, node: yaml.nodes.Node
-) -> NodeDictClass: ...
-
-
-def _add_reference(
- obj: dict | list | str | NodeDictClass | NodeListClass | NodeStrClass,
- loader: LoaderType,
- node: yaml.nodes.Node,
-) -> NodeDictClass | NodeListClass | NodeStrClass:
- """Add file reference information to an object."""
- if isinstance(obj, list):
- obj = NodeListClass(obj)
- elif isinstance(obj, str):
- obj = NodeStrClass(obj)
- elif isinstance(obj, dict):
- obj = NodeDictClass(obj)
- return _add_reference_to_node_class(obj, loader, node)
-
-
-@overload
-def _add_reference_to_node_class(
- obj: NodeListClass, loader: LoaderType, node: yaml.nodes.Node
-) -> NodeListClass: ...
-
-
-@overload
-def _add_reference_to_node_class(
- obj: NodeStrClass, loader: LoaderType, node: yaml.nodes.Node
-) -> NodeStrClass: ...
-
-
-@overload
-def _add_reference_to_node_class(
- obj: NodeDictClass, loader: LoaderType, node: yaml.nodes.Node
-) -> NodeDictClass: ...
-
-
-def _add_reference_to_node_class(
- obj: NodeDictClass | NodeListClass | NodeStrClass,
- loader: LoaderType,
- node: yaml.nodes.Node,
-) -> NodeDictClass | NodeListClass | NodeStrClass:
- """Add file reference information to a node class object."""
- try: # suppress is much slower
- obj.__config_file__ = loader.get_name
- obj.__line__ = node.start_mark.line + 1
- except AttributeError:
- pass
- return obj
-
-
-def _raise_if_no_value[NodeT: yaml.nodes.Node, _R](
- func: Callable[[LoaderType, NodeT], _R],
-) -> Callable[[LoaderType, NodeT], _R]:
- def wrapper(loader: LoaderType, node: NodeT) -> _R:
- if not node.value:
- raise HomeAssistantError(
- f"{node.start_mark}: {node.tag} needs an argument."
- )
- return func(loader, node)
-
- return wrapper
-
-
-@_raise_if_no_value
-def _include_yaml(loader: LoaderType, node: yaml.nodes.Node) -> JSON_TYPE:
- """Load another YAML file and embed it using the !include tag.
-
- Example:
- device_tracker: !include device_tracker.yaml
-
- """
- fname = os.path.join(os.path.dirname(loader.get_name), node.value)
- try:
- loaded_yaml = load_yaml(fname, loader.secrets)
- if loaded_yaml is None:
- loaded_yaml = NodeDictClass()
- return _add_reference(loaded_yaml, loader, node)
- except FileNotFoundError as exc:
- raise HomeAssistantError(
- f"{node.start_mark}: Unable to read file {fname}"
- ) from exc
-
-
-def _is_file_valid(name: str) -> bool:
- """Decide if a file is valid."""
- return not name.startswith(".")
-
-
-def _find_files(directory: str, pattern: str) -> Iterator[str]:
- """Recursively load files in a directory."""
- for root, dirs, files in os.walk(directory, topdown=True):
- dirs[:] = [d for d in dirs if _is_file_valid(d)]
- for basename in sorted(files):
- if _is_file_valid(basename) and fnmatch.fnmatch(basename, pattern):
- filename = os.path.join(root, basename)
- yield filename
-
-
-@_raise_if_no_value
-def _include_dir_named_yaml(loader: LoaderType, node: yaml.nodes.Node) -> NodeDictClass:
- """Load multiple files from directory as a dictionary."""
- mapping = NodeDictClass()
- loc = os.path.join(os.path.dirname(loader.get_name), node.value)
- for fname in _find_files(loc, "*.yaml"):
- filename = os.path.splitext(os.path.basename(fname))[0]
- if os.path.basename(fname) == SECRET_YAML:
- continue
- loaded_yaml = load_yaml(fname, loader.secrets)
- if loaded_yaml is None:
- # Special case, an empty file included by !include_dir_named is treated
- # as an empty dictionary
- loaded_yaml = NodeDictClass()
- mapping[filename] = loaded_yaml
- return _add_reference_to_node_class(mapping, loader, node)
-
-
-@_raise_if_no_value
-def _include_dir_merge_named_yaml(
- loader: LoaderType, node: yaml.nodes.Node
-) -> NodeDictClass:
- """Load multiple files from directory as a merged dictionary."""
- mapping = NodeDictClass()
- loc = os.path.join(os.path.dirname(loader.get_name), node.value)
- for fname in _find_files(loc, "*.yaml"):
- if os.path.basename(fname) == SECRET_YAML:
- continue
- loaded_yaml = load_yaml(fname, loader.secrets)
- if isinstance(loaded_yaml, dict):
- mapping.update(loaded_yaml)
- return _add_reference_to_node_class(mapping, loader, node)
-
-
-@_raise_if_no_value
-def _include_dir_list_yaml(
- loader: LoaderType, node: yaml.nodes.Node
-) -> list[JSON_TYPE]:
- """Load multiple files from directory as a list."""
- loc = os.path.join(os.path.dirname(loader.get_name), node.value)
- return [
- loaded_yaml
- for f in _find_files(loc, "*.yaml")
- if os.path.basename(f) != SECRET_YAML
- and (loaded_yaml := load_yaml(f, loader.secrets)) is not None
- ]
-
-
-@_raise_if_no_value
-def _include_dir_merge_list_yaml(
- loader: LoaderType, node: yaml.nodes.Node
-) -> JSON_TYPE:
- """Load multiple files from directory as a merged list."""
- loc: str = os.path.join(os.path.dirname(loader.get_name), node.value)
- merged_list: list[JSON_TYPE] = []
- for fname in _find_files(loc, "*.yaml"):
- if os.path.basename(fname) == SECRET_YAML:
- continue
- loaded_yaml = load_yaml(fname, loader.secrets)
- if isinstance(loaded_yaml, list):
- merged_list.extend(loaded_yaml)
- return _add_reference(merged_list, loader, node)
-
-
-def _handle_mapping_tag(
- loader: LoaderType, node: yaml.nodes.MappingNode
-) -> NodeDictClass:
- """Load YAML mappings into an ordered dictionary to preserve key order."""
- loader.flatten_mapping(node)
- nodes = loader.construct_pairs(node)
-
- seen: dict = {}
- for (key, _), (child_node, _) in zip(nodes, node.value, strict=False):
- line = child_node.start_mark.line
-
- try:
- hash(key)
- except TypeError as exc:
- fname = loader.get_stream_name
- raise yaml.MarkedYAMLError(
- context=f'invalid key: "{key}"',
- context_mark=yaml.Mark(
- fname,
- 0,
- line,
- -1,
- None,
- None, # type: ignore[arg-type]
- ),
- ) from exc
-
- if key in seen:
- fname = loader.get_stream_name
- _LOGGER.warning(
- 'YAML file %s contains duplicate key "%s". Check lines %d and %d',
- fname,
- key,
- seen[key],
- line,
- )
- seen[key] = line
-
- return _add_reference_to_node_class(NodeDictClass(nodes), loader, node)
-
-
-def _construct_seq(loader: LoaderType, node: yaml.nodes.Node) -> JSON_TYPE:
- """Add line number and file name to Load YAML sequence."""
- (obj,) = loader.construct_yaml_seq(node)
- return _add_reference(obj, loader, node)
-
-
-def _handle_scalar_tag(
- loader: LoaderType, node: yaml.nodes.ScalarNode
-) -> str | int | float | None:
- """Add line number and file name to Load YAML sequence."""
- obj = node.value
- if not isinstance(obj, str):
- return obj
- return _add_reference_to_node_class(NodeStrClass(obj), loader, node)
-
-
-def _env_var_yaml(loader: LoaderType, node: yaml.nodes.Node) -> str:
- """Load environment variables and embed it into the configuration YAML."""
- args = node.value.split()
-
- # Check for a default value
- if len(args) > 1:
- return os.getenv(args[0], " ".join(args[1:]))
- if args[0] in os.environ:
- return os.environ[args[0]]
- _LOGGER.error("Environment variable %s not defined", node.value)
- raise HomeAssistantError(node.value)
+ return parse_annotated_yaml(content, secrets)
+ except YAMLException as exc:
+ raise HomeAssistantError(str(exc)) from exc
def secret_yaml(loader: LoaderType, node: yaml.nodes.Node) -> JSON_TYPE:
"""Load secrets and embed it into the configuration YAML."""
- if loader.secrets is None:
- raise HomeAssistantError("Secrets not supported in this YAML file")
-
- return loader.secrets.get(loader.get_name, node.value)
-
-
-def add_constructor(tag: Any, constructor: Any) -> None:
- """Add to constructor to all loaders."""
- for yaml_loader in (FastSafeLoader, PythonSafeLoader):
- yaml_loader.add_constructor(tag, constructor)
-
-
-add_constructor("!include", _include_yaml)
-add_constructor(yaml.resolver.BaseResolver.DEFAULT_MAPPING_TAG, _handle_mapping_tag)
-add_constructor(yaml.resolver.BaseResolver.DEFAULT_SCALAR_TAG, _handle_scalar_tag)
-add_constructor(yaml.resolver.BaseResolver.DEFAULT_SEQUENCE_TAG, _construct_seq)
-add_constructor("!env_var", _env_var_yaml)
-add_constructor("!secret", secret_yaml)
-add_constructor("!include_dir_list", _include_dir_list_yaml)
-add_constructor("!include_dir_merge_list", _include_dir_merge_list_yaml)
-add_constructor("!include_dir_named", _include_dir_named_yaml)
-add_constructor("!include_dir_merge_named", _include_dir_merge_named_yaml)
-add_constructor("!input", Input.from_node)
+ try:
+ return annotated_secret_yaml(loader, node)
+ except YAMLException as exc:
+ raise HomeAssistantError(str(exc)) from exc
diff --git a/homeassistant/util/yaml/objects.py b/homeassistant/util/yaml/objects.py
index 7e4019331c6..26714b0fdd4 100644
--- a/homeassistant/util/yaml/objects.py
+++ b/homeassistant/util/yaml/objects.py
@@ -2,52 +2,6 @@
from __future__ import annotations
-from dataclasses import dataclass
-from typing import Any
+from annotatedyaml import Input, NodeDictClass, NodeListClass, NodeStrClass
-import voluptuous as vol
-from voluptuous.schema_builder import _compile_scalar
-import yaml
-
-
-class NodeListClass(list):
- """Wrapper class to be able to add attributes on a list."""
-
- __slots__ = ("__config_file__", "__line__")
-
- __config_file__: str
- __line__: int | str
-
-
-class NodeStrClass(str):
- """Wrapper class to be able to add attributes on a string."""
-
- __slots__ = ("__config_file__", "__line__")
-
- __config_file__: str
- __line__: int | str
-
- def __voluptuous_compile__(self, schema: vol.Schema) -> Any:
- """Needed because vol.Schema.compile does not handle str subclasses."""
- return _compile_scalar(self) # type: ignore[no-untyped-call]
-
-
-class NodeDictClass(dict):
- """Wrapper class to be able to add attributes on a dict."""
-
- __slots__ = ("__config_file__", "__line__")
-
- __config_file__: str
- __line__: int | str
-
-
-@dataclass(slots=True, frozen=True)
-class Input:
- """Input that should be substituted."""
-
- name: str
-
- @classmethod
- def from_node(cls, loader: yaml.Loader, node: yaml.nodes.Node) -> Input:
- """Create a new placeholder from a node."""
- return cls(node.value)
+__all__ = ["Input", "NodeDictClass", "NodeListClass", "NodeStrClass"]
diff --git a/mypy.ini b/mypy.ini
index 2d9821b1c64..852678677bb 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -785,6 +785,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.azure_storage.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.backup.*]
check_untyped_defs = true
disallow_incomplete_defs = true
@@ -1105,6 +1115,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.comelit.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.command_line.*]
check_untyped_defs = true
disallow_incomplete_defs = true
@@ -2096,6 +2116,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.home_connect.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.homeassistant.*]
check_untyped_defs = true
disallow_incomplete_defs = true
@@ -3696,6 +3726,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.pyload.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.python_script.*]
check_untyped_defs = true
disallow_incomplete_defs = true
@@ -3816,6 +3856,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.remember_the_milk.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.remote.*]
check_untyped_defs = true
disallow_incomplete_defs = true
@@ -3826,6 +3876,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.remote_calendar.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.renault.*]
check_untyped_defs = true
disallow_incomplete_defs = true
@@ -4126,6 +4186,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.sensorpush_cloud.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.sensoterra.*]
check_untyped_defs = true
disallow_incomplete_defs = true
@@ -4999,6 +5069,16 @@ disallow_untyped_defs = true
warn_return_any = true
warn_unreachable = true
+[mypy-homeassistant.components.vodafone_station.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
[mypy-homeassistant.components.wake_on_lan.*]
check_untyped_defs = true
disallow_incomplete_defs = true
diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py
index f76e0b43c10..ca7777da959 100644
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@@ -252,7 +252,7 @@ _FUNCTION_MATCH: dict[str, list[TypeHintMatch]] = {
arg_types={
0: "HomeAssistant",
1: "ConfigEntry",
- 2: "AddEntitiesCallback",
+ 2: "AddConfigEntryEntitiesCallback",
},
return_type=None,
),
@@ -1410,6 +1410,16 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
],
),
],
+ "entity": [
+ ClassTypeHintMatch(
+ base_class="Entity",
+ matches=_ENTITY_MATCH,
+ ),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
+ ],
"fan": [
ClassTypeHintMatch(
base_class="Entity",
@@ -2558,7 +2568,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
),
TypeHintMatch(
function_name="in_progress",
- return_type=["bool", "int", None],
+ return_type=["bool", None],
),
TypeHintMatch(
function_name="latest_version",
@@ -2580,6 +2590,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
function_name="title",
return_type=["str", None],
),
+ TypeHintMatch(
+ function_name="update_percentage",
+ return_type=["int", "float", None],
+ ),
TypeHintMatch(
function_name="install",
arg_types={1: "str | None", 2: "bool"},
diff --git a/pyproject.toml b/pyproject.toml
index 3936fdb3a1e..1bd74791a18 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,11 +1,12 @@
[build-system]
-requires = ["setuptools==75.1.0"]
+requires = ["setuptools==77.0.1"]
build-backend = "setuptools.build_meta"
[project]
name = "homeassistant"
-version = "2025.3.0.dev0"
-license = {text = "Apache-2.0"}
+version = "2025.4.0.dev0"
+license = "Apache-2.0"
+license-files = ["LICENSE*", "homeassistant/backports/LICENSE*"]
description = "Open-source home automation platform running on Python 3."
readme = "README.rst"
authors = [
@@ -16,7 +17,6 @@ classifiers = [
"Development Status :: 5 - Production/Stable",
"Intended Audience :: End Users/Desktop",
"Intended Audience :: Developers",
- "License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3.13",
"Topic :: Home Automation",
@@ -28,13 +28,14 @@ dependencies = [
# change behavior based on presence of supervisor. Deprecated with #127228
# Lib can be removed with 2025.11
"aiohasupervisor==0.3.0",
- "aiohttp==3.11.12",
+ "aiohttp==3.11.14",
"aiohttp_cors==0.7.0",
- "aiohttp-fast-zlib==0.2.2",
- "aiohttp-asyncmdnsresolver==0.1.0",
+ "aiohttp-fast-zlib==0.2.3",
+ "aiohttp-asyncmdnsresolver==0.1.1",
"aiozoneinfo==0.2.3",
+ "annotatedyaml==0.4.4",
"astral==2.2",
- "async-interrupt==1.2.1",
+ "async-interrupt==1.2.2",
"attrs==25.1.0",
"atomicwrites-homeassistant==1.4.1",
"audioop-lts==0.2.1",
@@ -43,46 +44,46 @@ dependencies = [
"certifi>=2021.5.30",
"ciso8601==2.3.2",
"cronsim==2.6",
- "fnv-hash-fast==1.2.2",
+ "fnv-hash-fast==1.4.0",
# hass-nabucasa is imported by helpers which don't depend on the cloud
# integration
- "hass-nabucasa==0.89.0",
+ "hass-nabucasa==0.94.0",
# When bumping httpx, please check the version pins of
# httpcore, anyio, and h11 in gen_requirements_all
"httpx==0.28.1",
"home-assistant-bluetooth==1.13.1",
"ifaddr==0.2.0",
- "Jinja2==3.1.5",
+ "Jinja2==3.1.6",
"lru-dict==1.3.0",
"PyJWT==2.10.1",
# PyJWT has loose dependency. We want the latest one.
- "cryptography==44.0.0",
+ "cryptography==44.0.1",
"Pillow==11.1.0",
- "propcache==0.2.1",
+ "propcache==0.3.0",
"pyOpenSSL==25.0.0",
- "orjson==3.10.12",
+ "orjson==3.10.15",
"packaging>=23.1",
"psutil-home-assistant==0.0.1",
"python-slugify==8.0.4",
"PyYAML==6.0.2",
"requests==2.32.3",
- "securetar==2025.1.4",
- "SQLAlchemy==2.0.38",
+ "securetar==2025.2.1",
+ "SQLAlchemy==2.0.39",
"standard-aifc==3.13.0",
"standard-telnetlib==3.13.0",
"typing-extensions>=4.12.2,<5.0",
- "ulid-transform==1.2.0",
+ "ulid-transform==1.4.0",
# Constrain urllib3 to ensure we deal with CVE-2020-26137 and CVE-2021-33503
# Temporary setting an upper bound, to prevent compat issues with urllib3>=2
# https://github.com/home-assistant/core/issues/97248
"urllib3>=1.26.5,<2",
- "uv==0.5.27",
+ "uv==0.6.8",
"voluptuous==0.15.2",
"voluptuous-serialize==2.6.0",
"voluptuous-openapi==0.0.6",
"yarl==1.18.3",
"webrtc-models==0.3.0",
- "zeroconf==0.143.0"
+ "zeroconf==0.146.0"
]
[project.urls]
@@ -700,17 +701,12 @@ exclude_lines = [
]
[tool.ruff]
-required-version = ">=0.9.1"
+required-version = ">=0.11.0"
[tool.ruff.lint]
select = [
"A001", # Variable {name} is shadowing a Python builtin
- "ASYNC210", # Async functions should not call blocking HTTP methods
- "ASYNC220", # Async functions should not create subprocesses with blocking methods
- "ASYNC221", # Async functions should not run processes with blocking methods
- "ASYNC222", # Async functions should not wait on processes with blocking methods
- "ASYNC230", # Async functions should not open files with blocking methods like open
- "ASYNC251", # Async functions should not call time.sleep
+ "ASYNC", # flake8-async
"B002", # Python does not support the unary prefix increment
"B005", # Using .strip() with multi-character strings is misleading
"B007", # Loop control variable {name} not used within loop body
@@ -789,7 +785,6 @@ select = [
"S317", # suspicious-xml-sax-usage
"S318", # suspicious-xml-mini-dom-usage
"S319", # suspicious-xml-pull-dom-usage
- "S320", # suspicious-xmle-tree-usage
"S601", # paramiko-call
"S602", # subprocess-popen-with-shell-equals-true
"S604", # call-with-shell-equals-true
@@ -810,6 +805,8 @@ select = [
]
ignore = [
+ "ASYNC109", # Async function definition with a `timeout` parameter Use `asyncio.timeout` instead
+ "ASYNC110", # Use `asyncio.Event` instead of awaiting `asyncio.sleep` in a `while` loop
"D202", # No blank lines allowed after function docstring
"D203", # 1 blank line required before class docstring
"D213", # Multi-line docstring summary should start at the second line
@@ -839,6 +836,8 @@ ignore = [
"TC001", # Move application import {} into a type-checking block
"TC002", # Move third-party import {} into a type-checking block
"TC003", # Move standard library import {} into a type-checking block
+ # Quotes for typing.cast generally not necessary, only for performance critical paths
+ "TC006", # Add quotes to type expression in typing.cast()
"TRY003", # Avoid specifying long messages outside the exception class
"TRY400", # Use `logging.exception` instead of `logging.error`
@@ -857,7 +856,7 @@ ignore = [
"COM819",
# Disabled because ruff does not understand type of __all__ generated by a function
- "PLE0605"
+ "PLE0605",
]
[tool.ruff.lint.flake8-import-conventions.extend-aliases]
diff --git a/requirements.txt b/requirements.txt
index f0ff3b8054a..0735e38c89c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,13 +5,14 @@
# Home Assistant Core
aiodns==3.2.0
aiohasupervisor==0.3.0
-aiohttp==3.11.12
+aiohttp==3.11.14
aiohttp_cors==0.7.0
-aiohttp-fast-zlib==0.2.2
-aiohttp-asyncmdnsresolver==0.1.0
+aiohttp-fast-zlib==0.2.3
+aiohttp-asyncmdnsresolver==0.1.1
aiozoneinfo==0.2.3
+annotatedyaml==0.4.4
astral==2.2
-async-interrupt==1.2.1
+async-interrupt==1.2.2
attrs==25.1.0
atomicwrites-homeassistant==1.4.1
audioop-lts==0.2.1
@@ -20,35 +21,35 @@ bcrypt==4.2.0
certifi>=2021.5.30
ciso8601==2.3.2
cronsim==2.6
-fnv-hash-fast==1.2.2
-hass-nabucasa==0.89.0
+fnv-hash-fast==1.4.0
+hass-nabucasa==0.94.0
httpx==0.28.1
home-assistant-bluetooth==1.13.1
ifaddr==0.2.0
-Jinja2==3.1.5
+Jinja2==3.1.6
lru-dict==1.3.0
PyJWT==2.10.1
-cryptography==44.0.0
+cryptography==44.0.1
Pillow==11.1.0
-propcache==0.2.1
+propcache==0.3.0
pyOpenSSL==25.0.0
-orjson==3.10.12
+orjson==3.10.15
packaging>=23.1
psutil-home-assistant==0.0.1
python-slugify==8.0.4
PyYAML==6.0.2
requests==2.32.3
-securetar==2025.1.4
-SQLAlchemy==2.0.38
+securetar==2025.2.1
+SQLAlchemy==2.0.39
standard-aifc==3.13.0
standard-telnetlib==3.13.0
typing-extensions>=4.12.2,<5.0
-ulid-transform==1.2.0
+ulid-transform==1.4.0
urllib3>=1.26.5,<2
-uv==0.5.27
+uv==0.6.8
voluptuous==0.15.2
voluptuous-serialize==2.6.0
voluptuous-openapi==0.0.6
yarl==1.18.3
webrtc-models==0.3.0
-zeroconf==0.143.0
+zeroconf==0.146.0
diff --git a/requirements_all.txt b/requirements_all.txt
index 306e5891c86..d30280144a3 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -19,10 +19,10 @@ DoorBirdPy==3.0.8
HAP-python==4.9.2
# homeassistant.components.tasmota
-HATasmota==0.9.2
+HATasmota==0.10.0
# homeassistant.components.mastodon
-Mastodon.py==1.8.1
+Mastodon.py==2.0.1
# homeassistant.components.doods
# homeassistant.components.generic
@@ -45,7 +45,7 @@ ProgettiHWSW==0.1.3
# PyBluez==0.22
# homeassistant.components.cast
-PyChromecast==14.0.5
+PyChromecast==14.0.6
# homeassistant.components.flick_electric
PyFlick==1.1.3
@@ -54,10 +54,10 @@ PyFlick==1.1.3
PyFlume==0.6.5
# homeassistant.components.fronius
-PyFronius==0.7.3
+PyFronius==0.7.7
# homeassistant.components.pyload
-PyLoadAPI==1.3.2
+PyLoadAPI==1.4.2
# homeassistant.components.met_eireann
PyMetEireann==2024.11.0
@@ -84,7 +84,7 @@ PyQRCode==1.2.1
PyRMVtransport==0.3.3
# homeassistant.components.switchbot
-PySwitchbot==0.56.0
+PySwitchbot==0.57.1
# homeassistant.components.switchmate
PySwitchmate==0.5.1
@@ -100,7 +100,7 @@ PyTransportNSW==0.1.1
PyTurboJPEG==1.7.5
# homeassistant.components.vicare
-PyViCare==2.41.0
+PyViCare==2.43.1
# homeassistant.components.xiaomi_aqara
PyXiaomiGateway==0.14.3
@@ -116,7 +116,7 @@ RtmAPI==0.7.2
# homeassistant.components.recorder
# homeassistant.components.sql
-SQLAlchemy==2.0.38
+SQLAlchemy==2.0.39
# homeassistant.components.tami4
Tami4EdgeAPI==3.0
@@ -131,7 +131,7 @@ TwitterAPI==2.7.12
WSDiscovery==2.1.2
# homeassistant.components.accuweather
-accuweather==4.0.0
+accuweather==4.2.0
# homeassistant.components.adax
adax==0.4.0
@@ -140,7 +140,7 @@ adax==0.4.0
adb-shell[async]==0.4.4
# homeassistant.components.alarmdecoder
-adext==0.4.3
+adext==0.4.4
# homeassistant.components.adguard
adguardhome==0.7.0
@@ -201,7 +201,7 @@ aioaseko==1.0.0
aioasuswrt==1.4.0
# homeassistant.components.husqvarna_automower
-aioautomower==2025.1.1
+aioautomower==2025.3.1
# homeassistant.components.azure_devops
aioazuredevops==2.2.1
@@ -213,13 +213,13 @@ aiobafi6==0.9.0
aiobotocore==2.13.1
# homeassistant.components.comelit
-aiocomelit==0.10.1
+aiocomelit==0.11.2
# homeassistant.components.dhcp
-aiodhcpwatcher==1.1.0
+aiodhcpwatcher==1.1.1
# homeassistant.components.dhcp
-aiodiscover==2.2.2
+aiodiscover==2.6.1
# homeassistant.components.dnsip
aiodns==3.2.0
@@ -234,7 +234,7 @@ aioeafm==0.1.2
aioeagle==1.1.0
# homeassistant.components.ecowitt
-aioecowitt==2024.2.1
+aioecowitt==2025.3.1
# homeassistant.components.co2signal
aioelectricitymaps==0.4.0
@@ -243,7 +243,7 @@ aioelectricitymaps==0.4.0
aioemonitor==1.0.5
# homeassistant.components.esphome
-aioesphomeapi==29.0.0
+aioesphomeapi==29.7.0
# homeassistant.components.flo
aioflo==2021.11.0
@@ -258,22 +258,22 @@ aiogithubapi==24.6.0
aioguardian==2022.07.0
# homeassistant.components.harmony
-aioharmony==0.4.1
+aioharmony==0.5.2
# homeassistant.components.hassio
aiohasupervisor==0.3.0
# homeassistant.components.home_connect
-aiohomeconnect==0.12.3
+aiohomeconnect==0.16.3
# homeassistant.components.homekit_controller
-aiohomekit==3.2.7
+aiohomekit==3.2.8
# homeassistant.components.mcp_server
aiohttp_sse==2.2.0
# homeassistant.components.hue
-aiohue==4.7.3
+aiohue==4.7.4
# homeassistant.components.imap
aioimaplib==2.0.1
@@ -291,7 +291,7 @@ aiolifx-effects==0.3.2
aiolifx-themes==0.6.4
# homeassistant.components.lifx
-aiolifx==1.1.2
+aiolifx==1.1.4
# homeassistant.components.lookin
aiolookin==1.0.0
@@ -371,7 +371,7 @@ aioruuvigateway==0.1.0
aiosenz==1.0.0
# homeassistant.components.shelly
-aioshelly==12.4.1
+aioshelly==13.4.0
# homeassistant.components.skybell
aioskybell==22.7.0
@@ -386,7 +386,7 @@ aiosolaredge==0.2.0
aiosteamist==1.0.1
# homeassistant.components.cambridge_audio
-aiostreammagic==2.10.0
+aiostreammagic==2.11.0
# homeassistant.components.switcher_kis
aioswitcher==6.0.0
@@ -404,7 +404,7 @@ aiotedee==0.2.20
aiotractive==0.6.0
# homeassistant.components.unifi
-aiounifi==81
+aiounifi==83
# homeassistant.components.usb
aiousbwatcher==1.1.1
@@ -421,17 +421,20 @@ aiowaqi==3.1.0
# homeassistant.components.watttime
aiowatttime==0.1.1
+# homeassistant.components.webdav
+aiowebdav2==0.4.2
+
# homeassistant.components.webostv
-aiowebostv==0.6.1
+aiowebostv==0.7.3
# homeassistant.components.withings
-aiowithings==3.1.5
+aiowithings==3.1.6
# homeassistant.components.yandex_transport
aioymaps==1.2.5
# homeassistant.components.airgradient
-airgradient==0.9.1
+airgradient==0.9.2
# homeassistant.components.airly
airly==1.1.0
@@ -461,7 +464,7 @@ amcrest==1.9.8
androidtv[async]==0.0.75
# homeassistant.components.androidtv_remote
-androidtvremote2==0.1.2
+androidtvremote2==0.2.0
# homeassistant.components.anel_pwrctrl
anel-pwrctrl-homeassistant==0.0.1.dev2
@@ -473,10 +476,10 @@ anova-wifi==0.17.0
anthemav==1.4.1
# homeassistant.components.anthropic
-anthropic==0.44.0
+anthropic==0.47.2
# homeassistant.components.mcp_server
-anyio==4.8.0
+anyio==4.9.0
# homeassistant.components.weatherkit
apple_weatherkit==1.1.3
@@ -497,7 +500,7 @@ aqualogic==2.6
aranet4==2.5.1
# homeassistant.components.arcam_fmj
-arcam-fmj==1.5.2
+arcam-fmj==1.8.1
# homeassistant.components.arris_tg2492lg
arris-tg2492lg==2.2.0
@@ -554,7 +557,7 @@ av==13.1.0
axis==64
# homeassistant.components.fujitsu_fglair
-ayla-iot-unofficial==1.4.5
+ayla-iot-unofficial==1.4.7
# homeassistant.components.azure_event_hub
azure-eventhub==5.11.1
@@ -568,6 +571,9 @@ azure-kusto-ingest==4.5.1
# homeassistant.components.azure_service_bus
azure-servicebus==7.10.0
+# homeassistant.components.azure_storage
+azure-storage-blob==12.24.0
+
# homeassistant.components.holiday
babel==2.15.0
@@ -597,10 +603,10 @@ bizkaibus==0.1.1
# homeassistant.components.eq3btsmart
# homeassistant.components.esphome
-bleak-esphome==2.7.1
+bleak-esphome==2.12.0
# homeassistant.components.bluetooth
-bleak-retry-connector==3.8.1
+bleak-retry-connector==3.9.0
# homeassistant.components.bluetooth
bleak==0.22.3
@@ -628,13 +634,13 @@ bluemaestro-ble==0.2.3
bluetooth-adapters==0.21.4
# homeassistant.components.bluetooth
-bluetooth-auto-recovery==1.4.2
+bluetooth-auto-recovery==1.4.5
# homeassistant.components.bluetooth
# homeassistant.components.ld2410_ble
# homeassistant.components.led_ble
# homeassistant.components.private_ble_device
-bluetooth-data-tools==1.23.4
+bluetooth-data-tools==1.26.1
# homeassistant.components.bond
bond-async==0.2.1
@@ -650,7 +656,7 @@ boto3==1.34.131
botocore==1.34.131
# homeassistant.components.bring
-bring-api==1.0.1
+bring-api==1.1.0
# homeassistant.components.broadlink
broadlink==0.19.0
@@ -680,7 +686,7 @@ btsmarthub-devicelist==0.2.3
buienradar==1.0.6
# homeassistant.components.dhcp
-cached-ipaddress==0.8.0
+cached-ipaddress==0.10.0
# homeassistant.components.caldav
caldav==1.3.9
@@ -738,10 +744,10 @@ datadog==0.15.0
datapoint==0.9.9
# homeassistant.components.bluetooth
-dbus-fast==2.33.0
+dbus-fast==2.41.1
# homeassistant.components.debugpy
-debugpy==1.8.11
+debugpy==1.8.13
# homeassistant.components.decora_wifi
# decora-wifi==1.4
@@ -750,7 +756,7 @@ debugpy==1.8.11
# decora==0.6
# homeassistant.components.ecovacs
-deebot-client==12.0.0
+deebot-client==12.3.1
# homeassistant.components.ihc
# homeassistant.components.namecheapdns
@@ -863,7 +869,7 @@ enocean==0.50
enturclient==0.2.4
# homeassistant.components.environment_canada
-env-canada==0.7.2
+env-canada==0.8.0
# homeassistant.components.season
ephem==4.1.6
@@ -893,7 +899,7 @@ eufylife-ble-client==0.1.8
# evdev==1.6.1
# homeassistant.components.evohome
-evohome-async==0.4.21
+evohome-async==1.0.4
# homeassistant.components.bryant_evolution
evolutionhttp==0.0.18
@@ -930,7 +936,7 @@ fixerio==1.0.0a0
fjaraskupan==2.3.2
# homeassistant.components.flexit_bacnet
-flexit_bacnet==2.2.1
+flexit_bacnet==2.2.3
# homeassistant.components.flipr
flipr-api==1.6.1
@@ -940,7 +946,7 @@ flux-led==1.1.3
# homeassistant.components.homekit
# homeassistant.components.recorder
-fnv-hash-fast==1.2.2
+fnv-hash-fast==1.4.0
# homeassistant.components.foobot
foobot_async==1.0.0
@@ -962,13 +968,13 @@ freesms==0.2.0
fritzconnection[qr]==1.14.0
# homeassistant.components.fyta
-fyta_cli==0.7.0
+fyta_cli==0.7.2
# homeassistant.components.google_translate
gTTS==2.5.3
# homeassistant.components.gardena_bluetooth
-gardena-bluetooth==1.5.0
+gardena-bluetooth==1.6.0
# homeassistant.components.google_assistant_sdk
gassist-text==0.0.11
@@ -980,7 +986,7 @@ gcal-sync==7.0.0
geniushub-client==0.7.1
# homeassistant.components.geocaching
-geocachingapi==0.2.1
+geocachingapi==0.3.0
# homeassistant.components.aprs
geopy==2.3.0
@@ -1002,7 +1008,7 @@ georss-qld-bushfire-alert-client==0.8
getmac==0.9.5
# homeassistant.components.gios
-gios==5.0.0
+gios==6.0.0
# homeassistant.components.gitter
gitterpy==0.1.7
@@ -1024,7 +1030,7 @@ goodwe==0.3.6
google-api-python-client==2.71.0
# homeassistant.components.google_pubsub
-google-cloud-pubsub==2.23.0
+google-cloud-pubsub==2.29.0
# homeassistant.components.google_cloud
google-cloud-speech==2.27.0
@@ -1033,10 +1039,10 @@ google-cloud-speech==2.27.0
google-cloud-texttospeech==2.17.2
# homeassistant.components.google_generative_ai_conversation
-google-generativeai==0.8.2
+google-genai==1.1.0
# homeassistant.components.nest
-google-nest-sdm==7.1.3
+google-nest-sdm==7.1.4
# homeassistant.components.google_photos
google-photos-library-api==0.12.1
@@ -1052,10 +1058,10 @@ goslide-api==0.7.0
gotailwind==0.3.0
# homeassistant.components.govee_ble
-govee-ble==0.43.0
+govee-ble==0.43.1
# homeassistant.components.govee_light_local
-govee-local-api==2.0.1
+govee-local-api==2.1.0
# homeassistant.components.remote_rpi_gpio
gpiozero==1.6.2
@@ -1099,14 +1105,17 @@ ha-iotawattpy==0.1.2
# homeassistant.components.philips_js
ha-philipsjs==3.2.2
+# homeassistant.components.homeassistant_hardware
+ha-silabs-firmware-client==0.2.0
+
# homeassistant.components.habitica
-habiticalib==0.3.5
+habiticalib==0.3.7
# homeassistant.components.bluetooth
-habluetooth==3.21.1
+habluetooth==3.37.0
# homeassistant.components.cloud
-hass-nabucasa==0.89.0
+hass-nabucasa==0.94.0
# homeassistant.components.splunk
hass-splunk==0.1.1
@@ -1115,7 +1124,7 @@ hass-splunk==0.1.1
hassil==2.2.3
# homeassistant.components.jewish_calendar
-hdate==0.11.1
+hdate[astral]==1.0.3
# homeassistant.components.heatmiser
heatmiserV3==2.0.3
@@ -1143,13 +1152,13 @@ hole==0.8.0
# homeassistant.components.holiday
# homeassistant.components.workday
-holidays==0.66
+holidays==0.68
# homeassistant.components.frontend
-home-assistant-frontend==20250205.0
+home-assistant-frontend==20250306.0
# homeassistant.components.conversation
-home-assistant-intents==2025.2.5
+home-assistant-intents==2025.3.5
# homeassistant.components.homematicip_cloud
homematicip==1.1.7
@@ -1184,7 +1193,8 @@ ibmiotf==0.3.4
# homeassistant.components.google
# homeassistant.components.local_calendar
# homeassistant.components.local_todo
-ical==8.3.0
+# homeassistant.components.remote_calendar
+ical==9.0.1
# homeassistant.components.caldav
icalendar==6.1.0
@@ -1220,7 +1230,7 @@ influxdb-client==1.24.0
influxdb==5.3.1
# homeassistant.components.inkbird
-inkbird-ble==0.5.8
+inkbird-ble==0.9.0
# homeassistant.components.insteon
insteon-frontend-home-assistant==0.5.0
@@ -1275,7 +1285,7 @@ kiwiki-client==0.1.1
knocki==0.4.2
# homeassistant.components.knx
-knx-frontend==2025.1.30.194235
+knx-frontend==2025.3.8.214559
# homeassistant.components.konnected
konnected==1.2.0
@@ -1417,7 +1427,7 @@ minio==7.1.12
moat-ble==0.1.1
# homeassistant.components.moehlenhoff_alpha2
-moehlenhoff-alpha2==1.3.1
+moehlenhoff-alpha2==1.4.0
# homeassistant.components.monzo
monzopy==1.4.2
@@ -1426,7 +1436,7 @@ monzopy==1.4.2
mopeka-iot-ble==0.8.0
# homeassistant.components.motion_blinds
-motionblinds==0.6.25
+motionblinds==0.6.26
# homeassistant.components.motionblinds_ble
motionblindsble==0.1.3
@@ -1441,7 +1451,7 @@ mozart-api==4.1.1.116.4
mullvad-api==1.0.0
# homeassistant.components.music_assistant
-music-assistant-client==1.0.8
+music-assistant-client==1.1.1
# homeassistant.components.tts
mutagen==1.47.0
@@ -1471,13 +1481,13 @@ netdata==1.3.0
netmap==0.7.0.2
# homeassistant.components.nam
-nettigo-air-monitor==4.0.0
+nettigo-air-monitor==4.1.0
# homeassistant.components.neurio_energy
neurio==0.3.1
# homeassistant.components.nexia
-nexia==2.0.9
+nexia==2.4.0
# homeassistant.components.nextcloud
nextcloudmonitor==1.5.1
@@ -1489,10 +1499,10 @@ nextcord==2.6.0
nextdns==4.0.0
# homeassistant.components.niko_home_control
-nhc==0.4.4
+nhc==0.4.10
# homeassistant.components.nibe_heatpump
-nibe==2.14.0
+nibe==2.17.0
# homeassistant.components.nice_go
nice-go==1.0.1
@@ -1501,7 +1511,7 @@ nice-go==1.0.1
niluclient==0.1.2
# homeassistant.components.noaa_tides
-noaa-coops==0.1.9
+noaa-coops==0.4.0
# homeassistant.components.nfandroidtv
notifications-android-tv==0.1.5
@@ -1547,7 +1557,7 @@ odp-amsterdam==6.0.2
oemthermostat==1.1.1
# homeassistant.components.ohme
-ohme==1.2.8
+ohme==1.4.1
# homeassistant.components.ollama
ollama==0.4.7
@@ -1559,7 +1569,7 @@ omnilogic==0.4.5
ondilo==0.5.0
# homeassistant.components.onedrive
-onedrive-personal-sdk==0.0.8
+onedrive-personal-sdk==0.0.13
# homeassistant.components.onvif
onvif-zeep-async==3.2.5
@@ -1571,7 +1581,7 @@ open-garage==0.2.0
open-meteo==0.3.2
# homeassistant.components.openai_conversation
-openai==1.61.0
+openai==1.68.2
# homeassistant.components.openerz
openerz-api==0.3.0
@@ -1595,7 +1605,7 @@ openwrt-luci-rpc==1.1.17
openwrt-ubus-rpc==0.0.2
# homeassistant.components.opower
-opower==0.8.9
+opower==0.9.0
# homeassistant.components.oralb
oralb-ble==0.17.6
@@ -1663,7 +1673,7 @@ plexauth==0.0.6
plexwebsocket==0.0.14
# homeassistant.components.plugwise
-plugwise==1.6.4
+plugwise==1.7.2
# homeassistant.components.plum_lightpad
plumlightpad==0.0.11
@@ -1749,7 +1759,7 @@ py-schluter==0.1.7
py-sucks==0.9.10
# homeassistant.components.synology_dsm
-py-synologydsm-api==2.6.0
+py-synologydsm-api==2.7.1
# homeassistant.components.atome
pyAtome==0.1.1
@@ -1773,7 +1783,7 @@ pyEmby==1.10
pyHik==0.3.2
# homeassistant.components.homee
-pyHomee==1.2.5
+pyHomee==1.2.8
# homeassistant.components.rfxtrx
pyRFXtrx==0.31.1
@@ -1807,7 +1817,7 @@ pyairnow==1.2.1
pyairvisual==2023.08.1
# homeassistant.components.aprilaire
-pyaprilaire==0.7.7
+pyaprilaire==0.8.1
# homeassistant.components.asuswrt
pyasuswrt==0.1.21
@@ -1825,7 +1835,7 @@ pyatv==0.16.0
pyaussiebb==0.1.5
# homeassistant.components.balboa
-pybalboa==1.0.2
+pybalboa==1.1.3
# homeassistant.components.bbox
pybbox==0.0.5-alpha
@@ -1837,7 +1847,7 @@ pyblackbird==0.6
pyblu==2.0.0
# homeassistant.components.neato
-pybotvac==0.0.25
+pybotvac==0.0.26
# homeassistant.components.braviatv
pybravia==0.3.4
@@ -1876,7 +1886,7 @@ pycsspeechtts==1.0.8
# pycups==2.0.4
# homeassistant.components.daikin
-pydaikin==2.13.8
+pydaikin==2.14.1
# homeassistant.components.danfoss_air
pydanfossair==0.1.0
@@ -1900,7 +1910,7 @@ pydiscovergy==3.0.2
pydoods==1.0.2
# homeassistant.components.hydrawise
-pydrawise==2025.1.0
+pydrawise==2025.3.0
# homeassistant.components.android_ip_webcam
pydroid-ipcam==2.0.0
@@ -1912,7 +1922,7 @@ pyebox==1.1.4
pyecoforest==0.4.0
# homeassistant.components.econet
-pyeconet==0.1.26
+pyeconet==0.1.28
# homeassistant.components.ista_ecotrend
pyecotrend-ista==3.3.1
@@ -1933,7 +1943,7 @@ pyeiscp==0.0.7
pyemoncms==0.1.1
# homeassistant.components.enphase_envoy
-pyenphase==1.23.1
+pyenphase==1.25.1
# homeassistant.components.envisalink
pyenvisalink==4.7
@@ -1951,7 +1961,7 @@ pyevilgenius==2.0.0
pyezviz==0.2.1.2
# homeassistant.components.fibaro
-pyfibaro==0.8.0
+pyfibaro==0.8.2
# homeassistant.components.fido
pyfido==2.1.2
@@ -1972,7 +1982,7 @@ pyforked-daapd==0.1.14
pyfreedompro==1.1.0
# homeassistant.components.fritzbox
-pyfritzhome==0.6.14
+pyfritzhome==0.6.17
# homeassistant.components.ifttt
pyfttt==0.3
@@ -1990,10 +2000,10 @@ pygti==0.9.4
pyhaversion==22.8.0
# homeassistant.components.heos
-pyheos==1.0.1
+pyheos==1.0.4
# homeassistant.components.hive
-pyhive-integration==1.0.1
+pyhive-integration==1.0.2
# homeassistant.components.homematic
pyhomematic==0.1.77
@@ -2014,7 +2024,7 @@ pyinsteon==1.6.3
pyintesishome==1.8.0
# homeassistant.components.ipma
-pyipma==3.0.8
+pyipma==3.0.9
# homeassistant.components.ipp
pyipp==0.17.0
@@ -2071,7 +2081,7 @@ pykwb==0.0.8
pylacrosse==0.4
# homeassistant.components.lamarzocco
-pylamarzocco==1.4.6
+pylamarzocco==1.4.9
# homeassistant.components.lastfm
pylast==5.1.0
@@ -2092,7 +2102,7 @@ pylitejet==0.6.3
pylitterbot==2024.0.0
# homeassistant.components.lutron_caseta
-pylutron-caseta==0.23.0
+pylutron-caseta==0.24.0
# homeassistant.components.lutron
pylutron==0.2.16
@@ -2134,7 +2144,7 @@ pymsteams==0.1.12
pymysensors==0.24.0
# homeassistant.components.iron_os
-pynecil==4.0.1
+pynecil==4.1.0
# homeassistant.components.netgear
pynetgear==0.10.10
@@ -2173,7 +2183,7 @@ pyombi==0.1.10
pyopenuv==2023.02.0
# homeassistant.components.openweathermap
-pyopenweathermap==0.2.1
+pyopenweathermap==0.2.2
# homeassistant.components.opnsense
pyopnsense==0.4.0
@@ -2193,7 +2203,7 @@ pyotgw==2.2.2
pyotp==2.8.0
# homeassistant.components.overkiz
-pyoverkiz==1.16.0
+pyoverkiz==1.16.4
# homeassistant.components.onewire
pyownet==0.10.0.post1
@@ -2207,6 +2217,9 @@ pypca==0.0.7
# homeassistant.components.lcn
pypck==0.8.5
+# homeassistant.components.pglab
+pypglab==0.0.3
+
# homeassistant.components.pjlink
pypjlink2==1.2.1
@@ -2220,7 +2233,7 @@ pypoint==3.0.0
pyprof2calltree==1.4.5
# homeassistant.components.prosegur
-pyprosegur==0.0.9
+pyprosegur==0.0.14
# homeassistant.components.prusalink
pyprusalink==2.1.1
@@ -2235,7 +2248,7 @@ pyqvrpro==0.52
pyqwikswitch==0.93
# homeassistant.components.nmbs
-pyrail==0.0.3
+pyrail==0.4.1
# homeassistant.components.rainbird
pyrainbird==6.0.1
@@ -2247,7 +2260,7 @@ pyrecswitch==1.0.2
pyrepetierng==0.1.0
# homeassistant.components.risco
-pyrisco==0.6.5
+pyrisco==0.6.7
# homeassistant.components.rituals_perfume_genie
pyrituals==0.0.6
@@ -2256,7 +2269,7 @@ pyrituals==0.0.6
pyroute2==0.7.5
# homeassistant.components.rympro
-pyrympro==0.0.8
+pyrympro==0.0.9
# homeassistant.components.sabnzbd
pysabnzbd==1.1.1
@@ -2283,7 +2296,7 @@ pyserial==3.5
pysesame2==1.0.1
# homeassistant.components.seventeentrack
-pyseventeentrack==1.0.1
+pyseventeentrack==1.0.2
# homeassistant.components.sia
pysiaalarm==3.1.1
@@ -2301,13 +2314,10 @@ pysma==0.7.5
pysmappee==0.2.29
# homeassistant.components.smartthings
-pysmartapp==0.3.5
-
-# homeassistant.components.smartthings
-pysmartthings==0.7.8
+pysmartthings==2.7.4
# homeassistant.components.smarty
-pysmarty2==0.10.1
+pysmarty2==0.10.2
# homeassistant.components.smhi
pysmhi==1.0.0
@@ -2334,13 +2344,13 @@ pyspcwebgw==0.7.0
pyspeex-noise==1.0.2
# homeassistant.components.squeezebox
-pysqueezebox==0.11.1
+pysqueezebox==0.12.0
# homeassistant.components.stiebel_eltron
pystiebeleltron==0.0.1.dev2
# homeassistant.components.suez_water
-pysuezV2==2.0.3
+pysuezV2==2.0.4
# homeassistant.components.switchbee
pyswitchbee==1.8.3
@@ -2391,7 +2401,7 @@ python-gc100==1.0.3a0
python-gitlab==1.6.0
# homeassistant.components.google_drive
-python-google-drive-api==0.0.2
+python-google-drive-api==0.1.0
# homeassistant.components.analytics_insights
python-homeassistant-analytics==0.9.0
@@ -2412,10 +2422,10 @@ python-join-api==0.0.9
python-juicenet==1.1.0
# homeassistant.components.tplink
-python-kasa[speedups]==0.10.1
+python-kasa[speedups]==0.10.2
# homeassistant.components.linkplay
-python-linkplay==0.1.3
+python-linkplay==0.2.0
# homeassistant.components.lirc
# python-lirc==1.2.3
@@ -2443,10 +2453,10 @@ python-opensky==1.0.1
python-otbr-api==2.7.0
# homeassistant.components.overseerr
-python-overseerr==0.7.0
+python-overseerr==0.7.1
# homeassistant.components.picnic
-python-picnic-api==1.1.0
+python-picnic-api2==1.2.2
# homeassistant.components.rabbitair
python-rabbitair==0.0.8
@@ -2455,10 +2465,13 @@ python-rabbitair==0.0.8
python-ripple-api==0.0.3
# homeassistant.components.roborock
-python-roborock==2.11.1
+python-roborock==2.16.1
# homeassistant.components.smarttub
-python-smarttub==0.0.38
+python-smarttub==0.0.39
+
+# homeassistant.components.snoo
+python-snoo==0.6.4
# homeassistant.components.songpal
python-songpal==0.16.2
@@ -2467,7 +2480,7 @@ python-songpal==0.16.2
python-tado==0.18.6
# homeassistant.components.technove
-python-technove==1.3.1
+python-technove==2.0.0
# homeassistant.components.telegram_bot
python-telegram-bot[socks]==21.5
@@ -2485,7 +2498,7 @@ pytile==2024.12.0
pytomorrowio==0.3.6
# homeassistant.components.touchline
-pytouchline==0.7
+pytouchline_extended==0.4.5
# homeassistant.components.touchline_sl
pytouchlinesl==0.3.0
@@ -2519,7 +2532,7 @@ pyvera==0.3.15
pyversasense==0.0.6
# homeassistant.components.vesync
-pyvesync==2.1.17
+pyvesync==2.1.18
# homeassistant.components.vizio
pyvizio==0.1.61
@@ -2570,7 +2583,7 @@ pyzerproc==0.4.8
qbittorrent-api==2024.2.59
# homeassistant.components.qbus
-qbusmqttapi==1.2.4
+qbusmqttapi==1.3.0
# homeassistant.components.qingping
qingping-ble==0.10.0
@@ -2609,7 +2622,7 @@ renault-api==0.2.9
renson-endura-delta==1.7.2
# homeassistant.components.reolink
-reolink-aio==0.11.10
+reolink-aio==0.12.3
# homeassistant.components.idteck_prox
rfk101py==0.0.1
@@ -2678,14 +2691,14 @@ screenlogicpy==0.10.0
scsgate==0.1.0
# homeassistant.components.backup
-securetar==2025.1.4
+securetar==2025.2.1
# homeassistant.components.sendgrid
sendgrid==6.8.2
# homeassistant.components.emulated_kasa
# homeassistant.components.sense
-sense-energy==0.13.4
+sense-energy==0.13.7
# homeassistant.components.sensirion_ble
sensirion-ble==0.1.1
@@ -2693,14 +2706,20 @@ sensirion-ble==0.1.1
# homeassistant.components.sensorpro
sensorpro-ble==0.5.3
+# homeassistant.components.sensorpush_cloud
+sensorpush-api==2.1.1
+
# homeassistant.components.sensorpush
sensorpush-ble==1.7.1
+# homeassistant.components.sensorpush_cloud
+sensorpush-ha==1.3.2
+
# homeassistant.components.sensoterra
sensoterra==2.0.1
# homeassistant.components.sentry
-sentry-sdk==1.40.3
+sentry-sdk==1.45.1
# homeassistant.components.sfr_box
sfrbox-api==0.0.11
@@ -2745,7 +2764,7 @@ smart-meter-texas==0.5.5
snapcast==2.3.6
# homeassistant.components.sonos
-soco==0.30.8
+soco==0.30.9
# homeassistant.components.solaredge_local
solaredge-local==0.2.3
@@ -2793,7 +2812,7 @@ statsd==3.2.1
steamodd==4.21
# homeassistant.components.stookwijzer
-stookwijzer==1.5.1
+stookwijzer==1.6.1
# homeassistant.components.streamlabswater
streamlabswater==1.0.1
@@ -2857,7 +2876,7 @@ temperusb==1.6.1
# homeassistant.components.tesla_fleet
# homeassistant.components.teslemetry
# homeassistant.components.tessie
-tesla-fleet-api==0.9.8
+tesla-fleet-api==0.9.13
# homeassistant.components.powerwall
tesla-powerwall==0.5.2
@@ -2866,7 +2885,7 @@ tesla-powerwall==0.5.2
tesla-wall-connector==1.0.2
# homeassistant.components.teslemetry
-teslemetry-stream==0.6.10
+teslemetry-stream==0.6.12
# homeassistant.components.tessie
tessie-api==0.1.1
@@ -2875,7 +2894,7 @@ tessie-api==0.1.1
# tf-models-official==2.5.0
# homeassistant.components.thermobeacon
-thermobeacon-ble==0.7.0
+thermobeacon-ble==0.8.1
# homeassistant.components.thermopro
thermopro-ble==0.11.0
@@ -2884,7 +2903,7 @@ thermopro-ble==0.11.0
thingspeak==1.0.0
# homeassistant.components.lg_thinq
-thinqconnect==1.0.4
+thinqconnect==1.0.5
# homeassistant.components.tikteck
tikteck==0.4
@@ -2911,7 +2930,7 @@ total-connect-client==2025.1.4
tp-connected==0.0.4
# homeassistant.components.tplink_omada
-tplink-omada-client==1.4.3
+tplink-omada-client==1.4.4
# homeassistant.components.transmission
transmission-rpc==7.0.3
@@ -2959,10 +2978,10 @@ unifi_ap==0.0.2
unifiled==0.11
# homeassistant.components.homeassistant_hardware
-universal-silabs-flasher==0.0.25
+universal-silabs-flasher==0.0.30
# homeassistant.components.upb
-upb-lib==0.6.0
+upb-lib==0.6.1
# homeassistant.components.upcloud
upcloud-api==2.6.0
@@ -2985,7 +3004,7 @@ vallox-websocket-api==5.3.0
vehicle==2.2.2
# homeassistant.components.velbus
-velbus-aio==2025.1.1
+velbus-aio==2025.3.1
# homeassistant.components.venstar
venstarcolortouch==0.19
@@ -3019,7 +3038,7 @@ vultr==0.1.2
wakeonlan==2.1.0
# homeassistant.components.wallbox
-wallbox==0.7.0
+wallbox==0.8.0
# homeassistant.components.folder_watcher
watchdog==6.0.0
@@ -3031,7 +3050,7 @@ waterfurnace==1.1.0
watergate-local-api==2024.4.1
# homeassistant.components.weatherflow_cloud
-weatherflow4py==1.0.6
+weatherflow4py==1.3.1
# homeassistant.components.cisco_webex_teams
webexpythonsdk==2.0.1
@@ -3043,10 +3062,10 @@ webio-api==0.1.11
webmin-xmlrpc==0.0.2
# homeassistant.components.weheat
-weheat==2025.1.15
+weheat==2025.2.26
# homeassistant.components.whirlpool
-whirlpool-sixth-sense==0.18.12
+whirlpool-sixth-sense==0.19.1
# homeassistant.components.whois
whois==0.9.27
@@ -3061,7 +3080,7 @@ wirelesstagpy==0.8.1
wled==0.21.0
# homeassistant.components.wolflink
-wolf-comm==0.0.15
+wolf-comm==0.0.23
# homeassistant.components.wyoming
wyoming==1.5.4
@@ -3073,10 +3092,10 @@ xbox-webapi==2.1.0
xiaomi-ble==0.33.0
# homeassistant.components.knx
-xknx==3.5.0
+xknx==3.6.0
# homeassistant.components.knx
-xknxproject==3.8.1
+xknxproject==3.8.2
# homeassistant.components.fritz
# homeassistant.components.rest
@@ -3107,7 +3126,7 @@ yeelight==0.7.16
yeelightsunflower==0.0.10
# homeassistant.components.yolink
-yolink-api==0.4.7
+yolink-api==0.4.8
# homeassistant.components.youless
youless-api==2.2.0
@@ -3116,7 +3135,7 @@ youless-api==2.2.0
youtubeaio==1.1.5
# homeassistant.components.media_extractor
-yt-dlp[default]==2025.01.26
+yt-dlp[default]==2025.02.19
# homeassistant.components.zabbix
zabbix-utils==2.0.2
@@ -3128,13 +3147,13 @@ zamg==0.3.6
zengge==0.2
# homeassistant.components.zeroconf
-zeroconf==0.143.0
+zeroconf==0.146.0
# homeassistant.components.zeversolar
zeversolar==0.3.2
# homeassistant.components.zha
-zha==0.0.48
+zha==0.0.53
# homeassistant.components.zhong_hong
zhong-hong-hvac==1.0.13
@@ -3146,7 +3165,7 @@ ziggo-mediabox-xl==1.1.0
zm-py==0.5.4
# homeassistant.components.zwave_js
-zwave-js-server-python==0.60.0
+zwave-js-server-python==0.62.0
# homeassistant.components.zwave_me
zwave-me-ws==0.4.3
diff --git a/requirements_test.txt b/requirements_test.txt
index 2731114043b..baf72265c40 100644
--- a/requirements_test.txt
+++ b/requirements_test.txt
@@ -7,17 +7,17 @@
-c homeassistant/package_constraints.txt
-r requirements_test_pre_commit.txt
-astroid==3.3.8
-coverage==7.6.10
+astroid==3.3.9
+coverage==7.6.12
freezegun==1.5.1
license-expression==30.4.1
mock-open==1.4.0
-mypy-dev==1.16.0a2
+mypy-dev==1.16.0a5
pre-commit==4.0.0
pydantic==2.10.6
-pylint==3.3.4
+pylint==3.3.6
pylint-per-file-ignores==1.4.0
-pipdeptree==2.25.0
+pipdeptree==2.25.1
pytest-asyncio==0.25.3
pytest-aiohttp==1.1.0
pytest-cov==6.0.0
@@ -29,7 +29,7 @@ pytest-timeout==2.3.1
pytest-unordered==0.6.1
pytest-picked==0.5.1
pytest-xdist==3.6.1
-pytest==8.3.4
+pytest==8.3.5
requests-mock==1.12.1
respx==0.22.0
syrupy==4.8.1
diff --git a/requirements_test_all.txt b/requirements_test_all.txt
index efe7b9d5418..5384d917e15 100644
--- a/requirements_test_all.txt
+++ b/requirements_test_all.txt
@@ -19,10 +19,10 @@ DoorBirdPy==3.0.8
HAP-python==4.9.2
# homeassistant.components.tasmota
-HATasmota==0.9.2
+HATasmota==0.10.0
# homeassistant.components.mastodon
-Mastodon.py==1.8.1
+Mastodon.py==2.0.1
# homeassistant.components.doods
# homeassistant.components.generic
@@ -42,7 +42,7 @@ PlexAPI==4.15.16
ProgettiHWSW==0.1.3
# homeassistant.components.cast
-PyChromecast==14.0.5
+PyChromecast==14.0.6
# homeassistant.components.flick_electric
PyFlick==1.1.3
@@ -51,10 +51,10 @@ PyFlick==1.1.3
PyFlume==0.6.5
# homeassistant.components.fronius
-PyFronius==0.7.3
+PyFronius==0.7.7
# homeassistant.components.pyload
-PyLoadAPI==1.3.2
+PyLoadAPI==1.4.2
# homeassistant.components.met_eireann
PyMetEireann==2024.11.0
@@ -81,7 +81,7 @@ PyQRCode==1.2.1
PyRMVtransport==0.3.3
# homeassistant.components.switchbot
-PySwitchbot==0.56.0
+PySwitchbot==0.57.1
# homeassistant.components.syncthru
PySyncThru==0.8.0
@@ -94,7 +94,7 @@ PyTransportNSW==0.1.1
PyTurboJPEG==1.7.5
# homeassistant.components.vicare
-PyViCare==2.41.0
+PyViCare==2.43.1
# homeassistant.components.xiaomi_aqara
PyXiaomiGateway==0.14.3
@@ -110,7 +110,7 @@ RtmAPI==0.7.2
# homeassistant.components.recorder
# homeassistant.components.sql
-SQLAlchemy==2.0.38
+SQLAlchemy==2.0.39
# homeassistant.components.tami4
Tami4EdgeAPI==3.0
@@ -119,7 +119,7 @@ Tami4EdgeAPI==3.0
WSDiscovery==2.1.2
# homeassistant.components.accuweather
-accuweather==4.0.0
+accuweather==4.2.0
# homeassistant.components.adax
adax==0.4.0
@@ -128,7 +128,7 @@ adax==0.4.0
adb-shell[async]==0.4.4
# homeassistant.components.alarmdecoder
-adext==0.4.3
+adext==0.4.4
# homeassistant.components.adguard
adguardhome==0.7.0
@@ -189,7 +189,7 @@ aioaseko==1.0.0
aioasuswrt==1.4.0
# homeassistant.components.husqvarna_automower
-aioautomower==2025.1.1
+aioautomower==2025.3.1
# homeassistant.components.azure_devops
aioazuredevops==2.2.1
@@ -201,13 +201,13 @@ aiobafi6==0.9.0
aiobotocore==2.13.1
# homeassistant.components.comelit
-aiocomelit==0.10.1
+aiocomelit==0.11.2
# homeassistant.components.dhcp
-aiodhcpwatcher==1.1.0
+aiodhcpwatcher==1.1.1
# homeassistant.components.dhcp
-aiodiscover==2.2.2
+aiodiscover==2.6.1
# homeassistant.components.dnsip
aiodns==3.2.0
@@ -222,7 +222,7 @@ aioeafm==0.1.2
aioeagle==1.1.0
# homeassistant.components.ecowitt
-aioecowitt==2024.2.1
+aioecowitt==2025.3.1
# homeassistant.components.co2signal
aioelectricitymaps==0.4.0
@@ -231,7 +231,7 @@ aioelectricitymaps==0.4.0
aioemonitor==1.0.5
# homeassistant.components.esphome
-aioesphomeapi==29.0.0
+aioesphomeapi==29.7.0
# homeassistant.components.flo
aioflo==2021.11.0
@@ -243,22 +243,22 @@ aiogithubapi==24.6.0
aioguardian==2022.07.0
# homeassistant.components.harmony
-aioharmony==0.4.1
+aioharmony==0.5.2
# homeassistant.components.hassio
aiohasupervisor==0.3.0
# homeassistant.components.home_connect
-aiohomeconnect==0.12.3
+aiohomeconnect==0.16.3
# homeassistant.components.homekit_controller
-aiohomekit==3.2.7
+aiohomekit==3.2.8
# homeassistant.components.mcp_server
aiohttp_sse==2.2.0
# homeassistant.components.hue
-aiohue==4.7.3
+aiohue==4.7.4
# homeassistant.components.imap
aioimaplib==2.0.1
@@ -273,7 +273,7 @@ aiolifx-effects==0.3.2
aiolifx-themes==0.6.4
# homeassistant.components.lifx
-aiolifx==1.1.2
+aiolifx==1.1.4
# homeassistant.components.lookin
aiolookin==1.0.0
@@ -353,7 +353,7 @@ aioruuvigateway==0.1.0
aiosenz==1.0.0
# homeassistant.components.shelly
-aioshelly==12.4.1
+aioshelly==13.4.0
# homeassistant.components.skybell
aioskybell==22.7.0
@@ -368,7 +368,7 @@ aiosolaredge==0.2.0
aiosteamist==1.0.1
# homeassistant.components.cambridge_audio
-aiostreammagic==2.10.0
+aiostreammagic==2.11.0
# homeassistant.components.switcher_kis
aioswitcher==6.0.0
@@ -386,7 +386,7 @@ aiotedee==0.2.20
aiotractive==0.6.0
# homeassistant.components.unifi
-aiounifi==81
+aiounifi==83
# homeassistant.components.usb
aiousbwatcher==1.1.1
@@ -403,17 +403,20 @@ aiowaqi==3.1.0
# homeassistant.components.watttime
aiowatttime==0.1.1
+# homeassistant.components.webdav
+aiowebdav2==0.4.2
+
# homeassistant.components.webostv
-aiowebostv==0.6.1
+aiowebostv==0.7.3
# homeassistant.components.withings
-aiowithings==3.1.5
+aiowithings==3.1.6
# homeassistant.components.yandex_transport
aioymaps==1.2.5
# homeassistant.components.airgradient
-airgradient==0.9.1
+airgradient==0.9.2
# homeassistant.components.airly
airly==1.1.0
@@ -437,7 +440,7 @@ amberelectric==2.0.12
androidtv[async]==0.0.75
# homeassistant.components.androidtv_remote
-androidtvremote2==0.1.2
+androidtvremote2==0.2.0
# homeassistant.components.anova
anova-wifi==0.17.0
@@ -446,10 +449,10 @@ anova-wifi==0.17.0
anthemav==1.4.1
# homeassistant.components.anthropic
-anthropic==0.44.0
+anthropic==0.47.2
# homeassistant.components.mcp_server
-anyio==4.8.0
+anyio==4.9.0
# homeassistant.components.weatherkit
apple_weatherkit==1.1.3
@@ -467,7 +470,7 @@ apsystems-ez1==2.4.0
aranet4==2.5.1
# homeassistant.components.arcam_fmj
-arcam-fmj==1.5.2
+arcam-fmj==1.8.1
# homeassistant.components.dlna_dmr
# homeassistant.components.dlna_dms
@@ -503,7 +506,7 @@ av==13.1.0
axis==64
# homeassistant.components.fujitsu_fglair
-ayla-iot-unofficial==1.4.5
+ayla-iot-unofficial==1.4.7
# homeassistant.components.azure_event_hub
azure-eventhub==5.11.1
@@ -514,6 +517,9 @@ azure-kusto-data[aio]==4.5.1
# homeassistant.components.azure_data_explorer
azure-kusto-ingest==4.5.1
+# homeassistant.components.azure_storage
+azure-storage-blob==12.24.0
+
# homeassistant.components.holiday
babel==2.15.0
@@ -528,10 +534,10 @@ bimmer-connected[china]==0.17.2
# homeassistant.components.eq3btsmart
# homeassistant.components.esphome
-bleak-esphome==2.7.1
+bleak-esphome==2.12.0
# homeassistant.components.bluetooth
-bleak-retry-connector==3.8.1
+bleak-retry-connector==3.9.0
# homeassistant.components.bluetooth
bleak==0.22.3
@@ -552,13 +558,13 @@ bluemaestro-ble==0.2.3
bluetooth-adapters==0.21.4
# homeassistant.components.bluetooth
-bluetooth-auto-recovery==1.4.2
+bluetooth-auto-recovery==1.4.5
# homeassistant.components.bluetooth
# homeassistant.components.ld2410_ble
# homeassistant.components.led_ble
# homeassistant.components.private_ble_device
-bluetooth-data-tools==1.23.4
+bluetooth-data-tools==1.26.1
# homeassistant.components.bond
bond-async==0.2.1
@@ -570,7 +576,7 @@ boschshcpy==0.2.91
botocore==1.34.131
# homeassistant.components.bring
-bring-api==1.0.1
+bring-api==1.1.0
# homeassistant.components.broadlink
broadlink==0.19.0
@@ -591,7 +597,7 @@ bthome-ble==3.12.4
buienradar==1.0.6
# homeassistant.components.dhcp
-cached-ipaddress==0.8.0
+cached-ipaddress==0.10.0
# homeassistant.components.caldav
caldav==1.3.9
@@ -634,13 +640,13 @@ datadog==0.15.0
datapoint==0.9.9
# homeassistant.components.bluetooth
-dbus-fast==2.33.0
+dbus-fast==2.41.1
# homeassistant.components.debugpy
-debugpy==1.8.11
+debugpy==1.8.13
# homeassistant.components.ecovacs
-deebot-client==12.0.0
+deebot-client==12.3.1
# homeassistant.components.ihc
# homeassistant.components.namecheapdns
@@ -732,7 +738,7 @@ energyzero==2.1.1
enocean==0.50
# homeassistant.components.environment_canada
-env-canada==0.7.2
+env-canada==0.8.0
# homeassistant.components.season
ephem==4.1.6
@@ -759,7 +765,7 @@ eternalegypt==0.0.16
eufylife-ble-client==0.1.8
# homeassistant.components.evohome
-evohome-async==0.4.21
+evohome-async==1.0.4
# homeassistant.components.bryant_evolution
evolutionhttp==0.0.18
@@ -789,7 +795,7 @@ fivem-api==0.1.2
fjaraskupan==2.3.2
# homeassistant.components.flexit_bacnet
-flexit_bacnet==2.2.1
+flexit_bacnet==2.2.3
# homeassistant.components.flipr
flipr-api==1.6.1
@@ -799,7 +805,7 @@ flux-led==1.1.3
# homeassistant.components.homekit
# homeassistant.components.recorder
-fnv-hash-fast==1.2.2
+fnv-hash-fast==1.4.0
# homeassistant.components.foobot
foobot_async==1.0.0
@@ -815,13 +821,13 @@ freebox-api==1.2.2
fritzconnection[qr]==1.14.0
# homeassistant.components.fyta
-fyta_cli==0.7.0
+fyta_cli==0.7.2
# homeassistant.components.google_translate
gTTS==2.5.3
# homeassistant.components.gardena_bluetooth
-gardena-bluetooth==1.5.0
+gardena-bluetooth==1.6.0
# homeassistant.components.google_assistant_sdk
gassist-text==0.0.11
@@ -833,7 +839,7 @@ gcal-sync==7.0.0
geniushub-client==0.7.1
# homeassistant.components.geocaching
-geocachingapi==0.2.1
+geocachingapi==0.3.0
# homeassistant.components.aprs
geopy==2.3.0
@@ -855,7 +861,7 @@ georss-qld-bushfire-alert-client==0.8
getmac==0.9.5
# homeassistant.components.gios
-gios==5.0.0
+gios==6.0.0
# homeassistant.components.glances
glances-api==0.8.0
@@ -874,7 +880,7 @@ goodwe==0.3.6
google-api-python-client==2.71.0
# homeassistant.components.google_pubsub
-google-cloud-pubsub==2.23.0
+google-cloud-pubsub==2.29.0
# homeassistant.components.google_cloud
google-cloud-speech==2.27.0
@@ -883,10 +889,10 @@ google-cloud-speech==2.27.0
google-cloud-texttospeech==2.17.2
# homeassistant.components.google_generative_ai_conversation
-google-generativeai==0.8.2
+google-genai==1.1.0
# homeassistant.components.nest
-google-nest-sdm==7.1.3
+google-nest-sdm==7.1.4
# homeassistant.components.google_photos
google-photos-library-api==0.12.1
@@ -902,10 +908,10 @@ goslide-api==0.7.0
gotailwind==0.3.0
# homeassistant.components.govee_ble
-govee-ble==0.43.0
+govee-ble==0.43.1
# homeassistant.components.govee_light_local
-govee-local-api==2.0.1
+govee-local-api==2.1.0
# homeassistant.components.gpsd
gps3==0.33.3
@@ -941,19 +947,19 @@ ha-iotawattpy==0.1.2
ha-philipsjs==3.2.2
# homeassistant.components.habitica
-habiticalib==0.3.5
+habiticalib==0.3.7
# homeassistant.components.bluetooth
-habluetooth==3.21.1
+habluetooth==3.37.0
# homeassistant.components.cloud
-hass-nabucasa==0.89.0
+hass-nabucasa==0.94.0
# homeassistant.components.conversation
hassil==2.2.3
# homeassistant.components.jewish_calendar
-hdate==0.11.1
+hdate[astral]==1.0.3
# homeassistant.components.here_travel_time
here-routing==1.0.1
@@ -972,13 +978,13 @@ hole==0.8.0
# homeassistant.components.holiday
# homeassistant.components.workday
-holidays==0.66
+holidays==0.68
# homeassistant.components.frontend
-home-assistant-frontend==20250205.0
+home-assistant-frontend==20250306.0
# homeassistant.components.conversation
-home-assistant-intents==2025.2.5
+home-assistant-intents==2025.3.5
# homeassistant.components.homematicip_cloud
homematicip==1.1.7
@@ -1004,7 +1010,8 @@ ibeacon-ble==1.2.0
# homeassistant.components.google
# homeassistant.components.local_calendar
# homeassistant.components.local_todo
-ical==8.3.0
+# homeassistant.components.remote_calendar
+ical==9.0.1
# homeassistant.components.caldav
icalendar==6.1.0
@@ -1034,7 +1041,7 @@ influxdb-client==1.24.0
influxdb==5.3.1
# homeassistant.components.inkbird
-inkbird-ble==0.5.8
+inkbird-ble==0.9.0
# homeassistant.components.insteon
insteon-frontend-home-assistant==0.5.0
@@ -1077,7 +1084,7 @@ kegtron-ble==0.4.0
knocki==0.4.2
# homeassistant.components.knx
-knx-frontend==2025.1.30.194235
+knx-frontend==2025.3.8.214559
# homeassistant.components.konnected
konnected==1.2.0
@@ -1189,7 +1196,7 @@ minio==7.1.12
moat-ble==0.1.1
# homeassistant.components.moehlenhoff_alpha2
-moehlenhoff-alpha2==1.3.1
+moehlenhoff-alpha2==1.4.0
# homeassistant.components.monzo
monzopy==1.4.2
@@ -1198,7 +1205,7 @@ monzopy==1.4.2
mopeka-iot-ble==0.8.0
# homeassistant.components.motion_blinds
-motionblinds==0.6.25
+motionblinds==0.6.26
# homeassistant.components.motionblinds_ble
motionblindsble==0.1.3
@@ -1213,7 +1220,7 @@ mozart-api==4.1.1.116.4
mullvad-api==1.0.0
# homeassistant.components.music_assistant
-music-assistant-client==1.0.8
+music-assistant-client==1.1.1
# homeassistant.components.tts
mutagen==1.47.0
@@ -1237,10 +1244,10 @@ nessclient==1.1.2
netmap==0.7.0.2
# homeassistant.components.nam
-nettigo-air-monitor==4.0.0
+nettigo-air-monitor==4.1.0
# homeassistant.components.nexia
-nexia==2.0.9
+nexia==2.4.0
# homeassistant.components.nextcloud
nextcloudmonitor==1.5.1
@@ -1252,10 +1259,10 @@ nextcord==2.6.0
nextdns==4.0.0
# homeassistant.components.niko_home_control
-nhc==0.4.4
+nhc==0.4.10
# homeassistant.components.nibe_heatpump
-nibe==2.14.0
+nibe==2.17.0
# homeassistant.components.nice_go
nice-go==1.0.1
@@ -1295,7 +1302,7 @@ objgraph==3.5.0
odp-amsterdam==6.0.2
# homeassistant.components.ohme
-ohme==1.2.8
+ohme==1.4.1
# homeassistant.components.ollama
ollama==0.4.7
@@ -1307,7 +1314,7 @@ omnilogic==0.4.5
ondilo==0.5.0
# homeassistant.components.onedrive
-onedrive-personal-sdk==0.0.8
+onedrive-personal-sdk==0.0.13
# homeassistant.components.onvif
onvif-zeep-async==3.2.5
@@ -1319,7 +1326,7 @@ open-garage==0.2.0
open-meteo==0.3.2
# homeassistant.components.openai_conversation
-openai==1.61.0
+openai==1.68.2
# homeassistant.components.openerz
openerz-api==0.3.0
@@ -1331,7 +1338,7 @@ openhomedevice==2.2.0
openwebifpy==4.3.1
# homeassistant.components.opower
-opower==0.8.9
+opower==0.9.0
# homeassistant.components.oralb
oralb-ble==0.17.6
@@ -1376,7 +1383,7 @@ plexauth==0.0.6
plexwebsocket==0.0.14
# homeassistant.components.plugwise
-plugwise==1.6.4
+plugwise==1.7.2
# homeassistant.components.plum_lightpad
plumlightpad==0.0.11
@@ -1447,7 +1454,7 @@ py-nightscout==1.2.2
py-sucks==0.9.10
# homeassistant.components.synology_dsm
-py-synologydsm-api==2.6.0
+py-synologydsm-api==2.7.1
# homeassistant.components.hdmi_cec
pyCEC==0.5.2
@@ -1462,7 +1469,7 @@ pyDuotecno==2024.10.1
pyElectra==1.2.4
# homeassistant.components.homee
-pyHomee==1.2.5
+pyHomee==1.2.8
# homeassistant.components.rfxtrx
pyRFXtrx==0.31.1
@@ -1487,7 +1494,7 @@ pyairnow==1.2.1
pyairvisual==2023.08.1
# homeassistant.components.aprilaire
-pyaprilaire==0.7.7
+pyaprilaire==0.8.1
# homeassistant.components.asuswrt
pyasuswrt==0.1.21
@@ -1505,7 +1512,7 @@ pyatv==0.16.0
pyaussiebb==0.1.5
# homeassistant.components.balboa
-pybalboa==1.0.2
+pybalboa==1.1.3
# homeassistant.components.blackbird
pyblackbird==0.6
@@ -1514,7 +1521,7 @@ pyblackbird==0.6
pyblu==2.0.0
# homeassistant.components.neato
-pybotvac==0.0.25
+pybotvac==0.0.26
# homeassistant.components.braviatv
pybravia==0.3.4
@@ -1535,7 +1542,7 @@ pycountry==24.6.1
pycsspeechtts==1.0.8
# homeassistant.components.daikin
-pydaikin==2.13.8
+pydaikin==2.14.1
# homeassistant.components.deako
pydeako==0.6.0
@@ -1550,7 +1557,7 @@ pydexcom==0.2.3
pydiscovergy==3.0.2
# homeassistant.components.hydrawise
-pydrawise==2025.1.0
+pydrawise==2025.3.0
# homeassistant.components.android_ip_webcam
pydroid-ipcam==2.0.0
@@ -1559,7 +1566,7 @@ pydroid-ipcam==2.0.0
pyecoforest==0.4.0
# homeassistant.components.econet
-pyeconet==0.1.26
+pyeconet==0.1.28
# homeassistant.components.ista_ecotrend
pyecotrend-ista==3.3.1
@@ -1577,7 +1584,7 @@ pyeiscp==0.0.7
pyemoncms==0.1.1
# homeassistant.components.enphase_envoy
-pyenphase==1.23.1
+pyenphase==1.25.1
# homeassistant.components.everlights
pyeverlights==0.1.0
@@ -1589,7 +1596,7 @@ pyevilgenius==2.0.0
pyezviz==0.2.1.2
# homeassistant.components.fibaro
-pyfibaro==0.8.0
+pyfibaro==0.8.2
# homeassistant.components.fido
pyfido==2.1.2
@@ -1607,7 +1614,7 @@ pyforked-daapd==0.1.14
pyfreedompro==1.1.0
# homeassistant.components.fritzbox
-pyfritzhome==0.6.14
+pyfritzhome==0.6.17
# homeassistant.components.ifttt
pyfttt==0.3
@@ -1619,10 +1626,10 @@ pygti==0.9.4
pyhaversion==22.8.0
# homeassistant.components.heos
-pyheos==1.0.1
+pyheos==1.0.4
# homeassistant.components.hive
-pyhive-integration==1.0.1
+pyhive-integration==1.0.2
# homeassistant.components.homematic
pyhomematic==0.1.77
@@ -1640,7 +1647,7 @@ pyicloud==1.0.0
pyinsteon==1.6.3
# homeassistant.components.ipma
-pyipma==3.0.8
+pyipma==3.0.9
# homeassistant.components.ipp
pyipp==0.17.0
@@ -1685,7 +1692,7 @@ pykrakenapi==0.1.8
pykulersky==0.5.2
# homeassistant.components.lamarzocco
-pylamarzocco==1.4.6
+pylamarzocco==1.4.9
# homeassistant.components.lastfm
pylast==5.1.0
@@ -1706,7 +1713,7 @@ pylitejet==0.6.3
pylitterbot==2024.0.0
# homeassistant.components.lutron_caseta
-pylutron-caseta==0.23.0
+pylutron-caseta==0.24.0
# homeassistant.components.lutron
pylutron==0.2.16
@@ -1739,7 +1746,7 @@ pymonoprice==0.4
pymysensors==0.24.0
# homeassistant.components.iron_os
-pynecil==4.0.1
+pynecil==4.1.0
# homeassistant.components.netgear
pynetgear==0.10.10
@@ -1772,7 +1779,7 @@ pyoctoprintapi==0.1.12
pyopenuv==2023.02.0
# homeassistant.components.openweathermap
-pyopenweathermap==0.2.1
+pyopenweathermap==0.2.2
# homeassistant.components.opnsense
pyopnsense==0.4.0
@@ -1789,7 +1796,7 @@ pyotgw==2.2.2
pyotp==2.8.0
# homeassistant.components.overkiz
-pyoverkiz==1.16.0
+pyoverkiz==1.16.4
# homeassistant.components.onewire
pyownet==0.10.0.post1
@@ -1800,6 +1807,9 @@ pypalazzetti==0.1.19
# homeassistant.components.lcn
pypck==0.8.5
+# homeassistant.components.pglab
+pypglab==0.0.3
+
# homeassistant.components.pjlink
pypjlink2==1.2.1
@@ -1813,7 +1823,7 @@ pypoint==3.0.0
pyprof2calltree==1.4.5
# homeassistant.components.prosegur
-pyprosegur==0.0.9
+pyprosegur==0.0.14
# homeassistant.components.prusalink
pyprusalink==2.1.1
@@ -1825,13 +1835,13 @@ pyps4-2ndscreen==1.3.1
pyqwikswitch==0.93
# homeassistant.components.nmbs
-pyrail==0.0.3
+pyrail==0.4.1
# homeassistant.components.rainbird
pyrainbird==6.0.1
# homeassistant.components.risco
-pyrisco==0.6.5
+pyrisco==0.6.7
# homeassistant.components.rituals_perfume_genie
pyrituals==0.0.6
@@ -1840,7 +1850,7 @@ pyrituals==0.0.6
pyroute2==0.7.5
# homeassistant.components.rympro
-pyrympro==0.0.8
+pyrympro==0.0.9
# homeassistant.components.sabnzbd
pysabnzbd==1.1.1
@@ -1858,7 +1868,7 @@ pysensibo==1.1.0
pyserial==3.5
# homeassistant.components.seventeentrack
-pyseventeentrack==1.0.1
+pyseventeentrack==1.0.2
# homeassistant.components.sia
pysiaalarm==3.1.1
@@ -1873,13 +1883,10 @@ pysma==0.7.5
pysmappee==0.2.29
# homeassistant.components.smartthings
-pysmartapp==0.3.5
-
-# homeassistant.components.smartthings
-pysmartthings==0.7.8
+pysmartthings==2.7.4
# homeassistant.components.smarty
-pysmarty2==0.10.1
+pysmarty2==0.10.2
# homeassistant.components.smhi
pysmhi==1.0.0
@@ -1906,10 +1913,10 @@ pyspcwebgw==0.7.0
pyspeex-noise==1.0.2
# homeassistant.components.squeezebox
-pysqueezebox==0.11.1
+pysqueezebox==0.12.0
# homeassistant.components.suez_water
-pysuezV2==2.0.3
+pysuezV2==2.0.4
# homeassistant.components.switchbee
pyswitchbee==1.8.3
@@ -1936,7 +1943,7 @@ python-fullykiosk==0.0.14
# python-gammu==3.2.4
# homeassistant.components.google_drive
-python-google-drive-api==0.0.2
+python-google-drive-api==0.1.0
# homeassistant.components.analytics_insights
python-homeassistant-analytics==0.9.0
@@ -1951,10 +1958,10 @@ python-izone==1.2.9
python-juicenet==1.1.0
# homeassistant.components.tplink
-python-kasa[speedups]==0.10.1
+python-kasa[speedups]==0.10.2
# homeassistant.components.linkplay
-python-linkplay==0.1.3
+python-linkplay==0.2.0
# homeassistant.components.matter
python-matter-server==7.0.0
@@ -1979,19 +1986,22 @@ python-opensky==1.0.1
python-otbr-api==2.7.0
# homeassistant.components.overseerr
-python-overseerr==0.7.0
+python-overseerr==0.7.1
# homeassistant.components.picnic
-python-picnic-api==1.1.0
+python-picnic-api2==1.2.2
# homeassistant.components.rabbitair
python-rabbitair==0.0.8
# homeassistant.components.roborock
-python-roborock==2.11.1
+python-roborock==2.16.1
# homeassistant.components.smarttub
-python-smarttub==0.0.38
+python-smarttub==0.0.39
+
+# homeassistant.components.snoo
+python-snoo==0.6.4
# homeassistant.components.songpal
python-songpal==0.16.2
@@ -2000,7 +2010,7 @@ python-songpal==0.16.2
python-tado==0.18.6
# homeassistant.components.technove
-python-technove==1.3.1
+python-technove==2.0.0
# homeassistant.components.telegram_bot
python-telegram-bot[socks]==21.5
@@ -2037,7 +2047,7 @@ pyuptimerobot==22.2.0
pyvera==0.3.15
# homeassistant.components.vesync
-pyvesync==2.1.17
+pyvesync==2.1.18
# homeassistant.components.vizio
pyvizio==0.1.61
@@ -2082,7 +2092,7 @@ pyzerproc==0.4.8
qbittorrent-api==2024.2.59
# homeassistant.components.qbus
-qbusmqttapi==1.2.4
+qbusmqttapi==1.3.0
# homeassistant.components.qingping
qingping-ble==0.10.0
@@ -2112,7 +2122,7 @@ renault-api==0.2.9
renson-endura-delta==1.7.2
# homeassistant.components.reolink
-reolink-aio==0.11.10
+reolink-aio==0.12.3
# homeassistant.components.rflink
rflink==0.0.66
@@ -2160,11 +2170,11 @@ sanix==1.0.6
screenlogicpy==0.10.0
# homeassistant.components.backup
-securetar==2025.1.4
+securetar==2025.2.1
# homeassistant.components.emulated_kasa
# homeassistant.components.sense
-sense-energy==0.13.4
+sense-energy==0.13.7
# homeassistant.components.sensirion_ble
sensirion-ble==0.1.1
@@ -2172,14 +2182,20 @@ sensirion-ble==0.1.1
# homeassistant.components.sensorpro
sensorpro-ble==0.5.3
+# homeassistant.components.sensorpush_cloud
+sensorpush-api==2.1.1
+
# homeassistant.components.sensorpush
sensorpush-ble==1.7.1
+# homeassistant.components.sensorpush_cloud
+sensorpush-ha==1.3.2
+
# homeassistant.components.sensoterra
sensoterra==2.0.1
# homeassistant.components.sentry
-sentry-sdk==1.40.3
+sentry-sdk==1.45.1
# homeassistant.components.sfr_box
sfrbox-api==0.0.11
@@ -2212,7 +2228,7 @@ smart-meter-texas==0.5.5
snapcast==2.3.6
# homeassistant.components.sonos
-soco==0.30.8
+soco==0.30.9
# homeassistant.components.solarlog
solarlog_cli==0.4.0
@@ -2254,7 +2270,7 @@ statsd==3.2.1
steamodd==4.21
# homeassistant.components.stookwijzer
-stookwijzer==1.5.1
+stookwijzer==1.6.1
# homeassistant.components.streamlabswater
streamlabswater==1.0.1
@@ -2297,7 +2313,7 @@ temperusb==1.6.1
# homeassistant.components.tesla_fleet
# homeassistant.components.teslemetry
# homeassistant.components.tessie
-tesla-fleet-api==0.9.8
+tesla-fleet-api==0.9.13
# homeassistant.components.powerwall
tesla-powerwall==0.5.2
@@ -2306,19 +2322,19 @@ tesla-powerwall==0.5.2
tesla-wall-connector==1.0.2
# homeassistant.components.teslemetry
-teslemetry-stream==0.6.10
+teslemetry-stream==0.6.12
# homeassistant.components.tessie
tessie-api==0.1.1
# homeassistant.components.thermobeacon
-thermobeacon-ble==0.7.0
+thermobeacon-ble==0.8.1
# homeassistant.components.thermopro
thermopro-ble==0.11.0
# homeassistant.components.lg_thinq
-thinqconnect==1.0.4
+thinqconnect==1.0.5
# homeassistant.components.tilt_ble
tilt-ble==0.2.3
@@ -2336,7 +2352,7 @@ toonapi==0.3.0
total-connect-client==2025.1.4
# homeassistant.components.tplink_omada
-tplink-omada-client==1.4.3
+tplink-omada-client==1.4.4
# homeassistant.components.transmission
transmission-rpc==7.0.3
@@ -2378,7 +2394,7 @@ ultraheat-api==0.5.7
unifi-discovery==1.2.0
# homeassistant.components.upb
-upb-lib==0.6.0
+upb-lib==0.6.1
# homeassistant.components.upcloud
upcloud-api==2.6.0
@@ -2401,7 +2417,7 @@ vallox-websocket-api==5.3.0
vehicle==2.2.2
# homeassistant.components.velbus
-velbus-aio==2025.1.1
+velbus-aio==2025.3.1
# homeassistant.components.venstar
venstarcolortouch==0.19
@@ -2429,7 +2445,7 @@ vultr==0.1.2
wakeonlan==2.1.0
# homeassistant.components.wallbox
-wallbox==0.7.0
+wallbox==0.8.0
# homeassistant.components.folder_watcher
watchdog==6.0.0
@@ -2438,7 +2454,7 @@ watchdog==6.0.0
watergate-local-api==2024.4.1
# homeassistant.components.weatherflow_cloud
-weatherflow4py==1.0.6
+weatherflow4py==1.3.1
# homeassistant.components.nasweb
webio-api==0.1.11
@@ -2447,10 +2463,10 @@ webio-api==0.1.11
webmin-xmlrpc==0.0.2
# homeassistant.components.weheat
-weheat==2025.1.15
+weheat==2025.2.26
# homeassistant.components.whirlpool
-whirlpool-sixth-sense==0.18.12
+whirlpool-sixth-sense==0.19.1
# homeassistant.components.whois
whois==0.9.27
@@ -2462,7 +2478,7 @@ wiffi==1.1.2
wled==0.21.0
# homeassistant.components.wolflink
-wolf-comm==0.0.15
+wolf-comm==0.0.23
# homeassistant.components.wyoming
wyoming==1.5.4
@@ -2474,10 +2490,10 @@ xbox-webapi==2.1.0
xiaomi-ble==0.33.0
# homeassistant.components.knx
-xknx==3.5.0
+xknx==3.6.0
# homeassistant.components.knx
-xknxproject==3.8.1
+xknxproject==3.8.2
# homeassistant.components.fritz
# homeassistant.components.rest
@@ -2502,7 +2518,7 @@ yalexs==8.10.0
yeelight==0.7.16
# homeassistant.components.yolink
-yolink-api==0.4.7
+yolink-api==0.4.8
# homeassistant.components.youless
youless-api==2.2.0
@@ -2511,22 +2527,22 @@ youless-api==2.2.0
youtubeaio==1.1.5
# homeassistant.components.media_extractor
-yt-dlp[default]==2025.01.26
+yt-dlp[default]==2025.02.19
# homeassistant.components.zamg
zamg==0.3.6
# homeassistant.components.zeroconf
-zeroconf==0.143.0
+zeroconf==0.146.0
# homeassistant.components.zeversolar
zeversolar==0.3.2
# homeassistant.components.zha
-zha==0.0.48
+zha==0.0.53
# homeassistant.components.zwave_js
-zwave-js-server-python==0.60.0
+zwave-js-server-python==0.62.0
# homeassistant.components.zwave_me
zwave-me-ws==0.4.3
diff --git a/requirements_test_pre_commit.txt b/requirements_test_pre_commit.txt
index 1cf3d91defa..ff86915bbf3 100644
--- a/requirements_test_pre_commit.txt
+++ b/requirements_test_pre_commit.txt
@@ -1,5 +1,5 @@
# Automatically generated from .pre-commit-config.yaml by gen_requirements_all.py, do not edit
codespell==2.4.1
-ruff==0.9.1
+ruff==0.11.0
yamllint==1.35.1
diff --git a/script/gen_requirements_all.py b/script/gen_requirements_all.py
index 25602918240..aa719d7f8eb 100755
--- a/script/gen_requirements_all.py
+++ b/script/gen_requirements_all.py
@@ -139,7 +139,7 @@ uuid==1000000000.0.0
# these requirements are quite loose. As the entire stack has some outstanding issues, and
# even newer versions seem to introduce new issues, it's useful for us to pin all these
# requirements so we can directly link HA versions to these library versions.
-anyio==4.8.0
+anyio==4.9.0
h11==0.14.0
httpcore==1.0.7
diff --git a/script/hassfest/__main__.py b/script/hassfest/__main__.py
index c93d8fd4499..277696c669b 100644
--- a/script/hassfest/__main__.py
+++ b/script/hassfest/__main__.py
@@ -107,7 +107,13 @@ def get_config() -> Config:
"--plugins",
type=validate_plugins,
default=ALL_PLUGIN_NAMES,
- help="Comma-separate list of plugins to run. Valid plugin names: %(default)s",
+ help="Comma-separated list of plugins to run. Valid plugin names: %(default)s",
+ )
+ parser.add_argument(
+ "--skip-plugins",
+ type=validate_plugins,
+ default=[],
+ help=f"Comma-separated list of plugins to skip. Valid plugin names: {ALL_PLUGIN_NAMES}",
)
parser.add_argument(
"--core-path",
@@ -131,6 +137,9 @@ def get_config() -> Config:
):
raise RuntimeError("Run from Home Assistant root")
+ if parsed.skip_plugins:
+ parsed.plugins = set(parsed.plugins) - set(parsed.skip_plugins)
+
return Config(
root=parsed.core_path.absolute(),
specific_integrations=parsed.integration_path,
diff --git a/script/hassfest/dependencies.py b/script/hassfest/dependencies.py
index d29571eaa83..b22027500dd 100644
--- a/script/hassfest/dependencies.py
+++ b/script/hassfest/dependencies.py
@@ -153,8 +153,6 @@ ALLOWED_USED_COMPONENTS = {
}
IGNORE_VIOLATIONS = {
- # Has same requirement, gets defaults.
- ("sql", "recorder"),
# Sharing a base class
("lutron_caseta", "lutron"),
("ffmpeg_noise", "ffmpeg_motion"),
@@ -175,6 +173,10 @@ IGNORE_VIOLATIONS = {
"logbook",
# Temporary needed for migration until 2024.10
("conversation", "assist_pipeline"),
+ # The onboarding integration provides a limited backup API used during
+ # onboarding. The onboarding integration waits for the backup manager
+ # to be ready before calling any backup functionality.
+ ("onboarding", "backup"),
}
diff --git a/script/hassfest/docker.py b/script/hassfest/docker.py
index edc47e2f9d7..4bf6c3bb0a6 100644
--- a/script/hassfest/docker.py
+++ b/script/hassfest/docker.py
@@ -26,6 +26,24 @@ ENV \
ARG QEMU_CPU
+# Home Assistant S6-Overlay
+COPY rootfs /
+
+# Needs to be redefined inside the FROM statement to be set for RUN commands
+ARG BUILD_ARCH
+# Get go2rtc binary
+RUN \
+ case "${{BUILD_ARCH}}" in \
+ "aarch64") go2rtc_suffix='arm64' ;; \
+ "armhf") go2rtc_suffix='armv6' ;; \
+ "armv7") go2rtc_suffix='arm' ;; \
+ *) go2rtc_suffix=${{BUILD_ARCH}} ;; \
+ esac \
+ && curl -L https://github.com/AlexxIT/go2rtc/releases/download/v{go2rtc}/go2rtc_linux_${{go2rtc_suffix}} --output /bin/go2rtc \
+ && chmod +x /bin/go2rtc \
+ # Verify go2rtc can be executed
+ && go2rtc --version
+
# Install uv
RUN pip3 install uv=={uv}
@@ -56,24 +74,6 @@ RUN \
&& python3 -m compileall \
homeassistant/homeassistant
-# Home Assistant S6-Overlay
-COPY rootfs /
-
-# Needs to be redefined inside the FROM statement to be set for RUN commands
-ARG BUILD_ARCH
-# Get go2rtc binary
-RUN \
- case "${{BUILD_ARCH}}" in \
- "aarch64") go2rtc_suffix='arm64' ;; \
- "armhf") go2rtc_suffix='armv6' ;; \
- "armv7") go2rtc_suffix='arm' ;; \
- *) go2rtc_suffix=${{BUILD_ARCH}} ;; \
- esac \
- && curl -L https://github.com/AlexxIT/go2rtc/releases/download/v{go2rtc}/go2rtc_linux_${{go2rtc_suffix}} --output /bin/go2rtc \
- && chmod +x /bin/go2rtc \
- # Verify go2rtc can be executed
- && go2rtc --version
-
WORKDIR /config
"""
diff --git a/script/hassfest/docker/Dockerfile b/script/hassfest/docker/Dockerfile
index 5598c839257..79716b6fec3 100644
--- a/script/hassfest/docker/Dockerfile
+++ b/script/hassfest/docker/Dockerfile
@@ -14,7 +14,7 @@ WORKDIR "/github/workspace"
COPY . /usr/src/homeassistant
# Uv is only needed during build
-RUN --mount=from=ghcr.io/astral-sh/uv:0.5.27,source=/uv,target=/bin/uv \
+RUN --mount=from=ghcr.io/astral-sh/uv:0.6.8,source=/uv,target=/bin/uv \
# Uv creates a lock file in /tmp
--mount=type=tmpfs,target=/tmp \
# Required for PyTurboJPEG
@@ -24,8 +24,8 @@ RUN --mount=from=ghcr.io/astral-sh/uv:0.5.27,source=/uv,target=/bin/uv \
--no-cache \
-c /usr/src/homeassistant/homeassistant/package_constraints.txt \
-r /usr/src/homeassistant/requirements.txt \
- stdlib-list==0.10.0 pipdeptree==2.25.0 tqdm==4.67.1 ruff==0.9.1 \
- PyTurboJPEG==1.7.5 go2rtc-client==0.1.2 ha-ffmpeg==3.2.2 hassil==2.2.3 home-assistant-intents==2025.2.5 mutagen==1.47.0 pymicro-vad==1.0.1 pyspeex-noise==1.0.2
+ stdlib-list==0.10.0 pipdeptree==2.25.1 tqdm==4.67.1 ruff==0.11.0 \
+ PyTurboJPEG==1.7.5 go2rtc-client==0.1.2 ha-ffmpeg==3.2.2 hassil==2.2.3 home-assistant-intents==2025.3.5 mutagen==1.47.0 pymicro-vad==1.0.1 pyspeex-noise==1.0.2
LABEL "name"="hassfest"
LABEL "maintainer"="Home Assistant "
diff --git a/script/hassfest/manifest.py b/script/hassfest/manifest.py
index 6e9cd8bdedc..02c96930bf5 100644
--- a/script/hassfest/manifest.py
+++ b/script/hassfest/manifest.py
@@ -19,6 +19,7 @@ from voluptuous.humanize import humanize_error
from homeassistant.const import Platform
from homeassistant.helpers import config_validation as cv
+from script.util import sort_manifest as util_sort_manifest
from .model import Config, Integration, ScaledQualityScaleTiers
@@ -376,20 +377,20 @@ def validate_manifest(integration: Integration, core_components_dir: Path) -> No
validate_version(integration)
-_SORT_KEYS = {"domain": ".domain", "name": ".name"}
-
-
-def _sort_manifest_keys(key: str) -> str:
- return _SORT_KEYS.get(key, key)
-
-
def sort_manifest(integration: Integration, config: Config) -> bool:
"""Sort manifest."""
- keys = list(integration.manifest.keys())
- if (keys_sorted := sorted(keys, key=_sort_manifest_keys)) != keys:
- manifest = {key: integration.manifest[key] for key in keys_sorted}
+ if integration.manifest_path is None:
+ integration.add_error(
+ "manifest",
+ "Manifest path not set, unable to sort manifest keys",
+ )
+ return False
+
+ if util_sort_manifest(integration.manifest):
if config.action == "generate":
- integration.manifest_path.write_text(json.dumps(manifest, indent=2))
+ integration.manifest_path.write_text(
+ json.dumps(integration.manifest, indent=2) + "\n"
+ )
text = "have been sorted"
else:
text = "are not sorted correctly"
diff --git a/script/hassfest/model.py b/script/hassfest/model.py
index 08ded687096..1ca4178d9c2 100644
--- a/script/hassfest/model.py
+++ b/script/hassfest/model.py
@@ -157,8 +157,10 @@ class Integration:
@property
def core(self) -> bool:
"""Core integration."""
- return self.path.as_posix().startswith(
- self._config.core_integrations_path.as_posix()
+ return (
+ self.path.absolute()
+ .as_posix()
+ .startswith(self._config.core_integrations_path.as_posix())
)
@property
diff --git a/script/hassfest/quality_scale.py b/script/hassfest/quality_scale.py
index a1ad52e6aa8..d74011801d5 100644
--- a/script/hassfest/quality_scale.py
+++ b/script/hassfest/quality_scale.py
@@ -335,7 +335,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"egardia",
"eight_sleep",
"electrasmart",
- "electric_kiwi",
"eliqonline",
"elkm1",
"elmax",
@@ -391,7 +390,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"fjaraskupan",
"fleetgo",
"flexit",
- "flexit_bacnet",
"flic",
"flick_electric",
"flipr",
@@ -517,7 +515,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"ihc",
"imgw_pib",
"improv_ble",
- "incomfort",
"influxdb",
"inkbird",
"insteon",
@@ -669,7 +666,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"motion_blinds",
"motionblinds_ble",
"motioneye",
- "motionmount",
"mpd",
"mqtt_eventstream",
"mqtt_json",
@@ -815,7 +811,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"pushsafer",
"pvoutput",
"pvpc_hourly_pricing",
- "pyload",
"qbittorrent",
"qingping",
"qld_bushfire",
@@ -859,7 +854,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"risco",
"rituals_perfume_genie",
"rmvtransport",
- "roborock",
"rocketchat",
"roku",
"romy",
@@ -926,7 +920,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"sma",
"smappee",
"smart_meter_texas",
- "smartthings",
"smarttub",
"smarty",
"smhi",
@@ -1093,7 +1086,6 @@ INTEGRATIONS_WITHOUT_QUALITY_SCALE_FILE = [
"vizio",
"vlc",
"vlc_telnet",
- "vodafone_station",
"voicerss",
"voip",
"volkszaehler",
@@ -1287,7 +1279,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"brottsplatskartan",
"browser",
"brunt",
- "bring",
"bryant_evolution",
"bsblan",
"bt_home_hub_5",
@@ -1397,7 +1388,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"egardia",
"eight_sleep",
"electrasmart",
- "electric_kiwi",
"elevenlabs",
"eliqonline",
"elkm1",
@@ -1457,7 +1447,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"fjaraskupan",
"fleetgo",
"flexit",
- "flexit_bacnet",
"flic",
"flick_electric",
"flipr",
@@ -1537,7 +1526,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"gstreamer",
"gtfs",
"guardian",
- "habitica",
"harman_kardon_avr",
"harmony",
"hassio",
@@ -1588,7 +1576,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"imap",
"imgw_pib",
"improv_ble",
- "incomfort",
"influxdb",
"inkbird",
"insteon",
@@ -1596,7 +1583,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"intellifire",
"intesishome",
"ios",
- "iron_os",
"iotawatt",
"iotty",
"iperf3",
@@ -1729,7 +1715,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"mikrotik",
"mill",
"min_max",
- "minecraft_server",
"minio",
"mjpeg",
"moat",
@@ -1748,7 +1733,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"motion_blinds",
"motionblinds_ble",
"motioneye",
- "motionmount",
"mpd",
"mqtt_eventstream",
"mqtt_json",
@@ -1899,7 +1883,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"pushsafer",
"pvoutput",
"pvpc_hourly_pricing",
- "pyload",
"qbittorrent",
"qingping",
"qld_bushfire",
@@ -1974,7 +1957,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"season",
"sendgrid",
"sense",
- "sensibo",
"sensirion_ble",
"sensorpro",
"sensorpush",
@@ -2013,7 +1995,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"sma",
"smappee",
"smart_meter_texas",
- "smartthings",
"smarttub",
"smarty",
"smhi",
@@ -2175,7 +2156,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"velux",
"venstar",
"vera",
- "velbus",
"verisure",
"versasense",
"version",
@@ -2187,7 +2167,6 @@ INTEGRATIONS_WITHOUT_SCALE = [
"vizio",
"vlc",
"vlc_telnet",
- "vodafone_station",
"voicerss",
"voip",
"volkszaehler",
diff --git a/script/hassfest/translations.py b/script/hassfest/translations.py
index b3d397dbd55..f4c05f504ca 100644
--- a/script/hassfest/translations.py
+++ b/script/hassfest/translations.py
@@ -29,6 +29,7 @@ ALLOW_NAME_TRANSLATION = {
"cert_expiry",
"cpuspeed",
"emulated_roku",
+ "energenie_power_sockets",
"faa_delays",
"garages_amsterdam",
"generic",
@@ -40,6 +41,7 @@ ALLOW_NAME_TRANSLATION = {
"local_ip",
"local_todo",
"nmap_tracker",
+ "remote_calendar",
"rpi_power",
"swiss_public_transport",
"waze_travel_time",
@@ -185,6 +187,8 @@ def gen_data_entry_schema(
vol.Optional("abort"): {str: translation_value_validator},
vol.Optional("progress"): {str: translation_value_validator},
vol.Optional("create_entry"): {str: translation_value_validator},
+ vol.Optional("initiate_flow"): {str: translation_value_validator},
+ vol.Optional("entry_type"): translation_value_validator,
}
if flow_title == REQUIRED:
schema[vol.Required("title")] = translation_value_validator
@@ -285,6 +289,15 @@ def gen_strings_schema(config: Config, integration: Integration) -> vol.Schema:
"user" if integration.integration_type == "helper" else None
),
),
+ vol.Optional("config_subentries"): cv.schema_with_slug_keys(
+ gen_data_entry_schema(
+ config=config,
+ integration=integration,
+ flow_title=REMOVED,
+ require_step_title=False,
+ ),
+ slug_validator=vol.Any("_", cv.slug),
+ ),
vol.Optional("options"): gen_data_entry_schema(
config=config,
integration=integration,
diff --git a/script/licenses.py b/script/licenses.py
index aa15a58f3bd..448e9dd2a67 100644
--- a/script/licenses.py
+++ b/script/licenses.py
@@ -180,7 +180,6 @@ EXCEPTIONS = {
"PyMicroBot", # https://github.com/spycle/pyMicroBot/pull/3
"PySwitchmate", # https://github.com/Danielhiversen/pySwitchmate/pull/16
"PyXiaomiGateway", # https://github.com/Danielhiversen/PyXiaomiGateway/pull/201
- "aioecowitt", # https://github.com/home-assistant-libs/aioecowitt/pull/180
"chacha20poly1305", # LGPL
"commentjson", # https://github.com/vaidik/commentjson/pull/55
"crownstone-cloud", # https://github.com/crownstone/crownstone-lib-python-cloud/pull/5
diff --git a/script/scaffold/__main__.py b/script/scaffold/__main__.py
index 93c787df50f..243ea9507f7 100644
--- a/script/scaffold/__main__.py
+++ b/script/scaffold/__main__.py
@@ -8,6 +8,7 @@ import sys
from script.util import valid_integration
from . import docs, error, gather_info, generate
+from .model import Info
TEMPLATES = [
p.name for p in (Path(__file__).parent / "templates").glob("*") if p.is_dir()
@@ -28,6 +29,40 @@ def get_arguments() -> argparse.Namespace:
return parser.parse_args()
+def run_process(name: str, cmd: list[str], info: Info) -> None:
+ """Run a sub process and handle the result.
+
+ :param name: The name of the sub process used in reporting.
+ :param cmd: The sub process arguments.
+ :param info: The Info object.
+ :raises subprocess.CalledProcessError: If the subprocess failed.
+
+ If the sub process was successful print a success message, otherwise
+ print an error message and raise a subprocess.CalledProcessError.
+ """
+ print(f"Command: {' '.join(cmd)}")
+ print()
+ result: subprocess.CompletedProcess = subprocess.run(cmd, check=False)
+ if result.returncode == 0:
+ print()
+ print(f"Completed {name} successfully.")
+ print()
+ return
+
+ print()
+ print(f"Fatal Error: {name} failed with exit code {result.returncode}")
+ print()
+ if info.is_new:
+ print("This is a bug, please report an issue!")
+ else:
+ print(
+ "This may be an existing issue with your integration,",
+ "if so fix and run `script.scaffold` again,",
+ "otherwise please report an issue.",
+ )
+ result.check_returncode()
+
+
def main() -> int:
"""Scaffold an integration."""
if not Path("requirements_all.txt").is_file():
@@ -64,20 +99,32 @@ def main() -> int:
if args.template != "integration":
generate.generate(args.template, info)
- pipe_null = {} if args.develop else {"stdout": subprocess.DEVNULL}
-
+ # Always output sub commands as the output will contain useful information if a command fails.
print("Running hassfest to pick up new information.")
- subprocess.run(["python", "-m", "script.hassfest"], **pipe_null, check=True)
- print()
+ run_process(
+ "hassfest",
+ [
+ "python",
+ "-m",
+ "script.hassfest",
+ "--integration-path",
+ str(info.integration_dir),
+ "--skip-plugins",
+ "quality_scale", # Skip quality scale as it will fail for newly generated integrations.
+ ],
+ info,
+ )
print("Running gen_requirements_all to pick up new information.")
- subprocess.run(
- ["python", "-m", "script.gen_requirements_all"], **pipe_null, check=True
+ run_process(
+ "gen_requirements_all",
+ ["python", "-m", "script.gen_requirements_all"],
+ info,
)
- print()
- print("Running script/translations_develop to pick up new translation strings.")
- subprocess.run(
+ print("Running translations to pick up new translation strings.")
+ run_process(
+ "translations",
[
"python",
"-m",
@@ -86,15 +133,13 @@ def main() -> int:
"--integration",
info.domain,
],
- **pipe_null,
- check=True,
+ info,
)
- print()
if args.develop:
print("Running tests")
- print(f"$ python3 -b -m pytest -vvv tests/components/{info.domain}")
- subprocess.run(
+ run_process(
+ "pytest",
[
"python3",
"-b",
@@ -103,9 +148,8 @@ def main() -> int:
"-vvv",
f"tests/components/{info.domain}",
],
- check=True,
+ info,
)
- print()
docs.print_relevant_docs(args.template, info)
@@ -115,6 +159,8 @@ def main() -> int:
if __name__ == "__main__":
try:
sys.exit(main())
+ except subprocess.CalledProcessError as err:
+ sys.exit(err.returncode)
except error.ExitApp as err:
print()
print(f"Fatal Error: {err.reason}")
diff --git a/script/scaffold/model.py b/script/scaffold/model.py
index 3b5a5e50fe4..e3a7be210ab 100644
--- a/script/scaffold/model.py
+++ b/script/scaffold/model.py
@@ -4,9 +4,12 @@ from __future__ import annotations
import json
from pathlib import Path
+from typing import Any
import attr
+from script.util import sort_manifest
+
from .const import COMPONENT_DIR, TESTS_DIR
@@ -44,16 +47,19 @@ class Info:
"""Path to the manifest."""
return COMPONENT_DIR / self.domain / "manifest.json"
- def manifest(self) -> dict:
+ def manifest(self) -> dict[str, Any]:
"""Return integration manifest."""
return json.loads(self.manifest_path.read_text())
def update_manifest(self, **kwargs) -> None:
"""Update the integration manifest."""
print(f"Updating {self.domain} manifest: {kwargs}")
- self.manifest_path.write_text(
- json.dumps({**self.manifest(), **kwargs}, indent=2) + "\n"
- )
+
+ # Sort keys in manifest so we don't trigger hassfest errors.
+ manifest: dict[str, Any] = {**self.manifest(), **kwargs}
+ sort_manifest(manifest)
+
+ self.manifest_path.write_text(json.dumps(manifest, indent=2) + "\n")
@property
def strings_path(self) -> Path:
diff --git a/script/scaffold/templates/config_flow_helper/integration/sensor.py b/script/scaffold/templates/config_flow_helper/integration/sensor.py
index 741b2e85eb2..9c00dd568eb 100644
--- a/script/scaffold/templates/config_flow_helper/integration/sensor.py
+++ b/script/scaffold/templates/config_flow_helper/integration/sensor.py
@@ -7,13 +7,13 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ENTITY_ID
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Initialize NEW_NAME config entry."""
registry = er.async_get(hass)
diff --git a/script/scaffold/templates/integration/integration/manifest.json b/script/scaffold/templates/integration/integration/manifest.json
index 7235500391d..15bc84a9b5e 100644
--- a/script/scaffold/templates/integration/integration/manifest.json
+++ b/script/scaffold/templates/integration/integration/manifest.json
@@ -7,6 +7,7 @@
"documentation": "https://www.home-assistant.io/integrations/NEW_DOMAIN",
"homekit": {},
"iot_class": "IOT_CLASS",
+ "quality_scale": "bronze",
"requirements": [],
"ssdp": [],
"zeroconf": []
diff --git a/script/util.py b/script/util.py
index b7c37c72102..c9fada38c80 100644
--- a/script/util.py
+++ b/script/util.py
@@ -1,6 +1,7 @@
"""Utility functions for the scaffold script."""
import argparse
+from typing import Any
from .const import COMPONENT_DIR
@@ -13,3 +14,23 @@ def valid_integration(integration):
)
return integration
+
+
+_MANIFEST_SORT_KEYS = {"domain": ".domain", "name": ".name"}
+
+
+def _sort_manifest_keys(key: str) -> str:
+ """Sort manifest keys."""
+ return _MANIFEST_SORT_KEYS.get(key, key)
+
+
+def sort_manifest(manifest: dict[str, Any]) -> bool:
+ """Sort manifest."""
+ keys = list(manifest)
+ if (keys_sorted := sorted(keys, key=_sort_manifest_keys)) != keys:
+ sorted_manifest = {key: manifest[key] for key in keys_sorted}
+ manifest.clear()
+ manifest.update(sorted_manifest)
+ return True
+
+ return False
diff --git a/tests/common.py b/tests/common.py
index 0315ee6d845..f426d2aebd2 100644
--- a/tests/common.py
+++ b/tests/common.py
@@ -29,6 +29,7 @@ from typing import Any, Literal, NoReturn
from unittest.mock import AsyncMock, Mock, patch
from aiohttp.test_utils import unused_port as get_test_instance_port # noqa: F401
+from annotatedyaml import load_yaml_dict, loader as yaml_loader
import pytest
from syrupy import SnapshotAssertion
import voluptuous as vol
@@ -86,7 +87,10 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_send,
)
from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity_platform import (
+ AddConfigEntryEntitiesCallback,
+ AddEntitiesCallback,
+)
from homeassistant.helpers.json import JSONEncoder, _orjson_default_encoder, json_dumps
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util, ulid as ulid_util
@@ -106,7 +110,6 @@ from homeassistant.util.json import (
)
from homeassistant.util.signal_type import SignalType
from homeassistant.util.unit_system import METRIC_SYSTEM
-from homeassistant.util.yaml import load_yaml_dict, loader as yaml_loader
from .testing_config.custom_components.test_constant_deprecation import (
import_deprecated_constant,
@@ -407,6 +410,25 @@ def async_mock_intent(hass: HomeAssistant, intent_typ: str) -> list[intent.Inten
return intents
+class MockMqttReasonCode:
+ """Class to fake a MQTT ReasonCode."""
+
+ value: int
+ is_failure: bool
+
+ def __init__(
+ self, value: int = 0, is_failure: bool = False, name: str = "Success"
+ ) -> None:
+ """Initialize the mock reason code."""
+ self.value = value
+ self.is_failure = is_failure
+ self._name = name
+
+ def getName(self) -> str:
+ """Return the name of the reason code."""
+ return self._name
+
+
@callback
def async_fire_mqtt_message(
hass: HomeAssistant,
@@ -1004,6 +1026,7 @@ class MockConfigEntry(config_entries.ConfigEntry):
reason=None,
source=config_entries.SOURCE_USER,
state=None,
+ subentries_data=None,
title="Mock Title",
unique_id=None,
version=1,
@@ -1020,6 +1043,7 @@ class MockConfigEntry(config_entries.ConfigEntry):
"options": options or {},
"pref_disable_new_entities": pref_disable_new_entities,
"pref_disable_polling": pref_disable_polling,
+ "subentries_data": subentries_data or (),
"title": title,
"unique_id": unique_id,
"version": version,
@@ -1092,6 +1116,28 @@ class MockConfigEntry(config_entries.ConfigEntry):
},
)
+ async def start_subentry_reconfigure_flow(
+ self,
+ hass: HomeAssistant,
+ subentry_flow_type: str,
+ subentry_id: str,
+ *,
+ show_advanced_options: bool = False,
+ ) -> ConfigFlowResult:
+ """Start a subentry reconfiguration flow."""
+ if self.entry_id not in hass.config_entries._entries:
+ raise ValueError(
+ "Config entry must be added to hass to start reconfiguration flow"
+ )
+ return await hass.config_entries.subentries.async_init(
+ (self.entry_id, subentry_flow_type),
+ context={
+ "source": config_entries.SOURCE_RECONFIGURE,
+ "subentry_id": subentry_id,
+ "show_advanced_options": show_advanced_options,
+ },
+ )
+
async def start_reauth_flow(
hass: HomeAssistant,
@@ -1789,7 +1835,7 @@ def setup_test_component_platform(
async def _async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
- async_add_entities: AddEntitiesCallback,
+ async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
"""Set up a test component platform."""
async_add_entities(entities)
@@ -1821,23 +1867,6 @@ async def snapshot_platform(
assert state == snapshot(name=f"{entity_entry.entity_id}-state")
-def reset_translation_cache(hass: HomeAssistant, components: list[str]) -> None:
- """Reset translation cache for specified components.
-
- Use this if you are mocking a core component (for example via
- mock_integration), to ensure that the mocked translations are not
- persisted in the shared session cache.
- """
- translations_cache = translation._async_get_translations_cache(hass)
- for loaded_components in translations_cache.cache_data.loaded.values():
- for component_to_unload in components:
- loaded_components.discard(component_to_unload)
- for loaded_categories in translations_cache.cache_data.cache.values():
- for loaded_components in loaded_categories.values():
- for component_to_unload in components:
- loaded_components.pop(component_to_unload, None)
-
-
@lru_cache
def get_quality_scale(integration: str) -> dict[str, QualityScaleStatus]:
"""Load quality scale for integration."""
diff --git a/tests/components/acaia/snapshots/test_binary_sensor.ambr b/tests/components/acaia/snapshots/test_binary_sensor.ambr
index 113b5f1501e..a9c52c052a3 100644
--- a/tests/components/acaia/snapshots/test_binary_sensor.ambr
+++ b/tests/components/acaia/snapshots/test_binary_sensor.ambr
@@ -6,6 +6,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
diff --git a/tests/components/acaia/snapshots/test_button.ambr b/tests/components/acaia/snapshots/test_button.ambr
index cd91ca1a17a..11827c0997f 100644
--- a/tests/components/acaia/snapshots/test_button.ambr
+++ b/tests/components/acaia/snapshots/test_button.ambr
@@ -6,6 +6,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -52,6 +53,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -98,6 +100,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
diff --git a/tests/components/acaia/snapshots/test_init.ambr b/tests/components/acaia/snapshots/test_init.ambr
index 7011b20f68c..c7a11cb58df 100644
--- a/tests/components/acaia/snapshots/test_init.ambr
+++ b/tests/components/acaia/snapshots/test_init.ambr
@@ -3,6 +3,7 @@
DeviceRegistryEntrySnapshot({
'area_id': 'kitchen',
'config_entries': ,
+ 'config_entries_subentries': ,
'configuration_url': None,
'connections': set({
tuple(
diff --git a/tests/components/acaia/snapshots/test_sensor.ambr b/tests/components/acaia/snapshots/test_sensor.ambr
index c3c8ce966ee..9214db4f102 100644
--- a/tests/components/acaia/snapshots/test_sensor.ambr
+++ b/tests/components/acaia/snapshots/test_sensor.ambr
@@ -8,6 +8,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -59,6 +60,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -113,6 +115,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
diff --git a/tests/components/accuweather/snapshots/test_sensor.ambr b/tests/components/accuweather/snapshots/test_sensor.ambr
index 3468d638bc0..cbd2e14207e 100644
--- a/tests/components/accuweather/snapshots/test_sensor.ambr
+++ b/tests/components/accuweather/snapshots/test_sensor.ambr
@@ -7,14 +7,14 @@
'capabilities': dict({
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -49,11 +49,10 @@
'friendly_name': 'Home Air quality day 0',
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'context': ,
@@ -72,14 +71,14 @@
'capabilities': dict({
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -114,11 +113,10 @@
'friendly_name': 'Home Air quality day 1',
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'context': ,
@@ -137,14 +135,14 @@
'capabilities': dict({
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -179,11 +177,10 @@
'friendly_name': 'Home Air quality day 2',
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'context': ,
@@ -202,14 +199,14 @@
'capabilities': dict({
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -244,11 +241,10 @@
'friendly_name': 'Home Air quality day 3',
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'context': ,
@@ -267,14 +263,14 @@
'capabilities': dict({
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -309,11 +305,10 @@
'friendly_name': 'Home Air quality day 4',
'options': list([
'good',
- 'hazardous',
- 'high',
- 'low',
'moderate',
'unhealthy',
+ 'very_unhealthy',
+ 'hazardous',
]),
}),
'context': ,
@@ -333,6 +328,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -385,6 +381,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -440,6 +437,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -489,6 +487,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -537,6 +536,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -585,6 +585,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -633,6 +634,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -681,6 +683,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -729,6 +732,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -777,6 +781,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -825,6 +830,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -873,6 +879,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -921,6 +928,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -969,6 +977,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1016,6 +1025,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1063,6 +1073,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1110,6 +1121,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1157,6 +1169,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1204,6 +1217,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1251,6 +1265,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1298,6 +1313,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1345,6 +1361,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1392,6 +1409,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1441,6 +1459,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1491,6 +1510,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1540,6 +1560,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1589,6 +1610,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1638,6 +1660,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1687,6 +1710,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1736,6 +1760,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1784,6 +1809,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1832,6 +1858,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1880,6 +1907,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1928,6 +1956,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -1978,6 +2007,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2028,6 +2058,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2077,6 +2108,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2126,6 +2158,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2175,6 +2208,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2224,6 +2258,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2275,6 +2310,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2328,6 +2364,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2387,6 +2424,7 @@
]),
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2440,6 +2478,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2489,6 +2528,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2538,6 +2578,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2587,6 +2628,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2636,6 +2678,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2687,6 +2730,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2737,6 +2781,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2786,6 +2831,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2835,6 +2881,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2884,6 +2931,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2933,6 +2981,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -2982,6 +3031,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3031,6 +3081,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3080,6 +3131,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3129,6 +3181,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3178,6 +3231,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3229,6 +3283,7 @@
'state_class': ,
}),
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3279,6 +3334,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3328,6 +3384,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3377,6 +3434,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3426,6 +3484,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id': ,
'disabled_by': None,
@@ -3475,6 +3534,7 @@
'area_id': None,
'capabilities': None,
'config_entry_id': ,
+ 'config_subentry_id': ,
'device_class': None,
'device_id':