150 lines
4.1 KiB
Python
150 lines
4.1 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", ItemListing.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 TownList(generic.ListView):
|
|
model = Town
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['current_page'] = "Towns"
|
|
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')
|
|
|
|
|
|
class TownInfo(View):
|
|
def get(self, request, id):
|
|
try:
|
|
town = Town.objects.get(pk=id)
|
|
|
|
residents = town.residents.all()
|
|
|
|
context = {
|
|
"loc": town,
|
|
"residents": residents
|
|
}
|
|
return render(request, 'GeoffreyApp/town.html', context=context)
|
|
except Player.DoesNotExist:
|
|
return render(request, 'GeoffreyApp/error.html')
|