twitools/twitools/streaming.py

import dbtools, html, tweepy

class BotStreamListener(tweepy.StreamListener):
 def __init__(self, bot, cid, *args, **kwargs):
  tweepy.StreamListener.__init__(self, *args, **kwargs)
  self.bot = bot
  self.cid = cid

 def on_status(self, status):
  db = dbtools.dbHelper()
  try:
   db.executeQuery("SELECT MAX(nr) FROM timelines WHERE cid = %i;" % self.cid)
   i = int(db.getNext()[0]) + 1
  except:
   i = 1

  db.executeQuery("INSERT INTO timelines VALUES(%i, %i, %i);" % (self.cid, i, status.id))
  db.commit()

  self.bot.sendMessage(chat_id=self.cid, text="Tweet %i:\n%s (@%s) at %s:\n%s" % (i, status.author.name, status.author.screen_name, status.created_at, html.unescape(status.text)))

 def on_error(self, status):
  if status == 420:
   return False