我有一个带有功能的python模块：

def do_stuff(param1 = 'a'):
    if type(param1) == int:
        # enter python interpreter here
        do_something()
    else:
        do_something_else()

有没有办法放入我有评论的命令行解释器？因此，如果我在python中运行以下命令：

>>> import my_module
>>> do_stuff(1)

我会在下一个评论的范围和上下文中得到我的下一个提示do_stuff()？