1
This commit is contained in:
80
_script/remove_custom_ids.js
Normal file
80
_script/remove_custom_ids.js
Normal file
@@ -0,0 +1,80 @@
|
||||
const fs = require('fs');
|
||||
const path = require('path');
|
||||
|
||||
// 递归查找所有 .js 文件
|
||||
function findJsFiles(dir) {
|
||||
let results = [];
|
||||
const list = fs.readdirSync(dir);
|
||||
|
||||
list.forEach(file => {
|
||||
const filePath = path.join(dir, file);
|
||||
const stat = fs.statSync(filePath);
|
||||
|
||||
if (stat && stat.isDirectory()) {
|
||||
results = results.concat(findJsFiles(filePath));
|
||||
} else if (file.endsWith('.js')) {
|
||||
results.push(filePath);
|
||||
}
|
||||
});
|
||||
|
||||
return results;
|
||||
}
|
||||
|
||||
// 移除自定义主键字段
|
||||
function removeCustomIdsInFile(filePath) {
|
||||
try {
|
||||
let content = fs.readFileSync(filePath, 'utf8');
|
||||
const originalContent = content;
|
||||
|
||||
// 匹配自定义主键字段模式
|
||||
const customIdPatterns = [
|
||||
// 模式1: applyId, taskId, chatId, resumeId, deviceId 等
|
||||
/(\w+Id):\s*\{\s*comment:\s*'[^']*',\s*type:\s*Sequelize\.STRING\(\d+\),\s*allowNull:\s*false,\s*primaryKey:\s*true\s*\},?\s*/g,
|
||||
// 模式2: 带 autoIncrement 的 id 字段
|
||||
/(\w+Id):\s*\{\s*comment:\s*'[^']*',\s*type:\s*Sequelize\.INTEGER,\s*primaryKey:\s*true,\s*autoIncrement:\s*true\s*\},?\s*/g
|
||||
];
|
||||
|
||||
customIdPatterns.forEach(pattern => {
|
||||
content = content.replace(pattern, '');
|
||||
});
|
||||
|
||||
// 清理多余的空行
|
||||
content = content.replace(/\n\s*\n\s*\n/g, '\n\n');
|
||||
|
||||
if (content !== originalContent) {
|
||||
fs.writeFileSync(filePath, content, 'utf8');
|
||||
console.log(`已修复: ${filePath}`);
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
} catch (error) {
|
||||
console.error(`处理文件失败 ${filePath}:`, error.message);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// 主函数
|
||||
function main() {
|
||||
const modelDir = path.join(__dirname, 'api', 'model');
|
||||
|
||||
if (!fs.existsSync(modelDir)) {
|
||||
console.error('模型目录不存在:', modelDir);
|
||||
return;
|
||||
}
|
||||
|
||||
const jsFiles = findJsFiles(modelDir);
|
||||
let fixedCount = 0;
|
||||
|
||||
console.log(`找到 ${jsFiles.length} 个模型文件`);
|
||||
|
||||
jsFiles.forEach(filePath => {
|
||||
if (removeCustomIdsInFile(filePath)) {
|
||||
fixedCount++;
|
||||
}
|
||||
});
|
||||
|
||||
console.log(`修复完成,共修复 ${fixedCount} 个文件`);
|
||||
}
|
||||
|
||||
main();
|
||||
Reference in New Issue
Block a user