warehack.ing/WireViz
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Extend unknown color error message (#147)

Browse Source
This commit is contained in:
Marc Brückner 2020-08-10 20:13:13 +02:00 committed by GitHub
parent 4802705427
commit 84edda64df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/wireviz/wv_colors.py
View File

@ -111,15 +111,18 @@ color_default = '#ffffff'
def get_color_hex(input, pad=False): def get_color_hex(input, pad=False):
if input is None or input == '': if input is None or input == '':
return [color_default] return [color_default]
if len(input) == 4: # give wires with EXACTLY 2 colors that striped/banded look if len(input) == 4: # give wires with EXACTLY 2 colors that striped/banded look
input = input + input[:2] padded = input + input[:2]
# hacky style fix: give single color wires a triple-up so that wires are the same size elif pad and len(input) == 2: # hacky style fix: give single color wires a triple-up so that wires are the same size
if pad and len(input) == 2: padded = input + input + input
input = input + input + input else:
padded = input
try: try:
output = [_color_hex[input[i:i + 2]] for i in range(0, len(input), 2)] output = [_color_hex[padded[i:i + 2]] for i in range(0, len(input), 2)]
except KeyError: except KeyError:
print("Unknown color specified") print(f'Unknown color specified: {input}')
output = [color_default] output = [color_default]
return output return output