[CalendarServer-changes] [12112] twext/trunk/twext/who
source_changes at macosforge.org
source_changes at macosforge.org
Wed Mar 12 11:17:35 PDT 2014
Revision: 12112
http://trac.calendarserver.org//changeset/12112
Author: wsanchez at apple.com
Date: 2013-12-16 14:37:53 -0800 (Mon, 16 Dec 2013)
Log Message:
-----------
Add (incomplete) OpenDirectory implementation.
Modified Paths:
--------------
twext/trunk/twext/who/expression.py
twext/trunk/twext/who/idirectory.py
twext/trunk/twext/who/opendirectory/__init__.py
twext/trunk/twext/who/opendirectory/dsquery.py
Added Paths:
-----------
twext/trunk/twext/who/opendirectory/
twext/trunk/twext/who/opendirectory/service.py
Modified: twext/trunk/twext/who/expression.py
===================================================================
--- twext/trunk/twext/who/expression.py 2013-12-16 17:12:05 UTC (rev 12111)
+++ twext/trunk/twext/who/expression.py 2013-12-16 22:37:53 UTC (rev 12112)
@@ -78,6 +78,9 @@
startsWith = NamedConstant()
startsWith.description = u"starts with"
+ endsWith = NamedConstant()
+ startsWith.description = u"ends with"
+
contains = NamedConstant()
contains.description = u"contains"
@@ -184,6 +187,10 @@
fieldName, fieldValue,
matchType=MatchType.equals, flags=None
):
+ if flags is None:
+ # Gross way to get a FlagsConstant with no flags in it:
+ flags = MatchFlags.NOT & MatchFlags.caseInsensitive
+
self.fieldName = fieldName
self.fieldValue = fieldValue
self.matchType = matchType
Modified: twext/trunk/twext/who/idirectory.py
===================================================================
--- twext/trunk/twext/who/idirectory.py 2013-12-16 17:12:05 UTC (rev 12111)
+++ twext/trunk/twext/who/idirectory.py 2013-12-16 22:37:53 UTC (rev 12112)
@@ -107,10 +107,10 @@
"""
Constants for common directory record types.
"""
- user = NamedConstant()
+ user = NamedConstant()
+ user.description = u"user"
+
group = NamedConstant()
-
- user.description = u"user"
group.description = u"group"
Modified: twext/trunk/twext/who/opendirectory/__init__.py
===================================================================
--- CalendarServer/trunk/calendarserver/platform/darwin/od/__init__.py 2013-12-16 17:12:05 UTC (rev 12111)
+++ twext/trunk/twext/who/opendirectory/__init__.py 2013-12-16 22:37:53 UTC (rev 12112)
@@ -13,3 +13,11 @@
# See the License for the specific language governing permissions and
# limitations under the License.
##
+
+__all__ = [
+ "DirectoryService",
+ "DirectoryRecord",
+]
+
+
+from .service import DirectoryService, DirectoryRecord
Modified: twext/trunk/twext/who/opendirectory/dsquery.py
===================================================================
--- CalendarServer/trunk/calendarserver/platform/darwin/od/dsquery.py 2013-12-16 17:12:05 UTC (rev 12111)
+++ twext/trunk/twext/who/opendirectory/dsquery.py 2013-12-16 22:37:53 UTC (rev 12112)
@@ -25,6 +25,8 @@
import dsattributes
+
+
class match(object):
"""
Represents and attribute/value match operation.
Added: twext/trunk/twext/who/opendirectory/service.py
===================================================================
--- twext/trunk/twext/who/opendirectory/service.py (rev 0)
+++ twext/trunk/twext/who/opendirectory/service.py 2013-12-16 22:37:53 UTC (rev 12112)
@@ -0,0 +1,418 @@
+# -*- test-case-name: twext.who.test.test_util -*-
+##
+# Copyright (c) 2013 Apple Inc. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+##
+
+from __future__ import print_function
+
+"""
+OpenDirectory directory service implementation.
+"""
+
+__all__ = [
+ "OpenDirectoryError",
+ "DirectoryService",
+ "DirectoryRecord",
+]
+
+from itertools import chain
+
+from twext.python.log import Logger
+from twisted.python.constants import Names, NamedConstant
+from twisted.python.constants import Values, ValueConstant
+
+from twext.who.idirectory import (
+ DirectoryServiceError, QueryNotSupportedError,
+ FieldName as BaseFieldName,
+ RecordType as BaseRecordType,
+)
+from twext.who.directory import (
+ DirectoryService as BaseDirectoryService,
+ DirectoryRecord as BaseDirectoryRecord,
+)
+from twext.who.expression import CompoundExpression, Operand
+from twext.who.expression import MatchExpression, MatchType, MatchFlags
+from twext.who.util import iterFlags, ConstantsContainer
+
+import dsattributes
+from opendirectory import (
+ ODError, odInit,
+ getNodeAttributes,
+ queryRecordsWithAttribute_list,
+)
+
+
+
+#
+# Exceptions
+#
+
+class OpenDirectoryError(DirectoryServiceError):
+ """
+ OpenDirectory error.
+ """
+
+
+
+#
+# Constants
+#
+
+class FieldName(Names):
+ metaNodeLocation = NamedConstant()
+ metaNodeLocation.description = "source OD node"
+ metaNodeLocation.multiValue = False
+
+
+#
+# Constants
+#
+
+class ODRecordType(Values):
+ user = ValueConstant(dsattributes.kDSStdRecordTypeUsers)
+ user.recordType = BaseRecordType.user
+
+ group = ValueConstant(dsattributes.kDSStdRecordTypeGroups)
+ group.recordType = BaseRecordType.group
+
+
+ @classmethod
+ def recordTypeForRecordType(cls, recordType):
+ if not hasattr(cls, "_recordTypeByRecordType"):
+ cls._recordTypeByRecordType = dict((
+ (recordType.recordType, recordType)
+ for recordType in cls.iterconstants()
+ ))
+
+ return cls._recordTypeByRecordType.get(recordType, None)
+
+
+
+class ODAttribute(Values):
+ recordType = ValueConstant(dsattributes.kDSNAttrRecordType)
+ recordType.fieldName = BaseFieldName.recordType
+
+ uid = ValueConstant(dsattributes.kDS1AttrGeneratedUID)
+ uid.fieldName = BaseFieldName.uid
+
+ guid = ValueConstant(dsattributes.kDS1AttrGeneratedUID)
+ guid.fieldName = BaseFieldName.guid
+
+ shortName = ValueConstant(dsattributes.kDSNAttrRecordName)
+ shortName.fieldName = BaseFieldName.shortNames
+
+ fullName = ValueConstant(dsattributes.kDS1AttrDistinguishedName)
+ fullName.fieldName = BaseFieldName.fullNames
+
+ emailAddress = ValueConstant(dsattributes.kDSNAttrEMailAddress)
+ emailAddress.fieldName = BaseFieldName.emailAddresses
+
+ metaNodeLocation = ValueConstant(dsattributes.kDSNAttrMetaNodeLocation)
+ metaNodeLocation.fieldName = FieldName.metaNodeLocation
+
+
+ @classmethod
+ def attributeForFieldName(cls, fieldName):
+ if not hasattr(cls, "_attributesByFieldName"):
+ cls._attributesByFieldName = dict((
+ (attribute.fieldName, attribute)
+ for attribute in cls.iterconstants()
+ if hasattr(attribute, "fieldName")
+ ))
+
+ return cls._attributesByFieldName.get(fieldName, None)
+
+
+
+class ODMatchType(Values):
+ equals = ValueConstant(dsattributes.eDSExact)
+ equals.matchType = MatchType.equals
+
+ startsWith = ValueConstant(dsattributes.eDSStartsWith)
+ startsWith.matchType = MatchType.startsWith
+
+ endsWith = ValueConstant(dsattributes.eDSEndsWith)
+ endsWith.matchType = MatchType.endsWith
+
+ contains = ValueConstant(dsattributes.eDSContains)
+ contains.matchType = MatchType.contains
+
+
+ @classmethod
+ def matchTypeForMatchType(cls, matchType):
+ if not hasattr(cls, "_matchTypeByMatchType"):
+ cls._matchTypeByMatchType = dict((
+ (matchType.matchType, matchType)
+ for matchType in cls.iterconstants()
+ ))
+
+ return cls._matchTypeByMatchType.get(matchType, None)
+
+
+
+#
+# Directory Service
+#
+
+class DirectoryService(BaseDirectoryService):
+ """
+ OpenDirectory directory service.
+ """
+ log = Logger()
+
+ fieldName = ConstantsContainer(chain(
+ BaseDirectoryService.fieldName.iterconstants(),
+ FieldName.iterconstants()
+ ))
+
+
+ def __init__(self, nodeName="/Search"):
+ """
+ @param nodeName: the OpenDirectory node to query against.
+ @type nodeName: bytes
+ """
+ self._nodeName = nodeName
+
+
+ @property
+ def nodeName(self):
+ return self._nodeName
+
+
+ @property
+ def realmName(self):
+ return "OpenDirectory Node {self.nodeName!r}".format(self=self)
+
+
+ @property
+ def node(self):
+ """
+ Get the underlying (network) directory node.
+ """
+ if not hasattr(self, "_node"):
+ try:
+ self._node = odInit(self.nodeName)
+ except ODError, e:
+ self.log.error(
+ "OpenDirectory initialization error"
+ "(node={source.nodeName}): {error}",
+ error=e
+ )
+ raise OpenDirectoryError(e)
+
+ return self._node
+
+
+ @property
+ def localNode(self):
+ """
+ Get the local node from the search path (if any), so that we can handle
+ it specially.
+ """
+ if not hasattr(self, "_localNode"):
+ if self.nodeName == "/Search":
+ result = getNodeAttributes(
+ self.node, "/Search",
+ (dsattributes.kDS1AttrSearchPath,)
+ )
+ if (
+ "/Local/Default" in
+ result[dsattributes.kDS1AttrSearchPath]
+ ):
+ try:
+ self._localNode = odInit("/Local/Default")
+ except ODError, e:
+ self.log.error(
+ "Failed to open local node: {error}}",
+ error=e,
+ )
+ raise OpenDirectoryError(e)
+ else:
+ self._localNode = None
+
+ elif self.nodeName == "/Local/Default":
+ self._localNode = self.node
+
+ else:
+ self._localNode = None
+
+ return self._localNode
+
+
+ def recordsFromMatchExpression(self, expression):
+ if not isinstance(expression, MatchExpression):
+ raise TypeError(expression)
+
+ matchType = ODMatchType.matchTypeForMatchType(expression.matchType)
+ if matchType is None:
+ raise QueryNotSupportedError(
+ "Unknown match type: {0}".format(matchType)
+ )
+
+ caseInsensitive = (
+ MatchFlags.caseInsensitive in iterFlags(expression.flags)
+ )
+
+ if expression.fieldName is self.fieldName.recordType:
+ raise NotImplementedError()
+ else:
+ results = queryRecordsWithAttribute_list(
+ self.node,
+ ODAttribute.attributeForFieldName(expression.fieldName).value,
+ expression.fieldValue.encode("utf-8"),
+ matchType.value,
+ caseInsensitive,
+ [
+ recordType.value
+ for recordType in ODRecordType.iterconstants()
+ ],
+ [attr.value for attr in ODAttribute.iterconstants()],
+ )
+
+ # def uniqueTupleFromAttribute(self, attribute):
+ # if attribute:
+ # if isinstance(attribute, bytes):
+ # return (attribute,)
+ # else:
+ # s = set()
+ # return tuple((
+ # (s.add(x), x)[1] for x in attribute if x not in s
+ # ))
+ # else:
+ # return ()
+
+ for (shortName, attributes) in results:
+ fields = {}
+
+ for (key, value) in attributes.iteritems():
+ if key == "dsAttrTypeStandard:AppleMetaRecordName":
+ # We get this even though we did not ask for it...
+ continue
+
+ try:
+ attribute = ODAttribute.lookupByValue(key)
+ except ValueError:
+ self.log.debug(
+ "Got back unexpected attribute {attribute} "
+ "for record with short name {shortName}",
+ attribute=key, shortName=shortName
+ )
+ continue
+ fieldName = attribute.fieldName
+
+ try:
+ if BaseFieldName.isMultiValue(fieldName):
+ if type(value) is bytes:
+ value = (value,)
+ elif type(value) is not list:
+ raise TypeError()
+
+ fields[fieldName] = tuple(
+ x.decode("utf-8") for x in value
+ )
+
+ else:
+ if type(value) is list:
+ assert len(value) == 1
+ value = value[0]
+ elif type(value) is not bytes:
+ raise TypeError()
+
+ if fieldName is self.fieldName.recordType:
+ fields[fieldName] = ODRecordType.lookupByValue(
+ value
+ ).recordType
+ else:
+ fields[fieldName] = value.decode("utf-8")
+
+ except TypeError:
+ raise AssertionError(
+ "Unexpected type {0} for attribute {1}"
+ .format(type(value), fieldName)
+ )
+
+
+ yield DirectoryRecord(self, fields)
+
+
+ def recordsFromExpression(self, expression):
+ """
+ This implementation can handle L{MatchExpression} expressions; other
+ expressions are passed up to the superclass.
+ """
+ if isinstance(expression, CompoundExpression):
+ raise NotImplementedError(Operand)
+
+ elif isinstance(expression, MatchExpression):
+ try:
+ return self.recordsFromMatchExpression(expression)
+ except QueryNotSupportedError:
+ return BaseDirectoryService.recordsFromExpression(
+ self, expression
+ )
+
+ else:
+ return BaseDirectoryService.recordsFromExpression(
+ self, expression
+ )
+
+
+
+
+class DirectoryRecord(BaseDirectoryRecord):
+ """
+ OpenDirectory directory record.
+ """
+
+ def __init__(self, service, fields):
+ # Make sure that uid and guid are both set and equal
+ uid = fields.get(service.fieldName.uid, None)
+ guid = fields.get(service.fieldName.guid, None)
+
+ if uid is not None and guid is not None:
+ if uid != guid:
+ raise ValueError(
+ "uid and guid must be equal ({uid} != {guid})"
+ .format(uid=uid, guid=guid)
+ )
+ elif uid is None:
+ fields[service.fieldName.uid] = guid
+ elif guid is None:
+ fields[service.fieldName.guid] = uid
+
+ super(DirectoryRecord, self).__init__(service, fields)
+
+
+ requiredFields = BaseDirectoryRecord.requiredFields + (BaseFieldName.guid,)
+
+
+
+
+if __name__ == "__main__":
+ service = DirectoryService()
+ print(
+ "Service = {service}\n"
+ "Node = {service.node}\n"
+ "Local node = {service.localNode}\n"
+ .format(service=service)
+ )
+
+ matchMorgen = MatchExpression(
+ service.fieldName.shortNames, u"sagen",
+ matchType=MatchType.equals,
+ )
+ for record in service.recordsFromExpression(matchMorgen):
+ print("*" * 80)
+ print(record.description())
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://lists.macosforge.org/pipermail/calendarserver-changes/attachments/20140312/ce307aad/attachment.html>
More information about the calendarserver-changes
mailing list