Как сделать Python интерактивной консолью на основе блоков, а не строк

Я хочу, чтобы мой Python интерактивная консоль в терминале работал как тот в pycharm или чем-то как jupyter ноутбук, но я не могу найти как.

Например, когда я делаю:

>>> def foo():
...     print('Make it work nice')
...
>>>

И затем я нажимаю стрелку вверх, я хочу видеть что-то вроде этого:

>>> def foo():
...     print('Make it work nice')
...

Где я могу отредактировать обе строки. но я получаю только это:

>>>     print('Make it work nice')

Так это возможный заставить его работать, как я хочу и если не там какой-либо чрезвычайно легкий инструмент (что-то меньшее, чем IDE или Браузер. что-то столь же тяжелое как VIM, например)