Source code for mdvtools.dbutils.safe_mdv_app
from flask import Flask
import os
from mdvtools.logging_config import get_logger
# Setup logging
[docs]
logger = get_logger(__name__)
logger.info("safe_mdv_app.py starting")
# Import the app at the module level (outside any conditionals)
try:
from mdvtools.dbutils.mdv_server_app import app
if app is None:
logger.error("app from mdv_server_app is None - using fallback")
except Exception as e:
logger.exception(f'Error importing mdv_app: {e}')
app = Flask(__name__)
# Keep the existing __main__ block for direct execution
if __name__ == '__main__':
"""
Running the app in local debugger with a path prefix
This can be usefull for testing behaviour app running on a sub-path
"""
# https://stackoverflow.com/a/36033627/279703
[docs]
class PrefixMiddleware(object):
def __init__(self, app, prefix=''):
[docs]
def __call__(self, environ, start_response):
if environ['PATH_INFO'].startswith(self.prefix):
environ['PATH_INFO'] = environ['PATH_INFO'][len(self.prefix):]
environ['SCRIPT_NAME'] = self.prefix
return self.app(environ, start_response)
else:
# It's good practice to ensure the Content-Length header is set.
# The Werkzeug BaseResponse object (which Flask uses) handles this automatically
# if the response is a string or a list of strings, but here we are returning a list of bytes.
response_body = b"This url does not belong to the app."
headers = [('Content-Type', 'text/plain'), ('Content-Length', str(len(response_body)))]
start_response('404 Not Found', headers)
return [response_body]
logger.info("running as __main__")
os.environ['MDV_API_ROOT'] = '/test/'
# Get the prefix from the environment variable
prefix = os.environ.get('MDV_API_ROOT', '')
# Remove trailing slash from prefix if it exists, as PATH_INFO usually starts with a slash
if prefix.endswith('/'):
prefix = prefix[:-1]
if prefix:
logger.info(f"Applying PrefixMiddleware with prefix: {prefix}")
app.wsgi_app = PrefixMiddleware(app.wsgi_app, prefix=prefix)
app.run(host='0.0.0.0', debug=False, port=5056)
