o d6i0@sdZddlZddlmZddlmZmZmZmZddlm Z m Z m Z m Z ddl mZddlmZeeZded ed eefd d Zded ed ed eefddZ d.ded ed eded eef ddZded ed edeeefd ef ddZded ed ed efddZd.ded eed efddZded ed ed efddZded eeeffddZ  d/ded ed ed ed!ed"ed efd#d$Z ded ed ed eefd%d&Z!ded eefd'd(Z"d)ed eefd*d+Z#d)ed efd,d-Z$dS)0z Utility functions for managing user puja sessions in the database. Phone numbers are now stored in collected_info JSON field. N)datetime)OptionalDictAnyList)Sessionselectand_or_)engine)UserPujaSessionuser_id session_idreturnc Csz;tt,}ttttj|ktj|ktjdk tj }| | }|WdWS1s4wYWdStyY}ztdt|jWYd}~dSd}~ww)z Get the active (incomplete) puja session for a user in a specific session. Returns the first incomplete puja found for the user+session combination. FNz#Error getting active puja session: )rr rr wherer r r completedorder_by updated_atdescexecfirst Exceptionloggererrortype__name__)r rsession statementresulter @/var/www/html/karishye-ai-python/app/utils/puja_session_utils.pyget_active_puja_sessions(   (r" puja_typec Csz;tt,}ttttj|ktj|ktj|k tj }| | }|WdWS1s4wYWdStyY}ztdt|jWYd}~dSd}~ww)zL Get a specific puja session by user_id, session_id, and puja_type. NzError getting puja session: )rr rr rr r rr#rrrrrrrrrr)r rr#rrrrr r r!get_puja_session,s(   (r$ puja_namec Csz;tt,}t||||ttd}|||||t d|WdWS1s4wYWdSt yY}zt dt |j WYd}~dSd}~ww)z4 Create a new puja session in the database. )r rr#r% created_atrzCreated puja sessionNzError creating puja session: )rr r rnowaddcommitrefreshrdebugrrrr)r rr#r%r puja_sessionrr r r!create_puja_sessionDs*    (r-updatesc Cszmtt^}ttttj|ktj|ktj|k}| | }|s2t d WdWdS| D]\}}t||rEt|||q6t|_|||t d WdWdS1sfwYWdSty} zt dt| jWYd} ~ dSd} ~ ww)z. Update a puja session with new data. z!Puja session not found for updateNFzUpdated puja sessionTzError updating puja session: )rr rr rr r rr#rrrr+itemshasattrsetattrrr'rr(r)rrrr) r rr#r.rrr,keyvaluerr r r!update_puja_sessionbs>        (r4c CszOtt@}ttttj|ktj|ktj|k}| | }|r;| || t d WdWdS WdWdS1sHwYWdStym}zt dt|jWYd}~dSd}~ww)zT Delete a puja session (e.g., when user starts a new puja in same session). zDeleted puja sessionNTFzError deleting puja session: )rr rr rr r rr#rrdeleter)rr+rrrr)r rr#rrr,rr r r!delete_puja_sessions2    (r6c CszJtt;}|rttttj|ktj|ktj }ntttj|ktj }| | }|WdWS1sCwYWdSt yi}ztdt|jgWYd}~Sd}~ww)zN Get all puja sessions for a user, optionally filtered by session_id. Nz!Error getting all puja sessions: )rr rr rr r rrrrrallrrrrr)r rrrresultsrr r r!get_all_user_puja_sessionss0    (r9cCst|||ddiS)z+ Mark a puja session as completed. rT)r4)r rr#r r r!mark_puja_completedsr:r,cCsJ|siS|j|j|j|j|j|j|j|j|j|j |j |j |j d S)zW Convert a UserPujaSession model to a dictionary (for backward compatibility). ) r rr#r%current_questioncollected_infor kb_checkedinvalid_attemptsawaiting_phonephone_collectedr&r)r rr#r%r;r<rr=r>r?r@r& isoformatr)r,r r r!puja_session_to_dicts rB+91 phone_numberpuja_idphone_country_codec Csrzt|||d|iddd}|rWdStdWdSty8}ztdt|jWYd}~dSd}~ww) z Save user's phone number to puja session's collected_info JSON field. NOTE: Phone numbers are now stored in collected_info, not a separate table. rDT)r<r@)r rr#r.z*Failed to update phone - session not foundFzError saving phone: N)r4rr+rrrr)r rrDr#rErF update_resultrr r r!save_user_phone_numbers$   rHc Cshzt|||}|r|jr|jdWSWdSty3}ztdt|jWYd}~dSd}~ww)z Get user's phone number from puja session's collected_info JSON field. Returns: Phone number string if found, None otherwise rDNzError getting phone: )r$r<getrrrrr)r rr#r,rr r r!get_user_phone_numbers  rJc CszPttA}tttj|ktj}| | }g}|D]}|j r;d|j vr;|j d}|r;||vr;| |q!|WdWS1sIwYWdSt yo}ztdt|jgWYd}~Sd}~ww)z{ Get all phone numbers for a user across all puja sessions. Returns: List of unique phone numbers rDNzError getting phones: )rr rr rr rr&rrr7r<appendrrrrr)r rrsessionsphonesr,phonerr r r!get_all_user_phones s*      (rOrNcCsddl}|r |dkr dS|dd|}|dr!|dd}n/|dr-|d d}n#|d r?t|d kr?|d d}n|d rPt|dkrP|dd}|d|rX|SdS)a Clean and normalize Indian phone number to 10-digit format. Handles various formats and validates against Indian mobile number rules. Accepts formats: - 10 digits: 9876543210 - With country code: +91-9876543210, 0091-9876543210, 91-9876543210 - With leading zero: 09876543210 - With spaces/dashes/parentheses: (987) 654-3210 Args: phone: Phone number string to clean Returns: 10-digit cleaned phone number if valid, None otherwise rNz@phonez [\s\-\(\)]rC009191 0 z ^[6-9]\d{9}$)resubstrip startswithlenmatch)rNrZcleanedr r r!clean_indian_phone_numberAs     racCs t|duS)a- Validate if a phone number is a valid Indian mobile number. Accepts formats: 10 digits, +91-10digits, 0091-10digits, with/without spaces/dashes Args: phone: Phone number string to validate Returns: True if valid Indian mobile number, False otherwise N)ra)rNr r r!validate_indian_phone_numberls rb)N)NrC)%__doc__loggingrtypingrrrrsqlmodelrrr r app.dbr app.schemasr getLoggerrrstrr"r$r-boolr4r6listr9r:rBintrHrJrOrarbr r r r!st       ,! "!+