84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import csv
|
|
from datetime import date
|
|
from typing import List
|
|
|
|
from django.contrib.admin.views.decorators import staff_member_required
|
|
from django.http import HttpResponse
|
|
|
|
from vbv_lernwelt.core.models import User
|
|
from vbv_lernwelt.course.consts import COURSE_UK, COURSE_UK_FR, COURSE_UK_IT
|
|
from vbv_lernwelt.course.models import CourseSessionUser
|
|
|
|
UK_COURSE_IDS = [COURSE_UK, COURSE_UK_FR, COURSE_UK_IT]
|
|
|
|
|
|
@staff_member_required
|
|
def export_students(request):
|
|
course_session_users = fetch_course_session_users(UK_COURSE_IDS)
|
|
return generate_export_response(course_session_users)
|
|
|
|
|
|
def export_trainers(request):
|
|
course_session_users = fetch_course_session_users(
|
|
UK_COURSE_IDS, role=CourseSessionUser.Role.EXPERT
|
|
)
|
|
return generate_export_response(course_session_users, role="Trainer")
|
|
|
|
|
|
def fetch_course_session_users(courses: List[int], role=CourseSessionUser.Role.MEMBER):
|
|
# if a user is in multiple courses, he should be exported multiple times
|
|
# todo: check if this is the case otherwise use .distinct("user")
|
|
return (
|
|
User.objects.filter(
|
|
coursesessionuser__course_session__course__id__in=courses,
|
|
coursesessionuser__role=role,
|
|
)
|
|
.order_by("email")
|
|
.distinct()
|
|
)
|
|
|
|
|
|
def generate_export_response(cs_users: List[User], role="Lernende") -> HttpResponse:
|
|
response = HttpResponse(content_type="text/csv")
|
|
response[
|
|
"Content-Disposition"
|
|
] = f"attachment; filename=edoniq_user_export_{date.today().strftime('%Y%m%d')}.csv"
|
|
|
|
writer = csv.writer(response)
|
|
writer.writerow(
|
|
[
|
|
"login",
|
|
"firstname",
|
|
"lastname",
|
|
"gender",
|
|
"preferred_language",
|
|
"email",
|
|
"birthday",
|
|
"group",
|
|
"region",
|
|
"division",
|
|
"oe",
|
|
"superior_login",
|
|
]
|
|
)
|
|
|
|
for cs_user in cs_users:
|
|
writer.writerow(
|
|
[
|
|
cs_user.id,
|
|
cs_user.first_name,
|
|
cs_user.last_name,
|
|
"m",
|
|
cs_user.language,
|
|
cs_user.email,
|
|
cs_user.additional_json_data.get("Geburtsdatum", ""),
|
|
role,
|
|
cs_user.additional_json_data.get("Firmenname", ""),
|
|
cs_user.additional_json_data.get("Lehrvertragsnummer", ""),
|
|
cs_user.coursesessionuser_set.first().course_session.import_id,
|
|
"",
|
|
]
|
|
)
|
|
|
|
return response
|