# -*- coding: utf-8 -*-
from django.contrib.auth.decorators import user_passes_test
from django.core.urlresolvers  import reverse

def license_required(function=None, redirect_field_name=None, login_url='/my/license/'):
    """
    Проверка активности лицензии
    """
    actual_decorator = user_passes_test(
        lambda u: u.profile.active,
        login_url=login_url,
        redirect_field_name=redirect_field_name
    )
    if function:
        return actual_decorator(function)
    return actual_decorator