chore: optional basic auth refactored

requirements.txt

@@ -1,4 +1,5 @@
 ###### Requirements without Version Specifiers ######
-flask
-gevent
-
+flask~=3.0.3
+gevent~=24.2.1
+Werkzeug~=3.0.3
+Flask-HTTPAuth~=4.8.0

table-cell-auto-height-setter/table_cell_auto_height_setter.py

@@ -3,16 +3,31 @@
 import re
 
 from flask import Flask, Response, request, json, abort
+from flask_httpauth import HTTPBasicAuth
 from gevent.pywsgi import WSGIServer
+from werkzeug.security import generate_password_hash, check_password_hash
 
 app = Flask(__name__)
+auth = HTTPBasicAuth()
+users = {
+    "username1": generate_password_hash("userpass1"),
+    "username2": generate_password_hash("userpass2")
+}
+
+
+@auth.verify_password
+def verify_password(username, password):
+    if username in users and check_password_hash(users.get(username), password):
+        return username
+    return None
+
 
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s [%(levelname)s] - %(message)s')
 
 
 @app.route("/td-height/<height>", methods=["POST"])
+@auth.login_required
 def process_html(height):
-    authorize()
     export_params, html = validate_request()
 
     logging.info(f"Changing HTML table cell height to '{height}'")
@@ -31,15 +46,6 @@ def change_height(html, height):
     return re.sub(pattern, lambda m: m.group(1) + height, html)
 
 
-def authorize():
-    if args.username and args.password:
-        auth = request.authorization
-        if not auth:
-            abort(Response('No authorization data', 401))
-        if auth.username != args.username or auth.password != args.password:
-            abort(Response('Invalid username/password provided', 401))
-
-
 def validate_request():
     export_params_json = request.form.get('exportParams')
     if export_params_json is None: