瀏覽代碼

HUE-2520 [pig] Enable a user to save an unsaved pig script after execution

Luca Natali 11 年之前
父節點
當前提交
cba7cf4
共有 2 個文件被更改,包括 8 次插入0 次删除
  1. 4 0
      apps/pig/src/pig/models.py
  2. 4 0
      desktop/core/src/desktop/models.py

+ 4 - 0
apps/pig/src/pig/models.py

@@ -94,6 +94,10 @@ def create_or_update_script(id, name, script, user, parameters, resources, hadoo
     if not is_design:
       pig_script.doc.get().add_to_history()
 
+  # A user decided eventually to save an unsaved script after execution:
+  if is_design and pig_script.doc.get().is_historic():
+    pig_script.doc.get().remove_from_history()
+
   pig_script.update_from_dict({
       'name': name,
       'script': script,

+ 4 - 0
desktop/core/src/desktop/models.py

@@ -394,6 +394,10 @@ class Document(models.Model):
     tag = DocumentTag.objects.get_history_tag(user=self.owner)
     self.tags.add(tag)
 
+  def remove_from_history(self):
+    tag = DocumentTag.objects.get_history_tag(user=self.owner)
+    self.tags.remove(tag)
+
   def share_to_default(self, name='read'):
     DocumentPermission.objects.share_to_default(self, name=name)