from django.db import models
from django.dispatch import receiver

from openslides.utils.person.models import PersonMixin, PersonField
from openslides.utils.person.signals import receive_persons
from openslides.utils.person.api import Person


class TestPerson(PersonMixin, models.Model):
    person_prefix = 'test'
    name = models.CharField(max_length='255')

    def __unicode__(self):
        return self.name


class TestPersonToPerson(object):
    def __init__(self, person_prefix_filter=None, id_filter=None):
        self.person_prefix_filter = person_prefix_filter
        self.id_filter = id_filter

    def __iter__(self):
        if (not self.person_prefix_filter or
                self.person_prefix_filter == TestPerson.person_prefix):
            if self.id_filter:
                try:
                    yield TestPerson.objects.get(pk=self.id_filter)
                except TestPerson.DoesNotExist:
                    pass
            else:
                for user in TestPerson.objects.all():
                    yield user


@receiver(receive_persons, dispatch_uid="test_person")
def receive_persons(sender, **kwargs):
    return TestPersonToPerson(
        person_prefix_filter=kwargs['person_prefix_filter'],
        id_filter=kwargs['id_filter'])


class TestModel(models.Model):
    person = PersonField()

    def __unicode__(self):
        return self.person