"""add is_locked and locked_at to user_sessions

Revision ID: 052
Revises: 051
"""
from alembic import op
import sqlalchemy as sa

revision = "052"
down_revision = "051"

def upgrade():
    op.add_column("user_sessions", sa.Column("is_locked", sa.Boolean(), server_default="false", nullable=False))
    op.add_column("user_sessions", sa.Column("locked_at", sa.DateTime(), nullable=True))

def downgrade():
    op.drop_column("user_sessions", "locked_at")
    op.drop_column("user_sessions", "is_locked")