from django.contrib.auth.models import AbstractUser from django.core.validators import FileExtensionValidator from django.db import models class User(AbstractUser): is_citizen = models.BooleanField(default=False) is_moderator = models.BooleanField(default=False) is_resolver = models.BooleanField(default=False) phone = models.CharField(max_length=15, blank=True, null=True) groups = models.ManyToManyField( 'auth.Group', verbose_name='groups', blank=True, help_text='The groups this user belongs to.', related_name='core_user_groups', related_query_name='core_user', ) user_permissions = models.ManyToManyField( 'auth.Permission', verbose_name='user permissions', blank=True, help_text='Specific permissions for this user.', related_name='core_user_permissions', related_query_name='core_user', ) class IssueCategory(models.Model): name = models.CharField(max_length=100) description = models.TextField(blank=True) icon = models.CharField(max_length=50, default='fas fa-exclamation-circle') def __str__(self): return self.name class Issue(models.Model): STATUS_CHOICES = [ ('reported', 'Reported'), ('acknowledged', 'Acknowledged'), ('in_progress', 'In Progress'), ('resolved', 'Resolved'), ] title = models.CharField(max_length=200) description = models.TextField() category = models.ForeignKey(IssueCategory, on_delete=models.SET_NULL, null=True, blank=True) reporter = models.ForeignKey(User, on_delete=models.CASCADE, related_name='reported_issues') location = models.CharField(max_length=200) latitude = models.FloatField() longitude = models.FloatField() photo = models.ImageField(upload_to='issue_photos/', blank=True, null=True, validators=[FileExtensionValidator(['jpg', 'jpeg', 'png', 'gif'])]) status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='reported') created_at = models.DateTimeField(auto_now_add=True) updated_at = models.DateTimeField(auto_now=True) def __str__(self): return self.title def vote_count(self): return self.votes.count() def has_user_voted(self, user): if user.is_authenticated: return self.votes.filter(user=user).exists() return False class Vote(models.Model): user = models.ForeignKey(User, on_delete=models.CASCADE) issue = models.ForeignKey(Issue, on_delete=models.CASCADE, related_name='votes') created_at = models.DateTimeField(auto_now_add=True) class Meta: unique_together = ('user', 'issue') # Prevent duplicate votes per user def __str__(self): return f"{self.user.username} voted on {self.issue.title}" class Comment(models.Model): issue = models.ForeignKey(Issue, on_delete=models.CASCADE, related_name="comments") user = models.ForeignKey(User, on_delete=models.CASCADE) content = models.TextField() # <--- field name is "content", not "text" parent = models.ForeignKey("self", null=True, blank=True, related_name="replies", on_delete=models.CASCADE) created_at = models.DateTimeField(auto_now_add=True) class Meta: ordering = ["created_at"] def __str__(self): return f"Comment by {self.user} on {self.issue}" @property def is_reply(self): return self.parent is not None