Brad Bishop | 7bc6d8d | 2016-08-29 22:19:51 -0400 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | # Contributors Listed Below - COPYRIGHT 2016 |
| 4 | # [+] International Business Machines Corp. |
| 5 | # |
| 6 | # |
| 7 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 8 | # you may not use this file except in compliance with the License. |
| 9 | # You may obtain a copy of the License at |
| 10 | # |
| 11 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 12 | # |
| 13 | # Unless required by applicable law or agreed to in writing, software |
| 14 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 15 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or |
| 16 | # implied. See the License for the specific language governing |
| 17 | # permissions and limitations under the License. |
| 18 | |
| 19 | |
| 20 | import sys |
| 21 | import os |
| 22 | import gevent |
| 23 | from gevent.pywsgi import WSGIServer |
Deepak Kodihalli | 0fe213f | 2017-10-11 00:08:48 -0500 | [diff] [blame^] | 24 | have_wsock = True |
| 25 | try: |
| 26 | from geventwebsocket.handler import WebSocketHandler |
| 27 | except ImportError: |
| 28 | have_wsock = False |
Brad Bishop | 7bc6d8d | 2016-08-29 22:19:51 -0400 | [diff] [blame] | 29 | |
| 30 | if __name__ == '__main__': |
| 31 | if len(sys.argv) < 2: |
| 32 | sys.stderr.write('WSGI application required!') |
| 33 | sys.exit(1) |
| 34 | |
| 35 | exec 'from obmc.wsgi.apps.%s import App' % sys.argv[1] |
| 36 | |
| 37 | default_cert = os.path.join( |
| 38 | sys.prefix, 'share', os.path.basename(__file__), 'cert.pem') |
| 39 | |
Deepak Kodihalli | 0fe213f | 2017-10-11 00:08:48 -0500 | [diff] [blame^] | 40 | kw = {} |
| 41 | if have_wsock: |
| 42 | kw['have_wsock'] = True |
| 43 | app = App(**kw) |
Brad Bishop | 7bc6d8d | 2016-08-29 22:19:51 -0400 | [diff] [blame] | 44 | |
| 45 | if os.environ.get('LISTEN_PID', None) == str(os.getpid()): |
| 46 | FIRST_SYSTEMD_SOCKET_FD = 3 |
| 47 | bind = gevent.socket.fromfd(FIRST_SYSTEMD_SOCKET_FD, |
| 48 | gevent.socket.AF_INET, |
| 49 | gevent.socket.SOCK_STREAM) |
| 50 | else: |
| 51 | bind = ('', 443) |
| 52 | |
Deepak Kodihalli | 0fe213f | 2017-10-11 00:08:48 -0500 | [diff] [blame^] | 53 | kw = {} |
| 54 | if have_wsock: |
| 55 | kw['handler_class'] = WebSocketHandler |
Brad Bishop | 7bc6d8d | 2016-08-29 22:19:51 -0400 | [diff] [blame] | 56 | server = WSGIServer( |
Deepak Kodihalli | 0fe213f | 2017-10-11 00:08:48 -0500 | [diff] [blame^] | 57 | bind, app, keyfile=default_cert, certfile=default_cert, **kw) |
Brad Bishop | 7bc6d8d | 2016-08-29 22:19:51 -0400 | [diff] [blame] | 58 | server.serve_forever() |