Enable merging chunks only for neighbours

Marcel Kawski
1 parent 8ba15019
Showing 1 changed file with 9 additions and 2 deletions
collector/storage/views.py
@@ -412,8 +412,15 @@ class ChunkMergeView(UpdateView):
  
     def get_form_kwargs(self):
         kwargs = super(ChunkMergeView, self).get_form_kwargs()
-        poss_target_chunks = self.object.document.chunks.all()
-        poss_target_chunks = poss_target_chunks.exclude(id=self.object.id)
+        all_target_chunks = self.object.document.chunks.all().order_by('sequence')
+        poss_target_chunks = Chunk.objects.none()
+        for index, ch in enumerate(all_target_chunks):
+            if ch.id == self.object.id:
+                if index > 0:
+                    poss_target_chunks |= Chunk.objects.filter(pk=all_target_chunks[index-1].pk)
+                if index < len(all_target_chunks)-1:
+                    poss_target_chunks |= Chunk.objects.filter(pk=all_target_chunks[index+1].pk)
+                break
         kwargs['poss_target_chunks'] = poss_target_chunks
         return kwargs