"""Add geolocation fields to audit logs

Revision ID: 005_add_geo_location
Revises: 004_enhance_audit_logs
Create Date: 2025-03-07

"""
from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = '005_add_geo_location'
down_revision = '004_enhance_audit_logs'
branch_labels = None
depends_on = None


def upgrade():
    op.add_column('audit_logs', sa.Column('country', sa.String(100), nullable=True))
    op.add_column('audit_logs', sa.Column('city', sa.String(100), nullable=True))


def downgrade():
    op.drop_column('audit_logs', 'city')
    op.drop_column('audit_logs', 'country')