BioArchLinux/nvchecker
2
0
Fork 0
mirror of https://github.com/lilydjwg/nvchecker.git synced 2025-03-10 06:14:02 +00:00
Code Issues Projects Releases Packages Wiki Activity

Replace delta if/elif/elif... with a dictionary lookup, and use "equal" to avoid awkward delta=None case

Browse source
This commit is contained in:
Tom Rathborne 2022-02-11 11:13:11 +01:00
parent 6c527a8a7a
commit e75e2cd3e5
1 changed files with 26 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
nvchecker/tools.py
View file

@ -139,6 +139,7 @@ def cmp() -> None:
differences.append(diff) differences.append(diff)
elif args.all: elif args.all:
diff['delta'] = 'equal'
differences.append(diff) differences.append(diff)
if args.json: if args.json:
@ -154,23 +155,30 @@ def cmp() -> None:
from .lib.nicelogger import Colors, support_color from .lib.nicelogger import Colors, support_color
c = Colors(support_color(sys.stdout)) c = Colors(support_color(sys.stdout))
diffstyles = {
'new': {
'symbol': '->',
'oldc': c.red
},
'old': {
'symbol': f'{c.red}<-{c.normal}',
'oldc': c.red
},
'added': {
'symbol': '++',
'oldc': c.red
},
'gone': {
'symbol': f'{c.red}--{c.normal}',
'oldc': c.green
},
'equal': {
'symbol': '==',
'oldc': c.green
}
}
for diff in differences: for diff in differences:
delta = diff.get('delta', None) style = diffstyles[diff['delta']]
if delta == 'new': print(f'{diff["name"]} {style["oldc"]}{diff["oldver"]}{c.normal} {style["symbol"]} {c.green}{diff["newver"]}{c.normal}')
arrow = '->'
oldc = c.red
elif delta == 'old':
arrow = f'{c.red}<-{c.normal}'
oldc = c.red
if delta == 'added':
arrow = '++'
oldc = c.red
elif delta == 'gone':
arrow = f'{c.red}--{c.normal}'
oldc = c.green
else:
arrow = '=='
oldc = c.green
print(f'{diff["name"]} {oldc}{diff["oldver"]}{c.normal} {arrow} {c.green}{diff["newver"]}{c.normal}')