How to simplify the process of saving files in python