OwlCyberSecurity - MANAGER

Edit File: xmlaugmentsparser.py

##
# Copyright (c) 2009-2017 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.
##

"""
XML based augment configuration file handling.
"""

__all__ = [
    "XMLAugmentsParser",
]

import types

from twext.python.log import Logger

from twistedcaldav.xmlutil import readXML

log = Logger()

ELEMENT_AUGMENTS = "augments"
ELEMENT_RECORD = "record"

ELEMENT_UID = "uid"
ELEMENT_ENABLE = "enable"
ELEMENT_SERVERID = "server-id"
ELEMENT_PARTITIONID = "partition-id"   # Backwards compatibility
ELEMENT_HOSTEDAT = "hosted-at"   # Backwards compatibility
ELEMENT_ENABLECALENDAR = "enable-calendar"
ELEMENT_ENABLEADDRESSBOOK = "enable-addressbook"
ELEMENT_ENABLELOGIN = "enable-login"
ELEMENT_AUTOSCHEDULE = "auto-schedule"
ELEMENT_AUTOSCHEDULE_MODE = "auto-schedule-mode"
ELEMENT_AUTOACCEPTGROUP = "auto-accept-group"

ATTRIBUTE_REPEAT = "repeat"

VALUE_TRUE = "true"
VALUE_FALSE = "false"

ELEMENT_AUGMENTRECORD_MAP = {
    ELEMENT_UID: "uid",
    ELEMENT_ENABLE: "enabled",
    ELEMENT_SERVERID: "serverID",
    ELEMENT_ENABLECALENDAR: "enabledForCalendaring",
    ELEMENT_ENABLEADDRESSBOOK: "enabledForAddressBooks",
    ELEMENT_ENABLELOGIN: "enabledForLogin",
    ELEMENT_AUTOSCHEDULE: "autoSchedule",
    ELEMENT_AUTOSCHEDULE_MODE: "autoScheduleMode",
    ELEMENT_AUTOACCEPTGROUP: "autoAcceptGroup",
}

class XMLAugmentsParser(object):
    """
    XML augments configuration file parser.
    """

def __repr__(self):
        return "<%s %r>" % (self.__class__.__name__, self.xmlFile)

def __init__(self, xmlFile, items):

self.items = items
        self.xmlFile = xmlFile

# Read in XML
        try:
            _ignore_tree, augments_node = readXML(self.xmlFile, ELEMENT_AUGMENTS)
        except ValueError, e:
            raise RuntimeError("XML parse error for '%s' because: %s" % (self.xmlFile, e,))

self._parseXML(augments_node)

def _parseXML(self, rootnode):
        """
        Parse the XML root node from the augments configuration document.
        @param rootnode: the L{Element} to parse.
        """
        for child in rootnode:

if child.tag != ELEMENT_RECORD:
                raise RuntimeError("Unknown augment type: '%s' in augment file: '%s'" % (child.tag, self.xmlFile,))

repeat = int(child.get(ATTRIBUTE_REPEAT, "1"))

fields = {}
            for node in child:

if node.tag in (
                    ELEMENT_UID,
                    ELEMENT_SERVERID,
                    ELEMENT_PARTITIONID,
                    ELEMENT_HOSTEDAT,
                    ELEMENT_AUTOSCHEDULE_MODE,
                    ELEMENT_AUTOACCEPTGROUP,
                ):
                    fields[node.tag] = node.text if node.text else ""
                elif node.tag in (
                    ELEMENT_ENABLE,
                    ELEMENT_ENABLECALENDAR,
                    ELEMENT_ENABLEADDRESSBOOK,
                    ELEMENT_ENABLELOGIN,
                    ELEMENT_AUTOSCHEDULE,
                ):
                    fields[node.tag] = node.text == VALUE_TRUE
                else:
                    raise RuntimeError("Invalid element '%s' in augment file: '%s'" % (node.tag, self.xmlFile,))

# Must have at least a uid
            if ELEMENT_UID not in fields:
                raise RuntimeError("Invalid record '%s' without a uid in augment file: '%s'" % (child, self.xmlFile,))

if repeat > 1:
                for i in xrange(1, repeat + 1):
                    self.buildRecord(fields, i)
            else:
                self.buildRecord(fields)

def buildRecord(self, fields, count=None):

from twistedcaldav.directory.augment import AugmentRecord

def expandCount(value, count):

if type(value) in types.StringTypes:
                return value % (count,) if count and "%" in value else value
            elif type(value) == set:
                return set([item % (count,) if count and "%" in item else item for item in value])
            else:
                return value

actualFields = {}
        for k, v in fields.iteritems():
            if k in ELEMENT_AUGMENTRECORD_MAP:
                actualFields[ELEMENT_AUGMENTRECORD_MAP[k]] = expandCount(v, count)

record = AugmentRecord(**actualFields)
        self.items[record.uid] = record