ei}dZddlZddlmZmZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZmZmZdd lmZdd lmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(ee)Z*GddZ+dS)z Garmin API client wrapper for health data retrieval. Supports Garmin Connect China (connect.garmin.cn) with authentication, data fetching, error handling, and retry logic. N)datedatetime)OptionalDictAnyList)Path)Garmin)GarthHTTPError)config)GarminAuthErrorGarminAPIErrorRateLimitError) setup_logger) StepsData HeartRateData SleepData StressDataBodyBatteryDataSpO2DataRespirationData HydrationData FloorsDataIntensityMinutesDataHRVDataRHRDataLifestyleLoggingData)Activity) WeightDatac .eZdZdZ ddeedeedeeddfdZd dZd d Z e j d d e de fd Zd edeefdZd edeefdZd edeefdZd edeefdZd edeefdZdededeefdZd edeefdZd edeefdZ d edee!fdZ"d edee#fdZ$d edee%fdZ&d edee'fdZ(d edee)fdZ*d edee+fdZ,d edee-fdZ.dS)!GarminHealthClientz:Client for fetching health data from Garmin Connect China.Nemailpassword token_storereturnc|p tj|_|p tj|_|p tj|_|jr|jstjd|_d|_ dS)a(Initialize Garmin client. Args: email: Garmin account email (defaults to config.GARMIN_EMAIL) password: Garmin account password (defaults to config.GARMIN_PASSWORD) token_store: Path to token storage directory (defaults to config.TOKEN_STORE) NF) r GARMIN_EMAILr"GARMIN_PASSWORDr# TOKEN_STOREr$validate_credentialsclient_authenticated)selfr"r#r$s 6/root/projects/butler/health/services/garmin_client.py__init__zGarminHealthClient.__init__/si1f1  :F$: &<&*<z * *  ' ) ) )(, #cftd|j t|j|jd|_|jdz }|jdz }|o|}|rtdntd|j t|j|sKtd |jj t|jd|_ td d S#t$r5}td |t!d ||d }~wt"$r}td| |j |jj t|jd|_ tdnB#t&$r5}td|t!d ||d }~wwxYwYd }~d Sd }~wt&$r5}td|t!d||d }~wwxYw)zvAuthenticate with Garmin Connect China. Raises: GarminAuthError: If authentication fails z-Authenticating with Garmin Connect China for T)r"r#is_cnzoauth1_token.jsonzoauth2_token.jsonz.Found existing tokens, attempting to load themz9No existing tokens found, will use credential-based login) tokenstorezSaving tokens for future usez4Successfully authenticated with Garmin Connect ChinazGarmin authentication failed: zAuthentication failed: Nz=Token files not found, retrying with credential-based login: z)>  ? ? ?  > ;<<< !&&s4+;'<'<==="&D  KKN O O O O O H H H LL=!== > > >!"?A"?"?@@a G ` ` ` NN^[\^^ _ _ _ ` !!### !&&s4+;'<'<===&*# Z[[[[ ` ` ` J[JJKKK%&M &M&MNNT__ `\[[[[[ R R R LL@Q@@ A A A!"Ia"I"IJJPQ Q RsUD/E J00F J0I.:A+H&%I.& I%00I  I%%I.. J0;0J++J0c@|jr|jstddS)zEnsure client is authenticated before making API calls. Raises: GarminAuthError: If not authenticated z4Client not authenticated. Call authenticate() first.N)r,r+r )r-s r._ensure_authenticatedz(GarminHealthClient._ensure_authenticateds7 " $+ !F   r0) max_retriesrHc t|D]G} ||i|cS#t$r}|jdkrtd||||dz kri|jdks |jdvrUtjd|zz}t d|dzd|d |d |d tj |Yd }~td ||j|d }~wt$rw}||dz krVtjd|zz}t d|dzd|d |d |d tj |Yd }~.td||d }~wwxYwtd|d)aRetry API call with exponential backoff. Args: func: Function to call max_retries: Maximum retry attempts *args: Positional arguments for func **kwargs: Keyword arguments for func Returns: Result of function call Raises: GarminAPIError: If all retries fail RateLimitError: If rate limit is exceeded zAPI rate limit exceeded: i)irJzAPI call failed (attempt /z): z. Retrying in zs...NzAPI call failed: ) status_codezUnexpected error (attempt zUnexpected API error: z Max retries (z ) exceeded) ranger statusrr RETRY_DELAY_SECONDSr4r>timesleeprr?)r-funcrHargskwargsattemptrCdelays r._retry_api_callz"GarminHealthClient._retry_api_calls5$[))" J" JG! JtT,V,,,,,! [ [ [8s??()HQ)H)HIIqP[1_,,HOOqx:'='="6!W*EENN3GaK33+33RS33',333Ju%%%HHHH%%<%<%EE target_datec |td| ||jj||}|r(t|trt|dkrdS|d}t|| d| d| d| d|S#t$rt$r(}td |Yd}~dSd}~wwxYw) zFetch daily step data. Args: target_date: Date to fetch data for Returns: StepsData model or None if no data available zFetching steps data for rN totalSteps totalDistancecaloriesstepGoal)r total_stepstotal_distance_meterscalories_burned step_goalraw_datazError parsing steps data: )rGr4r7rYr+get_daily_steps isoformat isinstancelistlenrgetrr?r<r-rZ raw_data_listrdrCs r. fetch_stepszGarminHealthClient.fetch_stepsse ""$$$ = ==>>>  00 +%%''%%''M !  =$(G(G 3}K]K]abKbKbt$Q'H $LL66&.ll?&C&C ( Z 8 8",,z22!          LL9a99 : : :44444 sA/D $A(D E !EE c :|td| ||jj|}|sdSt||d|d|d|d|d|S#t$rt$r(}t d |Yd}~dSd}~wwxYw) zFetch daily heart rate data. Args: target_date: Date to fetch data for Returns: HeartRateData model or None if no data available zFetching heart rate data for NrestingHeartRate minHeartRate maxHeartRateaverageHeartRateheartRateZones)rresting_heart_ratemin_heart_ratemax_heart_rateaverage_heart_rateheart_rate_zonesrdzError parsing heart rate data: ) rGr4r7rYr+get_heart_ratesrfrrjrr?r<r-rZrdrCs r.fetch_heart_ratez#GarminHealthClient.fetch_heart_rates9 ""$$$ B[BBCCC ++ +[-B-B-D-DH t  #+<<0B#C#C'||N;;'||N;;#+<<0B#C#C!).>!?!?!         LL>1>> ? ? ?44444 s4C)A4CD2DDc |td| ||jj|}|rd|vrdS|d}t||d|d|d|d|d|d id id |d | S#t$rt$r(}t d|Yd}~dSd}~wwxYw)zFetch daily sleep data. Args: target_date: Date to fetch data for Returns: SleepData model or None if no data available zFetching sleep data for dailySleepDTONsleepTimeSecondsdeepSleepSecondslightSleepSecondsremSleepSecondsawakeSleepSeconds sleepScoresoverallvalue sleepLevels) rtotal_sleep_secondsdeep_sleep_secondslight_sleep_secondsrem_sleep_seconds awake_seconds sleep_score sleep_levelsrdzError parsing sleep data: ) rGr4r7rYr+get_sleep_datarfrrjrr?r<)r-rZrd sleep_dtorCs r. fetch_sleepzGarminHealthClient.fetch_sleeps ""$$$ = ==>>> ++ *K,A,A,C,CH h>>t 1I $-MM2D$E$E#,==1C#D#D$-MM2E$F$F"+--0A"B"B'mm,?@@%MM-<<@@BOOSST[\\%\\-88!            LL9a99 : : :44444 s8D:-C D::E6E11E6c |td| ||jj|}|sdSt||d|d|d|d|d|d|d |d | S#t$rt$r(}t d |Yd}~dSd}~wwxYw) zFetch daily stress data. Args: target_date: Date to fetch data for Returns: StressData model or None if no data available zFetching stress data for NavgStressLevelmaxStressLevelrestStressDurationactivityStressDurationlowStressDurationmediumStressDurationhighStressDurationstressValuesArray) raverage_stress_levelmax_stress_levelrest_stress_duration_seconds activity_stress_duration_secondslow_stress_duration_secondsmedium_stress_duration_secondshigh_stress_duration_secondsstress_timelinerdzError parsing stress data: ) rGr4r7rYr+get_stress_datarfrrjrr?r<rzs r. fetch_stresszGarminHealthClient.fetch_stress:sk ""$$$ >>>??? ++ +[-B-B-D-DH t %-\\2B%C%C!).>!?!?-5\\:N-O-O19>V1W1W,4LL9L,M,M/7||??!            LL@Q@@ A A A44444 sA/D5$BD55E1 E,,E1 start_dateend_datec .|td|d| ||jj||}|sgSg}|D]%} td)idt| dd| di ddd | d d tj | d  d d dtj | d  d dd| dd| dd| dd| dd| dd| dd| dd| dd| d d!| d"d#|}||#t$r(}td$|Yd%}~d%}~wwxYwtd&t%|d'|S#t&$rt$r)}td(|gcYd%}~Sd%}~wwxYw)*zFetch activities within a date range. Args: start_date: Start date end_date: End date Returns: List of Activity models zFetching activities from z to activity_id activityId activity_type activityTypetypeKeyunknown activity_name activityNamerstartTimeLocalZ start_timeduration_secondsdurationdistance_metersdistanceelevation_gain_meters elevationGainelevation_loss_meters elevationLossr^rw averageHRrvmaxHRaverage_speed_mps averageSpeed max_speed_mpsmaxSpeedaverage_cadence%averageRunningCadenceInStepsPerMinuterdzFailed to parse activity: NzFetched z activitieszError fetching activities: )rGr4r7rYr+get_activities_by_daterfrr9rjr fromisoformatreplacerappendr?r>r5rirr<)r-rractivities_data activities raw_activityactivityrCs r.fetch_activitiesz#GarminHealthClient.fetch_activitiess ""$$$ KKKKKLLL0 "22 2$$&&""$$O #  J /   '   $' (8(8(F(F$G$G$G &2&6&6~r&J&J&N&NyZc&d&d&d '3&6&6~&F&F&F &3(,,-=>>FFsBOO$&&&  $,#9(,,-=>>FFsBOO$$$ *6)9)9*)E)E)E )5(8(8(D(D(D /;.>.>.O.O.O /;.>.>.O.O.O ".!1!1*!=!=!= ,8+;+;K+H+H+H  (4'7'7'@'@'@! "+7*:*:>*J*J*J# $'3&6&6z&B&B&B% &)5(8(89`(a(a(a' (".) H,%%h//// NN#C#C#CDDDHHHH KK?3z????? @ @ @         LL:q:: ; ; ;IIIIII sOAK?KG,I53K5 J'?J"K"J''/KL+L LLc |td| ||jj||}|rd|vrdS|d}|sdS|d}|d|}t||dr|ddz nd|d|d |d |d r|d dz nd|d r|d dz nd|d | S#t$rt$r(}t d|Yd}~dSd}~wwxYw)zFetch weight data for a specific date. Args: target_date: Date to fetch data for Returns: WeightData model or None if no data available zFetching weight data for dailyWeightSummariesNr latestWeightweightibmibodyFat bodyWaterboneMass muscleMass sourceType) r weight_kgrbody_fat_percentagebody_water_percentage bone_mass_kgmuscle_mass_kgsourcerdzError parsing weight data: ) rGr4r7rYr+ get_weigh_insrfrjrrr?r<)r-rZrd summariessummary latest_weightrCs r. fetch_weightzGarminHealthClient.fetch_weights ""$$$ >>>???% ++ )%%''%%''H  5XEEt !78I t lG$KK@@M @M@Q@QRZ@[@[e-++H55<>EEnr$((66!            LL:q:: ; ; ;44444 s%A F F DFG)G  Gc |td| ||jj|}|sdSt||d|d|d|d|S#t$rt$r(}t d|Yd}~dSd}~wwxYw) zFetch SpO2 (blood oxygen saturation) data. Args: target_date: Date to fetch data for Returns: SpO2Data model or None if no data available zFetching SpO2 data for N averageSpO2 lowestSpO2 latestSpO2spO2ValueDescriptorsDTOList)r average_spo2min_spo2max_spo2readingsrdzError parsing SpO2 data: ) rGr4r7rYr+ get_spo2_datarfrrjrr?r<rzs r. fetch_spo2zGarminHealthClient.fetch_spo2s* ""$$$ <{<<=== ++ );+@+@+B+BH t %\\-88!l33!l33!&CDD!          LL8Q88 9 9 944444 s4C )A C DDDc|td| ||jj|}|sdSt||d|d|d|S#t$rt$r(}t d|Yd}~dSd}~wwxYw)zFetch respiration rate data. Args: target_date: Date to fetch data for Returns: RespirationData model or None if no data available zFetching respiration data for NavgWakingRespirationValuelowestRespirationValuehighestRespirationValue)raverage_respiration_ratemin_respiration_ratemax_respiration_raterdz Error parsing respiration data: ) rGr4r7rYr+get_respiration_datarfrrjrr?r<rzs r.fetch_respirationz$GarminHealthClient.fetch_respiration)s ""$$$ CkCCDDD ++ 0+2G2G2I2IH t# )16Q)R)R%-\\2J%K%K%-\\2K%L%L!          LL?A?? @ @ @44444 s4B6)A B66C2 C--C2c|td| ||jj|}|sdSt||d|d|S#t$rt$r(}t d|Yd}~dSd}~wwxYw)zFetch hydration tracking data. Args: target_date: Date to fetch data for Returns: HydrationData model or None if no data available zFetching hydration data for N valueInML sweatLossInML)rtotal_intake_mlgoal_mlrdzError parsing hydration data: ) rGr4r7rYr+get_hydration_datarfrrjrr?r<rzs r.fetch_hydrationz"GarminHealthClient.fetch_hydrationLs  ""$$$ AKAABBB ++ . 0E0E0G0GH t! ( [ 9 9 _55!          LL=!== > > >44444 s4B")8B""C6CCc|td| ||jj|}|sdSd}d}|dg}|D]A}|r=t|dkr*||dr|dndz }||dr|dndz }Bt||dkrt|nd|dkrt|nd|d| S#t$rt$r(}t d |Yd}~dSd}~wwxYw) zFetch floors climbed data. Args: target_date: Date to fetch data for Returns: FloorsData model or None if no data available zFetching floors data for NrfloorValuesArrayrLuserFloorsAscendedGoal)rfloors_climbedfloors_descended floor_goalrdzError parsing floors data: )rGr4r7rYr+ get_floorsrfrjririntrr?r<)r-rZrdrrfloor_values_arrayentryrCs r. fetch_floorszGarminHealthClient.fetch_floorsns ""$$$ >>>??? ++ & (=(=(?(?H tN !).@"!E!E , D DDSZZ1__"%(&AeAhhAN$E!H(Ca!C$ 6Dq6H6Hs>222d:JQ:N:N%5!6!6!6TX#<<(@AA!          LL:q:: ; ; ;44444 s4D)B.DE,EEc |td| ||jj|}|sdS|d}|d}d}| |||dzz}t|||||d|S#t$rt$r(}t d|Yd}~dSd}~wwxYw) zFetch intensity minutes data. Args: target_date: Date to fetch data for Returns: IntensityMinutesData model or None if no data available z$Fetching intensity minutes data for NmoderateMinutesvigorousMinutesrLweekGoal)rmoderate_minutesvigorous_minutes total_minutes weekly_goalrdz&Error parsing intensity minutes data: ) rGr4r7rYr+get_intensity_minutes_datarfrjrrr?r<)r-rZrdrrrrCs r.fetch_intensity_minutesz*GarminHealthClient.fetch_intensity_minutessN ""$$$ IKIIJJJ ++ 6 8M8M8O8OH t (||,=>> '||,=>> !M+0@0L 03Ca3G G ' !1!1+$LL44!          LLE!EE F F F44444 s4C )AC DDDc|td| ||jj|}|sdS|di}t|dtr|dini}t||d|d|d|S#t$rt$r(}t d |Yd}~dSd}~wwxYw) zFetch Heart Rate Variability (HRV) data. Args: target_date: Date to fetch data for Returns: HRVData model or None if no data available zFetching HRV data for N hrvSummarybaseline lastNightAvglowUpperrP)r hrv_value baseline_hrvrPrdzError parsing HRV data: )rGr4r7rYr+ get_hrv_datarfrjrgdictrrr?r<)r-rZrd hrv_summaryr#rCs r. fetch_hrvzGarminHealthClient.fetch_hrvsc ""$$$ ;k;;<<< ++ (+*?*?*A*AH t#,,|R88K:D[__U_E`E`bf:g:go{z2666moH %//.99%\\*55"x00!          LL7A77 8 8 844444 s4D )B"D E EEcx|td| ||jj|}|sdSd}|di}|di}|dg}|r)>)@)@H t"& ",,|R88K%//,;;K%//*GLLK Is;//!33%(Q););GQ)G)G%H%H" #5!          LL7A77 8 8 844444 s4C=)BC==D9D44D9c  |td| ||jj|}|sdS|dg sdStd D}|sdSdtdttf fd }dtd tdtfd }|d }|d }|d }td/id|d|dud|r ||dndd|r ||dndd|r ||dndd|r ||dndd|dud|r ||dndd|r ||dndd|r ||dndd|dud |r ||dndd!|r ||dndd"|r ||dndd#|d$dud%|d&dud'|d(dud)|d*dud+|d,dud-|S#t$rt$r(} td.| Yd} ~ dSd} ~ wwxYw)0azFetch lifestyle logging data from Garmin Lifestyle Logging feature. Parses alcohol, caffeine, meal quality/timing, light exercise, and intermittent fasting behaviors logged by the user in Garmin Connect. Args: target_date: Date to fetch data for Returns: LifestyleLoggingData model or None if no data available z$Fetching lifestyle logging data for NdailyLogsReportc3HK|]}|ddkVdS) logStatusYESNrj).0rCs r. z=GarminHealthClient.fetch_lifestyle_logging..5s3GGQQUU;//58GGGGGGr0namer%c~D]8}|d|kr|ddkr|cS9dS)Nr=r8r9r:)r=rlogss r._findz9GarminHealthClient.fetch_lifestyle_logging.._find:sN!%%Eyy((D00UYY{5K5Ku5T5T$ tr0rsubtypec|dgD]@}|d|kr%t|ddcSAdS)Ndetails subTypeNameamountr)rjr)rrAds r._amountz;GarminHealthClient.fetch_lifestyle_logging.._amountAsb9b1177Auu]++w66"1551#5#5666667qr0AlcoholzMorning Caffeinez Late Caffeineralcohol_logged alcohol_beerBEERr alcohol_wineWINEalcohol_spiritSPIRIT alcohol_otherOTHERmorning_caffeine_loggedmorning_caffeine_coffeeCOFFEEmorning_caffeine_teaTEAmorning_caffeine_otherlate_caffeine_loggedlate_caffeine_coffeelate_caffeine_tealate_caffeine_otherlight_exercisezLight Exercise healthy_mealsz Healthy Meals heavy_mealsz Heavy Meals late_mealsz Late Mealsintermittent_fastingzIntermittent Fastingrdz&Error parsing lifestyle logging data: r)rGr4r7rYr+get_lifestyle_logging_datarfrjanyr9rr)rrrr?r<) r-rZrd any_loggedr@rG alcohol_entrymorning_caf_entrylate_caf_entryrCr?s @r.fetch_lifestyle_loggingz*GarminHealthClient.fetch_lifestyle_loggings ""$$$ IKIIJJJC ++ 6 8M8M8O8OH t<< 1266D tGG$GGGGGJ t C HTN       t c c     "E),,M %&8 9 9 "U?33N' [ -D88@MSWW]F;;;RS  @MSWW]F;;;RS  DQWww}h???VW BOUggmW===TU):(E(EQb(h0A8(L(L(LghK\%bWW->%F%F%FabO`'fww/@''J'J'Jef&44%?%?KY%_WW^X%F%F%F^_ ES"Y''.%"@"@"@XY!"IW$]GGNG$D$D$D\]#& %u%566dBB'($eO44D@@)*"E-00<<+,!5..d::-.&+U+A%B%B$%N%N/0"1 6        LLE!EE F F F44444 s*4H *H H !E'H III)NNN)r%N)/__name__ __module__ __qualname____doc__rr9r r/rErGr MAX_RETRIESrrrYrrrmrr{rrrrrrrrrrrrrrrrr rrrr rr+rr4rrgrr0r.r!r!,sFDD $"&&* $$}$3-$d^ $  $$$$.9R9R9R9Rv    /5.@6F6F6F(+6F 6F6F6F6Ft&t&0C&&&&P"D"Xm5L""""H&t&0C&&&&P%%*1E%%%%N(d(x7P((((T??*.? h????B11*1E1111p#d#x/A####J!T!h6O!!!!F 4 H]4K    D,,*1E,,,,\*4*HEYrws ########,,,,,,,,,,,, $$$$$$ 544444,+++++111111 h  ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ r0