Check tvOS instead of specific atv models

This commit is contained in:
Raymond Ha
2023-05-25 22:02:26 -07:00
parent f06e2b1490
commit 230024c179

View File

@@ -1,7 +1,7 @@
import pyatv import pyatv
import json import json
import asyncio import asyncio
from pyatv.const import DeviceModel from pyatv.const import OperatingSystem
import sys import sys
import aiohttp import aiohttp
import asyncio import asyncio
@@ -28,12 +28,7 @@ async def find_atvs(loop):
atvs = [] atvs = []
for i in devices: for i in devices:
# Only get Apple TV's # Only get Apple TV's
if i.device_info.model in [ if i.device_info.operating_system == OperatingSystem.TvOS:
DeviceModel.Gen4,
DeviceModel.Gen4K,
DeviceModel.AppleTV4KGen2,
]:
# if i.device_info.model in [DeviceModel.AppleTV4KGen2]: #FOR TESTING
if input("Found {}. Do you want to add it? (y/n) ".format(i.name)) == "y": if input("Found {}. Do you want to add it? (y/n) ".format(i.name)) == "y":
identifier = i.identifier identifier = i.identifier