lab_server/poem_server/app/views.py

57 lines
1.8 KiB
Python

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_about_line(params):
try:
line = Line.objects.filter(clean_text_contains=params['topic']).random()
return line.to_dict()
except Line.DoesNotExist:
raise NotFound("Line not found")
@route_get('lines/rhyme', args={'word' : str})
def get_rhyme_line(params):
try:
rhyme = Rhyme.get_rhyme_for_word(params['word']).lines.exclude(clean_text__endswidth=" "+params['word']).first()
return rhyme.to_dict()
except Rhyme.DoesNotExist:
raise NotFound("Rhyme not found")
@route_get('couplets/random', args={})
def get_random_couplet(params):
try:
line = Line.objects.random()
for rl in line.rhyming_lines().sample(1):
rhyme = rl
return {'lines' : [line, rhyme]}
except Rhyme.DoesNotExist:
raise NotFound("Rhyme Not Found")
@route_get('couplets/about', args={'topic' : str})
def get_about_couplet(params):
try:
line = Line.objects.filter(clean_text__contains=params['topics']).random()
for rl in line.rhyming_lines().sample(1):
rhyme = rl
return {'lines' : [line.text, rhyme.text]}
except:
pass
@route_get('couplets/rhyme', args={'word' : str})
def get_rhyme_couplet(params):
try:
toRhyme = Rhyme.get_rhyme_for_word(params['word']).lines.exclude(clean_text__endswith= " " + params['word']).first()
#line = Line.
for rl in line.rhyming_lines().sample(1):
rhyme = rl
return {lines : [lines.text, rhyme.text]}
except:
pass