Making code robust w/ respect to non-existing tables.

3-doc-enricher.py

@@ -8,26 +8,24 @@ from utils.exit_gracefully import exit_gracefully
 from utils.my_logging import logging
 from utils.postgis_helper import Remote
 from utils.serializers import encode_datetime
+from sqlalchemy.exc import NoSuchTableError
 
 
 def get_entries_from_postgis( link, cfg, no_features_per_page=1000 ):
-    #print('here', link)
 
     dbname = link['url'].split('/')[-1]
-    schema, table = link['name'].split('.')
-    print(dbname, schema, table)
-    #print(cfg)
-    #exit(1)
+    schema, table_name = link['name'].split('.')
 
     logging.info('Getting data from database %s...' % dbname)
     logging.info('Establishing a database connection...')
     pg = Remote(hostname=cfg['host'], dbname=dbname, username=cfg['username'], password=cfg['password'])
     logging.info('Done.')
 
-    table = pg.get_table(table, schema=schema)
-
-
-    #print(pg.get_tables('bruit'))
+    try:
+        table = pg.get_table(table_name, schema=schema)
+    except NoSuchTableError:
+        logging.debug('Table %s in schema % s not found :-(' % (table_name, schema))
+        return
 
     count = pg.count_entries(table)
 
@@ -313,6 +311,7 @@ if __name__ == '__main__':
     with open("config.yaml", 'r') as yamlfile:
         cfg = yaml.load(yamlfile)
 
+    main(cfg)
     while True:
         try:
             main(cfg)