From 9cb3608b27f893d261d8d29e946bec1ebcf35de7 Mon Sep 17 00:00:00 2001
From: Gabe R <Marsfan@users.noreply.github.com>
Date: Sun, 28 Jun 2020 23:52:52 -0500
Subject: [PATCH] Cleaned up wv_colors.py *Reformatted with autopep8 *renamed
 input->inp to avoid shadowing input() function

---
 src/wireviz/wv_colors.py | 106 ++++++++++++++++++++-------------------
 1 file changed, 55 insertions(+), 51 deletions(-)

diff --git a/src/wireviz/wv_colors.py b/src/wireviz/wv_colors.py
index 32c90c9..90aeb66 100644
--- a/src/wireviz/wv_colors.py
+++ b/src/wireviz/wv_colors.py
@@ -1,74 +1,78 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
 COLOR_CODES = {
-               'DIN': ['WH','BN','GN','YE','GY','PK','BU','RD','BK','VT'], # ,'GYPK','RDBU','WHGN','BNGN','WHYE','YEBN','WHGY','GYBN','WHPK','PKBN'],
-               'IEC': ['BN','RD','OG','YE','GN','BU','VT','GY','WH','BK'],
-               'BW':  ['BK','WH']
-              }
+    # ,'GYPK','RDBU','WHGN','BNGN','WHYE','YEBN','WHGY','GYBN','WHPK','PKBN'],
+    'DIN': ['WH', 'BN', 'GN', 'YE', 'GY', 'PK', 'BU', 'RD', 'BK', 'VT'],
+    'IEC': ['BN', 'RD', 'OG', 'YE', 'GN', 'BU', 'VT', 'GY', 'WH', 'BK'],
+    'BW': ['BK', 'WH'],
+}
 
 color_hex = {
-             'BK': '#000000',
-             'WH': '#ffffff',
-             'GY': '#999999',
-             'PK': '#ff66cc',
-             'RD': '#ff0000',
-             'OG': '#ff8000',
-             'YE': '#ffff00',
-             'GN': '#00ff00',
-             'TQ': '#00ffff',
-             'BU': '#0066ff',
-             'VT': '#8000ff',
-             'BN': '#666600',
-              }
+    'BK': '#000000',
+    'WH': '#ffffff',
+    'GY': '#999999',
+    'PK': '#ff66cc',
+    'RD': '#ff0000',
+    'OG': '#ff8000',
+    'YE': '#ffff00',
+    'GN': '#00ff00',
+    'TQ': '#00ffff',
+    'BU': '#0066ff',
+    'VT': '#8000ff',
+    'BN': '#666600',
+}
 
 color_full = {
-             'BK': 'black',
-             'WH': 'white',
-             'GY': 'grey',
-             'PK': 'pink',
-             'RD': 'red',
-             'OG': 'orange',
-             'YE': 'yellow',
-             'GN': 'green',
-             'TQ': 'turquoise',
-             'BU': 'blue',
-             'VT': 'violet',
-             'BN': 'brown',
+    'BK': 'black',
+    'WH': 'white',
+    'GY': 'grey',
+    'PK': 'pink',
+    'RD': 'red',
+    'OG': 'orange',
+    'YE': 'yellow',
+    'GN': 'green',
+    'TQ': 'turquoise',
+    'BU': 'blue',
+    'VT': 'violet',
+    'BN': 'brown',
 }
 
 color_ger = {
-             'BK': 'sw',
-             'WH': 'ws',
-             'GY': 'gr',
-             'PK': 'rs',
-             'RD': 'rt',
-             'OG': 'or',
-             'YE': 'ge',
-             'GN': 'gn',
-             'TQ': 'tk',
-             'BU': 'bl',
-             'VT': 'vi',
-             'BN': 'br',
+    'BK': 'sw',
+    'WH': 'ws',
+    'GY': 'gr',
+    'PK': 'rs',
+    'RD': 'rt',
+    'OG': 'or',
+    'YE': 'ge',
+    'GN': 'gn',
+    'TQ': 'tk',
+    'BU': 'bl',
+    'VT': 'vi',
+    'BN': 'br',
 }
 
-def translate_color(input, color_mode):
-    if input == '':
+
+def translate_color(inp, color_mode):
+    if inp == '':
         output = ''
     else:
         if color_mode == 'full':
-            output = color_full[input].lower()
+            output = color_full[inp].lower()
         elif color_mode == 'FULL':
-            output = color_full[input].upper()
+            output = color_full[inp].upper()
         elif color_mode == 'hex':
-            output = color_hex[input].lower()
+            output = color_hex[inp].lower()
         elif color_mode == 'HEX':
-            output = color_hex[input].upper()
+            output = color_hex[inp].upper()
         elif color_mode == 'ger':
-            output = color_ger[input].lower()
+            output = color_ger[inp].lower()
         elif color_mode == 'GER':
-            output = color_ger[input].upper()
+            output = color_ger[inp].upper()
         elif color_mode == 'short':
-            output = input.lower()
+            output = inp.lower()
         elif color_mode == 'SHORT':
-            output = input.upper()
+            output = inp.upper()
         else:
             raise Exception('Unknown color mode')
     return output