X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/468c2e650c854ffa691afc73aa1f41ce61283cdb..4e361d55331351f56077e5f9999ecb4ee7e93819:/src/paypal/models.py

diff --git a/src/paypal/models.py b/src/paypal/models.py
index 527e80458..2ce3635a7 100644
--- a/src/paypal/models.py
+++ b/src/paypal/models.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
@@ -14,7 +13,11 @@ class BillingPlan(models.Model):
 
 class BillingAgreement(models.Model):
     agreement_id = models.CharField(max_length=32)
-    user = models.ForeignKey(User)
-    plan = models.ForeignKey(BillingPlan)
+    user = models.ForeignKey(User, models.PROTECT)
+    plan = models.ForeignKey(BillingPlan, models.PROTECT)
     active = models.BooleanField(max_length=32)
     token = models.CharField(max_length=32)
+
+    def check_agreement(self):
+        from .rest import check_agreement
+        return check_agreement(self.agreement_id)