from django.db import models
from classes.models import Classes


    
# Create your models here.
class University(models.Model):
    name = models.CharField(max_length=200)

    def __str__(self) -> str:
        return self.name

class Agreement(models.Model):
    major = models.CharField(max_length=200)
    university = models.ForeignKey(University, related_name="agreements", on_delete=models.CASCADE)

    def __str__(self) -> str:
        return f"{self.major} {self.university.name}"

class Section(models.Model):
    area_name = models.CharField(max_length=200)
    number_of_required_courses = models.IntegerField()
    classes = models.ManyToManyField(Classes, related_name="classes")
    agreement = models.ForeignKey(Agreement, related_name="sections", on_delete=models.CASCADE)

    def __str__(self) -> str:
        return f"{self.area_name} {self.agreement.major} {self.agreement.university}"