from django.db.models import OneToOneField

from django.db.models.fields.related import SingleRelatedObjectDescriptor

class AutoSingleRelatedObjectDescriptor(SingleRelatedObjectDescriptor):

    def __get__(self, instance, instance_type=None):

        try:

            return super(AutoSingleRelatedObjectDescriptor, self).__get__(instance, instance_type)

        except self.related.model.DoesNotExist:

            obj = self.related.model(**{self.related.field.name: instance})

            obj.save()

            return obj

class AutoOneToOneField(OneToOneField):

    '''

    OneToOneField creates related object on first call if it doesnt exists yet.

    '''

    def contribute_to_related_class(self, cls, related):

        setattr(cls, related.get_accessor_name(), AutoSingleRelatedObjectDescriptor(related))

    version = "0.3",

    long_description = """

        **Features:**

            - render_to decorator - reduce typing in django views.

            - signals decorator - allow use signals as decorators.

            - get_object_or_None function - similar to get_object_or_404, but returns None if object not found.

            - AutoOneToOne field - creates related object on first call if it doesnt exists yet.

        **Installation instruction:**

             - Copy annoying directory to your django project or put in on PYTHONPATH

             - Also you can run sudo python setup.py install or sudo easy_install django-annoying

        **Download:**

            - hg clone http://hg.assembla.com/django-annoying

    """,