freeze() now accepts either db-urls or db instances

Browse Source
This commit is contained in:
Gregor Aisch 2013-04-09 00:16:09 +02:00
parent fb18961e32
commit bb94b1b343
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dataset/freeze/app.py
View File

@ -26,6 +26,8 @@ def freeze(database, query, format='csv', filename='freeze.csv',
assignment, and file name templating to dump each record (or a set assignment, and file name templating to dump each record (or a set
of records) into individual files. of records) into individual files.
""" """
if isinstance(database, (str, unicode)):
database = Database(database)
kw.update({ kw.update({
'database': database, 'database': database,
'query': query, 'query': query,
@ -42,7 +44,7 @@ def freeze(database, query, format='csv', filename='freeze.csv',
def freeze_export(export): def freeze_export(export):
try: try:
database = Database(export.get('database')) database = export.get('database')
query = database.query(export.get('query')) query = database.query(export.get('query'))
serializer_cls = get_serializer(export) serializer_cls = get_serializer(export)
serializer = serializer_cls(export, query) serializer = serializer_cls(export, query)