# -*- coding: utf-8 -*- #!/usr/bin/env python import os import csv from datetime import datetime, time, date import calebasse.settings import django.core.management from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned django.core.management.setup_environ(calebasse.settings) from django.contrib.auth.models import User from calebasse.actes.models import EventAct from calebasse.agenda.models import Event, EventType from calebasse.dossiers.models import PatientRecord, Status, FileState from calebasse.ressources.models import Service from calebasse.personnes.models import Worker, Holiday, TimeTable, PERIODICITIES from calebasse.personnes.forms import PERIOD_LIST_TO_FIELDS from calebasse.ressources.models import WorkerType, HolidayType, SchoolType, School, SchoolLevel # Configuration db_path = "./scripts/20121221-192258" dbs = ["F_ST_ETIENNE_SESSAD_TED", "F_ST_ETIENNE_CMPP", "F_ST_ETIENNE_CAMSP", "F_ST_ETIENNE_SESSAD"] def _to_date(str_date): if not str_date: return None return datetime.strptime(str_date[:-13], "%Y-%m-%d") def _to_int(str_int): if not str_int: return None return int(str_int) def discipline_mapper(tables_data, service): for line in tables_data['discipline']: # Insert workertype if not WorkerType.objects.filter(name=line['libelle']): WorkerType.objects.create(name=line['libelle']) def intervenants_mapper(tables_data, service): for line in tables_data['intervenants']: # Insert workers for disp in tables_data['discipline']: if disp['id'] == line['discipline']: type = WorkerType.objects.get(name=disp['libelle']) # TODO : import actif or not worker, created = Worker.objects.get_or_create( type=type, last_name=line['nom'], first_name=line['prenom'], email=line['email'], phone=line['tel'], gender=int(line['titre']), ) worker.services.add(service) def dossiers_mapper(tables_data, service): global dossiers for line in tables_data['dossiers']: status = Status.objects.filter(type="ACCUEIL").filter(services=service) creator = User.objects.get(id=1) gender = _to_int(line['nais_sexe']) if gender == 0: gender = None # TODO: add more fields patient, created = PatientRecord.objects.get_or_create(first_name=line['prenom'], last_name=line['nom'], birthdate=_to_date(line['nais_date']), twinning_rank=_to_int(line['nais_rang']), gender=gender, service=service, creator=creator) dossiers[line['id']] = patient if not created: if not line['ins_date']: # Hack when there is no inscription date put 01/01/1970 line['ins_date'] = "1970-01-01 00:00:00.000" fs = FileState.objects.create(status=status[0], author=creator, date_selected=_to_date(line['ins_date']), previous_state=None, patient=patient) patient.last_state = fs patient.save() if line['sor_date']: status = Status.objects.filter(type="CLOS").filter(services=service) fs = FileState.objects.create(status=status[0], author=creator, date_selected=_to_date(line['sor_date']), previous_state=None, patient=patient) patient.last_state = fs patient.save() def rs_mapper(tables_data, service): global dossiers event_type = EventType.objects.get( label=u"Rendez-vous patient" ) for line in tables_data['rs']: if dossiers.has_key(line['enfant_id']): patient = dossiers[line['enfant_id']] strdate = line['date_rdv'][:-13] + ' ' + line['heure'][11:-4] date = datetime.strptime(strdate, "%Y-%m-%d %H:%M:%S") # TODO: add act_type # act_event = EventAct.objects.get_or_create( # title=line['libelle'], # event_type=event_type, # patient=patient, # act_type=act_type, # date=date # ) else: # TODO: if no patient add event pass def conge_mapper(tables_data, service): """ """ for line in tables_data['conge']: pass def ev_mapper(tables_data, service): """ """ pass def notes_mapper(tables_data, service): """ """ pass def _get_dict(cols, line): """""" res = {} for i, data in enumerate(line): res[cols[i]] = data.decode('utf-8') return res PERIOD_FAURE_NOUS = {1 : 1, 2 : 2, 3 : 3, 4 : 4, 5: 6, 6: 7, 7: 8, 8: 9, 9: None, 10: 10, 12: 11, 13: 12, } JOURS = {1: 'lundi', 2: 'mardi', 3: 'mercredi', 4: 'jeudi', 5: 'vendredi' } dic_worker = {} def main(): """ """ t_inconnu = SchoolType.objects.get(name='Inconnu') t_maternelle = SchoolType.objects.get(name='Ecole maternelle') t_primaire = SchoolType.objects.get(name='Ecole primaire') t_college = SchoolType.objects.get(name='Collège') t_lycee = SchoolType.objects.get(name='Lycée') for db in dbs: if "F_ST_ETIENNE_CMPP" == db: service = Service.objects.get(name="CMPP") elif "F_ST_ETIENNE_CAMSP" == db: service = Service.objects.get(name="CAMSP") elif "F_ST_ETIENNE_SESSAD_TED" == db: service = Service.objects.get(name="SESSAD TED") elif "F_ST_ETIENNE_SESSAD" == db: service = Service.objects.get(name="SESSAD DYS") l = [] csvfile = open(os.path.join(db_path, db, 'classes.csv'), 'rb') csvlines = csv.reader(csvfile, delimiter=';', quotechar='|') cols = csvlines.next() for line in csvlines: data = _get_dict(cols, line) l.append(data) csvfile.close() for classe in l: SchoolLevel(name=classe['libelle'], old_id=classe['id'], old_service=service.name).save() l = [] csvfile = open(os.path.join(db_path, db, 'ecoles.csv'), 'rb') csvlines = csv.reader(csvfile, delimiter=';', quotechar='|') cols = csvlines.next() for line in csvlines: data = _get_dict(cols, line) l.append(data) csvfile.close() for ecole in l: t = t_inconnu ecole_nom = ecole['nom'].encode('utf-8') if 'mat' in ecole_nom or 'Mat' in ecole_nom or 'Maternelle' in ecole_nom or 'maternelle' in ecole_nom or 'MATERNELLE' in ecole_nom: t = t_maternelle elif 'Primaire' in ecole_nom or 'primaire' in ecole_nom or 'PRIMAIRE' in ecole_nom: t = t_primaire elif 'COLLEGE' in ecole_nom or 'COLLÈGE' in ecole_nom or 'collège' in ecole_nom or 'college' in ecole_nom or 'Collège' in ecole_nom or 'College' in ecole_nom: t = t_college elif 'LYCEE' in ecole_nom or 'LYCÉE' in ecole_nom or 'lycée' in ecole_nom or 'lycee' in ecole_nom or 'Lycée' in ecole_nom or 'Lycée' in ecole_nom: t = t_lycee School(name=ecole['nom'], school_type = t, address = ecole['voie'], address_complement = ecole['suite'], zip_code = ecole['codepostal'], city = ecole['ville'], phone = ecole['tel'], fax = ecole['fax'], email = ecole['email'], director_name = ecole['nom_dir'], description = ecole['infos'], old_id=ecole['id'], old_service=service.name).save() if __name__ == "__main__": main()