98 lines
4.0 KiB
Python
98 lines
4.0 KiB
Python
# web/models.py
|
|
from django.db import models
|
|
from django.contrib.auth.models import AbstractUser
|
|
|
|
|
|
class Category(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
description = models.TextField()
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Service(models.Model):
|
|
name = models.CharField(max_length=200)
|
|
description = models.TextField()
|
|
price = models.DecimalField(max_digits=10, decimal_places=2)
|
|
category = models.ForeignKey(Category, on_delete=models.SET_NULL, null=True, related_name='services')
|
|
image = models.ImageField(upload_to='static/img/services/', blank=True, null=True)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def average_rating(self):
|
|
reviews = self.reviews.all()
|
|
if reviews:
|
|
return sum(review.rating for review in reviews) / reviews.count()
|
|
return 0
|
|
|
|
def review_count(self):
|
|
return self.reviews.count()
|
|
|
|
class Client(models.Model):
|
|
first_name = models.CharField(max_length=100)
|
|
last_name = models.CharField(max_length=100)
|
|
email = models.EmailField()
|
|
phone_number = models.CharField(max_length=15)
|
|
image = models.ImageField(upload_to='static/img/customer/', blank=True, null=True)
|
|
|
|
def __str__(self):
|
|
return f"{self.first_name} {self.last_name}"
|
|
|
|
class BlogPost(models.Model):
|
|
title = models.CharField(max_length=200)
|
|
content = models.TextField()
|
|
published_date = models.DateTimeField(auto_now_add=True)
|
|
image = models.ImageField(upload_to='static/img/blog/', blank=True, null=True)
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
class Order(models.Model):
|
|
service = models.ForeignKey(Service, on_delete=models.CASCADE, related_name='orders')
|
|
client = models.ForeignKey(Client, on_delete=models.CASCADE, related_name='orders')
|
|
message = models.TextField(blank=True, null=True)
|
|
order_date = models.DateTimeField(auto_now_add=True)
|
|
status = models.CharField(max_length=50, choices=[('pending', 'Pending'), ('in_progress', 'In Progress'), ('completed', 'Completed'), ('cancelled', 'Cancelled')], default='pending')
|
|
|
|
def __str__(self):
|
|
return f"Order #{self.id} by {self.client.first_name}"
|
|
|
|
class Project(models.Model):
|
|
name = models.CharField(max_length=200)
|
|
description = models.TextField()
|
|
completion_date = models.DateField(blank=True, null=True)
|
|
client = models.ForeignKey(Client, on_delete=models.CASCADE, related_name='projects')
|
|
service = models.ForeignKey(Service, on_delete=models.CASCADE, related_name='projects')
|
|
order = models.OneToOneField(Order, on_delete=models.CASCADE, related_name='project', null=True, blank=True)
|
|
category = models.ForeignKey('Category', on_delete=models.SET_NULL, null=True, blank=True)
|
|
image = models.ImageField(upload_to='static/img/project/', blank=True, null=True)
|
|
status = models.CharField(max_length=50, choices=[('in_progress', 'In Progress'), ('completed', 'Completed')], default='in_progress')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Review(models.Model):
|
|
client = models.ForeignKey(Client, on_delete=models.CASCADE, related_name='reviews')
|
|
service = models.ForeignKey(Service, on_delete=models.CASCADE, related_name='reviews')
|
|
project = models.ForeignKey(Project, on_delete=models.CASCADE, related_name='reviews', blank=True, null=True)
|
|
rating = models.IntegerField()
|
|
comment = models.TextField()
|
|
review_date = models.DateTimeField(auto_now_add=True)
|
|
image = models.ImageField(upload_to='static/img/review/', blank=True, null=True)
|
|
|
|
def __str__(self):
|
|
return f"Review by {self.client.first_name} {self.client.last_name} for {self.service.name}"
|
|
|
|
COMMUNICATION_METHODS = [
|
|
('email', 'Email'),
|
|
('telegram', 'Telegram'),
|
|
('sms', 'SMS'),
|
|
]
|
|
|
|
class User(AbstractUser):
|
|
telegram_id = models.CharField(max_length=50, blank=True, null=True, unique=True)
|
|
preferred_communication = models.CharField(
|
|
max_length=20,
|
|
choices=COMMUNICATION_METHODS,
|
|
default='email',
|
|
) |