from datetime import date, timedelta from django.db import models from django.utils import timezone from django.utils.translation import ugettext_lazy as _ from django.conf import settings from django.core.exceptions import ValidationError from django.core.urlresolvers import reverse from django.contrib.auth.forms import PasswordResetForm from core.models import User import MySQLdb def validate_type(value): if value not in settings.SITH_SUBSCRIPTIONS.keys(): raise ValidationError(_('Bad subscription type')) def validate_payment(value): if value not in settings.SITH_SUBSCRIPTION_PAYMENT_METHOD: raise ValidationError(_('Bad payment method')) class Subscriber(User): class Meta: proxy = True def is_subscribed(self): s = self.subscriptions.last() return s.is_valid_now() if s is not None else False def save(self): create = False if not self.id: create = True super(Subscriber, self).save() if create and settings.IS_OLD_MYSQL_PRESENT: try: # Create user on the old site: TODO remove me! db = MySQLdb.connect(**settings.OLD_MYSQL_INFOS) c = db.cursor() c.execute("""INSERT INTO utilisateurs (id_utilisateur, nom_utl, prenom_utl, email_utl, hash_utl, ae_utl) VALUES (%s, %s, %s, %s, %s, %s)""", (self.id, self.last_name, self.first_name, self.email, "valid", "1")) db.commit() except Exception as e: with open(settings.BASE_DIR+"/user_fail.log", "a") as f: print("FAIL to add user %s (%s %s - %s) to old site" % (self.id, self.first_name.encode('utf-8'), self.last_name.encode('utf-8'), self.email), file=f) print("Reason: %s" % (repr(e)), file=f) db.rollback() class Subscription(models.Model): member = models.ForeignKey(Subscriber, related_name='subscriptions') subscription_type = models.CharField(_('subscription type'), max_length=255, choices=((k, v['name']) for k,v in sorted(settings.SITH_SUBSCRIPTIONS.items()))) subscription_start = models.DateField(_('subscription start')) subscription_end = models.DateField(_('subscription end')) payment_method = models.CharField(_('payment method'), max_length=255, choices=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD) location = models.CharField(choices=settings.SITH_SUBSCRIPTION_LOCATIONS, max_length=20, verbose_name=_('location')) class Meta: ordering = ['subscription_start',] def clean(self): try: for s in Subscription.objects.filter(member=self.member).exclude(pk=self.pk).all(): if s.is_valid_now(): raise ValidationError(_("You can not subscribe many time for the same period")) except: # This should not happen, because the form should have handled the data before, but sadly, it still # calls the model validation :'( # TODO see SubscriptionForm's clean method raise ValidationError(_("Subscription error")) def save(self): super(Subscription, self).save() from counter.models import Customer if not Customer.objects.filter(user=self.member).exists(): last_id = Customer.objects.count() + 1504 # Number to keep a continuity with the old site Customer(user=self.member, account_id=Customer.generate_account_id(last_id+1), amount=0).save() form = PasswordResetForm({'email': self.member.email}) if form.is_valid(): form.save(use_https=True, email_template_name='core/new_user_email.jinja', subject_template_name='core/new_user_email_subject.jinja', from_email="ae@utbm.fr") self.member.make_home() if settings.IS_OLD_MYSQL_PRESENT: try: # Create subscription on the old site: TODO remove me! LOCATION = { "SEVENANS": 5, "BELFORT": 6, "MONTBELIARD": 9, "EBOUTIC": 5, } TYPE = { 'un-semestre' : 0, 'deux-semestres' : 1, 'cursus-tronc-commun' : 2, 'cursus-branche' : 3, 'membre-honoraire' : 4, 'assidu' : 5, 'amicale/doceo' : 6, 'reseau-ut' : 7, 'crous' : 8, 'sbarro/esta' : 9, 'cursus-alternant' : 10, } PAYMENT = { "CHECK" : 1, "CARD" : 2, "CASH" : 3, "OTHER" : 4, "EBOUTIC" : 5, "OTHER" : 0, } db = MySQLdb.connect(**settings.OLD_MYSQL_INFOS) c = db.cursor() c.execute("""INSERT INTO ae_cotisations (id_utilisateur, date_cotis, date_fin_cotis, mode_paiement_cotis, type_cotis, id_comptoir) VALUES (%s, %s, %s, %s, %s, %s)""", (self.member.id, self.subscription_start, self.subscription_end, PAYMENT[self.payment_method], TYPE[self.subscription_type], LOCATION[self.location])) db.commit() except Exception as e: with open(settings.BASE_DIR+"/subscription_fail.log", "a") as f: print("FAIL to add subscription to %s to old site" % (self.member), file=f) print("Reason: %s" % (repr(e)), file=f) db.rollback() def get_absolute_url(self): return reverse('core:user_edit', kwargs={'user_id': self.member.pk}) def __str__(self): if hasattr(self, "member") and self.member is not None: return self.member.username+' - '+str(self.pk) else: return 'No user - '+str(self.pk) @staticmethod def compute_start(d=date.today(), duration=1): """ This function computes the start date of the subscription with respect to the given date (default is today), and the start date given in settings.SITH_START_DATE. It takes the nearest past start date. Exemples: with SITH_START_DATE = (8, 15) Today -> Start date 2015-03-17 -> 2015-02-15 2015-01-11 -> 2014-08-15 """ if duration <= 2: # Sliding subscriptions for 1 or 2 semesters return d today = d year = today.year start = date(year, settings.SITH_START_DATE[0], settings.SITH_START_DATE[1]) start2 = start.replace(month=(start.month+6)%12) if start > start2: start, start2 = start2, start if today < start: return start2.replace(year=year-1) elif today < start2: return start else: return start2 @staticmethod def compute_end(duration, start=None): """ This function compute the end date of the subscription given a start date and a duration in number of semestre Exemple: Start - Duration -> End date 2015-09-18 - 1 -> 2016-03-18 2015-09-18 - 2 -> 2016-09-18 2015-09-18 - 3 -> 2017-03-18 2015-09-18 - 4 -> 2017-09-18 """ if start is None: start = Subscription.compute_start(duration=duration) # This can certainly be simplified, but it works like this try: return start.replace(month=(start.month-1+6*duration)%12+1, year=start.year+int(duration/2)+(1 if start.month > 6 and duration%2 == 1 else 0)) except ValueError as e: return start.replace(day=1, month=(start.month+6*duration)%12+1, year=start.year+int(duration/2)+(1 if start.month > 6 and duration%2 == 1 else 0)) def can_be_edited_by(self, user): return user.is_in_group(settings.SITH_MAIN_BOARD_GROUP) or user.is_root def is_valid_now(self): return self.subscription_start <= date.today() and date.today() <= self.subscription_end def guy_test(date, duration=4): print(str(date)+" - "+str(duration)+" -> "+str(Subscription.compute_start(date, duration))) def bibou_test(duration, date=date.today()): print(str(date)+" - "+str(duration)+" -> "+str(Subscription.compute_end(duration, Subscription.compute_start(date, duration)))) def guy(): guy_test(date(2015, 7, 11)) guy_test(date(2015, 8, 11)) guy_test(date(2015, 2, 17)) guy_test(date(2015, 3, 17)) guy_test(date(2015, 1, 11)) guy_test(date(2015, 2, 11)) guy_test(date(2015, 8, 17)) guy_test(date(2015, 9, 17)) print('='*80) guy_test(date(2015, 7, 11), 1) guy_test(date(2015, 8, 11), 2) guy_test(date(2015, 2, 17), 3) guy_test(date(2015, 3, 17), 4) guy_test(date(2015, 1, 11), 1) guy_test(date(2015, 2, 11), 2) guy_test(date(2015, 8, 17), 3) guy_test(date(2015, 9, 17), 4) print('='*80) bibou_test(1, date(2015, 2, 18)) bibou_test(2, date(2015, 2, 18)) bibou_test(3, date(2015, 2, 18)) bibou_test(4, date(2015, 2, 18)) bibou_test(1, date(2015, 9, 18)) bibou_test(2, date(2015, 9, 18)) bibou_test(3, date(2015, 9, 18)) bibou_test(4, date(2015, 9, 18)) print('='*80) bibou_test(1, date(2000, 2, 29)) bibou_test(2, date(2000, 2, 29)) bibou_test(1, date(2000, 5, 31)) bibou_test(1, date(2000, 7, 31)) bibou_test(1) bibou_test(2) bibou_test(3) bibou_test(4) if __name__ == "__main__": guy()