feat: program restrictions

lms/lms/utils.py

@@ -1763,12 +1763,12 @@ def get_programs():
 		programs = frappe.get_all("LMS Program", fields=["name"])
 	else:
 		programs = frappe.get_all(
-			"LMS Program Member", {"member": frappe.session.user}, ["parent as name"]
+			"LMS Program Member", {"member": frappe.session.user}, ["parent as name", "progress"]
 		)
 
 	for program in programs:
 		program_courses = frappe.get_all(
-			"LMS Program Course", {"parent": program.name}, ["course"]
+			"LMS Program Course", {"parent": program.name}, ["course"], order_by="idx"
 		)
 		program.courses = []
 		for course in program_courses:
@@ -1777,3 +1777,55 @@ def get_programs():
 		program.members = frappe.db.count("LMS Program Member", {"parent": program.name})
 
 	return programs
+
+
+@frappe.whitelist()
+def enroll_in_program_course(program, course):
+	enrollment = frappe.db.exists(
+		"LMS Enrollment", {"member": frappe.session.user, "course": course}
+	)
+
+	if enrollment:
+		enrollment = frappe.db.get_value(
+			"LMS Enrollment", enrollment, ["name", "current_lesson"], as_dict=1
+		)
+		enrollment.current_lesson = get_lesson_index(enrollment.current_lesson)
+		return enrollment
+
+	program_courses = frappe.get_all(
+		"LMS Program Course", {"parent": program}, ["course", "idx"], order_by="idx"
+	)
+	current_course_idx = [
+		program_course.idx
+		for program_course in program_courses
+		if program_course.course == course
+	][0]
+
+	for program_course in program_courses:
+		if program_course.idx < current_course_idx:
+			enrollment = frappe.db.get_value(
+				"LMS Enrollment",
+				{"member": frappe.session.user, "course": program_course.course},
+				["name", "progress"],
+				as_dict=1,
+			)
+			if enrollment and enrollment.progress != 100:
+				frappe.throw(
+					_("Please complete the previous courses in the program to enroll in this course.")
+				)
+			elif not enrollment:
+				frappe.throw(
+					_("Please complete the previous courses in the program to enroll in this course.")
+				)
+			else:
+				continue
+
+	enrollment = frappe.new_doc("LMS Enrollment")
+	enrollment.update(
+		{
+			"member": frappe.session.user,
+			"course": course,
+		}
+	)
+	enrollment.save()
+	return enrollment