U
    [Mhz                     @   s   d Z ddlZddlmZmZ ddlmZ e  dd Zedkre	d e	d	 e re	d
 e	d e	d e	d e	d e	d e	d ne	d e	d dS )zP
Migration script to add detailed information columns to activity_records table
    N)create_enginetext)load_dotenvc            	      C   sJ  t d} t| } z| }| }ztd ddddg}|D ] }td|  |t| qB|  td |td	}d
d |	 D }td|  W W 5 Q R  W dS  t
k
 r } z,|  td|  W Y W 5 Q R  W dS d}~X Y nX W 5 Q R X W n8 t
k
rD } ztd|  W Y dS d}~X Y nX dS )z.Add new columns for detailed activity trackingZDATABASE_URLu4   🔄 Adding new columns to activity_records table...zDALTER TABLE activity_records ADD COLUMN IF NOT EXISTS file_path TEXTzVALTER TABLE activity_records ADD COLUMN IF NOT EXISTS database_connection VARCHAR(255)zSALTER TABLE activity_records ADD COLUMN IF NOT EXISTS specific_process VARCHAR(255)zLALTER TABLE activity_records ADD COLUMN IF NOT EXISTS detailed_activity TEXTz  Executing: u.   ✅ Database migration completed successfully!aG  
                    SELECT column_name 
                    FROM information_schema.columns 
                    WHERE table_name = 'activity_records' 
                    AND column_name IN ('file_path', 'database_connection', 'specific_process', 'detailed_activity')
                    ORDER BY column_name
                c                 S   s   g | ]}|d  qS )r    ).0rowr   r   ./migrate_detailed_info.py
<listcomp>/   s     z$migrate_database.<locals>.<listcomp>u   ✅ Verified new columns: Tu   ❌ Error during migration: FNu   ❌ Database connection error: )osgetenvr   ZconnectZbeginprintZexecuter   ZcommitZfetchall	ExceptionZrollback)	Zdatabase_urlZengineZconnZtransZnew_columnsZsqlresultZadded_columnser   r   r   migrate_database   s8    

2r   __main__z5Database Migration: Adding Detailed Activity Trackingz<============================================================u'   
🎉 Migration completed successfully!z
New features available:u$   • 📁 File path tracking for IDEsu'   • 🗄️ Database connection detailsu'   • ⚙️ Specific process informationu'   • 📋 Enhanced activity descriptionsz5
Restart your backend server to use the new features!u   
❌ Migration failed!z4Please check the error messages above and try again.)
__doc__r
   Z
sqlalchemyr   r   Zdotenvr   r   __name__r   r   r   r   r   <module>   s$   2
