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