做域名出租

from flask import Flask, request, jsonify
import datetime

app = Flask(__name__)

domains = {
    'example.com': {'owner': 'Alice', 'rented_by': None, 'expiry_date': None}
}

@app.route('/rent_domain', methods=['POST'])
def rent_domain():
    data = request.json
    domain_name = data['domain']
    renter = data['renter']
    duration = data['duration']

    if domain_name not in domains:
        return jsonify({'error': 'Domain not found'}), 404

    if domains[domain_name]['rented_by'] is not None:
        return jsonify({'error': 'Domain already rented'}), 400

    expiry_date = datetime.datetime.now() + datetime.timedelta(days=duration)
    domains[domain_name]['rented_by'] = renter
    domains[domain_name]['expiry_date'] = expiry_date

    return jsonify({'message': 'Domain rented successfully', 'expiry_date': expiry_date.isoformat()}), 200

@app.route('/check_domain', methods=['GET'])
def check_domain():
    domain_name = request.args.get('domain')
    if domain_name not in domains:
        return jsonify({'error': 'Domain not found'}), 404

    domain_info = domains[domain_name]
    return jsonify({
        'owner': domain_info['owner'],
        'rented_by': domain_info['rented_by'],
        'expiry_date': domain_info['expiry_date'].isoformat() if domain_info['expiry_date'] else None
    }), 200

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