3cc36e1573
- Use `#!/usr/bin/env python3` on executable modules - Remove shebangs on all other `.py` files - Add UTF-8 encoding info where mising
185 lines
6.7 KiB
Python
185 lines
6.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Dict, List
|
|
|
|
COLOR_CODES = {
|
|
'DIN': ['WH', 'BN', 'GN', 'YE', 'GY', 'PK', 'BU', 'RD', 'BK', 'VT', 'GYPK', 'RDBU', 'WHGN', 'BNGN', 'WHYE', 'YEBN',
|
|
'WHGY', 'GYBN', 'WHPK', 'PKBN', 'WHBU', 'BNBU', 'WHRD', 'BNRD', 'WHBK', 'BNBK', 'GYGN', 'YEGY', 'PKGN',
|
|
'YEPK', 'GNBU', 'YEBU', 'GNRD', 'YERD', 'GNBK', 'YEBK', 'GYBU', 'PKBU', 'GYRD', 'PKRD', 'GYBK', 'PKBK',
|
|
'BUBK', 'RDBK', 'WHBNBK', 'YEGNBK', 'GYPKBK', 'RDBUBK', 'WHGNBK', 'BNGNBK', 'WHYEBK', 'YEBNBK', 'WHGYBK',
|
|
'GYBNBK', 'WHPKBK', 'PKBNBK', 'WHBUBK', 'BNBUBK', 'WHRDBK', 'BNRDBK'],
|
|
'IEC': ['BN', 'RD', 'OG', 'YE', 'GN', 'BU', 'VT', 'GY', 'WH', 'BK'],
|
|
'BW': ['BK', 'WH'],
|
|
# 25-pair color code - see also https://en.wikipedia.org/wiki/25-pair_color_code
|
|
# 5 major colors (WH,RD,BK,YE,VT) combined with 5 minor colors (BU,OG,GN,BN,SL).
|
|
# Each POTS pair tip (+) had major/minor color, and ring (-) had minor/major color.
|
|
'TEL': [ # 25x2: Ring and then tip of each pair
|
|
'BUWH', 'WHBU', 'OGWH', 'WHOG', 'GNWH', 'WHGN', 'BNWH', 'WHBN', 'SLWH', 'WHSL',
|
|
'BURD', 'RDBU', 'OGRD', 'RDOG', 'GNRD', 'RDGN', 'BNRD', 'RDBN', 'SLRD', 'RDSL',
|
|
'BUBK', 'BKBU', 'OGBK', 'BKOG', 'GNBK', 'BKGN', 'BNBK', 'BKBN', 'SLBK', 'BKSL',
|
|
'BUYE', 'YEBU', 'OGYE', 'YEOG', 'GNYE', 'YEGN', 'BNYE', 'YEBN', 'SLYE', 'YESL',
|
|
'BUVT', 'VTBU', 'OGVT', 'VTOG', 'GNVT', 'VTGN', 'BNVT', 'VTBN', 'SLVT', 'VTSL'],
|
|
'TELALT': [ # 25x2: Tip and then ring of each pair
|
|
'WHBU', 'BU', 'WHOG', 'OG', 'WHGN', 'GN', 'WHBN', 'BN', 'WHSL', 'SL',
|
|
'RDBU', 'BURD', 'RDOG', 'OGRD', 'RDGN', 'GNRD', 'RDBN', 'BNRD', 'RDSL', 'SLRD',
|
|
'BKBU', 'BUBK', 'BKOG', 'OGBK', 'BKGN', 'GNBK', 'BKBN', 'BNBK', 'BKSL', 'SLBK',
|
|
'YEBU', 'BUYE', 'YEOG', 'OGYE', 'YEGN', 'GNYE', 'YEBN', 'BNYE', 'YESL', 'SLYE',
|
|
'VTBU', 'BUVT', 'VTOG', 'OGVT', 'VTGN', 'GNVT', 'VTBN', 'BNVT', 'VTSL', 'SLVT'],
|
|
'T568A': ['WHGN', 'GN', 'WHOG', 'BU', 'WHBU', 'OG', 'WHBN', 'BN'],
|
|
'T568B': ['WHOG', 'OG', 'WHGN', 'BU', 'WHBU', 'GN', 'WHBN', 'BN'],
|
|
}
|
|
|
|
# Convention: Color names should be 2 letters long, to allow for multicolored wires
|
|
|
|
_color_hex = {
|
|
'BK': '#000000',
|
|
'WH': '#ffffff',
|
|
'GY': '#999999',
|
|
'PK': '#ff66cc',
|
|
'RD': '#ff0000',
|
|
'OG': '#ff8000',
|
|
'YE': '#ffff00',
|
|
'OL': '#708000', # olive green
|
|
'GN': '#00ff00',
|
|
'TQ': '#00ffff',
|
|
'LB': '#a0dfff', # light blue
|
|
'BU': '#0066ff',
|
|
'VT': '#8000ff',
|
|
'BN': '#895956',
|
|
'BG': '#ceb673', # beige
|
|
'IV': '#f5f0d0', # ivory
|
|
'SL': '#708090',
|
|
'CU': '#d6775e', # Faux-copper look, for bare CU wire
|
|
'SN': '#aaaaaa', # Silvery look for tinned bare wire
|
|
'SR': '#84878c', # Darker silver for silvered wire
|
|
'GD': '#ffcf80', # Golden color for gold
|
|
}
|
|
|
|
_color_full = {
|
|
'BK': 'black',
|
|
'WH': 'white',
|
|
'GY': 'grey',
|
|
'PK': 'pink',
|
|
'RD': 'red',
|
|
'OG': 'orange',
|
|
'YE': 'yellow',
|
|
'OL': 'olive green',
|
|
'GN': 'green',
|
|
'TQ': 'turquoise',
|
|
'LB': 'light blue',
|
|
'BU': 'blue',
|
|
'VT': 'violet',
|
|
'BN': 'brown',
|
|
'BG': 'beige',
|
|
'IV': 'ivory',
|
|
'SL': 'slate',
|
|
'CU': 'copper',
|
|
'SN': 'tin',
|
|
'SR': 'silver',
|
|
'GD': 'gold',
|
|
}
|
|
|
|
_color_ger = {
|
|
'BK': 'sw',
|
|
'WH': 'ws',
|
|
'GY': 'gr',
|
|
'PK': 'rs',
|
|
'RD': 'rt',
|
|
'OG': 'or',
|
|
'YE': 'ge',
|
|
'OL': 'ol', # olivgrün
|
|
'GN': 'gn',
|
|
'TQ': 'tk',
|
|
'LB': 'hb', # hellblau
|
|
'BU': 'bl',
|
|
'VT': 'vi',
|
|
'BN': 'br',
|
|
'BG': 'bg', # beige
|
|
'IV': 'eb', # elfenbeinfarben
|
|
'SL': 'si', # Schiefer
|
|
'CU': 'ku', # Kupfer
|
|
'SN': 'vz', # verzinkt
|
|
'SR': 'ag', # Silber
|
|
'GD': 'au', # Gold
|
|
}
|
|
|
|
|
|
color_default = '#ffffff'
|
|
|
|
_hex_digits = set('0123456789abcdefABCDEF')
|
|
|
|
|
|
# Literal type aliases below are commented to avoid requiring python 3.8
|
|
Color = str # Two-letter color name = Literal[_color_hex.keys()]
|
|
Colors = str # One or more two-letter color names (Color) concatenated into one string
|
|
ColorMode = str # = Literal['full', 'FULL', 'hex', 'HEX', 'short', 'SHORT', 'ger', 'GER']
|
|
ColorScheme = str # Color scheme name = Literal[COLOR_CODES.keys()]
|
|
|
|
|
|
def get_color_hex(input: Colors, pad: bool = False) -> List[str]:
|
|
"""Return list of hex colors from either a string of color names or :-separated hex colors."""
|
|
if input is None or input == '':
|
|
return [color_default]
|
|
elif input[0] == '#': # Hex color(s)
|
|
output = input.split(':')
|
|
for i, c in enumerate(output):
|
|
if c[0] != '#' or not all(d in _hex_digits for d in c[1:]):
|
|
if c != input:
|
|
c += f' in input: {input}'
|
|
print(f'Invalid hex color: {c}')
|
|
output[i] = color_default
|
|
else: # Color name(s)
|
|
def lookup(c: str) -> str:
|
|
try:
|
|
return _color_hex[c]
|
|
except KeyError:
|
|
if c != input:
|
|
c += f' in input: {input}'
|
|
print(f'Unknown color name: {c}')
|
|
return color_default
|
|
|
|
output = [lookup(input[i:i + 2]) for i in range(0, len(input), 2)]
|
|
|
|
if len(output) == 2: # Give wires with EXACTLY 2 colors that striped look.
|
|
output += output[:1]
|
|
elif pad and len(output) == 1: # Hacky style fix: Give single color wires
|
|
output *= 3 # a triple-up so that wires are the same size.
|
|
|
|
return output
|
|
|
|
|
|
def get_color_translation(translate: Dict[Color, str], input: Colors) -> List[str]:
|
|
"""Return list of colors translations from either a string of color names or :-separated hex colors."""
|
|
def from_hex(hex_input: str) -> str:
|
|
for color, hex in _color_hex.items():
|
|
if hex == hex_input:
|
|
return translate[color]
|
|
return f'({",".join(str(int(hex_input[i:i+2], 16)) for i in range(1, 6, 2))})'
|
|
|
|
return [from_hex(h) for h in input.lower().split(':')] if input[0] == '#' else \
|
|
[translate.get(input[i:i+2], '??') for i in range(0, len(input), 2)]
|
|
|
|
|
|
def translate_color(input: Colors, color_mode: ColorMode) -> str:
|
|
if input == '' or input is None:
|
|
return ''
|
|
upper = color_mode.isupper()
|
|
if not (color_mode.isupper() or color_mode.islower()):
|
|
raise Exception('Unknown color mode capitalization')
|
|
|
|
color_mode = color_mode.lower()
|
|
if color_mode == 'full':
|
|
output = "/".join(get_color_translation(_color_full, input))
|
|
elif color_mode == 'hex':
|
|
output = ':'.join(get_color_hex(input, pad=False))
|
|
elif color_mode == 'ger':
|
|
output = "".join(get_color_translation(_color_ger, input))
|
|
elif color_mode == 'short':
|
|
output = input
|
|
else:
|
|
raise Exception('Unknown color mode')
|
|
if upper:
|
|
return output.upper()
|
|
else:
|
|
return output.lower()
|