Files
proyecto-final/tienda/models.py
T
elordenador 162b63cae9 feat: Add user purchase and receipt management
- Implemented 'Mis Compras' and 'Mis Recibos' pages for users to view their orders and payment receipts.
- Enhanced address validation in 'editar_direccion.html' to ensure cities and postal codes belong to Almería.
- Added shipping address display in seller order details on 'pedidos_vendedor.html'.
- Updated user portal to include links to purchases and receipts.
- Introduced email verification functionality during user registration.
- Refactored email sending utility for better error handling and logging.
- Improved session management for checkout processes with selected shipping addresses.
2026-03-10 13:08:10 +01:00

212 lines
8.2 KiB
Python

from django.db import models
from django.contrib.auth.models import User, AbstractUser
from .vars import VAT_RATE
import random, string
class User(AbstractUser):
class RegisterStatus(models.TextChoices):
CONFIRMATION_REQUIRED = "CR", "Confirmation Required"
ACTIVE = "AC", "Active"
BANNED = "BN", "Banned"
registration_status = models.CharField(
max_length = 2,
choices = RegisterStatus.choices,
default = RegisterStatus.CONFIRMATION_REQUIRED
)
class VerificationCode(models.Model):
class VerificationModes(models.TextChoices):
VERIFY_ACCOUNT = "VA"
RESET_PASSWORD = "RP"
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="user_belongsto", null=False, blank=False)
code = models.TextField(default = "", unique=True)
code_mode = models.CharField(
max_length=2,
choices = VerificationModes.choices,
default = VerificationModes.VERIFY_ACCOUNT
)
# Create your models here.
class Category(models.Model):
name = models.CharField(max_length=200)
def __str__(self):
return self.name
class Image(models.Model):
name = models.CharField(max_length=200, default="")
image = models.ImageField(upload_to='images/')
alt = models.CharField(max_length=255, default="", blank=True, verbose_name="Texto alternativo")
def __str__(self):
return self.name
class Product(models.Model):
name = models.CharField(max_length=200, default="")
description = models.TextField(default = "")
briefdesc = models.TextField(default = "")
price = models.FloatField(default = 0)
category = models.ForeignKey(Category, on_delete=models.CASCADE)
primary_image = models.ForeignKey(Image, on_delete=models.SET_NULL, null=True)
secondary_images = models.ManyToManyField(Image, related_name='products_secondary', blank=True)
creator = models.ForeignKey(User, on_delete=models.CASCADE, related_name='created_products', null=True, blank=True)
def __str__(self):
return self.name + " " + str(self.price)
def get_price_with_vat(self):
"""Retorna el precio con IVA incluido"""
return round(self.price * (1 + VAT_RATE), 2)
def get_vat_amount(self):
"""Retorna la cantidad de IVA"""
return round(self.price * VAT_RATE, 2)
class Cart(models.Model):
user = models.ForeignKey(User, on_delete=models.CASCADE, null=True, blank=True)
session_key = models.CharField(max_length=40, null=True, blank=True)
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
def __str__(self):
return f"Cart {self.id} - {self.user or self.session_key}"
def get_total(self):
return sum(item.get_subtotal() for item in self.items.all())
def get_total_with_vat(self):
"""Retorna el total del carrito con IVA incluido"""
return round(self.get_total() * (1 + VAT_RATE), 2)
def get_vat_amount(self):
"""Retorna la cantidad total de IVA"""
return round(self.get_total() * VAT_RATE, 2)
def get_items_count(self):
return sum(item.quantity for item in self.items.all())
class CartItem(models.Model):
cart = models.ForeignKey(Cart, on_delete=models.CASCADE, related_name='items')
product = models.ForeignKey(Product, on_delete=models.CASCADE)
quantity = models.PositiveIntegerField(default=1)
added_at = models.DateTimeField(auto_now_add=True)
class Meta:
unique_together = ('cart', 'product')
def __str__(self):
return f"{self.quantity}x {self.product.name}"
def get_subtotal(self):
return self.product.price * self.quantity
def get_subtotal_with_vat(self):
"""Retorna el subtotal del item con IVA incluido"""
return round(self.get_subtotal() * (1 + VAT_RATE), 2)
def get_vat_amount(self):
"""Retorna la cantidad de IVA de este item"""
return round(self.get_subtotal() * VAT_RATE, 2)
class Order(models.Model):
STATUS_PAID = "paid"
STATUS_CANCELLED = "cancelled"
STATUS_CHOICES = [
(STATUS_PAID, "Pagado"),
(STATUS_CANCELLED, "Cancelado"),
]
PAYMENT_STRIPE = "stripe"
PAYMENT_PAYPAL = "paypal"
PAYMENT_MANUAL = "manual"
PAYMENT_CHOICES = [
(PAYMENT_STRIPE, "Stripe"),
(PAYMENT_PAYPAL, "PayPal"),
(PAYMENT_MANUAL, "Manual"),
]
buyer = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True, related_name='orders')
shipping_address = models.ForeignKey('ShippingAddress', on_delete=models.SET_NULL, null=True, blank=True, related_name='orders')
session_key = models.CharField(max_length=40, null=True, blank=True)
total = models.FloatField(default=0)
status = models.CharField(max_length=20, choices=STATUS_CHOICES, default=STATUS_PAID)
payment_method = models.CharField(max_length=20, choices=PAYMENT_CHOICES, default=PAYMENT_MANUAL)
payment_reference = models.CharField(max_length=200, blank=True, default="")
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
def __str__(self):
return f"Pedido {self.id} - {self.buyer or self.session_key}"
def get_items_count(self):
return sum(item.quantity for item in self.items.all())
class OrderItem(models.Model):
STATUS_PENDING = "pending"
STATUS_PROCESSING = "processing"
STATUS_SHIPPED = "shipped"
STATUS_CHOICES = [
(STATUS_PENDING, "Pendiente"),
(STATUS_PROCESSING, "En preparación"),
(STATUS_SHIPPED, "Enviado"),
]
order = models.ForeignKey(Order, on_delete=models.CASCADE, related_name='items')
product = models.ForeignKey(Product, on_delete=models.SET_NULL, null=True, blank=True)
product_name = models.CharField(max_length=200, default="")
seller = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True, related_name='order_items_to_fulfill')
quantity = models.PositiveIntegerField(default=1)
unit_price = models.FloatField(default=0)
total_price = models.FloatField(default=0)
status = models.CharField(max_length=20, choices=STATUS_CHOICES, default=STATUS_PENDING)
created_at = models.DateTimeField(auto_now_add=True)
def __str__(self):
return f"{self.quantity}x {self.product_name} (Pedido {self.order_id})"
class OrderMessage(models.Model):
order_item = models.ForeignKey(OrderItem, on_delete=models.CASCADE, related_name='messages')
sender = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True, related_name='sent_messages')
message = models.TextField()
created_at = models.DateTimeField(auto_now_add=True)
class Meta:
ordering = ['created_at']
def __str__(self):
return f"Mensaje de {self.sender} - {self.created_at}"
class ShippingAddress(models.Model):
"""Direcciones de entrega de los usuarios"""
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='shipping_addresses')
full_name = models.CharField(max_length=200, verbose_name="Nombre completo")
address_line_1 = models.CharField(max_length=250, verbose_name="Dirección")
address_line_2 = models.CharField(max_length=250, blank=True, verbose_name="Dirección (línea 2)")
city = models.CharField(max_length=100, verbose_name="Ciudad")
postal_code = models.CharField(max_length=20, verbose_name="Código postal")
country = models.CharField(max_length=100, default="España", verbose_name="País")
phone = models.CharField(max_length=20, verbose_name="Teléfono")
is_default = models.BooleanField(default=False, verbose_name="Dirección predeterminada")
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
class Meta:
verbose_name = "Dirección de envío"
verbose_name_plural = "Direcciones de envío"
ordering = ['-is_default', '-created_at']
def __str__(self):
return f"{self.full_name} - {self.city}"
def save(self, *args, **kwargs):
# Si se marca como predeterminada, desmarcar las demás del usuario
if self.is_default:
ShippingAddress.objects.filter(user=self.user, is_default=True).update(is_default=False)
super().save(*args, **kwargs)