Make verse numbers, footnote anchors, themes, dynamic inserts unselectable. Fixes...

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

diff --git a/src/club/models.py b/src/club/models.py
index f36aada..6fd58e0 100644 (file)
--- a/src/club/models.py
+++ b/src/club/models.py
@@ -12,6 +12,7 @@ from django import template
 from django.utils.timezone import now
 from django.utils.translation import ugettext_lazy as _, ungettext, ugettext, get_language
 from catalogue.utils import get_random_hash
 from django.utils.timezone import now
 from django.utils.translation import ugettext_lazy as _, ungettext, ugettext, get_language
 from catalogue.utils import get_random_hash

+from messaging.states import Level

 from .payment_methods import recurring_payment_method, single_payment_method
 from .payu import models as payu_models
 from . import utils
 from .payment_methods import recurring_payment_method, single_payment_method
 from .payu import models as payu_models
 from . import utils


@@ -113,14 +114,14 @@ class Schedule(models.Model):
     def update_contact(self):
         Contact = apps.get_model('messaging', 'Contact')
         if not self.payed_at:
     def update_contact(self):
         Contact = apps.get_model('messaging', 'Contact')
         if not self.payed_at:

-            level = Contact.TRIED
+            level = Level.TRIED

             since = self.started_at
         else:
             since = self.payed_at
             if self.is_recurring():
             since = self.started_at
         else:
             since = self.payed_at
             if self.is_recurring():

-                level = Contact.RECURRING
+                level = Level.RECURRING

             else:
             else:

-                level = Contact.SINGLE
+                level = Level.SINGLE

         Contact.update(self.email, level, since, self.expires_at)
 
 
         Contact.update(self.email, level, since, self.expires_at)