source: rc/puzzlebox_brainstorms_server.py @ 8

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

turn_reverse function added

File size: 5.1 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                                command = 'python puzzlebox_brainstorms_rc.py --command=test_drive'
72                                os.system(command)
73                       
74                       
75                        elif instruction['command'] == 'drive_forward':
76                                command = 'python puzzlebox_brainstorms_rc.py --command=drive_forward'
77                                os.system(command)                     
78                       
79                        elif instruction['command'] == 'drive_reverse':
80                                command = 'python puzzlebox_brainstorms_rc.py --command=drive_reverse'
81                                os.system(command)
82                       
83                       
84                        elif instruction['command'] == 'turn_in_reverse':
85                                command = 'python puzzlebox_brainstorms_rc.py --command=turn_in_reverse'
86                                os.system(command)
87                       
88                       
89                        elif instruction['command'] == 'turn_left':
90                                command = 'python puzzlebox_brainstorms_rc.py --command=turn_left'
91                                os.system(command)
92                       
93                       
94                        elif instruction['command'] == 'turn_right':
95                                command = 'python puzzlebox_brainstorms_rc.py --command=turn_right'
96                                os.system(command)
97                       
98                       
99                        else:
100                               
101                                #self.log.error("Unrecognized command received: %s" % instruction)
102                                if self.DEBUG:
103                                        print "Unrecognized command received: %s" % instruction
104                                reponse = 'unrecognized instruction'
105               
106               
107                if response:
108                        d.callback(response)
109               
110               
111                return d
112       
113       
114        ##################################################################
115       
116        def process_connection_lost(self, instruction):
117               
118                if not instruction:
119                       
120                        #self.log.debug("Connection lost with no instruction")
121                       
122                        if self.DEBUG:
123                                print "Connection lost with no instruction"
124               
125               
126                else:
127                       
128                        #self.log.debug("Connection lost with no instruction")
129                       
130                        if self.DEBUG:
131                                print "Connection lost instruction"
132
133
134#####################################################################
135# Protocol
136#####################################################################
137
138class puzzlebox_master_control_server_protocol(protocol.Protocol):
139       
140        def __init__(self):
141                self.instruction = {}
142                self.data_chunk = ""
143       
144       
145        ##################################################################
146       
147        def dataReceived(self, data):
148               
149                self.data_chunk += data
150                try:
151                        self.instruction = pickle.loads(self.data_chunk)
152                except Exception, e:
153                        self.factory.log.error("Partial data received (or error: %s)." % e)
154                else:
155                        self.data_chunk = ""
156                       
157                        d = self.factory.process_instruction(self.instruction.copy())
158                        d.addCallback(self.send_response)
159       
160       
161        ##################################################################
162       
163        def send_response(self, response):
164               
165               
166                response = pickle.dumps(response)
167               
168                self.transport.write(response)
169       
170       
171        ##################################################################
172       
173        def connectionLost(self, reason):
174               
175                self.factory.process_connection_lost(self.instruction)
176
177
178#####################################################################
179# Main
180#####################################################################
181
182if __name__ == '__main__':
183
184        #log = puzzlebox_logger.puzzlebox_logger(logfile='master_control')
185        log = None
186       
187        # Collect default settings and command line parameters
188        server_host = SERVER_HOST
189        server_port = SERVER_PORT
190       
191        for each in sys.argv:
192               
193                if each.startswith("--host="):
194                        server_host = each[ len("--host="): ]
195                if each.startswith("--port="):
196                        server_port = each[ len("--port="): ]
197       
198       
199        mcp = puzzlebox_master_control(log, DEBUG)
200        reactor.listenTCP(port=server_port, factory=mcp, interface=server_host)
201        reactor.run()
202
Note: See TracBrowser for help on using the repository browser.