大家好,我是前端西瓜哥。最近我有个个人需求,要写个小脚本。
我有一个基于 hexo 生成的个人静态博客网站,想要给博客的所有的篇文章的尾部加一段广告文案,我不想写这种没啥意思的脚本,就找 ChatGPT 帮我写一段小脚本。
因为我是前端开发,所以我选择让 ChatGPT 给我生成 nodejs 脚本。其他 sh、Python/ target=_blank class=infotextkey>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 = 'nn这是添加的字符串。'; // 将要添加的字符串放在这里
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 就给你写好了,我们复制一下,改改就能用,甚至不用改!