3
7^N                 @   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j| _t j| d S )N)type_get_storage_pathstorage_pathr   SESSION_COOKIE_NAMEfile_prefixsuper__init__)selfsession_key)	__class__ G/tmp/pip-build-8app2_gc/Django/django/contrib/sessions/backends/file.pyr      s    zSessionStore.__init__c             C   sR   y| j S  tk
rL   ttdd p(tj }tjj|sBt	d| || _ |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| j  }t|jts&tdtj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   r   _key_to_file)   s    zSessionStore._key_to_filec             C   s<   t j| j j}tjr0tjj|}|jt	j
dS tj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    |j dp| j tjtjd S )zS
        Return the expiry time of the file storing the session's content.
        Z_session_expiry)seconds)getr,   r'   	timedeltar   ZSESSION_COOKIE_AGE)r   session_datar   r   r   _expiry_dateC   s    
zSessionStore._expiry_datec             C   s   i }yt | j ddd}|j }W d Q R X |ry| j|}W nX ttfk
r } z8t|tr~tjd|j	j
 }|jt| | j  W Y d d }~X nX | j| j|d}|dkri }| j  | j  W n ttfk
r   d | _Y nX |S )Nrascii)encodingzdjango.security.%s)Zexpiryr   )openr#   readdecodeEOFErrorr	   
isinstancelogging	getLoggerr   __name__warningstrcreateZget_expiry_ager1   deleteIOError_session_key)r   r0   session_fileZ	file_dataeloggerZ
expiry_ager   r   r   loadK   s(    
zSessionStore.loadc             C   sB   x<| j  | _y| jdd W n tk
r2   wY nX d| _d S d S )NT)must_create)Z_get_new_session_keyrB   saver   modified)r   r   r   r   r?   e   s    
zSessionStore.createFc             C   s>  | j d kr| j S | j|d}| j }y@tjttddB }|rN|tjtjB O }tj	||}tj
| W n6 tk
r   |s~tY n tk
r   |rtY nX tjj|\}}yntj||d d\}}	d}
z<ztj|| j|j  W d tj
| X tj|	| d}
W d |
stj|	 X W n tttfk
r8   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_CREATr5   closeFileNotFoundErrorr   FileExistsErrorr   r   splitr   mkstempwriteencodeshutilmoveunlinkOSErrorrA   r8   )r   rG   r0   Zsession_file_nameflagsfdrK   rL   Zoutput_file_fdZoutput_file_nameZrenamedr   r   r   rH   o   s<    

zSessionStore.savec             C   s   t jj| j|S )N)r   r   existsr#   )r   r   r   r   r   r]      s    zSessionStore.existsc             C   sJ   |d kr| j d krd S | j }ytj| j| W n tk
rD   Y nX d S )N)r   r   rY   r#   rZ   )r   r   r   r   r   r@      s    
zSessionStore.deletec             C   s   d S )Nr   )r   r   r   r   clean   s    zSessionStore.cleanc             C   s\   | j  }tj}xHtj|D ]:}|j|s*q|t|d  }| |}dd |_|j  qW d S )Nc               S   s   d S )Nr   r   r   r   r   <lambda>   s    z,SessionStore.clear_expired.<locals>.<lambda>)	r   r   r   r   listdir
startswithlenr?   rF   )r   r   r   rC   r   sessionr   r   r   clear_expired   s    

zSessionStore.clear_expired)N)N)F)N)r<   
__module____qualname____doc__r   classmethodr   r#   r,   r1   rF   r?   rH   r]   r@   r^   rd   __classcell__r   r   )r   r   r      s   



=

r   )r'   r:   r   rW   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   