Files
iconograph/server/https_server.py
2016-05-03 14:09:45 -07:00

77 lines
1.5 KiB
Python
Executable File

#!/usr/bin/python3
import argparse
from gevent import pywsgi
import ssl
parser = argparse.ArgumentParser(description='iconograph https_server')
parser.add_argument(
'--ca-cert',
dest='ca_cert',
action='store',
required=True)
parser.add_argument(
'--listen-host',
dest='listen_host',
action='store',
default='::')
parser.add_argument(
'--listen-port',
dest='listen_port',
type=int,
action='store',
default=443)
parser.add_argument(
'--server-key',
dest='server_key',
action='store',
required=True)
parser.add_argument(
'--server-cert',
dest='server_cert',
action='store',
required=True)
FLAGS = parser.parse_args()
class ImageRequestHandler(object):
def __call__(self, env, start_response):
print(env['PATH_INFO'])
start_response('200 OK', [('Content-Type', 'text/plain')])
return [b'foo']
class ImageServer(object):
def __init__(self, listen_host, listen_port, server_key, server_cert, ca_cert):
self._handler = ImageRequestHandler()
self._httpd = pywsgi.WSGIServer(
(listen_host, listen_port),
self._handler,
keyfile=server_key,
certfile=server_cert,
ca_certs=ca_cert,
cert_reqs=ssl.CERT_REQUIRED,
ssl_version=ssl.PROTOCOL_TLSv1_2)
def Serve(self):
self._httpd.serve_forever()
def main():
server = ImageServer(
FLAGS.listen_host,
FLAGS.listen_port,
FLAGS.server_key,
FLAGS.server_cert,
FLAGS.ca_cert)
server.Serve()
if __name__ == '__main__':
main()