This repository has been archived on 2023-02-21. You can view files and clone it, but cannot push or open issues or pull requests.
PcfAnnuaire/PcfAnnuaire.py

76 lines
2.4 KiB
Python

""" PcfAnnuaire Class """
import os
import time
import calendar
import ldap
from interfaces import IPcfAnnuaire
from zope.interface import implements
from Products.Archetypes.public import *
from Products.CMFCore.utils import getToolByName
from Acquisition import aq_base
schema = BaseSchema
class PcfAnnuaireEntry:
def __init__(self, ldap_entry):
def get_attr(x):
try:
return ldap_entry.get(x)[0]
except (IndexError, TypeError):
return None
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.building = get_attr('location')
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')
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 = luf.users_base
return users_base
def getEntries(self):
ldap_conn = self.get_ldap_conn()
result = ldap_conn.search_s(self.get_users_base(),
ldap.SCOPE_SUBTREE, 'objectClass=person')
return [PcfAnnuaireEntry(x[1]) for x in result]
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)
registerType(PcfAnnuaire)