钩子函数atexit.register

from atexit import register

def new_connection():
    def close():
        logging.info("close connection.")

    logging.info("Hello")
    db = 1
    register(close)
    logging.info("World")
    return db