python-django中间件session源码

时间:2023-03-08 23:22:10
python-django中间件session源码

settings.py

MIDDLEWARE = [
'django.contrib.sessions.middleware.SessionMiddleware',
]

1. 看看SessionMiddleware类   from django.contrib.sessions.middleware import SessionMiddleware

class SessionMiddleware(MiddlewareMixin):
   #1.1
def __init__(self, get_response=None):
self.get_response = get_response
engine = import_module(settings.SESSION_ENGINE)
self.SessionStore = engine.SessionStore
   #2
def process_request(self, request):
session_key = request.COOKIES.get(settings.SESSION_COOKIE_NAME)
request.session = self.SessionStore(session_key) def process_response(self, request, response):
"""
If request.session was modified, or if the configuration is to save the
session every time, save the changes and set a session cookie or delete
the session cookie if the session has been emptied.
"""
try:
accessed = request.session.accessed
modified = request.session.modified
empty = request.session.is_empty()
except AttributeError:
pass
else:
# First check if we need to delete this cookie.
# The session should be deleted only if the session is entirely empty
if settings.SESSION_COOKIE_NAME in request.COOKIES and empty:
response.delete_cookie(
settings.SESSION_COOKIE_NAME,
path=settings.SESSION_COOKIE_PATH,
domain=settings.SESSION_COOKIE_DOMAIN,
)
else:
if accessed:
patch_vary_headers(response, ('Cookie',))
if (modified or settings.SESSION_SAVE_EVERY_REQUEST) and not empty:
if request.session.get_expire_at_browser_close():
max_age = None
expires = None
else:
max_age = request.session.get_expiry_age()
expires_time = time.time() + max_age
expires = cookie_date(expires_time)
# Save the session data and refresh the client cookie.
# Skip session save for 500 responses, refs #3881.
if response.status_code != 500:
try:
request.session.save()
except UpdateError:
raise SuspiciousOperation(
"The request's session was deleted before the "
"request completed. The user may have logged "
"out in a concurrent request, for example."
)
response.set_cookie(
settings.SESSION_COOKIE_NAME,
request.session.session_key, max_age=max_age,
expires=expires, domain=settings.SESSION_COOKIE_DOMAIN,
path=settings.SESSION_COOKIE_PATH,
secure=settings.SESSION_COOKIE_SECURE or None,
httponly=settings.SESSION_COOKIE_HTTPONLY or None,
)
return response

1.1步:每个中间件的类在调用的时候都会先执行init 构造方法

class SessionMiddleware(MiddlewareMixin):
def __init__(self, get_response=None):
self.get_response = get_response
#1.1.1 到配置文件中获取一个类
engine = import_module(settings.SESSION_ENGINE)
self.SessionStore = engine.SessionStore

1.1.1步:django自带的settings中没有SESSION_ENGINE这个属性,那就需要去全局的global_settings文件中找

        SESSION_ENGINE = 'django.contrib.sessions.backends.db'     #这个时候self.SessionStore =db 文件中的 SessionStore类
class SessionStore(SessionBase):
"""
Implements database session store.
"""
def __init__(self, session_key=None):
super(SessionStore, self).__init__(session_key) @classmethod
def get_model_class(cls):
# Avoids a circular import and allows importing SessionStore when
# django.contrib.sessions is not in INSTALLED_APPS.
from django.contrib.sessions.models import Session
return Session @cached_property
def model(self):
return self.get_model_class() def load(self):
try:
s = self.model.objects.get(
session_key=self.session_key,
expire_date__gt=timezone.now()
)
return self.decode(s.session_data)
except (self.model.DoesNotExist, SuspiciousOperation) as e:
if isinstance(e, SuspiciousOperation):
logger = logging.getLogger('django.security.%s' % e.__class__.__name__)
logger.warning(force_text(e))
self._session_key = None
return {} def exists(self, session_key):
return self.model.objects.filter(session_key=session_key).exists() def create(self):
while True:
self._session_key = self._get_new_session_key()
try:
# Save immediately to ensure we have a unique entry in the
# database.
self.save(must_create=True)
except CreateError:
# Key wasn't unique. Try again.
continue
self.modified = True
return def create_model_instance(self, data):
"""
Return a new instance of the session model object, which represents the
current session state. Intended to be used for saving the session data
to the database.
"""
return self.model(
session_key=self._get_or_create_session_key(),
session_data=self.encode(data),
expire_date=self.get_expiry_date(),
) def save(self, must_create=False):
"""
Saves the current session data to the database. If 'must_create' is
True, a database error will be raised if the saving operation doesn't
create a *new* entry (as opposed to possibly updating an existing
entry).
"""
if self.session_key is None:
return self.create()
data = self._get_session(no_load=must_create)
obj = self.create_model_instance(data)
using = router.db_for_write(self.model, instance=obj)
try:
with transaction.atomic(using=using):
obj.save(force_insert=must_create, force_update=not must_create, using=using)
except IntegrityError:
if must_create:
raise CreateError
raise
except DatabaseError:
if not must_create:
raise UpdateError
raise def delete(self, session_key=None):
if session_key is None:
if self.session_key is None:
return
session_key = self.session_key
try:
self.model.objects.get(session_key=session_key).delete()
except self.model.DoesNotExist:
pass @classmethod
def clear_expired(cls):
cls.get_model_class().objects.filter(expire_date__lt=timezone.now()).delete()

SessionStore类

2.这时候就该走中间件的process_request方法了

    def process_request(self, request):
session_key = request.COOKIES.get(settings.SESSION_COOKIE_NAME) #从cookies中获取 'sessionid' 对应的值
request.session = self.SessionStore(session_key) #把获取的值给db中的SessionStore类进行实例化并赋值给request.session
class SessionStore(SessionBase):
def __init__(self, session_key=None):
super(SessionStore, self).__init__(session_key) #调用父类的 init 构造方
class SessionBase(object):
   session_key = property(_get_session_key)
   _session_key = property(_get_session_key, _set_session_key) #在代码加载的时候要类的字段在方法前面加载,这时绑定了property方法,如果是获取操作就走_get_session_key方法,设置操作就走_set_session_key方法

   def __init__(self, session_key=None):
     #2.1
self._session_key = session_key #给_session_key赋值,走_set_session_key方法,把session_key传进去
self.accessed = False
self.modified = False
self.serializer = import_string(settings.SESSION_SERIALIZER) #SESSION_SERIALIZER = 'django.contrib.sessions.serializers.JSONSerializer'
     #表示 django 的session中间件是用Json进行序列化

2.1步:

class SessionBase(object):
def _set_session_key(self, value): if self._validate_session_key(value): #value = None
self.__session_key = value
else:
self.__session_key = None def _validate_session_key(self, key):
#如果sessionid有值并且长度大于等于8
return key and len(key) >= 8

3. 然后该走视图中 对request.session进行操作,如果对session就行设置 值就执行 SessionStore 类中的__setitem__方法,如果是取session中的值就执行 __getitem__方法,删除执行__delitem__方法

,但是在SessionStore类中没有发现这个方法,那就去它的父类中找

class SessionBase(object):
   #self 是 request.session 这个SessionBase类对象
def __getitem__(self, key):
return self._session[key] #到字典中获取值 def __setitem__(self, key, value):
self._session[key] = value #给字典设置值,并且改变modified 状态为 True
self.modified = True def __delitem__(self, key):
del self._session[key] #删除字典中的键值对,并且改变modified 状态为 True
self.modified = True    

在上面的代码中,我们看到self._session,点进去看看

class SessionBase(object):
_session = property(_get_session)
class SessionBase(object):
   #self 是 request.session 这个SessionBase类对象
 
def _get_session(self, no_load=False): self.accessed = True #在这步给session设置值的时候 把这个accessed 状态改成了 True
try:
return self._session_cache # 获取self 的_session_cache 属性,如果没有就走下面的except方法
except AttributeError:
if self.session_key is None or no_load:
self._session_cache = {} #赋值操作
else:
self._session_cache = self.load()
return self._session_cache

4. 走中间件的process_response 方法

class SessionMiddleware(MiddlewareMixin):
def process_response(self, request, response): try:
accessed = request.session.accessed # TRUE
modified = request.session.modified #当设置或删除操作时 为TRUE
  
       #4.1
empty = request.session.is_empty() # Flase
except AttributeError:
pass
else:
# 4.2 如果sessionid在cookies中 并且值为空,从cookie中把sessionid键值删除
if settings.SESSION_COOKIE_NAME in request.COOKIES and empty:
response.delete_cookie(
settings.SESSION_COOKIE_NAME,
path=settings.SESSION_COOKIE_PATH,
domain=settings.SESSION_COOKIE_DOMAIN,
)
else:if accessed:
patch_vary_headers(response, ('Cookie',))
          4.3 有效期和 过期时间相关
if (modified or settings.SESSION_SAVE_EVERY_REQUEST) and not empty: #True and True
            #在浏览器关闭的时候 session 的有效期为 None , 过期时间 为 None
if request.session.get_expire_at_browser_close():
max_age = None
expires = None
else:
               # 设置 有效期 和 过期时间
max_age = request.session.get_expiry_age()
expires_time = time.time() + max_age
expires = cookie_date(expires_time)
# Save the session data and refresh the client cookie.
# Skip session save for 500 responses, refs #3881.
            
            
if response.status_code != 500:
try:
                 #4.4 访问正常
request.session.save()
except UpdateError:
raise SuspiciousOperation(
"The request's session was deleted before the "
"request completed. The user may have logged "
"out in a concurrent request, for example."
)
               #4.5 对 response 设置 cookie
response.set_cookie(
settings.SESSION_COOKIE_NAME,
request.session.session_key, max_age=max_age,
expires=expires, domain=settings.SESSION_COOKIE_DOMAIN,
path=settings.SESSION_COOKIE_PATH,
secure=settings.SESSION_COOKIE_SECURE or None,
httponly=settings.SESSION_COOKIE_HTTPONLY or None,
)
return response

4.1步:

class SessionBase(object):
def is_empty(self):
"Returns True when there is no session_key and the session is empty"
try:
return not bool(self._session_key) and not self._session_cache # 获取_session_key 执行property(_get_session_key)方法
                                              # 此时 self._session_cache 中已经有值了
        # True and Flase
except AttributeError:
return True
class SessionBase(object):
def _get_session_key(self):
return self.__session_key #从2.1步可以得出 为None

4.4步:

class SessionStore(SessionBase):
def save(self, must_create=False):
     #4.4.1
if self.session_key is None: return self.create() data = self._get_session(no_load=must_create)
obj = self.create_model_instance(data)
using = router.db_for_write(self.model, instance=obj)
try:
with transaction.atomic(using=using):
obj.save(force_insert=must_create, force_update=not must_create, using=using)
except IntegrityError:
if must_create:
raise CreateError
raise
except DatabaseError:
if not must_create:
raise UpdateError
raise

4.4.1步:

#session_key 获取,执行session_key = property(_get_session_key)  _get 方法

class SessionBase(object):
def _get_session_key(self):
return self.__session_key # None

执行 self.create()  方法

class SessionStore(SessionBase):
def create(self):
while True:
       #4.4.1.1
self._session_key = self._get_new_session_key() #对_session_key 赋值操作,执行_set 方法
try:
          #4.4.1.2 传参 must_create=True
self.save(must_create=True)
except CreateError:
# Key wasn't unique. Try again.
continue
self.modified = True
return

4.4.1.1步:

class SessionBase(object):
def _get_new_session_key(self):
while True:
session_key = get_random_string(32, VALID_KEY_CHARS)
if not self.exists(session_key):
break
return session_key #随机字符串
class SessionBase(object):
def _set_session_key(self, value): if self._validate_session_key(value): #此时 value 是一个随机字符串
self.__session_key = value #self.__session_key 是 一个随机字符串
else:
self.__session_key = None

4.4.1.2步:

class SessionStore(SessionBase):
def save(self, must_create=False): if self.session_key is None: #获取session_key 操作,执行_get方法 见下面
return self.create()
data = self._get_session(no_load=must_create) #must_create =True 见下面
obj = self.create_model_instance(data) #django创建session表对象
using = router.db_for_write(self.model, instance=obj)
try:
with transaction.atomic(using=using):
          # session对象 执行save方法 参数must_create = True 表示是创建 一条数据
obj.save(force_insert=must_create, force_update=not must_create, using=using)
except IntegrityError:
if must_create:
raise CreateError
raise
except DatabaseError:
if not must_create:
raise UpdateError
raise class SessionBase(object): session_key = property(_get_session_key) def _get_session_key(self):
return self.__session_key #此时是一个随机字符串 class SessionBase(object):
def _get_session(self, no_load=False): self.accessed = True
try:
return self._session_cache #此时 字典中已经有值
except AttributeError:
if self.session_key is None or no_load:
self._session_cache = {}
else:
self._session_cache = self.load()
return self._session_cache

python-django中间件session源码