缩短连接防域名

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

app = Flask(__name__)

# 存储长网址和短网址的映射
url_map = {}

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_map:
        short_url = ''.join(random.choice(characters) for _ in range(6))
    return short_url

@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        long_url = request.form['long_url']
        short_url = generate_short_url()
        url_map[short_url] = long_url
        return render_template('result.html', short_url=short_url)
    return render_template('index.html')

@app.route('/<short_url>')
def redirect_to_long_url(short_url):
    if short_url in url_map:
        return redirect(url_map[short_url])
    else:
        return "Short URL not found", 404

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