c8aed27755
- Set up project structure with modular organization - Implement database models for users, organizations, clients, invoices - Create Alembic migration scripts for database setup - Implement JWT-based authentication and authorization - Create API endpoints for users, organizations, clients, invoices - Add PDF generation for invoices using ReportLab - Add comprehensive documentation in README
163 lines
7.2 KiB
Python
163 lines
7.2 KiB
Python
"""Initial tables
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
Create Date: 2023-12-01
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
import enum
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '001'
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
class InvoiceStatus(str, enum.Enum):
|
|
DRAFT = "draft"
|
|
SENT = "sent"
|
|
PAID = "paid"
|
|
OVERDUE = "overdue"
|
|
CANCELLED = "cancelled"
|
|
|
|
|
|
def upgrade():
|
|
# Create organization table
|
|
op.create_table(
|
|
'organization',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.Column('name', sa.String(), nullable=False),
|
|
sa.Column('address', sa.Text(), nullable=True),
|
|
sa.Column('city', sa.String(), nullable=True),
|
|
sa.Column('state', sa.String(), nullable=True),
|
|
sa.Column('postal_code', sa.String(), nullable=True),
|
|
sa.Column('country', sa.String(), nullable=True),
|
|
sa.Column('phone', sa.String(), nullable=True),
|
|
sa.Column('email', sa.String(), nullable=True),
|
|
sa.Column('website', sa.String(), nullable=True),
|
|
sa.Column('tax_id', sa.String(), nullable=True),
|
|
sa.Column('logo_url', sa.String(), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_organization_id'), 'organization', ['id'], unique=False)
|
|
op.create_index(op.f('ix_organization_name'), 'organization', ['name'], unique=False)
|
|
|
|
# Create user table
|
|
op.create_table(
|
|
'user',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.Column('email', sa.String(), nullable=False),
|
|
sa.Column('hashed_password', sa.String(), nullable=False),
|
|
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('organization_id', sa.Integer(), nullable=True),
|
|
sa.ForeignKeyConstraint(['organization_id'], ['organization.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_user_email'), 'user', ['email'], unique=True)
|
|
op.create_index(op.f('ix_user_full_name'), 'user', ['full_name'], unique=False)
|
|
op.create_index(op.f('ix_user_id'), 'user', ['id'], unique=False)
|
|
|
|
# Create client table
|
|
op.create_table(
|
|
'client',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.Column('name', sa.String(), nullable=False),
|
|
sa.Column('contact_name', sa.String(), nullable=True),
|
|
sa.Column('email', sa.String(), nullable=True),
|
|
sa.Column('phone', sa.String(), nullable=True),
|
|
sa.Column('address', sa.Text(), nullable=True),
|
|
sa.Column('city', sa.String(), nullable=True),
|
|
sa.Column('state', sa.String(), nullable=True),
|
|
sa.Column('postal_code', sa.String(), nullable=True),
|
|
sa.Column('country', sa.String(), nullable=True),
|
|
sa.Column('tax_id', sa.String(), nullable=True),
|
|
sa.Column('notes', sa.Text(), nullable=True),
|
|
sa.Column('organization_id', sa.Integer(), nullable=False),
|
|
sa.Column('created_by_id', sa.Integer(), nullable=True),
|
|
sa.ForeignKeyConstraint(['created_by_id'], ['user.id'], ondelete='SET NULL'),
|
|
sa.ForeignKeyConstraint(['organization_id'], ['organization.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_client_id'), 'client', ['id'], unique=False)
|
|
op.create_index(op.f('ix_client_name'), 'client', ['name'], unique=False)
|
|
|
|
# Create invoice table
|
|
op.create_table(
|
|
'invoice',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.Column('invoice_number', sa.String(), nullable=False),
|
|
sa.Column('status', sa.Enum('draft', 'sent', 'paid', 'overdue', 'cancelled', name='invoicestatus'), nullable=True),
|
|
sa.Column('issue_date', sa.Date(), nullable=False),
|
|
sa.Column('due_date', sa.Date(), nullable=False),
|
|
sa.Column('subtotal', sa.Float(precision=2), nullable=False),
|
|
sa.Column('tax_rate', sa.Float(precision=2), nullable=True),
|
|
sa.Column('tax_amount', sa.Float(precision=2), nullable=True),
|
|
sa.Column('discount', sa.Float(precision=2), nullable=True),
|
|
sa.Column('total', sa.Float(precision=2), nullable=False),
|
|
sa.Column('notes', sa.Text(), nullable=True),
|
|
sa.Column('terms', sa.Text(), nullable=True),
|
|
sa.Column('is_recurring', sa.Boolean(), nullable=True),
|
|
sa.Column('recurring_interval', sa.String(), nullable=True),
|
|
sa.Column('client_id', sa.Integer(), nullable=False),
|
|
sa.Column('organization_id', sa.Integer(), nullable=False),
|
|
sa.Column('created_by_id', sa.Integer(), nullable=True),
|
|
sa.ForeignKeyConstraint(['client_id'], ['client.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['created_by_id'], ['user.id'], ondelete='SET NULL'),
|
|
sa.ForeignKeyConstraint(['organization_id'], ['organization.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_invoice_id'), 'invoice', ['id'], unique=False)
|
|
op.create_index(op.f('ix_invoice_invoice_number'), 'invoice', ['invoice_number'], unique=False)
|
|
|
|
# Create invoice_item table
|
|
op.create_table(
|
|
'invoiceitem',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.Column('description', sa.String(), nullable=False),
|
|
sa.Column('quantity', sa.Float(precision=2), nullable=False),
|
|
sa.Column('unit_price', sa.Float(precision=2), nullable=False),
|
|
sa.Column('amount', sa.Float(precision=2), nullable=False),
|
|
sa.Column('invoice_id', sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(['invoice_id'], ['invoice.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_invoiceitem_id'), 'invoiceitem', ['id'], unique=False)
|
|
|
|
|
|
def downgrade():
|
|
# Drop tables in reverse order
|
|
op.drop_index(op.f('ix_invoiceitem_id'), table_name='invoiceitem')
|
|
op.drop_table('invoiceitem')
|
|
|
|
op.drop_index(op.f('ix_invoice_invoice_number'), table_name='invoice')
|
|
op.drop_index(op.f('ix_invoice_id'), table_name='invoice')
|
|
op.drop_table('invoice')
|
|
|
|
op.drop_index(op.f('ix_client_name'), table_name='client')
|
|
op.drop_index(op.f('ix_client_id'), table_name='client')
|
|
op.drop_table('client')
|
|
|
|
op.drop_index(op.f('ix_user_id'), table_name='user')
|
|
op.drop_index(op.f('ix_user_full_name'), table_name='user')
|
|
op.drop_index(op.f('ix_user_email'), table_name='user')
|
|
op.drop_table('user')
|
|
|
|
op.drop_index(op.f('ix_organization_name'), table_name='organization')
|
|
op.drop_index(op.f('ix_organization_id'), table_name='organization')
|
|
op.drop_table('organization') |