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