return feedback from cli and add basic error check to bcdata2pg

smnorris · Jan 20, 2018 · 6c3ebb4 · 6c3ebb4
1 parent b9eddc5
commit 6c3ebb4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pgdata/cli.py b/pgdata/cli.py
@@ -23,7 +23,8 @@ def cli(dataset, email, db_url):
     """Mirror a DataBC Catalogue dataset in postgres
     """
     db = pgdata.connect(db_url)
-    db.bcdata2pg(dataset, email)
+    info = db.bcdata2pg(dataset, email)
+    click.echo(info['schema']+'.'+info['name'] + ' loaded')
 
 
 if __name__ == '__main__':

diff --git a/pgdata/database.py b/pgdata/database.py
@@ -372,3 +372,9 @@ def bcdata2pg(self, url, email, table_name=None, schema='public',
                     schema=schema,
                     sql=sql,
                     dim=dim)
+
+        # check that all went well
+        if schema+'.'+table in self.tables:
+            return info
+        else:
+            raise IOError(schema+'.'+table+' was not loaded')