Odstránený zbytočný kód pri genrovaní halvičiek diskusie

LinuxOSsk · Oct 5, 2019 · 6779655 · 6779655
1 parent c2925f2
commit 6779655
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/comments/models.py b/comments/models.py
@@ -119,14 +119,9 @@ class Comment(MPTTModel, TimestampModelMixin):
 	def get_or_create_root_header(self):
 		try:
 			header = RootHeader.objects.get(content_type=self.content_type, object_id=self.object_id)
-			return header
 		except RootHeader.DoesNotExist:
-			with transaction.atomic():
-				header, created = RootHeader.objects.get_or_create(content_type=self.content_type, object_id=self.object_id)
-				if created:
-					header.pub_date = self.created
-					header.save()
-				return header
+			header = RootHeader.objects.get_or_create(content_type=self.content_type, object_id=self.object_id, defaults={'pub_date': self.created})[0]
+		return header
 
 	def get_absolute_url(self):
 		return '%s#link_%d' % (reverse('comments:comments', args=(self.get_or_create_root_header().pk,), kwargs={}), self.id)