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