Use a markdown editor

ram/portal/admin.py

@@ -1,5 +1,7 @@
+from django.db import models
 from django.contrib import admin
 from solo.admin import SingletonModelAdmin
+from martor.widgets import AdminMartorWidget
 
 from portal.models import SiteConfiguration, Flatpage
 
@@ -12,6 +14,10 @@ class FlatpageAdmin(admin.ModelAdmin):
     list_display = ("name", "path")
     search_fields = ("name",)
 
+    formfield_overrides = {
+        models.TextField: {'widget': AdminMartorWidget},
+    }
+
     fieldsets = (
         (
             None,

ram/portal/models.py

@@ -3,6 +3,7 @@ from django.db import models
 from django.urls import reverse
 from django.dispatch.dispatcher import receiver
 from solo.models import SingletonModel
+from martor.models import MartorField
 
 from ram import __version__ as app_version
 from ram.utils import slugify
@@ -49,7 +50,7 @@ class Flatpage(models.Model):
     name = models.CharField(max_length=256, unique=True)
     path = models.CharField(max_length=256, unique=True)
     draft = models.BooleanField(default=True)
-    content = models.TextField(blank=True)
+    content = MartorField()
     creation_time = models.DateTimeField(auto_now_add=True)
     updated_time = models.DateTimeField(auto_now=True)
 

ram/portal/urls.py

@@ -4,6 +4,7 @@ from portal.views import (
     GetHome,
     GetHomeFiltered,
     GetFlatpage,
+    MDUploader,
     GetRollingStock,
     GetConsist,
     Consists,
@@ -13,6 +14,7 @@ from portal.views import (
 
 urlpatterns = [
     path("", GetHome.as_view(), name="index"),
+    path("uploader/", MDUploader.as_view(), name="markdown_uploader"),
     path("<int:page>", GetHome.as_view(), name="index_pagination"),
     path(
         "page/<str:flatpage>",

ram/portal/utils.py

@@ -1,4 +1,72 @@
+import os
+import json
+import uuid
+
 from django.apps import apps
+from django.conf import settings
+from django.http import HttpResponse
+from django.utils.translation import gettext_lazy as _
+from django.contrib.auth.decorators import login_required
+from django.core.files.storage import default_storage
+from django.core.files.base import ContentFile
+
+from martor.utils import LazyEncoder
+
+
+@login_required
+def markdown_uploader(request):
+    """
+    Makdown image upload for locale storage
+    and represent as json to markdown editor.
+    """
+    if request.META.get("HTTP_X_REQUESTED_WITH") == "XMLHttpRequest":
+        if "markdown-image-upload" in request.FILES:
+            image = request.FILES["markdown-image-upload"]
+            image_types = [
+                "image/png",
+                "image/jpg",
+                "image/jpeg",
+                "image/pjpeg",
+                "image/gif",
+            ]
+            if image.content_type not in image_types:
+                data = json.dumps(
+                    {"status": 405, "error": _("Bad image format.")},
+                    cls=LazyEncoder,
+                )
+                return HttpResponse(
+                    data, content_type="application/json", status=405
+                )
+
+            if image.size > settings.MAX_IMAGE_UPLOAD_SIZE:
+                to_MB = settings.MAX_IMAGE_UPLOAD_SIZE / (1024 * 1024)
+                data = json.dumps(
+                    {
+                        "status": 405,
+                        "error": _("Maximum image file is %(size)s MB.")
+                        % {"size": to_MB},
+                    },
+                    cls=LazyEncoder,
+                )
+                return HttpResponse(
+                    data, content_type="application/json", status=405
+                )
+
+            img_uuid = "{0}-{1}".format(
+                uuid.uuid4().hex[:10], image.name.replace(" ", "-")
+            )
+            tmp_file = os.path.join(settings.MARTOR_UPLOAD_PATH, img_uuid)
+            def_path = default_storage.save(
+                tmp_file, ContentFile(image.read())
+            )
+            img_url = os.path.join(settings.MEDIA_URL, def_path)
+
+            data = json.dumps(
+                {"status": 200, "link": img_url, "name": image.name}
+            )
+            return HttpResponse(data, content_type="application/json")
+        return HttpResponse(_("Invalid request!"))
+    return HttpResponse(_("Invalid request!"))
 
 
 def get_site_conf():

ram/portal/views.py

@@ -8,7 +8,7 @@ from django.shortcuts import render
 from django.core.exceptions import ObjectDoesNotExist
 from django.core.paginator import Paginator, PageNotAnInteger
 
-from portal.utils import get_site_conf
+from portal.utils import get_site_conf, markdown_uploader
 from portal.models import Flatpage
 from roster.models import RollingStock
 from consist.models import Consist
@@ -248,7 +248,9 @@ class Scales(View):
 class GetFlatpage(View):
     def get(self, request, flatpage):
         try:
-            flatpage = Flatpage.objects.get(path=flatpage)
+            flatpage = Flatpage.objects.get(
+                Q(Q(path=flatpage) & Q(draft=False))
+            )
         except ObjectDoesNotExist:
             raise Http404
 
@@ -257,3 +259,8 @@ class GetFlatpage(View):
             "flatpage.html",
             {"flatpage": flatpage},
         )
+
+
+class MDUploader(View):
+    def post(self, request):
+        return markdown_uploader(request)