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