diff --git a/gluon/particleGenerator/DataBaseModelGenerator.py b/gluon/particleGenerator/DataBaseModelGenerator.py index cab9b87..9abea86 100644 --- a/gluon/particleGenerator/DataBaseModelGenerator.py +++ b/gluon/particleGenerator/DataBaseModelGenerator.py @@ -39,7 +39,7 @@ class DataBaseModelProcessor(object): def get_table_class(self, api_name, table_name): try: return self.db_models.get(api_name)[table_name] - except ValueError: + except(TypeError, KeyError): raise Exception('Unknown table name %s' % table_name) def build_sqla_models(self, api_name, base=None):