society-self-service/ssmembership/management/commands/sendmdemail.py

48 lines
1.9 KiB
Python

# Society Self-Service
# Copyright © 2018-2019 Yingtong Li (RunasSudo)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
from django.core.management.base import BaseCommand, CommandError
from django.utils import timezone
from ssmain.email import Emailer
import ssmembership.models
class Command(BaseCommand):
help = 'Send Markdown emails'
def add_arguments(self, parser):
parser.add_argument('template', help='Template name')
parser.add_argument('subject', help='Email subject')
parser.add_argument('ids', nargs='*', type=int, help='Members with ID numbers equal to these values will be emailed (default all)')
def handle(self, *args, **options):
today = timezone.localtime(timezone.now()).date()
members = ssmembership.models.Member.objects.all()
if len(options['ids']) > 0:
members = [member for member in members if member.id in options['ids']]
else:
raise Exception('Must provide IDs')
emailer = Emailer()
for member in members:
if member.member_type != 1 or member.expires < today:
self.stdout.write('Skipping {} at {}'.format(member.id, member.email))
continue
self.stdout.write('Emailing {} at {}'.format(member.id, member.email))
emailer.send_mail([member.email], options['subject'], 'ssmembership/email/' + options['template'] + '.md', {})