import frappe
from frappe.utils import getdate
from lms.lms.utils import has_course_moderator_role, has_course_evaluator_role


def get_context(context):
	context.no_cache = 1
	context.is_moderator = has_course_moderator_role()
	context.is_evaluator = has_course_evaluator_role()
	batches = frappe.get_all(
		"LMS Batch",
		fields=[
			"name",
			"title",
			"description",
			"start_date",
			"end_date",
			"paid_batch",
			"amount",
			"currency",
			"seat_count",
			"published",
		],
		order_by="start_date",
	)

	past_batches, upcoming_batches, private_batches = [], [], []
	for batch in batches:
		batch.student_count = frappe.db.count("Batch Student", {"parent": batch.name})
		batch.course_count = frappe.db.count("Batch Course", {"parent": batch.name})
		batch.seats_left = (
			batch.seat_count - batch.student_count if batch.seat_count else None
		)
		if not batch.published:
			private_batches.append(batch)
		elif getdate(batch.start_date) < getdate():
			past_batches.append(batch)
		else:
			upcoming_batches.append(batch)

	context.past_batches = sorted(past_batches, key=lambda d: d.start_date)
	context.upcoming_batches = sorted(upcoming_batches, key=lambda d: d.start_date)
	context.private_batches = sorted(private_batches, key=lambda d: d.start_date)

	if frappe.session.user != "Guest":
		my_batches_info = []
		my_batches = frappe.get_all(
			"Batch Student", {"student": frappe.session.user}, pluck="parent"
		)

		for batch in my_batches:
			batchinfo = frappe.db.get_value(
				"LMS Batch",
				batch,
				[
					"name",
					"title",
					"description",
					"start_date",
					"end_date",
					"paid_batch",
					"amount",
					"currency",
					"seat_count",
				],
				as_dict=True,
			)

			batchinfo.student_count = frappe.db.count(
				"Batch Student", {"parent": batchinfo.name}
			)
			batchinfo.course_count = frappe.db.count("Batch Course", {"parent": batchinfo.name})
			batchinfo.seats_left = batchinfo.seat_count - batchinfo.student_count

			my_batches_info.append(batchinfo)

		context.my_batches = my_batches_info