U : iI@sddlmZmZmZmZddlmZddlmZddl m Z m Z ddl m Z m Z mZddlmZddlmZddlZddlZeZeed d d Zeed d dZeedddZeedddZeeeedddZedededeefee ee eedddZedededeefee ee eedddZ edededeefe ee eed d!d"Z!eed#d$d%Z"ed&eefeed'd(d)Z#dS)*) APIRouterDepends HTTPExceptionQuery)Session)text)datetimetimezone)OptionalListDict)get_db)ActivityCategorizerN)secreturncCsZt|pd}t|d\}}t|d\}}|r<|d|dS|rP|d|dS|dS)Nr<zh mzm s)intdivmod)rhrrrr ./activity_categorization_api.py_format_secondss r)rrcCs|pddS)Nu–—-)strip)rrrr _clean_tokensr)pathrcs|sdS|dd}dd|dD}|s0dSd|dkrH|dd}|sPdSd d d d d ddhfdd|D}|s||}|dS)zT Guess project from file path by taking the last meaningful directory name. r\/cSsg|] }|r|qSrr.0prrr $sz0infer_project_from_file_path...NsrcZappZappslibZdistZbuildZpubliccsg|]}|kr|qSrlowerr"Zgenericrrr%-s )replacesplit)rpartscandrr,rinfer_project_from_file_paths   r1)titlerc s|sdS|dddd}dd|dD}ddd d d d d ddh fdd|D}|s`dS|d}|dkrxdStd|rdS|S)u Extract project/site/workspace name from window title patterns. Examples: 'crud.py - mahindra-manulife-retail - Visual Studio Code' 'Plugins « Naturals — WordPress – Google Chrome' ru—-u–cSsg|]}t|rt|qSr)rr"rrrr%<sz3infer_project_from_window_title..z google chromeZchromezmozilla firefoxZfirefoxzmicrosoft edgeZedgezvisual studio codecodeZvscodecsg|]}|kr|qSrr*r"Zdroprrr%As r'>untitlednew tabz\.[a-zA-Z0-9]{1,6}$)r-r.r+research)r2tr/r0rr5rinfer_project_from_window_title2s.  r;)existing file_path window_titlercCsL|pd}|r |dkr |St|p(d}|r4|St|pfpZwtrrrchoose_project_nameKs   r@z'/api/activity-categories/{developer_id}) developer_id start_dateend_datedbc!szt}|r"t|dd}nttjjddddd}|rTt|dd}n ttj}td}|||||d }t d|dd|d d |d d |d d|d td} || |||d } iiiid} ddddddddddddd} d} | D]}|d}|d|d |d p|D}|| | d?| t| d0d ||t| d@WStk r} ztdAdBt| dCW5d} ~ XYnXdS)Da Get activities categorized into Productive, Browser, Server, Non-work. - Groups raw events by (category + window_title) for detail. - Ensures every activity has a reasonable project_name (inferred if needed). Durations are in SECONDS. Z+00:00rZhourZminutesecondZ microseconda SELECT COUNT(*) as total_records, SUM(duration) as total_duration, AVG(duration) as avg_duration, MIN(duration) as min_duration, MAX(duration) as max_duration FROM activity_records WHERE developer_id = :dev_id AND timestamp >= :start_date AND timestamp <= :end_date dev_idrBrCzDebug - Total records: z, Total duration: z, Avg: z, Min: z, Max: a  SELECT id, developer_id, application_name, window_title, duration, timestamp, url, file_path, project_name, project_type, category FROM activity_records WHERE developer_id = :dev_id AND timestamp >= :start_date AND timestamp <= :end_date ORDER BY timestamp DESC ) productivebrowserservernon-work)countdurationrN ) idrAapplication_namer>rT timestampurlr= project_name project_typeZexisting_categoryr>r\categoryrPZUntitledr] subcategory confidencer^r=r_r`)r>r\rTactivity_countZfirst_timestamplast_timestamprarbrcr^r=r_r`individual_activitiesrTrdrerfr[)r[rTr]rSrZduration_hoursrZduration_displayrrrcSs|dS)NrTr)xrrr z,get_categorized_activities..T)keyreversed percentagecSsi|]\}}||ddqS)NrZr)r#catitemsrrr sz.get_categorized_activities..startend)rA date_rangeZ statisticstotal_duration_hoursactivities_by_categorytop_activities_by_categoryZproductivity_scorezError categorizing activities: Z status_codeZdetail)rr fromisoformatr-nowr utcrexecutefetchoneprintfetchall isoformatr@get_detailed_categoryrappendroroundpopsortcalculate_productivity_score Exceptionrstr)!rArBrCrD categorizerrrrsZ debug_queryZ debug_resultZqueryresultZactivities_groupedcategory_statsZtotal_durationrowZ raw_projectZactivityZcat_inforaZwindow_title_keygrprvrnZ grouped_dictro_gdZminsZdurZhoursrwerrrget_categorized_activitiesas    4              "    rz./api/update-activity-categories/{developer_id}c s6zt}|rt|ddnttjjddddd}|rNt|ddn ttj}|td|||d }d}|D]R} | | dpd| d pd} |td | d | d | d | dd|d7}q~| d|d|ddWSt k r0} z"| tddt| dW5d} ~ XYnXdS)zJUpdate the category/subcategory/confidence columns for the selected range.rErFrrGz SELECT id, window_title, application_name FROM activity_records WHERE developer_id = :dev_id AND timestamp >= :start_date AND timestamp <= :end_date rIrKrrLz UPDATE activity_records SET category = :category, subcategory = :subcategory, category_confidence = :confidence WHERE id = :id rarbrc)rarbrcr[TzSuccessfully updated z activities)successZ updated_countmessagerxzError updating categories: ryN)rrrzr-r{r r|r}rrrZcommitrZrollbackrr) rArBrCrDrrrrsrupdatedrrnrrrrupdate_activity_categories+s0"     rz/api/category-summary)rBrCrDc sTz|rt|ddnttjjddddd}|rJt|ddn ttj}|td||d}gd}}|D]D}|dpd} | |d|d| t | d d|d d || 7}q~|D]&} |rt | d |d dnd| d<q| | d|t |d ddWSt k rN} zt ddt| dW5d} ~ XYnXdS)z;Aggregate across all developers by category for the period.rErFrrGa SELECT CASE WHEN category IN ('productive', 'browser', 'server', 'non-work') THEN category ELSE 'browser' END as category, COUNT(*) as activity_count, SUM(duration) as total_duration, COUNT(DISTINCT developer_id) as developer_count FROM activity_records WHERE timestamp >= :start_date AND timestamp <= :end_date GROUP BY CASE WHEN category IN ('productive', 'browser', 'server', 'non-work') THEN category ELSE 'browser' END ORDER BY total_duration DESC )rBrCrLrKrrM)rardtotal_duration_secondsruZdeveloper_countrrlrmrq)rtsummaryrurxz Error getting category summary: ryN)rrzr-r{r r|r}rrrrrrrr) rBrCrDrrrsrrtotalrritemrrrrget_category_summaryUs6"     $ r)rrc Cs|didd}|didd}|didd}|didd}|||d}||||}||}|dkr||dnd}tt|dd S) NrOrTrrQrPrRg?rlrL)getrmin) rZproductive_timeZ server_timeZ browser_timeZ non_work_timeZproductive_totalZ total_timeZ work_timeZscorerrrrsrz#/api/debug-durations/{developer_id})rArDc spz.|tdd|i}g}|D]}|dp2d}||d|dt|d|dkrht|dddnt|ddt|d pdd|d r|d nd |d r|d nd d q$|tdd|i}|dpdd}|d|d|dpdt|dt|d pdd|d |d dd|ddWStk rj}ztddt |dW5d }~XYnXd S)z.Quick look at durations & top windows (today).a SELECT window_title, COUNT(*) as count, SUM(duration) as total_duration, AVG(duration) as avg_duration, MIN(timestamp) as first_seen, MAX(timestamp) as last_seen FROM activity_records WHERE developer_id = :dev_id AND timestamp >= CURRENT_DATE GROUP BY window_title ORDER BY total_duration DESC LIMIT 20 rJrLrrKrrrrMrNNrV)r>rdrZtotal_duration_displayZavg_duration_secondsZ first_seenZ last_seena SELECT COUNT(*) as total_records, COUNT(DISTINCT window_title) as unique_windows, SUM(duration) as total_duration_seconds, AVG(duration) as avg_duration_seconds, MIN(duration) as min_duration_seconds, MAX(duration) as max_duration_seconds FROM activity_records WHERE developer_id = :dev_id AND timestamp >= CURRENT_DATE rz0ActivityWatch records activities every 6 seconds)Z total_recordsZunique_windowsrruZavg_duration_per_record_secondsZmin_duration_secondsZmax_duration_secondsZrecording_intervalz@Durations are in seconds. ActivityWatch polls every few seconds.)rZtop_activities_groupedZnoterxzError debugging durations: ry) r}rrrrrr~rrr) rArDrZ activitiesrZ total_secondsZstatsZ total_hoursrrrrdebug_activity_durationssD  *    r)$ZfastapirrrrZsqlalchemy.ormrZ sqlalchemyrrr typingr r r Zdatabaser Zactivity_categorizerrosr8Zrouterfloatrrrr1r;r@rrZpostrrrrrrrrs^     J)7