from flask import Flask, jsonify, request

app = Flask(__name__)

# 销售情况历史数据
salesInfo = [
    {
        "id": 1,
        "year": "2024",
        "month": "01",
        "xse": 200000,
        "tbzz": 23,
        "ml": 12,
        "hb": 60000,
        "db": 120000,
        "hd": 20000
    },
    {
        "id": 2,
        "year": "2024",
        "month": "02",
        "xse": 160000,
        "tbzz": 6,
        "ml": 6,
        "hb": 60000,
        "db": 50000,
        "hd": 50000
    },
    {
        "id": 3,
        "year": "2024",
        "month": "03",
        "xse": 320000,
        "tbzz": 43,
        "ml": 23,
        "hb": 160000,
        "db": 120000,
        "hd": 40000
    },
    {
        "id": 4,
        "year": "2024",
        "month": "04",
        "xse": 450000,
        "tbzz": 67,
        "ml": 44,
        "hb": 260000,
        "db": 120000,
        "hd": 70000
    },
    {
        "id": 5,
        "year": "2024",
        "month": "05",
        "xse": 340000,
        "tbzz": 23,
        "ml": 12,
        "hb": 80000,
        "db": 220000,
        "hd": 40000
    },
    {
        "id": 6,
        "year": "2024",
        "month": "06",
        "xse": 120000,
        "tbzz": 3,
        "ml": 2,
        "hb": 60000,
        "db": 40000,
        "hd": 20000
    },
    {
        "id": 7,
        "year": "2024",
        "month": "07",
        "xse": 670000,
        "tbzz": 54,
        "ml": 56,
        "hb": 260000,
        "db": 220000,
        "hd": 190000
    },
    {
        "id": 8,
        "year": "2024",
        "month": "08",
        "xse": 150000,
        "tbzz": 4,
        "ml": 12,
        "hb": 60000,
        "db": 20000,
        "hd": 70000
    },
    {
        "id": 9,
        "year": "2024",
        "month": "09",
        "xse": 340000,
        "tbzz": 22,
        "ml": 11,
        "hb": 160000,
        "db": 120000,
        "hd": 60000
    },
    {
        "id": 10,
        "year": "2024",
        "month": "10",
        "xse": 560000,
        "tbzz": 33,
        "ml": 12,
        "hb": 160000,
        "db": 120000,
        "hd": 180000
    }
]

@app.route('/')
def hello_world():  # put application's code here
    return 'Hello World!'

@app.route('/sales', methods=['GET'])
def get_sales():
    return jsonify(salesInfo)

@app.route('/sale/<int:year>/<int:month>', methods=['GET'])
def query_sale(year, month):
    sale = next((s for s in salesInfo if (s['year']==str(year) and s['month']=='0'+str(month))), None)
    if sale:
        return jsonify(sale)
    else:
        return jsonify({'error': 'Sale not found'}), 404

if __name__ == '__main__':
    app.run()