U %h>@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?) development productivitydatabasebrowser entertainmentsystemotherg333333?g?gffffff?g?) peak_hourseveningnight late_nightg?g?g?) short_focus medium_focus long_focus deep_focus)category_weightstime_multipliers focus_bonusesselfr:/var/www/html/timesheet/backend/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_partrrr format_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/rrr get_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_timedurationr1r2i,r3r4r1i) r2r*r4rangelenabs total_secondsappendcopy) rr1focus_sessionsZcurrent_sessioniactivityZ prev_activityZtime_gapZ same_categoryrrr calculate_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_minutesrrr get_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_breakdownr<longest_focus_sessionlongest_focus_session_formattedaverage_focus_sessionaverage_focus_session_formattedrecommendationsN+No activity data available for this period.Z MockActivityrr2r4application_namer*)r2r4rTr*)rget_activity_data Exceptionprinttyper:calculate_productivity_score) rrErF aw_client activity_dataeZmock_activitiesr>Z mock_activityrrr /calculate_productivity_score_from_activitywatchsX  zFProductivityCalculator.calculate_productivity_score_from_activitywatchN)dbuser_idrErFr#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 scoreNrrS)rGrHrIrLrMr<rRcss|] }|jVqdSNr4).0r>rrr szFProductivityCalculator.calculate_productivity_score..r r r4r2d) time_spentrIZproductivity_rateZpercentage_of_totalr5icSsg|] }|dqSrarrbsrrr szGProductivityCalculator.calculate_productivity_score..cSsg|] }|dqSrarrfrrr rhsr$cSsg|] }|dqSrarrfrrr rhscSsg|] }|dqSrarrfrrr rh s)rGrHrIrJrKrLrMr<rNrOrPrQrRZtime_distributionZproductivity_trend)querymodelsActivityRecordfilterrr_r*order_byascallsumr2r4rgetr0rr?rDminitems_generate_recommendationsroundr)r7max_get_time_distribution_get_productivity_trend)rr^r_rErFr1rHrIZcategory_scoresZcategory_timesr>r2r4Z base_weightZ time_categoryZtime_multiplierZ weighted_timer<Zfocus_bonus_timesessionZ focus_bonusZ bonus_timerLrGrMscorerRrrr rYs              z3ProductivityCalculator.calculate_productivity_score)rz categoriesr<rHr#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 recommendationsrCuI🔴 Low productivity detected. Consider reducing distracting activities.r%u9🟡 Moderate productivity. There's room for improvement.u3🟢 Good productivity! Keep up the excellent work.rrerrg333333?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.)r:rqr7) rrzr{r<rHrRZ browser_timeZentertainment_timeZdev_timerrr rt(s,              z0ProductivityCalculator._generate_recommendationscCs<i}|D].}|jj}||kr$d||<|||j7<q|S)z7Get time distribution across different hours of the dayr)r*r/r4)rr1Zhour_distributionr>r/rrr rwOsz-ProductivityCalculator._get_time_distributionc s|sdddSt|d}|d|}||d}fdd}||}||}||}t|dkrhd} n|dkrvd } nd } | t|d dS) z&Calculate productivity trend over timeZstabler)trendchanger|NcsH|sdStdd|D}tfdd|D}|dkrD||dSdS)Nrcss|] }|jVqdSr`rarbarrr rciszaProductivityCalculator._get_productivity_trend..calc_half_productivity..c3s$|]}|jj|jdVqdS)r N)r4rrqr2rrrr rcjsrd)rp)Zhalf_activitiesrHrIrrr calc_half_productivityes zNProductivityCalculator._get_productivity_trend..calc_half_productivityZ improvingZ decliningr5)r7r8ru) rr1Z mid_pointZ first_halfZ second_halfrZfirst_productivityZsecond_productivityrr~rrr rx[s      z.ProductivityCalculator._get_productivity_trend)N)__name__ __module__ __qualname__r!floatstrr)rr0rrjrkrr?rDr]rr(rYrtrwrxrrrr rs& .7 m ' r) __doc__rtypingrrrjmy_activitywatch_clientr sqlalchemyrsqlalchemy.ormrrrrrr s