5637

[auf_rh_dae.git] / project / dae / management / commands / dae.py

# -*- encoding: utf-8 -*-

import sys
import codecs
from django.core.management.base import BaseCommand
from project.rh import models as rh
from project.dae import models as dae

class Command(BaseCommand):

    def handle(self, *args, **options):

        stdout = options.get('stdout', sys.stdout)
        self.stdout = codecs.getwriter('utf8')(stdout)

        if args[0] == "liste_poste":
            for poste in dae.Poste.objects.filter(id_rh__isnull=False):
                self.stdout.write(u"%s: %s %s [%s]\n" % (poste.id, poste.nom,
                        poste.implantation, poste.etat))

        if args[0] == "importer_poste":
            poste = dae.Poste.objects.get(id=args[1])
            poste.importer_dans_rh()
            self.stdout.write(u"DAE:%s => RH:%s\n" % (poste.id, poste.id_rh.id))