1# Copyright (C) 2010, 2011 Apple Inc. All rights reserved. 2# 3# Redistribution and use in source and binary forms, with or without 4# modification, are permitted provided that the following conditions 5# are met: 6# 1. Redistributions of source code must retain the above copyright 7# notice, this list of conditions and the following disclaimer. 8# 2. Redistributions in binary form must reproduce the above copyright 9# notice, this list of conditions and the following disclaimer in the 10# documentation and/or other materials provided with the distribution. 11# 12# THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND 13# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 14# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 15# DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR 16# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 17# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 18# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 19# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 20# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 21# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 22 23import itertools 24 25 26class MessageReceiver(object): 27 def __init__(self, name, superclass, attributes, messages, condition): 28 self.name = name 29 self.superclass = superclass 30 self.attributes = frozenset(attributes or []) 31 self.messages = messages 32 self.condition = condition 33 34 def iterparameters(self): 35 return itertools.chain((parameter for message in self.messages for parameter in message.parameters), 36 (reply_parameter for message in self.messages if message.reply_parameters for reply_parameter in message.reply_parameters)) 37 38 def has_attribute(self, attribute): 39 return attribute in self.attributes 40 41class Message(object): 42 def __init__(self, name, parameters, reply_parameters, attributes, condition): 43 self.name = name 44 self.parameters = parameters 45 self.reply_parameters = reply_parameters 46 self.attributes = frozenset(attributes or []) 47 self.condition = condition 48 49 def has_attribute(self, attribute): 50 return attribute in self.attributes 51 52 53class Parameter(object): 54 def __init__(self, type, name, attributes=None, condition=None): 55 self.type = type 56 self.name = name 57 self.attributes = frozenset(attributes or []) 58 self.condition = condition 59 60 def has_attribute(self, attribute): 61 return attribute in self.attributes 62