U e@sddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZGd d d eZGd d d eZdS)N)urlparse)settings)PermissionDenied) mail_managers)HttpResponsePermanentRedirect) is_valid_path)MiddlewareMixin)escape_leading_slashesc@s4eZdZdZeZddZddZddZdd Z d S) CommonMiddlewarea  "Common" middleware for taking care of some basic operations: - Forbid access to User-Agents in settings.DISALLOWED_USER_AGENTS - URL rewriting: Based on the APPEND_SLASH and PREPEND_WWW settings, append missing slashes and/or prepends missing "www."s. - If APPEND_SLASH is set and the initial URL doesn't end with a slash, and it is not found in urlpatterns, form a new URL by appending a slash at the end. If this new URL is found in urlpatterns, return an HTTP redirect to this new URL; otherwise process the initial URL as usual. This behavior can be customized by subclassing CommonMiddleware and overriding the response_redirect_class attribute. cCs|jd}|dk r2tjD]}||rtdq|}tjr|r|ds| |rd| |}n| }| |j d||SdS)z Check for denied User-Agents and rewrite the URL based on settings.APPEND_SLASH and settings.PREPEND_WWW HTTP_USER_AGENTNzForbidden user agentzwww.z://www.)METAgetrZDISALLOWED_USER_AGENTSsearchrget_hostZ PREPEND_WWW startswithshould_redirect_with_slashget_full_path_with_slash get_full_pathresponse_redirect_classscheme)selfrequest user_agentZuser_agent_regexhostpathrs   z+CommonMiddleware.should_redirect_with_slashcCsD|jdd}t|}tjr@|jdkr@td|j||d|S)z Return the full path of the request with a trailing slash appended. Raise a RuntimeError if settings.DEBUG is True and request.method is POST, PUT, or PATCH. T)Zforce_append_slash)POSTPUTPATCHa"You called this URL via %(method)s, but the URL doesn't end in a slash and you have APPEND_SLASH set. Django can't redirect to the slash URL while maintaining %(method)s data. Change your form to point to %(url)s (note the trailing slash), or set APPEND_SLASH=False in your Django settings.)methodurl)rr rDEBUGr) RuntimeErrorr)rrnew_pathrrrrLs   z)CommonMiddleware.get_full_path_with_slashcCsL|jdkr$||r$|||S|jsH|dsHtt|j|j d<|S)z When the status code of the response is 404, it may redirect to a path with an appended slash if should_redirect_with_slash() returns True. zContent-Length) status_coderrrZ streaming has_headerstrlencontentheaders)rrresponserrrprocess_responseds z!CommonMiddleware.process_responseN) __name__ __module__ __qualname____doc__rrrrrr6rrrrr s r c@s$eZdZddZddZddZdS)BrokenLinkEmailsMiddlewarecCs|jdkrtjs|}|}|jdd}|||||s|jdd}|jdd}td| ||rndnd|fd ||||fd d |S) z=Send broken link emails for relevant 404 NOT FOUND responses.r.Z HTTP_REFERERr zZ REMOTE_ADDRzBroken %slink on %sz INTERNAL z=Referrer: %s Requested URL: %s User agent: %s IP address: %s T)Z fail_silently) r/rr+rrr r is_ignorable_requestris_internal_request)rrr5domainrrefererZuaiprrrr6ws*   z+BrokenLinkEmailsMiddleware.process_responsecCsttdt||S)ze Return True if the referring URL is the same domain as the current request. z ^https?://%s/)boolrer$escape)rr?r@rrrr>sz.BrokenLinkEmailsMiddleware.is_internal_requestcs|sdStjr,dr,|ddkr,dS|||sDd|krDdSt|}|jd|fkrh|jkrhdStfddtjDS) z Return True if the given request *shouldn't* notify the site managers according to project settings or in situations outlined by the inline comments. TrN?r<c3s|]}|VqdS)N)r).0patternurirr szBBrokenLinkEmailsMiddleware.is_ignorable_request..) rr r!r>rnetlocranyZIGNORABLE_404_URLS)rrrJr?r@Zparsed_refererrrIrr=s z/BrokenLinkEmailsMiddleware.is_ignorable_requestN)r7r8r9r6r>r=rrrrr;vsr;)rC urllib.parserZ django.confrZdjango.core.exceptionsrZdjango.core.mailrZ django.httprZ django.urlsrZdjango.utils.deprecationrZdjango.utils.httpr r r;rrrrs        i