python之接口開發

#注冊接口
import flask
from flask import request #想獲取到請求參數的話，就得用這個
server = flask.Flask(__name__) #把這個python文件當做一個web服務
def md5_passwd(str,salt='123456'):
    #satl是鹽值，默認是123456
    str=str+salt
    import hashlib
    md = hashlib.md5()  # 構造一個md5對象
    md.update(str.encode())
    res = md.hexdigest()
    return res

def conn_mysql(sql):
    import pymysql
    conn = pymysql.connect(host='211.149.218.16',user='jxz',password='123456',db='jxz',charset='utf8')
    cur = conn.cursor(cursor=pymysql.cursors.DictCursor)
    cur.execute(sql)
    res = cur.fetchone()
    conn.commit()
    cur.close()
    conn.close()
    return res
@server.route('/register',methods=['get','post'])#router里面第一個參數，是接口的路徑
def reg():
    # username = request.values.get('username')#這里就是你調用接口的是傳入的參數
    # password = request.values.get('password')#這里就是你調用接口的是傳入的參數
    print(request.json)
    username = request.json.get("username")#入參類型是json的話，那么必須得用.json方法才能獲取到數據
    password = request.json.get("password")
    if username and password:
        sql = 'select username,password from user where username="%s";'%username
        res = conn_mysql(sql)#執行sql
        if res:
            return '{"code":300,"msg":"你注冊的用戶已經存在"}'
        else:
            password = md5_passwd(password)#調用加密的函數
            sql = 'insert into user  (username,password) values ("%s","%s");'%(username,password)
            conn_mysql(sql)
            return '{"code":200,"msg":"注冊成功！"}'
    else:
        return '{"code":938,"msg":"必填參數未填，請看接口文檔！"}'

@server.route('/login')
def login():

    return '{"msg":"登錄成功"}'

server.run(port=8000,debug=True,host='0.0.0.0')