47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
|
|
class UserProfile(models.Model):
|
|
user = models.OneToOneField(User, on_delete=models.CASCADE)
|
|
is_seller = models.BooleanField(default=False)
|
|
phone_number = models.CharField(max_length=20, blank=True)
|
|
address = models.TextField(blank=True)
|
|
|
|
def __str__(self):
|
|
return self.user.username
|
|
|
|
class Category(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
slug = models.SlugField(unique=True)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Product(models.Model):
|
|
seller = models.ForeignKey(User, on_delete=models.CASCADE, related_name='products')
|
|
category = models.ForeignKey(Category, on_delete=models.SET_NULL, null=True, related_name='products')
|
|
name = models.CharField(max_length=200)
|
|
description = models.TextField()
|
|
price = models.DecimalField(max_digits=10, decimal_places=2)
|
|
stock = models.IntegerField(default=0)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Order(models.Model):
|
|
buyer = models.ForeignKey(User, on_delete=models.CASCADE, related_name='orders')
|
|
status = models.CharField(max_length=20, choices=[('pending', 'Pending'), ('shipped', 'Shipped'), ('delivered', 'Delivered')], default='pending')
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"Order {self.id} by {self.buyer.username}"
|
|
|
|
class OrderItem(models.Model):
|
|
order = models.ForeignKey(Order, on_delete=models.CASCADE, related_name='items')
|
|
product = models.ForeignKey(Product, on_delete=models.CASCADE)
|
|
quantity = models.IntegerField()
|
|
price = models.DecimalField(max_digits=10, decimal_places=2)
|
|
|
|
def __str__(self):
|
|
return f"{self.quantity} x {self.product.name} in Order {self.order.id}" |