summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to '')
-rwxr-xr-xsrc/vgstash_cli.py8
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"))