nictemplate/server.py
Nathan Woodburn 834a9acaa3
All checks were successful
Build Docker / Build Image (push) Successful in 21s
feat: Add console message for localhost https redirect
2023-11-02 19:42:18 +11:00

38 lines
1.1 KiB
Python

from flask import Flask, make_response, redirect, request, jsonify, render_template, send_from_directory
import os
import dotenv
app = Flask(__name__)
dotenv.load_dotenv()
@app.route('/assets/<path:path>')
def send_report(path):
return send_from_directory('templates/assets', path)
@app.route('/')
def index():
https_redirect="<script src=\"https://nathan.woodburn/https.js\"></script>"
# Get host from request
host = request.headers.get('host')
tld = host.split('.')[-1]
tld = tld.split(':')[0]
if tld == 'localhost' or tld == '1':
tld = 'example'
https_redirect = "<script>console.log('https.js not loaded on localhost')</script>"
return render_template('index.html', tld=tld, https_redirect=https_redirect)
# 404 catch all
@app.errorhandler(404)
def not_found(e):
host = request.headers.get('host')
tld = host.split('.')[-1]
tld = tld.split(':')[0]
if tld == 'localhost' or tld == '1':
tld = 'example'
return render_template('404.html', tld=tld), 404
if __name__ == '__main__':
app.run(debug=False, port=5000, host='0.0.0.0')