chore: add backend/prisma/seed.js

2026-04-29 10:53:17 +08:00
parent 3bc5c5f152
commit 2cd70ef581
1 changed files with 172 additions and 0 deletions
@@ -0,0 +1,172 @@
+/**
+ * Prisma Seed Script - 初始化测试数据
+ * 用于恢复账务系统的测试数据
+ */
+import { PrismaClient } from '@prisma/client';
+
+const prisma = new PrismaClient();
+
+async function main() {
+  console.log('开始恢复测试数据...');
+
+  // 清空现有数据（按依赖顺序）
+  await prisma.record.deleteMany();
+  await prisma.budget.deleteMany();
+  await prisma.account.deleteMany();
+  await prisma.user.deleteMany();
+
+  // 1. 创建测试用户
+  const user = await prisma.user.create({
+    data: {
+      name: '测试用户',
+      email: 'test@example.com',
+    },
+  });
+  console.log(`创建用户: ${user.name} (ID: ${user.id})`);
+
+  // 2. 创建账户（支付宝、微信、银行卡）
+  const accounts = await Promise.all([
+    prisma.account.create({
+      data: {
+        userId: user.id,
+        name: '支付宝',
+        balance: 5000.00,
+        type: 'Alipay',
+        color: '#1677FF',
+      },
+    }),
+    prisma.account.create({
+      data: {
+        userId: user.id,
+        name: '微信支付',
+        balance: 3000.00,
+        type: 'WeChat',
+        color: '#07C160',
+      },
+    }),
+    prisma.account.create({
+      data: {
+        userId: user.id,
+        name: '银行卡',
+        balance: 10000.00,
+        type: 'BankCard',
+        color: '#722ED1',
+      },
+    }),
+  ]);
+  console.log(`创建 ${accounts.length} 个账户: ${accounts.map(a => a.name).join(', ')}`);
+
+  // 3. 创建交易记录（至少6条）
+  const records = await Promise.all([
+    // 支付宝 - 支出
+    prisma.record.create({
+      data: {
+        userId: user.id,
+        accountId: accounts[0].id, // 支付宝
+        type: 'expense',
+        amount: 128.50,
+        category: '餐饮',
+        description: '午餐',
+        date: new Date('2026-04-25'),
+      },
+    }),
+    // 微信 - 支出
+    prisma.record.create({
+      data: {
+        userId: user.id,
+        accountId: accounts[1].id, // 微信
+        type: 'expense',
+        amount: 45.00,
+        category: '交通',
+        description: '地铁',
+        date: new Date('2026-04-25'),
+      },
+    }),
+    // 银行卡 - 收入
+    prisma.record.create({
+      data: {
+        userId: user.id,
+        accountId: accounts[2].id, // 银行卡
+        type: 'income',
+        amount: 15000.00,
+        category: '工资',
+        description: '月薪',
+        date: new Date('2026-04-20'),
+      },
+    }),
+    // 支付宝 - 支出
+    prisma.record.create({
+      data: {
+        userId: user.id,
+        accountId: accounts[0].id, // 支付宝
+        type: 'expense',
+        amount: 299.00,
+        category: '购物',
+        description: '网购商品',
+        date: new Date('2026-04-23'),
+      },
+    }),
+    // 微信 - 收入
+    prisma.record.create({
+      data: {
+        userId: user.id,
+        accountId: accounts[1].id, // 微信
+        type: 'income',
+        amount: 500.00,
+        category: '转账',
+        description: '朋友还款',
+        date: new Date('2026-04-22'),
+      },
+    }),
+    // 银行卡 - 支出
+    prisma.record.create({
+      data: {
+        userId: user.id,
+        accountId: accounts[2].id, // 银行卡
+        type: 'expense',
+        amount: 2000.00,
+        category: '房租',
+        description: '月租',
+        date: new Date('2026-04-01'),
+      },
+    }),
+    // 支付宝 - 支出（额外）
+    prisma.record.create({
+      data: {
+        userId: user.id,
+        accountId: accounts[0].id,
+        type: 'expense',
+        amount: 56.80,
+        category: '娱乐',
+        description: '电影票',
+        date: new Date('2026-04-24'),
+      },
+    }),
+  ]);
+  console.log(`创建 ${records.length} 条交易记录`);
+
+  // 4. 创建预算
+  await prisma.budget.create({
+    data: {
+      userId: user.id,
+      category: '餐饮',
+      amount: 2000.00,
+      month: '2026-04',
+    },
+  });
+  console.log('创建预算数据');
+
+  console.log('\n✅ 测试数据恢复完成!');
+  console.log(`- 用户: 1 个`);
+  console.log(`- 账户: ${accounts.length} 个`);
+  console.log(`- 交易记录: ${records.length} 条`);
+}
+
+main()
+  .catch((e) => {
+    console.error('❌ 数据恢复失败:', e);
+    process.exit(1);
+  })
+  .finally(async () => {
+    await prisma.$disconnect();
+  });