237 lines
8.4 KiB
Python
237 lines
8.4 KiB
Python
# myapp/services/course_service.py
|
||
import re
|
||
from typing import Dict, List
|
||
from flask import current_app
|
||
from datetime import datetime
|
||
from bson import ObjectId
|
||
|
||
from ..services.cos_service import upload_file, delete_file
|
||
|
||
from ..models.material import Material
|
||
|
||
# 假设你已有这两个 loader
|
||
from db.course import load_course_from_json
|
||
|
||
def load_course(course_id: str):
|
||
return load_course_from_json(
|
||
course_id, course_data_dir=current_app.config["COURSE_DATA_DIR"]
|
||
)
|
||
|
||
def user_selected_course_briefs(user_obj):
|
||
"""根据用户对象的选课列表，拼装课程简要信息数组"""
|
||
course_list = current_app.extensions["course_list"]
|
||
briefs = []
|
||
for cid in getattr(user_obj, "select_course", []):
|
||
course_obj = load_course(cid)
|
||
brief = course_list.get_course_brief_info(cid, course_obj)
|
||
briefs.append(brief)
|
||
return briefs
|
||
|
||
def create_material(teacher_id, material_name, description, chapters, image_url):
|
||
material = Material(
|
||
name=material_name,
|
||
description=description,
|
||
teacher_id=teacher_id,
|
||
created_at=datetime.now(),
|
||
updated_at=datetime.now(),
|
||
chapters=chapters,
|
||
image_url=image_url
|
||
)
|
||
mongo = current_app.extensions["mongo"]
|
||
material_id = mongo.db.materials.insert_one(material.model_dump()).inserted_id
|
||
return str(material_id)
|
||
|
||
def load_material(material_id):
|
||
mongo = current_app.extensions["mongo"]
|
||
material = mongo.db.materials.find_one({"_id": ObjectId(material_id)})
|
||
return Material(**material)
|
||
|
||
def update_material(material_id, chapters):
|
||
mongo = current_app.extensions["mongo"]
|
||
material = mongo.db.materials.find_one({'_id': ObjectId(material_id)})
|
||
if material:
|
||
mongo.db.materials.update_one(
|
||
{'_id': ObjectId(material_id)},
|
||
{'$set': {'chapters': chapters, 'updated_at': datetime.now()}}
|
||
)
|
||
return True
|
||
return False
|
||
|
||
def add_material_chapter(material_id, chapter_name, teacher_id):
|
||
mongo = current_app.extensions["mongo"]
|
||
material = mongo.db.materials.find_one({'_id': ObjectId(material_id)})
|
||
if material['teacher_id'] != teacher_id:
|
||
return False
|
||
if material:
|
||
mongo.db.materials.update_one(
|
||
{'_id': ObjectId(material_id)},
|
||
{'$push': {'chapters': {'chapter_name': chapter_name, 'lessons': []}}})
|
||
return True
|
||
|
||
def delete_material_chapter_lesson(material_id, chapter_name, lesson_name, teacher_id):
|
||
"""
|
||
删除指定课程的课时。
|
||
删除后，更新 MongoDB 中对应的章节中的 lesson 列表。
|
||
"""
|
||
mongo = current_app.extensions["mongo"]
|
||
|
||
# 查询 material 是否存在，且确保该 teacher_id 对应的 material
|
||
material = mongo.db.materials.find_one({'_id': ObjectId(material_id), 'teacher_id': teacher_id})
|
||
|
||
for chapter in material['chapters']:
|
||
if chapter['chapter_name'] == chapter_name:
|
||
for lesson in chapter['lessons']:
|
||
if lesson['lesson_name'] == lesson_name:
|
||
if lesson['markdown_lesson_file_link']!='':
|
||
delete_file(lesson['markdown_lesson_file_link'])
|
||
if lesson['markdown_prompt_file_name']!='':
|
||
delete_file(lesson['markdown_prompt_file_name'])
|
||
if lesson['markdown_score_prompt_file_link']!='':
|
||
delete_file(lesson['markdown_score_prompt_file_link'])
|
||
if not material:
|
||
return False # 找不到对应的 material 或 teacher_id 不匹配
|
||
|
||
# 删除 lesson 的条件
|
||
res = mongo.db.materials.update_one(
|
||
{
|
||
"_id": ObjectId(material_id),
|
||
"teacher_id": teacher_id # 确保是该教师的课程
|
||
},
|
||
{
|
||
"$pull": {
|
||
"chapters.$[c].lessons": {
|
||
"lesson_name": lesson_name # 根据 lesson_name 来删除
|
||
}
|
||
},
|
||
"$set": {"updated_at": datetime.now()} # 更新修改时间
|
||
},
|
||
array_filters=[{"c.chapter_name": chapter_name}] # 找到匹配的章节
|
||
)
|
||
|
||
# 如果删除操作成功，res.modified_count 应为 1
|
||
return res.modified_count == 1
|
||
|
||
|
||
def add_material_chapter_lesson(material_id, chapter_name, lesson_name, teacher_id):
|
||
"""
|
||
1) 生成占位 Markdown（lesson_prompt_score_prompt）
|
||
2) 上传到对象存储（返回 URL）
|
||
3) 将 lesson 文档（含三份 URL）push 到指定章节
|
||
"""
|
||
mongo = current_app.extensions["mongo"]
|
||
|
||
# 组织对象存储的文件路径
|
||
chapter_slug = _slugify(chapter_name or "")
|
||
lesson_slug = _slugify(lesson_name or "")
|
||
timestamp = datetime.now().strftime('%Y%m%dT%H%M%SZ')
|
||
base_prefix = f"materials_{material_id}_{chapter_slug}_{lesson_slug}_{timestamp}"
|
||
|
||
# 占位文本
|
||
lesson_md, prompt_md, score_prompt_md = _default_markdowns(lesson_name or "未命名课时")
|
||
|
||
# 生成文件名（带层级路径）
|
||
lesson_key = f"{base_prefix}_lesson.md"
|
||
prompt_key = f"{base_prefix}_prompt.md"
|
||
score_key = f"{base_prefix}_score_prompt.md"
|
||
|
||
# 上传（任意一个失败则终止，不写库）
|
||
try:
|
||
lesson_url = upload_file(lesson_md.encode("utf-8"), lesson_key)
|
||
prompt_url = upload_file(prompt_md.encode("utf-8"), prompt_key)
|
||
score_url = upload_file(score_prompt_md.encode("utf-8"), score_key)
|
||
except Exception as e:
|
||
current_app.logger.exception("Upload default markdown failed: %s", e)
|
||
return False
|
||
|
||
# 组装 lesson_doc
|
||
lesson_doc = {
|
||
"lesson_name": lesson_name or "",
|
||
"markdown_lesson_file_link": lesson_url, # lesson 主体
|
||
"markdown_prompt_file_name": prompt_url, # 给大模型的 prompt
|
||
"markdown_score_prompt_file_link": score_url, # 评分 prompt
|
||
}
|
||
|
||
# 将 lesson push 到指定章节
|
||
res = mongo.db.materials.update_one(
|
||
{
|
||
"_id": ObjectId(material_id),
|
||
"teacher_id": teacher_id, # 注意：若存的是 ObjectId，请确保传入的是同类型
|
||
},
|
||
{
|
||
"$push": {"chapters.$[c].lessons": lesson_doc},
|
||
"$set": {"updated_at": datetime.now()},
|
||
},
|
||
array_filters=[{"c.chapter_name": chapter_name}],
|
||
)
|
||
|
||
# 可选：如果 res.modified_count != 1，说明 push 失败（可能章节名不匹配）
|
||
# 这里可以考虑是否回滚 COS 上刚上传的 3 个文件（需实现删除逻辑）。
|
||
return res.modified_count == 1
|
||
|
||
|
||
def get_materials_by_teacher(teacher_id)->List[Material]:
|
||
mongo = current_app.extensions["mongo"]
|
||
materials = mongo.db.materials.find({'teacher_id': teacher_id})
|
||
return [Material(**material) for material in materials]
|
||
|
||
def get_materials_by_teacher_dict(teacher_id)->List[Dict]:
|
||
mongo = current_app.extensions["mongo"]
|
||
materials = mongo.db.materials.find({'teacher_id': teacher_id})
|
||
result = []
|
||
for material in materials:
|
||
material['_id'] = str(material['_id'])
|
||
result.append(material)
|
||
return result
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
def _slugify(name: str) -> str:
|
||
"""将章节/课时名转为 URL 友好的 slug。"""
|
||
s = name.strip().lower()
|
||
s = re.sub(r'\s+', '-', s) # 空白 -> -
|
||
s = re.sub(r'[^a-z0-9\-_]+', '', s) # 仅保留安全字符
|
||
return s or 'untitled'
|
||
|
||
def _default_markdowns(lesson_name: str):
|
||
"""三份占位 Markdown 文本。可按需调整模板内容。"""
|
||
ts = datetime.now().strftime('%Y-%m-%d %H:%M')
|
||
lesson_md = f"""# {lesson_name}
|
||
> Auto-generated at {ts}
|
||
|
||
## 阶段一：示例阶段
|
||
### 步骤A
|
||
在这里编写该步骤的教学内容（文字/代码/图片链接等）。
|
||
|
||
### 步骤B
|
||
继续补充该阶段的其它步骤内容。
|
||
"""
|
||
|
||
prompt_md = f"""# {lesson_name}
|
||
> Prompt for LLM (auto-generated at {ts})
|
||
|
||
## 阶段一：示例阶段
|
||
### 步骤A
|
||
请基于“步骤A”的教学目标与材料，生成指导性提示词（尽量结构化，给出评估标准和示例）。
|
||
|
||
### 步骤B
|
||
请基于“步骤B”的教学目标与材料，生成指导性提示词。
|
||
"""
|
||
|
||
score_prompt_md = f"""# {lesson_name}
|
||
> Scoring Prompt for LLM (auto-generated at {ts})
|
||
|
||
## 评分准则
|
||
- 正确性、完整性、可读性、效率（按需调整权重）
|
||
|
||
## 阶段一：示例阶段
|
||
### 步骤A
|
||
对学员在“步骤A”的输出进行评分与文字反馈，分数区间 0-100，并输出 JSON：```{{"score": <int>, "reasons": "<string>", "advices": "<string>"}}```
|
||
### 步骤B
|
||
同上，对“步骤B”输出进行评分与反馈。
|
||
"""
|
||
return lesson_md, prompt_md, score_prompt_md
|