cleanup

tufedtm · tufedtm · commit 844ae2cb3366 · 2024-05-24T18:07:57.000Z
diff --git a/debug_toolbar_line_profiler/__init__.py b/debug_toolbar_line_profiler/__init__.py
@@ -1,6 +1,6 @@
-
 def profile_additional(*subfuncs):
     def inner(func):
         func.profile_additional = subfuncs
         return func
+
     return inner
diff --git a/debug_toolbar_line_profiler/panel.py b/debug_toolbar_line_profiler/panel.py
@@ -1,139 +1,41 @@
-from colorsys import hsv_to_rgb
 import cProfile
 import inspect
-import os
+from io import StringIO
 from pstats import Stats
 
+from debug_toolbar import settings as dt_settings
+from debug_toolbar.panels import Panel
+from debug_toolbar.panels.profiling import FunctionCall as DjDTFunctionCall
 from django.urls import resolve
 from django.utils.translation import gettext_lazy as _
-from django.utils.safestring import mark_safe
-from io import StringIO
-from debug_toolbar.panels import Panel
 from django.views.generic.base import View
-
 from line_profiler import LineProfiler, show_func
 
 from . import signals
 
 
-class DjangoDebugToolbarStats(Stats):
-    __root = None
-
-    def get_root_func(self, view_func):
-        if self.__root is None:
-            filename = view_func.__code__.co_filename
-            firstlineno = view_func.__code__.co_firstlineno
-            for func, (_, __, ___, ____, callers) in self.stats.items():
-                if (len(callers) >= 0
-                        and func[0] == filename
-                        and func[1] == firstlineno):
-                    self.__root = func
-                    break
-        return self.__root
-
-
-class FunctionCall:
-    """
-    The FunctionCall object is a helper object that encapsulates some of the
-    complexity of working with pstats/cProfile objects
-
-    """
-    def __init__(self, statobj, func, depth=0, stats=None,
-                 _id="0", parent_ids=[], hsv=(0, 0.5, 1)):
-        self.statobj = statobj
-        self.func = func
-        if stats:
-            self.stats = stats
-        else:
-            self.stats = statobj.stats[func][:4]
-        self.depth = depth
-        self.id = _id
-        self.parent_ids = parent_ids
-        self.hsv = hsv
+class FunctionCall(DjDTFunctionCall):
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+        self.has_subfuncs = False
         self._line_stats_text = None
 
-    def parent_classes(self):
-        return self.parent_classes
-
-    def background(self):
-        r, g, b = hsv_to_rgb(*self.hsv)
-        return 'rgb(%f%%,%f%%,%f%%)' % (r * 100, g * 100, b * 100)
-
-    def func_std_string(self):  # match what old profile produced
-        func_name = self.func
-        if func_name[:2] == ('~', 0):
-            # special case for built-in functions
-            name = func_name[2]
-            if name.startswith('<') and name.endswith('>'):
-                return '{%s}' % name[1:-1]
-            else:
-                return name
-        else:
-            file_name, line_num, method = self.func
-            idx = file_name.find('/site-packages/')
-            if idx > -1:
-                file_name = file_name[(idx + 14):]
-
-            file_items = file_name.rsplit(os.sep, 1)
-            file_path, file_name = file_items if len(file_items) > 1 else [
-                None, file_name]
-
-            return mark_safe(
-                '<span class="path">{0}/</span>'
-                '<span class="file">{1}</span>'
-                ' in <span class="func">{3}</span>'
-                '(<span class="lineno">{2}</span>)'.format(
-                    file_path,
-                    file_name,
-                    line_num,
-                    method))
-
     def subfuncs(self):
-        i = 0
         h, s, _ = self.hsv
         count = len(self.statobj.all_callees[self.func])
-        for func, stats in self.statobj.all_callees[self.func].items():
-            i += 1
+        for i, (func, stats) in enumerate(self.statobj.all_callees[self.func].items(), 1):
             h1 = h + (i / count) / (self.depth + 1)
-            if stats[3] == 0 or self.stats[3] == 0:
-                s1 = 0
-            else:
-                s1 = s * (stats[3] / self.stats[3])
-            yield FunctionCall(self.statobj,
-                               func,
-                               self.depth + 1,
-                               stats=stats,
-                               _id=self.id + '_' + str(i),
-                               parent_ids=self.parent_ids + [self.id],
-                               hsv=(h1, s1, 1))
-
-    def count(self):
-        return self.stats[1]
-
-    def tottime(self):
-        return self.stats[2]
-
-    def cumtime(self):
-        return self.stats[3]
-
-    def tottime_per_call(self):
-        cc, nc, tt, ct = self.stats
-
-        if nc == 0:
-            return 0
-
-        return tt / nc
-
-    def cumtime_per_call(self):
-        cc, _, __, ct = self.stats
-
-        if cc == 0:
-            return 0
-
-        return ct / cc
-
-    def indent(self):
-        return 16 * self.depth
+            s1 = 0 if stats[3] == 0 or self.stats[3] == 0 else s * (stats[3] / self.stats[3])
+
+            yield FunctionCall(
+                self.statobj,
+                func,
+                self.depth + 1,
+                stats=stats,
+                id=f'{self.id}_{i}',
+                parent_ids=[*self.parent_ids, self.id],
+                hsv=(h1, s1, 1),
+            )
 
     def line_stats_text(self):
         if self._line_stats_text is None:
@@ -142,29 +44,29 @@ def line_stats_text(self):
                 out = StringIO()
                 fn, lineno, name = self.func
                 try:
-                    show_func(fn,
-                              lineno,
-                              name,
-                              lstats.timings[self.func],
-                              lstats.unit, stream=out)
+                    show_func(fn, lineno, name, lstats.timings[self.func], lstats.unit, stream=out)
                     self._line_stats_text = out.getvalue()
                 except ZeroDivisionError:
-                    self._line_stats_text = ("There was a ZeroDivisionError, "
-                                             "total_time was probably zero")
+                    self._line_stats_text = 'There was a ZeroDivisionError, total_time was probably zero'
             else:
                 self._line_stats_text = False
         return self._line_stats_text
 
 
 class ProfilingPanel(Panel):
-    """
-    Panel that displays profiling information.
-    """
-    title = _('Profiling')
+    """Panel that displays profiling information."""
+
+    capture_project_code = dt_settings.get_config()['PROFILER_CAPTURE_PROJECT_CODE']
+
+    @property
+    def title(self) -> str:
+        return _('Profiling')
 
-    template = 'debug_toolbar_line_profiler/panels/profiling.html'
+    @property
+    def template(self) -> str:
+        return 'debug_toolbar_line_profiler/panels/profiling.html'
 
-    def _unwrap_closure_and_profile(self, func):
+    def _unwrap_closure_and_profile(self, func) -> None:
         if not hasattr(func, '__code__') or func in self.added:
             return
 
@@ -173,96 +75,87 @@ def _unwrap_closure_and_profile(self, func):
         self.line_profiler.add_function(func)
         for subfunc in getattr(func, 'profile_additional', []):
             self._unwrap_closure_and_profile(subfunc)
-        func_closure = func.__closure__
 
-        if func_closure:
+        if func_closure := func.__closure__:
             for cell in func_closure:
                 target = cell.cell_contents
                 if hasattr(target, '__code__'):
-                    self._unwrap_closure_and_profile(cell.cell_contents)
+                    self._unwrap_closure_and_profile(target)
                 if inspect.isclass(target) and View in inspect.getmro(target):
                     for name, value in inspect.getmembers(target):
-                        if not name.startswith('__') and (
-                            inspect.ismethod(value) or
-                            inspect.isfunction(value)
-                        ):
+                        if not name.startswith('__') and (inspect.ismethod(value) or inspect.isfunction(value)):
                             self._unwrap_closure_and_profile(value)
 
     def process_request(self, request):
-        view_func, view_args, view_kwargs = resolve(request.path)
-        self.view_func = view_func
+        self.view_func, view_args, view_kwargs = resolve(request.path)
         self.profiler = cProfile.Profile()
         self.line_profiler = LineProfiler()
         self.added = set()
         self._unwrap_closure_and_profile(self.view_func)
-        signals.profiler_setup.send(sender=self,
-                                    profiler=self.line_profiler,
-                                    view_func=view_func,
-                                    view_args=view_args,
-                                    view_kwargs=view_kwargs)
+        signals.profiler_setup.send(
+            sender=self,
+            profiler=self.line_profiler,
+            view_func=self.view_func,
+            view_args=view_args,
+            view_kwargs=view_kwargs,
+        )
         self.line_profiler.enable_by_count()
         out = self.profiler.runcall(super().process_request, request)
         self.line_profiler.disable_by_count()
         return out
 
-    def add_node(self, func_list, func, max_depth, cum_time=0.1):
-        """
-        add_node does a depth first traversal of the call graph, appending a
-        FunctionCall object to func_list, so that the Django template only
-        has to do a single for loop over func_list that can render a tree
-        structure
+    def add_node(self, func_list: list, func: FunctionCall, max_depth: int, cum_time: float):
+        """add_node does a depth first traversal of the call graph, appending a FunctionCall object to func_list, so
+        that the Django template only has to do a single for loop over func_list that can render a tree structure.
 
-        Parameters:
-            func_list is an array that will have a FunctionCall for each call
-                added to it
+        Parameters
+        ----------
+            func_list is an array that will have a FunctionCall for each call added to it
             func is a FunctionCall object that will have all its callees added
             max_depth is the maximum depth we should recurse
-            cum_time is the minimum cum_time a function should have to be
-                included in the output
+            cum_time is the minimum cum_time a function should have to be included in the output
+
         """
         func_list.append(func)
         func.has_subfuncs = False
-        # this function somewhat dangerously relies on FunctionCall to set its
-        # subfuncs' depth argument correctly
+        # this function somewhat dangerously relies on FunctionCall to set its subfuncs' depth argument correctly
         if func.depth >= max_depth:
             return
 
-        # func.subfuncs returns FunctionCall objects
         subs = sorted(func.subfuncs(), key=FunctionCall.cumtime, reverse=True)
         for subfunc in subs:
-            # a sub function is important if it takes a long time or it has
-            # line_stats
-            if (subfunc.cumtime() >= cum_time or
-                    (hasattr(self.stats, 'line_stats') and
-                     subfunc.func in self.stats.line_stats.timings)):
+            # a sub function is important if it takes a long time or it has line_stats
+            is_project_code = bool(self.capture_project_code and subfunc.is_project_func())
+            if (
+                subfunc.cumtime() >= cum_time
+                or (is_project_code and subfunc.cumtime() > 0)
+                or (hasattr(self.stats, 'line_stats') and subfunc.func in self.stats.line_stats.timings)
+            ):
                 func.has_subfuncs = True
                 self.add_node(
                     func_list=func_list,
                     func=subfunc,
                     max_depth=max_depth,
-                    cum_time=subfunc.cumtime()/16)
+                    cum_time=subfunc.cumtime() / dt_settings.get_config()['PROFILER_THRESHOLD_RATIO'] / 2,
+                )
 
     def generate_stats(self, request, response):
         if not hasattr(self, 'profiler'):
-            return None
+            return
         # Could be delayed until the panel content is requested (perf. optim.)
         self.profiler.create_stats()
-        self.stats = DjangoDebugToolbarStats(self.profiler)
+        self.stats = Stats(self.profiler)
         self.stats.line_stats = self.line_profiler.get_stats()
         self.stats.calc_callees()
 
         func_list = []
-        root_func = self.stats.get_root_func(self.view_func)
-
-        if root_func is not None:
-            root_node = FunctionCall(statobj=self.stats,
-                                     func=root_func,
-                                     depth=0)
+        if root_func := cProfile.label(self.view_func.__code__):
+            root_node = FunctionCall(statobj=self.stats, func=root_func, depth=0)
             self.add_node(
                 func_list=func_list,
                 func=root_node,
-                max_depth=10,
-                cum_time=root_node.cumtime() / 8
+                max_depth=dt_settings.get_config()['PROFILER_MAX_DEPTH'],
+                cum_time=root_node.cumtime() / dt_settings.get_config()['PROFILER_THRESHOLD_RATIO'],
             )
         # else:
         # what should we do if we didn't detect a root function? It's not
diff --git a/debug_toolbar_line_profiler/signals.py b/debug_toolbar_line_profiler/signals.py
@@ -1,4 +1,3 @@
 import django.dispatch
 
-
 profiler_setup = django.dispatch.Signal()
diff --git a/debug_toolbar_line_profiler/templates/debug_toolbar_line_profiler/panels/profiling.html b/debug_toolbar_line_profiler/templates/debug_toolbar_line_profiler/panels/profiling.html
@@ -1,5 +1,5 @@
-{% load i18n %}{% load static %}
-<table width="100%">
+{% load i18n %}
+<table>
 	<thead>
 		<tr>
 			<th>{% trans "Call" %}</th>
@@ -12,16 +12,15 @@
 	</thead>
 	<tbody>
 		{% for call in func_list %}
-			<!--  style="background:{{ call.background }}" -->
-			<tr class="djDebugProfileRow{% for parent_id in call.parent_ids %} djToggleDetails_{{ parent_id }}{% endfor %}" depth="{{ call.depth }}" id="_{{ call.id }}">
+			<tr class="djdt-profile-row {% if call.is_project_func %}djdt-highlighted {% endif %} {% for parent_id in call.parent_ids %} djToggleDetails_{{ parent_id }}{% endfor %}" id="profilingMain_{{ call.id }}">
 				<td>
-					<div style="padding-left: {{ call.indent }}px;">
+					<div data-djdt-styles="paddingLeft:{{ call.indent }}px">
 						{% if call.has_subfuncs %}
-							<a class="djProfileToggleDetails djToggleSwitch" data-toggle-id="{{ call.id }}" data-toggle-open="+" data-toggle-close="-" data-toggle-name="" href="javascript:void(0)">-</a>
+							<button type="button" class="djProfileToggleDetails djToggleSwitch" data-toggle-name="profilingMain" data-toggle-id="{{ call.id }}">-</button>
 						{% else %}
 							<span class="djNoToggleSwitch"></span>
 						{% endif %}
-						<span class="stack">{{ call.func_std_string }}</span>
+						<span class="djdt-stack">{{ call.func_std_string }}</span>
 					</div>
 				</td>
 				<td>{{ call.cumtime|floatformat:3 }}</td>
@@ -33,7 +32,7 @@
 			{% if call.line_stats_text %}
 				<tr class="djToggleDetails_{{ call.id }}{% for parent_id in call.parent_ids %} djToggleDetails_{{ parent_id }}{% endfor %}">
 					<td colspan="6">
-						<div style="padding-left: {{ call.indent }}px;"><pre>{{ call.line_stats_text }}</pre></div>
+						<div style="padding-left:{{ call.indent }}px"><pre>{{ call.line_stats_text }}</pre></div>
 					</td>
 				</tr>
 			{% endif %}
diff --git a/setup.py b/setup.py

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`	`1`	`import django.dispatch`
`2`	`2`
`3`		`-`
`4`	`3`	`profiler_setup = django.dispatch.Signal()`