def get_email(request): try: token = request.META['HTTP_TOKEN'] jwt_string = jwt.decode(token, get_secret_key(), algorithms=['HS256']) email = jwt_string["email"] return email except: return INVALID_TOKEN
def create_token(email): token = jwt.encode({ 'email': email, 'date': str(date.today()) }, get_secret_key(), algorithm='HS256') token = token.decode() return token
def setUp(self): self.email = "*****@*****.**" self.token = create_token(self.email) self.expired_token = jwt.encode( { 'email': self.email, 'date': '2011-01-01' }, get_secret_key(), algorithm='HS256').decode()
def check_token(token): try: jwt_string = jwt.decode(token, get_secret_key(), algorithms=['HS256']) # email = jwt_string["email"] date_login = jwt_string["date"] date_login = datetime.strptime(date_login, '%Y-%m-%d').date() diff = calcuate_day_difference(date_login) if diff <= TOKEN_EXPIRY_DAYS: return VALID_TOKEN else: return TOKEN_EXPIRED except: return INVALID_TOKEN
def test_secret_key(self): self.assertEqual(self.secret_key, get_secret_key())
def setUp(self): self.secret_key = get_secret_key()