Fix KeyError in friends coordinator (#149684)

This commit is contained in:
Manu 2025-07-30 19:59:01 +02:00 committed by GitHub
parent a76af50c10
commit 8d27ca1e21
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 9 deletions

View File

@ -6,7 +6,7 @@ from abc import abstractmethod
from dataclasses import dataclass
from datetime import timedelta
import logging
from typing import Any
from typing import TYPE_CHECKING, Any
from psnawp_api.core.psnawp_exceptions import (
PSNAWPAuthenticationError,
@ -29,7 +29,7 @@ from homeassistant.exceptions import (
)
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import CONF_ACCOUNT_ID, DOMAIN
from .const import DOMAIN
from .helpers import PlaystationNetwork, PlaystationNetworkData
_LOGGER = logging.getLogger(__name__)
@ -176,7 +176,9 @@ class PlaystationNetworkFriendDataCoordinator(
def _setup(self) -> None:
"""Set up the coordinator."""
self.user = self.psn.psn.user(account_id=self.subentry.data[CONF_ACCOUNT_ID])
if TYPE_CHECKING:
assert self.subentry.unique_id
self.user = self.psn.psn.user(account_id=self.subentry.unique_id)
self.profile = self.user.profile()
async def _async_setup(self) -> None:

View File

@ -14,11 +14,7 @@ from psnawp_api.models.trophies import (
)
import pytest
from homeassistant.components.playstation_network.const import (
CONF_ACCOUNT_ID,
CONF_NPSSO,
DOMAIN,
)
from homeassistant.components.playstation_network.const import CONF_NPSSO, DOMAIN
from homeassistant.config_entries import ConfigSubentryData
from tests.common import MockConfigEntry
@ -40,7 +36,7 @@ def mock_config_entry() -> MockConfigEntry:
unique_id=PSN_ID,
subentries_data=[
ConfigSubentryData(
data={CONF_ACCOUNT_ID: "fren-psn-id"},
data={},
subentry_id="ABCDEF",
subentry_type="friend",
title="PublicUniversalFriend",