mirror of
https://github.com/DRYTRIX/TimeTracker.git
synced 2026-05-24 07:10:21 -05:00
Dries Peeters
f3a3a40480
This commit introduces two major features: 1. Project Custom Fields: Add custom_fields JSON column to projects table (migration 085), support for flexible custom data storage, display and edit in project views 2. File Attachments System: Add project_attachments and client_attachments tables (migration 086), new ProjectAttachment and ClientAttachment models, full CRUD operations, file upload/download/delete, client-visible attachments support Additional improvements: Enhanced data tables, updated project/client/invoice/timer views, improved UI for attachments and custom fields management
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
"""Add custom fields to projects
|
|
|
|
Revision ID: 085_add_project_custom_fields
|
|
Revises: 084_add_custom_field_definitions
|
|
Create Date: 2025-01-28
|
|
|
|
This migration adds:
|
|
- custom_fields JSON column to projects table for flexible custom data storage
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '085_add_project_custom_fields'
|
|
down_revision = '084_custom_field_definitions'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def _has_column(inspector, table_name: str, column_name: str) -> bool:
|
|
"""Check if a column exists in a table"""
|
|
try:
|
|
return column_name in [col['name'] for col in inspector.get_columns(table_name)]
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
def upgrade():
|
|
"""Add custom_fields to projects table"""
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
|
|
# Add custom_fields column to projects table if it doesn't exist
|
|
if 'projects' in inspector.get_table_names():
|
|
if not _has_column(inspector, 'projects', 'custom_fields'):
|
|
# Use JSONB for PostgreSQL, JSON for SQLite
|
|
try:
|
|
op.add_column('projects', sa.Column('custom_fields', postgresql.JSONB(astext_type=sa.Text()), nullable=True))
|
|
except Exception:
|
|
# Fallback to JSON for SQLite
|
|
op.add_column('projects', sa.Column('custom_fields', sa.JSON(), nullable=True))
|
|
|
|
|
|
def downgrade():
|
|
"""Remove custom_fields from projects table"""
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
|
|
# Remove custom_fields column from projects table
|
|
if 'projects' in inspector.get_table_names():
|
|
if _has_column(inspector, 'projects', 'custom_fields'):
|
|
op.drop_column('projects', 'custom_fields')
|
|
|