短网址源码,

🗣 SL提问: 短网址源码

✍️ Ai批改: 短网址（Short URL）服务的核心功能是将一个较长的 URL 转换为一个较短、易传播的 URL。例如：`https://example.com/very-long-url` → `https://short.url/abc123`

下面是一个简单的短网址生成系统源码示例，使用 Python 和 Flask 框架实现，并使用 SQLite 数据库存储映射关系。

-

🧩 技术栈

后端：Python + Flask
数据库：SQLite（也可以换成 MySQL、PostgreSQL 等）
可选：Redis 缓存提升性能

-

📁 项目结构

short_url/
│
├── app.py 主程序
├── database.py 数据库初始化
├── models.py 数据模型定义
└── README.md

-

📄 示例代码

1、安装依赖

bash
pip install flask

-

2、`database.py` 初始化数据库

python
import sqlite3

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

init_db()

-

3、`models.py` 数据操作函数

python
import sqlite3
import random
import string

def get_db_connection():
conn = sqlite3.connect('short_url.db')
conn.row_factory = sqlite3.Row
return conn

def generate_short_code(length=6):
chars = string.ascii_letters + string.digits
return ''.join(random.choices(chars, k=length))

def create_short_url(long_url):
conn = get_db_connection()
cursor = conn.cursor()
short_code = generate_short_code()
try:
cursor.execute('INSERT INTO urls (long_url, short_code) VALUES (?, ?)', (long_url, short_code))
conn.commit()
except sqlite3.IntegrityError:
如果生成的 short_code 冲突，则递归重试
return create_short_url(long_url)
finally:
conn.close()
return short_code

def get_long_url(short_code):
conn = get_db_connection()
cursor = conn.cursor()
cursor.execute('SELECT long_url FROM urls WHERE short_code = ?', (short_code,))
result = cursor.fetchone()
conn.close()
return result['long_url'] if result else None

-

4、`app.py` Flask 应用主程序

python
from flask import Flask, request, redirect, jsonify
from models import create_short_url, get_long_url

app = Flask(__name__)

@app.route('/shorten', methods=['POST'])
def shorten():
data = request.get_json()
long_url = data.get('url')
if not long_url:
return jsonify({'error': 'Missing URL'}), 400
short_code = create_short_url(long_url)
return jsonify({'short_url': f'http://localhost:5000/{short_code}'})

@app.route('/<short_code')
def redirect_to_url(short_code):
long_url = get_long_url(short_code)
if long_url:
return redirect(long_url)
return jsonify({'error': 'URL not found'}), 404

if __name__ == '__main__':
app.run(debug=True)

-

✅ 使用方式

启动服务：

bash
python app.py

发送 POST 请求创建短链接：

bash
curl -X POST http://localhost:5000/shorten -H Content-Type: application/json -d '{url:https://example.com}'

返回示例：

json
{
short_url: http://localhost:5000/abc123
}

访问该短网址就会跳转到原始地址。

-

🔒 可扩展功能（进阶）

增加用户注册与登录系统
添加短链过期时间
使用 Redis 缓存热点链接
提供统计分析（点击次数、来源等）
支持自定义短码
部署到服务器 + Nginx + HTTPS

-

如果你需要使用其他语言（如 Java、Node.js、PHP 等）实现的短网址系统，我也可以提供对应版本。欢迎继续提问！