pbp_com/apps/documents/models.py

from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes.generic import GenericForeignKey
from django.db import models, IntegrityError, transaction
from django.template.defaultfilters import slugify
from django.utils.translation import ugettext_lazy as _, ugettext

from directory.models import Place
from taggit.managers import TaggableManager
from django_extensions.db.models import TimeStampedModel
from documents.managers import PublishedManager


class Document(TimeStampedModel):
    ACCESS_CHOICES = (
        ("public", "Public"),
        ("private", "Private"),
        ("organization", "Organization"),
    )
    title = models.CharField(verbose_name=_("Title"), max_length=255)
    slug = models.SlugField(verbose_name=_("Slug"), unique=True, max_length=200)
    description = models.TextField(_("Description"), blank=True, null=True)
    source = models.CharField(_("Source"), max_length=100)
    access = models.CharField(
        _("Access"), max_length=12, choices=ACCESS_CHOICES, default="public"
    )
    published = models.BooleanField(_("published"), default=False)
    published_on = models.DateTimeField(_("published on"))
    preview = models.ImageField(
        _("preview"), upload_to="documents/previews", blank=True, null=True
    )
    doc_cloud_id = models.SlugField(
        max_length=80, editable=False, null=True, blank=True
    )

    tags = TaggableManager()
    objects = models.Manager()
    published_objects = PublishedManager()

    def __unicode__(self):
        return self.title

    class Meta:
        verbose_name = _("Document")
        verbose_name_plural = _("Documents")

    @models.permalink
    def get_absolute_url(self):
        return (
            "dc-document-detail",
            (),
            {
                "year": self.published_on.year,
                "month": self.published_on.strftime("%b").lower(),
                "day": self.published_on.day,
                "slug": self.slug,
            },
        )

    @property
    def display(self):
        return True


class TextDocument(Document):
    content = models.TextField(
        _("content"),
        blank=True,
        null=True,
        help_text="If document is plaintext, insert it here.",
    )

    class Meta:
        verbose_name = _("Text Document")
        verbose_name_plural = _("Text Documents")

    @models.permalink
    def get_absolute_url(self):
        return (
            "dc-text-document-detail",
            (),
            {
                "year": self.published_on.year,
                "month": self.published_on.strftime("%b").lower(),
                "day": self.published_on.day,
                "slug": self.slug,
            },
        )


class PDFDocument(Document):
    file = models.FileField(_("file"), upload_to="documents/pdf/%Y/%b/%d")

    class Meta:
        verbose_name = _("PDF Document")
        verbose_name_plural = _("PDF Documents")

    @models.permalink
    def get_absolute_url(self):
        return (
            "dc-pdf-document-detail",
            (),
            {
                "year": self.published_on.year,
                "month": self.published_on.strftime("%b").lower(),
                "day": self.published_on.day,
                "slug": self.slug,
            },
        )


class ExcelDocument(Document):
    file = models.FileField(_("file"), upload_to="documents/excel/%Y/%b/%d")

    class Meta:
        verbose_name = _("Excel Document")
        verbose_name_plural = _("Excel Documents")

    @models.permalink
    def get_absolute_url(self):
        return (
            "dc-xcl-document-detail",
            (),
            {
                "year": self.published_on.year,
                "month": self.published_on.strftime("%b").lower(),
                "day": self.published_on.day,
                "slug": self.slug,
            },
        )


class WordDocument(Document):
    file = models.FileField(_("file"), upload_to="documents/word/%Y/%b/%d")

    class Meta:
        verbose_name = _("Word Document")
        verbose_name_plural = _("Word Documents")

    @models.permalink
    def get_absolute_url(self):
        return (
            "dc-word-document-detail",
            (),
            {
                "year": self.published_on.year,
                "month": self.published_on.strftime("%b").lower(),
                "day": self.published_on.day,
                "slug": self.slug,
            },
        )