* refactor: replace datetime.now with timezone.now * refactor: migrate model fields to use timezone.now as default * refactor: replace datetime.today with timezone.now datetime.datetime.today() is equivalent to datetime.datetime.now(); both return a naive datetime with the current local time. * refactor: rephrase datetime.now(tz) as timezone.now().astimezone(tz) This is effectively the same, but is less likely to encourage accidental use of naive datetimes. * refactor: revert datetime.today() change to old migrations * refactor: change a missed datetime.now to timezone.now * chore: renumber timezone_now migration * chore: renumber migrations
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
# Copyright The IETF Trust 2012-2020, All Rights Reserved
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
import email.utils
|
|
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
|
|
import debug # pyflakes:ignore
|
|
|
|
from ietf.person.models import Person
|
|
from ietf.group.models import Group
|
|
from ietf.doc.models import Document
|
|
from ietf.name.models import RoleName
|
|
from ietf.utils.models import ForeignKey
|
|
from ietf.utils.mail import get_email_addresses_from_text
|
|
|
|
class Message(models.Model):
|
|
time = models.DateTimeField(default=timezone.now)
|
|
by = ForeignKey(Person)
|
|
|
|
subject = models.CharField(max_length=255)
|
|
frm = models.CharField(max_length=255)
|
|
to = models.CharField(max_length=1024)
|
|
cc = models.CharField(max_length=1024, blank=True)
|
|
bcc = models.CharField(max_length=255, blank=True)
|
|
reply_to = models.CharField(max_length=255, blank=True)
|
|
body = models.TextField()
|
|
content_type = models.CharField(default="text/plain", max_length=255, blank=True)
|
|
msgid = models.CharField(max_length=255, blank=True, null=True, default=email.utils.make_msgid)
|
|
|
|
related_groups = models.ManyToManyField(Group, blank=True)
|
|
related_docs = models.ManyToManyField(Document, blank=True)
|
|
|
|
sent = models.DateTimeField(blank=True, null=True)
|
|
|
|
class Meta:
|
|
ordering = ['time']
|
|
indexes = [
|
|
models.Index(fields=['time',]),
|
|
]
|
|
|
|
def __str__(self):
|
|
return "'%s' %s -> %s" % (self.subject, self.frm, self.to)
|
|
|
|
def get(self, field):
|
|
r = getattr(self, field)
|
|
return r if isinstance(r, list) else get_email_addresses_from_text(r)
|
|
|
|
|
|
class MessageAttachment(models.Model):
|
|
message = ForeignKey(Message)
|
|
filename = models.CharField(max_length=255, db_index=True, blank=True)
|
|
content_type = models.CharField(max_length=255, blank=True)
|
|
encoding = models.CharField(max_length=255, blank=True)
|
|
removed = models.BooleanField(default=False)
|
|
body = models.TextField()
|
|
|
|
def __str__(self):
|
|
return self.filename
|
|
|
|
|
|
class SendQueue(models.Model):
|
|
time = models.DateTimeField(default=timezone.now)
|
|
by = ForeignKey(Person)
|
|
|
|
message = ForeignKey(Message)
|
|
|
|
send_at = models.DateTimeField(blank=True, null=True)
|
|
sent_at = models.DateTimeField(blank=True, null=True)
|
|
|
|
note = models.TextField(blank=True)
|
|
|
|
class Meta:
|
|
ordering = ['time']
|
|
indexes = [
|
|
models.Index(fields=['time',]),
|
|
]
|
|
|
|
def __str__(self):
|
|
return "'%s' %s -> %s (sent at %s)" % (self.message.subject, self.message.frm, self.message.to, self.sent_at or "<not yet>")
|
|
|
|
|
|
class AnnouncementFrom(models.Model):
|
|
name = ForeignKey(RoleName)
|
|
group = ForeignKey(Group)
|
|
address = models.CharField(max_length=255)
|
|
|
|
def __str__(self):
|
|
return self.address
|
|
|
|
class Meta:
|
|
verbose_name_plural='Announcement From addresses'
|
|
|