from banjo.urls import route_get, route_post
from banjo.http import BadRequest, NotFound
from app.models import Line, Poem, Rhyme
from random import choice, sample

@route_get('lines/random', args={})
def get_random_line(params):
    return {'line': Line.objects.random().text}

@route_get('lines/about', args={'topic': str})
def get_line_about_topic(params):
    try:
        lines = Line.objects.filter(clean_text__contains=params['topic'])
        if len(lines) == 1:
            return {'line' : lines[0], 'topic': params['topic']}
        else:
            return {'lines' : lines, 'topic': params['topic']}
    except Line.DoesNotExist:
        raise NotFound("No lines with that topic")
    
@route_get('lines/rhyme', args={'rhyme': str})
def get_line_about_topic(params):
    try:
        lines = Rhyme.get_rhyme_for_word(params['rhyme']).lines.exclude(clean_text__endswith=params['rhyme'])
        if len(lines) == 1:
            return {'line' : lines[0], 'rhyme': params['rhyme']}
        else:
            return {'lines' : lines, 'rhyme': params['rhyme']}
    except Line.DoesNotExist:
        raise NotFound("No lines that rhyme")