U
    g<                     @   s   d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZm	Z	m
Z
mZ d dlmZ d dlmZmZ d dlmZ G dd de
ZdS )	    N)settings)VALID_KEY_CHARSCreateErrorSessionBaseUpdateError)InvalidSessionKey)ImproperlyConfiguredSuspiciousOperation)timezonec                       s   e Zd ZdZd fdd	Zedd ZdddZd	d
 Zdd Z	dd Z
dd ZdddZdd ZdddZdd Zedd Z  ZS ) SessionStorez/
    Implement a file based session store.
    Nc                    s"   |   | _tj| _t | d S N)_get_storage_pathstorage_pathr   SESSION_COOKIE_NAMEfile_prefixsuper__init__selfsession_key	__class__ I/tmp/pip-unpacked-wheel-ye1bytdm/django/contrib/sessions/backends/file.pyr      s    
zSessionStore.__init__c                 C   sX   z| j W S  tk
rR   ttdd p*t }tj|sDt	d| || _ | Y S X d S )NZSESSION_FILE_PATHzThe session storage path %r doesn't exist. Please set your SESSION_FILE_PATH setting to an existing directory in which Django can store session data.)
Z_storage_pathAttributeErrorgetattrr   tempfile
gettempdirospathisdirr   )clsr   r   r   r   r      s    zSessionStore._get_storage_pathc                 C   s<   |dkr|   }t|ts&tdtj| j| j	| S )z@
        Get the file associated with this session key.
        Nz!Invalid characters in session key)
Z_get_or_create_session_keysetissubsetr   r   r   r   joinr   r   r   r   r   r   _key_to_file)   s    zSessionStore._key_to_filec                 C   s<   t |  j}tjr0tj|}|jt	j
dS tj|S )zY
        Return the modification time of the file storing the session's content.
        )tzinfo)r   statr%   st_mtimer   ZUSE_TZdatetimeutcfromtimestampreplacer
   utcfromtimestamp)r   Zmodificationr   r   r   _last_modification9   s
    zSessionStore._last_modificationc                 C   s"   | dp |  tj|  d S )zS
        Return the expiry time of the file storing the session's content.
        Z_session_expiry)seconds)getr.   r)   	timedeltaZget_session_cookie_age)r   session_datar   r   r   _expiry_dateC   s    
zSessionStore._expiry_datec              
   C   s   i }zt |  dd}| }W 5 Q R X |rz| |}W nX ttfk
r } z6t|tr|td|j	j
 }|t| |   W 5 d }~X Y nX | j| |d}|dkri }|   |   W n ttfk
r   d | _Y nX |S )Nascii)encodingzdjango.security.%s)Zexpiryr   )openr%   readdecodeEOFErrorr	   
isinstancelogging	getLoggerr   __name__warningstrcreateZget_expiry_ager3   deleteOSError_session_key)r   r2   session_fileZ	file_dataeloggerZ
expiry_ager   r   r   loadK   s(    
zSessionStore.loadc                 C   s>   |   | _z| jdd W n tk
r2   Y q Y nX d| _d S )NT)must_create)Z_get_new_session_keyrC   saver   modifiedr   r   r   r   r@   e   s    

zSessionStore.createFc              	   C   s<  | j d kr|  S | j|d}|  }z@tjttddB }|rN|tjtjB O }t	||}t
| W n6 tk
r   |s~tY n tk
r   |rtY nX tj|\}}zntj||d d\}}	d}
z<zt|| |  W 5 t
| X t|	| d}
W 5 |
st|	 X W n ttfk
r6   Y nX d S )N)Zno_loadO_BINARYr   Z_out_)dirprefixFT)r   r@   Z_get_sessionr%   r   O_WRONLYr   O_EXCLO_CREATr6   closeFileNotFoundErrorr   FileExistsErrorr   r   splitr   mkstempunlinkwriteencodeshutilmover9   rB   )r   rH   r2   Zsession_file_nameflagsfdrM   rN   Zoutput_file_fdZoutput_file_nameZrenamedr   r   r   rI   o   s<    

zSessionStore.savec                 C   s   t j| |S r   )r   r   existsr%   r   r   r   r   r^      s    zSessionStore.existsc                 C   sJ   |d kr| j d krd S | j }zt| | W n tk
rD   Y nX d S r   )r   r   rW   r%   rB   r   r   r   r   rA      s    
zSessionStore.deletec                 C   s   d S r   r   rK   r   r   r   clean   s    zSessionStore.cleanc                 C   sX   |   }tj}t|D ]:}||s(q|t|d  }| |}dd |_|  qd S )Nc                   S   s   d S r   r   r   r   r   r   <lambda>       z,SessionStore.clear_expired.<locals>.<lambda>)	r   r   r   r   listdir
startswithlenr@   rG   )r!   r   r   rD   r   sessionr   r   r   clear_expired   s    

zSessionStore.clear_expired)N)N)F)N)r=   
__module____qualname____doc__r   classmethodr   r%   r.   r3   rG   r@   rI   r^   rA   r_   rf   __classcell__r   r   r   r   r      s   




=

r   )r)   r;   r   rZ   r   Zdjango.confr   Z%django.contrib.sessions.backends.baser   r   r   r   Z"django.contrib.sessions.exceptionsr   Zdjango.core.exceptionsr   r	   Zdjango.utilsr
   r   r   r   r   r   <module>   s   