Add more typing

This commit is contained in:
Daniel Egger 2023-06-26 14:44:46 +02:00
parent 46b14bd4e7
commit b9c9b009ff
4 changed files with 52 additions and 10 deletions

View File

@ -4,14 +4,20 @@ from rest_framework.exceptions import PermissionDenied
from vbv_lernwelt.core.models import User
from vbv_lernwelt.course.permissions import has_course_access
from vbv_lernwelt.course_session.graphql.types import CourseSessionAttendanceCourseType
from vbv_lernwelt.course_session.graphql.types import (
CourseSessionAttendanceCourseType,
)
from vbv_lernwelt.course_session.models import CourseSessionAttendanceCourse
from vbv_lernwelt.course_session.services.attendance import AttendanceUserStatus
logger = structlog.get_logger(__name__)
class AttendanceUserInputType(graphene.InputObjectType):
user_id = graphene.ID(required=True)
status = graphene.Field(
graphene.Enum.from_enum(AttendanceUserStatus), required=True
)
class AttendanceCourseUserMutation(graphene.Mutation):

View File

@ -2,10 +2,14 @@ import graphene
from graphene_django import DjangoObjectType
from vbv_lernwelt.course_session.models import CourseSessionAttendanceCourse
from vbv_lernwelt.course_session.services.attendance import AttendanceUserStatus
class AttendanceUserType(graphene.ObjectType):
user_id = graphene.ID()
user_id = graphene.ID(required=True)
status = graphene.Field(
graphene.Enum.from_enum(AttendanceUserStatus), required=True
)
first_name = graphene.String()
last_name = graphene.String()
email = graphene.String()

View File

@ -1,12 +1,26 @@
import enum
from typing import List, TypedDict
from vbv_lernwelt.core.models import User
from vbv_lernwelt.course.services import mark_course_completion
from vbv_lernwelt.course_session.models import CourseSessionAttendanceCourse
class AttendanceUserStatus(enum.Enum):
PRESENT = "PRESENT"
ABSENT = "ABSENT"
class AttendanceUser(TypedDict):
user_id: str
status: AttendanceUserStatus
def update_attendance_list(
attendance_course: CourseSessionAttendanceCourse, attendance_user_list: list
attendance_course: CourseSessionAttendanceCourse,
attendance_user_list: List[AttendanceUser],
):
user_id_list_before = attendance_course.attendance_user_list or []
user_id_list_before = [u["user_id"] for u in attendance_course.attendance_user_list]
result_user_list = []
for attendance_user in attendance_user_list:
@ -15,22 +29,26 @@ def update_attendance_list(
result_user_list.append(
{
"user_id": u.id,
"status": attendance_user.get("status"),
"email": u.email,
"first_name": u.first_name,
"last_name": u.last_name,
}
)
completion_status = (
"success" if attendance_user.get("status") == "PRESENT" else "fail"
)
mark_course_completion(
page_key=attendance_course.learning_content.translation_key,
user=u,
course_session=attendance_course.course_session,
completion_status="success",
completion_status=completion_status,
)
attendance_course.attendance_user_list = result_user_list
attendance_course.save()
user_id_list_after = attendance_course.attendance_user_list or []
user_id_list_after = [u["user_id"] for u in attendance_course.attendance_user_list]
user_id_list_removed = [
user_id for user_id in user_id_list_before if user_id not in user_id_list_after

View File

@ -23,7 +23,9 @@ class AttendanceServicesTestCase(TestCase):
def test_updateAttendanceList_withSingleUserId_findsDetailsAndStoresResult(self):
student = User.objects.get(username="test-student1@example.com")
update_attendance_list(self.attendance_course, [{"user_id": student.id}])
update_attendance_list(
self.attendance_course, [{"user_id": student.id, "status": "PRESENT"}]
)
attendance_course = CourseSessionAttendanceCourse.objects.get(
id=self.attendance_course.id
@ -36,6 +38,7 @@ class AttendanceServicesTestCase(TestCase):
[
{
"email": "test-student1@example.com",
"status": "PRESENT",
"user_id": -21,
"last_name": "Student1",
"first_name": "Test",
@ -45,8 +48,9 @@ class AttendanceServicesTestCase(TestCase):
def test_updateAttendanceList_willUpdateUserCourseCompletion(self):
student = User.objects.get(username="test-student1@example.com")
update_attendance_list(self.attendance_course, [{"user_id": student.id}])
update_attendance_list(
self.attendance_course, [{"user_id": student.id, "status": "PRESENT"}]
)
self.assertEqual(CourseCompletion.objects.count(), 1)
cc = CourseCompletion.objects.first()
self.assertEqual(cc.user, student)
@ -60,6 +64,16 @@ class AttendanceServicesTestCase(TestCase):
def test_updateAttendanceList_withRemovedUser_willUpdateUserCourseCompletion(self):
student = User.objects.get(username="test-student1@example.com")
self.attendance_course.attendance_user_list = [
{
"email": "test-student1@example.com",
"status": "PRESENT",
"user_id": "-21",
"last_name": "Student1",
"first_name": "Test",
}
]
self.attendance_course.save()
mark_course_completion(
page_key=self.attendance_course.learning_content.translation_key,
user=student,
@ -71,7 +85,7 @@ class AttendanceServicesTestCase(TestCase):
self.assertEqual(CourseCompletion.objects.count(), 1)
cc = CourseCompletion.objects.first()
self.assertEqual(cc.user, student)
self.assertEqual(cc.completion_status, "success")
self.assertEqual(cc.completion_status, "fail")
self.assertEqual(
cc.page_key, self.attendance_course.learning_content.translation_key
)