Revert static file fix
This commit is contained in:
+1
-39
@@ -7,34 +7,9 @@ from logging.handlers import RotatingFileHandler
|
||||
from flask import render_template
|
||||
from flask_migrate import upgrade
|
||||
import awsgi
|
||||
import base64
|
||||
import mimetypes
|
||||
|
||||
from flask import send_from_directory, request, make_response
|
||||
app = create_app()
|
||||
|
||||
# Patch static file serving to base64-encode for Lambda/awsgi
|
||||
@app.route('/static/<path:filename>')
|
||||
def static_files(filename):
|
||||
static_folder = os.path.join(app.root_path, 'static')
|
||||
file_path = os.path.join(static_folder, filename)
|
||||
content_type, _ = mimetypes.guess_type(file_path)
|
||||
if not content_type:
|
||||
content_type = "application/octet-stream"
|
||||
binary_types = ["application/octet-stream", "image/", "application/pdf", "font/"]
|
||||
if any(content_type.startswith(b) for b in binary_types):
|
||||
# Read file as bytes and base64 encode
|
||||
with open(file_path, "rb") as f:
|
||||
data = f.read()
|
||||
encoded = base64.b64encode(data).decode("utf-8")
|
||||
response = make_response(encoded)
|
||||
response.headers["Content-Type"] = content_type
|
||||
response.headers["Content-Transfer-Encoding"] = "base64"
|
||||
response.headers["X-Is-Binary"] = "1"
|
||||
return response
|
||||
# For non-binary, use normal send_from_directory
|
||||
return send_from_directory(static_folder, filename)
|
||||
|
||||
@app.cli.command("init-db")
|
||||
def init_db_command():
|
||||
"""Initialize the database."""
|
||||
@@ -69,20 +44,7 @@ if __name__ == '__main__':
|
||||
app.run(host='0.0.0.0', port=5004)
|
||||
|
||||
def lambda_handler(event, context):
|
||||
response = awsgi.response(app, event, context)
|
||||
headers = response.get("headers", {})
|
||||
# Only base64-encode if our custom header is set (for /static)
|
||||
if headers.get("X-Is-Binary") == "1":
|
||||
if not response.get("isBase64Encoded"):
|
||||
body = response["body"]
|
||||
if isinstance(body, str):
|
||||
body = body.encode("latin1")
|
||||
import base64
|
||||
response["body"] = base64.b64encode(body).decode("utf-8")
|
||||
response["isBase64Encoded"] = True
|
||||
# Remove the custom header from the response
|
||||
response["headers"].pop("X-Is-Binary", None)
|
||||
return response
|
||||
return awsgi.response(app, event, context)
|
||||
|
||||
def lambda_handler_upgrade_db(event, context):
|
||||
with app.app_context():
|
||||
|
||||
Reference in New Issue
Block a user