from django.db.models.signals import post_save from django.conf import settings from django.db import models from django.db.models import Sum from django.shortcuts import reverse from django_countries.fields import CountryField import uuid from datetime import datetime

CATEGORY_CHOICES = ( ('M', 'Men'), ('L', 'Ladies'), ('C', 'Children') )

LABEL_CHOICES = ( ('P', 'primary'), ('S', 'secondary'), ('D', 'danger'), ('W', 'warning') )

ADDRESS_CHOICES = ( ('B', 'Billing'), ('S', 'Shipping'), )

ITEM_STATUS = ( ('a', 'Available'), ('n', 'Not available'), ('r', 'Reserved'), )

class UserProfile(models.Model): user = models.OneToOneField( settings.AUTH_USER_MODEL, on_delete=models.CASCADE) stripe_customer_id = models.CharField(max_length=50, blank=True, null=True) one_click_purchasing = models.BooleanField(default=False)

def __str__(self):
    return self.user.username

class Item(models.Model): ITEM_STATUS = ( ('Available', 'Available'), ('Not available', 'Not available'), ('Reserved', 'Reserved'), ) item_status = models.CharField(max_length=50, choices=ITEM_STATUS) price = models.FloatField() discount_price = models.FloatField(blank=True, null=True) category = models.CharField(choices=CATEGORY_CHOICES, max_length=2) label = models.CharField(choices=LABEL_CHOICES, max_length=1) slug = models.SlugField(unique=True) description = models.TextField() image = models.ImageField()

def get_absolute_url(self):
    return reverse("ecommerce:product", kwargs={
        'slug': self.slug
    })

def get_add_to_cart_url(self):
    return reverse("ecommerce:add-to-cart", kwargs={
        'slug': self.slug
    })

def get_remove_from_cart_url(self):
    return reverse("ecommerce:remove-from-cart", kwargs={
        'slug': self.slug
    })

class OrderItem(models.Model): user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE) ordered = models.BooleanField(default=False) item = models.ForeignKey(Item, on_delete=models.CASCADE) quantity = models.IntegerField(default=1)

def __str__(self):
    return f"{self.quantity} of {self.item.title}"

def get_total_item_price(self):
    return self.quantity * self.item.price

def get_total_discount_item_price(self):
    return self.quantity * self.item.discount_price

def get_amount_saved(self):
    return self.get_total_item_price() - self.get_total_discount_item_price()

def get_final_price(self):
    if self.item.discount_price:
        return self.get_total_discount_item_price()
    return self.get_total_item_price()

class Order(models.Model): user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE) ref_code = models.CharField(max_length=20, blank=True, null=True) items = models.ManyToManyField(OrderItem) start_date = models.DateTimeField(auto_now_add=True) ordered_date = models.DateTimeField() ordered = models.BooleanField(default=False) shipping_address = models.ForeignKey( 'Address', related_name='shipping_address', on_delete=models.SET_NULL, blank=True, null=True) billing_address = models.ForeignKey( 'Address', related_name='billing_address', on_delete=models.SET_NULL, blank=True, null=True) payment = models.ForeignKey( 'Payment', on_delete=models.SET_NULL, blank=True, null=True) coupon = models.ForeignKey( 'Coupon', on_delete=models.SET_NULL, blank=True, null=True) being_delivered = models.BooleanField(default=False) received = models.BooleanField(default=False) refund_requested = models.BooleanField(default=False) refund_granted = models.BooleanField(default=False)

'''
1. Item added to cart
2. Adding a billing address
(Failed checkout)
3. Payment
(Preprocessing, processing, packaging etc.)
4. Being delivered
5. Received
6. Refunds
'''

def __str__(self):
    return self.user.username

def get_total(self):
    total = 0
    for order_item in self.items.all():
        total += order_item.get_final_price()
    if self.coupon:
        total -= self.coupon.amount
    return total

class Address(models.Model): user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE) street_address = models.CharField(max_length=100) apartment_address = models.CharField(max_length=100,default='street1') country = CountryField(multiple=False, default='Kenya') zip = models.CharField(max_length=100) address_type = models.CharField(max_length=1, choices=ADDRESS_CHOICES) default = models.BooleanField(default=False)

def __str__(self):
    return self.user.username

class Meta:
    verbose_name_plural = 'Addresses'

class Payment(models.Model): stripe_charge_id = models.CharField(max_length=50) user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.SET_NULL, blank=True, null=True) amount = models.FloatField(default='100') timestamp = models.DateTimeField(auto_now_add=True)

def __str__(self):
    return self.user.username

class Coupon(models.Model): code = models.CharField(max_length=15) amount = models.FloatField()

def __str__(self):
    return self.code

class Refund(models.Model): order = models.ForeignKey(Order, on_delete=models.CASCADE) reason = models.TextField() accepted = models.BooleanField(default=False) email = models.EmailField(default='example@gmail.com')

def __str__(self):
    return f"{self.pk}"

def userprofile_receiver(sender, instance, created, *args, **kwargs): if created: userprofile = UserProfile.objects.create(user=instance)

post_save.connect(userprofile_receiver, sender=settings.AUTH_USER_MODEL)