短域名系统

from flask import Flask, request, redirect, jsonify
import string
import random

app = Flask(__name__)

# 存储长网址和短链接的映射关系
url_mapping = {}

def generate_short_url():
    characters = string.ascii_letters + string.digits
    short_url = ''.join(random.choice(characters) for _ in range(6))
    while short_url in url_mapping:
        short_url = ''.join(random.choice(characters) for _ in range(6))
    return short_url

@app.route('/shorten', methods=['POST'])
def shorten_url():
    long_url = request.json.get('long_url')
    if not long_url:
        return jsonify({'error': 'Long URL is required'}), 400
    
    short_url = generate_short_url()
    url_mapping[short_url] = long_url
    
    return jsonify({'short_url': short_url}), 200

@app.route('/<short_url>')
def redirect_to_long_url(short_url):
    long_url = url_mapping.get(short_url)
    if not long_url:
        return jsonify({'error': 'Short URL not found'}), 404
    
    return redirect(long_url, code=302)

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