forked from public/pyceo
24 lines
752 B
Python
24 lines
752 B
Python
from typing import Union
|
|
|
|
import ldap
|
|
from zope import component
|
|
from zope.interface import implementer
|
|
|
|
from .UWLDAPRecord import UWLDAPRecord
|
|
from ceo_common.interfaces import IUWLDAPService, IConfig
|
|
|
|
|
|
@implementer(IUWLDAPService)
|
|
class UWLDAPService:
|
|
def __init__(self):
|
|
cfg = component.getUtility(IConfig)
|
|
self.uwldap_server_url = cfg.get('uwldap_server_url')
|
|
self.uwldap_base = cfg.get('uwldap_base')
|
|
|
|
def get(self, username: str) -> Union[UWLDAPRecord, None]:
|
|
conn = ldap.initialize(self.uwldap_server_url)
|
|
results = conn.search_s(self.uwldap_base, ldap.SCOPE_SUBTREE, f'uid={username}')
|
|
if not results:
|
|
return None
|
|
return UWLDAPRecord.deserialize_from_dict(results[0])
|