minor update

civicfix/core/models.py

@@ -10,7 +10,7 @@ class User(AbstractUser):
     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',
@@ -27,25 +27,33 @@ class User(AbstractUser):
         related_name='core_user_permissions',
         related_query_name='core_user',
     )
+
+    # Ban-related fields
     is_banned = models.BooleanField(default=False)
     banned_until = models.DateTimeField(null=True, blank=True)
 
-    def ban(self, days=1):
-        """Ban user for given days (default 7 days)."""
+    def ban(self, days=7):
+        """Ban user for given number of days (default = 7)."""
         self.is_banned = True
         self.banned_until = timezone.now() + timedelta(days=days)
         self.save()
 
     def unban(self):
+        """Unban user immediately."""
         self.is_banned = False
         self.banned_until = None
         self.save()
 
     def is_currently_banned(self):
+        """Check if user is still banned (auto-unban if expired)."""
         if self.is_banned and self.banned_until:
-            return timezone.now() < self.banned_until
+            if timezone.now() >= self.banned_until:
+                # Auto unban if ban expired
+                self.unban()
+                return False
+            return True
         return False
-
+    
 class Department(models.Model):
     name = models.CharField(max_length=100, unique=True)
     description = models.TextField(blank=True, null=True)