Update dev_membership.py

This commit is contained in:
zainab2097 2024-08-13 10:53:57 +03:00 committed by GitHub
parent 177a0c49de
commit 335f6157cb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions

View File

@ -143,11 +143,14 @@ class DevMembership(models.Model):
if not self.invoice_id and not self.is_free:
raise ValidationError(_('Please Create Membership Invoice'))
if self.invoice_id.payment_state not in ['paid','in_payment'] and not self.is_free:
raise ValidationError(_('Membership Invoice is not paid.\nPlease Paid membership invoice and active the membership.'))
raise ValidationError(_('Membership Invoice is not paid.\nPlease pay the membership invoice and activate the membership.'))
else:
self.invoice_id = self.invoice_id.payment_state
# Assuming you have a field to track payment state
self.payment_state = self.invoice_id.payment_state
self.state = 'active'
self.partner_id.product_id = self.product_id.id
if not self.partner_id.join_date:
self.partner_id.join_date = self.from_date
if not self.partner_id.membrship_level:
@ -155,6 +158,7 @@ class DevMembership(models.Model):
if not self.partner_id.memebership_end_date:
self.partner_id.memebership_end_date = self.to_date
def action_cancel_membership(self):
self.state = 'cancel'