from django.db import models
from subjects.models import Subject
from django.contrib.gis.db import models as gis_models
from django.contrib.gis.geos import Point
# Create your models here.


class StudyGroup(models.Model):
    name = models.CharField(max_length=48)
    users = models.ManyToManyField(
        'student_status.StudentStatus', through='StudyGroupMembership')
    location = gis_models.PointField(blank=True, null=True, srid=4326)
    subject = models.ForeignKey(Subject, on_delete=models.CASCADE)
    active = models.BooleanField(default=False)
    timestamp = models.DateField(auto_now_add=True)

    def __str__(self):
        return self.name


class StudyGroupMembership(models.Model):
    user = models.ForeignKey(
        'student_status.StudentStatus', on_delete=models.CASCADE)
    study_group = models.ForeignKey(
        'study_groups.StudyGroup', on_delete=models.CASCADE)

    def __str__(self):
        return f'StudyGroupMembership: User={self.user}, StudyGroup={self.study_group.name}'