接口测试学习-python第六课（登录判断后存入数据）

很多时候，一个接口的功能实现是需要前提条件的。比如推送数据需要先登录系统等。这里用一个文章发布接口来进行示例。想要通过接口发表文章，首先需要登录，要如何判断这个账号已经登录了呢。登录的账号可能有多个，总不能每次都验证不同的数据吧。这时候就可以通过登录接口存储一些登录信息，然后在文章发表接口时判断登录信息是否符合且有效，就可以判断该账号是登录状态的。这样不论登录的是哪一个账号，文章发票接口都能顺利验证。

import flask,time,json
from lib import tools
server = flask.Flask(__name__)


@server.route('/login', methods=['get'])
def login():
    username = flask.request.values.get('username')
    pwd = flask.request.values.get('pwd')
    if username == 'gyx' and pwd=='123456':
        session_id = tools.my_md5(username+time.strftime('%Y%m%d%H%M%S'))
        key = 'txz_session:%s'%username
        tools.op_redis(key,session_id,3600)
        res = {'session_id':session_id,'error_code':0,'msg':'登录成功',
               'login_time':time.strftime('%Y%m%d%H%M%S') }
        json_res = json.dumps(res, ensure_ascii=False)  # 返回结果弄成一个json
        res = flask.make_response(json_res)  # 构造成一个对象
        res.set_cookie(key, session_id, 600)
        return res



@server.route('/posts', methods=['get'])
def posts():
    cookies = flask.request.cookies  # 获取到所有的cookie
    username = ''  # 定义username和session是为了在没有传cookie的时候用的
    session = ''
    for key,value in cookies.items():
        if key.startswith('txz_session'):
            username = key
            session = value  # 调用接口时得到的session
    redis_session = tools.op_redis(username)  # 从redis里面获取到cookie
    if redis_session == session:  # 判断是否登录成功
        title = flask.request.values.get('title')  # 获取文章标题
        content = flask.request.values.get('content')  #获取文章内容
        article_key = 'article:%s'%title
        tools.op_redis(article_key,content)  # 登录成功就存入文章标题和内容
        res = {'msg':'文章发表成功', 'msg_code': 0}
    else:
        res = {'msg':'用户未登录','msg_code':2009}
    return res