How to files based on file type Python