from vladiate import Vlad
from vladiate.validators import UniqueValidator, SetValidator
from vladiate.inputs import LocalFile

class CustomFieldnamesValidator(Vlad):
    source = LocalFile('vampires.csv')
    fieldnames = ['Column A', 'Column B']
    validators = {
        'Column A': [
            UniqueValidator()
        ],
        'Column B': [
            SetValidator(['Vampire', 'Not A Vampire'])
        ]
    }