如何使用 Python 创建简单的 URL 缩短工具-Python教程-PHP中文网

如何使用 Python 创建简单的 URL 缩短工具

WBOY

发布： 2024-07-23 12:30:01

转载

861人浏览过

如何使用 python 创建简单的 url 缩短工具

url 缩短工具允许用户将长 url 转换为更短、更易于管理的链接。我们可以使用 python 和 flask（一个轻量级 web 框架）构建该工具的简单版本。

先决条件

开始之前，请确保您具备以下条件：

您的系统上已安装python（推荐python 3.6+）。
flask 安装完毕。您可以使用 pip 安装 flask。

第 1 步：设置您的环境

首先需要创建虚拟环境并安装flask。打开终端并运行以下命令：

# create a virtual environment
python -m venv myenv

# activate the virtual environment
# on windows
myenv\scripts\activate
# on macos/linux
source myenv/bin/activate

# install flask
pip install flask

登录后复制

第 2 步：编写代码

为您的项目创建一个目录，例如 url_shortener，并在其中创建一个名为 app.py 的 python 文件。我们将把代码分成几个部分。

导入所需的库

from flask import flask, request, redirect, url_for
import string
import 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，可用于测试重定向功能。