from helpers.cluster import ClickHouseCluster


def get_table_path(node, table, database):
    return node.query(
        sql=f"SELECT data_paths FROM system.tables WHERE table = '{table}' and database = '{database}'"
    ).strip("'[]\n")


def check_flags_deleted(node, database_name, tables):
    for table in tables:
        assert "convert_to_replicated" not in node.exec_in_container(
            [
                "bash",
                "-c",
                f"ls {get_table_path(node, table, database_name)}",
            ]
        )


def set_convert_flags(node, database_name, tables):
    for table in tables:
        node.exec_in_container(
            [
                "bash",
                "-c",
                f"touch {get_table_path(node, table, database_name)}convert_to_replicated",
            ]
        )