74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
from flask import Flask, make_response, redirect, render_template_string, request, jsonify, render_template, send_from_directory
|
|
import os
|
|
import dotenv
|
|
import requests
|
|
import json
|
|
import schedule
|
|
import time
|
|
import db
|
|
import website
|
|
|
|
|
|
app = Flask(__name__)
|
|
dotenv.load_dotenv()
|
|
|
|
main_domain = "cities.hnshosting.au"
|
|
if os.getenv('MAIN_DOMAIN') != None:
|
|
main_domain = os.getenv('MAIN_DOMAIN')
|
|
|
|
#Assets routes
|
|
@app.route('/assets/<path:path>')
|
|
def assets(path):
|
|
return send_from_directory('templates/assets', path)
|
|
|
|
@app.route('/')
|
|
def index():
|
|
host = request.host
|
|
if len(host.split('.')) != 2:
|
|
return redirect('https://'+main_domain)
|
|
host = host.split('.')[0]
|
|
|
|
# Get website data
|
|
data = db.get_website_data(host)
|
|
db_object = db.get_website_data_raw(host)
|
|
# Render as HTML
|
|
return website.render(data,db_object)
|
|
|
|
|
|
@app.route('/.well-known/wallets/<token>')
|
|
def wallet(token):
|
|
address = db.get_website_wallet(request.host.split('.')[0],token)
|
|
if address == "":
|
|
return redirect('/')
|
|
# Plain text
|
|
response = make_response(address)
|
|
response.mimetype = "text/plain"
|
|
return response
|
|
|
|
@app.route('/<path:path>')
|
|
def catch_all(path):
|
|
return redirect('/') # 404 catch all
|
|
|
|
# 404 catch all
|
|
@app.errorhandler(404)
|
|
def not_found(e):
|
|
return redirect('/')
|
|
|
|
def clean_templates():
|
|
# Clean template
|
|
for file in os.listdir('templates'):
|
|
if file.endswith('.html'):
|
|
with open('templates/' + file) as f:
|
|
data = f.read()
|
|
|
|
data = data.replace('#f1ffff', '{{fg_colour}}')
|
|
data = data.replace('#1fffff', '{{text_colour}}')
|
|
data = data.replace('#000000', '{{bg_colour}}')
|
|
# Save
|
|
with open('templates/' + file, 'w') as f:
|
|
f.write(data)
|
|
print("Cleaned " + file, flush=True)
|
|
print("Cleaned template", flush=True)
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=False, port=5000, host='0.0.0.0') |