79 lines
3.2 KiB
Python
79 lines
3.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
Import des mails de GestioCOF dans la base de donnée
|
||
|
|
"""
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
from custommail.models import VariableType, CustomMail, CustomMailVariable
|
||
|
|
|
||
|
|
from django.core.management.base import BaseCommand
|
||
|
|
from django.contrib.contenttypes.models import ContentType
|
||
|
|
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = ("Va chercher les données mails de GestioCOF stocké au format json "
|
||
|
|
"dans /gestioncof/management/data/custommails.json. Le format des "
|
||
|
|
"données est celui donné par la commande :"
|
||
|
|
" `python manage.py dumpdata custommail --natural-foreign` "
|
||
|
|
"La bonne façon de mettre à jour ce fichier est donc de le "
|
||
|
|
"charger à l'aide de syncmails, le faire les modifications à "
|
||
|
|
"l'aide de l'interface administration et/ou du shell puis de le "
|
||
|
|
"remplacer par le nouveau résultat de la commande précédente.")
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
path = os.path.join(
|
||
|
|
os.path.dirname(os.path.dirname(__file__)),
|
||
|
|
'data', 'custommail.json')
|
||
|
|
with open(path, 'r') as jsonfile:
|
||
|
|
mail_data = json.load(jsonfile)
|
||
|
|
|
||
|
|
# On se souvient à quel objet correspond quel pk du json
|
||
|
|
assoc = {'types': {}, 'mails': {}}
|
||
|
|
status = {'synced': 0, 'unchanged': 0}
|
||
|
|
|
||
|
|
for obj in mail_data:
|
||
|
|
model = obj['model']
|
||
|
|
fields = obj['fields']
|
||
|
|
|
||
|
|
# Pour les trois types d'objets :
|
||
|
|
# - On récupère les objets référencés par les clefs étrangères
|
||
|
|
# - On crée l'objet si nécessaire
|
||
|
|
# - On le stocke éventuellement dans les deux dictionnaires définis
|
||
|
|
# plus haut
|
||
|
|
|
||
|
|
# Variable types
|
||
|
|
if model == 'custommail.variabletype':
|
||
|
|
fields['inner1'] = assoc['types'].get(fields['inner1'])
|
||
|
|
fields['inner2'] = assoc['types'].get(fields['inner2'])
|
||
|
|
if fields['typ'] == 'model':
|
||
|
|
fields['content_type'] = (
|
||
|
|
ContentType.objects
|
||
|
|
.get_by_natural_key(*fields['content_type'])
|
||
|
|
)
|
||
|
|
var_type, _ = VariableType.objects.get_or_create(**fields)
|
||
|
|
assoc['types'][obj['pk']] = var_type
|
||
|
|
|
||
|
|
# Custom mails
|
||
|
|
if model == 'custommail.custommail':
|
||
|
|
mail, created = CustomMail.objects.get_or_create(**fields)
|
||
|
|
assoc['mails'][obj['pk']] = mail
|
||
|
|
if created:
|
||
|
|
self.stdout.write(
|
||
|
|
'SYNCED {:s}'.format(fields['shortname']))
|
||
|
|
status['synced'] += 1
|
||
|
|
else:
|
||
|
|
status['unchanged'] += 1
|
||
|
|
|
||
|
|
# Variables
|
||
|
|
if model == 'custommail.custommailvariable':
|
||
|
|
fields['custommail'] = assoc['mails'].get(fields['custommail'])
|
||
|
|
fields['typ'] = assoc['types'].get(fields['typ'])
|
||
|
|
CustomMailVariable.objects.get_or_create(**fields)
|
||
|
|
|
||
|
|
# C'est agréable d'avoir le résultat affiché
|
||
|
|
self.stdout.write(
|
||
|
|
'{synced:d} mails synchronized {unchanged:d} unchanged'
|
||
|
|
.format(**status)
|
||
|
|
)
|