Make Markov bot use NLTK.
This commit is contained in:
parent
94cfb56611
commit
6df747c303
1 changed files with 13 additions and 3 deletions
16
markov.py
16
markov.py
|
@ -1,16 +1,26 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
import dbtools, twitools
|
import dbtools, twitools
|
||||||
import argparse, markovify, operator, random, re, sys
|
import argparse, markovify, nltk, operator, random, re, sys
|
||||||
|
|
||||||
|
class Possy(markovify.NewlineText):
|
||||||
|
def word_split(self, sentence):
|
||||||
|
words = re.split(self.word_split_pattern, sentence)
|
||||||
|
words = [ "::".join(tag) for tag in nltk.pos_tag(words) ]
|
||||||
|
return words
|
||||||
|
|
||||||
|
def word_join(self, words):
|
||||||
|
sentence = " ".join(word.split("::")[0] for word in words)
|
||||||
|
return sentence
|
||||||
|
|
||||||
def getText(db = dbtools.dbHelper()):
|
def getText(db = dbtools.dbHelper()):
|
||||||
text = ""
|
text = ""
|
||||||
for string in db.executeQuery("SELECT text FROM tweets;"):
|
for string in db.executeQuery("SELECT text FROM tweets;"):
|
||||||
text += string[0] + "\n"
|
text += string[0] + "\n"
|
||||||
return text
|
return "".join([s for s in text.strip().splitlines(True) if s.strip()])
|
||||||
|
|
||||||
def markovifyText(text):
|
def markovifyText(text):
|
||||||
return markovify.NewlineText(text).make_short_sentence(130).replace("@", "@")
|
return Possy(text).make_short_sentence(130).replace("@", "@")
|
||||||
|
|
||||||
def tweet(text, ref = 0, two = twitools.twObject()):
|
def tweet(text, ref = 0, two = twitools.twObject()):
|
||||||
return two.tweet(text, ref).id
|
return two.tweet(text, ref).id
|
||||||
|
|
Loading…
Reference in a new issue