记录一个JWT验证装饰器

最新推荐文章于 2024-09-05 22:53:42 发布

南瓜头777

最新推荐文章于 2024-09-05 22:53:42 发布

阅读量1.9k

点赞数

分类专栏： python 文章标签：安全 python django flask jwt

本文链接：https://blog.youkuaiyun.com/weixin_44840774/article/details/122456295

版权

python 专栏收录该内容

18 篇文章

订阅专栏

本文记录了如何创建一个JWT验证装饰器，用于接口的安全验证。内容涉及Python、Django、Flask和JWT技术，详细阐述了装饰器的编写过程及模拟验证的步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

记录一个JWT验证装饰器，可用于做接口验证。
用到的包：

import time
import jwt
import uuid
import functools

用到的加密解密方法：

class Token:
    @classmethod
    def encrpyt_token(cls, username, exp_time=None, secret=None):
        """
        jwt加密
        :param username:
        :param exp_time:
        :param secret:
        :return:
        """
        if not exp_time:
            exp_time = time.time() + 60 * 30

        if not secret:
            secret = 'ngt777'

        payloads = {
            'iss': 'ngt777 Is The JWT\'s Builder',  # 该JWT的签发者，是否使用是可以选择的。
            # 'sub': '',  # sub：该JWT所面向的用户，是否使用是可选的。
            # 'aud': '',  # aud：接收该JWT的一方，是否使用是可选的。
            'iat': time.time(),  # iat（issued at）：在什么时候签发UNIX时间，是否使用是可选的。
            'username': username,
            'uuid': str(uuid.uuid1()),
            "exp": exp_time,  # exp（expires）：什么时候过期，是一个UNIX的时间戳，是否使用是可选的。默认设置为：30分钟
            # "nbf": ""  # nbf（not before）：如果当前时间在nbf的时间之前，则Token不被接受，一般都会留几分钟，是否使用是可选的。
        }
        encoded_jwt = jwt.encode(payloads, secret, algorithm='HS256')
        return encoded_jwt

    @classmethod
    def decrypt_token(cls, token, secret=None):
        """
        jwt解密
        :param token:
        :param secret:
        :return:
        """
        try:
            if not secret:
                secret = 'ngt777'
            decode_jwt = jwt.decode(token, secret, algorithms=['HS256'])

            return decode_jwt
        except jwt.exceptions.InvalidSignatureError as e:
            return {'error': 'Signature verification failed'}

编写装饰器：

def access_token_required():
    def decorator(f):
    	@functools.wraps(f)
    	def wrapper(*args, **kwargs):
	        token = Token().encrpyt_token('martin')
	        decryptToken = Token().decrypt_token(token)
	        if token and decryptToken.get('username') == 'martin':
	            print(222)
	            return f(*args, **kwargs)
	        else:
	            raise {'error': 'X-Access-Token verification failed!'}
	    return wrapper
    return decorator

模拟验证：

@access_token_required()
def martin():
    print(111)

输出：

222
111