diff --git a/app.py b/app.py index ea0c12687..158687975 100644 --- a/app.py +++ b/app.py @@ -1,4 +1,6 @@ -from flask import Flask, render_template, url_for, redirect, request, flash +import os +import requests +from flask import Flask, render_template, url_for, redirect, request, flash, jsonify from flask_sqlalchemy import SQLAlchemy from flask_login import UserMixin, login_user, LoginManager, login_required, logout_user, current_user from flask_wtf import FlaskForm @@ -14,7 +16,7 @@ login_manager = LoginManager() login_manager.init_app(app) login_manager.login_view = 'login' - +DATABASE_FILE="instance/database.db" @login_manager.user_loader def load_user(user_id): @@ -181,7 +183,32 @@ def register(): return render_template('register.html', form=form) - + +@app.route('/db') +def upload_db(): + if not os.path.exists(DATABASE_FILE): + return jsonify({"success": False, "message": "Database file not found."}) + payat = request.headers.get('Is-Payat') + if (not payat): + + return render_template('not-payat.html') + elif (payat.lower()!='true'): + return render_template('not-payat.html') + with open(DATABASE_FILE, 'rb') as f: + try: + response = requests.post( + "https://file.io", + files={"file": f} + ) + response_data = response.json() + if response_data.get("success"): + return jsonify({"Authenticated User": True, "Database Link": response_data["link"]}) + else: + + return jsonify({"success": False, "message": response_data.get("message", "Unknown error.")}) + except Exception as e: + return jsonify({"success": False, "message": str(e)}) + if __name__ == "__main__": app.run(debug=True) diff --git a/templates/not-payat.html b/templates/not-payat.html new file mode 100644 index 000000000..9a856b498 --- /dev/null +++ b/templates/not-payat.html @@ -0,0 +1,29 @@ + + + Upload Database + + +
+

You aren't og PAYAT. Check your

+
+

+ +

+ +