Migration of events to use group_id instead of
group_name
This commit is contained in:
@@ -0,0 +1,91 @@
|
||||
"""Migrate events to use group_id instead of client_uuid
|
||||
|
||||
Revision ID: d490cbfdea65
|
||||
Revises: 8a45ec34f84d
|
||||
Create Date: 2025-06-30 19:16:29.138440
|
||||
|
||||
ACHTUNG:
|
||||
Ein Downgrade dieser Migration ist NICHT verlustfrei möglich, wenn mehrere Clients einer Gruppe zugeordnet sind.
|
||||
Beim Downgrade wird jedem Event willkürlich ein (der erste gefundene) Client der Gruppe zugeordnet.
|
||||
Die ursprüngliche Zuordnung von Events zu Clients kann dadurch NICHT wiederhergestellt werden!
|
||||
"""
|
||||
from typing import Sequence, Union
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision: str = 'd490cbfdea65'
|
||||
down_revision: Union[str, None] = '8a45ec34f84d'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
"""Upgrade schema."""
|
||||
# 1. Neue Spalte group_id (vorübergehend nullable)
|
||||
op.add_column('events', sa.Column('group_id', sa.Integer(), nullable=True))
|
||||
op.create_foreign_key(
|
||||
'fk_events_group_id_client_groups',
|
||||
'events', 'client_groups',
|
||||
['group_id'], ['id']
|
||||
)
|
||||
op.create_index('ix_events_group_id', 'events', ['group_id'])
|
||||
|
||||
# 2. group_id für alle Events anhand client_uuid setzen
|
||||
op.execute("""
|
||||
UPDATE events
|
||||
SET group_id = (
|
||||
SELECT group_id FROM clients WHERE clients.uuid = events.client_uuid
|
||||
)
|
||||
""")
|
||||
|
||||
# 3. client_uuid entfernen
|
||||
op.drop_constraint('events_ibfk_1', 'events',
|
||||
type_='foreignkey') # Name ggf. anpassen!
|
||||
# Name ggf. anpassen!
|
||||
op.drop_index('ix_events_client_uuid', table_name='events')
|
||||
op.drop_column('events', 'client_uuid')
|
||||
|
||||
# 4. group_id auf NOT NULL setzen
|
||||
op.alter_column(
|
||||
'events',
|
||||
'group_id',
|
||||
existing_type=sa.Integer(),
|
||||
nullable=False
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
"""Downgrade schema.
|
||||
|
||||
ACHTUNG:
|
||||
Ein Downgrade ist nicht verlustfrei möglich, wenn mehrere Clients pro Gruppe existieren.
|
||||
Es wird jeweils ein beliebiger (erster) Client der Gruppe für die Zuordnung gewählt.
|
||||
"""
|
||||
# 1. client_uuid wieder hinzufügen
|
||||
op.add_column('events', sa.Column(
|
||||
'client_uuid', sa.String(length=36), nullable=True))
|
||||
op.create_foreign_key(
|
||||
'fk_events_client_uuid_clients',
|
||||
'events', 'clients',
|
||||
['client_uuid'], ['uuid']
|
||||
)
|
||||
op.create_index('ix_events_client_uuid', 'events', ['client_uuid'])
|
||||
|
||||
# 2. client_uuid anhand group_id zurücksetzen (nur möglich, wenn 1:1-Beziehung!)
|
||||
# Falls mehrere Clients pro Gruppe: Datenverlust möglich!
|
||||
# Hier ggf. eine Strategie überlegen oder leerlassen.
|
||||
op.execute("""
|
||||
UPDATE events
|
||||
SET client_uuid = (
|
||||
SELECT uuid FROM clients WHERE clients.group_id = events.group_id LIMIT 1
|
||||
)
|
||||
""")
|
||||
|
||||
# 3. group_id entfernen
|
||||
op.drop_constraint('fk_events_group_id_client_groups',
|
||||
'events', type_='foreignkey')
|
||||
op.drop_index('ix_events_group_id', table_name='events')
|
||||
op.drop_column('events', 'group_id')
|
||||
Reference in New Issue
Block a user