diff --git a/iSponsorBlockTV/config_setup.py b/iSponsorBlockTV/config_setup.py index 1208116..cfadf75 100644 --- a/iSponsorBlockTV/config_setup.py +++ b/iSponsorBlockTV/config_setup.py @@ -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