from django_cas.decorators import user_passes_test

def is_cof(user):
    try:
        profile = user.get_profile()
        return profile.is_cof
    except:
        return False

def cof_required(login_url = None):
    return user_passes_test(lambda u: is_cof(u), login_url=login_url)

def is_buro(user):
    try:
        profile = user.get_profile()
        return profile.is_buro
    except:
        return False

def buro_required(login_url = None):
    return user_passes_test(lambda u: is_buro(u), login_url=login_url)