dss.api/spa/models/basemodel.py

import logging
import os
from datetime import datetime
from dirtyfields import DirtyFieldsMixin
from django.db import models
from django.utils import simplejson
from core.utils import url
from dss import localsettings, settings



class BaseModel(models.Model):
    logger = logging.getLogger(__name__)

    object_created = models.DateTimeField(auto_now_add=True, default=datetime.now())
    object_updated = models.DateTimeField(auto_now=True, default=datetime.now(), db_index=True)

    class Meta:
        abstract = True
        app_label = 'spa'

    def tosimplejson(self):
        return simplejson.dump(self)

    @classmethod
    def get_lookup(cls, filter_field, transform=None, filter=None):
        if filter is not None:
            filter_dict = {'%s__startswith' % filter_field: filter}
            return cls.objects.all().filter(filter_dict).extra(select=transform)
        else:
            return cls.objects.all()

    def get_image_url(self, image, default):
        try:
            if os.path.isfile(image.path):
                images_root = localsettings.IMAGE_URL if hasattr(localsettings,
                                                                 'IMAGE_URL') else "%s" % settings.MEDIA_URL
                ret = "%s/%s/%s" % (settings.STATIC_URL, images_root, image)
                return url.urlclean(ret)

        except Exception, ex:
            pass

        return default

    @classmethod
    def get_lookup_filter_field(cls):
        field_list = cls._meta.get_all_field_names()
        for field in field_list:
            if field.endswith("title") or field.endswith("name") or field.endswith("description"):
                return field
        return "description"

    def base_pre_save(self, sender, instance, created, **kwargs):
        pass

    def clean_image(self, image_field, sender):

        # TODO(fergal.moran@gmail.com): stop tastypie nulling image field if not sent

        if self.__dict__[image_field] == "DONOTSEND":
            try:
                old_instance = sender.objects.get(pk=self.pk)
                if old_instance is not None:
                    self.__dict__[image_field] = old_instance.__dict__[image_field]
            except Exception, ex:
                pass