Skip to content

Commit

Permalink
Fixed timezone support for django 1.4
Browse files Browse the repository at this point in the history
  • Loading branch information
Psyton committed Apr 4, 2012
1 parent 1bd21d4 commit 476ad03
Showing 1 changed file with 14 additions and 6 deletions.
20 changes: 14 additions & 6 deletions django_mailer/models.py
@@ -1,6 +1,14 @@
from django.db import models
from django_mailer import constants, managers
import datetime


def default_now():
from django import VERSION
if (VERSION[0] >= 1 and VERSION[1] >= 4):
from django.utils import timezone
return timezone.now()
import datetime
return datetime.datetime.now()


PRIORITIES = (
Expand Down Expand Up @@ -31,7 +39,7 @@ class Message(models.Model):
subject = models.CharField(max_length=255)

encoded_message = models.TextField()
date_created = models.DateTimeField(default=datetime.datetime.now)
date_created = models.DateTimeField(default=default_now)

class Meta:
ordering = ('date_created',)
Expand All @@ -53,15 +61,15 @@ class QueuedMessage(models.Model):
default=constants.PRIORITY_NORMAL)
deferred = models.DateTimeField(null=True, blank=True)
retries = models.PositiveIntegerField(default=0)
date_queued = models.DateTimeField(default=datetime.datetime.now)
date_queued = models.DateTimeField(default=default_now)

objects = managers.QueueManager()

class Meta:
ordering = ('priority', 'date_queued')

def defer(self):
self.deferred = datetime.datetime.now()
self.deferred = default_now()
self.save()


Expand All @@ -74,7 +82,7 @@ class Blacklist(models.Model):
"""
email = models.EmailField(max_length=200)
date_added = models.DateTimeField(default=datetime.datetime.now)
date_added = models.DateTimeField(default=default_now)

class Meta:
ordering = ('-date_added',)
Expand All @@ -89,7 +97,7 @@ class Log(models.Model):
"""
message = models.ForeignKey(Message, editable=False)
result = models.PositiveSmallIntegerField(choices=RESULT_CODES)
date = models.DateTimeField(default=datetime.datetime.now)
date = models.DateTimeField(default=default_now)
log_message = models.TextField()

class Meta:
Expand Down

0 comments on commit 476ad03

Please sign in to comment.