125 lines
5.6 KiB
Python
125 lines
5.6 KiB
Python
"""initial schema
|
|
|
|
Revision ID: 202605110001
|
|
Revises:
|
|
Create Date: 2026-05-11
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
|
|
from alembic import op
|
|
|
|
revision: str = "202605110001"
|
|
down_revision: str | None = None
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
service_type = sa.Enum(
|
|
"maintenance",
|
|
"repair",
|
|
"fluid",
|
|
"tire",
|
|
"inspection",
|
|
"insurance",
|
|
"tax",
|
|
"other",
|
|
name="servicetype",
|
|
create_type=False,
|
|
)
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("telegram_id", sa.BigInteger(), nullable=False),
|
|
sa.Column("username", sa.String(length=128), nullable=True),
|
|
sa.Column("first_name", sa.String(length=128), nullable=True),
|
|
sa.Column("last_name", sa.String(length=128), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_users_telegram_id"), "users", ["telegram_id"], unique=True)
|
|
|
|
op.create_table(
|
|
"cars",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("owner_id", sa.Integer(), nullable=False),
|
|
sa.Column("name", sa.String(length=160), nullable=False),
|
|
sa.Column("make", sa.String(length=80), nullable=True),
|
|
sa.Column("model", sa.String(length=80), nullable=True),
|
|
sa.Column("year", sa.Integer(), nullable=True),
|
|
sa.Column("plate_number", sa.String(length=32), nullable=True),
|
|
sa.Column("vin", sa.String(length=32), nullable=True),
|
|
sa.Column("fuel_type", sa.String(length=32), nullable=True),
|
|
sa.Column("purchase_date", sa.Date(), nullable=True),
|
|
sa.Column("purchase_price", sa.Numeric(12, 2), nullable=True),
|
|
sa.Column("current_odometer", sa.Integer(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["owner_id"], ["users.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_cars_owner_id"), "cars", ["owner_id"], unique=False)
|
|
|
|
op.create_table(
|
|
"fuel_entries",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("car_id", sa.Integer(), nullable=False),
|
|
sa.Column("entry_date", sa.Date(), nullable=False),
|
|
sa.Column("odometer", sa.Integer(), nullable=False),
|
|
sa.Column("liters", sa.Numeric(8, 3), nullable=False),
|
|
sa.Column("price_per_liter", sa.Numeric(10, 2), nullable=False),
|
|
sa.Column("total_cost", sa.Numeric(12, 2), nullable=False),
|
|
sa.Column("station", sa.String(length=160), nullable=True),
|
|
sa.Column("fuel_brand", sa.String(length=80), nullable=True),
|
|
sa.Column("is_full_tank", sa.Boolean(), nullable=False),
|
|
sa.Column("notes", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["car_id"], ["cars.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_fuel_entries_car_id"), "fuel_entries", ["car_id"], unique=False)
|
|
op.create_index(op.f("ix_fuel_entries_entry_date"), "fuel_entries", ["entry_date"], unique=False)
|
|
|
|
op.create_table(
|
|
"service_entries",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("car_id", sa.Integer(), nullable=False),
|
|
sa.Column("entry_date", sa.Date(), nullable=False),
|
|
sa.Column("odometer", sa.Integer(), nullable=True),
|
|
sa.Column("service_type", service_type, nullable=False),
|
|
sa.Column("title", sa.String(length=180), nullable=False),
|
|
sa.Column("category", sa.String(length=80), nullable=True),
|
|
sa.Column("vendor", sa.String(length=160), nullable=True),
|
|
sa.Column("total_cost", sa.Numeric(12, 2), nullable=False),
|
|
sa.Column("next_due_date", sa.Date(), nullable=True),
|
|
sa.Column("next_due_odometer", sa.Integer(), nullable=True),
|
|
sa.Column("notes", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.ForeignKeyConstraint(["car_id"], ["cars.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_service_entries_car_id"), "service_entries", ["car_id"], unique=False)
|
|
op.create_index(op.f("ix_service_entries_entry_date"), "service_entries", ["entry_date"], unique=False)
|
|
op.create_index(op.f("ix_service_entries_service_type"), "service_entries", ["service_type"], unique=False)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index(op.f("ix_service_entries_service_type"), table_name="service_entries")
|
|
op.drop_index(op.f("ix_service_entries_entry_date"), table_name="service_entries")
|
|
op.drop_index(op.f("ix_service_entries_car_id"), table_name="service_entries")
|
|
op.drop_table("service_entries")
|
|
op.drop_index(op.f("ix_fuel_entries_entry_date"), table_name="fuel_entries")
|
|
op.drop_index(op.f("ix_fuel_entries_car_id"), table_name="fuel_entries")
|
|
op.drop_table("fuel_entries")
|
|
op.drop_index(op.f("ix_cars_owner_id"), table_name="cars")
|
|
op.drop_table("cars")
|
|
op.drop_index(op.f("ix_users_telegram_id"), table_name="users")
|
|
op.drop_table("users")
|
|
service_type.drop(op.get_bind(), checkfirst=True)
|