From b7c1faff911c6943d379ce6ad3f44bba4027aedc Mon Sep 17 00:00:00 2001 From: Gregor Kikelj Date: Fri, 27 Jan 2023 16:08:18 +0100 Subject: Public cleanup --- server.py | 82 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 82 insertions(+) create mode 100644 server.py (limited to 'server.py') diff --git a/server.py b/server.py new file mode 100644 index 0000000..aebf98c --- /dev/null +++ b/server.py @@ -0,0 +1,82 @@ +from flask import Flask, jsonify, render_template, request, send_from_directory +import os +from helpers import aux +from discord_webhook import DiscordWebhook +import datetime +from flask_limiter import Limiter +from flask_limiter.util import get_remote_address +import logging + +# Logs to both console and app.log file +logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(name)s %(threadName)s : %(message)s") +formatter = logging.Formatter("%(asctime)s %(levelname)s %(name)s %(threadName)s : %(message)s") +logger = logging.getLogger('server') +handler = logging.FileHandler('app.log') +handler.setFormatter(formatter) +handler.setLevel(logging.INFO) +logger.addHandler(handler) + +def niceDateTime(): + time = datetime.datetime.now() + return time.strftime("%H:%M:%S -- %d/%m/%Y") +def niceTime(): + time = datetime.datetime.now() + return time.strftime("%H:%M:%S") + +def spamDiscord(logs = ""): + content = f"@everyone, napaka na strani ob {niceDateTime()}: \n {logs}" + webhook = DiscordWebhook(url='https://discord.com/api/webhooks/1062432166950219838/tgl04gFFIMkrASeWX2AlETdsgdTHU3CWufzE1NZQMLQ2eZGVeQyGXj4AAt0k4Dj467B'+'d', content=content) + webhook.execute() + + +def discordLog(msg): + DiscordWebhook(url="https://discord.com/api/webhooks/1062474351196262571/V6M_vfhPy9QEEbM-b1W7qxBr2k5olpyeOY1O2RoKZ178c0Fs9_vYzKqJwES3o3suLSF"+"o", content=msg).execute() + + +app = Flask(__name__, static_folder="static") +limiter = Limiter( + get_remote_address, + app=app, + default_limits=["50 per minute"], + storage_uri="memory://", +) + + +def setHeaders(request): + headers = request.form["headers"] + with open("headers.txt", "w") as f: + f.write(headers) + discordLog(f"Headers so bili spremenjeni ob {niceDateTime()}") + return "OK" + +@app.route('/favicon.ico') +@app.route('/robots.txt') +@app.route('/sitemap.xml') +def static_from_root(): + return send_from_directory(app.static_folder, request.path[1:]) + +@app.route('/', methods=['POST']) +def apcall(): + if "headers" in request.form: + setHeaders(request) + return "OK" + registrska = request.form["registrska"] + try: + ar = aux(registrska) + app.logger.info(f"Registrska \"{registrska}\" je veljavna do {ar} ob {niceTime()}") + # discordLog(f"Registrska \"{registrska}\" je veljavna do {ar} ob {niceTime()}") + return render_template('index.html', valid_until = ar, license_plate = registrska) + except Exception as e: + spamDiscord(str(e)) + app.logger.error(f"\nNapaka pri {registrska} ob {niceTime()}") + # discordLog(f"!!!!!!!!!!!!!!!!!!!!!!!!!!!\nNapaka pri {registrska} ob {niceTime()}") + return render_template('failure.html') + +@app.route('/', methods=['GET']) +def index(): + app.logger.info("Zahteva za domačo stran") + # discordLog(f"Zahteva za domačo stran ob {niceTime()}") + return render_template('index.html') + +if __name__ == '__main__': + app.run() \ No newline at end of file -- cgit v1.2.3