mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-18 08:29:14 +00:00
Add basic API auth
This commit is contained in:
parent
767c504a1d
commit
6c3609f590
|
@ -1,6 +1,7 @@
|
|||
#!/usr/bin/env python3
|
||||
from flask import Flask
|
||||
from flask import Flask, request, jsonify
|
||||
from flask_restful import Api
|
||||
import os
|
||||
|
||||
from selfprivacy_api.resources.users import Users
|
||||
from selfprivacy_api.resources.common import DecryptDisk
|
||||
|
@ -10,6 +11,19 @@ def create_app():
|
|||
app = Flask(__name__)
|
||||
api = Api(app)
|
||||
|
||||
app.config['AUTH_TOKEN'] = os.environ.get('AUTH_TOKEN')
|
||||
|
||||
# Check bearer token
|
||||
@app.before_request
|
||||
def check_auth():
|
||||
auth = request.headers.get("Authorization")
|
||||
if auth is None:
|
||||
return jsonify({"error": "Missing Authorization header"}), 401
|
||||
|
||||
# Check if token is valid
|
||||
if auth != "Bearer " + app.config['AUTH_TOKEN']:
|
||||
return jsonify({"error": "Invalid token"}), 401
|
||||
|
||||
api.add_resource(Users, "/users")
|
||||
api.add_resource(DecryptDisk, "/decryptDisk")
|
||||
from selfprivacy_api.resources.system import api_system
|
||||
|
|
Loading…
Reference in a new issue