mirror of
https://github.com/dmunozv04/iSponsorBlockTV.git
synced 2025-12-06 11:56:45 +03:00
105 lines
3.6 KiB
Python
105 lines
3.6 KiB
Python
import pyatv
|
|
import json
|
|
import asyncio
|
|
from pyatv.const import DeviceModel
|
|
from pyatv.scripts import TransformProtocol
|
|
import sys
|
|
|
|
def load_config(config_file="config.json"):
|
|
try:
|
|
with open(config_file) as f:
|
|
config = json.load(f)
|
|
except:
|
|
config = {}
|
|
return config
|
|
|
|
def save_config(config, config_file="config.json"):
|
|
with open(config_file, "w") as f:
|
|
json.dump(config, f)
|
|
|
|
#Taken from postlund/pyatv atvremote.py
|
|
async def _read_input(loop: asyncio.AbstractEventLoop, prompt: str):
|
|
sys.stdout.write(prompt)
|
|
sys.stdout.flush()
|
|
user_input = await loop.run_in_executor(None, sys.stdin.readline)
|
|
return user_input.strip()
|
|
|
|
async def find_atvs(loop):
|
|
devices = await pyatv.scan(loop)
|
|
if not devices:
|
|
print("No devices found")
|
|
return
|
|
atvs = []
|
|
for i in devices:
|
|
#Only get Apple TV's
|
|
if i.device_info.model in [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":
|
|
|
|
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)
|
|
await pairing.finish()
|
|
creds = pairing.service.credentials
|
|
atvs.append({"identifier": identifier, "airplay_credentials": creds})
|
|
print("Pairing successful")
|
|
await pairing.close()
|
|
return atvs
|
|
|
|
|
|
|
|
|
|
def main():
|
|
config = load_config()
|
|
try: num_atvs = len(config["atvs"])
|
|
except: num_atvs = 0
|
|
if input("Found {} Apple TV(s) in config.json. Add more? (y/n) ".format(num_atvs)) == "y":
|
|
loop = asyncio.get_event_loop_policy().get_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
task = loop.create_task(find_atvs(loop))
|
|
loop.run_until_complete(task)
|
|
atvs = task.result()
|
|
try:
|
|
for i in atvs:
|
|
config["atvs"].append(i)
|
|
print("done adding")
|
|
except:
|
|
print("rewriting atvs (don't worry if none were saved before)")
|
|
config["atvs"] = atvs
|
|
|
|
try : apikey = config["apikey"]
|
|
except:
|
|
apikey = ""
|
|
if apikey != "" :
|
|
if input("Apikey already specified. Change it? (y/n) ") == "y":
|
|
apikey = input("Enter your API key: ")
|
|
config["apikey"] = apikey
|
|
else:
|
|
apikey = input("Enter your API key: ")
|
|
config["apikey"] = apikey
|
|
|
|
try: skip_categories = config["skip_categories"]
|
|
except:
|
|
skip_categories = []
|
|
|
|
if skip_categories != []:
|
|
if input("Skip categories already specified. Change them? (y/n) ") == "y":
|
|
categories = input("Enter skip categories (space sepparated) Options: [sponsor, selfpromo, exclusive_access, interaction, poi_highlight, intro, outro, preview, filler, music_offtopic:\n")
|
|
skip_categories = categories.split(" ")
|
|
else:
|
|
categories = input("Enter skip categories (space sepparated) Options: [sponsor, selfpromo, exclusive_access, interaction, poi_highlight, intro, outro, preview, filler, music_offtopic:\n")
|
|
skip_categories = categories.split(" ")
|
|
config["skip_categories"] = skip_categories
|
|
print("config finished")
|
|
save_config(config)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("starting")
|
|
main()
|