kopia lustrzana https://github.com/longclawshop/longclaw
106 wiersze
3.4 KiB
Python
106 wiersze
3.4 KiB
Python
'''
|
|
Shipping logic and payment capture API
|
|
'''
|
|
from django.utils import timezone
|
|
from django.db import transaction
|
|
from rest_framework.decorators import api_view, permission_classes
|
|
from rest_framework import permissions, status
|
|
from rest_framework.response import Response
|
|
from longclaw.longclawbasket.utils import destroy_basket
|
|
from longclaw.longclawcheckout.utils import create_order, GATEWAY
|
|
from longclaw.longclawcheckout.errors import PaymentError
|
|
|
|
@api_view(['GET'])
|
|
@permission_classes([permissions.AllowAny])
|
|
def create_token(request):
|
|
''' Generic function for creating a payment token from the
|
|
payment backend. Some payment backends (e.g. braintree) support creating a payment
|
|
token, which should be imported from the backend as 'get_token'
|
|
'''
|
|
token = GATEWAY.get_token(request)
|
|
return Response({'token': token}, status=status.HTTP_200_OK)
|
|
|
|
@transaction.atomic
|
|
@api_view(['POST'])
|
|
@permission_classes([permissions.AllowAny])
|
|
def create_order_with_token(request):
|
|
'''
|
|
Create an order using an existing transaction ID.
|
|
This is useful for capturing the payment outside of
|
|
longclaw - e.g. using paypals' express checkout or
|
|
similar
|
|
'''
|
|
# Get the request data
|
|
try:
|
|
address = request.data['address']
|
|
shipping_option = request.data.get('shipping_option', None)
|
|
email = request.data['email']
|
|
transaction_id = request.data['transaction_id']
|
|
except KeyError:
|
|
return Response(data={"message": "Missing parameters from request data"},
|
|
status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
# Create the order
|
|
order = create_order(
|
|
email,
|
|
request,
|
|
addresses=address,
|
|
shipping_option=shipping_option
|
|
)
|
|
|
|
order.payment_date = timezone.now()
|
|
order.transaction_id = transaction_id
|
|
order.save()
|
|
# Once the order has been successfully taken, we can empty the basket
|
|
destroy_basket(request)
|
|
|
|
return Response(data={"order_id": order.id}, status=status.HTTP_201_CREATED)
|
|
|
|
@transaction.atomic
|
|
@api_view(['POST'])
|
|
@permission_classes([permissions.AllowAny])
|
|
def capture_payment(request):
|
|
'''
|
|
Capture the payment for a basket and create an order
|
|
|
|
request.data should contain:
|
|
|
|
'address': Dict with the following fields:
|
|
shipping_name
|
|
shipping_address_line1
|
|
shipping_address_city
|
|
shipping_address_zip
|
|
shipping_address_country
|
|
billing_name
|
|
billing_address_line1
|
|
billing_address_city
|
|
billing_address_zip
|
|
billing_address_country
|
|
|
|
'email': Email address of the customer
|
|
'shipping': The shipping rate (in the sites' currency)
|
|
'''
|
|
# get request data
|
|
address = request.data['address']
|
|
email = request.data.get('email', None)
|
|
shipping_option = request.data.get('shipping_option', None)
|
|
|
|
# Capture the payment
|
|
try:
|
|
order = create_order(
|
|
email,
|
|
request,
|
|
addresses=address,
|
|
shipping_option=shipping_option,
|
|
capture_payment=True
|
|
)
|
|
response = Response(data={"order_id": order.id},
|
|
status=status.HTTP_201_CREATED)
|
|
except PaymentError as err:
|
|
order.status = Order.CANCELLED
|
|
order.note = "Payment failed"
|
|
response = Response(data={"message": err.message, "order_id": order.id},
|
|
status=status.HTTP_400_BAD_REQUEST)
|
|
order.save()
|
|
return response
|