【Python】学习笔记5-利用flask来mock接口

# 1、mock接口
目的：

mock|muk|接口，就是模拟接口
1、暂时代替第三方接口测试，自己模拟一个接口
2、辅助测试，用来代替没有开发好的接口
3、查看数据

# import flask #python的轻量级的开发框架
# # 接口，后台服务的开发
# # 在浏览器运行http://127.0.0.1:8080/get_user即可，或者其他访问接口的方式
# server = flask.Flask(__name__) #__name__当前文件名，把咱们这个app python当做一个server
# @server.route('/get_user',methods=['get','post']) #将下面函数变成一个接口
# def get_all_user():
#     return 'cm的接口' #return只能返回字符串
# #启动服务,加debug自动帮忙重启
# server.run(port = 8080,debug = True)


# 2、调用自己的模块，连接数据库，浏览器请求接口
# import tools #使用时，需要用tools.op_mysql()
from lib.tools import op_mysql  #使用时，直接用op_mysql()
# 以上两种方式都可以

3、写接口前先要形成标准的目录结构，分解一下各个目录的文件

  4、start.py运行的启动程序，在这里启动 服务

#这是程序的入口文件，将启动服务的命令放里面

#增加根目录为环境变量，方便底层目录执行时目录错误
import sys,os
BASE_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) #取到api目录,根目录
sys.path.insert(0,BASE_PATH)

#加环境变量才能导入，否则导入会异常
from lib.main  import server
server.run(port = 8080,host = '0.0.0.0',debug = True) #默认端口号是5000
#host = '0.0.0.0' 代表局域网内别人都可以通ip访问自己的接口

5、setting 设置常规变量等

MYSQL_HOST = 'X.X.X.X'
PASSWORD = '123456'
PORT = 3306
USER = 'jxz'
DB = 'jxz'

6、main.py......自己的主程序

# import tools #使用时，需要用tools.op_mysql()
from lib.tools import op_mysql  #使用时，直接用op_mysql()
# 以上两种方式都可以

import flask,json #python的轻量级的开发框架
# 接口，后台服务的开发
# 在浏览器运行http://127.0.0.1:8080/get_user即可，或者其他访问接口的方式
server = flask.Flask(__name__) #__name__当前文件名，把咱们这个app python当做一个server


  #浏览器输出bt_stu表中前5条的数据
@server.route('/get_user',methods=['get','post']) #将下面函数变成一个接口
def get_all_user():
    sql = 'select * from bt_stu limit 5;'
    res = op_mysql(sql = sql)
    response = json.dumps(res,ensure_ascii=False)
    return response #return只能返回字符串


# 浏览器输入用户id、姓名,插入数据到数据库stu表，
@server.route('/add_user',methods=['post']) #代码支持什么类型的接口
def add_user():
    user_id = flask.request.values.get('id')
    user_name = flask.request.values.get('u')
    sql = "insert into stu values ('%s','%s');"%(user_id,user_name)
    if user_id and user_name:
        res = op_mysql(sql = sql)
        response = {"code":"308","message":"success"}
    else:
        response = {"code":"503","message":"参数不能为空"}
    return json.dumps(response,ensure_ascii=False) #return只能返回字符串

7、tools工具类文件，相关的函数在这里面

import pymysql
from conf import setting

# 1、操作mysql
def op_mysql(sql):
    conn = pymysql.connect(host=setting.MYSQL_HOST,user= setting.USER,
                           password=setting.PASSWORD,
                           port=setting.PORT,
                           charset = 'utf8',
                           db = setting.DB)
    cur = conn.cursor(cursor=pymysql.cursors.DictCursor)
    #判断是否需要commit，根据select update delete insert的类型
    cur.execute(sql)
    sql_start = sql[:6].upper()# select不区分大小写,取前6位转换成大写
    if sql_start == 'SELECT':
        res = cur.fetchall()
    else:
        conn.commit()
        res = 'ok'
    cur.close()
    conn.close()
    return res