def test_zip_is():
zf = open(os.path.join(RESOURCEPATH, 'speggs.zip'), 'rb')
inpl = factory(zf, zipcheck=True)
inp = next(inpl)
assert inp.iri is None
assert inp.stream.read() == b'python\n'
inp = next(inpl)
assert inp.iri is None
assert inp.stream.read() == b'monty\n'
def test_zip_is():
zf = open(os.path.join(RESOURCEPATH, 'speggs.zip'), 'rb')
inpl = factory(zf, zipcheck=True)
inp = next(inpl)
assert inp.iri is None
assert inp.stream.read() == b'python\n'
inp = next(inpl)
assert inp.iri is None
assert inp.stream.read() == b'monty\n'
def test_filelist_is():
files = [ open(os.path.join(RESOURCEPATH, f)) for f in ('spam.txt', 'eggs.txt') ]
inpl = factory(files)
inp = inpl[0]
assert inp.iri is None
assert inp.stream.read() == 'monty\n'
inp = inpl[1]
assert inp.iri is None
assert inp.stream.read() == 'python\n'
def test_filelist_is():
files = [
open(os.path.join(RESOURCEPATH, f)) for f in ('spam.txt', 'eggs.txt')
]
inpl = factory(files)
inp = inpl[0]
assert inp.iri is None
assert inp.stream.read() == 'monty\n'
inp = inpl[1]
assert inp.iri is None
assert inp.stream.read() == 'python\n'
def test_stringlist_is():
inpl = factory(['abc', 'def', 'ghi'])
inp = inpl[0]
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'abc'
inp = inpl[1]
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'def'
inp = inpl[2]
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'ghi'
def test_stringlist_is():
inpl = factory(['abc', 'def', 'ghi'])
inp = inpl[0]
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'abc'
inp = inpl[1]
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'def'
inp = inpl[2]
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'ghi'
def test_factory_file_is():
fname = os.path.join(RESOURCEPATH, 'spam.txt')
inpl = factory(open(fname))
for inp in inpl:
assert inp.iri is None
assert inp.stream.read() == 'monty\n'
def test_factory_string_is():
inpl = factory('abc')
for inp in inpl:
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'abc'
def bfconvert(inputs, handle_marc_source=handle_marcxml_source, entbase=None, model=None,
out=None, limit=None, rdfttl=None, rdfxml=None, xml=None, config=None,
verbose=False, logger=logging, loop=None, canonical=False,
lax=False, defaultsourcetype=inputsourcetype.unknown):
'''
inputs - One or more open file-like object, string with MARC content, or filename or IRI. If filename or
IRI it's a good idea to indicate this via the defaultsourcetype parameter
handle_marc_source - Function to turn a source of MARC data (e.g. XML or JSON) into the internal format for processing
entbase - Base IRI to be used for creating resources.
model - model instance for internal use
out - file where raw Versa JSON dump output should be written (default: write to stdout)
limit - Limit the number of records processed to this number. If omitted, all records will be processed.
rdfttl - stream to where RDF Turtle output should be written
rdfxml - stream to where RDF/XML output should be written
config - configuration information
verbose - If true show additional messages and information (default: False)
logger - logging object for messages
loop - optional asyncio event loop to use
canonical - output Versa's canonical form?
lax - If True signal to the handle_marc_source function that relaxed syntax rules should be applied
(e.g. accept XML with namespace problems)
defaultsourcetype - Signal indicating how best to interpret inputs to create an inputsource
'''
#if stats:
# register_service(statsgen.statshandler)
config = config or {}
if limit is not None:
try:
limit = int(limit)
except ValueError:
logger.debug('Limit must be a number, not "{0}". Ignoring.'.format(limit))
def resolve_class(fullname):
'''
Given a full name for a Python class, return the class object
'''
import importlib
modpath, name = fullname.rsplit('.', 1)
module = importlib.import_module(modpath)
cls = getattr(module, name)
return cls
attr_cls = resolve_class(config.get('versa-attr-cls', 'builtins.dict'))
attr_list_cls = resolve_class(config.get('versa-attr-list-cls', 'builtins.list'))
#attr_ordered_cls = resolve_class(config.get('versa-attr-cls', 'collections.OrderedDict'))
model_factory = functools.partial(memory.connection, attr_cls=attr_cls) #,logger=logger)
model_factory.attr_list_cls = attr_list_cls
#model_odict_factory = functools.partial(memory.connection, attr_cls=attr_ordered_cls) #,logger=logger)
#model_odict_factory.attr_list_cls = attr_list_cls
if 'marc_record_handler' in config:
handle_marc_source = AVAILABLE_MARC_HANDLERS[config['marc_record_handler']]
readmode = handle_marc_source.readmode
#inputs = ( inputsource(open(i, readmode)) for i in inputs )
#if not isinstance(inputs[0], inputsource):
# inputs = ( inputsource(i, streamopenmode=readmode) for i in inputs )
if handle_marc_source.makeinputsource:
inputs = factory(inputs, defaultsourcetype=defaultsourcetype, streamopenmode=readmode)
#inputs = ( inputsource(i, streamopenmode=readmode) for i in inputs )
ids = marc.idgen(entbase)
if model is None: model = model_factory()
g = rdflib.Graph()
#Intentionally not using either factory
if canonical: global_model = memory.connection() #logger=logger)
if xml is not None:
xmlw = writer.raw(xml, indent=' ')
xmlw.start_element('bibframe')
extant_resources = None
#extant_resources = set()
def postprocess():
#No need to bother with Versa -> RDF translation if we were not asked to generate Turtle
if any((rdfttl, rdfxml)): rdf.process(model, g, to_ignore=extant_resources, logger=logger)
if canonical: global_model.add_many([(o,r,t,a) for (rid,(o,r,t,a)) in model])
if xml is not None:
microxml.process(model, xmlw, to_ignore=extant_resources, logger=logger)
model.create_space()
#Set up event loop if not provided
if not loop:
loop = asyncio.get_event_loop()
#Allow configuration of a separate base URI for vocab items (classes & properties)
#XXX: Is this the best way to do this, or rather via a post-processing plug-in
vb = config.get('vocab-base-uri', BL)
transform_iris = config.get('transforms', {})
if transform_iris:
transforms = {}
for tiri in transform_iris:
try:
transforms.update(AVAILABLE_TRANSFORMS[tiri])
except KeyError:
raise Exception('Unknown transforms set {0}'.format(tiri))
else:
transforms = TRANSFORMS
marcextras_vocab = config.get('marcextras-vocab')
#Initialize auxiliary services (i.e. plugins)
plugins = []
for pc in config.get('plugins', []):
try:
pinfo = g_services[pc['id']]
plugins.append(pinfo)
pinfo[BF_INIT_TASK](pinfo, config=pc)
except KeyError:
raise Exception('Unknown plugin {0}'.format(pc['id']))
limiting = [0, limit]
#logger=logger,
#raise(Exception(repr(inputs)))
for source in inputs:
@asyncio.coroutine
#Wrap the parse operation to make it a task in the event loop
def wrap_task(): #source=source
sink = marc.record_handler( loop,
model,
entbase=entbase,
vocabbase=vb,
limiting=limiting,
plugins=plugins,
ids=ids,
postprocess=postprocess,
out=out,
logger=logger,
transforms=transforms,
extra_transforms=extra_transforms(marcextras_vocab),
canonical=canonical,
model_factory=model_factory)
args = dict(lax=lax)
handle_marc_source(source, sink, args, logger, model_factory)
sink.close()
yield
task = asyncio.async(wrap_task(), loop=loop)
try:
loop.run_until_complete(task)
except Exception as ex:
raise ex
finally:
loop.close()
if canonical:
out.write(repr(global_model))
if vb == BFZ:
g.bind('bf', BFNS)
g.bind('bfc', BFCNS)
g.bind('bfd', BFDNS)
else:
g.bind('vb', rdflib.Namespace(vb))
if entbase:
g.bind('ent', entbase)
if rdfttl is not None:
logger.debug('Converting to RDF (Turtle).')
rdfttl.write(g.serialize(format="turtle"))
if rdfxml is not None:
logger.debug('Converting to RDF (XML).')
rdfxml.write(g.serialize(format="pretty-xml"))
if xml is not None:
logger.debug('Converting to XML.')
xmlw.end_element('bibframe')
return
def test_factory_file_is():
fname = os.path.join(RESOURCEPATH, 'spam.txt')
inpl = factory(open(fname))
for inp in inpl:
assert inp.iri is None
assert inp.stream.read() == 'monty\n'
def test_factory_string_is():
inpl = factory('abc')
for inp in inpl:
assert inp.stream.__class__ == io.StringIO
assert inp.iri is None
assert inp.stream.read() == 'abc'
def bfconvert(inputs,
handle_marc_source=handle_marcxml_source,
entbase=None,
model=None,
out=None,
limit=None,
rdfttl=None,
rdfxml=None,
xml=None,
config=None,
verbose=False,
logger=logging,
loop=None,
canonical=False,
lax=False,
defaultsourcetype=inputsourcetype.unknown):
'''
inputs - One or more open file-like object, string with MARC content, or filename or IRI. If filename or
IRI it's a good idea to indicate this via the defaultsourcetype parameter
handle_marc_source - Function to turn a source of MARC data (e.g. XML or JSON) into the internal format for processing
entbase - Base IRI to be used for creating resources.
model - model instance for internal use
out - file where raw Versa JSON dump output should be written (default: write to stdout)
limit - Limit the number of records processed to this number. If omitted, all records will be processed.
rdfttl - stream to where RDF Turtle output should be written
rdfxml - stream to where RDF/XML output should be written
config - configuration information
verbose - If true show additional messages and information (default: False)
logger - logging object for messages
loop - optional asyncio event loop to use
canonical - output Versa's canonical form?
lax - If True signal to the handle_marc_source function that relaxed syntax rules should be applied
(e.g. accept XML with namespace problems)
defaultsourcetype - Signal indicating how best to interpret inputs to create an inputsource
'''
#if stats:
# register_service(statsgen.statshandler)
config = config or {}
if limit is not None:
try:
limit = int(limit)
except ValueError:
logger.debug(
'Limit must be a number, not "{0}". Ignoring.'.format(limit))
def resolve_class(fullname):
'''
Given a full name for a Python class, return the class object
'''
import importlib
modpath, name = fullname.rsplit('.', 1)
module = importlib.import_module(modpath)
cls = getattr(module, name)
return cls
attr_cls = resolve_class(config.get('versa-attr-cls', 'builtins.dict'))
model_factory = functools.partial(memory.connection,
attr_cls=attr_cls) #,logger=logger)
if 'marc_record_handler' in config:
handle_marc_source = AVAILABLE_MARC_HANDLERS[
config['marc_record_handler']]
readmode = handle_marc_source.readmode
#inputs = ( inputsource(open(i, readmode)) for i in inputs )
#if not isinstance(inputs[0], inputsource):
# inputs = ( inputsource(i, streamopenmode=readmode) for i in inputs )
if handle_marc_source.makeinputsource:
inputs = factory(inputs,
defaultsourcetype=defaultsourcetype,
streamopenmode=readmode)
#inputs = ( inputsource(i, streamopenmode=readmode) for i in inputs )
ids = marc.idgen(entbase)
if model is None: model = model_factory()
if any((rdfttl, rdfxml)):
import rdflib
BFNS = rdflib.Namespace(BFZ)
BFCNS = rdflib.Namespace(BFZ + 'cftag/')
BFDNS = rdflib.Namespace(BFZ + 'dftag/')
g = rdflib.Graph()
#Intentionally not using either factory
if canonical: global_model = memory.connection() #logger=logger)
if xml is not None:
xmlw = writer.raw(xml, indent=' ')
xmlw.start_element('bibframe')
extant_resources = None
#extant_resources = set()
def postprocess():
#No need to bother with Versa -> RDF translation if we were not asked to generate Turtle
if any((rdfttl, rdfxml)):
rdf.process(model, g, to_ignore=extant_resources, logger=logger)
if canonical:
global_model.add_many([(o, r, t, a)
for (rid, (o, r, t, a)) in model])
if xml is not None:
microxml.process(model,
xmlw,
to_ignore=extant_resources,
logger=logger)
model.create_space()
#Set up event loop if not provided
if not loop:
loop = asyncio.get_event_loop()
#Allow configuration of a separate base URI for vocab items (classes & properties)
#XXX: Is this the best way to do this, or rather via a post-processing plug-in
vb = config.get('vocab-base-uri', BL)
transform_iris = config.get('transforms', [])
marcspecials_vocab = config.get('marcspecials-vocab')
transforms = transform_set(transform_iris, marcspecials_vocab)
lookups = config.get('lookups', {})
#Initialize auxiliary services (i.e. plugins)
plugins = []
for pc in config.get('plugins', []):
try:
pinfo = g_services[pc['id']]
plugins.append(pinfo)
pinfo[BF_INIT_TASK](pinfo, config=pc)
except KeyError:
raise Exception('Unknown plugin {0}'.format(pc['id']))
limiting = [0, limit]
#logger=logger,
#raise(Exception(repr(inputs)))
for source in inputs:
@asyncio.coroutine
#Wrap the parse operation to make it a task in the event loop
def wrap_task(): #source=source
sink = marc.record_handler(loop,
model,
entbase=entbase,
vocabbase=vb,
limiting=limiting,
plugins=plugins,
ids=ids,
postprocess=postprocess,
out=out,
logger=logger,
transforms=transforms,
canonical=canonical,
lookups=lookups,
model_factory=model_factory)
args = dict(lax=lax)
handle_marc_source(source, sink, args, logger, model_factory)
sink.close()
yield
task = asyncio. async (wrap_task(), loop=loop)
try:
loop.run_until_complete(task)
except Exception as ex:
raise ex
finally:
loop.close()
if canonical:
out.write(repr(global_model))
if any((rdfttl, rdfxml)):
if vb == BFZ:
g.bind('bf', BFNS)
g.bind('bfc', BFCNS)
g.bind('bfd', BFDNS)
else:
g.bind('vb', rdflib.Namespace(vb))
if entbase:
g.bind('ent', entbase)
if rdfttl is not None:
logger.debug('Converting to RDF (Turtle).')
rdfttl.write(g.serialize(format="turtle"))
if rdfxml is not None:
logger.debug('Converting to RDF (XML).')
rdfxml.write(g.serialize(format="pretty-xml"))
if xml is not None:
logger.debug('Converting to XML.')
xmlw.end_element('bibframe')
return
