def solr_print(args):
from utils.db import session_autocommit
from models import BearerToken
from sqlalchemy.orm.exc import NoResultFound
if args.bearertoken_id:
print(" * Printing the content of the Solr index for bearertoken_id: {}".format(
args.bearertoken_id))
with session_autocommit() as sex:
try:
bearertoken = sex.query(BearerToken).filter_by(id=args.bearertoken_id).one()
except NoResultFound:
print('There is no bearertoken_id {} in the database.'.format(args.bearertoken_id))
return
provider_name = bearertoken.provider.name
query = 'bearertoken_id:{}'.format(args.bearertoken_id)
if args.provider:
print(" * Printing the Solr index for provider: {}".format(args.provider))
provider_name = args.provider
query = '*:*'
solr = Solr(CORE_NAMES[provider_name])
r = solr.search(q=query)
for doc in r.documents:
del doc['content']
print(doc)
print("\n * {} documents.".format(r.total_results))
print("Done.")
def __init__(self, bearertoken):
self.bearertoken = bearertoken
with session_autocommit() as sex:
# Add bearertoken to the current session.
bearertoken = sex.merge(self.bearertoken)
self._bearertoken_id = bearertoken.id
self._access_token = bearertoken.access_token
def solr_reset(args):
from utils.db import session_autocommit
from models import BearerToken
from sqlalchemy.orm.exc import NoResultFound
if args.bearertoken_id:
print(" * Resetting the Solr index for bearertoken_id: {}".format(args.bearertoken_id))
with session_autocommit() as sex:
try:
bearertoken = sex.query(BearerToken).filter_by(id=args.bearertoken_id).one()
except NoResultFound:
print('There is no bearertoken_id {} in the database.'.format(args.bearertoken_id))
return
provider_name = bearertoken.provider.name
query = 'bearertoken_id:{}'.format(args.bearertoken_id)
if args.provider:
print(" * Resetting the Solr index for provider: {}".format(args.provider))
provider_name = args.provider
query = '*:*'
solr = Solr(CORE_NAMES[provider_name])
solr.delete_by_query(query)
solr.commit()
print("Done.")
def __init__(self, bearertoken_id, reset_cursor=False):
with session_autocommit() as sex:
self.bearertoken = sex.query(BearerToken).filter_by(id=bearertoken_id).one()
self.provider_name = self.bearertoken.provider.name
if reset_cursor:
self.bearertoken.updates_cursor = None
