from import_export import resources, fields
from import_export.widgets import BooleanWidget
from .models import JioHotstarCode

class JioHotstarCodeResource(resources.ModelResource):
    code = fields.Field(attribute='code', column_name='code')
    type = fields.Field(attribute='type', column_name='type')

    class Meta:
        model = JioHotstarCode
        import_id_fields = ('code',)
        fields = ('code', 'type')

    def before_import_row(self, row, **kwargs):
        # Set default values for fields not present in the import
        row.setdefault('activated_flag', False)
        row.setdefault('activated_date', None)
        row.setdefault('activated_client_id', '')
        row.setdefault('client_phone_number', '')
        return row