from flask import Flask, request, Response, jsonify, abort
import json

app = Flask(__name__)

def main(query, k):
    """業(yè)務(wù)邏輯處理函數(shù)
    
    Args:
        query (str): 查詢文本
        k (int): 返回結(jié)果數(shù)量
    
    Returns:
        dict: 包含查詢參數(shù)和結(jié)果的字典
    """
    return {"query": query, "k": k, "status": "success"}

@app.route('/api/search', methods=['GET'])
def handle_get():
    """處理GET請求
    
    參數(shù)通過URL查詢字符串傳遞:
    - query: 必需，字符串類型
    - k: 可選，整數(shù)類型，默認(rèn)為1
    
    Returns:
        Response: JSON格式的響應(yīng)
    """
    # 獲取并驗證參數(shù)
    query = request.args.get("query")
    if not query:
        abort(400, description="query參數(shù)不能為空")
    
    try:
        k = int(request.args.get("k", 1))  # 默認(rèn)為1
    except ValueError:
        abort(400, description="k必須是整數(shù)")
    
    # 調(diào)用業(yè)務(wù)邏輯
    result = main(query, k)
    
    # 返回JSON響應(yīng)
    return jsonify(result)

import requests

# 構(gòu)造URL參數(shù)
params = {
    'query': 'Python Flask',
    'k': 3
}

# 發(fā)送GET請求
response = requests.get(
    url="http://127.0.0.1:8000/api/search",
    params=params
)

# 處理響應(yīng)
if response.status_code == 200:
    print(response.json())
else:
    print(f"請求失敗，狀態(tài)碼: {response.status_code}, 錯誤: {response.text}")

@app.route('/api/form-submit', methods=['POST'])
def handle_post_form():
    """處理表單POST請求
    
    參數(shù)通過表單數(shù)據(jù)傳遞:
    - query: 必需，字符串類型
    - k: 可選，整數(shù)類型，默認(rèn)為1
    
    Returns:
        Response: JSON格式的響應(yīng)
    """
    # 獲取并驗證參數(shù)
    query = request.form.get("query")
    if not query:
        abort(400, description="query參數(shù)不能為空")
    
    try:
        k = int(request.form.get("k", 1))  # 默認(rèn)為1
    except ValueError:
        abort(400, description="k必須是整數(shù)")
    
    # 調(diào)用業(yè)務(wù)邏輯
    result = main(query, k)
    
    # 返回JSON響應(yīng)
    return jsonify(result)

import requests

# 準(zhǔn)備表單數(shù)據(jù)
form_data = {
    'query': '表單提交示例',
    'k': 2
}

# 發(fā)送POST請求
response = requests.post(
    url="http://127.0.0.1:8000/api/form-submit",
    data=form_data
)

# 處理響應(yīng)
if response.status_code == 200:
    print(response.json())
else:
    print(f"請求失敗，狀態(tài)碼: {response.status_code}, 錯誤: {response.text}")

@app.route('/api/json-submit', methods=['POST'])
def handle_post_json():
    """處理JSON POST請求
    
    參數(shù)通過JSON格式傳遞:
    - query: 必需，字符串類型
    - k: 可選，整數(shù)類型，默認(rèn)為1
    
    Returns:
        Response: JSON格式的響應(yīng)
    """
    # 驗證請求內(nèi)容類型
    if not request.is_json:
        abort(415, description="請求必須是JSON格式")
    
    # 獲取JSON數(shù)據(jù)
    data = request.get_json()
    
    # 驗證參數(shù)
    query = data.get("query")
    if not query:
        abort(400, description="query參數(shù)不能為空")
    
    try:
        k = int(data.get("k", 1))  # 默認(rèn)為1
    except ValueError:
        abort(400, description="k必須是整數(shù)")
    
    # 調(diào)用業(yè)務(wù)邏輯
    result = main(query, k)
    
    # 返回JSON響應(yīng)
    return jsonify(result)

import requests

# 準(zhǔn)備JSON數(shù)據(jù)
json_data = {
    'query': 'JSON數(shù)據(jù)示例',
    'k': 4
}

# 發(fā)送POST請求
response = requests.post(
    url="http://127.0.0.1:8000/api/json-submit",
    json=json_data,  # 自動設(shè)置Content-Type為application/json
    headers={'Accept': 'application/json'}  # 明確要求JSON響應(yīng)
)

# 處理響應(yīng)
if response.status_code == 200:
    print(response.json())
else:
    print(f"請求失敗，狀態(tài)碼: {response.status_code}, 錯誤: {response.text}")

@app.errorhandler(400)
def bad_request(error):
    return jsonify({
        'status': 'error',
        'message': error.description
    }), 400

@app.errorhandler(404)
def not_found(error):
    return jsonify({
        'status': 'error',
        'message': '資源不存在'
    }), 404

@app.errorhandler(415)
def unsupported_media_type(error):
    return jsonify({
        'status': 'error',
        'message': error.description
    }), 415

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=8000, debug=True)