# -*- coding: utf-8 -*-
from openai import OpenAI
import markdown2
import pdfkit
from django.conf import settings
from celery import shared_task
from django.utils import timezone
from django.core.exceptions import ObjectDoesNotExist
from .models import Papers
# ============== PYdoc ===================
# from docx import Document
# import pypandoc
# pypandoc.download_pandoc()
# ================ weasyprint ===========
# 库用不了
# from weasyprint import HTML
# HTML(string=html).write_pdf("output.pdf")
def generate_html_paper(md_result, file_path_name):
# 写到md中
md_file = open(f'{file_path_name}.md', 'w', encoding='utf-8')
md_file.write(md_result)
md_file.close()
# 从md中读出来再写到html中
md_read_file = open(f'{file_path_name}.md', 'r', encoding='utf-8')
md_text = md_read_file.read()
md_read_file.close()
html = markdown2.markdown(md_text)
html_write_file = open(f'{file_path_name}.html', 'w', encoding='utf-8')
html_write_file.write(
"""
"""+html)
html_write_file.close()
def convert_to_pdf(file_path_name):
# =============== pdfkit ================
# 从html中读出来再写到pdf中
f = open(f'{file_path_name}.html', 'r', encoding='utf-8')
content = f.read()
f.close()
config = pdfkit.configuration(wkhtmltopdf=settings.WKHTMLTOPDF_PATH)
pdfkit.from_string(content, f'{file_path_name}.pdf', configuration=config)
@shared_task
def paper_gen_async_wraper(file_path_name, uuid_str, subject, textbook_version, start_grade, \
start_semester, start_chapter, end_grade, end_semester, end_chapter, student):
# client = OpenAI(
# api_key = settings.OPAI_API_KEY,
# base_url = settings.OPAI_BASE_URL,
# )
# completion = client.chat.completions.create(
# model = "deepseek-r1-250120", # your model endpoint ID
# messages = [
# {"role": "system", "content": "你是教学助手"},
# # {"role": "user",
# # "content": f"请根据以下信息生成一份检测卷：学科为{subject}，教材版本是{textbook_version},\
# # 检测范围从{start_grade}年级{start_semester}学期{start_chapter}章\
# # 到{end_grade}年级{end_semester}学期{end_chapter}章，学生年级为{student.get('grade')},\
# # 年龄为{student.get('age')}，所在学校是{student.get('school')}。\
# # 根据答题所需时间一个半小时设置题量。题目有梯度。\
# # 试卷题目为{subject}{textbook_version}检测卷。只生成题目，不输出答案及说明等其他内容。"
# # },
# {"role": "user", "content": f"""请严格按照以下要求生成一份{subject}检测卷：
# 【试卷信息】
# 1. 标题：**{subject}{textbook_version}综合检测卷**
# 2. 检测范围：{start_grade}年级{start_semester}学期第{start_chapter}章 至
# {end_grade}年级{end_semester}学期第{end_chapter}章
# 3. 适用对象：{student.get('school')} {student.get('grade')}年级学生（年龄{student.get('age')}）
# 【命题要求】
# 1. 题型结构：
# - 按难度梯度分为基础题（60%）、中档题（30%）、提高题（10%）
# - 根据学科特点设计合理的题型（如数学可设置选择题/填空题/计算题/应用题，语文可设置阅读理解/文言文/写作等）
# - 同一知识点不重复考查
# 2. 题量控制：
# - 总题数控制在18-22题（以优秀学生完成时间约75分钟为标准）
# - 选择题不超过5题，填空题不超过6题
# - 需包含至少1道综合应用题（理科）或材料分析题（文科）
# 3. 内容要求：
# - 严格按照指定章节范围命题
# - 题目表述清晰无歧义，数字单位标注完整
# - 禁止出现需图片辅助的题目
# - 数学类题目需给出必要公式位置（如"（提示：可用公式____）"）
# 【输出格式】
# 仅输出以下内容：
# 1. 试卷标题（加粗居中）
# 2. 考试说明（含总分值100分、考试时长、姓名班级填写处）
# 3. 分模块的题目内容（标注题型、分值和题号）
# 4. "——以下为题目区域——" 分隔线
# 禁止包含：
# - 答案及解析
# - 评分标准
# - 额外说明文字
# - markdown格式及特殊符号"""}
# ],
# )
# generate_html_paper(completion.choices[0].message.content, file_path_name)
import time
time.sleep(30)
generate_html_paper(""" # Test md""", file_path_name)
try:
obj = Papers.objects.get(uuid=uuid_str) # 只获取一个对象
obj.status = 'D'
obj.updated_at = timezone.now()
obj.save()
except ObjectDoesNotExist:
print("warning: record not found")