1# Run a WSGI application in a daemon thread 2 3import bottle 4import threading 5import os.path 6 7from . import util 8 9global_stop = False 10 11class Server(bottle.WSGIRefServer): 12 def run(self, handler): # pragma: no cover 13 self.srv = self.make_server(handler) 14 self.serve() 15 16 def make_server(self, handler): 17 from wsgiref.simple_server import make_server, WSGIRequestHandler 18 if self.quiet: 19 base = self.options.get('handler_class', WSGIRequestHandler) 20 class QuietHandler(base): 21 def log_request(*args, **kw): 22 pass 23 self.options['handler_class'] = QuietHandler 24 srv = make_server(self.host, self.port, handler, **self.options) 25 return srv 26 27 def serve(self): 28 self.srv.serve_forever(poll_interval=0.1) 29 30# http://www.socouldanyone.com/2014/01/bottle-with-ssl.html 31# https://github.com/mfm24/miscpython/blob/master/bottle_ssl.py 32class SslServer(Server): 33 def run(self, handler): # pragma: no cover 34 self.srv = self.make_server(handler) 35 36 import ssl 37 cert_dir = os.path.join(os.path.dirname(__file__), 'certs') 38 self.srv.socket = ssl.wrap_socket( 39 self.srv.socket, 40 keyfile=os.path.join(cert_dir, 'server.key'), 41 certfile=os.path.join(cert_dir, 'server.crt'), 42 server_side=True) 43 44 self.serve() 45 46def start_bottle_server(app, port, server, **kwargs): 47 server_thread = ServerThread(app, port, server, kwargs) 48 server_thread.daemon = True 49 server_thread.start() 50 51 ok = util.wait_for_network_service(('127.0.0.1', port), 0.1, 10) 52 if not ok: 53 import warnings 54 warnings.warn('Server did not start after 1 second') 55 56 return server_thread.server 57 58class ServerThread(threading.Thread): 59 def __init__(self, app, port, server, server_kwargs): 60 threading.Thread.__init__(self) 61 self.app = app 62 self.port = port 63 self.server_kwargs = server_kwargs 64 self.server = server(host='127.0.0.1', port=self.port, **self.server_kwargs) 65 66 def run(self): 67 bottle.run(self.app, server=self.server, quiet=True) 68 69started_servers = {} 70 71def app_runner_setup(*specs): 72 '''Returns setup and teardown methods for running a list of WSGI 73 applications in a daemon thread. 74 75 Each argument is an (app, port) pair. 76 77 Return value is a (setup, teardown) function pair. 78 79 The setup and teardown functions expect to be called with an argument 80 on which server state will be stored. 81 82 Example usage with nose: 83 84 >>> setup_module, teardown_module = \ 85 runwsgi.app_runner_setup((app_module.app, 8050)) 86 ''' 87 88 def setup(self): 89 self.servers = [] 90 for spec in specs: 91 if len(spec) == 2: 92 app, port = spec 93 kwargs = {} 94 else: 95 app, port, kwargs = spec 96 if port in started_servers: 97 assert started_servers[port] == (app, kwargs) 98 else: 99 server = Server 100 if 'server' in kwargs: 101 server = kwargs['server'] 102 del kwargs['server'] 103 elif 'ssl' in kwargs: 104 if kwargs['ssl']: 105 server = SslServer 106 del kwargs['ssl'] 107 self.servers.append(start_bottle_server(app, port, server, **kwargs)) 108 started_servers[port] = (app, kwargs) 109 110 def teardown(self): 111 return 112 for server in self.servers: 113 # if no tests from module were run, there is no server to shut down 114 if hasattr(server, 'srv'): 115 server.srv.shutdown() 116 117 return [setup, teardown] 118