diff --git a/school/lms/doctype/cohort/cohort.py b/school/lms/doctype/cohort/cohort.py index 91ba2004..1a7a3025 100644 --- a/school/lms/doctype/cohort/cohort.py +++ b/school/lms/doctype/cohort/cohort.py @@ -5,6 +5,9 @@ import frappe from frappe.model.document import Document class Cohort(Document): + def get_url(self): + return f"{frappe.utils.get_url()}/courses/{self.course}/cohorts/{self.slug}" + def get_subgroups(self, include_counts=False): names = frappe.get_all("Cohort Subgroup", filters={"cohort": self.name}, pluck="name") subgroups = [frappe.get_doc("Cohort Subgroup", name) for name in names] diff --git a/school/lms/doctype/cohort_mentor/cohort_mentor.py b/school/lms/doctype/cohort_mentor/cohort_mentor.py index a385819c..71bf9c4b 100644 --- a/school/lms/doctype/cohort_mentor/cohort_mentor.py +++ b/school/lms/doctype/cohort_mentor/cohort_mentor.py @@ -1,8 +1,12 @@ # Copyright (c) 2021, FOSS United and contributors # For license information, please see license.txt -# import frappe +import frappe from frappe.model.document import Document class CohortMentor(Document): - pass + def get_subgroup(self): + return frappe.get_doc("Cohort Subgroup", self.subgroup) + + def get_user(self): + return frappe.get_doc("User", self.email) diff --git a/school/lms/doctype/cohort_subgroup/cohort_subgroup.py b/school/lms/doctype/cohort_subgroup/cohort_subgroup.py index aa675206..68550bd0 100644 --- a/school/lms/doctype/cohort_subgroup/cohort_subgroup.py +++ b/school/lms/doctype/cohort_subgroup/cohort_subgroup.py @@ -10,6 +10,10 @@ class CohortSubgroup(Document): if not self.invite_code: self.invite_code = random_string(8) + def get_url(self): + cohort = frappe.get_doc("Cohort", self.cohort) + return f"{frappe.utils.get_url()}/courses/{self.course}/subgroups/{cohort.slug}/{self.slug}" + def get_invite_link(self): cohort = frappe.get_doc("Cohort", self.cohort) return f"{frappe.utils.get_url()}/courses/{self.course}/join/{cohort.slug}/{self.slug}/{self.invite_code}"