Many files to book - improvements
[wolnelektury.git] / apps / lesmianator / views.py
1 # Create your views here.
2
3 import cPickle
4 from django.shortcuts import render_to_response
5 from django.template import RequestContext
6 from random import randint
7
8
9 def _choose_word(word):
10     try:
11         choices = sum((_dictionary[word][post] for post in _dictionary[word]))
12         r = randint(0, choices - 1)
13
14         for post in _dictionary[word]:
15             r -= _dictionary[word][post]
16             if r < 0:
17                 return post
18     except KeyError:
19         return ''
20
21 # load dictionary on start, it won't change
22 from django.conf import settings
23
24 try:
25     f = open(settings.LESMIANATOR_PICKLE)
26     _dictionary = cPickle.load(f)
27 except:
28     _dictionary = {}
29
30
31 def poem(request):
32     letters = []
33     word = u''
34     empty = -10
35     left = 1000
36     lines = 0
37     if not _dictionary:
38         left = 0
39     # want at least two lines, but let Lesmianator end his stanzas
40     while (empty < 2 or lines < 2) and left:
41         letter = _choose_word(word)
42         letters.append(letter)
43         word = word[-2:] + letter
44         if letter == u'\n':
45             # count non-empty lines
46             if empty == 0:
47                 lines += 1
48             # 
49             if lines >= 2:
50                 empty += 1
51             lines += 1
52         else:
53             empty = 0
54         left -= 1
55
56     poem = ''.join(letters).strip()
57
58     return render_to_response('lesmianator/poem.html', 
59                 {"object": poem},
60                 context_instance=RequestContext(request))