自定义redis session

1.思路

2.程序实现

1.用户系统类

这里模拟一个蹩脚的用户系统类(userSystem),如下:

#coding=utf-8
#Redis实现用户系统 __author__ = 'beginman' import redis import datetime import hashlib r = redis.StrictRedis(host='localhost', port='6379', db=0) class usSystem(object): def __init__(self, request,response=None, uid=0, **kwargs): self.request = request self.response = response self.kwargs = kwargs self.uid = uid # user id self.sessionid = None def testCookie(self): """事先在登陆方法中下了request.session.set_test_cookie()的套子""" if self.request.session.test_cookie_worked(): self.request.session.delete_test_cookie() return True return False def getUsObj(self): """返回用户对象,有则说明用户已登陆,无则注销""" self.sessionid = self.request.COOKIES.get('sessionid', None) if r.exists(self.sessionid): if r.exists('sessionid_%s' %self.sessionid): return r.hget('sessionid_%s' %self.sessionid, 'uid') return None def setCookieAndSession(self): """cookie在登陆成功后已经写入""" self.sessionid = self.request.COOKIES.get('sessionid', None) if not self.sessionid: # set cookie h = hashlib.md5() h.update(datetime.datetime.now()) self.response.set_cookie('sessionid', h.hexdigest()) self.sessionid = h.hexdigest() if not r.exists('sessionid_%s' %self.sessionid): #set session r.hset('sessionid_%s' %self.sessionid,'uid', self.uid) return True

然后在登陆方法中这样写:

# coding=utf-8
__author__ = 'beginman' from django.shortcuts import render from django.http import HttpResponseRedirect from form import LoginForm from common.userSystem import usSystem import redis import datetime r = redis.StrictRedis(host='localhost', port='6379', db=0) def home(request): return render(request, 'index.html') def usLogin(request): context = {} if request.method == 'POST': form = LoginForm(request.POST) if form.is_valid(): us = form.cleaned_data['us'] pwd = form.cleaned_data['pwd'] if r.exists('us:%s:id' %us): # 检查是否存在该用户关系键值 uid = r.get('us:%s:id' %us) # 获取该用户在user表中对应的id if r.exists('user:%s' %uid): # 检查是否存在该用户键值(如user:1) us_, pwd_ = r.hmget('user:%s' %uid, 'username', 'pwd') # 获取该用户的用户名密码 if us_ == us and pwd_ == pwd: # 校验成功 r.hincrby('user:%s' %uid, 'login_count', 1) # 登陆次数累加 r.hset('user:%s' %uid, 'last_login_date', datetime.datetime.now()) # 添加最近登陆 # set Cookies res = HttpResponseRedirect('/') ussys = usSystem(request, res, uid) if ussys.testCookie() and ussys.setCookieAndSession(): return res context['msg'] = u'账号或密码错误' context['form'] = form request.session.set_test_cookie() form = LoginForm() context['form'] = form return render(request, 'login.html', context)

同时要注意中间件处理:

#coding=utf-8
#中间件扩展 __author__ = 'beginman' from django.http import HttpResponseRedirect from django.conf import settings from common.userSystem import usSystem class Mymiddleware(object): def process_request(self, request): """Request预处理函数""" path = str(request.path) request.session['domain'] = settings.DOMAIN if path.startswith('/site_media/'): return None #验证登陆 ussys = usSystem(request) if ussys.getUsObj(): pass

在一些需要登陆后才能访问的可以写在验证登陆后面.这里还需要慢慢改进.

转载于:https://www.cnblogs.com/ExMan/p/10432642.html