"""Model for target_group table."""
from product_review.api import db


class TargetGroup(db.Model):
    """Target group model."""

    __tablename__ = "target_group"
    target_group_id = db.Column("id", db.Integer, primary_key=True, nullable=False)
    display_name = db.Column(db.String(80), nullable=False)

    def to_dict(self):
        """Convert the TargetGroup object to a dictionary representation."""
        items = {}
        for c in self.__table__.columns:
            column_name = "target_group_id" if c.name == "id" else c.name
            items[column_name] = getattr(self, column_name)
        return items


def get_items():
    """Get all target groups."""
    return [target_group.to_dict() for target_group in TargetGroup.query.all()]