mirror of
https://github.com/haiwen/seahub.git
synced 2025-04-27 02:51:00 +00:00
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
# encoding: utf-8
|
|
import logging
|
|
import string
|
|
from datetime import datetime
|
|
from django.core.mail import send_mail
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
from notifications.models import UserNotification
|
|
import settings
|
|
|
|
# Get an instance of a logger
|
|
logger = logging.getLogger(__name__)
|
|
|
|
email_template = u'''${username}您好：
|
|
|
|
您有${cnt}条新消息，请点击下面的链接查看：
|
|
${msg_url}
|
|
|
|
感谢使用我们的网站！
|
|
|
|
${site_name}团队
|
|
'''
|
|
|
|
site_name = settings.SITE_NAME
|
|
subject = u'%s：新消息' % site_name
|
|
url = settings.SITE_BASE.rstrip('/') + settings.SITE_ROOT + 'home/my/'
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Send Email notifications to user if he/she has a new group message.'
|
|
|
|
def handle(self, *args, **options):
|
|
logger.info('Start sending group notification...')
|
|
self.do_action()
|
|
logger.info('Finish sending group notification.\n')
|
|
|
|
def do_action(self):
|
|
today = datetime.now()
|
|
notifications = UserNotification.objects.all()
|
|
|
|
d = {}
|
|
for e in notifications:
|
|
if today.year != e.timestamp.year or today.month != e.timestamp.month or \
|
|
today.day != e.timestamp.day: # Only send today's notifications
|
|
continue
|
|
if d.has_key(e.to_user):
|
|
d[e.to_user] += 1
|
|
else:
|
|
d[e.to_user] = 1
|
|
|
|
for user,cnt in d.items():
|
|
template = string.Template(email_template)
|
|
content = template.substitute(username=user, cnt=cnt, msg_url=url, \
|
|
site_name=site_name)
|
|
try:
|
|
send_mail(subject, content, settings.DEFAULT_FROM_EMAIL, [user], \
|
|
fail_silently=False)
|
|
logger.info('Succesfuuly sended email to %s' % user)
|
|
except Exception, e:
|
|
logger.error('Failed to send email to %s, error detail: %s' % (user, e))
|
|
|