utils.py

import os.path

from functools import wraps
from flask import request, Response



def check_auth(username, password):
    """This function is called to check if a username /
    password combination is valid.
    """
    with open(os.path.join(os.path.dirname(__file__), "users.txt")) as f:
        for l in f:
            l = l.strip()
            if l == ("%s:%s" % (username, password)):
                return True
    return False

def authenticate():
    """Sends a 401 response that enables basic auth"""
    return Response(
    'Could not verify your access level for that URL.\n'
    'You have to login with proper credentials', 401,
    {'WWW-Authenticate': 'Basic realm="Login Required"'})

def requires_auth(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        auth = request.authorization
        if not auth or not check_auth(auth.username, auth.password):
            return authenticate()
        return f(*args, **kwargs)
    return decorated

def socket_requires_auth(f):
    @wraps(f)
    def wrapped(*args, **kwargs):
        auth = request.authorization
        if not auth or not check_auth(auth.username, auth.password):
            disconnect()
        else:
            return f(*args, **kwargs)
    return wrapped