Fix type error
This commit is contained in:
parent
5560ec2ede
commit
9a060cb59f
|
|
@ -98,8 +98,17 @@ def create_course_session_dict(course_session_object, my_role, user_role):
|
|||
def create_person_list_with_roles(
|
||||
user, course_session_ids=None, include_private_data=False
|
||||
):
|
||||
def create_user_dict(csu: CourseSessionUser):
|
||||
user_object = csu.user
|
||||
def create_csu_dict(csu: CourseSessionUser):
|
||||
user_data = create_user_dict(csu.user)
|
||||
user_data["chosen_profile"] = (
|
||||
csu.chosen_profile.code if csu.chosen_profile else "all"
|
||||
)
|
||||
user_data["paid_datetime"] = (
|
||||
csu.paid_datetime.isoformat() if csu.paid_datetime else None
|
||||
)
|
||||
return user_data
|
||||
|
||||
def create_user_dict(user_object: User):
|
||||
user_data = {
|
||||
"user_id": user_object.id,
|
||||
"first_name": user_object.first_name,
|
||||
|
|
@ -108,10 +117,6 @@ def create_person_list_with_roles(
|
|||
"avatar_url_small": user_object.avatar_url_small,
|
||||
"avatar_url": user_object.avatar_url,
|
||||
"course_sessions": [],
|
||||
"chosen_profile": csu.chosen_profile.code if csu.chosen_profile else "all",
|
||||
"paid_datetime": csu.paid_datetime.isoformat()
|
||||
if csu.paid_datetime
|
||||
else None,
|
||||
}
|
||||
if include_private_data:
|
||||
user_data["phone_number"] = user_object.phone_number
|
||||
|
|
@ -135,7 +140,7 @@ def create_person_list_with_roles(
|
|||
).select_related("user")
|
||||
my_role = user_role(cs.roles)
|
||||
for csu in course_session_users:
|
||||
person_data = result_persons.get(csu.user.id, create_user_dict(csu))
|
||||
person_data = result_persons.get(csu.user.id, create_csu_dict(csu))
|
||||
person_data["course_sessions"].append(
|
||||
create_course_session_dict(cs, my_role, csu.role)
|
||||
)
|
||||
|
|
@ -149,7 +154,7 @@ def create_person_list_with_roles(
|
|||
participant_user = relation.participant.user
|
||||
|
||||
if participant_user.id not in result_persons:
|
||||
person_data = create_user_dict(relation.participant)
|
||||
person_data = create_csu_dict(relation.participant)
|
||||
person_data["course_sessions"] = [course_session_entry]
|
||||
result_persons[participant_user.id] = person_data
|
||||
else:
|
||||
|
|
|
|||
Loading…
Reference in New Issue