source: rc/puzzlebox_brainstorms_server.py @ 6

Last change on this file since 6 was 6, checked in by sc, 12 years ago

whitespace and comment cleanup

File size: 4.4 KB
Line 
1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3#
4# Puzzlebox - Brainstorms - Server
5#
6# Copyright Puzzlebox Productions, LLC (2010)
7#
8# Portions of this code have been previously
9# released under the GNU Pulic License (GPL) version 2
10# and is Copyright Steven M. Castellotti (2010)
11# For more information please refer to http://www.gnu.org/copyleft/gpl.htm
12#
13# Last Update: 2010.01.28
14#
15#####################################################################
16
17import os, signal, sys, time
18import cPickle as pickle
19
20from twisted.internet import reactor, protocol, defer
21
22import puzzlebox_brainstorms_configuration as configuration
23import puzzlebox_brainstorms_client
24#import puzzlebox_logger
25
26#####################################################################
27# Globals
28#####################################################################
29
30DEBUG = 1
31
32SERVER_HOST = configuration.SERVER_HOST
33SERVER_PORT = configuration.SERVER_PORT
34
35#####################################################################
36# Classes
37#####################################################################
38
39class puzzlebox_master_control(protocol.ServerFactory):
40       
41        def __init__(self, log, DEBUG=DEBUG):
42               
43                self.protocol = puzzlebox_master_control_server_protocol
44               
45                self.log = log
46                self.DEBUG = DEBUG
47               
48                self.registry = {}
49       
50       
51        ##################################################################
52       
53        def process_instruction(self, instruction):
54       
55                #self.log.debug("Received instruction: %s" % instruction)
56                #if self.DEBUG:
57                        #print "Received instruction: %s" % instruction
58               
59                d = defer.Deferred()
60                response = 'instruction received'
61               
62                if 'command' in instruction:
63                        response = '%s instruction received' % instruction['command']
64               
65               
66                        if 'information' in instruction:
67                                information = instruction['information']
68                       
69                       
70                        if instruction['command'] == 'test_drive':
71                               
72                                command = 'python puzzlebox_brainstorms_rc.py'
73                               
74                                os.system(command)
75                       
76                       
77                        else:
78                               
79                                #self.log.error("Unrecognized command received: %s" % instruction)
80                                if self.DEBUG:
81                                        print "Unrecognized command received: %s" % instruction
82                                reponse = 'unrecognized instruction'
83               
84               
85                if response:
86                        d.callback(response)
87               
88                return d
89       
90       
91        ##################################################################
92       
93        def process_connection_lost(self, instruction):
94               
95                if not instruction:
96                       
97                        #self.log.debug("Connection lost with no instruction")
98                       
99                        if self.DEBUG:
100                                print "Connection lost with no instruction"
101               
102               
103                else:
104                       
105                        #self.log.debug("Connection lost with no instruction")
106                       
107                        if self.DEBUG:
108                                print "Connection lost instruction"
109
110
111#####################################################################
112# Protocol
113#####################################################################
114
115class puzzlebox_master_control_server_protocol(protocol.Protocol):
116       
117        def __init__(self):
118                self.instruction = {}
119                self.data_chunk = ""
120       
121       
122        ##################################################################
123       
124        def dataReceived(self, data):
125               
126                self.data_chunk += data
127                try:
128                        self.instruction = pickle.loads(self.data_chunk)
129                except Exception, e:
130                        self.factory.log.error("Partial data received (or error: %s)." % e)
131                else:
132                        self.data_chunk = ""
133                       
134                        d = self.factory.process_instruction(self.instruction.copy())
135                        d.addCallback(self.send_response)
136       
137       
138        ##################################################################
139       
140        def send_response(self, response):
141               
142                #if response == "browser exit":
143                        #self.instruction['command'] = response
144               
145                response = pickle.dumps(response)
146               
147                self.transport.write(response)
148       
149       
150        ##################################################################
151       
152        def connectionLost(self, reason):
153               
154                self.factory.process_connection_lost(self.instruction)
155
156
157#####################################################################
158# Main
159#####################################################################
160
161if __name__ == '__main__':
162
163        #log = puzzlebox_logger.puzzlebox_logger(logfile='master_control')
164        log = None
165       
166        # Collect default settings and command line parameters
167        server_host = SERVER_HOST
168        server_port = SERVER_PORT
169       
170        for each in sys.argv:
171               
172                if each.startswith("--host="):
173                        server_host = each[ len("--host="): ]
174                if each.startswith("--port="):
175                        server_port = each[ len("--port="): ]
176       
177       
178        mcp = puzzlebox_master_control(log, DEBUG)
179        reactor.listenTCP(port=server_port, factory=mcp, interface=server_host)
180        reactor.run()
181
Note: See TracBrowser for help on using the repository browser.