Преглед на файлове

[notebook] Add a notebook and editor type

Romain Rigaux преди 10 години
родител
ревизия
0c22de3

+ 3 - 2
desktop/libs/notebook/src/notebook/api.py

@@ -202,12 +202,13 @@ def save_notebook(request):
   response = {'status': -1}
 
   notebook = json.loads(request.POST.get('notebook', '{}'))
+  notebook_type = notebook.get('type', 'notebook')
 
   if notebook.get('id'):
     notebook_doc = Document2.objects.get(id=notebook['id'])
   else:
-    notebook_doc = Document2.objects.create(name=notebook['name'], type='notebook', owner=request.user)
-    Document.objects.link(notebook_doc, owner=notebook_doc.owner, name=notebook_doc.name, description=notebook_doc.description, extra='notebook')
+    notebook_doc = Document2.objects.create(name=notebook['name'], type=notebook_type, owner=request.user)
+    Document.objects.link(notebook_doc, owner=notebook_doc.owner, name=notebook_doc.name, description=notebook_doc.description, extra=notebook_type)
 
   notebook_doc1 = notebook_doc.doc.get()
   notebook_doc.update_data(notebook)

+ 0 - 1
desktop/libs/notebook/src/notebook/connectors/base.py

@@ -59,7 +59,6 @@ class Notebook(object):
           'name': 'My Notebook',
           'description': '',
           'snippets': [],
-          'type': 'notebook'
       })
 
   def get_json(self):

+ 2 - 2
desktop/libs/notebook/src/notebook/views.py

@@ -78,8 +78,8 @@ def editor(request):
   else:
     editor = Notebook()
     data = editor.get_data()
-    data['name'] = '%s Query' % editor_type.title()
-    data['type'] = 'query'    
+    data['name'] = '%s Query' % editor_type.title()    
+    data['type'] = editor_type
     editor.data = json.dumps(data)
 
   autocomplete_base_url = ''