Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 17 additions & 0 deletions account/migrations/0002_alter_user_options.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# Generated by Django 3.2.5 on 2021-11-04 08:07

from django.db import migrations


class Migration(migrations.Migration):

dependencies = [
('account', '0001_initial'),
]

operations = [
migrations.AlterModelOptions(
name='user',
options={'verbose_name': 'user', 'verbose_name_plural': 'users'},
),
]
255 changes: 145 additions & 110 deletions commerce/controllers.py
Original file line number Diff line number Diff line change
@@ -1,30 +1,35 @@
import random
import string
from typing import List

from django.contrib.auth import get_user_model
import string
import random
from django.contrib.auth.models import User
from django.db.models import Q
from django.shortcuts import get_object_or_404
from ninja import Router
from pydantic import UUID4

from account.authorization import GlobalAuth
from commerce.models import Product, Category, City, Vendor, Item, Order, OrderStatus
from commerce.schemas import ProductOut, CitiesOut, CitySchema, VendorOut, ItemOut, ItemSchema, ItemCreate
from config.utils.schemas import MessageOut
from account.authorization import GlobalAuth, get_tokens_for_user
from django.contrib.auth import get_user_model, authenticate


from commerce.models import Product, Category, City, Vendor, Item, Address, Order, OrderStatus
from commerce.schemas import MessageOut, ProductOut, CitiesOut, CitySchema, VendorOut, ItemOut, ItemSchema, ItemCreate, AddressOut, AddressCreate, OrderOut, orderstatus, checkout

User = get_user_model()


products_controller = Router(tags=['products'])
address_controller = Router(tags=['addresses'])
city_controller = Router(tags=['cities'])
vendor_controller = Router(tags=['vendors'])
order_controller = Router(tags=['orders'])
checkout_controller = Router(tags=['checkout'])

User = get_user_model()

@vendor_controller.get('', response=List[VendorOut])
def list_vendors(request):
return Vendor.objects.all()


# ===================products=============================
@products_controller.get('', response={
200: List[ProductOut],
404: MessageOut
Expand Down Expand Up @@ -58,73 +63,50 @@ def list_products(
return products_qs


"""
# product = Product.objects.all().select_related('merchant', 'category', 'vendor', 'label')
# print(product)
#
# order = Product.objects.all().select_related('address', 'user').prefetch_related('items')

# try:
# one_product = Product.objects.get(id='8d3dd0f1-2910-457c-89e3-1b0ed6aa720a')
# except Product.DoesNotExist:
# return {"detail": "Not found"}
# print(one_product)
#
# shortcut_function = get_object_or_404(Product, id='8d3dd0f1-2910-457c-89e3-1b0ed6aa720a')
# print(shortcut_function)

# print(type(product))
# print(product.merchant.name)
# print(type(product.merchant))
# print(type(product.category))


Product <- Merchant, Label, Category, Vendor

Retrieve 1000 Products form DB

products = Product.objects.all()[:1000] (select * from product limit 1000)

for p in products:
print(p)

for every product, we retrieve (Merchant, Label, Category, Vendor) records

Merchant.objects.get(id=p.merchant_id) (select * from merchant where id = 'p.merchant_id')
Label.objects.get(id=p.label_id) (select * from merchant where id = 'p.label_id')
Category.objects.get(id=p.category_id) (select * from merchant where id = 'p.category_id')
Vendor.objects.get(id=p.vendor_id) (select * from merchant where id = 'p.vendor_id')

4*1000+1

Solution: Eager loading

products = (select * from product limit 1000)
# ===================address=========================
@address_controller.get('', response=List[AddressOut])
def list_addresses(request):
return Address.objects.all()

mids = [p1.merchant_id, p2.merchant_id, ...]
[p1.label_id, p2.label_id, ...]
.
.
.

select * from merchant where id in (mids) * 4 for (label, category and vendor)
@address_controller.post('add-addresses', auth=GlobalAuth(), response={
200: AddressOut
})
def create_address(request, address_in: AddressCreate):
current_user = get_object_or_404(User, id=request.auth['pk'])
address = Address(**address_in.dict(), user= current_user)
address.save()
return 200, address

4+1

"""
@address_controller.put('update-addresses/{id}', response={
200: AddressOut
})
def update_address(request,id: UUID4, address_in: AddressCreate):
address = get_object_or_404(Address, id=id)
address.work_address = address_in.work_address
address.address1 = address_in.address1
address.address2 = address_in.address2
address.city_id = address_in.city_id
address.phone = address_in.phone
address.save()
return 200, address


@address_controller.get('')
def list_addresses(request):
pass

@address_controller.delete('delete-addresses/{id}', auth=GlobalAuth(), response={
204: MessageOut
})
def delete_address(request, id: UUID4):
current_user = get_object_or_404(User, id=request.auth['pk'])
address = get_object_or_404(Address, id=id, user= current_user)
address.delete()

# @products_controller.get('categories', response=List[CategoryOut])
# def list_categories(request):
# return Category.objects.all()
return 204, {'detail': 'Address deleted!'}


@address_controller.get('cities', response={
# =============city===================
@city_controller.get('cities', response={
200: List[CitiesOut],
404: MessageOut
})
Expand All @@ -137,15 +119,15 @@ def list_cities(request):
return 404, {'detail': 'No cities found'}


@address_controller.get('cities/{id}', response={
@city_controller.get('cities/{id}', response={
200: CitiesOut,
404: MessageOut
})
def retrieve_city(request, id: UUID4):
return get_object_or_404(City, id=id)


@address_controller.post('cities', response={
@city_controller.post('cities', response={
201: CitiesOut,
400: MessageOut
})
Expand All @@ -155,7 +137,7 @@ def create_city(request, city_in: CitySchema):
return 201, city


@address_controller.put('cities/{id}', response={
@city_controller.put('cities/{id}', response={
200: CitiesOut,
400: MessageOut
})
Expand All @@ -166,48 +148,117 @@ def update_city(request, id: UUID4, city_in: CitySchema):
return 200, city


@address_controller.delete('cities/{id}', response={
@city_controller.delete('cities/{id}', response={
204: MessageOut
})
def delete_city(request, id: UUID4):
city = get_object_or_404(City, id=id)
city.delete()
return 204, {'detail': ''}

# ==================order==============================
@order_controller.post('create-order', auth=GlobalAuth(), response=MessageOut)
def create_order(request):
current_user = get_object_or_404(User, id=request.auth['pk'])
user_items=Item.objects.filter(user=current_user,ordered=False)
if user_items:
order = Order.objects.create(
user=User.objects.first(),
status=OrderStatus.objects.get(is_default=True),
ref_code = ''.join(random.sample(string.ascii_letters+string.digits,6)),
ordered=False)
user_items.update(ordered=True)
order.items.add(*user_items)
order.total =order.order_total
order.save()
return {'detail': 'order created'}
else:
return {'detail': 'cart is empty'}




@order_controller.get('Orders', response={
200: List[OrderOut],
404: MessageOut
})
def orderout(request):
orderout = Order.objects.all()

if orderout:
return orderout

return 404, {'detail': 'Not found'}


"""
[
{
"NEW_id": "dcb384f7-98de-4b35-9451-22f8975218df"
},
{
"REFUNDED_id": "5357bec5-4ce1-45b6-8700-2c9986fa7f2b"
},
{
"COMPLETED_id": "5c8f472e-79a1-46fa-900f-3c975dd71d6c"
},
{
"SHIPPED_id": "87d1f525-bf34-4469-8bd0-7437ea3915a0"
},
{
"PROCESSING_id": "a1207dc4-1052-4ee9-bff7-da67397aeee4"
}
]
"""
@checkout_controller.put('checkout', response={
200: OrderOut
})
def checkout(request,id: UUID4, checkout_in: checkout):
checkout = get_object_or_404(Order, id=id)
checkout.address_id = checkout_in.address_id
checkout.note = checkout_in.note
checkout.ordered = True
checkout.save()
return 200, checkout



@order_controller.get('cart', response={
@order_controller.get('cart', auth=GlobalAuth(), response={
200: List[ItemOut],
404: MessageOut
})
def view_cart(request):
cart_items = Item.objects.filter(user=User.objects.first(), ordered=False)
current_user = get_object_or_404(User, id=request.auth['pk'])
cart_items = Item.objects.filter(user=current_user, ordered=False)

if cart_items:
return cart_items

return 404, {'detail': 'Your cart is empty, go shop like crazy!'}


@order_controller.post('add-to-cart', response={
@order_controller.post('add-to-cart', auth=GlobalAuth(), response={
200: MessageOut,
# 400: MessageOut
})
def add_update_cart(request, item_in: ItemCreate):
current_user = get_object_or_404(User, id=request.auth['pk'])
try:
item = Item.objects.get(product_id=item_in.product_id, user=User.objects.first())
item = Item.objects.get(product_id=item_in.product_id, user=current_user)
item.item_qty += 1
item.save()
except Item.DoesNotExist:
Item.objects.create(**item_in.dict(), user=User.objects.first())
Item.objects.create(**item_in.dict(), user=current_user)

return 200, {'detail': 'Added to cart successfully'}


@order_controller.post('item/{id}/reduce-quantity', response={
@order_controller.post('item/{id}/reduce-quantity', auth=GlobalAuth(), response={
200: MessageOut,
})
def reduce_item_quantity(request, id: UUID4):
item = get_object_or_404(Item, id=id, user=User.objects.first())
current_user = get_object_or_404(User, id=request.auth['pk'])
item = get_object_or_404(Item, id=id, user=current_user)
if item.item_qty <= 1:
item.delete()
return 200, {'detail': 'Item deleted!'}
Expand All @@ -217,41 +268,25 @@ def reduce_item_quantity(request, id: UUID4):
return 200, {'detail': 'Item quantity reduced successfully!'}


@order_controller.delete('item/{id}', response={

@order_controller.post('item/{id}/increase-quantity', auth=GlobalAuth(), response={
200: MessageOut,
})
def increase_item_quantity(request, id: UUID4):
current_user = get_object_or_404(User, id=request.auth['pk'])
item = get_object_or_404(Item, id=id, user=current_user)
item.item_qty += 1
item.save()

return 200, {'detail': 'Item quantity increased successfully!'}


@order_controller.delete('item/{id}', auth=GlobalAuth(), response={
204: MessageOut
})
def delete_item(request, id: UUID4):
item = get_object_or_404(Item, id=id, user=User.objects.first())
current_user = get_object_or_404(User, id=request.auth['pk'])
item = get_object_or_404(Item, id=id, user=current_user)
item.delete()

return 204, {'detail': 'Item deleted!'}


def generate_ref_code():
return ''.join(random.sample(string.ascii_letters + string.digits, 6))


@order_controller.post('create-order', auth=GlobalAuth(), response=MessageOut)
def create_order(request):
'''
* add items and mark (ordered) field as True
* add ref_number
* add NEW status
* calculate the total
'''

order_qs = Order.objects.create(
user=User.objects.first(),
status=OrderStatus.objects.get(is_default=True),
ref_code=generate_ref_code(),
ordered=False,
)

user_items = Item.objects.filter(user=User.objects.first()).filter(ordered=False)

order_qs.items.add(*user_items)
order_qs.total = order_qs.order_total
user_items.update(ordered=True)
order_qs.save()

return {'detail': 'order created successfully'}
Loading