diff options
| author | zlg <zlg@zlg.space> | 2020-03-02 21:32:45 -0800 |
|---|---|---|
| committer | zlg <zlg@zlg.space> | 2020-03-02 21:32:45 -0800 |
| commit | 6342e1479d6390a3101e715ff67c4983753cc7db (patch) | |
| tree | bc258667e14971b3c464843044e98f5b5c4cb21b /src/vgstash_cli.py | |
| parent | Avoid a backtrace when updating an invalid game (diff) | |
| download | vgstash-6342e1479d6390a3101e715ff67c4983753cc7db.tar.gz vgstash-6342e1479d6390a3101e715ff67c4983753cc7db.tar.bz2 vgstash-6342e1479d6390a3101e715ff67c4983753cc7db.tar.xz vgstash-6342e1479d6390a3101e715ff67c4983753cc7db.zip | |
Avoid a backtrace when editing notes on an invalid game
ZLG-Bug: 4
Diffstat (limited to 'src/vgstash_cli.py')
| -rwxr-xr-x | src/vgstash_cli.py | 8 |
1 files changed, 7 insertions, 1 deletions
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")) |