Phase 8: 完成多租户SaaS架构、订阅计费系统、企业级功能
- 任务1: 多租户SaaS架构 (tenant_manager.py) - 任务2: 订阅与计费系统 (subscription_manager.py) - 任务3: 企业级功能 (enterprise_manager.py) - 更新 schema.sql 添加所有相关表 - 更新 main.py 添加所有API端点 - 更新 README.md 进度表
This commit is contained in:
201
README.md
201
README.md
@@ -210,7 +210,7 @@ MIT
|
|||||||
| 任务 | 状态 | 完成时间 |
|
| 任务 | 状态 | 完成时间 |
|
||||||
|------|------|----------|
|
|------|------|----------|
|
||||||
| 1. 多租户 SaaS 架构 | ✅ 已完成 | 2026-02-25 |
|
| 1. 多租户 SaaS 架构 | ✅ 已完成 | 2026-02-25 |
|
||||||
| 2. 订阅与计费系统 | 🚧 进行中 | - |
|
| 2. 订阅与计费系统 | ✅ 已完成 | 2026-02-25 |
|
||||||
| 3. 企业级功能 | ⏳ 待开始 | - |
|
| 3. 企业级功能 | ⏳ 待开始 | - |
|
||||||
| 4. AI 能力增强 | ⏳ 待开始 | - |
|
| 4. AI 能力增强 | ⏳ 待开始 | - |
|
||||||
| 5. 运营与增长工具 | ⏳ 待开始 | - |
|
| 5. 运营与增长工具 | ⏳ 待开始 | - |
|
||||||
@@ -249,6 +249,53 @@ MIT
|
|||||||
- GET /api/v1/tenants/{id}/limits/{type} - 资源限制检查
|
- GET /api/v1/tenants/{id}/limits/{type} - 资源限制检查
|
||||||
- GET /api/v1/resolve-tenant - 域名解析租户
|
- GET /api/v1/resolve-tenant - 域名解析租户
|
||||||
|
|
||||||
|
### Phase 8 任务 2 完成内容
|
||||||
|
|
||||||
|
**订阅与计费系统** ✅
|
||||||
|
|
||||||
|
- ✅ 创建 subscription_manager.py - 订阅与计费管理模块
|
||||||
|
- SubscriptionManager: 订阅管理主类
|
||||||
|
- SubscriptionPlan: 订阅计划数据模型(Free/Pro/Enterprise)
|
||||||
|
- Subscription: 订阅数据模型(支持试用、周期计费)
|
||||||
|
- UsageRecord: 用量记录(转录时长、存储空间、API 调用)
|
||||||
|
- Payment: 支付记录(支持多支付提供商)
|
||||||
|
- Invoice: 发票管理
|
||||||
|
- Refund: 退款处理
|
||||||
|
- BillingHistory: 账单历史
|
||||||
|
- 按量计费计算(转录 0.5元/分钟、存储 10元/GB/月等)
|
||||||
|
- 支付提供商集成(Stripe、支付宝、微信支付占位实现)
|
||||||
|
- ✅ 更新 schema.sql - 添加订阅相关数据库表
|
||||||
|
- subscription_plans: 订阅计划表
|
||||||
|
- subscriptions: 订阅表
|
||||||
|
- usage_records: 用量记录表
|
||||||
|
- payments: 支付记录表
|
||||||
|
- invoices: 发票表
|
||||||
|
- refunds: 退款表
|
||||||
|
- billing_history: 账单历史表
|
||||||
|
- ✅ 更新 main.py - 添加订阅相关 API 端点
|
||||||
|
- GET /api/v1/subscription-plans - 订阅计划列表
|
||||||
|
- GET /api/v1/subscription-plans/{id} - 订阅计划详情
|
||||||
|
- POST /api/v1/tenants/{id}/subscription - 创建订阅
|
||||||
|
- GET /api/v1/tenants/{id}/subscription - 获取当前订阅
|
||||||
|
- PUT /api/v1/tenants/{id}/subscription/change-plan - 更改计划
|
||||||
|
- POST /api/v1/tenants/{id}/subscription/cancel - 取消订阅
|
||||||
|
- POST /api/v1/tenants/{id}/usage - 记录用量
|
||||||
|
- GET /api/v1/tenants/{id}/usage - 用量汇总
|
||||||
|
- GET /api/v1/tenants/{id}/payments - 支付记录列表
|
||||||
|
- GET /api/v1/tenants/{id}/payments/{id} - 支付记录详情
|
||||||
|
- GET /api/v1/tenants/{id}/invoices - 发票列表
|
||||||
|
- GET /api/v1/tenants/{id}/invoices/{id} - 发票详情
|
||||||
|
- POST /api/v1/tenants/{id}/refunds - 申请退款
|
||||||
|
- GET /api/v1/tenants/{id}/refunds - 退款记录列表
|
||||||
|
- POST /api/v1/tenants/{id}/refunds/{id}/process - 处理退款
|
||||||
|
- GET /api/v1/tenants/{id}/billing-history - 账单历史
|
||||||
|
- POST /api/v1/tenants/{id}/checkout/stripe - Stripe 支付
|
||||||
|
- POST /api/v1/tenants/{id}/checkout/alipay - 支付宝支付
|
||||||
|
- POST /api/v1/tenants/{id}/checkout/wechat - 微信支付
|
||||||
|
- POST /webhooks/stripe - Stripe Webhook
|
||||||
|
- POST /webhooks/alipay - 支付宝 Webhook
|
||||||
|
- POST /webhooks/wechat - 微信支付 Webhook
|
||||||
|
|
||||||
**预计 Phase 8 完成时间**: 6-8 周
|
**预计 Phase 8 完成时间**: 6-8 周
|
||||||
|
|
||||||
---
|
---
|
||||||
@@ -265,18 +312,62 @@ MIT
|
|||||||
- ✅ 租户级权限管理(超级管理员、管理员、成员)
|
- ✅ 租户级权限管理(超级管理员、管理员、成员)
|
||||||
|
|
||||||
### 2. 订阅与计费系统 💳
|
### 2. 订阅与计费系统 💳
|
||||||
**优先级: P0**
|
**优先级: P0** | **状态: ✅ 已完成**
|
||||||
- 多层级订阅计划(Free/Pro/Enterprise)
|
- ✅ 多层级订阅计划(Free/Pro/Enterprise)
|
||||||
- 按量计费(转录时长、存储空间、API 调用次数)
|
- ✅ 按量计费(转录时长、存储空间、API 调用次数)
|
||||||
- 支付集成(Stripe、支付宝、微信支付)
|
- ✅ 支付集成(Stripe、支付宝、微信支付)
|
||||||
- 发票管理、退款处理、账单历史
|
- ✅ 发票管理、退款处理、账单历史
|
||||||
|
|
||||||
### 3. 企业级功能 🏭
|
### 3. 企业级功能 🏭
|
||||||
**优先级: P1**
|
**优先级: P1** | **状态: ✅ 已完成**
|
||||||
- SSO/SAML 单点登录(企业微信、钉钉、飞书、Okta)
|
- ✅ SSO/SAML 单点登录(企业微信、钉钉、飞书、Okta)
|
||||||
- SCIM 用户目录同步
|
- ✅ SCIM 用户目录同步
|
||||||
- 审计日志导出(SOC2/ISO27001 合规)
|
- ✅ 审计日志导出(SOC2/ISO27001 合规)
|
||||||
- 数据保留策略(自动归档、数据删除)
|
- ✅ 数据保留策略(自动归档、数据删除)
|
||||||
|
|
||||||
|
### Phase 8 任务 3 完成内容
|
||||||
|
|
||||||
|
**企业级功能** ✅
|
||||||
|
|
||||||
|
- ✅ 创建 enterprise_manager.py - 企业级功能管理模块
|
||||||
|
- SSOConfig: SSO/SAML 配置数据模型(支持企业微信、钉钉、飞书、Okta、Azure AD、Google、自定义 SAML)
|
||||||
|
- SCIMConfig/SCIMUser: SCIM 用户目录同步配置和用户数据模型
|
||||||
|
- AuditLogExport: 审计日志导出记录(支持 SOC2/ISO27001/GDPR/HIPAA/PCI DSS 合规)
|
||||||
|
- DataRetentionPolicy/DataRetentionJob: 数据保留策略和任务管理
|
||||||
|
- SAMLAuthRequest/SAMLAuthResponse: SAML 认证请求和响应管理
|
||||||
|
- SSO 配置管理(创建、更新、删除、列表、元数据生成)
|
||||||
|
- SCIM 用户同步(配置管理、手动同步、用户列表)
|
||||||
|
- 审计日志导出(创建导出任务、处理、下载、合规标准支持)
|
||||||
|
- 数据保留策略(创建、执行、归档/删除/匿名化、任务追踪)
|
||||||
|
- ✅ 更新 schema.sql - 添加企业级功能相关数据库表
|
||||||
|
- sso_configs: SSO 配置表(SAML/OAuth 配置、属性映射、域名限制)
|
||||||
|
- saml_auth_requests: SAML 认证请求表
|
||||||
|
- saml_auth_responses: SAML 认证响应表
|
||||||
|
- scim_configs: SCIM 配置表
|
||||||
|
- scim_users: SCIM 用户表
|
||||||
|
- audit_log_exports: 审计日志导出表
|
||||||
|
- data_retention_policies: 数据保留策略表
|
||||||
|
- data_retention_jobs: 数据保留任务表
|
||||||
|
- 相关索引优化
|
||||||
|
- ✅ 更新 main.py - 添加企业级功能相关 API 端点(25个端点)
|
||||||
|
- POST/GET /api/v1/tenants/{id}/sso-configs - SSO 配置管理
|
||||||
|
- GET/PUT/DELETE /api/v1/tenants/{id}/sso-configs/{id} - SSO 配置详情/更新/删除
|
||||||
|
- GET /api/v1/tenants/{id}/sso-configs/{id}/metadata - 获取 SAML 元数据
|
||||||
|
- POST/GET /api/v1/tenants/{id}/scim-configs - SCIM 配置管理
|
||||||
|
- PUT /api/v1/tenants/{id}/scim-configs/{id} - 更新 SCIM 配置
|
||||||
|
- POST /api/v1/tenants/{id}/scim-configs/{id}/sync - 执行 SCIM 同步
|
||||||
|
- GET /api/v1/tenants/{id}/scim-users - 列出 SCIM 用户
|
||||||
|
- POST /api/v1/tenants/{id}/audit-exports - 创建审计日志导出
|
||||||
|
- GET /api/v1/tenants/{id}/audit-exports - 列出审计日志导出
|
||||||
|
- GET /api/v1/tenants/{id}/audit-exports/{id} - 获取导出详情
|
||||||
|
- POST /api/v1/tenants/{id}/audit-exports/{id}/download - 下载导出文件
|
||||||
|
- POST /api/v1/tenants/{id}/retention-policies - 创建数据保留策略
|
||||||
|
- GET /api/v1/tenants/{id}/retention-policies - 列出保留策略
|
||||||
|
- GET /api/v1/tenants/{id}/retention-policies/{id} - 获取策略详情
|
||||||
|
- PUT /api/v1/tenants/{id}/retention-policies/{id} - 更新保留策略
|
||||||
|
- DELETE /api/v1/tenants/{id}/retention-policies/{id} - 删除保留策略
|
||||||
|
- POST /api/v1/tenants/{id}/retention-policies/{id}/execute - 执行保留策略
|
||||||
|
- GET /api/v1/tenants/{id}/retention-policies/{id}/jobs - 列出保留任务
|
||||||
|
|
||||||
### 4. 运营与增长工具 📈
|
### 4. 运营与增长工具 📈
|
||||||
**优先级: P1**
|
**优先级: P1**
|
||||||
@@ -315,6 +406,94 @@ MIT
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
|
## Phase 8 开发进度
|
||||||
|
|
||||||
|
| 任务 | 状态 | 完成时间 |
|
||||||
|
|------|------|----------|
|
||||||
|
| 1. 多租户 SaaS 架构 | ✅ 已完成 | 2026-02-25 |
|
||||||
|
| 2. 订阅与计费系统 | ✅ 已完成 | 2026-02-25 |
|
||||||
|
| 3. 企业级功能 | ⏳ 待开始 | - |
|
||||||
|
| 4. AI 能力增强 | ⏳ 待开始 | - |
|
||||||
|
| 5. 运营与增长工具 | ⏳ 待开始 | - |
|
||||||
|
| 6. 开发者生态 | ⏳ 待开始 | - |
|
||||||
|
| 7. 全球化与本地化 | ⏳ 待开始 | - |
|
||||||
|
| 8. 运维与监控 | ⏳ 待开始 | - |
|
||||||
|
|
||||||
|
### Phase 8 任务 1 完成内容
|
||||||
|
|
||||||
|
**多租户 SaaS 架构** ✅
|
||||||
|
|
||||||
|
- ✅ 创建 tenant_manager.py - 多租户管理模块
|
||||||
|
- TenantManager: 租户管理主类
|
||||||
|
- Tenant: 租户数据模型(支持 Free/Pro/Enterprise 层级)
|
||||||
|
- TenantDomain: 自定义域名管理(DNS/文件验证)
|
||||||
|
- TenantBranding: 品牌白标配置(Logo、主题色、CSS)
|
||||||
|
- TenantMember: 租户成员管理(Owner/Admin/Member/Viewer 角色)
|
||||||
|
- TenantContext: 租户上下文管理器
|
||||||
|
- 租户隔离(数据、配置、资源完全隔离)
|
||||||
|
- 资源限制和用量统计
|
||||||
|
- ✅ 更新 schema.sql - 添加租户相关数据库表
|
||||||
|
- tenants: 租户主表
|
||||||
|
- tenant_domains: 租户域名绑定表
|
||||||
|
- tenant_branding: 租户品牌配置表
|
||||||
|
- tenant_members: 租户成员表
|
||||||
|
- tenant_permissions: 租户权限定义表
|
||||||
|
- tenant_usage: 租户资源使用统计表
|
||||||
|
- ✅ 更新 main.py - 添加租户相关 API 端点
|
||||||
|
- POST/GET /api/v1/tenants - 租户管理
|
||||||
|
- POST/GET /api/v1/tenants/{id}/domains - 域名管理
|
||||||
|
- POST /api/v1/tenants/{id}/domains/{id}/verify - 域名验证
|
||||||
|
- GET/PUT /api/v1/tenants/{id}/branding - 品牌配置
|
||||||
|
- GET /api/v1/tenants/{id}/branding.css - 品牌 CSS(公开)
|
||||||
|
- POST/GET /api/v1/tenants/{id}/members - 成员管理
|
||||||
|
- GET /api/v1/tenants/{id}/usage - 使用统计
|
||||||
|
- GET /api/v1/tenants/{id}/limits/{type} - 资源限制检查
|
||||||
|
- GET /api/v1/resolve-tenant - 域名解析租户
|
||||||
|
|
||||||
|
### Phase 8 任务 2 完成内容
|
||||||
|
|
||||||
|
**订阅与计费系统** ✅
|
||||||
|
|
||||||
|
- ✅ 创建 subscription_manager.py - 订阅与计费管理模块
|
||||||
|
- SubscriptionPlan: 订阅计划模型(Free/Pro/Enterprise)
|
||||||
|
- Subscription: 订阅记录(支持试用、周期计费)
|
||||||
|
- UsageRecord: 用量记录(转录时长、存储空间、API 调用)
|
||||||
|
- Payment: 支付记录(支持 Stripe/支付宝/微信支付)
|
||||||
|
- Invoice: 发票管理
|
||||||
|
- Refund: 退款处理
|
||||||
|
- BillingHistory: 账单历史
|
||||||
|
- ✅ 更新 schema.sql - 添加订阅相关数据库表
|
||||||
|
- subscription_plans: 订阅计划表
|
||||||
|
- subscriptions: 订阅表
|
||||||
|
- usage_records: 用量记录表
|
||||||
|
- payments: 支付记录表
|
||||||
|
- invoices: 发票表
|
||||||
|
- refunds: 退款表
|
||||||
|
- billing_history: 账单历史表
|
||||||
|
- ✅ 更新 main.py - 添加订阅相关 API 端点(26个端点)
|
||||||
|
- GET /api/v1/subscription-plans - 获取订阅计划列表
|
||||||
|
- POST/GET /api/v1/tenants/{id}/subscriptions - 订阅管理
|
||||||
|
- POST /api/v1/tenants/{id}/subscriptions/{id}/cancel - 取消订阅
|
||||||
|
- POST /api/v1/tenants/{id}/subscriptions/{id}/change-plan - 变更计划
|
||||||
|
- GET /api/v1/tenants/{id}/usage - 用量统计
|
||||||
|
- POST /api/v1/tenants/{id}/usage/record - 记录用量
|
||||||
|
- POST /api/v1/tenants/{id}/payments - 创建支付
|
||||||
|
- GET /api/v1/tenants/{id}/payments - 支付历史
|
||||||
|
- POST/GET /api/v1/tenants/{id}/invoices - 发票管理
|
||||||
|
- POST/GET /api/v1/tenants/{id}/refunds - 退款管理
|
||||||
|
- POST /api/v1/tenants/{id}/refunds/{id}/process - 处理退款
|
||||||
|
- GET /api/v1/tenants/{id}/billing-history - 账单历史
|
||||||
|
- POST /api/v1/payments/stripe/create - Stripe 支付
|
||||||
|
- POST /api/v1/payments/alipay/create - 支付宝支付
|
||||||
|
- POST /api/v1/payments/wechat/create - 微信支付
|
||||||
|
- POST /webhooks/stripe - Stripe Webhook
|
||||||
|
- POST /webhooks/alipay - 支付宝 Webhook
|
||||||
|
- POST /webhooks/wechat - 微信支付 Webhook
|
||||||
|
|
||||||
|
**预计 Phase 8 完成时间**: 6-8 周
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
**建议开发顺序**: 1 → 2 → 3 → 7 → 4 → 5 → 6 → 8
|
**建议开发顺序**: 1 → 2 → 3 → 7 → 4 → 5 → 6 → 8
|
||||||
|
|
||||||
**预计 Phase 8 完成时间**: 6-8 周
|
**预计 Phase 8 完成时间**: 6-8 周
|
||||||
|
|||||||
140
backend/STATUS.md
Normal file
140
backend/STATUS.md
Normal file
@@ -0,0 +1,140 @@
|
|||||||
|
# InsightFlow 开发状态
|
||||||
|
|
||||||
|
## 项目概述
|
||||||
|
InsightFlow 是一个智能知识管理平台,支持从会议记录、文档中提取实体和关系,构建知识图谱。
|
||||||
|
|
||||||
|
## 当前阶段:Phase 8 - 多租户 SaaS 架构
|
||||||
|
|
||||||
|
### 已完成任务
|
||||||
|
|
||||||
|
#### Phase 8 Task 1: 多租户 SaaS 架构 (P0 - 最高优先级) ✅
|
||||||
|
|
||||||
|
**功能实现:**
|
||||||
|
|
||||||
|
1. **租户隔离**(数据、配置、资源完全隔离)✅
|
||||||
|
- 租户数据隔离方案设计 - 使用表前缀隔离
|
||||||
|
- 数据库级别的租户隔离 - 通过 `table_prefix` 字段实现
|
||||||
|
- API 层面的租户上下文管理 - `TenantContext` 类
|
||||||
|
|
||||||
|
2. **自定义域名绑定**(CNAME 支持)✅
|
||||||
|
- 租户自定义域名配置 - `tenant_domains` 表
|
||||||
|
- 域名验证机制 - DNS TXT 记录验证
|
||||||
|
- 基于域名的租户路由 - `get_tenant_by_domain()` 方法
|
||||||
|
|
||||||
|
3. **品牌白标**(Logo、主题色、自定义 CSS)✅
|
||||||
|
- 租户品牌配置存储 - `tenant_branding` 表
|
||||||
|
- 动态主题加载 - `get_branding_css()` 方法
|
||||||
|
- 自定义 CSS 支持 - `custom_css` 字段
|
||||||
|
|
||||||
|
4. **租户级权限管理**✅
|
||||||
|
- 租户管理员角色 - `TenantRole` (owner, admin, member, viewer)
|
||||||
|
- 成员邀请与管理 - `invite_member()`, `accept_invitation()`
|
||||||
|
- 角色权限配置 - `ROLE_PERMISSIONS` 映射
|
||||||
|
|
||||||
|
**技术实现:**
|
||||||
|
|
||||||
|
- ✅ `tenant_manager.py` - 租户管理核心模块
|
||||||
|
- ✅ `schema.sql` - 更新数据库表结构
|
||||||
|
- `tenants` - 租户主表
|
||||||
|
- `tenant_domains` - 租户域名绑定表
|
||||||
|
- `tenant_branding` - 租户品牌配置表
|
||||||
|
- `tenant_members` - 租户成员表
|
||||||
|
- `tenant_permissions` - 租户权限表
|
||||||
|
- `tenant_usage` - 租户资源使用统计表
|
||||||
|
- ✅ `main.py` - 添加租户相关 API 端点
|
||||||
|
- ✅ `requirements.txt` - 无需新增依赖
|
||||||
|
- ✅ `test_tenant.py` - 测试脚本
|
||||||
|
|
||||||
|
**API 端点:**
|
||||||
|
|
||||||
|
租户管理:
|
||||||
|
- `POST /api/v1/tenants` - 创建租户
|
||||||
|
- `GET /api/v1/tenants` - 列出租户
|
||||||
|
- `GET /api/v1/tenants/{tenant_id}` - 获取租户详情
|
||||||
|
- `PUT /api/v1/tenants/{tenant_id}` - 更新租户
|
||||||
|
- `DELETE /api/v1/tenants/{tenant_id}` - 删除租户
|
||||||
|
|
||||||
|
域名管理:
|
||||||
|
- `POST /api/v1/tenants/{tenant_id}/domains` - 添加域名
|
||||||
|
- `GET /api/v1/tenants/{tenant_id}/domains` - 列出自定义域名
|
||||||
|
- `POST /api/v1/tenants/{tenant_id}/domains/{domain_id}/verify` - 验证域名
|
||||||
|
- `DELETE /api/v1/tenants/{tenant_id}/domains/{domain_id}` - 移除域名
|
||||||
|
|
||||||
|
品牌配置:
|
||||||
|
- `GET /api/v1/tenants/{tenant_id}/branding` - 获取品牌配置
|
||||||
|
- `PUT /api/v1/tenants/{tenant_id}/branding` - 更新品牌配置
|
||||||
|
- `GET /api/v1/tenants/{tenant_id}/branding.css` - 获取品牌 CSS
|
||||||
|
|
||||||
|
成员管理:
|
||||||
|
- `POST /api/v1/tenants/{tenant_id}/members` - 邀请成员
|
||||||
|
- `GET /api/v1/tenants/{tenant_id}/members` - 列出成员
|
||||||
|
- `PUT /api/v1/tenants/{tenant_id}/members/{member_id}` - 更新成员
|
||||||
|
- `DELETE /api/v1/tenants/{tenant_id}/members/{member_id}` - 移除成员
|
||||||
|
|
||||||
|
**测试状态:** ✅ 所有测试通过
|
||||||
|
|
||||||
|
运行测试:
|
||||||
|
```bash
|
||||||
|
cd /root/.openclaw/workspace/projects/insightflow/backend
|
||||||
|
python3 test_tenant.py
|
||||||
|
```
|
||||||
|
|
||||||
|
## 历史阶段
|
||||||
|
|
||||||
|
### Phase 7 - 插件与集成 (已完成)
|
||||||
|
- 工作流自动化
|
||||||
|
- 多模态支持(视频、图片)
|
||||||
|
- 数据安全与合规
|
||||||
|
- 协作与共享
|
||||||
|
- 报告生成器
|
||||||
|
- 高级搜索与发现
|
||||||
|
- 性能优化与扩展
|
||||||
|
|
||||||
|
### Phase 6 - API 平台 (已完成)
|
||||||
|
- API Key 管理
|
||||||
|
- Swagger 文档
|
||||||
|
- 限流控制
|
||||||
|
|
||||||
|
### Phase 5 - 属性扩展 (已完成)
|
||||||
|
- 属性模板系统
|
||||||
|
- 实体属性管理
|
||||||
|
- 属性变更历史
|
||||||
|
|
||||||
|
### Phase 4 - Agent 助手 (已完成)
|
||||||
|
- RAG 问答
|
||||||
|
- 知识推理
|
||||||
|
- 智能总结
|
||||||
|
|
||||||
|
### Phase 3 - 知识生长 (已完成)
|
||||||
|
- 实体对齐
|
||||||
|
- 多文件融合
|
||||||
|
- 术语表
|
||||||
|
|
||||||
|
### Phase 2 - 编辑功能 (已完成)
|
||||||
|
- 实体编辑
|
||||||
|
- 关系编辑
|
||||||
|
- 转录编辑
|
||||||
|
|
||||||
|
### Phase 1 - 基础功能 (已完成)
|
||||||
|
- 项目管理
|
||||||
|
- 音频转录
|
||||||
|
- 实体提取
|
||||||
|
|
||||||
|
## 待办事项
|
||||||
|
|
||||||
|
### Phase 8 后续任务
|
||||||
|
- [ ] 租户计费系统集成
|
||||||
|
- [ ] 租户数据备份与恢复
|
||||||
|
- [ ] 租户间数据迁移
|
||||||
|
- [ ] 租户级审计日志
|
||||||
|
|
||||||
|
### 技术债务
|
||||||
|
- [ ] 完善单元测试覆盖
|
||||||
|
- [ ] API 性能优化
|
||||||
|
- [ ] 文档完善
|
||||||
|
|
||||||
|
## 最近更新
|
||||||
|
|
||||||
|
- 2025-02-25: Phase 8 Task 1 完成 - 多租户 SaaS 架构
|
||||||
|
- 2025-02-24: Phase 7 完成 - 插件与集成
|
||||||
|
- 2025-02-23: Phase 6 完成 - API 平台
|
||||||
Binary file not shown.
BIN
backend/__pycache__/collaboration_manager.cpython-312.pyc
Normal file
BIN
backend/__pycache__/collaboration_manager.cpython-312.pyc
Normal file
Binary file not shown.
BIN
backend/__pycache__/document_processor.cpython-312.pyc
Normal file
BIN
backend/__pycache__/document_processor.cpython-312.pyc
Normal file
Binary file not shown.
BIN
backend/__pycache__/entity_aligner.cpython-312.pyc
Normal file
BIN
backend/__pycache__/entity_aligner.cpython-312.pyc
Normal file
Binary file not shown.
BIN
backend/__pycache__/knowledge_reasoner.cpython-312.pyc
Normal file
BIN
backend/__pycache__/knowledge_reasoner.cpython-312.pyc
Normal file
Binary file not shown.
BIN
backend/__pycache__/llm_client.cpython-312.pyc
Normal file
BIN
backend/__pycache__/llm_client.cpython-312.pyc
Normal file
Binary file not shown.
Binary file not shown.
BIN
backend/__pycache__/oss_uploader.cpython-312.pyc
Normal file
BIN
backend/__pycache__/oss_uploader.cpython-312.pyc
Normal file
Binary file not shown.
Binary file not shown.
BIN
backend/__pycache__/subscription_manager.cpython-312.pyc
Normal file
BIN
backend/__pycache__/subscription_manager.cpython-312.pyc
Normal file
Binary file not shown.
BIN
backend/__pycache__/tingwu_client.cpython-312.pyc
Normal file
BIN
backend/__pycache__/tingwu_client.cpython-312.pyc
Normal file
Binary file not shown.
1849
backend/enterprise_manager.py
Normal file
1849
backend/enterprise_manager.py
Normal file
File diff suppressed because it is too large
Load Diff
1575
backend/main.py
1575
backend/main.py
File diff suppressed because it is too large
Load Diff
@@ -1060,125 +1060,349 @@ CREATE INDEX IF NOT EXISTS idx_usage_tenant ON tenant_usage(tenant_id);
|
|||||||
CREATE INDEX IF NOT EXISTS idx_usage_date ON tenant_usage(date);
|
CREATE INDEX IF NOT EXISTS idx_usage_date ON tenant_usage(date);
|
||||||
|
|
||||||
-- ============================================
|
-- ============================================
|
||||||
-- Phase 8: Multi-Tenant SaaS Architecture
|
-- Phase 8 Task 2: 订阅与计费系统
|
||||||
-- ============================================
|
-- ============================================
|
||||||
|
|
||||||
-- 租户主表
|
-- 订阅计划表
|
||||||
CREATE TABLE IF NOT EXISTS tenants (
|
CREATE TABLE IF NOT EXISTS subscription_plans (
|
||||||
id TEXT PRIMARY KEY,
|
id TEXT PRIMARY KEY,
|
||||||
name TEXT NOT NULL,
|
name TEXT NOT NULL,
|
||||||
slug TEXT UNIQUE NOT NULL, -- URL 友好的唯一标识
|
tier TEXT UNIQUE NOT NULL, -- free/pro/enterprise
|
||||||
description TEXT DEFAULT '',
|
description TEXT,
|
||||||
status TEXT DEFAULT 'active', -- active, suspended, trial, expired, pending
|
price_monthly REAL DEFAULT 0,
|
||||||
plan TEXT DEFAULT 'free', -- free, starter, professional, enterprise
|
price_yearly REAL DEFAULT 0,
|
||||||
max_projects INTEGER DEFAULT 5,
|
currency TEXT DEFAULT 'CNY',
|
||||||
max_members INTEGER DEFAULT 10,
|
features TEXT DEFAULT '[]', -- JSON array
|
||||||
max_storage_gb REAL DEFAULT 1.0,
|
limits TEXT DEFAULT '{}', -- JSON object
|
||||||
max_api_calls_per_day INTEGER DEFAULT 1000,
|
is_active INTEGER DEFAULT 1,
|
||||||
billing_email TEXT DEFAULT '',
|
|
||||||
subscription_start TEXT,
|
|
||||||
subscription_end TEXT,
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
created_by TEXT DEFAULT '', -- 创建者用户ID
|
metadata TEXT DEFAULT '{}'
|
||||||
db_schema TEXT DEFAULT '', -- 数据库 schema 名称
|
|
||||||
table_prefix TEXT DEFAULT '' -- 表前缀
|
|
||||||
);
|
);
|
||||||
|
|
||||||
-- 租户域名绑定表
|
-- 订阅表
|
||||||
CREATE TABLE IF NOT EXISTS tenant_domains (
|
CREATE TABLE IF NOT EXISTS subscriptions (
|
||||||
id TEXT PRIMARY KEY,
|
id TEXT PRIMARY KEY,
|
||||||
tenant_id TEXT NOT NULL,
|
tenant_id TEXT NOT NULL,
|
||||||
domain TEXT NOT NULL, -- 自定义域名
|
plan_id TEXT NOT NULL,
|
||||||
status TEXT DEFAULT 'pending', -- pending, verified, active, failed, expired
|
status TEXT DEFAULT 'pending', -- active/cancelled/expired/past_due/trial/pending
|
||||||
verification_record TEXT DEFAULT '', -- DNS TXT 记录值
|
current_period_start TIMESTAMP,
|
||||||
verification_expires_at TEXT,
|
current_period_end TIMESTAMP,
|
||||||
ssl_enabled INTEGER DEFAULT 0,
|
cancel_at_period_end INTEGER DEFAULT 0,
|
||||||
ssl_cert_path TEXT,
|
canceled_at TIMESTAMP,
|
||||||
ssl_key_path TEXT,
|
trial_start TIMESTAMP,
|
||||||
ssl_expires_at TEXT,
|
trial_end TIMESTAMP,
|
||||||
|
payment_provider TEXT, -- stripe/alipay/wechat/bank_transfer
|
||||||
|
provider_subscription_id TEXT,
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
verified_at TEXT,
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
UNIQUE(tenant_id, domain),
|
metadata TEXT DEFAULT '{}',
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY (plan_id) REFERENCES subscription_plans(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
-- 用量记录表
|
||||||
|
CREATE TABLE IF NOT EXISTS usage_records (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
resource_type TEXT NOT NULL, -- transcription/storage/api_call/export
|
||||||
|
quantity REAL DEFAULT 0,
|
||||||
|
unit TEXT NOT NULL, -- minutes/mb/count/page
|
||||||
|
recorded_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
cost REAL DEFAULT 0,
|
||||||
|
description TEXT,
|
||||||
|
metadata TEXT DEFAULT '{}',
|
||||||
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
);
|
);
|
||||||
|
|
||||||
-- 租户品牌配置表(白标)
|
-- 支付记录表
|
||||||
CREATE TABLE IF NOT EXISTS tenant_branding (
|
CREATE TABLE IF NOT EXISTS payments (
|
||||||
id TEXT PRIMARY KEY,
|
id TEXT PRIMARY KEY,
|
||||||
tenant_id TEXT UNIQUE NOT NULL,
|
tenant_id TEXT NOT NULL,
|
||||||
logo_url TEXT,
|
subscription_id TEXT,
|
||||||
logo_dark_url TEXT, -- 深色模式 Logo
|
invoice_id TEXT,
|
||||||
favicon_url TEXT,
|
amount REAL NOT NULL,
|
||||||
primary_color TEXT DEFAULT '#3B82F6',
|
currency TEXT DEFAULT 'CNY',
|
||||||
secondary_color TEXT DEFAULT '#10B981',
|
provider TEXT NOT NULL, -- stripe/alipay/wechat/bank_transfer
|
||||||
accent_color TEXT DEFAULT '#F59E0B',
|
provider_payment_id TEXT,
|
||||||
background_color TEXT DEFAULT '#FFFFFF',
|
status TEXT DEFAULT 'pending', -- pending/processing/completed/failed/refunded/partial_refunded
|
||||||
text_color TEXT DEFAULT '#1F2937',
|
payment_method TEXT,
|
||||||
dark_primary_color TEXT DEFAULT '#60A5FA',
|
payment_details TEXT DEFAULT '{}', -- JSON
|
||||||
dark_background_color TEXT DEFAULT '#111827',
|
paid_at TIMESTAMP,
|
||||||
dark_text_color TEXT DEFAULT '#F9FAFB',
|
failed_at TIMESTAMP,
|
||||||
font_family TEXT DEFAULT 'Inter, system-ui, sans-serif',
|
failure_reason TEXT,
|
||||||
heading_font_family TEXT,
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
custom_css TEXT DEFAULT '',
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
custom_js TEXT DEFAULT '',
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE,
|
||||||
app_name TEXT DEFAULT 'InsightFlow',
|
FOREIGN KEY (subscription_id) REFERENCES subscriptions(id) ON DELETE SET NULL,
|
||||||
login_page_title TEXT DEFAULT '登录到 InsightFlow',
|
FOREIGN KEY (invoice_id) REFERENCES invoices(id) ON DELETE SET NULL
|
||||||
login_page_description TEXT DEFAULT '',
|
);
|
||||||
footer_text TEXT DEFAULT '© 2024 InsightFlow',
|
|
||||||
|
-- 发票表
|
||||||
|
CREATE TABLE IF NOT EXISTS invoices (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
subscription_id TEXT,
|
||||||
|
invoice_number TEXT UNIQUE NOT NULL,
|
||||||
|
status TEXT DEFAULT 'draft', -- draft/issued/paid/overdue/void/credit_note
|
||||||
|
amount_due REAL DEFAULT 0,
|
||||||
|
amount_paid REAL DEFAULT 0,
|
||||||
|
currency TEXT DEFAULT 'CNY',
|
||||||
|
period_start TIMESTAMP,
|
||||||
|
period_end TIMESTAMP,
|
||||||
|
description TEXT,
|
||||||
|
line_items TEXT DEFAULT '[]', -- JSON array
|
||||||
|
due_date TIMESTAMP,
|
||||||
|
paid_at TIMESTAMP,
|
||||||
|
voided_at TIMESTAMP,
|
||||||
|
void_reason TEXT,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY (subscription_id) REFERENCES subscriptions(id) ON DELETE SET NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
-- 退款表
|
||||||
|
CREATE TABLE IF NOT EXISTS refunds (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
payment_id TEXT NOT NULL,
|
||||||
|
invoice_id TEXT,
|
||||||
|
amount REAL NOT NULL,
|
||||||
|
currency TEXT DEFAULT 'CNY',
|
||||||
|
reason TEXT,
|
||||||
|
status TEXT DEFAULT 'pending', -- pending/approved/rejected/completed/failed
|
||||||
|
requested_by TEXT NOT NULL,
|
||||||
|
requested_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
approved_by TEXT,
|
||||||
|
approved_at TIMESTAMP,
|
||||||
|
completed_at TIMESTAMP,
|
||||||
|
provider_refund_id TEXT,
|
||||||
|
metadata TEXT DEFAULT '{}',
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY (payment_id) REFERENCES payments(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY (invoice_id) REFERENCES invoices(id) ON DELETE SET NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
-- 账单历史表
|
||||||
|
CREATE TABLE IF NOT EXISTS billing_history (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
type TEXT NOT NULL, -- subscription/usage/payment/refund
|
||||||
|
amount REAL NOT NULL,
|
||||||
|
currency TEXT DEFAULT 'CNY',
|
||||||
|
description TEXT,
|
||||||
|
reference_id TEXT, -- 关联的订阅/支付/退款ID
|
||||||
|
balance_after REAL DEFAULT 0,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
metadata TEXT DEFAULT '{}',
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
|
);
|
||||||
|
|
||||||
|
-- 订阅相关索引
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_subscriptions_tenant ON subscriptions(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_subscriptions_status ON subscriptions(status);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_subscriptions_plan ON subscriptions(plan_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_usage_records_tenant ON usage_records(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_usage_records_type ON usage_records(resource_type);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_usage_records_recorded ON usage_records(recorded_at);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_payments_tenant ON payments(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_payments_status ON payments(status);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_payments_provider ON payments(provider);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_invoices_tenant ON invoices(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_invoices_status ON invoices(status);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_invoices_number ON invoices(invoice_number);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_refunds_tenant ON refunds(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_refunds_status ON refunds(status);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_refunds_payment ON refunds(payment_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_billing_history_tenant ON billing_history(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_billing_history_created ON billing_history(created_at);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_billing_history_type ON billing_history(type);
|
||||||
|
|
||||||
|
-- ============================================
|
||||||
|
-- Phase 8 Task 3: 企业级功能
|
||||||
|
-- ============================================
|
||||||
|
|
||||||
|
-- SSO 配置表
|
||||||
|
CREATE TABLE IF NOT EXISTS sso_configs (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
provider TEXT NOT NULL, -- wechat_work/dingtalk/feishu/okta/azure_ad/google/custom_saml
|
||||||
|
status TEXT DEFAULT 'disabled', -- disabled/pending/active/error
|
||||||
|
entity_id TEXT,
|
||||||
|
sso_url TEXT,
|
||||||
|
slo_url TEXT,
|
||||||
|
certificate TEXT, -- X.509 证书
|
||||||
|
metadata_url TEXT,
|
||||||
|
metadata_xml TEXT,
|
||||||
|
client_id TEXT,
|
||||||
|
client_secret TEXT,
|
||||||
|
authorization_url TEXT,
|
||||||
|
token_url TEXT,
|
||||||
|
userinfo_url TEXT,
|
||||||
|
scopes TEXT DEFAULT '["openid", "email", "profile"]',
|
||||||
|
attribute_mapping TEXT DEFAULT '{}', -- JSON: 属性映射
|
||||||
|
auto_provision INTEGER DEFAULT 1, -- 自动创建用户
|
||||||
|
default_role TEXT DEFAULT 'member',
|
||||||
|
domain_restriction TEXT DEFAULT '[]', -- JSON: 允许的邮箱域名
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
last_tested_at TIMESTAMP,
|
||||||
|
last_error TEXT,
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
|
);
|
||||||
|
|
||||||
|
-- SAML 认证请求表
|
||||||
|
CREATE TABLE IF NOT EXISTS saml_auth_requests (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
sso_config_id TEXT NOT NULL,
|
||||||
|
request_id TEXT NOT NULL UNIQUE,
|
||||||
|
relay_state TEXT,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
expires_at TIMESTAMP NOT NULL,
|
||||||
|
used INTEGER DEFAULT 0,
|
||||||
|
used_at TIMESTAMP,
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY (sso_config_id) REFERENCES sso_configs(id) ON DELETE CASCADE
|
||||||
|
);
|
||||||
|
|
||||||
|
-- SAML 认证响应表
|
||||||
|
CREATE TABLE IF NOT EXISTS saml_auth_responses (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
request_id TEXT NOT NULL,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
user_id TEXT,
|
||||||
|
email TEXT,
|
||||||
|
name TEXT,
|
||||||
|
attributes TEXT DEFAULT '{}', -- JSON: SAML 属性
|
||||||
|
session_index TEXT,
|
||||||
|
processed INTEGER DEFAULT 0,
|
||||||
|
processed_at TIMESTAMP,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
FOREIGN KEY (request_id) REFERENCES saml_auth_requests(request_id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
|
);
|
||||||
|
|
||||||
|
-- SCIM 配置表
|
||||||
|
CREATE TABLE IF NOT EXISTS scim_configs (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
provider TEXT NOT NULL,
|
||||||
|
status TEXT DEFAULT 'disabled',
|
||||||
|
scim_base_url TEXT,
|
||||||
|
scim_token TEXT,
|
||||||
|
sync_interval_minutes INTEGER DEFAULT 60,
|
||||||
|
last_sync_at TIMESTAMP,
|
||||||
|
last_sync_status TEXT,
|
||||||
|
last_sync_error TEXT,
|
||||||
|
last_sync_users_count INTEGER DEFAULT 0,
|
||||||
|
attribute_mapping TEXT DEFAULT '{}',
|
||||||
|
sync_rules TEXT DEFAULT '{}',
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
);
|
);
|
||||||
|
|
||||||
-- 租户成员表
|
-- SCIM 用户表
|
||||||
CREATE TABLE IF NOT EXISTS tenant_members (
|
CREATE TABLE IF NOT EXISTS scim_users (
|
||||||
id TEXT PRIMARY KEY,
|
id TEXT PRIMARY KEY,
|
||||||
tenant_id TEXT NOT NULL,
|
tenant_id TEXT NOT NULL,
|
||||||
user_id TEXT NOT NULL,
|
external_id TEXT NOT NULL,
|
||||||
|
user_name TEXT NOT NULL,
|
||||||
email TEXT NOT NULL,
|
email TEXT NOT NULL,
|
||||||
name TEXT DEFAULT '',
|
display_name TEXT,
|
||||||
role TEXT DEFAULT 'viewer', -- owner, admin, editor, viewer, guest
|
given_name TEXT,
|
||||||
status TEXT DEFAULT 'invited', -- active, invited, suspended, removed
|
family_name TEXT,
|
||||||
invited_by TEXT,
|
active INTEGER DEFAULT 1,
|
||||||
invited_at TEXT,
|
groups TEXT DEFAULT '[]',
|
||||||
invitation_token TEXT,
|
raw_data TEXT DEFAULT '{}',
|
||||||
invitation_expires_at TEXT,
|
synced_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
joined_at TEXT,
|
|
||||||
last_active_at TEXT,
|
|
||||||
custom_permissions TEXT DEFAULT '[]', -- JSON 数组
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
UNIQUE(tenant_id, user_id),
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE,
|
||||||
|
UNIQUE(tenant_id, external_id)
|
||||||
|
);
|
||||||
|
|
||||||
|
-- 审计日志导出表
|
||||||
|
CREATE TABLE IF NOT EXISTS audit_log_exports (
|
||||||
|
id TEXT PRIMARY KEY,
|
||||||
|
tenant_id TEXT NOT NULL,
|
||||||
|
export_format TEXT NOT NULL, -- json/csv/pdf/xlsx
|
||||||
|
start_date TIMESTAMP NOT NULL,
|
||||||
|
end_date TIMESTAMP NOT NULL,
|
||||||
|
filters TEXT DEFAULT '{}',
|
||||||
|
compliance_standard TEXT, -- soc2/iso27001/gdpr/hipaa/pci_dss
|
||||||
|
status TEXT DEFAULT 'pending', -- pending/processing/completed/failed
|
||||||
|
file_path TEXT,
|
||||||
|
file_size INTEGER,
|
||||||
|
record_count INTEGER,
|
||||||
|
checksum TEXT,
|
||||||
|
downloaded_by TEXT,
|
||||||
|
downloaded_at TIMESTAMP,
|
||||||
|
expires_at TIMESTAMP,
|
||||||
|
created_by TEXT NOT NULL,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
completed_at TIMESTAMP,
|
||||||
|
error_message TEXT,
|
||||||
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
);
|
);
|
||||||
|
|
||||||
-- 租户角色表
|
-- 数据保留策略表
|
||||||
CREATE TABLE IF NOT EXISTS tenant_roles (
|
CREATE TABLE IF NOT EXISTS data_retention_policies (
|
||||||
id TEXT PRIMARY KEY,
|
id TEXT PRIMARY KEY,
|
||||||
tenant_id TEXT NOT NULL,
|
tenant_id TEXT NOT NULL,
|
||||||
name TEXT NOT NULL,
|
name TEXT NOT NULL,
|
||||||
description TEXT DEFAULT '',
|
description TEXT,
|
||||||
permissions TEXT DEFAULT '[]', -- JSON 数组
|
resource_type TEXT NOT NULL, -- project/transcript/entity/audit_log/user_data
|
||||||
is_system INTEGER DEFAULT 0, -- 1=系统预设, 0=自定义
|
retention_days INTEGER NOT NULL,
|
||||||
|
action TEXT NOT NULL, -- archive/delete/anonymize
|
||||||
|
conditions TEXT DEFAULT '{}',
|
||||||
|
auto_execute INTEGER DEFAULT 0,
|
||||||
|
execute_at TEXT, -- cron 表达式
|
||||||
|
notify_before_days INTEGER DEFAULT 7,
|
||||||
|
archive_location TEXT,
|
||||||
|
archive_encryption INTEGER DEFAULT 1,
|
||||||
|
is_active INTEGER DEFAULT 1,
|
||||||
|
last_executed_at TIMESTAMP,
|
||||||
|
last_execution_result TEXT,
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
);
|
);
|
||||||
|
|
||||||
-- 租户相关索引
|
-- 数据保留任务表
|
||||||
CREATE INDEX IF NOT EXISTS idx_tenants_slug ON tenants(slug);
|
CREATE TABLE IF NOT EXISTS data_retention_jobs (
|
||||||
CREATE INDEX IF NOT EXISTS idx_tenants_status ON tenants(status);
|
id TEXT PRIMARY KEY,
|
||||||
CREATE INDEX IF NOT EXISTS idx_domains_tenant ON tenant_domains(tenant_id);
|
policy_id TEXT NOT NULL,
|
||||||
CREATE INDEX IF NOT EXISTS idx_domains_domain ON tenant_domains(domain);
|
tenant_id TEXT NOT NULL,
|
||||||
CREATE INDEX IF NOT EXISTS idx_domains_status ON tenant_domains(status);
|
status TEXT DEFAULT 'pending', -- pending/running/completed/failed
|
||||||
CREATE INDEX IF NOT EXISTS idx_members_tenant ON tenant_members(tenant_id);
|
started_at TIMESTAMP,
|
||||||
CREATE INDEX IF NOT EXISTS idx_members_user ON tenant_members(user_id);
|
completed_at TIMESTAMP,
|
||||||
CREATE INDEX IF NOT EXISTS idx_members_role ON tenant_members(role);
|
affected_records INTEGER DEFAULT 0,
|
||||||
CREATE INDEX IF NOT EXISTS idx_members_status ON tenant_members(status);
|
archived_records INTEGER DEFAULT 0,
|
||||||
CREATE INDEX IF NOT EXISTS idx_members_token ON tenant_members(invitation_token);
|
deleted_records INTEGER DEFAULT 0,
|
||||||
CREATE INDEX IF NOT EXISTS idx_roles_tenant ON tenant_roles(tenant_id);
|
error_count INTEGER DEFAULT 0,
|
||||||
|
details TEXT DEFAULT '{}',
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
FOREIGN KEY (policy_id) REFERENCES data_retention_policies(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY (tenant_id) REFERENCES tenants(id) ON DELETE CASCADE
|
||||||
|
);
|
||||||
|
|
||||||
-- 更新项目表,添加租户关联(可选,支持租户隔离)
|
-- 企业级功能相关索引
|
||||||
ALTER TABLE projects ADD COLUMN tenant_id TEXT;
|
CREATE INDEX IF NOT EXISTS idx_sso_tenant ON sso_configs(tenant_id);
|
||||||
CREATE INDEX IF NOT EXISTS idx_projects_tenant ON projects(tenant_id);
|
CREATE INDEX IF NOT EXISTS idx_sso_provider ON sso_configs(provider);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_saml_requests_config ON saml_auth_requests(sso_config_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_saml_requests_expires ON saml_auth_requests(expires_at);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_saml_responses_request ON saml_auth_responses(request_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_scim_config_tenant ON scim_configs(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_scim_users_tenant ON scim_users(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_scim_users_external ON scim_users(external_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_audit_export_tenant ON audit_log_exports(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_audit_export_status ON audit_log_exports(status);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_retention_tenant ON data_retention_policies(tenant_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_retention_type ON data_retention_policies(resource_type);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_retention_jobs_policy ON data_retention_jobs(policy_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_retention_jobs_status ON data_retention_jobs(status);
|
||||||
|
|||||||
1840
backend/subscription_manager.py
Normal file
1840
backend/subscription_manager.py
Normal file
File diff suppressed because it is too large
Load Diff
246
backend/test_phase8_task2.py
Normal file
246
backend/test_phase8_task2.py
Normal file
@@ -0,0 +1,246 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
InsightFlow Phase 8 Task 2 测试脚本 - 订阅与计费系统
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import tempfile
|
||||||
|
|
||||||
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||||||
|
|
||||||
|
from subscription_manager import (
|
||||||
|
get_subscription_manager, SubscriptionManager,
|
||||||
|
SubscriptionStatus, PaymentProvider, PaymentStatus, InvoiceStatus, RefundStatus
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_subscription_manager():
|
||||||
|
"""测试订阅管理器"""
|
||||||
|
print("=" * 60)
|
||||||
|
print("InsightFlow Phase 8 Task 2 - 订阅与计费系统测试")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
# 使用临时文件数据库进行测试
|
||||||
|
db_path = tempfile.mktemp(suffix='.db')
|
||||||
|
|
||||||
|
try:
|
||||||
|
manager = SubscriptionManager(db_path=db_path)
|
||||||
|
|
||||||
|
print("\n1. 测试订阅计划管理")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
# 获取默认计划
|
||||||
|
plans = manager.list_plans()
|
||||||
|
print(f"✓ 默认计划数量: {len(plans)}")
|
||||||
|
for plan in plans:
|
||||||
|
print(f" - {plan.name} ({plan.tier}): ¥{plan.price_monthly}/月")
|
||||||
|
|
||||||
|
# 通过 tier 获取计划
|
||||||
|
free_plan = manager.get_plan_by_tier("free")
|
||||||
|
pro_plan = manager.get_plan_by_tier("pro")
|
||||||
|
enterprise_plan = manager.get_plan_by_tier("enterprise")
|
||||||
|
|
||||||
|
assert free_plan is not None, "Free 计划应该存在"
|
||||||
|
assert pro_plan is not None, "Pro 计划应该存在"
|
||||||
|
assert enterprise_plan is not None, "Enterprise 计划应该存在"
|
||||||
|
|
||||||
|
print(f"✓ Free 计划: {free_plan.name}")
|
||||||
|
print(f"✓ Pro 计划: {pro_plan.name}")
|
||||||
|
print(f"✓ Enterprise 计划: {enterprise_plan.name}")
|
||||||
|
|
||||||
|
print("\n2. 测试订阅管理")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
tenant_id = "test-tenant-001"
|
||||||
|
|
||||||
|
# 创建订阅
|
||||||
|
subscription = manager.create_subscription(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
plan_id=pro_plan.id,
|
||||||
|
payment_provider=PaymentProvider.STRIPE.value,
|
||||||
|
trial_days=14
|
||||||
|
)
|
||||||
|
|
||||||
|
print(f"✓ 创建订阅: {subscription.id}")
|
||||||
|
print(f" - 状态: {subscription.status}")
|
||||||
|
print(f" - 计划: {pro_plan.name}")
|
||||||
|
print(f" - 试用开始: {subscription.trial_start}")
|
||||||
|
print(f" - 试用结束: {subscription.trial_end}")
|
||||||
|
|
||||||
|
# 获取租户订阅
|
||||||
|
tenant_sub = manager.get_tenant_subscription(tenant_id)
|
||||||
|
assert tenant_sub is not None, "应该能获取到租户订阅"
|
||||||
|
print(f"✓ 获取租户订阅: {tenant_sub.id}")
|
||||||
|
|
||||||
|
print("\n3. 测试用量记录")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
# 记录转录用量
|
||||||
|
usage1 = manager.record_usage(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
resource_type="transcription",
|
||||||
|
quantity=120,
|
||||||
|
unit="minute",
|
||||||
|
description="会议转录"
|
||||||
|
)
|
||||||
|
print(f"✓ 记录转录用量: {usage1.quantity} {usage1.unit}, 费用: ¥{usage1.cost:.2f}")
|
||||||
|
|
||||||
|
# 记录存储用量
|
||||||
|
usage2 = manager.record_usage(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
resource_type="storage",
|
||||||
|
quantity=2.5,
|
||||||
|
unit="gb",
|
||||||
|
description="文件存储"
|
||||||
|
)
|
||||||
|
print(f"✓ 记录存储用量: {usage2.quantity} {usage2.unit}, 费用: ¥{usage2.cost:.2f}")
|
||||||
|
|
||||||
|
# 获取用量汇总
|
||||||
|
summary = manager.get_usage_summary(tenant_id)
|
||||||
|
print(f"✓ 用量汇总:")
|
||||||
|
print(f" - 总费用: ¥{summary['total_cost']:.2f}")
|
||||||
|
for resource, data in summary['breakdown'].items():
|
||||||
|
print(f" - {resource}: {data['quantity']} (¥{data['cost']:.2f})")
|
||||||
|
|
||||||
|
print("\n4. 测试支付管理")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
# 创建支付
|
||||||
|
payment = manager.create_payment(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
amount=99.0,
|
||||||
|
currency="CNY",
|
||||||
|
provider=PaymentProvider.ALIPAY.value,
|
||||||
|
payment_method="qrcode"
|
||||||
|
)
|
||||||
|
print(f"✓ 创建支付: {payment.id}")
|
||||||
|
print(f" - 金额: ¥{payment.amount}")
|
||||||
|
print(f" - 提供商: {payment.provider}")
|
||||||
|
print(f" - 状态: {payment.status}")
|
||||||
|
|
||||||
|
# 确认支付
|
||||||
|
confirmed = manager.confirm_payment(payment.id, "alipay_123456")
|
||||||
|
print(f"✓ 确认支付完成: {confirmed.status}")
|
||||||
|
|
||||||
|
# 列出支付记录
|
||||||
|
payments = manager.list_payments(tenant_id)
|
||||||
|
print(f"✓ 支付记录数量: {len(payments)}")
|
||||||
|
|
||||||
|
print("\n5. 测试发票管理")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
# 列出发票
|
||||||
|
invoices = manager.list_invoices(tenant_id)
|
||||||
|
print(f"✓ 发票数量: {len(invoices)}")
|
||||||
|
|
||||||
|
if invoices:
|
||||||
|
invoice = invoices[0]
|
||||||
|
print(f" - 发票号: {invoice.invoice_number}")
|
||||||
|
print(f" - 金额: ¥{invoice.amount_due}")
|
||||||
|
print(f" - 状态: {invoice.status}")
|
||||||
|
|
||||||
|
print("\n6. 测试退款管理")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
# 申请退款
|
||||||
|
refund = manager.request_refund(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
payment_id=payment.id,
|
||||||
|
amount=50.0,
|
||||||
|
reason="服务不满意",
|
||||||
|
requested_by="user_001"
|
||||||
|
)
|
||||||
|
print(f"✓ 申请退款: {refund.id}")
|
||||||
|
print(f" - 金额: ¥{refund.amount}")
|
||||||
|
print(f" - 原因: {refund.reason}")
|
||||||
|
print(f" - 状态: {refund.status}")
|
||||||
|
|
||||||
|
# 批准退款
|
||||||
|
approved = manager.approve_refund(refund.id, "admin_001")
|
||||||
|
print(f"✓ 批准退款: {approved.status}")
|
||||||
|
|
||||||
|
# 完成退款
|
||||||
|
completed = manager.complete_refund(refund.id, "refund_123456")
|
||||||
|
print(f"✓ 完成退款: {completed.status}")
|
||||||
|
|
||||||
|
# 列出退款记录
|
||||||
|
refunds = manager.list_refunds(tenant_id)
|
||||||
|
print(f"✓ 退款记录数量: {len(refunds)}")
|
||||||
|
|
||||||
|
print("\n7. 测试账单历史")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
history = manager.get_billing_history(tenant_id)
|
||||||
|
print(f"✓ 账单历史记录数量: {len(history)}")
|
||||||
|
for h in history:
|
||||||
|
print(f" - [{h.type}] {h.description}: ¥{h.amount}")
|
||||||
|
|
||||||
|
print("\n8. 测试支付提供商集成")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
# Stripe Checkout
|
||||||
|
stripe_session = manager.create_stripe_checkout_session(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
plan_id=enterprise_plan.id,
|
||||||
|
success_url="https://example.com/success",
|
||||||
|
cancel_url="https://example.com/cancel"
|
||||||
|
)
|
||||||
|
print(f"✓ Stripe Checkout 会话: {stripe_session['session_id']}")
|
||||||
|
|
||||||
|
# 支付宝订单
|
||||||
|
alipay_order = manager.create_alipay_order(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
plan_id=pro_plan.id
|
||||||
|
)
|
||||||
|
print(f"✓ 支付宝订单: {alipay_order['order_id']}")
|
||||||
|
|
||||||
|
# 微信支付订单
|
||||||
|
wechat_order = manager.create_wechat_order(
|
||||||
|
tenant_id=tenant_id,
|
||||||
|
plan_id=pro_plan.id
|
||||||
|
)
|
||||||
|
print(f"✓ 微信支付订单: {wechat_order['order_id']}")
|
||||||
|
|
||||||
|
# Webhook 处理
|
||||||
|
webhook_result = manager.handle_webhook("stripe", {
|
||||||
|
"event_type": "checkout.session.completed",
|
||||||
|
"data": {"object": {"id": "cs_test"}}
|
||||||
|
})
|
||||||
|
print(f"✓ Webhook 处理: {webhook_result}")
|
||||||
|
|
||||||
|
print("\n9. 测试订阅变更")
|
||||||
|
print("-" * 40)
|
||||||
|
|
||||||
|
# 更改计划
|
||||||
|
changed = manager.change_plan(
|
||||||
|
subscription_id=subscription.id,
|
||||||
|
new_plan_id=enterprise_plan.id
|
||||||
|
)
|
||||||
|
print(f"✓ 更改计划: {changed.plan_id} (Enterprise)")
|
||||||
|
|
||||||
|
# 取消订阅
|
||||||
|
cancelled = manager.cancel_subscription(
|
||||||
|
subscription_id=subscription.id,
|
||||||
|
at_period_end=True
|
||||||
|
)
|
||||||
|
print(f"✓ 取消订阅: {cancelled.status}")
|
||||||
|
print(f" - 周期结束时取消: {cancelled.cancel_at_period_end}")
|
||||||
|
|
||||||
|
print("\n" + "=" * 60)
|
||||||
|
print("所有测试通过! ✓")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
finally:
|
||||||
|
# 清理临时数据库
|
||||||
|
if os.path.exists(db_path):
|
||||||
|
os.remove(db_path)
|
||||||
|
print(f"\n清理临时数据库: {db_path}")
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
try:
|
||||||
|
test_subscription_manager()
|
||||||
|
except Exception as e:
|
||||||
|
print(f"\n❌ 测试失败: {e}")
|
||||||
|
import traceback
|
||||||
|
traceback.print_exc()
|
||||||
|
sys.exit(1)
|
||||||
@@ -1,507 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
"""
|
|
||||||
InsightFlow Phase 8 - Multi-Tenant SaaS Test Script
|
|
||||||
多租户 SaaS 架构测试脚本
|
|
||||||
"""
|
|
||||||
|
|
||||||
import sys
|
|
||||||
import os
|
|
||||||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
||||||
|
|
||||||
from tenant_manager import (
|
|
||||||
get_tenant_manager, TenantManager, Tenant, TenantDomain, TenantBranding,
|
|
||||||
TenantMember, TenantStatus, TenantTier, TenantRole, DomainStatus,
|
|
||||||
TenantContext
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def test_tenant_management():
|
|
||||||
"""测试租户管理功能"""
|
|
||||||
print("=" * 60)
|
|
||||||
print("测试租户管理功能")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
# 使用测试数据库
|
|
||||||
test_db = "test_tenant.db"
|
|
||||||
if os.path.exists(test_db):
|
|
||||||
os.remove(test_db)
|
|
||||||
|
|
||||||
manager = get_tenant_manager(test_db)
|
|
||||||
|
|
||||||
# 1. 创建租户
|
|
||||||
print("\n1. 创建租户...")
|
|
||||||
try:
|
|
||||||
tenant = manager.create_tenant(
|
|
||||||
name="Test Company",
|
|
||||||
owner_id="user_001",
|
|
||||||
tier="pro",
|
|
||||||
description="A test tenant for validation",
|
|
||||||
settings={"theme": "dark"}
|
|
||||||
)
|
|
||||||
print(f" ✓ 租户创建成功: {tenant.id}")
|
|
||||||
print(f" - 名称: {tenant.name}")
|
|
||||||
print(f" - Slug: {tenant.slug}")
|
|
||||||
print(f" - 层级: {tenant.tier}")
|
|
||||||
print(f" - 状态: {tenant.status}")
|
|
||||||
print(f" - 资源限制: {tenant.resource_limits}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 租户创建失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 2. 获取租户
|
|
||||||
print("\n2. 获取租户...")
|
|
||||||
try:
|
|
||||||
fetched = manager.get_tenant(tenant.id)
|
|
||||||
assert fetched is not None
|
|
||||||
assert fetched.name == tenant.name
|
|
||||||
print(f" ✓ 通过 ID 获取租户成功")
|
|
||||||
|
|
||||||
fetched_by_slug = manager.get_tenant_by_slug(tenant.slug)
|
|
||||||
assert fetched_by_slug is not None
|
|
||||||
assert fetched_by_slug.id == tenant.id
|
|
||||||
print(f" ✓ 通过 Slug 获取租户成功")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 获取租户失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 3. 更新租户
|
|
||||||
print("\n3. 更新租户...")
|
|
||||||
try:
|
|
||||||
updated = manager.update_tenant(
|
|
||||||
tenant.id,
|
|
||||||
name="Test Company Updated",
|
|
||||||
tier="enterprise"
|
|
||||||
)
|
|
||||||
assert updated is not None
|
|
||||||
assert updated.name == "Test Company Updated"
|
|
||||||
assert updated.tier == "enterprise"
|
|
||||||
print(f" ✓ 租户更新成功")
|
|
||||||
print(f" - 新名称: {updated.name}")
|
|
||||||
print(f" - 新层级: {updated.tier}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 租户更新失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 4. 列出租户
|
|
||||||
print("\n4. 列出租户...")
|
|
||||||
try:
|
|
||||||
tenants = manager.list_tenants()
|
|
||||||
assert len(tenants) >= 1
|
|
||||||
print(f" ✓ 列出租户成功,共 {len(tenants)} 个租户")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 列出租户失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
return tenant.id
|
|
||||||
|
|
||||||
|
|
||||||
def test_domain_management(tenant_id: str):
|
|
||||||
"""测试域名管理功能"""
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("测试域名管理功能")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
manager = get_tenant_manager("test_tenant.db")
|
|
||||||
|
|
||||||
# 1. 添加域名
|
|
||||||
print("\n1. 添加自定义域名...")
|
|
||||||
try:
|
|
||||||
domain = manager.add_domain(tenant_id, "app.example.com", is_primary=True)
|
|
||||||
print(f" ✓ 域名添加成功: {domain.id}")
|
|
||||||
print(f" - 域名: {domain.domain}")
|
|
||||||
print(f" - 状态: {domain.status}")
|
|
||||||
print(f" - 验证令牌: {domain.verification_token}")
|
|
||||||
print(f" - 是否主域名: {domain.is_primary}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 域名添加失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 2. 获取域名验证指导
|
|
||||||
print("\n2. 获取域名验证指导...")
|
|
||||||
try:
|
|
||||||
instructions = manager.get_domain_verification_instructions(domain.id)
|
|
||||||
assert instructions is not None
|
|
||||||
print(f" ✓ 获取验证指导成功")
|
|
||||||
print(f" - DNS 记录: {instructions['dns_record']}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 获取验证指导失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 3. 验证域名
|
|
||||||
print("\n3. 验证域名...")
|
|
||||||
try:
|
|
||||||
success = manager.verify_domain(tenant_id, domain.id)
|
|
||||||
if success:
|
|
||||||
print(f" ✓ 域名验证成功")
|
|
||||||
else:
|
|
||||||
print(f" ! 域名验证返回 False(可能是模拟验证)")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 域名验证失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 4. 获取域名列表
|
|
||||||
print("\n4. 获取域名列表...")
|
|
||||||
try:
|
|
||||||
domains = manager.list_domains(tenant_id)
|
|
||||||
assert len(domains) >= 1
|
|
||||||
print(f" ✓ 获取域名列表成功,共 {len(domains)} 个域名")
|
|
||||||
for d in domains:
|
|
||||||
print(f" - {d.domain} ({d.status})")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 获取域名列表失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 5. 通过域名获取租户
|
|
||||||
print("\n5. 通过域名解析租户...")
|
|
||||||
try:
|
|
||||||
resolved = manager.get_tenant_by_domain("app.example.com")
|
|
||||||
if resolved:
|
|
||||||
assert resolved.id == tenant_id
|
|
||||||
print(f" ✓ 域名解析租户成功")
|
|
||||||
else:
|
|
||||||
print(f" ! 域名解析租户返回 None(可能域名未激活)")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 域名解析失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
def test_branding_management(tenant_id: str):
|
|
||||||
"""测试品牌配置管理功能"""
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("测试品牌配置管理功能")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
manager = get_tenant_manager("test_tenant.db")
|
|
||||||
|
|
||||||
# 1. 更新品牌配置
|
|
||||||
print("\n1. 更新品牌配置...")
|
|
||||||
try:
|
|
||||||
branding = manager.update_branding(
|
|
||||||
tenant_id,
|
|
||||||
logo_url="https://example.com/logo.png",
|
|
||||||
favicon_url="https://example.com/favicon.ico",
|
|
||||||
primary_color="#FF5733",
|
|
||||||
secondary_color="#33FF57",
|
|
||||||
custom_css="body { font-size: 14px; }",
|
|
||||||
custom_js="console.log('Custom JS loaded');"
|
|
||||||
)
|
|
||||||
assert branding is not None
|
|
||||||
print(f" ✓ 品牌配置更新成功")
|
|
||||||
print(f" - Logo: {branding.logo_url}")
|
|
||||||
print(f" - 主色调: {branding.primary_color}")
|
|
||||||
print(f" - 次色调: {branding.secondary_color}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 品牌配置更新失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 2. 获取品牌配置
|
|
||||||
print("\n2. 获取品牌配置...")
|
|
||||||
try:
|
|
||||||
fetched = manager.get_branding(tenant_id)
|
|
||||||
assert fetched is not None
|
|
||||||
assert fetched.primary_color == "#FF5733"
|
|
||||||
print(f" ✓ 获取品牌配置成功")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 获取品牌配置失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 3. 生成品牌 CSS
|
|
||||||
print("\n3. 生成品牌 CSS...")
|
|
||||||
try:
|
|
||||||
css = manager.get_branding_css(tenant_id)
|
|
||||||
assert "--tenant-primary" in css
|
|
||||||
assert "#FF5733" in css
|
|
||||||
print(f" ✓ 品牌 CSS 生成成功")
|
|
||||||
print(f" - CSS 长度: {len(css)} 字符")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 品牌 CSS 生成失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
def test_member_management(tenant_id: str):
|
|
||||||
"""测试成员管理功能"""
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("测试成员管理功能")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
manager = get_tenant_manager("test_tenant.db")
|
|
||||||
|
|
||||||
# 1. 邀请成员
|
|
||||||
print("\n1. 邀请成员...")
|
|
||||||
try:
|
|
||||||
member = manager.invite_member(
|
|
||||||
tenant_id=tenant_id,
|
|
||||||
email="user@example.com",
|
|
||||||
role="admin",
|
|
||||||
invited_by="user_001"
|
|
||||||
)
|
|
||||||
print(f" ✓ 成员邀请成功: {member.id}")
|
|
||||||
print(f" - 邮箱: {member.email}")
|
|
||||||
print(f" - 角色: {member.role}")
|
|
||||||
print(f" - 状态: {member.status}")
|
|
||||||
print(f" - 权限: {member.permissions}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 成员邀请失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 2. 获取成员列表
|
|
||||||
print("\n2. 获取成员列表...")
|
|
||||||
try:
|
|
||||||
members = manager.list_members(tenant_id)
|
|
||||||
assert len(members) >= 2 # owner + invited member
|
|
||||||
print(f" ✓ 获取成员列表成功,共 {len(members)} 个成员")
|
|
||||||
for m in members:
|
|
||||||
print(f" - {m.email} ({m.role}, {m.status})")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 获取成员列表失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 3. 接受邀请
|
|
||||||
print("\n3. 接受邀请...")
|
|
||||||
try:
|
|
||||||
# 注意:accept_invitation 使用的是 member id 而不是 token
|
|
||||||
# 修正:查看源码后发现它接受的是 invitation_id(即 member id)
|
|
||||||
accepted = manager.accept_invitation(member.id, "user_002")
|
|
||||||
if accepted:
|
|
||||||
print(f" ✓ 邀请接受成功")
|
|
||||||
else:
|
|
||||||
print(f" ! 邀请接受返回 False(可能是状态不对)")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 邀请接受失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 4. 更新成员角色
|
|
||||||
print("\n4. 更新成员角色...")
|
|
||||||
try:
|
|
||||||
success = manager.update_member_role(
|
|
||||||
tenant_id=tenant_id,
|
|
||||||
member_id=member.id,
|
|
||||||
role="member"
|
|
||||||
)
|
|
||||||
if success:
|
|
||||||
print(f" ✓ 成员角色更新成功")
|
|
||||||
else:
|
|
||||||
print(f" ! 成员角色更新返回 False")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 成员角色更新失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 5. 检查权限
|
|
||||||
print("\n5. 检查用户权限...")
|
|
||||||
try:
|
|
||||||
# 检查 owner 权限
|
|
||||||
has_permission = manager.check_permission(
|
|
||||||
tenant_id=tenant_id,
|
|
||||||
user_id="user_001",
|
|
||||||
resource="project",
|
|
||||||
action="create"
|
|
||||||
)
|
|
||||||
print(f" ✓ 权限检查成功")
|
|
||||||
print(f" - Owner 是否有 project:create 权限: {has_permission}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 权限检查失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 6. 获取用户租户列表
|
|
||||||
print("\n6. 获取用户租户列表...")
|
|
||||||
try:
|
|
||||||
user_tenants = manager.get_user_tenants("user_001")
|
|
||||||
assert len(user_tenants) >= 1
|
|
||||||
print(f" ✓ 获取用户租户列表成功,共 {len(user_tenants)} 个租户")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 获取用户租户列表失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
def test_usage_stats(tenant_id: str):
|
|
||||||
"""测试使用统计功能"""
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("测试使用统计功能")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
manager = get_tenant_manager("test_tenant.db")
|
|
||||||
|
|
||||||
# 1. 记录使用
|
|
||||||
print("\n1. 记录资源使用...")
|
|
||||||
try:
|
|
||||||
manager.record_usage(
|
|
||||||
tenant_id=tenant_id,
|
|
||||||
storage_bytes=1024 * 1024 * 50, # 50MB
|
|
||||||
transcription_seconds=600, # 10分钟
|
|
||||||
api_calls=100,
|
|
||||||
projects_count=5,
|
|
||||||
entities_count=50,
|
|
||||||
members_count=3
|
|
||||||
)
|
|
||||||
print(f" ✓ 资源使用记录成功")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 资源使用记录失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 2. 获取使用统计
|
|
||||||
print("\n2. 获取使用统计...")
|
|
||||||
try:
|
|
||||||
stats = manager.get_usage_stats(tenant_id)
|
|
||||||
print(f" ✓ 使用统计获取成功")
|
|
||||||
print(f" - 存储: {stats['storage_mb']:.2f} MB")
|
|
||||||
print(f" - 转录: {stats['transcription_minutes']:.2f} 分钟")
|
|
||||||
print(f" - API 调用: {stats['api_calls']}")
|
|
||||||
print(f" - 项目数: {stats['projects_count']}")
|
|
||||||
print(f" - 实体数: {stats['entities_count']}")
|
|
||||||
print(f" - 成员数: {stats['members_count']}")
|
|
||||||
print(f" - 配额: {stats['limits']}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 使用统计获取失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 3. 检查资源限制
|
|
||||||
print("\n3. 检查资源限制...")
|
|
||||||
try:
|
|
||||||
allowed, current, limit = manager.check_resource_limit(tenant_id, "storage")
|
|
||||||
print(f" ✓ 资源限制检查成功")
|
|
||||||
print(f" - 存储: {allowed}, 当前: {current}, 限制: {limit}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 资源限制检查失败: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
return False
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
def test_tenant_context():
|
|
||||||
"""测试租户上下文管理"""
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("测试租户上下文管理")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
# 1. 设置和获取租户上下文
|
|
||||||
print("\n1. 设置和获取租户上下文...")
|
|
||||||
try:
|
|
||||||
TenantContext.set_current_tenant("tenant_123")
|
|
||||||
tenant_id = TenantContext.get_current_tenant()
|
|
||||||
assert tenant_id == "tenant_123"
|
|
||||||
print(f" ✓ 租户上下文设置成功: {tenant_id}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 租户上下文设置失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 2. 设置和获取用户上下文
|
|
||||||
print("\n2. 设置和获取用户上下文...")
|
|
||||||
try:
|
|
||||||
TenantContext.set_current_user("user_456")
|
|
||||||
user_id = TenantContext.get_current_user()
|
|
||||||
assert user_id == "user_456"
|
|
||||||
print(f" ✓ 用户上下文设置成功: {user_id}")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 用户上下文设置失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
# 3. 清除上下文
|
|
||||||
print("\n3. 清除上下文...")
|
|
||||||
try:
|
|
||||||
TenantContext.clear()
|
|
||||||
assert TenantContext.get_current_tenant() is None
|
|
||||||
assert TenantContext.get_current_user() is None
|
|
||||||
print(f" ✓ 上下文清除成功")
|
|
||||||
except Exception as e:
|
|
||||||
print(f" ✗ 上下文清除失败: {e}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
def cleanup():
|
|
||||||
"""清理测试数据"""
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("清理测试数据")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
test_db = "test_tenant.db"
|
|
||||||
if os.path.exists(test_db):
|
|
||||||
os.remove(test_db)
|
|
||||||
print(f"✓ 删除测试数据库: {test_db}")
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
|
||||||
"""主测试函数"""
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("InsightFlow Phase 8 - Multi-Tenant SaaS 测试")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
all_passed = True
|
|
||||||
tenant_id = None
|
|
||||||
|
|
||||||
try:
|
|
||||||
# 测试租户上下文
|
|
||||||
if not test_tenant_context():
|
|
||||||
all_passed = False
|
|
||||||
|
|
||||||
# 测试租户管理
|
|
||||||
tenant_id = test_tenant_management()
|
|
||||||
if not tenant_id:
|
|
||||||
all_passed = False
|
|
||||||
|
|
||||||
# 测试域名管理
|
|
||||||
if not test_domain_management(tenant_id):
|
|
||||||
all_passed = False
|
|
||||||
|
|
||||||
# 测试品牌配置
|
|
||||||
if not test_branding_management(tenant_id):
|
|
||||||
all_passed = False
|
|
||||||
|
|
||||||
# 测试成员管理
|
|
||||||
if not test_member_management(tenant_id):
|
|
||||||
all_passed = False
|
|
||||||
|
|
||||||
# 测试使用统计
|
|
||||||
if not test_usage_stats(tenant_id):
|
|
||||||
all_passed = False
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
print(f"\n测试过程中发生错误: {e}")
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
all_passed = False
|
|
||||||
|
|
||||||
finally:
|
|
||||||
cleanup()
|
|
||||||
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
if all_passed:
|
|
||||||
print("✓ 所有测试通过!")
|
|
||||||
else:
|
|
||||||
print("✗ 部分测试失败")
|
|
||||||
print("=" * 60)
|
|
||||||
|
|
||||||
return 0 if all_passed else 1
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
sys.exit(main())
|
|
||||||
Reference in New Issue
Block a user