如何使用 Python 创建简单的 URL 缩短工具
url 缩短工具允许用户将长 url 转换为更短、更易于管理的链接。我们可以使用 python 和 flask(一个轻量级 web 框架)构建该工具的简单版本。
先决条件
开始之前,请确保您具备以下条件:
第 1 步:设置您的环境
首先需要创建虚拟环境并安装flask。打开终端并运行以下命令:
# create a virtual environmentpython -m venv myenv# activate the virtual environment# on windowsmyenvscriptsctivate# on macos/linuxsource myenv/bin/activate# install flaskpip install flask
第 2 步:编写代码
为您的项目创建一个目录,例如 url_shortener,并在其中创建一个名为 app.py 的 python 文件。我们将把代码分成几个部分。
导入所需的库
from flask import flask, request, redirect, url_forimport stringimport random
设置 flask 应用程序和数据库
我们将使用一个简单的内存字典来存储 url 映射。为了获得更强大的解决方案,您可以使用 sqlite 或 postgresql 等数据库。
app = flask(__name__)url_mapping = {}
生成短网址的函数
我们将创建一个生成短 url 密钥的函数。
def generate_short_url(): characters = string.ascii_letters + string.digits short_url = ''.join(random.choice(characters) for _ in range(6)) return short_url
处理 url 缩短的路由
此路由将从用户处获取长 url 并返回缩短的版本。
@app.route('/shorten', methods=['post'])def shorten_url(): original_url = request.form['url'] short_url = generate_short_url() url_mapping[short_url] = original_url return f'shortened url: {request.host_url}{short_url}'
重定向短 url 的路由
此路由会将用户从短网址重定向到原始长网址。
@app.route('/<short_url>')def redirect_url(short_url): original_url = url_mapping.get(short_url) if original_url: return redirect(original_url) else: return 'url not found', 404
运行应用程序的主要功能
if __name__ == "__main__": app.run(debug=true)
第 3 步:运行该工具
保存您的 app.py 文件并从终端运行它:
python app.py
flask 开发服务器将启动,您可以通过在网络浏览器中导航到 http://127.0.0.1:5000 来访问 url 缩短工具。
立即学习“Python免费学习笔记(深入)”;
第 4 步:测试工具
要缩短 url,您可以使用curl 等工具或简单的 html 表单。这是用于测试的基本 html 表单:
<!DOCTYPE html><html><head> <title>URL Shortener</title></head><body> <form action="http://127.0.0.1:5000/shorten" method="post"> <label for="url">Enter URL to shorten:</label> <input type="text" id="url" name="url"> <input type="submit" value="Shorten"> </form></body></html>
将此 html 代码保存为 index.html 并在网络浏览器中打开它。输入要缩短的 url 并提交表单。您应该会看到一个缩短的 url,可用于测试重定向功能。
结论
按照以下步骤,您可以使用 python 和 flask 创建一个简单的 url 缩短工具。
可以通过用户身份验证、分析和用于持久存储 url 映射的数据库等附加功能进一步增强此工具。
想了解更多吗?在 zerobytecode 上探索编程文章、提示和技巧。