New fast recursion-free tree render engine

boards/models/post/__init__.py

@@ -58,6 +58,7 b' POST_VIEW_PARAMS = ('
     'truncated',
     'mode_tree',
     'perms',
+    'tree_depth',
 )
 
 

boards/models/thread.py

@@ -279,3 +279,20 b' class Thread(models.Model):'
             if len(parents) > 0:
                 self.tags.add(*parents)
 
+    def get_reply_tree(self):
+        replies = self.get_replies().prefetch_related('refposts')
+        tree = []
+        for reply in replies:
+            parents = reply.refposts.all()
+            found_parent = False
+            if len(parents) > 0:
+                index = 0
+                for depth, element in tree:
+                    index += 1
+                    if element in parents:
+                        tree.insert(index, (depth + 1, reply))
+                        found_parent = True
+            if not found_parent:
+                tree.append((0, reply))
+
+        return tree

boards/templates/boards/post.html

@@ -3,7 +3,7 b''
 
 {% get_current_language as LANGUAGE_CODE %}
 
-<div class="{{ css_class }}" id="{{ post.id }}" data-uid="{{ post.uid }}">
+<div class="{{ css_class }}" id="{{ post.id }}" data-uid="{{ post.uid }}" {% if tree_depth %}style="margin-left: {{ tree_depth }}em;"{% endif %}>
     <div class="post-info">
         <a class="post_id" href="{{ post.get_absolute_url }}">#{{ post.get_absolute_id }}</a>
         <span class="title">{{ post.title }}</span>
@@ -82,13 +82,7 b''
         {% endautoescape %}
     </div>
     {% if post.is_referenced %}
-        {% if mode_tree %}
-            <div class="tree_reply">
-                {% for refpost in post.get_referenced_posts %}
-                    {% post_view refpost mode_tree=True %}
-                {% endfor %}
-            </div>
-        {% else %}
+        {% if not mode_tree %}
             <div class="refmap">
                 {% trans "Replies" %}: {{ post.refmap|safe }}
             </div>

boards/templates/boards/thread_tree.html

@@ -10,8 +10,8 b''
     {% get_current_timezone as TIME_ZONE %}
 
     <div class="thread">
-        {% for post in thread.get_top_level_replies %}
-            {% post_view post mode_tree=True %}
+        {% for depth, post in thread.get_reply_tree %}
+        {% post_view post mode_tree=True tree_depth=depth%}
         {% endfor %}
     </div>