222 lines
8.4 KiB
Python
222 lines
8.4 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
|
|
from django.conf import settings
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
|
|
|
|
class Business(models.Model):
|
|
name = models.CharField(max_length=160)
|
|
slug = models.SlugField(unique=True)
|
|
industry = models.CharField(max_length=120, default='Home Services')
|
|
primary_city = models.CharField(max_length=120)
|
|
primary_state = models.CharField(max_length=2)
|
|
google_review_url = models.URLField(blank=True)
|
|
is_active = models.BooleanField(default=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
ordering = ['name']
|
|
verbose_name_plural = 'businesses'
|
|
|
|
def __str__(self) -> str:
|
|
return self.name
|
|
|
|
@property
|
|
def initials(self) -> str:
|
|
words = [chunk for chunk in self.name.split() if chunk]
|
|
if not words:
|
|
return 'TF'
|
|
return ''.join(word[0] for word in words[:2]).upper()
|
|
|
|
|
|
class BusinessMembership(models.Model):
|
|
class Role(models.TextChoices):
|
|
OWNER = 'owner', 'Owner'
|
|
ADMIN = 'admin', 'Admin'
|
|
MANAGER = 'manager', 'Manager'
|
|
TECHNICIAN = 'technician', 'Technician'
|
|
|
|
business = models.ForeignKey(Business, on_delete=models.CASCADE, related_name='memberships')
|
|
user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='business_memberships')
|
|
role = models.CharField(max_length=24, choices=Role.choices, default=Role.OWNER)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
ordering = ['business__name', 'user__email']
|
|
unique_together = ('business', 'user')
|
|
|
|
def __str__(self) -> str:
|
|
identity = self.user.get_full_name() or self.user.email or self.user.username
|
|
return f'{identity} · {self.business.name} ({self.get_role_display()})'
|
|
|
|
@property
|
|
def can_manage_workspace(self) -> bool:
|
|
return self.role in {self.Role.OWNER, self.Role.ADMIN}
|
|
|
|
@property
|
|
def can_manage_proof(self) -> bool:
|
|
return self.role in {self.Role.OWNER, self.Role.ADMIN, self.Role.MANAGER}
|
|
|
|
|
|
class Customer(models.Model):
|
|
business = models.ForeignKey(Business, on_delete=models.CASCADE, related_name='customers')
|
|
full_name = models.CharField(max_length=160)
|
|
email = models.EmailField(blank=True)
|
|
phone = models.CharField(max_length=40, blank=True)
|
|
city = models.CharField(max_length=120)
|
|
state = models.CharField(max_length=2)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
ordering = ['full_name']
|
|
|
|
def __str__(self) -> str:
|
|
return f'{self.full_name} · {self.city}, {self.state}'
|
|
|
|
|
|
class Job(models.Model):
|
|
class Status(models.TextChoices):
|
|
COMPLETED = 'completed', 'Completed'
|
|
REVIEW_REQUESTED = 'review_requested', 'Review requested'
|
|
PROOF_READY = 'proof_ready', 'Proof ready'
|
|
|
|
business = models.ForeignKey(Business, on_delete=models.CASCADE, related_name='jobs')
|
|
customer = models.ForeignKey(Customer, on_delete=models.CASCADE, related_name='jobs')
|
|
service_type = models.CharField(max_length=120)
|
|
description = models.TextField(blank=True)
|
|
technician_name = models.CharField(max_length=120, blank=True)
|
|
city = models.CharField(max_length=120)
|
|
state = models.CharField(max_length=2)
|
|
completed_at = models.DateField(default=timezone.localdate)
|
|
project_value = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True)
|
|
status = models.CharField(max_length=32, choices=Status.choices, default=Status.COMPLETED)
|
|
is_verified = models.BooleanField(default=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
ordering = ['-completed_at', '-created_at']
|
|
|
|
def __str__(self) -> str:
|
|
return f'{self.service_type} for {self.customer.full_name}'
|
|
|
|
def _get_media_by_type(self, media_type: str):
|
|
prefetched_media = getattr(self, '_prefetched_objects_cache', {}).get('media')
|
|
if prefetched_media is not None:
|
|
return next((media for media in prefetched_media if media.media_type == media_type), None)
|
|
return self.media.filter(media_type=media_type).first()
|
|
|
|
@property
|
|
def before_media(self):
|
|
return self._get_media_by_type(JobMedia.MediaType.BEFORE)
|
|
|
|
@property
|
|
def after_media(self):
|
|
return self._get_media_by_type(JobMedia.MediaType.AFTER)
|
|
|
|
|
|
def job_media_upload_path(instance: 'JobMedia', filename: str) -> str:
|
|
return f'jobs/job_{instance.job_id}/{instance.media_type}_{filename}'
|
|
|
|
|
|
class JobMedia(models.Model):
|
|
class MediaType(models.TextChoices):
|
|
BEFORE = 'before', 'Before'
|
|
AFTER = 'after', 'After'
|
|
|
|
job = models.ForeignKey(Job, on_delete=models.CASCADE, related_name='media')
|
|
media_type = models.CharField(max_length=16, choices=MediaType.choices)
|
|
file = models.FileField(upload_to=job_media_upload_path, blank=True)
|
|
caption = models.CharField(max_length=140, blank=True)
|
|
display_order = models.PositiveSmallIntegerField(default=0)
|
|
|
|
class Meta:
|
|
ordering = ['display_order', 'id']
|
|
verbose_name_plural = 'job media'
|
|
|
|
def __str__(self) -> str:
|
|
return f'{self.job} · {self.get_media_type_display()}'
|
|
|
|
|
|
class ReviewRequest(models.Model):
|
|
class Status(models.TextChoices):
|
|
SENT = 'sent', 'Sent'
|
|
VIEWED = 'viewed', 'Viewed'
|
|
RESPONDED = 'responded', 'Responded'
|
|
|
|
class Channel(models.TextChoices):
|
|
EMAIL = 'email', 'Email'
|
|
SMS = 'sms', 'SMS'
|
|
MANUAL = 'manual', 'Manual share'
|
|
|
|
job = models.OneToOneField(Job, on_delete=models.CASCADE, related_name='review_request')
|
|
token = models.UUIDField(default=uuid.uuid4, unique=True, editable=False)
|
|
status = models.CharField(max_length=20, choices=Status.choices, default=Status.SENT)
|
|
channel = models.CharField(max_length=16, choices=Channel.choices, default=Channel.EMAIL)
|
|
sent_at = models.DateTimeField(default=timezone.now)
|
|
last_opened_at = models.DateTimeField(null=True, blank=True)
|
|
reviewed_at = models.DateTimeField(null=True, blank=True)
|
|
delivery_note = models.CharField(max_length=200, blank=True)
|
|
|
|
class Meta:
|
|
ordering = ['-sent_at']
|
|
|
|
def __str__(self) -> str:
|
|
return f'Review request for {self.job}'
|
|
|
|
|
|
class Feedback(models.Model):
|
|
class Experience(models.TextChoices):
|
|
GREAT = 'great', 'Great'
|
|
GOOD = 'good', 'Good'
|
|
OKAY = 'okay', 'Okay'
|
|
BAD = 'bad', 'Bad'
|
|
|
|
review_request = models.OneToOneField(ReviewRequest, on_delete=models.CASCADE, related_name='feedback')
|
|
experience = models.CharField(max_length=12, choices=Experience.choices)
|
|
rating = models.PositiveSmallIntegerField(null=True, blank=True)
|
|
testimonial = models.TextField(blank=True)
|
|
follow_up_required = models.BooleanField(default=False)
|
|
is_public_approved = models.BooleanField(default=False)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
ordering = ['-created_at']
|
|
|
|
def __str__(self) -> str:
|
|
return f'{self.get_experience_display()} feedback for {self.review_request.job}'
|
|
|
|
|
|
class ProofCard(models.Model):
|
|
class Status(models.TextChoices):
|
|
DRAFT = 'draft', 'Draft'
|
|
PUBLISHED = 'published', 'Published'
|
|
HIDDEN = 'hidden', 'Hidden'
|
|
|
|
job = models.OneToOneField(Job, on_delete=models.CASCADE, related_name='proof_card')
|
|
customer_display_name = models.CharField(max_length=160)
|
|
is_anonymized = models.BooleanField(default=False)
|
|
testimonial_quote = models.TextField(blank=True)
|
|
rating = models.PositiveSmallIntegerField(null=True, blank=True)
|
|
status = models.CharField(max_length=16, choices=Status.choices, default=Status.DRAFT)
|
|
is_featured = models.BooleanField(default=False)
|
|
attached_widget_label = models.CharField(max_length=120, blank=True)
|
|
attached_pages = models.CharField(max_length=200, blank=True)
|
|
published_at = models.DateTimeField(null=True, blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
ordering = ['-is_featured', '-updated_at']
|
|
|
|
def __str__(self) -> str:
|
|
return f'Proof card · {self.job.service_type} · {self.customer_display_name}'
|
|
|
|
@property
|
|
def verified_label(self) -> str:
|
|
return 'Verified job' if self.job.is_verified else 'Pending verification'
|