source: remote_control/puzzlebox_brainstorms_client_thinkgear.py @ 52

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Puzzlebox - Brainstorms - Client - ThinkGear Connector
#
# Copyright Puzzlebox Productions, LLC (2010)
#
# This code is released under the GNU Pulic License (GPL) version 2
# For more information please refer to http://www.gnu.org/copyleft/gpl.html
#
# Last Update: 2010.06.16
#
#####################################################################

import os, sys
import simplejson as json

from twisted.internet import reactor, protocol, defer
from twisted.protocols import basic

import puzzlebox_brainstorms_configuration as configuration
import puzzlebox_brainstorms_client as client

#####################################################################
# Globals
#####################################################################

DEBUG = 1

SERVER_HOST = configuration.THINKGEAR_SERVER_HOST
SERVER_PORT = configuration.THINKGEAR_SERVER_PORT

MAX_CONNECTION_ATTEMPTS = configuration.MAX_CONNECTION_ATTEMPTS
NO_REPLY_WAIT = configuration.NO_REPLY_WAIT


AUTHORIZATION_REQUEST = configuration.THINKGEAR_AUTHORIZATION_REQUEST

THINKGEAR_PARAMETERS = {"enableRawOutput": False, "format": "Json"}
#THINKGEAR_PARAMETERS = {"enableRawOutput": True, "format": "Json"}

#####################################################################
# Classes
#####################################################################

class puzzlebox_brainstorms_client_thinkgear(client.puzzlebox_brainstorms_client):
        
        def __init__(self, log, \
                          authorization_request, \
                          server_host=SERVER_HOST, \
                          server_port=SERVER_PORT, \
                          DEBUG=DEBUG):
                
                self.log = log
                self.DEBUG=DEBUG
                
                self.authorization_request = authorization_request
                self.server_host = server_host
                self.server_port = server_port
                self.max_connection_attempts = MAX_CONNECTION_ATTEMPTS

        
        ##################################################################
        
        def send_command(self, \
                         command, \
                         max_connection_attempts=MAX_CONNECTION_ATTEMPTS):
                
                factory = puzzlebox_brainstorms_client_thinkgear_factory(self.log, \
                                                       command, \
                                                       self.server_host, \
                                                       self.server_port, \
                                                       max_connection_attempts, \
                                                       self.DEBUG)
                
                reactor.connectTCP(self.server_host, self.server_port, factory)
                
                return factory.replyDefer
        
        
        ##################################################################
        
        def send_command_and_print_response(self, command):

                if self.DEBUG:
                        print "---> [Client] Sending:",
                        print command

                        
                d = self.send_command(command)
                d.addCallback(self.print_response)
        
        
        ##################################################################
        
        def print_response(self, response):
                
                if self.DEBUG:
                        print "---> [Client] Server Response:",
                        print response
                

#####################################################################
# ThinkGear Client Protocol class
#####################################################################

class puzzlebox_brainstorms_client_thinkgear_protocol( \
        basic.LineReceiver):

        delimiter='\r'
        
        def __init__(self):
                
                self.DEBUG = DEBUG

        
        ##################################################################
        
        def connectionMade(self):
                
                #data = pickle.dumps(self.factory.command)
                data = json.dumps(self.factory.command)
                self.transport.write(data)
                
                self.factory.noReply = reactor.callLater(NO_REPLY_WAIT, self.noReply)
        
        
        ##################################################################
        
        def noReply(self):
                
                try:
                        self.factory.replyDefer.callback('NO_REPLY')
                except:
                        if self.DEBUG:
                                print "noReply failed to call callback"
                        #self.factory.log.error("noReply failed to call callback")
                
                self.transport.loseConnection()
        
        
        ##################################################################

        def lineReceived(self, line):

                # Ignore blank lines
                if not line:
                        return
                

                data = json.loads(line)

                if self.DEBUG:
                        if ('rawEeg' in data.keys()):
                                if self.DEBUG > 1:
                                        print "data:",
                                        print data
                        else:
                                print "data:",
                                print data


#####################################################################
# ThinkGear Client Factory class
#####################################################################

class puzzlebox_brainstorms_client_thinkgear_factory( \
        client.puzzlebox_brainstorms_client_send_command_factory):
        
        def __init__(self, log, \
                     command, \
                     server_host=SERVER_HOST, \
                     server_port=SERVER_PORT, \
                     max_connection_attempts=MAX_CONNECTION_ATTEMPTS, \
                     DEBUG=DEBUG):
                
                self.log = log
                self.DEBUG = DEBUG
                self.server_host = server_host
                self.server_port = server_port
                self.command = command
                
                self.max_connection_attempts = max_connection_attempts
                self.connection_attempt = 1
                
                self.protocol = \
                   puzzlebox_brainstorms_client_thinkgear_protocol
                
                self.replyDefer = defer.Deferred()


#####################################################################
# Main
#####################################################################

if __name__ == '__main__':
        
        #log = puzzlebox_logger.puzzlebox_logger(logfile='client')
        log = None
        
        #authorization_request = AUTHORIZATION_REQUEST
        authorization_request = None # ThinkGear authentication not working
        thinkgear_parameters = THINKGEAR_PARAMETERS
        
        #log.info("Command parameters: %s" % command_parameters)

        thinkgear_client = puzzlebox_brainstorms_client_thinkgear( \
                    log, \
                    authorization_request, \
                    server_host=SERVER_HOST, \
                    server_port=SERVER_PORT, \
                    DEBUG=DEBUG)

        reactor.callWhenRunning( \
                thinkgear_client.send_command_and_print_response, \
                        thinkgear_parameters)


        reactor.run()