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.

2 files changed, 65 insertions, 18 deletions

diff --git a/src/vgstash/__init__.py b/src/vgstash/__init__.py
index 71b5981..ee07c78 100755
--- a/src/vgstash/__init__.py
+++ b/src/vgstash/__init__.py
@@ -79,6 +79,22 @@ def kvmatch(arg, dict_map, fallback):
     return ret
 
 
+def vtok(arg, dict_map):
+    """
+    Match an integer value to a key name in the mapping dictionary. Returns a
+    string (the key name) if found, False if not found.
+    """
+    if isinstance(arg, int):
+        for k, v in dict_map.items():
+            if arg == v:
+                return k
+    elif isinstance(arg, str):
+        for k, v in dict_map.items():
+            if k.startswith(arg):
+                return k
+        return False
+
+
 class DB(object):
     """
     The central class of vgstash. It handles everything relating to storing the
@@ -208,6 +224,15 @@ class DB(object):
         else:
             return False
 
+    def get_game(self, title, system):
+        """
+        Fetches a game's information from the database. Returns a Game object.
+        """
+        stmt = "SELECT * FROM games WHERE title=? AND system=?"
+        res = self.conn.execute(stmt, (title, system)).fetchone()
+        if bool(res):
+            return Game(res['title'], res['system'], res['ownership'], res['progress'], res['notes'])
+
     def has_game(self, game, fuzzy=False):
         """
         Returns whether or not the game is in the database.
diff --git a/src/vgstash/test_vgstash_cli.py b/src/vgstash/test_vgstash_cli.py
index cbbd151..5ba7d30 100644
--- a/src/vgstash/test_vgstash_cli.py
+++ b/src/vgstash/test_vgstash_cli.py
@@ -98,13 +98,13 @@ def test_list_pretty():
         print(result.output)
     assert result.exit_code == 0
     assert result.output == '\n'.join((
-        'Title                                               | System   | Own | Progress ',
+        'Title                                               | System   | Own | Progress',
         '--------------------------------------------------------------------------------',
-        'Sonic the Hedgehog 2                                | Genesis  |     |       B',
-        'Vectorman                                           | Genesis  |     |       B',
-        'Vectorman 2                                         | Genesis  | P   |     P  ',
-        'Super Mario Bros.                                   |   NES    | P   |     P  ',
-        'The Legend of Zelda                                 |   NES    |   D |     P  \n',
+        'Sonic the Hedgehog 2                                | Genesis  |     |     B',
+        'Vectorman                                           | Genesis  |     |     B',
+        'Vectorman 2                                         | Genesis  | P   |   P',
+        'Super Mario Bros.                                   |   NES    | P   |   P',
+        'The Legend of Zelda                                 |   NES    |   D |   P\n',
     ))
 
 
@@ -116,13 +116,13 @@ def test_list_pretty_smaller():
         print(result.output)
     assert result.exit_code == 0
     assert result.output == '\n'.join((
-        'Title                           | System   | Own | Progress ',
+        'Title                           | System   | Own | Progress',
         '------------------------------------------------------------',
-        'Sonic the Hedgehog 2            | Genesis  |     |       B',
-        'Vectorman                       | Genesis  |     |       B',
-        'Vectorman 2                     | Genesis  | P   |     P  ',
-        'Super Mario Bros.               |   NES    | P   |     P  ',
-        'The Legend of Zelda             |   NES    |   D |     P  \n'
+        'Sonic the Hedgehog 2            | Genesis  |     |     B',
+        'Vectorman                       | Genesis  |     |     B',
+        'Vectorman 2                     | Genesis  | P   |   P',
+        'Super Mario Bros.               |   NES    | P   |   P',
+        'The Legend of Zelda             |   NES    |   D |   P\n'
     ))
 
 
@@ -134,13 +134,13 @@ def test_list_pretty_tiny():
         print(result.output)
     assert result.exit_code == 0
     assert result.output == '\n'.join((
-        'Title                 | System   | Own | Progress ',
+        'Title                 | System   | Own | Progress',
         '--------------------------------------------------',
-        'Sonic the Hedgehog 2  | Genesis  |     |       B',
-        'Vectorman             | Genesis  |     |       B',
-        'Vectorman 2           | Genesis  | P   |     P  ',
-        'Super Mario Bros.     |   NES    | P   |     P  ',
-        'The Legend of Zelda   |   NES    |   D |     P  \n'
+        'Sonic the Hedgehog 2  | Genesis  |     |     B',
+        'Vectorman             | Genesis  |     |     B',
+        'Vectorman 2           | Genesis  | P   |   P',
+        'Super Mario Bros.     |   NES    | P   |   P',
+        'The Legend of Zelda   |   NES    |   D |   P\n'
     ))
 
 
@@ -151,3 +151,25 @@ def test_delete():
         print(result.output)
     assert result.exit_code == 0
     assert result.output == "Removed Vectorman 2 for Genesis from your collection.\n"
+
+
+def test_update():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['update', 'Super Mario Bros.', 'NES', 'progress', 'c'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == 'Updated Super Mario Bros. for NES. Its progress is now complete.\n'
+
+    list_result = runner.invoke(vgstash_cli.cli, ['list', '-w', '40'])
+    if verbose:
+        print(list_result.output)
+    assert list_result.exit_code == 0
+    assert list_result.output == "\n".join((
+        'Title       | System   | Own | Progress',
+        '----------------------------------------',
+        'Sonic the H | Genesis  |     |     B',
+        'Vectorman   | Genesis  |     |     B',
+        'Super Mario |   NES    | P   |       C',
+        'The Legend  |   NES    |   D |   P\n'
+    ))