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

src: wv_bom: simplify partnumbers2list

Browse Source
This commit is contained in:
Laurier Loiselle 2023-03-27 15:31:48 -04:00
parent 8343f62f41
commit 850695409f
No known key found for this signature in database
GPG Key ID: 345920CC72089A3F
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/wireviz/wv_bom.py
View File

@ -20,18 +20,19 @@ def partnumbers2list(
partnumbers: PartNumberInfo, parent_partnumbers: PartNumberInfo = None partnumbers: PartNumberInfo, parent_partnumbers: PartNumberInfo = None
) -> List[str]: ) -> List[str]:
if not isinstance(partnumbers, list): if not isinstance(partnumbers, list):
return partnumbers.str_list partnumbers = [partnumbers]
pn = None
for p in partnumbers:
if pn is None:
pn = p.copy()
pn = pn.keep_only_eq(p)
if pn.str_list is None and parent_partnumbers is not None: # if there's no parent, fold
return parent_partnumbers.str_list if parent_partnumbers is None:
else: return PartNumberInfo.list_keep_only_eq(partnumbers).str_list
return pn.str_list
if isinstance(parent_partnumbers, list):
parent_partnumbers = PartNumberInfo.list_keep_only_eq(parent_partnumbers)
partnumbers = [p.remove_eq(parent_partnumbers) for p in partnumbers]
return [p.str_list for p in partnumbers if p]
def bom_list(bom): def bom_list(bom):