models.py

import re
import datetime

from django.db import models
import jsonfield

import debug                            # pyflakes:ignore

from ietf.doc.models import Document
from ietf.person.models import Person
from ietf.group.models import Group
from ietf.message.models import Message
from ietf.name.models import DraftSubmissionStateName
from ietf.utils.accesstoken import generate_random_key, generate_access_token


def parse_email_line(line):
    """Split line on the form 'Some Name <email@example.com>'"""
    m = re.match("([^<]+) <([^>]+)>$", line)
    if m:
        return dict(name=m.group(1), email=m.group(2))
    else:
        return dict(name=line, email="")

class Submission(models.Model):
    state = models.ForeignKey(DraftSubmissionStateName)
    remote_ip = models.CharField(max_length=100, blank=True)

    access_key = models.CharField(max_length=255, default=generate_random_key)
    auth_key = models.CharField(max_length=255, blank=True)

    # draft metadata
    name = models.CharField(max_length=255, db_index=True)
    group = models.ForeignKey(Group, null=True, blank=True)
    title = models.CharField(max_length=255, blank=True)
    abstract = models.TextField(blank=True)
    rev = models.CharField(max_length=3, blank=True)
    pages = models.IntegerField(null=True, blank=True)
    authors = models.TextField(blank=True, help_text="List of author names and emails, one author per line, e.g. \"John Doe &lt;john@example.org&gt;\".")
    note = models.TextField(blank=True)
    replaces = models.CharField(max_length=1000, blank=True)

    first_two_pages = models.TextField(blank=True)
    file_types = models.CharField(max_length=50, blank=True)
    file_size = models.IntegerField(null=True, blank=True)
    document_date = models.DateField(null=True, blank=True)
    submission_date = models.DateField(default=datetime.date.today)

    submitter = models.CharField(max_length=255, blank=True, help_text="Name and email of submitter, e.g. \"John Doe &lt;john@example.org&gt;\".")

    draft = models.ForeignKey(Document, null=True, blank=True)

    def __unicode__(self):
        return u"%s-%s" % (self.name, self.rev)

    def authors_parsed(self):
        if not hasattr(self, '_cached_authors_parsed'):
            from ietf.submit.utils import ensure_person_email_info_exists
            res = []
            for line in self.authors.replace("\r", "").split("\n"):
                line = line.strip()
                if line:
                    parsed = parse_email_line(line)
                    if not parsed["email"]:
                        parsed["email"] = ensure_person_email_info_exists(**parsed).address
                    res.append(parsed)
            self._cached_authors_parsed = res
        return self._cached_authors_parsed

    def submitter_parsed(self):
        return parse_email_line(self.submitter)

    def access_token(self):
        return generate_access_token(self.access_key)

    def existing_document(self):
        return Document.objects.filter(name=self.name).first()

class SubmissionCheck(models.Model):
    time = models.DateTimeField(auto_now=True)
    submission = models.ForeignKey(Submission, related_name='checks')
    checker = models.CharField(max_length=256, blank=True)
    passed = models.NullBooleanField(default=False)
    message = models.TextField(null=True, blank=True)
    errors = models.IntegerField(null=True, blank=True, default=None)
    warnings = models.IntegerField(null=True, blank=True, default=None)
    items = jsonfield.JSONField(null=True, blank=True, default='{}')
    symbol = models.CharField(max_length=64, default='')
    #
    def __unicode__(self):
        return "%s submission check: %s: %s" % (self.checker, 'Passed' if self.passed else 'Failed', self.message[:48]+'...')
    def has_warnings(self):
        return self.warnings != '[]'
    def has_errors(self):
        return self.errors != '[]'

class SubmissionEvent(models.Model):
    submission = models.ForeignKey(Submission)
    time = models.DateTimeField(default=datetime.datetime.now)
    by = models.ForeignKey(Person, null=True, blank=True)
    desc = models.TextField()

    def __unicode__(self):
        return u"%s %s by %s at %s" % (self.submission.name, self.desc, self.by.plain_name() if self.by else "(unknown)", self.time)

    class Meta:
        ordering = ("-time", "-id")


class Preapproval(models.Model):
    """Pre-approved draft submission name."""
    name = models.CharField(max_length=255, db_index=True)
    by = models.ForeignKey(Person)
    time = models.DateTimeField(default=datetime.datetime.now)

    def __unicode__(self):
        return self.name

class SubmissionEmailEvent(SubmissionEvent):
    message     = models.ForeignKey(Message, null=True, blank=True,related_name='manualevents')
    msgtype     = models.CharField(max_length=25)
    in_reply_to = models.ForeignKey(Message, null=True, blank=True,related_name='irtomanual')

    def __unicode__(self):
        return u"%s %s by %s at %s" % (self.submission.name, self.desc, self.by.plain_name() if self.by else "(unknown)", self.time)

    class Meta:
        ordering = ['-time', '-id']