Migrate clubs

club/migrations/0003_auto_20160813_1448.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('club', '0002_club_home'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='club',
+            name='name',
+            field=models.CharField(verbose_name='name', max_length=64),
+        ),
+    ]

club/migrations/0004_auto_20160813_1551.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('club', '0003_auto_20160813_1448'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='membership',
+            name='description',
+            field=models.CharField(verbose_name='description', blank=True, max_length=128),
+        ),
+    ]

club/models.py

@@ -15,7 +15,7 @@ class Club(models.Model):
     """
     The Club class, made as a tree to allow nice tidy organization
     """
-    name = models.CharField(_('name'), max_length=30)
+    name = models.CharField(_('name'), max_length=64)
     parent = models.ForeignKey('Club', related_name='children', null=True, blank=True)
     unix_name = models.CharField(_('unix name'), max_length=30, unique=True,
             validators=[
@@ -72,10 +72,10 @@ class Club(models.Model):
     def save(self, *args, **kwargs):
         with transaction.atomic():
             creation = False
-            if self.id is None:
+            old = Club.objects.filter(id=self.id).first()
+            if not old:
                 creation = True
             else:
-                old = Club.objects.filter(id=self.id).first()
                 if old.unix_name != self.unix_name:
                     self._change_unixname(self.unix_name)
             super(Club, self).save(*args, **kwargs)
@@ -148,7 +148,7 @@ class Membership(models.Model):
     end_date = models.DateField(_('end date'), null=True, blank=True)
     role = models.IntegerField(_('role'), choices=sorted(settings.SITH_CLUB_ROLES.items()),
             default=sorted(settings.SITH_CLUB_ROLES.items())[0][0])
-    description = models.CharField(_('description'), max_length=30, null=False, blank=True)
+    description = models.CharField(_('description'), max_length=128, null=False, blank=True)
 
     def clean(self):
         sub = Subscriber.objects.filter(pk=self.user.pk).first()

club/templates/club/club_detail.jinja

@@ -16,7 +16,7 @@
     <h3>{{ club.name }}</h3>
     <p>{{ club.address }}</p>
     <ul>
-        {% for m in club.members.all() %}
+        {% for m in club.members.filter(end_date=None).all() %}
         <li>{{ m }}</li>
         {% endfor %}
     </ul>

club/templates/club/club_tools.jinja

@@ -19,6 +19,9 @@
     {% if object.club_account %}
     <li>Accouting: <a href="{{ url('accounting:club_details', c_account_id=object.club_account.id) }}">{{ object }}</a></li>
     {% endif %}
+    {% if object.unix_name == settings.SITH_LAUNDERETTE_MANAGER['unix_name'] %}
+    <li><a href="{{ url('launderette:launderette_list') }}">{% trans %}Launderette{% endtrans %}</a></li>
+    {% endif %}
 </ul>
 {% endblock %}
 

core/management/commands/populate.py

@@ -43,13 +43,13 @@ class Command(BaseCommand):
         home_root.save()
         club_root = SithFile(parent=None, name="clubs", is_folder=True, owner=root)
         club_root.save()
-        main_club = Club(name=settings.SITH_MAIN_CLUB['name'], unix_name=settings.SITH_MAIN_CLUB['unix_name'],
+        main_club = Club(id=1, name=settings.SITH_MAIN_CLUB['name'], unix_name=settings.SITH_MAIN_CLUB['unix_name'],
                 address=settings.SITH_MAIN_CLUB['address'])
         main_club.save()
-        bar_club = Club(name=settings.SITH_BAR_MANAGER['name'], unix_name=settings.SITH_BAR_MANAGER['unix_name'],
+        bar_club = Club(id=2, name=settings.SITH_BAR_MANAGER['name'], unix_name=settings.SITH_BAR_MANAGER['unix_name'],
                 address=settings.SITH_BAR_MANAGER['address'])
         bar_club.save()
-        launderette_club = Club(name=settings.SITH_LAUNDERETTE_MANAGER['name'],
+        launderette_club = Club(id=84, name=settings.SITH_LAUNDERETTE_MANAGER['name'],
                 unix_name=settings.SITH_LAUNDERETTE_MANAGER['unix_name'],
                 address=settings.SITH_LAUNDERETTE_MANAGER['address'])
         launderette_club.save()

migrate.py

@@ -2,15 +2,20 @@ import MySQLdb
 import os
 import django
 import random
+from io import StringIO
 
 os.environ["DJANGO_SETTINGS_MODULE"] = "sith.settings"
+os.environ['DJANGO_COLORS'] = 'nocolor'
 django.setup()
 
-from core.models import User, SithFile
 from django.db import IntegrityError
 from django.conf import settings
+from django.core.management import call_command
+from django.db import connection
 
-PROFILE_ROOT = "/data/matmatronch/"
+
+from core.models import User, SithFile
+from club.models import Club, Membership
 
 db = MySQLdb.connect(
         host="ae-db",
@@ -19,26 +24,20 @@ db = MySQLdb.connect(
         db="ae2-taiste",
         charset='utf8',
         use_unicode=True)
-c = db.cursor(MySQLdb.cursors.DictCursor)
-c.execute("""
-SELECT *
-FROM utilisateurs utl
-JOIN utl_etu ue
-ON ue.id_utilisateur = utl.id_utilisateur
-JOIN utl_etu_utbm ueu
-ON ueu.id_utilisateur = utl.id_utilisateur
-JOIN utl_extra uxtra
-ON uxtra.id_utilisateur = utl.id_utilisateur
-JOIN loc_ville ville
-ON utl.id_ville = ville.id_ville
-WHERE utl.id_utilisateur > 9000
-""")
-User.objects.filter(id__gt=0).delete()
-print("Users deleted")
-# guy = c.fetchone()
-# for k,v in sorted(guy.items()):
-#     print("%40s | %40s" % (k, v))
 
+def reset_index(*args):
+    sqlcmd = StringIO()
+    call_command("sqlsequencereset", *args, stdout=sqlcmd)
+    cursor = connection.cursor()
+    cursor.execute(sqlcmd.getvalue())
+
+def to_unicode(s):
+    if s:
+        return bytes(s, 'cp1252', errors="replace").decode('utf-8', errors='replace')
+    return ""
+
+
+def migrate_users():
     SEX = {'1': 'MAN', '2': 'WOMAN', None: 'MAN'}
     TSHIRT = {
             None: '-',
@@ -77,11 +76,6 @@ DEPARTMENTS = {
             None: 'NA',
             }
 
-def to_unicode(s):
-    if s:
-        return bytes(s, 'cp1252', errors="replace").decode('utf-8', errors='replace')
-    return ""
-
     def get_random_free_email():
         id = random.randrange(4000)
         email = "no_email_%s@git.an" % random.randrange(4000, 40000)
@@ -89,6 +83,23 @@ def get_random_free_email():
             email = "no_email_%s@git.an" % random.randrange(4000, 40000)
         return email
 
+    c = db.cursor(MySQLdb.cursors.DictCursor)
+    c.execute("""
+    SELECT *
+    FROM utilisateurs utl
+    JOIN utl_etu ue
+    ON ue.id_utilisateur = utl.id_utilisateur
+    JOIN utl_etu_utbm ueu
+    ON ueu.id_utilisateur = utl.id_utilisateur
+    JOIN utl_extra uxtra
+    ON uxtra.id_utilisateur = utl.id_utilisateur
+    JOIN loc_ville ville
+    ON utl.id_ville = ville.id_ville
+    -- WHERE utl.id_utilisateur > 9000
+    """)
+    User.objects.filter(id__gt=0).delete()
+    print("Users deleted")
+
     for u in c.fetchall():
         try:
             new = User(
@@ -127,11 +138,17 @@ for u in c.fetchall():
         except Exception as e:
             print("FAIL for user %s: %s" % (u['id_utilisateur'], repr(e)))
     c.close()
+    reset_index('core')
+
+def migrate_profile_pict():
+    PROFILE_ROOT = "/data/matmatronch/"
 
     from os import listdir
     from django.core.files import File
 
     profile = SithFile.objects.filter(parent=None, name="profiles").first()
+    profile.children.all().delete()
+    print("Profiles pictures deleted")
     for filename in listdir(PROFILE_ROOT):
         if filename.split('.')[-2] != "mini":
             try:
@@ -139,6 +156,7 @@ for filename in listdir(PROFILE_ROOT):
                 user = User.objects.filter(id=int(uid)).first()
                 if user:
                     f = File(open(PROFILE_ROOT + '/' + filename, 'rb'))
+                    f.name = f.name.split('/')[-1]
                     t = filename.split('.')[1]
                     new_file = SithFile(parent=profile, name=filename,
                             file=f, owner=user, is_folder=False, mime_type="image/jpeg", size=f.size)
@@ -157,6 +175,74 @@ for filename in listdir(PROFILE_ROOT):
             except Exception as e:
                 print(repr(e))
 
-#     profile_pict = models.OneToOneField('SithFile', related_name='profile_of', verbose_name=_("profile"), null=True, blank=True)
-#     avatar_pict = models.OneToOneField('SithFile', related_name='avatar_of', verbose_name=_("avatar"), null=True, blank=True)
-#     scrub_pict = models.OneToOneField('SithFile', related_name='scrub_of', verbose_name=_("scrub"), null=True, blank=True)
+def migrate_clubs():
+    cur = db.cursor(MySQLdb.cursors.DictCursor)
+    cur.execute("""
+    SELECT *
+    FROM asso asso
+    WHERE nom_unix_asso <> "ae"
+        AND nom_unix_asso <> "bdf"
+        AND nom_unix_asso <> "laverie"
+    """)
+    # club = cur.fetchone()
+    # for k,v in club.items():
+    #     print("%40s | %40s" % (k, v))
+
+    for c in cur.fetchall():
+        try:
+            new = Club(
+                    id=c['id_asso'],
+                    name=to_unicode(c['nom_asso']),
+                    unix_name=to_unicode(c['nom_unix_asso']),
+                    address=to_unicode(c['adresse_postale']),
+                    )
+            new.save()
+        except Exception as e:
+                print("FAIL for club %s: %s" % (c['nom_unix_asso'], repr(e)))
+    cur.execute("""
+    SELECT *
+    FROM asso
+    """)
+    for c in cur.fetchall():
+        club = Club.objects.filter(id=c['id_asso']).first()
+        parent = Club.objects.filter(id=c['id_asso_parent']).first()
+        club.parent = parent
+        club.save()
+    cur.close()
+
+def migrate_club_memberships():
+    cur = db.cursor(MySQLdb.cursors.DictCursor)
+    cur.execute("""
+    SELECT *
+    FROM asso_membre
+    """)
+
+    Membership.objects.all().delete()
+    print("Memberships deleted")
+    for m in cur.fetchall():
+        try:
+            club = Club.objects.filter(id=m['id_asso']).first()
+            user = User.objects.filter(id=m['id_utilisateur']).first()
+            if club and user:
+                new = Membership(
+                        club=club,
+                        user=user,
+                        start_date=m['date_debut'],
+                        end_date=m['date_fin'],
+                        role=m['role'],
+                        description=to_unicode(m['desc_role']),
+                        )
+                new.save()
+        except Exception as e:
+                print("FAIL for club membership %s: %s" % (m['id_asso'], repr(e)))
+    cur.close()
+
+
+def main():
+    # migrate_users()
+    # migrate_profile_pict()
+    # migrate_clubs()
+    migrate_club_memberships()
+
+if __name__ == "__main__":
+    main()