adding snapshot function

1ccb2da4 · Guba Sándor · c4589229 · 1ccb2da4
Commit 1ccb2da4 authored Sep 26, 2013 by Guba Sándor
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

storagedriver.py
+8 -2

No files found.
--- a/storagedriver.py
+++ b/storagedriver.py
@@ -20,6 +20,12 @@ def delete_disk(json_data):


 @celery.task()
-def get_disk(json_data):
+def snapshot(json_data):
    disk = Disk.import_from_json(json_data)
-    return jsonpickle.encode(disk, unpicklable=False)
+    disk.snapshot()
+
+
+@celery.task()
+def get_disk(json_data):
+    disk = Disk.get(dir=json_data['dir'], name=json_data['name'])
+    return disk.get_desc()