Add enviroment variables

This commit is contained in:
Michał Gdula 2023-05-04 13:11:08 +03:00
parent dbd6a81c78
commit fda3fcc28c
5 changed files with 17 additions and 9 deletions

View file

@ -1,4 +1,7 @@
SECRET_KEY = 'dev'
import os
SECRET_KEY = os.getenv('SECRET_KEY')
BEARER_TOKEN = os.getenv('BEARER_TOKEN')
SQLALCHEMY_DATABASE_URI = 'sqlite:///db.sqlite'
SQLALCHEMY_TRACK_MODIFICATIONS = False

View file

@ -4,4 +4,4 @@ from flask_caching import Cache
db = SQLAlchemy()
migrate = Migrate()
cache = Cache()
cache = Cache(config={'CACHE_TYPE': 'simple'})

View file

@ -5,6 +5,8 @@ from wtforms.validators import DataRequired
from models import Scores, Users
from extensions import db, cache
from config import BEARER_TOKEN
blueprint = Blueprint('views', __name__)
@ -63,7 +65,7 @@ def post():
if not form:
return "Invalid form", 400
if request.headers.get('Authentication') != 'Bearer 1234':
if request.headers.get('Authentication') != 'Bearer ' + BEARER_TOKEN:
return "Invalid authentication", 401
if not isinstance(form.score.data, int):
@ -91,5 +93,3 @@ def post():
db.session.add(score)
db.session.commit()
return jsonify({'message': 'Success!'})