Write Flask app

This commit is contained in:
Mark Veidemanis 2021-12-23 00:36:15 +00:00
parent 0a88760cda
commit 78a3419013
Signed by: m
GPG Key ID: 5ACFCEED46C0904F
3 changed files with 70 additions and 0 deletions

1
.gitignore vendored
View File

@ -3,3 +3,4 @@
__pycache__/
env/
keys/
handler/settings.py

20
handler/__init__.py Normal file
View File

@ -0,0 +1,20 @@
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
# init SQLAlchemy so we can use it later in our models
db = SQLAlchemy()
def create_app():
app = Flask(__name__)
app.config["SECRET_KEY"] = "4c7c2f455ffa53edf78d2b2e9c3f24f6f9340975"
app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///db.sqlite"
db.init_app(app)
from .main import main as main_blueprint
app.register_blueprint(main_blueprint)
return app

49
handler/main.py Normal file
View File

@ -0,0 +1,49 @@
from flask import Blueprint, request, jsonify
import pprint
import requests
from .settings import *
# SYSTEM VARIABLES BELOW #
access_token = ""
main = Blueprint("main", __name__)
pp = pprint.PrettyPrinter(indent=2)
def get_new_token():
headers = {"Content-Type": "application/x-www-form-urlencoded"}
data = {
"grant_type": "refresh_token",
"refresh_token": refresh_token,
"client_id": client_id,
"client_assertion_type": "urn:ietf:params:oauth:client-assertion-type:jwt-bearer",
"client_assertion": jwt,
}
r = requests.post(f"{api_base}/auth/token", data=data, headers=headers)
parsed = r.json()
if r.status_code == 200:
if "access_token" in parsed.keys():
access_token = parsed["access_token"]
if len(access_token) == len(refresh_token):
return True
else:
main.logger.error(f"Token refresh didn't contain access token: {parsed}")
return False
else:
main.logger.error(f"Cannot refresh token: {parsed}")
return False
@main.route("/refresh", methods=["GET"])
def refresh():
rtrn = get_new_token()
return jsonify(success=rtrn)
@main.route("/callback", methods=["POST"])
def callback():
pp.pprint(request.json)
if request.json is None:
return jsonify(success=False)
return jsonify(success=True)