怎样优雅的创建 log？

比如我有如下的 python 程序，为了方便调试，每个函数都加了很多打印，有没有办法把这些打印都自动保存到一个 log.txt 里 最好能将进入函数、函数中执行步骤、退出函数的 print 的输出都记录下来保存进 log.txt 里

def aa():
    print("enter aa")
    # do sth
    print("quit aa")

def bb():
    print("enter bb")
    # do sth
    print("quit bb")

def cc():
    print("enter cc")
    # do sth
    print("quit cc")

def main()
    print("begin")
    aa()
    bb()
    cc()
    print("all end")

main()