Geoffrey-Django/views.py

125 lines
3.4 KiB
Python

from django.shortcuts import render
from django.views.generic import View
from GeoffreyApp.models import *
from django.views import generic
from django.db.models import Q
# Create your views here.
class Home(View):
def get(self, request):
stats = []
stats.append(("Players", Player.objects.count()))
stats.append(("Bases", Base.objects.count()))
stats.append(("Shops", Shop.objects.count()))
stats.append(("Items for Sale", Player.objects.count()))
context = {
"stat_list": stats,
"current_page": "Home",
}
return render(request, 'GeoffreyApp/home.html', context=context)
class SearchList(View):
def get(self, request):
context = {}
query = request.GET.get('search')
context["search"] = query
context["player_list"] = Player.objects.filter(Q(name__icontains=query)).all()
context["loc_list"] = Location.objects.filter(Q(name__icontains=query) | Q(owner__name__icontains=query)).all()
return render(request, 'GeoffreyApp/search.html', context=context)
class PlayerList(generic.ListView):
model = Player
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context['current_page'] = "Players"
return context
class ShopList(generic.ListView):
model = Shop
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context['current_page'] = "Shops"
return context
def get_queryset(self):
qs = Shop.objects.all()
search = self.request.GET.get('q')
if search:
qs = qs.filter(Q(name__icontains=search) | Q(owner__name__icontains=search))
return qs
class BaseList(generic.ListView):
model = Base
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context['current_page'] = "Bases"
return context
class ItemListingList(generic.ListView):
model = ItemListing
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context['current_page'] = "Item Listings"
return context
class PlayerInfo(View):
def get(self, request, id):
try:
player = Player.objects.get(pk=id)
loc_list = Location.objects.filter(owner=player).all()
context = {
"player": player,
"loc_list": loc_list
}
return render(request, 'GeoffreyApp/player.html', context=context)
except Player.DoesNotExist:
return render(request, 'GeoffreyApp/error.html')
class ShopInfo(View):
def get(self, request, id):
try:
shop = Shop.objects.get(pk=id)
inventory = ItemListing.objects.filter(shop=shop).all()
context = {
"loc": shop,
"inventory": inventory
}
return render(request, 'GeoffreyApp/shop.html', context=context)
except Player.DoesNotExist:
return render(request, 'GeoffreyApp/error.html')
class BaseInfo(View):
def get(self, request, id):
try:
base = Base.objects.get(pk=id)
context = {
"loc": base,
}
return render(request, 'GeoffreyApp/location.html', context=context)
except Player.DoesNotExist:
return render(request, 'GeoffreyApp/error.html')