def test_write_settings_to_file(self):
"""> Test if cache file writing works"""
# Build file path
home_dir = os.path.expanduser("~")
dir_name = settings.CACHE_DIR
file_name = settings.CACHE_FILE
path_and_file = os.path.join(home_dir, dir_name, file_name)
# Backup original file if this is running locally
if Path(path_and_file).exists():
shutil.copyfile(path_and_file, path_and_file + "_backup")
# Save random device settings to cache
device = MagicMock()
device.name = "test_device"
device.serial = "test_serial"
util.write_settings_to_file(device, dpi="1234")
util.write_settings_to_file(device, effect=["multicolor"])
# Check if file has been written
self.assertTrue(os.path.isfile(path_and_file))
# Check contents
data = json.loads(Path(path_and_file).read_text())[0]
self.assertEqual(data["device_name"], "test_device")
self.assertEqual(data["serial"], "test_serial")
self.assertEqual(data["dpi"], "1234")
self.assertEqual(data["effect"], ["multicolor"])
# Restore original file
if Path(path_and_file + "_backup").exists():
shutil.copyfile(path_and_file + "_backup", path_and_file)
Path(path_and_file + "_backup").unlink()
def set(self):
# Iterate over each device and set Polling Rate
for device in self.device_manager.devices:
# If -d argument is set, only set those devices
if (self.args.device and (device.name in self.args.device or device.serial in self.args.device)) or (not self.args.device):
if device.has("poll_rate"):
if self.args.poll == "print":
if self.args.dpi == "print" or self.args.battery and len(self.args.battery) == 1:
print('poll_rate:', device.poll_rate)
else:
print(device.poll_rate)
else:
if self.args.verbose:
print(
"Setting polling rate of device {} to {}".format(
device.name,
self.args.poll))
# Actually set Polling Rate
device.poll_rate = int(self.args.poll)
# Save used settings for this device to a file
util.write_settings_to_file(device, poll=self.args.poll)
else:
print("Device does not support setter the polling rate")
def set(self):
# Iterate over each device and set DPI
for device in self.device_manager.devices:
# If -d argument is set, only set those devices
if (self.args.device and
(device.name in self.args.device or device.serial
in self.args.device)) or (not self.args.device):
if self.args.verbose:
print("Setting brightness of device {}:".format(
device.name))
brightness = self.args.brightness
if 'all' in brightness.keys():
for i in settings.ZONES:
brightness[i] = brightness['all']
del brightness['all']
if 'wheel' in brightness.keys(
) and not device.has('lighting_wheel_brightness'):
brightness['scroll'] = brightness['wheel']
del brightness['wheel']
if self.args.verbose:
print(
' Device does not support "wheel" assuming "scroll"'
)
if self.args.verbose:
print(' Input data:', brightness)
for i in brightness:
if i == 'generic':
if device.has('brightness'):
if self.args.verbose:
print(' Setting brightness to:',
brightness[i])
device.brightness = int(brightness['generic'])
elif self.args.verbose:
print(' Device does not support brightness')
elif util.rgb_support(device, i, 'brightness'):
if self.args.verbose:
print(
' Setting lighting_' + i +
'_brightness to', brightness[i])
val = int(brightness[i])
if i == 'scroll':
i = 'scroll_wheel'
getattr(device.fx.misc, i).brightness = val
elif self.args.verbose:
print(' Device does not support lighting_' + i +
'_brightness')
# Save used settings for this device to a file
util.write_settings_to_file(device, brightness=brightness)
def set(self):
for device in self.device_manager.devices:
if (self.args.device and
(device.name in self.args.device or device.serial
in self.args.device)) or (not self.args.device):
if device.has("battery"):
if len(self.args.battery) < 2:
print(device.name, 'battery:')
print(" charge:", device.battery_level)
print(" charging:", device.is_charging)
print(" low threshold:",
device.get_low_battery_threshold(), '%')
print(" idle delay", device.get_idle_time(),
'seconds')
else:
i = 0
stop = len(self.args.battery)
bat = {}
while i < stop:
if self.args.battery[i] == "low" and stop > i + 1:
bat['low'] = int(self.args.battery[i + 1])
device.set_low_battery_threshold(bat['low'])
if self.args.verbose:
print(device.name, 'low battery =',
device.get_low_battery_threshold(),
'%')
elif self.args.battery[
i] == "idle" and stop > i + 1:
bat['idle'] = int(self.args.battery[i + 1])
device.set_idle_time(bat['idle'])
if self.args.verbose:
print(device.name, 'idle delay =',
device.get_idle_time(), 'seconds')
i += 1
# Save used settings for this device to a file
util.write_settings_to_file(device, battery=bat)
else:
print('Does', device.name, 'have a battery?')
def set(self):
# Iterate over each device and set DPI
for device in self.device_manager.devices:
# If -d argument is set, only set those devices
if (self.args.device and
(device.name in self.args.device or device.serial
in self.args.device)) or (not self.args.device):
if not device.has('dpi'):
if self.args.verbose:
print("Device {} is not have a DPI setter".format(
device.name))
elif self.args.dpi == "print":
dpi = device.dpi
if self.args.poll == "print":
if dpi[0] == dpi[1] or self.args.battery and len(
self.args.battery) == 1:
print('DPI:', dpi[0])
else:
print('DPI:', *dpi)
else:
if dpi[0] == dpi[1]:
print(dpi[0])
else:
print(*dpi)
else:
if self.args.verbose:
print("Setting DPI of device {} to {}".format(
device.name, self.args.dpi))
# Actually set DPI
dpi = self.args.dpi.split(',')
if len(dpi) == 1:
dpi.append(int(dpi[0]))
device.dpi = (int(dpi[0]), int(dpi[1]))
# Save used settings for this device to a file
util.write_settings_to_file(device, dpi=self.args.dpi)
def set_effect_to_device(self, device, effects, color, zones):
# Known Effects: advanced, breath_dual, breath_random, breath_single,
# breath_triple, blinking, none, reactive, ripple, ripple_random,
# pulsate, spectrum, starlight_dual, starlight_random,
# starlight_single, static, wave
if self.args.verbose:
debug_msg = {}
og_color_len = len(color)
c_used = 0
i = 0
stop = len(zones)
if len(effects) < stop:
stop = len(effects)
if self.args.verbose:
print(" Warning: Only", stop, "effects provided for",
len(zones), "zones")
while i < stop:
arg = effects[i].split(',')
effect = arg[0]
if effect not in settings.EFFECTS and effect not in settings.CUSTOM_EFFECTS:
print(" Warning:", effect, "is not a known effect.")
arg.pop(0)
used = 0
for zone in zones[i]:
if self.args.verbose:
if zone not in debug_msg:
debug_msg[zone] = []
if not util.rgb_support(device, zone, effect):
if self.args.verbose:
if util.rgb_support(device, zone):
debug_msg[zone].append(
['Does not support', effect])
else:
debug_msg[zone].append([zone, 'is not supported'])
continue
if zone == 'generic':
prop = device.fx
else:
if zone in ["scroll", "wheel"]:
if self.args.verbose:
debug_msg[zone].append(
['not valid, assuming scroll_wheel'])
zone = "scroll_wheel"
zones[i].append(zone)
continue
prop = getattr(device.fx.misc, zone)
if og_color_len == 0 and effect not in settings.COLORLESS_EFFECTS and not (
effect == 'multicolor' and
(len(arg) == 0 or arg[0] == '0')):
# Lets get default colors
# We can skip this if the effect does not need colors
# * multicolor does NOT use consume colors if set to 0 (default)
try:
# hasattr will error anyway if does not work, no point testing it
color = parse_color(
util.bytes_array_to_hex_array(
getattr(prop, 'colors')), self.args)
except:
color = [util.get_x_color(self.args.verbose)]
c_used = 0
if self.args.verbose:
debug_msg[zone].append(
["No color given, using:", color])
if effect not in settings.ALL_EFFECTS:
debug_msg[zone].append([
"Sorry", effect,
"is supported by the hardware, but this software does not\n",
" Consider makeing a bug report:\n",
" https://github.com/LoLei/razer-cli/issues"
])
continue
# Prep colors
used = 0
if effect not in settings.COLORLESS_EFFECTS:
# Custom effects do this in there own block
used = 1
if effect == 'breath_triple':
used = 3
elif effect in ['breath_dual', 'starlight_dual']:
used = 2
while len(color) < c_used + used:
color.append(util.get_random_color_rgb())
# Do I really need this message?
# if self.args.verbose:
# debug_msg[zone].append(["Using color(s):",
# [*color[c_used:c_used+used]],
# "\n from:",color])
# Set Effect
if effect in ['none', 'breath_random', 'spectrum']:
# Effects that use no parameters
if getattr(prop, effect)() and self.args.verbose:
debug_msg[zone].append(["Setting to", effect])
elif effect in [
'static', 'breath_single', 'blinking', 'pulsate'
]:
# Effects that use 1 color
rgb = color[c_used]
if getattr(prop, effect)(*rgb) and self.args.verbose:
debug_msg[zone].append(
["Setting", effect, "to:", *rgb])
elif effect == 'breath_dual':
# Effects that use 2 colors
rgb = color[c_used]
rgb2 = color[c_used + 1]
if getattr(prop, effect)(*rgb, *
rgb2) and self.args.verbose:
debug_msg[zone].append([
"Setting", effect, "to:\n [", *rgb,
']\n [', *rgb2, ']'
])
elif effect == 'breath_triple':
# Effects that use 3 colors
rgb = color[c_used]
rgb2 = color[c_used + 1]
rgb3 = color[c_used + 2]
if getattr(prop, effect)(*rgb, *rgb2, *
rgb3) and self.args.verbose:
debug_msg[zone].append([
"Setting", effect, "to:\n [", *rgb,
']\n [', *rgb2, ']\n [', *rgb3, ']'
])
elif effect == 'reactive':
# These are just numbers 1-4
# time = [razer_constants.REACTIVE_500MS, razer_constants.REACTIVE_1000MS,
# razer_constants.REACTIVE_1500MS, razer_constants.REACTIVE_2000MS]
if len(arg) == 0:
time = util.randint(1, 4)
else:
time = int(arg[0])
if getattr(prop, effect)(*color[c_used],
time) and self.args.verbose:
debug_msg[zone].append([
"Setting", effect, "to:", *color[c_used],
'@ timing level', time
])
elif effect == 'ripple':
# Effects that use 1 color and a refresh rate
if len(arg) == 0:
refresh = razer_constants.RIPPLE_REFRESH_RATE
else:
refresh = float(arg[0])
rgb = color[c_used]
if getattr(prop, effect)(*rgb,
refresh) and self.args.verbose:
debug_msg[zone].append(["Setting", effect, "to:", rgb])
elif effect == 'ripple_random':
if len(arg) == 0:
refresh = razer_constants.RIPPLE_REFRESH_RATE
else:
refresh = float(arg[0])
if getattr(prop, effect)(refresh) and self.args.verbose:
debug_msg[zone].append(["Setting to", effect])
elif effect == 'starlight_single':
# These are just numbers 1-3
# time = [razer_constants.STARLIGHT_FAST, razer_constants.STARLIGHT_NORMAL,
# razer_constants.STARLIGHT_SLOW]
if len(arg) == 0:
time = util.randint(1, 3)
else:
time = int(arg[0])
if getattr(prop, effect)(*color[c_used],
time) and self.args.verbose:
debug_msg[zone].append([
"Setting", effect, "to:", *color[c_used],
'@ timing level', time
])
elif effect == 'starlight_dual':
# These are just numbers 1-3
# time = [razer_constants.STARLIGHT_FAST, razer_constants.STARLIGHT_NORMAL,
# razer_constants.STARLIGHT_SLOW]
if len(arg) == 0:
time = util.randint(1, 3)
else:
time = int(arg[0])
if getattr(prop,
effect)(*color[c_used], *color[c_used + 1],
time) and self.args.verbose:
debug_msg[zone].append([
"Setting", effect, " @ timing level", time,
"to:\n [", *color[c_used], ']\n [',
*color[c_used + 1]
])
elif effect == 'starlight_random':
if len(arg) == 0:
time = util.randint(1, 3)
else:
time = int(arg[0])
if getattr(prop, effect)(time) and self.args.verbose:
debug_msg[zone].append(
["Setting", effect, '@ timing level', time])
elif effect == 'wave':
# These are just numbers 1-2
# direction=[razer_constants.WAVE_LEFT, razer_constants.WAVE_RIGHT]
if len(arg) == 0:
direction = util.randint(1, 2)
else:
direction = int(arg[0])
if getattr(prop, effect)(direction) and self.args.verbose:
debug_msg[zone].append(
["Setting", effect, ' in direction', direction])
elif effect == 'brightness':
if len(arg) == 0:
b = 100
else:
b = int(arg[0])
prop.brightness = b
if self.args.verbose:
debug_msg[zone].append(["Setting", effect, 'to', b])
elif effect == 'active':
if len(arg) == 0:
b = True
else:
b = bool(arg[0])
prop.active = b
if self.args.verbose:
debug_msg[zone].append(["Setting", effect, 'to', b])
elif effect == 'multicolor':
if len(arg) > 0:
used = int(arg[0])
else:
used = 0
while len(color) < c_used + used:
color.append(util.get_random_color_rgb())
cols = prop.advanced.cols
rows = prop.advanced.rows
if used == 0:
colors_to_dist = [
util.get_random_color_rgb()
for _ in range(cols * rows)
]
else:
colors_to_dist = []
end = len(color)
start = c_used
while start < end:
colors_to_dist.append(color[start])
start += 1
try:
counter = 0
for row in range(rows):
for col in range(cols):
prop.advanced.matrix.set(
row, col,
colors_to_dist[counter %
len(colors_to_dist)])
counter += 1
# prop.advanced.draw_fb_or()
prop.advanced.draw()
if self.args.verbose:
debug_msg[zone].append([
"Setting", effect, "\n ", colors_to_dist
])
except (AssertionError, ValueError) as e:
if self.args.verbose:
debug_msg[zone].append(["Warning:", str(e)])
else:
debug_msg[zone].append([
"Looks like someone forget to program a action for",
effect, "after adding it to the settings file"
])
c_used += used
i += 1
if self.args.verbose:
for i in debug_msg:
print(' ' + i + ':')
for x in debug_msg[i]:
print(' ', *x)
# Save used settings for this device to a file
if og_color_len == 0:
color = []
util.write_settings_to_file(device, effects, color, zones=zones)