139 lines
4.5 KiB
Python
139 lines
4.5 KiB
Python
from django.shortcuts import get_object_or_404, render
|
|
from django.urls import reverse
|
|
from django.http import HttpResponse
|
|
|
|
from .models import (
|
|
LegislativeText,
|
|
LegislationBook,
|
|
LegislationClassification,
|
|
School,
|
|
Country,
|
|
Sponsor,
|
|
models_in_index,
|
|
)
|
|
|
|
from random import sample
|
|
|
|
def index(request):
|
|
legislative_texts = list(LegislativeText.objects.all())
|
|
try:
|
|
legislative_texts = sample(legislative_texts, 5)
|
|
except ValueError:
|
|
# there's not enough texts, so just return nothing
|
|
legislative_texts = []
|
|
context = {
|
|
"legislative_texts": legislative_texts,
|
|
}
|
|
return render(request, "explorer/index.html", context)
|
|
|
|
def all(request):
|
|
legislative_texts = list(LegislativeText.objects.all())
|
|
context = {
|
|
"legislative_texts": legislative_texts,
|
|
}
|
|
return render(request, "explorer/all.html", context)
|
|
|
|
def view_legislation(request, legislation_id):
|
|
legislation = get_object_or_404(LegislativeText, pk=legislation_id)
|
|
context = {
|
|
"legislation": legislation,
|
|
"lines": legislation.get_lines()
|
|
}
|
|
return render(request, "explorer/legislation.html", context)
|
|
|
|
def view_conference(request, conference_id):
|
|
book = get_object_or_404(LegislationBook, pk=conference_id)
|
|
results = LegislativeText.objects.filter(from_book=book)
|
|
context = {
|
|
"book": book,
|
|
"legislation": results,
|
|
"sample": results[0]
|
|
}
|
|
return render(request, "explorer/conference.html", context)
|
|
|
|
def stats(request):
|
|
all_legislation = len(LegislativeText.objects.all())
|
|
context = {
|
|
"all": all_legislation,
|
|
"red_senate": len(LegislativeText.objects.filter(assembly="RSB")),
|
|
"blue_senate": len(LegislativeText.objects.filter(assembly="BSB")),
|
|
"white_senate": len(LegislativeText.objects.filter(assembly="WSB")),
|
|
"red_house": len(LegislativeText.objects.filter(assembly="RHB")),
|
|
"blue_house": len(LegislativeText.objects.filter(assembly="BHB")),
|
|
"white_house": len(LegislativeText.objects.filter(assembly="WHB")),
|
|
"red_ga": len(LegislativeText.objects.filter(assembly="RGA")),
|
|
"blue_ga": len(LegislativeText.objects.filter(assembly="BGA")),
|
|
"white_ga": len(LegislativeText.objects.filter(assembly="WGA")),
|
|
}
|
|
return render(request, "explorer/stats.html", context)
|
|
|
|
def get_all_classified_by_id(request, model_id):
|
|
classification = get_object_or_404(LegislationClassification, pk=model_id)
|
|
# this is very expensive; make a way for this to be cached please?
|
|
|
|
all_texts = LegislativeText.objects.all()
|
|
all_terms = classification.text_to_match.split(',')
|
|
all_terms = [i.lower() for i in all_terms]
|
|
|
|
matches = []
|
|
|
|
for text in all_texts:
|
|
for term in all_terms:
|
|
if term in text.text.lower():
|
|
matches.append(text)
|
|
break
|
|
|
|
return render(request, "explorer/results.html", {
|
|
"legislation": matches,
|
|
"result_name": "All legislation in topic {}".format(classification.name)
|
|
})
|
|
|
|
def get_all_by_x(model):
|
|
def wrapped(request, model_id):
|
|
instance = get_object_or_404(model, pk=model_id)
|
|
return render(request, "explorer/results.html", {
|
|
"result_name": "All legislation by {}".format(instance.name),
|
|
"legislation": instance.legislativetext_set.all()
|
|
})
|
|
|
|
return wrapped
|
|
|
|
def get_all_xs(model):
|
|
def wrapper(request):
|
|
instances = model.objects.all()
|
|
try:
|
|
# what the heck, django?????
|
|
plural = model._meta.verbose_name_plural
|
|
except:
|
|
plural = model.__name__ + "s"
|
|
|
|
plural = plural.lower()
|
|
|
|
return render(request, "explorer/listing.html", {
|
|
"result_name": "All {}".format(plural),
|
|
"instances": instances,
|
|
})
|
|
|
|
return wrapper
|
|
|
|
def return_groups(request):
|
|
listing = {}
|
|
for model in models_in_index:
|
|
try:
|
|
name = model._meta.verbose_name.lower()
|
|
except:
|
|
name = model.__name__.lower()
|
|
|
|
listing[name] = reverse(model.__name__)
|
|
|
|
print(listing)
|
|
return render(request, "explorer/by_group.html", { "listing": listing })
|
|
|
|
get_all_by_school = get_all_by_x(School)
|
|
get_all_by_country = get_all_by_x(Country)
|
|
get_all_by_sponsor = get_all_by_x(Sponsor)
|
|
|
|
get_all_schools = get_all_xs(School)
|
|
get_all_countries = get_all_xs(Country)
|
|
get_all_sponsors = get_all_xs(Sponsor)
|
|
get_all_classifications = get_all_xs(LegislationClassification) |