Yi%dZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZdd lmZmZdd lmZdd lmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'ee(Z)GddZ*dS)zv Storage service for health data. Handles JSON file storage and database indexing for health metrics and activities. N)date)Path)OptionalUnionDictAny) BaseModel)config)HealthRepository) StorageErrorDataValidationError) setup_logger) StepsData HeartRateData SleepData StressDataBodyBatteryDataSpO2DataRespirationData HydrationData FloorsDataIntensityMinutesDataHRVDataRHRDataLifestyleLoggingData)Activity) WeightDatac4eZdZdZddeeddfdZ ddeded e defd Z dd ed ed e defd Z de e eeeeeeeeeeeeefdedefdZdededeeeeffdZdedefdZ d edeeeeffdZ!dedede fdZ"d ede fdZ#dS) HealthStoragezCService for storing health data to JSON files and database indexes.Ndata_dirreturncR|p tj|_t|_dS)zInitialize storage service. Args: data_dir: Optional data directory path (defaults to config.DATA_DIR) N)r DATA_DIRr r repo)selfr s 0/root/projects/butler/health/services/storage.py__init__zHealthStorage.__init__*s# !3FO $&& T storage_path target_date create_dirsct|j}|jd}|d}|j|z |z |z |z }|r|jdd|S)a.Get file path for a data item. Args: storage_path: Relative storage path (e.g., "daily_metrics/steps") target_date: Date of the data create_dirs: Whether to create directories if they don't exist Returns: Full path to JSON file 02d.jsonTparentsexist_ok)stryearmonth isoformatr parentmkdir)r%r)r*r+r3r4filename file_paths r&_get_file_pathzHealthStorage._get_file_path3s;#$$$**!++--444ML047%?(J  @   " "4$ " ? ? ?r( activity_id activity_datect|j}|jd}|d}|jdz |z |z |z }|r|jdd|S)aGet file path for an activity. Args: activity_id: Activity ID activity_date: Date of the activity create_dirs: Whether to create directories Returns: Full path to activity JSON file r-r. activitiesTr/)r2r3r4r r6r7)r%r;r<r+r3r4r8r9s r&_get_activity_file_pathz%HealthStorage._get_activity_file_pathLsv=%&& &,,!(((ML047%?(J  @   " "4$ " ? ? ?r(data metric_typec t|tstdt||tjvrt d|tj|d}|||j}| dd}t|dd 5}tj ||d d d d d n #1swxYwYtd|d|jd||j||j|d|S#t$rt"$r;}td|d|t d|d||d }~wwxYw)abSave a daily metric to JSON file and update database index. Args: data: Pydantic model instance metric_type: Type of metric (e.g., "steps", "sleep") Returns: Path to saved file Raises: StorageError: If save fails DataValidationError: If data validation fails z#Data must be a Pydantic model, got zUnknown metric type: r)jsonTmode exclude_nonewutf-8encodingFindent ensure_asciiNzSaved data for z to )rA metric_dater9has_datazFailed to save z data: ) isinstancer r typer DATA_TYPE_CONFIGr r:r model_dumpopenrCdumploggerdebugr$index_daily_metric Exceptionerror)r%r@rAr)r9 data_dictfes r&save_daily_metriczHealthStorage.save_daily_metricdsB$ QdI.. ^)*\PTUYPZPZ*\*\]]]&"999"#H;#H#HIII!2;?OL++L$)DDIV$GGIiw777 F1 )QquEEEE F F F F F F F F F F F F F F F LLS+SSSS SS T T T I ( (' I# )    "     Q Q Q LLB;BBqBB C C CHHHQHHIIq P Qs=B0D/2C D/CD/CAD//E>6E99E>c   |j||}|rt|}n0tj|d}|||d}|sdSt|dd5}tj |}dddn #1swxYwYt d|d ||S#t$r.}t d |d |d |Yd}~dSd}~wwxYw) zLoad a daily metric from JSON file. Args: metric_type: Type of metric target_date: Date of the data Returns: Data dictionary or None if not found r)F)r+NrrHrIzLoaded rOzFailed to load : )r$get_daily_metric_pathrr rTr:existsrVrCloadrXrYr[r\) r%rAr* indexed_pathr9r)r^r@r_s r&load_daily_metriczHealthStorage.load_daily_metrics{ 9::; TTL  ..  &6{CNS  // +50 ##%% tiw777 $1y|| $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ LLG;GG+GG H H HK    LLT;TT+TTQRTT U U U44444 sBA1C5CB( C(B,,C/B,0$C D #DD activitycZ ||j|j}|dd}t |dd5}t j||dd d d d n #1swxYwYtd |jd |j d ||j |j|j |j||j |j |S#t$r=}td|jd|t!d||d }~wwxYw)zSave an activity to JSON file and update database index. Args: activity: Activity model instance Returns: Path to saved file Raises: StorageError: If save fails rCTrDrGrHrIrKFrLNzSaved activity z (z) to )r; activity_typer<r9duration_secondsdistance_meterszFailed to save activity rczFailed to save activity: )r?r;rrUrVrCrWrXrYrkr$index_activityrlrmr[r\r )r%rir9r]r^r_s r& save_activityzHealthStorage.save_activitys G44$hmI !++d+KKIiw777 F1 )QquEEEE F F F F F F F F F F F F F F F LLb("6bb(:PbbW`bb    I $ $$0&4&m#!)!: ( 8 %     G G G LLOH4HOOAOO P P P>1>>??Q F Gs=A C# A0$ C#0A44C#7A48A*C## D*-8D%%D*c |j5}|}|d|f|}dddn #1swxYwY|sdSt |d}|s"td|d|dSt|dd5}tj |}dddn #1swxYwYt d ||S#t$r+}td |d |Yd}~dSd}~wwxYw) zLoad an activity from JSON file. Args: activity_id: Activity ID Returns: Activity data dictionary or None if not found z:SELECT file_path FROM activity_index WHERE activity_id = ?Nr9z Activity z indexed but file not found: rbrHrIzLoaded activity zFailed to load activity rc)r$ _get_conncursorexecutefetchonerrerXwarningrVrCrfrYr[r\) r%r;connrrrowr9r^r@r_s r& load_activityzHealthStorage.load_activitys $$&& ($P Noo''  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( tS-..I##%% U UU)UUtiw777 $1y|| $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ LL9K99 : : :K    LLFKFF1FF G G G44444 slDAA' D'A++D.A+/D6A DDC4( D4C88D;C8,>,E,E,G,GGr(c6|j|S)zCheck if an activity already exists. Args: activity_id: Activity ID Returns: True if activity exists )r$activity_exists)r%r;s r&r|zHealthStorage.activity_exists2sy((555r()N)T)$__name__ __module__ __qualname____doc__rrr'r2rboolr:r?rrrrrrrrrrrrrrrr`rrrhrrorxrzr|r(r&rr'sMM''$'4''''IM.2AE 4JN/3BF 0EQ              " EQ$%EQ& 'EQEQEQEQN$$-1$ $sCx. !$$$$L*Gh*G4*G*G*G*GX&&$sCx.1I&&&&P H H4 HD H H H H 63 64 6 6 6 6 6 6r(r)+rrCdatetimerpathlibrtypingrrrrpydanticr healthr health.db.repositoryr health.utils.exceptionsr r health.utils.logging_configrhealth.models.daily_metricsrrrrrrrrrrrrrhealth.models.activityrhealth.models.body_metricsrr}rXrrr(r&rs  ------------111111EEEEEEEE444444,+++++111111 h  T6T6T6T6T6T6T6T6T6T6r(