Изменение формата печати даты по умолчанию в Python - PullRequest
1 голос
/ 28 июня 2010

Могу ли я изменить функцию по умолчанию __str__() объекта datetime.datetime? По умолчанию он возвращает что-то вроде '2010-06-28 12:43:56.985790', и мне нужно напечатать что-то вроде '2010-06-28T12:44:21.241228' (что является функцией isoformat ()).

Мне это нужно для сериализации JSON модели django.

Моя модель:


class Transport(models.Model):
    user = models.ForeignKey(User)
    source = models.ForeignKey(Marker, related_name="source_marker")
    destination = models.ForeignKey(Marker, related_name="destination_marker")
    object = models.CharField(choices=possesion_choices, max_length=2**6)
    quantity = models.IntegerField()
    time_sent = models.DateTimeField()
    time_arrived = models.DateTimeField()

И когда я сериализую его (используя модуль wadofstuff ), он печатает что-то вроде


print serializers.serialize('json', Transport.objects.all(), relations=('source', 'destination',)  indent=4)
[
    {
        "pk": 1, 
        "model": "main.transport", 
        "fields": {
            [.. bla bla ..]
            "time_sent": "2010-06-28 12:18:05", 
            "time_arrived": "2010-06-28 12:38:36", 
            [.. bla bla ..]
        }
    }
]

1 Ответ

2 голосов
/ 28 июня 2010

Django использует метод поля value_to_string для предоставления строкового представления в сериализаторе. Таким образом, вы можете определить подкласс настраиваемого поля, который переопределяет этот метод:

class MyDateTimeField(DateTimeField)
    def value_to_string(self, obj):
        val = self._get_val_from_obj(obj)
        if val is None:
            data = ''
        else:
            data = val.isoformat()
        return data

Отредактированный

Ааа, похоже, я искал не в том месте. Метод serializers.python.Serializer.handle_field фактически проверяет поля даты и времени и передает их без изменений в кодировщик JSON. Так что это тот кодер, который нам на самом деле нужно переопределить.

class MyJSONEncoder(DjangoJSONEncoder):
    def default(self, o):
        if isinstance(o, datetime.datetime):
            return o.isoformat()
        else:
            return super(MyJSONEncoder, self).default(o)

К сожалению, wadofstuff жестко кодирует исходный кодер DjangoJSONEncoder, поэтому нам также потребуется переопределить сериализатор.

from wadofstuff.django.serializers.json import Serializer
class BetterSerializer(Serializer):
    """
    Convert a queryset to JSON.
    """
    def end_serialization(self):
        """Output a JSON encoded queryset."""
        self.options.pop('stream', None)
        self.options.pop('fields', None)
        self.options.pop('excludes', None)
        self.options.pop('relations', None)
        self.options.pop('extras', None)
        simplejson.dump(self.objects, self.stream, cls=MyJSONEncoder,
            **self.options)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...