用 AI 助手 WorkBuddy 10 天开发一套进销存软件—全程实录

进销存软件
├── 基础档案
│   ├── 商品管理（含规格/单位）
│   ├── 供应商管理
│   └── 客户管理
├── 采购管理
│   ├── OCR 扫描进项发票
│   └── 发票明细入库
├── 销售管理
│   ├── OCR 扫描销项发票
│   └── 发票明细出库（含出库成本）
├── 库存管理
│   ├── 期初库存导入（Excel）
│   └── 库存台账（数量 + 金额合计）
└── 系统设置
    ├── 多公司管理
    └── 用户权限（JWT 鉴权）

帮我在 D:\workbuddy\inventory-system 目录下初始化一个
Flask + Vue3 前后端分离项目，后端端口 5000，前端端口 5173，
MySQL 数据库名 inventory_db。

inventory-system/
├── backend/
│   ├── app.py              # Flask 主入口
│   ├── start_backend.py    # 一键启动脚本
│   ├── config.py           # 数据库/JWT 配置
│   ├── routes/
│   │   ├── auth.py         # 登录鉴权
│   │   ├── inventory.py    # 库存管理（含期初导入）
│   │   ├── purchase.py     # 采购发票
│   │   ├── sales.py        # 销售发票
│   │   ├── products.py     # 商品档案
│   │   ├── suppliers.py    # 供应商
│   │   └── customers.py    # 客户
│   ├── utils/
│   │   ├── baidu_ocr.py    # 百度 OCR 封装
│   │   └── tencent_ocr.py  # 腾讯 OCR 封装
│   └── requirements.txt
├── frontend/
│   ├── src/
│   │   ├── views/          # 14 个页面组件
│   │   ├── router/         # Vue Router 路由配置
│   │   ├── stores/         # Pinia 状态管理
│   │   └── api/            # Axios API 模块
│   └── package.json
└── .workbuddy/             # WorkBuddy 工作目录

# routes/auth.py（WorkBuddy 生成）
@auth_bp.route('/api/auth/login', methods=['POST'])
def login():
    data = request.json
    username = data.get('username')
    password = data.get('password')

    user = db.execute_one(
        "SELECT * FROM users WHERE username = %s AND password = %s",
        (username, md5(password))
    )
    if not user:
        return jsonify({'code': 401, 'msg': '用户名或密码错误'}), 401

    token = jwt.encode({
        'user_id': user['id'],
        'exp': datetime.utcnow() + timedelta(hours=24)
    }, SECRET_KEY, algorithm='HS256')

    return jsonify({'code': 0, 'token': token, 'user': user})

@purchase_bp.route('/api/purchase/invoices', methods=['POST'])
@require_auth
def create_purchase_invoice():
    data = request.json
    company_id = g.company_id  # 从 JWT 中取多公司 ID

    # 幂等检查：发票号唯一
    existing = db.execute_one(
        "SELECT id FROM purchase_invoices WHERE invoice_no = %s AND company_id = %s",
        (data['invoice_no'], company_id)
    )
    if existing:
        return jsonify({'code': 400, 'msg': '该发票号已存在'}), 400

    # 事务入库
    with db.transaction():
        invoice_id = db.insert('purchase_invoices', {...})
        for item in data['items']:
            db.insert('purchase_invoice_items', {
                'invoice_id': invoice_id, ...
            })
            # 写入库存流水
            db.insert('inventory_transactions', {
                'reference_type': 'purchase',
                'reference_id': invoice_id,
                'product_id': item['product_id'],
                'quantity': item['quantity'],
                'unit_price': item['unit_price'],
                'company_id': company_id
            })

    return jsonify({'code': 0, 'msg': '入库成功'})

我需要上传一张发票图片，系统自动识别出：
- 发票号、开票日期
- 供应商名称和税号
- 商品明细列表（名称、规格、数量、单价、金额）

百度 OCR 和腾讯 OCR 各自的 API Key 我已经申请好了。
请帮我封装两个 OCR 函数，优先用百度，失败时自动切换腾讯。

# utils/baidu_ocr.py
def recognize_invoice(image_path: str) -> dict:
    """调用百度增值税发票识别 API"""
    client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
    with open(image_path, 'rb') as f:
        image_data = f.read()

    result = client.vatInvoice(image_data)

    if 'words_result' not in result:
        raise OCRException(f"百度 OCR 识别失败: {result}")

    # 结构化解析
    words = result['words_result']
    items = []
    for item in words.get('CommodityName', {}).get('word', []):
        items.append({
            'name': item,
            'quantity': ...,
            'unit_price': ...,
            'amount': ...
        })

    return {
        'invoice_no': words.get('InvoiceNum', {}).get('word', ''),
        'date': words.get('InvoiceDate', {}).get('word', ''),
        'seller_name': words.get('SellerName', {}).get('word', ''),
        'seller_tax_no': words.get('SellerRegisterNum', {}).get('word', ''),
        'items': items,
        'total_amount': words.get('TotalAmount', {}).get('word', '0')
    }

def ocr_invoice_with_fallback(image_path: str) -> dict:
    """优先百度，失败切换腾讯"""
    try:
        return baidu_ocr.recognize_invoice(image_path)
    except Exception as e:
        logger.warning(f"百度 OCR 失败，切换腾讯: {e}")
        return tencent_ocr.recognize_invoice(image_path)

@inventory_bp.route('/api/inventory/initial-import', methods=['POST'])
@require_auth
def initial_import():
    """Excel 期初库存导入"""
    file = request.files.get('file')
    if not file:
        return jsonify({'code': 400, 'msg': '请上传文件'}), 400

    # 检查是否已有期初数据（防重复导入）
    existing = db.execute_one(
        "SELECT COUNT(*) as cnt FROM inventory_transactions "
        "WHERE reference_type = 'initial' AND company_id = %s",
        (g.company_id,)
    )
    if existing['cnt'] > 0:
        return jsonify({'code': 400, 'msg': '期初库存已导入，如需重新导入请先清除'}), 400

    # 用 openpyxl 解析 Excel
    wb = openpyxl.load_workbook(file)
    ws = wb.active

    success_count = 0
    errors = []

    with db.transaction():
        for row_num, row in enumerate(ws.iter_rows(min_row=2, values_only=True), start=2):
            product_name, spec, unit, quantity, unit_price = row[:5]

            if not product_name or quantity is None:
                continue

            try:
                # 模糊匹配商品档案
                product = find_or_create_product(product_name, spec, unit)

                db.insert('inventory_transactions', {
                    'reference_type': 'initial',
                    'product_id': product['id'],
                    'quantity': float(quantity),
                    'unit_price': float(unit_price or 0),
                    'company_id': g.company_id,
                    'remark': '期初导入'
                })
                success_count += 1
            except Exception as e:
                errors.append(f"第 {row_num} 行：{str(e)}")

    return jsonify({
        'code': 0,
        'success_count': success_count,
        'errors': errors
    })

<!-- PurchaseForm.vue 核心部分 -->
<template>
  <div class="purchase-form">
    <!-- 发票上传区域 -->
    <el-upload
      action="/api/ocr/invoice"
      :headers="authHeaders"
      :on-success="handleOCRSuccess"
      accept=".jpg,.png,.pdf"
      drag
    >
      <el-icon class="el-icon--upload"><upload-filled /></el-icon>
      <div class="el-upload__text">拖拽发票图片到此处，或 <em>点击上传</em></div>
    </el-upload>

    <!-- OCR 识别结果（可编辑） -->
    <el-form v-if="invoiceData" :model="invoiceData" label-width="120px">
      <el-form-item label="发票号码">
        <el-input v-model="invoiceData.invoice_no" />
      </el-form-item>
      <el-form-item label="供应商">
        <el-select v-model="invoiceData.supplier_id" filterable allow-create>
          <el-option v-for="s in suppliers" :key="s.id" :label="s.name" :value="s.id" />
        </el-select>
      </el-form-item>

      <!-- 商品明细表格（可编辑） -->
      <el-table :data="invoiceData.items" border>
        <el-table-column label="商品名称">
          <template #default="{ row }">
            <el-input v-model="row.product_name" @blur="handleFuzzyMatch(row)" />
          </template>
        </el-table-column>
        <!-- ... 数量、单价、金额列 -->
      </el-table>

      <el-button type="primary" @click="submitInvoice">确认入库</el-button>
    </el-form>
  </div>
</template>

<!-- InventoryLedger.vue 核心查询逻辑 -->
<script setup>
const ledgerData = ref([])
const totalQty = computed(() => ledgerData.value.reduce((s, r) => s + r.quantity, 0))
const totalAmount = computed(() => ledgerData.value.reduce((s, r) => s + r.amount, 0))

// 查询库存台账
const fetchLedger = async () => {
  const res = await api.get('/api/inventory/ledger', {
    params: { company_id: companyStore.currentCompanyId, ...filters }
  })
  ledgerData.value = res.data.list
}
</script>

商品名称模糊匹配规则：
1. 名称中包含相同核心词（如"钢筋"）
2. 规格相同（如"HRB400"、"16mm"）
3. 单位相同（如"吨"）
以上三个条件满足则视为同一商品，自动匹配并关联

@inventory_bp.route('/api/inventory/batch-fuzzy-match', methods=['POST'])
@require_auth
def batch_fuzzy_match():
    """OCR 识别后批量模糊匹配商品"""
    items = request.json.get('items', [])
    products = db.execute_many(
        "SELECT id, name, spec, unit FROM products WHERE company_id = %s",
        (g.company_id,)
    )

    results = []
    for item in items:
        matched = None
        best_score = 0

        for product in products:
            score = 0
            # 名称包含匹配（核心词提取）
            if extract_core_name(item['name']) in product['name']:
                score += 2
            # 规格完全匹配
            if normalize_spec(item.get('spec', '')) == normalize_spec(product['spec']):
                score += 2
            # 单位匹配
            if item.get('unit', '') == product['unit']:
                score += 1

            if score > best_score and score >= 3:
                best_score = score
                matched = product

        results.append({
            'input': item,
            'matched_product': matched,
            'confidence': best_score / 5
        })

    return jsonify({'code': 0, 'results': results})

-- 先修改为 VARCHAR，再约束
ALTER TABLE inventory_transactions
  MODIFY COLUMN reference_type VARCHAR(20) NOT NULL;

-- 添加 CHECK 约束
ALTER TABLE inventory_transactions
  ADD CONSTRAINT chk_ref_type
  CHECK (reference_type IN ('purchase', 'sale', 'initial', 'adjustment'));

# 负库存放行策略
def check_stock_before_sale(product_id, quantity, company_id):
    current_stock = get_current_stock(product_id, company_id)
    if current_stock < quantity:
        # 不阻断，记录警告
        logger.warning(f"商品 {product_id} 库存不足: 当前 {current_stock}，出库 {quantity}")
        return {'allow': True, 'warning': f'库存不足，当前余量 {current_stock}'}
    return {'allow': True, 'warning': None}

写一个 Python 函数 insert_into_db(data)

我需要在采购发票保存时，把每条商品明细同时写入库存流水表，
要求事务原子性，如果任何一条失败则全部回滚，
数据库是 MySQL 8，用 PyMySQL 连接池。

添加一个 if 判断

同一张发票号，在同一个公司下只能录入一次，
如果重复提交要返回友好的错误提示（不是 500 报错）

# 安装 Python 依赖
pip install flask pymysql pyjwt python-dotenv openpyxl aip

# 创建 MySQL 数据库
mysql -u root -p
CREATE DATABASE inventory_db CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;

# 初始化表结构
python backend/create_tables.py

# backend/config.py
DB_CONFIG = {
    'host': '127.0.0.1',
    'port': 3306,
    'user': 'root',
    'password': 'your_password',
    'database': 'inventory_db',
    'charset': 'utf8mb4'
}

JWT_SECRET = 'your-secret-key-change-in-production'
BAIDU_OCR_APP_ID = 'your_app_id'
BAIDU_OCR_API_KEY = 'your_api_key'
BAIDU_OCR_SECRET_KEY = 'your_secret_key'

# start_backend.py — WorkBuddy 生成的启动脚本
import subprocess
import sys
import os

os.chdir(os.path.dirname(os.path.abspath(__file__)))

print("启动后端服务...")
process = subprocess.Popen(
    [sys.executable, 'app.py'],
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT
)

print(f"后端服务已启动，PID: {process.pid}")
print("访问地址：http://127.0.0.1:5000")

# 启动前端
cd frontend && npm run dev

# 访问系统
open http://localhost:5173