"""
A simple Flask application for interacting with WHSPAH shockers.
"""

import argparse
import logging

import waitress.server
import flask

app = flask.Flask(__name__)

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-d', '--debug', action='store_true')
    parser.add_argument('--ip', type=str, default='0.0.0.0')
    parser.add_argument('--port', type=int, default=8000)
    args = parser.parse_args()

    if args.debug:
        logging.basicConfig(level=logging.DEBUG)
    else:
        logging.basicConfig(level=logging.INFO)

    if args.debug:
        app.run(host=args.ip, port=args.port, debug=True)
    else:
        server = waitress.server.create_server(app, host=args.ip, port=args.port)
        print(f'Serving at http://{args.ip}:{args.port}/')
        try:
            server.run()
        except KeyboardInterrupt:
            pass
        server.close()