使用Python Flask构建高效简洁的URL缩短服务
一、设置 Flask 应用程序
首先,为您的项目创建一个新目录并在该目录中打开一个终端。然后,运行以下命令为您的项目创建一个新的虚拟环境:
# For windows:virtualenv venv# For linux:python3 -m venv venv通过运行以下命令激活虚拟环境:
# For windows
venv/Scripts/activate
# For linux
source venv/bin/activate
接下来,通过运行以下命令安装 Flask:
pip install Flask
立即学习“Python免费学习笔记(深入)”;
在您的项目目录中创建一个名为的新文件app.py并添加以下代码以设置基本的 Flask 应用程序:
from flask import Flask, request, redirectapp = Flask(__name__)@app.route('/')def index(): return 'Welcome to the URL Shortener'if __name__ == '__main__': app.run(debug=True)
运行以下命令启动 Flask 开发服务器:
python app.py
在您的网络浏览器中访问http://localhost:5000以查看“欢迎使用 URL 缩短器”消息。
二、将 URL 存储在字典中
接下来,我们会将原始 URL 及其相应的缩短 URL 存储在 Python 字典中。将以下代码添加到您的app.py文件中:
url_map = {}@app.route('/shorten', methods=['POST'])def shorten_url(): original_url = request.form['url'] short_url = generate_short_url(original_url) url_map[short_url] = original_url return short_urldef generate_short_url(original_url): # Generate a unique short URL for the original URL # (We'll implement this in the next step) pass
三、生成唯一的短 URL
要生成唯一的短 URL,我们将使用哈希函数。哈希函数接受输入(在本例中为原始 URL)并返回固定长度的输出(短 URL)。我们将使用 SHA-1 哈希函数,它是 Pythonhashlib库的一部分。将以下代码添加到您的app.py文件中:
import hashlibdef generate_short_url(original_url): hash = hashlib.sha1(original_url.encode()) short_url = hash.hexdigest()[:8] return short_url
四、重定向到原始 URL
最后,我们将添加一个路由,以便在用户访问缩短的 URL 时将用户重定向到原始 URL。将以下代码添加到您的app.py文件中:
@app.route('/<short_url>')def redirect_url(short_url): original_url = url_map.get(short_url) if original_url: return redirect(original_url</short_url>