Sinkler/django-radiogrid

# -*- coding: utf-8 -*-

from django.forms import ChoiceField, MultiValueField

from .widgets import RadioGridWidget


class RadioGridFormField(MultiValueField):
    def __init__(self, rows, values, *args, **kwargs):
        kwargs["widget"] = RadioGridWidget(rows, values)
        kwargs["fields"] = [ChoiceField(choices=values, required=False) for _ in rows]
        super(RadioGridFormField, self).__init__(*args, **kwargs)

    def compress(self, data_list):
        return "" if data_list is None else ",".join(data_list)