609e7fb237
This API provides endpoints for: - Product and inventory management - Customer management - Order processing - Payment processing with Stripe integration - User authentication generated with BackendIM... (backend.im)
152 lines
7.1 KiB
Python
152 lines
7.1 KiB
Python
"""Initial migration
|
|
|
|
Revision ID: 00001
|
|
Revises:
|
|
Create Date: 2025-05-12
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '00001'
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
# Create user table
|
|
op.create_table(
|
|
'user',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('email', sa.String(), nullable=True),
|
|
sa.Column('username', sa.String(), nullable=True),
|
|
sa.Column('hashed_password', sa.String(), nullable=True),
|
|
sa.Column('full_name', sa.String(), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=True),
|
|
sa.Column('is_superuser', sa.Boolean(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_user_email'), 'user', ['email'], unique=True)
|
|
op.create_index(op.f('ix_user_id'), 'user', ['id'], unique=False)
|
|
op.create_index(op.f('ix_user_username'), 'user', ['username'], unique=True)
|
|
|
|
# Create product table
|
|
op.create_table(
|
|
'product',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(), nullable=True),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('sku', sa.String(), nullable=True),
|
|
sa.Column('price', sa.Float(), nullable=False),
|
|
sa.Column('image_url', sa.String(), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_product_id'), 'product', ['id'], unique=False)
|
|
op.create_index(op.f('ix_product_name'), 'product', ['name'], unique=False)
|
|
op.create_index(op.f('ix_product_sku'), 'product', ['sku'], unique=True)
|
|
|
|
# Create inventory table
|
|
op.create_table(
|
|
'inventory',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('product_id', sa.Integer(), nullable=True),
|
|
sa.Column('quantity', sa.Integer(), nullable=False),
|
|
sa.Column('location', sa.String(), nullable=True),
|
|
sa.Column('last_restock_date', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['product_id'], ['product.id'], ),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('product_id')
|
|
)
|
|
op.create_index(op.f('ix_inventory_id'), 'inventory', ['id'], unique=False)
|
|
|
|
# Create customer table
|
|
op.create_table(
|
|
'customer',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=True),
|
|
sa.Column('first_name', sa.String(), nullable=False),
|
|
sa.Column('last_name', sa.String(), nullable=False),
|
|
sa.Column('email', sa.String(), nullable=True),
|
|
sa.Column('phone', sa.String(), nullable=True),
|
|
sa.Column('address', sa.String(), nullable=True),
|
|
sa.Column('city', sa.String(), nullable=True),
|
|
sa.Column('state', sa.String(), nullable=True),
|
|
sa.Column('zip_code', sa.String(), nullable=True),
|
|
sa.Column('country', sa.String(), nullable=True),
|
|
sa.Column('stripe_customer_id', sa.String(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['user_id'], ['user.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_customer_email'), 'customer', ['email'], unique=True)
|
|
op.create_index(op.f('ix_customer_id'), 'customer', ['id'], unique=False)
|
|
|
|
# Create order table
|
|
op.create_table(
|
|
'order',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('customer_id', sa.Integer(), nullable=True),
|
|
sa.Column('order_number', sa.String(), nullable=True),
|
|
sa.Column('status', sa.String(), nullable=True),
|
|
sa.Column('total_amount', sa.Float(), nullable=True),
|
|
sa.Column('payment_intent_id', sa.String(), nullable=True),
|
|
sa.Column('payment_status', sa.String(), nullable=True),
|
|
sa.Column('shipping_address', sa.String(), nullable=True),
|
|
sa.Column('shipping_city', sa.String(), nullable=True),
|
|
sa.Column('shipping_state', sa.String(), nullable=True),
|
|
sa.Column('shipping_zip', sa.String(), nullable=True),
|
|
sa.Column('shipping_country', sa.String(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['customer_id'], ['customer.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_order_id'), 'order', ['id'], unique=False)
|
|
op.create_index(op.f('ix_order_order_number'), 'order', ['order_number'], unique=True)
|
|
|
|
# Create order_item table
|
|
op.create_table(
|
|
'orderitem',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('order_id', sa.Integer(), nullable=True),
|
|
sa.Column('product_id', sa.Integer(), nullable=True),
|
|
sa.Column('quantity', sa.Integer(), nullable=True),
|
|
sa.Column('unit_price', sa.Float(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=True),
|
|
sa.ForeignKeyConstraint(['order_id'], ['order.id'], ),
|
|
sa.ForeignKeyConstraint(['product_id'], ['product.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_orderitem_id'), 'orderitem', ['id'], unique=False)
|
|
|
|
|
|
def downgrade():
|
|
op.drop_index(op.f('ix_orderitem_id'), table_name='orderitem')
|
|
op.drop_table('orderitem')
|
|
op.drop_index(op.f('ix_order_order_number'), table_name='order')
|
|
op.drop_index(op.f('ix_order_id'), table_name='order')
|
|
op.drop_table('order')
|
|
op.drop_index(op.f('ix_customer_id'), table_name='customer')
|
|
op.drop_index(op.f('ix_customer_email'), table_name='customer')
|
|
op.drop_table('customer')
|
|
op.drop_index(op.f('ix_inventory_id'), table_name='inventory')
|
|
op.drop_table('inventory')
|
|
op.drop_index(op.f('ix_product_sku'), table_name='product')
|
|
op.drop_index(op.f('ix_product_name'), table_name='product')
|
|
op.drop_index(op.f('ix_product_id'), table_name='product')
|
|
op.drop_table('product')
|
|
op.drop_index(op.f('ix_user_username'), table_name='user')
|
|
op.drop_index(op.f('ix_user_id'), table_name='user')
|
|
op.drop_index(op.f('ix_user_email'), table_name='user')
|
|
op.drop_table('user') |