twitools/markov.py

#!/usr/bin/env python3

import dbtools, setuptools, twitools
import argparse, datetime, html, markovify, nltk, operator, os, random, re, string, sys, time

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 sanitizeText(text):
 split = text.split()
 try:
  if "@" in (text[0], text[1]):
   if split[1][0] not in string.ascii_lowercase:
    return sanitizeText(text.partition(" ")[2])
  if split[-1][0] == "@":
   return sanitizeText(" ".join(split[:-1]))
  if text[:4] == "RT @":
   return sanitizeText(text.partition(":")[2])
 except:
  return ""
 return text

def getText(db = dbtools.dbHelper()):
 text = ""
 for string in db.executeQuery('SELECT text FROM tweets;'):
  text += sanitizeText(string[0]) + "\n"
 return html.unescape("".join([s for s in text.strip().splitlines(True) if s.strip()]))

def markovifyText(text):
 return Possy(text).make_short_sentence(130).replace("@", "@​")

def getTime():
 now = datetime.datetime.now()

 thenminute = 15 if now.minute < 15 else 30 if now.minute < 30 else 45 if now.minute < 45 else 0
 thenhour = now.hour + 1 if thenminute == 0 else now.hour

 then = datetime.datetime(now.year, now.month, now.day, thenhour, thenminute, 0)
 return (then - now).seconds

if __name__ == "__main__":
 os.nice(5)
 text = markovifyText(getText())
 time.sleep(getTime())
 twitools.tweet(text, section = setuptools.MARKOV)