source: TI07-MOLES/trunk/PythonCode/wsgi/DiscoveryState.py @ 2340

from paste.request import parse_querystring,construct_url
class DiscoveryState:
    ''' This class holds the state associated with presenting multiple slices
    of a large result set '''
    def __init__(self,sessionID,environ,hits,offset=1,stride=10):
        ''' On instantiation, provide
                the backend sessionID
                the application URL that produced this query
                the stride through the result set '''
        self.environ=environ # the wsgi environment
        self.sessID=sessionID
        self.hits=hits
        self.offset=offset
        self.stride=stride
    def geturl(self,**kw):
        ''' Get a url from the wsgi environment, modified by the keyword arguments offset and stride 
        which are to be part of the querystring '''
        args=dict(parse_querystring(self.environ))
        offset,stride=kw.get('offset'),kw.get('stride')
        if offset is not None:args['start']=offset
        if stride is not None:args['howmany']=stride
        q=''
        for i in args: q+='%s=%s&'%(i,args[i])
        q=q[0:-1]
        url=construct_url(self.environ, with_query_string=True, with_path_info=True, querystring=q)
        return url
        
import unittest

class TestCase(unittest.TestCase):
    
    def testDiscoveryState(self):
        ''' Test creation of a discovery state variable '''
        DummyEnviron={'QUERY_STRING':'start=10&howmany=10','HTTP_HOST':'example.ndg',
                      'PATH_INFO':'/discovery','wsgi.url_scheme':'http','SERVER_PORT':'80'}
        d=DiscoveryState('123',DummyEnviron,12)
        self.assertEqual(d.geturl(offset='11',stride='20'),
                         'http://example.ndg/discovery?start=11&howmany=20')

 
        
if __name__=="__main__":
    unittest.main()