python简说（二十一）开发接口

一.flask举例

import flask
server = flask.Flask(__name__)
#新建一个服务，把当前这个python文件当做一个服务
@server.route('/login',methods=['get'])
def hello():
    uname = flask.request.values.get('username')
    pd = flask.request.values.get('passwd')
    sql = 'select * from app_myuser where username="%s"'%uname
    res = tools.my_db(sql)
    if res:
        if tools.my_md5(pd) == res.get('passwd'):
            res = {"code":0,"msg":"登录成功！"}
        else:
            res = {"code":1,"msg":"密码错误！"}
    else:
        res = {'code':2,"msg":"用户不存在"}
    return json.dumps(res,ensure_ascii=False,indent=4)
server.run()
二.按条件查询

@server.route('/api/stu')
def get_stu():
    username = flask.request.values.get('name')#默认get不到的话，返回的值就是None
    age = flask.request.values.get('age')
    if username and age:
        sql = "select * from app_student where name='%s' and age='%s'" % (username, age)
    elif not username and age:
        sql = "select * from app_student where age='%s'" % age
    elif username and not age:
        sql = "select * from app_student where name='%s'" % username
    else:
        sql = "select * from app_student"
    res = tools.my_db2(sql)
    return json.dumps(res,ensure_ascii=False,indent=4)

三.带session的接口

@server.route('/api/login',methods=['get','post'])
def login():
    uname = flask.request.values.get('username')
    pd = flask.request.values.get('passwd')
    sql = 'select * from app_myuser where username="%s"'%uname
    log.debug('登录执行的sql是：%s'%sql)
    res = tools.my_db(sql)
    if res:
        if tools.my_md5(pd) == res.get('passwd'):
            uname_md5 = tools.my_md5(uname+str(time.time()))
            # uname_md5 = tools.my_md5(uname)
            key='session:%s'%(uname_md5)
            user_info = {'user_id':res.get('id'),'user_name':uname}
            r = tools.get_redis()
            log.debug('连接redis成功')
            r.set(key,json.dumps(user_info),setting.SESSION_EXPIRE)
            res = {"code":0,"msg":"登录成功！",'session_id':uname_md5,
                   'login_time':time.strftime('%Y-%m-%d %H:%M:%S')}
            log.warning('%s 用户登陆成功'%uname)
        else:
            res = {"code":1,"msg":"密码错误！"}
    else:
        log.warning('%s用户不存在'%uname)
        res = {'code':2,"msg":"用户不存在"}
    return json.dumps(res,ensure_ascii=False)
四.入参是json

@server.route('/api/add_stu',methods=['post'])
def add_stu():
    if flask.request.is_json:
        session_id = flask.request.json.get('session_id')
        name= flask.request.json.get('name')
        phone = flask.request.json.get('phone')
        grade = flask.request.json.get('grade')
        addr = flask.request.json.get('addr','北京')
        sex = flask.request.json.get('sex','男')
        age = flask.request.json.get('age',18)
        gold = flask.request.json.get('gold',1000)
        if session_id and name and phone and grade:
            login_tag , data = check_session(session_id)
            if login_tag:
                insert_sql="insert into `jxz`.`app_student` ( `name`, `sex`," \
                           " `age`, `addr`, `grade`, `phone`, `gold`)" \
                           " values ( '%s', '%s', '%s', '%s', '%s', '%s', '%s');"%(name,sex,age,addr,grade,phone,gold)
                tools.my_db(insert_sql)
                res = {'code':0,'msg':'学生信息添加成功'}
            else:
                res = data

        else:
            res = {'code':3,'msg':'必填参数未填，请查看接口文档！'}
    else:
        res = {'code':5,'msg':'入参是json类型的！'}
    return json.dumps(res, ensure_ascii=False)
五.上传文件

@server.route('/api/upload',methods=['post'])
def upload():
    session_id = flask.request.values.get('session_id')
    file = flask.request.files.get('file_name')
    if session_id and file:
        login_tag, data = check_session(session_id)
        if login_tag:
            new_file_name = '%s_%s'%(int(time.time()),file.filename)
            #防止文件名重复，所以把文件名前面加上了时间戳
            abs_file_path = os.path.join(setting.UPLOAD_DIR,new_file_name)
            #拼文件的绝对路径，这样子就放到了uploads目录下
            file.save(abs_file_path)#保存文件
            res = {'code':0,'msg':'上传文件成功！'}
        else:
            res = data
    else:
        res = {'code':3,'msg':'必填参数未填，请查看接口文档！'}
    return json.dumps(res, ensure_ascii=False)