Refactor active session handling in PlaystationNetwork (#149559)

This commit is contained in:
Manu 2025-07-28 13:13:08 +02:00 committed by GitHub
parent 140f56aeaa
commit 95c5a91f01
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -107,30 +107,34 @@ class PlaystationNetwork:
data.shareable_profile_link = self.shareable_profile_link
data.availability = data.presence["basicPresence"]["availability"]
session = SessionData()
session.platform = PlatformType(
data.presence["basicPresence"]["primaryPlatformInfo"]["platform"]
)
if session.platform in SUPPORTED_PLATFORMS:
session.status = data.presence.get("basicPresence", {}).get(
"primaryPlatformInfo"
)["onlineStatus"]
game_title_info = data.presence.get("basicPresence", {}).get(
"gameTitleInfoList"
if "platform" in data.presence["basicPresence"]["primaryPlatformInfo"]:
primary_platform = PlatformType(
data.presence["basicPresence"]["primaryPlatformInfo"]["platform"]
)
game_title_info: dict[str, Any] = next(
iter(
data.presence.get("basicPresence", {}).get("gameTitleInfoList", [])
),
{},
)
status = data.presence.get("basicPresence", {}).get("primaryPlatformInfo")[
"onlineStatus"
]
title_format = (
PlatformType(fmt) if (fmt := game_title_info.get("format")) else None
)
if game_title_info:
session.title_id = game_title_info[0]["npTitleId"]
session.title_name = game_title_info[0]["titleName"]
session.format = PlatformType(game_title_info[0]["format"])
if session.format in {PlatformType.PS5, PlatformType.PSPC}:
session.media_image_url = game_title_info[0]["conceptIconUrl"]
else:
session.media_image_url = game_title_info[0]["npTitleIconUrl"]
data.active_sessions[session.platform] = session
data.active_sessions[primary_platform] = SessionData(
platform=primary_platform,
status=status,
title_id=game_title_info.get("npTitleId"),
title_name=game_title_info.get("titleName"),
format=title_format,
media_image_url=(
game_title_info.get("conceptIconUrl")
or game_title_info.get("npTitleIconUrl")
),
)
if self.legacy_profile:
presence = self.legacy_profile["profile"].get("presences", [])