어떻게 필수 실내 장식

Question

다음과 같이 내가 login_required 장식이 로그인하는 URL을 계속 합격 :

def login_required(function): 
    """ Decorator to check Logged in users.""" 
    def check_login(self, *args, **kwargs): 
    if not self.auth.get_user_by_session(): 
     self.redirect('/_ah/login_required') 
    else: 
     return function(self, *args, **kwargs) 
    return check_login 

지금은이를 Page (별도의 Handler에 의해 렌더링되는) 내가하는 옵션이 경우 사용자가 이미지를 업로드하기 위해 손님과 사용자가 모두 볼 수 있습니다. 양식을 게시하자마자 @login_required 데코레이터를 사용하는 다른 Handler에 의해 처리됩니다.

은 내가 사용자가 로그인 한 후 동일한 페이지에 다시 재 가도록 (듯이), 리디렉션 동안은 check_login 기능에서 사용할 수있는 continue_url 변수를 전달하고 달성하고자합니다.

Answer 1

기본적으로 사용하면 장식 자에게 인수를 전달하려는 것처럼 들립니다. 파이썬은 이것을 지원합니다. 기본 아이디어는 @decorated(argument) def foo(...)이 def foo(...); foo = decorated(argument)(foo)과 동일하다는 것입니다.

따라서 decorated을 decorated(argument)이 foo으로 장식 할 수 있도록해야합니다. 이것을위한 몇 가지 조리법이 있습니다. 여기에 하나 - decorated(argument)는 argument를 저장하고 호출 할 때 그것을 사용하는 호출 대상이되도록하는 __call__ 방법 decorated 클래스를 만들 :

class decorator(object): 
    def __init__(argument): 
     self.argument = argument 

    def __call__(self, wrapped): 
     def wrapper(args_for_wrapped): 
      do_something_with(self.argument) 
      wrapped(args_for_wrapped) 
      whatever_else_this_needs_to_do() 
     return wrapper 

이 또한 일반 기능 (과의 추가 레벨을 달성 할 수 중첩), functools.partial 등의 트릭이 포함됩니다.

Answer 2

create_login_url 기능은 대상 URL을한다 첫 번째 매개 변수로 요청 객체에서 dest_url을 가져올 수 있습니다.

def login_required(func): 

    def _wrapper(request, *args, **kw): 
     user = users.get_current_user() 
     if user: 
      return func(request, *args, **kw) 
     else: 
      return request.redirect(users.create_login_url(request.get_url())) 

    return _wrapper