124 lines
3.2 KiB
JavaScript
124 lines
3.2 KiB
JavaScript
const sqlite3 = require('sqlite3').verbose();
|
||
|
||
// 数据库连接
|
||
const db = new sqlite3.Database('goguryeo_video.db');
|
||
|
||
// 重新分配视频到不同分类
|
||
const redistributionPlan = [
|
||
// 将一些动物视频重新分类为传统艺术（表演类）
|
||
{
|
||
videoIds: [17], // 巨肺狗子（表演性质）
|
||
newCategory: 2, // 传统艺术
|
||
reason: '表演性质的内容'
|
||
},
|
||
// 将一些动物视频重新分类为语言学习（教育性质）
|
||
{
|
||
videoIds: [11, 15], // 小牛叫妈妈、水獭
|
||
newCategory: 3, // 语言学习
|
||
reason: '教育和学习性质的内容'
|
||
},
|
||
// 将一些动物视频重新分类为考古发现（自然探索）
|
||
{
|
||
videoIds: [9, 18], // 小老虎、小象
|
||
newCategory: 4, // 考古发现
|
||
reason: '自然探索和发现类内容'
|
||
},
|
||
// 将一些动物视频重新分类为其他
|
||
{
|
||
videoIds: [16, 20], // 小肥啾、小猫咪
|
||
newCategory: 6, // 其他
|
||
reason: '综合性内容'
|
||
}
|
||
];
|
||
|
||
// 更新视频分类
|
||
function updateVideoCategory(videoId, newCategory) {
|
||
return new Promise((resolve, reject) => {
|
||
const sql = 'UPDATE videos SET category = ? WHERE id = ?';
|
||
db.run(sql, [newCategory, videoId], function(err) {
|
||
if (err) {
|
||
reject(err);
|
||
} else {
|
||
resolve(this.changes);
|
||
}
|
||
});
|
||
});
|
||
}
|
||
|
||
// 获取视频信息
|
||
function getVideoInfo(videoId) {
|
||
return new Promise((resolve, reject) => {
|
||
const sql = 'SELECT id, title, category FROM videos WHERE id = ?';
|
||
db.get(sql, [videoId], (err, row) => {
|
||
if (err) {
|
||
reject(err);
|
||
} else {
|
||
resolve(row);
|
||
}
|
||
});
|
||
});
|
||
}
|
||
|
||
// 主函数
|
||
async function main() {
|
||
try {
|
||
console.log('开始重新分配视频分类...');
|
||
|
||
for (const plan of redistributionPlan) {
|
||
console.log(`\n重新分配到分类 ${plan.newCategory}: ${plan.reason}`);
|
||
|
||
for (const videoId of plan.videoIds) {
|
||
try {
|
||
// 获取视频信息
|
||
const videoInfo = await getVideoInfo(videoId);
|
||
if (!videoInfo) {
|
||
console.log(`视频 ID ${videoId} 不存在`);
|
||
continue;
|
||
}
|
||
|
||
console.log(` ${videoInfo.title} (ID: ${videoId}) 从分类 ${videoInfo.category} -> ${plan.newCategory}`);
|
||
|
||
// 更新分类
|
||
await updateVideoCategory(videoId, plan.newCategory);
|
||
|
||
} catch (error) {
|
||
console.error(`更新视频 ${videoId} 失败:`, error.message);
|
||
}
|
||
}
|
||
}
|
||
|
||
console.log('\n重新分配完成！');
|
||
|
||
// 显示最终的分类统计
|
||
console.log('\n=== 最终分类统计 ===');
|
||
const sql = `
|
||
SELECT c.name, COUNT(v.id) as video_count
|
||
FROM categories c
|
||
LEFT JOIN videos v ON c.id = v.category
|
||
GROUP BY c.id, c.name
|
||
ORDER BY c.id
|
||
`;
|
||
|
||
db.all(sql, [], (err, rows) => {
|
||
if (err) {
|
||
console.error('查询统计失败:', err);
|
||
} else {
|
||
rows.forEach(row => {
|
||
console.log(`${row.name}: ${row.video_count} 个视频`);
|
||
});
|
||
}
|
||
db.close();
|
||
});
|
||
|
||
} catch (error) {
|
||
console.error('重新分配失败:', error);
|
||
db.close();
|
||
}
|
||
}
|
||
|
||
// 运行脚本
|
||
if (require.main === module) {
|
||
main();
|
||
}
|
||
|
||
module.exports = { main }; |