115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
import structlog
|
|
|
|
from vbv_lernwelt.core.models import Organisation, User
|
|
from vbv_lernwelt.course.consts import (
|
|
COURSE_UK,
|
|
COURSE_UK_FR,
|
|
COURSE_UK_IT,
|
|
VV_COURSE_IDS,
|
|
)
|
|
from vbv_lernwelt.course.models import CourseSessionUser
|
|
from vbv_lernwelt.learning_mentor.models import (
|
|
AgentParticipantRelation,
|
|
AgentParticipantRoleType,
|
|
)
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
UK_COURSES = [COURSE_UK, COURSE_UK_FR, COURSE_UK_IT]
|
|
|
|
|
|
def uk_cs_users_by_org(org: Organisation) -> set[CourseSessionUser]:
|
|
return set(
|
|
CourseSessionUser.objects.filter(user__organisation=org)
|
|
.filter(course_session__course__configuration__is_uk=True)
|
|
.filter(role=CourseSessionUser.Role.MEMBER.value)
|
|
.filter(course_session__course_id__in=UK_COURSES)
|
|
.exclude(course_session_id__in=[4, 5, 6])
|
|
)
|
|
|
|
|
|
def vv_cs_users_by_org(org: Organisation) -> set[CourseSessionUser]:
|
|
return set(
|
|
CourseSessionUser.objects.filter(user__organisation=org)
|
|
.filter(course_session__course__configuration__is_uk=False)
|
|
.filter(role=CourseSessionUser.Role.MEMBER.value)
|
|
.filter(course_session__course_id__in=VV_COURSE_IDS)
|
|
)
|
|
|
|
|
|
def create_or_sync_berufsbildner(berufsbildner: User, organisation: Organisation) -> bool:
|
|
org_members = uk_cs_users_by_org(organisation)
|
|
return create_or_sync_learning_mentor(berufsbildner, org_members)
|
|
|
|
|
|
def create_or_sync_ausbildungsverantwortlicher(
|
|
ausbildungsverantwortlicher: User,
|
|
organisation: Organisation
|
|
) -> bool:
|
|
org_members = vv_cs_users_by_org(organisation)
|
|
return create_or_sync_learning_mentor(ausbildungsverantwortlicher, org_members)
|
|
|
|
|
|
def create_or_sync_learning_mentor(
|
|
agent: User, org_members: set[CourseSessionUser]
|
|
) -> bool:
|
|
logger.info(
|
|
"Creating or syncing berufsbildner/ausbildungsverantwortlicher",
|
|
berufsbildner=agent,
|
|
org=agent.organisation.name_de,
|
|
)
|
|
|
|
# Check if it is a valid organisation
|
|
if agent.organisation and agent.organisation.organisation_id < 4:
|
|
logger.error("Invalid organisation", org=agent.organisation)
|
|
return False
|
|
|
|
# Get existing connections (full relation objects)
|
|
existing_relations = set(agent.agentparticipantrelation_set.all())
|
|
|
|
# Add new relations that are not in existing relations
|
|
existing_members = {relation.participant for relation in existing_relations}
|
|
for csu in org_members:
|
|
if csu not in existing_members:
|
|
AgentParticipantRelation.objects.get_or_create(
|
|
agent=agent,
|
|
participant=csu,
|
|
role=AgentParticipantRoleType.BERUFSBILDNER.value,
|
|
)
|
|
|
|
# Remove old relations that are not in the new relations
|
|
for relation in existing_relations:
|
|
if relation.participant not in org_members:
|
|
relation.delete()
|
|
|
|
return True
|
|
|
|
|
|
def delete_berufsbildner_relation(berufsbildner: User, organisation: Organisation) -> bool:
|
|
org_members = uk_cs_users_by_org(organisation)
|
|
delete_org_supervisor_relation(berufsbildner, org_members)
|
|
|
|
|
|
def delete_ausbildungsverantwortlicher_relation(ausbildungsverantwortlicher: User, organisation: Organisation) -> bool:
|
|
org_members = vv_cs_users_by_org(organisation)
|
|
delete_org_supervisor_relation(ausbildungsverantwortlicher, org_members)
|
|
|
|
|
|
def delete_org_supervisor_relation(
|
|
agent: User,
|
|
org_members: set[CourseSessionUser],
|
|
):
|
|
# As the key berufsbildner is used in several courses, we use org_members to select the ones from the correct
|
|
# course sessions
|
|
relations_to_delete = agent.agentparticipantrelation_set.filter(participant__in=org_members)
|
|
|
|
# Bulk delete the identified relations
|
|
deleted_count, _ = relations_to_delete.delete()
|
|
|
|
# Log the result
|
|
logger.info(
|
|
"Deleted ausbildungsverantwortlicher relations",
|
|
agent=agent,
|
|
deleted_count=deleted_count,
|
|
)
|