大家好,我是前端西瓜哥。最近我有个个人需求,要写个小脚本。
我有一个基于 hexo 生成的个人静态博客网站,想要给博客的所有的篇文章的尾部加一段广告文案,我不想写这种没啥意思的脚本,就找 ChatGPT 帮我写一段小脚本。
初版因为我是前端开发,所以我选择让 ChatGPT 给我生成 nodejs 脚本。其他 sh、python 其实也可以,但我真不熟,不好确认代码是否符合需求。
(资料图片)
描述很简洁,ChatGPT 给的结果看着不错。
加上细节要求但我接着发现我有个细节没有提供,就是文件路径需要在文件里提供。我期待通过命令行参数传入路径,因为在 bash 环境下,可以通过 tab 键来补全路径,既方便又不容易出错。
得益于 ChatGPT 的上下文联想能力,我们可以基于上面的问题,加入新的要求。
还提供了使用案例:
完整代码const fs = require("fs");const path = require("path");// 从命令行参数中获取文件夹路径const directoryPath = process.argv[2];if (!directoryPath) { console.error("请提供要处理的文件夹路径"); process.exit(1);}const suffix = "\n\n这是添加的字符串。"; // 将要添加的字符串放在这里fs.readdir(directoryPath, (err, files) => { if (err) { console.error(`无法读取目录:${directoryPath}`, err); return; } // 仅处理 Markdown 文件 const markdownFiles = files.filter(file => path.extname(file).toLowerCase() === ".md"); markdownFiles.forEach(file => { const filePath = path.join(directoryPath, file); // 读取文件内容 fs.readFile(filePath, "utf8", (err, data) => { if (err) { console.error(`无法读取文件:${filePath}`, err); return; } // 将字符串添加到文件末尾 const content = `${data.trim()}${suffix}`; // 将更改保存回原始文件 fs.writeFile(filePath, content, "utf8", err => { if (err) { console.error(`无法写入文件:${filePath}`, err); return; } console.log(`已更新文件:${filePath}`); }); }); });});
使用将代码拷贝到我创建的 js 文件下,将里面的 suffix 替换为自己需要填充的内容。
然后执行:
node scripts/add-footer-qrcode.js source/_posts
可以看到,文件夹下的 100 多个 md 文件的末尾都加上了这段文字。
结尾我是前端西瓜哥,欢迎关注我,一起感受 AI 的日新月异的变化。
ChatGPT 用来写简单的脚本或算法,那叫一个牛逼。
你要是让我来写,还是挺花时间的,里面小细节挺多,快的话写加测试也得半小时,但 ChatGPT 它来了,它 10s 就给你写好了,我们复制一下,改改就能用,甚至不用改!