Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F80862231
utils.py
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Mon, Sep 2, 23:26
Size
2 KB
Mime Type
text/x-python
Expires
Wed, Sep 4, 23:26 (1 d, 22 h)
Engine
blob
Format
Raw Data
Handle
20463254
Attached To
R3600 invenio-infoscience
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
,
0
o644
)
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
Log In to Comment