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

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

WBOY

发布： 2024-07-23 20:57:04

原创

576 人浏览过

How to Create a Simple URL Shortener Tool with Python

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 开发服务器将启动，您可以通过在 Web 浏览器中导航到 http://127.0.0.1:5000 来访问 URL 缩短工具。

第 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，可用于测试重定向功能。