Можно ли задать первичный ключ вручную?

Задание первичного ключа вручную в Python и SQL

Да, можно и часто нужно! Первичный ключ (Primary Key) можно задать вручную вместо автоинкремента. Это полезно в разных сценариях.

Способы задания Primary Key

1. SQL: Явное определение PK

-- Вариант 1: Auto-increment (по умолчанию)
CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(100),
    email VARCHAR(100)
);

-- Вариант 2: Вручную указываем PRIMARY KEY
CREATE TABLE users (
    id INT PRIMARY KEY,
    name VARCHAR(100),
    email VARCHAR(100)
);

-- Вариант 3: UUID (уникальный идентификатор)
CREATE TABLE users (
    id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
    name VARCHAR(100),
    email VARCHAR(100)
);

-- Вариант 4: Строка как PK
CREATE TABLE users (
    username VARCHAR(50) PRIMARY KEY,
    email VARCHAR(100),
    created_at TIMESTAMP
);

SQLAlchemy: Задание PK вручную

Пример 1: Integer PK без автоинкремента

from sqlalchemy import Column, Integer, String, create_engine
from sqlalchemy.orm import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    
    # Первичный ключ БЕЗ автоинкремента
    id = Column(Integer, primary_key=True, autoincrement=False)
    name = Column(String(100))
    email = Column(String(100))

# Вставка — нужно самому задать id
db = SessionLocal()
user = User(id=1, name='John', email='john@example.com')
db.add(user)
db.commit()

Пример 2: UUID как первичный ключ

from sqlalchemy import Column, String, create_engine
from sqlalchemy.orm import declarative_base
import uuid

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    
    # UUID как PK
    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    name = Column(String(100))
    email = Column(String(100))

# Использование
db = SessionLocal()
user = User(name='John', email='john@example.com')  # id сгенерируется автоматически
db.add(user)
db.commit()
print(user.id)  # f67556b8-54c9-4971-96ea-4fdefb25afd8

Пример 3: Строка (username) как первичный ключ

from sqlalchemy import Column, String, DateTime, create_engine
from sqlalchemy.orm import declarative_base
from datetime import datetime

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    
    # Username как первичный ключ
    username = Column(String(50), primary_key=True)
    email = Column(String(100))
    created_at = Column(DateTime, default=datetime.utcnow)

# Использование
db = SessionLocal()
user = User(username='john_doe', email='john@example.com')
db.add(user)
db.commit()

# Поиск
user = db.query(User).filter_by(username='john_doe').first()

Пример 4: Составной первичный ключ (Composite Key)

from sqlalchemy import Column, Integer, String, PrimaryKeyConstraint, create_engine
from sqlalchemy.orm import declarative_base

Base = declarative_base()

class OrderItem(Base):
    __tablename__ = 'order_items'
    
    # Два поля составляют PK
    order_id = Column(Integer, primary_key=True)
    item_id = Column(Integer, primary_key=True)
    quantity = Column(Integer)
    
    # Или через constraint:
    # __table_args__ = (PrimaryKeyConstraint('order_id', 'item_id'),)

# Использование
db = SessionLocal()
order_item = OrderItem(order_id=1, item_id=1, quantity=5)
db.add(order_item)
db.commit()

# Поиск
item = db.query(OrderItem).filter_by(
    order_id=1, item_id=1
).first()

PyMySQL: Вставка вручную заданного PK

import pymysql

connection = pymysql.connect(
    host='localhost',
    user='root',
    password='password',
    database='mydb'
)

cursor = connection.cursor()

# Вставка с явным id
sql = "INSERT INTO users (id, name, email) VALUES (%s, %s, %s)"
cursor.execute(sql, (42, 'John', 'john@example.com'))
connection.commit()

print(f'Вставлена строка с id=42')

cursor.close()
connection.close()

Когда задавать PK вручную?

1. UUID (распределённые системы)

import uuid

class Post(Base):
    __tablename__ = 'posts'
    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    title = Column(String(200))
    content = Column(String(5000))

# Преимущества:
# ✓ Уникален гарантированно
# ✓ Не зависит от последовательности в БД
# ✓ Хорош для микросервисов
# ✓ Распределённые системы

2. Строка (username, email)

class Account(Base):
    __tablename__ = 'accounts'
    username = Column(String(50), primary_key=True)
    password_hash = Column(String(256))
    email = Column(String(100))

# Преимущества:
# ✓ Естественный идентификатор
# ✓ Поиск быстрее (не нужен JOIN)
# ✓ Понятнее для пользователя
# ✓ Нет дублирования username

3. Составной ключ (отношение многие-ко-многим)

class StudentCourse(Base):
    __tablename__ = 'student_courses'
    student_id = Column(Integer, primary_key=True, ForeignKey('students.id'))
    course_id = Column(Integer, primary_key=True, ForeignKey('courses.id'))
    grade = Column(String(1))

# Преимущества:
# ✓ Предотвращает дубликаты
# ✓ Естественное описание отношения
# ✓ Экономит место в БД

4. Кастомная логика (код сотрудника)

from datetime import datetime

class Employee(Base):
    __tablename__ = 'employees'
    # id формируется как EMP-2024-0001
    emp_id = Column(String(15), primary_key=True)
    name = Column(String(100))
    email = Column(String(100))

def generate_emp_id():
    year = datetime.now().year
    # SELECT MAX(id) WHERE id LIKE 'EMP-2024-%'
    return f'EMP-{year}-0001'

db = SessionLocal()
emp = Employee(emp_id=generate_emp_id(), name='John')
db.add(emp)
db.commit()

Сравнение вариантов

Проблемы при вручном задании PK

# ❌ Проблема 1: Дубликаты
db.add(User(id=1, name='John'))  # id=1
db.add(User(id=1, name='Jane'))  # Дубликат! IntegrityError
db.commit()

# ✅ Решение: Проверка перед вставкой
if not db.query(User).filter_by(id=1).exists():
    db.add(User(id=1, name='John'))
    db.commit()

# ❌ Проблема 2: Пропуск ID
db.add(User(id=1, name='John'))
db.add(User(id=3, name='Jane'))  # id=2 пропущен
db.commit()

# ✅ Решение: UUID или наличие логики для генерации

Лучшие практики

1. Используй AUTO_INCREMENT для простых таблиц
2. Используй UUID для микросервисов и распределённых систем
3. Используй строку как PK, если это естественный идентификатор
4. Защищай от дубликатов при вручном задании
5. Документируй логику генерации PK

class User(Base):
    __tablename__ = 'users'
    
    # Явное указание: не AUTO_INCREMENT
    id = Column(
        String(36),
        primary_key=True,
        default=lambda: str(uuid.uuid4()),
        doc='UUID, генерируется автоматически'
    )
    username = Column(String(50), unique=True, nullable=False)
    email = Column(String(100), unique=True, nullable=False)

Итог: Первичный ключ можно задавать вручную в любом формате — UUID, строки, составные ключи. Главное — гарантировать уникальность и правильную генерацию. Для большинства случаев подходит AUTO_INCREMENT, для сложных систем — UUID.