def __init__(self, now=datetime.now()):
self.now = now + timedelta(hours=(1 if settings.DST else 0))
self.template = Template()
self.mailclient = MailClient()
self.failure = 'Failed to send email with subject "{}" to {}.'
self.success = 'Sent email with subject "{}" to {}.\n'
class Cron(object):
def __init__(self, now=datetime.now()):
self.now = now + timedelta(hours=(1 if settings.DST else 0))
self.template = Template()
self.mailclient = MailClient()
self.failure = 'Failed to send email with subject "{}" to {}.'
self.success = 'Sent email with subject "{}" to {}.\n'
def job(self):
# This where clause should be (User.is_active is True) to comply
# with PEP-8, but peewee doesn't support that syntax.
users = User.select().where(User.is_active == True)
# Send Email Digests (if necessary)
for user in filter(lambda user: user.is_due(self.now), users):
if self.send_email(user, self.create_digest, Subjects.DIGEST):
user.update_last_sent(self.now)
# Send Reminders (if necessary)
for user in filter(lambda user: user.is_late(self.now), users):
if self.send_email(user, self.create_reminder, Subjects.REMINDER):
user.update_last_reminded(self.now)
def send_email(self, user, create_email, subject):
try:
email = self.mailclient.send(user, create_email(user), subject)
logging.info(self.success.format(subject, user.name))
if settings.DEBUG:
print("\n".join(email))
return True
except Exception as e:
logging.error(self.failure.format(subject, user.name))
logging.error(e)
if settings.DEBUG:
raise
def create_digest(self, user):
last_sent = user.last_sent or user.created
tasks = Task.select()\
.where((Task.created > last_sent))\
.order_by(Task.user.name, Task.date)
email_params = {
'user': user,
'start': last_sent,
'end': self.now,
'tasks': self.format_tasks(tasks, user),
'sender': settings.SENDER,
'timezone_correct': timedelta(hours=user.timezone)
}
return self.template.render('email_digest.tpl', **email_params)
def create_reminder(self, user):
return self.template.render('email_reminder.tpl', user=user)
def format_tasks(self, all_tasks, recipient):
formatted_tasks = []
timezone = timedelta(hours=recipient.timezone)
usergen = groupby(all_tasks, lambda task: task.user.name)
for user, user_tasks in usergen:
date_tasks = []
dategen = groupby(
user_tasks,
lambda task: (task.date + timezone).strftime("%a %D, %Y")
)
for date, tasks in dategen:
date_tasks.append({
'date': date,
'tasks': [t.description for t in tasks]
})
formatted_tasks.append({'user': user, 'tasks': date_tasks})
return formatted_tasks
