#models.py
from django import models


class Contract(models.Model):
    document_type = models.CharField(max_length=50, choices=DOCUMENT_TYPES)
    document_name = models.CharField(max_length=255)
    duration = models.CharField(max_length=50)
    start_date = models.DateField(null=True, blank=True)
    expiry_date = models.DateField(null=True, blank=True)
    issuing_body = models.CharField(max_length=255)
    process_owner = models.CharField(max_length=255)

    def __str__(self):
        return self.document_name

#tasks.py
from celery import shared_task
from datetime import datetime, timedelta
from django.core.mail import send_mail
from .models import Contract

@shared_task(bind=True)
def send_contract_expiry_notification(self):

    # Calculate the date one month from now
    one_month_from_now = datetime.now() + timedelta(days=30)

    # Find contracts that are expiring within one month
    expiring_contracts = Contract.objects.filter(expiry_date__lte=one_month_from_now)
    print(expiring_contracts)

    # If there's no contract then don't send any email
    if expiring_contracts:
        # Initialize the message body
        message_body = "Dear user,\n\nThe following contracts are expiring soon:\n\n"

        for contract in expiring_contracts:
            # Calculate the number of days until the expiry date
            days_until_expiry = (contract.expiry_date - datetime.now().date()).days

            # Add contract details to the message body
            message_body += (
                f"Contract: {contract.document_name}\n"
                f"Expiry Date: {contract.expiry_date}\n"
                f"Days Until Expiry: {days_until_expiry}\n\n"
            )

        # Finalize the message body
        message_body += "Please take the necessary actions."

        # Customize your notification message here
        subject = "Contract Expiry Notification"
        recipient_list = ['***@gmail.com', '***@gmail.com']  # Change this to your desired recipient email(s)

        # Send email notification
        send_mail(subject, message_body, '***@gmail.com', recipient_list)