"""Add user streak tracking

Revision ID: 51a2c48d2f9a
Revises: head
Create Date: 2023-09-20 10:00:00.000000

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '51a2c48d2f9a'
down_revision = None  # Update this to the current head revision
branch_labels = None
depends_on = None


def upgrade():
    # Add new columns to users table
    op.add_column('users', sa.Column('last_login', sa.DateTime(), nullable=True))
    op.add_column('users', sa.Column('current_streak', sa.Integer(), nullable=True, server_default='0'))
    op.add_column('users', sa.Column('longest_streak', sa.Integer(), nullable=True, server_default='0'))
    op.add_column('users', sa.Column('last_streak_date', sa.Date(), nullable=True))


def downgrade():
    # Remove columns from users table
    op.drop_column('users', 'last_login')
    op.drop_column('users', 'current_streak')
    op.drop_column('users', 'longest_streak')
    op.drop_column('users', 'last_streak_date')