Switch a few ssdp calls to use get_lower (#99931)

get_lower avoids lower casing already lower-cased strings
This commit is contained in:
J. Nick Koston 2023-09-08 12:59:25 -05:00 committed by GitHub
parent be4ea32049
commit d1ac4c9c46
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -606,7 +606,7 @@ def discovery_info_from_headers_and_description(
) -> SsdpServiceInfo: ) -> SsdpServiceInfo:
"""Convert headers and description to discovery_info.""" """Convert headers and description to discovery_info."""
ssdp_usn = combined_headers["usn"] ssdp_usn = combined_headers["usn"]
ssdp_st = combined_headers.get("st") ssdp_st = combined_headers.get_lower("st")
if isinstance(info_desc, CaseInsensitiveDict): if isinstance(info_desc, CaseInsensitiveDict):
upnp_info = {**info_desc.as_dict()} upnp_info = {**info_desc.as_dict()}
else: else:
@ -626,11 +626,11 @@ def discovery_info_from_headers_and_description(
return SsdpServiceInfo( return SsdpServiceInfo(
ssdp_usn=ssdp_usn, ssdp_usn=ssdp_usn,
ssdp_st=ssdp_st, ssdp_st=ssdp_st,
ssdp_ext=combined_headers.get("ext"), ssdp_ext=combined_headers.get_lower("ext"),
ssdp_server=combined_headers.get("server"), ssdp_server=combined_headers.get_lower("server"),
ssdp_location=combined_headers.get("location"), ssdp_location=combined_headers.get_lower("location"),
ssdp_udn=combined_headers.get("_udn"), ssdp_udn=combined_headers.get_lower("_udn"),
ssdp_nt=combined_headers.get("nt"), ssdp_nt=combined_headers.get_lower("nt"),
ssdp_headers=combined_headers, ssdp_headers=combined_headers,
upnp=upnp_info, upnp=upnp_info,
) )