mi*dZddlZddlmZmZddlmZmZmZmZddl m Z ddl m Z ddl mZeeZGdd ZdS) zr Data access layer for health database. Provides CRUD operations for sync records, sync state, and data indexes. N)datedatetime)ListOptionalDictAny)Path)get_connection) setup_loggerceZdZdZd&deeddfdZdejfdZ d'de d e d e d e d e d ee de fdZ d(dee de deee effdZde dee fdZ d&de de dee ddfdZdee ee efffdZ d)de de dededdf dZde de dee fdZ d*de de d e ded!ee d"eeddfd#Zd e d e deee effd$Zde defd%ZdS)+HealthRepositoryz/Repository for health data database operations.Ndb_pathreturnc||_dS)zbInitialize repository. Args: db_path: Optional path to database file N)r)selfrs -/root/projects/butler/health/db/repository.py__init__zHealthRepository.__init__s  c*t|jS)zQGet database connection. Returns: SQLite connection )r r)rs r _get_connzHealthRepository._get_conns dl+++rr data_type start_dateend_datestatusrecords_synced error_messagec <|5}|}|d||||||f||jcdddS#1swxYwYdS)aCreate a new sync record. Args: data_type: Type of data synced start_date: Start date of sync range end_date: End date of sync range status: Sync status ('success', 'failed', 'partial') records_synced: Number of records successfully synced error_message: Optional error message if sync failed Returns: ID of created record z INSERT INTO sync_records (data_type, start_date, end_date, status, records_synced, error_message) VALUES (?, ?, ?, ?, ?, ?) N)rcursorexecute isoformatcommit lastrowid) rrrrrrrconnrs rcreate_sync_recordz#HealthRepository.create_sync_record's,^^   $[[]]F NN ((**&&(("!      KKMMM#% $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $sA/BBBdlimitc&|5}|}|r|d||fn|d|fd|DcdddS#1swxYwYdS)zGet sync records, optionally filtered by data type. Args: data_type: Optional data type filter limit: Maximum number of records to return Returns: List of sync record dictionaries z SELECT * FROM sync_records WHERE data_type = ? ORDER BY created_at DESC LIMIT ? z SELECT * FROM sync_records ORDER BY created_at DESC LIMIT ? c,g|]}t|Sdict.0rows r z5HealthRepository.get_sync_records..v;;;#DII;;;rNrrrfetchall)rrr&r#rs rget_sync_recordsz!HealthRepository.get_sync_recordsQs^^   <[[]]F  & H <;):):;;;/ < < < < < < < < < < < < < < < < < <A$BB  B cT|5}|}|d|f|}|r8t j|dcdddS ddddS#1swxYwYdS)zGet the last sync date for a data type. Args: data_type: Type of data Returns: Last sync date, or None if never synced z>SELECT last_sync_date FROM last_sync_state WHERE data_type = ?last_sync_dateN)rrrfetchoner fromisoformatr)rrr#rr.s rget_last_sync_datez#HealthRepository.get_last_sync_datezs^^   [[]]F NNP    //##C L-c2B.CDDIIKK                          sA-BBB!$B!r6 total_recordsc|5}|}|d|f|}|rY|,|d|||fnX|d||fn-|d|||pdf|ddddS#1swxYwYdS)zUpdate the last sync state for a data type. Args: data_type: Type of data last_sync_date: Last successfully synced date total_records: Optional total record count to set z=SELECT total_records FROM last_sync_state WHERE data_type = ?Nz UPDATE last_sync_state SET last_sync_date = ?, total_records = ?, updated_at = CURRENT_TIMESTAMP WHERE data_type = ? z UPDATE last_sync_state SET last_sync_date = ?, updated_at = CURRENT_TIMESTAMP WHERE data_type = ? z INSERT INTO last_sync_state (data_type, last_sync_date, total_records) VALUES (?, ?, ?) r)rrrr7r r!)rrr6r:r#rr.s rupdate_last_sync_statez'HealthRepository.update_last_sync_states{^^  ( [[]]F NNO    //##C  ,NN (1133]IN NN (1133Y?  8 8 : :M>DDc(|5}|}|di}|D]#}|d|d|dd||d<$|cdddS#1swxYwYdS)zyGet last sync state for all data types. Returns: Dictionary mapping data_type to state info zSELECT * FROM last_sync_stater6r: updated_at)r6r:r>rNr1)rr#rresultr.s rget_all_last_sync_statesz)HealthRepository.get_all_last_sync_statess ^^   [[]]F NN: ; ; ;F((  &)*:&;%(%9"%l"3,,s;'((                   sA%BB B T metric_type metric_date file_pathhas_datac "|5}|}|d||t ||f|ddddS#1swxYwYdS)zIndex a daily metric file. Args: metric_type: Type of metric metric_date: Date of the metric file_path: Path to JSON file has_data: Whether the file contains valid data z INSERT OR REPLACE INTO daily_metrics_index (metric_type, date, file_path, has_data) VALUES (?, ?, ?, ?) Nrrrr strr!)rrArBrCrDr#rs rindex_daily_metricz#HealthRepository.index_daily_metrics^^   [[]]F NN  ))++ NN     KKMMM                  sA"BB Bc|5}|}|d||f|}|r|dndcdddS#1swxYwYdS)zGet file path for a daily metric. Args: metric_type: Type of metric metric_date: Date of the metric Returns: File path string, or None if not found z~ SELECT file_path FROM daily_metrics_index WHERE metric_type = ? AND date = ? rCN)rrrr r7)rrArBr#rr.s rget_daily_metric_pathz&HealthRepository.get_daily_metric_paths^^   5[[]]F NNk33556    //##C'*43{## 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5sABBB activity_id activity_type activity_dateduration_secondsdistance_metersc &|5}|}|d|||||t |f|ddddS#1swxYwYdS)aQIndex an activity file. Args: activity_id: Unique activity ID activity_type: Type of activity activity_date: Date of the activity file_path: Path to JSON file duration_seconds: Optional activity duration distance_meters: Optional activity distance z INSERT OR REPLACE INTO activity_index (activity_id, activity_type, date, duration_seconds, distance_meters, file_path) VALUES (?, ?, ?, ?, ?, ?) NrF) rrKrLrMrCrNrOr#rs rindex_activityzHealthRepository.index_activitys&^^   [[]]F NN  !!++--$# NN      KKMMM#                  r4c:|5}|}|d||fd|DcdddS#1swxYwYdS)zGet all activities within a date range. Args: start_date: Start date end_date: End date Returns: List of activity index records z SELECT * FROM activity_index WHERE date >= ? AND date <= ? ORDER BY date DESC c,g|]}t|Sr)r*r,s rr/zAHealthRepository.get_activities_by_date_range..Tr0rN)rrrr r2)rrrr#rs rget_activities_by_date_rangez-HealthRepository.get_activities_by_date_range=s^^   <[[]]F NN %%''););)=)=>    <;):):;;; < < < < < < < < < < < < < < < < < res ########,,,,,,,,,,,,++++++444444 h  S1S1S1S1S1S1S1S1S1S1r