source: TI01-discovery/trunk/python/DiscoveryWS.py @ 932

#/usr/bin/env python
from ZSI import ServiceProxy
from ZSI.wstools.WSDLTools import WSDLReader
import sys
import os
import pdb

# This version known to work (in part) with ZSI 1.6 patched for NDG ...

DEBUG=1

class DiscoveryWS:
    
    ''' This is the class which provides an interface to the NDG web discovery
    web services '''
    
    def __init__(self,
        wsdl='http://superglue.badc.rl.ac.uk/exist/services/Discovery?WSDL',
        tracefile=None,
        user=None,
        passwd=None):
        ''' Establish connection to wsdl proxy at self.srv with self.sessID '''

        try:
            wr=WSDLReader()
            self.srv=ServiceProxy(wsdl,use_wsdl=True,tracefile=tracefile)
            if user is not None:
                conResp=self.srv.connect(in0=user,in1=passwd)
            else:
                conResp=self.srv.connect()
            self.sessID=conResp['connectReturn']
            if DEBUG: print 'Connection Made to Session: ',self.sessID
        except Exception, e:
            raise Exception(str(e)+os.linesep)


    def SearchFullText(self,query):
        ''' Produce a result set from a full text query on the NDG database '''
        result = self.srv.fullText(in0=self.sessID, in1=query)
        self.result=result['fullTextReturn']
        if DEBUG: self.__performance()
        
        
    def __performance(self):
        print 'QueryTime='+str(self.result['queryTime'])
        print 'Hits='+str(self.result['hits'])

        
    def GetResults(self,
            number=1, # number of records to return
            offset=1, # offset index within result set (begins at 1)
            highlight=u'none',  # string to highlight in result set
            xinclude=1,
            indent=0):
        ''' Return <number> of records from the result set beginning
        at <offset> '''

        sets=self.srv.retrieve(in0=self.sessID,
                              in1=offset,
                              in2=number,
                              in3=indent,
                              in4=xinclude,
                              in5=highlight)

        return sets['retrieveReturn']   
        
    def release(self):
        self.srv.disconnect(in0=self.sessID)

if __name__ == '__main__':

    ws=DiscoveryWS()
    ws.SearchFullText('acsoe')
    results=ws.GetResults(number=2)
    ws.release()
    f=file('output.html','w')
    for r in results:f.write(r+'\n')
    f.close()