97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
from datetime import datetime
|
|
from io import BytesIO
|
|
from itertools import groupby
|
|
from operator import attrgetter
|
|
|
|
import djclick as click
|
|
import structlog
|
|
from openpyxl import Workbook
|
|
|
|
from vbv_lernwelt.feedback.models import FeedbackResponse
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
VV_FEEDBACK_QUESTIONS = [
|
|
("satisfaction", "Zufriedenheit insgesamt"),
|
|
("goal_attainment", "Zielerreichung insgesamt"),
|
|
(
|
|
"proficiency",
|
|
"Wie beurteilst du deine Sicherheit bezüglichen den Themen nach dem Circle?",
|
|
),
|
|
("preparation_task_clarity", "Waren die Praxisaufträge klar und verständlich?"),
|
|
("would_recommend", "Würdest du den Circle weiterempfehlen?"),
|
|
("course_positive_feedback", "Was hat dir besonders gut gefallen?"),
|
|
("course_negative_feedback", "Wo siehst du Verbesserungspotential?"),
|
|
]
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
|
|
@click.command()
|
|
@click.argument("course_session_id")
|
|
@click.option(
|
|
"--save-as-file/--no-save-as-file",
|
|
default=True,
|
|
help="`save-as-file` to save the file, `no-save-as-file` returns bytes. Default is `save-as-file`.",
|
|
)
|
|
def command(course_session_id, save_as_file):
|
|
# using the output from call_command was a bit cumbersome, so this is just a wrapper for the actual function
|
|
export_feedback(course_session_id, save_as_file)
|
|
|
|
|
|
def export_feedback(course_session_id: str, save_as_file: bool):
|
|
wb = Workbook()
|
|
|
|
# remove the first sheet is just easier than keeping track of the active sheet
|
|
wb.remove_sheet(wb.active)
|
|
|
|
feedbacks = FeedbackResponse.objects.filter(
|
|
course_session_id=course_session_id
|
|
).order_by("circle")
|
|
grouped_feedbacks = groupby(feedbacks, key=attrgetter("circle"))
|
|
|
|
for circle, group_feedbacks in grouped_feedbacks:
|
|
group_feedbacks = list(group_feedbacks)
|
|
logger.debug(
|
|
"export_feedback_for_circle",
|
|
data={
|
|
"circle": circle.title,
|
|
"course_session_id": course_session_id,
|
|
"count": len(group_feedbacks),
|
|
},
|
|
label="feedback_export",
|
|
)
|
|
create_sheet(wb, circle.title, group_feedbacks)
|
|
|
|
if save_as_file:
|
|
wb.save(make_export_filename())
|
|
else:
|
|
output = BytesIO()
|
|
wb.save(output)
|
|
|
|
output.seek(0)
|
|
return output.getvalue()
|
|
|
|
|
|
def create_sheet(wb: Workbook, title: str, data: list[FeedbackResponse]):
|
|
sheet = wb.create_sheet(title=title)
|
|
# add header
|
|
questions = [q[1] for q in VV_FEEDBACK_QUESTIONS]
|
|
for col_idx, title in enumerate(questions, start=2):
|
|
sheet.cell(row=1, column=col_idx, value=title)
|
|
|
|
add_rows(sheet, data)
|
|
return sheet
|
|
|
|
|
|
def add_rows(sheet, data):
|
|
for row_idx, feedback in enumerate(data, start=2):
|
|
for col_idx, question in enumerate(VV_FEEDBACK_QUESTIONS, start=2):
|
|
response = feedback.data.get(question[0], "")
|
|
sheet.cell(row=row_idx, column=col_idx, value=response)
|
|
|
|
|
|
def make_export_filename():
|
|
today_date = datetime.today().strftime("%Y-%m-%d")
|
|
return f"feedback_export_{today_date}.xlsx"
|