X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/875efc3090b43807d41438a98f3e84ceafecab51..c36c3891e56611e2ad4c518daa46b8cc3f28188e:/src/paypal/models.py

diff --git a/src/paypal/models.py b/src/paypal/models.py
index 80aa357a6..6d45bcdb5 100644
--- a/src/paypal/models.py
+++ b/src/paypal/models.py
@@ -1,6 +1,5 @@
-# -*- 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.
+# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
 #
 # from django.contrib.auth import get_user_model
 from django.contrib.auth.models import User
@@ -14,15 +13,15 @@ class BillingPlan(models.Model):
 
 class BillingAgreement(models.Model):
     agreement_id = models.CharField(max_length=32)
-    user = models.ForeignKey(User)
-    plan = models.ForeignKey(BillingPlan)
+    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)
 
-    def get_agreement(self):
-        from .rest import get_agreement
-        return get_agreement(self.agreement_id)
-
     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)