Bold trace file names, insert traces

debug_toolbar_mongo/operation_tracker.py

@@ -42,10 +42,12 @@ def _insert(collection_self, doc_or_docs, manipulate=True,
     )
     total_time = (time.time() - start_time) * 1000
 
+    __traceback_hide__ = True
     inserts.append({
         'document': doc_or_docs,
         'safe': safe,
         'time': total_time,
+        'stack_trace': _tidy_stacktrace(reversed(inspect.stack())),
     })
     return result
 
@@ -65,6 +67,7 @@ def _update(collection_self, spec, document, upsert=False,
     )
     total_time = (time.time() - start_time) * 1000
 
+    __traceback_hide__ = True
     updates.append({
         'document': document,
         'upsert': upsert,
@@ -72,6 +75,7 @@ def _update(collection_self, spec, document, upsert=False,
         'spec': spec,
         'safe': safe,
         'time': total_time,
+        'stack_trace': _tidy_stacktrace(reversed(inspect.stack())),
     })
     return result
 
@@ -87,10 +91,12 @@ def _remove(collection_self, spec_or_id, safe=False, **kwargs):
     )
     total_time = (time.time() - start_time) * 1000
 
+    __traceback_hide__ = True
     removes.append({
         'spec_or_id': spec_or_id,
         'safe': safe,
         'time': total_time,
+        'stack_trace': _tidy_stacktrace(reversed(inspect.stack())),
     })
     return result
 

debug_toolbar_mongo/templates/mongo-panel.html

@@ -48,11 +48,11 @@ table.mongo-op-table tbody {
                 <td><pre class="mongo-highlight">{% if query.ordering %}{{ query.ordering }}{% endif %}</pre></td>
                 <td>{% if query.skip %}{{ query.skip }}{% endif %}</td>
                 <td>{% if query.limit %}{{ query.limit }}{% endif %}</td>
-                <td><a href="javascript:void(0);" class="mongo-toggle-trace" data-row="{{ forloop.counter }}">Toggle</a></td>
+                <td><a href="javascript:void(0);" class="mongo-toggle-trace" data-row="queries-{{ forloop.counter }}">Toggle</a></td>
             </tr>
             <tr class="{% cycle 'djDebugOdd' 'djDebugEven' %}">
                 <td colspan="8">
-                    <table class="mongo-stack-trace "id="mongo-stack-trace-{{ forloop.counter }}">
+                    <table class="mongo-stack-trace "id="mongo-stack-trace-queries-{{ forloop.counter }}">
                         <thead>
                             <tr>
                                 <th>Line</th>
@@ -65,14 +65,13 @@ table.mongo-op-table tbody {
                         {% for line in query.stack_trace %}
                             <tr>
                                 <td class="lineno">{{ line.1 }}</td>
-                                <td class="file">{{ line.0 }}</td>
+                                <td class="file">{{ line.0|embolden_file }}</td>
                                 <td class="function">{{ line.2 }}</td>
                                 <td class="code">{{ line.3|highlight:"python"|safe }}</td>
                             </tr>
                         {% endfor %}
                         </tbody>
                     </table>
-                    <!--<pre class="mongo-stack-trace">{{ query.stack_trace|format_stack_trace }}</pre>-->
                 </td>
             </tr>
         {% endfor %}
@@ -90,6 +89,7 @@ table.mongo-op-table tbody {
             <th>Time (ms)</th>
             <th>Document</th>
             <th>Safe</th>
+            <th>Stack Trace</th>
         </tr>
     </thead>
     <tbody>
@@ -100,6 +100,31 @@ table.mongo-op-table tbody {
                     <pre class="mongo-highlight">{{ insert.document|format_dict:120|highlight:"javascript"|safe }}</pre>
                 </td>
                 <td>{{ insert.safe }}</td>
+                <td><a href="javascript:void(0);" class="mongo-toggle-trace" data-row="inserts-{{ forloop.counter }}">Toggle</a></td>
+            </tr>
+            <tr class="{% cycle 'djDebugOdd' 'djDebugEven' %}">
+                <td colspan="4">
+                    <table class="mongo-stack-trace "id="mongo-stack-trace-inserts-{{ forloop.counter }}">
+                        <thead>
+                            <tr>
+                                <th>Line</th>
+                                <th>File</th>
+                                <th>Function</th>
+                                <th>Code</th>
+                            </tr>
+                        </thead>
+                        <tbody>
+                        {% for line in insert.stack_trace %}
+                            <tr>
+                                <td class="lineno">{{ line.1 }}</td>
+                                <td class="file">{{ line.0|embolden_file }}</td>
+                                <td class="function">{{ line.2 }}</td>
+                                <td class="code">{{ line.3|highlight:"python"|safe }}</td>
+                            </tr>
+                        {% endfor %}
+                        </tbody>
+                    </table>
+                </td>
             </tr>
         {% endfor %}
     </tbody>

debug_toolbar_mongo/templatetags/mongo_debug_tags.py

@@ -3,6 +3,7 @@ from django.utils.html import escape
 from django.utils.safestring import mark_safe
 
 import pprint
+import os
 
 register = template.Library()
 
@@ -20,6 +21,11 @@ def format_stack_trace(value):
     return mark_safe('\n'.join(stack_trace))
 
 @register.filter
+def embolden_file(path):
+    head, tail = os.path.split(escape(path))
+    return mark_safe(os.sep.join([head, '<strong>{0}</strong>'.format(tail)]))
+
+@register.filter
 def format_dict(value, width=60):
     return pprint.pformat(value, width=int(width))