在线短网址生成

from flask import Flask, request, redirect
import hashlib
import sqlite3

app = Flask(__name__)

def create_table():
    conn = sqlite3.connect('short_urls.db')
    c = conn.cursor()
    c.execute('''CREATE TABLE IF NOT EXISTS urls (short_url TEXT PRIMARY KEY, long_url TEXT)''')
    conn.commit()
    conn.close()

def generate_short_url(long_url):
    hash_object = hashlib.md5(long_url.encode())
    short_url = hash_object.hexdigest()[: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(long_url)
        conn = sqlite3.connect('short_urls.db')
        c = conn.cursor()
        c.execute("INSERT INTO urls (short_url, long_url) VALUES (?, ?)", (short_url, long_url))
        conn.commit()
        conn.close()
        return f'Your short URL is: http://yourdomain.com/{short_url}'
    return '''
        <form method="POST">
            Long URL: <input type="text" name="long_url"><br>
            <input type="submit" value="Generate Short URL">
        </form>
    '''

@app.route('/<short_url>')
def redirect_to_long_url(short_url):
    conn = sqlite3.connect('short_urls.db')
    c = conn.cursor()
    c.execute("SELECT long_url FROM urls WHERE short_url=?", (short_url,))
    long_url = c.fetchone()
    conn.close()
    if long_url:
        return redirect(long_url[0])
    else:
        return "Short URL not found", 404

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