122 lines
5.0 KiB
Python
122 lines
5.0 KiB
Python
|
# database/models.py - 数据模型定义（SQLite 建表 DDL）
|
|||
|
|
from dataclasses import dataclass, field
|
|||
|
|
from datetime import datetime
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
# ══════════════════════════════════════════════════════
|
|||
|
|
# DDL 建表语句
|
|||
|
|
# ══════════════════════════════════════════════════════
|
|||
|
|
|
|||
|
|
CREATE_TABLES_SQL = """
|
|||
|
|
PRAGMA foreign_keys = ON;
|
|||
|
|
|
|||
|
|
-- 项目表
|
|||
|
|
CREATE TABLE IF NOT EXISTS projects (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
name TEXT NOT NULL UNIQUE, -- 项目名称
|
|||
|
|
description TEXT, -- 项目描述
|
|||
|
|
language TEXT NOT NULL DEFAULT 'python', -- 目标代码语言
|
|||
|
|
output_dir TEXT NOT NULL, -- 输出目录路径
|
|||
|
|
created_at TEXT NOT NULL,
|
|||
|
|
updated_at TEXT NOT NULL
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
-- 原始需求表
|
|||
|
|
CREATE TABLE IF NOT EXISTS raw_requirements (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
project_id INTEGER NOT NULL, -- 关联项目
|
|||
|
|
content TEXT NOT NULL, -- 需求原文
|
|||
|
|
source_type TEXT NOT NULL DEFAULT 'text', -- text | file
|
|||
|
|
source_name TEXT, -- 文件名（文件输入时）
|
|||
|
|
knowledge TEXT, -- 合并后的知识库内容
|
|||
|
|
created_at TEXT NOT NULL,
|
|||
|
|
FOREIGN KEY (project_id) REFERENCES projects(id) ON DELETE CASCADE
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
-- 功能需求表
|
|||
|
|
CREATE TABLE IF NOT EXISTS functional_requirements (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
project_id INTEGER NOT NULL,
|
|||
|
|
raw_req_id INTEGER NOT NULL, -- 关联原始需求
|
|||
|
|
index_no INTEGER NOT NULL, -- 序号
|
|||
|
|
title TEXT NOT NULL, -- 功能标题
|
|||
|
|
description TEXT NOT NULL, -- 功能描述
|
|||
|
|
function_name TEXT NOT NULL, -- 对应函数名
|
|||
|
|
priority TEXT NOT NULL DEFAULT 'medium', -- high|medium|low
|
|||
|
|
status TEXT NOT NULL DEFAULT 'pending', -- pending|generated|skipped
|
|||
|
|
is_custom INTEGER NOT NULL DEFAULT 0, -- 是否用户自定义 (0/1)
|
|||
|
|
created_at TEXT NOT NULL,
|
|||
|
|
updated_at TEXT NOT NULL,
|
|||
|
|
FOREIGN KEY (project_id) REFERENCES projects(id) ON DELETE CASCADE,
|
|||
|
|
FOREIGN KEY (raw_req_id) REFERENCES raw_requirements(id) ON DELETE CASCADE
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
-- 代码文件表
|
|||
|
|
CREATE TABLE IF NOT EXISTS code_files (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
project_id INTEGER NOT NULL,
|
|||
|
|
func_req_id INTEGER NOT NULL UNIQUE, -- 关联功能需求（1对1）
|
|||
|
|
file_name TEXT NOT NULL, -- 文件名
|
|||
|
|
file_path TEXT NOT NULL, -- 完整路径
|
|||
|
|
language TEXT NOT NULL, -- 代码语言
|
|||
|
|
content TEXT NOT NULL, -- 代码内容
|
|||
|
|
created_at TEXT NOT NULL,
|
|||
|
|
updated_at TEXT NOT NULL,
|
|||
|
|
FOREIGN KEY (project_id) REFERENCES projects(id) ON DELETE CASCADE,
|
|||
|
|
FOREIGN KEY (func_req_id) REFERENCES functional_requirements(id) ON DELETE CASCADE
|
|||
|
|
);
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# ══════════════════════════════════════════════════════
|
|||
|
|
# 数据类（Python 对象映射）
|
|||
|
|
# ══════════════════════════════════════════════════════
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class Project:
|
|||
|
|
name: str
|
|||
|
|
output_dir: str
|
|||
|
|
language: str = "python"
|
|||
|
|
description: str = ""
|
|||
|
|
id: Optional[int] = None
|
|||
|
|
created_at: str = field(default_factory=lambda: datetime.now().isoformat())
|
|||
|
|
updated_at: str = field(default_factory=lambda: datetime.now().isoformat())
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class RawRequirement:
|
|||
|
|
project_id: int
|
|||
|
|
content: str
|
|||
|
|
source_type: str = "text" # text | file
|
|||
|
|
source_name: Optional[str] = None
|
|||
|
|
knowledge: Optional[str] = None
|
|||
|
|
id: Optional[int] = None
|
|||
|
|
created_at: str = field(default_factory=lambda: datetime.now().isoformat())
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class FunctionalRequirement:
|
|||
|
|
project_id: int
|
|||
|
|
raw_req_id: int
|
|||
|
|
index_no: int
|
|||
|
|
title: str
|
|||
|
|
description: str
|
|||
|
|
function_name: str
|
|||
|
|
priority: str = "medium"
|
|||
|
|
status: str = "pending"
|
|||
|
|
is_custom: bool = False
|
|||
|
|
id: Optional[int] = None
|
|||
|
|
created_at: str = field(default_factory=lambda: datetime.now().isoformat())
|
|||
|
|
updated_at: str = field(default_factory=lambda: datetime.now().isoformat())
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class CodeFile:
|
|||
|
|
project_id: int
|
|||
|
|
func_req_id: int
|
|||
|
|
file_name: str
|
|||
|
|
file_path: str
|
|||
|
|
language: str
|
|||
|
|
content: str
|
|||
|
|
id: Optional[int] = None
|
|||
|
|
created_at: str = field(default_factory=lambda: datetime.now().isoformat())
|
|||
|
|
updated_at: str = field(default_factory=lambda: datetime.now().isoformat())
|