Minor stuff
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user