110 lines
4.9 KiB
Python
Executable File
110 lines
4.9 KiB
Python
Executable File
from django.shortcuts import render
|
||
from django_tables2 import RequestConfig
|
||
from .models import Speech, Protocol
|
||
from .tables import SpeechTable, ProtocolTable
|
||
from django.http import Http404
|
||
from .utils import create_html_speech
|
||
from .forms import SearchForm, SearchFormSpeech
|
||
from watson import search as watson
|
||
from collections import Counter
|
||
|
||
|
||
def speech(request, speech_id):
|
||
try:
|
||
current_speech = Speech.objects.get(pk=speech_id)
|
||
if(current_speech.previous_speech_id is not None):
|
||
previous_speech = Speech.objects.get(pk=current_speech.previous_speech_id)
|
||
previous_speech_html = create_html_speech(previous_speech.speech_content)[0]
|
||
else:
|
||
previous_speech = None
|
||
previous_speech_html = None
|
||
if(current_speech.next_speech_id is not None):
|
||
next_speech = Speech.objects.get(pk=current_speech.next_speech_id)
|
||
next_speech_html = create_html_speech(next_speech.speech_content)[0]
|
||
else:
|
||
next_speech = None
|
||
next_speech_html = None
|
||
current_speech_html, raw_text, interruptions = create_html_speech(current_speech.speech_content)
|
||
vocabulary = Counter(raw_text.split()).most_common()
|
||
unique_words = len(vocabulary)
|
||
tmp_str = []
|
||
for pair in vocabulary:
|
||
tmp_str.append("<li>{}: {}</li>".format(pair[0], pair[1]))
|
||
vocabulary = "".join(tmp_str)
|
||
except Speech.DoesNotExist:
|
||
raise Http404("Speech does not exist")
|
||
context = {"title": ("Rede – "
|
||
+ " "
|
||
+ current_speech.speech_id),
|
||
"current_speech": current_speech,
|
||
"current_speech_html": current_speech_html,
|
||
"previous_speech_html": previous_speech_html,
|
||
"next_speech_html": next_speech_html,
|
||
"previous_speech": previous_speech,
|
||
"next_speech": next_speech,
|
||
"interruptions": interruptions,
|
||
"words": len(raw_text.split()),
|
||
"vocabulary": vocabulary,
|
||
"unique_words": unique_words}
|
||
return render(request, "speeches/speech.html", context)
|
||
|
||
|
||
def speeches(request):
|
||
if(request.method == "GET"):
|
||
form = SearchFormSpeech(request.GET)
|
||
if(form.is_valid()):
|
||
query = form.cleaned_data["query"]
|
||
search_results = watson.filter(Speech, query)
|
||
table = SpeechTable(search_results)
|
||
RequestConfig(request, paginate={'per_page': 20}).configure(table)
|
||
context = {"title": "Suchergebnisse für " + query,
|
||
"form": form, "table": table}
|
||
return render(request, "speeches/speeches.html", context)
|
||
else:
|
||
form = SearchFormSpeech()
|
||
table = SpeechTable(Speech.objects.all().order_by("speech_id"))
|
||
RequestConfig(request, paginate={'per_page': 20}).configure(table)
|
||
context = {"title": "Suche", "table": table, "form": form}
|
||
return render(request, "speeches/speeches.html", context)
|
||
|
||
|
||
def protocol(request, protocol_id):
|
||
try:
|
||
current_protocol = Protocol.objects.get(pk=protocol_id)
|
||
related_speeches = Speech.objects.filter(foreign_protocol=protocol_id).order_by("speech_id")
|
||
speakers = []
|
||
speeches_html = []
|
||
for speech in related_speeches:
|
||
speakers.append(speech.foreign_speaker)
|
||
speech_html = create_html_speech(speech.speech_content)[0]
|
||
speeches_html.append(speech_html)
|
||
speaker_speech_html = zip(speakers, speeches_html, related_speeches)
|
||
except Protocol.DoesNotExist:
|
||
raise Http404("Protocol does not exist")
|
||
context = {"title": ("Protokoll – " + str(current_protocol.protocol_id)),
|
||
"current_protocol": current_protocol,
|
||
"related_speeches": related_speeches,
|
||
"speeches_html": speeches_html,
|
||
"speakers": set(speakers),
|
||
"speaker_speech_html": speaker_speech_html}
|
||
return render(request, "speeches/protocol.html", context)
|
||
|
||
|
||
def protocols(request):
|
||
if(request.method == "GET"):
|
||
form = SearchForm(request.GET)
|
||
if(form.is_valid()):
|
||
query = form.cleaned_data["query"]
|
||
search_results = watson.filter(Protocol, query)
|
||
table = ProtocolTable(search_results)
|
||
RequestConfig(request, paginate={'per_page': 20}).configure(table)
|
||
context = {"title": "Suchergebnisse für " + query,
|
||
"form": form, "table": table}
|
||
return render(request, "speeches/protocols.html", context)
|
||
else:
|
||
form = SearchForm()
|
||
table = ProtocolTable(Protocol.objects.all().order_by("session_date"))
|
||
RequestConfig(request, paginate={'per_page': 20}).configure(table)
|
||
context = {"title": "Suche", "table": table, "form": form}
|
||
return render(request, "speeches/protocols.html", context)
|