views.py 1.65 KB

from rest_framework.decorators import api_view
from rest_framework.response import Response

from webapp.models import Expression, Segment

# wywolanie
#http://localhost:8000/get_expressions/?expression=kot
@api_view(['GET'])
def get_expressions(request):
    expressions = Expression.objects.filter(text=request.GET['expression'].upper())
    meanings = [expr.meaning for expr in expressions]
    expressions_pks = []
    for meaning in meanings:
        for expr in meaning.expressions.filter(main_expression=None):
            expressions_pks.append(expr.pk)
    expressions = Expression.objects.filter(pk__in=expressions_pks).distinct('text')
    expressions_text_reps = [expr.lemmatized_form() for expr in expressions.order_by('text')]
    #expressions_text_reps = [unicode(expr) for expr in expressions.order_by('text')]
    return Response({'expressions': expressions_text_reps})


# wywolanie
#http://localhost:8000/get_synonyms/?expression=kot
@api_view(['GET'])
def get_synonyms(request):
    #segments = Segment.objects.filter(base=request.GET['expression'])
    expressions = Expression.objects.filter(text=request.GET['expression'].upper())

    meanings = [expr.meaning for expr in expressions if expr.segments.count() == 1]
    expressions_pks = []
    for meaning in meanings:
        for expr in meaning.expressions.filter(main_expression=None):
            if expr.segments.count() == 1:
                expressions_pks.append(expr.pk)
    expressions = Expression.objects.filter(pk__in=expressions_pks).distinct('text')
    expressions_text_reps = [expr.lemmatized_form() for expr in expressions.order_by('text')]
    return Response({'synonyms': expressions_text_reps})