Pokus o operavenie nekonzistentných komentárov v db

LinuxOSsk · Oct 5, 2019 · 3a2bbc1 · 3a2bbc1
1 parent e7d8cde
commit 3a2bbc1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/comments/templatetags/comments_tags.py b/comments/templatetags/comments_tags.py
@@ -131,7 +131,11 @@ def target_ctype(self):
 	def root_header(self):
 		object_id = self.target.pk
 		ctype = self.target_ctype
-		return RootHeader.objects.get(content_type=ctype, object_id=object_id)
+		try:
+			return RootHeader.objects.get(content_type=ctype, object_id=object_id)
+		except RootHeader.DoesNotExist:
+			Comment.objects.get_or_create_root_comment(ctype, object_id)
+			return RootHeader.objects.get(content_type=ctype, object_id=object_id)
 
 	def get_queryset(self):
 		object_id = self.target.pk