from django.shortcuts import get_object_or_404, render from django.http import HttpResponse from .models import LegislativeText, LegislationBook, LegislationClassification from random import sample def index(request): legislative_texts = list(LegislativeText.objects.all()) legislative_texts = sample(legislative_texts, 5) 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_classifications(request): classifications = LegislationClassification.objects.all() return render(request, "explorer/classifications.html", { "classifications": classifications, }) def get_all_classified_by_id(request, classification_id): classification = get_object_or_404(LegislationClassification, pk=classification_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) })