82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
|
from datetime import datetime
|
||
|
|
|
||
|
|
from django.db import models
|
||
|
|
from django.utils.translation import ugettext_lazy as _
|
||
|
|
|
||
|
|
from markitup.fields import MarkupField
|
||
|
|
|
||
|
|
from taggit.managers import TaggableManager
|
||
|
|
|
||
|
|
from mptt.models import MPTTModel, TreeForeignKey
|
||
|
|
from mptt.utils import drilldown_tree_for_node
|
||
|
|
|
||
|
|
|
||
|
|
class ContentBase(models.Model):
|
||
|
|
|
||
|
|
STATUS_CHOICES = (
|
||
|
|
(1, _("Draft")),
|
||
|
|
(2, _("Public")),
|
||
|
|
)
|
||
|
|
|
||
|
|
title = models.CharField(max_length=100)
|
||
|
|
slug = models.CharField(max_length=100, blank=True, null=True)
|
||
|
|
body = MarkupField()
|
||
|
|
|
||
|
|
tags = TaggableManager(blank=True)
|
||
|
|
|
||
|
|
status = models.IntegerField(choices=STATUS_CHOICES, default=2)
|
||
|
|
published = models.DateTimeField(default=datetime.now)
|
||
|
|
created = models.DateTimeField(editable=False, default=datetime.now)
|
||
|
|
updated = models.DateTimeField(editable=False, default=datetime.now)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
abstract = True
|
||
|
|
|
||
|
|
|
||
|
|
class Page(MPTTModel, ContentBase):
|
||
|
|
|
||
|
|
parent = TreeForeignKey("self", null=True, blank=True, related_name="children")
|
||
|
|
ordering = models.PositiveIntegerField(default=1)
|
||
|
|
path = models.TextField(blank=True, editable=False)
|
||
|
|
|
||
|
|
def __unicode__(self):
|
||
|
|
return self.title
|
||
|
|
|
||
|
|
def save(self, calculate_path=True, *args, **kwargs):
|
||
|
|
super(Page, self).save(*args, **kwargs)
|
||
|
|
if calculate_path:
|
||
|
|
self.calculate_path()
|
||
|
|
|
||
|
|
def calculate_path(self):
|
||
|
|
self.path = ""
|
||
|
|
for page in drilldown_tree_for_node(self):
|
||
|
|
if page == self:
|
||
|
|
self.path += page.slug
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
self.path += "%s/" % page.slug
|
||
|
|
self.save(calculate_path=False)
|
||
|
|
|
||
|
|
class MPTTMeta:
|
||
|
|
order_insertion_by = ["title"]
|
||
|
|
|
||
|
|
|
||
|
|
class MenuItem(MPTTModel):
|
||
|
|
|
||
|
|
name = models.CharField(max_length=50, unique=True)
|
||
|
|
slug = models.SlugField()
|
||
|
|
parent = TreeForeignKey("self", null=True, blank=True, related_name="children")
|
||
|
|
|
||
|
|
url = models.CharField(max_length=200)
|
||
|
|
#FIELDNAME = models.ForeignKey()
|
||
|
|
|
||
|
|
published = models.BooleanField(default=True)
|
||
|
|
login_required = models.BooleanField(default=False)
|
||
|
|
|
||
|
|
def __unicode__(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
|
||
|
|
class MPTTMeta:
|
||
|
|
order_insertion_by = ["name"]
|