Python学习之==>有依赖关系的接口开发

一、接口需求

1、登录接口

　　（1）登录成功后将session信息存入redis数据库并设置失效时间为600秒

　　（2）构造返回结果的对象flask.make_response()

　　（3）产生cookie，失效时间同样设定为600秒

2、发帖接口

　　（1）根据登录接口成功写入的cookie来判断用户是否登录

　　（2）判断用户传过来的session和redis数据库中存入的session是否一致

　　（3）如果一致的话则进行发帖操作

二、代码实现

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

 @server.route('/login')
 def login():
     username = flask.request.values.get('username')
     pwd = flask.request.values.get('pwd')
     if username == 'luolei' and pwd == '':
         session_id = my_md5(username+time.strftime('%Y%m%d%H%M%S'))
         key = 'session:%s'%username
         op_redis(key,session_id,600)
         res = {'sessionid':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)#600是cookie的失效时间
         return res

 # 有依赖关系的接口
 # 根据上面登录成功写入的cookie来判断用户是否登录，然后进行发帖操作
 @server.route('/posts')
 def posts():
     username = ''
     session = '' # 定义这两个变量是为了在没有传cookie的时候用的
     cookies = flask.request.cookies # 获取所有的cookie
     for key,value in cookies.items():
         if key.startswith('session:'): # 判断cookie是否以session开头
             username = key
             session = value # 调用接口的时候用户传过来的session，从cookie中获取到的
     redis_session = op_redis(username) # 从redis中获取到的cookie
     if redis_session == session: # 判断用户传过来的session和redis里面的session是否一样
         title = flask.request.values.get('title')
         content = flask.request.values.get('content')
         article_key = 'article_key:%s'%title
         op_redis(article_key,content) # 把文章写入redis
         res = {'msg':'文章发表成功','code':0000}
     else:
         res = {'msg':'用户未登录','code':2001}
     return json.dumps(res,ensure_ascii=False)

 三、项目添加环境变量

 import sys,os
 os.path.abspath(__file__) # 获取到当前文件的绝对路径
 os.path.dirname(__file__) # 获取父目录，获取它的上一级目录
 BASE_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) # 获取到当前程序的主目录
 sys.path.insert(0,BASE_PATH) # 将程序主目录加入python环境变量