diff --git a/digits/__main__.py b/digits/__main__.py index 0bbfac5bf..c0c7fbf62 100644 --- a/digits/__main__.py +++ b/digits/__main__.py @@ -18,6 +18,12 @@ def main(): parser = argparse.ArgumentParser(description='DIGITS server') + parser.add_argument( + '-H', '--host', + type=str, + default='0.0.0.0', + help='Host to run app on (default 0.0.0.0)' + ) parser.add_argument( '-p', '--port', type=int, @@ -59,7 +65,7 @@ def main(): print 'ERROR: Scheduler would not start' else: digits.webapp.app.debug = args['debug'] - digits.webapp.socketio.run(digits.webapp.app, '0.0.0.0', args['port']) + digits.webapp.socketio.run(digits.webapp.app, args['host'], args['port']) except KeyboardInterrupt: pass finally: