fix

[wolnelektury.git] / src / paypal / models.py

diff --git a/src/paypal/models.py b/src/paypal/models.py
index 2ce3635..3baf99e 100644 (file)
--- a/src/paypal/models.py
+++ b/src/paypal/models.py
@@ -13,7 +13,7 @@ class BillingPlan(models.Model):
 
 class BillingAgreement(models.Model):
     agreement_id = models.CharField(max_length=32)
-    user = models.ForeignKey(User, models.PROTECT)
+    schedule = models.ForeignKey('club.Schedule', models.PROTECT)
     plan = models.ForeignKey(BillingPlan, models.PROTECT)
     active = models.BooleanField(max_length=32)
     token = models.CharField(max_length=32)
@@ -21,3 +21,7 @@ class BillingAgreement(models.Model):
     def check_agreement(self):
         from .rest import check_agreement
         return check_agreement(self.agreement_id)
+
+    def get_donations(self, year):
+        from .rest import get_donations
+        return get_donations(self.agreement_id, year)