#!/usr/bin/env python

import BaseHTTPServer

import base64

TEST_LINES = ['hello',

              'world',

              'how are',

              'you?']

NEWLINE = '\n'

class TwitterStreamRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):

    def is_authorized(self):

        try:

            authentication = self.headers['Authorization']

            decoded_auth = base64.b64decode(authentication.split()[1])

            username = decoded_auth.split(':')[0]

            password = decoded_auth.split(':')[1]

            print("username: %s" % username)

            print("password: %s" % password)

            return True

        except KeyError:

            return False

    def echo_request(self):

        if(self.is_authorized()):

            request = "GET: %s" % self.path

            self.send_response(200)

            self.send_header('content-type', 'text/plain')

            self.end_headers()

            for line in TEST_LINES:                

                self.wfile.write(line)

                self.wfile.write(NEWLINE)

        else:

            self.send_error(401, "Not Authorized")

    def do_GET(self):

        #print(self.headers)

        self.echo_request()

def run(server_class=BaseHTTPServer.HTTPServer,

        handler_class=BaseHTTPServer.BaseHTTPRequestHandler):

    server_address = ('', 8000)

    httpd = server_class(server_address, handler_class)

    httpd.serve_forever()

if __name__ == '__main__':

    run(handler_class=TwitterStreamRequestHandler)

from twisted.internet import protocol, reactor

from twisted.web.client import HTTPPageGetter, HTTPClientFactory, _makeGetterFactory

import base64

class StreamingTwitter(HTTPPageGetter):

    receiving_tweets = False

    def lineReceived(self, line):

        if(self.receiving_tweets is True):

            print "got line: %s" % line

        else:

            return HTTPPageGetter.lineReceived(self, line)

    def handleEndHeaders(self):

        self.receiving_tweets = True

        return HTTPPageGetter.handleEndHeaders(self)

class StreamingTwitterClientFactory(HTTPClientFactory):

    protocol = StreamingTwitter

def getTwitterStream(url, username, password, contextFactory=None, *args, **kwargs):

    encoded_auth = base64.encodestring("%s:%s" % (username, password))

    authorization_header = "Basic %s" % encoded_auth

    kwargs.update({'headers' : {'authorization': authorization_header}})

    return _makeGetterFactory(

        url,

        StreamingTwitterClientFactory,

        contextFactory=contextFactory,

        *args, **kwargs).deferred    

if __name__ == '__main__':

    #d = getTwitterStream('http://localhost:8000', 'mbone', 'helloworld')

    from passwords import TWITTER_USERNAME, TWITTER_PASSWORD

    d = getTwitterStream('http://stream.twitter.com/1/statuses/sample.json', TWITTER_USERNAME, TWITTER_PASSWORD)

    def echo_page(page):

        print(page)

        reactor.stop()

    d.addCallback(echo_page)

    reactor.run()