python学习笔记19：mock接口-flask模块

mock接口就是模拟接口的意思。在接口测试中，一个接口可能依赖另一个接口的功能。模拟第三方数据交互的接口，当第三方的接口没有提供之前，我们自己写mock接口和自己的系统数据对接测试，如果不想让别人连接数据库，可以mock接口，让别人调这个接口看数据。例：测试支付功能接口，而它需要调用第三方服务的接口，如微信支付或支付宝支付接口，因此，可以开发mock接口进行模拟支付时调用第三方接口。

一、安装第三方模块flask

安装命令为：

pip install flask

注：flask是一个轻量级的web开发框架

二、开发接口

1.实例化一个服务server:flask.Flask(__name__)
2.函数前加装饰器@server.route(), 将函数变为一个接口
3.启动服务server:server.run()

import flask
import json
import pymysql
server = flask.Flask(__name__)  # __name__代表当前python文件, 把当前python文件当成一个服务

# 连接数据库
def op_mysql(sql):
    connect=pymysql.connect(host='127.0.0.1',
                    user='xxx',
                    password='123456',
                    db='xxx',
                    port=3306,
                    charset='utf8',
                    autocommit=True
                    )
    cursor = connect.cursor(pymysql.cursors.DictCursor)
    try:
        cursor.execute(sql)
    except Exception as e:
        result = {'error':'sql错误'}
    else:
        result = cursor.fetchall()
    finally:
        cursor.close()
        connect.close()
    return result

# 第一种：get请求
@server.route('/api/login')  #装饰器，函数就不是普通的函数，变成接口
# route里面第一个参数是路径名，第二个参数是请求类型，必须用list这种方式，methods默认不写就是get请求
def login():
    data ={"code":0,"msg":"登录成功"}
    return json.dumps(data,ensure_ascii=False)

# 第二种：获取请求参数
@server.route('/api/payment')
def payment():
    status = flask.request.values.get('status')  #获取请求参数
    if status == 'success':
        data ={"code":0,"msg":"支付成功","amount":30000}
    elif status == 'process':
        data = {"code": 305, "msg": "支付处理中", "amount": 30000}
    elif status == 'fail':
        data = {"code": -1, "msg": "失败", "amount": 30000}
    else:
        data = {"code": 400, "msg": "支付状态错误"}
    return json.dumps(data,ensure_ascii=False)

# 第三种：数据库操作
@server.route('/account_data',methods=['post'])
def account_data():
    account_name = flask.request.values.get('account_name')  #获取json
    result = op_mysql('select * from gtm_account where account="%s";'%account_name)
    data = {"code":0,"msg":"success","data":result}
    return json.dumps(data,ensure_ascii=False)

#启动服务： 必须在所有接口定义完后再定义，否则检测不到之后定义的接口, 接口是无法被运行的
server.run(host='0.0.0.0',port=9999,debug=True)  
# host=0.0.0.0同一局域网内写上你真实的IP就可以访问
# debug=True是自动重启的意思，修改代码后服务会重启

# server.run(port=9999,debug=True)  #这种启动方式只能本地IP访问：127.0.0.1

 三、小练习

import flask,json,pymysql,hashlib,redis,time
server = flask.Flask(__name__)

def op_mysql(sql,one_tag):
    connect=pymysql.connect(host='127.0.0.1',
                    user='xxx',
                    password='123456',
                    db='xxx',
                    port=3306,
                    charset='utf8',
                    autocommit=True
                    )
    cursor = connect.cursor(pymysql.cursors.DictCursor)
    cursor.execute(sql)
    if one_tag:
        result = cursor.fetchone()
    else:
        result = cursor.fetchall()
    cursor.close()
    connect.close()
    return result

def md5(s,salt='zx_session'):
    s = str(s)+salt  #加盐
    m = hashlib.md5(s.encode())  #加密
    return m.hexdigest()

def op_redis(key,value=None,expire=60*60*2):
    r = redis.Redis(host='127.0.0.1',password='123456',decode_responses=True)
    if value:
        r.set(key,value,expire)
    else:
        return r.get(key)

# 注册接口
@server.route('/register',methods=['post'])
def register(): #app_myuser
    username = flask.request.values.get('username')
    password = flask.request.values.get('password')
    cpwd = flask.request.values.get('cpwd')
    sql = 'select * from app_myuser where username ="%s";'%username
    if username and password and cpwd:
        if password !=cpwd:
            data = {'code': 400, 'msg': '两次输入密码不一致'}
        elif op_mysql(sql):
            data = {'code':401,'msg':'用户已经存在'}
        else:
            password = md5(password)
            insert_sql = 'insert into app_myuser (username,passwd) values ("%s","%s");'%(username,password)
            op_mysql(insert_sql)
            data = {'code':0,'msg':'注册成功'}
    else:
        data = {'code':400,'msg':'必填参数不能为空'}
    return json.dumps(data,ensure_ascii=False)

# 登录接口
@server.route('/login',methods=['post'])
def login():
    username = flask.request.values.get('username')
    password = flask.request.values.get('password')
    if username and password:
        sql = 'select * from app_myuser where username ="%s";' % username
        result = op_mysql(sql,True)
        if result:
            if md5(password) == result.get('passwd'):
                token = md5(username + str(time.time()))  #获取token：用户名+时间戳
                info = {'username':username,'id':result.get('id')}
                op_redis(token,json.dumps(info))  #token存到redis里
                data = {'code':0,'msg':'登录成功','token':token}
            else:
                data = {'code':403,'msg':'账号/密码错误'}
        else:
            data = {'code': 401, 'msg': '用户不存在'}
    else:
        data = {'code': 400, 'msg': '必填参数不能为空'}
    return json.dumps(data,ensure_ascii=False)

# 支付接口
@server.route('/api/payment',methods=['get'])
def payment():
    token = flask.request.values.get('token')
    amount = flask.request.values.get('amount')
    if token:
        result = op_redis(token)
        if result:
            amount = float(amount)
            result = json.loads(result)
            userid = result.get('id')
            sql = 'update app_myuser set balance=balance-%s where id = %s;'%(amount,userid)
            op_mysql(sql)
            data = {'code':0,'msg':'支付成功'}
        else:
            data = {'code': 401, 'msg': '未登录'}
    else:
        data = {'code':401,'msg':'未登录'}
    return json.dumps(data,ensure_ascii=False)

server.run(port=9999,debug=True)