from api.utils import get_object
from users.models import SchoolClass


def set_hidden_for(block, visibility_list):
    for v in visibility_list:
        school_class = get_object(SchoolClass, v.school_class_id)
        if v.hidden:
            block.hidden_for.add(school_class)
        else:
            block.hidden_for.remove(school_class)


def set_visible_for(block, visibility_list):
    for v in visibility_list:
        school_class = get_object(SchoolClass, v.school_class_id)
        if v.hidden:
            block.visible_for.remove(school_class)
        else:
            block.visible_for.add(school_class)