h(oigdZddlZddlmZmZddlmZddlmZmZddl m Z ddl m Z m Z mZmZmZddlmZeeZGd d ZdS) z| Storage service for manual health logs. Handles reading/writing user-entered diet, alcohol, supplement, and feeling data. N)datedatetime)Path)OptionalList)config)DailyManualLog DietEntry AlcoholEntrySupplementEntryBodyFeelingEntry) setup_loggerceZdZdZddeeddfdZddededefd Z dede fd Z d e defd Z ded e de fdZded ede fdZded ede fdZded ede fdZdedede fdZdededee fdZdS)ManualLogStoragez6Service for storing and retrieving manual health logs.Ndata_dirreturnc|p tj|_|jdz |_|jdddS)zInitialize manual log storage service. Args: data_dir: Optional data directory path (defaults to config.DATA_DIR) manual_logsTparentsexist_okN)rDATA_DIRrmanual_log_dirmkdir)selfrs ;/root/projects/butler/health/services/manual_log_storage.py__init__zManualLogStorage.__init__sC !3FO "mm; !!$!>>>>>T target_date create_dirsct|j}|jd}|d}|j|z |z |z }|r|jdd|S)aGet file path for a manual log. Args: target_date: Date of the log create_dirs: Whether to create directories if they don't exist Returns: Full path to JSON file (data/health/manual_logs/YYYY/MM/YYYY-MM-DD.json) 02dz.jsonTr)stryearmonth isoformatrparentr)rrr r$r%filename file_paths r_get_file_pathzManualLogStorage._get_file_path&s~;#$$$**!++--444'$.6A  @   " "4$ " ? ? ?rc||d}|s.td|dt |S t |dd5}t j|}d d d n #1swxYwYt d i|}td ||S#t$r:}t d |d |t |cYd }~Sd }~wwxYw)zLoad manual log for a specific date. Args: target_date: Date to load Returns: DailyManualLog instance (creates new if doesn't exist) F)r zNo manual log found for z, creating new)log_daterutf-8encodingNzLoaded manual log for zFailed to load manual log for : ) r*existsloggerdebugr openjsonload Exceptionerror)rrr)fdataloges rload_logzManualLogStorage.load_log;so'' 'GG !! 8 LLOKOOO P P P!;777 7 8iw777 $1y|| $ $ $ $ $ $ $ $ $ $ $ $ $ $ $!((4((C LL?+?? @ @ @J 8 8 8 LLL+LLLL M M M!;777 7 7 7 7 7 7 8sBC-B CBCB-C D/D=DDr=c tj|_||j}|d}t |dd5}tj||ddd d d n #1swxYwYt d |jd ||S#t$r+}t d |jd |d }~wwxYw)zSave manual log to JSON file. Args: log: DailyManualLog instance to save Returns: Path to saved file r7)modewr.r/F)indent ensure_asciiNzSaved manual log for z to zFailed to save manual log for r1) rnow updated_atr*r, model_dumpr6r7dumpr4infor9r:)rr=r) data_dictr;r>s rsave_logzManualLogStorage.save_logXsB %\^^CN++CL99IF33Iiw777 F1 )QquEEEE F F F F F F F F F F F F F F F KKM MM)MM N N N     LLM#,MM!MM N N N  s<AB3B5 B3BB3B )B33 C(=&C##C(entryc||}|j|||td|d|j|S)zAdd a diet entry to the log. Args: target_date: Date of the entry entry: DietEntry to add Returns: Updated DailyManualLog zAdded diet entry for  at )r? diet_entriesappendrLr4rJtimerrrMr=s radd_diet_entryzManualLogStorage.add_diet_entrytshmmK(( &&& c IKIIUZIIJJJ rc||}|j|||td|d|j|S)zAdd an alcohol entry to the log. Args: target_date: Date of the entry entry: AlcoholEntry to add Returns: Updated DailyManualLog zAdded alcohol entry for rO)r?alcohol_entriesrQrLr4rJrRrSs radd_alcohol_entryz"ManualLogStorage.add_alcohol_entryshmmK(( ""5))) c L{LL LLMMM rc||}|j|||td|d|j|S)zAdd a supplement entry to the log. Args: target_date: Date of the entry entry: SupplementEntry to add Returns: Updated DailyManualLog zAdded supplement entry for r1)r?supplement_entriesrQrLr4rJsupplement_namerSs radd_supplement_entryz%ManualLogStorage.add_supplement_entrysimmK(( %%e,,, c X+XXAVXXYYY rc||}|j|||td|d|j|S)zAdd a body feeling entry to the log. Args: target_date: Date of the entry entry: BodyFeelingEntry to add Returns: Updated DailyManualLog zAdded feeling entry for r1)r?feeling_entriesrQrLr4rJ feeling_typerSs radd_feeling_entryz"ManualLogStorage.add_feeling_entrysimmK(( ""5))) c R{RRe>PRRSSS rrAc||}||_||td|d||S)zSet the fasting mode for a day. Args: target_date: Date to set mode: Fasting mode (PSMF/OMAD/Water Fast/Normal) Returns: Updated DailyManualLog zSet fasting mode for r1)r? fasting_moderLr4rJ)rrrAr=s rset_fasting_modez!ManualLogStorage.set_fasting_modesWmmK(( c AKAA4AABBB r start_dateend_datecg}|}||krh||}|js|js|js|js|jr||ddlm}||dz }||kh|S)zGet all manual logs in a date range. Args: start_date: Start date (inclusive) end_date: End date (inclusive) Returns: List of DailyManualLog instances r) timedelta)days) r?rPrVrYr]rarQrrf)rrcrdlogs current_dater=rfs rget_logs_in_rangez"ManualLogStorage.get_logs_in_ranges! h&&-- --C  !& !) !&  ! #  ! C    + * * * * * II1--- -Lh&&  r)N)T)__name__ __module__ __qualname____doc__rrrrboolr*r r?rLr rTr rWr r[r r_r#rbrrkr2rrrrs@@??$?4????$TT*8D8^8888:Nt8(1 $(4 $(7 $(8 $D *. n rr)ror7rrpathlibrtypingrrhealthrhealth.models.manual_logr r r r r health.utils.logging_configrrlr4rr2rrrvs  ########!!!!!!!!544444 h  RRRRRRRRRRr