import frappe
from school.lms.utils import slugify, get_membership, get_lessons, get_batch, get_lesson_url
def get_common_context(context):
    context.no_cache = 1

    try:
        batch_name = frappe.form_dict["batch"]
    except KeyError:
        batch_name = None

    course = frappe.db.get_value("LMS Course",
        frappe.form_dict["course"], ["name", "title", "video_link"], as_dict=True)
    if not course:
        context.template = "www/404.html"
        return
    context.course = course
    context.lessons = get_lessons(course.name)
    membership = get_membership(course.name, frappe.session.user, batch_name)
    context.membership = membership
    if membership:
        batch = get_batch(course.name, membership.batch)

        if batch:
            context.batch = batch

    context.course.query_parameter = "?batch=" + membership.batch if membership and membership.batch else ""
    context.livecode_url = get_livecode_url()

def get_livecode_url():
    return frappe.db.get_single_value("LMS Settings", "livecode_url")

def redirect_to_lesson(course, index_="1.1"):
    frappe.local.flags.redirect_location = get_lesson_url(course.name, index_) + course.query_parameter
    raise frappe.Redirect