import MySQLdb import os import django import random from io import StringIO from pytz import timezone os.environ["DJANGO_SETTINGS_MODULE"] = "sith.settings" os.environ['DJANGO_COLORS'] = 'nocolor' django.setup() from django.db import IntegrityError from django.conf import settings from django.core.management import call_command from django.db import connection from django.forms import ValidationError from core.models import User, SithFile from club.models import Club, Membership from counter.models import Customer, Counter, Selling, Refilling, Product, ProductType, Permanency from subscription.models import Subscription, Subscriber from eboutic.models import Invoice, InvoiceItem db = MySQLdb.connect( host="ae-db", user="taiste_rw", passwd=input("password: "), db="ae2-taiste", charset='utf8', use_unicode=True) 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: '-', '': '-', 'NULL': '-', 'XS': 'XS', 'S': 'S', 'M': 'M', 'L': 'L', 'XL': 'XL', 'XXL': 'XXL', 'XXXL': 'XXXL', } ROLE = { 'doc': 'DOCTOR', 'etu': 'STUDENT', 'anc': 'FORMER STUDENT', 'ens': 'TEACHER', 'adm': 'ADMINISTRATIVE', 'srv': 'SERVICE', 'per': 'AGENT', None: '', } DEPARTMENTS = { 'tc': 'TC', 'gi': 'GI', 'gesc': 'GESC', 'na': 'NA', 'mc': 'MC', 'imap': 'IMAP', 'huma': 'HUMA', 'edim': 'EDIM', 'ee': 'EE', 'imsi': 'IMSI', 'truc': 'NA', None: 'NA', } def get_random_free_email(): id = random.randrange(4000) email = "no_email_%s@git.an" % random.randrange(4000, 40000) while User.objects.filter(email=email).exists(): email = "no_email_%s@git.an" % random.randrange(4000, 40000) return email c = db.cursor(MySQLdb.cursors.SSDictCursor) c.execute(""" SELECT * FROM utilisateurs utl LEFT JOIN utl_etu ue ON ue.id_utilisateur = utl.id_utilisateur LEFT JOIN utl_etu_utbm ueu ON ueu.id_utilisateur = utl.id_utilisateur LEFT JOIN utl_extra uxtra ON uxtra.id_utilisateur = utl.id_utilisateur LEFT JOIN loc_ville ville ON utl.id_ville = ville.id_ville -- WHERE utl.id_utilisateur = 9360 """) User.objects.filter(id__gt=0).delete() print("Users deleted") for u in c: try: new = User( id=u['id_utilisateur'], last_name=to_unicode(u['nom_utl']) or "Bou", first_name=to_unicode(u['prenom_utl']) or "Bi", email=u['email_utl'], second_email=u['email_utbm'] or "", date_of_birth=u['date_naissance_utl'], last_update=u['date_maj_utl'], nick_name=to_unicode(u['surnom_utbm']), sex=SEX[u['sexe_utl']], tshirt_size=TSHIRT[u['taille_tshirt_utl']], role=ROLE[u['role_utbm']], department=DEPARTMENTS[u['departement_utbm']], dpt_option=to_unicode(u['filiere_utbm']), semester=u['semestre_utbm'] or 0, quote=to_unicode(u['citation']), school=to_unicode(u['nom_ecole_etudiant']), promo=u['promo_utbm'] or 0, forum_signature=to_unicode(u['signature_utl']), address=(to_unicode(u['addresse_utl']) + ", " + to_unicode(u['cpostal_ville']) + " " + to_unicode(u['nom_ville'])), parent_address=(to_unicode(u['adresse_parents']) + ", " + to_unicode(u['cpostal_parents']) + " " + to_unicode(u['ville_parents'])), phone=u['tel_portable_utl'] or "", parent_phone=u['tel_parents'] or "", ) new.generate_username() new.save() except IntegrityError as e: if "Key (email)" in repr(e): new.email = get_random_free_email() new.save() print("New email generated") else: print("FAIL for user %s: %s" % (u['id_utilisateur'], repr(e))) except Exception as e: print("FAIL for user %s: %s" % (u['id_utilisateur'], repr(e))) c.close() 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: uid = filename.split('.')[0].split('-')[0] 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) if t == "identity": new_file.save() user.profile_pict = new_file user.save() elif t == "blouse": new_file.save() user.scrub_pict = new_file user.save() else: new_file.save() user.avatar_pict = new_file user.save() except Exception as e: print(repr(e)) def migrate_clubs(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) 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: 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: 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.SSDictCursor) cur.execute(""" SELECT * FROM asso_membre """) Membership.objects.all().delete() print("Memberships deleted") for m in cur: 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 migrate_subscriptions(): LOCATION = { 5: "SEVENANS", 6: "BELFORT", 9: "MONTBELIARD", None: "SEVENANS", } TYPE = { 0: 'un-semestre', 1: 'deux-semestres', 2: 'cursus-tronc-commun', 3: 'cursus-branche', 4: 'membre-honoraire', 5: 'assidu', 6: 'amicale/doceo', 7: 'reseau-ut', 8: 'crous', 9: 'sbarro/esta', 10: 'cursus-alternant', None: 'un-semestre', } PAYMENT = { 1: "CHECK", 2: "CARD", 3: "CASH", 4: "OTHER", 5: "EBOUTIC", 0: "OTHER", } cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM ae_cotisations """) Subscription.objects.all().delete() print("Subscriptions deleted") Customer.objects.all().delete() print("Customers deleted") for r in cur: try: user = Subscriber.objects.filter(id=r['id_utilisateur']).first() if user: new = Subscription( id=r['id_cotisation'], member=user, subscription_start=r['date_cotis'], subscription_end=r['date_fin_cotis'], subscription_type=TYPE[r['type_cotis']], payment_method=PAYMENT[r['mode_paiement_cotis']], location=LOCATION[r['id_comptoir']], ) new.save() except Exception as e: print("FAIL for subscription %s: %s" % (r['id_cotisation'], repr(e))) cur.close() def update_customer_account(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM ae_carte carte JOIN ae_cotisations cotis ON carte.id_cotisation = cotis.id_cotisation """) for r in cur: try: user = Customer.objects.filter(user_id=r['id_utilisateur']).first() if user: user.account_id = str(r['id_carte_ae']) + r['cle_carteae'].lower() user.save() except Exception as e: print("FAIL to update customer account for %s: %s" % (r['id_cotisation'], repr(e))) cur.close() def migrate_counters(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_comptoir """) Counter.objects.all().delete() for r in cur: try: club = Club.objects.filter(id=r['id_assocpt']).first() new = Counter( id=r['id_comptoir'], name=to_unicode(r['nom_cpt']), club=club, type="OFFICE", ) new.save() except Exception as e: print("FAIL to migrate counter %s: %s" % (r['id_comptoir'], repr(e))) cur.close() def reset_customer_amount(): Refilling.objects.all().delete() Selling.objects.all().delete() Invoice.objects.all().delete() for c in Customer.objects.all(): c.amount = 0 c.save() print("Customer amount reset") def migrate_refillings(): BANK = { 0: "OTHER", 1: "SOCIETE-GENERALE", 2: "BANQUE-POPULAIRE", 3: "BNP", 4: "CAISSE-EPARGNE", 5: "CIC", 6: "CREDIT-AGRICOLE", 7: "CREDIT-MUTUEL", 8: "CREDIT-LYONNAIS", 9: "LA-POSTE", 100: "OTHER", None: "OTHER", } cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_rechargements """) Refilling.objects.filter(payment_method="SITH_ACCOUNT").delete() print("Sith account refillings deleted") fail = 100 root_cust = Customer.objects.filter(user__id=0).first() mde = Counter.objects.filter(id=1).first() for r in cur: try: cust = Customer.objects.filter(user__id=r['id_utilisateur']).first() user = User.objects.filter(id=r['id_utilisateur']).first() if not cust: if not user: cust = root_cust else: cust = Customer(user=user, amount=0, account_id=Customer.generate_account_id(fail)) cust.save() fail += 1 op = User.objects.filter(id=r['id_utilisateur_operateur']).first() counter = Counter.objects.filter(id=r['id_comptoir']).first() new = Refilling( id=r['id_rechargement'], counter=counter or mde, customer=cust, operator=op or root_cust.user, amount=r['montant_rech']/100, bank=BANK[r['banque_rech']], date=r['date_rech'].replace(tzinfo=timezone('Europe/Paris')), ) new.save() except Exception as e: print("FAIL to migrate refilling %s for %s: %s" % (r['id_rechargement'], r['id_utilisateur'], repr(e))) cur.close() def migrate_typeproducts(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_type_produit """) ProductType.objects.all().delete() print("Product types deleted") for r in cur: try: new = ProductType( id=r['id_typeprod'], name=to_unicode(r['nom_typeprod']), description=to_unicode(r['description_typeprod']), ) new.save() except Exception as e: print("FAIL to migrate product type %s: %s" % (r['nom_typeprod'], repr(e))) cur.close() def migrate_products(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_produits """) Product.objects.all().delete() print("Product deleted") for r in cur: try: type = ProductType.objects.filter(id=r['id_typeprod']).first() club = Club.objects.filter(id=r['id_assocpt']).first() new = Product( id=r['id_produit'], product_type=type, name=to_unicode(r['nom_prod']), description=to_unicode(r['description_prod']), code=to_unicode(r['cbarre_prod']), purchase_price=r['prix_achat_prod']/100, selling_price=r['prix_vente_prod']/100, special_selling_price=r['prix_vente_barman_prod']/100, club=club, ) new.save() except Exception as e: print("FAIL to migrate product %s: %s" % (r['nom_prod'], repr(e))) cur.close() def migrate_products_to_counter(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_mise_en_vente """) for r in cur: try: product = Product.objects.filter(id=r['id_produit']).first() counter = Counter.objects.filter(id=r['id_comptoir']).first() counter.products.add(product) counter.save() except Exception as e: print("FAIL to set product %s in counter %s: %s" % (product, counter, repr(e))) cur.close() def migrate_invoices(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_vendu ven LEFT JOIN cpt_debitfacture fac ON ven.id_facture = fac.id_facture WHERE fac.mode_paiement = 'SG' """) Invoice.objects.all().delete() print("Invoices deleted") Refilling.objects.filter(payment_method="CARD").delete() print("Card refillings deleted") Selling.objects.filter(payment_method="CARD").delete() print("Card sellings deleted") root = User.objects.filter(id=0).first() for r in cur: try: product = Product.objects.filter(id=r['id_produit']).first() user = User.objects.filter(id=r['id_utilisateur_client']).first() i = Invoice.objects.filter(id=r['id_facture']).first() or Invoice(id=r['id_facture']) i.user = user or root for f in i._meta.local_fields: if f.name == "date": f.auto_now = False i.date = r['date_facture'].replace(tzinfo=timezone('Europe/Paris')) i.save() InvoiceItem(invoice=i, product_id=product.id, product_name=product.name, type_id=product.product_type.id, product_unit_price=r['prix_unit']/100, quantity=r['quantite']).save() except ValidationError as e: print(repr(e) + " for %s (%s)" % (customer, customer.user.id)) except Exception as e: print("FAIL to migrate invoice %s: %s" % (r['id_facture'], repr(e))) cur.close() for i in Invoice.objects.all(): for f in i._meta.local_fields: if f.name == "date": f.auto_now = False i.validate() def migrate_sellings(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_vendu ven LEFT JOIN cpt_debitfacture fac ON ven.id_facture = fac.id_facture WHERE fac.mode_paiement = 'AE' """) Selling.objects.filter(payment_method="SITH_ACCOUNT").delete() print("Sith account selling deleted") for r in cur: try: product = Product.objects.filter(id=r['id_produit']).first() club = Club.objects.filter(id=r['id_assocpt']).first() counter = Counter.objects.filter(id=r['id_comptoir']).first() op = User.objects.filter(id=r['id_utilisateur']).first() customer = Customer.objects.filter(user__id=r['id_utilisateur_client']).first() new = Selling( label=product.name, counter=counter, club=club, product=product, seller=op, customer=customer, unit_price=r['prix_unit']/100, quantity=r['quantite'], payment_method="SITH_ACCOUNT", date=r['date_facture'].replace(tzinfo=timezone('Europe/Paris')), ) new.save() except ValidationError as e: print(repr(e) + " for %s (%s)" % (customer, customer.user.id)) except Exception as e: print("FAIL to migrate selling %s: %s" % (r['id_facture'], repr(e))) cur.close() def migrate_permanencies(): cur = db.cursor(MySQLdb.cursors.SSDictCursor) cur.execute(""" SELECT * FROM cpt_tracking """) Permanency.objects.all().delete() print("Permanencies deleted") for r in cur: try: counter = Counter.objects.filter(id=r['id_comptoir']).first() user = User.objects.filter(id=r['id_utilisateur']).first() new = Permanency( user=user, counter=counter, start=r['logged_time'].replace(tzinfo=timezone('Europe/Paris')), end=r['closed_time'].replace(tzinfo=timezone('Europe/Paris')), ) new.save() except Exception as e: print("FAIL to migrate permanency: %s" % (repr(e))) cur.close() def main(): # migrate_users() # migrate_profile_pict() # migrate_clubs() # migrate_club_memberships() # migrate_subscriptions() # update_customer_account() # migrate_counters() # migrate_typeproducts() # migrate_products() # migrate_products_to_counter() # reset_customer_amount() # migrate_invoices() # migrate_refillings() # migrate_sellings() # migrate_permanencies() reset_index('core', 'counter') if __name__ == "__main__": main()