Ji>ddZddlmZddlmZmZddlZddlmZddlm Z ddl m Z Gdd Z y) zP Productivity Calculator - Analyze user productivity based on activity patterns )datetime)DictListN)ActivityWatchClient)and_)Sessionc eZdZdZdedefdZdedefdZde e jde e fdZ d edefd Zd ed ede fd Z ddeded ed ede f dZdede de dede ef dZde e jde fdZde e jde fdZy)ProductivityCalculatorcRdddddddd|_dd d d d |_d dddd|_y)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_bonuses)selfs =E:\timesheet\timesheet_new\backend\productivity_calculator.py__init__zProductivityCalculator.__init__sS  !  !   hoursreturncv|dk(ryt|dz}|dz}|dz}|dk(r|dS|dk(r|dS|d|dS)z@Convert decimal hours to readable format (e.g., 5.5 -> '5h 30m')r0h<mhzh )int)r r$ total_minutes hours_part minutes_parts r!format_time_readablez+ProductivityCalculator.format_time_readable+si A:EBJ "b( $r) ?"^1% % Q  \# # \L>3 3r# timestampcl|j}d|cxkrdkryd|cxkrdkryd|cxkrdkryyy) z0Categorize time of day for productivity analysis rrrr)hour)r r0r6s r!get_time_categoryz(ProductivityCalculator.get_time_category;sI~~ >r>  4_"_ 4_"_ r# activitiesc|sgSg}|dj|dj|dj|dj|dgd}tdt |D]}||}||dz }t |j|jz j }|j|dk(}|dkr=|r;|j|d<|dxx|jz cc<|dj||dd k\r|j|j|j|j|j|j|gd}|dd k\r|j||S) z:Identify focus sessions (continuous work on same category)r)category start_timeend_timedurationr8r:i,r<r=r8i) r:r0r=rangelenabs total_secondsappendcopy) r r8focus_sessionscurrent_sessioniactivity prev_activitytime_gap same_categorys r!calculate_focus_sessionsz/ProductivityCalculator.calculate_focus_sessionsHs{I"1 ..$Q-11"1 //"1 ..%a=/  q#j/*A!!}H&q1u-MH..1H1HHWWYZH$--1LLM3=.6.@.@ + +x/@/@@+ -44X>#:.$6"))/*>*>*@A!) 1 1"*"4"4 ( 2 2 ( 1 1#+* #%+6 : &$ .  ! !/ 2r#duration_secondsc|dz }|dk\r|jdS|dk\r|jdS|dk\r|jdS|dk\r|jdSy ) z4Get focus bonus multiplier based on session durationr(rxrrrr )r)r rMduration_minutess r!get_focus_bonusz&ProductivityCalculator.get_focus_bonusvst+b0 s "%%l3 3  $%%l3 3  #%%n5 5  #%%m4 4r# start_dateend_datecpt} |j||}|sddddddiddddddgd Sg}|D]6}t dd |d |d |d |d d}|j |8|j dd|||S#t$r)}td|ddddddiddddddgd cYd}~Sd}~wwxYw)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_breakdownrElongest_focus_sessionlongest_focus_session_formattedaverage_focus_sessionaverage_focus_session_formattedrecommendationsN+No activity data available for this period. MockActivityr:r=application_namer0)r:r=rfr0)rget_activity_data ExceptionprinttyperCcalculate_productivity_score) r rTrU aw_client activity_dataemock_activitiesrH mock_activitys r!/calculate_productivity_score_from_activitywatchzFProductivityCalculator.calculate_productivity_score_from_activitywatchs+')  %77 HMM&!"#$(,-1+,&("#)*37)*37$Q#R "%HD$Z0$Z0$,-?$@%k2 6M  " "= 1&00tZSbcc]  7s; zFProductivityCalculator.calculate_productivity_score..sF:x**:r r r=r:d) time_spentrYproductivity_ratepercentage_of_totalr>ir')rWrXrYrZr[r\r]rEr^r_r`rarbtime_distributionproductivity_trend)querymodelsActivityRecordfilterrrsr0order_byascallsumr:r=rgetr7rrLrSminitems_generate_recommendationsroundr/r@max_get_time_distribution_get_productivity_trend)r rrrsrTrUr8rXrYcategory_scorescategory_timesrHr:r= base_weight time_categorytime_multiplier weighted_timerEfocus_bonus_timesession focus_bonus bonus_timer\rWr]scorerbss r!rkz3ProductivityCalculator.calculate_productivity_scoresb  &"7"78??))11W<))33zA))33x? hv,,66::<=cce !"#$+,&("$$Q#R F:FF #H((H((H//33HcBK!2283E3EFM"3377 sKO%{2_DM } ,O.,-)+,x( H % 6 % 8 $ 0 $+#066zB%G..wz/BCK//33GJ4GMK ,{:kC>OPJ  *  & ++KUWX.?Z#?##E^_C!89  .446OHe,X6#(Q_`hQilmQmenX.F&F&LstXbefXfx(@:(MPS(Slm , x ( 788 -~z  #=!4$.$($=$=j4>O$P)-)B)B?UYCY)Z',-Da'H"4!.1UcS)PA!J-)P%QijHV04/H/HN;NqQz]N;">z"J'  *Q;)PO_s N8 N$ N) N."r categoriesrErXcg}|dkr|jdn(|dkr|jdn|jd|jdijdd}|jd ijdd}||d zkDr|jd ||d zkDr|jd t|dkr|jdnt|dkDr|jd|jdijdd}|dkDr,||dzkr|jd|S|jd|S)z2Generate personalized productivity recommendationsrQuI🔴 Low productivity detected. Consider reducing distracting activities.r(u9🟡 Moderate productivity. There's room for improvement.u3🟢 Good productivity! Keep up the excellent work.rr|rrg333333?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.rg?u?💻 Consider dedicating more time to focused development work.uB💻 Great development focus! You're spending quality time coding.)rCrr@) r rrrErXrb browser_timeentertainment_timedev_times r!rz0ProductivityCalculator._generate_recommendations(sW 2:  " "#n o RZ  " "#^ _  " "#X Y"~~i488qI '^^OR@DD\STU *s* *  " "#y z  S 0 0  " "#n o ~  "  " "#t u  1 $  " "#v w>>-488qI a<*s**&&'hi &&'klr#ci}|D]8}|jj}||vrd||<||xx|jz cc<:|S)z7Get time distribution across different hours of the dayr)r0r6r=)r r8hour_distributionrHr6s r!rz-ProductivityCalculator._get_time_distributionOsU"H%%**D,,*+!$' d #x'8'8 8 # # ! r#c|sdddSt|dz}|d|}||d}fd}||}||}||z }t|dkrd} n |dkDrd} nd } | t|d dS) z&Calculate productivity trend over timestabler)trendchangerNct|sytd|D}tfd|D}|dkDr||z dzSdS)Nrc34K|]}|jywrvrw)rxas r!ryzaProductivityCalculator._get_productivity_trend..calc_half_productivity..isAAQZZrzc3K|]7}|jjj|jdz9yw)r N)r=rrr:)rxrr s r!ryzaProductivityCalculator._get_productivity_trend..calc_half_productivity..js9"(A T2266qzz3GG(s=Ar{)r)half_activitiesrXrYr s r!calc_half_productivityzNProductivityCalculator._get_productivity_trend..calc_half_productivityesN"AAAJ!"("OOj036 Pq Pr# improving decliningr>)r@rAr) r r8 mid_point first_half second_halfrfirst_productivitysecond_productivityrrs ` r!rz.ProductivityCalculator._get_productivity_trend[s%3 3 Oq(  +  ,  Q4J?4[A$'99 v;?E aZEE%*:;;r#rv)__name__ __module__ __qualname__r"floatstrr/rr7rrrrrLrSrqrr+rkrrrrer#r!r r s/ 84%4C4  8    ,48M8M3N,SWX\S],\  % 4d(4d^f4dko4dn[_k wk k 19k EMk cgk Z%u%$%26%DI%NRSVi%N !f6K6K1L !QU !!<$v7L7L2M!rs* 7"n<n