diff --git a/management/commands/extract_ml_registrations.py b/management/commands/extract_ml_registrations.py index 034737c..5a5a0de 100644 --- a/management/commands/extract_ml_registrations.py +++ b/management/commands/extract_ml_registrations.py @@ -10,7 +10,7 @@ from member.models import Club, Membership class Command(BaseCommand): help = "Get mailing list registrations from the last wei. " \ - "Usage: manage.py extract_ml_registrations -t {events,art,sport} -t {fr, en}. " \ + "Usage: manage.py extract_ml_registrations -t {events,art,sport} -t {fr, en} -y {0, 1, ...}. " \ "You can write this into a file with a pipe, then paste the document into your mail manager." def add_arguments(self, parser): @@ -19,6 +19,8 @@ class Command(BaseCommand): parser.add_argument('--lang', '-l', type=str, choices=['fr', 'en'], default='fr', help='Select the registred users of the ML of the given language. Useful only for the ' 'events mailing list.') + parser.add_argument('--years', '-y', type=int, default=0, + help='Select the cumulative registred users of a membership from years ago. 0 means the current users') def handle(self, *args, **options): # TODO: Improve the mailing list extraction system, and link it automatically with Mailman. @@ -28,10 +30,12 @@ class Command(BaseCommand): return if options["type"] == "members": + today_date = date.today() + selected_date = date(today_date.year - options["years"], today_date.month, today_date.day) for membership in Membership.objects.filter( club__name="BDE", - date_start__lte=date.today(), - date_end__gte=date.today(), + date_start__lte=today_date, + date_end__gte=selected_date, ).all(): self.stdout.write(membership.user.email) return