52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from datetime import datetime
|
|
from django.utils import timezone
|
|
from django.contrib.auth.models import Group
|
|
from django.forms.models import model_to_dict
|
|
from django.core.management.base import BaseCommand
|
|
from mhackspace.subscriptions.payments import select_provider
|
|
from mhackspace.users.models import Membership, User
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Update user subscriptions'
|
|
|
|
def handle(self, *args, **options):
|
|
provider = select_provider('gocardless')
|
|
|
|
self.stdout.write(
|
|
self.style.NOTICE(
|
|
'== Gocardless subscriptions =='))
|
|
|
|
Membership.objects.all().delete()
|
|
subscriptions = []
|
|
|
|
group = Group.objects.get(name='member')
|
|
|
|
for sub in provider.fetch_subscriptions():
|
|
try:
|
|
user_model = User.objects.get(email=sub.get('email'))
|
|
if sub.get('status') == 'active':
|
|
user_model.groups.add(group)
|
|
except User.DoesNotExist:
|
|
user_model = None
|
|
|
|
self.stdout.write(sub.get('status'))
|
|
subscriptions.append(
|
|
Membership(
|
|
user= user_model,
|
|
email= sub.get('email'),
|
|
reference= sub.get('reference'),
|
|
payment= 10.00,
|
|
# 'date'= sub.get('start_date'),
|
|
date=timezone.now(),
|
|
status=Membership.lookup_status(name=sub.get('status'))
|
|
)
|
|
)
|
|
|
|
self.stdout.write(
|
|
self.style.SUCCESS(
|
|
'\t{reference} - {payment} - {status} - {email}'.format(**model_to_dict(subscriptions[-1]))))
|
|
|
|
Membership.objects.bulk_create(subscriptions)
|
|
|