• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django_audit_trail-master
  • CHANGES.txt
  • .landscape.yaml
  • bump_version.py
  • test_project
    • models.py
    • views.py
    • __init__.py
    • tests
      • test_audit_trail.py
      • test_stringifier.py
      • __init__.py
    • settings.py
    • admin.py
  • setup.py
  • LICENSE.md
  • runtests.py
  • .travis.yml
  • README.md
  • pylintrc
  • .gitignore
  • MANIFEST.in
  • audit_trail
    • south_migrations
      • 0002_auto__chg_field_audittrail_user_ip.py
      • 0001_initial.py
      • __init__.py
    • app.py
    • templates
      • admin
        • audit_trail
          • audittrail
            • change_form.html
      • audit_trail
        • change_row.html
        • changes.html
    • models.py
    • views.py
    • locale
      • ru
        • LC_MESSAGES
          • django.mo
          • django.po
    • forms.py
    • watcher.py
    • __init__.py
    • utils.py
    • migrations
      • 0006_auto_20150124_0822.py
      • 0005_auto_20150114_1622.py
      • 0004_auto_20150112_1321.py
      • 0001_initial.py
      • 0002_auto_20150112_1228.py
      • __init__.py
      • 0007_auto_20150422_0548.py
      • 0003_auto_20150112_1236.py
      • 0008_auto_20150824_1934.py
    • stringifier.py
    • admin.py
    • signals.py
  • tox.ini

from django.contrib.postgres.fields import ArrayField
from django.core.exceptions import ObjectDoesNotExist
from django.db import models
from django.db.models import IntegerField
from django.utils import formats
from django.utils.encoding import force_text
from django.utils import six


class ModelFieldStringifier(object):

    custom_stringify_methods = {
        models.DateTimeField: 'stringify_datetime',
        models.DateField: 'stringify_date',
        models.ForeignKey: 'stringify_fk'
    }

    @classmethod
    def stringify(cls, field, value=None):
        value = value
        field_class = field.__class__

        if field_class in cls.custom_stringify_methods:
            stringifier = cls.custom_stringify_methods[field_class]

            if isinstance(stringifier, six.string_types):
                stringifier = getattr(cls, stringifier)

            return stringifier(value, field)

        if value is None:
            return None

        if isinstance(field, IntegerField):
            value = int(value)

        if isinstance(field, ArrayField):
            return force_text(', '.join(value))

        if getattr(field, 'choices', None):
            try:
                choices_dict = dict(field.choices)
                value = choices_dict[value]
                return force_text(value)
            except (KeyError, TypeError):
                return force_text(value)

        return force_text(value)

    @staticmethod
    def stringify_datetime(value, *args):
        if value is None:
            return None

        # Like models.DateField(default=date.today)
        if hasattr(value, '__call__'):
            value = value()

        return formats.date_format(value, "DATETIME_FORMAT")

    @staticmethod
    def stringify_date(value, *args):
        if value is None:
            return None

        # Like models.DateField(default=date.today)
        if hasattr(value, '__call__'):
            value = value()

        return formats.date_format(value, "DATE_FORMAT")

    @staticmethod
    def stringify_fk(value, field):
        if value is None:
            return None

        # iterate over to fields trying to look up the fk
        if not isinstance(value, field.related_model):
            success = False
            to_fields = field.to_fields or []
            if 'pk' not in to_fields:
                to_fields.append('pk')

            for to_field in to_fields:
                if not to_field:
                    continue

                try:
                    value = field.related_model.objects.get(**{to_field: value})
                    success = True
                    break
                except ObjectDoesNotExist:
                    pass
            if not success:
                return None

        return force_text(value)

    @classmethod
    def add_stringifier(cls, field_class, callback):
        cls.custom_stringify_methods[field_class] = callback