154 lines
4.6 KiB
Python
154 lines
4.6 KiB
Python
"""
|
||
新建项目对话框
|
||
"""
|
||
from PyQt5.QtWidgets import (
|
||
QDialog, QVBoxLayout, QHBoxLayout, QLabel,
|
||
QLineEdit, QTextEdit, QPushButton, QMessageBox
|
||
)
|
||
from PyQt5.QtCore import Qt
|
||
from gui_ai_developer.models import Project, Requirement, Module, Document
|
||
|
||
|
||
class NewProjectDialog(QDialog):
|
||
"""新建项目对话框"""
|
||
|
||
def __init__(self, parent=None):
|
||
"""
|
||
初始化对话框
|
||
|
||
Args:
|
||
parent: 父窗口
|
||
"""
|
||
super().__init__(parent)
|
||
self.project = None
|
||
self._setup_ui()
|
||
|
||
def _setup_ui(self):
|
||
"""设置UI"""
|
||
self.setWindowTitle("新建项目")
|
||
self.setModal(True)
|
||
self.resize(700, 500)
|
||
|
||
layout = QVBoxLayout(self)
|
||
layout.setSpacing(15)
|
||
|
||
# 标题
|
||
title_label = QLabel("<h2>📁 新建项目</h2>")
|
||
layout.addWidget(title_label)
|
||
|
||
# 项目名称
|
||
layout.addWidget(QLabel("项目名称 *"))
|
||
self.name_edit = QLineEdit()
|
||
self.name_edit.setPlaceholderText("请输入项目名称")
|
||
layout.addWidget(self.name_edit)
|
||
|
||
# 项目描述
|
||
layout.addWidget(QLabel("项目描述"))
|
||
self.desc_edit = QTextEdit()
|
||
self.desc_edit.setPlaceholderText("请输入项目描述（可选）")
|
||
self.desc_edit.setMaximumHeight(80)
|
||
layout.addWidget(self.desc_edit)
|
||
|
||
# 需求描述
|
||
layout.addWidget(QLabel("需求描述 *"))
|
||
self.req_edit = QTextEdit()
|
||
self.req_edit.setPlaceholderText(
|
||
"请输入项目需求...\n\n"
|
||
"建议格式：\n"
|
||
"1. 功能描述\n"
|
||
"2. 性能要求\n"
|
||
"3. 安全要求"
|
||
)
|
||
layout.addWidget(self.req_edit)
|
||
|
||
# 按钮区域
|
||
button_layout = QHBoxLayout()
|
||
button_layout.addStretch()
|
||
|
||
cancel_btn = QPushButton("取消")
|
||
cancel_btn.setStyleSheet("""
|
||
QPushButton {
|
||
background: rgba(51, 65, 85, 0.5);
|
||
border: 1px solid rgba(59, 130, 246, 0.2);
|
||
}
|
||
""")
|
||
cancel_btn.clicked.connect(self.reject)
|
||
button_layout.addWidget(cancel_btn)
|
||
|
||
create_btn = QPushButton("创建项目")
|
||
create_btn.clicked.connect(self._create_project)
|
||
button_layout.addWidget(create_btn)
|
||
|
||
layout.addLayout(button_layout)
|
||
|
||
def _create_project(self):
|
||
"""创建项目"""
|
||
# 验证输入
|
||
name = self.name_edit.text().strip()
|
||
if not name:
|
||
QMessageBox.warning(self, "提示", "请输入项目名称")
|
||
return
|
||
|
||
req_text = self.req_edit.toPlainText().strip()
|
||
if not req_text:
|
||
QMessageBox.warning(self, "提示", "请输入需求描述")
|
||
return
|
||
|
||
# 创建项目对象
|
||
self.project = Project(
|
||
name=name,
|
||
description=self.desc_edit.toPlainText().strip()
|
||
)
|
||
self.project.original_requirement = req_text
|
||
|
||
# 模拟生成需求
|
||
self._generate_requirements()
|
||
|
||
# 模拟生成模块
|
||
self._generate_modules()
|
||
|
||
# 模拟生成文档
|
||
self._generate_documents()
|
||
|
||
self.accept()
|
||
|
||
def _generate_requirements(self):
|
||
"""生成需求（模拟AI分析）"""
|
||
requirements = [
|
||
Requirement("REQ-001", "核心功能", "从需求文本提取的核心功能", 90, "高"),
|
||
Requirement("REQ-002", "数据管理", "数据的增删改查功能", 85, "中"),
|
||
Requirement("REQ-003", "用户管理", "用户认证和权限管理", 88, "高")
|
||
]
|
||
|
||
for req in requirements:
|
||
self.project.add_requirement(req)
|
||
|
||
def _generate_modules(self):
|
||
"""生成模块（模拟AI生成）"""
|
||
modules = [
|
||
Module("核心业务模块", "实现主要业务逻辑", ["REQ-001"]),
|
||
Module("数据管理模块", "数据存储和查询", ["REQ-002"]),
|
||
Module("用户认证模块", "用户登录和权限控制", ["REQ-003"])
|
||
]
|
||
|
||
for module in modules:
|
||
module.add_tech("Spring Boot")
|
||
module.add_tech("MySQL")
|
||
module.add_tech("Redis")
|
||
self.project.add_module(module)
|
||
|
||
def _generate_documents(self):
|
||
"""生成文档"""
|
||
documents = [
|
||
Document("需求规格说明书.docx", "SRS", "1.5 MB"),
|
||
Document("功能设计文档.docx", "功能设计", "1.2 MB"),
|
||
Document("接口文档.pdf", "API", "800 KB")
|
||
]
|
||
|
||
for doc in documents:
|
||
self.project.add_document(doc)
|
||
|
||
def get_project(self):
|
||
"""获取创建的项目"""
|
||
return self.project
|