cocitedwith.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	# -- coding: utf-8 --
	#
	# This file is part of Invenio.
	# Copyright (C) 2014, 2015 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.

	"""Cocitedwith search unit."""

	from intbitset import intbitset


	def search_unit(query, f, m, wl=None):
	"""Search for records in citation index."""
	from invenio.legacy.search_engine import record_exists
	from invenio.legacy.bibrank.citation_searcher import \
	calculate_co_cited_with_list

	results = intbitset([])

	if query:
	if isinstance(query, intbitset):
	ahitset = query
	else:
	recid = int(query)
	ahitset = [recid] if record_exists(recid) == 1 else []

	if len(ahitset):
	for recid in ahitset:
	results \|= intbitset([
	x[0] for x in calculate_co_cited_with_list(recid)])

	return results