# -*- coding: utf-8 -*-
from flask import Flask, jsonify, request
from data import *

# 初始化 Flask 应用
app = Flask(__name__)

# 初始化数据库和表
initialize_database()
initialize_table()

@app.route('/')
def home():
    return "Welcome to the Fruit API!"

# 添加水果数据
@app.route('/add_fruit', methods=['POST'])
def add_fruit():
    data = request.json
    name = data.get('name')
    origin = data.get('origin')

    if not name or not origin:
        return jsonify({"error": "Missing name or origin"}), 400

    try:
        insert_data(name, origin)
        return jsonify({"message": f"Fruit {name} from {origin} added successfully!"}), 200
    except Exception as e:
        return jsonify({"error": str(e)}), 500

# 获取所有水果数据
@app.route('/fruits', methods=['GET'])
def get_fruits():
    try:
        fruits = fetch_data()
        return jsonify({"fruits": fruits})
    except Exception as e:
        return jsonify({"error": str(e)}), 500

@app.route('/delete_fruit', methods=['POST'])
def delete_fruit():
    data = request.json
    name = data.get('name')

    if not name:
        return jsonify({"error": "Missing name"}), 400

    try:
        delete_data(name)
        return jsonify({"message": f"Fruit {name} deleted successfully!"}), 200
    except Exception as e:
        return jsonify({"error": str(e)}), 500


# 运行 Flask 应用
if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, debug=True)