Skip to content

Commit fa9a799

Browse files
jath03jath03
committed
Formatting
1 parent a148c74 commit fa9a799

File tree

1 file changed

+46
-23
lines changed

1 file changed

+46
-23
lines changed

‎openrgb/orgb.py‎

Lines changed: 46 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,8 @@ def _update(self, data: utils.ZoneData):
6161
self.leds = [None for led in data.leds]
6262
for x in range(len(data.leds)):
6363
if self.leds[x] is None:
64-
self.leds[x] = LED(data.leds[x], data.colors[x], data.start_idx + x, self.device_id, self.comms)
64+
self.leds[x] = LED(data.leds[x], data.colors[x],
65+
data.start_idx + x, self.device_id, self.comms)
6566
else:
6667
self.leds[x]._update(data.leds[x], data.colors[x])
6768
self.mat_width = data.mat_width
@@ -82,7 +83,8 @@ def set_color(self, color: utils.RGBColor, fast: bool = False):
8283
utils.PacketType.RGBCONTROLLER_UPDATEZONELEDS,
8384
struct.calcsize(f"iIH{3*(len(self.leds))}b{len(self.leds)}x")
8485
)
85-
buff = struct.pack("iH", self.id, len(self.leds)) + (color.pack())*len(self.leds)
86+
buff = struct.pack("iH", self.id, len(self.leds)) + \
87+
(color.pack())*len(self.leds)
8688
buff = struct.pack("I", len(buff)) + buff
8789
self.comms.send_data(buff)
8890
if not fast:
@@ -96,13 +98,15 @@ def set_colors(self, colors: list[utils.RGBColor], fast: bool = False):
9698
:param fast: If you care more about quickly setting colors than having correct internal state data, then set :code:`fast` to :code:`True`
9799
'''
98100
if len(colors) != len(self.leds):
99-
raise IndexError("Number of colors doesn't match number of LEDs in the zone")
101+
raise IndexError(
102+
"Number of colors doesn't match number of LEDs in the zone")
100103
self.comms.send_header(
101104
self.device_id,
102105
utils.PacketType.RGBCONTROLLER_UPDATEZONELEDS,
103106
struct.calcsize(f"iIH{3*(len(self.leds))}b{len(self.leds)}x")
104107
)
105-
buff = struct.pack("iH", self.id, len(self.leds)) + b''.join((color.pack() for color in colors))
108+
buff = struct.pack("iH", self.id, len(self.leds)) + \
109+
b''.join((color.pack() for color in colors))
106110
buff = struct.pack("I", len(buff)) + buff
107111
self.comms.send_data(buff)
108112
if not fast:
@@ -144,12 +148,14 @@ def _update(self, data: utils.ControllerData):
144148
self.leds = [None for i in data.leds] # type: ignore
145149
for x in range(len(data.leds)):
146150
if self.leds[x] is None:
147-
self.leds[x] = LED(data.leds[x], data.colors[x], x, self.device_id, self.comms)
151+
self.leds[x] = LED(data.leds[x], data.colors[x],
152+
x, self.device_id, self.comms)
148153
else:
149154
self.leds[x]._update(data.leds[x], data.colors[x])
150155
for x in range(len(data.zones)):
151156
if self.zones[x] is None:
152-
self.zones[x] = Zone(data.zones[x], x, self.device_id, self.comms)
157+
self.zones[x] = Zone(
158+
data.zones[x], x, self.device_id, self.comms)
153159
else:
154160
self.zones[x]._update(data.zones[x]) # type: ignore
155161
self.modes = data.modes
@@ -190,7 +196,8 @@ def _set_device_colors(self, colors: list[utils.RGBColor], fast: bool = False):
190196
utils.PacketType.RGBCONTROLLER_UPDATELEDS,
191197
struct.calcsize(f"IH{3*(len(self.leds))}b{len(self.leds)}x")
192198
)
193-
buff = struct.pack("H", len(self.leds)) + b''.join((color.pack() for color in colors))
199+
buff = struct.pack("H", len(self.leds)) + \
200+
b''.join((color.pack() for color in colors))
194201
buff = struct.pack("I", len(buff)) + buff
195202
self.comms.send_data(buff)
196203
if not fast:
@@ -217,7 +224,8 @@ def _set_mode_colors(self, colors: list[utils.RGBColor]):
217224
active_mode = self.modes[self.active_mode]
218225
assert active_mode.color_mode == utils.ModeColors.MODE_SPECIFIC
219226
assert active_mode.colors is not None
220-
assert active_mode.colors_min <= len(colors) <= active_mode.colors_max # type: ignore
227+
assert active_mode.colors_min <= len(
228+
colors) <= active_mode.colors_max # type: ignore
221229
active_mode.colors = colors
222230
self.set_mode(active_mode)
223231

@@ -261,9 +269,11 @@ def set_mode(self, mode: Union[int, str, utils.ModeData], save: bool = False):
261269
mode = self.modes[mode]
262270
elif type(mode) == str:
263271
try:
264-
mode = next((m for m in self.modes if m.name.lower() == mode.lower()))
272+
mode = next(
273+
(m for m in self.modes if m.name.lower() == mode.lower()))
265274
except StopIteration as e:
266-
raise ValueError(f"Mode `{mode}` not found for device `{self.name}`") from e
275+
raise ValueError(
276+
f"Mode `{mode}` not found for device `{self.name}`") from e
267277
data = mode.pack(self.comms._protocol_version) # type: ignore
268278
self.comms.send_header(
269279
self.id,
@@ -330,7 +340,8 @@ def __init__(self, address: str = "127.0.0.1", port: int = 6742, name: str = "op
330340
self.device_num = 0
331341
self.devices: list[Device] = []
332342
self.profiles: list[utils.Profile] = []
333-
self.comms = NetworkClient(self._callback, address, port, name, protocol_version)
343+
self.comms = NetworkClient(
344+
self._callback, address, port, name, protocol_version)
334345
self.address = address
335346
self.port = port
336347
self.name = name
@@ -343,7 +354,8 @@ def _callback(self, device: int, type: int, data: Any):
343354
if type == utils.PacketType.REQUEST_CONTROLLER_COUNT:
344355
if data != self.device_num or data != len(self.devices):
345356
self.device_num = data
346-
self.devices = [None for x in range(self.device_num)] # type: ignore
357+
self.devices = [None for x in range(
358+
self.device_num)] # type: ignore
347359
for x in range(self.device_num):
348360
self.comms.requestDeviceData(x)
349361
elif type == utils.PacketType.REQUEST_CONTROLLER_DATA:
@@ -401,7 +413,8 @@ def load_profile(self, name: Union[str, int, utils.Profile], local: bool = False
401413
assert type(name) is str
402414
if directory == '':
403415
if platform.system() == "Linux":
404-
directory = environ['HOME'].rstrip("/") + "/.config/OpenRGB"
416+
directory = environ['HOME'].rstrip(
417+
"/") + "/.config/OpenRGB"
405418
elif platform.system() == "Windows":
406419
directory = environ['APPDATA'].rstrip("\\") + "\\OpenRGB"
407420
else:
@@ -427,15 +440,18 @@ def load_profile(self, name: Union[str, int, utils.Profile], local: bool = False
427440
else:
428441
if type(name) is str:
429442
try:
430-
name = next(p for p in self.profiles if p.name.lower() == name.lower())
443+
name = next(
444+
p for p in self.profiles if p.name.lower() == name.lower())
431445
except StopIteration as e:
432-
raise ValueError(f"`{name}` is not an existing profile") from e
446+
raise ValueError(
447+
f"`{name}` is not an existing profile") from e
433448
elif type(name) is int:
434449
name = self.profiles[name]
435450
elif type(name) is utils.Profile:
436451
pass
437452
raw_name = name.pack() # type: ignore
438-
self.comms.send_header(0, utils.PacketType.REQUEST_LOAD_PROFILE, len(raw_name))
453+
self.comms.send_header(
454+
0, utils.PacketType.REQUEST_LOAD_PROFILE, len(raw_name))
439455
self.comms.send_data(raw_name)
440456

441457
def save_profile(self, name: Union[str, int, utils.Profile], local: bool = False, directory: str = ''):
@@ -451,25 +467,29 @@ def save_profile(self, name: Union[str, int, utils.Profile], local: bool = False
451467
self.update()
452468
if directory == '':
453469
if platform.system() == "Linux":
454-
directory = environ['HOME'].rstrip("/") + "/.config/OpenRGB"
470+
directory = environ['HOME'].rstrip(
471+
"/") + "/.config/OpenRGB"
455472
elif platform.system() == "Windows":
456473
directory = environ['APPDATA'].rstrip("\\") + "\\OpenRGB"
457474
else:
458475
directory = '.'
459476
with open(f'{directory.rstrip("/")}/{name}.orp', 'wb') as f:
460-
f.write(utils.LocalProfile([dev.data for dev in self.devices]).pack())
477+
f.write(utils.LocalProfile(
478+
[dev.data for dev in self.devices]).pack())
461479
else:
462480
if type(name) is str:
463481
try:
464-
name = next(p for p in self.profiles if p.name.lower() == name.lower())
482+
name = next(
483+
p for p in self.profiles if p.name.lower() == name.lower())
465484
except StopIteration:
466485
name = utils.Profile(name) # type: ignore
467486
elif type(name) is int:
468487
name = self.profiles[name]
469488
elif type(name) is utils.Profile:
470489
pass
471490
raw_name = name.pack() # type: ignore
472-
self.comms.send_header(0, utils.PacketType.REQUEST_SAVE_PROFILE, len(raw_name))
491+
self.comms.send_header(
492+
0, utils.PacketType.REQUEST_SAVE_PROFILE, len(raw_name))
473493
self.comms.send_data(raw_name)
474494
self.update_profiles()
475495

@@ -481,15 +501,17 @@ def delete_profile(self, name: Union[str, int, utils.Profile]):
481501
'''
482502
if type(name) is str:
483503
try:
484-
name = next(p for p in self.profiles if p.name.lower() == name.lower())
504+
name = next(
505+
p for p in self.profiles if p.name.lower() == name.lower())
485506
except StopIteration as e:
486507
raise ValueError(f"`{name}` is not an existing profile") from e
487508
elif type(name) is int:
488509
name = self.profiles[name]
489510
elif type(name) is utils.Profile:
490511
pass
491512
raw_name = name.pack() # type: ignore
492-
self.comms.send_header(0, utils.PacketType.REQUEST_DELETE_PROFILE, len(raw_name))
513+
self.comms.send_header(
514+
0, utils.PacketType.REQUEST_DELETE_PROFILE, len(raw_name))
493515
self.comms.send_data(raw_name)
494516
self.update_profiles()
495517

@@ -542,7 +564,8 @@ def protocol_version(self, version: int):
542564
if version <= self.comms.max_protocol_version:
543565
self.comms._protocol_version = version
544566
else:
545-
raise ValueError(f"version {version} is greater than maximum supported version {self.comms.max_protocol_version}")
567+
raise ValueError(
568+
f"version {version} is greater than maximum supported version {self.comms.max_protocol_version}")
546569

547570
@property
548571
def ee_devices(self):

0 commit comments

Comments
 (0)