python_tornado_session用户验证

什么是session？

       -- Django中带有session，tornado中自己写

       -- 逻辑整理

　　　　　　用户请求过来，验证通过，随机生成一个字符串当作value返回给浏览器，

　　　　　　在服务器中用户信息与随机生成一个字符串建立对应关系，

　　　　　　下次用户请求带来了cookie，通过字符串找到对应的用户信息。

　　-- 优点：一切与用户相关的信息都保存在客户端中，

　　　　　　　用户端无法获取除了随机字符串其他信息，解决了信息隐患

基于session做用户验证如何做？

       -- 定义一个全局字典

       -- 验证用户，失败跳转到登录页面

       -- 成功 >>生成随机字符串 映射用户信息，登陆状态为True

　　　　　 >>设置cookie，value=随机字符串

　　-- 下次访问，通过cookie中value，找到映射的用户信息，

　　　　找打登录状态，为Ture跳转到成功页面，为False，跳转到登录界面

#!/usr/bin/python3

import tornado.web
import tornado.ioloop
import time
import hashlib

user_login = {}
class HomeHandle(tornado.web.RequestHandler):

    def get(self, *args, **kwargs):
        if self.get_argument('u') in ['yizhihua', 'beimenchuixue']:      # 判断是合法用户

            m = hashlib.md5()                                   # 通过md5获得随机值
            m.update(bytes(str(time.time()), encoding='utf-8'))
            random_str = m.hexdigest()

            user_login[random_str] = {}                         # 全局变量新增一个用户信息字典
            user_login[random_str]['username'] = self.get_argument('u')

            if self.get_argument('u') == 'yizhihua':            # 把yizhihua用户信息设置不可登录状态
                user_login[random_str]['login_status'] = False
            else:
                user_login[random_str]['login_status'] = True   # 其他用户可以登录

            self.set_cookie('yi_ye', random_str)
        else:
            self.write('请登陆')

class ManHandle(tornado.web.RequestHandler):

    def get(self, *args, **kwargs):

        random_str = self.get_cookie('yi_ye')                   # 获得cookie对应的随机字符串

        if user_login.get(random_str, None):                    # 尝试取值，没有相当于没有验证
            if user_login[random_str].get('login_status', None):    # 判断用户是否可登录状态
                temp = '%s-%s' % (user_login[random_str].get('username', ''),
                                  user_login[random_str].get('login_status', ''))
                self.write(temp)                                # 打印登录成功界面
            else:
                self.redirect('/home')                          # 返回验证
        else:
            self.redirect('/home')                              # 返回验证

setting = {
    'template_path': 'template',
    'cookie_secret': 'yes'

}

Appliaction = tornado.web.Application(
    [(r'/home', HomeHandle),
     (r'/manager', ManHandle),
     ], **setting
)

if __name__ == '__main__':
    Appliaction.listen(9999)
    tornado.ioloop.IOLoop.instance().start()

　　-- 启动，通过http://127.0.0.1:9999/home?u=yizhihua 进行验证

　　上面代码有问题，session部分应该很多地方需要验证，

　　每个函数都写一个，显得太low了，如何解决？通过类

如何把session定义成一个类？

　　逻辑整理：

       class Session():

              def __init__(self,)

              def __random_str()

                     -- 生成随机字符串

              def set_value(self):

                     -- 获得一个随机的数据

                     -- 定义自己唯一的数据

                     -- 写入随机字符串

              def get_value()

                     -- 获得客户端随机字符串

                     -- 取自己的数据

                     -- 专属信息key

#!/usr/bin/python3

import tornado.web
import tornado.ioloop

user_info = {}
# 输入 self
# 内部逻辑 验证用户 生成随机字符串  添加对应用户信息  更改登录状态  发送cookie  获得cookie ，验证，检查登录状态  取出用户信息
# 输出 通过外部参数获得对应的参数
class Session(object):
    """Session验证"""
    def __init__(self, handler, cookie_key, url_home):  # self对象，cookie的key，url_home跳转主页url
        self.handler = handler                          # self 对象
        self.cookie_key = cookie_key                    # cookie的key
        self.url_home = url_home                        # 验证主页

    def __get_random_str(self):
        """获得随机字符串"""
        import time
        import hashlib
        m = hashlib.md5()
        m.update(bytes(str(time.time()), encoding='utf-8'))
        return m.hexdigest()

    def set_value(self, get_name, user_status):
        """用户验证"""
        if self.handler.get_argument(get_name) in ['yizhihua', 'beimenchuixue']:    # 验证用户
            random_str = self.__get_random_str()
            user_info[random_str] = {}                                              # 初始化用户信息表
            user_info[random_str]['username'] = self.handler.get_argument(get_name)
            user_info[random_str]['user_status'] = user_status
            user_info[random_str]['work'] = 'student'
            user_info[random_str]['addr'] = '湖南'

            self.handler.set_cookie(self.cookie_key, random_str)        # 设置cookie
            self.handler.redirect('/manager')                           # 验证成功跳转到/manager页面
        else:
            self.handler.redirect(self.url_home)                        # 验证失败重新验证

    def get_value(self, user_key):
        """取cookie值，取用户信息"""
        random_str = self.handler.get_cookie(self.cookie_key, None)
        if random_str:                                                  # 判断是否有对应的cookie值
            if random_str in user_info:                                 # 验证cookie
                if user_info[random_str].get('user_status', None):      # 判断可登录状态
                    return user_info[random_str].get(user_key, None)    # 取值
                else:
                    self.handler.write('不可登录状态')
            else:
                self.handler.redirect(self.url_home)
        else:
            self.handler.redirect(self.url_home)


class Home(tornado.web.RequestHandler):
    
    def get(self, *args, **kwargs):
        self.render('home.html')                                        # form表单


class HomeHandle(tornado.web.RequestHandler):

    def get(self, *args, **kwargs):
        session = Session(self, 'yi_ye', '/home/')
        session.set_value('u', True)

class ManHandle(tornado.web.RequestHandler):

    def get(self, *args, **kwargs):

        session = Session(self, 'yi_ye', '/home')
        work = session.get_value('work')
        addr = session.get_value('addr')
        print(work, addr)



setting = {
    'template_path': 'views',
    'cookie_secret': 'yes'

}


Appliaction = tornado.web.Application(
    [(r'/home', HomeHandle),
    (r'/home/', Home),
     (r'/manager', ManHandle),
     ], **setting
)


if __name__ == '__main__':
    Appliaction.listen(9999)
    tornado.ioloop.IOLoop.instance().start()

　　

如何在这个session类中调用处理函数中的方法？

　　-- 可以把class逻辑处理类中的self对象当做参数传入进来，

　　　　然后通过session类__init__析构函数给self对象进行重新命名，

　　　　然后就通过新的名字调用class逻辑处理类中的任何方法

上面的方式并没有实现面向过程，如何实现？

　　--start.py  　　写入

#!/usr/bin/python3
import tornado.web
import tornado.ioloop
user_info = {}      # 临时存放用户信息列表
user_eyes = [{'yizhihua': '18'}, {'hehe': '19'}]
# 入 self 验证 生成cookie 更改用户可登录状态
class Session(object):
    """session验证"""
    def __init__(self, handler):
        self.handler = handler

    def __get_random_str(self):
        """生产随机数"""
        import time, hashlib
        m = hashlib.md5()
        m.update(bytes(str(time.time()), encoding='utf-8'))
        random_str = m.hexdigest()
        return random_str
        pass

    def __setitem__(self, key, value):
        """验证登录，设置值"""
        user_name = self.handler.get_argument('user_name')      # 获取前端验证
        pwd = self.handler.get_argument('pwd')
        print(user_name, pwd)
        if {user_name: pwd} in user_eyes:                       # 判断用户合法
            if not self.handler.get_cookie('yi_ye', None):      # 如果cookie存在，不生成随机字符串
                random_str = self.__get_random_str()                # 生成随机字符串
            else:
                random_str = self.handler.get_cookie('yi_ye', None)  # 就算是用户伪造cookie也得是验证通过的情况下
            user_info[random_str] = {}                          # 添加用户信息
            user_info[random_str]['user_name'] = user_name
            user_info[random_str]['pwd'] = pwd
            user_info[random_str][key] = value                  # 通过接口设置登录状态
            print(user_info)

            self.handler.set_cookie('yi_ye', random_str)            # 生成cookie
            self.handler.render('home.html', user_name=user_name)   # 登录页面
        else:
            self.handler.redirect('/login')                         # 返回登录

    def __getitem__(self, item):
        """验证session，获取值"""
        random_str = self.handler.get_cookie('yi_ye', None)         # 获取cookie
        if user_info:                                               # 判断user_info是否有值
            if random_str:                                          # 判断cookie使用有值
                server_info = user_info.get(random_str, None)       # 判断user_info对应的值
                if server_info.get('status'):                       # 判断状态
                    return server_info.get(item, None)              # 获取值
                else:
                    self.handler.redirect('/login')                 # 以上任意一个条件不满足，退回登录界面
            else:
                self.handler.redirect('/login')
        else:
            self.handler.redirect('/login')

class BaseHandler(tornado.web.RequestHandler):
    """让其他处理类每次都先实例session对象"""
    def initialize(self):
        self.session = Session(self)                                # 实例Session类，self为每个类处理对象


# 登录界面不需要session验证，打开隐私界面需要session验证

class Login(BaseHandler):

    def get(self, *args, **kwargs):
        self.render('login.html')

    def post(self, *args, **kwargs):
        self.session['status'] = True                              # 设置登录状态为True

class Home(BaseHandler):

    def get(self, *args, **kwargs):
        if self.session['status']:                                  # 取登录状态
            self.render('home.html', user_name=self.session['user_name'])   # 验证，并模版渲染


setting ={
    'template_path': 'views',
    'static_path': 'statics'
}

Application = tornado.web.Application(
    [
        (r'/login', Login),
        (r'/home', Home),
    ], **setting
)

if __name__ == '__main__':
    Application.listen(9999)
    tornado.ioloop.IOLoop.instance().start()

　　-- login.html　　写入

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>login</title>
</head>
<body>
    <form action="/login"  method="post">
        <p>账户：<input name='user_name' type="text" ></p>
        <p>密码：<input name='pwd' type="password"></p>
        <p><input type="submit" value="提交"></p>
    </form>
</body>
</html>

 　　-- home.html　　写入

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>主页</title>
</head>
<body>
    <p><h>欢迎来到{{user_name}}主页</h></p>
    <p><input type="button" value="退出登录"></p>
</body>
</html>

 如何逻辑整理？

　　-- 用户行为：进入登录界面、输入验证信息、提交、进入隐私页面

　　-- 服务器行为：验证合法性、成功跳转到隐私页面、设置session_cookie、请求隐私页面验证session，登录界面提交验证不需要验证session

　　-- 整个流程：输入验证、提交、后台验证合法性、合法生成session_cookie并且进入隐私页面、失败重新跳转到登录、

　　　　　　　　用户请求隐私页面、验证session_cookie、通过则跳转、不通过则跳转到登录界面