U Sh{ @sddlmZmZddlZddlZddlmZddlmZddlZee dZ e se ddZ e dd Z e d d Z e d d Ze ddZdedede de de Z ee ZddZddZddZedkredS)) create_enginetextN) load_dotenv)datetime DATABASE_URLDB_HOSTZ localhostDB_PORTZ5432DB_NAMEZ timesheet_dbDB_USERpostgres DB_PASSWORDz postgresql://:@/c Cszt|trt|}n|}t|ts,WdSd|krBt|dWSd|krvt|dtrvd|dkrvt|ddWSd|krd|krz,t|d}t|d}||WWSYnXd|krt|d}| drt|ddWSWn$t k r}zW5d}~XYnXdS)zExtract duration from JSON dataNdurationdataZ timestampZ end_timestamps) isinstancestrjsonloadsdictfloatrZ fromisoformatZ total_secondsendswith Exception) activity_datarstartendZdur_strer!./fix_duration_data.pyextract_duration_from_jsons2      r#csd}|s |S|tfdddDr8tdd}nxtfdddDr\td d }nTtfd dd Drtd d}n0tfdddDrtdd}n td d}|S)z2Generate realistic duration based on activity typec3s|]}|kVqdSNr!).0ZideZ app_lowerr!r" Nsz.generate_realistic_duration..)codeZvscodeZpycharmZsublimeZ developmenti,ic3s|]}|kVqdSr%r!)r&browserr'r!r"r(Qs)r*ZchromeZfirefoxxiXc3s|]}|kVqdSr%r!)r&Zcommr'r!r"r(Ts)ZslackZteamsZdiscord<c3s|]}|kVqdSr%r!)r&Zdbr'r!r"r(Ws)Zmysqlr Zdatabasei)loweranyrandomZrandint)app_name window_titleZ base_durationr!r'r"generate_realistic_durationCs r2c Cstdtv}td|td}|}td|dtd|d|ddkrxtdW5QRd Std |td }g}d}d}|D]}|\}}} } d } |rt|} | r| dkr|d7}| r| dkrt| | } |d7}|| |ft |d krtd |d| d| ddqtdt |dtd|td||r4| } zd} t dt || D]d}|||| }|D] \} }|td| |dqtdt || t |dt |dq~| tdWn<tk r2}z| td|W5d }~XYnXtd|td}|D]4\}}}}td|d|d |d!d"|dd# qNW5QRXd S)$zFix missing duration dataz=== Fixing Duration Data === z%1. Checking current duration state...z SELECT COUNT(*) as total, COUNT(CASE WHEN duration IS NULL OR duration = 0 THEN 1 END) as needs_fix FROM activity_records z Total records: rz Records needing fix: u& ✅ All records have duration values!Nz4 2. Processing records with missing/zero duration...z SELECT id, activity_data, application_name, window_title FROM activity_records WHERE duration IS NULL OR duration = 0 z ID z: z -> z.0fz secondsz 3. Updating z records...z Extracted from JSON: z Generated realistic: iz UPDATE activity_records SET duration = :duration WHERE id = :id )ridz Updated ru ✅ Duration fix completed!u ❌ Error during update: z 4. Verification...aG SELECT developer_id, COUNT(*) as records, SUM(duration) / 3600 as total_hours, AVG(duration) as avg_seconds FROM activity_records WHERE duration > 0 GROUP BY developer_id ORDER BY developer_id z z records, z.1fz hours total, z s average)printengineZconnectZexecuterZfetchoner#r2appendlenZbeginrangeminZcommitrZrollback)ZconnresultZstatsZupdatesZextracted_countZgenerated_countrecordZ record_idrr0r1rZtransZ batch_sizeiZbatchr Zdev_idZrecordsZhoursZavg_secr!r!r"fix_duration_data^sh          ,  r?__main__)Z sqlalchemyrrrosZdotenvrrr/getenvrrrr r r r7r#r2r?__name__r!r!r!r"s(        ")f