"""Functions to work with bit flag."""

from update_market_vendors.constants import Vendor


def set_vendor(vendor: Vendor, data: int) -> int:
    """Set bit corresponding to particular vendor in data."""
    return data | vendor.value


def remove_vendor(vendor: Vendor, data: int) -> int:
    """Reset bit corresponding to particular vendor in data."""
    return data & (~vendor.value)


def check_vendor(vendor: Vendor, data: int) -> int:
    """Check if bit corresponding to particulate vendor is set in data."""
    return data & vendor