fix

Author: JohananOppongAmoateng

debug_toolbar/panels/profiling.py

@@ -205,7 +205,8 @@ def generate_stats(self, request, response):
                 # If writing to the file fails, we don't want to break the
                 # whole page.
 
-        root_func = cProfile.label(super().process_request.__code__)
+        code = super().process_request.__code__
+        root_func = (code.co_filename, code.co_firstlineno, code.co_name)
         if root_func in self.stats.stats:
             root = FunctionCall(self.stats, root_func, depth=0)
             func_list = []

tests/panels/test_profiling.py

@@ -84,7 +84,8 @@ def test_generate_stats_no_profiler(self):
         self.assertIsNone(self.panel.generate_stats(self.request, response))
 
     def test_generate_stats_signed_path(self):
-        with tempfile.TemporaryDirectory() as tmpdir:
+        tmpdir = tempfile.mkdtemp()
+        try:
             with self.settings(DEBUG_TOOLBAR_CONFIG={"PROFILER_PROFILE_ROOT": tmpdir}):
                 response = self.panel.process_request(self.request)
                 self.panel.generate_stats(self.request, response)
@@ -93,6 +94,8 @@ def test_generate_stats_signed_path(self):
                 # Check that it's a valid signature
                 filename = signing.loads(path)
                 self.assertTrue(filename.endswith(".prof"))
+        finally:
+            shutil.rmtree(tmpdir)
 
     def test_generate_stats_no_root(self):
         response = self.panel.process_request(self.request)
@@ -114,7 +117,8 @@ def test_generate_stats_no_root_func(self):
     @unittest.mock.patch("cProfile.Profile.dump_stats")
     def test_generate_stats_oserror(self, mock_dump_stats):
         mock_dump_stats.side_effect = OSError
-        with tempfile.TemporaryDirectory() as tmpdir:
+        tmpdir = tempfile.mkdtemp()
+        try:
             with self.settings(DEBUG_TOOLBAR_CONFIG={"PROFILER_PROFILE_ROOT": tmpdir}):
                 response = self.panel.process_request(self.request)
                 with self.assertLogs(
@@ -124,6 +128,8 @@ def test_generate_stats_oserror(self, mock_dump_stats):
                 self.assertIn("Failed to dump profiling stats", cm.output[0])
                 # Ensure prof_file_path is not set/updated if dump fails
                 self.assertFalse(hasattr(self.panel, "prof_file_path"))
+        finally:
+            shutil.rmtree(tmpdir)
 
 
 @override_settings(