moved str to Database conversion to freeze_export

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

4
dataset/freeze/app.py
View File

@ -26,8 +26,6 @@ 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,
@ -45,6 +43,8 @@ def freeze(database, query, format='csv', filename='freeze.csv',
def freeze_export(export): def freeze_export(export):
try: try:
database = export.get('database') database = export.get('database')
if isinstance(database, (str, unicode)):
database = Database(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)