修改生成模板逻辑