source: TI01-discovery/trunk/OAIInfoEditor/oai_info_editor/controllers/harvest.py @ 5253

'''
 Class representing pylons controller for the harvesting of provider info
  
 @author: C Byrom, Tessella Apr 2009
'''
import logging, urllib
from oai_info_editor.lib.base import *
from oai_info_editor.controllers.master import MasterController
import oai_info_editor.lib.constants as const

class HarvestController(MasterController):

    def harvestProviderInfo(self, providerName, repositoryName):
        '''
        Harvest the repository info associated with the input name and provider
        @param providerName: name of provider whose info should be displayed
        @param repositoryName: name of repository whose info should be displayed
        '''
        providerName = urllib.unquote(providerName)
        repositoryName = urllib.unquote(repositoryName)
        self._setUpController(providerName = providerName)
        logging.info("Setting up page to display info for provider, '%s'" %providerName)

        pi = g.dao.getProviderInfo(providerName)
        ri = g.dao.getRepositoryInfo(providerName, repositoryName)
        if not ri:
            raise ValueError("Could not find repository, '%s' for provider, '%s'" \
                             %(repositoryName, providerName))

        try:
            status, result = g.harvester.runHarvestAndIngest(providerName, ri,
                                                             ingestAsynch = True,
                                                             userEmail = pi.email)
        except Exception, e:
            c.errors = {'System error': e}
            return render('genshi', 'error')

        outMessage = "Repository data, %s, for provider with name, %s, successfully harvested. (Details: %s)" \
            %(repositoryName, providerName, result)
        if not status:
            outMessage = "A problem occurred whilst harvesting the data - please retry later."
            c.errors = {'Harvest/ingest error': result}
            return render('genshi', 'error')

        # now return to the home page - and display pop up
        session[const.POP_UP_MESSAGE_SESSION_KEY] = outMessage
        session.save()

        # now, return to the view page
        h.redirect_to('home')