datasette_connectors/connectors.py

   1 import pkg_resources
   2
   3 db_connectors = {}
   4
   5 def load():
   6     for entry_point in pkg_resources.iter_entry_points('datasette.connectors'):
   7         db_connectors[entry_point.name] = entry_point.load()
   8
   9 def inspect(path):
  10     for connector in db_connectors.values():
  11         try:
  12             return connector.inspect(path)
  13         except:
  14             pass
  15     else:
  16         raise Exception("No database connector found for %s" % path)
  17
  18 def connect(path, dbtype):
  19     try:
  20         return db_connectors[dbtype].Connection(path)
  21     except:
  22         raise Exception("No database connector found for %s" % path)