Ejemplo n.º 1
0
def register_hub(request):
    """Hub creation page

    Returns the hub registration page

    Args:
        request: The HTTP request

    Returns: The hub registration page

    """
    error = None
    if request.method == 'POST':
        airport_id = request.POST['airport']
        airline = request.user.airline.first()
        airport = Airport.objects.filter(pk=airport_id)
        if airport.exists():
            airport = airport.first()
            if not (Hub.objects.filter(owner=airline, airport=airport).exists()):
                hub = Hub(owner=airline, airport=airport)
                hub.save()
                return redirect('home')
            else:
                error = "Own this hub"

    airports = Airport.objects.all()

    return render(request, 'registration/registration-hub.html', {'airports': airports, 'error': error})
Ejemplo n.º 2
0
def buy_hub_save(request):
    """Buy a hub

    Buy a given hub if possible

    Args:
        request: the HTTP request

    Returns:

    """
    if request.method == 'POST':
        airport_id = request.POST['airport']
        airline = request.user.airline.first()
        airport = Airport.objects.filter(pk=airport_id)
        error = None
        if airport.exists():
            airport = airport.first()
            if not (Hub.objects.filter(owner=airline, airport_id=airport_id).exists()):
                if not airline.hubs.exists():
                    hub = Hub()
                    hub.owner_id = airline.pk
                    hub.airport_id = airport_id
                    hub.save()
                    return redirect('home')
                elif airline.money > airport.price:
                    hub = Hub()
                    hub.owner_id = airline.pk
                    hub.airport_id = airport_id
                    airline.credit(airport.price)
                    hub.save()
                    return redirect('home')
                else:
                    error = _("You do not have the necessary funds")
            else:
                error = _("You already own this hub")
            hubs = airline.hubs.all().values_list('airport_id', flat=True)
            airports = Airport.objects.all().exclude(id__in=hubs)
            return render(request, 'buy-hub.html', {'airports': airports, 'error': error})
    else:
        return redirect('buy-hub')