byceps/byceps

"""
byceps.services.consent.brand_requirements_service
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

:Copyright: 2014-2024 Jochen Kupperschmidt
:License: Revised BSD (see `LICENSE` file for details)
"""

from sqlalchemy import delete

from byceps.database import db
from byceps.services.brand.models import BrandID

from .dbmodels import DbConsentBrandRequirement
from .models import ConsentSubjectID


def create_brand_requirement(
    brand_id: BrandID, subject_id: ConsentSubjectID
) -> None:
    """Create a brand requirement."""
    db_brand_requirement = DbConsentBrandRequirement(brand_id, subject_id)

    db.session.add(db_brand_requirement)
    db.session.commit()


def delete_brand_requirement(
    brand_id: BrandID, subject_id: ConsentSubjectID
) -> None:
    """Delete a brand requirement."""
    db.session.execute(
        delete(DbConsentBrandRequirement)
        .filter_by(brand_id=brand_id)
        .filter_by(subject_id=subject_id)
    )
    db.session.commit()