Simplest python Bdb extension(just for study)

mibdb-test.py

import pdb
import wdb
from mibdb import MiBdb
d = MiBdb()
d.set_trace()

d.set_breakpoint("simplecall.py", 12)
d.set_breakpoint("simplecall.py", 16)

def a():
    double_value = 2
    return b() * double_value


def b():
    return 'response from b'


print('hello')

a()

mibdb.py

from bdb import Bdb
import inspect
import sys


def default_trace(frame):
    print("breakpoint log : ", frame.f_locals)
    local_vars: dict = frame.f_locals
    ui = input("MiBDB> ")


class MiBdb(Bdb):
    def do_clear(self, arg):
        pass

    def __init__(self):
        Bdb.__init__(self)
        self.breakpoints = dict()
        self.set_trace()

    def end(self):
        self.reset()

        # bdb does not seem to actually support clearing, so we do it manually
        sys.settrace(None)

    def set_breakpoint(self, filename=None, lineno=None, method=None):
        if filename is None and lineno is None and method is None:
            return
        self.set_break(filename, lineno)

        self.breakpoints.get((filename, lineno), {})

        if method is None:
            method = default_trace
        try :
            self.breakpoints[(filename, lineno)].add(method)
        except KeyError:
            self.breakpoints[(filename, lineno)] = [method]

    def unset_breakpoint(self, filename=None, lineno=None):
        if filename is None and lineno is None:
            return
        del self.breakpoints[(filename, lineno)]

    def user_line(self, frame):
        if not self.break_here(frame):
            return

        (filename, lineno, function, code_context, index) = inspect.getframeinfo(frame)
        filename = filename.split("/")[-1]
        callbacks = self.breakpoints[((filename, lineno))]
        for callback in callbacks:
            callback(frame)

#
# class MiDbg(Bdb):
#     def user_call(self, frame, argument_list):
#         print('user call')
#     def user_exception(self, frame, exc_info):
#         print('user exception')
#     def user_line(self, frame):
#         print('user line')
#     def user_return(self, frame, return_value):
#         print('user return')