wagtail-longclaw/longclaw/longclawshipping/api.py

from rest_framework.decorators import api_view, permission_classes
from rest_framework import permissions, status
from rest_framework.response import Response
from django_countries import countries
from longclaw.longclawshipping import serializers, models
from longclaw.longclawsettings.models import LongclawSettings

class InvalidShippingRate(Exception):
    pass


class InvalidShippingCountry(Exception):
    pass


def get_shipping_cost(country_code, option, settings):
    try:
        qrs = models.ShippingRate.objects.filter(countries__contains=country_code)
        try:
            if qrs.count() > 1:
                shipping_rate = qrs.filter(name=option)[0]
            else:
                shipping_rate = qrs[0]
            return {
                "rate": shipping_rate.rate,
                "description": shipping_rate.description,
                "carrier": shipping_rate.carrier
            }
        except models.ShippingRate.DoesNotExist:
            raise InvalidShippingRate

    except models.ShippingRate.DoesNotExist:
        if settings.default_shipping_enabled:
            return {"rate": settings.default_shipping_rate,
                    "description": "Standard shipping to rest of world",
                    "carrier": settings.default_shipping_rate}
        else:
            raise InvalidShippingCountry


@api_view(['GET'])
@permission_classes({permissions.AllowAny})
def shipping_cost(request):
    ''' Returns the shipping cost for a given country
    If the shipping cost for the given country has not been set, it will
    fallback to the default shipping cost if it has been enabled in the app
    settings
    '''
    try:
        code = request.query_params.get('country_code')
    except AttributeError:
        return Response(data={"message": "No country code supplied"},
                        status=status.HTTP_400_BAD_REQUEST)

    option = request.query_params.get('shipping_rate_name', 'standard')
    try:
        settings = LongclawSettings.for_site(request.site)
        data = get_shipping_cost(code, option, settings)
        response = Response(data=data, status=status.HTTP_200_OK)
    except InvalidShippingRate:
        response = Response(data={"message": "Shipping option {} is invalid".format(option)},
                            status=status.HTTP_400_BAD_REQUEST)
    except InvalidShippingCountry:
        response = Response(data={"message": "Shipping to {} is not available".format(code)},
                            status=status.HTTP_400_BAD_REQUEST)

    return response


@api_view(["GET"])
@permission_classes([permissions.AllowAny])
def shipping_countries(request):
    ''' Get all shipping countries
    '''
    queryset = models.ShippingRate.objects.all()
    country_data = [(c.name, c.code) for obj in queryset for c in obj.countries]
    return Response(data=country_data, status=status.HTTP_200_OK)