Combine ifs

This commit is contained in:
Raymond Ha
2023-05-26 01:14:53 -07:00
parent 230024c179
commit 49cd69de8b

View File

@@ -28,27 +28,28 @@ async def find_atvs(loop):
atvs = []
for i in devices:
# Only get Apple TV's
if i.device_info.operating_system == OperatingSystem.TvOS:
if input("Found {}. Do you want to add it? (y/n) ".format(i.name)) == "y":
if (
i.device_info.operating_system == OperatingSystem.TvOS
and input("Found {}. Do you want to add it? (y/n) ".format(i.name)) == "y"
):
identifier = i.identifier
identifier = i.identifier
pairing = await pyatv.pair(
i, loop=loop, protocol=pyatv.Protocol.AirPlay
)
await pairing.begin()
if pairing.device_provides_pin:
pin = await _read_input(loop, "Enter PIN on screen: ")
pairing.pin(pin)
pairing = await pyatv.pair(
i, loop=loop, protocol=pyatv.Protocol.AirPlay
await pairing.finish()
if pairing.has_paired:
creds = pairing.service.credentials
atvs.append(
{"identifier": identifier, "airplay_credentials": creds}
)
await pairing.begin()
if pairing.device_provides_pin:
pin = await _read_input(loop, "Enter PIN on screen: ")
pairing.pin(pin)
await pairing.finish()
if pairing.has_paired:
creds = pairing.service.credentials
atvs.append(
{"identifier": identifier, "airplay_credentials": creds}
)
print("Pairing successful")
await pairing.close()
print("Pairing successful")
await pairing.close()
return atvs