datatracker/ietf/message/models.py
Jennifer Richards ebebdbed3e
refactor: replace datetime.now and datetime.today with timezone.now (#4211)
* 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
2022-08-25 13:45:16 -03:00

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'