source: remote_control/puzzlebox_brainstorms_network_server.py @ 107

Last change on this file since 107 was 107, checked in by sc, 11 years ago
  • support for user-configurable NXT bluetooth device port in interface
  • window icon for interface manually set
  • Property svn:executable set to *
File size: 6.7 KB
Line 
1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3#
4# Puzzlebox - Brainstorms - Network - Server
5#
6# Copyright Puzzlebox Productions, LLC (2010)
7#
8# This code is released under the GNU Pulic License (GPL) version 2
9# For more information please refer to http://www.gnu.org/copyleft/gpl.html
10#
11# Last Update: 2010.07.10
12#
13#####################################################################
14
15import os, sys
16import signal
17
18import simplejson as json
19
20from PyQt4 import QtCore, QtNetwork
21
22import puzzlebox_brainstorms_configuration as configuration
23import puzzlebox_brainstorms_remote_control as remote_control
24#import puzzlebox_logger
25
26#####################################################################
27# Globals
28#####################################################################
29
30DEBUG = 1
31
32SERVER_INTERFACE = configuration.BRAINSTORMS_SERVER_INTERFACE
33SERVER_PORT = configuration.BRAINSTORMS_SERVER_PORT
34
35CLIENT_NO_REPLY_WAIT = configuration.CLIENT_NO_REPLY_WAIT * 1000
36
37DELIMITER = configuration.BRAINSTORMS_DELIMITER
38
39BLUETOOTH_DEVICE = configuration.NXT_BLUETOOTH_DEVICE
40
41VARIABLE_CONTROL_DURATION = configuration.BRAINSTORMS_VARIABLE_CONTROL_DURATION
42
43#####################################################################
44# Classes
45#####################################################################
46
47class puzzlebox_brainstorms_network_server:
48       
49        def __init__(self, log, \
50                          server_interface=SERVER_INTERFACE, \
51                          server_port=SERVER_PORT, \
52                          embedded_mode=False, \
53                          DEBUG=DEBUG, \
54                          parent=None):
55               
56                self.log = log
57                self.DEBUG = DEBUG
58               
59                self.server_interface = server_interface
60                self.server_port = server_port
61               
62                self.embedded_mode=embedded_mode
63               
64                self.configureNetwork()
65               
66                self.rc = None
67               
68                #self.rc = remote_control.puzzlebox_brainstorms_remote_control( \
69                            #device=BLUETOOTH_DEVICE, \
70                            #DEBUG=DEBUG)
71       
72       
73        ##################################################################
74       
75        def configureNetwork(self):
76       
77                #self.blockSize = 0
78                self.socket = QtNetwork.QTcpServer()
79                self.socket.name = 'Brainstorms Server'
80               
81                if self.DEBUG:
82                        print "<---- [%s] Initializing server on %s:%i" % \
83                                (self.socket.name, self.server_interface, self.server_port)
84       
85                if (self.server_interface == ''):
86                        result = self.socket.listen(port=self.server_port)
87                else:
88                        result = self.socket.listen(address=self.server_interface, \
89                                           port=self.server_port)
90               
91                if not result:
92                        if self.DEBUG:
93                                print "ERROR [%s] Unable to start the server:", self.socket.name,
94                                print self.socket.errorString()
95                               
96                        self.socket.close()
97                        return
98               
99               
100                self.socket.newConnection.connect(self.processConnection)
101       
102       
103        ##################################################################
104       
105        def processData(self, clientConnection=None):
106               
107                if (clientConnection == None):
108                        clientConnection = self.clientConnection
109               
110                socket_buffer = clientConnection.readAll()
111               
112                for packet in socket_buffer.split(DELIMITER):
113                       
114                        if packet != '':
115                               
116                                try:
117                                        data = json.loads(packet.data())
118                                except:
119                                        data = packet
120                               
121                                if self.DEBUG:
122                                        print "--> [%s] Received:" % self.socket.name,
123                                        print data
124                               
125                                response = self.processCommand(data)
126                               
127                                if response != None:
128                               
129                                        data = json.dumps(response)
130                                       
131                                       
132                                        if clientConnection.waitForConnected(CLIENT_NO_REPLY_WAIT):
133                                                clientConnection.write(data)
134       
135       
136        ##################################################################
137       
138        def processConnection(self):
139               
140                if self.DEBUG:
141                        print "--> [%s]: Client connected" % self.socket.name
142               
143                clientConnection = self.socket.nextPendingConnection()
144                clientConnection.disconnected.connect(clientConnection.deleteLater)
145               
146                if (self.embedded_mode):
147                       
148                        # Only one client connection to server will work in embedded mode
149                        self.clientConnection = clientConnection
150                        clientConnection.readyRead.connect(self.processData)
151               
152               
153                else:
154               
155                        if not clientConnection.waitForReadyRead(CLIENT_NO_REPLY_WAIT):
156                                if self.DEBUG:
157                                        print "WARNING [%s] Timeout waiting for client to transmit data" % \
158                                                self.socket.name
159                                        #print "State:",
160                                        #print clientConnection.state()
161                                clientConnection.disconnectFromHost()
162                                return
163                       
164                       
165                        self.processData(clientConnection)
166                       
167                        clientConnection.disconnectFromHost()
168       
169       
170        ##################################################################
171       
172        def processCommand(self, data):
173               
174                response = {}
175               
176                if not VARIABLE_CONTROL_DURATION:
177                       
178                        response['status'] = self.executeCommand(data['command'], data['power'])
179                       
180                       
181                else:
182                       
183                        if (self.rc == None):
184                               
185                                if ('bluetooth_device' in data.keys()):
186                                        bluetooth_device = data['bluetooth_device']
187                                else:
188                                        bluetooth_device == BLUETOOTH_DEVICE
189                               
190                                self.rc = remote_control.puzzlebox_brainstorms_remote_control( \
191                                             device=bluetooth_device, \
192                                             DEBUG=DEBUG)
193                               
194                                if self.rc.connection != None:
195                                       
196                                        self.rc.run(data['command'], data['power'])
197                                        response['status'] = self.rc.get_battery_voltage(self.rc.connection)
198                               
199                                else:
200                                       
201                                        self.rc = None
202                                        response = {}
203                                        response['status'] = 'LEGO NXT N/A'
204               
205               
206                return(response)
207       
208       
209        ##################################################################
210       
211        def executeCommand(self, command):
212               
213                #command_line = 'python puzzlebox_brainstorms_remote_control.py --command=%s' % command
214               
215                #os.system(command_line)
216               
217                status = 'N/A'
218               
219                rc = remote_control.puzzlebox_brainstorms_remote_control( \
220                        device=BLUETOOTH_DEVICE, \
221                        command=command, \
222                        DEBUG=DEBUG)
223               
224                if rc.connection != None:
225                        rc.run(rc.command)
226                        rc.stop()
227                        status = rc.get_battery_voltage(rc.connection)
228               
229               
230                return(status)
231
232
233#####################################################################
234# Main
235#####################################################################
236
237if __name__ == '__main__':
238       
239        # Perform correct KeyboardInterrupt handling
240        signal.signal(signal.SIGINT, signal.SIG_DFL)
241       
242        #log = puzzlebox_logger.puzzlebox_logger(logfile='server_brainstorms')
243        log = None
244       
245        # Collect default settings and command line parameters
246        server_interface = SERVER_INTERFACE
247        server_port = SERVER_PORT
248       
249        for each in sys.argv:
250               
251                if each.startswith("--interface="):
252                        server_interface = each[ len("--interface="): ]
253                if each.startswith("--port="):
254                        server_port = each[ len("--port="): ]
255       
256       
257        app = QtCore.QCoreApplication(sys.argv)
258       
259        server = puzzlebox_brainstorms_network_server(log, \
260                                                      server_interface, \
261                                                      server_port, \
262                                                      embedded_mode=False, \
263                                                      DEBUG=DEBUG)
264       
265        sys.exit(app.exec_())
266
Note: See TracBrowser for help on using the repository browser.