Homec4science
Diffusion invenio-infoscience (maint-2.1) Browse

R3600/invenio/modules/workflows/tasks/marcxml_tasks.py
1201a27cedffmaint-2.1

Last Change
Raw File

marcxml_tasks.py
View Options

## This file is part of Invenio.
## Copyright (C) 2012 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
def approve_record(obj, eng):
"""
Will add the approval widget to the record
"""
obj.extra_data["last_task_name"] = 'Record Approval'
try:
eng.log.info("Adding the approval widget to %s" % (obj.id))
obj.extra_data['widget'] = 'approval_widget'
eng.halt("Record needs approval")
except KeyError:
# Log the error
obj.extra_data["error_msg"] = 'Could not assign widget'
eng.log.error("Error: %s" % (obj.extra_data["error_msg"],))
approve_record.__title__ = "Record Approval"
approve_record.__description__ = "This task assigns the approval widget to a record."
def convert_record(stylesheet="oaiarxiv2marcxml.xsl"):
def _convert_record(obj, eng):
"""
Will convert the object data, if XML, using the given stylesheet
"""
from invenio.bibconvert_xslt_engine import convert
obj.extra_data["last_task_name"] = 'Convert Record'
eng.log.info("Starting conversion using %s stylesheet" %
(stylesheet,))
eng.log.info("Type of data: %s" % (obj.data_type,))
try:
obj.data = convert(obj.data, stylesheet)
except:
obj.extra_data["error_msg"] = 'Could not convert record'
eng.log.error("Error: %s" % (obj.extra_data["error_msg"],))
raise
_convert_record.__title__ = "Convert Record"
_convert_record.__description__ = "This task converts a XML record."
return _convert_record
def download_fulltext(obj, eng):
"""
Will download the fulltext document
"""
from invenio.bibdocfile import download_url
obj.extra_data["last_task_name"] = 'Download Fulltext'
try:
eng.log.info("Starting download of %s" % (obj.data['url']))
url = download_url(obj.data['url'])
obj.extra_data['tasks_results']['fulltext_url'] = url
except KeyError:
# Log the error
obj.extra_data["error_msg"] = 'Record does not include url'
eng.log.error("Error: %s" % (obj.extra_data["error_msg"],))
download_fulltext.__title__ = "Fulltext Download"
download_fulltext.__description__ = "This task downloads fulltext."
def match_record(obj, eng):
"""
Will try to find matches in stored records
"""
from invenio.legacy.bibrecord import create_record
from invenio.bibmatch_engine import match_records
obj.extra_data["last_task_name"] = 'Bibmatch Record'
rec = create_record(obj.data)
matches = match_records(records=[rec],
qrystrs=[("title", "[245__a]")])
obj.extra_data['tasks_results']['match_record'] = matches
if matches[2] or matches[3]:
# we have ambiguous or fuzzy results
# render holding pen corresponding template
eng.halt("Match resolution needed")
elif matches[0]:
eng.log.info("Matching: new record")
else:
results = matches[1][0][1]
eng.log.info("Matching: existing record %s" % (results,))
obj.extra_data['widget'] = 'bibmatch_widget'
match_record.__title__ = "Bibmatch Record"
match_record.__description__ = "This task matches a XML record."
def print_record(obj, eng):
eng.log.info(obj.get_data())
print_record.__title__ = "Print Record"
print_record.__description__ = "Prints the record data to engine log"
def upload_record(mode="ir"):
def _upload_record(obj, eng):
from invenio.bibtask import task_low_level_submission
obj.extra_data["last_task_name"] = 'Upload Record'
eng.log.info("Saving data to temporary file for upload")
filename = obj.save_to_file()
params = ["-%s" % (mode,), filename]
task_id = task_low_level_submission("bibupload", "bibworkflow",
*tuple(params))
eng.log.info("Submitted task #%s" % (task_id,))
_upload_record.__title__ = "Upload Record"
_upload_record.__description__ = "Uploads the record using BibUpload"
return _upload_record
c4science · Help