o ѷ6i6@sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZd ZGddde ZdS)N) timedelta)settings)mail) BaseCommand)Q)render_to_string)timezone)activate) deactivate)models) app_settingsxc@sPeZdZdZdZddZddZddZd d Zd d Z d dZ dddZ dS)CommandTz[Sends notification emails to subscribed users taking into account the subscription intervalcCs|jdddddd|jdddd d d|jd d d ddd|jdd dddd|jdddddd|jdd dddd|jddddd|jd d d!d"td|jd#d d$d%ddS)&Nz--daemonz-d store_truedaemonzGo to daemon mode and exit)actiondesthelpz--cronz-ccronz/Do not loop, just send out emails once and exitz --pid-filestorepidz!Where to write PID before exitingz/tmp/nyt_daemon.pid)rrrdefaultz--domaindomainz6Base domain to use for URLs (excluding HTTP protocol).z --http-onlyhttpz(Use http:// in URLs instead of https://.Fz --log-filelogz!Where daemon should write its logz/tmp/nyt_daemon.logz --no-sys-exit no_sys_exitz@Skip sys-exit after forking daemon (mainly for testing purposes)z--daemon-sleep-interval sleep_timez3Minimum sleep between each polling of the database.z--nownowzASimulate when to start sending from (mainly for testing purposes)) add_argument SLEEP_TIME)selfparserr#n/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/django_nyt/management/commands/notifymail.py add_arguments sz zCommand.add_argumentscCsztjr ttj}nt||}|dd}t||}tj||tj|dj g|d}|j d|dj |j dddS)N ruser) connectionzSending to: %sF) fail_silently) r NYT_EMAIL_SUBJECTstrrreplacestripr EmailMessageNYT_EMAIL_SENDERemailloggerinfosend)r! template_nametemplate_subject_namecontextr(subjectmessager0r#r#r$_render_and_send_s    zCommand._render_and_sendc Cs|jdzDt}|dkrE|jdt|t|jdd}|t|Wdn1s3wY|jdsHt dWdSWdSWdSt yn}ztj d|j |j ft dWYd}~dSd}~ww) NzDaemon mode enabled, forkingrzPID: %srwrzfork failed: %d (%s) )r1r2osforkr+openoptionswritesysexitOSErrorstderrerrnostrerror)r!fpidpid_fileer#r#r$ _daemonizews"  zCommand._daemonizecOsttj|dd|dd|dd||_|d}|d}|r*|r*Jdtd|_|jjs[|r?tj |dd}nt |j }| t d d |j||jtj|jd tjsn|jd t|rt|t}|r|jd r|jd }|jd|nt}|||dS|stdz ||t|dWn t ytdYnwt!dS)NrFrrz.You cannot both choose cron and daemon options django_nytr)filenamez(%(asctime)s: %(levelname)s - %(message)sz%Y-%m-%d %H:%M:%Sz%Starting django_nyt e-mail dispatcherzE-mails disabled - quitting.rz using now: z"Entering send-loop, CTRL+C to exitrz Quitting...)"r r LANGUAGE_CODE setdefaultr?logging getLoggerr1handlers FileHandler StreamHandlerstdout setFormatter Formatter addHandlersetLevelINFOr2r NYT_SEND_EMAILSrArBrJrget_connectiongetrr send_mailsprint send_loopintKeyboardInterruptr )r!argsr?rrhandlerr(rr#r#r$handlesT               zCommand.handlecCsd} t}|jdt||r0tjjj||j ddd d}|j dp.t}nt}d}|j ||||d|t}t|q)NTzStarting send loop at %s<) interval__lter'r) last_sent user_settings)rrr1r2r+r Settingsobjectsfiltersecondsorder_byr?r\r]closetimesleep)r!r(rrgstarted_sending_atrhrr#r#r$r_s( zCommand.send_loopc CsB|d}t|ddkrdS dd|D}z9|jd|||||||D]}d|_|t|j_ |jq*t j j j |djtd WdStjyd|jd |jjYqtjy} z|jd | |jd td WYd} ~ nd} ~ wty} z |jd| d} ~ wwq)zn Loops through emails in a list of notifications and tries to send to each recipient notificationsrNTcSsg|]}|jqSr#)id).0nr#r#r$ sz,Command._send_with_retry..zSending to notification ids )notification__id__inrgz-E-mail refused by SMTP server ({}), skipping!z@You have an error with your SMTP server connection, error is: {}z!Sleeping for 30s then retrying...z0Unhandled exception while sending, giving up: {})lenr1r2r9 is_emailedsaverr subscriptionrgr SubscriptionrjrkupdatesmtplibSMTPSenderRefusederrorformatr'r0 SMTPExceptionrorp Exception) r!r4subject_template_namer6r(settingrrnotification_idsrurIr#r#r$_send_with_retrysX    zCommand._send_with_retryNc Cs|jd|d|d||s#tjjddd d}|j dr0d}|j d}nd d l m }|j }|j}|j d rFd }nd }|D]} |t| jd } dddd|||d} | j| d<t| j| jj| d<ddtjD| j} tj| d| d<i} t}| jrtt| dtdd@tt| dtdd@B}| jj|dd dD]=}z|j}Wntjjy|jd|jYqw|j }| !||fg| ||ft"|j#jdd7<q| $D]\\}}}|sq|| d<|%||| || qqJdS)z Does the lookups and sends out email digests to anyone who has them due. Since the system may have different templates depending on which notification is being sent, we will generate a call for each template. zEntering send_mails(now=z , last_sent=z, ...)r'subscription_set#subscription_set__notification_typerNr)Siterhttps)minutes)r'usernamerrdigestsiter http_schemercSsg|]}|dqS)rr#)rtyr#r#r$rvQsz&Command.send_mails..r;r) created__lterx)last_sent__lteF)latest__is_emailedT) send_emailsnotification_type__keyz0Subscription has non-existent notification type )r{rr)&r1debugr>r rirjallselect_relatedprefetch_relatedrmr?django.contrib.sites.modelsr get_currentrrintervalr'getattrUSERNAME_FIELDr NYT_INTERVALSindexrrrknotification_typeget_email_template_nameNotificationType DoesNotExistwarningnotification_type_idget_email_subject_template_namerNlistnotification_setitemsr)r!r(rrgrh site_objectrrrr thresholdr6idxemails_per_template filter_qsr}r4rrrr#r#r$r]s            zCommand.send_mails)NN) __name__ __module__ __qualname__can_import_settingsrr%r9rJrdr_rr]r#r#r#r$rs??5r)rOr<rrArodatetimer django.confr django.corerdjango.core.management.baserdjango.db.modelsrdjango.template.loaderr django.utilsrdjango.utils.translationr r rKr django_nyt.confr r rr#r#r#r$s$