from simple_sso.sso_server.server import Server
from authentification.models import Employe

class CerfigSSOServer(Server):
    def get_user_data(self, user, consumer, extra_data=None):
        user_data = super().get_user_data(user, consumer, extra_data)  
        employe = Employe.objects.get(user=user)   
        
        user_data.update({
            'departement': employe.departement.nom,
            'chef_departement': employe.chef,
            'groups': [group.name for group in user.groups.all()],
        })

        return user_data