U
    mÇf?
  ã                   @   s@   d dl Z d dlmZ d dlmZ d dlmZ G dd„ deƒZdS )é    N)Úsettings)ÚHttpResponsePermanentRedirect)ÚMiddlewareMixinc                       s,   e Zd Z‡ fdd„Zdd„ Zdd„ Z‡  ZS )ÚSecurityMiddlewarec                    sb   t ƒ  |¡ tj| _tj| _tj| _tj	| _
tj| _tj| _dd„ tjD ƒ| _tj| _tj| _d S )Nc                 S   s   g | ]}t  |¡‘qS © )ÚreÚcompile)Ú.0Úrr   r   ú>/tmp/pip-unpacked-wheel-siwebuq3/django/middleware/security.pyÚ
<listcomp>   s     z/SecurityMiddleware.__init__.<locals>.<listcomp>)ÚsuperÚ__init__r   ZSECURE_HSTS_SECONDSÚsts_secondsZSECURE_HSTS_INCLUDE_SUBDOMAINSÚsts_include_subdomainsZSECURE_HSTS_PRELOADÚsts_preloadZSECURE_CONTENT_TYPE_NOSNIFFÚcontent_type_nosniffZSECURE_SSL_REDIRECTÚredirectZSECURE_SSL_HOSTÚredirect_hostZSECURE_REDIRECT_EXEMPTÚredirect_exemptZSECURE_REFERRER_POLICYÚreferrer_policyZ!SECURE_CROSS_ORIGIN_OPENER_POLICYÚcross_origin_opener_policy)ÚselfZget_response©Ú	__class__r   r   r   	   s    zSecurityMiddleware.__init__c                    sX   |j  d¡‰ | jrT| ¡ sTt‡ fdd„| jD ƒƒsT| jp>| ¡ }td|| 	¡ f ƒS d S )Nú/c                 3   s   | ]}|  ˆ ¡V  qd S )N)Úsearch)r	   Úpattern©Úpathr   r   Ú	<genexpr>   s     z5SecurityMiddleware.process_request.<locals>.<genexpr>zhttps://%s%s)
r   Úlstripr   Ú	is_secureÚanyr   r   Úget_hostr   Zget_full_path)r   ÚrequestÚhostr   r   r   Úprocess_request   s    ÿþýÿz"SecurityMiddleware.process_requestc              	   C   s°   | j rF| ¡ rFd|krFd| j  }| jr.|d }| jr<|d }||jd< | jrZ|j dd¡ | jr˜|j dd t	| jt
ƒrŽd	d
„ | j d¡D ƒn| j¡¡ | jr¬| d| j¡ |S )NzStrict-Transport-Securityz
max-age=%sz; includeSubDomainsz	; preloadzX-Content-Type-OptionsZnosniffzReferrer-Policyú,c                 S   s   g | ]}|  ¡ ‘qS r   )Ústrip)r	   Úvr   r   r   r   7   s     z7SecurityMiddleware.process_response.<locals>.<listcomp>zCross-Origin-Opener-Policy)r   r"   r   r   Úheadersr   Ú
setdefaultr   ÚjoinÚ
isinstanceÚstrÚsplitr   )r   r%   ÚresponseZ
sts_headerr   r   r   Úprocess_response!   s:    ÿþý


ÿýþ	þz#SecurityMiddleware.process_response)Ú__name__Ú
__module__Ú__qualname__r   r'   r2   Ú__classcell__r   r   r   r   r      s   r   )r   Zdjango.confr   Zdjango.httpr   Zdjango.utils.deprecationr   r   r   r   r   r   Ú<module>   s   