From 284b5946c61cc39d8aa880554ce5ba363285c5fc Mon Sep 17 00:00:00 2001
From: zlg <zlg@zlg.space>
Date: Mon, 2 Mar 2020 21:09:00 -0800
Subject: Avoid a backtrace when updating an invalid game

ZLG-Bug: 2
---
 src/vgstash_cli.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)
 mode change 100644 => 100755 src/vgstash_cli.py

(limited to 'src/vgstash_cli.py')

diff --git a/src/vgstash_cli.py b/src/vgstash_cli.py
old mode 100644
new mode 100755
index b92201f..323f1d9
--- a/src/vgstash_cli.py
+++ b/src/vgstash_cli.py
@@ -158,7 +158,10 @@ def delete_game(title, system):
 def update_game(title, system, attr, val):
     # TODO: Consider namedtuple as a solution
     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.")
     if attr == 'ownership':
         val = vgstash.vtok(val, vgstash.OWNERSHIP)
     if attr == 'progress':
@@ -172,7 +175,6 @@ def update_game(title, system, attr, val):
     )
     if db.update_game(target_game, updated_game):
         click.echo("Updated {} for {}. Its {} is now {}.".format(title, system, attr, val))
-    pass
 
 
 @cli.command('notes')
-- 
cgit v1.2.3-54-g00ecf