Files
zhuiguang-ai/prisma/seed.ts
T

42 lines
1.5 KiB
TypeScript

import { PrismaClient } from "@prisma/client";
import { PrismaMariaDb } from "@prisma/adapter-mariadb";
const adapter = new PrismaMariaDb(process.env.DATABASE_URL!);
const prisma = new PrismaClient({ adapter });
const categories = [
{ name: "文本生成", slug: "text-generation", icon: "", sortOrder: 1 },
{ name: "图像生成", slug: "image-generation", icon: "", sortOrder: 2 },
{ name: "视频生成", slug: "video-generation", icon: "", sortOrder: 3 },
{ name: "音频处理", slug: "audio-processing", icon: "", sortOrder: 4 },
{ name: "代码助手", slug: "code-assistant", icon: "", sortOrder: 5 },
{ name: "办公效率", slug: "office-efficiency", icon: "", sortOrder: 6 },
{ name: "AI 角色", slug: "ai-character", icon: "", sortOrder: 7 },
{ name: "设计工具", slug: "design-tools", icon: "", sortOrder: 8 },
{ name: "数据分析", slug: "data-analysis", icon: "", sortOrder: 9 },
{ name: "内容检测", slug: "content-detection", icon: "", sortOrder: 10 },
];
async function main() {
console.log("Seeding categories...");
for (const category of categories) {
await prisma.category.upsert({
where: { slug: category.slug },
update: { name: category.name, sortOrder: category.sortOrder },
create: category,
});
}
console.log(`Seeded ${categories.length} categories.`);
}
main()
.catch((e) => {
console.error(e);
process.exit(1);
})
.finally(async () => {
await prisma.$disconnect();
});