util.py revision 6657:ef5fae93a3b2 
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
27import os
28import sys
29
30def makeDir(path):
31    if os.path.exists(path):
32        if not os.path.isdir(path):
33            raise AttributeError, "%s exists but is not directory" % path
34    else:
35        os.mkdir(path)
36
37class PairContainer(object):
38    def __init__(self, pairs=None):
39        self.pairs = {}
40        if pairs:
41            self.pairs.update(pairs)
42
43    def __contains__(self, item):
44        return item in self.pairs
45
46    def __getitem__(self, item):
47        return self.pairs[item]
48
49    def __setitem__(self, item, value):
50        self.pairs[item] = value
51
52    def get(self, item, failobj=None):
53        return self.pairs.get(item, failobj)
54
55class Location(object):
56    def __init__(self, filename, lineno):
57        self.filename = filename
58        self.lineno = lineno
59
60    def __str__(self):
61        return '%s:%d' % (os.path.basename(self.filename), self.lineno)
62
63    def warning(self, message, *args):
64        if args:
65            message = message % args
66        #raise Exception, "%s: Warning: %s" % (self, message)
67        print >>sys.stderr, "%s: Warning: %s" % (self, message)
68
69    def error(self, message, *args):
70        if args:
71            message = message % args
72        raise Exception, "%s: Error: %s" % (self, message)
73        sys.exit("\n%s: Error: %s" % (self, message))
74
75__all__ = [ 'makeDir', 'PairContainer', 'Location' ]
76