短域名源码,

Wd提问: 短域名源码

Ai批改: 短域名系统（Short URL）是一种将长网址转换为较短形式的服务，用户点击短链接后会被重定向到原始的长网址。实现一个简单的短域名系统可以使用多种编程语言和框架，下面我将用 Python + Flask + SQLite 提供一个基础版本的示例代码。

-

🧩 功能需求

1、接收长 URL。
2、生成唯一的短码（如：abc123）。
3、将短码与原 URL 存入数据库。
4、用户访问短链接时跳转至原 URL。

-

📦 技术栈

后端：Python + Flask
数据库：SQLite（可替换为 MySQL / PostgreSQL）
短码生成方式：Base64 编码或随机字符串

-

🗂️ 目录结构

short_url/
├── app.py
├── database.db
└── templates/
└── index.html

-

🔧 app.py 示例代码

python
from flask import Flask, request, redirect, render_template
import sqlite3
import random
import string

app = Flask(__name__)
DB_NAME = 'database.db'

初始化数据库
def init_db():
with sqlite3.connect(DB_NAME) as conn:
cur = conn.cursor()
cur.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()

init_db()

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

插入新URL
def insert_url(long_url):
short_code = generate_short_code()
while True:
try:
with sqlite3.connect(DB_NAME) as conn:
cur = conn.cursor()
cur.execute(INSERT INTO urls (long_url, short_code) VALUES (?, ?), (long_url, short_code))
conn.commit()
break
except sqlite3.IntegrityError:
short_code = generate_short_code() 避免冲突
return short_code

查询长URL
def get_long_url(short_code):
with sqlite3.connect(DB_NAME) as conn:
cur = conn.cursor()
cur.execute(SELECT long_url FROM urls WHERE short_code=?, (short_code,))
result = cur.fetchone()
return result[0] if result else None

主页
@app.route(/, methods=[GET, POST])
def index():
short_url = None
if request.method == POST:
long_url = request.form.get(url)
if long_url:
short_code = insert_url(long_url)
short_url = request.host_url + short_code
return render_template(index.html, short_url=short_url)

短链跳转
@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 URL not found, 404

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

-

🌐 templates/index.html 示例页面

html
<!DOCTYPE html
<html lang=en
<head
<meta charset=UTF-8
<title短链接生成器</title
</head
<body
<h1短链接生成器</h1
<form method=post
<input type=text name=url placeholder=输入你的长链接 required
<button type=submit生成短链</button
</form

{% if short_url %}
<p你的短链接是：<a href={{ short_url }}{{ short_url }}</a</p
{% endif %}
</body
</html

-

🚀 运行方法

1、安装依赖：

bash
pip install flask

2、执行程序：

bash
python app.py

3、打开浏览器访问：

http://localhost:5000

-

🔄 可扩展功能

使用 Redis 缓存提高性能
添加后台管理界面
统计点击次数
自定义短码
使用哈希算法生成更短的唯一码（如 Hashids）

-

如果你需要部署版、带数据库迁移脚本或者使用其他语言（如 PHP、Node.js）的实现，请告诉我，我可以继续提供完整源码和部署文档。