lingr.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import simplejson
import urllib
import urllib2


class Lingr(object):

    __URL_BASE__ = 'http://lingr.com/api/'
    __URL_BASE_OBSERVE__ = 'http://lingr.com:8080/api/'

    def __init__(self, user, password):
        self.user = user
        self.password = password
        self.counter = 0

    def create_session(self):
        data = self.post('session/create', {
            'user': self.user,
            'password': self.password
        })
        if data:
            self.session = data['session']
            self.nickname = data['nickname']
        return data

    def get_rooms(self):
        data = self.get('user/get_rooms', {
            'session': self.session
        })
        if data:
            self.rooms = data['rooms']
        return data

    def subscribe(self, room=None, reset='true'):
        if not room:
            room = ','.join(self.rooms)
        data = self.post('room/subscribe', {
            'session': self.session,
            'room': room,
            'reset': reset
        })
        if data:
            self.counter = data['counter']
        return data

    def observe(self):
        data = self.get('event/observe', {
            'session': self.session,
            'counter': self.counter
        })
        if 'counter' in data:
            self.counter = data['counter']
        return data

    def say(self, room, text):
        data = self.post('room/say', {
            'session': self.session,
            'room': room,
            'nickname': self.nickname,
            'text': text
        })
        return data

    def post(self, path, params):
        r = self.get_opener().open(self.get_url(path),
                                   urllib.urlencode(params))
        return self.loads(r.read())

    def get(self, path, params):
        full_url = self.get_url(path) + '?' + urllib.urlencode(params)
        r = self.get_opener().open(full_url)
        return self.loads(r.read())

    def loads(self, json):
        data = simplejson.loads(json)
        if data['status'] == 'ok':
            return data
        else:
            print 'error'
            print data
        return None

    def get_url(self, path):
        url = self.__URL_BASE__
        if path == 'event/observe':
            url = self.__URL_BASE_OBSERVE__
        return url + path

    def get_opener(self):
        opener = urllib2.build_opener()
        opener.addheaders = [
            ('User-agent', 'lingrpy(https://github.com/narusemotoki/lingrpy)')
        ]
        return opener

    def stream(self):
        self.create_session()
        self.get_rooms()
        self.subscribe()
        while True:
            obj = self.observe()
            if 'events' in obj:
                for event in obj['events']:
                    yield event