232 lines
12 KiB
Python
232 lines
12 KiB
Python
"""
|
|
Initial database setup
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
Create Date: 2023-07-25 00:00:00.000000
|
|
|
|
"""
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import sqlite
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '001'
|
|
down_revision: str | None = None
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create enum types for SQLite
|
|
user_role_type = sa.Enum('customer', 'seller', 'admin', name='userroletype')
|
|
product_status_type = sa.Enum('draft', 'published', 'out_of_stock', 'discontinued', name='productstatustype')
|
|
order_status_type = sa.Enum('pending', 'processing', 'shipped', 'delivered', 'cancelled', 'refunded', name='orderstatustype')
|
|
shipping_method_type = sa.Enum('standard', 'express', 'overnight', 'pickup', 'digital', name='shippingmethodtype')
|
|
payment_status_type = sa.Enum('pending', 'processing', 'completed', 'failed', 'refunded', name='paymentstatustype')
|
|
payment_method_type = sa.Enum('credit_card', 'paypal', 'bank_transfer', 'cash_on_delivery', 'stripe', 'apple_pay', 'google_pay', name='paymentmethodtype')
|
|
|
|
# Users table
|
|
op.create_table(
|
|
'users',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('email', sa.String(255), nullable=False, unique=True, index=True),
|
|
sa.Column('hashed_password', sa.String(255), nullable=False),
|
|
sa.Column('first_name', sa.String(100), nullable=True),
|
|
sa.Column('last_name', sa.String(100), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), default=True),
|
|
sa.Column('role', user_role_type, default='customer'),
|
|
sa.Column('phone_number', sa.String(20), nullable=True),
|
|
sa.Column('profile_image', sa.String(255), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
sa.Column('address_line1', sa.String(255), nullable=True),
|
|
sa.Column('address_line2', sa.String(255), nullable=True),
|
|
sa.Column('city', sa.String(100), nullable=True),
|
|
sa.Column('state', sa.String(100), nullable=True),
|
|
sa.Column('postal_code', sa.String(20), nullable=True),
|
|
sa.Column('country', sa.String(100), nullable=True),
|
|
sa.Column('email_verified', sa.Boolean(), default=False),
|
|
sa.Column('verification_token', sa.String(255), nullable=True),
|
|
sa.Column('reset_password_token', sa.String(255), nullable=True),
|
|
sa.Column('reset_token_expires_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('bio', sa.Text(), nullable=True),
|
|
)
|
|
|
|
# Categories table
|
|
op.create_table(
|
|
'categories',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('name', sa.String(100), nullable=False, index=True),
|
|
sa.Column('slug', sa.String(120), nullable=False, unique=True),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('image', sa.String(255), nullable=True),
|
|
sa.Column('parent_id', sa.String(36), sa.ForeignKey('categories.id'), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), default=True),
|
|
sa.Column('display_order', sa.Integer(), default=0),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Tags table
|
|
op.create_table(
|
|
'tags',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('name', sa.String(50), nullable=False, unique=True, index=True),
|
|
sa.Column('slug', sa.String(60), nullable=False, unique=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Products table
|
|
op.create_table(
|
|
'products',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('name', sa.String(255), nullable=False, index=True),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('price', sa.Float(), nullable=False),
|
|
sa.Column('sku', sa.String(100), unique=True, nullable=True),
|
|
sa.Column('barcode', sa.String(100), unique=True, nullable=True),
|
|
sa.Column('stock_quantity', sa.Integer(), default=0),
|
|
sa.Column('weight', sa.Float(), nullable=True),
|
|
sa.Column('dimensions', sa.String(100), nullable=True),
|
|
sa.Column('status', product_status_type, default='draft'),
|
|
sa.Column('is_featured', sa.Boolean(), default=False),
|
|
sa.Column('is_digital', sa.Boolean(), default=False),
|
|
sa.Column('digital_download_link', sa.String(512), nullable=True),
|
|
sa.Column('slug', sa.String(255), nullable=False, unique=True),
|
|
sa.Column('tax_rate', sa.Float(), default=0.0),
|
|
sa.Column('discount_price', sa.Float(), nullable=True),
|
|
sa.Column('discount_start_date', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('discount_end_date', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('category_id', sa.String(36), sa.ForeignKey('categories.id'), nullable=True),
|
|
sa.Column('seller_id', sa.String(36), sa.ForeignKey('users.id'), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Product Images table
|
|
op.create_table(
|
|
'product_images',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('product_id', sa.String(36), sa.ForeignKey('products.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('image_url', sa.String(512), nullable=False),
|
|
sa.Column('alt_text', sa.String(255), nullable=True),
|
|
sa.Column('is_primary', sa.Boolean(), default=False),
|
|
sa.Column('display_order', sa.Integer(), default=0),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
|
|
# Product-Tag association table
|
|
op.create_table(
|
|
'product_tags',
|
|
sa.Column('product_id', sa.String(36), sa.ForeignKey('products.id', ondelete='CASCADE'), primary_key=True),
|
|
sa.Column('tag_id', sa.String(36), sa.ForeignKey('tags.id', ondelete='CASCADE'), primary_key=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
|
|
# Cart Items table
|
|
op.create_table(
|
|
'cart_items',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('user_id', sa.String(36), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('product_id', sa.String(36), sa.ForeignKey('products.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('quantity', sa.Integer(), default=1, nullable=False),
|
|
sa.Column('price_at_addition', sa.Float(), nullable=False),
|
|
sa.Column('custom_properties', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Orders table
|
|
op.create_table(
|
|
'orders',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('user_id', sa.String(36), sa.ForeignKey('users.id'), nullable=False),
|
|
sa.Column('order_number', sa.String(50), nullable=False, unique=True, index=True),
|
|
sa.Column('status', order_status_type, default='pending'),
|
|
sa.Column('total_amount', sa.Float(), nullable=False),
|
|
sa.Column('subtotal', sa.Float(), nullable=False),
|
|
sa.Column('tax_amount', sa.Float(), nullable=False),
|
|
sa.Column('shipping_amount', sa.Float(), nullable=False),
|
|
sa.Column('discount_amount', sa.Float(), default=0.0),
|
|
sa.Column('shipping_method', shipping_method_type, nullable=True),
|
|
sa.Column('tracking_number', sa.String(100), nullable=True),
|
|
sa.Column('notes', sa.Text(), nullable=True),
|
|
sa.Column('shipping_address', sqlite.JSON(), nullable=True),
|
|
sa.Column('billing_address', sqlite.JSON(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Order Items table
|
|
op.create_table(
|
|
'order_items',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('order_id', sa.String(36), sa.ForeignKey('orders.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('product_id', sa.String(36), sa.ForeignKey('products.id'), nullable=False),
|
|
sa.Column('quantity', sa.Integer(), default=1, nullable=False),
|
|
sa.Column('unit_price', sa.Float(), nullable=False),
|
|
sa.Column('subtotal', sa.Float(), nullable=False),
|
|
sa.Column('discount', sa.Float(), default=0.0),
|
|
sa.Column('tax_amount', sa.Float(), default=0.0),
|
|
sa.Column('product_name', sa.String(255), nullable=False),
|
|
sa.Column('product_sku', sa.String(100), nullable=True),
|
|
sa.Column('product_options', sqlite.JSON(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
|
|
# Payments table
|
|
op.create_table(
|
|
'payments',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('order_id', sa.String(36), sa.ForeignKey('orders.id'), nullable=False),
|
|
sa.Column('amount', sa.Float(), nullable=False),
|
|
sa.Column('payment_method', payment_method_type, nullable=False),
|
|
sa.Column('status', payment_status_type, default='pending'),
|
|
sa.Column('transaction_id', sa.String(255), nullable=True, unique=True),
|
|
sa.Column('payment_details', sqlite.JSON(), nullable=True),
|
|
sa.Column('error_message', sa.String(512), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Reviews table
|
|
op.create_table(
|
|
'reviews',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('product_id', sa.String(36), sa.ForeignKey('products.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('user_id', sa.String(36), sa.ForeignKey('users.id'), nullable=False),
|
|
sa.Column('rating', sa.Integer(), nullable=False),
|
|
sa.Column('title', sa.String(255), nullable=True),
|
|
sa.Column('comment', sa.Text(), nullable=True),
|
|
sa.Column('is_verified_purchase', sa.Boolean(), default=False),
|
|
sa.Column('is_approved', sa.Boolean(), default=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Create indexes
|
|
op.create_index('ix_users_email', 'users', ['email'], unique=True)
|
|
op.create_index('ix_categories_name', 'categories', ['name'])
|
|
op.create_index('ix_products_name', 'products', ['name'])
|
|
op.create_index('ix_tags_name', 'tags', ['name'], unique=True)
|
|
op.create_index('ix_orders_order_number', 'orders', ['order_number'], unique=True)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Drop tables in reverse order of creation
|
|
op.drop_table('reviews')
|
|
op.drop_table('payments')
|
|
op.drop_table('order_items')
|
|
op.drop_table('orders')
|
|
op.drop_table('cart_items')
|
|
op.drop_table('product_tags')
|
|
op.drop_table('product_images')
|
|
op.drop_table('products')
|
|
op.drop_table('tags')
|
|
op.drop_table('categories')
|
|
op.drop_table('users')
|