54 lines
1.1 KiB
Python
54 lines
1.1 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
'''
|
||
|
|
The demo server init script. Environment could be configured
|
||
|
|
via the MEMGRAPH_DEMO environtment variable. Available environments
|
||
|
|
are: debug, prod.
|
||
|
|
'''
|
||
|
|
|
||
|
|
import os
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from simulation.web_server import SimulationWebServer
|
||
|
|
|
||
|
|
|
||
|
|
def fetch_env(env_name, default=None):
|
||
|
|
'''
|
||
|
|
Fetches environment variable.
|
||
|
|
'''
|
||
|
|
if env_name in os.environ:
|
||
|
|
return os.environ[env_name]
|
||
|
|
|
||
|
|
return default
|
||
|
|
|
||
|
|
|
||
|
|
environment = fetch_env('MEMGRAPH_DEMO', 'debug')
|
||
|
|
wsgi = fetch_env('MEMGRAPH_DEMO_WSGI', 'werkzeug')
|
||
|
|
|
||
|
|
|
||
|
|
def _init():
|
||
|
|
'''
|
||
|
|
Initialzies logging level and server.
|
||
|
|
'''
|
||
|
|
if environment == 'prod':
|
||
|
|
logging.basicConfig(level=logging.WARNING)
|
||
|
|
elif environment == 'test':
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
else:
|
||
|
|
logging.basicConfig(level=logging.DEBUG)
|
||
|
|
|
||
|
|
return SimulationWebServer().server
|
||
|
|
|
||
|
|
|
||
|
|
app = _init()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
if wsgi == 'gevent':
|
||
|
|
from gevent.wsgi import WSGIServer
|
||
|
|
http_server = WSGIServer(('', 8080), app)
|
||
|
|
http_server.serve_forever()
|
||
|
|
else:
|
||
|
|
app.run(host="0.0.0.0", port=8080)
|