import frappe
from community.lms.models import Course

def get_context(context):
    context.no_cache = 1
    course_name = frappe.form_dict["course"]
    batch_name = frappe.form_dict["batch"]

    course = Course.find(course_name)
    if not course:
        context.template = "www/404.html"
        return

    batch = course.get_batch(batch_name)
    if not batch:
        frappe.local.flags.redirect_location = "/courses/" + course_name
        raise frappe.Redirect

    context.course = course
    context.batch = batch