def post(self):
body = json.loads(self.request.body)
query = body['query']
n = body['n']
limit = body['limit']
ngrams = Ngrams(db=self.db)
res = ngrams.find(query=query, n=n, limit=limit)
self.write(json.dumps(res))
def post_actions(seq, h=1, adres_slice=3):
sessions = dict()
nums = dict()
for key in seq:
sessions[key] = [(i[0][:adres_slice], i[1])
for i in sorted(seq[key], key=lambda x: x[1])]
nums[key] = [
i[1] for i in sorted(Counter(sessions[key]).items(),
key=lambda x: x[0][1])
]
sessions[key] = [
i[0] for i in sorted(set(sessions[key]), key=lambda x: x[1])
]
# ngrams
print str(h) + 'grams'
ng = Ngrams(sessions=sessions, n=h)
# price-list
vals = list(
map(lambda x: (x[0], x[1] * 10**(len(x[0])) - 1), ng.ngrams.items()))
tag_list = sorted(list(set(vals)), key=lambda x: x[1], reverse=True)
# tagging
res = actions(sessions, tag_list, ng)
act = []
for key in seq:
superactions = []
for i in range(len(res[key])):
superactions += [res[key][i] for j in range(nums[key][i])]
if len(seq[key]) > h:
for j in range(len(seq[key])):
i = seq[key][j]
action = {
'_op_type': 'update',
u'_id': i[2],
u'_type': u'auditd-parent',
u'_index': u'wailt'
}
if superactions[j] == (-1, ):
superactions[j] = (-1, -1)
action['doc'] = {
'action': superactions[j][0],
'action_separator': superactions[j][1]
}
act.append(action)
if len(act) > 10000:
print 'acted'
bulk(es, actions=act)
act = []
bulk(es, actions=act)
class Poem(object):
"""
Main class responsible for generating poems
"""
QUERY_LIMIT = 1024 * 4
def __init__(self, ptype='haiku', smoothing='backoff', db=None):
self.ngrams = Ngrams(db)
self.type = ptype
self.smoothing = smoothing
self.pattern = []
if self.type == 'haiku':
self.pattern = ['*****', '*******', '*****']
self.rs = None
if self.type == 'tanka':
self.pattern = ['*****', '*******', '*****', '*******', '*******']
self.rs = None
if self.type == 'limerick':
self.rs = 'AABBA'
lengths = []
lengths.append(randint(8, 11))
lengths.append(randint(8, 11))
lengths.append(randint(5, 7))
lengths.append(randint(5, 7))
lengths.append(randint(8, 11))
for length in lengths:
self.pattern.append('*' * length)
if self.type == 'quatrain':
mean = randint(5, 12)
lengths = []
for x in range(0, 3):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
#What we're doing now is giving the AAAA rhyme scene a 1/5 chance of happening
p = randint(0,3)
if p == 0:
self.rs = 'AAAA'
if p == 1:
self.rs = 'AABB'
if p == 2:
self.rs = 'ABAB'
if p == 3:
self.rs = 'ABBA'
if self.type == 'spens_sonnet':
mean = randint(7, 12)
lengths = []
for x in range(0, 13):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
self.rs = 'ABABBCBCCDCDEE'
if self.type == 'ital_sonnet':
mean = randint(7, 12)
lengths = []
for x in range(0, 13):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
octave = 'ABBAABA'
sestet = ''
p = randint(0,3)
if p == 0:
sestet = 'CDCDCD'
if p == 1:
sestet= 'CDDCDC'
if p == 2:
sestet = 'CDECDE'
if p == 3:
sestet = 'CDECED'
self.rs = octave + sestet
if self.type == 'shaks_sonnet':
mean = randint(7, 12)
lengths = []
for x in range(0, 13):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
self.rs = 'ABABCDCDEFEFGG'
self.template = pt.PoemTemplate(self.pattern, self.rs)
def generate(self):
"""
Generates a new poem with a new rhyming and template
"""
rhymesch = RhymeScheme(self.rs) if self.rs else None
sentences = self.template.createTemplate()
while len(sentences) == 0:
sentences = self.template.createTemplate()
text = ''
chosen_words = []
for sentence in sentences:
for word in sentence:
if not isinstance(word, pt.Word):
# if not Word, is a punctuation
text = text[:-1] + word + ' '
continue
word.text = self._next_word(chosen_words, word, rhymesch)
if not word.text or not len(word.text):
# TODO: throw this SENTENCE away, not the entire poem!!!
return
text += word.text + ' '
chosen_words.append(word)
text += '\n'
# final formatting fix
text = text[0].upper() + text[1:-2] + '.\n'
return text
def _weighted_choice(self, choices):
""" Choose one of the possible tuples """
total = sum(w for c, w in choices)
r = random.uniform(0, total)
upto = 0
for c, w in choices:
if upto + w > r:
return c
upto += w
return [], []
def _weighted_tuples(self, listOfNGrams, key, multiplier=1):
""" Weight each possible tuple by its frequency and by its multiplier """
tuples = []
for item in listOfNGrams:
tuple = []
tuple.append(item[key])
tuple.append(multiplier * item[u'freq'])
tuples.append(tuple)
return tuples
def _smoothed_generation(self, unigrams, bigrams=[], trigrams=[], fourgrams=[]):
""" Returns the smoothed choice over all ngrams """
if self.smoothing == 'linear':
multiplierUnigrams = 1
multiplierBigrams = 50
multiplerTrigrams = 200
multiplierFourgrams = 1000
tuples1 = self._weighted_tuples(unigrams, u'word1', multiplierUnigrams)
tuples2 = self._weighted_tuples(bigrams, u'word1', multiplierBigrams)
tuples3 = self._weighted_tuples(trigrams, u'word2', multiplerTrigrams)
tuples4 = self._weighted_tuples(fourgrams, u'word3', multiplierFourgrams)
return self._weighted_choice(tuples1+tuples2+tuples3+tuples4)
if self.smoothing == 'backoff':
if(len(fourgrams) > 0):
tuples4 = self._weighted_tuples(fourgrams, u'word3')
return self._weighted_choice(tuples4)
if(len(trigrams) > 0):
tuples3 = self._weighted_tuples(trigrams, u'word2')
return self._weighted_choice(tuples3)
if(len(bigrams) > 0):
tuples2 = self._weighted_tuples(bigrams, u'word1')
return self._weighted_choice(tuples2)
tuples1 = self._weighted_tuples(unigrams, u'word1')
if (len(unigrams) > 0):
tuples1 = self._weighted_tuples(unigrams, u'word1')
return self._weighted_choice(tuples1)
return []
def _next_word(self, words, nextword, rhymesch):
""" Fetch the next word of the poem """
N = len(words)
query = {}
# if end of line, we need to add the rhyme constraint to the queries
if nextword.rhyme != '*' and rhymesch:
rhyme = rhymesch.get_rhyme(nextword.rhyme)
query['rhyme'] = rhyme
query['syllables'] = nextword.syllables
query['type'] = nextword.typespeech
unigrams = self.ngrams.find(query, n=2, limit=Poem.QUERY_LIMIT)
if(N<=1):
choice = self._smoothed_generation(unigrams)
if len(choice) == 0:
return []
return choice
query['word0'] = words[N-1].text
bigrams = self.ngrams.find(query, n=2, limit=Poem.QUERY_LIMIT)
if(N<=2):
choice = self._smoothed_generation(unigrams, bigrams)
return choice
query['word0'] = words[N-2].text
query['word1'] = words[N-1].text
trigrams = self.ngrams.find(query, n=3, limit=Poem.QUERY_LIMIT)
if(N<=3):
choice = self._smoothed_generation(unigrams, bigrams, trigrams)
return choice
query['word0'] = words[N-3].text
query['word1'] = words[N-2].text
query['word2'] = words[N-1].text
fourgrams = self.ngrams.find(query, n=4, limit=Poem.QUERY_LIMIT)
choice = self._smoothed_generation(unigrams, bigrams, trigrams, fourgrams)
return choice
def __init__(self, ptype='haiku', smoothing='backoff', db=None):
self.ngrams = Ngrams(db)
self.type = ptype
self.smoothing = smoothing
self.pattern = []
if self.type == 'haiku':
self.pattern = ['*****', '*******', '*****']
self.rs = None
if self.type == 'tanka':
self.pattern = ['*****', '*******', '*****', '*******', '*******']
self.rs = None
if self.type == 'limerick':
self.rs = 'AABBA'
lengths = []
lengths.append(randint(8, 11))
lengths.append(randint(8, 11))
lengths.append(randint(5, 7))
lengths.append(randint(5, 7))
lengths.append(randint(8, 11))
for length in lengths:
self.pattern.append('*' * length)
if self.type == 'quatrain':
mean = randint(5, 12)
lengths = []
for x in range(0, 3):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
#What we're doing now is giving the AAAA rhyme scene a 1/5 chance of happening
p = randint(0,3)
if p == 0:
self.rs = 'AAAA'
if p == 1:
self.rs = 'AABB'
if p == 2:
self.rs = 'ABAB'
if p == 3:
self.rs = 'ABBA'
if self.type == 'spens_sonnet':
mean = randint(7, 12)
lengths = []
for x in range(0, 13):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
self.rs = 'ABABBCBCCDCDEE'
if self.type == 'ital_sonnet':
mean = randint(7, 12)
lengths = []
for x in range(0, 13):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
octave = 'ABBAABA'
sestet = ''
p = randint(0,3)
if p == 0:
sestet = 'CDCDCD'
if p == 1:
sestet= 'CDDCDC'
if p == 2:
sestet = 'CDECDE'
if p == 3:
sestet = 'CDECED'
self.rs = octave + sestet
if self.type == 'shaks_sonnet':
mean = randint(7, 12)
lengths = []
for x in range(0, 13):
lengths.append(randint(mean-1, mean+1))
for length in lengths:
self.pattern.append('*' * length)
self.rs = 'ABABCDCDEFEFGG'
self.template = pt.PoemTemplate(self.pattern, self.rs)