Files
Hotel-Booking/Backend/alembic/versions/add_guest_profile_crm_tables.py
Iliyan Angelov be07802066 updates
2025-11-21 22:40:44 +02:00

161 lines
7.7 KiB
Python

"""add guest profile crm tables
Revision ID: add_guest_profile_crm
Revises: ff515d77abbe
Create Date: 2024-01-01 12:00:00.000000
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import mysql
# revision identifiers, used by Alembic.
revision = 'add_guest_profile_crm'
down_revision = '9bb08492a382'
branch_labels = None
depends_on = None
def upgrade() -> None:
# Add new fields to users table
op.add_column('users', sa.Column('is_vip', sa.Boolean(), nullable=False, server_default='0'))
op.add_column('users', sa.Column('lifetime_value', sa.Numeric(10, 2), nullable=True, server_default='0'))
op.add_column('users', sa.Column('satisfaction_score', sa.Numeric(3, 2), nullable=True))
op.add_column('users', sa.Column('last_visit_date', sa.DateTime(), nullable=True))
op.add_column('users', sa.Column('total_visits', sa.Integer(), nullable=False, server_default='0'))
# Create guest_preferences table
op.create_table(
'guest_preferences',
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
sa.Column('user_id', sa.Integer(), nullable=False),
sa.Column('preferred_room_location', sa.String(length=100), nullable=True),
sa.Column('preferred_floor', sa.Integer(), nullable=True),
sa.Column('preferred_room_type_id', sa.Integer(), nullable=True),
sa.Column('preferred_amenities', sa.JSON(), nullable=True),
sa.Column('special_requests', sa.Text(), nullable=True),
sa.Column('preferred_services', sa.JSON(), nullable=True),
sa.Column('preferred_contact_method', sa.String(length=50), nullable=True),
sa.Column('preferred_language', sa.String(length=10), nullable=True, server_default='en'),
sa.Column('dietary_restrictions', sa.JSON(), nullable=True),
sa.Column('additional_preferences', sa.JSON(), nullable=True),
sa.Column('created_at', sa.DateTime(), nullable=False),
sa.Column('updated_at', sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
sa.ForeignKeyConstraint(['preferred_room_type_id'], ['room_types.id'], ),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_guest_preferences_id'), 'guest_preferences', ['id'], unique=False)
op.create_index(op.f('ix_guest_preferences_user_id'), 'guest_preferences', ['user_id'], unique=False)
# Create guest_notes table
op.create_table(
'guest_notes',
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
sa.Column('user_id', sa.Integer(), nullable=False),
sa.Column('created_by', sa.Integer(), nullable=False),
sa.Column('note', sa.Text(), nullable=False),
sa.Column('is_important', sa.Boolean(), nullable=False, server_default='0'),
sa.Column('is_private', sa.Boolean(), nullable=False, server_default='0'),
sa.Column('created_at', sa.DateTime(), nullable=False),
sa.Column('updated_at', sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
sa.ForeignKeyConstraint(['created_by'], ['users.id'], ),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_guest_notes_id'), 'guest_notes', ['id'], unique=False)
op.create_index(op.f('ix_guest_notes_user_id'), 'guest_notes', ['user_id'], unique=False)
# Create guest_tags table
op.create_table(
'guest_tags',
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
sa.Column('name', sa.String(length=50), nullable=False),
sa.Column('color', sa.String(length=7), nullable=True, server_default='#3B82F6'),
sa.Column('description', sa.String(length=255), nullable=True),
sa.Column('created_at', sa.DateTime(), nullable=False),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('name')
)
op.create_index(op.f('ix_guest_tags_id'), 'guest_tags', ['id'], unique=False)
op.create_index(op.f('ix_guest_tags_name'), 'guest_tags', ['name'], unique=True)
# Create guest_tag_associations table
op.create_table(
'guest_tag_associations',
sa.Column('user_id', sa.Integer(), nullable=False),
sa.Column('tag_id', sa.Integer(), nullable=False),
sa.Column('created_at', sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(['tag_id'], ['guest_tags.id'], ),
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
sa.PrimaryKeyConstraint('user_id', 'tag_id')
)
# Create guest_communications table
op.create_table(
'guest_communications',
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
sa.Column('user_id', sa.Integer(), nullable=False),
sa.Column('staff_id', sa.Integer(), nullable=True),
sa.Column('communication_type', sa.Enum('email', 'phone', 'sms', 'chat', 'in_person', 'other', name='communicationtype'), nullable=False),
sa.Column('direction', sa.Enum('inbound', 'outbound', name='communicationdirection'), nullable=False),
sa.Column('subject', sa.String(length=255), nullable=True),
sa.Column('content', sa.Text(), nullable=False),
sa.Column('booking_id', sa.Integer(), nullable=True),
sa.Column('is_automated', sa.Boolean(), nullable=False, server_default='0'),
sa.Column('communication_metadata', sa.Text(), nullable=True),
sa.Column('created_at', sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(['booking_id'], ['bookings.id'], ),
sa.ForeignKeyConstraint(['staff_id'], ['users.id'], ),
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_guest_communications_id'), 'guest_communications', ['id'], unique=False)
op.create_index(op.f('ix_guest_communications_user_id'), 'guest_communications', ['user_id'], unique=False)
# Create guest_segments table
op.create_table(
'guest_segments',
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
sa.Column('name', sa.String(length=100), nullable=False),
sa.Column('description', sa.Text(), nullable=True),
sa.Column('criteria', sa.JSON(), nullable=True),
sa.Column('is_active', sa.Boolean(), nullable=False, server_default='1'),
sa.Column('created_at', sa.DateTime(), nullable=False),
sa.Column('updated_at', sa.DateTime(), nullable=False),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('name')
)
op.create_index(op.f('ix_guest_segments_id'), 'guest_segments', ['id'], unique=False)
op.create_index(op.f('ix_guest_segments_name'), 'guest_segments', ['name'], unique=True)
# Create guest_segment_associations table
op.create_table(
'guest_segment_associations',
sa.Column('user_id', sa.Integer(), nullable=False),
sa.Column('segment_id', sa.Integer(), nullable=False),
sa.Column('assigned_at', sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(['segment_id'], ['guest_segments.id'], ),
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
sa.PrimaryKeyConstraint('user_id', 'segment_id')
)
def downgrade() -> None:
# Drop tables in reverse order
op.drop_table('guest_segment_associations')
op.drop_table('guest_segments')
op.drop_table('guest_communications')
op.drop_table('guest_tag_associations')
op.drop_table('guest_tags')
op.drop_table('guest_notes')
op.drop_table('guest_preferences')
# Remove columns from users table
op.drop_column('users', 'total_visits')
op.drop_column('users', 'last_visit_date')
op.drop_column('users', 'satisfaction_score')
op.drop_column('users', 'lifetime_value')
op.drop_column('users', 'is_vip')