src/yarf/authentication/base_auth.py

   1 # Copyright 2019 Nokia
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14 #
  15
  16 from flask import abort, request
  17 from functools import wraps
  18
  19 def login_required(func):
  20     @wraps(func)
  21     def wrapper(*args, **kwargs):
  22         auth_method = func.__self__.authentication_method
  23         if auth_method is None:
  24             return func(*args, **kwargs)
  25
  26         if isinstance(auth_method, BaseAuthMethod) and auth_method.get_authentication(request)[0]:
  27             return func(*args, **kwargs)
  28         else:
  29             abort(401)
  30             return None
  31     return wrapper
  32
  33 class BaseAuthMethod(object):
  34     def __init__(self):
  35         pass
  36     def get_authentication(self, req):
  37         raise NotImplementedError("Function get_authentication not implemented")