Allow to use url defined as env variable.

Browse Source
This commit is contained in:
mattack108 2013-07-14 21:50:52 +01:00
parent 6df8870234
commit cf8f3dc4d2
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
dataset/__init__.py
View File

@ -1,3 +1,4 @@
import os
# shut up useless SA warning: # shut up useless SA warning:
import warnings import warnings
warnings.filterwarnings( warnings.filterwarnings(
@ -10,10 +11,11 @@ from dataset.freeze.app import freeze
__all__ = ['Database', 'Table', 'freeze', 'connect'] __all__ = ['Database', 'Table', 'freeze', 'connect']
def connect(url, reflectMetadata=True): def connect(url=None, reflectMetadata=True):
""" """
Opens a new connection to a database. *url* can be any valid `SQLAlchemy engine URL`_. Returns Opens a new connection to a database. *url* can be any valid `SQLAlchemy engine URL`_.
an instance of :py:class:`Database <dataset.Database>`. Set *reflectMetadata* to False if you If *url* is not defined it will try to use *DATABASE_URL* from environment variable.
Returns an instance of :py:class:`Database <dataset.Database>`. Set *reflectMetadata* to False if you
don't want the entire database schema to be pre-loaded. This significantly speeds up don't want the entire database schema to be pre-loaded. This significantly speeds up
connecting to large databases with lots of tables. connecting to large databases with lots of tables.
:: ::
@ -22,4 +24,5 @@ def connect(url, reflectMetadata=True):
.. _SQLAlchemy Engine URL: http://docs.sqlalchemy.org/en/latest/core/engines.html#sqlalchemy.create_engine .. _SQLAlchemy Engine URL: http://docs.sqlalchemy.org/en/latest/core/engines.html#sqlalchemy.create_engine
""" """
url = os.environ.get('DATABASE_URL', url)
return Database(url, reflectMetadata) return Database(url, reflectMetadata)