def setUp(self): try: os.makedirs(self.temp_dir) except: pass delete_index() init_search() db.drop_all() db.create_all() create_system_roles()
def index(foreign_id=None): """Index documents in the given source (or throughout).""" q = Document.all_ids() if foreign_id: source = Source.by_foreign_id(foreign_id) if source is None: raise ValueError("No such source: %r" % foreign_id) q = q.filter(Document.source_id == source.id) else: delete_index() init_search() for doc_id, in q: index_document.delay(doc_id)
def evilshit(): """EVIL: Delete all data and recreate the database.""" delete_index() db.drop_all() from sqlalchemy import MetaData, inspect from sqlalchemy.dialects.postgresql import ENUM metadata = MetaData() metadata.bind = db.engine metadata.reflect() for table in metadata.sorted_tables: table.drop(checkfirst=True) for enum in inspect(db.engine).get_enums(): enum = ENUM(name=enum['name']) enum.drop(bind=db.engine, checkfirst=True) init()
def setUp(self): if not hasattr(TestCase, '_global_test_state'): TestCase._global_test_state = True delete_index() upgrade_search() else: indexes = [collection_index(), entity_index(), record_index()] es.delete_by_query(index=indexes, body={'query': { 'match_all': {} }}, refresh=True, conflicts='proceed') destroy_db() db.create_all() create_system_roles()
def evilshit(): """EVIL: Delete all data and recreate the database.""" delete_index() from sqlalchemy import MetaData, inspect from sqlalchemy.exc import InternalError from sqlalchemy.dialects.postgresql import ENUM metadata = MetaData() metadata.bind = db.engine metadata.reflect() tables = list(metadata.sorted_tables) while len(tables): for table in tables: try: table.drop(checkfirst=True) tables.remove(table) except InternalError: pass for enum in inspect(db.engine).get_enums(): enum = ENUM(name=enum['name']) enum.drop(bind=db.engine, checkfirst=True) upgrade()
def resetindex(): """Re-create the ES index configuration, dropping all data.""" delete_index() upgrade_search()
def evilshit(): """Delete all data and recreate the database.""" db.drop_all() upgrade_db() delete_index() init_search()
def resetindex(): """Re-create the ES index configuration, dropping all data.""" delete_index() init_search()
def setUp(self): delete_index() upgrade_search() db.drop_all() db.create_all() create_system_roles()
def reset(): db.drop_all() delete_index() upgrade()