From 6342e1479d6390a3101e715ff67c4983753cc7db Mon Sep 17 00:00:00 2001 From: zlg Date: Mon, 2 Mar 2020 21:32:45 -0800 Subject: Avoid a backtrace when editing notes on an invalid game ZLG-Bug: 4 --- src/vgstash_cli.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'src') diff --git a/src/vgstash_cli.py b/src/vgstash_cli.py index 323f1d9..d00f339 100755 --- a/src/vgstash_cli.py +++ b/src/vgstash_cli.py @@ -162,6 +162,7 @@ def update_game(title, system, attr, val): target_game = db.get_game(title, system) except: click.echo("Game not found. Please try again.") + return if attr == 'ownership': val = vgstash.vtok(val, vgstash.OWNERSHIP) if attr == 'progress': @@ -183,7 +184,12 @@ def update_game(title, system, attr, val): @click.option('--edit', '-e', is_flag=True, default=False) def notes(title, system, edit): db = get_db() - target_game = db.get_game(title, system) + try: + target_game = db.get_game(title, system) + except: + click.echo("Game not found. Please try again.") + return + if edit: with tempfile.NamedTemporaryFile() as tmpfile: tmpfile.write(target_game.notes.encode("UTF-8")) -- cgit v1.2.3-54-g00ecf