python基础 接口

#pip install flask
#flask 开源的web框架，可以启动服务
#mock服务
import flask,json
from day_7.my_pysql import my_db
server=flask.Flask(__name__)#把当前这个python文件当作一个服务
@server.route('/login',methods=['get','post'])#装饰器，加上此装饰器就变成了接口
def denglu():
    username=flask.request.values.get('username')#从客户端传过来的数据
    pwd=flask.request.values.get('password')
    #json_data=flask.request.json.get('username')#获取入参是json类型的
    # cookies_data=flask.request.cookies.get('xxx')#获取cookies
    # header_data=flask.request.headers.get('xxx')#获取header
    if username=='jing' and pwd=='123456':
        res={'code':1000,'msg':'登录成功','sign':'sdfnfk38fnfd'}
    else:
        res={'code':2000,'msg':'账号或密码不正确'}
    return json.dumps(res,ensure_ascii=False)
@server.route('/reg',methods=['get','post'])
def reg():
    username=flask.request.values.get('username')
    pwd=flask.request.values.get('password')
    cpwd=flask.request.values.get('cpassword')
    if username and pwd and cpwd:
        sql="select * from nhy where name='%s';"%username
        sql1="insert into nhy(name,password)values('%s','%s');"%(username,pwd)
        if my_db(sql):
            data={'code':100,'msg':'username+"账号已存在"'}
        elif pwd!=cpwd:
            data={'code':220,'msg':'账号密码不一致'}
        else:
            my_db(sql1)
            data={'code':3000,'msg':'注册成功'}
    else:
        data={'code':400,'msg':'username,password and cpassword不能为空'}
    return json.dumps(data,ensure_ascii=False)
# server.run(port=8989,debug=True)#启动服务，括号里也可以指定host,port
#host=‘0.0.0.0’就可以在同一个局域网里，别人拿你的ip地址去访问了
#debug=True调试模式
#端口号不能重复，所以不能每次都重新运行
#http://127.0.0.1:8989/login?username=jing&password=123456


my_pysql.py

import pymysql
def my_db(sql,port=3306,charset='utf8'):
    ip,user,password,db='118.24.3','jxz','11111','jxz'
    coon=pymysql.connect(host=ip,port=port,user=user,password=password,
                         db=db,autocommit=True)
    cur=coon.cursor()
    sql=sql.strip()
    cur.execute(sql)
    sql_start=sql[:6].lower()#取sql的开头，转成小写
    if sql_start.startswith('select') or sql_start.startswith('show'):
        data=cur.fetchall()
    else:
        data='ok'
    cur.close()
    coon.close()
    return data