import django from django.db import models from django.urls import reverse from django.dispatch.dispatcher import receiver from django.utils.safestring import mark_safe from solo.models import SingletonModel from ckeditor.fields import RichTextField from ckeditor_uploader.fields import RichTextUploadingField from ram import __version__ as app_version from ram.utils import slugify class SiteConfiguration(SingletonModel): site_name = models.CharField( max_length=256, default="Railroad Assets Manager" ) site_author = models.CharField(max_length=256, blank=True) about = RichTextField(blank=True) items_per_page = models.CharField( max_length=2, choices=[(str(x * 3), str(x * 3)) for x in range(2, 11)], default="6", ) items_ordering = models.CharField( max_length=10, choices=[ ("type", "By rolling stock type"), ("company", "By company name"), ("identifier", "By rolling stock class"), ], default="type", ) footer = RichTextField(blank=True) footer_extended = RichTextField(blank=True) show_version = models.BooleanField(default=True) use_cdn = models.BooleanField(default=True) extra_head = models.TextField(blank=True) class Meta: verbose_name = "Site Configuration" def __str__(self): return "Site Configuration" def version(self): return app_version def django_version(self): return django.get_version() class Flatpage(models.Model): name = models.CharField(max_length=256, unique=True) path = models.CharField(max_length=256, unique=True) published = models.BooleanField(default=False) content = RichTextUploadingField() creation_time = models.DateTimeField(auto_now_add=True) updated_time = models.DateTimeField(auto_now=True) def __str__(self): return self.name def get_absolute_url(self): return reverse("flatpage", kwargs={"flatpage": self.path}) def get_link(self): if self.published: return mark_safe( 'Link'.format( self.get_absolute_url() ) ) @receiver(models.signals.pre_save, sender=Flatpage) def tag_pre_save(sender, instance, **kwargs): instance.path = slugify(instance.name)