kforney/hyfetch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[+] Encapsulate set_light and lighten

Browse source
This commit is contained in:
Azalea (on HyDEV-Daisy) 2022-06-19 17:37:04 -04:00
parent 1c67333b45
commit b46d0e96f4
2 changed files with 26 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
hyfetch/main.py
View file

@ -118,13 +118,15 @@ def create_config() -> Config:
##############################
# 2. Select light/dark mode
light_dark = literal_input(f'1. Is your terminal in &gf(#85e7e9)light mode&r or &gf(#c471ed)dark mode&r?',
clear_screen(title)
light_dark = literal_input(f'2. Is your terminal in &gf(#85e7e9)light mode&r or &gf(#c471ed)dark mode&r?',
['light', 'dark'], 'dark')
##############################
# 3. Choose preset
clear_screen(title)
print('2. Let\'s choose a flag! Available flags:\n')
print('3. Let\'s choose a flag!\n'
'Available flags:\n')
# Create flags = [[lines]]
flags = []
@ -147,7 +149,7 @@ def create_config() -> Config:
print()
tmp = PRESETS['rainbow'].color_text('preset')
preset = literal_input(f'Which {tmp} do you want to use?', PRESETS.keys(), 'rainbow')
preset = literal_input(f'Which {tmp} do you want to use?', PRESETS.keys(), 'rainbow', show_ops=False)
# Create config
c = Config(preset, color_system, light_dark)
@ -201,9 +203,9 @@ def run():
# Lighten
if args.scale:
preset = ColorProfile([c.lighten(args.scale) for c in preset.colors])
preset = preset.lighten(args.scale)
if args.light:
preset = ColorProfile([c.set_light(args.light) for c in preset.colors])
preset = preset.set_light(args.light)
# Test distro ascii art
if args.test_distro:

21
hyfetch/presets.py
View file

@ -17,7 +17,6 @@ class ColorProfile:
else:
self.colors = colors
def with_weights(self, weights: list[int]) -> list[RGB]:
"""
Map colors based on weights
@ -76,11 +75,29 @@ class ColorProfile:
result += '\033[0m'
result += t
else:
result += colors[i].to_ansi_rgb(foreground) + t
result += colors[i].to_ansi(foreground=foreground) + t
result += '\033[0m'
return result
def lighten(self, multiplier: float) -> ColorProfile:
"""
Lighten the color profile by a multiplier
:param multiplier: Multiplier
:return: Lightened color profile (original isn't modified)
"""
return ColorProfile([c.lighten(multiplier) for c in self.colors])
def set_light(self, light: int):
"""
Set HSL lightness value
:param light: Lightness value
:return: New color profile (original isn't modified)
"""
return ColorProfile([c.set_light(light) for c in self.colors])
PRESETS: dict[str, ColorProfile] = {
'rainbow': ColorProfile([