diff --git a/iSponsorBlockTV/main.py b/iSponsorBlockTV/main.py index eb9fd29..ac43df2 100644 --- a/iSponsorBlockTV/main.py +++ b/iSponsorBlockTV/main.py @@ -7,11 +7,12 @@ import traceback class DeviceListener: - def __init__(self, api_helper, config, screen_id, offset): + def __init__(self, api_helper, config, device): self.task: asyncio.Task = None self.api_helper = api_helper - self.lounge_controller = ytlounge.YtLoungeApi(screen_id, config, api_helper) - self.offset = offset + self.lounge_controller = ytlounge.YtLoungeApi(device.screen_id, config, api_helper) + self.offset = device.offset + self.name = device.name self.cancelled = False # Ensures that we have a valid auth token @@ -54,7 +55,7 @@ class DeviceListener: await lounge_controller.connect() except: pass - print(f"Connected to device {lounge_controller.screen_name}") + print(f"Connected to device {lounge_controller.screen_name} ({self.name})") try: #print("Subscribing to lounge") sub = await lounge_controller.subscribe_monitored(self) @@ -136,7 +137,7 @@ def main(config, debug): web_session = aiohttp.ClientSession(loop=loop, connector=tcp_connector) api_helper = api_helpers.ApiHelper(config, web_session) for i in config.devices: - device = DeviceListener(api_helper, config, i.screen_id, i.offset) + device = DeviceListener(api_helper, config, i) devices.append(device) tasks.append(loop.create_task(device.loop())) tasks.append(loop.create_task(device.refresh_auth_loop()))