Cleaned up wv_helper.py

* Formatted with autopep8
* Removed unused import typing.Any
* Adding Coding Magic
* Renamed argument input->inp to avoid shadowing python builtin
* Removed unnecessary dummy variable creation
This commit is contained in:
Gabe R 2020-06-28 23:57:55 -05:00
parent 9cb3608b27
commit 6a68790653
No known key found for this signature in database
GPG Key ID: F96D83D8D7ED67D0

View File

@ -1,33 +1,37 @@
from typing import Any, List #!/usr/bin/env python3
# -*- coding: utf-8 -*-
from typing import List
def awg_equiv(mm2): def awg_equiv(mm2):
awg_equiv_table = { awg_equiv_table = {
'0.09': 28, '0.09': 28,
'0.14': 26, '0.14': 26,
'0.25': 24, '0.25': 24,
'0.34': 22, '0.34': 22,
'0.5': 21, '0.5': 21,
'0.75': 20, '0.75': 20,
'1': 18, '1': 18,
'1.5': 16, '1.5': 16,
'2.5': 14, '2.5': 14,
'4': 12, '4': 12,
'6': 10, '6': 10,
'10': 8, '10': 8,
'16': 6, '16': 6,
'25': 4, '25': 4,
'35': 2, '35': 2,
'50': 1, '50': 1,
} }
k = str(mm2) k = str(mm2)
if k in awg_equiv_table: if k in awg_equiv_table:
return awg_equiv_table[k] return awg_equiv_table[k]
else: else:
return 'unknown' return 'unknown'
def nested(input):
def nested(inp):
l = [] l = []
for x in input: for x in inp:
if isinstance(x, list): if isinstance(x, list):
if len(x) > 0: if len(x) > 0:
n = nested(x) n = nested(x)
@ -37,25 +41,26 @@ def nested(input):
if x is not None: if x is not None:
if x != '': if x != '':
l.append(str(x)) l.append(str(x))
s = '|'.join(l) return '|'.join(l)
return s
def int2tuple(input):
if isinstance(input, tuple): def int2tuple(inp):
output = input if isinstance(inp, tuple):
output = inp
else: else:
output = (input,) output = (inp,)
return output return output
def flatten2d(input):
output = [[str(item) if not isinstance(item, List) else ', '.join(item) for item in row] for row in input]
return output
def tuplelist2tsv(input, header=None): def flatten2d(inp):
return [[str(item) if not isinstance(item, List) else ', '.join(item) for item in row] for row in inp]
def tuplelist2tsv(inp, header=None):
output = '' output = ''
if header is not None: if header is not None:
input.insert(0, header) inp.insert(0, header)
input = flatten2d(input) inp = flatten2d(inp)
for row in input: for row in inp:
output = output + '\t'.join(str(item) for item in row) + '\n' output = output + '\t'.join(str(item) for item in row) + '\n'
return output return output