Minor stuff

This commit is contained in:
lvrossem
2023-04-01 10:03:18 -06:00
parent 65d1a2a6e4
commit d2933a95ba
9 changed files with 91 additions and 54 deletions

View File

@@ -2,7 +2,8 @@ from fastapi import HTTPException
from sqlalchemy.orm import Session
from enums import CourseEnum, course_enum_list
from models import User, CourseProgress
from models import CourseProgress, User
from schemas.courseprogress import CourseProgressBase
def get_course_progress(db: Session, user: User, course: CourseEnum):
@@ -15,21 +16,25 @@ def get_course_progress(db: Session, user: User, course: CourseEnum):
)
.first()
)
if course_progress:
return [
CourseProgressBase(
progress_value = course_progress.progress_value,
course = course_progress.course,
progress_value=course_progress.progress_value,
course=course_progress.course,
)
]
else:
return [CourseProgressBase(progress_value = 0, course=course)]
db.add(
CourseProgress(progress_value=0.0, course=course, owner_id=user.user_id)
)
db.commit()
return [CourseProgressBase(progress_value=0.0, course=course)]
return []
def initialize_user(db: Session, user: User):
"""Create CourseProgress records with a value of 0 for a new user"""
for course in course_enum_list:
db.add(CourseProgress(progress_value = 0.0, course = course, owner_id = user.user_id))
db.add(CourseProgress(progress_value=0.0, course=course, owner_id=user.user_id))
db.commit()