281 lines
10 KiB
Python
281 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
#!/usr/bin/env python
|
|
|
|
import os
|
|
import csv
|
|
|
|
from datetime import datetime, time
|
|
|
|
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
|
|
from calebasse.ressources.models import WorkerType, HolidayType
|
|
|
|
# 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
|
|
|
|
tables_data = {}
|
|
|
|
def main():
|
|
""" """
|
|
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")
|
|
|
|
csvfile = open(os.path.join(db_path, db, 'conge.csv'), 'rb')
|
|
csvlines = csv.reader(csvfile, delimiter=';', quotechar='|')
|
|
cols = csvlines.next()
|
|
tables_data[service.name] = []
|
|
for line in csvlines:
|
|
data = _get_dict(cols, line)
|
|
tables_data[service.name].append(data)
|
|
csvfile.close()
|
|
|
|
motif_unknown = HolidayType.objects.get(pk=18)
|
|
|
|
conges = {}
|
|
|
|
sname = 'CAMSP'
|
|
l_camsp = []
|
|
old_conges = tables_data['CAMSP']
|
|
for old_conge in old_conges:
|
|
worker = None
|
|
date_debut = _to_date(old_conge['date_debut'])
|
|
if date_debut > datetime(year=2012, month=1, day=1):
|
|
try:
|
|
worker = Worker.objects.get(old_camsp_id=old_conge['thera_id'])
|
|
if worker.enabled:
|
|
motif = int(old_conge['motif'])
|
|
if motif == 0:
|
|
motif = motif_unknown
|
|
else:
|
|
motif = HolidayType.objects.get(pk=motif)
|
|
end_time = _to_date(old_conge['date_fin'])
|
|
if not worker in conges:
|
|
conges[worker] = []
|
|
conges[worker].append((date_debut, end_time, motif))
|
|
except MultipleObjectsReturned:
|
|
print 'Plusieurs workers avec id : %s' % old_conge['thera_id']
|
|
print Worker.objects.filter(old_camsp_id=old_conge['thera_id'])
|
|
except ObjectDoesNotExist:
|
|
if not old_conge['thera_id'] in l_camsp:
|
|
l_camsp.append(old_conge['thera_id'])
|
|
|
|
sname = 'CMPP'
|
|
l_cmpp = []
|
|
l_cmpp_dbl = []
|
|
old_conges = tables_data['CMPP']
|
|
for old_conge in old_conges:
|
|
worker = None
|
|
date_debut = _to_date(old_conge['date_debut'])
|
|
if date_debut > datetime(year=2012, month=1, day=1):
|
|
try:
|
|
worker = Worker.objects.get(old_cmpp_id=old_conge['thera_id'])
|
|
if worker.enabled:
|
|
motif = int(old_conge['motif'])
|
|
if motif == 0:
|
|
motif = motif_unknown
|
|
else:
|
|
motif = HolidayType.objects.get(pk=motif)
|
|
end_time = _to_date(old_conge['date_fin'])
|
|
if not worker in conges:
|
|
conges[worker] = []
|
|
conges[worker].append((date_debut, end_time, motif))
|
|
except MultipleObjectsReturned:
|
|
print 'Plusieurs workers avec id : %s' % old_conge['thera_id']
|
|
print Worker.objects.filter(old_cmpp_id=old_conge['thera_id'])
|
|
except ObjectDoesNotExist:
|
|
if not old_conge['thera_id'] in l_cmpp:
|
|
l_cmpp.append(old_conge['thera_id'])
|
|
|
|
sname = 'SESSAD DYS'
|
|
l_dys = []
|
|
old_conges = tables_data['SESSAD DYS']
|
|
for old_conge in old_conges:
|
|
worker = None
|
|
date_debut = _to_date(old_conge['date_debut'])
|
|
if date_debut > datetime(year=2012, month=1, day=1):
|
|
try:
|
|
worker = Worker.objects.get(old_sessad_dys_id=old_conge['thera_id'])
|
|
if worker.enabled:
|
|
motif = int(old_conge['motif'])
|
|
if motif == 0:
|
|
motif = motif_unknown
|
|
else:
|
|
motif = HolidayType.objects.get(pk=motif)
|
|
end_time = _to_date(old_conge['date_fin'])
|
|
if not worker in conges:
|
|
conges[worker] = []
|
|
conges[worker].append((date_debut, end_time, motif))
|
|
except MultipleObjectsReturned:
|
|
print 'Plusieurs workers avec id : %s' % old_conge['thera_id']
|
|
print Worker.objects.filter(old_sessad_dys_id=old_conge['thera_id'])
|
|
except ObjectDoesNotExist:
|
|
if not old_conge['thera_id'] in l_dys:
|
|
l_dys.append(old_conge['thera_id'])
|
|
|
|
sname = 'SESSAD TED'
|
|
l_ted = []
|
|
old_conges = tables_data['SESSAD TED']
|
|
for old_conge in old_conges:
|
|
worker = None
|
|
date_debut = _to_date(old_conge['date_debut'])
|
|
if date_debut > datetime(year=2012, month=1, day=1):
|
|
try:
|
|
worker = Worker.objects.get(old_sessad_ted_id=old_conge['thera_id'])
|
|
if worker.enabled:
|
|
motif = int(old_conge['motif'])
|
|
if motif == 0:
|
|
motif = motif_unknown
|
|
else:
|
|
motif = HolidayType.objects.get(pk=motif)
|
|
end_time = _to_date(old_conge['date_fin'])
|
|
if not worker in conges:
|
|
conges[worker] = []
|
|
conges[worker].append((date_debut, end_time, motif))
|
|
except MultipleObjectsReturned:
|
|
print 'Plusieurs workers avec id : %s' % old_conge['thera_id']
|
|
print Worker.objects.filter(old_sessad_ted_id=old_conge['thera_id'])
|
|
except ObjectDoesNotExist:
|
|
if not old_conge['thera_id'] in l_ted:
|
|
l_ted.append(old_conge['thera_id'])
|
|
|
|
for worker, ccs in conges.items():
|
|
for date_debut, end_time, motif in set(ccs):
|
|
Holiday(worker=worker, holiday_type=motif, start_date=date_debut, end_date=end_time).save()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|