U [Mh>@s^dZddlmZddlmZmZddlZddlmZddlm Z ddl m Z Gdd d Z dS) zP Productivity Calculator - Analyze user productivity based on activity patterns )datetime)DictListN)ActivityWatchClient)and_)Sessionc@seZdZddZeedddZeedddZe e j e e d d d Z eed d dZeee dddZdeeeee dddZee e ee edddZe e j e d ddZe e j e d ddZdS)ProductivityCalculatorcCs:dddddddd|_dd d d d |_d dddd|_dS)N?gffffff?皙?gg?g333333?) developmentZ productivityZdatabasebrowser entertainmentsystemotherg333333?g?gffffff?g?) peak_hourseveningnight late_nightg?g?g?) short_focus medium_focus long_focus deep_focus)category_weightstime_multipliers focus_bonusesselfr./productivity_calculator.py__init__s$  zProductivityCalculator.__init__)hoursreturncCs`|dkr dSt|d}|d}|d}|dkr:|dS|dkrL|dS|d|dSdS)z@Convert decimal hours to readable format (e.g., 5.5 -> '5h 30m')r0h<mhzh N)int)rr Z total_minutesZ hours_partZ minutes_partrrrformat_time_readable+s   z+ProductivityCalculator.format_time_readable) timestampr!cCsb|j}d|krdkr"nndSd|kr6dkr>nndSd|krRdkrZnndSdSd S) z0Categorize time of day for productivity analysis rrrrN)hour)rr(r-rrrget_time_category;sz(ProductivityCalculator.get_time_category) activitiesr!c Cs |sgSg}|dj|dj|dj|dj|dgd}tdt|D]}||}||d}t|j|j}|j|dk}|dkr|r|j|d<|d|j7<|d|qH|dd kr|||j|j|j|j|gd}qH|dd kr|||S) z:Identify focus sessions (continuous work on same category)r)categoryZ start_timeend_timedurationr/r0i,r1r2r/i) r0r(r2rangelenabsZ total_secondsappendcopy) rr/focus_sessionsZcurrent_sessioniactivityZ prev_activityZtime_gapZ same_categoryrrrcalculate_focus_sessionsHs:      z/ProductivityCalculator.calculate_focus_sessions)duration_secondsr!cCsX|d}|dkr|jdS|dkr,|jdS|dkr>|jdS|dkrP|jdSd Sd S) z4Get focus bonus multiplier based on session durationr#rxrrrr N)r)rr=Zduration_minutesrrrget_focus_bonusvs    z&ProductivityCalculator.get_focus_bonus) start_dateend_dater!c Cst}z|||}WnVtk rl}z8td|ddddddiddddddgd WYSd}~XYnX|sddddddiddddddgd Sg}|D]6}tdd |d |d |d |d d}||q|dd|||S)z=Calculate productivity score directly from ActivityWatch dataz#Error fetching ActivityWatch data: rr"z"Error fetching ActivityWatch data.) overall_score total_timeproductive_timetotal_time_formattedproductive_time_formattedproductivity_percentagecategory_breakdownr9longest_focus_sessionlongest_focus_session_formattedaverage_focus_sessionaverage_focus_session_formattedrecommendationsN+No activity data available for this period.Z MockActivityrr0r2application_namer()r0r2rQr()rZget_activity_data Exceptionprinttyper7calculate_productivity_score) rrBrCZ aw_clientZ activity_dataeZmock_activitiesr;Z mock_activityrrr/calculate_productivity_score_from_activitywatchsX  zFProductivityCalculator.calculate_productivity_score_from_activitywatchN)dbuser_idrBrCr!cCs|dkrL|tjttjj|ktjj|ktjj|ktjj }|sfddddigdgdSt dd|D}d}i}i} |D]} | j } | j } |j | d} || j}|j|d}| | |}||7}| |krd|| <d| | <|| |7<| | | 7<q||}d}|D]B}||d }|j |d d} |d | |d}||7}q ||7}|dkr||d nd}td |}i}|D]X\} }| | || | dkr|| | d nd|dkr| | |d ndd || <q|||||}t|d ||||d||dt|d |t||rRtdd|Dnd|rv|tdd|Ddnd|rt dd|Dt|nd|r|t dd|Dt|dnd|||||dS)z*Calculate comprehensive productivity scoreNrrP)rDrErFrIrJr9rOcss|] }|jVqdSNr2).0r;rrr szFProductivityCalculator.calculate_productivity_score..r r r2r0d) time_spentrFZproductivity_rateZpercentage_of_totalr3icSsg|] }|dqSr[rr\srrr szGProductivityCalculator.calculate_productivity_score..cSsg|] }|dqSr[rr`rrrrbsr"cSsg|] }|dqSr[rr`rrrrbscSsg|] }|dqSr[rr`rrrrb s)rDrErFrGrHrIrJr9rKrLrMrNrOZtime_distributionZproductivity_trend)ZquerymodelsActivityRecordfilterrrYr(Zorder_byZascallsumr0r2rgetr.rr<rAminitems_generate_recommendationsroundr'r5max_get_time_distribution_get_productivity_trend)rrXrYrBrCr/rErFZcategory_scoresZcategory_timesr;r0r2Z base_weightZ time_categoryZtime_multiplierZ weighted_timer9Zfocus_bonus_timeZsessionZ focus_bonusZ bonus_timerIrDrJscorerOrrrrUs              z3ProductivityCalculator.calculate_productivity_score)rp categoriesr9rEr!c Csg}|dkr|dn|dkr,|dn |d|didd}|d idd}||d krt|d ||d kr|d t|dkr|dnt|dkr|d|didd}|dkr||dkr|dn |d|S)z2Generate personalized productivity recommendationsr@uI🔴 Low productivity detected. Consider reducing distracting activities.r#u9🟡 Moderate productivity. There's room for improvement.u3🟢 Good productivity! Keep up the excellent work.r r_rr g333333?uT🌐 High browser usage detected. Consider using website blockers during work hours.r uI🎮 Significant entertainment time. Try scheduling specific break times.uO🎯 Try to create longer focus sessions (30+ minutes) for better productivity.uQ⚡ Many short sessions detected. Consider consolidating work into longer blocks.r g?u?💻 Consider dedicating more time to focused development work.uB💻 Great development focus! You're spending quality time coding.)r7rhr5) rrprqr9rErOZ browser_timeZentertainment_timeZdev_timerrrrk(s,              z0ProductivityCalculator._generate_recommendationscCs<i}|D].}|jj}||kr$d||<|||j7<q|S)z7Get time distribution across different hours of the dayr)r(r-r2)rr/Zhour_distributionr;r-rrrrnOsz-ProductivityCalculator._get_time_distributionc s|sdddSt|d}|d|}||d}fdd}||}||}||}t|dkrhd} n|dkrvd } nd } | t|d dS) z&Calculate productivity trend over timeZstabler)trendchangerrNcsH|sdStdd|D}tfdd|D}|dkrD||dSdS)Nrcss|] }|jVqdSrZr[r\arrrr]iszaProductivityCalculator._get_productivity_trend..calc_half_productivity..c3s$|]}|jj|jdVqdS)r N)r2rrhr0rvrrrr]jsr^)rg)Zhalf_activitiesrErFrrrcalc_half_productivityes zNProductivityCalculator._get_productivity_trend..calc_half_productivityZ improvingZ decliningr3)r5r6rl) rr/Z mid_pointZ first_halfZ second_halfrxZfirst_productivityZsecond_productivityrurtrrrro[s      z.ProductivityCalculator._get_productivity_trend)N)__name__ __module__ __qualname__rfloatstrr'rr.rrcrdrr<rArWrr&rUrkrnrorrrrrs& .7 m ' r) __doc__rtypingrrrcZmy_activitywatch_clientrZ sqlalchemyrZsqlalchemy.ormrrrrrrs