|
| 1 | +from django.db import models |
| 2 | +from django.contrib.auth import get_user_model |
| 3 | +from django.utils.html import format_html |
| 4 | +from utils.images import upload_image_to |
| 5 | +from utils.slugs import generate_unique_slug |
| 6 | + |
| 7 | +User = get_user_model() |
| 8 | + |
| 9 | + |
| 10 | +class Event(models.Model): |
| 11 | + name = models.CharField(max_length=256) |
| 12 | + slug = models.SlugField(unique=True, blank=True) |
| 13 | + description = models.TextField(max_length=4096) |
| 14 | + cover_image = models.ImageField(upload_to=upload_image_to, blank=True, null=True) |
| 15 | + venue = models.CharField(max_length=128) |
| 16 | + start_time = models.DateField() |
| 17 | + end_time = models.DateField(null=True, blank=True) |
| 18 | + accept_reg = models.BooleanField(verbose_name="Accepting registrations ?") |
| 19 | + show_on_home = models.BooleanField(verbose_name="Show on Home Page ?") |
| 20 | + |
| 21 | + def __str__(self): |
| 22 | + return self.name + str(self.start_time) |
| 23 | + |
| 24 | + def save(self, *args, **kwargs): |
| 25 | + if self.slug == "": |
| 26 | + self.slug = generate_unique_slug(self.name, Event) |
| 27 | + super().save(args, kwargs) |
| 28 | + |
| 29 | + def get_absolute_url(self): |
| 30 | + return f"/events/e/{self.slug}" |
| 31 | + |
| 32 | + def event_page(self): |
| 33 | + return format_html(f'<a href="{self.get_absolute_url()}" target="_blank">View Page</a>') |
| 34 | + |
| 35 | + |
| 36 | +class EventRegistration(models.Model): |
| 37 | + event = models.ForeignKey(Event, on_delete=models.CASCADE) |
| 38 | + user = models.ForeignKey(User, on_delete=models.CASCADE) |
| 39 | + |
| 40 | + def __str__(self) -> str: |
| 41 | + return self.user.__str__() + "%" + self.event.__str__() |
0 commit comments