Tool black: auto-format Python code

This commit is contained in:
Odd Stråbø 2024-06-10 09:19:47 +00:00
commit 7763762212
54 changed files with 6545 additions and 4723 deletions

View file

@ -14,7 +14,8 @@ from ..helper import LazyLogger, settings
# If numpy is installed, the websockets library tries to use it, and then
# kodi hard crashes for reasons I don't even want to pretend to understand
import sys # noqa: E402,I100
sys.modules['numpy'] = None
sys.modules["numpy"] = None
import websocket # noqa: E402,I201
##################################################################################################
@ -41,23 +42,29 @@ class WSClient(threading.Thread):
if self.wsc is None:
raise ValueError("The websocket client is not started.")
self.wsc.send(json.dumps({'MessageType': message, "Data": data}))
self.wsc.send(json.dumps({"MessageType": message, "Data": data}))
def run(self):
monitor = xbmc.Monitor()
token = self.client.config.data['auth.token']
device_id = self.client.config.data['app.device_id']
server = self.client.config.data['auth.server']
server = server.replace('https://', 'wss://') if server.startswith('https') else server.replace('http://', 'ws://')
token = self.client.config.data["auth.token"]
device_id = self.client.config.data["app.device_id"]
server = self.client.config.data["auth.server"]
server = (
server.replace("https://", "wss://")
if server.startswith("https")
else server.replace("http://", "ws://")
)
wsc_url = "%s/socket?api_key=%s&device_id=%s" % (server, token, device_id)
LOG.info("Websocket url: %s", wsc_url)
self.wsc = websocket.WebSocketApp(wsc_url,
on_open=lambda ws: self.on_open(ws),
on_message=lambda ws, message: self.on_message(ws, message),
on_error=lambda ws, error: self.on_error(ws, error))
self.wsc = websocket.WebSocketApp(
wsc_url,
on_open=lambda ws: self.on_open(ws),
on_message=lambda ws, message: self.on_message(ws, message),
on_error=lambda ws, error: self.on_error(ws, error),
)
while not self.stop:
@ -73,41 +80,42 @@ class WSClient(threading.Thread):
LOG.info("--->[ websocket opened ]")
# Avoid a timing issue where the capabilities are not correctly registered
time.sleep(5)
if settings('remoteControl.bool'):
self.client.jellyfin.post_capabilities({
'PlayableMediaTypes': "Audio,Video",
'SupportsMediaControl': True,
'SupportedCommands': (
"MoveUp,MoveDown,MoveLeft,MoveRight,Select,"
"Back,ToggleContextMenu,ToggleFullscreen,ToggleOsdMenu,"
"GoHome,PageUp,NextLetter,GoToSearch,"
"GoToSettings,PageDown,PreviousLetter,TakeScreenshot,"
"VolumeUp,VolumeDown,ToggleMute,SendString,DisplayMessage,"
"SetAudioStreamIndex,SetSubtitleStreamIndex,"
"SetRepeatMode,Mute,Unmute,SetVolume,"
"Play,Playstate,PlayNext,PlayMediaSource"
),
})
if settings("remoteControl.bool"):
self.client.jellyfin.post_capabilities(
{
"PlayableMediaTypes": "Audio,Video",
"SupportsMediaControl": True,
"SupportedCommands": (
"MoveUp,MoveDown,MoveLeft,MoveRight,Select,"
"Back,ToggleContextMenu,ToggleFullscreen,ToggleOsdMenu,"
"GoHome,PageUp,NextLetter,GoToSearch,"
"GoToSettings,PageDown,PreviousLetter,TakeScreenshot,"
"VolumeUp,VolumeDown,ToggleMute,SendString,DisplayMessage,"
"SetAudioStreamIndex,SetSubtitleStreamIndex,"
"SetRepeatMode,Mute,Unmute,SetVolume,"
"Play,Playstate,PlayNext,PlayMediaSource"
),
}
)
else:
self.client.jellyfin.post_capabilities({
"PlayableMediaTypes": "Audio, Video",
"SupportsMediaControl": False
})
self.client.jellyfin.post_capabilities(
{"PlayableMediaTypes": "Audio, Video", "SupportsMediaControl": False}
)
def on_message(self, ws, message):
message = json.loads(message)
data = message.get('Data', {})
data = message.get("Data", {})
if message['MessageType'] in ('RefreshProgress',):
if message["MessageType"] in ("RefreshProgress",):
LOG.debug("Ignoring %s", message)
return
if not self.client.config.data['app.default']:
data['ServerId'] = self.client.auth.server_id
if not self.client.config.data["app.default"]:
data["ServerId"] = self.client.auth.server_id
self.client.callback(message['MessageType'], data)
self.client.callback(message["MessageType"], data)
def stop_client(self):