coin/coin/billing/management/commands/charge_subscriptions.py

# -*- coding: utf-8 -*-
import datetime
from django.core.management.base import BaseCommand, CommandError
from django.conf import settings

from coin.utils import respect_language
from coin.billing.create_subscriptions_invoices import create_all_members_invoices_for_a_period


class Command(BaseCommand):
    args = '[date=2011-07-04]'
    help = 'Create invoices for members subscriptions for date specified (or today if no date passed)'

    def handle(self, *args, **options):
        try:
            date = datetime.datetime.strptime(args[0], '%Y-%m-%d').date()
        except IndexError:
            date = datetime.date.today()
        except ValueError:
            raise CommandError(
                'Please enter a valid date : YYYY-mm-dd (ex: 2011-07-04)')

        if options['verbosity'] >= 2:
            self.stdout.write(
                'Create invoices for all members for the date : %s' % date)
        with respect_language(settings.LANGUAGE_CODE):
            invoices = create_all_members_invoices_for_a_period(date)

        if invoices > 0 or options['verbosity'] >= 2:
            self.stdout.write(
                u'%d invoices were created' % len(invoices))