mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-20 18:51:05 +01:00
26 lines
827 B
Python
26 lines
827 B
Python
from base64 import b64decode
|
|
from flask import jsonify, g, request
|
|
from functools import wraps
|
|
|
|
from lnbits.core.crud import get_wallet_for_key
|
|
|
|
|
|
def check_wallet(requires_admin=False):
|
|
def wrap(view):
|
|
@wraps(view)
|
|
def wrapped_view(**kwargs):
|
|
token = request.headers["Authorization"].split("Bearer ")[1]
|
|
key_type, key = b64decode(token).decode("utf-8").split(":")
|
|
|
|
if requires_admin and key_type != "admin":
|
|
return jsonify({"error": True, "code": 2, "message": "insufficient permissions"})
|
|
|
|
g.wallet = get_wallet_for_key(key, key_type)
|
|
if not g.wallet:
|
|
return jsonify({"error": True, "code": 2, "message": "insufficient permissions"})
|
|
return view(**kwargs)
|
|
|
|
return wrapped_view
|
|
|
|
return wrap
|