Python廖雪峰实战web开发(Day10(下)-用户登录)

用户登录比用户注册复杂。由于HTTP协议是一种无状态协议,而服务器要跟踪用户状态,就只能通过cookie实现。

大多数Web框架提供了Session功能来封装保存用户状态的cookie,使用Session的Web App很难扩展。
这里采用直接读取cookie的方式来验证用户登录,每次用户访问任意URL,都会对cookie进行验证(比方说,假如不验证左上角的用户栏会一直是请登录状态等等),这种方式的好处是保证服务器处理任意的URL都是无状态的,可以扩展到多台服务器。


1. 用户登录

1.1 编写用户登录API

用户登录之后服务器会生成一cookie发送给浏览器,所以,要保证这个cookie不会被客户端伪造出来。
由上一章也知道,实现防伪造cookie的关键是通过一个单向算法
查看上一章的代码,并不难发现当一个正确的密码输入,会经历以下步骤:

  • 首先会由网页中的javascript取得对应的值,并按照如下组合方式,进行摘要算法计算出一个字符串(A):

    `"email" + ":" + "passwd"`
    
  • 然后字符串(A)被以密码的形式传递到API内,在API内,字符串(A)再一次按照如下组合方式,进行摘要算法计算出一个字符串(B),并保存到服务器数据库上

    `"用户id" + ":" + 字串符(A)`
    
  • 到了最后制作cookie发送给浏览器时,又使用字符串(B)按照如下组合方式,进行摘要算法计算出一个字符串(C):

    `"用户id" + 字串符(B) + "到期时间" + "密匙"`
    

    最后再按照如下组合方式,生成一个字串符(D)发送给浏览器

    `"用户id" + "到期时间" + 字符串(C)`
    

所以浏览器实际收到的信息有:用户id、过期时间、SHA1值(字串符(C))
cookie未到过期的期间,当服务器验证cookie是否伪造时,其实只需根据用户信息在数据库查找相应的用户口令(即字符串(B)),再使用其进行摘要算法cookie中的字符串(C)比较是否等价,就可以知道是否伪造了。

先编写用户登录API:

#导入模块。。。

#显示登录页面
@get('/signin')
def signin():
    return {
        '__template__': 'signin.html'
    }

COOKIE_NAME = 'awesession'#用来在set_cookie中命名
_COOKIE_KEY = config['session']['secret']#导入默认设置


#验证登录信息
@post('/api/authenticate')
async def authenticate(*,email,passwd):
    if not email:
        raise APIValueError('email')
    if not passwd:
        raise APIValueError('passwd')
    users = await User.findall(where='email=?',args=[email])
    if len(users) == 0:
        raise APIValueError('email','Email not exist.')
    user = user[0]#此时finall得出来的数值是一个仅含一个dict的list,就是sql语句返回什么类型的数据自己忘了
    #把登录密码转化格式并进行摘要算法
    sha1 = hashlib.sha1()
    sha1.update(user.id.encode('utf-8'))
    sha1.update(b':')
    sha1.update(passwd.encode('utf-8'))
    if sha1.hexdigest() != user.passwd:#与数据库密码比较
        raise APIValueError('password','Invaild password')
    #制作cookie发送给浏览器,这步骤与注册用户一样
    r = web.Response()
    r.set_cookie(COOKIE_NAME, user2cookie(user,86400), max_age=86400, httponly=True)
    user.passwd = "******"
    r.content_type = 'application/json'
    r.body = json.dumps(user, ensure_ascii = False).encode('utf-8')
    return r

1.2 编写解析cookie代码

由于每次用户访问任意URL,都会对cookie进行验证,也就是说这对于每个URL处理函数,如果我们都要去写解析cookie的代码,那会导致代码重复很多次。
还记得Day5编写web框架时引入的middleware吗?

利用middleware在处理URL之前,把cookie解析出来,并将登录用户绑定到request对象上,这样,后续的URL处理函数就可以直接拿到登录用户。

编写解析cookie代码:

#解释cookie
async def cookie2user(cookie_str):
    if not cookie_str:
        return None
    try:
        L = cookie_str.split('-') #拆分字符串(D)
        if len(L) !=3:
            return None
        uid, expires, sha1 = L
        if float(expires) < time.time():#查看是否过期,这里廖大用的是int,但是字符串用int的时候,只能全是数字,不能含小数点
            return None
        user = await User.find(uid)
        if not user:
            return None
        #用数据库来生字符串(C)与cookie的比较
        s = '%s-%s-%s-%s'%(uid, user.passwd, expires, _COOKIE_KEY)
        if sha1 != hashlib.sha1(s.encode('utf-8')).hexdigest():
            logging.info('invalid sha1')
            return None
        user.passwd = "******"
        return user
    except Exception as e:
        logging.exception(e)
        return None

编写middleware解析cookie工厂函数:

#提取并解析cookie并绑定在request对象上
async def auth_factory(app,handler):
    async def auth(request):
        logging.info('check user: %s %s' % (request.method, request.path))
        request.__user__ = None #初始化
        cookie_str = request.cookies.get(COOKIE_NAME) #读取cookie
        if cookie_str:
            user = await cookie2user(cookie_str)
            if user:
                logging.info('set current user: %s' % user.email)
                request.__user__ = user
        return await handler(request)
    return auth

最后附上登录页面的源码:


<html class="uk-height-1-1">
<head>
    <meta charset="utf-8" />
    <title>登录 - Awesome Python Webapptitle>
    <link rel="stylesheet" href="/static/css/uikit.min.css">
    <link rel="stylesheet" href="/static/css/uikit.gradient.min.css">
    <script src="/static/js/jquery.min.js">script>
    <script src="/static/js/sha1.min.js">script>
    <script src="/static/js/uikit.min.js">script>
    <script src="/static/js/vue.min.js">script>
    <script src="/static/js/awesome.js">script>
    <script>
$(function() {
    var vmAuth = new Vue({
        el: '#vm',
        data: {
            email: '',
            passwd: ''
        },
        methods: {
            submit: function(event) {
                event.preventDefault();
                var
                    $form = $('#vm'),
                    email = this.email.trim().toLowerCase(),
                    data = {
                        email: email,
                        passwd: this.passwd==='' ? '' : CryptoJS.SHA1(email + ':' + this.passwd).toString()
                    };
                $form.postJSON('/api/authenticate', data, function(err, result) {
                    if (! err) {
                        location.assign('/');
                    }
                });
            }
        }
    });
});
    script>
head>
<body class="uk-height-1-1">
    <div class="uk-vertical-align uk-text-center uk-height-1-1">
        <div class="uk-vertical-align-middle" style="width: 320px">
            <p><a href="/" class="uk-icon-button"><i class="uk-icon-html5">i>a> <a href="/">Awesome Python Webappa>p>
            <form id="vm" v-on="submit: submit" class="uk-panel uk-panel-box uk-form">
                <div class="uk-alert uk-alert-danger uk-hidden">div>
                <div class="uk-form-row">
                    <div class="uk-form-icon uk-width-1-1">
                        <i class="uk-icon-envelope-o">i>
                        <input v-model="email" name="email" type="text" placeholder="电子邮件" maxlength="50" class="uk-width-1-1 uk-form-large">
                    div>
                div>
                <div class="uk-form-row">
                    <div class="uk-form-icon uk-width-1-1">
                        <i class="uk-icon-lock">i>
                        <input v-model="passwd" name="passwd" type="password" placeholder="口令" maxlength="50" class="uk-width-1-1 uk-form-large">
                    div>
                div>
                <div class="uk-form-row">
                    <button type="submit" class="uk-width-1-1 uk-button uk-button-primary uk-button-large"><i class="uk-icon-sign-in">i> 登录button>
                div>
            form>
        div>
    div>
body>
html>

编辑完代码后,进入网站进行登录,登录过后返回首页,发现左上角的用户状态还是请登录状态,这时查看一下Jinja2父模板,发现原来用户状态还需自己导进去,这时在URL处理函数上加多一个request参数以及编写导进去的代码,例如:

@get('/')
def index(request):
    summary = 'Hello,World.'
    blogs = [
        Blog(id='1', name='Test Blog', summary=summary, create_at=time.time()-120),
        Blog(id='2', name='Something New', summary=summary, create_at=time.time()-3600),
        Blog(id='3', name='Learn Swift', summary=summary, create_at=time.time()-7200)
    ]
    return {
        '__template__': 'blogs.html',
        'blogs': blogs,
        '__user__': request.__user__
    }

转载于:https://www.cnblogs.com/AB786883603/p/8325438.html

你可能感兴趣的:(Python廖雪峰实战web开发(Day10(下)-用户登录))