cli: Add zero-game import/export messages

1 files changed, 14 insertions, 7 deletions

diff --git a/src/vgstash_cli.py b/src/vgstash_cli.py
index 17fd301..22cfc5a 100644
--- a/src/vgstash_cli.py
+++ b/src/vgstash_cli.py
@@ -38,13 +38,14 @@ def row_format(row, width, header):
     Prints a row from the result set into a nice table.
     """
     # The magic number comes from:
-    #    game ID (4)
-    #    system name (8)
-    #    ownership (3)
-    #    progress (9)
+    #    3 chars per separator (9 chars total)
+    #    8 chars for "System" (so there's an additional space on each side)
+    #    3 chars for "Own"
+    #    9 chars for "Progress" and an additional space
+    #    Total is 29 characters
     twidth = int(width) - 29
     if header == True:
-        click.echo("{:<{w}s} | {:<8s} | {:^3s} | {}".format(
+        click.echo("{:<{w}s} | {:^8s} | {:^3s} | {}".format(
             "Title",
             "System",
             "Own",
@@ -235,7 +236,10 @@ def import_file(format, filepath, update):
                 )
             except sqlite3.IntegrityError as e:
                 count -= 1
-        click.echo("Successfully imported {} games from {}.".format(count, filepath))
+        if count > 0:
+            click.echo("Successfully imported {} games from {}.".format(count, filepath))
+        else:
+            click.echo("Couldn't import any games. Is the YAML file formatted correctly?")
 
 
 @cli.command("export")
@@ -270,5 +274,8 @@ def export_file(format, filepath):
         with open(filepath, "w") as fp:
             yaml.dump(game_set, fp, default_flow_style=False,
                              indent=4, allow_unicode=True)
-        click.echo("Successfully exported {} games to {}.".format(len(game_set), filepath))
+        if len(game_set) > 0:
+            click.echo("Successfully exported {} games to {}.".format(len(game_set), filepath))
+        else:
+            click.echo("Could not export any games; have you made sure your collection has games in it?")