start.py

#!/usr/bin/env python
import httplib
import urllib
import json
import sys
from modules.basic_module import BasicModule
from modules.dice import DiceModule
from modules.lunch import LunchModule 
from modules.finance import FinanceModule
from modules.wordcheck import WordCheckModule
from modules.kvdb import KVDBModule
from modules.memo import MemoModule
from modules.event import EventModule
from modules.aqi import AqiModule
from modules.rss import RssModule
from modules.hiking import HikingModule
#from modules.github import GithubModule
from commands import Command, CommandParseException

class RJJBot:

  SERVER = 'api.telegram.org'
  with open('api-key.txt', 'r') as f:
    key = f.read().strip()
  BOT = '/bot%s/' % (key)
  OFFSET_FILE = 'offset.txt'

  def __init__(self):
    self.modules = []
    self.print_reply = False

  def send_request(self, method, data={}):
    c = httplib.HTTPSConnection(self.SERVER)
    headers = {"Content-type": "application/json"}
    try:
      c.request('POST', self.BOT + method, json.dumps(data), headers)
      r = c.getresponse()
      if r.status != 200:
        raise Exception('HTTP response %s: %s' % (r.status, r.reason))
      try:
        res = r.read()
        o = json.loads(res)
        if o['ok'] is not True:
          raise Exception('Method returned failure: %s' % (o.get('result')))
        return o['result']
      except Exception, e:
        raise Exception('Error parsing response JSON: %s' % (res))
    except Exception, e:
      print "ERROR SENDING REQUEST: %s" % (str(e))


  def update_offset(self, offset):
    x = self.get_offset()
    if offset > x:
      with open(self.OFFSET_FILE, 'w') as f:
        f.write(str(offset))

  def get_offset(self):
    with open(self.OFFSET_FILE, 'r') as f:
      x = f.read()
      return int(x)

  def process_messages(self, messages):
    if messages is None:
      return
    for message in messages:
      self.process_message(message)

  def send_message(self, text, chat_id):
    if isinstance(text, basestring):
      self.send_request('sendMessage', {'chat_id': chat_id, 'text': text.encode('utf-8'), 'disable_web_page_preview': True})
    elif text.get('type') == 'html':
      self.send_request('sendMessage', {'chat_id': chat_id, 'text': text.get('content').encode('utf-8'), 'parse_mode': 'HTML', 'disable_web_page_preview': True if text.get('disable_preview') == None else text.get('disable_preview')})
    elif text.get('type') == 'sticker':
      self.send_request('sendSticker', { 'chat_id': chat_id, 'sticker': text.get('content', '') })

###### helpers above #######


  def process_message(self, msg):
    update_id = int(msg.get('update_id'))
    self.update_offset(update_id)
    print 'Now process %s' % (update_id)
    m = msg.get('message')
    if m is not None and m.get('message_id') is not None:
      m_id = m.get('mesesage_id')
      chat_id = m['chat']['id']
      command = None
      if 'text' in m:
        try:
          command = Command(m['text'])
        except CommandParseException, e:
          command = None
      # Delegate messages to modules
      for module in self.modules:
        try:
          reply = module.process_message(m, command)
        except TypeError, e:
          reply = module.process_message(m)
        if reply != None:
          if self.print_reply:
            print reply
          else:
            self.send_message(reply, chat_id)
          return


  def start(self):
    print "RJJ Standby"
    while True:
      try:
        #if chat_id is not None:
        #    self.send_message(message, chat_id)
        update_id = self.get_offset()
        res = self.send_request('getUpdates', { 'offset': update_id + 1 })
        self.process_messages(res)
      except Exception, e:
        raise
        print 'Exception caught: ', type(e)
      for module in self.modules:
        ret = module.tick()
        if ret is not None:
          self.send_message(ret, '-8311114')
      import time
      time.sleep(1)
    print "RJJ Close"

  def start_local(self):
    self.print_reply = True
    print ("> "),
    while True:
      line = sys.stdin.readline()
      msg = {
        'update_id': 1,
        'message': {
          'text': line,
          'message_id': 1,
          'chat': { 'id': 9999 },
          'from': {
            'id': 'local',
            'first_name': 'Local'
          }
        }
      }
      if len(line) == 0:
        break
      self.process_message(msg)
      print ("> "),
    print "RJJ Close"



if __name__ == '__main__':
    rjj = RJJBot()
    rjj.modules = [ AqiModule(), BasicModule(), MemoModule(), DiceModule(), LunchModule(), FinanceModule(), WordCheckModule(), KVDBModule(), EventModule(), RssModule(), HikingModule() ]
    if (len(sys.argv) > 1 and sys.argv[1] == "local"):
      rjj.start_local()
    else:
      rjj.start()