"""Add referrer_stats_daily and browser_stats_daily (Method A, Ch08)."""
from alembic import op
import sqlalchemy as sa

revision = "0003_referrer_and_browser_stats"
down_revision = "0002_ip_registry_verification_result"
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        "referrer_stats_daily",
        sa.Column("date", sa.Date, primary_key=True),
        sa.Column("referrer_domain", sa.String(255), primary_key=True),
        sa.Column("count", sa.Integer, nullable=False, server_default="0"),
    )
    op.create_table(
        "browser_stats_daily",
        sa.Column("date", sa.Date, primary_key=True),
        sa.Column("browser", sa.String(64), primary_key=True),
        sa.Column("os", sa.String(64), primary_key=True),
        sa.Column("count", sa.Integer, nullable=False, server_default="0"),
    )


def downgrade():
    op.drop_table("browser_stats_daily")
    op.drop_table("referrer_stats_daily")