source: TI07-MOLES/trunk/StubB/XSLT/browse/portal/cgi/browse/Utilities.py @ 1173

from collections import deque # python 2.4
import ElementTree as ET
import ConfigParser
import os
import re

class myConfig:
        ''' Handle missing sections and variables in a config file a bit gracefully '''
        def __init__(self,configfile):
                self.config=ConfigParser.ConfigParser()
                if not os.path.exists(configfile):
                        raise 'No config file: %s'%configfile
                self.config.read(configfile)
                logfile=self.get('logging','debugLog',None)
                self.logfile=None
                if logfile is not None:
                        self.logfile=open(logfile,'a')

        def get(self,section,key,default=None):
                try:
                        return self.config.get(section,key)
                except:
                        return default
        
        def log(self,string):
            ''' Log some debug information '''
            if self.logfile is None: return
            if string is not None:
                self.logfile.write('%s\n'%string)
            else:
                self.logfile.write('empty log entry\n')
                
class Response:
        ''' Holds all the parts of the response to an HTML get or post '''
        def __init__(self,content='text/html'):
                self.headers=[]
                self.cookie=None
                self.contentType=content
                self.content=''
        def write(self,out):
                for name,value in self.headers:
                        out.write("%s: %s\n"( name,value))
                if self.cookie:
                        out.write("%s\n"%self.cookie)
                out.write("Content-type: %s\n"%self.contentType)
                out.write("\n")
                out.write(self.content)

class RingBuffer(deque):
    #deque is a python 2.4 class!
    #credit http://www.onlamp.com/pub/a/python/excerpt/pythonckbk_chap1/index1.html
    def __init__(self, size_max):
        deque.__init__(self)
        self.size_max = size_max
    def append(self, datum):
        deque.append(self, datum)
        if len(self) > self.size_max:
            self.popleft( )
    def tolist(self):
        return list(self)

def wrapGetText(element,xpathExpression,multiple=0):
        ''' Wraps a call to ET to get a text object in an error handler '''
        if multiple:
                r=element.findall(xpathExpression)
        else:
                r=[element.find(xpathExpression),]
        try:
                rr=[i.text for i in r]
        except:
                rr=['',]
        if multiple: 
                return rr
        else: return rr[0] 

def getURLdict(cgiFieldStorage):
    ''' takes a cgi field storage object and converts it to a dictionary '''
    result={}
    for item in cgiFieldStorage:
        result[item]=cgiFieldStorage[item].value
    return result
##
### convert the followign two methods into one class that can handle
## xml directly too if necessary
##
def DIFid2NDGid(string):
    ''' takes a dif thing parses it and produces an ET ndg element id ...
    and use this in dif ... '''
    s=string.split(':')
    try:
        r='''<DIFid><schemeIdentifier>%s</schemeIdentifier>
         <repositoryIdentifier>%s</repositoryIdentifier>
         <localIdentifier>%s</localIdentifier></DIFid>'''%(s[1],s[0],s[2])
        return ET.fromstring(r)
    except:
        r='''<DIFid><schemeIdentifier>DIF</schemeIdentifier>
        <repositoryIdentifier>Unknown</repositoryIdentifier>
        <localIdentifier>%s</localIdentifier></DIFid>'''%string
        return ET.fromstring(r)
def idconvert(e):
    ''' Converts an XML ndg identifier into a uri version '''
    s='%s:%s:%s'%(wrapGetText(e,'repositoryIdentifier'),
                    wrapGetText(e,'schemeIdentifier'),
                    wrapGetText(e,'localIdentifier'))
    return s

if __name__=="__main__":
    x='''<test attr='something'>some content with <a href="target"> display</a> url</test>'''
    import ElementTree as ET
    y=ET.fromstring(x)
    print PrettyPrintHTML(y,['content',])
    print PrettyPrint(y)