cli: add 'update' command

Two helper functions were also added to the vgstash package to ease
client workflows.

This commit marks the final core function necessary to manipulate a
vgstash DB on the command line.

1 files changed, 27 insertions, 2 deletions

diff --git a/src/vgstash_cli.py b/src/vgstash_cli.py
index f88187c..29c1de8 100644
--- a/src/vgstash_cli.py
+++ b/src/vgstash_cli.py
@@ -40,7 +40,7 @@ def row_format(row, width, header):
     #    progress (9)
     twidth = int(width) - 29
     if header == True:
-        click.echo("{:<{w}s} | {:<8s} | {:^3s} | {:<9s}".format(
+        click.echo("{:<{w}s} | {:<8s} | {:^3s} | {}".format(
             "Title",
             "System",
             "Own",
@@ -61,7 +61,7 @@ def row_format(row, width, header):
         3: 'B',
         4: 'C'
     }
-    progstr = "{: <7s}".format((" " * row['progress'] * 2) + progltr[row['progress']])
+    progstr = "{}".format((" " * (row['progress'] - 1) * 2) + progltr[row['progress']])
     print(" | ".join((titlestr, systemstr, ownstr, progstr)))
 
 
@@ -139,3 +139,28 @@ def delete_game(title, system):
     target_game = vgstash.Game(title, system)
     if db.delete_game(target_game):
         click.echo("Removed {} for {} from your collection.".format(title, system))
+
+
+@cli.command('update')
+@click.argument('title', required=True)
+@click.argument('system', required=True)
+@click.argument('attr', type=click.Choice(['title', 'system', 'ownership', 'progress']), required=True)
+@click.argument('val', required=True)
+def update_game(title, system, attr, val):
+    # TODO: Consider namedtuple as a solution
+    db = get_db()
+    target_game = db.get_game(title, system)
+    if attr == 'ownership':
+        val = vgstash.vtok(val, vgstash.OWNERSHIP)
+    if attr == 'progress':
+        val = vgstash.vtok(val, vgstash.PROGRESS)
+    updated_game = vgstash.Game(
+        val if attr == 'title' else target_game.title,
+        val if attr == 'system' else target_game.system,
+        val if attr == 'ownership' else target_game.ownership,
+        val if attr == 'progress' else target_game.progress,
+        target_game.notes
+    )
+    if db.update_game(target_game, updated_game):
+        click.echo("Updated {} for {}. Its {} is now {}.".format(title, system, attr, val))
+    pass