From bb94b1b343d9d6f7e25bf819f1890b974c7717e3 Mon Sep 17 00:00:00 2001 From: Gregor Aisch Date: Tue, 9 Apr 2013 00:16:09 +0200 Subject: [PATCH] freeze() now accepts either db-urls or db instances --- dataset/freeze/app.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dataset/freeze/app.py b/dataset/freeze/app.py index dee4404..be6a7d5 100644 --- a/dataset/freeze/app.py +++ b/dataset/freeze/app.py @@ -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 of records) into individual files. """ + if isinstance(database, (str, unicode)): + database = Database(database) kw.update({ 'database': database, 'query': query, @@ -42,7 +44,7 @@ def freeze(database, query, format='csv', filename='freeze.csv', def freeze_export(export): try: - database = Database(export.get('database')) + database = export.get('database') query = database.query(export.get('query')) serializer_cls = get_serializer(export) serializer = serializer_cls(export, query)