Page MenuHomec4science
Files F121691858

utils.py
No OneTemporary
Actions

File Metadata

Created
Sun, Jul 13, 05:10

utils.py
View Options

# This file is part of Invenio.
# Copyright (C) 2013, 2014 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.
"""Helper module to bibupload a record."""
import os
import uuid
import six
from tempfile import mkstemp
from invenio.legacy.bibrecord import record_xml_output
from invenio.legacy.bibsched.bibtask import task_low_level_submission
from invenio.config import CFG_TMPSHAREDDIR
CFG_MAX_RECORDS = 500
def open_temp_file(prefix):
"""Create a temporary file to write MARC XML in."""
# Prepare to save results in a tmp file
(fd, filename) = mkstemp(
dir=CFG_TMPSHAREDDIR,
prefix=prefix + str(uuid.uuid4()),
suffix='.xml'
)
file_out = os.fdopen(fd, "w")
return (file_out, filename)
def close_temp_file(file_out, filename):
"""Close temporary file again."""
file_out.close()
os.chmod(filename, 0o644)
def create_marcxml(record):
"""Create MARCXML based on type of input variable."""
from invenio.modules.records.api import Record
if isinstance(record, six.string_types):
return record
elif isinstance(record, Record):
return record.legacy_export_as_marc()
else:
return record_xml_output(record)
def bibupload_record(record=None, collection=None,
file_prefix="bibuploadutils", mode="-c",
alias='bibuploadutils', opts=[]):
"""Write a MARCXML file and bibupload it."""
if collection is None and record is None:
return
(file_out, filename) = open_temp_file(file_prefix)
if collection is not None:
file_out.write("<collection>")
tot = 0
for rec in collection:
file_out.write(create_marcxml(record))
tot += 1
if tot == CFG_MAX_RECORDS:
file_out.write("</collection>")
close_temp_file(file_out, filename)
task_low_level_submission(
'bibupload', alias, mode, filename, *opts
)
(file_out, filename) = open_temp_file(file_prefix)
file_out.write("<collection>")
tot = 0
file_out.write("</collection>")
elif record is not None:
tot = 1
file_out.write(create_marcxml(record))
close_temp_file(file_out, filename)
if tot > 0:
task_low_level_submission('bibupload', alias, mode, filename, *opts)

Event Timeline

c4science · Help