一个简单的短链API程序

以下是一个使用 FastAPI 和 SQLite 实现的简单短链 API 程序：

🔗 功能说明：

通过 POST 请求访问 /u?url=原网址，可以创建一个短链（如 /u/abc123）。
通过 GET 请求访问 /u/短码（如 /u/abc123），可以跳转到原网址。

📦 项目结构（简单版，所有代码在一个文件中）：

你可以将以下代码保存为 main.py，然后运行：

pip install fastapi uvicorn

然后运行：

uvicorn main:app –reload

📄 main.py：

from fastapi import FastAPI, Request, HTTPException, Query
from fastapi.responses import RedirectResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from pydantic import BaseModel
import sqlite3
import secrets
import string

app = FastAPI()

# SQLite 数据库初始化
DB_NAME = "shortener.db"

def init_db():
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS urls (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            short_code TEXT UNIQUE NOT NULL,
            original_url TEXT NOT NULL
        )
    ''')
    conn.commit()
    conn.close()

init_db()

# 生成随机短码，比如 6 位字母数字组合
def generate_short_code(length=6):
    chars = string.ascii_letters + string.digits
    return ''.join(secrets.choice(chars) for _ in range(length))

# 短链创建接口：POST /u?url=原网址 或 GET /u?url=原网址（推荐用 POST，但为了简单，这里用 GET 参数也支持）
@app.get("/u/")
async def create_short_url(url: str = Query(..., description="要缩短的原始 URL")):
    if not url.startswith(('http://', 'https://')):
        url = 'http://' + url  # 自动补全协议

    short_code = generate_short_code()
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    try:
        cursor.execute("INSERT INTO urls (short_code, original_url) VALUES (?, ?)", (short_code, url))
        conn.commit()
    except sqlite3.IntegrityError:
        # 如果短码冲突（极小概率），重新生成
        conn.close()
        return await create_short_url(url)
    conn.close()

    short_url = f"http://localhost:8000/u/{short_code}"  # 假设你的服务运行在本地 8000 端口
    return {"original_url": url, "short_url": short_url, "short_code": short_code}

# 短链访问接口：GET /u/{short_code} --> 重定向到原网址
@app.get("/u/{short_code}")
async def redirect_to_original(short_code: str):
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute("SELECT original_url FROM urls WHERE short_code = ?", (short_code,))
    result = cursor.fetchone()
    conn.close()

    if result:
        original_url = result[0]
        return RedirectResponse(url=original_url)
    else:
        raise HTTPException(status_code=404, detail="短链不存在")

# 可选：一个简单的界面来展示创建短链的表单（可选功能）
@app.get("/")
async def home():
    return {"message": "欢迎使用短链服务！请通过 /u/?url=你的网址 来创建短链，或访问 /u/短码 来跳转。"}

# 可选：如果你想用 POST 方法创建短链，可以添加如下接口（推荐方式）
@app.post("/u/create/")
async def create_short_url_post(original_url: str):
    if not original_url.startswith(('http://', 'https://')):
        original_url = 'http://' + original_url

    short_code = generate_short_code()
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    try:
        cursor.execute("INSERT INTO urls (short_code, original_url) VALUES (?, ?)", (short_code, original_url))
        conn.commit()
    except sqlite3.IntegrityError:
        conn.close()
        return await create_short_url_post(original_url)
    conn.close()

    short_url = f"http://localhost:8000/u/{short_code}"
    return {"original_url": original_url, "short_url": short_url, "short_code": short_code}

✅ 使用方式：

创建短链（推荐 POST，但本示例 GET 参数也支持）：

GET 请求示例（在浏览器访问或 curl）：

http://localhost:8000/u/?url=https://www.example.com

返回结果类似：
```
{
  "original_url": "https://www.example.com",
  "short_url": "http://localhost:8000/u/AbC123",
  "short_code": "AbC123"
}
```
之后你访问 http://localhost:8000/u/AbC123 就会跳转到 https://www.example.com
或者你也可以用 POST 方式（更标准）：

请求：

POST http://localhost:8000/u/create/
Body (form-data 或 x-www-form-urlencoded): original_url=https://www.example.com

返回结果同上。
访问短链：

直接在浏览器访问：

http://localhost:8000/u/你生成的短码

就会自动跳转到原始网址。

📌 注意事项：

本示例使用内存安全的 SQLite，数据保存在当前目录的 shortener.db 文件中。
短码生成采用 secrets 模块，随机且较安全。
默认服务运行在 http://localhost:8000，你可以部署到线上并修改代码中的域名。
若你想支持 xxx.com/u/xxx 这样的自定义域名访问，请确保你的服务器配置了正确的域名和反向代理（如 Nginx），并将代码中的 localhost:8000 替换为你的真实域名。