You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
939 B
Python
27 lines
939 B
Python
from django.test import TestCase
|
|
from django.utils import timezone
|
|
from datetime import timedelta
|
|
from .models import *
|
|
# Create your tests here.
|
|
|
|
|
|
@shared_task
|
|
def create_charges_for_recurring_services():
|
|
today = timezone.now().date()
|
|
active_recurring_services = Service.objects.filter(active=True, recurring=True)
|
|
|
|
for service in active_recurring_services:
|
|
# Get the last charge for the service
|
|
last_charge = service.charge_set.order_by('-due_date').first()
|
|
|
|
if last_charge and last_charge.due_date <= today:
|
|
# Calculate the new due date for the charge based on the recurring cycle
|
|
new_due_date = today + timedelta(days=service.recurring_cycle.number_of_months * 30)
|
|
|
|
# Create a new charge for the service
|
|
Charge.objects.create(
|
|
service=service,
|
|
amount=service.amount,
|
|
due_date=new_due_date
|
|
)
|