More refactors

This commit is contained in:
lvrossem
2023-04-17 07:51:53 -06:00
parent 0bf764a0f4
commit 38eb9027d6
14 changed files with 142 additions and 89 deletions

View File

@@ -26,14 +26,10 @@ def get_course_progress(db: Session, user: User, course: CourseEnum):
if course_progress:
result.append(
CourseProgressParent(
progress=course_progress.progress, course=course
)
CourseProgressParent(progress=course_progress.progress, course=course)
)
else:
db.add(
CourseProgress(progress=0.0, course=course, owner_id=user.user_id)
)
db.add(CourseProgress(progress=0.0, course=course, owner_id=user.user_id))
db.commit()
result.append(CourseProgressParent(progress=0.0, course=course))
@@ -44,9 +40,7 @@ def initialize_user(db: Session, user: User):
"""Create CourseProgress records with a value of 0 for a new user"""
for course in CourseEnum:
if course != CourseEnum.All:
db.add(
CourseProgress(progress=0.0, course=course, owner_id=user.user_id)
)
db.add(CourseProgress(progress=0.0, course=course, owner_id=user.user_id))
db.commit()