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(
|
def create_person_list_with_roles(
|
||||||
user, course_session_ids=None, include_private_data=False
|
user, course_session_ids=None, include_private_data=False
|
||||||
):
|
):
|
||||||
def create_user_dict(csu: CourseSessionUser):
|
def create_csu_dict(csu: CourseSessionUser):
|
||||||
user_object = csu.user
|
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_data = {
|
||||||
"user_id": user_object.id,
|
"user_id": user_object.id,
|
||||||
"first_name": user_object.first_name,
|
"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_small": user_object.avatar_url_small,
|
||||||
"avatar_url": user_object.avatar_url,
|
"avatar_url": user_object.avatar_url,
|
||||||
"course_sessions": [],
|
"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:
|
if include_private_data:
|
||||||
user_data["phone_number"] = user_object.phone_number
|
user_data["phone_number"] = user_object.phone_number
|
||||||
|
|
@ -135,7 +140,7 @@ def create_person_list_with_roles(
|
||||||
).select_related("user")
|
).select_related("user")
|
||||||
my_role = user_role(cs.roles)
|
my_role = user_role(cs.roles)
|
||||||
for csu in course_session_users:
|
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(
|
person_data["course_sessions"].append(
|
||||||
create_course_session_dict(cs, my_role, csu.role)
|
create_course_session_dict(cs, my_role, csu.role)
|
||||||
)
|
)
|
||||||
|
|
@ -149,7 +154,7 @@ def create_person_list_with_roles(
|
||||||
participant_user = relation.participant.user
|
participant_user = relation.participant.user
|
||||||
|
|
||||||
if participant_user.id not in result_persons:
|
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]
|
person_data["course_sessions"] = [course_session_entry]
|
||||||
result_persons[participant_user.id] = person_data
|
result_persons[participant_user.id] = person_data
|
||||||
else:
|
else:
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue