util.py revision 12563:8d59ed22ae79 
1# Copyright (c) 2009 The Hewlett-Packard Development Company
2# All rights reserved.
3#
4# Redistribution and use in source and binary forms, with or without
5# modification, are permitted provided that the following conditions are
6# met: redistributions of source code must retain the above copyright
7# notice, this list of conditions and the following disclaimer;
8# 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# neither the name of the copyright holders nor the names of its
12# contributors may be used to endorse or promote products derived from
13# this software without specific prior written permission.
14#
15# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
19# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
21# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26
27from __future__ import print_function
28
29import os
30import sys
31
32class PairContainer(object):
33    def __init__(self, pairs=None):
34        self.pairs = {}
35        if pairs:
36            self.pairs.update(pairs)
37
38    def __contains__(self, item):
39        return item in self.pairs
40
41    def __getitem__(self, item):
42        return self.pairs[item]
43
44    def __setitem__(self, item, value):
45        self.pairs[item] = value
46
47    def get(self, item, failobj=None):
48        return self.pairs.get(item, failobj)
49
50class Location(object):
51    def __init__(self, filename, lineno, no_warning=False):
52        if not isinstance(filename, basestring):
53            raise AttributeError, \
54                "filename must be a string, found '%s'" % (type(filename), )
55        if not isinstance(lineno, (int, long)):
56            raise AttributeError, \
57                "filename must be an integer, found '%s'" % (type(lineno), )
58        self.filename = filename
59        self.lineno = lineno
60        self.no_warning = no_warning
61
62    def __str__(self):
63        return '%s:%d' % (os.path.basename(self.filename), self.lineno)
64
65    def warning(self, message, *args):
66        if self.no_warning:
67            return
68        if args:
69            message = message % args
70        #raise Exception, "%s: Warning: %s" % (self, message)
71        print("%s: Warning: %s" % (self, message), file=sys.stderr)
72
73    def error(self, message, *args):
74        if args:
75            message = message % args
76        raise Exception, "%s: Error: %s" % (self, message)
77        sys.exit("\n%s: Error: %s" % (self, message))
78
79__all__ = [ 'PairContainer', 'Location' ]
80