)h5ddlmZddlmZmZddlmZmZmZmZddl m Z m Z m Z m Z ddlZddlZddlmZddlmZddlmZdd lmZdd lmZdd lmZej8eZGd d ZeZ dZ!dZ"dZ#dde$de%fdZ&y))Session)funcand_)datetimedate timedeltatimezone)DictAnyListOptionalN)BackgroundScheduler) CronTrigger)PersonaGenerationService)Persona)ConnectedAccount)get_db_sessionc eZdZdZdZdZdZdZdede e fdZ ded e de e effd Zded e de efd Zd Zdd e d ede e effdZde e effdZy)PersonaScheduledServicezXScheduled service for automatically refreshing personas based on social media analytics.c>t|_d|_d|_y)NF)r schedulerservice is_runningselfs K/var/www/html/hubwallet-dev/src/marketing/apps/persona/scheduled_service.py__init__z PersonaScheduledService.__init__s,. c |js|jj|jddddd|jj d|_t j dt j d|jy t j d y #t$r'}t jd t|d }~wwxYw) z$Start the persona refresh scheduler.intervalpersona_refresh_jobz.Refresh all personas every 5 minutes (testing)T)rtriggerminutesidnamereplace_existingz.Persona scheduled service started successfullyz)Personas will be refreshed every 2-3 daysz,Persona scheduled service is already runningz*Error starting persona scheduled service: N) rradd_jobrefresh_all_personas_jobstartloggerinforun_initial_refresh Exceptionerrorstrres rstart_schedulerz'PersonaScheduledService.start_schedulers$ ??&&66&,I%) '&$$&"& LM GH((* JK  LLEc!fXN O  sBB+B++ C4"CCc |jr7|jjd|_tj dytj dy#t $r'}tj dt|d}~wwxYw)z#Stop the persona refresh scheduler.Fz!Persona scheduled service stoppedz(Persona scheduled service is not runningz*Error stopping persona scheduled service: N)rrshutdownr+r,r.r/r0r1s rstop_schedulerz&PersonaScheduledService.stop_schedulerBsj '')"' ?@ FG  LLEc!fXN O  sAAA B $"BB c tjdtj} t}t ||_|j |}|s4tjd dtvr|jyytjdt|dt|dddgd}|D]b} |j||}|d j||d d k(r|d xxd z cc<n#|d dk(r|dxxd z cc<n |dxxd z cc<dtj|z }tjd|ddtjd|d d|dd|dd|d D]} | d d k(r"tjd| dd| d-| d dk(r"tjd| dd| dW| d dk(s`tjd| dd| d dtvrjyy#t$r]}tjd|dt||dxxd z cc<|d j|dt|dYd}~d}~wwxYw#t$r+}tjdt|Yd}~d}~wwxYw#dtvrjwwxYw)z*Main job function to refresh all personas.z&Starting scheduled persona refresh jobz)No stores with personas found for refreshdbNzFound z stores with personas to refreshr) total_stores successfulfailedskippeddetailsr=statussuccessr:r;r<zError processing store : r/store_idr>messagez!Persona refresh job completed in z.2fz secondsz Results: z successful, z failed, z skippedzStore rCrDzError in persona refresh job: )r+r,timerrr_get_stores_with_personaslocalscloselen_refresh_store_personasappendr.r/r0warning) r start_timer8stores_with_personasresultsrC store_resultr2execution_timedetails rr)z0PersonaScheduledService.refresh_all_personas_jobPs <=YY[ @ !B4B7DL$(#A#A"#E ' GHfvx  c KK&%9!: ;;[\ ]!$$8 9 G1 #'#?#?H#MLI&--l;#H-: -2-%h/8;)Q.) *a/* ,"YY[:5N KK;N3;OxX Y KK)GL$9#:-PXHYGZZcdkludvcwwA B"), T(#y0KK& );(Q RSH%1NNVF:,>+?r&BSAT#UVH%2KK& );(Q RS  Tvx  5!LL#:8*Bs1vh!OPH%*%I&--$,")#&q6/. D LL9#a&B C C Dvx  siAJ 8JAH&$B?J$#JK& J /AJJJ  J K!J>9K>KKK%r8returnc |jtjjj }|Dcgc]}|d c}Scc}w#t $r-}t jdt|gcYd}~Sd}~wwxYw)z%Get all store IDs that have personas.rz$Error getting stores with personas: N) queryrrCdistinctallr.r+r/r0)rr8 store_idsrCr2s rrFz1PersonaScheduledService._get_stores_with_personassr !1!12;;=AACI09:HHQK: ::  LL?AxH II s/?A A AA B "BB B rCc  tjd||j||}|s|dddS|jtj tj |k(j}|s|dddSg}|D]}|j}|j |jtj}tjtj|z j}|dk\sq|j!||s|dd dSd } d } |D]} |j"j%|j |j&d d } | j(r+| dz } tjd|j*d|n7| dz } tj-d|j*d|d| j.| d k(r d} d| d}n| d kDr d} d| d| d}nd} d| d}|| || | t7|dS#t0$rA} | dz } tj3d|j*d|dt5| Yd} ~ %d} ~ wwxYw#t0$r@} tj3d|dt5| |dd t5| dcYd} ~ Sd} ~ wwxYw)!z*Refresh all personas for a specific store.zRefreshing personas for store r<z%No active social media accounts foundrBzNo personas found for storeN)tzinfoz.All personas are recent (less than 2 days old)rT)days_to_analyze force_refreshr@zSuccessfully refreshed persona z for store zFailed to refresh persona rAzError refreshing persona r?z Refreshed z personas successfullypartialz personas, z failedr;z Failed to refresh any personas (z failed))rCr>rDrefreshed_count failed_counttotal_personasz$Error refreshing personas for store r/zError: )r+r,_get_active_social_accountsrUrfilterrCrW updated_atrZreplacer utcrnowdaysrKrgenerate_persona_for_store branch_idr?r%rLrDr.r/r0rI)rr8rCactive_accountsstore_personaspersonas_to_refreshpersonaredays_since_updater`raresultr2r>rDs rrJz/PersonaScheduledService._refresh_store_personass[  KK8 C D#>>r8LO" ('F XXg.55g6F6F(6RSWWYN! ('<#% ) 8$// $$,!+!3!38<6H:4I:H??I J  5JJ J c B |jtjtj|k(tjj dj }|S#t$r0}tjd|dt|gcYd}~Sd}~wwxYw)z-Get active social media accounts for a store.Fz/Error getting active social accounts for store rAN) rUrrdrC is_deletedis_rWr.r+r/r0)rr8rCaccountsr2s rrcz3PersonaScheduledService._get_active_social_accountss xx 0188 ))X5 ++//6ce  O  LLJ8*TVWZ[\W]V^_ `I sA"A%% B.%BBBcNtjd|jy)zARun initial refresh for all stores (useful for first-time setup).z.Running initial persona refresh for all storesN)r+r,r)rs rr-z+PersonaScheduledService.run_initial_refresh s DE %%'rr^c tjd| t}t||_|j ||}d||ddt vr|jSS#t$rZ}tjd|dt|d|t|dcYd }~dt vrjSSd }~wwxYw#dt vrjwwxYw) z/Manually refresh personas for a specific store.z#Manual refresh requested for store T)r?rCrqr8z"Error in manual refresh for store rAF)r?rCr/N) r+r,rrrrJrGrHr.r/r0)rrCr^r8rqr2s rmanual_refresh_storez,PersonaScheduledService.manual_refresh_stores 9(DE !B3B7DL11"h?F $ vx    LL=hZr#a&R S $Q  vx   vx  s/1A)) C 22C$C %CC  CC.c  g}|jjD]e}|j|j|j|j r|j j ndt|jdg|j|jjt||dS#t$r9}tjdt|dt|dcYd}~Sd}~wwxYw)z-Get current scheduler status and information.N)r%r& next_run_timer#)rscheduler_state active_jobsjobsz Error getting scheduler status: F)rr/)rget_jobsrKr%r&rz isoformatr0r#rstaterIr.r+r/)rr}jobr2s rget_scheduler_statusz,PersonaScheduledService.get_scheduler_status-s D~~..0  &&HHFIFWFWS%6%6%@%@%B]a"3;;/  #oo#'>>#7#7"4y    LL;CF8D E#Q  sB1B44 C6=.C1+C61C6NF)__name__ __module__ __qualname____doc__rr3r6r)rr intrFr r0r rJrrcr-boolrxrrrrrsb &P ENGS ]']S]T#s(^]~ g  N^I_ ( SRVWZ\_W_R`6d38nrrc,tjy)z$Start the persona scheduler service.N)persona_schedulerr3rrrstart_persona_schedulerrJs%%'rc,tjy)z#Stop the persona scheduler service.N)rr6rrrstop_persona_schedulerrNs$$&rc*tjS)z0Get the current status of the persona scheduler.)rrrrrget_persona_scheduler_statusrRs  1 1 33rrCr^c.tj||S)z6Manually trigger persona refresh for a specific store.)rrx)rCr^s rmanual_refresh_store_personasrVs  1 1(M JJrr)'sqlalchemy.ormr sqlalchemyrrrrrr typingr r r r loggingrE!apscheduler.schedulers.backgroundrapscheduler.triggers.cronr"src.marketing.apps.persona.servicer src.marketing.apps.persona.modelr src.marketing.apps.Account.modelr src.utils.dbr getLoggerrr+rrrrrrrrrrrrs~"!88,, A1G4='   8 $ssl ,-('4KCKKr