feat: 个人记账与预算管理系统 MVP 初始版本

2026-04-28 22:57:38 +08:00
commit d2b2e9887a
88 changed files with 29512 additions and 0 deletions
@@ -0,0 +1,66 @@
+// This is your Prisma schema file,
+// learn more about it in the docs: https://pris.ly/d/prisma-schema
+
+// Looking for ways to speed up your queries, or scale easily with your serverless or edge functions?
+// Try Prisma Accelerate: https://pris.ly/cli/accelerate-init
+
+generator client {
+  provider = "prisma-client-js"
+}
+
+datasource db {
+  provider = "sqlite"
+  url      = env("DATABASE_URL")
+}
+
+// 用户表
+model User {
+  id        Int      @id @default(autoincrement())
+  name      String
+  email     String   @unique
+  createdAt DateTime @default(now())
+  accounts  Account[]
+  budgets   Budget[]
+  records   Record[]
+}
+
+// 账户表
+model Account {
+  id        Int      @id @default(autoincrement())
+  userId    Int
+  name      String   // 账户名称(支付宝、微信、银行卡等)
+  balance   Decimal  @default(0)  // 当前余额
+  type      String   // 账户类型
+  color     String   // 显示颜色
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+  user      User     @relation(fields: [userId], references: [id])
+  records   Record[]
+}
+
+// 交易记录表
+model Record {
+  id          Int      @id @default(autoincrement())
+  userId      Int
+  accountId   Int
+  type        String   // 'income' | 'expense'
+  amount      Decimal
+  category    String   // 消费类别
+  description String?  // 备注
+  date        DateTime
+  createdAt   DateTime @default(now())
+  user        User     @relation(fields: [userId], references: [id])
+  account     Account  @relation(fields: [accountId], references: [id])
+}
+
+// 预算表
+model Budget {
+  id        Int      @id @default(autoincrement())
+  userId    Int
+  category  String   // 预算类别
+  amount    Decimal  // 预算金额
+  month     String   // 预算月份(YYYY-MM)
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+  user      User     @relation(fields: [userId], references: [id])
+}
@@ -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();
+  });