Flask-SQLAlchemy模型：安全高效地生成唯一6位ID

import uuid

def generate_truncated_uuid_id():
    """
    生成一个截断的UUID字符串作为ID。
    注意：截断会大幅降低UUID的唯一性。
    """
    return str(uuid.uuid4())[:6]

# 示例用法
# print(generate_truncated_uuid_id()) # 示例输出: 'a1b2c3'

import secrets
import string

def generate_random_6_char_id():
    """
    生成一个由大小写字母和数字组成的6位随机字符串ID。
    此函数不进行数据库唯一性检查，依赖数据库的unique约束。
    """
    characters = string.ascii_letters + string.digits # 包含大小写字母和数字 (共62种字符)
    return ''.join(secrets.choice(characters) for _ in range(6))

# 示例用法
# print(generate_random_6_char_id()) # 示例输出: 'QkABL0'

import secrets
import string
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.exc import IntegrityError # 用于捕获数据库唯一性约束错误

# Flask 和 SQLAlchemy 初始化
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///app.db' # 使用文件型SQLite数据库
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)

# ID生成函数
def generate_random_6_char_id():
    """
    生成一个由大小写字母和数字组成的6位随机字符串ID。
    此函数仅负责生成随机字符串，不进行数据库唯一性检查。
    唯一性由数据库的unique约束和应用层的重试机制保证。
    """
    characters = string.ascii_letters + string.digits # 62种字符
    return ''.join(secrets.choice(characters) for _ in range(6))

# SQLAlchemy 模型定义
class Item(db.Model):
    __tablename__ = 'items' # 明确表名

    id = db.Column(db.String(6), primary_key=True, unique=True, default=generate_random_6_char_id)
    name = db.Column(db.String(100), nullable=False)
    description = db.Column(db.Text, nullable=True)

    def __repr__(self):
        return f'<Item id={self.id}, name={self.name}>'

# 数据库操作示例
with app.app_context():
    db.create_all() # 创建所有表

    # 尝试创建新的Item实例，并处理潜在的ID冲突
    max_retries = 5 # 设置最大重试次数
    for i in range(max_retries):
        try:
            new_item = Item(name=f"商品 {i+1}", description=f"这是第 {i+1} 个商品")
            db.session.add(new_item)
            db.session.commit()
            print(f"成功创建商品，ID: {new_item.id}, 名称: {new_item.name}")
            break # 成功创建，跳出重试循环
        except IntegrityError:
            db.session.rollback() # 回滚事务
            print(f"检测到ID冲突，尝试重新生成 (第 {i+1} 次尝试)...")
            if i == max_retries - 1:
                # 如果达到最大重试次数仍失败，则抛出运行时错误
                raise RuntimeError("未能生成唯一的6位ID，所有重试均失败。可能ID空间已耗尽或冲突率过高。")
        except Exception as e:
            db.session.rollback() # 捕获其他异常也回滚
            raise e # 重新抛出其他意外错误

    print("\n--- 当前数据库中的所有商品 ---")
    all_items = Item.query.all()
    if all_items:
        for item in all_items:
            print(item)
    else:
        print("数据库中没有商品。")

    # 也可以手动指定ID（如果需要）
    try:
        specific_item = Item(id="CUSTOM", name="定制商品")
        db.session.add(specific_item)
        db.session.commit()
        print(f"成功创建定制商品，ID: {specific_item.id}")
    except IntegrityError:
        db.session.rollback()
        print("定制ID 'CUSTOM' 已存在或发生冲突。")