PcfAnnuaire/PcfAnnuaire.py

# -*- coding: UTF-8 -*-

""" PcfAnnuaire Class """

import os
import time
import calendar
import ldap
import urllib

from interfaces import IPcfAnnuaire
from zope.interface import implements
from Products.Archetypes.public import *
from Products.CMFCore.utils import getToolByName
from plone.memoize.volatile import cache, DontCache
from Acquisition import aq_base

schema = BaseSchema

def get_sn(x):
    return getattr(x, 'sn')

class PcfAnnuaireEntry:
    def __init__(self, ldap_dn, ldap_entry):
        def get_attr(x):
            try:
                return ldap_entry.get(x)[0].strip()
            except (IndexError, TypeError):
                return None
        self._debug_entry = ldap_entry
        self.dn = ldap_dn
        self.cn = get_attr('cn')
        self.sn = get_attr('sn')
        self.telephoneNumbers = ldap_entry.get('telephoneNumber')
        if self.telephoneNumbers:
            self.shortTelephoneNumber = self.telephoneNumbers[0][-3:]
        else:
            self.shortTelephoneNumber = None
        self.givenName = get_attr('givenName')
        self.mobile = get_attr('mobile')
        self.service = get_attr('department')
        self.building = get_attr('destinationIndicator')
        self.floor = get_attr('extensionAttribute1')
        self.office = get_attr('physicalDeliveryOfficeName')
        self.fax = get_attr('facsimileTelephoneNumber')
        self.email = get_attr('mail')
        self.division = get_attr('division')
        self.title = get_attr('title')
        self.company = get_attr('company')
        self.manager = get_attr('manager')
        if self.manager:
            self.manager_url = urllib.quote(self.manager)
            self.manager = self.manager.split('=')[1].split(',')[0]

    def __str__(self):
        return urllib.quote(self.dn)

class PcfAnnuaireService:
    pass

def cache_key(method, self):
    # append time, so old entries do not get returned forever; (they will still
    # get removed as stale entries are automatically removed after 3 days)
    # (cf plone/memoize/volatile.py)
    return '%s_%s' % (hash(self), (int(time.time()) / 300))

class PcfAnnuaire(BaseContent):

    implements(IPcfAnnuaire)

    schema = schema
    _at_rename_after_creation = True

    def get_ldap_conn(self):
        portal = getToolByName(self, 'portal_url').getPortalObject()
        ldapdir = aq_base(portal.acl_users.ldap)
        luf = ldapdir._getLDAPUserFolder()
        server = luf._delegate.getServers()[0]
        ldap_conn = ldap.initialize('%(protocol)s://%(host)s:%(port)s' % server)
        ldap_conn.simple_bind_s(luf._binduid, luf._bindpwd)
        return ldap_conn

    def get_users_base(self):
        portal = getToolByName(self, 'portal_url').getPortalObject()
        ldapdir = aq_base(portal.acl_users.ldap)
        luf = ldapdir._getLDAPUserFolder()
        users_base = 'OU=ouPCF,' + luf.users_base
        return users_base

    def getObjectEntries(self):
        ldap_conn = self.get_ldap_conn()
        result = ldap_conn.search_s(self.get_users_base(),
                        ldap.SCOPE_SUBTREE, 'objectClass=person')
        return [PcfAnnuaireEntry(*x) for x in result]

    @cache(cache_key)
    def getEntries(self):
        return [x for x in self.getObjectEntries() if \
                x.givenName and x.company == 'Parlement de la Communauté française de Belgique']

    def getServices(self):
        entries = self.getObjectEntries()
        services = {}
        for entry in entries:
            if not entry.division:
                continue
            if not entry.division in services:
                service = Service()
                service.name = entry.division
                service.agents = []
                services[division] = service
            service = services.get(entry.division)
            service.agents.append(entry)
        def get_name(x): return getattr(x, 'name')
        def get_sn(x): return getattr(x, 'sn')
        services = sorted([x for x in services.values()], key=get_name)
        for service in services:
            services.agents.sort(key=get_sn)
        return services

    def getEntriesWithPhone(self):
        def get_sn(x): return getattr(x, 'sn')
        return sorted([x for x in self.getEntries() if x.telephoneNumbers], key=get_sn)

    def getSecretaireGeneral(self):
        try:
            return [x for x in self.getEntries() if x.title == 'Secrétaire Général'][0]
        except IndexError:
            raise KeyError

    def getServices(self):
        services = {}
        for entry in self.getEntries():
            services[entry.service] = True
        services = sorted(services.keys())
        return services

    def getServiceManager(self, service):
        sec_gen = self.getSecretaireGeneral()
        if service == 'Service du Greffier':
            return sec_gen
        try:
            return [x for x in self.getEntries() if x.service == service and x.manager == sec_gen.dn][0]
        except IndexError:
            return None

    def getServiceNonManagers(self, service):
        service_manager = self.getServiceManager(service)
        entries = [x for x in self.getEntries() if \
                   x.service == service and x.manager == service_manager.dn and x.division in (None, ' ')]
        return sorted(entries, key=get_sn)

    def getCellules(self, service):
        service_manager = self.getServiceManager(service)
        entries = [x for x in self.getEntries() if x.manager == service_manager.dn]
        cells = {}
        for entry in entries:
            if entry.division and entry.division != ' ':
                cells[entry.division] = True
        cells = sorted(cells.keys())
        return cells

    def getCelluleManagers(self, service, cellule):
        service_manager = self.getServiceManager(service)
        entries = [x for x in self.getEntries() if \
                   x.service == service and x.division == cellule and x.manager == service_manager.dn]
        return sorted(entries, key=get_sn)

    def getCellulePeople(self, service, cellule):
        cellule_managers = [x.dn for x in self.getCelluleManagers(service, cellule)]
        entries = [x for x in self.getEntries() if \
                   x.service == service and x.division == cellule and x.dn not in cellule_managers]
        return sorted(entries, key=get_sn)

    def getPerson(self, dn):
        person = [x for x in self.getEntries() if x.dn == dn][0]
        return person

registerType(PcfAnnuaire)