Fix print_stack_entry for celery

cansarigol · Jan 27, 2023 · ffd9e3c · ffd9e3c
1 parent 05aa637
commit ffd9e3c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/pdbr/_pdbr.py b/pdbr/_pdbr.py
@@ -419,22 +419,20 @@ def _print_layout(self, val, **kwargs):
             )
 
         def print_stack_entry(self, frame_lineno, prompt_prefix="\n-> ", context=None):
-            def print_syntax(*args):
-                # Remove color format.
+            if base == Pdb or is_celery:
+                Pdb.print_stack_entry(self, frame_lineno, prompt_prefix)
+            else:
                 self._print(
                     Syntax(
-                        ANSI_ESCAPE.sub("", self.format_stack_entry(*args)),
+                        ANSI_ESCAPE.sub(
+                            "", self.format_stack_entry(frame_lineno, "", context)
+                        ),
                         "python",
                         theme=self._theme or DEFAULT_THEME,
                     ),
                     print_layout=False,
                 )
 
-            if base == Pdb or is_celery:
-                print_syntax(frame_lineno, prompt_prefix)
-            else:
-                print_syntax(frame_lineno, "", context)
-
                 # vds: >>
                 frame, lineno = frame_lineno
                 filename = frame.f_code.co_filename

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pdbr"
-version = "0.7.8"
+version = "0.8.0"
 description = "Pdb with Rich library."
 authors = ["Can Sarigol <ertugrulsarigol@gmail.com>"]
 packages = [
@@ -55,7 +55,7 @@ verbose = false
 
 [project]
 name = "pdbr"
-version = "0.7.8"
+version = "0.8.0"
 
 [tool.setuptools]
 py-modules = []