uw-it-aca / pathways / 17604285327

from django.core.management.base import BaseCommand

from pathways.models.major import Major, SimilarMajor

from pathways.models.course import Course

from multiprocessing import Pool, cpu_count

class Command(BaseCommand):

    def handle(self, *args, **options):

        self.validate_majors()

        self.validate_courses()

    def validate_majors(self)   :

        count_majors = Major.objects.count()

        count_gpa_2yr = (Major.objects.filter(gpa_2yr__isnull=False)

        count_gpa_5yr = (Major.objects.filter(gpa_5yr__isnull=False)

        count_common_courses = 0

        for major in Major.objects.all():

            cc = major.get_common_with_coi_and_flags()

            if cc and len(cc) > 0:

                count_common_courses += 1

        count_similar_majors = SimilarMajor.objects.values('source_major')\

        print(f"Total Majors: {count_majors}")

        print(f"Majors with GPA Distro (2, 5): {count_gpa_2yr} "

        print(f"Majors with Common Courses: {count_common_courses}"

        print(f"Majors with Similar Majors: {count_similar_majors} "

    def validate_courses(self):

        count_courses = Course.objects.count()

        print(count_courses)

        count_with_grade_distro = \

        count_with_prereq = (Course.objects.filter(prereq_graph__isnull=False))

        pool = Pool(processes=cpu_count()-1)

        results = pool.map(self._process_course, Course.objects.all())

        pool.close()

        pool.join()

        count_with_coi = len([r for r in results if r[1]])

        count_with_concurrent = len([r for r in results if r[2]])

        print(f"Total Courses: {count_courses}")

        print(f"Courses with Grade Distro: {count_with_grade_distro}"

        print(f"Courses with Prereq Map: {count_with_prereq.count()}"

        print(f"Courses with COI: {count_with_coi}"

        print(f"Courses with Concurrent: {count_with_concurrent}"

    @staticmethod

    def _process_course(course):

        coi = course.get_coi_data()

        concurrent = course.get_concurrent_with_coi_and_flags()

        return (course.course_id, bool(coi), bool(concurrent))