# -*- coding: utf-8 -*- ########################################################################### # Copyright (c), The AiiDA team. All rights reserved. # # This file is part of the AiiDA code. # # # # The code is hosted on GitHub at https://github.com/aiidateam/aiida-core # # For further information on the license, see the LICENSE.txt file # # For further information please visit http://www.aiida.net # ########################################################################### # pylint: disable=invalid-name """Invalidating node hash - User should rehash nodes for caching Revision ID: e797afa09270 Revises: 26d561acd560 Create Date: 2019-07-01 19:39:33.605457 """ from alembic import op # Remove when https://github.com/PyCQA/pylint/issues/1931 is fixed # pylint: disable=no-name-in-module,import-error from sqlalchemy.sql import text from aiida.cmdline.utils import echo # revision identifiers, used by Alembic. revision = 'e797afa09270' down_revision = '26d561acd560' branch_labels = None depends_on = None # Currently valid hash key _HASH_EXTRA_KEY = '_aiida_hash' [docs]def drop_hashes(conn): # pylint: disable=unused-argument """Drop hashes of nodes. Print warning only if the DB actually contains nodes. """ n_nodes = conn.execute(text("""SELECT count(*) FROM db_dbnode;""")).fetchall()[0][0] if n_nodes > 0: echo.echo_warning('Invalidating the hashes of all nodes. Please run "verdi rehash".', bold=True) statement = text(f"UPDATE db_dbnode SET extras = extras #- '{{{_HASH_EXTRA_KEY}}}'::text[];") conn.execute(statement) [docs]def upgrade(): """drop the hashes when upgrading""" drop_hashes(op.get_bind()) # pylint: disable=no-member [docs]def downgrade(): """drop the hashes also when downgrading""" drop_hashes(op.get_bind()) # pylint: disable=no-member