python create json and write to file