Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def get_next_five(last_word, starter=None):
- query = mm.WordLink.select().join(mm.Word).where(mm.Word.name == last_word)
- result = query.execute()
- l = list(result)
- if starter is not None:
- l = [wl for wl in l if wl.suggestion.name.startswith(starter)]
- ret = sorted(l, key=lambda k: k.weight, reverse=True)[:5]
- p = []
- for r in ret:
- p.append(r.suggestion.name)
- return p
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement